/**********************************************************************  FileName            : socket.cpp  Description         : 套接字處理函數(shù)庫  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)  //生成監(jiān)聽本機某端口的Socketint 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;	//超時秒數(shù)(連接會被自動關(guān)閉) RTimeOut.tv_usec=0;	//微秒數(shù) 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)  //生成連接某服務(wù)器某端口的Socketint 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;	//超時秒數(shù)(連接會被自動關(guān)閉) RTimeOut.tv_usec=0;	//微秒數(shù) 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字節(jié)的數(shù)據(jù)存到某指針指向的內(nèi)存,被信號中斷后自動重啟*///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字節(jié)的數(shù)據(jù)發(fā)送到指定的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);  }//=======================================================//=======================================================//接收一個完整包,入口參數(shù)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;        }        //動態(tài)為所收數(shù)據(jù)包分配內(nèi)存，包被處理后內(nèi)存必須被釋放！！！！！！		if(PackLen>30)    		Pack = (char *)malloc(PackLen+141);	//為收到的包增加空間以便編碼轉(zhuǎn)換		else			Pack = (char *)malloc(PackLen);    	memset(Pack,0,PackLen);    	memcpy(Pack,PackHead,HEAD_LEN);    	    	if(PackLen>HEAD_LEN)	//繼續(xù)接收包的剩余部分    	{    		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;}//=======================================================//=======================================================//接收一個完整包,入口參數(shù)Packet指向一片預(yù)先分配的內(nèi)存區(qū),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)	//繼續(xù)接收包的剩余部分    {    	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;}