基于UDP的TFTP文件传输

功能：下载、上传、退出

#include <myhead.h>                                   
#define IP "192.168.8.100"                            
#define PORT 69                                       int download_file(int cfd,struct sockaddr_in sin);    
int upload_file(int cfd,struct sockaddr_in sin); int main(int argc, const char *argv[])                
{                                                     //创建套接字文件                                  int cfd=socket(AF_INET,SOCK_DGRAM,0);             if(cfd<0){                                        ERR_MSG("socket");                            return -1;                                    }                                                 printf("socket create success\n");                //填充地址信息体                                  struct sockaddr_in sin;                           sin.sin_family=AF_INET;                           sin.sin_port=htons(PORT);                         sin.sin_addr.s_addr=inet_addr(IP);                //定义输入内容调用的函数                          char choose=0;                                    while(1){                                         printf("--------------------\n");             printf("-------1.下载-------\n");             printf("-------2.上传-------\n");             printf("-------3.退出-------\n");             printf("--------------------\n");             scanf("%c",&choose);                          while(getchar()!='\n'); //当不为\n时循环      switch(choose){                               case '1':                                     download_file(cfd,sin);                   break;                                    case '2':                                     upload_file(cfd,sin);                     break;                                    case '3':                                     goto END;                                 default:                                      printf("输入错误请重新输入\n");           }                                             }                                                 
END:                                                  //关闭文件                                        close(cfd);                                       return 0;                                         
}                                                     

 包的结构：

下载：

1、拼接 读操作 请求体，向服务器发送

2、服务器返回数据包，保存其中的块编号，并通过操作码判断是否是数据包，打印出错误码

3、将数据包的数据写入本地文件中

4、返回ACK包，ACK块编号为数据包的块编号

5、直到数据包小于516时，文件下载完毕

注意：ACK包总共就4个字节，发送给服务器的时候只能发4Bytes的包

//下载
int download_file(int cfd,struct sockaddr_in sin)
{char filename[20]="";printf("请输入需要下载的文件>>>");scanf("%s",filename);while(getchar()!='\n');//组包发送请求char buf[516]="";short* p1=(short*)buf;*p1=htons(1);  //操作码char* p2=buf+2;strcpy(p2,filename);//已经包含了'\0'就是后面那个0char* p3=p2+strlen(p2)+1;strcpy(p3,"octet");int size=2+strlen(p2)+1+strlen(p3)+1;if(sendto(cfd,buf,sizeof(buf),0,\(struct sockaddr*)&sin,sizeof(sin))<0){ERR_MSG("sendto");return -1;}printf("发送下载请求成功\n");//收数据struct sockaddr_in recvin;socklen_t len=sizeof(recvin);ssize_t res=0;short* ack=NULL;char Ack[4]="";int pd=0;//打开要写入的文件pd=open(filename,\O_CREAT|O_WRONLY|0664);while(1){bzero(buf,sizeof(buf));if((res=recvfrom(cfd,buf,sizeof(buf),0,\(struct sockaddr*)&recvin,&len))<0){ERR_MSG("recvfrom");return -1;}printf("res=%ld\n",res);printf("成功接收 [%s:%d] 数据\n",\inet_ntoa(recvin.sin_addr),ntohs(recvin.sin_port));p1=(short*)buf; //服务器返回的操作值ack=p1+1; //服务器返回的数据包编号//	printf("ack=%d\n",ntohs(*ack));if(ntohs(*p1)!=3){ //数据包操作码不对printf("数据包错误\n");if(ntohs(*p1)==5){ //5是报错包printf("ERRNO:%d\n",ntohs(*(p1+1)));}return -1; }else{ //将接收的数据写到文件里if(pd<0 && errno!=17){//	printf("ERR:%d\n",errno);ERR_MSG("open");return -1;}if(write(pd,buf+4,res-4)<0){ERR_MSG("write");return -1;}//返回ack包bzero(Ack,sizeof(Ack));p1=(short*)Ack;*p1=htons(4);*(p1+1)=*ack;if(sendto(cfd,Ack,sizeof(Ack),0,\(struct sockaddr*)&recvin,\sizeof(recvin))<0){ERR_MSG("sendto");return -1;}printf("返回ack包=%d成功\n",ntohs(*ack));}if(res<516){printf("下载完成\n");close(pd);break;}}
}

 

上传：

1、拼接 写操作 请求体，向服务器发送

2、服务器返回ACK包，通过操作码判断是否是ACK包，打印出错误码

3、拼接数据包，数据包编号从1开始ACK块编号是跟着数据包走的

4、将数据从本地文件中读出来，发送给服务器

5、直到从本地读出的数据小于512时，文件上传完毕

注意：用res控制包的大小，更灵活不会传空数据

//上传
int upload_file(int cfd,struct sockaddr_in sin)
{char buf[516]="";char name[100]="";printf("请输入要上传的文件名>>>");scanf("%s",name);while(getchar()!='\n');int pd_r=open(name,O_RDONLY);if(pd_r<0){ERR_MSG("open");return -1;}//组装请求数据包short* p1=(short*)buf;*p1=htons(2);char* p2=buf+2;strcpy(p2,name);char* p3=p2+strlen(name)+1;strcpy(p3,"octet");size_t size=2+strlen(name)+1+strlen(p3)+1;if(sendto(cfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0){ERR_MSG("sendto");return -1;}struct sockaddr_in recvin;socklen_t len=sizeof(recvin);short acknum=1;ssize_t res=0;char ack[4]="";while(1){//接收服务器返回数据//	bzero(buf,sizeof(buf));if(recvfrom(cfd,ack,sizeof(ack),0,\(struct sockaddr*)&recvin,&len)<0){ERR_MSG("recvfrom");return -1;}//	printf("%d\n",ntohs(*(short*)ack)); //4 服务器给的ack包if(ntohs(*(short*)ack)==5){printf("请求错误,ERRNO:%d\n",ntohs(*(short*)ack+1));return -1;}//发文件数据包bzero(buf,sizeof(buf));p1=(short*)buf;*p1=htons(3);*(p1+1)=htons(acknum);res=read(pd_r,buf+4,512);if(sendto(cfd,buf,res+4,0,\(struct sockaddr*)&recvin,len)<0){ERR_MSG("sendto");return -1;}if(res<512){printf("上传完成\n");close(pd_r);break;}acknum++;}
}