利用流式套接字实现文件传输

Server端代码：

#include "stdafx.h"  #include <winsock2.h>  #include <cstdio>#include <wincrypt.h>#include <time.h>#include <cstring>#include <iostream>  #include <string.h> #define PORT 2345#pragma comment(lib, "user32.lib")#pragma comment(lib, "shlwapi.lib")#pragma comment(lib, "ws2_32.lib")   #pragma comment(lib, "crypt32.lib")  using namespace std;const int MAX1=1000;typedef long long LL;typedef unsigned long long ULL;int recvn(SOCKET s, char * recvbuf, unsigned int fixedlen){int iResult; int cnt=fixedlen; //剩余多少字节尚未接收 while ( cnt > 0 ) {iResult = recv(s, recvbuf, cnt, 0);if ( iResult < 0 ){printf("error: %d\n", WSAGetLastError());return -1;}if ( iResult == 0 )//对方关闭连接，返回已接收到的小于fixedlen的字节数 return fixedlen - cnt;recvbuf +=iResult;cnt -=iResult;}return fixedlen;}int recvvl(SOCKET s) { int iResult;unsigned int reclen; //用于存储报文头部存储的长度信息 unsigned int recvbuflen=0;char filename[100];TCHAR name[100];char tmp[1000],recvbuf[1000];memset(tmp,0,sizeof(tmp));int i=0;memset(filename,0,sizeof(filename));while(true){iResult = recvn(s, tmp, sizeof( char )); if(tmp[0]=='#') break;filename[i++]=tmp[0];if ( iResult !=sizeof (char ))//没有成功接收到一个整型数据{ if ( iResult == -1 ) {printf("get the filename error: %d\n", WSAGetLastError());return -1; } else {printf("connect closed when get the filename\n");return 0;}}}cout<<"filename is "<<filename<<endl;memset(name,0,sizeof(name));for(int i=0;filename[i];i++)name[i]=filename[i];HANDLE hFile;               // 文件句柄    DWORD count;                // 写入的数据计数char path_name[200];    hFile = CreateFile(        name, // 文件名        GENERIC_WRITE,          // 写入权限        0,                      // 阻止其他进程访问        NULL,                   // 子进程不可继承本句柄CREATE_ALWAYS,             // 仅不存在时创建新文件        FILE_ATTRIBUTE_NORMAL,  // 普通文件        NULL    ); if(hFile==INVALID_HANDLE_VALUE)cout<<"fail to creat file"<<endl;iResult = recvn(s, ( char * )&reclen, sizeof( unsigned int )); if ( iResult !=sizeof ( unsigned int )) { if ( iResult == -1 ) {printf("get the size of file error: %d\n", WSAGetLastError());CloseHandle(hFile);return -1; } else {printf("connect closed when get the size of file\n");CloseHandle(hFile);return 0;}} reclen = ntohl( reclen );cout<<"the size of the file is "<<reclen<<endl;recvbuflen=min(1000,reclen);//缓存区int packnum = 1;while ( reclen > 0){memset(recvbuf,0,sizeof(recvbuf));iResult = recvn(s, recvbuf, recvbuflen);cout<< "packet " << packnum++<<" is:"<<recvbuf<<endl;WriteFile(hFile,recvbuf,strlen(recvbuf),&count,0);//将recvbuflen内容写入文件reclen -= recvbuflen; //处理最后一段数据长度 recvbuflen = min(reclen,recvbuflen);cout<<"reclen is "<<reclen<<endl;}char endsign[10]={"OK"};cout << "send OK to client\n get file!" << endl;send(s,endsign,strlen(endsign),0);CloseHandle(hFile);return 0 ;}DWORD WINAPI ClientThread(LPVOID lpParameter){SOCKET socketClient = (SOCKET) lpParameter;cout<<"begin to receive file"<<endl;recvvl(socketClient);cout << "closed" << endl;return 0;}int main(int argc, const char* argv[]){WORD sockVersion = MAKEWORD(2,2); WSADATA wsaData; int error= WSAStartup(sockVersion,&wsaData);if(error!=0){cout << "fail to startup! " << WSAGetLastError()<<endl;return 0;}SOCKET socketSever=socket(AF_INET,SOCK_STREAM,0);if(socketSever==INVALID_SOCKET){cout << "socket error! " << WSAGetLastError() << endl;WSACleanup();closesocket(socketSever);return 0;}//服务端socket 地址sockaddr_in addrServer;addrServer.sin_addr.S_un.S_addr = htonl(INADDR_ANY);addrServer.sin_family=AF_INET;addrServer.sin_port=htons(PORT);//将socket绑定在本地端口if(bind(socketSever,(SOCKADDR*)&addrServer,sizeof(SOCKADDR))==SOCKET_ERROR){cout<<"bind error! "<<WSAGetLastError()<<endl;;closesocket(socketSever);WSACleanup();return 0;}cout<<"listening the client"<<endl;if(listen(socketSever,10)==SOCKET_ERROR)cout<<"Listen failed with error "<<WSAGetLastError()<<endl;int len=sizeof(SOCKADDR);SOCKET socketClient;sockaddr_in addrClient;char* ip;while(true){socketClient = accept(socketSever,(SOCKADDR *)&addrClient,&len); ip = inet_ntoa(addrClient.sin_addr);cout<<"succeed connection      ip: "<<ip<<"    port: "<<addrClient.sin_port<<endl;HANDLE hThread=NULL;hThread = CreateThread(NULL,0,ClientThread,(LPVOID)socketClient,0,NULL);if(hThread==NULL){cout<<"Creat Thread Failed! "<<WSAGetLastError()<<endl;break;}CloseHandle(hThread);}closesocket(socketSever);return 0;}

