Tooltron

#include <packet.h>

/**

 * @brief Packet handler states

 * @param sd Looking for a start delimiter

 * @param src Looking for a source

 * @param dest Looking for a destination

 * @param comd Looking for a command

 * @param read Reading data payload

 * @param cs Calcualting the checksum and returning

 */

typedef enum {

    sd,   

    src,  

    dest, 

    comd, 

    read, 

    cs    

} state_t;

/**

 * @brief Parses a tooltron packet

 * 

 * This is a state machine that parses the packet. It uses a software 

 * counter to track timeout status. The timeout is not reset unless

 * a fully valid packet is found. It will fail if there is a timeout,

 * crc error, or any error in the packet. All data will be placed in

 * mbuf. Note that mbuf must be able to handle PROGD_PACKET_SIZE

 * bytes.

 *

 * @pre mbuf is PROGD_PACKET_SIZE bytes large

 * @param mbuf Buffer to write the packet data payload to if there is one

 * @param addr The address of the sending node

 * @return the command received or TT_BAD on any error

 */

char parse_packet(uint8_t *mbuf, uint8_t addr) {

  uint8_t r;        // Byte from the network

  uint8_t crc;      // Running checksum of the packet

  uint8_t cmd;      // The command received

  uint8_t pos;      // Position in the message buffer

  uint8_t lim;      // Max number of bytes to read into the message buf

  state_t state;    // State machine

  uint16_t count;

  r = 0;

  crc = 0;

  cmd = 0;

  pos = 0;

  lim = 0;

  state = sd;

  count = 0;

  while (1) {

    // Wait for the next byte

    while ((rs485_get_byte(&r)) < 0) {

        if (count >= MAX_TIMEOUT) {

            return TT_BAD;

        } else {

            count++;

        }

    }

    switch (state) {

        case sd:

            if (r == DELIM) {

                state = src;

            }

            break;

        case src:

            if (r == DELIM) {

                state = src;

            } else {

                crc = r;

                state = dest;

            }

            break;

        case dest:

            if (r == DELIM) {

                state = src;

            } else if (r == addr) {

                crc ^= r;

                state = comd;

            } else {

                state = sd;

            }

            break;

        case comd:

            cmd = r;

            crc ^= r;

            if (r == DELIM) {

                state = src;

            } else if (r == TT_PROGM) {

                lim = PROGM_PACKET_SIZE;

                state = read;

            } else if (r == TT_PROGD) {

                lim = PROGD_PACKET_SIZE;

                state = read;

            } else if (r == TT_SEND_KEY) {

                lim = SEND_KEY_PACKET_SIZE;

                state = read;

            } else {

                state = cs;

            }

            break;

        case read:

            mbuf[pos] = r;

            crc ^= r;

            pos++;

            if (pos == lim) {

                state = cs;

            }

            break;

        case cs:

            if (r == crc) {

              return cmd;

            } else {

                return TT_BAD;

            }

            break;

        default:

            return TT_BAD;

    }

  }

}

/**

 * @brief Sends a packet of type cmd onto the network

 * @param cmd The command to send

 * @param addr The address of the sending node

 */

void send_packet(uint8_t cmd, uint8_t addr) {

    rs485_send_byte(DELIM);

    rs485_send_byte(addr);

    rs485_send_byte(SERVER);

    rs485_send_byte(cmd);

    rs485_send_byte(addr ^ SERVER ^ cmd);

}

/**

 * @brief Sends a packet of type cmd onto the network

 * @param cmd The command to send

 * @param addr The address of the sending node

 * @param data Pointer to data

 * @param len Number of bytes to send

 */

void send_packet_data(uint8_t cmd, uint8_t addr, uint8_t* data, uint8_t len) {

    uint8_t i;

    uint8_t crc = 0;

    rs485_send_byte(DELIM);

    rs485_send_byte(addr);

    rs485_send_byte(SERVER);

    rs485_send_byte(cmd);

    for (i = 0; i < len; i++) {

        crc  ^= data[i];

        rs485_send_byte(data[i]);

    }

    rs485_send_byte(addr ^ SERVER ^ cmd ^ crc);

}