be more carful wrt. signed vs. unsigned argument passing
[paraslash.git] / ortp_recv.c
1 /*
2 * Copyright (C) 2005-2007 Andre Noll <maan@systemlinux.org>
3 *
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
17 */
18 /** \file ortp_recv.c paraslash's ortp receiver */
19
20 #include <ortp/ortp.h>
21 #include "para.h"
22
23 #include "ortp.h"
24 #include "list.h"
25 #include "sched.h"
26 #include "recv.h"
27 #include "ortp_recv.cmdline.h"
28
29 #include "error.h"
30 #include "audiod.h"
31 #include "string.h"
32
33 #define CHUNK_SIZE 128 * 1024
34
35 /**
36 * data specific to the ortp receiver
37 *
38 * \sa receiver receiver_node
39 */
40 struct private_ortp_recv_data {
41 /**
42 *
43 *
44 * whether a header was received
45 *
46 * A flag indicating whether the ortp receiver already received a packet which
47 * contains the audio file header.
48 *
49 * This flag has no effect if the audio stream indicates that no extra headers
50 * will be sent (mp3). Otherwise, all data packets are dropped until the
51 * header is received.
52 */
53 int have_header;
54 /** the ortp session this instance of the receiver belongs to */
55 RtpSession *session;
56 /** the time the first data or header packet was received */
57 struct timeval start;
58 /** ETA of the next chunk */
59 struct timeval next_chunk;
60 /** number of consecutive bad chunks */
61 int c_bad;
62 /** the current timestamp which is passed to the receiving function of libortp */
63 uint32_t timestamp;
64 /** \a timestamp increases by this amount */
65 uint32_t chunk_ts;
66 };
67
68
69 static int msg_to_buf(mblk_t *mp, char *buffer, int len)
70 {
71 mblk_t *m, *mprev;
72 size_t mlen, rlen = len;
73
74 m = mp->b_cont;
75 mprev = mp;
76 while (m != NULL) {
77 mlen = m->b_wptr - m->b_rptr;
78 if (mlen <= rlen) {
79 mblk_t *consumed = m;
80 memcpy(buffer, m->b_rptr, mlen);
81 /* go to next mblk_t */
82 mprev->b_cont = m->b_cont;
83 m = m->b_cont;
84 consumed->b_cont = NULL;
85 freeb (consumed);
86 buffer += mlen;
87 rlen -= mlen;
88 } else {
89 memcpy (buffer, m->b_rptr, rlen);
90 m->b_rptr += rlen;
91 return len;
92 }
93 }
94 return len - rlen;
95 }
96
97
98 static void ortp_recv_pre_select(struct sched *s, struct task *t)
99 {
100 struct receiver_node *rn = t->private_data;
101 struct private_ortp_recv_data *pord = rn->private_data;
102 struct timeval tmp;
103
104 if (tv_diff(now, &pord->next_chunk, &tmp) >= 0) {
105 tmp.tv_sec = 0;
106 tmp.tv_usec = 1000;
107 }
108 if (tv_diff(&s->timeout, &tmp, NULL) > 0)
109 s->timeout = tmp;
110 t->ret = 1;
111 }
112
113 static void compute_next_chunk(unsigned chunk_time,
114 struct private_ortp_recv_data *pord)
115 {
116 struct timeval chunk_tv = {0, chunk_time};
117 struct timeval tmp;
118
119 tv_add(&chunk_tv, &pord->next_chunk, &tmp);
120 pord->next_chunk = tmp;
121 pord->timestamp += pord->chunk_ts;
122 // PARA_DEBUG_LOG("next chunk (ts = %d) due at %lu:%lu\n",
123 // pord->timestamp, pord->next_chunk.tv_sec,
124 // pord->next_chunk.tv_usec);
125 }
126
127 static void ortp_recv_post_select(__a_unused struct sched *s, struct task *t)
128 {
129 struct receiver_node *rn = t->private_data;
130 struct private_ortp_recv_data *pord = rn->private_data;
131 mblk_t *mp;
132 int packet_type, stream_type;
133 char tmpbuf[CHUNK_SIZE + 3];
134 unsigned chunk_time;
135 size_t packet_size;
136
137 // PARA_INFO_LOG("rn: %p, pord: %p, session: %p\n", rn, pord, pord->session);
138 t->ret = -E_ORTP_RECV_EOF;
139 if (rn->output_eof && *rn->output_eof) {
140 rn->eof = 1;
141 return;
142 }
143 t->ret = 1;
144 if (pord->start.tv_sec)
145 if (tv_diff(now, &pord->next_chunk, NULL) < 0)
146 return;
147 mp = rtp_session_recvm_with_ts(pord->session, pord->timestamp);
148 if (!mp) {
149 struct timeval min_delay = {0, 100};
150 // PARA_INFO_LOG("nope, chunk_ts = %d, loaded: %d, bad: %d\n",
151 // pord->timestamp, rn->loaded, pord->c_bad);
152 pord->c_bad++;
153 t->ret = -E_TOO_MANY_BAD_CHUNKS;
154 if ((pord->c_bad > 5000 && pord->start.tv_sec) || pord->c_bad > 10000)
155 return;
156 t->ret = 1;
157 tv_add(now, &min_delay, &pord->next_chunk);
158 return;
159 }
160 /* okay, we have a chunk of data */
161 if (!pord->start.tv_sec)
162 pord->start = *now;
163 t->ret = msg_to_buf(mp, tmpbuf, CHUNK_SIZE);
164 if (t->ret < ORTP_AUDIO_HEADER_LEN) {
165 rn->eof = 1;
166 if (t->ret < 0)
167 t->ret = -E_MSG_TO_BUF;
168 else
169 t->ret = -E_ORTP_RECV_EOF;
170 goto err_out;
171 }
172 packet_size = t->ret;
173 packet_type = READ_PACKET_TYPE(tmpbuf);
174 stream_type = READ_STREAM_TYPE(tmpbuf);
175 chunk_time = READ_CHUNK_TIME(tmpbuf);
176 pord->chunk_ts = READ_CHUNK_TS(tmpbuf);
177 // PARA_INFO_LOG("packet type: %d, stream type: %d, chunk time: %u,"
178 // " chunk_ts: %u, loaded: %u, bad: %d\n", packet_type,
179 // stream_type, chunk_time, pord->chunk_ts, rn->loaded, pord->c_bad);
180 switch (packet_type) {
181 unsigned header_len, payload_len;
182 case ORTP_EOF:
183 rn->eof = 1;
184 t->ret = -E_ORTP_RECV_EOF;
185 goto err_out;
186 case ORTP_BOF:
187 PARA_INFO_LOG("bof (%zu)\n", packet_size);
188 pord->have_header = 1;
189 /* fall through */
190 case ORTP_DATA:
191 if (!pord->have_header && stream_type)
192 /* can't use the data, wait for header */
193 goto success;
194 t->ret = -E_OVERRUN;
195 if (packet_size + rn->loaded >= CHUNK_SIZE + ORTP_AUDIO_HEADER_LEN)
196 goto err_out;
197 if (packet_size > ORTP_AUDIO_HEADER_LEN) {
198 memcpy(rn->buf + rn->loaded, tmpbuf + ORTP_AUDIO_HEADER_LEN,
199 packet_size - ORTP_AUDIO_HEADER_LEN);
200 rn->loaded += packet_size - ORTP_AUDIO_HEADER_LEN;
201 }
202 goto success;
203 case ORTP_HEADER:
204 header_len = READ_HEADER_LEN(tmpbuf);
205 PARA_DEBUG_LOG("header packet (%zu bytes), header len: %d\n",
206 packet_size, header_len);
207 if (!pord->have_header) {
208 pord->have_header = 1;
209 memcpy(rn->buf, tmpbuf + ORTP_AUDIO_HEADER_LEN,
210 packet_size - ORTP_AUDIO_HEADER_LEN);
211 rn->loaded = packet_size - ORTP_AUDIO_HEADER_LEN;
212 goto success;
213 }
214 t->ret = -E_INVALID_HEADER;
215 if (header_len + ORTP_AUDIO_HEADER_LEN > packet_size)
216 goto err_out;
217 payload_len = packet_size - ORTP_AUDIO_HEADER_LEN - header_len;
218 t->ret = -E_OVERRUN;
219 if (rn->loaded + payload_len > CHUNK_SIZE)
220 goto err_out;
221 if (payload_len)
222 memcpy(rn->buf + rn->loaded, tmpbuf
223 + (packet_size - payload_len), payload_len);
224 rn->loaded += payload_len;
225 goto success;
226 }
227 success:
228 t->ret = 1;
229 freemsg(mp);
230 if (pord->c_bad) {
231 pord->c_bad = 0;
232 pord->next_chunk = *now;
233 }
234 compute_next_chunk(chunk_time, pord);
235 return;
236 err_out:
237 rn->eof = 1;
238 freemsg(mp);
239 }
240
241 static void ortp_shutdown(void)
242 {
243 // ortp_global_stats_display();
244 ortp_exit();
245 }
246
247 static void ortp_recv_close(struct receiver_node *rn)
248 {
249 struct private_ortp_recv_data *pord = rn->private_data;
250
251 rtp_session_destroy(pord->session);
252 free(rn->private_data);
253 free(rn->buf);
254 }
255
256 static void *ortp_recv_parse_config(int argc, char **argv)
257 {
258 int ret;
259
260 struct ortp_recv_args_info *tmp =
261 para_calloc(sizeof(struct ortp_recv_args_info));
262
263 ret = ortp_recv_cmdline_parser(argc, argv, tmp)? -E_ORTP_SYNTAX : 1;
264 if (ret > 0)
265 return tmp;
266 free(tmp);
267 return NULL;
268 }
269
270 static int ortp_recv_open(struct receiver_node *rn)
271 {
272 struct private_ortp_recv_data *pord;
273 struct ortp_recv_args_info *c = rn->conf;
274
275 rn->buf = para_calloc(CHUNK_SIZE);
276
277 rn->private_data = para_calloc(sizeof(struct private_ortp_recv_data));
278 pord = rn->private_data;
279 pord->session = rtp_session_new(RTP_SESSION_RECVONLY);
280 PARA_NOTICE_LOG("receiving from %s:%d\n", c->host_arg, c->port_arg);
281 rtp_session_set_local_addr(pord->session, c->host_arg, c->port_arg);
282 rtp_session_set_remote_addr(pord->session, c->host_arg, c->port_arg);
283 rtp_session_set_payload_type(pord->session, PAYLOAD_AUDIO_CONTINUOUS);
284 if (c->jitter_compensation_arg) {
285 rtp_session_enable_adaptive_jitter_compensation(pord->session, TRUE);
286 rtp_session_set_jitter_compensation(pord->session,
287 c->jitter_compensation_arg);
288 }
289 return 1;
290 }
291
292 /**
293 * the init function of the ortp receiver
294 *
295 * \param r pointer to the receiver struct to initialize
296 *
297 * Initialize all function pointers of \a r and call libortp's ortp_init().
298 */
299 void ortp_recv_init(struct receiver *r)
300 {
301 r->shutdown = ortp_shutdown;
302 r->open = ortp_recv_open;
303 r->close = ortp_recv_close;
304 r->pre_select = ortp_recv_pre_select;
305 r->post_select = ortp_recv_post_select;
306 r->parse_config = ortp_recv_parse_config;
307
308 ortp_init();
309 }