下载
#include<myhead.h>
#define IP "192.168.115.124"
int do_download(int socfd,struct sockaddr_in cin,FILE *fp);
int main(int argc, const char *argv[])
{
FILE *fp;
fp=fopen(argv[1],"w+");
if(fp==NULL)
{
perror("fopen error");
return -1;
}
//创建报式套接字 socket
int socfd=socket(AF_INET,SOCK_DGRAM,0);
if(socfd<0)
{
perror("socket error");
return -1;
}
printf("socket success,socfd=%d\n",socfd);
char buf[256]="";
struct sockaddr_in cin;
cin.sin_family=AF_INET;
cin.sin_port=htons(69);
cin.sin_addr.s_addr=inet_addr(IP);
char choose=0;
while(1)
{
system("clear");
printf("------------------\n");
printf("-----1、下载------\n");
printf("-----2、上传------\n");
printf("-----3、退出------\n");
printf("------------------\n");
printf("请输入>>>");
scanf("%c",&choose);
while(getchar()!=10);
switch(choose)
{
case '1':
do_download(socfd,cin,fp);
break;
case '2':
break;
case '3':
goto END;
break;
default:
printf("输入错误,请从重新输入\n");
break;
}
printf("输入任意字符清屏>>>>");
while(getchar()!=10);
}
END:
fclose(fp);
//关闭套接字,释放资源 close
close(socfd);
return 0;
}
int do_download(int socfd,struct sockaddr_in cin,FILE *fp)
{
char filename[20]="";
printf("请输入要下载的文件名>>");
scanf("%s",filename);
while(getchar()!=10);
//组下载请求包
char buf[516];
unsigned short* ptr1=(unsigned short*)buf;
*ptr1=htons(1);
char* ptr2=buf+2;
strcpy(ptr2,filename); //组名字
char*ptr4=ptr2+strlen(filename)+1;
strcpy(ptr4,"octet"); //组模式
int size=2+strlen(ptr2)+1+strlen(ptr4)+1;
if(sendto(socfd,buf,size,0,(struct sockaddr *)&cin,sizeof(cin))<0)
{
perror("sendto 请求 error");
return -1;
}
printf("sendto download request success\n");
socklen_t addrlen=sizeof(cin);
//发送下载请求
while(1)
{
bzero(buf,sizeof(buf));
//接收数据,接收地址信息
ssize_t a=recvfrom(socfd,buf,sizeof(buf),0,(struct sockaddr *)&cin,&addrlen);
if(a<0)
{
perror("recvfrom error");
return -1;
}
unsigned short* ptr1=(unsigned short*)buf;
if(fwrite((buf+4),(a-4),1,fp)<0)
{
printf("fwrite error\n");
return -1;
}
//组ACK包
*ptr1=htons(4);
size=2+2;
//发送ACK
if(sendto(socfd,buf,size,0,(struct sockaddr *)&cin,sizeof(cin))<0)
{
perror("sendto ACK error");
return -1;
}
printf("sendto download success\n");
//判断是否小于512字节,若小于则下载完成
if((a-4)<512)
{
printf("下载完成\n");
break;
}
}
}