socket.cpp

SMS gateway. SMS protocol for CHINA mobile, unicom, lingtong. Using mysql to exchange message.

/**********************************************************************
  FileName            : socket.cpp
  Description         : 套接字处理函数库
  Version             : 1.0
  Date                : 2003年6月13日
  Author              : 刘荣辉
  Other               : 
***********************************************************************/
#include "../include/socket.h"

//=======================================================
//=======================================================
TCPsocket::TCPsocket()
{  
	TimeOut = 100;	//默认100秒超时
}

TCPsocket::TCPsocket(int nTimeOut)
{  
	TimeOut = nTimeOut;
}

TCPsocket::~TCPsocket()
{  
 if (sock >= 0)
       close(sock);
}

void TCPsocket::SetTimeOut(int nTimeOut)
{  
	TimeOut = nTimeOut;
}

int TCPsocket::Close_sock()
{
  int retcode=0;
  if(sock >= 0)
     retcode = close(sock);
  return retcode;
}
//=======================================================
//=======================================================
//int Make_Listen_Sock(const int port, const char *ServerIp)  //生成监听本机某端口的Socket
int TCPsocket::Make_Listen_Sock(const char *ServerIp,const int port)
{
  //int sock;
  int reuse;
  struct sockaddr_in name;
  
  sock=0;  
  bzero(&name,sizeof(name));   
  name.sin_family=AF_INET;
  name.sin_port=htons(port);
  
  if(strlen(ServerIp)>6 && strlen(ServerIp)<16)
      name.sin_addr.s_addr=inet_addr(ServerIp);
  else
      name.sin_addr.s_addr=htonl(INADDR_ANY);/*本机的所有合法IP*/ 
  
  sock=socket(AF_INET,SOCK_STREAM,0);
  if(sock<0) 
     {
      printf("[Make_Listen_Sock]: Socket create failed!");
      return (-1);
     } 
  reuse = 1;
  if (setsockopt(sock,SOL_SOCKET,SO_REUSEADDR,(char *)&reuse,sizeof(reuse)) < 0)     
      {
       printf("[Make_Listen_Sock]: TCP setsockopt failed\n");
       if (sock >= 0)
          close(sock);
       return (-2);
      } 
 struct timeval RTimeOut;
 RTimeOut.tv_sec=TimeOut;	//超时秒数(连接会被自动关闭)
 RTimeOut.tv_usec=0;	//微秒数
 if(setsockopt(sock,SOL_SOCKET,SO_RCVTIMEO,&RTimeOut,sizeof(RTimeOut))!=0)
 {
	  printf("[Make_Connect_Sock]: setsockopt failed!\n");
      if (sock >= 0)
	       close(sock);
	  return (-7);
 }
  if(bind(sock,(struct sockaddr *)&name,sizeof(name))<0)
      {
      	printf("[Make_Listen_Sock]: Socket bind failed!");
        if (sock >= 0)
           close(sock);
      	return (-3);
      } 
  if (listen(sock, 10) == -1) 
     {
        printf("[Make_Listen_Sock]: Socket listen failed");
        if (sock >= 0)
           close(sock);
      	return (-4);
     }    
  return sock;  	
}

