udpsocket_sendto_repeat.cpp

00001 /*
00002  * Copyright (c) 2018, 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 #include "mbed.h"
00019 #include "UDPSocket.h"
00020 #include "greentea-client/test_env.h"
00021 #include "unity/unity.h"
00022 #include "utest.h"
00023 #include "udp_tests.h"
00024 
00025 using namespace utest::v1;
00026 
00027 void UDPSOCKET_SENDTO_REPEAT()
00028 {
00029     UDPSocket sock;
00030     TEST_ASSERT_EQUAL(NSAPI_ERROR_OK , sock.open(get_interface()));
00031 
00032     SocketAddress udp_addr;
00033     get_interface()->gethostbyname(MBED_CONF_APP_ECHO_SERVER_ADDR, &udp_addr);
00034     udp_addr.set_port(9);
00035 
00036     int sent;
00037     Timer timer;
00038     int i;
00039     static const char tx_buffer[] = {'h', 'e', 'l', 'l', 'o'};
00040     bool oom_earlier = false; // 2 times in a row -> time to give up
00041     for (i = 0; i < 100; i++) {
00042         sent = sock.sendto(udp_addr, tx_buffer, sizeof(tx_buffer));
00043         if (sent == NSAPI_ERROR_NO_MEMORY ) {
00044             if (oom_earlier) {
00045                 break;
00046             }
00047             oom_earlier = true;
00048             wait(1);
00049             continue;
00050         }
00051         oom_earlier = false;
00052         TEST_ASSERT_EQUAL(sizeof(tx_buffer), sent);
00053     }
00054     TEST_ASSERT_EQUAL(NSAPI_ERROR_OK , sock.close());
00055 }