libosmo-netif/tests/osmo-pcap-test/osmux_test.c

/*
 * (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;
}