简单的TFTP客户端

   上一篇博文已经详细介绍tftp协议了，下边联系写一个简单的tftp客户端，首先tftp客户端与服务端通过udp通信，udp不是可靠的通信服务，为了保持通信中数据的可靠传输，我们采用简单的确认机制，服务端对每一个数据报文从01开始编号【0号报文为系统保留】，客户端收到数据报文后，发送确认报文，确认报文应该包含服务器发送报文的编号，这样，当服务端收到确认报文并检查确认数据报文编号，如果是以前发送的最新报文则发送下一个报文，这里需要注意的是，如果服务端在一定时间内没有收到确认报文，则会发生超时，具体超时处理看具体的服务端，一般的默认处理是重新发送报文，当超时次数达到一定的数字，则退出，客户端处理是如果没有收到数据报文，则发送最新收到数据报文的确认报文，如果在一定时间没有收到，则发生超时，通常的超时处理是重新发送最新收到数据报文的确认报文，如果超时次数超过一定的数字，则退出。在tftp通信过程，可以随时的中断通信。这就是tftp通信的整个过程。下边是自己写的一个简单的客户端，没有加入超时机制，只是简单的数据接受。

• main.c  main.h    
• socket_init.c  socket_init.h //建立网络连接，
• file_save.c file_save.c  //文件传输
  
  测试命令格式：　　cmd 192.168.220.63 xxx.xxx


MAIN.H


#ifndef __MAIN_H__
#define __MAIN_H__
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <fcntl.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/select.h>
#include <sys/time.h>
#endif
#include "main.h"
#include "socket_init.h"
#include "file_save.h"
// tftp ip file (space)



MAIN.C                   //命令格式：  cmd  ip  filename 

int main(int argc,char *argv[])//argv[1]:ip地址 argv[2]:传输的文件名
{    
    int fd;
    fd=socket_init(argc,argv,69);//初始化网络，建立网络连接，返回socket_fd 
  //默认的连接端口为69
    file_save(argc,argv,69,fd);//进行UDP通信，保存文件
    return 1;
    
}    


网络初始化：

 

socket_init.h

 

#ifndef __SOCKET__
#define __SOCKET__
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <fcntl.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/select.h>
#include <sys/time.h>
extern int socket_init(int argc,char *argv[],unsigned short int port);
#endif

socket_init.c

#include "socket_init.h"
int socket_init(int argc,char *argv[],unsigned short int port)
{    
    int sock_fd;
    char request[512];//请求报文
    int len;
    char rec_buf[512];//接受缓存
    unsigned long ip;
    struct sockaddr_in tftp_addr;
    bzero(&tftp_addr,sizeof(struct sockaddr_in));
    bzero(rec_buf,512);
    bzero(request,512);
    //inet_pton(AF_INET,argv[1],&tftp_addr.sin_addr.s_addr);

    inet_pton(AF_INET,argv[1],&ip);//字符串ip转为二进制ip存入ip变量
    printf("ip=%d\n",ip);
    tftp_addr.sin_addr.s_addr=htonl(ip);//大端转换
    tftp_addr.sin_family=AF_INET;//ipv4
    tftp_addr.sin_port=htons(port);
    tftp_addr.sin_addr.s_addr=htonl(tftp_addr.sin_addr.s_addr); // ip

    sprintf(request," %s %s",argv[2],"octet");//字符串格式输出到request
    request[0]=0;
    request[1]=1;
    request[2+strlen(argv[2])]=0;
    request[2+strlen(argv[2])+5+1]=0;//请求报文
    puts(request);
    if( (sock_fd=socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP))<0)
    {
        perror("socket error");
        exit(-1);
    }    
    len=sizeof(struct sockaddr_in);
    sendto(sock_fd,request,2+5+2+strlen(argv[2]),0,(struct sockaddr *)&tftp_addr,len);//2+5+2+strlen(argv[2]):没有好的计算大小的方法，因为在request中存在多个'\0'，不能使用strlen(),而使用sizeof()，输入宏操作，只能返回request数组的大小，同样不能知道实际的大小，通过对request报文格式的观察，发现操作码占2bytes，模式占5bytes，'\0'有两个，占2个byets，在加入文件名大小strlen(argv[2])就可以获取request报文的实际长度了。
    puts("udp init ok ");
    return sock_fd; //返回网络描述符，实际代表一个网络连接
}    

文件传输保存：

 

 

file_save.h

 

#ifndef __FILE_H__
#define __FILE_H__
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <fcntl.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/select.h>
#include <sys/time.h>

extern int file_save(int argc,char *argv[],unsigned short int port,int fd);
#endif


fille.c


