#include <myhead.h>
#define ERR_LOG(msg) do{\
perror(msg);\
printf("%d %s %s\n",__LINE__,__func__,__FILE__);\
}while(0)
#define PORT 69
#define N 516
//实现下载功能
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 do_upload(int cfd,struct sockaddr_in sin)
{
char filename[20]="";
printf("请输入要上传的文件>>>");
fgets(filename,20,stdin);
filename[strlen(filename)-1]='\0';
//判断文件是否存在
int fd=open(filename,O_RDONLY);
if(fd<0)
{
if(errno==ENOENT)
{
printf("文件不存在,请重新输入\n");
return -1;
}
else
{
ERR_LOG("open");
return -1;
}
//发送上传请求
//上传协议
char buf[N]="";
int size = sprintf(buf, "%c%c%s%c%s%c", 0, 2, filename, 0, "octet", 0);
if(sendto(cfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin))<0)
{
ERR_LOG("sendto");
return -1;
}
//循环接收发送数据包
int recv_len;
unsigned short num = 0;
socklen_t addrlen = sizeof(sin);
while(1)
{
bzero(buf, N);
recv_len = recvfrom(cfd, buf, sizeof(buf), 0, (struct sockaddr*)&sin, &addrlen);
if(recv_len < 0)
{
ERR_LOG("recvfrom");
return -1;
}
//操作码的范围是1-5,因为是网络字节序
//所以有效操作吗存储在高位,即buf[1]的位置.
//printf("buf[1] = %d\n", buf[1]); //4 服务器返回应答包
if(4 == buf[1])
{
//判断当前数据包的编号是否等于应答包的编号
//防止数据包在传送过程丢包或重复收包
if(num == ntohs(*(unsigned short*)(buf+2)))
{
//修改操作码为数据包
buf[1] = 3;
//填充块编号
num++;
*(unsigned short*)(buf+2) = htons(num);
//读取数据
//发送数据
int res = read(fd, buf+4, N-4);
if(res < 0)
{
ERR_LOG("read");
return -1;
}
else if(0 == res)
{
printf("-----文件上传完毕-----\n");
break;
}
//发送数据包
//发送的数据包大小为,读取到的字节数(res)+操作码(2byte)+快编号(2byte)
if(sendto(cfd, buf, res+4, 0, (struct sockaddr*)&sin, sizeof(sin)) <0)
{
ERR_LOG("sendto");
return -1;
}
}
else
{
printf("-----文件上传失败,请检查网络环境-----\n");
break;
}
}
else if(5 == buf[1])
{
printf("-----ERROR:%s-----\n", buf+4);
break;
}
}
}
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(cfd,sin);
}
break;
case 0:
goto POS;
default:printf("输入功能有误,请重新输入\n");
}
//阻塞
printf("请输入任意键,按回车清空:");
while(getchar()!='\n');
}
POS:
close(cfd);
return 0;
}
11.21
最新推荐文章于 2023-11-29 13:21:07 发布