blob: 95729ef054e99057844217f6cffe30fd8cd9e880 [file] [log] [blame]
Harald Welteead7a7b2009-07-28 00:01:58 +02001/* RTP proxy handling for ip.access nanoBTS */
2
Harald Welte647db842013-02-03 12:06:58 +01003/* (C) 2009-2013 by Harald Welte <laforge@gnumonks.org>
Harald Welteead7a7b2009-07-28 00:01:58 +02004 * All Rights Reserved
5 *
6 * This program is free software; you can redistribute it and/or modify
Harald Welte9af6ddf2011-01-01 15:25:50 +01007 * it under the terms of the GNU Affero General Public License as published by
8 * the Free Software Foundation; either version 3 of the License, or
Harald Welteead7a7b2009-07-28 00:01:58 +02009 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
Harald Welte9af6ddf2011-01-01 15:25:50 +010014 * GNU Affero General Public License for more details.
Harald Welteead7a7b2009-07-28 00:01:58 +020015 *
Harald Welte9af6ddf2011-01-01 15:25:50 +010016 * You should have received a copy of the GNU Affero General Public License
17 * along with this program. If not, see <http://www.gnu.org/licenses/>.
Harald Welteead7a7b2009-07-28 00:01:58 +020018 *
19 */
20
21#include <errno.h>
22#include <unistd.h>
23#include <sys/socket.h>
24#include <netinet/in.h>
Harald Welte805f6442009-07-28 18:25:29 +020025#include <arpa/inet.h>
Harald Welteda7ab742009-12-19 22:23:05 +010026#include <sys/time.h> /* gettimeofday() */
27#include <unistd.h> /* get..() */
28#include <time.h> /* clock() */
29#include <sys/utsname.h> /* uname() */
Harald Welteead7a7b2009-07-28 00:01:58 +020030
Pablo Neira Ayuso136f4532011-03-22 16:47:59 +010031#include <osmocom/core/talloc.h>
Harald Welteead7a7b2009-07-28 00:01:58 +020032#include <openbsc/gsm_data.h>
Pablo Neira Ayuso136f4532011-03-22 16:47:59 +010033#include <osmocom/core/msgb.h>
34#include <osmocom/core/select.h>
Harald Welte805f6442009-07-28 18:25:29 +020035#include <openbsc/debug.h>
Harald Welteead7a7b2009-07-28 00:01:58 +020036#include <openbsc/rtp_proxy.h>
Harald Weltef142c972011-05-24 13:25:38 +020037#include <openbsc/mncc.h>
Alexander Huemer0a654612011-09-06 00:09:49 +020038#include <openbsc/trau_upqueue.h>
Harald Welteead7a7b2009-07-28 00:01:58 +020039
Holger Hans Peter Freyther3cb28792010-10-12 15:39:46 +020040/* attempt to determine byte order */
Holger Hans Peter Freyther3cb28792010-10-12 15:39:46 +020041#include <sys/param.h>
42#include <limits.h>
43
44#ifndef __BYTE_ORDER
Tobias Engelaff20712012-10-24 17:53:50 +020045# ifdef __APPLE__
46# define __BYTE_ORDER __DARWIN_BYTE_ORDER
47# define __LITTLE_ENDIAN __DARWIN_LITTLE_ENDIAN
48# define __BIG_ENDIAN __DARWIN_BIG_ENDIAN
49# else
50# error "__BYTE_ORDER should be defined by someone"
51# endif
Holger Hans Peter Freyther3cb28792010-10-12 15:39:46 +020052#endif
53
Harald Welteead7a7b2009-07-28 00:01:58 +020054static LLIST_HEAD(rtp_sockets);
55
Harald Welte805f6442009-07-28 18:25:29 +020056/* should we mangle the CNAME inside SDES of RTCP packets? We disable
57 * this by default, as it seems to be not needed */
58static int mangle_rtcp_cname = 0;
59
Harald Welteead7a7b2009-07-28 00:01:58 +020060enum rtp_bfd_priv {
61 RTP_PRIV_NONE,
62 RTP_PRIV_RTP,
63 RTP_PRIV_RTCP
64};
65
66#define RTP_ALLOC_SIZE 1500
67
Harald Welte805f6442009-07-28 18:25:29 +020068/* according to RFC 1889 */
69struct rtcp_hdr {
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +020070 uint8_t byte0;
71 uint8_t type;
72 uint16_t length;
Harald Welte805f6442009-07-28 18:25:29 +020073} __attribute__((packed));
74
75#define RTCP_TYPE_SDES 202
76
77#define RTCP_IE_CNAME 1
78
Harald Welteda7ab742009-12-19 22:23:05 +010079/* according to RFC 3550 */
80struct rtp_hdr {
Holger Hans Peter Freyther2890d102010-02-26 13:12:41 +010081#if __BYTE_ORDER == __LITTLE_ENDIAN
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +020082 uint8_t csrc_count:4,
Harald Welteda7ab742009-12-19 22:23:05 +010083 extension:1,
84 padding:1,
85 version:2;
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +020086 uint8_t payload_type:7,
Harald Welteda7ab742009-12-19 22:23:05 +010087 marker:1;
Holger Hans Peter Freyther2890d102010-02-26 13:12:41 +010088#elif __BYTE_ORDER == __BIG_ENDIAN
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +020089 uint8_t version:2,
Holger Hans Peter Freyther2890d102010-02-26 13:12:41 +010090 padding:1,
91 extension:1,
92 csrc_count:4;
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +020093 uint8_t marker:1,
Holger Hans Peter Freyther2890d102010-02-26 13:12:41 +010094 payload_type:7;
95#endif
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +020096 uint16_t sequence;
97 uint32_t timestamp;
98 uint32_t ssrc;
Harald Welteda7ab742009-12-19 22:23:05 +010099} __attribute__((packed));
100
101struct rtp_x_hdr {
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +0200102 uint16_t by_profile;
103 uint16_t length;
Harald Welteda7ab742009-12-19 22:23:05 +0100104} __attribute__((packed));
105
106#define RTP_VERSION 2
107
Harald Welteda7ab742009-12-19 22:23:05 +0100108/* decode an rtp frame and create a new buffer with payload */
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +0200109static int rtp_decode(struct msgb *msg, uint32_t callref, struct msgb **data)
Harald Welteda7ab742009-12-19 22:23:05 +0100110{
111 struct msgb *new_msg;
112 struct gsm_data_frame *frame;
113 struct rtp_hdr *rtph = (struct rtp_hdr *)msg->data;
114 struct rtp_x_hdr *rtpxh;
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +0200115 uint8_t *payload;
Harald Welteda7ab742009-12-19 22:23:05 +0100116 int payload_len;
117 int msg_type;
118 int x_len;
119
120 if (msg->len < 12) {
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200121 DEBUGPC(DLMUX, "received RTP frame too short (len = %d)\n",
Harald Welteda7ab742009-12-19 22:23:05 +0100122 msg->len);
123 return -EINVAL;
124 }
125 if (rtph->version != RTP_VERSION) {
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200126 DEBUGPC(DLMUX, "received RTP version %d not supported.\n",
Harald Welteda7ab742009-12-19 22:23:05 +0100127 rtph->version);
128 return -EINVAL;
129 }
130 payload = msg->data + sizeof(struct rtp_hdr) + (rtph->csrc_count << 2);
131 payload_len = msg->len - sizeof(struct rtp_hdr) - (rtph->csrc_count << 2);
132 if (payload_len < 0) {
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200133 DEBUGPC(DLMUX, "received RTP frame too short (len = %d, "
Harald Welteda7ab742009-12-19 22:23:05 +0100134 "csrc count = %d)\n", msg->len, rtph->csrc_count);
135 return -EINVAL;
136 }
137 if (rtph->extension) {
138 if (payload_len < sizeof(struct rtp_x_hdr)) {
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200139 DEBUGPC(DLMUX, "received RTP frame too short for "
Harald Welteda7ab742009-12-19 22:23:05 +0100140 "extension header\n");
141 return -EINVAL;
142 }
143 rtpxh = (struct rtp_x_hdr *)payload;
144 x_len = ntohs(rtpxh->length) * 4 + sizeof(struct rtp_x_hdr);
145 payload += x_len;
146 payload_len -= x_len;
147 if (payload_len < 0) {
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200148 DEBUGPC(DLMUX, "received RTP frame too short, "
Harald Welteda7ab742009-12-19 22:23:05 +0100149 "extension header exceeds frame length\n");
150 return -EINVAL;
151 }
152 }
153 if (rtph->padding) {
Jacob Erlbeckd9e40392013-11-21 19:05:42 +0100154 if (payload_len < 1) {
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200155 DEBUGPC(DLMUX, "received RTP frame too short for "
Harald Welteda7ab742009-12-19 22:23:05 +0100156 "padding length\n");
157 return -EINVAL;
158 }
159 payload_len -= payload[payload_len - 1];
160 if (payload_len < 0) {
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200161 DEBUGPC(DLMUX, "received RTP frame with padding "
Harald Welteda7ab742009-12-19 22:23:05 +0100162 "greater than payload\n");
163 return -EINVAL;
164 }
165 }
166
167 switch (rtph->payload_type) {
168 case RTP_PT_GSM_FULL:
169 msg_type = GSM_TCHF_FRAME;
Andreas Eversberg88012b62014-01-22 10:05:51 +0100170 if (payload_len != RTP_LEN_GSM_FULL) {
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200171 DEBUGPC(DLMUX, "received RTP full rate frame with "
Andreas Eversberg88012b62014-01-22 10:05:51 +0100172 "payload length != %d (len = %d)\n",
173 RTP_LEN_GSM_FULL, payload_len);
Harald Welteda7ab742009-12-19 22:23:05 +0100174 return -EINVAL;
175 }
176 break;
Harald Welteaca8f152009-12-19 23:06:41 +0100177 case RTP_PT_GSM_EFR:
178 msg_type = GSM_TCHF_FRAME_EFR;
Andreas Eversberg88012b62014-01-22 10:05:51 +0100179 if (payload_len != RTP_LEN_GSM_EFR) {
180 DEBUGPC(DLMUX, "received RTP extended full rate frame "
181 "with payload length != %d (len = %d)\n",
182 RTP_LEN_GSM_EFR, payload_len);
183 return -EINVAL;
184 }
Harald Welteaca8f152009-12-19 23:06:41 +0100185 break;
Andreas Eversberg63bfdd82014-01-17 19:06:38 +0100186 case RTP_PT_GSM_HALF:
187 msg_type = GSM_TCHH_FRAME;
188 if (payload_len != RTP_LEN_GSM_HALF) {
189 DEBUGPC(DLMUX, "received RTP half rate frame with "
190 "payload length != %d (len = %d)\n",
191 RTP_LEN_GSM_HALF, payload_len);
192 return -EINVAL;
193 }
194 break;
Andreas Eversbergd8967f72012-03-08 14:39:19 +0100195 case RTP_PT_AMR:
196 break;
Harald Welteda7ab742009-12-19 22:23:05 +0100197 default:
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200198 DEBUGPC(DLMUX, "received RTP frame with unknown payload "
Harald Welteda7ab742009-12-19 22:23:05 +0100199 "type %d\n", rtph->payload_type);
200 return -EINVAL;
201 }
202
Andreas Eversbergd8967f72012-03-08 14:39:19 +0100203 if (payload_len > 33) {
204 DEBUGPC(DLMUX, "RTP payload too large (%d octets)\n",
205 payload_len);
206 return -EINVAL;
207 }
208
209 if (rtph->payload_type == RTP_PT_AMR) {
210 new_msg = msgb_alloc(sizeof(struct gsm_data_frame) + 1
211 + payload_len, "GSM-DATA (AMR)");
212 } else {
213 new_msg = msgb_alloc(sizeof(struct gsm_data_frame)
214 + payload_len, "GSM-DATA");
215 }
Harald Welteda7ab742009-12-19 22:23:05 +0100216 if (!new_msg)
217 return -ENOMEM;
218 frame = (struct gsm_data_frame *)(new_msg->data);
219 frame->msg_type = msg_type;
220 frame->callref = callref;
Andreas Eversbergd8967f72012-03-08 14:39:19 +0100221 if (rtph->payload_type == RTP_PT_AMR) {
222 frame->data[0] = payload_len;
223 msgb_put(new_msg, sizeof(struct gsm_data_frame) + 1
224 + payload_len);
225 memcpy(frame->data + 1, payload, payload_len);
226 } else {
227 msgb_put(new_msg, sizeof(struct gsm_data_frame) + payload_len);
228 memcpy(frame->data, payload, payload_len);
229 }
Harald Welteda7ab742009-12-19 22:23:05 +0100230
231 *data = new_msg;
232 return 0;
233}
234
Harald Welte392736d2009-12-20 13:16:14 +0100235/* "to - from" */
236static void tv_difference(struct timeval *diff, const struct timeval *from,
237 const struct timeval *__to)
238{
239 struct timeval _to = *__to, *to = &_to;
240
241 if (to->tv_usec < from->tv_usec) {
242 to->tv_sec -= 1;
243 to->tv_usec += 1000000;
244 }
245
246 diff->tv_usec = to->tv_usec - from->tv_usec;
247 diff->tv_sec = to->tv_sec - from->tv_sec;
248}
249
Harald Welte647db842013-02-03 12:06:58 +0100250/*! \brief encode and send a rtp frame
251 * \param[in] rs RTP socket through which we shall send
252 * \param[in] frame GSM RTP frame to be sent
253 */
Harald Welteda7ab742009-12-19 22:23:05 +0100254int rtp_send_frame(struct rtp_socket *rs, struct gsm_data_frame *frame)
255{
256 struct rtp_sub_socket *rss = &rs->rtp;
257 struct msgb *msg;
258 struct rtp_hdr *rtph;
259 int payload_type;
260 int payload_len;
261 int duration; /* in samples */
262
263 if (rs->tx_action != RTP_SEND_DOWNSTREAM) {
264 /* initialize sequences */
265 rs->tx_action = RTP_SEND_DOWNSTREAM;
266 rs->transmit.ssrc = rand();
267 rs->transmit.sequence = random();
268 rs->transmit.timestamp = random();
269 }
270
271 switch (frame->msg_type) {
272 case GSM_TCHF_FRAME:
273 payload_type = RTP_PT_GSM_FULL;
Andreas Eversberg88012b62014-01-22 10:05:51 +0100274 payload_len = RTP_LEN_GSM_FULL;
275 duration = RTP_GSM_DURATION;
Harald Welteda7ab742009-12-19 22:23:05 +0100276 break;
Harald Welteaca8f152009-12-19 23:06:41 +0100277 case GSM_TCHF_FRAME_EFR:
278 payload_type = RTP_PT_GSM_EFR;
Andreas Eversberg88012b62014-01-22 10:05:51 +0100279 payload_len = RTP_LEN_GSM_EFR;
280 duration = RTP_GSM_DURATION;
Harald Welteaca8f152009-12-19 23:06:41 +0100281 break;
Andreas Eversberg63bfdd82014-01-17 19:06:38 +0100282 case GSM_TCHH_FRAME:
283 payload_type = RTP_PT_GSM_HALF;
284 payload_len = RTP_LEN_GSM_HALF;
285 duration = RTP_GSM_DURATION;
286 break;
Andreas Eversbergd8967f72012-03-08 14:39:19 +0100287 case GSM_TCH_FRAME_AMR:
288 payload_type = RTP_PT_AMR;
289 payload_len = frame->data[0];
290 duration = RTP_GSM_DURATION;
291 break;
Harald Welteda7ab742009-12-19 22:23:05 +0100292 default:
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200293 DEBUGPC(DLMUX, "unsupported message type %d\n",
Harald Welteda7ab742009-12-19 22:23:05 +0100294 frame->msg_type);
295 return -EINVAL;
296 }
297
Harald Welte392736d2009-12-20 13:16:14 +0100298 {
299 struct timeval tv, tv_diff;
300 long int usec_diff, frame_diff;
301
302 gettimeofday(&tv, NULL);
303 tv_difference(&tv_diff, &rs->transmit.last_tv, &tv);
304 rs->transmit.last_tv = tv;
305
306 usec_diff = tv_diff.tv_sec * 1000000 + tv_diff.tv_usec;
307 frame_diff = (usec_diff / 20000);
308
309 if (abs(frame_diff) > 1) {
310 long int frame_diff_excess = frame_diff - 1;
311
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200312 LOGP(DLMUX, LOGL_NOTICE,
Harald Welte (local)9fcf6d72009-12-27 17:01:40 +0100313 "Correcting frame difference of %ld frames\n", frame_diff_excess);
Harald Welte392736d2009-12-20 13:16:14 +0100314 rs->transmit.sequence += frame_diff_excess;
315 rs->transmit.timestamp += frame_diff_excess * duration;
316 }
317 }
318
Andreas Eversbergd8967f72012-03-08 14:39:19 +0100319 if (payload_len > 33) {
320 DEBUGPC(DLMUX, "RTP payload too large (%d octets)\n",
321 payload_len);
322 return -EINVAL;
323 }
324
325 if (frame->msg_type == GSM_TCH_FRAME_AMR)
326 msg = msgb_alloc(sizeof(struct rtp_hdr) + payload_len,
327 "RTP-GSM (AMR)");
328 else
329 msg = msgb_alloc(sizeof(struct rtp_hdr) + payload_len,
330 "RTP-GSM");
Harald Welteda7ab742009-12-19 22:23:05 +0100331 if (!msg)
332 return -ENOMEM;
333 rtph = (struct rtp_hdr *)msg->data;
334 rtph->version = RTP_VERSION;
335 rtph->padding = 0;
336 rtph->extension = 0;
337 rtph->csrc_count = 0;
338 rtph->marker = 0;
339 rtph->payload_type = payload_type;
340 rtph->sequence = htons(rs->transmit.sequence++);
341 rtph->timestamp = htonl(rs->transmit.timestamp);
342 rs->transmit.timestamp += duration;
343 rtph->ssrc = htonl(rs->transmit.ssrc);
Andreas Eversbergd8967f72012-03-08 14:39:19 +0100344 if (frame->msg_type == GSM_TCH_FRAME_AMR)
345 memcpy(msg->data + sizeof(struct rtp_hdr), frame->data + 1,
346 payload_len);
347 else
348 memcpy(msg->data + sizeof(struct rtp_hdr), frame->data,
349 payload_len);
Harald Welteda7ab742009-12-19 22:23:05 +0100350 msgb_put(msg, sizeof(struct rtp_hdr) + payload_len);
351 msgb_enqueue(&rss->tx_queue, msg);
352 rss->bfd.when |= BSC_FD_WRITE;
353
354 return 0;
355}
356
Holger Hans Peter Freyther89acf062009-07-29 06:46:26 +0200357/* iterate over all chunks in one RTCP message, look for CNAME IEs and
Harald Welte805f6442009-07-28 18:25:29 +0200358 * replace all of those with 'new_cname' */
359static int rtcp_sdes_cname_mangle(struct msgb *msg, struct rtcp_hdr *rh,
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +0200360 uint16_t *rtcp_len, const char *new_cname)
Harald Welte805f6442009-07-28 18:25:29 +0200361{
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +0200362 uint8_t *rtcp_end;
363 uint8_t *cur = (uint8_t *) rh;
364 uint8_t tag, len = 0;
Harald Welte805f6442009-07-28 18:25:29 +0200365
366 rtcp_end = cur + *rtcp_len;
367 /* move cur to end of RTP header */
Harald Welte198f3f52009-07-29 10:46:41 +0200368 cur += sizeof(*rh);
Harald Welte805f6442009-07-28 18:25:29 +0200369
370 /* iterate over Chunks */
371 while (cur+4 < rtcp_end) {
372 /* skip four bytes SSRC/CSRC */
373 cur += 4;
374
375 /* iterate over IE's inside the chunk */
376 while (cur+1 < rtcp_end) {
377 tag = *cur++;
378 if (tag == 0) {
379 /* end of chunk, skip additional zero */
380 while (*cur++ == 0) { }
381 break;
382 }
383 len = *cur++;
384
385 if (tag == RTCP_IE_CNAME) {
386 /* we've found the CNAME, lets mangle it */
387 if (len < strlen(new_cname)) {
388 /* we need to make more space */
389 int increase = strlen(new_cname) - len;
390
391 msgb_push(msg, increase);
392 memmove(cur+len+increase, cur+len,
393 rtcp_end - (cur+len));
394 /* FIXME: we have to respect RTCP
395 * padding/alignment rules! */
396 len += increase;
397 *(cur-1) += increase;
398 rtcp_end += increase;
399 *rtcp_len += increase;
400 }
401 /* copy new CNAME into message */
402 memcpy(cur, new_cname, strlen(new_cname));
403 /* FIXME: zero the padding in case new CNAME
404 * is smaller than old one !!! */
405 }
406 cur += len;
407 }
408 }
409
410 return 0;
411}
412
413static int rtcp_mangle(struct msgb *msg, struct rtp_socket *rs)
414{
415 struct rtp_sub_socket *rss = &rs->rtcp;
416 struct rtcp_hdr *rtph;
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +0200417 uint16_t old_len;
Harald Welte805f6442009-07-28 18:25:29 +0200418 int rc;
419
420 if (!mangle_rtcp_cname)
421 return 0;
422
Harald Welteda7ab742009-12-19 22:23:05 +0100423 printf("RTCP\n");
Harald Welte805f6442009-07-28 18:25:29 +0200424 /* iterate over list of RTCP messages */
425 rtph = (struct rtcp_hdr *)msg->data;
Harald Welteda7ab742009-12-19 22:23:05 +0100426 while ((void *)rtph + sizeof(*rtph) <= (void *)msg->data + msg->len) {
Harald Welte805f6442009-07-28 18:25:29 +0200427 old_len = (ntohs(rtph->length) + 1) * 4;
Harald Welteda7ab742009-12-19 22:23:05 +0100428 if ((void *)rtph + old_len > (void *)msg->data + msg->len) {
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200429 DEBUGPC(DLMUX, "received RTCP packet too short for "
Harald Welteda7ab742009-12-19 22:23:05 +0100430 "length element\n");
431 return -EINVAL;
432 }
Harald Welte805f6442009-07-28 18:25:29 +0200433 if (rtph->type == RTCP_TYPE_SDES) {
434 char new_cname[255];
435 strncpy(new_cname, inet_ntoa(rss->sin_local.sin_addr),
436 sizeof(new_cname));
437 new_cname[sizeof(new_cname)-1] = '\0';
438 rc = rtcp_sdes_cname_mangle(msg, rtph, &old_len,
439 new_cname);
440 if (rc < 0)
441 return rc;
442 }
443 rtph = (void *)rtph + old_len;
444 }
445
446 return 0;
447}
448
Harald Welteead7a7b2009-07-28 00:01:58 +0200449/* read from incoming RTP/RTCP socket */
450static int rtp_socket_read(struct rtp_socket *rs, struct rtp_sub_socket *rss)
451{
452 int rc;
453 struct msgb *msg = msgb_alloc(RTP_ALLOC_SIZE, "RTP/RTCP");
Harald Welteda7ab742009-12-19 22:23:05 +0100454 struct msgb *new_msg;
Harald Welteead7a7b2009-07-28 00:01:58 +0200455 struct rtp_sub_socket *other_rss;
456
457 if (!msg)
458 return -ENOMEM;
459
460 rc = read(rss->bfd.fd, msg->data, RTP_ALLOC_SIZE);
461 if (rc <= 0) {
462 rss->bfd.when &= ~BSC_FD_READ;
463 return rc;
464 }
465
466 msgb_put(msg, rc);
467
468 switch (rs->rx_action) {
469 case RTP_PROXY:
Harald Welte805f6442009-07-28 18:25:29 +0200470 if (!rs->proxy.other_sock) {
471 rc = -EIO;
472 goto out_free;
473 }
Harald Welteead7a7b2009-07-28 00:01:58 +0200474 if (rss->bfd.priv_nr == RTP_PRIV_RTP)
475 other_rss = &rs->proxy.other_sock->rtp;
Harald Welte805f6442009-07-28 18:25:29 +0200476 else if (rss->bfd.priv_nr == RTP_PRIV_RTCP) {
Harald Welteead7a7b2009-07-28 00:01:58 +0200477 other_rss = &rs->proxy.other_sock->rtcp;
Harald Welte805f6442009-07-28 18:25:29 +0200478 /* modify RTCP SDES CNAME */
479 rc = rtcp_mangle(msg, rs);
480 if (rc < 0)
481 goto out_free;
482 } else {
483 rc = -EINVAL;
484 goto out_free;
Harald Welteead7a7b2009-07-28 00:01:58 +0200485 }
486 msgb_enqueue(&other_rss->tx_queue, msg);
487 other_rss->bfd.when |= BSC_FD_WRITE;
488 break;
Holger Hans Peter Freyther1ddb3562009-08-10 07:57:13 +0200489
490 case RTP_RECV_UPSTREAM:
Harald Welteda7ab742009-12-19 22:23:05 +0100491 if (!rs->receive.callref || !rs->receive.net) {
492 rc = -EIO;
493 goto out_free;
494 }
495 if (rss->bfd.priv_nr == RTP_PRIV_RTCP) {
496 if (!mangle_rtcp_cname) {
497 msgb_free(msg);
498 break;
499 }
500 /* modify RTCP SDES CNAME */
501 rc = rtcp_mangle(msg, rs);
502 if (rc < 0)
503 goto out_free;
504 msgb_enqueue(&rss->tx_queue, msg);
505 rss->bfd.when |= BSC_FD_WRITE;
506 break;
507 }
508 if (rss->bfd.priv_nr != RTP_PRIV_RTP) {
509 rc = -EINVAL;
510 goto out_free;
511 }
512 rc = rtp_decode(msg, rs->receive.callref, &new_msg);
513 if (rc < 0)
514 goto out_free;
515 msgb_free(msg);
Harald Welte31c00f72011-03-03 23:29:05 +0100516 trau_tx_to_mncc(rs->receive.net, new_msg);
Harald Welteda7ab742009-12-19 22:23:05 +0100517 break;
518
519 case RTP_NONE: /* if socket exists, but disabled by app */
520 msgb_free(msg);
Holger Hans Peter Freyther1ddb3562009-08-10 07:57:13 +0200521 break;
Harald Welteead7a7b2009-07-28 00:01:58 +0200522 }
523
Harald Welteda7ab742009-12-19 22:23:05 +0100524 return 0;
Harald Welte805f6442009-07-28 18:25:29 +0200525
526out_free:
527 msgb_free(msg);
528 return rc;
Harald Welteead7a7b2009-07-28 00:01:58 +0200529}
530
Harald Welte647db842013-02-03 12:06:58 +0100531/* \brief write from tx_queue to RTP/RTCP socket */
Harald Welteead7a7b2009-07-28 00:01:58 +0200532static int rtp_socket_write(struct rtp_socket *rs, struct rtp_sub_socket *rss)
533{
534 struct msgb *msg;
535 int written;
536
537 msg = msgb_dequeue(&rss->tx_queue);
538 if (!msg) {
539 rss->bfd.when &= ~BSC_FD_WRITE;
540 return 0;
541 }
542
543 written = write(rss->bfd.fd, msg->data, msg->len);
544 if (written < msg->len) {
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200545 LOGP(DLMIB, LOGL_ERROR, "short write");
Harald Welteead7a7b2009-07-28 00:01:58 +0200546 msgb_free(msg);
547 return -EIO;
548 }
549
550 msgb_free(msg);
551
552 return 0;
553}
554
555
Harald Welte647db842013-02-03 12:06:58 +0100556/*! \brief callback for the select.c:bfd_* layer */
Pablo Neira Ayuso4db92992011-05-06 12:11:23 +0200557static int rtp_bfd_cb(struct osmo_fd *bfd, unsigned int flags)
Harald Welteead7a7b2009-07-28 00:01:58 +0200558{
559 struct rtp_socket *rs = bfd->data;
560 struct rtp_sub_socket *rss;
561
562 switch (bfd->priv_nr) {
563 case RTP_PRIV_RTP:
564 rss = &rs->rtp;
565 break;
566 case RTP_PRIV_RTCP:
567 rss = &rs->rtcp;
568 break;
569 default:
570 return -EINVAL;
571 }
572
573 if (flags & BSC_FD_READ)
574 rtp_socket_read(rs, rss);
575
576 if (flags & BSC_FD_WRITE)
577 rtp_socket_write(rs, rss);
578
579 return 0;
580}
581
Harald Welte647db842013-02-03 12:06:58 +0100582/*! \brief initialize one rtp sub-socket */
Holger Hans Peter Freytheracf8a0c2010-03-29 08:47:44 +0200583static void init_rss(struct rtp_sub_socket *rss,
Harald Welteead7a7b2009-07-28 00:01:58 +0200584 struct rtp_socket *rs, int fd, int priv_nr)
585{
586 /* initialize bfd */
587 rss->bfd.fd = fd;
588 rss->bfd.data = rs;
589 rss->bfd.priv_nr = priv_nr;
590 rss->bfd.cb = rtp_bfd_cb;
591}
592
Harald Welte647db842013-02-03 12:06:58 +0100593/*! \brief create a new RTP/RTCP socket and bind it */
Harald Welteead7a7b2009-07-28 00:01:58 +0200594struct rtp_socket *rtp_socket_create(void)
595{
596 int rc;
597 struct rtp_socket *rs;
598
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200599 DEBUGP(DLMUX, "rtp_socket_create(): ");
Harald Welte805f6442009-07-28 18:25:29 +0200600
Harald Welteead7a7b2009-07-28 00:01:58 +0200601 rs = talloc_zero(tall_bsc_ctx, struct rtp_socket);
602 if (!rs)
603 return NULL;
604
605 INIT_LLIST_HEAD(&rs->rtp.tx_queue);
606 INIT_LLIST_HEAD(&rs->rtcp.tx_queue);
607
608 rc = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
609 if (rc < 0)
610 goto out_free;
611
612 init_rss(&rs->rtp, rs, rc, RTP_PRIV_RTP);
Pablo Neira Ayuso4db92992011-05-06 12:11:23 +0200613 rc = osmo_fd_register(&rs->rtp.bfd);
Harald Welteead7a7b2009-07-28 00:01:58 +0200614 if (rc < 0)
615 goto out_rtp_socket;
616
617 rc = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
618 if (rc < 0)
619 goto out_rtp_bfd;
620
621 init_rss(&rs->rtcp, rs, rc, RTP_PRIV_RTCP);
Pablo Neira Ayuso4db92992011-05-06 12:11:23 +0200622 rc = osmo_fd_register(&rs->rtcp.bfd);
Harald Welteead7a7b2009-07-28 00:01:58 +0200623 if (rc < 0)
624 goto out_rtcp_socket;
625
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200626 DEBUGPC(DLMUX, "success\n");
Harald Welte805f6442009-07-28 18:25:29 +0200627
628 rc = rtp_socket_bind(rs, INADDR_ANY);
629 if (rc < 0)
630 goto out_rtcp_bfd;
631
Harald Welteead7a7b2009-07-28 00:01:58 +0200632 return rs;
633
Harald Welte805f6442009-07-28 18:25:29 +0200634out_rtcp_bfd:
Pablo Neira Ayuso4db92992011-05-06 12:11:23 +0200635 osmo_fd_unregister(&rs->rtcp.bfd);
Harald Welteead7a7b2009-07-28 00:01:58 +0200636out_rtcp_socket:
637 close(rs->rtcp.bfd.fd);
638out_rtp_bfd:
Pablo Neira Ayuso4db92992011-05-06 12:11:23 +0200639 osmo_fd_unregister(&rs->rtp.bfd);
Harald Welteead7a7b2009-07-28 00:01:58 +0200640out_rtp_socket:
641 close(rs->rtp.bfd.fd);
642out_free:
643 talloc_free(rs);
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200644 DEBUGPC(DLMUX, "failed\n");
Harald Welteead7a7b2009-07-28 00:01:58 +0200645 return NULL;
646}
647
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +0200648static int rtp_sub_socket_bind(struct rtp_sub_socket *rss, uint32_t ip,
649 uint16_t port)
Harald Welteead7a7b2009-07-28 00:01:58 +0200650{
Harald Welte805f6442009-07-28 18:25:29 +0200651 int rc;
652 socklen_t alen = sizeof(rss->sin_local);
653
Harald Welteead7a7b2009-07-28 00:01:58 +0200654 rss->sin_local.sin_family = AF_INET;
655 rss->sin_local.sin_addr.s_addr = htonl(ip);
656 rss->sin_local.sin_port = htons(port);
657 rss->bfd.when |= BSC_FD_READ;
658
Harald Welte805f6442009-07-28 18:25:29 +0200659 rc = bind(rss->bfd.fd, (struct sockaddr *)&rss->sin_local,
660 sizeof(rss->sin_local));
661 if (rc < 0)
662 return rc;
663
664 /* retrieve the address we actually bound to, in case we
665 * passed INADDR_ANY as IP address */
666 return getsockname(rss->bfd.fd, (struct sockaddr *)&rss->sin_local,
667 &alen);
Harald Welteead7a7b2009-07-28 00:01:58 +0200668}
669
670#define RTP_PORT_BASE 30000
Harald Welte805f6442009-07-28 18:25:29 +0200671static unsigned int next_udp_port = RTP_PORT_BASE;
Harald Welteead7a7b2009-07-28 00:01:58 +0200672
Harald Welte647db842013-02-03 12:06:58 +0100673/*! \brief bind a RTP socket to a specific local address
674 * \param[in] rs RTP socket to be bound
675 * \param[in] ip local IP address to which socket is to be bound
676 */
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +0200677int rtp_socket_bind(struct rtp_socket *rs, uint32_t ip)
Harald Welteead7a7b2009-07-28 00:01:58 +0200678{
Harald Welte805f6442009-07-28 18:25:29 +0200679 int rc = -EIO;
680 struct in_addr ia;
681
682 ia.s_addr = htonl(ip);
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200683 DEBUGP(DLMUX, "rtp_socket_bind(rs=%p, IP=%s): ", rs,
Harald Welte805f6442009-07-28 18:25:29 +0200684 inet_ntoa(ia));
Harald Welteead7a7b2009-07-28 00:01:58 +0200685
686 /* try to bind to a consecutive pair of ports */
Harald Welte805f6442009-07-28 18:25:29 +0200687 for (next_udp_port = next_udp_port % 0xffff;
688 next_udp_port < 0xffff; next_udp_port += 2) {
Harald Welteead7a7b2009-07-28 00:01:58 +0200689 rc = rtp_sub_socket_bind(&rs->rtp, ip, next_udp_port);
690 if (rc != 0)
691 continue;
692
693 rc = rtp_sub_socket_bind(&rs->rtcp, ip, next_udp_port+1);
694 if (rc == 0)
695 break;
696 }
Harald Welte805f6442009-07-28 18:25:29 +0200697 if (rc < 0) {
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200698 DEBUGPC(DLMUX, "failed\n");
Harald Welteead7a7b2009-07-28 00:01:58 +0200699 return rc;
Harald Welte805f6442009-07-28 18:25:29 +0200700 }
Harald Welteead7a7b2009-07-28 00:01:58 +0200701
Harald Welte805f6442009-07-28 18:25:29 +0200702 ia.s_addr = rs->rtp.sin_local.sin_addr.s_addr;
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200703 DEBUGPC(DLMUX, "BOUND_IP=%s, BOUND_PORT=%u\n",
Harald Welte805f6442009-07-28 18:25:29 +0200704 inet_ntoa(ia), ntohs(rs->rtp.sin_local.sin_port));
Harald Welteead7a7b2009-07-28 00:01:58 +0200705 return ntohs(rs->rtp.sin_local.sin_port);
706}
707
708static int rtp_sub_socket_connect(struct rtp_sub_socket *rss,
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +0200709 uint32_t ip, uint16_t port)
Harald Welteead7a7b2009-07-28 00:01:58 +0200710{
Harald Welte805f6442009-07-28 18:25:29 +0200711 int rc;
712 socklen_t alen = sizeof(rss->sin_local);
713
Harald Welteead7a7b2009-07-28 00:01:58 +0200714 rss->sin_remote.sin_family = AF_INET;
715 rss->sin_remote.sin_addr.s_addr = htonl(ip);
716 rss->sin_remote.sin_port = htons(port);
717
Harald Welte805f6442009-07-28 18:25:29 +0200718 rc = connect(rss->bfd.fd, (struct sockaddr *) &rss->sin_remote,
719 sizeof(rss->sin_remote));
720 if (rc < 0)
721 return rc;
722
723 return getsockname(rss->bfd.fd, (struct sockaddr *)&rss->sin_local,
724 &alen);
Harald Welteead7a7b2009-07-28 00:01:58 +0200725}
726
Harald Welte647db842013-02-03 12:06:58 +0100727/*! \brief 'connect' a RTP socket to a remote peer
728 * \param[in] rs RTP socket to be connected
729 * \param[in] ip remote IP address to which to connect
730 * \param[in] port remote UDP port number to which to connect
731 */
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +0200732int rtp_socket_connect(struct rtp_socket *rs, uint32_t ip, uint16_t port)
Harald Welteead7a7b2009-07-28 00:01:58 +0200733{
734 int rc;
Harald Welte805f6442009-07-28 18:25:29 +0200735 struct in_addr ia;
736
737 ia.s_addr = htonl(ip);
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200738 DEBUGP(DLMUX, "rtp_socket_connect(rs=%p, ip=%s, port=%u)\n",
Harald Welte805f6442009-07-28 18:25:29 +0200739 rs, inet_ntoa(ia), port);
Harald Welteead7a7b2009-07-28 00:01:58 +0200740
741 rc = rtp_sub_socket_connect(&rs->rtp, ip, port);
742 if (rc < 0)
743 return rc;
744
745 return rtp_sub_socket_connect(&rs->rtcp, ip, port+1);
746}
747
Harald Welte647db842013-02-03 12:06:58 +0100748/*! \brief bind two RTP/RTCP sockets together in the proxy
749 * \param[in] this First RTP socket
750 * \param[in] other Second RTP socket
751 */
Harald Welteead7a7b2009-07-28 00:01:58 +0200752int rtp_socket_proxy(struct rtp_socket *this, struct rtp_socket *other)
753{
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200754 DEBUGP(DLMUX, "rtp_socket_proxy(this=%p, other=%p)\n",
Harald Welte805f6442009-07-28 18:25:29 +0200755 this, other);
756
Harald Welteead7a7b2009-07-28 00:01:58 +0200757 this->rx_action = RTP_PROXY;
758 this->proxy.other_sock = other;
759
760 other->rx_action = RTP_PROXY;
761 other->proxy.other_sock = this;
762
763 return 0;
764}
765
Harald Welte647db842013-02-03 12:06:58 +0100766/*! \brief bind RTP/RTCP socket to application, disabling proxy
767 * \param[in] this RTP socket
768 * \param[in] net gsm_network argument to trau_tx_to_mncc()
769 * \param[in] callref callref argument to trau_tx_to_mncc()
770 */
Harald Welte9fb1f102009-12-20 17:07:23 +0100771int rtp_socket_upstream(struct rtp_socket *this, struct gsm_network *net,
Holger Hans Peter Freytherc42ad8b2011-04-18 17:04:00 +0200772 uint32_t callref)
Harald Welteda7ab742009-12-19 22:23:05 +0100773{
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200774 DEBUGP(DLMUX, "rtp_socket_proxy(this=%p, callref=%u)\n",
Harald Welteda7ab742009-12-19 22:23:05 +0100775 this, callref);
776
777 if (callref) {
778 this->rx_action = RTP_RECV_UPSTREAM;
779 this->receive.net = net;
780 this->receive.callref = callref;
781 } else
782 this->rx_action = RTP_NONE;
783
784 return 0;
785}
786
Harald Welteead7a7b2009-07-28 00:01:58 +0200787static void free_tx_queue(struct rtp_sub_socket *rss)
788{
789 struct msgb *msg;
790
791 while ((msg = msgb_dequeue(&rss->tx_queue)))
792 msgb_free(msg);
793}
794
Harald Welte647db842013-02-03 12:06:58 +0100795/*! \brief Free/release a previously allocated RTP socket
796 * \param[in[] rs RTP/RTCP socket to be released
797 */
Harald Welteead7a7b2009-07-28 00:01:58 +0200798int rtp_socket_free(struct rtp_socket *rs)
799{
Pablo Neira Ayusoed5cacb2011-08-17 22:44:07 +0200800 DEBUGP(DLMUX, "rtp_socket_free(rs=%p)\n", rs);
Harald Welteead7a7b2009-07-28 00:01:58 +0200801
802 /* make sure we don't leave references dangling to us */
803 if (rs->rx_action == RTP_PROXY &&
804 rs->proxy.other_sock)
805 rs->proxy.other_sock->proxy.other_sock = NULL;
806
Pablo Neira Ayuso4db92992011-05-06 12:11:23 +0200807 osmo_fd_unregister(&rs->rtp.bfd);
Harald Welteead7a7b2009-07-28 00:01:58 +0200808 close(rs->rtp.bfd.fd);
809 free_tx_queue(&rs->rtp);
810
Pablo Neira Ayuso4db92992011-05-06 12:11:23 +0200811 osmo_fd_unregister(&rs->rtcp.bfd);
Harald Welteead7a7b2009-07-28 00:01:58 +0200812 close(rs->rtcp.bfd.fd);
813 free_tx_queue(&rs->rtcp);
814
815 talloc_free(rs);
816
817 return 0;
818}