3.14作业

m0_65619859

已于 2023-03-14 19:40:04 修改

阅读量35

点赞数

文章标签：网络 Powered by 金山文档

于 2023-03-14 19:36:57 首次发布

本文链接：https://blog.csdn.net/m0_65619859/article/details/129538247

版权

该代码示例展示了如何使用UDP套接字实现TFTP协议进行文件下载。用户可以选择下载、上传或退出。程序发送下载请求，接收数据包并将其写入本地文件，同时发送ACK确认。如果遇到错误包，程序会终止并显示错误信息。

摘要由CSDN通过智能技术生成

TFTP下载

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>

#define ERR_MSG(msg) do{\
    fprintf(stderr, "line:%d ", __LINE__);\
    perror(msg);\
}while(0)

#define PORT 69               
#define IP "192.168.31.123"

int do_download(int sfd, struct sockaddr_in sin);
int main(int argc, const char *argv[])
{
    //创建报式套接字
    int sfd=socket(AF_INET,SOCK_DGRAM);
    if(sfd<0)
    {
         ERR_MSG("socket");
         return -1;
    }
    //填充服务器的地址信息结构体
    //供给下面的sendto使用
    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");
        printf("请输入>>> ");
        choose=getchar();
        while(getchar()!=10);
        
        switch(choose)
        {
        case '1':
            do_download(sfd,sin);
            break;
        case '2':
            break;
        case '3':
            goto END;
            break;
        default:
            printf("输入错误，请重新输入\n");
            break;
        }
    }
END:
   //关闭所有文件描述符
   close(sfd);
   return 0;
}

int do_download(int sfd,struct sockaddr_in sin)
{
   int ret_val=0;
   char buf[516]="";
   
   char filename[20]="";
   printf("请输入要下载的文件名\n");
   scanf("%s",filename);
   while(getchar()!=10);
   
   //发送下载请求
   //组协议包
   int size=sprintf(buf,"%c%c%s%c%s%c",0,1,filename,0,"octet",0);
   if(sendto(sfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0)
   {
       ERR_MSG("sendto");
       return -1;
   }
   
   //本地创建并打开要下载的文件
   int fd=-1;     //初始化
   
   socklen_t addrlen=sizeof(sin);
   ssize_t res=0;
   unsigned short num=0;
   
   //循环接收数据包，回复ACk
   while(1)
   {
       bzero(buf,sizeof(buf));
       //接收数据包
       res=recvfrom(sfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&addrlen);
       if(res<0)
       {
           ERR_MSG("recvfrom");
           return -1;
           break;
       }
       if(3==buf[1])   //数据包
       {
           //防止重复收包
           if(htons(num+1)==*(unsigned short*)(buf+2))
           {
               num++;
               
               if(-1==fd)
               {
                   fd=open(filename,O_WRONLY|O_CREAT|O_TRUNC,0664);
                   if(fd<0)
                   {
                       ERR_MSG("open");
                       ret_val=-3;
                       break;
                   }
               }
               //将获取到的数据写入到文件中
               if(write(fd,buf+4,res-4)<0)
               {
                   ERR_MSG("write");
                   ret_val=-3;
                   break;
               }
               //回复ACK
               buf[1]=4;
               if(sendto(sfd,buf,4,0,(struct sockaddr*)&sin,sizeof(sin))<0)
               {
                   ERR_MSG("sendto");
                   ret_val=-1;
                   break;
               }
               
               if(res<516)
               {
                   printf("=======文件下载完毕=======\n");
                   break;
               }
           }
       }
       else if(5==buf[1])    //错误包
       {
           printf("MSG_ERR:code[%d] msg[%s] __%d__\n",\
                   ntohs(*(unsigned short*)(buf+2)),buf+4,__LINE__);
           ret_val=-1;
           break;
       }
   }
   close(fd);
   return ret_val;
}