knetrs 0.1.1

/*
 * Copyright (c) 2014-2016, dennis wang
 * All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in the
 *       documentation and/or other materials provided with the distribution.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#ifndef CONFIG_H
#define CONFIG_H

#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <stdio.h>
#include <limits.h>
#include <assert.h>

#if (defined(WIN32) || defined(_WIN64))
    #if defined(_WIN32_WINNT)
		#undef _WIN32_WINNT
        #define _WIN32_WINNT 0x0600
    #else
		#define _WIN32_WINNT 0x0600
    #endif /* defined(_WIN32_WINNT) */
#ifndef WIN32_LEAN_AND_MEAN
    #define WIN32_LEAN_AND_MEAN
#endif
    #include <windows.h>
    #include <winsock2.h>
    #include <mswsock.h>
    #include <process.h>
    #include <ws2tcpip.h> /* getaddrinfo */
    #if defined(_MSC_VER )
        #pragma comment(lib,"wsock32.lib")
    #endif /* defined(_MSC_VER) */
    typedef SOCKET socket_t;
    typedef int socket_len_t;
    typedef uintptr_t thread_id_t;
    typedef DWORD sys_error_t;
    typedef volatile LONG atomic_counter_t;
    typedef signed char int8_t;
    typedef unsigned char uint8_t;
    typedef unsigned short uint16_t;
    typedef signed short int16_t;
    typedef unsigned int uint32_t;
    typedef signed int int32_t;
    #ifndef CYGWIN
    typedef unsigned long long uint64_t ;
    typedef signed long long int64_t;
    #endif /* CYGWIN */
    #define vsnprintf _vsnprintf
    #ifndef PATH_MAX
        #define PATH_MAX MAX_PATH
    #endif /* PATH_MAX */
    #ifdef __cplusplus
    #define FuncExport extern "C" __declspec(dllexport)
    #else
    #define FuncExport __declspec(dllexport)
    #endif /* __cplusplus */
#else
    #include <stdint.h>
    #include <errno.h>
    #include <netdb.h>
    #include <sys/types.h>
    #include <sys/time.h>
    #include <netinet/in.h>
    #include <sys/socket.h>
    #include <arpa/inet.h>
    #include <fcntl.h>
    #include <unistd.h>
    #include <pthread.h>
    #ifndef __APPLE__
    #include <sys/epoll.h>
    #endif /*__APPLE__*/
    #define socket_len_t socklen_t
    #define thread_id_t pthread_t
    #define socket_t int
    #define sys_error_t int
    #define atomic_counter_t volatile int
    #ifdef __cplusplus
    #define FuncExport extern "C"
    #else
    #define FuncExport
    #endif
#endif /* defined(WIN32) || defined(_WIN64) */

#define float32_t float
#define float64_t double

#ifndef INT_MAX
/* from stdint.h */
#define INT_MAX  2147483647 /* maximum (signed) int value */
#endif /* INT_MAX */

#define knet_create(type)                         (type*)knet_malloc(sizeof(type))
#define knet_create_raw(size)                     (char*)knet_malloc(size)
#define knet_create_type(type, size)              (type*)knet_malloc(size)
#define knet_create_type_ptr_array(type, n)       (type**)knet_malloc((n) * sizeof(type*))
#define knet_rcreate_raw(ptr, size)               (char*)knet_realloc(ptr, size)
#define knet_rcreate_type(type, ptr, size)        (type*)knet_realloc(ptr, size)
#define knet_rcreate_type_ptr_array(type, ptr, n) (type**)knet_realloc(ptr, (n) * sizeof(type*))

extern void knet_free(void* ptr); /* free */
extern void* knet_malloc(size_t size); /* malloc */
extern void* knet_realloc(void* ptr, size_t size); /* realloc */

typedef void* (*knet_malloc_func_t)(size_t);
typedef void* (*knet_realloc_func_t)(void*, size_t);
typedef void  (*knet_free_func_t)(void*);

