]>
code.delx.au - pulseaudio/blob - src/modules/rtp/rtp.c
2 This file is part of PulseAudio.
4 Copyright 2006 Lennart Poettering
6 PulseAudio is free software; you can redistribute it and/or modify
7 it under the terms of the GNU Lesser General Public License as published
8 by the Free Software Foundation; either version 2 of the License,
9 or (at your option) any later version.
11 PulseAudio is distributed in the hope that it will be useful, but
12 WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 General Public License for more details.
16 You should have received a copy of the GNU Lesser General Public License
17 along with PulseAudio; if not, write to the Free Software
18 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
30 #include <arpa/inet.h>
32 #include <sys/ioctl.h>
34 #ifdef HAVE_SYS_FILIO_H
35 #include <sys/filio.h>
42 #include <pulsecore/core-error.h>
43 #include <pulsecore/log.h>
44 #include <pulsecore/macro.h>
45 #include <pulsecore/core-util.h>
49 pa_rtp_context
* pa_rtp_context_init_send(pa_rtp_context
*c
, int fd
, uint32_t ssrc
, uint8_t payload
, size_t frame_size
) {
54 c
->sequence
= (uint16_t) (rand()*rand());
56 c
->ssrc
= ssrc
? ssrc
: (uint32_t) (rand()*rand());
57 c
->payload
= (uint8_t) (payload
& 127U);
58 c
->frame_size
= frame_size
;
60 pa_memchunk_reset(&c
->memchunk
);
67 int pa_rtp_send(pa_rtp_context
*c
, size_t size
, pa_memblockq
*q
) {
68 struct iovec iov
[MAX_IOVECS
];
69 pa_memblock
* mb
[MAX_IOVECS
];
77 if (pa_memblockq_get_length(q
) < size
)
84 pa_memchunk_reset(&chunk
);
86 if ((r
= pa_memblockq_peek(q
, &chunk
)) >= 0) {
88 size_t k
= n
+ chunk
.length
> size
? size
- n
: chunk
.length
;
90 pa_assert(chunk
.memblock
);
92 iov
[iov_idx
].iov_base
= ((uint8_t*) pa_memblock_acquire(chunk
.memblock
) + chunk
.index
);
93 iov
[iov_idx
].iov_len
= k
;
94 mb
[iov_idx
] = chunk
.memblock
;
98 pa_memblockq_drop(q
, k
);
101 pa_assert(n
% c
->frame_size
== 0);
103 if (r
< 0 || n
>= size
|| iov_idx
>= MAX_IOVECS
) {
110 header
[0] = htonl(((uint32_t) 2 << 30) | ((uint32_t) c
->payload
<< 16) | ((uint32_t) c
->sequence
));
111 header
[1] = htonl(c
->timestamp
);
112 header
[2] = htonl(c
->ssrc
);
114 iov
[0].iov_base
= (void*)header
;
115 iov
[0].iov_len
= sizeof(header
);
120 m
.msg_iovlen
= (size_t) iov_idx
;
121 m
.msg_control
= NULL
;
122 m
.msg_controllen
= 0;
125 k
= sendmsg(c
->fd
, &m
, MSG_DONTWAIT
);
127 for (i
= 1; i
< iov_idx
; i
++) {
128 pa_memblock_release(mb
[i
]);
129 pa_memblock_unref(mb
[i
]);
136 c
->timestamp
+= (unsigned) (n
/c
->frame_size
);
139 if (errno
!= EAGAIN
&& errno
!= EINTR
) /* If the queue is full, just ignore it */
140 pa_log("sendmsg() failed: %s", pa_cstrerror(errno
));
144 if (r
< 0 || pa_memblockq_get_length(q
) < size
)
155 pa_rtp_context
* pa_rtp_context_init_recv(pa_rtp_context
*c
, int fd
, size_t frame_size
) {
159 c
->frame_size
= frame_size
;
161 pa_memchunk_reset(&c
->memchunk
);
165 int pa_rtp_recv(pa_rtp_context
*c
, pa_memchunk
*chunk
, pa_mempool
*pool
) {
176 pa_memchunk_reset(chunk
);
178 if (ioctl(c
->fd
, FIONREAD
, &size
) < 0) {
179 pa_log_warn("FIONREAD failed: %s", pa_cstrerror(errno
));
186 if (c
->memchunk
.length
< (unsigned) size
) {
189 if (c
->memchunk
.memblock
)
190 pa_memblock_unref(c
->memchunk
.memblock
);
192 l
= PA_MAX((size_t) size
, pa_mempool_block_size_max(pool
));
194 c
->memchunk
.memblock
= pa_memblock_new(pool
, l
);
195 c
->memchunk
.index
= 0;
196 c
->memchunk
.length
= pa_memblock_get_length(c
->memchunk
.memblock
);
199 pa_assert(c
->memchunk
.length
>= (size_t) size
);
201 chunk
->memblock
= pa_memblock_ref(c
->memchunk
.memblock
);
202 chunk
->index
= c
->memchunk
.index
;
204 iov
.iov_base
= (uint8_t*) pa_memblock_acquire(chunk
->memblock
) + chunk
->index
;
205 iov
.iov_len
= (size_t) size
;
211 m
.msg_control
= NULL
;
212 m
.msg_controllen
= 0;
215 r
= recvmsg(c
->fd
, &m
, 0);
216 pa_memblock_release(chunk
->memblock
);
219 if (r
< 0 && errno
!= EAGAIN
&& errno
!= EINTR
)
220 pa_log_warn("recvmsg() failed: %s", r
< 0 ? pa_cstrerror(errno
) : "size mismatch");
226 pa_log_warn("RTP packet too short.");
230 memcpy(&header
, iov
.iov_base
, sizeof(uint32_t));
231 memcpy(&c
->timestamp
, (uint8_t*) iov
.iov_base
+ 4, sizeof(uint32_t));
232 memcpy(&c
->ssrc
, (uint8_t*) iov
.iov_base
+ 8, sizeof(uint32_t));
234 header
= ntohl(header
);
235 c
->timestamp
= ntohl(c
->timestamp
);
236 c
->ssrc
= ntohl(c
->ssrc
);
238 if ((header
>> 30) != 2) {
239 pa_log_warn("Unsupported RTP version.");
243 if ((header
>> 29) & 1) {
244 pa_log_warn("RTP padding not supported.");
248 if ((header
>> 28) & 1) {
249 pa_log_warn("RTP header extensions not supported.");
253 cc
= (header
>> 24) & 0xF;
254 c
->payload
= (uint8_t) ((header
>> 16) & 127U);
255 c
->sequence
= (uint16_t) (header
& 0xFFFFU
);
257 if (12 + cc
*4 > (unsigned) size
) {
258 pa_log_warn("RTP packet too short. (CSRC)");
262 chunk
->index
+= 12 + cc
*4;
263 chunk
->length
= (size_t) size
- 12 + cc
*4;
265 if (chunk
->length
% c
->frame_size
!= 0) {
266 pa_log_warn("Bad RTP packet size.");
270 c
->memchunk
.index
= chunk
->index
+ chunk
->length
;
271 c
->memchunk
.length
= pa_memblock_get_length(c
->memchunk
.memblock
) - c
->memchunk
.index
;
273 if (c
->memchunk
.length
<= 0) {
274 pa_memblock_unref(c
->memchunk
.memblock
);
275 pa_memchunk_reset(&c
->memchunk
);
282 pa_memblock_unref(chunk
->memblock
);
287 uint8_t pa_rtp_payload_from_sample_spec(const pa_sample_spec
*ss
) {
290 if (ss
->format
== PA_SAMPLE_ULAW
&& ss
->rate
== 8000 && ss
->channels
== 1)
292 if (ss
->format
== PA_SAMPLE_ALAW
&& ss
->rate
== 8000 && ss
->channels
== 1)
294 if (ss
->format
== PA_SAMPLE_S16BE
&& ss
->rate
== 44100 && ss
->channels
== 2)
296 if (ss
->format
== PA_SAMPLE_S16BE
&& ss
->rate
== 44100 && ss
->channels
== 1)
302 pa_sample_spec
*pa_rtp_sample_spec_from_payload(uint8_t payload
, pa_sample_spec
*ss
) {
308 ss
->format
= PA_SAMPLE_ULAW
;
314 ss
->format
= PA_SAMPLE_ALAW
;
320 ss
->format
= PA_SAMPLE_S16BE
;
326 ss
->format
= PA_SAMPLE_S16BE
;
337 pa_sample_spec
*pa_rtp_sample_spec_fixup(pa_sample_spec
* ss
) {
340 if (!pa_rtp_sample_spec_valid(ss
))
341 ss
->format
= PA_SAMPLE_S16BE
;
343 pa_assert(pa_rtp_sample_spec_valid(ss
));
347 int pa_rtp_sample_spec_valid(const pa_sample_spec
*ss
) {
350 if (!pa_sample_spec_valid(ss
))
354 ss
->format
== PA_SAMPLE_U8
||
355 ss
->format
== PA_SAMPLE_ALAW
||
356 ss
->format
== PA_SAMPLE_ULAW
||
357 ss
->format
== PA_SAMPLE_S16BE
;
360 void pa_rtp_context_destroy(pa_rtp_context
*c
) {
363 pa_assert_se(pa_close(c
->fd
) == 0);
365 if (c
->memchunk
.memblock
)
366 pa_memblock_unref(c
->memchunk
.memblock
);
369 const char* pa_rtp_format_to_string(pa_sample_format_t f
) {
371 case PA_SAMPLE_S16BE
:
384 pa_sample_format_t
pa_rtp_string_to_format(const char *s
) {
387 if (!(strcmp(s
, "L16")))
388 return PA_SAMPLE_S16BE
;
389 else if (!strcmp(s
, "L8"))
391 else if (!strcmp(s
, "PCMA"))
392 return PA_SAMPLE_ALAW
;
393 else if (!strcmp(s
, "PCMU"))
394 return PA_SAMPLE_ULAW
;
396 return PA_SAMPLE_INVALID
;