smppsockold.cpp

移动

// SMPP Sock.cpp : implementation file
#include "stdafx.h"
#include "SmppSock.h"
#include "mbstring.h"
#include "Global.h"

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

int  fGetSockErr(char errText[MAX_ERRTEXT])
{
	ASSERT(errText!=NULL);
	strcpy(errText,"\0");
	int iRet=WSAGetLastError();

	LPVOID lpMsgBuf;                                                                  
	FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS,  
				NULL,iRet,MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR) &lpMsgBuf,0,NULL); 
	sprintf(errText,"SOCKET错误:%s",(LPCSTR)lpMsgBuf); //errText[strlen(errText)-2]=0;
	LocalFree(lpMsgBuf );
	return iRet;
}

void CSmppSock::network_head(struct Smpp_MESSAGE_HEAD* phead)
{
	phead->command_length=network_dword(phead->command_length);
	phead->command_id=network_dword(phead->command_id);
	phead->command_status=network_dword(phead->command_status);
	phead->sequence_no=network_dword(phead->sequence_no);
}
int CSmppSock::fill_message(BYTE* pbody,const char* msg)
{
	BYTE *p=pbody;
	BYTE len=strlen(msg);
	if(len>140) len=140;
	wchar_t bb[300]; memset(bb,0,sizeof(bb));
	MultiByteToWideChar(CP_ACP,0,(const char*)msg,len,bb,sizeof(bb));
	len=wcslen(bb)*2;
	*p++=len;		//sm_length
	memcpy(p,bb,len);
	BYTE temp;
	for(int i=0;i<len/2;i++)
	{
		temp=p[i*2];p[i*2]=p[i*2+1];p[i*2+1]=temp;
	}
	return len+1;

	/*BYTE *p=pbody;
	BSTR bb=msg.AllocSysString(); 
	BYTE len=wcslen(bb)*2;
	if(len>200)	len=200;
	*p++=len;		//sm_length
	memcpy(p,bb,len);
	::SysFreeString(bb); 
	BYTE temp;
	for(int i=0;i<len/2;i++)
	{
		temp=p[i*2];p[i*2]=p[i*2+1];p[i*2+1]=temp;
	}
	return len+1;*/
}

/////////////////////////////////////////////////////////////////////////////
CSmppSock::CSmppSock()
{
	m_sequence=1;
	//m_defaultPeroid=10;
	m_bConnected=m_bAlive=FALSE;

	strcpy(m_smscInfo.SMSCIP,"10.71.13.17");	//hz add
	m_smscInfo.uPort=5018;						//hz add
	strcpy(m_smscInfo.usrName,"roam");			//hz add
	strcpy(m_smscInfo.passWd,"8888");			//hz add

/*	strcpy(m_smscInfo.SMSCIP,SMEConfig.host);
	m_smscInfo.uPort=SMEConfig.port;
	strcpy(m_smscInfo.usrName,SMEConfig.user);
	strcpy(m_smscInfo.passWd,SMEConfig.pwd);
//	strcpy(SMEConfig.province,pApp->GetProfileString("SME","province"));
//	strcpy(SMEConfig.src_addr,pApp->GetProfileString("SME","src_addr"));
*/
	m_hSocket=INVALID_SOCKET;
//	m_hEventMainOver=CreateEvent(NULL,TRUE,FALSE,"exit");  //nosignle
}
int CSmppSock::CreateSocket()
{
	m_hSocket=socket(AF_INET,SOCK_STREAM,0);
	if(m_hSocket==INVALID_SOCKET)
	{
		Smpp_PRINTOUT("FAILED:socket create failed.....");
		//nErrorCode_=WSAGetLastError();
		return 0;
	}
	SOCKADDR_IN  sockaddr;
	memset(&sockaddr,0,sizeof(sockaddr));
	sockaddr.sin_family=AF_INET;
	sockaddr.sin_addr.s_addr=htonl(INADDR_ANY);
	sockaddr.sin_port=htons(0);
	if(bind(m_hSocket,(LPSOCKADDR)&sockaddr,sizeof(sockaddr)) == SOCKET_ERROR)
	{
		closesocket(m_hSocket);
		Smpp_PRINTOUT("FAILED:socket bind failed,close socket.....");
		//int nErrorCode_=WSAGetLastError();
		m_hSocket=INVALID_SOCKET;
		return 0;
	}
	m_sequence=1;
	return 1;
}

