代码:
#include <myhead.h>
int do_download(int cfd,struct sockaddr_in sin)
{
//定义字符数组
char buf[516]="";
char filename[40]="";
//得到文件名
printf("please enter filename:");
scanf("%s",filename);
getchar();
short *p1=(short *)buf;
//将下载操作码转化为网络字节序
*p1=htons(1);
char *p2=buf+2;
//将文件名放入buf中
strcpy(p2,filename);
char *p3=p2+strlen(p2)+1;
//将模式放入buf中
strcpy(p3,"octet");
int size=4+strlen(p2)+strlen(p3);
//向服务器发送下载请求
if(sendto(cfd,buf,size,0,(struct sockaddr *)&sin,sizeof(sin))==-1){
perror("sendto error");
return -1;
}
//以只写的形式打开文件
int fd=-1;
if((fd=open(filename,O_WRONLY|O_CREAT|O_TRUNC,0664))==-1){
perror("open error");
return -1;
}
socklen_t socklen=sizeof(sin);
while(1){
bzero(buf,sizeof(buf));
//接收服务器发来的数据包
if(recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr *)&sin,&socklen)==-1){
perror("sendto error");
return -1;
}
int len=strlen(buf+4);
//判断数据包中数据的大小,小于512则结束
//将数据包中的数据放入文件中
write(fd,buf+4,len);
//向服务器发送ACK
short *pb=(short *)buf;
*pb=htons(4);
if(sendto(cfd,buf,4,0,(struct sockaddr *)&sin,sizeof(sin))==-1){
perror("sendto error");
return -1;
}
if(len<512){
close(cfd);
close(fd);
break;
}
}
return 0;
}
int do_upload(int cfd,struct sockaddr_in sin)
{
//定义字符数组
char buf[516]="";
char filename[40]="";
//得到文件名
printf("please enter filename:");
scanf("%s",filename);
getchar();
short *p1=(short *)buf;
//将上传操作码转化为网络字节序
*p1=htons(2);
char *p2=buf+2;
//将文件名放入buf中
strcpy(p2,filename);
char *p3=p2+strlen(p2)+1;
//将模式放入buf中
strcpy(p3,"octet");
int size=4+strlen(p2)+strlen(p3);
//向服务器发送上传请求
if(sendto(cfd,buf,size,0,(struct sockaddr *)&sin,sizeof(sin))==-1){
perror("sendto error");
return -1;
}
//以只读的形式打开文件
int fd=-1;
if((fd=open(filename,O_RDONLY))==-1){
perror("open error");
return -1;
}
socklen_t socklen=sizeof(sin);
int n=1;
while(1){
bzero(buf,sizeof(buf));
//接收服务器发来的ACK
if(recvfrom(cfd,buf,4,0,(struct sockaddr *)&sin,&socklen)==-1){
perror("sendto error");
return -1;
}
short *pb=(short *)buf;
*pb=htons(3);
buf[3]=n;
//从文件中取数据
int res=read(fd,buf+4,512);
//向服务器发送数据包
if(sendto(cfd,buf,516,0,(struct sockaddr *)&sin,sizeof(sin))==-1){
perror("sendto error");
return -1;
}
//判断文件中的数据是否读完
n++;
if(res<512){
close(cfd);
close(fd);
break;
}
}
return 0;
}
int main(int argc, const char *argv[])
{
//判断终端执行输入是否正确
if(argc!=2){
printf("input error\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 muex;
while(1){
system("clear");
printf("\t\t========1.下载=========\n");
printf("\t\t========2.上传=========\n");
printf("\t\t========3.清空=========\n");
printf("please enter muex:");
scanf("%d",&muex);
switch(muex){
case 1:{
do_download(cfd,sin);
printf("下载成功\n");
break;
}
case 2:{
do_upload(cfd,sin);
printf("上传成功\n");
break;
}
case 3:goto EXIT;
default:printf("enter error");
}
printf("输入任意建,按回车清空:");
while(getchar()!='\n');
}
close(cfd);
EXIT:
return 0;
}
运行结果: