Super lightweight, not at all robust, TFTP server for FRDM-K64F eval board.

Dependencies:   EthernetInterface mbed-rtos mbed

Fork of FRDM_K64F-Ethernet by Rangel Alvarado

Super lightweight, not at all robust, TFTP server for FRDM-K64F eval board. This tool supports read-only access to two files. It does not support NACK responses or timeouts. The tool is intended for use by our test department to check out Ethernet functionality on our main processor board.

main.cpp

Committer:
audim
Date:
2018-10-11
Revision:
2:28ddb9b073ec
Parent:
1:2944c0d494ff

File content as of revision 2:28ddb9b073ec:

/*
** Super lightweight, not at all robust, TFTP server for FRDM-K64F eval board.
** This tool supports read-only access to a couple of files. It does not support
** NACK responses or timeouts. The tool is intended for use by our test department
** to check out Ethernet functionality on our main processor board.
**
** 11 October 2018
** - am
*/

#include "mbed.h"
#include "EthernetInterface.h"
#include "string.h"

#include "env.h"

#define IP_ADDR     "192.168.1.10"
#define NET_MASK    "255.255.255.0"
#define GATEWAY     "192.168.1.1"
#define TFTP_PORT   69

#define LED_ON  0
#define LED_OFF 1

// status LEDs
DigitalOut led_error(LED1);     // red
DigitalOut led_waiting(LED2);   // green
DigitalOut led_connected(LED3); // blue

// TFTP opcodes
enum opcode { RRQ=1, WRQ, DATA, ACK, ERROR };

// TFTP packet structure
typedef union {

    uint16_t opcode;

    struct {
        uint16_t opcode; // RRQ or WRQ
        char filename_and_mode[512];
    } request;

    struct {
        uint16_t opcode; // DATA
        uint16_t block_number;
        char data[512];
    } data;

    struct {
        uint16_t opcode; // ACK
        uint16_t block_number;
    } ack;

    struct {
        uint16_t opcode; // ERROR
        uint16_t error_code;
        char error_string[512];
    } error;

} tftp_packet_t;

int tftp_send_error(UDPSocket socket, Endpoint end_point, int error_code, char *error_string) {
    
    tftp_packet_t packet;
    char *packet_ptr = (char *)&packet;

    packet.opcode = htons(ERROR);
    packet.error.error_code = htons(error_code);
    strcpy(packet.error.error_string, error_string);

    return socket.sendTo(end_point, packet_ptr, 4 + strlen(error_string)); // 4 = sizeof(opcode + error_code)
}

int main (void) {

    // Ethernet infrastructure
    EthernetInterface eth;
    UDPSocket tftp_server;
    Endpoint client;

    // data packet storage    
    tftp_packet_t packet;
    char *packet_ptr = (char *)&packet;

    // assign ip address, net mask and gateway to Ethernet interface
    eth.init(IP_ADDR, NET_MASK, GATEWAY);
    eth.connect();
    
    // main loop    
    while (true) {

        // clear status LEDs
        led_error = LED_OFF;
        led_waiting = LED_OFF;
        led_connected = LED_OFF;
        wait(1); // one second

        // wait for request
        led_waiting = LED_ON;
        tftp_server.bind(TFTP_PORT);
        tftp_server.receiveFrom(client, packet_ptr, sizeof(packet));
        led_waiting = LED_OFF;

        // if it's a read request
        if (ntohs(packet.opcode) == RRQ) {
            // and it's for the environment file
            if (!strncmp(packet.request.filename_and_mode, "/production/u-boot-env_txt", 26)) {
                // send file
                int i_block = 1;
                int i_string = 0;
                int i_char = 0;
                int i_buff = 512;
                while (i_buff == 512) {
                    led_connected = LED_ON;
                    i_buff = 0;
                    packet.opcode = htons(DATA);
                    packet.data.block_number = htons(i_block++);
                    do {
                        packet.data.data[i_buff++] = env_string[i_string][i_char++];
                        // check for end of string
                        if (env_string[i_string][i_char] == '\0') {
                            i_char = 0;
                            i_string++;
                        }
                    } while ((i_buff < 512) && (env_string[i_string][0] != '\0'));
                    tftp_server.sendTo(client, packet_ptr, 4 + i_buff); // 4 = sizeof(opcode + block_number)
                    led_connected = LED_OFF;
                    led_error = LED_ON;
                    tftp_server.receiveFrom(client, packet_ptr, sizeof(packet)); // wait for ACK
                    led_error = LED_OFF;
                };
            }
            // or the binary test file
            else if (!strncmp(packet.request.filename_and_mode, "/production/ethtest_bin", 23)) {
                // send a 4M byte file (8192 512-byte blocks)
                for (int block = 1; block < 8194; block++) { // blocks 1-8192 are full, block 8193 is empty (end of file)
                    led_connected = LED_ON;
                    packet.opcode = htons(DATA);
                    packet.data.block_number = htons(block);
                    // fill packet with Gray code
                    for (uint16_t w=0; w<256; w++) {
                        packet.data.data[w] = (char)((w>>1)^w);
                        packet.data.data[256+w] = ~packet.data.data[w];
                    }
                    tftp_server.sendTo(client, packet_ptr, (block<8193) ? 516 : 4); // 4 = sizeof(opcode + block_number)
                    led_connected = LED_OFF;
                    led_error = LED_ON;
                    tftp_server.receiveFrom(client, packet_ptr, sizeof(packet)); // wait for ACK
                    led_error = LED_OFF;
                }
            }
            // error, we only support a couple of files
            else {
                led_error = LED_ON;
                tftp_send_error(tftp_server, client, 0, "file not found, this board only has u-boot-env_txt and ethtest_bin");
                wait(3); // three seconds
                led_error = LED_OFF;
            }
        }
        // error, we only support read requests
        else {
            led_error = LED_ON;
            tftp_send_error(tftp_server, client, 0, "this test board only supports read requests");
            wait(3); // three seconds
            led_error = LED_OFF;
        }

        led_connected = LED_ON;
        led_waiting = LED_ON;
        wait(1); // one second
        tftp_server.close();
    }
}