Client端代码:

#include "stdafx.h"  #include <stdio.h>  #include <winsock2.h>  #include <wincrypt.h>#include <time.h>#include <cstring>#include <iostream>  #include <string.h> #pragma comment(lib, "user32.lib")#pragma comment(lib, "shlwapi.lib")#pragma comment(lib, "ws2_32.lib")   #pragma comment(lib, "crypt32.lib")const int MAX1 = 1000;using namespace std;void FileSend(SOCKET s){char buf[MAX1];memset(buf,0,sizeof(buf));DWORD len = 0;cout<<"Please input the filename :";char filename[100];memset(filename,0,sizeof(filename));gets(filename);filename[strlen(filename)]='#';int err=send(s,filename,strlen(filename),0);if(err == SOCKET_ERROR){cout<<"send filename fail "<<GetLastError()<<endl;return;}TCHAR name[100];memset(name,0,sizeof(name));for(int i=1;filename[i];i++)name[i-1]=filename[i-1];HANDLE hFile;hFile = CreateFile(name,GENERIC_READ,0,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL);if(hFile == INVALID_HANDLE_VALUE){cout<<"cannot open the file"<<endl;CloseHandle(hFile);return;}DWORD dwBytesRead,dwBytesToRead,tmpLen;unsigned _size=GetFileSize(hFile,NULL);char tmpBuf[1<<15];memset(tmpBuf,0,sizeof(tmpBuf));dwBytesToRead = _size;    dwBytesRead = 0;cout<<"the size of the file is "<<_size<<endl;_size=htonl(_size);err=send(s,(char*)&_size,sizeof(int),0);if(err == SOCKET_ERROR){cout<<"send filesize fail "<<GetLastError()<<endl;CloseHandle(hFile);return;}//dwBytesRead=100;do{memset(tmpBuf, 0, sizeof(tmpBuf));ReadFile(hFile, tmpBuf, 1024, &dwBytesRead, NULL);cout << tmpBuf << endl;if (dwBytesRead == 0)break;dwBytesToRead -= dwBytesRead;send(s, tmpBuf, dwBytesRead, 0);cout << dwBytesRead << endl;} while (dwBytesToRead > 0);cout<<"wait ok size from server"<<endl;char revbuf[10];memset(revbuf,0,sizeof(revbuf));recv(s,revbuf,10,0);CloseHandle(hFile);return;if (strcmp("OK",revbuf)==0){cout<<"get OK and close the client"<<endl;CloseHandle(hFile);return ;}}int main(int argc,const char*argv[]){WORD sockVersion = MAKEWORD(2,2);WSADATA wsaData;int error= WSAStartup(sockVersion,&wsaData);if(error){cout<<"fail to startup"<<GetLastError()<<endl;WSACleanup();return 0;}SOCKET socketClient=socket(AF_INET,SOCK_STREAM,0);if(socketClient==INVALID_SOCKET){cout<<"socket error!  "<<GetLastError()<<endl;WSACleanup();closesocket(socketClient);return 0;}char ip[20];memset(ip,0,sizeof(ip));cout << "Please input the ip: ";gets(ip);cout << endl;sockaddr_in addrServer;addrServer.sin_addr.S_un.S_addr = inet_addr(ip);addrServer.sin_family = AF_INET;addrServer.sin_port = htons(2345);connect(socketClient,(SOCKADDR*)&addrServer,sizeof(SOCKADDR));FileSend(socketClient);closesocket(socketClient);getchar();return 0;}