12356

 #include <stdio.h>
 #include <sys/types.h>
 #include <sys/socket.h>
 #include <arpa/inet.h>
 #include <netinet/in.h>
 #include <string.h>
 
 #define ERR_MSG(msg) {\
     printf("LINE__%d__\n",__LINE__);\
     printf("FUNC:__%s__\n",__func__);\
     perror(msg);\
 }
 
 #define IP "192.168.3.15"
 #define PORT 69
 
 
 
 int do_download(int cfd, struct sockaddr_in sin, const char* argv)
 {
     char buf[516] = "";
 
     //操作码
     short *ptr1 = (short*)buf;
     *ptr1 = htons(1);
 
     //文件名
     char strname[20] = "";
     printf("输入文件名>>>");
     scanf("%s",strname);
     getchar();
     char *ptr2 = buf + 2;
     strcpy(ptr2, strname);
 
     //模式octet
     char* ptr3 = ptr2 + strlen(strname) + 1;
     strcpy(ptr3, "octet");
 
     int len = 2 + strlen(strname) + 1 + strlen("octet") + 1;
     if(sendto(cfd, buf, len, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
     {
         ERR_MSG("sendto");
         return -1;
     }
     printf("下载请求发送成功\n");
 
 
     //开一个文件用来保存数据包中的数据
     fopen(argv, "w");
     printf("文件已经清空\n");
     FILE* fp = fopen(argv, "a");
     if(fp == NULL)
     {
         ERR_MSG("fopen");
         return -1;
     }
 
     int res = 0, number = 0;
     struct sockaddr_in cin;
     socklen_t addrlen = sizeof(cin);
     short int arr[2] = {0};
 
     while(1)
     {
         bzero(buf, sizeof(buf));
         //接受来自服务器的数据包
         res = recvfrom(cfd, buf, sizeof(buf), 0, (struct sockaddr*)&cin, &addrlen);
         if(res < 0)
         {
             ERR_MSG("recvfrom");
             return -1;
         }
 
         arr[0] = htons(4);
         arr[1] = htons(number+1);
         number++;
 
 
 
         for(int i = 4; i < 516; i++)
         {
             fprintf(fp, "%c", buf[i]);
             printf("%c",buf[i]);
         }
 
         //sendto(cfd, arr, sizeof(arr), 0, (struct sockaddr*)&sin, sizeof(sin));
         printf("res = %d\n",res);
 
         if(res < 514)
         {
             printf("[%s:%d]数据传输完毕\n",inet_ntoa(cin.sin_addr), ntohs(cin.sin_port));
             break;
         }
 
 
 
     }
 
     fclose(fp);
     return 0;
 }
 
 
 
 int main(int argc, const char *argv[])
 {
     //创建套接字
     int cfd = socket(AF_INET, SOCK_DGRAM, 0);
     if(cfd< 0)
     {                                                                                                                                                                                 
         ERR_MSG("socket");
         return -1;
     }
 
 
 
     struct sockaddr_in sin;
     sin.sin_family      = AF_INET;
     sin.sin_port        = htons(PORT);
     sin.sin_addr.s_addr = inet_addr(IP);
     //功能选择
     while(1)
     {
         printf("----------1. 下载---------\n");
         printf("----------2. 上传---------\n");
         printf("----------3. 退出---------\n");
         printf("请输入>>>");
         int choice = 0;
         scanf("%d",&choice);
         getchar();
         switch(choice)
         {
         case 1:
             do_download(cfd, sin, argv[1]);
             break;
         case 2:
         //  do_upload();
             break;
         case 3:
             goto QWE;
             break;
         default:
             printf("输入错误,重新输入\n");
             break;
         }
     }
 
 QWE:
     return 0;
 }