自己写的一个客户端下载文件程序

1. /******************************************************
   
2. 功能：可以从tftp服务器下载文件
   
3. 命令格式：./tftp 192.168.221.x 文件名
   
4. 时间：2012-04-25
   
5. ******************************************************/
   
6. #include<stdio.h>
   
7. #include<unistd.h>
   
8. #include<stdlib.h>
   
9. #include<arpa/inet.h>
   
10. #include<netinet/in.h>
    
11. #include<sys/socket.h>
    
12. #include<string.h>
    
13. #include<errno.h>
    
14. #include <netdb.h>
    
15. #include<sys/types.h>
    
16. #include<sys/stat.h>
    
17. #include<fcntl.h>
    
18. 
    
19. int main(int argc, char *argv[])
    
20. {
    
21.     int fp;
    
22.     int sockfd = 0;
    
23.     int i=0,fd;
    
24.     unsigned short port = 69;
    
25.     struct sockaddr_in cliaddr;
    
26.     struct sockaddr_in recvddr;    
    
27.     char buf[1024] = "";
    
28.     struct hostent *host;
    
29.     int ret;
    
30.     fp = open(argv[2],O_WRONLY|O_CREAT);    //创建打开文件
    
31.     if((host = gethostbyname(argv[1])) == NULL)        //从输入端获取IP
    
32.     {
    
33.         fprintf(stderr, "Gethostname error\n");
    
34.         exit(1);
    
35.     }        
    
36.     if(argc < 1)
    
37.     {
    
38.         printf("You not input IP and filename!\n");
    
39.         exit(1);
    
40.     }    
    
41.     bzero(&cliaddr, sizeof(cliaddr));
    
42.     cliaddr.sin_family = AF_INET;
    
43.     cliaddr.sin_port = htons(port);
    
44.     cliaddr.sin_addr = *((struct in_addr *)host->h_addr);
    
45.     
    
46.     if((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) == -1)    //创建套接字
    
47.     {
    
48.         perror("socket error\n");
    
49.         exit(1);
    
50.     }    
    
51.     ret = sprintf(buf, "%c%c%s%c%s%c", 0, 1,argv[2], 0, "octet", 0);    //传送命令，请求下载文件
    
52.     sendto(sockfd, buf, ret, 0, (struct sockaddr *)&cliaddr, sizeof(cliaddr));
    
53.     while(1)
    
54.     {
    
55.         char sendbuf[1024] = "";
    
56.         size_t len = 0;
    
57.         unsigned int recvddr_len = sizeof(struct sockaddr);
    
58.         bzero(&sendbuf, sizeof(sendbuf));
    
59.         bzero(&buf, sizeof(buf));
    
60.         len = recvfrom(sockfd, buf, sizeof(buf), 0, (struct sockaddr *)&recvddr, &recvddr_len) ;    
    
61.         if(buf[1] == 5)
    
62.         {
    
63.             printf("%s\n",buf+4);
    
64.             unlink(argv[2]);    //如果服务器没有那个文件，防止创建空文件
    
65.             exit(1);
    
66.         }
    
67.         if(buf[1] == 3)
    
68.         {
    
69.             fd = sprintf(sendbuf, "%c%c%c%c", buf[0], buf[1]+1, buf[2], buf[3]);        //接到数据后发送ACK确认
    
70.             sendto(sockfd, sendbuf, fd, 0, (struct sockaddr *)&recvddr, sizeof(recvddr));
    
71.             if((unsigned short)(i+1) == ntohs(*(unsigned short*)(buf+2)))
    
72.             {
    
73.                 write(fp, (buf+4), (len-4));    //这个地方很重要，大小一定要确定好
    
74.                 i = ntohs(*(unsigned short*)(buf+2));
    
75.                 printf("recv:%d\n",i);
    
76.             }            
    
77.             
    
78.         }        
    
79.         if(len < 516)
    
80.         {
    
81.             printf("Accept finished!\n");
    
82.             close(fp);
    
83.             close(sockfd);
    
84.             exit(1);
    
85.         }
    
86.     }    
    
87.     return 0;
    
88. }