main.cpp

00001 /*
00002  * Copyright (c) 2013-2017, ARM Limited, All Rights Reserved
00003  * SPDX-License-Identifier: Apache-2.0
00004  *
00005  * Licensed under the Apache License, Version 2.0 (the "License"); you may
00006  * not use this file except in compliance with the License.
00007  * You may obtain a copy of the License at
00008  *
00009  * http://www.apache.org/licenses/LICENSE-2.0
00010  *
00011  * Unless required by applicable law or agreed to in writing, software
00012  * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
00013  * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
00014  * See the License for the specific language governing permissions and
00015  * limitations under the License.
00016  */
00017 
00018 #ifndef MBED_CONF_APP_CONNECT_STATEMENT
00019     #error [NOT_SUPPORTED] No network configuration found for this target.
00020 #endif
00021 
00022 #include "mbed.h"
00023 #include MBED_CONF_APP_HEADER_FILE
00024 #include "TCPSocket.h"
00025 #include "greentea-client/test_env.h"
00026 #include "unity/unity.h"
00027 #include "utest.h"
00028 
00029 using namespace utest::v1;
00030 
00031 #ifndef MBED_CONF_APP_TCP_CLIENT_ECHO_BUFFER_SIZE
00032 #define MBED_CONF_APP_TCP_CLIENT_ECHO_BUFFER_SIZE 256
00033 #endif
00034 
00035 namespace {
00036     char tx_buffer[MBED_CONF_APP_TCP_CLIENT_ECHO_BUFFER_SIZE] = {0};
00037     char rx_buffer[MBED_CONF_APP_TCP_CLIENT_ECHO_BUFFER_SIZE] = {0};
00038 }
00039 
00040 void prep_buffer(char *tx_buffer, size_t tx_size) {
00041     for (size_t i=0; i<tx_size; ++i) {
00042         tx_buffer[i] = (rand() % 10) + '0';
00043     }
00044 }
00045 
00046 void test_tcp_echo() {
00047     int n = 0;
00048     NetworkInterface* net = MBED_CONF_APP_OBJECT_CONSTRUCTION;
00049     int err =  MBED_CONF_APP_CONNECT_STATEMENT;
00050 
00051     if (err) {
00052         printf("MBED: failed to connect with an error of %d\r\n", err);
00053         TEST_ASSERT_EQUAL(0, err);
00054     }
00055 
00056     printf("MBED: TCPClient IP address is '%s'\n", net->get_ip_address());
00057 
00058     bool result = false;
00059 
00060     TCPSocket sock(net);
00061 
00062 #if defined(MBED_CONF_APP_ECHO_SERVER_ADDR) && defined(MBED_CONF_APP_ECHO_SERVER_PORT)
00063     printf("TCP: Connect to %s:%d\r\n", MBED_CONF_APP_ECHO_SERVER_ADDR, MBED_CONF_APP_ECHO_SERVER_PORT);
00064     SocketAddress tcp_addr(MBED_CONF_APP_ECHO_SERVER_ADDR, MBED_CONF_APP_ECHO_SERVER_PORT);
00065 #else /* MBED_CONF_APP_ECHO_SERVER_ADDR && MBED_CONF_APP_ECHO_SERVER_PORT */
00066     char recv_key[] = "host_port";
00067     char ipbuf[60] = {0};
00068     char portbuf[16] = {0};
00069     unsigned int port = 0;
00070 
00071     greentea_send_kv("target_ip", net->get_ip_address());
00072     greentea_send_kv("host_ip", " ");
00073     greentea_parse_kv(recv_key, ipbuf, sizeof(recv_key), sizeof(ipbuf));
00074 
00075     greentea_send_kv("host_port", " ");
00076     greentea_parse_kv(recv_key, portbuf, sizeof(recv_key), sizeof(ipbuf));
00077     sscanf(portbuf, "%u", &port);
00078 
00079     printf("TCP: Connect to %s:%d\r\n", ipbuf, port);
00080     SocketAddress tcp_addr(ipbuf, port);
00081 #endif /* MBED_CONF_APP_ECHO_SERVER_ADDR && MBED_CONF_APP_ECHO_SERVER_PORT */
00082 
00083     if (sock.connect(tcp_addr) == 0) {
00084         printf("tx_buffer buffer size: %u\r\n", sizeof(tx_buffer));
00085         printf("rx_buffer buffer size: %u\r\n", sizeof(rx_buffer));
00086 
00087         prep_buffer(tx_buffer, sizeof(tx_buffer));
00088 #if defined(MBED_CONF_APP_TCP_ECHO_PREFIX)
00089         n = sock.recv(rx_buffer, sizeof(MBED_CONF_APP_TCP_ECHO_PREFIX));
00090         if (n >= 0) {
00091             printf("recv-ed prefix: %d bytes - %.*s  \n", n, n, rx_buffer);
00092         } else {
00093             printf("Network error in receiving prefix: %d\n", n);
00094         }
00095 #endif /* MBED_CONF_APP_TCP_ECHO_PREFIX */
00096         const int ret = sock.send(tx_buffer, sizeof(tx_buffer));
00097         if (ret >= 0) {
00098             printf("sent %d bytes - %.*s  \n", ret, ret, tx_buffer);
00099         } else {
00100             printf("Network error %d\n", ret);
00101         }
00102 
00103         n = sock.recv(rx_buffer, sizeof(rx_buffer));
00104         if (n >= 0) {
00105             printf("recv %d bytes - %.*s  \n", n, n, rx_buffer);
00106         } else {
00107             printf("Network error %d\n", n);
00108         }
00109 
00110         result = !memcmp(tx_buffer, rx_buffer, sizeof(tx_buffer));
00111         TEST_ASSERT_EQUAL(ret, sizeof(rx_buffer));
00112         TEST_ASSERT_EQUAL(true, result);
00113     }
00114 
00115     sock.close();
00116     net->disconnect();
00117     TEST_ASSERT_EQUAL(true, result);
00118 }
00119 
00120 
00121 // Test setup
00122 utest::v1::status_t test_setup(const size_t number_of_cases) {
00123     GREENTEA_SETUP(240, "tcp_echo");
00124     return verbose_test_setup_handler(number_of_cases);
00125 }
00126 
00127 Case cases[] = {
00128     Case("TCP echo", test_tcp_echo),
00129 };
00130 
00131 Specification specification(test_setup, cases);
00132 
00133 int main() {
00134     return !Harness::run(specification);
00135 }