C语言 TCP 客户端与服务器文件上传

╰半橙微兮° 2022-10-21 03:53 171阅读 0赞

> TCPServer.cpp

#include "stdio.h"
    #include "winsock2.h"
    #pragma comment(lib,"ws2_32.lib")
    #define SERVERIP "127.0.0.1"
    #define SERVERPORT 3333
    #define SIZE 4096//4096
    #define BUFSIZE 4096
    
    int RecvFile(SOCKET sock);
    int main(void)
    {
    	int ret,len;
    	WSADATA data;
    	SOCKET sockListen,sockTx;
    	struct sockaddr_in addrServer;
    
    	ret=WSAStartup(MAKEWORD(2,2),&data);
    	if (SOCKET_ERROR==ret)
    	{
    		return -1;
    	}
    
    	sockListen=socket(AF_INET,SOCK_STREAM,0);
    	if (INVALID_SOCKET==sockListen)
    	{
    		WSACleanup();
    		return -2;
    	}
    
    	memset(&addrServer,0,sizeof(struct sockaddr_in));
    	addrServer.sin_family=AF_INET;
    	addrServer.sin_addr.S_un.S_addr=inet_addr(SERVERIP);
    	addrServer.sin_port=htons(SERVERPORT);
    
    	len=sizeof(struct sockaddr);
    	ret=bind(sockListen,(struct sockaddr*)&addrServer,len);
    	if (SOCKET_ERROR==ret)
    	{
    		closesocket(sockListen);
    		WSACleanup();
    		return -3;
    	}
    
    	ret=listen(sockListen,1);
    	if (SOCKET_ERROR==ret)
    	{
    		closesocket(sockListen);
    		WSACleanup();
    		return -4;
    	}
    
    	len=sizeof(struct sockaddr);
    	sockTx=accept(sockListen,NULL,&len);
    	if (INVALID_SOCKET==sockTx)
    	{
    		closesocket(sockListen);
    		WSACleanup();
    		return -5;
    	}
    
    	printf("接收到客户端的连接\n");
    
    	RecvFile(sockTx);
    
    	closesocket(sockListen);
    	closesocket(sockTx);
    	WSACleanup();
    	system("pause");
    	return 0;
    }
    
    int RecvFile(SOCKET sock)
    {
    	int ret;
    	FILE *fp;
    	char sendBuf[BUFSIZE],recvBuf[BUFSIZE],filename[512];
    
    	
    	memset(recvBuf,0,sizeof(recvBuf));
    	ret=recv(sock,recvBuf,sizeof(recvBuf),0);
    	if (SOCKET_ERROR==ret)
    	{
    		return -1;
    	}
    	//接收客户端发来的文件名
    	printf("成功接收客户端发来的文件名: %s\n",recvBuf);
    
    	memset(sendBuf,0,sizeof(sendBuf));
    	strcpy(sendBuf,recvBuf);
    	ret=send(sock,sendBuf,strlen(sendBuf),0);
    	if (SOCKET_ERROR==ret)
    	{
    		return -2;
    	}
    
    	printf("请输入你希望保存的文件名（带扩展名）\n");
    	scanf("%s",filename);
    
    	//以二进制写方式打开文件
        fp=fopen(filename,"wb");
    	if (NULL==fp)
    	{
    		return -3;
    	}
    
    	memset(recvBuf,0,sizeof(recvBuf));
    	ret=recv(sock,recvBuf,sizeof(recvBuf),0);
    	if (SOCKET_ERROR==ret)
    	{
    		return -4;
    	}
    
    	printf("成功接收客户端发来的文件长度: %s字节\n",recvBuf);
    
        int totlen = atoi(recvBuf);
    
    	memset(sendBuf,0,sizeof(sendBuf));
    	strcpy(sendBuf,recvBuf);
    	ret=send(sock,sendBuf,strlen(sendBuf),0);
    	if (SOCKET_ERROR==ret)
    	{
    		return -5;
    	}
    
    	printf("\n 下面开始接收文件具体内容...  \n\n");
    	int recvBytes = 0;
    	int progress,count;
    	while(recvBytes<totlen)
    	{
    		ret = recv(sock,recvBuf,SIZE,0);
    		if(SOCKET_ERROR==ret)
    		{	
    			return -6;
    		}
    		fwrite(recvBuf,ret,1,fp);
    		recvBytes += ret;
    
    		printf("\r");
    		printf("[");
    
    		progress = 50;
    		count = recvBytes/(totlen/progress);
    		for(int j=0; j<progress; j++)
    		{
    			if(j<=count)
    				printf("=");
    			else
    				printf("+");
    		}
    
    		printf("]");
    		printf(" %8ld/%ld %6.2f%%",recvBytes ,totlen ,(float)recvBytes/totlen*100);
    	}
    	printf("\n\n #######成功接收全部文件####### \n");
    
    	fclose(fp);
    	return 0;	
    }

> TCPClient.cpp

