windows环境下C++多线程文件传输

把上午写的传输字符串代码改了一下，本来是想实现客户端和服务器一对一多线程分块传输的，结果发现好像变成了一个服务器对多个客户端，每个客户端分配一条线程来处理文件传输任务的模式...

直接上代码，使用的是简单的fstream操作。
Sever代码：

#include "stdafx.h"
#include <WinSock2.h> //windows socket的头文件
#include <Windows.h>
#include <iostream>
#include <thread>
#include <mutex>
#include <process.h>
#include <fstream>
#include <string>

#pragma comment(lib, "ws2_32.lib") //连接winsock2.h的静态库文件

using namespace std;

mutex m;

//定义结构体用来设置
typedef struct my_file
{
	SOCKET clientSocket; //文件内部包含了一个SOCKET 用于和客户端进行通信
	sockaddr_in clientAddr; //用于保存客户端的socket地址
	int id; //文件块的序号
}F;

DWORD WINAPI transmmit(const LPVOID arg)
{
	//实际上这里也可以不加锁，上锁是为了方便看输出
	m.lock();

	F *temp = (F*)arg;
	//获取文件的序号
	//int file_id = temp->id;
	//获取客户机的端口号
	//ntohs(temp -> clientAddr.sin_port); 
	cout << "测试开始,等待客户端发送消息..." << endl;
	//从客户端处接受数据
	/*
	char Buffer[MAXBYTE] = { 0 }; //缓冲区
	recv(temp->clientSocket, Buffer, MAXBYTE, 0); //recv方法 从客户端通过clientScocket接收
	cout << "线程" << temp->id << "从客户端的" << ntohs(temp->clientAddr.sin_port) << "号端口收到:" << Buffer << endl;
	*/
	char file_name[100] = { 0 }; //文件路径
	cout << "输入传输文件路径: ";
	cin >> file_name;
	FILE *fp = fopen(file_name, "rb"); //将文件按二进制读取
	if (fp == NULL)
	{
		cout << "文件" << file_name << "出错或不存在" << endl;
	}
	else
	{
		char Buffer[MAXBYTE] = { 0 }; //文件缓冲区
		int size = 0; //读取的文件长度
		//每次读取完之后清空缓存区，以便下一块文件读入
		while ((size = fread(Buffer, sizeof(char), MAXBYTE, fp)) > 0) 
		{
			//返回非0值表示send错误
			if (send(temp->clientSocket, Buffer, size, NULL) < 0)
			{
				cout << "传输出错，请检查网络配置。" << endl;
			}
			memset(&Buffer, 0, MAXBYTE);
		}
		cout << temp->id << "线程已成功发送" << file_name << endl;
		fclose(fp);
	}


	/*
	//发送简单的字符串到客户端
	const char* s = "Server file";
	send(temp->clientSocket, s, strlen(s)*sizeof(char)+1, NULL);
	cout << "线程" << temp->id << "通过客户端的" << ntohs(temp->clientAddr.sin_port) << "号端口发送:" << s << endl;
	*/

	m.unlock();

	return 0;
}



