A sporadic test failure was happening when executing the AEAP Websocket transport tests. It was originally thought this was due to things not getting cleaned up fast enough, but upon further investigation I determined the underlying cause was poll() getting interrupted and this not being handled in all places. This change adds EINTR and EAGAIN handling to the Websocket client connect code as well as the AEAP Websocket transport code. If either occur then the code will just go back to waiting for data. The originally disabled failure test case has also been re-enabled. ASTERISK-30099 Change-Id: I1711a331ecf5d35cd542911dc6aaa9acf1e172ad
250 lines
6.9 KiB
C
250 lines
6.9 KiB
C
/*
|
|
* Asterisk -- An open source telephony toolkit.
|
|
*
|
|
* Copyright (C) 2021, Sangoma Technologies Corporation
|
|
*
|
|
* Kevin Harwell <kharwell@sangoma.com>
|
|
*
|
|
* See http://www.asterisk.org for more information about
|
|
* the Asterisk project. Please do not directly contact
|
|
* any of the maintainers of this project for assistance;
|
|
* the project provides a web site, mailing lists and IRC
|
|
* channels for your use.
|
|
*
|
|
* This program is free software, distributed under the terms of
|
|
* the GNU General Public License Version 2. See the LICENSE file
|
|
* at the top of the source tree.
|
|
*/
|
|
|
|
/*** MODULEINFO
|
|
<depend>TEST_FRAMEWORK</depend>
|
|
<depend>res_aeap</depend>
|
|
<support_level>core</support_level>
|
|
***/
|
|
|
|
#include "asterisk.h"
|
|
|
|
#include "asterisk/http.h"
|
|
#include "asterisk/test.h"
|
|
#include "asterisk/module.h"
|
|
|
|
#include "../res/res_aeap/transport.h"
|
|
|
|
#define CATEGORY "/res/aeap/transport/"
|
|
|
|
#define ADDR "127.0.0.1:8088"
|
|
#define TRANSPORT_URL "ws://" ADDR "/ws"
|
|
#define TRANSPORT_URL_INVALID "ws://" ADDR "/invalid"
|
|
#define TRANSPORT_PROTOCOL "echo"
|
|
#define TRANSPORT_PROTOCOL_INVALID "invalid"
|
|
#define TRANSPORT_TIMEOUT 2000
|
|
|
|
AST_TEST_DEFINE(transport_create_invalid)
|
|
{
|
|
RAII_VAR(struct aeap_transport *, transport, NULL, aeap_transport_destroy);
|
|
|
|
switch (cmd) {
|
|
case TEST_INIT:
|
|
info->name = __func__;
|
|
info->explicit_only = 0;
|
|
info->category = CATEGORY;
|
|
info->summary = "test creating an AEAP invalid transport type";
|
|
info->description = info->summary;
|
|
return AST_TEST_NOT_RUN;
|
|
case TEST_EXECUTE:
|
|
break;
|
|
}
|
|
|
|
/* Transport is expected to be NULL here */
|
|
ast_test_validate(test, !(transport = aeap_transport_create("invalid")));
|
|
|
|
return AST_TEST_PASS;
|
|
}
|
|
|
|
AST_TEST_DEFINE(transport_create)
|
|
{
|
|
RAII_VAR(struct aeap_transport *, transport, NULL, aeap_transport_destroy);
|
|
|
|
switch (cmd) {
|
|
case TEST_INIT:
|
|
info->name = __func__;
|
|
info->explicit_only = 0;
|
|
info->category = CATEGORY;
|
|
info->summary = "test creating an AEAP transport";
|
|
info->description = info->summary;
|
|
return AST_TEST_NOT_RUN;
|
|
case TEST_EXECUTE:
|
|
break;
|
|
}
|
|
|
|
/* Type is based off the scheme, so just pass in the URL here */
|
|
ast_test_validate(test, (transport = aeap_transport_create(TRANSPORT_URL)));
|
|
|
|
return AST_TEST_PASS;
|
|
}
|
|
|
|
AST_TEST_DEFINE(transport_connect)
|
|
{
|
|
RAII_VAR(struct aeap_transport *, transport, NULL, aeap_transport_destroy);
|
|
|
|
switch (cmd) {
|
|
case TEST_INIT:
|
|
info->name = __func__;
|
|
info->explicit_only = 0;
|
|
info->category = CATEGORY;
|
|
info->summary = "test connecting to an AEAP transport";
|
|
info->description = info->summary;
|
|
return AST_TEST_NOT_RUN;
|
|
case TEST_EXECUTE:
|
|
break;
|
|
}
|
|
|
|
/* Type is based off the scheme, so just pass in the URL for the type */
|
|
ast_test_validate(test, (transport = aeap_transport_create_and_connect(
|
|
TRANSPORT_URL, TRANSPORT_URL, TRANSPORT_PROTOCOL, TRANSPORT_TIMEOUT)));
|
|
|
|
ast_test_validate(test, aeap_transport_is_connected(transport));
|
|
ast_test_validate(test, !aeap_transport_disconnect(transport));
|
|
ast_test_validate(test, !aeap_transport_is_connected(transport));
|
|
|
|
return AST_TEST_PASS;
|
|
}
|
|
|
|
AST_TEST_DEFINE(transport_connect_fail)
|
|
{
|
|
RAII_VAR(struct aeap_transport *, transport, NULL, aeap_transport_destroy);
|
|
|
|
switch (cmd) {
|
|
case TEST_INIT:
|
|
info->name = __func__;
|
|
info->explicit_only = 0;
|
|
info->category = CATEGORY;
|
|
info->summary = "test connecting failure for an AEAP transport";
|
|
info->description = info->summary;
|
|
return AST_TEST_NOT_RUN;
|
|
case TEST_EXECUTE:
|
|
break;
|
|
}
|
|
|
|
/* Test invalid address */
|
|
ast_test_validate(test, (transport = aeap_transport_create(TRANSPORT_URL)));
|
|
|
|
ast_test_validate(test, aeap_transport_connect(transport,
|
|
TRANSPORT_URL_INVALID, TRANSPORT_PROTOCOL, TRANSPORT_TIMEOUT));
|
|
|
|
ast_test_validate(test, !aeap_transport_is_connected(transport));
|
|
|
|
aeap_transport_destroy(transport);
|
|
|
|
/* /\* Test invalid protocol *\/ */
|
|
ast_test_validate(test, (transport = aeap_transport_create(TRANSPORT_URL)));
|
|
|
|
ast_test_validate(test, aeap_transport_connect(transport,
|
|
TRANSPORT_URL, TRANSPORT_PROTOCOL_INVALID, TRANSPORT_TIMEOUT));
|
|
|
|
ast_test_validate(test, !aeap_transport_is_connected(transport));
|
|
|
|
return AST_TEST_PASS;
|
|
}
|
|
|
|
AST_TEST_DEFINE(transport_binary)
|
|
{
|
|
RAII_VAR(struct aeap_transport *, transport, NULL, aeap_transport_destroy);
|
|
int num = 38;
|
|
enum AST_AEAP_DATA_TYPE rtype;
|
|
|
|
switch (cmd) {
|
|
case TEST_INIT:
|
|
info->name = __func__;
|
|
info->explicit_only = 0;
|
|
info->category = CATEGORY;
|
|
info->summary = "test binary I/O from an AEAP transport";
|
|
info->description = info->summary;
|
|
return AST_TEST_NOT_RUN;
|
|
case TEST_EXECUTE:
|
|
break;
|
|
}
|
|
|
|
ast_test_validate(test, (transport = aeap_transport_create_and_connect(
|
|
TRANSPORT_URL, TRANSPORT_URL, TRANSPORT_PROTOCOL, TRANSPORT_TIMEOUT)));
|
|
|
|
ast_test_validate(test, aeap_transport_write(transport, &num, sizeof(num),
|
|
AST_AEAP_DATA_TYPE_BINARY) == sizeof(num));
|
|
ast_test_validate(test, aeap_transport_read(transport, &num,
|
|
sizeof(num), &rtype) == sizeof(num));
|
|
ast_test_validate(test, rtype == AST_AEAP_DATA_TYPE_BINARY);
|
|
ast_test_validate(test, num == 38);
|
|
|
|
return AST_TEST_PASS;
|
|
}
|
|
|
|
AST_TEST_DEFINE(transport_string)
|
|
{
|
|
RAII_VAR(struct aeap_transport *, transport, NULL, aeap_transport_destroy);
|
|
char buf[16];
|
|
enum AST_AEAP_DATA_TYPE rtype;
|
|
|
|
switch (cmd) {
|
|
case TEST_INIT:
|
|
info->name = __func__;
|
|
info->explicit_only = 0;
|
|
info->category = CATEGORY;
|
|
info->summary = "test string I/O from an AEAP transport";
|
|
info->description = info->summary;
|
|
return AST_TEST_NOT_RUN;
|
|
case TEST_EXECUTE:
|
|
break;
|
|
}
|
|
|
|
ast_test_validate(test, (transport = aeap_transport_create_and_connect(
|
|
TRANSPORT_URL, TRANSPORT_URL, TRANSPORT_PROTOCOL, TRANSPORT_TIMEOUT)));
|
|
|
|
ast_test_validate(test, aeap_transport_write(transport, "foo bar baz", 11,
|
|
AST_AEAP_DATA_TYPE_STRING) == 11);
|
|
ast_test_validate(test, aeap_transport_read(transport, buf,
|
|
sizeof(buf) / sizeof(char), &rtype) == 11);
|
|
ast_test_validate(test, rtype == AST_AEAP_DATA_TYPE_STRING);
|
|
ast_test_validate(test, !strcmp(buf, "foo bar baz"));
|
|
|
|
return AST_TEST_PASS;
|
|
}
|
|
|
|
static struct ast_http_server *http_server;
|
|
|
|
static int load_module(void)
|
|
{
|
|
if (!(http_server = ast_http_test_server_get("aeap transport http server", NULL))) {
|
|
return AST_MODULE_LOAD_DECLINE;
|
|
}
|
|
|
|
AST_TEST_REGISTER(transport_string);
|
|
AST_TEST_REGISTER(transport_binary);
|
|
AST_TEST_REGISTER(transport_connect_fail);
|
|
AST_TEST_REGISTER(transport_connect);
|
|
AST_TEST_REGISTER(transport_create);
|
|
AST_TEST_REGISTER(transport_create_invalid);
|
|
|
|
return AST_MODULE_LOAD_SUCCESS;
|
|
}
|
|
|
|
static int unload_module(void)
|
|
{
|
|
AST_TEST_UNREGISTER(transport_string);
|
|
AST_TEST_UNREGISTER(transport_binary);
|
|
AST_TEST_UNREGISTER(transport_connect_fail);
|
|
AST_TEST_UNREGISTER(transport_connect);
|
|
AST_TEST_UNREGISTER(transport_create);
|
|
AST_TEST_UNREGISTER(transport_create_invalid);
|
|
|
|
ast_http_test_server_discard(http_server);
|
|
|
|
return 0;
|
|
}
|
|
|
|
AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_DEFAULT, "Asterisk External Application Protocol Transport Tests",
|
|
.support_level = AST_MODULE_SUPPORT_CORE,
|
|
.load = load_module,
|
|
.unload = unload_module,
|
|
.requires = "res_aeap",
|
|
);
|