tftp客户端实现(一)-发出一条RRQ消息

一、简介

TFTP（Trivial File Transfer Protocol,简单文件传输协议）是TCP/IP协议族中的一个用来在客户机与服务器之间进行简单文件传输的协议，提供不复杂、开销不大的文件传输服务。端口号为69。

二、报文格式

TFTP支持5种类型的包，TFTP数据报的头两个字节为操作码,通过操作码来标识这5种类型的包：

1. Read request(RRQ)
2. Write request(WRQ)
3. Data(DATA)
4. Acknowledgment(ACK)
5. Error(ERROR)

而模式域包括了字符串”netascii”，”octet”或”mail”，名称不分大小写，这里先不一一细说。

三、发送过程

首先，作为tftp客户端，主机A从X端口发送一条RRQ消息（读请求）到tftp服务器监听的69端口。服务器收到这条RRQ消息后就开始往主机A回发文件数据，需注意的是：回发文件数据的端口不再是69，而是服务器上随机的某个端口，图中为Y端口。主机A从本地的X端口接收完服务器的数据后，就会发送一条ACK响应消息给服务器。这就是一次RRQ的请求过程。

四、主要代码：

tftp.h：

#ifndef __TFTPC_H__
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <strings.h>
#include <sys/stat.h>
#include <sys/socket.h>
#include <netdb.h>
#include <sys/ioctl.h>
#include <assert.h>

#define OPCODE_RRQ (1)
#define OPCODE_WRQ (2)
#define OPCODE_DATA (3)
#define OPCODE_ACK (4)
#define OPCODE_ERR (5)

#define BLOCKSIZE (512)

struct TFTPHeader{
    short opcode;
}__attribute__((packed));

struct TFTPWRRQ{
    struct TFTPHeader header;
    char *filename; 
    char *mode; 
}__attribute__((packed));

struct TFTPData{
    struct TFTPHeader header;
    short block;
    char data[];
}__attribute__((packed));

struct TFTPACK{
    struct TFTPHeader header;
    short block;
}__attribute__((packed));

struct TFTPERR{
    struct TFTPHeader header;
    short errcode;
    char *errmsg;   
}__attribute__((packed));
#endif

tftp_client.c：

#include "tftp.h"
#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <string.h>
#include <errno.h>

/************************************************************
FileName:    tftp_client.c
Date:        2017-03-05
Description: Linux Debian系统中，实现tftp协议的RRQ消息的发送，通过WireShark捕获并分析
***********************************************************/

int main(int argc, char **argv)
{
    struct hostent *serv_host;
    struct sockaddr_in serv_addr;

    int sockfd = -1;
    int opt = 1;
    int ret = -1;


    if(argc != 5)       /*判断入口参数是否为5个*/
    {
        fprintf(stderr,"Usage: %s host port [get|put] filename\n", argv[0]);
        exit(1);

    }

    char *host = argv[1];
    int serv_port = atoi(argv[2]);
    int opt_code = strcmp(argv[3], "get") == 0 ? 1 : 2;
    char *filename = argv[4];

    printf("DBG: host = %s , port = %d , opt = %d , filename = %s \n",
    host, serv_port, opt_code, filename);

    /*获取tftp服务器的信息*/
    if((serv_host = gethostbyname(argv[1])) == NULL)
    {
        printf("host is incorrect!\n");
        exit(1);
    }

    serv_addr.sin_family = PF_INET;
    serv_addr.sin_addr = *((struct in_addr *)serv_host->h_addr_list[0]);
    serv_addr.sin_port = htons(serv_port);

    /*获取udp的协议号*/
    struct protoent *proto;
    if((proto = getprotobyname("udp")) == NULL)
    {
        fprintf(stderr,"getprotoname(), failed to find udp protoco info\n");
        exit(1);
    }

    /*创建udp套接字*/
    if((sockfd = socket(PF_INET, SOCK_DGRAM, proto->p_proto)) < 0)
    {
        perror("ERROR opening socket\n");
        exit(1);
    }

    /*构造一条RRQ消息*/
    struct TFTPHeader header;
    header.opcode = htons(OPCODE_RRQ);

    int filenamelen = strlen(filename) + 1;
    int packetsize = sizeof(header) + filenamelen + 5 + 1;     /*5为数据报模式(octet)的长度，1为字符串结束符长度*/

    void *packet = malloc(packetsize);   
    memcpy(packet, &header, sizeof(header));
    memcpy(packet + sizeof(header), filename, filenamelen);

    char *mode = "octet";
    memcpy(packet + sizeof(header) + filenamelen, mode, strlen(mode) + 1);

    if(sendto(sockfd, packet, packetsize, 0,
        (struct sockaddr*)&serv_addr, sizeof(struct sockaddr)) < 0){
        perror("sendto()");
        return 1;
    }
    return 0;
}

五、结果分析：

Linux Debian下执行：

这里以本地主机作为tftp服务器端，1234为服务器监听端口，获取服务器上的testfile文件。

WireShark抓包结果分析：