typedef struct _loop_t kloop_t;
typedef struct _channel_t kchannel_t;
typedef struct _channel_ref_t kchannel_ref_t;
typedef struct _address_t kaddress_t;
typedef struct _lock_t klock_t;
typedef struct _loop_balancer_t kloop_balancer_t;
typedef struct _thread_runner_t kthread_runner_t;
typedef struct _stream_t kstream_t;
typedef struct _kdlist_t kdlist_t;
typedef struct _kdlist_node_t kdlist_node_t;
typedef struct _ringbuffer_t kringbuffer_t;
typedef struct _buffer_t kbuffer_t;
typedef struct _ktimer_loop_t ktimer_loop_t;
typedef struct _ktimer_t ktimer_t;
typedef struct _logger_t klogger_t;
typedef struct _hash_t khash_t;
typedef struct _hash_value_t khash_value_t;
typedef struct _loop_profile_t kloop_profile_t;
typedef struct _trie_t ktrie_t;
typedef struct _ip_filter_t kip_filter_t;
typedef struct _rwlock_t krwlock_t;
typedef struct _cond_t kcond_t;
typedef struct _rb_tree_t krbtree_t;
typedef struct _rb_node_t krbnode_t;

/* 管道可投递事件 */
typedef enum _channel_event_e {
    channel_event_recv = 1,
    channel_event_send = 2,
} knet_channel_event_e;

/*! 管道状态 */
typedef enum _channel_state_e {
    channel_state_connect = 1, /*! 主动发起连接，连接未完成 */
    channel_state_accept = 2,  /*! 监听 */
    channel_state_close = 4,   /*! 管道已关闭 */
    channel_state_active = 8,  /*! 管道已激活，可以收发数据 */
    channel_state_init = 16,   /*! 管道已建立，但未连接 */
} knet_channel_state_e;

/*! 定时器类型 */
typedef enum _ktimer_type_e {
    ktimer_type_once   = 1, /*! 运行一次 */
    ktimer_type_period = 2, /*! 无限 */
    ktimer_type_times  = 3, /*! 多次运行 */
} ktimer_type_e;

/*! 负载均衡配置 */
typedef enum _loop_balance_option_e {
    loop_balancer_in  = 1, /*! 开启其他kloop_t的管道在当前kloop_t负载 */
    loop_balancer_out = 2, /*! 开启当前kloop_t的管道到其他kloop_t内负载 */
} knet_loop_balance_option_e;

/*! 红黑树节点颜色 */
typedef enum _rb_color_e {
    rb_color_red = 1, /* 红色节点 */
    rb_color_black,   /* 黑色节点 */
} rb_color_e;

/* 错误码 */
typedef enum _error_e {
    error_ok = 0,
    error_fail,
    error_invalid_parameters,
    error_must_be_shared_channel_ref,
    error_invalid_channel,
    error_invalid_broadcast,
    error_no_memory,
    error_hash_not_found,
    error_recv_fail,
    error_send_fail,
    error_send_patial,
    error_recv_buffer_full,
    error_recv_nothing,
    error_connect_fail,
    error_connect_in_progress,
    error_channel_not_connect,
    error_accept_in_progress,
    error_bind_fail,
    error_listen_fail,
    error_ref_nonzero,
    error_loop_fail,
    error_loop_attached,
    error_loop_not_found,
    error_loop_impl_init_fail,
    error_thread_start_fail,
    error_already_close,
    error_impl_add_channel_ref_fail,
    error_broadcast_not_found,
    error_getpeername,
    error_getsockname,
    error_not_correct_domain,
    error_multiple_start,
    error_not_connected,
    error_logger_write,
    error_set_tls_fail,
    error_recvbuffer_not_enough,
    error_recvbuffer_locked,
    error_stream_enable,
    error_stream_disable,
    error_stream_flush,
    error_stream_buffer_overflow,
    error_trie_not_found,
    error_trie_key_exist,
    error_trie_for_each_fail,
    error_ip_filter_open_fail,
    error_router_wire_not_found,
    error_router_wire_exist,
    error_ringbuffer_not_found,
    error_getaddrinfo_fail,
} knet_error_e;

