123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589 |
- /*
- * Asterisk -- An open source telephony toolkit.
- *
- * Copyright (C) 2015, Digium, Inc.
- *
- * Joshua Colp <jcolp@digium.com>
- * Mark Michelson <mmichelson@digium.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>
- <support_level>core</support_level>
- ***/
- #include "asterisk.h"
- #include <arpa/nameser.h>
- #include "asterisk/test.h"
- #include "asterisk/module.h"
- #include "asterisk/dns_core.h"
- #include "asterisk/dns_resolver.h"
- #include "asterisk/dns_srv.h"
- #include "asterisk/dns_test.h"
- struct srv_record {
- uint16_t priority;
- uint16_t weight;
- uint16_t port;
- const char *host;
- unsigned int ignore_priority;
- unsigned int ignore_weight;
- unsigned int ignore_port;
- unsigned int ignore_host;
- };
- static int generate_srv_record(void *dns_record, char *buf)
- {
- struct srv_record *record = dns_record;
- uint16_t priority = htons(record->priority);
- uint16_t weight = htons(record->weight);
- uint16_t port = htons(record->port);
- char *ptr = buf;
- if (!record->ignore_priority) {
- memcpy(ptr, &priority, sizeof(priority));
- ptr += sizeof(priority);
- }
- if (!record->ignore_weight) {
- memcpy(ptr, &weight, sizeof(weight));
- ptr += sizeof(weight);
- }
- if (!record->ignore_port) {
- memcpy(ptr, &port, sizeof(port));
- ptr += sizeof(port);
- }
- if (!record->ignore_host) {
- ptr += ast_dns_test_write_domain(record->host, ptr);
- }
- return ptr - buf;
- }
- static struct srv_record *test_records;
- static int num_test_records;
- static char ans_buffer[1024];
- static void *srv_thread(void *dns_query)
- {
- struct ast_dns_query *query = dns_query;
- int i;
- int ans_size;
- ans_size = ast_dns_test_generate_result(query, test_records, num_test_records,
- sizeof(struct srv_record), generate_srv_record, ans_buffer);
- ast_dns_resolver_set_result(query, 0, 0, NOERROR, "goose.feathers", ans_buffer, ans_size);
- for (i = 0; i < num_test_records; ++i) {
- char record[128];
- int srv_size;
- srv_size = generate_srv_record(&test_records[i], record);
- ast_dns_resolver_add_record(query, T_SRV, C_IN, 12345, record, srv_size);
- }
- ast_dns_resolver_completed(query);
- ao2_ref(query, -1);
- return NULL;
- }
- static int srv_resolve(struct ast_dns_query *query)
- {
- pthread_t thread;
- return ast_pthread_create_detached(&thread, NULL, srv_thread, ao2_bump(query));
- }
- static int srv_cancel(struct ast_dns_query *query)
- {
- return -1;
- }
- static struct ast_dns_resolver srv_resolver = {
- .name = "srv_test",
- .priority = 0,
- .resolve = srv_resolve,
- .cancel = srv_cancel,
- };
- static enum ast_test_result_state nominal_test(struct ast_test *test, struct srv_record *records,
- int *srv_record_order, int num_records)
- {
- RAII_VAR(struct ast_dns_result *, result, NULL, ast_dns_result_free);
- const struct ast_dns_record *record;
- enum ast_test_result_state res = AST_TEST_PASS;
- int i;
- test_records = records;
- num_test_records = num_records;
- memset(ans_buffer, 0, sizeof(ans_buffer));
- ast_dns_resolver_register(&srv_resolver);
- if (ast_dns_resolve("goose.feathers", T_SRV, C_IN, &result)) {
- ast_test_status_update(test, "DNS resolution failed\n");
- res = AST_TEST_FAIL;
- goto cleanup;
- }
- if (!result) {
- ast_test_status_update(test, "DNS resolution returned no result\n");
- res = AST_TEST_FAIL;
- goto cleanup;
- }
- i = 0;
- for (record = ast_dns_result_get_records(result); record; record = ast_dns_record_get_next(record)) {
- if (ast_dns_srv_get_priority(record) != records[srv_record_order[i]].priority) {
- ast_test_status_update(test, "Unexpected priority in returned SRV record\n");
- res = AST_TEST_FAIL;
- }
- if (ast_dns_srv_get_weight(record) != records[srv_record_order[i]].weight) {
- ast_test_status_update(test, "Unexpected weight in returned SRV record\n");
- res = AST_TEST_FAIL;
- }
- if (ast_dns_srv_get_port(record) != records[srv_record_order[i]].port) {
- ast_test_status_update(test, "Unexpected port in returned SRV record\n");
- res = AST_TEST_FAIL;
- }
- if (strcmp(ast_dns_srv_get_host(record), records[srv_record_order[i]].host)) {
- ast_test_status_update(test, "Unexpected host in returned SRV record\n");
- res = AST_TEST_FAIL;
- }
- ++i;
- }
- if (i != num_records) {
- ast_test_status_update(test, "Unexpected number of records returned in SRV lookup\n");
- res = AST_TEST_FAIL;
- }
- cleanup:
- ast_dns_resolver_unregister(&srv_resolver);
- test_records = NULL;
- num_test_records = 0;
- memset(ans_buffer, 0, sizeof(ans_buffer));
- return res;
- }
- AST_TEST_DEFINE(srv_resolve_single_record)
- {
- struct srv_record records[] = {
- { 10, 10, 5060, "goose.down" },
- };
- int srv_record_order[] = { 0, };
- switch (cmd) {
- case TEST_INIT:
- info->name = "srv_resolve_single_record";
- info->category = "/main/dns/srv/";
- info->summary = "Test an SRV lookup which returns a single record";
- info->description = "This test defines a single SRV record and performs a\n"
- "resolution of the domain to which they belong. The test ensures that all\n"
- "fields of the SRV record are parsed correctly";
- return AST_TEST_NOT_RUN;
- case TEST_EXECUTE:
- break;
- }
- return nominal_test(test, records, srv_record_order, ARRAY_LEN(records));
- }
- AST_TEST_DEFINE(srv_resolve_sort_priority)
- {
- struct srv_record records[] = {
- { 20, 10, 5060, "tacos" },
- { 10, 10, 5060, "goose.down" },
- };
- int srv_record_order[] = { 1, 0};
- switch (cmd) {
- case TEST_INIT:
- info->name = "srv_resolve_sort_priority";
- info->category = "/main/dns/srv/";
- info->summary = "Test an SRV lookup which returns two records with differing priorities";
- info->description = "This test defines two SRV records with differing priorities and\n"
- "performs a resolution of the domain to which they belong. The test ensures that\n"
- "the two records are sorted according to priority and that all fields of the SRV\n"
- "records are parsed correctly";
- return AST_TEST_NOT_RUN;
- case TEST_EXECUTE:
- break;
- }
- return nominal_test(test, records, srv_record_order, ARRAY_LEN(records));
- }
- AST_TEST_DEFINE(srv_resolve_same_priority_zero_weight)
- {
- struct srv_record records[] = {
- { 10, 0, 5060, "tacos" },
- { 10, 10, 5060, "goose.down" },
- };
- int srv_record_order[] = { 1, 0};
- switch (cmd) {
- case TEST_INIT:
- info->name = "srv_resolve_same_priority_zero_weight";
- info->category = "/main/dns/srv/";
- info->summary = "Test an SRV lookup which returns two records with same priority but different weights";
- info->description = "This test defines two SRV records with same priority but different weights and\n"
- "performs a resolution of the domain to which they belong. The test ensures that\n"
- "the record with zero weight comes last and that all fields of the SRV\n"
- "records are parsed correctly";
- return AST_TEST_NOT_RUN;
- case TEST_EXECUTE:
- break;
- }
- return nominal_test(test, records, srv_record_order, ARRAY_LEN(records));
- }
- AST_TEST_DEFINE(srv_resolve_same_priority_different_weights)
- {
- struct srv_record records[] = {
- { 10, 10, 5060, "tacos" },
- { 10, 20, 5060, "goose.down" },
- };
- int srv_record_occurence[2] = { 0, };
- enum ast_test_result_state res = AST_TEST_PASS;
- int count = 0;
- switch (cmd) {
- case TEST_INIT:
- info->name = "srv_resolve_same_priority_different_weights";
- info->category = "/main/dns/srv/";
- info->summary = "Test an SRV lookup which returns two records with same priority but different weights";
- info->description = "This test defines two SRV records with same priority but different weights and\n"
- "performs a resolution of the domain to which they belong. The test ensures that\n"
- "the record with higher weight occurs more often than the one of lesser weight";
- return AST_TEST_NOT_RUN;
- case TEST_EXECUTE:
- break;
- }
- test_records = records;
- num_test_records = ARRAY_LEN(records);
- ast_dns_resolver_register(&srv_resolver);
- for (count = 0; count < 100; count++) {
- struct ast_dns_result *result;
- const struct ast_dns_record *record;
- int i;
- memset(ans_buffer, 0, sizeof(ans_buffer));
- if (ast_dns_resolve("goose.feathers", T_SRV, C_IN, &result)) {
- ast_test_status_update(test, "DNS resolution failed\n");
- res = AST_TEST_FAIL;
- goto cleanup;
- }
- if (!result) {
- ast_test_status_update(test, "DNS resolution returned no result\n");
- res = AST_TEST_FAIL;
- goto cleanup;
- }
- record = ast_dns_result_get_records(result);
- for (i = 0; i < ARRAY_LEN(records); i++) {
- if (ast_dns_srv_get_priority(record) != records[i].priority) {
- continue;
- }
- if (ast_dns_srv_get_weight(record) != records[i].weight) {
- continue;
- }
- if (ast_dns_srv_get_port(record) != records[i].port) {
- continue;
- }
- if (strcmp(ast_dns_srv_get_host(record), records[i].host)) {
- continue;
- }
- srv_record_occurence[i]++;
- break;
- }
- ast_dns_result_free(result);
- }
- if (srv_record_occurence[0] > srv_record_occurence[1]) {
- ast_test_status_update(test, "SRV sorting resulted in lesser weight being returned more often\n");
- res = AST_TEST_FAIL;
- }
- cleanup:
- ast_dns_resolver_unregister(&srv_resolver);
- test_records = NULL;
- num_test_records = 0;
- memset(ans_buffer, 0, sizeof(ans_buffer));
- return res;
- }
- AST_TEST_DEFINE(srv_resolve_different_priorities_different_weights)
- {
- struct srv_record records[] = {
- { 10, 10, 5060, "tacos" },
- { 10, 20, 5060, "goose.down" },
- { 5, 80, 5060, "moo" },
- { 5, 10, 5060, "Canada" },
- };
- int srv_record_priority[4] = { 5, 5, 10, 10 };
- int srv_record_occurence[4] = { 0, };
- enum ast_test_result_state res = AST_TEST_PASS;
- int count = 0;
- switch (cmd) {
- case TEST_INIT:
- info->name = "srv_resolve_different_priorities_different_weights";
- info->category = "/main/dns/srv/";
- info->summary = "Test an SRV lookup which returns four records with different priority and different weights";
- info->description = "This test defines four SRV records, two with one priority and two with another priority,\n"
- "and different weights and performs a resolution of the domain to which they belong.\n"
- "The test ensures that the priorities are sorted properly and that the records with higher weight\n"
- "occur more often than the ones of less weight.";
- return AST_TEST_NOT_RUN;
- case TEST_EXECUTE:
- break;
- }
- test_records = records;
- num_test_records = ARRAY_LEN(records);
- ast_dns_resolver_register(&srv_resolver);
- for (count = 0; count < 100; count++) {
- struct ast_dns_result *result;
- const struct ast_dns_record *record;
- int i;
- memset(ans_buffer, 0, sizeof(ans_buffer));
- if (ast_dns_resolve("goose.feathers", T_SRV, C_IN, &result)) {
- ast_test_status_update(test, "DNS resolution failed\n");
- res = AST_TEST_FAIL;
- goto cleanup;
- }
- if (!result) {
- ast_test_status_update(test, "DNS resolution returned no result\n");
- res = AST_TEST_FAIL;
- goto cleanup;
- }
- i = 0;
- for (record = ast_dns_result_get_records(result); record; record = ast_dns_record_get_next(record)) {
- if (ast_dns_srv_get_priority(record) != srv_record_priority[i]) {
- ast_test_status_update(test, "Unexpected priority in returned SRV record\n");
- res = AST_TEST_FAIL;
- }
- i++;
- }
- record = ast_dns_result_get_records(result);
- for (i = 0; i < ARRAY_LEN(records); i++) {
- if (ast_dns_srv_get_priority(record) != records[i].priority) {
- continue;
- }
- if (ast_dns_srv_get_weight(record) != records[i].weight) {
- continue;
- }
- if (ast_dns_srv_get_port(record) != records[i].port) {
- continue;
- }
- if (strcmp(ast_dns_srv_get_host(record), records[i].host)) {
- continue;
- }
- srv_record_occurence[i]++;
- break;
- }
- ast_dns_result_free(result);
- }
- if (srv_record_occurence[0] > srv_record_occurence[1]) {
- ast_test_status_update(test, "SRV sorting resulted in lesser weight being returned more often for priority 10\n");
- res = AST_TEST_FAIL;
- }
- if (srv_record_occurence[3] > srv_record_occurence[2]) {
- ast_test_status_update(test, "SRV sorting resulted in lesser weight being returned more often for priority 5\n");
- res = AST_TEST_FAIL;
- }
- cleanup:
- ast_dns_resolver_unregister(&srv_resolver);
- test_records = NULL;
- num_test_records = 0;
- memset(ans_buffer, 0, sizeof(ans_buffer));
- return res;
- }
- static enum ast_test_result_state invalid_record_test(struct ast_test *test, struct srv_record *records,
- int num_records)
- {
- RAII_VAR(struct ast_dns_result *, result, NULL, ast_dns_result_free);
- const struct ast_dns_record *record;
- enum ast_test_result_state res = AST_TEST_PASS;
- test_records = records;
- num_test_records = num_records;
- memset(ans_buffer, 0, sizeof(ans_buffer));
- ast_dns_resolver_register(&srv_resolver);
- if (ast_dns_resolve("goose.feathers", T_SRV, C_IN, &result)) {
- ast_test_status_update(test, "DNS resolution failed\n");
- res = AST_TEST_FAIL;
- goto cleanup;
- }
- if (!result) {
- ast_test_status_update(test, "DNS resolution returned no result\n");
- res = AST_TEST_FAIL;
- goto cleanup;
- }
- record = ast_dns_result_get_records(result);
- if (record) {
- ast_test_status_update(test, "Unexpected record returned from SRV query\n");
- res = AST_TEST_FAIL;
- }
- cleanup:
- ast_dns_resolver_unregister(&srv_resolver);
- test_records = NULL;
- num_test_records = 0;
- memset(ans_buffer, 0, sizeof(ans_buffer));
- return res;
- }
- AST_TEST_DEFINE(srv_resolve_record_missing_weight_port_host)
- {
- struct srv_record records[] = {
- { 10, 10, 5060, "tacos.com", 0, 1, 1, 1 },
- };
- switch (cmd) {
- case TEST_INIT:
- info->name = "srv_resolve_record_missing_weight_port_host";
- info->category = "/main/dns/srv/";
- info->summary = "Test an SRV lookup which returns a single invalid record";
- info->description = "This test defines a single SRV record and performs a\n"
- "resolution of the domain to which they belong. The test ensures that the\n"
- "record is determined to be corrupt as it contains only a priority";
- return AST_TEST_NOT_RUN;
- case TEST_EXECUTE:
- break;
- }
- return invalid_record_test(test, records, ARRAY_LEN(records));
- }
- AST_TEST_DEFINE(srv_resolve_record_missing_port_host)
- {
- struct srv_record records[] = {
- { 10, 10, 5060, "tacos.com", 0, 0, 1, 1 },
- };
- switch (cmd) {
- case TEST_INIT:
- info->name = "srv_resolve_record_missing_port_host";
- info->category = "/main/dns/srv/";
- info->summary = "Test an SRV lookup which returns a single invalid record";
- info->description = "This test defines a single SRV record and performs a\n"
- "resolution of the domain to which they belong. The test ensures that the\n"
- "record is determined to be corrupt as it contains only a priority and weight";
- return AST_TEST_NOT_RUN;
- case TEST_EXECUTE:
- break;
- }
- return invalid_record_test(test, records, ARRAY_LEN(records));
- }
- AST_TEST_DEFINE(srv_resolve_record_missing_host)
- {
- struct srv_record records[] = {
- { 10, 10, 5060, "tacos.com", 0, 0, 0, 1 },
- };
- switch (cmd) {
- case TEST_INIT:
- info->name = "srv_resolve_record_missing_host";
- info->category = "/main/dns/srv/";
- info->summary = "Test an SRV lookup which returns a single invalid record";
- info->description = "This test defines a single SRV record and performs a\n"
- "resolution of the domain to which they belong. The test ensures that the\n"
- "record is determined to be corrupt as it contains only a priority, weight,\n"
- "and port";
- return AST_TEST_NOT_RUN;
- case TEST_EXECUTE:
- break;
- }
- return invalid_record_test(test, records, ARRAY_LEN(records));
- }
- static int unload_module(void)
- {
- AST_TEST_UNREGISTER(srv_resolve_single_record);
- AST_TEST_UNREGISTER(srv_resolve_sort_priority);
- AST_TEST_UNREGISTER(srv_resolve_same_priority_zero_weight);
- AST_TEST_UNREGISTER(srv_resolve_same_priority_different_weights);
- AST_TEST_UNREGISTER(srv_resolve_different_priorities_different_weights);
- AST_TEST_UNREGISTER(srv_resolve_record_missing_weight_port_host);
- AST_TEST_UNREGISTER(srv_resolve_record_missing_port_host);
- AST_TEST_UNREGISTER(srv_resolve_record_missing_host);
- return 0;
- }
- static int load_module(void)
- {
- AST_TEST_REGISTER(srv_resolve_single_record);
- AST_TEST_REGISTER(srv_resolve_sort_priority);
- AST_TEST_REGISTER(srv_resolve_same_priority_zero_weight);
- AST_TEST_REGISTER(srv_resolve_same_priority_different_weights);
- AST_TEST_REGISTER(srv_resolve_different_priorities_different_weights);
- AST_TEST_REGISTER(srv_resolve_record_missing_weight_port_host);
- AST_TEST_REGISTER(srv_resolve_record_missing_port_host);
- AST_TEST_REGISTER(srv_resolve_record_missing_host);
- return AST_MODULE_LOAD_SUCCESS;
- }
- AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "DNS SRV Tests");
|