tcp_proxy.h

Go to the documentation of this file.

#ifndef TCP_PROXY_H
#define TCP_PROXY_H

#include <stdint.h>
#include <stdio.h>
#include <pbl.h>
#include <ev.h>

#include "config/conf.h"

typedef enum {
    
    WAIT_FOR_START, 
    WAIT_FOR_END, 
    PROCESS_PACKET 
    
} ksnTCPProxyBufferStage;

typedef enum {
    
    CMD_TCPP_PROXY      
//    CMD_TCPP_GET_IPS,   ///< Get host IPs list
//    CMD_TCPP_GET_PORT   ///< Get host UDP proxy port
    
} ksnTCPProxyCommand;

typedef struct ksnTCPProxyHeader {
    
    uint8_t checksum; 
    unsigned int version : 4; 
    unsigned int command : 4; 
    uint16_t port; 
    uint16_t packet_length; 
    uint8_t addr_length; 
    uint8_t packet_checksum; 
    
} ksnTCPProxyHeader;

typedef struct ksnTCPProxyPacketData {
    
    char buffer[KSN_BUFFER_DB_SIZE]; 
    ksnTCPProxyHeader* header; 
    size_t length; 
    size_t ptr;  
    int stage;   
    
} ksnTCPProxyPacketData;

typedef struct ksnTCPProxyData {
    
    int tcp_proxy_fd;   
    int udp_proxy_port; 
    int udp_proxy_fd;   
    ev_io w;            
    ev_io w_udp;        
    ksnTCPProxyPacketData packet; 
    
} ksnTCPProxyData;

typedef struct ksnTCPProxyClass {
    
    void *ke;       
    
    // Client
    int fd_client;  
    ev_io w_client; 
    ksnTCPProxyPacketData packet; 
    
    // Server
    int fd;         
    PblMap* map;    
    
} ksnTCPProxyClass;

#ifdef  __cplusplus
extern "C" {
#endif

ksnTCPProxyClass *ksnTCPProxyInit(void *ke);
void ksnTCPProxyDestroy(ksnTCPProxyClass *tp);

int ksnTCPProxyClientConnect(ksnTCPProxyClass *tp);

#ifdef  __cplusplus
}
#endif

#endif  /* TCP_PROXY_H */