71c8b1068ff3102c6d97b6c6159c1d01861ee94b
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 */
21 #include <ortp/ortp.h>
25 #include "ortp_recv.cmdline.h"
31 #define CHUNK_SIZE 128 * 1024
33 extern gint
msg_to_buf(mblk_t
*, char *, gint
);
36 * data specific to the ortp receiver
38 * \sa receiver receiver_node
40 struct private_ortp_recv_data
{
45 * whether a header was received
47 * A flag indicating whether the ortp receiver already received a packet which
48 * contains the audio file header.
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
55 /** the ortp session this instance of the receiver belongs to */
57 /** the time the first data or header packet was received */
59 /** ETA of the next chunk */
60 struct timeval next_chunk
;
61 /** number of consecutive bad chunks */
63 /** the current timestamp which is passed to the receiving function of libortp */
65 /** \a timestamp increases by this amount */
70 static void recalc_timestamp(struct receiver_node
*rn
)
72 struct timeval now
, tmp
, min_delay
= {0, 5000};
73 struct private_ortp_recv_data
*pord
= rn
->private_data
;
75 gettimeofday(&now
, NULL
);
76 tv_add(&now
, &min_delay
, &pord
->next_chunk
);
77 if (pord
->start
.tv_sec
) {
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
;
86 PARA_DEBUG_LOG("ts=%u, c_bad = %d\n", pord
->timestamp
, pord
->c_bad
);
91 static int ortp_recv_pre_select(struct receiver_node
*rn
,
92 __unused fd_set
*rfds
, __unused fd_set
*wfds
,
93 struct timeval
*timeout
)
95 struct private_ortp_recv_data
*pord
= rn
->private_data
;
96 struct timeval now
, tmp
;
98 gettimeofday(&now
, NULL
);
99 if (tv_diff(&now
, &pord
->next_chunk
, &tmp
) >= 0) {
103 if (tv_diff(timeout
, &tmp
, NULL
) > 0)
105 return -1; /* we did not modify the fd sets */
108 static void compute_next_chunk(struct timeval
*now
, unsigned chunk_time
,
109 struct private_ortp_recv_data
*pord
)
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);
118 tv_add(&chunk_tv
, &pord
->next_chunk
, &tmp
);
119 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
);
127 #define BUF_TO_VAL(buf) (((unsigned)(buf)[1] & 0xff) + 256 * ((unsigned)(buf)[0] & 0xff))
131 static int ortp_recv_post_select(struct receiver_node
*rn
, int select_ret
,
132 __unused fd_set
*rfds
, __unused fd_set
*wfds
)
134 struct private_ortp_recv_data
*pord
= rn
->private_data
;
136 int ret
, packet_type
, stream_type
;
137 char tmpbuf
[CHUNK_SIZE
+ 3];
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
) {
145 if (tv_diff(&now
, &pord
->next_chunk
, &diff
) < 0) {
146 long unsigned diff_ms
= tv2ms(&diff
);
148 PARA_DEBUG_LOG("too early: %lu ms left\n",
154 mp
= rtp_session_recvm_with_ts(pord
->session
, pord
->timestamp
);
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
);
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
)
170 ret
= msg_to_buf(mp
, tmpbuf
, CHUNK_SIZE
);
171 if (ret
< ORTP_AUDIO_HEADER_LEN
) {
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
;
190 PARA_INFO_LOG("bof (%d)\n", ret
);
191 pord
->have_header
= 1;
194 if (!pord
->have_header
&& stream_type
) /* can't use the data, wait for header */
196 if (ret
+ rn
->loaded
>= CHUNK_SIZE
+ ORTP_AUDIO_HEADER_LEN
) {
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
;
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
;
216 if (header_len
+ ORTP_AUDIO_HEADER_LEN
> ret
) {
217 ret
= -E_INVALID_HEADER
;
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
) {
228 memcpy(rn
->buf
+ rn
->loaded
, tmpbuf
229 + (ret
- payload_len
), payload_len
);
230 rn
->loaded
+= payload_len
;
236 compute_next_chunk(&now
, chunk_time
, pord
);
243 static void ortp_shutdown(void)
245 // ortp_global_stats_display();
249 static void ortp_recv_close(struct receiver_node
*rn
)
251 struct private_ortp_recv_data
*pord
= rn
->private_data
;
253 rtp_session_destroy(pord
->session
);
254 free(rn
->private_data
);
258 static void *ortp_recv_parse_config(int argc
, char **argv
)
262 struct gengetopt_args_info
*tmp
= para_calloc(sizeof(struct gengetopt_args_info
));
264 ret
= ortp_recv_cmdline_parser(argc
, argv
, tmp
)? -E_ORTP_SYNTAX
: 1;
271 static int ortp_recv_open(struct receiver_node
*rn
)
273 struct private_ortp_recv_data
*pord
;
274 struct gengetopt_args_info
*conf
= rn
->conf
;
276 rn
->buf
= para_calloc(CHUNK_SIZE
);
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
);
288 * the init function of the ortp receiver
290 * \param r pointer to the receiver struct to initialize
292 * Initialize all function pointers of \a r and call libortp's ortp_init().
294 void ortp_recv_init(struct receiver
*r
)
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
;
304 ortp_set_debug_file("oRTP", NULL
);