Easily add all supported connectivity methods to your mbed OS project
esp8266-driver/TESTS/net/udp_echo/main.cpp@0:615f90842ce8, 2017-07-12 (annotated)
- Committer:
- MACRUM
- Date:
- Wed Jul 12 10:52:58 2017 +0000
- Revision:
- 0:615f90842ce8
Initial commit
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
MACRUM | 0:615f90842ce8 | 1 | #include "mbed.h" |
MACRUM | 0:615f90842ce8 | 2 | #include "ESP8266Interface.h" |
MACRUM | 0:615f90842ce8 | 3 | #include "UDPSocket.h" |
MACRUM | 0:615f90842ce8 | 4 | #include "greentea-client/test_env.h" |
MACRUM | 0:615f90842ce8 | 5 | #include "unity/unity.h" |
MACRUM | 0:615f90842ce8 | 6 | #include "utest.h" |
MACRUM | 0:615f90842ce8 | 7 | |
MACRUM | 0:615f90842ce8 | 8 | using namespace utest::v1; |
MACRUM | 0:615f90842ce8 | 9 | |
MACRUM | 0:615f90842ce8 | 10 | |
MACRUM | 0:615f90842ce8 | 11 | #ifndef MBED_CFG_UDP_CLIENT_ECHO_BUFFER_SIZE |
MACRUM | 0:615f90842ce8 | 12 | #define MBED_CFG_UDP_CLIENT_ECHO_BUFFER_SIZE 64 |
MACRUM | 0:615f90842ce8 | 13 | #endif |
MACRUM | 0:615f90842ce8 | 14 | |
MACRUM | 0:615f90842ce8 | 15 | #ifndef MBED_CFG_UDP_CLIENT_ECHO_TIMEOUT |
MACRUM | 0:615f90842ce8 | 16 | #define MBED_CFG_UDP_CLIENT_ECHO_TIMEOUT 500 |
MACRUM | 0:615f90842ce8 | 17 | #endif |
MACRUM | 0:615f90842ce8 | 18 | |
MACRUM | 0:615f90842ce8 | 19 | #ifndef MBED_CFG_ESP8266_TX |
MACRUM | 0:615f90842ce8 | 20 | #define MBED_CFG_ESP8266_TX D1 |
MACRUM | 0:615f90842ce8 | 21 | #endif |
MACRUM | 0:615f90842ce8 | 22 | |
MACRUM | 0:615f90842ce8 | 23 | #ifndef MBED_CFG_ESP8266_RX |
MACRUM | 0:615f90842ce8 | 24 | #define MBED_CFG_ESP8266_RX D0 |
MACRUM | 0:615f90842ce8 | 25 | #endif |
MACRUM | 0:615f90842ce8 | 26 | |
MACRUM | 0:615f90842ce8 | 27 | #ifndef MBED_CFG_ESP8266_DEBUG |
MACRUM | 0:615f90842ce8 | 28 | #define MBED_CFG_ESP8266_DEBUG false |
MACRUM | 0:615f90842ce8 | 29 | #endif |
MACRUM | 0:615f90842ce8 | 30 | |
MACRUM | 0:615f90842ce8 | 31 | |
MACRUM | 0:615f90842ce8 | 32 | namespace { |
MACRUM | 0:615f90842ce8 | 33 | char tx_buffer[MBED_CFG_UDP_CLIENT_ECHO_BUFFER_SIZE] = {0}; |
MACRUM | 0:615f90842ce8 | 34 | char rx_buffer[MBED_CFG_UDP_CLIENT_ECHO_BUFFER_SIZE] = {0}; |
MACRUM | 0:615f90842ce8 | 35 | const char ASCII_MAX = '~' - ' '; |
MACRUM | 0:615f90842ce8 | 36 | const int ECHO_LOOPS = 16; |
MACRUM | 0:615f90842ce8 | 37 | char uuid[48] = {0}; |
MACRUM | 0:615f90842ce8 | 38 | } |
MACRUM | 0:615f90842ce8 | 39 | |
MACRUM | 0:615f90842ce8 | 40 | void prep_buffer(char *uuid, char *tx_buffer, size_t tx_size) { |
MACRUM | 0:615f90842ce8 | 41 | size_t i = 0; |
MACRUM | 0:615f90842ce8 | 42 | |
MACRUM | 0:615f90842ce8 | 43 | memcpy(tx_buffer, uuid, strlen(uuid)); |
MACRUM | 0:615f90842ce8 | 44 | i += strlen(uuid); |
MACRUM | 0:615f90842ce8 | 45 | |
MACRUM | 0:615f90842ce8 | 46 | tx_buffer[i++] = ' '; |
MACRUM | 0:615f90842ce8 | 47 | |
MACRUM | 0:615f90842ce8 | 48 | for (; i<tx_size; ++i) { |
MACRUM | 0:615f90842ce8 | 49 | tx_buffer[i] = (rand() % 10) + '0'; |
MACRUM | 0:615f90842ce8 | 50 | } |
MACRUM | 0:615f90842ce8 | 51 | } |
MACRUM | 0:615f90842ce8 | 52 | |
MACRUM | 0:615f90842ce8 | 53 | void test_udp_echo() { |
MACRUM | 0:615f90842ce8 | 54 | ESP8266Interface net(MBED_CFG_ESP8266_TX, MBED_CFG_ESP8266_RX, MBED_CFG_ESP8266_DEBUG); |
MACRUM | 0:615f90842ce8 | 55 | |
MACRUM | 0:615f90842ce8 | 56 | int err = net.connect(MBED_CFG_ESP8266_SSID, MBED_CFG_ESP8266_PASS); |
MACRUM | 0:615f90842ce8 | 57 | TEST_ASSERT_EQUAL(0, err); |
MACRUM | 0:615f90842ce8 | 58 | |
MACRUM | 0:615f90842ce8 | 59 | if (err) { |
MACRUM | 0:615f90842ce8 | 60 | printf("MBED: failed to connect with an error of %d\r\n", err); |
MACRUM | 0:615f90842ce8 | 61 | TEST_ASSERT_EQUAL(0, err); |
MACRUM | 0:615f90842ce8 | 62 | } |
MACRUM | 0:615f90842ce8 | 63 | |
MACRUM | 0:615f90842ce8 | 64 | printf("UDP client IP Address is %s\n", net.get_ip_address()); |
MACRUM | 0:615f90842ce8 | 65 | |
MACRUM | 0:615f90842ce8 | 66 | greentea_send_kv("target_ip", net.get_ip_address()); |
MACRUM | 0:615f90842ce8 | 67 | |
MACRUM | 0:615f90842ce8 | 68 | char recv_key[] = "host_port"; |
MACRUM | 0:615f90842ce8 | 69 | char ipbuf[60] = {0}; |
MACRUM | 0:615f90842ce8 | 70 | char portbuf[16] = {0}; |
MACRUM | 0:615f90842ce8 | 71 | unsigned int port = 0; |
MACRUM | 0:615f90842ce8 | 72 | |
MACRUM | 0:615f90842ce8 | 73 | UDPSocket sock; |
MACRUM | 0:615f90842ce8 | 74 | sock.open(&net); |
MACRUM | 0:615f90842ce8 | 75 | sock.set_timeout(MBED_CFG_UDP_CLIENT_ECHO_TIMEOUT); |
MACRUM | 0:615f90842ce8 | 76 | |
MACRUM | 0:615f90842ce8 | 77 | greentea_send_kv("host_ip", " "); |
MACRUM | 0:615f90842ce8 | 78 | greentea_parse_kv(recv_key, ipbuf, sizeof(recv_key), sizeof(ipbuf)); |
MACRUM | 0:615f90842ce8 | 79 | |
MACRUM | 0:615f90842ce8 | 80 | greentea_send_kv("host_port", " "); |
MACRUM | 0:615f90842ce8 | 81 | greentea_parse_kv(recv_key, portbuf, sizeof(recv_key), sizeof(ipbuf)); |
MACRUM | 0:615f90842ce8 | 82 | sscanf(portbuf, "%u", &port); |
MACRUM | 0:615f90842ce8 | 83 | |
MACRUM | 0:615f90842ce8 | 84 | printf("MBED: UDP Server IP address received: %s:%d \n", ipbuf, port); |
MACRUM | 0:615f90842ce8 | 85 | SocketAddress udp_addr(ipbuf, port); |
MACRUM | 0:615f90842ce8 | 86 | |
MACRUM | 0:615f90842ce8 | 87 | int success = 0; |
MACRUM | 0:615f90842ce8 | 88 | for (int i = 0; success < ECHO_LOOPS; i++) { |
MACRUM | 0:615f90842ce8 | 89 | prep_buffer(uuid, tx_buffer, sizeof(tx_buffer)); |
MACRUM | 0:615f90842ce8 | 90 | const int ret = sock.sendto(udp_addr, tx_buffer, sizeof(tx_buffer)); |
MACRUM | 0:615f90842ce8 | 91 | if (ret >= 0) { |
MACRUM | 0:615f90842ce8 | 92 | printf("[%02d] sent %d bytes - %.*s \n", i, ret, ret, tx_buffer); |
MACRUM | 0:615f90842ce8 | 93 | } else { |
MACRUM | 0:615f90842ce8 | 94 | printf("[%02d] Network error %d\n", i, ret); |
MACRUM | 0:615f90842ce8 | 95 | continue; |
MACRUM | 0:615f90842ce8 | 96 | } |
MACRUM | 0:615f90842ce8 | 97 | |
MACRUM | 0:615f90842ce8 | 98 | SocketAddress temp_addr; |
MACRUM | 0:615f90842ce8 | 99 | const int n = sock.recvfrom(&temp_addr, rx_buffer, sizeof(rx_buffer)); |
MACRUM | 0:615f90842ce8 | 100 | if (n >= 0) { |
MACRUM | 0:615f90842ce8 | 101 | printf("[%02d] recv %d bytes - %.*s \n", i, n, n, tx_buffer); |
MACRUM | 0:615f90842ce8 | 102 | } else { |
MACRUM | 0:615f90842ce8 | 103 | printf("[%02d] Network error %d\n", i, n); |
MACRUM | 0:615f90842ce8 | 104 | continue; |
MACRUM | 0:615f90842ce8 | 105 | } |
MACRUM | 0:615f90842ce8 | 106 | |
MACRUM | 0:615f90842ce8 | 107 | if ((temp_addr == udp_addr && |
MACRUM | 0:615f90842ce8 | 108 | n == sizeof(tx_buffer) && |
MACRUM | 0:615f90842ce8 | 109 | memcmp(rx_buffer, tx_buffer, sizeof(rx_buffer)) == 0)) { |
MACRUM | 0:615f90842ce8 | 110 | success += 1; |
MACRUM | 0:615f90842ce8 | 111 | |
MACRUM | 0:615f90842ce8 | 112 | printf("[%02d] success #%d\n", i, success); |
MACRUM | 0:615f90842ce8 | 113 | continue; |
MACRUM | 0:615f90842ce8 | 114 | } |
MACRUM | 0:615f90842ce8 | 115 | |
MACRUM | 0:615f90842ce8 | 116 | // failed, clean out any remaining bad packets |
MACRUM | 0:615f90842ce8 | 117 | sock.set_timeout(0); |
MACRUM | 0:615f90842ce8 | 118 | while (true) { |
MACRUM | 0:615f90842ce8 | 119 | err = sock.recvfrom(NULL, NULL, 0); |
MACRUM | 0:615f90842ce8 | 120 | if (err == NSAPI_ERROR_WOULD_BLOCK) { |
MACRUM | 0:615f90842ce8 | 121 | break; |
MACRUM | 0:615f90842ce8 | 122 | } |
MACRUM | 0:615f90842ce8 | 123 | } |
MACRUM | 0:615f90842ce8 | 124 | sock.set_timeout(MBED_CFG_UDP_CLIENT_ECHO_TIMEOUT); |
MACRUM | 0:615f90842ce8 | 125 | } |
MACRUM | 0:615f90842ce8 | 126 | |
MACRUM | 0:615f90842ce8 | 127 | sock.close(); |
MACRUM | 0:615f90842ce8 | 128 | net.disconnect(); |
MACRUM | 0:615f90842ce8 | 129 | TEST_ASSERT_EQUAL(ECHO_LOOPS, success); |
MACRUM | 0:615f90842ce8 | 130 | } |
MACRUM | 0:615f90842ce8 | 131 | |
MACRUM | 0:615f90842ce8 | 132 | |
MACRUM | 0:615f90842ce8 | 133 | // Test setup |
MACRUM | 0:615f90842ce8 | 134 | utest::v1::status_t test_setup(const size_t number_of_cases) { |
MACRUM | 0:615f90842ce8 | 135 | GREENTEA_SETUP_UUID(120, "udp_echo", uuid, 48); |
MACRUM | 0:615f90842ce8 | 136 | |
MACRUM | 0:615f90842ce8 | 137 | // create mac address based on uuid |
MACRUM | 0:615f90842ce8 | 138 | uint64_t mac = 0; |
MACRUM | 0:615f90842ce8 | 139 | for (int i = 0; i < sizeof(uuid); i++) { |
MACRUM | 0:615f90842ce8 | 140 | mac += uuid[i]; |
MACRUM | 0:615f90842ce8 | 141 | } |
MACRUM | 0:615f90842ce8 | 142 | mbed_set_mac_address((const char*)mac, /*coerce control bits*/ 1); |
MACRUM | 0:615f90842ce8 | 143 | |
MACRUM | 0:615f90842ce8 | 144 | return verbose_test_setup_handler(number_of_cases); |
MACRUM | 0:615f90842ce8 | 145 | } |
MACRUM | 0:615f90842ce8 | 146 | |
MACRUM | 0:615f90842ce8 | 147 | Case cases[] = { |
MACRUM | 0:615f90842ce8 | 148 | Case("UDP echo", test_udp_echo), |
MACRUM | 0:615f90842ce8 | 149 | }; |
MACRUM | 0:615f90842ce8 | 150 | |
MACRUM | 0:615f90842ce8 | 151 | Specification specification(test_setup, cases); |
MACRUM | 0:615f90842ce8 | 152 | |
MACRUM | 0:615f90842ce8 | 153 | int main() { |
MACRUM | 0:615f90842ce8 | 154 | return !Harness::run(specification); |
MACRUM | 0:615f90842ce8 | 155 | } |
MACRUM | 0:615f90842ce8 | 156 |