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[])
{
	//初始化WSA
	WORD 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;
}