C++ Winsock传送文件

Socket至少有一个客户端和一个服务器端。

Client端代码：client.cpp
#include <stdio.h>#include <winsock2.h>#pragma comment(lib, "ws2_32.lib")void main() {// 初始化 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;}// 连接到服务器.sockaddr_in clientService;clientService.sin_family = AF_INET;clientService.sin_addr.s_addr = inet_addr( "127.0.0.1" );clientService.sin_port = htons( 8000 );if ( connect( client, (SOCKADDR*) &clientService, sizeof(clientService) ) == SOCKET_ERROR) {printf( "Failed to connect./n" );WSACleanup();return;}char recvbuf[1024];//发送缓冲区int read;DWORD dwWrite;BOOL bRet;// 发送并接收数据.char filename[]="avatar.jpg";HANDLE hFile=CreateFile(filename,GENERIC_WRITE,0,0,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,0);while(true){read=recv( client, recvbuf, 1024, 0 );if(read==-1)break;bRet=WriteFile(hFile,recvbuf,read,&dwWrite,NULL);if(bRet==FALSE){MessageBox(NULL,"Write Buf ERROR!","Error",MB_OK);break;}}MessageBox(NULL,"Receive file OK!","OK",MB_OK);return;}  

Server端代码：server.cpp
#include <stdio.h>#include <winsock2.h>#pragma comment(lib, "ws2_32.lib")void main() {// 初始化WSADATA wsaData;int iResult = WSAStartup( MAKEWORD(2,2), &wsaData );if ( iResult != NO_ERROR )printf("Error at WSAStartup()/n");// 建立socketSOCKET server;server = socket( AF_INET, SOCK_STREAM, IPPROTO_TCP );if ( server == INVALID_SOCKET ) {printf( "Error at socket(): %ld/n", WSAGetLastError() );WSACleanup();return;}// 绑定socketsockaddr_in service;service.sin_family = AF_INET;service.sin_addr.s_addr = inet_addr( "127.0.0.1" );service.sin_port = htons(8000);if ( bind( server, (SOCKADDR*) &service, sizeof(service) ) == SOCKET_ERROR ) {printf( "bind() failed./n" );closesocket(server);return;}// 监听 socketif ( 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");server = AcceptSocket; break;}// 发送接受数据// int bytesSent;char sendbuf[1024];//发送缓冲区//DWORD dwFileSize;//文件大小DWORD dwRead;BOOL bRet;char filename[]="avatar02.jpg";HANDLE hFile=CreateFile(filename,GENERIC_READ,0,0,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,0);//dwFileSize=GetFileSize(hFile,NULL);while(true){bRet=ReadFile(hFile,sendbuf,1024,&dwRead,NULL);if(bRet==FALSE){MessageBox(NULL,"Read Buf ERROR!","Error",MB_OK);break;}else if(dwRead==0){MessageBox(NULL,"Send file OK!","OK",MB_OK);break;}else{send(server,sendbuf,dwRead,0);}};CloseHandle(hFile);return;}