#include "stdio.h"
    #include "stdlib.h"
    #include "winsock2.h"
    #pragma comment(lib,"ws2_32.lib")
    #define SERVERIP "127.0.0.1"
    #define SERVERPORT 3333
    #define SIZE 4096//4096
    #define BUFSIZE 4096
    
    int SendFile(SOCKET sock,char filename[512]);
    int main(void)
    {
    	int ret,len;
    	WSADATA data;
    	SOCKET sockClient;
    	struct sockaddr_in addrServer;
    	char fname[512];
    
    	ret=WSAStartup(MAKEWORD(2,2),&data);
    	if (SOCKET_ERROR==ret)
    	{
    		return -1;
    	}
    
    	sockClient=socket(AF_INET,SOCK_STREAM,0);
    	if (INVALID_SOCKET==sockClient)
    	{
    		WSACleanup();
    		return -2;
    	}
    
    	memset(&addrServer,0,sizeof(struct sockaddr_in));
    	addrServer.sin_family=AF_INET;
    	addrServer.sin_addr.S_un.S_addr=inet_addr(SERVERIP);
    	addrServer.sin_port=htons(SERVERPORT);
    
    	len=sizeof(struct sockaddr);
    	ret=connect(sockClient,(sockaddr*)&addrServer,len);
    	if (SOCKET_ERROR==ret)
    	{
    		closesocket(sockClient);
    		WSACleanup();
    		return -3;
    	}
    
    	printf("连接服务器成功\n");
    
    	printf("请输入你需要传送的文件名,带扩展名\n");
    	scanf("%s",fname);
    	SendFile(sockClient,fname);
    	
    	
    	closesocket(sockClient);
    	WSACleanup();
    
    	system("pause");
    	return 0;
    }
    
    int SendFile(SOCKET sock,char filename[512])
    {
    	int ret,cnt,lenlast;
    	char sendBuf[BUFSIZE],recvBuf[BUFSIZE];
    	FILE *fp;
    	int totlen;
    	fp=fopen(filename,"rb");
    	if (NULL==fp)
    	{
    		printf("open file error\n");
    		return -1;
    	}
    
        fseek(fp,0,SEEK_END);
    	totlen=ftell(fp);
    	printf("文件%s的长度为%d字节\n",filename,totlen);
    
    	char* purename;
    	purename=strrchr(filename,'\\');
    
    	memset(sendBuf,0,sizeof(sendBuf));
    	strcpy(sendBuf,purename+1);
    
    	ret=send(sock,sendBuf,strlen(sendBuf)+1,0);
    	if (SOCKET_ERROR==ret)
    	{
    		return -2;
    	}
    
    	memset(recvBuf,0,sizeof(recvBuf));
    	ret=recv(sock,recvBuf,sizeof(recvBuf),0);
    	if (SOCKET_ERROR==ret|| (strcmp(sendBuf,recvBuf)!=0) )
    	{
    		return -3;
    	} 
    
    	printf("向服务器发送文件名成功\n");
    
    	memset(sendBuf,0,sizeof(sendBuf));
    	itoa(totlen,sendBuf,10);
    	ret=send(sock,sendBuf,strlen(sendBuf)+1,0);
    	if (SOCKET_ERROR==ret)
    	{
    		return -4;
    	}
    
    	memset(recvBuf,0,sizeof(recvBuf));
    	ret=recv(sock,recvBuf,sizeof(recvBuf),0);
    	if (SOCKET_ERROR==ret|| (strcmp(sendBuf,recvBuf)!=0) )
    	{
    		return -5;
    	} 
    
    	printf("向服务器发送文件长度成功\n");	
    
    	int i, res;
    	cnt = totlen/SIZE;
    	res = totlen%SIZE;
    	lenlast = SIZE;
    	if(res)
    	{
    		lenlast = res;
    		cnt++;
    	}
    
    	rewind(fp);
    	printf("\n下面开始发送文件具体内容...\n\n");
    
    	int sendBytes = 0;
    	int progress,count;
    	bool flag = true;
    	for(i=1; i<cnt; i++)
    	{
    		fread(sendBuf,SIZE,1,fp);
    		ret = send(sock,sendBuf,SIZE,0);
    		if (SOCKET_ERROR==ret)
    		{
    			return -6;
    		}
    
    		printf("\r");
    		printf("[");
    
    		sendBytes += ret;
    		progress = 50;
    		count = sendBytes/(totlen/progress);
    		for(int j=0; j<progress; j++)
    		{
    			if(j<=count)
    				printf("=");
    			else
    				printf("+");
    		}
    
    		printf("]");
    
    		if(cnt==i+1)
    		{
    			sendBytes += lenlast;						
    		}
    
    		printf(" %8ld/%ld %6.2f%%",sendBytes ,totlen ,(float)sendBytes/totlen*100);
    	}
    	
    	fread(sendBuf,lenlast,1,fp);
    	ret = send(sock,sendBuf,lenlast,0);
    
    	if (SOCKET_ERROR==ret)
    	{
    		return -7;
    	}
    
    	printf("\n\n ###### 成功发送文件！###### \n");
    	
    	fclose(fp);
    	return 0;
    }