include/ascii-chat/network/tcp/server.h

Go to the documentation of this file.

#pragma once
 
#include <stdint.h>
#include <stdbool.h>
#include "../../atomic.h"
#include "../../common.h"
#include "../../platform/socket.h"
#include "../../platform/abstraction.h"
#include "../../thread_pool.h"
#include <ascii-chat/uthash.h> // For UT_hash_handle
 
// Forward declarations
typedef struct tcp_client_entry tcp_client_entry_t;
typedef struct tcp_server tcp_server_t;
 
typedef void (*tcp_client_cleanup_fn)(void *client_data);
 
typedef void (*tcp_client_foreach_fn)(socket_t socket, void *client_data, void *user_arg);
 
typedef void (*tcp_status_update_fn)(void *user_data);
 
typedef struct {
  socket_t client_socket;       
  struct sockaddr_storage addr; 
  socklen_t addr_len;           
  void *user_data;              
} tcp_client_context_t;
 
typedef void *(*tcp_client_handler_fn)(void *arg);
 
typedef struct {
  int port;                              
  const char *ipv4_address;              
  const char *ipv6_address;              
  bool bind_ipv4;                        
  bool bind_ipv6;                        
  double accept_timeout_sec;             
  tcp_client_handler_fn client_handler;  
  void *user_data;                       
  tcp_status_update_fn status_update_fn; 
  void *status_update_data;              
} tcp_server_config_t;
 
struct tcp_client_entry {
  socket_t socket;        
  void *client_data;      
  thread_pool_t *threads; 
  UT_hash_handle hh;      
};
 
struct tcp_server {
  socket_t listen_socket;     
  socket_t listen_socket6;    
  atomic_t running;           
  tcp_server_config_t config; 
 
  // Client registry (thread-safe with read-write lock)
  tcp_client_entry_t *clients;      
  rwlock_t clients_rwlock;          
  tcp_client_cleanup_fn cleanup_fn; 
};
 
asciichat_error_t tcp_server_init(tcp_server_t *server, const tcp_server_config_t *config);
 
asciichat_error_t tcp_server_run(tcp_server_t *server);
 
void tcp_server_destroy(tcp_server_t *server);
 
void tcp_server_set_cleanup_callback(tcp_server_t *server, tcp_client_cleanup_fn cleanup_fn);
 
asciichat_error_t tcp_server_add_client(tcp_server_t *server, socket_t socket, void *client_data);
 
asciichat_error_t tcp_server_remove_client(tcp_server_t *server, socket_t socket);
 
asciichat_error_t tcp_server_get_client(tcp_server_t *server, socket_t socket, void **out_data);
 
void tcp_server_foreach_client(tcp_server_t *server, tcp_client_foreach_fn callback, void *user_arg);
 
size_t tcp_server_get_client_count(tcp_server_t *server);
 
// ============================================================================
// Client Context Utilities
// ============================================================================
 
const char *tcp_client_context_get_ip(const tcp_client_context_t *ctx, char *buf, size_t len);
 
int tcp_client_context_get_port(const tcp_client_context_t *ctx);
 
void tcp_server_reject_client(socket_t socket, const char *reason);
 
// ============================================================================
// Client Thread Pool Management
// ============================================================================
 
asciichat_error_t tcp_server_spawn_thread(tcp_server_t *server, socket_t client_socket, void *(*thread_func)(void *),
                                          void *thread_arg, int stop_id, const char *thread_name);
 
asciichat_error_t tcp_server_stop_client_threads(tcp_server_t *server, socket_t client_socket);
 
asciichat_error_t tcp_server_get_thread_count(tcp_server_t *server, socket_t client_socket, size_t *count);