Colony

/* testbench for Lab 2 first checkpoint

 * determine whether robot under test complies with communication standard

 * to be conducted using XBee USB dongle

 */

/* The tests shall be as follows

 * 1. Receive a TAG, send an ACK

 * - Robot passes if it sends a tag and changes to prey state

 * 2. Send a TAG, receive an ACK

 * - Pass if sends ACK, changes to wait state, then hunter state

 * 3. Receive TAG, send slightly delayed (< 1s) ACK

 * - Pass if sends one and only one TAG packet (and changes state appropriately)

 * 4. Receive TAG, do not send ACK

 * - Pass if sends one and only one TAG packet (and does not change state)

 * 5. Receive TAG, send ACK to incorrect robot

 * - Pass if goes to wait state, then back to hunter state

 * 6. Simulate TAG from robot A to B, ACK from B to A

 * - Pass if ignores TAG, goes to wait then hunter in response to ACK

 */

#include <stdlib.h>

#include <stdio.h>

#include <unistd.h>

#include "../../libwireless/lib/wl_basic.h"

#include "../../libwireless/lib/wireless.h"

#include "hunter_prey.h"

#define CHANNEL 0xF // channel for wireless communication

#define TYPE 42        // packet type for wireless communication

#define ROBOTID 255 // make up a robot id because the PC doesn't have one

int main(int argc, char *argv[])

{

    char send_buffer[2];    // holds data to send

    int data_length;        // length of data received

    unsigned char *packet_data;        // data received

    // set up wireless

    wl_basic_init_default();

    wl_set_channel(CHANNEL);

    wl_set_com_port("/dev/ttyUSB0");

    printf("Testing communications\n\n");

    // Receive TAG, send ACK

    printf("Receive TAG, send ACK... ");

    fflush(stdout);

    // Wait until we receive a packet

    while (!(packet_data = wl_basic_do_default(&data_length)));

    if (data_length > 2)

    {

        printf("Excessive TAG packet length... ");

        fflush(stdout);

    }

    if (data_length >= 2 && packet_data[0] == HUNTER_PREY_ACTION_TAG)

    {

        // send back an ACK

        send_buffer[0] = HUNTER_PREY_ACTION_ACK;

        send_buffer[1] = packet_data[1];

        wl_basic_send_global_packet(TYPE, send_buffer, 2);

        printf("PASSED\n");

    }

    else

        printf("FAILED\n");

    // Send a TAG, receive an ACK

    printf("Send TAG, wait for ACK... ");

    fflush(stdout);

    send_buffer[0] = HUNTER_PREY_ACTION_TAG;

    send_buffer[1] = ROBOTID;

    // robot number stays the same

    wl_basic_send_global_packet(TYPE, send_buffer, 2);

    // Wait for an ACK

    usleep(800000);  // wait for 800 ms before sending ACK

    packet_data = wl_basic_do_default(&data_length);

    // Check ACK for presence and correctness

    if (!packet_data || data_length < 2 ||

            packet_data[0] != HUNTER_PREY_ACTION_ACK ||

            packet_data[1] != ROBOTID)

        printf("FAILED\n");

    else

        printf("PASSED\n");

    // Receive a TAG, send a delayed ACK

    printf("Receive a TAG, send a delayed ACK... ");

    fflush(stdout);

    while (!(packet_data = wl_basic_do_default(&data_length)));

    if (data_length >= 2 && packet_data[0] == HUNTER_PREY_ACTION_TAG)

    {

        // wait before sending ACK back

        usleep(900000);

        if (wl_basic_do_default(&data_length))

            printf("FAILED\n");

        else

        {

            // send packet

            send_buffer[0] = HUNTER_PREY_ACTION_ACK;

            send_buffer[1] = packet_data[1];

            wl_basic_send_global_packet(TYPE, send_buffer, 2);

            printf("PASSED\n");

        }

    }

    else

        printf("FAILED\n");

    // Receive a TAG, never send an ACK

    printf("Receive TAG, never send ACK... ");

    fflush(stdout);

    while (!(packet_data = wl_basic_do_default(&data_length)));

    if (data_length >= 2 && packet_data[0] == HUNTER_PREY_ACTION_TAG)

    {

        usleep(5000000);    // wait 5 seconds to see if they TAG again

        if (wl_basic_do_default(&data_length))

            printf("FAILED\n");

        else

            printf("PASSED\n");

    }

    else

        printf("FAILED\n");

    // Receive TAG, send ACK to incorrect robot

    printf("Receive TAG, send ACK to incorrect robot... ");

    fflush(stdout);

    // Wait until we receive a packet

    while (!(packet_data = wl_basic_do_default(&data_length)));

    if (data_length >= 2 && packet_data[0] == HUNTER_PREY_ACTION_TAG)

    {

        // send back an ACK

        send_buffer[0] = HUNTER_PREY_ACTION_ACK;

        send_buffer[1] = packet_data[1] + 1;

        wl_basic_send_global_packet(TYPE, send_buffer, 2);

        printf("PASSED\n");

    }

    else

        printf("FAILED\n");

    // Simulate TAG from robot A to B, ACK from robot B to A

    printf("Send TAG from robot A to B, ACK from B to A... ");

    fflush(stdout);

    // TAG

    send_buffer[0] = HUNTER_PREY_ACTION_TAG;

    send_buffer[1] = packet_data[1] - 1;    // robot other than testee

    wl_basic_send_global_packet(TYPE, send_buffer, 2);

    // ACK

    send_buffer[0] = HUNTER_PREY_ACTION_ACK;

    // send_buffer[1] stays the same

    wl_basic_send_global_packet(TYPE, send_buffer, 2);

    if (wl_basic_do_default(&data_length))

        printf("FAILED\n");

    else

        printf("PASSED\n");

    return 0;

}