#include "file_save.h"
int file_save(int argc,char *argv[],unsigned short int port,int socket_fd)
{    
    int fd;
    char *p;
    char rec_buf[550]; //接受数据报文缓存
    int len;
    char ack_buf[4];//应答报文
    unsigned long ip;
    int len_addr;
    struct sockaddr_in tftp_addr,rev_addr;//本地客户端信息
    struct sockaddr tmp_addr;//存储服务端信息
    bzero(&tftp_addr,sizeof(struct sockaddr_in));
    bzero(rec_buf,550);
    bzero(ack_buf,4);
    inet_pton(AF_INET,argv[1],&ip);
    tftp_addr.sin_family=AF_INET;
    tftp_addr.sin_port=htons(port);
    tftp_addr.sin_addr.s_addr=htonl(ip); // ip

    if( (fd=open(argv[2],O_CREAT|O_RDWR))<0 )
    {
        perror("file open error");
        exit(1);
    }    
    bzero(&rev_addr,sizeof(struct sockaddr_in));
    len_addr=sizeof(struct sockaddr_in);
    //while( ( len=recvfrom(socket_fd,rec_buf,512,0,(struct sockaddr *)&rev_addr,&len_addr) )==512 )

    while( ( len=recvfrom(socket_fd,rec_buf,516,0,&tmp_addr,&len_addr) )==516 )
    {    
        write(fd,&rec_buf[4],512);//写入文件，除去报文额外信息
        ack_buf[0]=0;
        ack_buf[1]=4;//应答报文操作码
        ack_buf[2]=rec_buf[2];
        ack_buf[3]=rec_buf[3];//获得报文编号  
        bzero(rec_buf,550);
      //sendto(socket_fd,ack_buf,4,0,(struct sockaddr *)&tftp_addr,len_addr);
        sendto(socket_fd,ack_buf,4,0,&tmp_addr,len_addr);//发送应答报文
    }
    write(fd,&rec_buf[4],len-4);//写入文件，除去报文额外信息。
    ack_buf[0]=0;
    ack_buf[1]=4;//应答报文操作码
    ack_buf[2]=rec_buf[2];
    ack_buf[3]=rec_buf[3];//获取报文编号
  //    sendto(socket_fd,ack_buf,4,0,(struct sockaddr *)&tftp_addr,len_addr);
    sendto(socket_fd,ack_buf,4,0,&tmp_addr,len_addr);//发送应答报文
    close(fd);
    return 0;
}    

上边是一个简单的tftp的客户端，没有加入超时机制，上边的程序都是调试通过的，没有问题的，下边总结下在编写过程中，犯的一些错误，总结下，先看代码吧：

错误一：

有逻辑错误的代码：

if( sock_fd=socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP)<0)
    {
        perror("socket error");
        exit(-1);
    }  
  
没有错误的代码：

if( (sock_fd=socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP))<0)
    {
        perror("socket error");
        exit(-1);
    }    

sock_fd=socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP)<0 这个逻辑错误比较隐蔽， '=' 的优先级要小与'<'的优先级，那么执行的结构就是socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP)<0这个逻辑式比较的结构存入到sock_fd,而不是把sock（）的结构存入到socket_fd中，在求socket_fd<0逻辑式的结构。这个错误的由来是对编写代码不够仔细，对与这个错误还是比较熟悉，往往在写代码时遗漏（），这样加入（）后就正常，同时这种类型的逻辑错误编译器是不会报错的，也就是编译是没有问题的，往往是程序运行的结构不对。需要特别的注意逻辑错误。

错误二：

struct sockaddr_in tftp_addr,rev_addr;
struct sockaddr tmp_addr;

while( ( len=recvfromsocket_fd,rec_buf,516,0,&tmp_addr,&len_addr) )==512 )
    {    
        write(fd,&rec_buf[4],512);//写文件
        ack_buf.... //处理应答报文，代码省略
        tftp_addr.sin_port=htons(rev_addr.sin_port);
       //sendto(socket_fd,ack_buf,4,0,(struct sockaddr *)&tftp_addr,len_addr);
        sendto(socket_fd,ack_buf,4,0,&tmp_addr,len_addr);
    }

recvfromsocket_fd,rec_buf,512,0,&tmp_addr,&len_addr)这条语句会将服务端的信息存入到tmp_addr中，这里的tmp_addr是struct sockaddr 类型的，而struct sockaddr_in类型和struct sockaddr是同大小的，可以之间进行安全的赋值转换，如果tmp_addr是struct sockaddr_in类型的，在recvfrom中需要类型转换(struct sockaddr *),如果不进行格式的转换，编译器换警告，在函数内部会进行自动的类型转换，在发送的时候在使用这个tmp_addr时，tmp_addr内部的转矩其实已经经过两次的类型转换，先从struct sockaddr_in转换为struct sockaddr,在从struct sockaddr 转换为struct sockaddr_in,内部的数据就会发生移位，实际的服务端返回的端口号。 比较安全的方法是使用sruct sockaddr类型，不进行类型的转换。