1 files changed, 142 insertions, 0 deletions
diff --git a/tests/osmo-pcap-test/osmux_test.c b/tests/osmo-pcap-test/osmux_test.c
new file mode 100644
index 0000000..0e47004
--- /dev/null
+++ b/tests/osmo-pcap-test/osmux_test.c
@@ -0,0 +1,142 @@
+/*
+ * (C) 2012 by Pablo Neira Ayuso <pablo@gnumonks.org>
+ * (C) 2012 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 <pcap.h>
+#include <stdlib.h>
+#include <stdint.h>
+#include <stdbool.h>
+#include <netinet/ip.h>
+#include <arpa/inet.h>
+#include <string.h>
+#include <dlfcn.h>
+#include <unistd.h>
+
+#include <osmocom/core/msgb.h>
+#include <osmocom/core/timer.h>
+#include <osmocom/core/select.h>
+
+#include <osmocom/netif/osmux.h>
+
+#include "osmo_pcap.h"
+
+/*
+ * This is the output handle for osmux, it stores last RTP sequence and
+ * timestamp that has been used. There should be one per circuit ID.
+ */
+static struct osmux_out_handle h_output = {
+	.rtp_seq	= 1000,
+	.rtp_timestamp	= 10,
+};
+
+static void tx_cb(struct msgb *msg, void *data)
+{
+	printf("now sending message scheduled [emulated], msg=%p\n", msg);
+	/*
+	 * Here we should call the real function that sends the message
+	 * instead of releasing it.
+	 */
+	msgb_free(msg);
+}
+
+static void deliver(struct msgb *batch_msg)
+{
+	struct osmux_hdr *osmuxh;
+	struct msgb *msg;
+	int i = 0;
+	struct timeval tv = { .tv_sec = 0, .tv_usec = 0 };
+	struct timeval delta = { .tv_sec = 0, .tv_usec = 160000 };
+
+	timerclear(&tv);
+
+	printf("sending batch (len=%d) [emulated]\n", batch_msg->len);
+
+	/* This code below belongs to the osmux receiver */
+	while((osmuxh = osmux_xfrm_output_pull(batch_msg)) != NULL) {
+		msg = osmux_xfrm_output(osmuxh, &h_output);
+		printf("scheduled transmision in %lu.%6lu seconds, "
+		       "msg=%p (%d in batch)\n",
+			tv.tv_sec, tv.tv_usec, msg, ++i);
+		osmux_tx_sched(msg, &tv, tx_cb, NULL);
+		timeradd(&tv, &delta, &tv);
+	}
+}
+
+/*
+ * This is the input handle for osmux. It stores the last osmux sequence that
+ * has been used and the deliver function that sends the osmux batch.
+ */
+struct osmux_in_handle h_input = {
+	.osmux_seq	= 0, /* sequence number to start OSmux message from */
+	.deliver	= deliver,
+};
+
+static int pcap_test_run(struct msgb *msg)
+{
+	int ret;
+retry:
+	ret = osmux_xfrm_input(msg);
+	switch(ret) {
+		case -1:
+			printf("something is wrong\n");
+			break;
+		case 0:
+			break;
+		case 1:
+			osmux_xfrm_input_deliver(&h_input);
+			goto retry;
+	}
+	return 0;
+}
+
+static struct osmo_pcap osmo_pcap;
+
+static void osmo_pcap_pkt_timer_cb(void *data)
+{
+	if (osmo_pcap_test_run(&osmo_pcap, IPPROTO_UDP, pcap_test_run) < 0) {
+		osmo_pcap_stats_printf();
+		printf("\e[1;34mDone.\e[0m\n");
+		osmo_pcap_test_close(osmo_pcap.h);
+		exit(EXIT_SUCCESS);
+	}
+}
+
+int main(int argc, char *argv[])
+{
+	int ret;
+
+	if (argc != 2) {
+		fprintf(stderr, "Wrong usage:\n");
+		fprintf(stderr, "%s <pcap_file>\n", argv[0]);
+		fprintf(stderr, "example: %s file.pcap\n", argv[0]);
+		exit(EXIT_FAILURE);
+	}
+
+	osmo_pcap_init();
+
+	printf("\e[1;34mStarting test...\e[0m\n");
+
+	osmo_pcap.h = osmo_pcap_test_open(argv[1]);
+	if (osmo_pcap.h == NULL)
+		exit(EXIT_FAILURE);
+
+	osmo_pcap.timer.cb = osmo_pcap_pkt_timer_cb;
+
+	/* first run */
+	osmo_pcap_pkt_timer_cb(NULL);
+
+	osmux_xfrm_input_init(&h_input);
+
+	while(1) {
+		osmo_select_main(0);
+	}
+
+	return ret;
+}