websocket_server.cpp

#include "netlib.h"
#include "EncDec.h"
#include "ConfigFileReader.h"
#include "MsgConn.h"
#include "LoginServConn.h"
#include "RouteServConn.h"
#include "DBServConn.h"
#include "PushServConn.h"
#include "FileServConn.h"
//#include "version.h"
#include "WebSocketConn.h"
//#define NEEDSSL 


#ifdef NEEDSSL
#include "SSLSocket.h"
#endif


#define DEFAULT_CONCURRENT_DB_CONN_CNT  10

CAes *pAes;

// for client connect in
void msg_serv_callback(void* callback_data, uint8_t msg, uint32_t handle, void* pParam)
{
    if (msg == NETLIB_MSG_CONNECT)
    {
        //CMsgConn* pConn = new CMsgConn();
        CMsgConn* pConn = new WebSocketConn();
        pConn->OnConnect(handle);
    }
    else
    {
        log("!!!error msg: %d ", msg);
    }
}

#ifdef NEEDSSL
void try_run_ssl_server(CConfigFileReader *config_file) {
    char* listen_ip = config_file->GetConfigName("SSLListenIP");
    char* listen_port_str = config_file->GetConfigName("SSLListenPort");
    char* CertFile = config_file->GetConfigName("SSLCertFile");
    char* KeyFile = config_file->GetConfigName("SSLKeyFile");
    if(listen_ip && listen_port_str && CertFile && KeyFile) {
        initSSLConfig(CertFile,KeyFile);
        uint16_t listen_port = atoi(listen_port_str);
        CStrExplode listen_ip_list(listen_ip, ';');
        for (uint32_t i = 0; i < listen_ip_list.GetItemCnt(); i++) {
             int ret = netlib_ssl_listen(listen_ip_list.GetItem(i), listen_port, msg_serv_callback, NULL);
        }
    }else {
        log("not config for ssl");
    }
}
#endif


int main(int argc, char* argv[])
{
    if ((argc == 2) && (strcmp(argv[1], "-v") == 0)) {
//        printf("Server Version: MsgServer/%s\n", VERSION);
        printf("Server Build: %s %s\n", __DATE__, __TIME__);
        return 0;
    }

    signal(SIGPIPE, SIG_IGN);
    srand(time(NULL));

    log("MsgServer max files can open: %d ", getdtablesize());

    CConfigFileReader config_file("websocketserver.conf");

    char* listen_ip = config_file.GetConfigName("ListenIP");
    char* str_listen_port = config_file.GetConfigName("ListenPort");
    char* ip_addr1 = config_file.GetConfigName("IpAddr1");    // 电信IP
    char* ip_addr2 = config_file.GetConfigName("IpAddr2");    // 网通IP
    char* str_max_conn_cnt = config_file.GetConfigName("MaxConnCnt");
    char* str_aes_key = config_file.GetConfigName("aesKey");
    uint32_t db_server_count = 0;
    serv_info_t* db_server_list = read_server_config(&config_file, "DBServerIP", "DBServerPort", db_server_count);

    uint32_t login_server_count = 0;
    serv_info_t* login_server_list = read_server_config(&config_file, "LoginServerIP", "LoginServerPort", login_server_count);

    uint32_t route_server_count = 0;
    serv_info_t* route_server_list = read_server_config(&config_file, "RouteServerIP", "RouteServerPort", route_server_count);

    uint32_t push_server_count = 0;
    serv_info_t* push_server_list = read_server_config(&config_file, "PushServerIP",
                                                       "PushServerPort", push_server_count);
    
    uint32_t file_server_count = 0;
    serv_info_t* file_server_list = read_server_config(&config_file, "FileServerIP",
                                                       "FileServerPort", file_server_count);
    
    if (!str_aes_key || strlen(str_aes_key)!=32) {
        log("aes key is invalied");
        return -1;
    }
 
    pAes = new CAes(str_aes_key);
    
    // 必须至少配置2个BusinessServer实例, 一个用于用户登录业务，一个用于其他业务
    // 这样当其他业务量非常繁忙时，也不会影响客服端的登录验证
    // 建议配置4个实例，这样更新BusinessServer时，不会影响业务
    if (db_server_count < 2) {
        log("DBServerIP need 2 instance at lest ");
        return 1;
    }

    // 到BusinessServer的开多个并发的连接
    uint32_t concurrent_db_conn_cnt = DEFAULT_CONCURRENT_DB_CONN_CNT;
    uint32_t db_server_count2 = db_server_count * DEFAULT_CONCURRENT_DB_CONN_CNT;
    char* concurrent_db_conn = config_file.GetConfigName("ConcurrentDBConnCnt");
    if (concurrent_db_conn) {
        concurrent_db_conn_cnt  = atoi(concurrent_db_conn);
        db_server_count2 = db_server_count * concurrent_db_conn_cnt;
    }

    serv_info_t* db_server_list2 = new serv_info_t [ db_server_count2];
    for (uint32_t i = 0; i < db_server_count2; i++) {
        db_server_list2[i].server_ip = db_server_list[i / concurrent_db_conn_cnt].server_ip.c_str();
        db_server_list2[i].server_port = db_server_list[i / concurrent_db_conn_cnt].server_port;
    }

    if (!listen_ip || !str_listen_port || !ip_addr1) {
        log("config file miss, exit... ");
        return -1;
    }

    // 没有IP2，就用第一个IP
    if (!ip_addr2) {
        ip_addr2 = ip_addr1;
    }

    uint16_t listen_port = atoi(str_listen_port);
    uint32_t max_conn_cnt = atoi(str_max_conn_cnt);

    int ret = netlib_init();

    if (ret == NETLIB_ERROR)
        return ret;

    CStrExplode listen_ip_list(listen_ip, ';');
    for (uint32_t i = 0; i < listen_ip_list.GetItemCnt(); i++) {
        ret = netlib_listen(listen_ip_list.GetItem(i), listen_port, msg_serv_callback, NULL);
        if (ret == NETLIB_ERROR)
            return ret;
    }

    printf("server start listen on: %s:%d\n", listen_ip, listen_port);

#ifdef NEEDSSL
    try_run_ssl_server(&config_file);
#endif
    init_msg_conn();

    init_file_serv_conn(file_server_list, file_server_count);

    init_db_serv_conn(db_server_list2, db_server_count2, concurrent_db_conn_cnt);

    init_login_serv_conn(login_server_list, login_server_count, ip_addr1, ip_addr2, listen_port, max_conn_cnt);

    init_route_serv_conn(route_server_list, route_server_count);

    init_push_serv_conn(push_server_list, push_server_count);
    printf("now enter the event loop...\n");
    
    writePid();

    netlib_eventloop();

    return 0;
}