cellular-infrastructure
/
osmo-hlr
mirror of https://gerrit.osmocom.org/osmo-hlr
9
0
Fork 0
Code Issues Releases Wiki Activity
osmo-hlr/tests/mslookup/mdns_test.c

460 lines
13 KiB
C
Raw Blame History

/* Copyright 2019 by sysmocom s.f.m.c. GmbH <info@sysmocom.de>
*
* All Rights Reserved
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
#include <assert.h>
#include <errno.h>
#include <string.h>
#include <osmocom/core/application.h>
#include <osmocom/core/logging.h>
#include <osmocom/core/utils.h>
#include <osmocom/mslookup/mdns_rfc.h>
#include <osmocom/mslookup/mdns_msg.h>
struct qname_enc_dec_test {
const char *domain;
const char *qname;
size_t qname_max_len; /* default: strlen(qname) + 1 */
};
#define PRINT_HDR(hdr, name) \
fprintf(stderr, "header %s:\n" \
".id = %i\n" \
".qr = %i\n" \
".opcode = %x\n" \
".aa = %i\n" \
".tc = %i\n" \
".rd = %i\n" \
".ra = %i\n" \
".z = %x\n" \
".rcode = %x\n" \
".qdcount = %u\n" \
".ancount = %u\n" \
".nscount = %u\n" \
".arcount = %u\n", \
name, hdr.id, hdr.qr, hdr.opcode, hdr.aa, hdr.tc, hdr.rd, hdr.ra, hdr.z, hdr.rcode, hdr.qdcount, \
hdr.ancount, hdr.nscount, hdr.arcount)
static const struct osmo_mdns_rfc_header header_enc_dec_test_data[] = {
{
/* Typical use case for mslookup */
.id = 1337,
.qdcount = 1,
},
{
/* Fill out everything */
.id = 42,
.qr = 1,
.opcode = 0x02,
.aa = 1,
.tc = 1,
.rd = 1,
.ra = 1,
.z = 0x02,
.rcode = 0x03,
.qdcount = 1234,
.ancount = 1111,
.nscount = 2222,
.arcount = 3333,
},
};
void test_enc_dec_rfc_header(void)
{
int i;
fprintf(stderr, "-- %s --\n", __func__);
for (i = 0; i< ARRAY_SIZE(header_enc_dec_test_data); i++) {
const struct osmo_mdns_rfc_header in = header_enc_dec_test_data[i];
struct osmo_mdns_rfc_header out = {0};
struct msgb *msg = msgb_alloc(4096, "dns_test");
PRINT_HDR(in, "in");
osmo_mdns_rfc_header_encode(msg, &in);
fprintf(stderr, "encoded: %s\n", osmo_hexdump(msgb_data(msg), msgb_length(msg)));
assert(osmo_mdns_rfc_header_decode(msgb_data(msg), msgb_length(msg), &out) == 0);
PRINT_HDR(out, "out");
fprintf(stderr, "in (hexdump): %s\n", osmo_hexdump((unsigned char *)&in, sizeof(in)));
fprintf(stderr, "out (hexdump): %s\n", osmo_hexdump((unsigned char *)&out, sizeof(out)));
assert(memcmp(&in, &out, sizeof(in)) == 0);
fprintf(stderr, "=> OK\n\n");
msgb_free(msg);
}
}
void test_enc_dec_rfc_header_einval(void)
{
struct osmo_mdns_rfc_header out = {0};
struct msgb *msg = msgb_alloc(4096, "dns_test");
fprintf(stderr, "-- %s --\n", __func__);
assert(osmo_mdns_rfc_header_decode(msgb_data(msg), 11, &out) == -EINVAL);
fprintf(stderr, "=> OK\n\n");
msgb_free(msg);
}
#define PRINT_QST(qst, name) \
fprintf(stderr, "question %s:\n" \
".domain = %s\n" \
".qtype = %i\n" \
".qclass = %i\n", \
name, (qst)->domain, (qst)->qtype, (qst)->qclass)
static const struct osmo_mdns_rfc_question question_enc_dec_test_data[] = {
{
.domain = "hlr.1234567.imsi",
.qtype = OSMO_MDNS_RFC_RECORD_TYPE_ALL,
.qclass = OSMO_MDNS_RFC_CLASS_IN,
},
{
.domain = "hlr.1234567.imsi",
.qtype = OSMO_MDNS_RFC_RECORD_TYPE_A,
.qclass = OSMO_MDNS_RFC_CLASS_ALL,
},
{
.domain = "hlr.1234567.imsi",
.qtype = OSMO_MDNS_RFC_RECORD_TYPE_AAAA,
.qclass = OSMO_MDNS_RFC_CLASS_ALL,
},
};
void test_enc_dec_rfc_question(void *ctx)
{
int i;
fprintf(stderr, "-- %s --\n", __func__);
for (i = 0; i< ARRAY_SIZE(question_enc_dec_test_data); i++) {
const struct osmo_mdns_rfc_question in = question_enc_dec_test_data[i];
struct osmo_mdns_rfc_question *out;
struct msgb *msg = msgb_alloc(4096, "dns_test");
PRINT_QST(&in, "in");
assert(osmo_mdns_rfc_question_encode(ctx, msg, &in) == 0);
fprintf(stderr, "encoded: %s\n", osmo_hexdump(msgb_data(msg), msgb_length(msg)));
out = osmo_mdns_rfc_question_decode(ctx, msgb_data(msg), msgb_length(msg));
assert(out);
PRINT_QST(out, "out");
if (strcmp(in.domain, out->domain) != 0)
fprintf(stderr, "=> ERROR: domain does not match\n");
else if (in.qtype != out->qtype)
fprintf(stderr, "=> ERROR: qtype does not match\n");
else if (in.qclass != out->qclass)
fprintf(stderr, "=> ERROR: qclass does not match\n");
else
fprintf(stderr, "=> OK\n");
fprintf(stderr, "\n");
msgb_free(msg);
talloc_free(out);
}
}
void test_enc_dec_rfc_question_null(void *ctx)
{
uint8_t data[5] = {0};
fprintf(stderr, "-- %s --\n", __func__);
assert(osmo_mdns_rfc_question_decode(ctx, data, sizeof(data)) == NULL);
fprintf(stderr, "=> OK\n\n");
}
#define PRINT_REC(rec, name) \
fprintf(stderr, "question %s:\n" \
".domain = %s\n" \
".type = %i\n" \
".class = %i\n" \
".ttl = %i\n" \
".rdlength = %i\n" \
".rdata = %s\n", \
name, (rec)->domain, (rec)->type, (rec)->class, (rec)->ttl, (rec)->rdlength, \
osmo_quote_str((char *)(rec)->rdata, (rec)->rdlength))
static const struct osmo_mdns_rfc_record record_enc_dec_test_data[] = {
{
.domain = "hlr.1234567.imsi",
.type = OSMO_MDNS_RFC_RECORD_TYPE_A,
.class = OSMO_MDNS_RFC_CLASS_IN,
.ttl = 1234,
.rdlength = 9,
.rdata = (uint8_t *)"10.42.2.1",
},
};
void test_enc_dec_rfc_record(void *ctx)
{
int i;
fprintf(stderr, "-- %s --\n", __func__);
for (i=0; i< ARRAY_SIZE(record_enc_dec_test_data); i++) {
const struct osmo_mdns_rfc_record in = record_enc_dec_test_data[i];
struct osmo_mdns_rfc_record *out;
struct msgb *msg = msgb_alloc(4096, "dns_test");
size_t record_len;
PRINT_REC(&in, "in");
assert(osmo_mdns_rfc_record_encode(ctx, msg, &in) == 0);
fprintf(stderr, "encoded: %s\n", osmo_hexdump(msgb_data(msg), msgb_length(msg)));
out = osmo_mdns_rfc_record_decode(ctx, msgb_data(msg), msgb_length(msg), &record_len);
fprintf(stderr, "record_len: %lu\n", record_len);
assert(out);
PRINT_REC(out, "out");
if (strcmp(in.domain, out->domain) != 0)
fprintf(stderr, "=> ERROR: domain does not match\n");
else if (in.type != out->type)
fprintf(stderr, "=> ERROR: type does not match\n");
else if (in.class != out->class)
fprintf(stderr, "=> ERROR: class does not match\n");
else if (in.ttl != out->ttl)
fprintf(stderr, "=> ERROR: ttl does not match\n");
else if (in.rdlength != out->rdlength)
fprintf(stderr, "=> ERROR: rdlength does not match\n");
else if (memcmp(in.rdata, out->rdata, in.rdlength) != 0)
fprintf(stderr, "=> ERROR: rdata does not match\n");
else
fprintf(stderr, "=> OK\n");
fprintf(stderr, "\n");
msgb_free(msg);
talloc_free(out);
}
}
static uint8_t ip_v4_n[] = {23, 42, 47, 11};
static uint8_t ip_v6_n[] = {0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88,
0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff, 0x00};
enum test_records {
RECORD_NONE,
RECORD_A,
RECORD_AAAA,
RECORD_TXT_AGE,
RECORD_TXT_PORT_444,
RECORD_TXT_PORT_666,
RECORD_TXT_INVALID_KEY,
RECORD_TXT_INVALID_NO_KEY_VALUE,
RECORD_INVALID,
};
struct result_from_answer_test {
const char *desc;
const enum test_records records[5];
bool error;
const struct osmo_mslookup_result res;
};
static void test_result_from_answer(void *ctx)
{
void *print_ctx = talloc_named_const(ctx, 0, __func__);
struct osmo_sockaddr_str test_host_v4 = {.af = AF_INET, .port=444, .ip = "23.42.47.11"};
struct osmo_sockaddr_str test_host_v6 = {.af = AF_INET6, .port=666,
.ip = "1122:3344:5566:7788:99aa:bbcc:ddee:ff00"};
struct osmo_mslookup_result test_result_v4 = {.rc = OSMO_MSLOOKUP_RC_RESULT, .age = 3,
.host_v4 = test_host_v4};
struct osmo_mslookup_result test_result_v6 = {.rc = OSMO_MSLOOKUP_RC_RESULT, .age = 3,
.host_v6 = test_host_v6};
struct osmo_mslookup_result test_result_v4_v6 = {.rc = OSMO_MSLOOKUP_RC_RESULT, .age = 3,
.host_v4 = test_host_v4, .host_v6 = test_host_v6};
struct result_from_answer_test result_from_answer_data[] = {
{
.desc = "IPv4",
.records = {RECORD_TXT_AGE, RECORD_A, RECORD_TXT_PORT_444},
.res = test_result_v4
},
{
.desc = "IPv6",
.records = {RECORD_TXT_AGE, RECORD_AAAA, RECORD_TXT_PORT_666},
.res = test_result_v6
},
{
.desc = "IPv4 + IPv6",
.records = {RECORD_TXT_AGE, RECORD_A, RECORD_TXT_PORT_444, RECORD_AAAA, RECORD_TXT_PORT_666},
.res = test_result_v4_v6
},
{
.desc = "A twice",
.records = {RECORD_TXT_AGE, RECORD_A, RECORD_TXT_PORT_444, RECORD_A},
.error = true
},
{
.desc = "AAAA twice",
.records = {RECORD_TXT_AGE, RECORD_AAAA, RECORD_TXT_PORT_444, RECORD_AAAA},
.error = true
},
{
.desc = "invalid TXT: no key/value pair",
.records = {RECORD_TXT_AGE, RECORD_AAAA, RECORD_TXT_INVALID_NO_KEY_VALUE},
.error = true
},
{
.desc = "age twice",
.records = {RECORD_TXT_AGE, RECORD_TXT_AGE},
.error = true
},
{
.desc = "port as first record",
.records = {RECORD_TXT_PORT_444},
.error = true
},
{
.desc = "port without previous ip record",
.records = {RECORD_TXT_AGE, RECORD_TXT_PORT_444},
.error = true
},
{
.desc = "invalid TXT: invalid key",
.records = {RECORD_TXT_AGE, RECORD_AAAA, RECORD_TXT_INVALID_KEY},
.error = true
},
{
.desc = "unexpected record type",
.records = {RECORD_TXT_AGE, RECORD_INVALID},
.error = true
},
{
.desc = "missing record: age",
.records = {RECORD_A, RECORD_TXT_PORT_444},
.error = true
},
{
.desc = "missing record: port for ipv4",
.records = {RECORD_TXT_AGE, RECORD_A},
.error = true
},
{
.desc = "missing record: port for ipv4 #2",
.records = {RECORD_TXT_AGE, RECORD_AAAA, RECORD_TXT_PORT_666, RECORD_A},
.error = true
},
};
int i = 0;
int j = 0;
fprintf(stderr, "-- %s --\n", __func__);
for (i = 0; i < ARRAY_SIZE(result_from_answer_data); i++) {
struct result_from_answer_test *t = &result_from_answer_data[i];
struct osmo_mdns_msg_answer ans = {0};
struct osmo_mslookup_result res = {0};
void *ctx_test = talloc_named_const(ctx, 0, t->desc);
bool is_error;
fprintf(stderr, "---\n");
fprintf(stderr, "test: %s\n", t->desc);
fprintf(stderr, "error: %s\n", t->error ? "true" : "false");
fprintf(stderr, "records:\n");
/* Build records list */
INIT_LLIST_HEAD(&ans.records);
for (j = 0; j < ARRAY_SIZE(t->records); j++) {
struct osmo_mdns_record *rec = NULL;
switch (t->records[j]) {
case RECORD_NONE:
break;
case RECORD_A:
fprintf(stderr, "- A 42.42.42.42\n");
rec = talloc_zero(ctx_test, struct osmo_mdns_record);
rec->type = OSMO_MDNS_RFC_RECORD_TYPE_A;
rec->data = ip_v4_n;
rec->length = sizeof(ip_v4_n);
break;
case RECORD_AAAA:
fprintf(stderr, "- AAAA 1122:3344:5566:7788:99aa:bbcc:ddee:ff00\n");
rec = talloc_zero(ctx_test, struct osmo_mdns_record);
rec->type = OSMO_MDNS_RFC_RECORD_TYPE_AAAA;
rec->data = ip_v6_n;
rec->length = sizeof(ip_v6_n);
break;
case RECORD_TXT_AGE:
fprintf(stderr, "- TXT age=3\n");
rec = osmo_mdns_record_txt_keyval_encode(ctx_test, "age", "3");
break;
case RECORD_TXT_PORT_444:
fprintf(stderr, "- TXT port=444\n");
rec = osmo_mdns_record_txt_keyval_encode(ctx_test, "port", "444");
break;
case RECORD_TXT_PORT_666:
fprintf(stderr, "- TXT port=666\n");
rec = osmo_mdns_record_txt_keyval_encode(ctx_test, "port", "666");
break;
case RECORD_TXT_INVALID_KEY:
fprintf(stderr, "- TXT hello=world\n");
rec = osmo_mdns_record_txt_keyval_encode(ctx_test, "hello", "world");
break;
case RECORD_TXT_INVALID_NO_KEY_VALUE:
fprintf(stderr, "- TXT 12345\n");
rec = osmo_mdns_record_txt_keyval_encode(ctx_test, "12", "45");
rec->data[3] = '3';
break;
case RECORD_INVALID:
fprintf(stderr, "- (invalid)\n");
rec = talloc_zero(ctx, struct osmo_mdns_record);
rec->type = OSMO_MDNS_RFC_RECORD_TYPE_UNKNOWN;
break;
}
if (rec)
llist_add_tail(&rec->list, &ans.records);
}
/* Verify output */
is_error = (osmo_mdns_result_from_answer(&res, &ans) != 0);
if (t->error != is_error) {
fprintf(stderr, "got %s\n", is_error ? "error" : "no error");
OSMO_ASSERT(false);
}
if (!t->error) {
fprintf(stderr, "exp: %s\n", osmo_mslookup_result_name_c(print_ctx, NULL, &t->res));
fprintf(stderr, "res: %s\n", osmo_mslookup_result_name_c(print_ctx, NULL, &res));
OSMO_ASSERT(t->res.rc == res.rc);
OSMO_ASSERT(!osmo_sockaddr_str_cmp(&t->res.host_v4, &res.host_v4));
OSMO_ASSERT(!osmo_sockaddr_str_cmp(&t->res.host_v6, &res.host_v6));
OSMO_ASSERT(t->res.age == res.age);
OSMO_ASSERT(t->res.last == res.last);
}
talloc_free(ctx_test);
fprintf(stderr, "=> OK\n");
}
}
int main(int argc, char **argv)
{
void *ctx = talloc_named_const(NULL, 0, "main");
osmo_init_logging2(ctx, NULL);
log_set_print_filename2(osmo_stderr_target, LOG_FILENAME_NONE);
log_set_print_level(osmo_stderr_target, 1);
log_set_print_category(osmo_stderr_target, 1);
log_set_print_category_hex(osmo_stderr_target, 0);
log_set_use_color(osmo_stderr_target, 0);
test_enc_dec_rfc_header();
test_enc_dec_rfc_header_einval();
test_enc_dec_rfc_question(ctx);
test_enc_dec_rfc_question_null(ctx);
test_enc_dec_rfc_record(ctx);
test_result_from_answer(ctx);
return 0;
}
View Git Blame Copy Permalink