//=======================================================
//=======================================================
//int TCPsocket::Make_Connect_Sock(char *ServerIp,int port)  //生成连接某服务器某端口的Socket
int TCPsocket::Make_Connect_Sock(const char *ServerIp,const int port)
{
 //int sock;
 struct sockaddr_in servaddr;
 sock=0;
 
 bzero((char*)&servaddr,sizeof(servaddr));
 servaddr.sin_family=AF_INET;
 servaddr.sin_addr.s_addr=inet_addr(ServerIp);
 servaddr.sin_port=htons(port);
 
 if((sock=socket(AF_INET,SOCK_STREAM,0))<0)
     {
      printf("[Make_Connect_Sock]: Socket create failed!");
      return (-5);
     } 
 struct timeval RTimeOut;
 RTimeOut.tv_sec=TimeOut;	//超时秒数(连接会被自动关闭)
 RTimeOut.tv_usec=0;	//微秒数
 if(setsockopt(sock,SOL_SOCKET,SO_RCVTIMEO,&RTimeOut,sizeof(RTimeOut))!=0)
 {
	  printf("[Make_Connect_Sock]: setsockopt failed!\n");
      if (sock >= 0)
	       close(sock);
	  return (-7);
 }
 if(connect(sock,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
     {
      printf("[Make_Connect_Sock]: connect '%s :%d' failed!\n",ServerIp,port);
      if (sock >= 0)
          close(sock);
      return (-6);
     } 
 return sock;       
}

//=======================================================
//=======================================================
/*[lib/readn.c][lib/writen.c]*/
/*从指定的socket中读取N字节的数据存到某指针指向的内存,被信号中断后自动重启*/
//int TCPsocket::Readn(int fd,void *vptr,int n)
int TCPsocket::Readn(void *vptr,const int n)
{
 int nleft;
 int nread;
 char *ptr;
 ptr=(char *)vptr;
 nleft=n;
 while(nleft>0)
    {
      if((nread=read(sock,ptr,nleft))<0)
        {
         if(errno==EINTR) nread=0;
         else return (-1);
        }
      else if(nread==0) break;
      nleft-=nread;
      ptr+=nread;
    }
 return(n-nleft);
}

//=======================================================
//=======================================================
/*将某指针开始的N字节的数据发送到指定的socket,被信号中断后自动重启*/
//int TCPsocket::Writen(int fd,const void *vptr,int n)
int TCPsocket::Writen(const void *vptr,const int n)
{
 int nleft;
 int nwriten;
 const char *ptr;
 ptr=(char *)vptr;
 nleft=n;
 while(nleft>0)
   {
    if((nwriten=write(sock,ptr,nleft))<=0)
      {
       if(errno==EINTR) nwriten=0;
       else return(-1);
      }
    nleft-=nwriten;
    ptr+=nwriten;  
   } 
 return(n);  
}

//=======================================================
//=======================================================
//接收一个完整包,入口参数Packet为包头指针的指针,MaxSize为允许接收的最大包长度
int TCPsocket::RecvPack(void **Packet,const int MaxSize)
{
  	char PackHead[HEAD_LEN]; 
  	char *Pack;
  	unsigned int iGet,iWant;
  	int PackLen;
  	
	iGet = Readn(PackHead,HEAD_LEN);	//首先读出包头
    	if(iGet!=HEAD_LEN)
    	{
    		Close_sock();
    		return -1;
    	}
        PackLen = ntohl( *(int *)PackHead );   
        //printf("\n [Socket.cpp]: PackLen=%d,HEAD_LEN=%d,\n",PackLen,HEAD_LEN);
        if((PackLen>MaxSize)||(PackLen<HEAD_LEN)) //包长度异常
        {
        	Close_sock();
    		return -2;
        }
        //动态为所收数据包分配内存，包被处理后内存必须被释放！！！！！！
		if(PackLen>30)
    		Pack = (char *)malloc(PackLen+141);	//为收到的包增加空间以便编码转换
		else
			Pack = (char *)malloc(PackLen);
    	memset(Pack,0,PackLen);
    	memcpy(Pack,PackHead,HEAD_LEN);
    	
    	if(PackLen>HEAD_LEN)	//继续接收包的剩余部分
    	{
    		iWant = PackLen-HEAD_LEN;
    		iGet = Readn(&Pack[HEAD_LEN],iWant);
    		if(iGet!=iWant)
    		{
    			Close_sock();
    			return -1;
    		}
    	}
    *Packet = Pack;
    
  /*/-----------------debug----------------------
  printf("\n===========checking Pack============= \n");
  unsigned char *ptmp;
  ptmp=(unsigned char *)Pack;
  int len;
  len = ntohl(*(int *)Pack);
  printf("PackLen=[%d]\n",len);
  for(int i=0;i<len;i++)
  {
     printf("%02x ",*(ptmp+i));
     if(((i+1)%4)==0) printf("  ");
     if((i+1)==12) printf("\n");
  }
  printf("\n=========================================== \n");
  for(int i=0;i<len;i++)
  {   
     printf("%d ",*(ptmp+i));
     if(((i+1)%4)==0) printf("  ");
     if((i+1)==12) printf("\n");
  }
  printf("\n=========================================== \n");
  //--------------------debug end-------------------*/
    
    return 0;
}

//=======================================================
//=======================================================
//接收一个完整包,入口参数Packet指向一片预先分配的内存区,MaxSize为允许接收的最大包长度
int TCPsocket::RecvPack2(void *Packet, const int MaxSize)
{	
  	char *Pack;
  	unsigned int iGet,iWant;
  	int PackLen;
  	
  	Pack = (char *)Packet;
	iGet = Readn(Pack,HEAD_LEN);	//首先读出包头
    if(iGet!=HEAD_LEN)
    {
    	Close_sock();
    	printf("\n[socket.cpp]: connection lost 1!\n");
    	return -1;
    }
    PackLen = ntohl( *(int *)Pack );

    if((PackLen>MaxSize)||(PackLen<HEAD_LEN)) //包长度异常
    {
    	Close_sock();
    	printf("\n[socket.cpp]: Abnormal PackLen=%d,MaxSize=%d,HEAD_LEN=%d,\n",PackLen,MaxSize,HEAD_LEN);
    	return -2;
    }
    //printf("\n[socket.cpp]: go on to receive...\n");
    if(PackLen>HEAD_LEN)	//继续接收包的剩余部分
    {
    	iWant = PackLen-HEAD_LEN;
    	iGet = Readn(&Pack[HEAD_LEN],iWant);
    	if(iGet!=iWant)
    	{
    		Close_sock();
    		printf("\n[socket.cpp]: connection lost 2!\n");
    		return -1;
    	}
    }
    
    return 0;
}