12a990c3fc322f8be59f5ed12f19fe08381ec96b
2 * Copyright (C) 2005-2006 Andre Noll <maan@systemlinux.org>
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.
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.
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.
18 /** \file ortp_recv.c paraslash's ortp receiver */
20 #include <ortp/ortp.h>
27 #include "ortp_recv.cmdline.h"
33 #define CHUNK_SIZE 128 * 1024
36 * data specific to the ortp receiver
38 * \sa receiver receiver_node
40 struct private_ortp_recv_data
{
44 * whether a header was received
46 * A flag indicating whether the ortp receiver already received a packet which
47 * contains the audio file header.
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
54 /** the ortp session this instance of the receiver belongs to */
56 /** the time the first data or header packet was received */
58 /** ETA of the next chunk */
59 struct timeval next_chunk
;
60 /** number of consecutive bad chunks */
62 /** the current timestamp which is passed to the receiving function of libortp */
64 /** \a timestamp increases by this amount */
69 static int msg_to_buf(mblk_t
*mp
, char *buffer
, int len
)
78 mlen
= (int) (m
->b_wptr
- m
->b_rptr
);
81 memcpy (buffer
, m
->b_rptr
, mlen
);
82 /* go to next mblk_t */
83 mprev
->b_cont
= m
->b_cont
;
85 consumed
->b_cont
= NULL
;
89 } else { /*if mlen>rlen */
90 memcpy (buffer
, m
->b_rptr
, rlen
);
99 static void ortp_recv_pre_select(struct sched
*s
, struct task
*t
)
101 struct receiver_node
*rn
= t
->private_data
;
102 struct private_ortp_recv_data
*pord
= rn
->private_data
;
105 if (tv_diff(now
, &pord
->next_chunk
, &tmp
) >= 0) {
109 if (tv_diff(&s
->timeout
, &tmp
, NULL
) > 0)
114 static void compute_next_chunk(unsigned chunk_time
,
115 struct private_ortp_recv_data
*pord
)
117 struct timeval chunk_tv
= {0, chunk_time
};
120 tv_add(&chunk_tv
, &pord
->next_chunk
, &tmp
);
121 pord
->next_chunk
= tmp
;
122 pord
->timestamp
+= pord
->chunk_ts
;
123 PARA_DEBUG_LOG("next chunk (ts = %d) due at %lu:%lu\n",
124 pord
->timestamp
, pord
->next_chunk
.tv_sec
,
125 pord
->next_chunk
.tv_usec
);
128 static void ortp_recv_post_select(__a_unused
struct sched
*s
, struct task
*t
)
130 struct receiver_node
*rn
= t
->private_data
;
131 struct private_ortp_recv_data
*pord
= rn
->private_data
;
133 int packet_type
, stream_type
;
134 char tmpbuf
[CHUNK_SIZE
+ 3];
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
) {
144 if (pord
->start
.tv_sec
)
145 if (tv_diff(now
, &pord
->next_chunk
, NULL
) < 0)
147 mp
= rtp_session_recvm_with_ts(pord
->session
, pord
->timestamp
);
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);
153 t
->ret
= -E_TOO_MANY_BAD_CHUNKS
;
154 if ((pord
->c_bad
> 5000 && pord
->start
.tv_sec
) || pord
->c_bad
> 10000)
157 tv_add(now
, &min_delay
, &pord
->next_chunk
);
160 /* okay, we have a chunk of data */
161 if (!pord
->start
.tv_sec
)
163 t
->ret
= msg_to_buf(mp
, tmpbuf
, CHUNK_SIZE
);
164 if (t
->ret
< ORTP_AUDIO_HEADER_LEN
) {
167 t
->ret
= -E_MSG_TO_BUF
;
169 t
->ret
= -E_ORTP_RECV_EOF
;
172 packet_type
= READ_PACKET_TYPE(tmpbuf
);
173 stream_type
= READ_STREAM_TYPE(tmpbuf
);
174 chunk_time
= READ_CHUNK_TIME(tmpbuf
);
175 pord
->chunk_ts
= READ_CHUNK_TS(tmpbuf
);
176 // PARA_INFO_LOG("packet type: %d, stream type: %d, chunk time: %u,"
177 // " chunk_ts: %u, loaded: %u, bad: %d\n", packet_type,
178 // stream_type, chunk_time, pord->chunk_ts, rn->loaded, pord->c_bad);
179 switch (packet_type
) {
180 unsigned header_len
, payload_len
;
183 t
->ret
= -E_ORTP_RECV_EOF
;
186 PARA_INFO_LOG("bof (%d)\n", t
->ret
);
187 pord
->have_header
= 1;
190 if (!pord
->have_header
&& stream_type
)
191 /* can't use the data, wait for header */
193 if (t
->ret
+ rn
->loaded
>= CHUNK_SIZE
+ ORTP_AUDIO_HEADER_LEN
) {
197 if (t
->ret
> ORTP_AUDIO_HEADER_LEN
) {
198 memcpy(rn
->buf
+ rn
->loaded
, tmpbuf
+ ORTP_AUDIO_HEADER_LEN
,
199 t
->ret
- ORTP_AUDIO_HEADER_LEN
);
200 rn
->loaded
+= t
->ret
- ORTP_AUDIO_HEADER_LEN
;
204 header_len
= READ_HEADER_LEN(tmpbuf
);
205 PARA_DEBUG_LOG("header packet (%d bytes), header len: %d\n",
207 if (!pord
->have_header
) {
208 pord
->have_header
= 1;
209 memcpy(rn
->buf
, tmpbuf
+ ORTP_AUDIO_HEADER_LEN
,
210 t
->ret
- ORTP_AUDIO_HEADER_LEN
);
211 rn
->loaded
= t
->ret
- ORTP_AUDIO_HEADER_LEN
;
214 if (header_len
+ ORTP_AUDIO_HEADER_LEN
> t
->ret
) {
215 t
->ret
= -E_INVALID_HEADER
;
218 payload_len
= t
->ret
- ORTP_AUDIO_HEADER_LEN
- header_len
;
219 // PARA_INFO_LOG("len: %d header_len: %d, payload_len: %d, loaded: %d\n", ret,
220 // header_len, payload_len, rn->loaded);
221 if (rn
->loaded
+ payload_len
> CHUNK_SIZE
) {
226 memcpy(rn
->buf
+ rn
->loaded
, tmpbuf
227 + (t
->ret
- payload_len
), payload_len
);
228 rn
->loaded
+= payload_len
;
236 pord
->next_chunk
= *now
;
238 compute_next_chunk(chunk_time
, pord
);
245 static void ortp_shutdown(void)
247 // ortp_global_stats_display();
251 static void ortp_recv_close(struct receiver_node
*rn
)
253 struct private_ortp_recv_data
*pord
= rn
->private_data
;
255 rtp_session_destroy(pord
->session
);
256 free(rn
->private_data
);
260 static void *ortp_recv_parse_config(int argc
, char **argv
)
264 struct ortp_recv_args_info
*tmp
= para_calloc(sizeof(struct ortp_recv_args_info
));
266 ret
= ortp_recv_cmdline_parser(argc
, argv
, tmp
)? -E_ORTP_SYNTAX
: 1;
273 static int ortp_recv_open(struct receiver_node
*rn
)
275 struct private_ortp_recv_data
*pord
;
276 struct ortp_recv_args_info
*conf
= rn
->conf
;
278 rn
->buf
= para_calloc(CHUNK_SIZE
);
280 rn
->private_data
= para_calloc(sizeof(struct private_ortp_recv_data
));
281 pord
= rn
->private_data
;
282 pord
->session
= rtp_session_new(RTP_SESSION_RECVONLY
);
283 PARA_NOTICE_LOG("receiving from %s:%d\n", conf
->host_arg
, conf
->port_arg
);
284 rtp_session_set_local_addr(pord
->session
, conf
->host_arg
, conf
->port_arg
);
285 rtp_session_set_payload_type(pord
->session
, PAYLOAD_AUDIO_CONTINUOUS
);
286 if (conf
->jitter_compensation_arg
) {
287 rtp_session_enable_adaptive_jitter_compensation(pord
->session
, TRUE
);
288 rtp_session_set_jitter_compensation(pord
->session
,
289 conf
->jitter_compensation_arg
);
295 * the init function of the ortp receiver
297 * \param r pointer to the receiver struct to initialize
299 * Initialize all function pointers of \a r and call libortp's ortp_init().
301 void ortp_recv_init(struct receiver
*r
)
303 r
->shutdown
= ortp_shutdown
;
304 r
->open
= ortp_recv_open
;
305 r
->close
= ortp_recv_close
;
306 r
->pre_select
= ortp_recv_pre_select
;
307 r
->post_select
= ortp_recv_post_select
;
308 r
->parse_config
= ortp_recv_parse_config
;