Windows Socket编程之TCP实现大文件的传输

前言：

UDP版本可参考被人博客：http://blog.csdn.net/luchengtao11/article/details/71016222

本文所述代码工程托管在Github:https://github.com/Wuchenwcf/MyCode/tree/master/Socket

本文实现以下功能：

在客户端，用户选择本地的某个文件，并发送到服务器端。
在服务器端，接收客户端传输的数据流，并按IP 地址保存在服务器端（文
件名重复的，可以覆盖）。
如果传输过程中服务器端发现客户端断开，服务器端应删除文件，并在屏幕
上提示，如“IP：1.2.3.4 发来abcd.txt 文件过程中失去连接。”。如果客户端发
现服务器端不工作，客户端应有提示“服务器1.2.3.5:62345 失去连接”。

一、Socket TCP传输的流程

二、代码实现

1.FileHelper用来进行与文件相关的操作

#include<stdio.h>#include<stdlib.h>#include <WINSOCK2.H>#include <STDIO.H>#pragma  comment(lib,"ws2_32.lib")class FileHelper{private:FILE *f;char path_buffer[_MAX_PATH];char drive[_MAX_DRIVE];char dir[_MAX_DIR];char fname[_MAX_FNAME];char ext[_MAX_EXT];public:FILE * selectfile(){printf("请输入要传送的文件名\n");scanf("%s",path_buffer);if (f=fopen(path_buffer,"rb")){printf("文件打开成功\n");return f;}else{printf("文件不存在，请重新输入\n");return selectfile();}}char * getFileName(){_splitpath(path_buffer, drive, dir, fname, ext);return strcat(fname, ext);}FILE * createFile(char *name){remove(name);if (f = fopen(name, "ab")){printf("文件创建成功\n");}else{printf("文件创建失败\n");}return f;}bool createDir(char *dir){char head[MAX_PATH] = "md ";return system(strcat(head, dir));}};

2.Client端

#include <stdio.h>#include <WINSOCK2.H>#include <STDIO.H>#include"FileHelper.h"#pragma  comment(lib,"ws2_32.lib")int main(int argc, char* argv[]){WORD sockVersion = MAKEWORD(2,2);//版本号WSADATA data;    //用来保存WSAStartup调用后返回的windows Sockets数据FileHelper fh;if(WSAStartup(sockVersion, &data) != 0){return 0;}sockaddr_in serAddr;serAddr.sin_family = AF_INET;serAddr.sin_port = htons(8888);serAddr.sin_addr.S_un.S_addr = inet_addr("59.77.13.189"); while (true){SOCKET sclient = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);if (sclient == INVALID_SOCKET){printf("invalid socket !");return 0;}if (connect(sclient, (sockaddr *)&serAddr, sizeof(serAddr)) == SOCKET_ERROR){printf("connect error !");closesocket(sclient);return 0;}FILE *f=fh.selectfile();char sendData[BUFSIZ];char recData[BUFSIZ];char over[BUFSIZ] = "Finnal";char * name = fh.getFileName();strcpy(sendData, name);printf("%s\n", sendData);int nCount;long long sum = 0;send(sclient, sendData, strlen(sendData)+1, 0);int ret = recv(sclient, recData, BUFSIZ, 0);printf(recData);while ((nCount=fread(sendData,1,BUFSIZ,f))>0){printf("%db\n",sum+=nCount);send(sclient, sendData, nCount, 0);int ret = recv(sclient, recData, BUFSIZ, 0);if (ret >0){//recData[ret] = 0x00;printf(recData);}else{printf("与服务器失去连接");break;}}send(sclient, over, BUFSIZ, 0);ret = recv(sclient, recData, BUFSIZ, 0);if (ret>0&&strcmp(recData,over)==0){printf("传输成功！");}fclose(f);closesocket(sclient);}WSACleanup();return 0;}

3.Server端

#include <stdio.h>#include <stdio.h>#include <winsock2.h>#include "FileHelper.h"#pragma comment(lib,"ws2_32.lib")int main(int argc, char* argv[]){//初始化WSAWORD sockVersion = MAKEWORD(2, 2);WSADATA wsaData;FileHelper fh;if (WSAStartup(sockVersion, &wsaData) != 0){return 0;}//创建套接字SOCKET slisten = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);if (slisten == INVALID_SOCKET){printf("socket error !");return 0;}//绑定IP和端口sockaddr_in sin;sin.sin_family = AF_INET;sin.sin_port = htons(8888);sin.sin_addr.S_un.S_addr = INADDR_ANY;if (bind(slisten, (LPSOCKADDR)&sin, sizeof(sin)) == SOCKET_ERROR){printf("bind error !");}//开始监听if (listen(slisten, 5) == SOCKET_ERROR){printf("listen error !");return 0;}//循环接收数据SOCKET sClient;sockaddr_in remoteAddr;int nAddrlen = sizeof(remoteAddr);char revData[BUFSIZ];while (true){printf("等待连接...\n");sClient = accept(slisten, (SOCKADDR *)&remoteAddr, &nAddrlen);if (sClient == INVALID_SOCKET){printf("accept error !");continue;}printf("接受到一个连接：%s \r\n", inet_ntoa(remoteAddr.sin_addr));/*if (fh.createDir(inet_ntoa(remoteAddr.sin_addr)))printf("文件夹创建成功！");*/int ret = 0;long long count = 0;char sendData[BUFSIZ] = "你好，TCP客户端！\n";ret = recv(sClient, revData, BUFSIZ, 0);char fromname[BUFSIZ] = {};strcpy(fromname, revData);char mid[3] = "\\";char finame[MAX_PATH] = {};char over[BUFSIZ] = "Finnal";strcat(finame, inet_ntoa(remoteAddr.sin_addr));printf(finame);strcat(finame, mid);strcat(finame, revData);//printf(finame);FILE *f = fh.createFile(finame);send(sClient, sendData, BUFSIZ, 0);while ((ret = recv(sClient, revData, BUFSIZ, 0)) > 0){//printf("%d\n", ret);printf("%db\n", count += ret);if (strcmp(revData,over)==0){printf("文件%s传输成功\n", fromname);break;send(sClient, over, BUFSIZ, 0);}fwrite(revData, 1, ret, f);send(sClient, sendData, BUFSIZ, 0);}fclose(f);if (strcmp(revData, over) != 0){printf("IP：%s发来的%s传输过程中失去连接\n", inet_ntoa(remoteAddr.sin_addr),fromname);remove(finame);}closesocket(sClient);}closesocket(slisten);WSACleanup();return 0;}