123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287 |
- /*
- * 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/test.h"
- #include "asterisk/module.h"
- #include "asterisk/file.h"
- #include "asterisk/format_cap.h"
- #include "asterisk/http.h"
- #include "asterisk/http_websocket.h"
- #include "asterisk/json.h"
- #include "asterisk/speech.h"
- #include "asterisk/res_aeap.h"
- #include "asterisk/res_aeap_message.h"
- #define ADDR "127.0.0.1:8088"
- static int speech_test_server_setup(struct ast_json *req, struct ast_json *resp)
- {
- struct ast_json *params;
- if (ast_json_object_set(resp, "codecs", ast_json_ref(ast_json_object_get(req, "codecs")))) {
- return -1;
- }
- params = ast_json_object_get(req, "params"); /* Optional */
- if (params && ast_json_object_set(resp, "params", ast_json_ref(params))) {
- return -1;
- }
- return 0;
- }
- #define TEST_SPEECH_RESULTS_TEXT "foo"
- #define TEST_SPEECH_RESULTS_SCORE 7
- #define TEST_SPEECH_RESULTS_GRAMMAR "bar"
- #define TEST_SPEECH_RESULTS_BEST 1
- static int speech_test_server_get(struct ast_json *req, struct ast_json *resp)
- {
- const char *param;
- struct ast_json *json = NULL;
- param = ast_json_string_get(ast_json_array_get(ast_json_object_get(req, "params"), 0));
- if (!param) {
- return -1;
- }
- if (!strcmp(param, "results")) {
- json = ast_json_pack("{s:[{s:s,s:i,s:s,s:i}]}",
- param,
- "text", TEST_SPEECH_RESULTS_TEXT,
- "score", TEST_SPEECH_RESULTS_SCORE,
- "grammar", TEST_SPEECH_RESULTS_GRAMMAR,
- "best", TEST_SPEECH_RESULTS_BEST);
- } else {
- /* Assume setting */
- json = ast_json_pack("{s:s}", param, "bar");
- }
- if (!json || ast_json_object_set(resp, "params", json)) {
- return -1;
- }
- return 0;
- }
- static int speech_test_server_set(struct ast_json *req, struct ast_json *resp)
- {
- if (ast_json_object_set(resp, "params", ast_json_ref(ast_json_object_get(req, "params")))) {
- return -1;
- }
- return 0;
- }
- static int speech_test_server_handle_request(struct ast_websocket *ws, const void *buf, uint64_t size)
- {
- struct ast_json *req;
- struct ast_json *resp;
- const char *name;
- char *resp_buf;
- int res = 0;
- req = ast_json_load_buf(buf, size, NULL);
- if (!req) {
- ast_log(LOG_ERROR, "speech test handle request: unable to load json\n");
- return -1;
- }
- name = ast_json_object_string_get(req, "request");
- if (!name) {
- ast_log(LOG_ERROR, "speech test handle request: no name\n");
- ast_json_unref(req);
- return -1;
- }
- resp = ast_json_pack("{s:s, s:s}", "response", name,
- "id", ast_json_object_string_get(req, "id"));
- if (!resp) {
- ast_log(LOG_ERROR, "speech test handle request: unable to create response '%s'\n", name);
- ast_json_unref(req);
- return -1;
- }
- if (!strcmp(name, "setup")) {
- res = speech_test_server_setup(req, resp);
- } else if (!strcmp(name, "get")) {
- res = speech_test_server_get(req, resp);
- } else if (!strcmp(name, "set")) {
- res = speech_test_server_set(req, resp);
- } else {
- ast_log(LOG_ERROR, "speech test handle request: unsupported request '%s'\n", name);
- return -1;
- }
- if (res) {
- ast_log(LOG_ERROR, "speech test handle request: unable to build response '%s'\n", name);
- ast_json_unref(resp);
- ast_json_unref(req);
- return -1;
- }
- resp_buf = ast_json_dump_string(resp);
- ast_json_unref(resp);
- if (!resp_buf) {
- ast_log(LOG_ERROR, "speech test handle request: unable to dump response '%s'\n", name);
- ast_json_unref(req);
- return -1;
- }
- res = ast_websocket_write_string(ws, resp_buf);
- if (res) {
- ast_log(LOG_ERROR, "speech test handle request: unable to write response '%s'\n", name);
- }
- ast_json_unref(req);
- ast_free(resp_buf);
- return res;
- }
- static void speech_test_server_cb(struct ast_websocket *ws, struct ast_variable *parameters,
- struct ast_variable *headers)
- {
- int res;
- if (ast_fd_set_flags(ast_websocket_fd(ws), O_NONBLOCK)) {
- ast_websocket_unref(ws);
- return;
- }
- while ((res = ast_websocket_wait_for_input(ws, -1)) > 0) {
- char *payload;
- uint64_t payload_len;
- enum ast_websocket_opcode opcode;
- int fragmented;
- if (ast_websocket_read(ws, &payload, &payload_len, &opcode, &fragmented)) {
- ast_log(LOG_ERROR, "speech test: Read failure in server loop\n");
- break;
- }
- switch (opcode) {
- case AST_WEBSOCKET_OPCODE_CLOSE:
- ast_websocket_unref(ws);
- return;
- case AST_WEBSOCKET_OPCODE_BINARY:
- ast_websocket_write(ws, opcode, payload, payload_len);
- break;
- case AST_WEBSOCKET_OPCODE_TEXT:
- ast_debug(3, "payload=%.*s\n", (int)payload_len, payload);
- if (speech_test_server_handle_request(ws, payload, payload_len)) {
- ast_websocket_unref(ws);
- return;
- }
- break;
- default:
- break;
- }
- }
- ast_websocket_unref(ws);
- }
- AST_TEST_DEFINE(res_speech_aeap_test)
- {
- RAII_VAR(struct ast_format_cap *, cap, NULL, ao2_cleanup);
- RAII_VAR(struct ast_speech_result *, results, NULL, ast_speech_results_free);
- struct ast_speech *speech = NULL;
- enum ast_test_result_state res = AST_TEST_PASS;
- char buf[8] = "";
- switch (cmd) {
- case TEST_INIT:
- info->name = __func__;
- info->explicit_only = 0;
- info->category = "/res/aeap/speech/";
- info->summary = "test the speech AEAP interface";
- info->description = info->summary;
- return AST_TEST_NOT_RUN;
- case TEST_EXECUTE:
- break;
- }
- ast_test_validate(test, !ast_websocket_add_protocol("_aeap_test_speech_", speech_test_server_cb));
- ast_test_validate(test, (cap = ast_format_cap_alloc(AST_FORMAT_CAP_FLAG_DEFAULT)));
- ast_test_validate(test, !ast_format_cap_update_by_allow_disallow(cap, "ulaw", 1));
- ast_test_validate_cleanup(test, (speech = ast_speech_new("_aeap_test_speech_", cap)), res, cleanup);
- ast_speech_start(speech);
- ast_test_validate_cleanup(test, !ast_speech_dtmf(speech, "1"), res, cleanup);
- ast_test_validate_cleanup(test, !ast_speech_change(speech, "foo", "bar"), res, cleanup);
- ast_test_validate_cleanup(test, !ast_speech_change_results_type(
- speech, AST_SPEECH_RESULTS_TYPE_NBEST), res, cleanup);
- ast_test_validate_cleanup(test, !ast_speech_get_setting(
- speech, "foo", buf, sizeof(buf)), res, cleanup);
- ast_test_validate_cleanup(test, !strcmp(buf, "bar"), res, cleanup);
- ast_test_validate_cleanup(test, (results = ast_speech_results_get(speech)), res, cleanup);
- ast_test_validate_cleanup(test, !strcmp(results->text, TEST_SPEECH_RESULTS_TEXT), res, cleanup);
- ast_test_validate_cleanup(test, results->score == TEST_SPEECH_RESULTS_SCORE, res, cleanup);
- ast_test_validate_cleanup(test, !strcmp(results->grammar, TEST_SPEECH_RESULTS_GRAMMAR), res, cleanup);
- ast_test_validate_cleanup(test, results->nbest_num == TEST_SPEECH_RESULTS_BEST, res, cleanup);
- cleanup:
- if (speech) {
- ast_speech_destroy(speech);
- }
- ast_websocket_remove_protocol("_aeap_test_speech_", speech_test_server_cb);
- return res;
- }
- 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(res_speech_aeap_test);
- return AST_MODULE_LOAD_SUCCESS;
- }
- static int unload_module(void)
- {
- AST_TEST_UNREGISTER(res_speech_aeap_test);
- ast_http_test_server_discard(http_server);
- return 0;
- }
- AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_DEFAULT, "Asterisk External Application Protocol Speech test(s)",
- .support_level = AST_MODULE_SUPPORT_CORE,
- .load = load_module,
- .unload = unload_module,
- .requires = "res_speech_aeap",
- );
|