connection_manager.h

#pragma once
 
#include "net_socket.h"
#include <stdint.h>
#include <stdbool.h>
#include <winsock2.h>
#include "config.h" 
// Include networked_entity.h to obtain the definition of interest_mask_t.
// This header does not depend on connection_manager, so including it here
// avoids circular dependencies and allows peers to have an interest mask.
#include "networked_entity.h"
 
 
#ifdef __cplusplus
extern "C" {
#endif
 

#define PROTOCOL_COUNT 2
#ifndef MAX_PACKET_SIZE
#define MAX_PACKET_SIZE 1024
#endif
#define MAX_PEERS 32

typedef struct protocol_handler_t protocol_handler_t;
typedef struct network_cs_t network_cs_t;
typedef struct connection_manager_t connection_manager_t;
 

typedef struct peer_t {
    char id[64];                            
    struct sockaddr_in addr_tcp;            
    struct sockaddr_in addr_udp;            
    net_socket_t net_sockets[PROTOCOL_COUNT]; 
    bool is_connected;                      
    bool udp_ready;                      

    interest_mask_t interest_mask;
 
    //  TCP reassembly (used for framing)
    uint8_t tcp_rx_buf[MAX_PACKET_SIZE * 2];
    int     tcp_rx_len;
} peer_t;
 

typedef struct connection_manager_t {
    peer_t peers[MAX_PEERS];                
    int peer_count;                         
    net_socket_t listen_sockets[PROTOCOL_COUNT]; 
    bool is_server;                         
    void* user_data;                        
 
    // Callbacks for different network events
    void (*on_receive)(void*, peer_t*, const void*, int);       
    void (*on_connect)(void*, peer_t*);                         
    void (*on_disconnect)(void*, peer_t*);                      
} connection_manager_t;

void connection_manager_init(connection_manager_t* connection_manager);

void connection_manager_add_listen_socket(connection_manager_t* connection_manager, net_socket_t socket, socket_type_t socket_type);

void connection_manager_remove_peer(connection_manager_t* connection_manager, const char* peer_id);

int connection_manager_send_to_peer(connection_manager_t* connection_manager, const char* peer_id, const void* data, int len);

int connection_manager_send_to_peer_udp(connection_manager_t* connection_manager, const char* peer_id, const void* data, int len);

int connection_manager_broadcast(connection_manager_t* connection_manager, const void* data, int len);

int connection_manager_connect_to_server(connection_manager_t* connection_manager, const char* ip, uint16_t port);

void connection_manager_update(connection_manager_t* connection_manager);

void connection_manager_destroy(connection_manager_t* connection_manager);

net_socket_t* connection_manager_get_listen_socket(connection_manager_t* connection_manager, socket_type_t type);
 
uint16_t connection_manager_get_udp_local_port(connection_manager_t* connection_manager);

int connection_manager_set_peer_udp_remote_port_by_id(connection_manager_t* connection_manager, const char* peer_id, uint16_t remote_udp_port);

peer_t* find_peer_by_addr(connection_manager_t* connection_manager, const struct sockaddr_in* addr, socket_type_t socket_type);

peer_t* connection_manager_get_peer(connection_manager_t* connection_manager, const char* peer_id);

void connection_manager_set_peer_interest(connection_manager_t* connection_manager, const char* peer_id, interest_mask_t mask);
 
#ifdef __cplusplus
}
#endif