基于UDP的titp文件传输下载功能

#include "myhead.h"
#define ERR_MSG(msg) do{\
	printf("__%d__:",__LINE__);\
	perror(msg);\
}while(0)
#define PORT 69//要发送的端口号
#define IP "192.168.114.198"//要发给谁的ip
int do_download(int cfd,struct sockaddr_in sin);


int main(int argc, const char *argv[])
{		
	//创建套接字 socket
	int cfd=socket(AF_INET,SOCK_DGRAM,0);
	if(cfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success cfd=%d\n",cfd);
	//填充服务器地址信息结构体，给bind函数使用
	//真实的地址信息结构体根据地址族指定，AF_INET：man 7 IP
	struct sockaddr_in sin;
	sin.sin_family =AF_INET;   //必须填AF_INET
	sin.sin_port =htons(PORT);
	sin.sin_addr.s_addr=inet_addr(IP);
	char choose=0;
	while(1)
	{
		system("clear");
		printf("--------------------------\n");
		printf("--                      --\n");
		printf("--       1.下载         --\n");
		printf("--       2.上传         --\n");
		printf("--       3.退出         --\n");
		printf("--                      --\n");
		printf("--------------------------\n");
		printf("请输入>>>>");
		scanf("%c",&choose);
		while(getchar()!=10);//循环吸收垃圾字符，直到吸收到\n字符
		switch(choose)
		{
			case '1':
				do_download(cfd,sin);
				break;
			case '2':
				//do_upload();
				break;
			case '3':
				goto END;
				break;
			default:
				printf("输入错误,重新输入\n");
				break;
		}
		printf("输入任意字符清屏>>>>");
		while(getchar()!=10);
	}
END:
	//关闭套接字
	close(cfd);
	return 0;
}
int do_download(int cfd,struct sockaddr_in sin)
{
	char buf[516]="";
	char filename[20]="";
	printf("请输入要下载的文件名>>>>");
	scanf("%s",filename);
	while(getchar()!=10);
	//组下载请求包
	/*unsigned short*ptr1=(unsigned short*)buf;
	*ptr1=htons(1);//组操作码
	char *ptr2=buf+2;//&buf[2]//(char*)(ptr1+1)
	strcpy(ptr2,filename);//组名字
	char* ptr4=ptr2+strlen(filename)+1;
	strcpy(ptr4,"octet");//组模式
	int size=2+strlen(ptr2)+1+strlen(ptr4)+1;*/
	int size=sprintf(buf,"%c%c%s%c%s%c",0,1,filename,0,"octet",0);
	//发送下载请求sendto
	if(sendto(cfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	printf("sendto success\n");
	int fd=-1;//打开要下载的文件
	struct sockaddr_in rcvaddr;//储存获取到的数据包的地址
	socklen_t addrlen=sizeof(rcvaddr);
	ssize_t res=0;
	int ret_value=0;
	unsigned short num=0;//本地记录的块编号
	//循环接收数据包，回复ACK 
		while(1)
		{
			bzero(buf,sizeof(buf));//清空
			//接收数据 recvfrom ，接收地址信息
			res=recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&addrlen);
			if(res<0)
			{
				ERR_MSG("recvfrom");
				ret_value=-1;
				break;
			}
			if(3==buf[1])//数据包
			{
				//UDO可能会数据重复，为了防止重复收包
				//在本地记录了一下服务器过来的编程块，每次处理前，先判断块编号是否正确
				if(htons(num+1)==*(unsigned short*)(buf+2))
				{
					num++;
					if(-1==fd)
					{
						//第一次循环执行
						fd=open(filename,O_WRONLY|O_CREAT|O_TRUNC,0664);//打开要下载的文件
						if(fd<0)
						{
							ERR_MSG("opem");
							ret_value=-3;
							break;
						}
					}
					if(write(fd,buf+4,res-4)<0)//将获取到的数据写入到文件中
					{
						ERR_MSG("write");
						ret_value=-3;
						break;
					}
					//回复ACK,数据包前4个字节与ACK包基本一致
					//只有操作码不同有效操作码放在buf[1]的位置，直接将buf[1]的3修改为4即可
					buf[1]=4;
					if(sendto(cfd,buf,4,0,(struct sockaddr*)&sin,sizeof(sin))<0)
					{
						ERR_MSG("sendto");
						return -1;
					}
					if(res-4<512)
					{
						printf("文件下载完毕\n");
						break;
					}
				}
			}
			else if(5==buf[1])//错误包
			{
					printf("错误包\n");
					ret_value=-1;
					break;
			}
				
		}
		close(fd);
		return 0;
}