keepalive.c

Go to the documentation of this file.

 
#include "keepalive.h"
#include "main.h"
#include "server.h"
#include "crypto.h"
#include "util/fps.h"
 
#include "common.h"
#include "platform/abstraction.h"
#include "thread_pool.h"
 
#include <stdatomic.h>
 
/* ============================================================================
 * Keepalive Thread Management
 * ============================================================================ */
 
__attribute__((unused)) static asciichat_thread_t g_ping_thread;
 
static bool g_ping_thread_created = false;
 
static atomic_bool g_ping_thread_exited = false;
 
/* ============================================================================
 * Keepalive Configuration
 * ============================================================================ */
 
#define PING_INTERVAL_SECONDS 3
 
#define PING_SLEEP_INTERVAL_SECONDS 1
 
/* ============================================================================
 * Ping Thread Implementation
 * ============================================================================ */
 
static void *ping_thread_func(void *arg) {
  (void)arg;
 
#ifdef DEBUG_THREADS
  log_debug("Ping thread started");
#endif
 
  // FPS tracking for keepalive thread (ping sent every 3 seconds = ~0.33 Hz)
  static fps_t fps_tracker = {0};
  static bool fps_tracker_initialized = false;
  if (!fps_tracker_initialized) {
    fps_init_with_interval(&fps_tracker, 1, "KEEPALIVE", 10000000ULL); // 1 "frame" per 3 seconds, report every 10s
    fps_tracker_initialized = true;
  }
 
  // Startup grace period: Allow connection initialization to complete before checking state
  // Prevents race condition where thread spawns before transport is fully configured
  platform_sleep_ms(100);
 
  while (!should_exit() && !server_connection_is_lost()) {
    // Check if connection is still active before sending
    if (!server_connection_is_active()) {
      log_debug("Connection inactive, exiting ping thread");
      break;
    }
 
    // Check if session rekeying should be triggered
    if (crypto_client_should_rekey()) {
      log_debug("Rekey threshold reached, initiating session rekey");
      if (crypto_client_initiate_rekey() < 0) {
        log_error("Failed to initiate rekey");
        // Don't break - continue with keepalive, rekey will be retried
      }
    }
 
    // Send ping packet every PING_INTERVAL_SECONDS to keep connection alive
    // Server timeout is 5 seconds, so 3-second pings provide safety margin
    if (threaded_send_ping_packet() < 0) {
      log_debug("Failed to send ping packet");
      // Set connection lost flag so main loop knows to reconnect
      server_connection_lost();
      break;
    }
 
    // Track ping for FPS reporting
    struct timespec current_time;
    (void)clock_gettime(CLOCK_MONOTONIC, &current_time);
    fps_frame(&fps_tracker, &current_time, "ping sent");
 
    // Sleep with early wake capability for responsive shutdown
    // Break sleep into 1-second intervals to check shutdown flags
    for (int i = 0;
         i < PING_INTERVAL_SECONDS && !should_exit() && !server_connection_is_lost() && server_connection_is_active();
         i++) {
      platform_sleep_usec(PING_SLEEP_INTERVAL_SECONDS * 1000 * 1000); // 1 second
    }
  }
 
#ifdef DEBUG_THREADS
  log_debug("Ping thread stopped");
#endif
 
  atomic_store(&g_ping_thread_exited, true);
 
  // Clean up thread-local error context before exit
  asciichat_errno_cleanup();
 
  return NULL;
}
 
/* ============================================================================
 * Public Interface Functions
 * ============================================================================ */
 
int keepalive_start_thread() {
  if (g_ping_thread_created) {
    log_warn("Ping thread already created");
    return 0;
  }
 
  // Start ping thread for keepalive
  atomic_store(&g_ping_thread_exited, false);
  if (thread_pool_spawn(g_client_worker_pool, ping_thread_func, NULL, 3, "keepalive_ping") != ASCIICHAT_OK) {
    log_error("Failed to spawn ping thread in worker pool");
    LOG_ERRNO_IF_SET("Ping thread creation failed");
    return -1;
  }
 
  g_ping_thread_created = true;
  return 0;
}
 
void keepalive_stop_thread() {
  if (!g_ping_thread_created) {
    return;
  }
 
  // Don't call signal_exit() here - that's for global shutdown only!
  // The ping thread monitors connection state and will exit when connection is lost
 
  // Wait for thread to exit gracefully
  int wait_count = 0;
  while (wait_count < 20 && !atomic_load(&g_ping_thread_exited)) {
    platform_sleep_usec(100000); // 100ms
    wait_count++;
  }
 
  if (!atomic_load(&g_ping_thread_exited)) {
    log_warn("Ping thread not responding - will be joined by thread pool");
  }
 
  // Thread will be joined by thread_pool_stop_all() in protocol_stop_connection()
  g_ping_thread_created = false;
 
  log_info("Ping thread stopped and joined");
}
 
bool keepalive_thread_exited() {
  return atomic_load(&g_ping_thread_exited);
}