int main()
{
	WSADATA wsaData;
	//第一个参数是winsocket load的版本号（2.2）
	WSAStartup(MAKEWORD(2, 3), &wsaData);
	//创建服务器端的socket（协议族， sokcet类型）
	SOCKET servSocket = socket(AF_INET, SOCK_STREAM, 0);//如果改成SOCK_DGRAM则使用UDP
	sockaddr_in servAddr; //服务器的socket地址，包含sin_addr表示IP地址，sin_port保持端口号和sin_zero填充字节
	memset(&servAddr, 0, sizeof(SOCKADDR)); //初始化socket地址
	servAddr.sin_family = PF_INET; //设置使用的协议族
	servAddr.sin_port = htons(2017); //设置使用的端口
	servAddr.sin_addr.s_addr = inet_addr("127.0.0.1"); //define s_addr = S_un.S_addr
	::bind(servSocket, (SOCKADDR *)&servAddr, sizeof(SOCKADDR)); //将之前创建的servSocket和端口，IP地址绑定

	HANDLE hThread[2]; //获取句柄
	listen(servSocket, 1); //监听服务器端口
	for (int i = 0; i < 1; i++)
	{
		F *temp = new F; //创建新的传输结构体
		sockaddr_in clntAddr;
		int nSize = sizeof(SOCKADDR);
		SOCKET clientSock = accept(servSocket, (SOCKADDR*)&clntAddr, &nSize);
		//temp数据成员赋值
		temp->clientSocket = clientSock;
		temp->id = i + 1;
		temp->clientAddr = clntAddr;
		//通过句柄创建子线程
		hThread[i] = CreateThread(NULL, 0, &transmmit, temp, 0, NULL);
	}
	//等待子线程完成
	WaitForMultipleObjects(1, hThread, TRUE, INFINITE);
	cout << "错误代码: " << WSAGetLastError() << endl;

	//关闭socket，释放winsock
	closesocket(servSocket);
	WSACleanup();

	cout << "服务器连接已关闭。" << endl;
	system("pause");

	return 0;
}

Client端代码：

#include "stdafx.h"
#include <WinSock2.h> //windows socket的头文件
#include <Windows.h>
#include <iostream>
#include <thread>
#include <process.h>

#pragma comment(lib, "ws2_32.lib") //连接winsock2.h的静态库文件

using namespace std;

int main()
{
	//加载winsock库
	WSADATA wsadata;
	WSAStartup(MAKEWORD(2, 3), &wsadata);

	//客户端socket
	SOCKET clientSock = socket(PF_INET, SOCK_STREAM, 0);
	//初始化socket信息
	sockaddr_in clientAddr;
	memset(&clientAddr, 0, sizeof(SOCKADDR));
	//clientAddr.sin_addr.s_addr = htonl(INADDR_ANY);
	clientAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
	clientAddr.sin_family = PF_INET;
	clientAddr.sin_port = htons(2017);
	//建立连接
	connect(clientSock, (SOCKADDR*)&clientAddr, sizeof(SOCKADDR));

	cout << "已建立连接。" << endl;

	/*
	char* s = new char[100];
	cout << "请输入你要发送的文字消息: ";
	cin >> s;
	send(clientSock, s, strlen(s)*sizeof(char) + 1, NULL);
	cout << "已发送:" << s << endl;
	*/

	char Buffer[MAXBYTE] = { 0 }; // 文件缓冲区
	char wb_file[100] = { 0 }; //写入的文件

	cout << "请输入想要写入的文件: ";
	cin >> wb_file;

	FILE *fp = fopen(wb_file, "wb");
	if (fp == NULL)
	{
		cout << "操作文件时出错" << endl;
		system("pause");
	}
	else
	{
		memset(&Buffer, 0, MAXBYTE);
		int size = 0;
		//当成功接收文件（size > 0）时，判断写入的时候文件长度是否等于接收的长度
		while ((size = recv(clientSock, Buffer, MAXBYTE, 0)) > 0)
		{
			if (fwrite(Buffer, sizeof(char), size, fp) < size)
			{
				cout << "写入出错，部分文件缺失。" << endl;
			}
			//清空缓存区以便下一次接收
			memset(&Buffer, 0, MAXBYTE);
		}
		cout << "接收完成" << endl;
		fclose(fp);
	}
	
	closesocket(clientSock);
	WSACleanup();

	cout << "客户端连接已关闭。" << endl;
	system("pause");

	return 0;
}

因为之前用了WaitForMultipleObjects，所以客户端即使已经接收完成，亦不会输出接收完成的信息，只有关闭服务器的时候才会输出...

接下来思考一下怎么实现大文件分块传输的问题吧_(:з)∠)_