lib/network/tcp/server.h

Go to the documentation of this file.

#pragma once
 
#include <stdint.h>
#include <stdbool.h>
#include <stdatomic.h>
#include "common.h"
#include "platform/socket.h"
#include "platform/abstraction.h"
#include "thread_pool.h"
#include "uthash.h"
 
// 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 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;                       
  int accept_timeout_sec;               
  tcp_client_handler_fn client_handler; 
  void *user_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_bool running;        
  tcp_server_config_t config; 
 
  // Client registry (thread-safe)
  tcp_client_entry_t *clients;      
  mutex_t clients_mutex;            
  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_shutdown(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);