cngp.cpp

短信防伪系统服务器端

// Cngp.cpp: implementation of the CCngp class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "EasyTradeService.h"
#include "Cngp.h"
#include "md5.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

void PK_CNGP(CNGP_SUBMIT * pSubmit,
               int*  pk_len,
               char *corpId,
               char* srcTermid,
               char* destTermid,
               char* chargeTermid,
               int   subType,
               int   feeuserType,
               char* feeType,
               char* feecode,
               char* serviceid,
               int   msgFormat,
               int   msgLength,
               char* msgContent)
               
{
    memset(pSubmit, 0, sizeof(CNGP_SUBMIT)); 
    pSubmit->nNeedReport=1;
    pSubmit->nDestTermIdCount= 1;
    pSubmit->nMsgFormat   = msgFormat;
    pSubmit->nMsgLength   = msgLength;
    pSubmit->nMsgType     = subType;
    pSubmit->sFeeuUserType =feeuserType;
    memcpy(pSubmit->spid,corpId,sizeof(pSubmit->spid));
    memcpy(pSubmit->sFeeType, feeType,sizeof(pSubmit->sFeeType)); 
    memcpy(pSubmit->sFeeCode, feecode,sizeof(pSubmit->sFeeCode)); 
    strcpy(pSubmit->sServerId, serviceid); 
    strcpy(pSubmit->sSrcTermid,srcTermid);
    strcpy(pSubmit->sDestTermId,destTermid);
    strcpy(pSubmit->sChargeTermId,chargeTermid);
    memcpy(pSubmit->sMsgContent,msgContent,msgLength);
    *pk_len=sizeof(CNGP_HEAD)+sizeof(CNGP_SUBMIT);
    *pk_len= *pk_len- (sizeof(pSubmit->sMsgContent)-msgLength);
}
CCngp::CCngp()
{

}

