华清远见上海中心22071班

简单文件传输协议，适用于在网络上进行文件传输的一套标准协议，使用UDP传输

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

int do_download(int sfd, struct sockaddr_in sin);
int do_upload(int sfd, struct sockaddr_in sin);
#define ERR_MSG(msg) do{\
    fprintf(stderr, "__%d__:", __LINE__);\
    perror(msg);\
}while(0)

#define PORT 	69 	//服务器绑定的端口号
#define IP 		"192.168.31.150" 	//服务器绑定的IP

int main(int argc, const char *argv[])
{
	//创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}

	//绑定客户端自身的IP和端口 --->非必须绑定
	
	
	//因为要发送给服务器，所以要先填充服务器的地址信息结构体
	//在sendto函数中使用。
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;
	sin.sin_port 		= htons(PORT); 	//服务器的端口号
	sin.sin_addr.s_addr = inet_addr(IP); 	//服务器自身的IP地址

	char c = 0;
	while(1)
	{
		printf("****************\n");
		printf("******1.下载*********\n");
		printf("******2.上传*********\n");
		printf("******3.退出*********\n");
		printf("****************\n");
		printf("请输入(1,2,3)>>>");
		c = getchar();
		while(getchar()!= 10);

		switch(c)
		{
		case '1': 	//下载 
			do_download(sfd, sin);
			break;
		case  '2': 	//上传
			do_upload(sfd, sin);
			break;
		case '3':
			goto END;
		default:
			printf("输入错误，请重新输入\n");
		}

	}

END:
	close(sfd);

	return 0;
}

int do_download(int sfd, struct sockaddr_in sin)
{
	
	char buf[1024] = ""; 	//初始化字符串
	//组下载请求包
	char* ptr = buf;

	unsigned short* ptr1 = (unsigned short*)ptr;
	*ptr1 = htons(1); 	//下载请求的网络字节序

	char* ptr2 = ptr+2;
	printf("请输入文件名》》");
	char str8[20]="";
	fgets(str8, sizeof(str8), stdin);
	str8[strlen(str8)-1]=0;
	strcpy(ptr2, str8);

	char* ptr3 = ptr2+strlen(ptr2);
	*ptr3 = 0;

	char* ptr4 = ptr3+1;
	strcpy(ptr4, "octet");

	int size = 2+strlen(ptr2)+1+strlen(ptr4)+1;

	//发送下载请求包
	if(sendto(sfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	printf("发送下载请求包成功\n");
	int fp =  open("./6.png", O_WRONLY|O_CREAT|O_TRUNC, 0664);

	if(fp<0)
	{
		ERR_MSG("open");
	}
	
	struct sockaddr_in rec;
	socklen_t a = sizeof(rec);

	//打开一个文件
	while(1)
	{
		//循环接收服务器返回的数据包
		bzero(buf, sizeof(buf));
		
	    ssize_t res1=recvfrom(sfd, buf, sizeof(buf), 0,(struct sockaddr*)&rec, &a);
		
		//如果是数据包，则将数据提取出来，数据在buf+4的位置上
		char* p=buf+4;
		char* p2=buf+2;
		unsigned short* p3=(unsigned short*)p2;
	
		char*buf1 =buf;
		unsigned short* buf2=(unsigned short*)buf1;
		//分析是数据包还是错误报
		if(3==ntohs(*buf2))
		{
		//将提取出来的数据写入到文件中
			if(write(fp, p, res1-4)<0);
			{
				ERR_MSG("write");
				return -1;
			}
			char str[20]="";
		//组ACK包，回复ACK给服务器
			char* str1=str;
			unsigned short* str2 =(unsigned short*) str1;
			*str2=htons(4);
			char* str3 = str+2;
			unsigned short* str4=(unsigned short *)str3;
			*str4=*p3;
	        if(sendto(sfd, str, 4, 0, (struct sockaddr*)&rec, sizeof(rec)) < 0)
			{
				ERR_MSG("sendto");
				return -1;
			}
		//直到数据内容的长度小于512个字节
		    if(res1<516)
			{
				break;
			}
		}
		//如果是错误包，
		else
		{
			printf("%s\n", p);
		}
	}


		//则打印错误信息，退出



	return 0;

}

int do_upload(int sfd, struct sockaddr_in sin)
{
	char arr[1024] = ""; 	//初始化字符串
	//组下载请求包
	char* brr = arr;

	unsigned short* brr1 = (unsigned short*)brr;
	*brr1 = htons(2); 	//下载请求的网络字节序

	char* brr2 = arr+2;
	char buf5[20]="";
	//终端获取文件名
	printf("请输入文件名>>");
	fgets(buf5, sizeof(buf5),stdin);
	buf5[strlen(buf5)-1]=0;
	strcpy(brr2, buf5);

	char* brr3 = brr2+strlen(brr2);
	*brr3 = 0;

	char* brr4 = brr3+1;
	strcpy(brr4, "octet");

	int size = 2+strlen(brr2)+1+strlen(brr4)+1;

	//发送上传请求包
	if(sendto(sfd, arr, size, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	bzero(arr, sizeof(arr));
	struct sockaddr_in rec;
	socklen_t size4 =  sizeof(rec);
	if(recvfrom(sfd, arr, sizeof(arr), 0,(struct sockaddr*)&rec,&size4)<0)
	{
		ERR_MSG("recvfromf");
		return -1;
	}
	printf("发送上传请求包成功\n");

	int mp = open("./fp.png", O_RDONLY);
	ssize_t ze;
	char drr[512]="";

	//组上传请求包
	//发送上传请求
		int num=1;

	    char arr8[128]="";
	while(1)
	{
		//组建发送包
		bzero(arr, sizeof(arr));
		char* p=arr;
		unsigned short* p1 = (unsigned short*)p;
		*p1 = htons(3);     //操作码
		unsigned short* p2 = p1+1;
		*p2 = htons(num);    //快编号

		ze = read(mp,drr, sizeof(drr));
		if(ze<0)
		{
			ERR_MSG("read");
			return -1;
		}
		else if(0==ze)
		{
			printf("读取完毕\n");
			break;
		}
		char* p3=arr+4;
		strcpy(p3, drr);       //文件内容
		int size3=4+strlen(p3);
		if(sendto(sfd, arr, ze+4, 0, (struct sockaddr*)&rec, sizeof(rec)) < 0)
		{
			ERR_MSG("sendto");
			return -1;
		}
	//	接收ACK应答包
		if(recvfrom(sfd, arr8, sizeof(arr8), 0,(struct sockaddr*)&rec, &size4)<0)
		{
			ERR_MSG("recvfromf");
			return -1;
		}

		//如果发送数据包不成功， 接收打印错误报信息
		char* str6 =arr8;
		unsigned short* str7 =(unsigned short*)str6;
		if(*str7==5)
		{
			printf("%s\n", arr8+4);
			
		}
		num ++;

		
	

	}
	return 0;
}