VC+CSocket文件传送示例

服务器端示例代码

#define PRE_AGREED_PORT 8686 //定义端口号
#define SEND_BUFFER_SIZE 4096 //buffer大小

BOOL CYourServerClass::SendFileToRemoteRecipient(CString fName)
{

AfxSocketInit(NULL);
CSocket sockSrvr;
sockSrvr.Create(PRE_AGREED_PORT);//创建一个Sokcet并监听
sockSrvr.Listen();//监听
CSocket sockConnection;
//用另一个Soket来收发数据
sockSrvr.Accept(sockConnection);
//接受从客户端发来的连接请求

BOOL bRet = TRUE;

int fileLength, cbLeftToSend;
BYTE* sendData = NULL;

CFile sourceFile;
CFileException fe;
BOOL bFileIsOpen = FALSE;

if( !( bFileIsOpen = sourceFile.Open( fName,
CFile::modeRead | CFile::typeBinary, &fe ) ) )
{
TCHAR strCause[256];
fe.GetErrorMessage( strCause, 255 );
TRACE( "SendFileToRemoteRecipient encountered
an error while opening the local file\n"
"\tFile name = %s\n\tCause = %s\n\tm_cause = %d\n\tm_IOsError = %d\n",
fe.m_strFileName, strCause, fe.m_cause, fe.m_lOsError );

//在这里处理错误

bRet = FALSE;
goto PreReturnCleanup;
}

//首先发送文件长度

fileLength = sourceFile.GetLength();
fileLength = htonl( fileLength );

cbLeftToSend = sizeof( fileLength );

do
{
int cbBytesSent;
BYTE* bp = (BYTE*)(&fileLength) + sizeof(fileLength) - cbLeftToSend;
cbBytesSent = sockConnection.Send( bp, cbLeftToSend );

// test for errors and get out if they occurred
if ( cbBytesSent == SOCKET_ERROR )
{
int iErr = ::GetLastError();
TRACE( "SendFileToRemoteRecipient returned a socket
error while sending file length\n"
"\tNumber of Bytes sent = %d\n"
"\tGetLastError = %d\n", cbBytesSent, iErr );

//在这里处理错误

bRet = FALSE;
goto PreReturnCleanup;
}

cbLeftToSend -= cbBytesSent;
}
while ( cbLeftToSend>0 );

//开始发送文件数据
sendData = new BYTE[SEND_BUFFER_SIZE];

cbLeftToSend = sourceFile.GetLength();

do
{

int sendThisTime, doneSoFar, buffOffset;

sendThisTime = sourceFile.Read( sendData, SEND_BUFFER_SIZE );
buffOffset = 0;

do
{
doneSoFar = sockConnection.Send( sendData + buffOffset,
sendThisTime );

//看执行是否成功
if ( doneSoFar == SOCKET_ERROR )
{
int iErr = ::GetLastError();
TRACE( "SendFileToRemoteRecipient returned a socket error
while sending chunked file data\n"
"\tNumber of Bytes sent = %d\n"
"\tGetLastError = %d\n", doneSoFar, iErr );

//处理错误

bRet = FALSE;
goto PreReturnCleanup;
}

buffOffset += doneSoFar;
sendThisTime -= doneSoFar;
cbLeftToSend -= doneSoFar;
}
while ( sendThisTime > 0 );

}
while ( cbLeftToSend > 0 );

PreReturnCleanup: // 出口点

delete[] sendData;

if ( bFileIsOpen )
sourceFile.Close();

sockConnection.Close();

return bRet;

}

客户端示例代码

#define PRE_AGREED_PORT 8686
#define RECV_BUFFER_SIZE 4096

BOOL CYourClientClass::GetFileFromRemoteSender(CString strIP,
CString fName)
{
AfxSocketInit(NULL);
CSocket sockClient;
sockClient.Create();
// PRE_AGREED_PORT 是 #define 8686
sockClient.Connect( strIP, PRE_AGREED_PORT );

//连接服务器

BOOL bRet = TRUE;

int dataLength, cbBytesRet, cbLeftToReceive;
BYTE* recdData = NULL;

CFile destFile;
CFileException fe;
BOOL bFileIsOpen = FALSE;

//打开或创建文件

if( !( bFileIsOpen = destFile.Open( fName, CFile::modeCreate |
CFile::modeWrite | CFile::typeBinary, &fe ) ) )
{
TCHAR strCause[256];
fe.GetErrorMessage( strCause, 255 );
TRACE( "GetFileFromRemoteSender encountered
an error while opening the local file\n"
"\tFile name = %s\n\tCause = %s\n\tm_cause = %d\n\tm_IOsError = %d\n",
fe.m_strFileName, strCause, fe.m_cause, fe.m_lOsError );

//在这里处理错误

bRet = FALSE;
goto PreReturnCleanup;
}

//得到文件的大小
cbLeftToReceive = sizeof( dataLength );

do
{
BYTE* bp = (BYTE*)(&dataLength) + sizeof(dataLength) - cbLeftToReceive;
cbBytesRet = sockClient.Receive( bp, cbLeftToReceive );

if ( cbBytesRet == SOCKET_ERROR || cbBytesRet == 0 )
{
int iErr = ::GetLastError();
TRACE( "GetFileFromRemoteSite returned
a socket error while getting file length\n"
"\tNumber of Bytes received (zero means connection was closed) = %d\n"
"\tGetLastError = %d\n", cbBytesRet, iErr );

//处理错误

bRet = FALSE;
goto PreReturnCleanup;
}

cbLeftToReceive -= cbBytesRet;

}
while ( cbLeftToReceive > 0 );

dataLength = ntohl( dataLength );

recdData = new byte[RECV_BUFFER_SIZE];
cbLeftToReceive = dataLength;

do
{
int iiGet, iiRecd;

iiGet = (cbLeftToReceive<RECV_BUFFER_SIZE) ?
cbLeftToReceive : RECV_BUFFER_SIZE ;
iiRecd = sockClient.Receive( recdData, iiGet );

if ( iiRecd == SOCKET_ERROR || iiRecd == 0 )
{
int iErr = ::GetLastError();
TRACE( "GetFileFromRemoteSite returned a socket error
while getting chunked file data\n"
"\tNumber of Bytes received (zero means connection was closed) = %d\n"
"\tGetLastError = %d\n", iiRecd, iErr );

//要在这里处理错误

bRet = FALSE;
goto PreReturnCleanup;
}

destFile.Write( recdData, iiRecd); // 写入数据
cbLeftToReceive -= iiRecd;

}
while ( cbLeftToReceive > 0 );

PreReturnCleanup: // 入口点

delete[] recdData;

if ( bFileIsOpen )
destFile.Close();

sockClient.Close();

return bRet;
}