Merge branch 'sched'
[paraslash.git] / ortp_recv.c
1 /*
2 * Copyright (C) 2005-2006 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 extern int msg_to_buf(mblk_t *, char *, int);
36
37 /**
38 * data specific to the ortp receiver
39 *
40 * \sa receiver receiver_node
41 */
42 struct private_ortp_recv_data {
43 /**
44 *
45 *
46 * whether a header was received
47 *
48 * A flag indicating whether the ortp receiver already received a packet which
49 * contains the audio file header.
50 *
51 * This flag has no effect if the audio stream indicates that no extra headers
52 * will be sent (mp3). Otherwise, all data packets are dropped until the
53 * header is received.
54 */
55 int have_header;
56 /** the ortp session this instance of the receiver belongs to */
57 RtpSession *session;
58 /** the time the first data or header packet was received */
59 struct timeval start;
60 /** ETA of the next chunk */
61 struct timeval next_chunk;
62 /** number of consecutive bad chunks */
63 int c_bad;
64 /** the current timestamp which is passed to the receiving function of libortp */
65 uint32_t timestamp;
66 /** \a timestamp increases by this amount */
67 uint32_t chunk_ts;
68 };
69
70 static void ortp_recv_pre_select(struct sched *s, struct task *t)
71 {
72 struct receiver_node *rn = t->private_data;
73 struct private_ortp_recv_data *pord = rn->private_data;
74 struct timeval tmp;
75
76 if (tv_diff(&s->now, &pord->next_chunk, &tmp) >= 0) {
77 tmp.tv_sec = 0;
78 tmp.tv_usec = 1000;
79 }
80 if (tv_diff(&s->timeout, &tmp, NULL) > 0)
81 s->timeout = tmp;
82 t->ret = 1;
83 }
84
85 static void compute_next_chunk(unsigned chunk_time,
86 struct private_ortp_recv_data *pord)
87 {
88 struct timeval chunk_tv = {0, chunk_time};
89 struct timeval tmp;
90
91 tv_add(&chunk_tv, &pord->next_chunk, &tmp);
92 pord->next_chunk = tmp;
93 pord->timestamp += pord->chunk_ts;
94 PARA_DEBUG_LOG("next chunk (ts = %d) due at %lu:%lu\n",
95 pord->timestamp, pord->next_chunk.tv_sec,
96 pord->next_chunk.tv_usec);
97 }
98
99 static void ortp_recv_post_select(struct sched *s, struct task *t)
100 {
101 struct receiver_node *rn = t->private_data;
102 struct private_ortp_recv_data *pord = rn->private_data;
103 mblk_t *mp;
104 int packet_type, stream_type;
105 char tmpbuf[CHUNK_SIZE + 3];
106 unsigned chunk_time;
107
108 // PARA_INFO_LOG("rn: %p, pord: %p, session: %p\n", rn, pord, pord->session);
109 t->ret = 1;
110 if (pord->start.tv_sec)
111 if (tv_diff(&s->now, &pord->next_chunk, NULL) < 0)
112 return;
113 mp = rtp_session_recvm_with_ts(pord->session, pord->timestamp);
114 if (!mp) {
115 struct timeval min_delay = {0, 100};
116 // PARA_INFO_LOG("nope, chunk_ts = %d, loaded: %d, bad: %d\n",
117 // pord->timestamp, rn->loaded, pord->c_bad);
118 pord->c_bad++;
119 t->ret = -E_TOO_MANY_BAD_CHUNKS;
120 if ((pord->c_bad > 5000 && pord->start.tv_sec) || pord->c_bad > 10000)
121 return;
122 t->ret = 1;
123 tv_add(&s->now, &min_delay, &pord->next_chunk);
124 return;
125 }
126 /* okay, we have a chunk of data */
127 if (!pord->start.tv_sec)
128 pord->start = s->now;
129 t->ret = msg_to_buf(mp, tmpbuf, CHUNK_SIZE);
130 if (t->ret < ORTP_AUDIO_HEADER_LEN) {
131 rn->eof = 1;
132 if (t->ret < 0)
133 t->ret = -E_MSG_TO_BUF;
134 else
135 t->ret = -E_ORTP_RECV_EOF;
136 goto err_out;
137 }
138 packet_type = READ_PACKET_TYPE(tmpbuf);
139 stream_type = READ_STREAM_TYPE(tmpbuf);
140 chunk_time = READ_CHUNK_TIME(tmpbuf);
141 pord->chunk_ts = READ_CHUNK_TS(tmpbuf);
142 // PARA_INFO_LOG("packet type: %d, stream type: %d, chunk time: %u,"
143 // " chunk_ts: %u, loaded: %u, bad: %d\n", packet_type,
144 // stream_type, chunk_time, pord->chunk_ts, rn->loaded, pord->c_bad);
145 switch (packet_type) {
146 unsigned header_len, payload_len;
147 case ORTP_EOF:
148 rn->eof = 1;
149 t->ret = -E_ORTP_RECV_EOF;
150 goto err_out;
151 case ORTP_BOF:
152 PARA_INFO_LOG("bof (%d)\n", t->ret);
153 pord->have_header = 1;
154 /* fall through */
155 case ORTP_DATA:
156 if (!pord->have_header && stream_type)
157 /* can't use the data, wait for header */
158 goto success;
159 if (t->ret + rn->loaded >= CHUNK_SIZE + ORTP_AUDIO_HEADER_LEN) {
160 t->ret = -E_OVERRUN;
161 goto err_out;
162 }
163 if (t->ret > ORTP_AUDIO_HEADER_LEN) {
164 memcpy(rn->buf + rn->loaded, tmpbuf + ORTP_AUDIO_HEADER_LEN,
165 t->ret - ORTP_AUDIO_HEADER_LEN);
166 rn->loaded += t->ret - ORTP_AUDIO_HEADER_LEN;
167 }
168 goto success;
169 case ORTP_HEADER:
170 header_len = READ_HEADER_LEN(tmpbuf);
171 PARA_DEBUG_LOG("header packet (%d bytes), header len: %d\n",
172 t->ret, header_len);
173 if (!pord->have_header) {
174 pord->have_header = 1;
175 memcpy(rn->buf, tmpbuf + ORTP_AUDIO_HEADER_LEN,
176 t->ret - ORTP_AUDIO_HEADER_LEN);
177 rn->loaded = t->ret - ORTP_AUDIO_HEADER_LEN;
178 goto success;
179 }
180 if (header_len + ORTP_AUDIO_HEADER_LEN > t->ret) {
181 t->ret = -E_INVALID_HEADER;
182 goto err_out;
183 }
184 payload_len = t->ret - ORTP_AUDIO_HEADER_LEN - header_len;
185 // PARA_INFO_LOG("len: %d header_len: %d, payload_len: %d, loaded: %d\n", ret,
186 // header_len, payload_len, rn->loaded);
187 if (rn->loaded + payload_len > CHUNK_SIZE) {
188 t->ret = -E_OVERRUN;
189 goto err_out;
190 }
191 if (payload_len)
192 memcpy(rn->buf + rn->loaded, tmpbuf
193 + (t->ret - payload_len), payload_len);
194 rn->loaded += payload_len;
195 goto success;
196 }
197 success:
198 t->ret = 1;
199 freemsg(mp);
200 if (pord->c_bad) {
201 pord->c_bad = 0;
202 pord->next_chunk = s->now;
203 }
204 compute_next_chunk(chunk_time, pord);
205 return;
206 err_out:
207 rn->eof = 1;
208 freemsg(mp);
209 }
210
211 static void ortp_shutdown(void)
212 {
213 // ortp_global_stats_display();
214 ortp_exit();
215 }
216
217 static void ortp_recv_close(struct receiver_node *rn)
218 {
219 struct private_ortp_recv_data *pord = rn->private_data;
220
221 rtp_session_destroy(pord->session);
222 free(rn->private_data);
223 free(rn->buf);
224 }
225
226 static void *ortp_recv_parse_config(int argc, char **argv)
227 {
228 int ret;
229
230 struct ortp_recv_args_info *tmp = para_calloc(sizeof(struct ortp_recv_args_info));
231
232 ret = ortp_recv_cmdline_parser(argc, argv, tmp)? -E_ORTP_SYNTAX : 1;
233 if (ret > 0)
234 return tmp;
235 free(tmp);
236 return NULL;
237 }
238
239 static int ortp_recv_open(struct receiver_node *rn)
240 {
241 struct private_ortp_recv_data *pord;
242 struct ortp_recv_args_info *conf = rn->conf;
243
244 rn->buf = para_calloc(CHUNK_SIZE);
245
246 rn->private_data = para_calloc(sizeof(struct private_ortp_recv_data));
247 pord = rn->private_data;
248 pord->session = rtp_session_new(RTP_SESSION_RECVONLY);
249 PARA_NOTICE_LOG("receiving from %s:%d\n", conf->host_arg, conf->port_arg);
250 rtp_session_set_local_addr(pord->session, conf->host_arg, conf->port_arg);
251 rtp_session_set_payload_type(pord->session, PAYLOAD_AUDIO_CONTINUOUS);
252 if (conf->jitter_compensation_arg) {
253 rtp_session_enable_adaptive_jitter_compensation(pord->session, TRUE);
254 rtp_session_set_jitter_compensation(pord->session,
255 conf->jitter_compensation_arg);
256 }
257 return 1;
258 }
259
260 /**
261 * the init function of the ortp receiver
262 *
263 * \param r pointer to the receiver struct to initialize
264 *
265 * Initialize all function pointers of \a r and call libortp's ortp_init().
266 */
267 void ortp_recv_init(struct receiver *r)
268 {
269 r->shutdown = ortp_shutdown;
270 r->open = ortp_recv_open;
271 r->close = ortp_recv_close;
272 r->pre_select = ortp_recv_pre_select;
273 r->post_select = ortp_recv_post_select;
274 r->parse_config = ortp_recv_parse_config;
275
276 ortp_init();
277 }