socket发送和接收固定长度数据的函数

#include "stdafx.h"
#include "ATLComTime.h"
#include "Functions.h"
#include   <rpc.h> 
#pragma   comment(lib,   "rpcrt4.lib ") 


CRITICAL_SECTION _criticalLog;


extern string _ExePath;


//接收數據包
int SocketRecvFunc(SOCKET s, char *buf, int len, int timeout)
{
fd_set fdread;
int iRecvlen, iRead, iRecError;
int iResult=0;
LONG lRecvSize = len;
timeval time={timeout,0};//超时时长5秒 

while( s != INVALID_SOCKET )
{
FD_ZERO(&fdread);
FD_SET(s,&fdread);


iRead = select( 0, &fdread, NULL, NULL, &time );


if( iRead == 0 )//接收超时
break;


if( iRead == SOCKET_ERROR )
{
iResult = iRead;
break;
}


if( iRead > 0 )
{
if( FD_ISSET(s, &fdread) )
{
iRecvlen = recv( s, buf+(len-lRecvSize), lRecvSize, NULL );


if( iRecvlen == 0 )
{
iResult = iRecvlen;
break;
}


if( iRecvlen == SOCKET_ERROR )
{
iRecError = WSAGetLastError();


if( (iRecError == WSAENOTCONN)//(WSABASEERR+57)
|| (iRecError == WSAESHUTDOWN)//(WSABASEERR+58)
|| (iRecError == WSAENOTSOCK)
|| (iRecError == WSAENETDOWN)
|| (iRecError == WSAECONNABORTED)//(WSABASEERR+53)
|| (iRecError == WSAETIMEDOUT)//(WSABASEERR+60)
|| (iRecError == WSAECONNRESET) )//(WSABASEERR+54)
{
iResult = iRecvlen;
break;
}
}


lRecvSize -= iRecvlen;
iResult += iRecvlen;


//接收到完整数据包后再处理（因为每次recv的数据长度不一定等于发送包的长度）
if (lRecvSize <= 0)
break;
}
}
}


return iResult;
}


//发送数据包
int SocketSendFunc(SOCKET s, const char *buf, int len)
{
fd_set fdwrite;
int iSendlen;
int lSendSize=0;
int iResult=0;
int iWrite;
timeval timeout={2,0};//超时时长


do
{
FD_ZERO(&fdwrite);
FD_SET(s,&fdwrite);

iWrite = select(0, NULL, &fdwrite, NULL, &timeout);

if (iWrite == 0)
{
iResult = iWrite;
break;
}
if (iWrite == SOCKET_ERROR)
{
iResult = iWrite;
break;
}
if (iWrite > 0)
if (FD_ISSET(s,&fdwrite))
{
iSendlen = send(s, buf+lSendSize, len-lSendSize, NULL);
if (iSendlen == SOCKET_ERROR)
{
iResult = iSendlen;
break;
}
else
lSendSize += iSendlen;


iResult = lSendSize;
}
}while((lSendSize < len) && (s != INVALID_SOCKET));//判断是否发送完毕


return iResult;
}


int SocketSendFunc(SOCKET s, const char *buf, int len, bool* blockflag)
{
fd_set fdwrite;
int iSendlen;
int lSendSize=0;
int iResult=0;
int iWrite;
timeval timeout={2,0};//超时时长


do
{
FD_ZERO(&fdwrite);
FD_SET(s,&fdwrite);

iWrite = select(0, NULL, &fdwrite, NULL, &timeout);

if (iWrite == 0)
{
if (*blockflag)
continue;


iResult = iWrite;
break;
}


if (iWrite == SOCKET_ERROR)
{
iResult = iWrite;
break;
}
if (iWrite > 0)
if (FD_ISSET(s,&fdwrite))
{
iSendlen = send(s, buf+lSendSize, len-lSendSize, NULL);
if (iSendlen == SOCKET_ERROR)
{
iResult = iSendlen;
break;
}
else
lSendSize += iSendlen;


iResult = lSendSize;
}
}while((lSendSize < len) && (s != INVALID_SOCKET));//判断是否发送完毕


return iResult;
}


