11.21

#include <myhead.h>
//实现下载功能
int do_download(int cfd,struct sockaddr_in sin)
{
	//定义变量存储下载请求包
	char buf[516]="";
	//定义变量存储文件名
	char filename[40]="";
 
	printf("请输入文件名：");
	scanf("%s",filename);
	getchar();
 
	//组装请求包
	short *p1=(short *)buf;
	*p1=htons(1);  //表明要下载
 
	char *p2=buf+2; //文件名段
	strcpy(p2,filename);
 
	char *p4=p2+strlen(p2)+1;//模式段
	strcpy(p4,"octet");
 
	int size=4+strlen(p2)+strlen(p4);//要发送数据的大小
 
	//向服务器发送下载请求
	if(sendto(cfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))==-1)
	{
		perror("sendto error");
		return -1;
	}
	printf("请求成功\n");
 
 
	//循环接收回复服务器发来的消息
    int fd=-1;
	if((fd=open(filename,O_WRONLY|O_CREAT|O_TRUNC,0666))==-1)
	{
		perror("open error");
		return -1;
	}
	struct sockaddr_in cin;
	cin.sin_family=AF_INET;
	socklen_t socklen=sizeof(cin);
	int rs;
	while(1)
	{ 
		bzero(buf,sizeof(buf));	
		if((rs=recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&cin,&socklen))==-1)
		{
			perror("recvfrom error");
			return -1;
		}	
		write(fd,&buf[4],rs-4);		
		buf[1]=4;
		if(sendto(cfd,buf,4,0,(struct sockaddr*)&cin,sizeof(cin))==-1)
		{
			perror("ack sendto error");
			return -1;
		}
		if(rs<516)
			break;
	}
close(cfd);
return 0;
	
}
int main(int argc, const char *argv[])
{
	if(argc!=2)
	{
		printf("input error\n");
		printf("usage:./a.out ip\n");
		return -1;
	}
 
	int cfd=socket(AF_INET,SOCK_DGRAM,0);
	if(cfd==-1)
	{
		perror("socket error");
		return -1;
	}
 
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(69);
	sin.sin_addr.s_addr=inet_addr(argv[1]);
 
	int mune=-1;
	while(1)
	{
		system("clear");
		printf("\t\t=========1、下载========\n");
		printf("\t\t=========2、上传========\n");
		printf("\t\t=========0、退出========\n");
 
		printf("请输入功能:");
		scanf("%d",&mune);
		getchar();
 
		switch(mune)
		{
		case 1:
			{
				do_download(cfd,sin);
			}
			break;
		case 2:
			{
				//do_upload();
			}
			break;
		case 0:
			goto POS;
		default:printf("输入功能有误，请重新输入\n");
		}
		//阻塞
		printf("请输入任意键，按回车清空：");
		while(getchar()!='\n');
	}
POS:
	close(cfd);
	
	return 0;
}