mirror of https://github.com/ARMmbed/mbed-os.git
122 lines
4.5 KiB
C++
122 lines
4.5 KiB
C++
/*
|
|
* Copyright (c) 2018, ARM Limited, All Rights Reserved
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
* not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#if defined(MBED_CONF_RTOS_PRESENT) && !defined(MBED_CONF_CELLULAR_OFFLOAD_DNS_QUERIES)
|
|
|
|
#include "mbed.h"
|
|
#include "greentea-client/test_env.h"
|
|
#include "unity.h"
|
|
#include "utest.h"
|
|
#include "dns_tests.h"
|
|
|
|
using namespace utest::v1;
|
|
|
|
namespace {
|
|
int result_number;
|
|
int result_no_mem;
|
|
int result_dns_failure;
|
|
int result_exp_timeout;
|
|
}
|
|
|
|
// Callback used for asynchronous DNS result
|
|
void getaddrinfo_cb(void *data, nsapi_error_t result, SocketAddress *address)
|
|
{
|
|
dns_application_data_multi_ip *app_data = static_cast<dns_application_data_multi_ip *>(data);
|
|
app_data->result = result;
|
|
for (unsigned int i = 0; i < result; i++) {
|
|
if (address) {
|
|
app_data->addr[i] = address[i];
|
|
}
|
|
}
|
|
app_data->semaphore->release();
|
|
app_data->value_set = true;
|
|
}
|
|
|
|
void do_getaddrinfo_async(const char hosts[][DNS_TEST_HOST_LEN], unsigned int op_count, int *exp_ok, int *exp_no_mem, int *exp_dns_failure, int *exp_timeout)
|
|
{
|
|
// Verify that there is enough hosts in the host list
|
|
TEST_ASSERT(op_count <= MBED_CONF_APP_DNS_TEST_HOSTS_NUM)
|
|
|
|
// Reset counters
|
|
(*exp_ok) = 0;
|
|
(*exp_no_mem) = 0;
|
|
(*exp_dns_failure) = 0;
|
|
(*exp_timeout) = 0;
|
|
|
|
// Create callback semaphore and data
|
|
rtos::Semaphore semaphore;
|
|
dns_application_data_multi_ip *data = new dns_application_data_multi_ip[op_count];
|
|
SocketAddress hints{{NSAPI_UNSPEC}, 80};
|
|
|
|
unsigned int count = 0;
|
|
for (unsigned int i = 0; i < op_count; i++) {
|
|
data[i].semaphore = &semaphore;
|
|
nsapi_error_t err = NetworkInterface::get_default_instance()->getaddrinfo_async(hosts[i], &hints, mbed::Callback<void(nsapi_error_t, SocketAddress *)>(getaddrinfo_cb, (void *) &data[i]));
|
|
TEST_ASSERT(err >= 0 || err == NSAPI_ERROR_NO_MEMORY || err == NSAPI_ERROR_BUSY);
|
|
if (err >= 0) {
|
|
// Callback will be called
|
|
count++;
|
|
} else {
|
|
// No memory to initiate DNS query, callback will not be called
|
|
data[i].result = err;
|
|
}
|
|
}
|
|
|
|
// Wait for callback(s) to complete
|
|
for (unsigned int i = 0; i < count; i++) {
|
|
semaphore.acquire();
|
|
}
|
|
// Print result
|
|
for (unsigned int i = 0; i < op_count; i++) {
|
|
TEST_ASSERT(data[i].result > 0 || data[i].result == NSAPI_ERROR_NO_MEMORY || data[i].result == NSAPI_ERROR_BUSY
|
|
|| data[i].result == NSAPI_ERROR_DNS_FAILURE || data[i].result == NSAPI_ERROR_TIMEOUT);
|
|
if (data[i].result > 0) {
|
|
(*exp_ok)++;
|
|
for (unsigned int results = 0; results < data[i].result; results++) {
|
|
printf("DNS: query \"%s\" => \"%s\"\n",
|
|
hosts[i], data[i].addr[results].get_ip_address());
|
|
}
|
|
} else if (data[i].result == NSAPI_ERROR_DNS_FAILURE) {
|
|
(*exp_dns_failure)++;
|
|
printf("DNS: query \"%s\" => DNS failure\n", hosts[i]);
|
|
} else if (data[i].result == NSAPI_ERROR_TIMEOUT) {
|
|
(*exp_timeout)++;
|
|
printf("DNS: query \"%s\" => timeout\n", hosts[i]);
|
|
} else if (data[i].result == NSAPI_ERROR_NO_MEMORY) {
|
|
(*exp_no_mem)++;
|
|
printf("DNS: query \"%s\" => no memory\n", hosts[i]);
|
|
} else if (data[i].result == NSAPI_ERROR_BUSY) {
|
|
(*exp_no_mem)++;
|
|
printf("DNS: query \"%s\" => busy\n", hosts[i]);
|
|
}
|
|
}
|
|
|
|
delete[] data;
|
|
}
|
|
|
|
void ASYNCHRONOUS_DNS_MULTI_IP()
|
|
{
|
|
nsapi_dns_reset();
|
|
do_getaddrinfo_async(dns_test_hosts_multi_ip, MBED_CONF_APP_DNS_SIMULT_QUERIES, &result_number, &result_no_mem, &result_dns_failure, &result_exp_timeout);
|
|
|
|
TEST_ASSERT_EQUAL(MBED_CONF_APP_DNS_SIMULT_QUERIES, result_number);
|
|
TEST_ASSERT_EQUAL(0, result_no_mem);
|
|
TEST_ASSERT_EQUAL(0, result_dns_failure);
|
|
TEST_ASSERT_EQUAL(0, result_exp_timeout);
|
|
}
|
|
#endif // defined(MBED_CONF_RTOS_PRESENT) && !defined(MBED_CONF_CELLULAR_OFFLOAD_DNS_QUERIES)
|