source: roaraudio/libroar/vio_rtp.c @ 5373:8da157c10483

Last change on this file since 5373:8da157c10483 was 5357:f3c705877ceb, checked in by phi, 12 years ago

fixed some compiler warnings

File size: 11.2 KB
Line 
1//vio_rtp.c:
2
3/*
4 *      Copyright (C) Philipp 'ph3-der-loewe' Schafft - 2010-2011
5 *
6 *  This file is part of libroar a part of RoarAudio,
7 *  a cross-platform sound system for both, home and professional use.
8 *  See README for details.
9 *
10 *  This file is free software; you can redistribute it and/or modify
11 *  it under the terms of the GNU General Public License version 3
12 *  as published by the Free Software Foundation.
13 *
14 *  libroar is distributed in the hope that it will be useful,
15 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
16 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17 *  GNU General Public License for more details.
18 *
19 *  You should have received a copy of the GNU General Public License
20 *  along with this software; see the file COPYING.  If not, write to
21 *  the Free Software Foundation, 51 Franklin Street, Fifth Floor,
22 *  Boston, MA 02110-1301, USA.
23 *
24 *  NOTE for everyone want's to change something and send patches:
25 *  read README and HACKING! There a addition information on
26 *  the license of this document you need to read before you send
27 *  any patches.
28 *
29 *  NOTE for uses of non-GPL (LGPL,...) software using libesd, libartsc
30 *  or libpulse*:
31 *  The libs libroaresd, libroararts and libroarpulse link this lib
32 *  and are therefore GPL. Because of this it may be illigal to use
33 *  them with any software that uses libesd, libartsc or libpulse*.
34 */
35
36#include "libroar.h"
37
38static const struct {
39 int pt;
40 struct roar_audio_info info;
41} _g_rtp_pt[] = {
42 {ROAR_RTP_PT_A_PCMU,      {.codec = ROAR_CODEC_MULAW,    .bits =  8, .rate =  8000, .channels = 1}},
43 {ROAR_RTP_PT_A_PCMA,      {.codec = ROAR_CODEC_ALAW,     .bits =  8, .rate =  8000, .channels = 1}},
44 {ROAR_RTP_PT_A_L16_441_2, {.codec = ROAR_CODEC_PCM_S_BE, .bits = 16, .rate = 44100, .channels = 2}},
45 {ROAR_RTP_PT_A_L16_441_1, {.codec = ROAR_CODEC_PCM_S_BE, .bits = 16, .rate = 44100, .channels = 1}},
46 {-1, {-1, -1, -1, -1}}
47};
48
49static int _info2pt (struct roar_audio_info * info) {
50 int i;
51
52 ROAR_DBG("_info2pt(info=%p{.codec=%s(%i), .bits=%i, .rate=%i, .channels=%i}) = ?", info, roar_codec2str(info->codec), info->codec, info->bits, info->rate, info->channels);
53
54 for (i = 0; _g_rtp_pt[i].pt != -1; i++) {
55  if ( info->codec    == _g_rtp_pt[i].info.codec    &&
56       info->bits     == _g_rtp_pt[i].info.bits     &&
57       info->rate     == _g_rtp_pt[i].info.rate     &&
58       info->channels == _g_rtp_pt[i].info.channels ) {
59   return _g_rtp_pt[i].pt;
60  }
61 }
62
63 ROAR_DBG("_info2pt(info=%p{.codec=%s(%i), .bits=%i, .rate=%i, .channels=%i}) = -1", info, roar_codec2str(info->codec), info->codec, info->bits, info->rate, info->channels);
64 return -1;
65}
66
67#if 0
68static const struct roar_audio_info * _pt2info (int pt) {
69 int i;
70
71 for (i = 0; _g_rtp_pt[i].pt != -1; i++) {
72  if ( _g_rtp_pt[i].pt == pt ) {
73   return &(_g_rtp_pt[i].info);
74  }
75 }
76
77 return NULL;
78}
79#endif
80
81int roar_vio_open_rtp        (struct roar_vio_calls * calls, struct roar_vio_calls * dst,
82                              char * dstr, struct roar_vio_defaults * odef) {
83 struct roar_rtp_inst * self = NULL;
84
85 ROAR_DBG("roar_vio_open_rtp(calls=%p, dst=%p, dstr='%s', odef=%p) = ?", calls, dst, dstr, odef);
86
87 if ( calls == NULL || dst == NULL )
88  return -1;
89
90 if ( (self = roar_mm_malloc(sizeof(struct roar_rtp_inst))) == NULL )
91  return -1;
92
93 ROAR_DBG("roar_vio_open_rtp(calls=%p, dst=%p, dstr='%s', odef=%p) = ?", calls, dst, dstr, odef);
94
95 memset(self, 0, sizeof(struct roar_rtp_inst));
96
97 self->vio                 = dst;
98 self->bpf                 = 0;
99 self->mtu                 = 768;
100
101 memset(&(self->info), 0, sizeof(struct roar_audio_info));
102
103 self->header.version      = 2;
104 self->header.payload_type = ROAR_RTP_PT_UNKNOWN;
105
106 // TODO: init with random values:
107 //       Sequence Number
108 //       ts
109 //       SSRC
110
111 memset(calls, 0, sizeof(struct roar_vio_calls));
112
113 calls->inst               = self;
114 calls->read               = roar_vio_rtp_read;
115 calls->write              = roar_vio_rtp_write;
116 calls->sync               = roar_vio_rtp_sync;
117 calls->ctl                = roar_vio_rtp_ctl;
118 calls->close              = roar_vio_rtp_close;
119
120 ROAR_DBG("roar_vio_open_rtp(calls=%p, dst=%p, dstr='%s', odef=%p) = 0", calls, dst, dstr, odef);
121
122 return 0;
123}
124
125ssize_t roar_vio_rtp_read    (struct roar_vio_calls * vio, void *buf, size_t count) {
126 struct roar_rtp_inst * self = vio->inst;
127 size_t len_need = self->mtu * 4 + sizeof(struct roar_rtp_header); // we hope to never get pkgs with size > 4*mtu
128 size_t len_have;
129 ssize_t have = 0;
130 ssize_t ret;
131 union {
132  void     * vp;
133  char     * cp;
134  uint16_t * u16;
135  uint32_t * u32;
136 } data;
137 size_t  dataoffset;
138 int     i;
139
140 ROAR_DBG("roar_vio_rtp_read(vio=%p, buf=%p, count=%llu) = ?", vio, buf, (long long unsigned)count);
141
142 if ( self->rx_decoded != NULL ) {
143  // handle this case and set to NULL if the buffer is empty
144  // set have;
145  // increment buf, decrement count
146 }
147
148 if ( count == 0 )
149  return have;
150
151 if ( self->io == NULL ) {
152  if ( roar_buffer_new(&(self->io), len_need) == -1 )
153   return have ? have : -1;
154
155  len_have = len_need;
156 } else {
157  if ( roar_buffer_get_len(self->io, &len_have) == -1 )
158   return have ? have : -1;
159
160  if ( len_have < len_need ) {
161   if ( roar_buffer_set_len(self->io, len_need) == -1 ) {
162    if ( roar_buffer_free(self->io) == -1 )
163     return have ? have : -1;
164
165    self->io = NULL;
166    if ( have != 0 ) {
167     return have;
168    } else {
169     return roar_vio_rtp_read(vio, buf, count); // restart our self from the beginning with no buffer
170    }
171   }
172  }
173 }
174
175 if ( roar_buffer_get_data(self->io, &(data.vp)) == -1 )
176  return have ? have : -1;
177
178 if ( (ret = roar_vio_read(self->vio, data.vp, len_need)) == -1 )
179  return have ? have : -1;
180
181 if ( (data.cp[0] & 0x02) == 0x02 ) /* version check */
182  return have ? have : -1;
183
184 self->header.csrc_count   = (data.cp[0] & 0xF0) >> 4;
185 self->header.payload_type = (data.cp[1] & 0xFE) >> 1;
186
187 // TODO: check old seqnum < new seqnum
188 self->header.seq_num      = ROAR_NET2HOST16(data.u16[1]);
189
190 // TODO: check timestamp:
191 self->header.ts           = ROAR_NET2HOST32(data.u32[1]);
192
193 self->header.ssrc         = ROAR_NET2HOST32(data.u32[2]);
194
195 for (i = 0; i < self->header.csrc_count; i++) {
196  self->header.csrc[i]     = ROAR_NET2HOST16(data.u32[3+i]);
197 }
198
199 dataoffset   = 3*4 + self->header.csrc_count*4;
200
201 ret     -= dataoffset;
202 data.vp += dataoffset;
203
204 if ( ret <= (ssize_t)count ) {
205  memcpy(buf, data.vp, ret);
206  ROAR_DBG("roar_vio_rtp_read(vio=%p, buf=%p, count=?) = %llu", vio, buf, (long long unsigned)(have+ret));
207  return have + ret;
208 } else {
209 }
210
211 ROAR_DBG("roar_vio_rtp_read(vio=%p, buf=%p, count=?) = -1", vio, buf);
212 return -1;
213}
214
215ssize_t roar_vio_rtp_write   (struct roar_vio_calls * vio, void *buf, size_t count) {
216 struct roar_rtp_inst * self = vio->inst;
217 size_t len_need = count + sizeof(struct roar_rtp_header); // this is a bit more than we need
218                                                           // we ignore this at the moment
219 size_t len_have;
220 union {
221  void     * vp;
222  char     * cp;
223  uint16_t * u16;
224  uint32_t * u32;
225 } data;
226 size_t  dataoffset;
227 ssize_t ret;
228 int     i;
229
230 ROAR_DBG("roar_vio_rtp_write(vio=%p, buf=%p, count=%llu) = ?", vio, buf, (long long unsigned)count);
231
232 if ( self->mtu < (sizeof(struct roar_rtp_header) + self->bpf) )
233  return -1;
234
235 if ( len_need > self->mtu ) {
236  len_have = 0;
237  ret      = 0;
238
239  while (count) {
240   len_need = self->mtu - sizeof(struct roar_rtp_header);
241
242   if ( count < len_need )
243    len_need = count;
244
245   if ( (ret = roar_vio_rtp_write(vio, buf, len_need)) == -1 )
246    break;
247
248   len_have += ret;
249   buf      += ret;
250   count    -= ret;
251  }
252
253  return len_have ? (ssize_t)len_have : ret;
254 }
255
256 if ( self->io == NULL ) {
257  if ( roar_buffer_new(&(self->io), len_need) == -1 )
258   return -1;
259
260  len_have = len_need;
261 } else {
262  if ( roar_buffer_get_len(self->io, &len_have) == -1 )
263   return -1;
264
265  if ( len_have < len_need ) {
266   if ( roar_buffer_set_len(self->io, len_need) == -1 ) {
267    if ( roar_buffer_free(self->io) == -1 )
268     return -1;
269
270    self->io = NULL;
271    return roar_vio_rtp_write(vio, buf, count); // restart ower self from the beginning with no buffer
272   }
273  }
274 }
275
276 ROAR_DBG("roar_vio_rtp_write(vio=%p, buf=%p, count=%llu) = ?", vio, buf, (long long unsigned)count);
277
278 if ( roar_buffer_get_data(self->io, &(data.vp)) == -1 )
279  return -1;
280
281 memset(data.vp, 0, len_need);
282
283 self->header.seq_num++;
284
285 data.cp[0]   = 2;
286 data.cp[0]  |= self->header.csrc_count   << 4;
287 data.cp[1]  |= self->header.payload_type << 1;
288
289 data.u16[1]  = ROAR_HOST2NET16(self->header.seq_num);
290
291 data.u32[1]  = ROAR_HOST2NET32(self->header.ts);
292 data.u32[2]  = ROAR_HOST2NET32(self->header.ssrc);
293
294 for (i = 0; i < self->header.csrc_count; i++) {
295  data.u32[3+i] = ROAR_HOST2NET32(self->header.csrc[i]);
296 }
297
298 dataoffset   = 3*4 + self->header.csrc_count*4;
299
300 memcpy(data.vp + dataoffset, buf, count);
301
302 ROAR_DBG("roar_vio_rtp_write(vio=%p, buf=%p, count=%llu) = ?", vio, buf, (long long unsigned)count);
303
304 if ( (ret = roar_vio_write(self->vio, data.vp, count+dataoffset)) == -1 )
305  return -1;
306
307 len_have = ret - dataoffset;
308
309 self->header.ts += len_have / self->bpf;
310
311 return len_have;
312}
313
314int     roar_vio_rtp_sync    (struct roar_vio_calls * vio) {
315 struct roar_rtp_inst * self = vio->inst;
316
317 ROAR_DBG("roar_vio_rtp_sync(vio=%p) = ?", vio);
318
319 return roar_vio_sync(self->vio);
320}
321
322int     roar_vio_rtp_ctl     (struct roar_vio_calls * vio, roar_vio_ctl_t cmd, void * data) {
323 struct roar_rtp_inst * self = vio->inst;
324 struct roar_stream          * s    = NULL;
325 struct roar_stream_server   * ss   = NULL;
326 struct roar_audio_info      * info = NULL;
327
328 ROAR_DBG("roar_vio_rtp_ctl(vio=%p, cmd=%i, data=%p) = ?", vio, cmd, data);
329
330 if (vio == NULL || cmd == -1)
331  return -1;
332
333 ROAR_DBG("roar_vio_rtp_ctl(vio=%p, cmd=%i, data=%p) = ?", vio, cmd, data);
334
335 switch (cmd) {
336  case ROAR_VIO_CTL_GET_NAME:
337    if ( data == NULL )
338     return -1;
339
340    *(char**)data = "rtp";
341    return 0;
342   break;
343  case ROAR_VIO_CTL_SET_SSTREAM:
344    s = ROAR_STREAM(ss = data);
345    info = &(s->info);
346   break;
347  case ROAR_VIO_CTL_SET_STREAM:
348    s = ROAR_STREAM(data);
349    info = &(s->info);
350   break;
351  case ROAR_VIO_CTL_SET_AUINFO:
352    info = data;
353   break;
354  case ROAR_VIO_CTL_GET_NEXT:
355    *(struct roar_vio_calls **)data = self->vio;
356    return 0;
357   break;
358  case ROAR_VIO_CTL_SET_NEXT:
359    self->vio = *(struct roar_vio_calls **)data;
360    return 0;
361   break;
362 }
363
364 ROAR_DBG("roar_vio_rtp_ctl(vio=%p, cmd=%i, data=%p) = ?", vio, cmd, data);
365
366 if ( info != NULL ) {
367  switch (info->codec) {
368   case ROAR_CODEC_PCM_S_LE:
369   case ROAR_CODEC_PCM_S_PDP:
370     info->codec = ROAR_CODEC_PCM_S_BE;
371    break;
372   case ROAR_CODEC_PCM_U_LE:
373   case ROAR_CODEC_PCM_U_PDP:
374     info->codec = ROAR_CODEC_PCM_U_BE;
375    break;
376  }
377
378  memcpy(&(self->info), info, sizeof(struct roar_audio_info));
379
380  self->header.payload_type = _info2pt(info);
381
382  self->bpf                 = info->channels * info->bits / 8;
383
384  _LIBROAR_IGNORE_RET(roar_vio_ctl(self->vio, cmd, data));
385  return 0;
386 }
387
388 return roar_vio_ctl(self->vio, cmd, data);
389}
390
391int     roar_vio_rtp_close   (struct roar_vio_calls * vio) {
392 struct roar_rtp_inst * self = vio->inst;
393 int ret;
394
395 ROAR_DBG("roar_vio_rtp_close(vio=%p) = ?", vio);
396
397 ret = roar_vio_close(self->vio);
398
399 if ( self->io != NULL )
400  if ( roar_buffer_free(self->io) == -1 )
401   ret = -1;
402
403 roar_mm_free(self);
404
405 ROAR_DBG("roar_vio_rtp_close(vio=%p) = %i", vio, ret);
406
407 return ret;
408}
409
410//ll
Note: See TracBrowser for help on using the repository browser.