windows网络编程（四）——不同主机之间传送文件应用

客户端

#include <stdio.h>
#include <stdlib.h>
#include <WinSock2.h>
#pragma comment(lib, "ws2_32.lib")  //加载 ws2_32.dll
int main(){
	char fullname[200] = "C:\\Users\\Administrator\\Desktop\\";


	//初始化DLL
	WSADATA wsaData;
	WSAStartup(MAKEWORD(2, 2), &wsaData);
	//创建套接字
	SOCKET sock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
	//向服务器发起请求
	sockaddr_in sockAddr;
	memset(&sockAddr, 0, sizeof(sockAddr));  //每个字节都用0填充
	sockAddr.sin_family = PF_INET;
	sockAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
	sockAddr.sin_port = htons(1234);

	
	if (connect(sock, (SOCKADDR*)&sockAddr, sizeof(SOCKADDR)) == SOCKET_ERROR)
	{
		closesocket(sock);
		printf("error connect!  %d\n",WSAGetLastError());
		WSACleanup();
	//	exit(1);
	}
	//接收服务器传回的数据
	char szBuffer[MAXBYTE] = "";
	//send(sock, szBuffer, MAXBYTE, NULL);
	if (recv(sock, szBuffer, MAXBYTE, NULL) == SOCKET_ERROR)
	{
		printf("接受失败！\n"); closesocket(sock); WSACleanup(); return 0;
	}
	//输出接收到的数据
	printf("Message form server: %s\n", szBuffer);
	char YN[10];
	scanf("%s", YN);//是否传输
	if (send(sock, YN, 2, NULL) == SOCKET_ERROR)
	{
		printf("发送失败！\n"); closesocket(sock); WSACleanup(); return 0;
	}
	int  len;
	if ((len=recv(sock, szBuffer, MAXBYTE, NULL)) == SOCKET_ERROR)
	{
		printf("接受失败！\n"); closesocket(sock); WSACleanup(); return 0;
	}
	printf("Message form server: %s\n", szBuffer);
	scanf("%s", YN);//发送类型
	if (send(sock, YN, 5, NULL) == SOCKET_ERROR)
	{
		printf("发送失败！\n"); closesocket(sock); WSACleanup(); return 0;
	}
	if ((len = recv(sock, szBuffer, MAXBYTE, NULL)) == SOCKET_ERROR)
	{
		printf("接受失败！\n"); closesocket(sock); WSACleanup(); return 0;
	}
	printf("Message form server: %s\n", szBuffer);
	char filename[50];
	printf( "输入待发送的文件名：");
	scanf("%s", filename);
	strcat(fullname, filename);
	FILE*p = fopen(fullname, "rb+");
	if (!p)
	{
		printf("打开文件失败！\n"); closesocket(sock); WSACleanup(); return 0;
	}
	while ((len=fread(szBuffer, 1, 99, p))>0)
	{
		send(sock, szBuffer, len, NULL);
	}
	fclose(p);
	//关闭套接字
	shutdown(sock, SD_SEND);
	closesocket(sock);
	//终止使用 DLL
	WSACleanup();
	system("pause");
	return 0;
}

服务器端

#include <stdio.h>
#include<iostream>
#include <winsock2.h>
#include<string>
#include<ctime>
using namespace std;
#pragma comment (lib, "ws2_32.lib")  //加载 ws2_32.dll
int main(){
	//初始化 DLL
	WSADATA wsaData;
	WSAStartup(MAKEWORD(2, 2), &wsaData);
	//创建套接字
	SOCKET servSock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
	//绑定套接字
	sockaddr_in sockAddr;
	memset(&sockAddr, 0, sizeof(sockAddr));  //每个字节都用0填充
	sockAddr.sin_family = PF_INET;  //使用IPv4地址
	sockAddr.sin_addr.s_addr = INADDR_ANY;  //具体的IP地址
	sockAddr.sin_port = htons(1234);  //端口
	bind(servSock, (SOCKADDR*)&sockAddr, sizeof(SOCKADDR));
	//进入监听状态
	listen(servSock, 20);
	//接收客户端请求
	while (1)
	{
		FILE*p;
		char type[100];
		char baseicpath[200] = "C:\\Users\\Administrator\\Desktop\\receivefiles\\";
		char *fullpath[200];
		int len;
		SOCKADDR clntAddr;
		int nSize = sizeof(SOCKADDR);
		SOCKET clntSock = accept(servSock, (SOCKADDR*)&clntAddr, &nSize);
		//向客户端发送数据
		char str[100] = "Are you sending a file?(sending Y/N)";
		if (clntSock == INVALID_SOCKET)
		{
			printf("无效socket!\n");
			continue;
		}
		if (send(clntSock, str, strlen(str) + sizeof(char), NULL) == SOCKET_ERROR)
		{
			printf("发送失败\n"); closesocket(clntSock); continue;
		}
		else
		{
			printf("已成功发送一条回复！\n");
		}
		if (recv(clntSock, str, 100, NULL) == SOCKET_ERROR)
		{
			printf("接受失败！\n"); closesocket(clntSock); continue;
		}
		switch (str[0])
		{
		case 'Y':
		case 'y':
			//传数据
		{strcpy(str, "please note the type of your file(.txt/.pdf/.PNG——only three type can be supported)");
		if (send(clntSock, str, strlen(str) + sizeof(char), NULL) == SOCKET_ERROR)//要求类型
		{
			printf("发送失败！\n"); closesocket(clntSock); continue;
		}
		if (recv(clntSock, type, 100, NULL) == SOCKET_ERROR)//收到类型
		{
			printf("接受失败！\n"); closesocket(clntSock); continue;
		}
		strcpy(str, "please send  your file(s)!I'm ready!");
		if (send(clntSock, str, strlen(str) + sizeof(char), NULL) == SOCKET_ERROR)//要求类型
		{
			printf("发送失败！\n"); closesocket(clntSock); continue;
		}
		char filename[50];//文件名
		itoa(time(0), filename, 10);
		strcat(filename, type);
		strcat(baseicpath, filename);//完整名
		p = fopen(baseicpath, "wb+");
		if (!p){ printf("文件打开失败！\n"); closesocket(clntSock); continue; }
		int len;
		while ((len = recv(clntSock, str, 99, NULL)) >0)
		{
			fwrite(str, 1, len, p);
		}
		break;
		}
		case'N':
		case'n':
		{
				   break;
		}
			break;
		}
		shutdown(clntSock, SD_RECEIVE);
		fclose(p);
		//关闭套接字
		closesocket(clntSock);
	}
	closesocket(servSock);
	//终止 DLL 的使用
	WSACleanup();
	system("pause");
	return 0;
}