2 * Copyright (C) 2005-2007 Andre Noll <maan@systemlinux.org>
4 * Licensed under the GPL v2. For licencing details see COPYING.
6 /** \file ortp_recv.c paraslash's ortp receiver */
15 #include "ortp_recv.cmdline.h"
21 #define CHUNK_SIZE 128 * 1024
24 * data specific to the ortp receiver
26 * \sa receiver receiver_node
28 struct private_ortp_recv_data {
32 * whether a header was received
34 * A flag indicating whether the ortp receiver already received a packet which
35 * contains the audio file header.
37 * This flag has no effect if the audio stream indicates that no extra headers
38 * will be sent (mp3). Otherwise, all data packets are dropped until the
42 /** the ortp session this instance of the receiver belongs to */
44 /** the time the first data or header packet was received */
46 /** ETA of the next chunk */
47 struct timeval next_chunk;
48 /** number of consecutive bad chunks */
50 /** the current timestamp which is passed to the receiving function of libortp */
52 /** \a timestamp increases by this amount */
57 static int msg_to_buf(mblk_t *mp, char *buffer, int len)
60 size_t mlen, rlen = len;
65 mlen = m->b_wptr - m->b_rptr;
68 memcpy(buffer, m->b_rptr, mlen);
69 /* go to next mblk_t */
70 mprev->b_cont = m->b_cont;
72 consumed->b_cont = NULL;
77 memcpy (buffer, m->b_rptr, rlen);
86 static void ortp_recv_pre_select(struct sched *s, struct task *t)
88 struct receiver_node *rn = t->private_data;
89 struct private_ortp_recv_data *pord = rn->private_data;
92 if (tv_diff(now, &pord->next_chunk, &tmp) >= 0) {
96 if (tv_diff(&s->timeout, &tmp, NULL) > 0)
101 static void compute_next_chunk(unsigned chunk_time,
102 struct private_ortp_recv_data *pord)
104 struct timeval chunk_tv = {0, chunk_time};
107 tv_add(&chunk_tv, &pord->next_chunk, &tmp);
108 pord->next_chunk = tmp;
109 pord->timestamp += pord->chunk_ts;
110 // PARA_DEBUG_LOG("next chunk (ts = %d) due at %lu:%lu\n",
111 // pord->timestamp, pord->next_chunk.tv_sec,
112 // pord->next_chunk.tv_usec);
115 static void ortp_recv_post_select(__a_unused struct sched *s, struct task *t)
117 struct receiver_node *rn = t->private_data;
118 struct private_ortp_recv_data *pord = rn->private_data;
120 int packet_type, stream_type;
121 char tmpbuf[CHUNK_SIZE + 3];
125 // PARA_INFO_LOG("rn: %p, pord: %p, session: %p\n", rn, pord, pord->session);
126 t->ret = -E_ORTP_RECV_EOF;
127 if (rn->output_eof && *rn->output_eof) {
132 if (pord->start.tv_sec)
133 if (tv_diff(now, &pord->next_chunk, NULL) < 0)
135 mp = rtp_session_recvm_with_ts(pord->session, pord->timestamp);
137 struct timeval min_delay = {0, 100};
138 // PARA_INFO_LOG("nope, chunk_ts = %d, loaded: %d, bad: %d\n",
139 // pord->timestamp, rn->loaded, pord->c_bad);
141 t->ret = -E_TOO_MANY_BAD_CHUNKS;
142 if ((pord->c_bad > 5000 && pord->start.tv_sec) || pord->c_bad > 10000)
145 tv_add(now, &min_delay, &pord->next_chunk);
148 /* okay, we have a chunk of data */
149 if (!pord->start.tv_sec)
151 t->ret = msg_to_buf(mp, tmpbuf, CHUNK_SIZE);
152 if (t->ret < ORTP_AUDIO_HEADER_LEN) {
155 t->ret = -E_MSG_TO_BUF;
157 t->ret = -E_ORTP_RECV_EOF;
160 packet_size = t->ret;
161 packet_type = READ_PACKET_TYPE(tmpbuf);
162 stream_type = READ_STREAM_TYPE(tmpbuf);
163 chunk_time = READ_CHUNK_TIME(tmpbuf);
164 pord->chunk_ts = READ_CHUNK_TS(tmpbuf);
165 // PARA_INFO_LOG("packet type: %d, stream type: %d, chunk time: %u,"
166 // " chunk_ts: %u, loaded: %u, bad: %d\n", packet_type,
167 // stream_type, chunk_time, pord->chunk_ts, rn->loaded, pord->c_bad);
168 switch (packet_type) {
169 unsigned header_len, payload_len;
172 t->ret = -E_ORTP_RECV_EOF;
175 PARA_INFO_LOG("bof (%zu)\n", packet_size);
176 pord->have_header = 1;
179 if (!pord->have_header && stream_type)
180 /* can't use the data, wait for header */
183 if (packet_size + rn->loaded >= CHUNK_SIZE + ORTP_AUDIO_HEADER_LEN)
185 if (packet_size > ORTP_AUDIO_HEADER_LEN) {
186 memcpy(rn->buf + rn->loaded, tmpbuf + ORTP_AUDIO_HEADER_LEN,
187 packet_size - ORTP_AUDIO_HEADER_LEN);
188 rn->loaded += packet_size - ORTP_AUDIO_HEADER_LEN;
192 header_len = READ_HEADER_LEN(tmpbuf);
193 PARA_DEBUG_LOG("header packet (%zu bytes), header len: %d\n",
194 packet_size, header_len);
195 if (!pord->have_header) {
196 pord->have_header = 1;
197 memcpy(rn->buf, tmpbuf + ORTP_AUDIO_HEADER_LEN,
198 packet_size - ORTP_AUDIO_HEADER_LEN);
199 rn->loaded = packet_size - ORTP_AUDIO_HEADER_LEN;
202 t->ret = -E_INVALID_HEADER;
203 if (header_len + ORTP_AUDIO_HEADER_LEN > packet_size)
205 payload_len = packet_size - ORTP_AUDIO_HEADER_LEN - header_len;
207 if (rn->loaded + payload_len > CHUNK_SIZE)
210 memcpy(rn->buf + rn->loaded, tmpbuf
211 + (packet_size - payload_len), payload_len);
212 rn->loaded += payload_len;
220 pord->next_chunk = *now;
222 compute_next_chunk(chunk_time, pord);
229 static void ortp_shutdown(void)
231 // ortp_global_stats_display();
235 static void ortp_recv_close(struct receiver_node *rn)
237 struct private_ortp_recv_data *pord = rn->private_data;
239 rtp_session_destroy(pord->session);
240 free(rn->private_data);
244 static void *ortp_recv_parse_config(int argc, char **argv)
248 struct ortp_recv_args_info *tmp =
249 para_calloc(sizeof(struct ortp_recv_args_info));
251 ret = ortp_recv_cmdline_parser(argc, argv, tmp)? -E_ORTP_SYNTAX : 1;
258 static int ortp_recv_open(struct receiver_node *rn)
260 struct private_ortp_recv_data *pord;
261 struct ortp_recv_args_info *c = rn->conf;
263 rn->buf = para_calloc(CHUNK_SIZE);
265 rn->private_data = para_calloc(sizeof(struct private_ortp_recv_data));
266 pord = rn->private_data;
267 pord->session = rtp_session_new(RTP_SESSION_RECVONLY);
268 PARA_NOTICE_LOG("receiving from %s:%d\n", c->host_arg, c->port_arg);
269 rtp_session_set_local_addr(pord->session, c->host_arg, c->port_arg);
270 rtp_session_set_remote_addr(pord->session, c->host_arg, c->port_arg);
271 rtp_session_set_payload_type(pord->session, PAYLOAD_AUDIO_CONTINUOUS);
272 rtp_session_enable_adaptive_jitter_compensation(pord->session, TRUE);
273 rtp_session_set_jitter_compensation(pord->session,
274 c->jitter_compensation_arg);
279 * the init function of the ortp receiver
281 * \param r pointer to the receiver struct to initialize
283 * Initialize all function pointers of \a r and call libortp's ortp_init().
285 void ortp_recv_init(struct receiver *r)
287 r->shutdown = ortp_shutdown;
288 r->open = ortp_recv_open;
289 r->close = ortp_recv_close;
290 r->pre_select = ortp_recv_pre_select;
291 r->post_select = ortp_recv_post_select;
292 r->parse_config = ortp_recv_parse_config;