src/discovery-service/server.h

Go to the documentation of this file.

#pragma once
 
#include <stdint.h>
#include <stdbool.h>
#include <sqlite3.h>
#include <ascii-chat/atomic.h>
#include <ascii-chat/network/tcp/server.h>
#include <ascii-chat/network/acip/acds.h>
#include <ascii-chat/network/acip/transport.h>
#include <ascii-chat/network/websocket/server.h>
#include <ascii-chat/options/options.h> // For MAX_IDENTITY_KEYS
#include <ascii-chat/thread_pool.h>
#include <ascii-chat/crypto/handshake/common.h>
#include "discovery-service/main.h"
#include <ascii-chat/discovery/session.h> // For host_lost_candidate_t and MAX_PARTICIPANTS
 
typedef struct {
  uint8_t session_id[16];     
  uint8_t participant_id[16]; 
  bool joined_session;        
 
  // Crypto handshake state
  crypto_handshake_context_t handshake_ctx; 
  bool handshake_complete;                  
 
  // Multi-key session creation state
  bool in_multikey_session_create;                     
  acip_session_create_t pending_session;               
  uint8_t pending_session_keys[MAX_IDENTITY_KEYS][32]; 
  size_t num_pending_keys;                             
 
  // Transport abstraction
  acip_transport_t *transport; 
  int registry_id;             
 
  // Protocol state
  bool session_established; 
} acds_client_data_t;
 
typedef struct {
  uint8_t session_id[16];      
  uint64_t migration_start_ns; 
} migration_context_t;
 
typedef struct {
  atomic_t bytes_sent;       
  atomic_t bytes_received;   
  atomic_t packets_sent;     
  atomic_t packets_received; 
  time_t start_time;         
} acds_stats_t;
 
typedef struct {
  tcp_server_t *tcp_server; 
 
  // Identity
  uint8_t identity_public[32]; 
  uint8_t identity_secret[64]; 
 
  // Persistence (SQLite as single source of truth for sessions)
  sqlite3 *db; 
 
  // Rate limiting
  struct rate_limiter_s *rate_limiter; 
 
  // Host migration tracking (in-memory during active migrations)
  migration_context_t active_migrations[32]; 
  size_t num_active_migrations;              
 
  // Background worker threads (cleanup, etc.)
  thread_pool_t *worker_pool; 
  atomic_t shutdown;          
 
  // Statistics (TCP + WebSocket)
  acds_stats_t stats; 
 
  // Configuration
  acds_config_t config; 
} acds_server_t;
 
asciichat_error_t acds_server_init(acds_server_t *server, const acds_config_t *config);
 
void acds_server_shutdown(acds_server_t *server);
 
void *acds_client_handler(void *arg);
 
void *acds_websocket_client_handler(void *arg);