CSmppSock::~CSmppSock()
{	
	fLogout();
	fDisconnect();
//	if(m_hEventMainOver) CloseHandle(m_hEventMainOver);
}
// Do not edit the following lines, which are needed by ClassWizard.
#if 0
BEGIN_MESSAGE_MAP(CSmppSock, CSocket)
	//{{AFX_MSG_MAP(CSmppSock)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()
#endif	// 0

/////////////////////////////////////////////////////////////////////////////
// CSmppSock member functions
/**************************************************************************************
return 
0 : already have login
-1: usrname is null
-2: no socket connected
-4: receive response timeout
-5: login be rejected
>0: success
**************************************************************************************/
int CSmppSock::fLogin(char *usrname, char *passwd)
{
	ClearBuffer();
	if(m_bAlive)     {Smpp_PRINTOUT("已登录."); return 0;}
	if(!m_bConnected){Smpp_PRINTOUT("FAILED:socket连接尚未建立."); return -2;}
	if(!usrname)     {Smpp_PRINTOUT("FAILED:用户名为空.") ; return -1;}	

	strcpy(m_smscInfo.usrName,usrname); 
	strcpy(m_smscInfo.passWd,passwd);

	Smpp_PRINTOUT("Logining.....");

	BYTE buffer[MAX_SMPP_COM_BUF];
	memset(buffer,0,MAX_SMPP_COM_BUF);
	struct Smpp_MESSAGE_HEAD *phead=(struct Smpp_MESSAGE_HEAD *)buffer;
	BYTE *p=buffer+sizeof(struct Smpp_MESSAGE_HEAD);

	strcpy((char*)p,usrname);	p+=strlen(usrname)+1;	//pbind->system_id
	strcpy((char*)p,passwd);	p+=strlen(passwd)+1;	//pbind->password
	*p++=0;				//pbind->system_type
	*p++=2;				//pbind->interface_version=1;
	*p++=0;				//pbind->addr_ton=0;
	*p++=0;				//pbind->addr_npi=0;
	*p++=0;				//pbind->address_range
	
	phead->command_id=ESME_BNDTRN;
	phead->command_status=0;
	UINT packet_len=p-buffer;
	phead->command_length=packet_len;
	phead->sequence_no=m_sequence++;
	UINT login_sequence=phead->sequence_no;
	network_head(phead);
	//if(Send(buffer,packet_len)==SOCKET_ERROR)
	if(send(m_hSocket,(const char*)buffer,packet_len,0)==SOCKET_ERROR)
	{
		if(send(m_hSocket,(const char*)buffer,packet_len,0)==SOCKET_ERROR)//重发
		{
			Smpp_PRINTOUT("FAILED:Login ,send packet failed.Disconnect..."); 
			fDisconnect();
			return -0xff;
		}
	}
	Sleep(1000);
	
	//receive resp	
	DWORD dRevLen;
	UINT starttime=time(NULL);	
	//MSG msg;
	while(1) //GetMessage(&msg, NULL, 0, 0) ) 
	{
		//TranslateMessage(&msg);		DispatchMessage(&msg);
		//IOCtl(FIONREAD,&dRevLen);	//get the socket on receive data length
		ioctlsocket(m_hSocket,FIONREAD,&dRevLen);
	    if(dRevLen>=sizeof(Smpp_MESSAGE_HEAD))
		{
			memset(buffer,0,MAX_SMPP_COM_BUF);
			dRevLen=fReceive(buffer,sizeof(MAX_SMPP_COM_BUF));
			if(login_sequence==phead->sequence_no && phead->command_id==ESME_BNDTRN_RESP)
			{
				if(((Smpp_MESSAGE_HEAD*)buffer)->command_status==E_SUCCESS)
				{
					m_bAlive=TRUE; Smpp_PRINTOUT("Login +OK.");
					return 1;
				}
				else
				{
					Smpp_PRINTOUT("FAILED: Login +FAILED.");
					return -5;
				}
			}
		}
		Sleep(100);
//		if(WaitForSingleObject(m_hEventMainOver,0)==WAIT_OBJECT_0) {break;}
		if(time(NULL)-starttime>MAX_LOGINTIME)    //5--10s  
		{
			Smpp_PRINTOUT("FAILED:LOGIN receive timeout");	
			fDisconnect();
			return -4;		
		}
	}	
	return -4;
}
/*****************************************************************************************
return 
0 : no login or socket connect
-1: destadd or sendmsg is null
-2: sendmsg error
-3: sendpack error
-4: receive response timeout
-5: 
-6: sendsm resperror
1: success

0xff  : no send sm alarm
*****************************************************************************************/
int CSmppSock::fSendSM(const char* destaddr,const char* srcaddr, const char* sendmsg,bool bResp)
{
	//Smpp_PRINTOUT((CString)destaddr + "  Submit +OK.\n");
	//return 1;

	if(!m_bConnected)
	{
		if(fConnect(m_smscInfo.SMSCIP,m_smscInfo.uPort)<1)	
		{  
			Smpp_PRINTOUT("FAILED:Submit , connect retrying");
			return 0xFF;
		}
	}
	if(!m_bAlive)
	{
		if(fLogin(m_smscInfo.usrName,m_smscInfo.passWd)<1)
		{
			Smpp_PRINTOUT("FAILED: Submit , login retrying");
			fDisconnect();
			Sleep(3000);
			return -2;
		}		 
	}
	if(destaddr[0]==0 || sendmsg[0]==0 )  
	{
		Smpp_PRINTOUT("FAILED: Submit , destadd or sendmsg is NULL") ;	
		return -1;
	}

	BYTE buffer[MAX_SMPP_COM_BUF];
	struct Smpp_MESSAGE_HEAD *phead=(struct Smpp_MESSAGE_HEAD *)buffer;
	BYTE* p=buffer+sizeof(struct Smpp_MESSAGE_HEAD);
	memset(buffer,0,MAX_SMPP_COM_BUF);
	
	*p++='0';	*p++=0;	//service_type
	*p++=1;		//pbody->source_address_ton=1;
	*p++=1;		//pbody->source_address_npi=1;
	strcpy((char*)p,srcaddr);	p+=strlen(srcaddr)+1;	//source_address
	*p++=1;		//pbody->dest_address_ton=1;
	*p++=1;		//pbody->dest_address_npi=1;
	strcpy((char*)p,destaddr);	p+=strlen(destaddr)+1;	//dest_address
	*p++=0;		//pbody->esm_class=0;
	*p++=0;		//pbody->protocol_ID=0;
	*p++=0;		//pbody->priority_flag=0; 
	*p++=0;		//schedule_delivery_time
	if(m_life.bUsed==1)///生存期限
		p+=FillTimeStrToStr((char *)p);
	else
		*p++=0;		//validity_peroid
	*p++=0;		//registered_delivery_flag
	*p++=0;		//replace_if_presend_flag
	if(m_bFlashMsg==1)
		*p++=24;		//快闪短信
	else
		*p++=8;		//data_codeing
	*p++=0;		//sme_default_msgid
	p+=fill_message(p,sendmsg);		//length and content
	
	phead->command_id=ESME_SUB_SM;
	phead->command_status=0;
	UINT packet_len=p-buffer;
	phead->command_length=packet_len;
	phead->sequence_no=m_sequence++;
	UINT send_sequence=phead->sequence_no; 

	network_head(phead);
	//if(Send(buffer,packet_len)==SOCKET_ERROR)
	if(send(m_hSocket,(const char*)buffer,packet_len,0)==SOCKET_ERROR)
	{
		if(send(m_hSocket,(const char*)buffer,packet_len,0)==SOCKET_ERROR)//重发
		{
			Smpp_PRINTOUT("FAILED: Send Short Message ,send packet failed.disconnect..."); 
			fDisconnect();
			return -0xff;
		}
	}
//	Sleep(50);
	if(!bResp)	return 1;
	FD_SET sockset = {1, m_hSocket};
	TIMEVAL tv = {10, 0};
    int result=select(0, &sockset, NULL, NULL, &tv);
	
	DWORD dRevLen;
//	UINT starttime=time(NULL);
	//MSG msg;
	if(result>0)//while(1)// GetMessage(&msg, NULL, 0, 0) ) 
	{
		//TranslateMessage(&msg);
		//DispatchMessage(&msg);
		//IOCtl(FIONREAD,&dRevLen); 
		if ( FD_ISSET( m_hSocket, &sockset))
		{
#if 0
			while(1){
				dRevLen=recv(m_hSocket,(char *)buffer,MAX_SMPP_COM_BUF,0);
				break;
				if(dRevLen<=0)
					break;
				result=select(0, &sockset, NULL, NULL, &tv);
				if(result>0)
				{
					if ( FD_ISSET( m_hSocket, &sockset))
						continue;
					else
						break;
				}
				else
					break;
			}
#else
			
			ioctlsocket(m_hSocket,FIONREAD,&dRevLen);
			if(dRevLen>=sizeof(Smpp_MESSAGE_HEAD))
			{
				memset(buffer,0,MAX_SMPP_COM_BUF);
				dRevLen=fReceive(buffer,sizeof(MAX_SMPP_COM_BUF));
//				if(send_sequence==phead->sequence_no && phead->command_id==ESME_SUB_SM_RESP)
				if(send_sequence==phead->sequence_no &&phead->command_id==ESME_SUB_SM_RESP)
				{
					if(phead->command_status==E_SUCCESS)
					{
						Smpp_PRINTOUT((CString)destaddr + "  Submit +OK.\n");
//						return 1;
					}
					else
					{
						Smpp_PRINTOUT("FAILED: Submit.\n");
//						fLogout(FALSE);
//						fDisconnect();
//						return -2; 
					}
					return phead->command_status;
				}
			}
			
#endif
		}
	}	
	else if(result<0)
	{
//		Sleep(50);
//		if(WaitForSingleObject(m_hEventMainOver,0)==WAIT_OBJECT_0) {break;}
//		if(time(NULL)-starttime>MAX_SUBTIME)    //5--10s  
		{
			Smpp_PRINTOUT("FAILED: SendSM receive timeout");	
			fLogout(FALSE);
			fDisconnect();
			Sleep(3000);
//			return -4;		
		}
	}
	FD_CLR(m_hSocket,&sockset);
//	return 1;
	return -4;
}


/*******************************************************************************
return 
0 : no login or socket connect
-1: send buf failed
-4: receive response timeout
-5: logout be rejected
>0: success
*******************************************************************************/
int CSmppSock::fLogout(bool bResp)
{
	Smpp_PRINTOUT("Logouting.....") ;
	if(!m_bAlive) {Smpp_PRINTOUT("FAILED: logout,but no login."); return 0;}
	
	Smpp_MESSAGE_HEAD head;
	UINT packet_len=sizeof(Smpp_MESSAGE_HEAD);
	head.command_length=packet_len;
	head.command_id=ESME_UBD;
	head.command_status=0;
	head.sequence_no=m_sequence++;
	UINT logout_sequence=head.sequence_no;
	network_head(&head);