CCngp::~CCngp()
{

}
char *CCngp::BCDToString(char *src,int len, char *dest)
{
    char *decode="0123456789ABCDEF";
    for(int i=0;i<len;i++)
    {
        dest[i*2]=decode[src[i]>>4];
        dest[i*2+1]=decode[src[i]&0X0F];
    }
    dest[i*2]='\0';
    return dest;
}
void CCngp::OnRecv()
{ 
	int	ret,len,cmdid=0;
	FD_SET	fdset;
	TIMEVAL	timeout;
	timeout.tv_sec = 1;
	timeout.tv_usec= 0;    
    CNGP_HEAD          head; 
    char               recvbuff[512];
    char               szMsgId[24];
    char               szResp[64];
    CNGP_SUBMIT_RESP  *pSubmitResp=0;
    CNGP_DELIVER      *pDeliver=0;
    CNGP_DELIVER_RESP *pDeliverResp=0;
    DELIVER_SMS        deliverSMS;
    MTREPORT_SMS       reportSMS;  
	while(!m_bQuit)
	{
        FD_ZERO( &fdset);
        FD_SET(m_Socket, &fdset);
		ret = select(0,&fdset,NULL,NULL,&timeout);
		if( ret == SOCKET_ERROR)
        {
		     OnSocketError();
             continue;
        }
		if( ret == 0)
        {
            continue; 
        }
        ret =::recv( m_Socket ,(char*)&head, sizeof( CNGP_HEAD ) ,0); 
		if( ret != sizeof( CNGP_HEAD))
        {
            if(ret==0||ret== SOCKET_ERROR)
            {
                OnSocketError();
            }
			continue;
        }
		len= ntohl(head.size ) - sizeof( CNGP_HEAD );
        ret =::recv(m_Socket, recvbuff, len ,0 );
		if( ret != len )
        {  
            if(ret==0||ret==SOCKET_ERROR)
            {
                OnSocketError();
            }
			continue;
        }
		int cmdid=ntohl(head.cmdid);
        switch(cmdid)
        {
            case nCNGP_SUBMIT_RESP: 
                pSubmitResp= (CNGP_SUBMIT_RESP*)recvbuff;
                BCDToString(pSubmitResp->msgid, 10, szMsgId);
                OnSubmitResp(ntohl(head.seqid),ntohl(head.command_status),szMsgId);
                break;
            case nCNGP_DELIVER:
                pDeliver=(CNGP_DELIVER*)recvbuff;
                *((CNGP_HEAD *)szResp)=head;
                len=  sizeof( CNGP_HEAD) + sizeof(CNGP_DELIVER_RESP);
                ((CNGP_HEAD *)szResp)->command_status=0;
                ((CNGP_HEAD *)szResp)->size=htonl(len);
                ((CNGP_HEAD *)szResp)->cmdid=htonl(nCNGP_DELIVER_RESP);
                memcpy(((CNGP_DELIVER_RESP *)(szResp+ sizeof( CNGP_HEAD)))->msgid, pDeliver->msgid,10);
                ret=Send(szResp,len);
                if(ret==len)
                {
                    if(pDeliver->isReport)      //状态报告
                    {
                        int nPos=0;char *p=recvbuff ; 
                        p=p+3;
                        BCDToString(p,10, reportSMS.smgMsgId);
                        p=p+10;
	                    //p=strstr(p,"submit_date:");
                        p=p+12+10;
	                    //p=strstr(p,"done_date:");
                        p=p+10+10;
	                    p=strstr(p,"stat:");
                        memcpy(reportSMS.state,p+5,7);p=p+6+7;
	                    p=strstr(p,"err:");           p=p+4;
                        *(p+3)=0;reportSMS.errorCode=atoi(p);      
                        reportSMS.protrol=2;
                        strcpy(reportSMS.corpId,m_szCorpId);
                        strcpy(reportSMS.srcTermid,pDeliver->srcTermID);
                        strcpy(reportSMS.destTermid,pDeliver->destTermID);
                        OnMTReportSMS(&reportSMS);
                    }
                    else
                    {
                        deliverSMS.protrol=2;
                        strcpy(deliverSMS.corpId,m_szCorpId);
                        deliverSMS.msgFormat=pDeliver->msgFormat;
                        strcpy(deliverSMS.srcTermid,pDeliver->srcTermID);
                        strcpy(deliverSMS.destTermid,pDeliver->destTermID);
                        memcpy(deliverSMS.msgContent,pDeliver->msgContent,pDeliver->msgLength);
                        deliverSMS.msgContent[pDeliver->msgLength]=0;
                        BCDToString(deliverSMS.smgMsgId,10,pDeliver->msgid);
                        OnDeliverSMS(&deliverSMS);
                    }
                }
                else
                {
                    if(ret==0||ret==SOCKET_ERROR)
                    {
                        OnSocketError();
                    }
                }
                break;
		    case nCNGP_ACTIVE_TEST:
                head.command_status=0;
                head.cmdid=htonl(nCNGP_ACTIVE_TEST_RESP);
                ret=Send((char*)&head,sizeof(CNGP_HEAD));
                if(ret==0||ret==SOCKET_ERROR)
                {
                    OnSocketError();
                }
               break;
            default:
			    break;
        }
    }
}
int CCngp:: SubmitSMS(SUBMIT_SMS *p ,int len,int seqid)
{
    ((CNGP_HEAD * )(p->pkg))->cmdid = htonl(nCNGP_SUBMIT);
    ((CNGP_HEAD * )(p->pkg))->size  = htonl( len );
    ((CNGP_HEAD * )(p->pkg))->seqid = htonl( seqid );
    return Send(p->pkg,len);
}
void CCngp::OnSocketError()
{
    int ret;
    EnterCriticalSection( &m_csSocket);
    ret=Connect();
    if(ret==0)
    {
        char szbuff[64];
        char authsrc[24]={0}, szTimeStamp[12]={0};
	    int  nsize = sizeof( CNGP_HEAD) + sizeof( CNGP_LOGIN);
	    ((CNGP_HEAD * )szbuff)->size  = htonl( nsize);
        ((CNGP_HEAD * )szbuff)->cmdid = htonl( nCNGP_LOGIN);
        ((CNGP_HEAD * )szbuff)->seqid = htonl(GetSeqid(0));
        GetTimeStamp(szTimeStamp);
        strcpy(((CNGP_LOGIN*)(szbuff+sizeof( CNGP_HEAD)))->clientID, m_szUser);
        MD5 ctx;
        ctx.update((unsigned char *)m_szUser, strlen(m_szUser));
        ctx.update((unsigned char *)"\0\0\0\0\0\0\0", 7);
        ctx.update((unsigned char *)m_szPwd, strlen(m_szPwd));
        ctx.update((unsigned char *)szTimeStamp, 10);
        ctx.finalize();
        ctx.raw_digest((unsigned char *)((CNGP_LOGIN*)(szbuff+sizeof( CNGP_HEAD)))->authenticatorClient);
        ((CNGP_LOGIN*)(szbuff+sizeof( CNGP_HEAD)))->version=0x10;
        ((CNGP_LOGIN*)(szbuff+sizeof( CNGP_HEAD)))->loginMode=m_nLoginMode;
        ((CNGP_LOGIN*)(szbuff+sizeof( CNGP_HEAD)))->timeStamp=htonl(atol(szTimeStamp));
        ret=::send(m_Socket,szbuff, nsize,0 );
    }
    LeaveCriticalSection( &m_csSocket);
}
int  CCngp::Login()
{
    char szbuff[64];
    char authsrc[24]={0}, szTimeStamp[12]={0};
	int  nsize = sizeof( CNGP_HEAD) + sizeof( CNGP_LOGIN);
	((CNGP_HEAD * )szbuff)->size  = htonl( nsize);
    ((CNGP_HEAD * )szbuff)->cmdid = htonl( nCNGP_LOGIN);
    ((CNGP_HEAD * )szbuff)->seqid = htonl(GetSeqid(0));
    GetTimeStamp(szTimeStamp);
    strcpy(((CNGP_LOGIN*)(szbuff+sizeof( CNGP_HEAD)))->clientID, m_szUser);
    MD5 ctx;
    ctx.update((unsigned char *)m_szUser, strlen(m_szUser));
    ctx.update((unsigned char *)"\0\0\0\0\0\0\0", 7);
    ctx.update((unsigned char *)m_szPwd, strlen(m_szPwd));
    ctx.update((unsigned char *)szTimeStamp, 10);
    ctx.finalize();
    ctx.raw_digest((unsigned char *)((CNGP_LOGIN*)(szbuff+sizeof( CNGP_HEAD)))->authenticatorClient);
    ((CNGP_LOGIN*)(szbuff+sizeof( CNGP_HEAD)))->version=0x10;
    ((CNGP_LOGIN*)(szbuff+sizeof( CNGP_HEAD)))->loginMode=m_nLoginMode;
    ((CNGP_LOGIN*)(szbuff+sizeof( CNGP_HEAD)))->timeStamp=htonl(atol(szTimeStamp));
    int ret=Send(szbuff, nsize);
    if(ret== nsize)
    {   
        nsize= sizeof( CNGP_HEAD) + sizeof(CNGP_LOGIN_RESP);
        ret=recv(m_Socket,szbuff,nsize,0);
        if(ret==nsize)
        {
            return ntohl( ((CNGP_HEAD * )szbuff)->command_status); 
        }
        else
        {
            return WSAGetLastError();
        }
    }
    return WSAGetLastError();
}
int  CCngp::Logout()
{
    CNGP_HEAD head;
    head.size=htonl(sizeof(CNGP_HEAD));
    head.cmdid=htonl(nCNGP_EXIT);
    head.seqid=htonl(GetSeqid(0));
    int ret=Send((char *)&head,sizeof(CNGP_HEAD));
    return 0;
}
int CCngp::Startup(char *szIp,int port, int login_mode,
            char *szUser,char *szPwd,
            char *szCorpId, char *szSeverNumber,char *szSpNode,
            int  threadpool,int wndsize,
            int  timeout,int trys,
            void (*OnDeliverSMS)(DELIVER_SMS   *pDeliver),
            void (*OnMTReportSMS)(MTREPORT_SMS *pReport),
            void (*OnSubmitedSMS)(SUBMIT_SMS   *pSubmited))
{
    m_OnDeliverSMS=OnDeliverSMS;
    m_OnMTReportSMS=OnMTReportSMS;
    m_OnSubmitedSMS=OnSubmitedSMS;
    strcpy(m_szIp, szIp);
    m_nPort=port ;
    m_nLoginMode= login_mode;
    strcpy(m_szUser,szUser) ;
    strcpy(m_szPwd,szPwd) ;
    if(szCorpId)      strcpy(m_szCorpId, szCorpId);
    if(szSeverNumber) strcpy(m_szSeverNumber,szSeverNumber);
    if(szSpNode)      strcpy(m_szSpNode,szSpNode);
    int ret=Connect();
    if(ret==0)
    {
        ret=Login();
    }
    Initialize(threadpool,wndsize,timeout,trys);
    return ret;
}
void CCngp::OnActive()
{
    int timeActive=0;
    CNGP_HEAD  head;
    while(!m_bQuit)
    {
        Sleep(1000);
        if(timeActive++>= 60 )
        {
            head.cmdid= htonl(nCNGP_ACTIVE_TEST);
            head.seqid= htonl(GetSeqid(0));
            head.size= htonl(sizeof(CNGP_HEAD));
            Send((char *)&head,sizeof(CNGP_HEAD));
            timeActive=0;
        }
    }
}