UINT ConnectToServer(char* szServer, WORD nPort)
{
SOCKET sSocket;
struct sockaddr_in serveraddr;
FD_SET mask;
u_long value=1;
TIMEVAL timeout;


sSocket = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
if (sSocket == INVALID_SOCKET)
{
sSocket = 0;
}else
{
serveraddr.sin_family = AF_INET;
serveraddr.sin_port = htons(nPort);
serveraddr.sin_addr.s_addr = inet_addr(szServer);

ioctlsocket(sSocket,FIONBIO,&value);//设置为非阻塞
connect(sSocket,(struct sockaddr *)&serveraddr,sizeof(serveraddr));
timeout.tv_sec=2;
timeout.tv_usec=0;
FD_ZERO(&mask);
FD_SET(sSocket,&mask);


value=select(NULL,NULL,&mask,NULL,&timeout);
if(value && value!=SOCKET_ERROR)
{ 
//连接成功
value = 0;
ioctlsocket(sSocket,FIONBIO,&value);//设置为阻塞
}
else
{
shutdown(sSocket,SD_BOTH);
closesocket(sSocket);
sSocket = 0;
}
}


return sSocket;
}


void WriteLogFile(char* szLog)
{
COleDateTime t(COleDateTime::GetCurrentTime());
FILE *file=NULL;
errno_t err;
char szTxt[128]={};


LockCS lock( &_criticalLog );
sprintf_s( szTxt, sizeof(szTxt), "%d-%d-%d %d:%d:%d  ", t.GetYear(), t.GetMonth(), t.GetDay(), t.GetHour(), t.GetMinute(), t.GetSecond() );


if( 0 == (err  = fopen_s( &file, "MRVideoServer.log", "a+" )))
{
if( file != NULL )
{
fprintf_s( file, "%s", szTxt );
fflush(file);
fprintf_s( file, "%s", szLog );
fflush(file);
fprintf_s( file, "\r\n" );
fflush(file);


fclose(file);
}
}
}


void WriteLogFileEx(char* szLog,char* szPath)
{
COleDateTime t(COleDateTime::GetCurrentTime());
FILE *file=NULL;
errno_t err;
char szTxt[128]={};


LockCS lock( &_criticalLog );
sprintf_s( szTxt, sizeof(szTxt), "%d-%d-%d %d:%d:%d  ", t.GetYear(), t.GetMonth(), t.GetDay(), t.GetHour(), t.GetMinute(), t.GetSecond() );


if( 0 == (err  = fopen_s( &file, szPath, "a+" )))
{
if( file != NULL )
{
fprintf_s( file, "%s", szTxt );
fflush(file);
fprintf_s( file, "%s", szLog );
fflush(file);
fprintf_s( file, "\r\n" );
fflush(file);


fclose(file);
}
}
}


int MakeVideoID(short ConnectID,short SubID)
{
int iResult = 0;


if ((0 > ConnectID ) || (0 > SubID))
iResult = -1;
else
iResult = (ConnectID << 16) | (SubID);


return iResult;
}


void AnalysisID(int iID,short* pConnectID,short* pSubID)
{
if (0 > iID)
{
*pConnectID = -1;
*pSubID = -1;
}else
{
*pConnectID = (iID >> 16);
*pSubID = (short)iID;
}
}


void DwordIPToChar(DWORD dwIP,char* szIP)
{
char* szTempIP = NULL;
struct sockaddr_in addr;
addr.sin_family = AF_INET;
addr.sin_addr.S_un.S_addr = dwIP;
szTempIP = inet_ntoa(addr.sin_addr);
strcpy_s(szIP,128,szTempIP);
}


DWORD CharIPToDWORD(const char* szIP)
{
return inet_addr(szIP);
}


bool compare(const GUID &guidl, const GUID &guidr)
{
char *pszGuid_l = NULL;
char *pszGuid_r = NULL;
string guid_l = "";
string guid_r = "";


UuidToStringA(&guidl,(RPC_CSTR*)&pszGuid_l);
UuidToStringA(&guidr,(RPC_CSTR*)&pszGuid_r);
guid_l = pszGuid_l;
guid_r = pszGuid_r;


if(guid_l.compare(guid_r) < 0)
return true;
else
return false;
}


bool MyIsEqualGuid(const GUID &guid1, const GUID &guid2)
{
return ((guid1.Data1 == guid2.Data1) && (guid1.Data2 == guid2.Data2));
}


bool IsFileExists(const char* lpszFileName)   
{   
WIN32_FIND_DATAA wfd;   
bool bRet;   
HANDLE hFind;   
hFind = FindFirstFileA(lpszFileName,   &wfd);   
bRet  = (hFind != INVALID_HANDLE_VALUE);   
FindClose(hFind);   
return   bRet;   
}