// tcp服务端，增加了心跳报文。
#include "_el.h"

using namespace eviwbh;

ctcpserver tcpserver;  // 创建服务端对象。
clogfile logfile;            // 服务程序的运行日志。

void FathEXIT(int sig);  // 父进程退出函数。
void ChldEXIT(int sig);  // 子进程退出函数。

string strsendbuffer;   // 发送报文的buffer。
string strrecvbuffer;    // 接收报文的buffer。

bool bizmain();    // 业务处理主函数。

int main(int argc,char *argv[])
{
    if (argc!=4)
    {
      printf("Using:./serverHeartbeat port logfile timeout\n");
      printf("Example:./serverHeartbeat 5005 /log/idc/demo06.log 10\n\n"); 
      return -1;
    }

    // 关闭全部的信号和输入输出。
    // 设置信号,在shell状态下可用 "kill + 进程号" 正常终止些进程
    // 但请不要用 "kill -9 +进程号" 强行终止
    //closeioandsignal(false); 
    signal(SIGINT,FathEXIT); signal(SIGTERM,FathEXIT);

    if (logfile.open(argv[2])==false) { printf("logfile.open(%s) failed.\n",argv[2]); return -1; }

    // 服务端初始化。
    if (tcpserver.initserver(atoi(argv[1]))==false)
    {
      logfile.write("tcpserver.initserver(%s) failed.\n",argv[1]); return -1;
    }

    while (true)
    {
        // 获取客户端的连接请求。
        if (tcpserver.accept()==false)
        {
            logfile.write("tcpserver.accept() failed.\n"); FathEXIT(-1);
        }

        logfile.write("客户端（%s）已连接。\n",tcpserver.getip());

        if (fork()>0) { tcpserver.closeclient(); continue; }  // 父进程继续回到accept()。
   
        // 子进程重新设置退出信号。
        signal(SIGINT,ChldEXIT); signal(SIGTERM,ChldEXIT);

        tcpserver.closelisten();     // 子进程关闭监听的socket。

        while (true)
        {
            // 子进程与客户端进行通讯，处理业务。
            if (tcpserver.read(strrecvbuffer,atoi(argv[3]))==false)
            {
                logfile.write("tcpserver.read() failed.\n"); ChldEXIT(0);
            }
            logfile.write("接收：%s\n",strrecvbuffer.c_str());

            bizmain();    // 业务处理主函数。

            if (tcpserver.write(strsendbuffer)==false)
            {
                logfile.write("tcpserver.send() failed.\n"); ChldEXIT(0);
            }
            logfile.write("发送：%s\n",strsendbuffer.c_str());
        }

        ChldEXIT(0);
    }
}

// 父进程退出函数。
void FathEXIT(int sig)  
{
    // 以下代码是为了防止信号处理函数在执行的过程中被信号中断。
    signal(SIGINT,SIG_IGN); signal(SIGTERM,SIG_IGN);

    logfile.write("父进程退出，sig=%d。\n",sig);

    tcpserver.closelisten();    // 关闭监听的socket。

    kill(0,15);     // 通知全部的子进程退出。

    exit(0);
}

// 子进程退出函数。
void ChldEXIT(int sig)  
{
    // 以下代码是为了防止信号处理函数在执行的过程中被信号中断。
    signal(SIGINT,SIG_IGN); signal(SIGTERM,SIG_IGN);

    logfile.write("子进程退出，sig=%d。\n",sig);

    tcpserver.closeclient();    // 关闭客户端的socket。

    exit(0);
}

void login();   // 登录。
void check_balance();   // 查询余额。
void transfer();   // 转帐。

bool businessmain()    // 业务处理主函数。
{
    int id;  // 业务代码。
    getxmlbuffer(strrecvbuffer,"id",id);

    switch(id)
    {
        case 1:    // 登录。
            login();
            break;
        case 2:    // 查询余额。
            check_balance();
            break;
        case 3:    // 转帐。
            transfer();
            break;
        default:   // 非法报文。
            strsendbuffer="<retcode>9</retcode><message>业务不存在。</message>";
            break;
    }

    return true;
}

void login()   // 登录。
{
    string username,password;
    getxmlbuffer(strrecvbuffer,"username",username);
    getxmlbuffer(strrecvbuffer,"password",password);

    if ( (username=="0369") && (password=="3690") )
        strsendbuffer="<retcode>0</retcode><message>成功。</message>";
    else
        strsendbuffer="<retcode>-1</retcode><message>用户名或密码不正确。</message>";
}

void check_balance()   // 查询余额。
{
    string cardid;
    getxmlbuffer(strrecvbuffer,"cardid",cardid);  // 获取卡号。

    // 假装操作了数据库，得到了卡的余额。

    strsendbuffer="<retcode>0</retcode><ye>128.83</ye>";
}

void transfer()   // 转帐。
{
    string cardid1,cardid2;
    getxmlbuffer(strrecvbuffer,"cardid1",cardid1);
    getxmlbuffer(strrecvbuffer,"cardid2",cardid2);
    double transfer;
    getxmlbuffer(strrecvbuffer,"transfer",transfer);

    // 假装操作了数据库，更新了两个账户的金额，完成了转帐操作。

    if ( transfer<100 )
        strsendbuffer="<retcode>0</retcode><message>成功。</message>";
    else
        strsendbuffer="<retcode>-1</retcode><message>余额不足。</message>";
}