mptunnel.h

#ifndef __MPTUNNEL_H__
#define __MPTUNNEL_H__ 1

#include <stdlib.h>
#include <stdio.h>
#include <sys/time.h>
#include <time.h>
#include <string.h>
#include <stdint.h>

#include <ev.h>

#include <locale.h>
#include <libintl.h>

#include "linklist.h"
#include "rbtree.h"

#define _(STR) gettext(STR)

// 转发时最大的包长度
#define MAX_PACKET_SIZE 8000

// 客户端检测的到桥端的连接超时时间
#define CLIENT_BRIDGE_TIMEOUT   60

#define LOG_ERROR 1
#define LOG_WARNING 2
#define LOG_NOTICE 3
#define LOG_INFO 4
#define LOG_LDEBUG 5
#define LOG_DEBUG 6

#define LOGD(FMT, ...) LOG(LOG_DEBUG, FMT, ##__VA_ARGS__)
#define LOGW(FMT, ...) LOG(LOG_WARNING, FMT, ##__VA_ARGS__)
#define LOGE(FMT, ...) LOG(LOG_ERROR, FMT, ##__VA_ARGS__)
#define LOGN(FMT, ...) LOG(LOG_NOTICE, FMT, ##__VA_ARGS__)
#define LOGI(FMT, ...) LOG(LOG_INFO, FMT, ##__VA_ARGS__)

#define LOG(level, FMT, ...) do {    \
    static struct tm *tmp = NULL; static time_t t1, t2 = (time_t)NULL; struct timeval tv; char timestr[128] = {0}; char ms[4] = {0};  \
    if ((t1 = time(NULL)) != t2) {t2 = t1; tmp = localtime(&t2); }     \
    gettimeofday(&tv, NULL);  \
    if (tmp == NULL) {  \
        strncpy(timestr, "unknow", sizeof(timestr) - 1);   \
    } else {    \
        strftime(timestr, sizeof(timestr), "%F %H:%M:%S.", tmp);  \
        sprintf(ms, "%03d", (int)tv.tv_usec / 1000);   \
        strncat(timestr, ms, sizeof(timestr) - strlen(timestr) - 1); \
    } \
    fputc('[', stderr); fputs(timestr, stderr); fputs("]", stderr); fprintf(stderr, "(%s:%d) ", __FILE__, __LINE__); fprintf(stderr, FMT, ##__VA_ARGS__); \
} while(0);


enum packet_type {
    PKT_TYPE_NONE,
    PKT_TYPE_CTL,
    PKT_TYPE_DATA
};

typedef struct packet_t {
    uint32_t iv;    /** 加密使用的 IV */
    enum packet_type type;
    int id;
    int buflen;     /** 数据长度（不包括 packet_t 自身） */
} packet_t;

typedef struct received_list_t {
    struct rb_node rbnode;
    long ctime;
    int id;
} received_list_t;
    
typedef struct received_t {
    int min_con_id;     // 连续收到的最小包编号
    int max_id;         // 目前已经收到的最大包编号
    time_t last_dropdead_time;
    struct list_head rlist;
    pthread_mutex_t rlist_mutex;
} received_t;

typedef struct connections_t {
    struct list_head list;
    int fd;
    char *host;
    int port;
    ev_io *watcher;
    int rc_time;    // 最后一次收到服务器端数据的时间
    int st_time;    // 最后一次向服务器发送数据包的时间
    unsigned char broken;   // 连接是否已经中断并需要重连，该记号由人工进行标记
} connections_t;

packet_t* packet_make(enum packet_type type, const char *buf, int buflen, int);
int packet_free(packet_t *p);
int packet_send(int fd, char *buf, int buflen, int);
    
int received_is_received(received_t *r, int id);
int received_try_dropdead(received_t *r, int ttl);
int received_init(received_t *r);
int received_add(received_t *r, int id);
int received_destroy(received_t *r);

received_list_t * received_rbtree_get(struct rb_root *, int);
int received_rbtree_add(struct rb_root * , received_list_t *);

void encrypt_lfsr(char *_buf, int _size, uint32_t *);
void decrypt_lfsr(char *_buf, int _size, uint32_t *);
void mpdecrypt(char *_buf);
void mpencrypt(char *_buf, int _buflen);

uint32_t lfsr_rand(uint32_t *);
#endif