diff options
author | Pablo Neira Ayuso <pablo@gnumonks.org> | 2013-05-23 19:58:28 +0200 |
---|---|---|
committer | Pablo Neira Ayuso <pablo@netfilter.org> | 2013-05-24 02:20:50 +0200 |
commit | 479fba171ba5d3877fc4024f4df03183ffc363b3 (patch) | |
tree | 25af744f32993b4ef2de3c50d85b738ac7763ae2 /tests/osmux/osmux_test.c | |
parent | dd34adea7a385b2246e6a7445d2bf71dc86e046b (diff) |
add testsuite infrastructure and osmux test
This patch adds the testsuite infrastructure and it populates it
with one test for osmux.
The osmux tests makes sure that:
* We get the same number of RTP messages in the input and the output path.
* The payload of the RTP message is reconstructed correctly.
* The reconstructed timing is correct.
Diffstat (limited to 'tests/osmux/osmux_test.c')
-rw-r--r-- | tests/osmux/osmux_test.c | 159 |
1 files changed, 159 insertions, 0 deletions
diff --git a/tests/osmux/osmux_test.c b/tests/osmux/osmux_test.c new file mode 100644 index 0000000..3f01d6e --- /dev/null +++ b/tests/osmux/osmux_test.c @@ -0,0 +1,159 @@ +/* + * (C) 2013 by Pablo Neira Ayuso <pablo@gnumonks.org> + * (C) 2013 by On Waves ehf <http://www.on-waves.com> + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + */ + +#include <stdio.h> +#include <stdlib.h> +#include <unistd.h> +#include <stdint.h> +#include <string.h> +#include <signal.h> +#include <arpa/inet.h> +#include <sys/time.h> + +#include <osmocom/core/select.h> +#include <osmocom/core/application.h> +#include <osmocom/core/logging.h> +#include <osmocom/core/msgb.h> +#include <osmocom/core/linuxlist.h> +#include <osmocom/netif/osmux.h> +#include <osmocom/netif/rtp.h> + +#define DOSMUX_TEST 0 + +struct log_info_cat osmux_test_cat[] = { + [DOSMUX_TEST] = { + .name = "DOSMUX_TEST", + .description = "osmux test", + .color = "\033[1;35m", + .enabled = 1, .loglevel = LOGL_DEBUG, + }, +}; + +const struct log_info osmux_test_log_info = { + .filter_fn = NULL, + .cat = osmux_test_cat, + .num_cat = ARRAY_SIZE(osmux_test_cat), +}; + +/* RTP packet with AMR payload */ +static uint8_t rtp_pkt[] = { + 0x80, 0x62, 0x3f, 0xcc, 0x00, 0x01, 0xa7, 0x6f, /* RTP */ + 0x07, 0x09, 0x00, 0x62, 0x20, 0x14, 0xff, 0xd4, /* AMR */ + 0xf9, 0xff, 0xfb, 0xe7, 0xeb, 0xf9, 0x9f, 0xf8, + 0xf2, 0x26, 0x33, 0x65, 0x54, +}; + +static int rtp_pkts; +static struct timeval last; + +static void tx_cb(struct msgb *msg, void *data) +{ + char buf[4096]; + struct timeval now, diff; + + gettimeofday(&now, NULL); + timersub(&now, &last, &diff); + last = now; + + if (diff.tv_usec > 17000) { + fprintf(stdout, "lagging to deliver reconstructed RTP\n"); + exit(EXIT_FAILURE); + } + + osmo_rtp_snprintf(buf, sizeof(buf), msg); + fprintf(stderr, "extracted packet: %s\n", buf); + + if (memcmp(msg->data + sizeof(struct rtp_hdr), + rtp_pkt + sizeof(struct rtp_hdr), + sizeof(rtp_pkt) - sizeof(struct rtp_hdr)) != 0) { + fprintf(stdout, "payload mismatch!\n"); + exit(EXIT_FAILURE); + } + + rtp_pkts--; +} + +static struct osmux_out_handle h_output; + +static void osmux_deliver(struct msgb *batch_msg, void *data) +{ + char buf[1024]; + struct osmux_hdr *osmuxh; + LLIST_HEAD(list); + + osmux_snprintf(buf, sizeof(buf), batch_msg); + fprintf(stderr, "OSMUX message (len=%d) %s\n", batch_msg->len, buf); + + /* For each OSMUX message, extract the RTP messages and put them + * in a list. Then, reconstruct transmission timing. + */ + while((osmuxh = osmux_xfrm_output_pull(batch_msg)) != NULL) { + osmux_xfrm_output(osmuxh, &h_output, &list); + osmux_tx_sched(&list, tx_cb, NULL); + } +} + +struct osmux_in_handle h_input = { + .osmux_seq = 0, /* sequence number to start OSmux message from */ + .batch_factor = 4, /* batch up to 4 RTP messages */ + .deliver = osmux_deliver, +}; + +int main(void) +{ + struct msgb *msg; + char buf[1024]; + struct rtp_hdr *rtph = (struct rtp_hdr *)rtp_pkt; + uint16_t seq; + int i, j; + + /* This test doesn't use it, but osmux requires it internally. */ + osmo_init_logging(&osmux_test_log_info); + log_set_log_level(osmo_stderr_target, LOGL_DEBUG); + + osmux_xfrm_input_init(&h_input); + osmux_xfrm_output_init(&h_output, 0x7000000); + + for (i=1; i<64; i++) { + msg = msgb_alloc(1500, "test"); + if (!msg) + return 0; + + memcpy(msg->data, rtp_pkt, sizeof(rtp_pkt)); + msgb_put(msg, sizeof(rtp_pkt)); + + seq = ntohs(rtph->sequence); + seq++; + rtph->sequence = htons(seq); + + osmo_rtp_snprintf(buf, sizeof(buf), msg); + fprintf(stderr, "adding %s\n", buf); + rtp_pkts++; + + osmux_xfrm_input(&h_input, msg, 0); + if (i % 4 == 0) { + gettimeofday(&last, NULL); + + /* After four RTP messages, squash them into the OSMUX + * batch and call the routine to deliver it. + */ + osmux_xfrm_input_deliver(&h_input); + + /* The first RTP message that is delivered immediately, + * wait until the three RTP messages that are extracted + * from OSMUX has been delivered. + */ + for (j=0; j<3; j++) + osmo_select_main(0); + } + } + fprintf(stdout, "OK: Test passed\n"); + return EXIT_SUCCESS; +} |