From b9cf903bbe0cd9b84a1226c345e68f0e826034e9 Mon Sep 17 00:00:00 2001
From: Pablo Neira Ayuso <pablo@gnumonks.org>
Date: Mon, 6 Aug 2012 18:56:36 +0200
Subject: [PATCH] osmux: add infrastructure to map RTP SSRC and osmux CCID

---
 include/osmocom/netif/osmux.h     |  2 ++
 src/osmux.c                       | 52 ++++++++++++++++++++++++++++---
 tests/osmo-pcap-test/osmux_test.c |  9 ++++++
 3 files changed, 59 insertions(+), 4 deletions(-)

diff --git a/include/osmocom/netif/osmux.h b/include/osmocom/netif/osmux.h
index 5fcc583..58ec40d 100644
--- a/include/osmocom/netif/osmux.h
+++ b/include/osmocom/netif/osmux.h
@@ -59,6 +59,8 @@ static inline uint8_t *osmux_get_payload(struct osmux_hdr *osmuxh)
 }
 
 void osmux_xfrm_input_init(struct osmux_in_handle *h);
+int osmux_xfrm_input_get_ccid(struct osmux_in_handle *h, uint32_t ssrc);
+void osmux_xfrm_input_register_ccid(struct osmux_in_handle *h, uint32_t ssrc);
 
 int osmux_xfrm_input(struct osmux_in_handle *h, struct msgb *msg);
 void osmux_xfrm_input_deliver(struct osmux_in_handle *h);
diff --git a/src/osmux.c b/src/osmux.c
index 9f8e25c..508f01c 100644
--- a/src/osmux.c
+++ b/src/osmux.c
@@ -58,6 +58,7 @@ osmux_rebuild_rtp(struct osmux_out_handle *h,
 	struct msgb *out_msg;
 	struct rtp_hdr *rtph;
 	struct amr_hdr *amrh;
+	uint32_t ssrc_from_ccid = osmuxh->circuit_id;
 	char buf[4096];
 
 	out_msg = msgb_alloc(sizeof(struct rtp_hdr) +
@@ -76,7 +77,7 @@ osmux_rebuild_rtp(struct osmux_out_handle *h,
 	/* ... emulate timestamp and ssrc */
 	rtph->timestamp = htonl(h->rtp_timestamp);
 	rtph->sequence = htons(h->rtp_seq);
-	rtph->ssrc = osmuxh->circuit_id;
+	rtph->ssrc = htonl(ssrc_from_ccid);
 
 	msgb_put(out_msg, sizeof(struct rtp_hdr));
 
@@ -133,13 +134,14 @@ struct osmux_batch {
 	struct llist_head	msgb_list;
 	unsigned int		remaining_bytes;
 	uint8_t			seq;
+	int64_t			ccid[8];
 };
 
 static int
 osmux_batch_add(struct osmux_in_handle *h, struct msgb *out_msg,
 		struct msgb *msg, struct rtp_hdr *rtph,
 		struct amr_hdr *amrh, uint32_t amr_payload_len,
-		uint8_t circuit_id, int add_osmux_header)
+		int add_osmux_header)
 {
 	struct osmux_batch *batch = (struct osmux_batch *)h->data;
 	struct osmux_hdr *osmuxh;
@@ -151,7 +153,7 @@ osmux_batch_add(struct osmux_in_handle *h, struct msgb *out_msg,
 		osmuxh->amr_f = amrh->f;
 		osmuxh->amr_q= amrh->q;
 		osmuxh->seq = batch->seq++;
-		osmuxh->circuit_id = circuit_id;
+		osmuxh->circuit_id = osmux_xfrm_input_get_ccid(h, rtph->ssrc);
 		osmuxh->amr_cmr = amrh->cmr;
 		osmuxh->amr_ft = amrh->ft;
 		msgb_put(out_msg, sizeof(struct osmux_hdr));
@@ -191,7 +193,7 @@ osmux_xfrm_encode_amr(struct osmux_in_handle *h,
 	amr_payload_len = amr_len - sizeof(struct amr_hdr);
 
 	if (osmux_batch_add(h, out_msg, msg, rtph, amrh, amr_payload_len,
-			    0, add_osmux_header) < 0)
+			    add_osmux_header) < 0)
 		return -1;
 
 	return 0;
@@ -370,6 +372,7 @@ int osmux_xfrm_input(struct osmux_in_handle *h, struct msgb *msg)
 void osmux_xfrm_input_init(struct osmux_in_handle *h)
 {
 	struct osmux_batch *batch;
+	int i;
 
 	LOGP(DOSMUX, LOGL_DEBUG, "initialized osmux input converter\n");
 
@@ -382,6 +385,9 @@ void osmux_xfrm_input_init(struct osmux_in_handle *h)
 	batch->timer.cb = osmux_batch_timer_expired;
 	batch->timer.data = h;
 
+	for (i=0; i<8; i++)
+		batch->ccid[i] = -1;
+
 	h->data = (void *)batch;
 }
 
@@ -463,3 +469,41 @@ osmux_tx_sched(struct llist_head *list,
 		llist_del(&cur->list);
 	}
 }
+
+void osmux_xfrm_input_register_ccid(struct osmux_in_handle *h, uint32_t ssrc)
+{
+	struct osmux_batch *batch = (struct osmux_batch *)h->data;;
+	int i, found = 0;
+
+	for (i=0; i<8; i++) {
+		if (batch->ccid[i] == ssrc)
+			continue;
+		if (batch->ccid[i] < 0) {
+			found = 1;
+			break;
+		}
+	}
+
+	if (found) {
+		batch->ccid[i] = ssrc;
+		LOGP(DOSMUX, LOGL_DEBUG, "mapping ssrc=%u to ccid=%d\n",
+			ntohl(ssrc), i);
+	} else {
+		LOGP(DOSMUX, LOGL_ERROR, "cannot map ssrc to ccid!\n");
+	}
+}
+
+int osmux_xfrm_input_get_ccid(struct osmux_in_handle *h, uint32_t ssrc)
+{
+	struct osmux_batch *batch = (struct osmux_batch *)h->data;;
+	int i, found = 0;
+
+	for (i=0; i<8; i++) {
+		if (batch->ccid[i] == ssrc) {
+			found = 1;
+			break;
+		}
+	}
+
+	return found ? i : -1;
+}
diff --git a/tests/osmo-pcap-test/osmux_test.c b/tests/osmo-pcap-test/osmux_test.c
index 157947a..8dab1bf 100644
--- a/tests/osmo-pcap-test/osmux_test.c
+++ b/tests/osmo-pcap-test/osmux_test.c
@@ -26,6 +26,7 @@
 #include <osmocom/core/application.h>
 #include <osmocom/core/talloc.h>
 
+#include <osmocom/netif/rtp.h>
 #include <osmocom/netif/osmux.h>
 
 #include "osmo_pcap.h"
@@ -77,6 +78,14 @@ struct osmux_in_handle h_input = {
 static int pcap_test_run(struct msgb *msg)
 {
 	int ret;
+	struct rtp_hdr *rtph;
+
+	rtph = osmo_rtp_get_hdr(msg);
+	if (rtph == NULL)
+		return 0;
+
+	if (osmux_xfrm_input_get_ccid(&h_input, rtph->ssrc) < 0)
+		osmux_xfrm_input_register_ccid(&h_input, rtph->ssrc);
 
 	while ((ret = osmux_xfrm_input(&h_input, msg)) > 1) {
 		/* batch full, deliver it */