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", ns_t_srv, ns_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;
}
static enum ast_test_result_state off_nominal_test(struct ast_test *test, struct naptr_record *records, int num_records)
{
RAII_VAR(struct ast_dns_result *, result, NULL, ast_dns_result_free);
enum ast_test_result_state res = AST_TEST_PASS;
const struct ast_dns_record *record;
test_records = records;
num_test_records = num_records;
memset(ans_buffer, 0, sizeof(ans_buffer));
ast_dns_resolver_register(&naptr_resolver);
if (ast_dns_resolve("goose.feathers", ns_t_naptr, ns_c_in, &result)) {
ast_test_status_update(test, "Failed to perform DNS resolution, despite using valid inputs\n");
res = AST_TEST_FAIL;
goto cleanup;
}
if (!result) {
ast_test_status_update(test, "Synchronous DNS resolution failed to set a result\n");
res = AST_TEST_FAIL;
goto cleanup;
}
record = ast_dns_result_get_records(result);
if (record) {
ast_test_status_update(test, "DNS resolution returned records when it was not expected to\n");
res = AST_TEST_FAIL;
goto cleanup;
}
cleanup:
ast_dns_resolver_unregister(&naptr_resolver);
test_records = NULL;
num_test_records = 0;
memset(ans_buffer, 0, sizeof(ans_buffer));
return res;
}
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", ns_t_srv, ns_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;
}