/*! 管道回调事件 */
typedef enum _channel_cb_event_e {
    channel_cb_event_connect = 1,          /*! 连接完成 */
    channel_cb_event_accept = 2,           /*! 管道监听到了新连接请求 */ 
    channel_cb_event_recv = 4,             /*! 管道有数据可以读 */
    channel_cb_event_send = 8,             /*! 管道发送了字节，保留 */
    channel_cb_event_close = 16,           /*! 管道关闭 */
    channel_cb_event_timeout = 32,         /*! 管道读空闲 */
    channel_cb_event_connect_timeout = 64, /*! 主动发起连接，但连接超时 */
} knet_channel_cb_event_e;

/* 日志等级 */
typedef enum _logger_level_e {
    logger_level_verbose = 1, /* verbose - 尽量输出 */
    logger_level_information, /* information - 提示信息 */
    logger_level_warning,     /* warning - 警告 */ 
    logger_level_error,       /* error - 错误 */
    logger_level_fatal,       /* fatal - 致命错误 */
} knet_logger_level_e;

/* 日志模式 */
typedef enum _logger_mode_e {
    logger_mode_file = 1,     /* 生成日志文件 */
    logger_mode_console = 2,  /* 打印到stderr */
    logger_mode_flush = 4,    /* 每次写日志同时清空缓存 */
    logger_mode_override = 8, /* 覆盖已存在的日志文件 */
} knet_logger_mode_e;

/*! 线程函数 */
typedef void (*knet_thread_func_t)(kthread_runner_t*);
/*! 管道事件回调函数 */
typedef void (*knet_channel_ref_cb_t)(kchannel_ref_t*, knet_channel_cb_event_e);
/*! 定时器回调函数 */
typedef void (*ktimer_cb_t)(ktimer_t*, void*);
/*! RPC加密回调函数, 返回 非零 加密后长度, 0 失败 */
typedef uint16_t (*krpc_encrypt_t)(void*, uint16_t, void*, uint16_t);
/*! RPC解密回调函数, 返回 非零 解密后长度, 0 失败 */
typedef uint16_t (*krpc_decrypt_t)(void*, uint16_t, void*, uint16_t);
/*! 哈希表元素销毁函数 */
typedef void (*knet_hash_dtor_t)(void*);
/*! trie元素销毁函数 */
typedef void (*knet_trie_dtor_t)(void*);
/*! trie遍历函数 */
typedef int (*knet_trie_for_each_func_t)(const char*, void*);
/*! 红黑树节点销毁回调函数 */
typedef void(*knet_rb_node_destroy_cb_t)(void*, uint64_t);

/* 根据需要， 开启不同选取器 */
#if (defined(WIN32) || defined(_WIN64))
    #define LOOP_IOCP 1    /* IOCP */
    #define LOOP_SELECT 0  /* select */
#elif __APPLE__
    #define LOOP_EPOLL 0   /* epoll */
    #define LOOP_SELECT 1  /* select */
#else
    #define LOOP_EPOLL 1   /* epoll */
    #define LOOP_SELECT 0  /* select */
#endif /* defined(WIN32) || defined(_WIN64) */

#if defined(DEBUG) || defined(_DEBUG) || !defined(NDEBUG)
    #define LOGGER_ON 1 /* 调试版本开启日志 */
#else
    #define LOGGER_ON 0 /* 发行版关闭日志 */
#endif /* defined(DEBUG) || defined(_DEBUG) */

#define LOGGER_MODE (logger_mode_file | logger_mode_console | logger_mode_flush | logger_mode_override) /* 日志模式 */
#define LOGGER_LEVEL logger_level_fatal /* 日志等级 */

#if defined(DEBUG) || defined(_DEBUG)
    #define verify(expr) assert(expr)
#elif defined(NDEBUG)
    #define verify(expr) \
        if (!(expr)) { \
            log_fatal("crash point abort, file:(%s:%d): cause:(%s)", __FILE__, __LINE__, #expr); \
            abort(); \
        }
#else
    #define verify(expr) \
        if (!(expr)) { \
            log_fatal("crash point abort, file:(%s:%d): cause:(%s)", __FILE__, __LINE__, #expr); \
            abort(); \
        }
#endif /* defined(DEBUG) || defined(_DEBUG) */

#endif /* CONFIG_H */