simplify open_audiod_pipe(), remove a trailing newline which caused
[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 "para.h"
21 #include <ortp/ortp.h>
22
23 #include "ortp.h"
24 #include "recv.h"
25 #include "ortp_recv.cmdline.h"
26
27 #include "error.h"
28 #include "audiod.h"
29 #include "string.h"
30
31 #define CHUNK_SIZE 128 * 1024
32
33 extern gint msg_to_buf(mblk_t *, char *, gint);
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 *
45 * whether a header was received
46 *
47 * A flag indicating whether the ortp receiver already received a packet which
48 * contains the audio file header.
49 *
50 * This flag has no effect if the audio stream indicates that no extra headers
51 * will be sent (mp3). Otherwise, all data packets are dropped until the
52 * header is received.
53 */
54 int have_header;
55 /** the ortp session this instance of the receiver belongs to */
56 RtpSession *session;
57 /** the time the first data or header packet was received */
58 struct timeval start;
59 /** ETA of the next chunk */
60 struct timeval next_chunk;
61 /** number of consecutive bad chunks */
62 int c_bad;
63 /** the current timestamp which is passed to the receiving function of libortp */
64 guint32 timestamp;
65 /** \a timestamp increases by this amount */
66 guint32 chunk_ts;
67 };
68
69 #if 1
70 static void recalc_timestamp(struct receiver_node *rn)
71 {
72 struct timeval now, tmp, min_delay = {0, 5000};
73 struct private_ortp_recv_data *pord = rn->private_data;
74
75 gettimeofday(&now, NULL);
76 tv_add(&now, &min_delay, &pord->next_chunk);
77 if (pord->start.tv_sec) {
78 int new_ts;
79
80 tv_diff(&now, &pord->start, &tmp);
81 new_ts = rtp_session_time_to_ts(pord->session, tv2ms(&tmp));
82 // if (pord->c_bad > 50)
83 pord->timestamp = new_ts;
84 } else
85 pord->timestamp = 0;
86 PARA_DEBUG_LOG("ts=%u, c_bad = %d\n", pord->timestamp, pord->c_bad);
87
88 }
89 #endif
90
91 static int ortp_recv_pre_select(struct receiver_node *rn,
92 __unused fd_set *rfds, __unused fd_set *wfds,
93 struct timeval *timeout)
94 {
95 struct private_ortp_recv_data *pord = rn->private_data;
96 struct timeval now, tmp;
97
98 gettimeofday(&now, NULL);
99 if (tv_diff(&now, &pord->next_chunk, &tmp) >= 0) {
100 tmp.tv_sec = 0;
101 tmp.tv_usec = 1000;
102 }
103 if (tv_diff(timeout, &tmp, NULL) > 0)
104 *timeout = tmp;
105 return -1; /* we did not modify the fd sets */
106 }
107
108 static void compute_next_chunk(struct timeval *now, unsigned chunk_time,
109 struct private_ortp_recv_data *pord)
110 {
111 struct timeval chunk_tv = {0, chunk_time};
112 // if (!pord->start.tv_sec) {
113 // pord->start = *now;
114 // tv_add(&chunk_tv, now, &pord->next_chunk);
115 // } else
116 {
117 struct timeval tmp;
118 tv_add(&chunk_tv, &pord->next_chunk, &tmp);
119 pord->next_chunk = tmp;
120 }
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 /** \cond */
127 #define BUF_TO_VAL(buf) (((unsigned)(buf)[1] & 0xff) + 256 * ((unsigned)(buf)[0] & 0xff))
128 /** \endcond */
129
130
131 static int ortp_recv_post_select(struct receiver_node *rn, int select_ret,
132 __unused fd_set *rfds, __unused fd_set *wfds)
133 {
134 struct private_ortp_recv_data *pord = rn->private_data;
135 mblk_t *mp;
136 int ret, packet_type, stream_type;
137 char tmpbuf[CHUNK_SIZE + 3];
138 struct timeval now;
139 unsigned chunk_time;
140
141 gettimeofday(&now, NULL);
142 PARA_DEBUG_LOG("rn: %p, pord: %p, session: %p\n", rn, pord, pord->session);
143 if (pord->start.tv_sec) {
144 struct timeval diff;
145 if (tv_diff(&now, &pord->next_chunk, &diff) < 0) {
146 long unsigned diff_ms = tv2ms(&diff);
147 if (diff_ms > 5) {
148 PARA_DEBUG_LOG("too early: %lu ms left\n",
149 diff_ms);
150 return 1;
151 }
152 }
153 }
154 mp = rtp_session_recvm_with_ts(pord->session, pord->timestamp);
155 if (!mp) {
156 pord->c_bad++;
157 if ((pord->c_bad > 500 && pord->start.tv_sec) || pord->c_bad > 10000)
158 return -E_TOO_MANY_BAD_CHUNKS;
159 // PARA_NOTICE_LOG("did not receive buffer, ts=%u, c_bad = %d, to = %lu\n",
160 // pord->timestamp, pord->c_bad, tv2ms(&tmp));
161 if (pord->c_bad > 100)
162 recalc_timestamp(rn);
163 return 1;
164 }
165 /* okay, we have a chunk of data */
166 PARA_DEBUG_LOG("have it ts = %d, chunk_ts = %d, loaded: %d, bad: %d\n",
167 pord->timestamp, pord->chunk_ts, rn->loaded, pord->c_bad);
168 if (!pord->start.tv_sec)
169 pord->start = now;
170 ret = msg_to_buf(mp, tmpbuf, CHUNK_SIZE);
171 if (ret < ORTP_AUDIO_HEADER_LEN) {
172 if (ret < 0)
173 ret = -E_MSG_TO_BUF;
174 else
175 ret = 0;
176 goto err_out;
177 }
178 packet_type = READ_PACKET_TYPE(tmpbuf);
179 stream_type = READ_STREAM_TYPE(tmpbuf);
180 chunk_time = READ_CHUNK_TIME(tmpbuf);
181 pord->chunk_ts = READ_CHUNK_TS(tmpbuf);
182 PARA_DEBUG_LOG("packet type: %d, stream type: %d, chunk time: %u, chunk_ts: %u\n", packet_type,
183 stream_type, chunk_time, pord->chunk_ts);
184 switch (packet_type) {
185 unsigned header_len, payload_len;
186 case ORTP_EOF:
187 ret = 0;
188 goto err_out;
189 case ORTP_BOF:
190 PARA_INFO_LOG("bof (%d)\n", ret);
191 pord->have_header = 1;
192 /* fall through */
193 case ORTP_DATA:
194 if (!pord->have_header && stream_type) /* can't use the data, wait for header */
195 goto success;
196 if (ret + rn->loaded >= CHUNK_SIZE + ORTP_AUDIO_HEADER_LEN) {
197 ret = -E_OVERRUN;
198 goto err_out;
199 }
200 if (ret > ORTP_AUDIO_HEADER_LEN) {
201 memcpy(rn->buf + rn->loaded, tmpbuf + ORTP_AUDIO_HEADER_LEN,
202 ret - ORTP_AUDIO_HEADER_LEN);
203 rn->loaded += ret - ORTP_AUDIO_HEADER_LEN;
204 }
205 goto success;
206 case ORTP_HEADER:
207 header_len = READ_HEADER_LEN(tmpbuf);
208 PARA_DEBUG_LOG("header packet (%d bytes), header len: %d\n", ret, header_len);
209 if (!pord->have_header) {
210 pord->have_header = 1;
211 memcpy(rn->buf, tmpbuf + ORTP_AUDIO_HEADER_LEN,
212 ret - ORTP_AUDIO_HEADER_LEN);
213 rn->loaded = ret - ORTP_AUDIO_HEADER_LEN;
214 goto success;
215 }
216 if (header_len + ORTP_AUDIO_HEADER_LEN > ret) {
217 ret = -E_INVALID_HEADER;
218 goto err_out;
219 }
220 payload_len = ret - ORTP_AUDIO_HEADER_LEN - header_len;
221 // PARA_INFO_LOG("len: %d header_len: %d, payload_len: %d, loaded: %d\n", ret,
222 // header_len, payload_len, rn->loaded);
223 if (rn->loaded + payload_len > CHUNK_SIZE) {
224 ret = -E_OVERRUN;
225 goto err_out;
226 }
227 if (payload_len)
228 memcpy(rn->buf + rn->loaded, tmpbuf
229 + (ret - payload_len), payload_len);
230 rn->loaded += payload_len;
231 goto success;
232 }
233 success:
234 freemsg(mp);
235 pord->c_bad = 0;
236 compute_next_chunk(&now, chunk_time, pord);
237 return 1;
238 err_out:
239 freemsg(mp);
240 return ret;
241 }
242
243 static void ortp_shutdown(void)
244 {
245 // ortp_global_stats_display();
246 ortp_exit();
247 }
248
249 static void ortp_recv_close(struct receiver_node *rn)
250 {
251 struct private_ortp_recv_data *pord = rn->private_data;
252
253 rtp_session_destroy(pord->session);
254 free(rn->private_data);
255 free(rn->buf);
256 }
257
258 static void *ortp_recv_parse_config(int argc, char **argv)
259 {
260 int ret;
261
262 struct gengetopt_args_info *tmp = para_calloc(sizeof(struct gengetopt_args_info));
263
264 ret = ortp_recv_cmdline_parser(argc, argv, tmp)? -E_ORTP_SYNTAX : 1;
265 if (ret > 0)
266 return tmp;
267 free(tmp);
268 return NULL;
269 }
270
271 static int ortp_recv_open(struct receiver_node *rn)
272 {
273 struct private_ortp_recv_data *pord;
274 struct gengetopt_args_info *conf = rn->conf;
275
276 rn->buf = para_calloc(CHUNK_SIZE);
277
278 rn->private_data = para_calloc(sizeof(struct private_ortp_recv_data));
279 pord = rn->private_data;
280 pord->session = rtp_session_new(RTP_SESSION_RECVONLY);
281 PARA_NOTICE_LOG("receiving from %s:%d\n", conf->host_arg, conf->port_arg);
282 rtp_session_set_local_addr(pord->session, conf->host_arg, conf->port_arg);
283 rtp_session_set_payload_type(pord->session, PAYLOAD_AUDIO_CONTINUOUS);
284 return 1;
285 }
286
287 /**
288 * the init function of the ortp receiver
289 *
290 * \param r pointer to the receiver struct to initialize
291 *
292 * Initialize all function pointers of \a r and call libortp's ortp_init().
293 */
294 void ortp_recv_init(struct receiver *r)
295 {
296 r->shutdown = ortp_shutdown;
297 r->open = ortp_recv_open;
298 r->close = ortp_recv_close;
299 r->pre_select = ortp_recv_pre_select;
300 r->post_select = ortp_recv_post_select;
301 r->parse_config = ortp_recv_parse_config;
302
303 ortp_init();
304 ortp_set_debug_file("oRTP", NULL);
305 }