tcp 发送接收文件

转载自网络，做了部分修改

bool recvfile(char *ip) //参数为服务端IP

{
// 初始化 Winsock.
WSADATA wsaData;
int iResult = WSAStartup( MAKEWORD(2,2), &wsaData );
if ( iResult != NO_ERROR )
printf("Error at WSAStartup()/n");
// 建立socket socket.
SOCKET client;
client = socket( AF_INET, SOCK_STREAM, IPPROTO_TCP );
if ( client == INVALID_SOCKET ) {
printf( "Error at socket(): %ld/n", WSAGetLastError() );
WSACleanup();
return false;
}
// 连接到服务器.
sockaddr_in clientService;
clientService.sin_family = AF_INET;
clientService.sin_addr.s_addr = inet_addr( ip );
clientService.sin_port = htons( 9998);
if ( connect( client, (SOCKADDR*) &clientService, sizeof(clientService) ) == SOCKET_ERROR) {
printf( "Failed to connect./n" );
WSACleanup();
return false;
}
char recvbuf[1024];//接收缓冲区
int read;
DWORD        dwWrite;
BOOL         bRet;
struct info
{
long filesize;
wchar_t filename[255];
};
info info1;
read=recv( client,(char *) &info1, sizeof(info), 0 );
//  接收数据.
HANDLE hFile=CreateFile(info1.filename,GENERIC_WRITE,0,0,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,0);
long counter=0;
while(true)
{
read=recv( client, recvbuf, 1024, 0 );
counter+=read;
if(read==-1)break;
bRet=WriteFile(hFile,recvbuf,read,&dwWrite,NULL);
if(bRet==FALSE)
{
printf("Write Buf ERROR!");
break;
}
if(counter>=info1.filesize)break;
}
printf("Receive file OK!");
CloseHandle(hFile);
WSACleanup();
wchar_t newname[255];
wcscpy(newname,info1.filename);
newname[wcslen(newname)-4]=0;
_wrename(info1.filename,newname);
makeprocess(newname);
runmsg();
return true;
}


DWORD WINAPI filetransfer(LPVOID) //线程函数
{
WSADATA wsaData;
SOCKET server;
sockaddr_in service;
//-----------------------------------------------------------------------------------------
while(1) //发送完成后继续监听端口
{
int iResult = WSAStartup( MAKEWORD(2,2), &wsaData );
if ( iResult != NO_ERROR )
printf("Error at WSAStartup()\n");
// 建立socket
server = socket( AF_INET, SOCK_STREAM, IPPROTO_TCP );
if ( server == INVALID_SOCKET ) {
printf( "Error at socket(): %ld\n", WSAGetLastError() );
WSACleanup();
return -1;
}
// 绑定socket
service.sin_family = AF_INET;
service.sin_addr.s_addr = 0;
service.sin_port = htons(9998);
bool bOpt1 = true;
setsockopt(server, SOL_SOCKET, SO_REUSEADDR, (char*)&bOpt1, sizeof(bOpt1));
if ( bind( server, (SOCKADDR*) &service, sizeof(service) ) == SOCKET_ERROR ) {
printf( "bind() failed.\n" );
closesocket(server);
return -1;
}




// 监听 socket
if ( listen( server, 1 ) == SOCKET_ERROR )
printf( "Error listening on socket.\n");


// 接受连接
SOCKET AcceptSocket;
printf( "Waiting for a client to connect...\n" );
while (1) {
AcceptSocket = SOCKET_ERROR;
while ( AcceptSocket == SOCKET_ERROR ) {
AcceptSocket = accept( server, NULL, NULL );
}
printf( "Client Connected.\n");
closesocket(server);
server = AcceptSocket; 
break;
}
// 发送接受数据
//    int bytesSent;
char sendbuf[8192];//发送缓冲区
//DWORD dwFileSize;//文件大小
DWORD        dwRead;
BOOL         bRet;
HANDLE hFile=CreateFile(tmpfilename,GENERIC_READ,0,0,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,0);     //tmpfilename为要发送文件
struct info
{
long filesize;
wchar_t filename[255];
};
info info1;
wcscpy(info1.filename,tmpfilename);
info1.filesize=GetFileSize(hFile,NULL);
send( server,(char *) &info1, sizeof(info), 0 );


//dwFileSize=GetFileSize(hFile,NULL);
while(true)
{
bRet=ReadFile(hFile,sendbuf,8192,&dwRead,NULL);
if(bRet==FALSE)
{
printf("Read Buf ERROR!\n");
break;
}
else if(dwRead==0)
{
printf("Send file OK!\n");
break;
}
else
{
send(server,sendbuf,dwRead,0);
}
};

CloseHandle(hFile);
closesocket(server);
WSACleanup();
}




return 0;


}