基于UDP协议的客户端与服务器端的文件传送

01
客户端

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<unistd.h>

#include <sys/types.h>
#include <sys/socket.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include<unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#define IP   　"127.0.0.1"
#define PORT　 8888

// 客户端
int main(int argc, char *argv[])
{
	int sockfd;
	int ret;
	char buf[32] = {0};
	
	if(argc != 2 )
	{
		printf("fault\n");
		
		exit(1);
	}

	struct sockaddr_in server_addr;

	// 打开网络通信端口
	sockfd = socket(PF_INET, SOCK_DGRAM, 0);
	
	if(-1 == sockfd)
	{
		perror("socket");
		exit(1);
	}
	
	// 清空
	bzero(&server_addr, sizeof(server_addr));
	
	server_addr.sin_family = PF_INET;
	server_addr.sin_port = htons(PORT);
	server_addr.sin_addr.s_addr = inet_addr(IP);
	int length = sizeof(server_addr);
	
	int fd = open(argv[1], O_RDONLY);
	
	if(-1 == fd)
	{
		perror("open");
		exit(1);
	}
	
	// 命令行参数
	// 第一个：可执行文件
	// 第二个：待发送文件的名称
	ret = sendto(sockfd, argv[1], strlen(argv[1]), 0, (struct sockaddr*)&server_addr, (socklen_t)length);
	
	if(-1 == ret)
	{
		perror("sendto");
		exit(1);
	}

	while(1)
	{
		// 从打开的文件里读一定字节内容
		ret = read(fd, buf, sizeof(buf)-1);
		
		if(-1 == ret)
		{
			perror("read");
		}
		
		if(0 == ret)
		{
			printf("文件传输完毕!\n");
			strcpy(buf, "bye");
			sendto(sockfd, buf, strlen(buf), 0, (struct sockaddr*)&server_addr, (socklen_t)length);
			
			close(fd);
			
			break;
		}
		
		ret = sendto(sockfd, buf, strlen(buf), 0, (struct sockaddr*)&server_addr, (socklen_t)length);
		
		if(-1 == ret)
		{
			perror("sendto");
			
		}
		
		bzero(buf, sizeof(buf));
	}
	
	close(sockfd);
	close(fd);
	
	return 0;
}

02
服务器端

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include<unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#define  PORT   8888
#define  IP     "127.0.0.1"

// 服务器
int main()
{
	int sockfd, ret;
	struct sockaddr_in server_addr;
	struct sockaddr_in client_addr;
	char buf[32] = {0};

	sockfd = socket( PF_INET, SOCK_DGRAM, 0);
	
	if(-1 == sockfd)
	{
		perror("socket");
		exit(1);
	}
	
	bzero(&server_addr, sizeof(server_addr));

	server_addr.sin_family = PF_INET;
	server_addr.sin_port = htons(PORT);
	server_addr.sin_addr.s_addr = inet_addr(IP);

	ret = bind(sockfd, (struct sockaddr*)&server_addr, sizeof(server_addr));
	
	if(-1 ==ret)
	{
		perror("bind");
		exit(1);
	}

	int length = sizeof(client_addr);

	ret =  recvfrom(sockfd, buf, sizeof(buf), 0, (struct sockaddr*)&client_addr, (socklen_t*)&length);
	if(-1 == ret)
	{
		perror("recvfrom");
		exit(1);
	}
	
	printf("收到文件%s, 接受中...\n", buf);

	
	int fd  = open(buf, O_WRONLY | O_CREAT |O_EXCL, 00700);
	if(-1 == fd)
	{
		perror("open");
		exit(1);
	}

	while(1)
	{
		bzero(buf, sizeof(buf));
		ret = recvfrom(sockfd, buf, sizeof(buf), 0, (struct sockaddr*)&client_addr, (socklen_t*)&length);
		if(-1 == ret)
		{
			perror("recvfrom");
		}

		if(!strcmp(buf, "bye"))
		{
			printf("文件接受完毕!\n");
			close(fd);
			break;
		}
		

		ret = write(fd, buf, strlen(buf));
		if(-1 == ret)
		{
			perror("write");
		}
		bzero(buf, sizeof(buf));
	}
	close(sockfd);
	return 0;
}