]> git.tuebingen.mpg.de Git - paraslash.git/blob - ortp_recv.c
kill unused BUF_TO_VAL macro
[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 "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 int msg_to_buf(mblk_t *, char *, int);
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 uint32_t timestamp;
65 /** \a timestamp increases by this amount */
66 uint32_t chunk_ts;
67 };
68
69 static int ortp_recv_pre_select(struct receiver_node *rn,
70                 __a_unused fd_set *rfds, __a_unused fd_set *wfds,
71                 struct timeval *timeout)
72 {
73         struct private_ortp_recv_data *pord = rn->private_data;
74         struct timeval now, tmp;
75
76         gettimeofday(&now, NULL);
77         if (tv_diff(&now, &pord->next_chunk, &tmp) >= 0) {
78                 tmp.tv_sec = 0;
79                 tmp.tv_usec = 1000;
80         }
81         if (tv_diff(timeout, &tmp, NULL) > 0)
82                 *timeout = tmp;
83         return -1; /* we did not modify the fd sets */
84 }
85
86 static void compute_next_chunk(struct timeval *now, unsigned chunk_time,
87                 struct private_ortp_recv_data *pord)
88 {
89         struct timeval chunk_tv = {0, chunk_time};
90         struct timeval tmp;
91
92         tv_add(&chunk_tv, &pord->next_chunk, &tmp);
93         pord->next_chunk = tmp;
94         pord->timestamp += pord->chunk_ts;
95         PARA_DEBUG_LOG("next chunk (ts = %d) due at %lu:%lu\n",
96                 pord->timestamp, pord->next_chunk.tv_sec,
97                 pord->next_chunk.tv_usec);
98 }
99
100 static int ortp_recv_post_select(struct receiver_node *rn, int select_ret,
101         __a_unused fd_set *rfds, __a_unused fd_set *wfds)
102 {
103         struct private_ortp_recv_data *pord = rn->private_data;
104         mblk_t *mp;
105         int ret, packet_type, stream_type;
106         char tmpbuf[CHUNK_SIZE + 3];
107         struct timeval now;
108         unsigned chunk_time;
109
110         gettimeofday(&now, NULL);
111 //      PARA_DEBUG_LOG("rn: %p, pord: %p, session: %p\n", rn, pord, pord->session);
112         if (pord->start.tv_sec) {
113                 struct timeval diff;
114                 if (tv_diff(&now, &pord->next_chunk, &diff) < 0)
115                                 return 1;
116         }
117         mp = rtp_session_recvm_with_ts(pord->session, pord->timestamp);
118         if (!mp) {
119                 struct timeval min_delay = {0, 100};
120 //              PARA_INFO_LOG("nope, chunk_ts = %d, loaded: %d, bad: %d\n",
121 //                      pord->timestamp, rn->loaded, pord->c_bad);
122                 pord->c_bad++;
123                 if ((pord->c_bad > 5000 && pord->start.tv_sec) || pord->c_bad > 10000)
124                         return -E_TOO_MANY_BAD_CHUNKS;
125                 tv_add(&now, &min_delay, &pord->next_chunk);
126                 return 1;
127         }
128         /* okay, we have a chunk of data */
129         if (!pord->start.tv_sec)
130                 pord->start = now;
131         ret = msg_to_buf(mp, tmpbuf, CHUNK_SIZE);
132 //      PARA_DEBUG_LOG("have it ts = %d, chunk_ts = %d, loaded: %d, "
133 //              "bad: %d, len: %d\n", pord->timestamp, pord->chunk_ts,
134 //              rn->loaded, pord->c_bad, ret);
135         if (ret < ORTP_AUDIO_HEADER_LEN) {
136                 if (ret < 0)
137                         ret = -E_MSG_TO_BUF;
138                 else
139                         ret = 0;
140                 goto err_out;
141         }
142         packet_type = READ_PACKET_TYPE(tmpbuf);
143         stream_type = READ_STREAM_TYPE(tmpbuf);
144         chunk_time = READ_CHUNK_TIME(tmpbuf);
145         pord->chunk_ts = READ_CHUNK_TS(tmpbuf);
146 //      PARA_INFO_LOG("packet type: %d, stream type: %d, chunk time: %u,"
147 //              " chunk_ts: %u, loaded: %u, bad: %d\n", packet_type,
148 //              stream_type, chunk_time, pord->chunk_ts, rn->loaded, pord->c_bad);
149         switch (packet_type) {
150         unsigned header_len, payload_len;
151         case ORTP_EOF:
152                 ret = 0;
153                 goto err_out;
154         case ORTP_BOF:
155                 PARA_INFO_LOG("bof (%d)\n", ret);
156                 pord->have_header = 1;
157                 /* fall through */
158         case ORTP_DATA:
159                 if (!pord->have_header && stream_type)
160                 /* can't use the data, wait for header */
161                         goto success;
162                 if (ret + rn->loaded >= CHUNK_SIZE + ORTP_AUDIO_HEADER_LEN) {
163                         ret = -E_OVERRUN;
164                         goto err_out;
165                 }
166                 if (ret > ORTP_AUDIO_HEADER_LEN) {
167                         memcpy(rn->buf + rn->loaded, tmpbuf + ORTP_AUDIO_HEADER_LEN,
168                                 ret - ORTP_AUDIO_HEADER_LEN);
169                         rn->loaded += ret - ORTP_AUDIO_HEADER_LEN;
170                 }
171                 goto success;
172         case ORTP_HEADER:
173                 header_len = READ_HEADER_LEN(tmpbuf);
174                 PARA_DEBUG_LOG("header packet (%d bytes), header len: %d\n",
175                         ret, header_len);
176                 if (!pord->have_header) {
177                         pord->have_header = 1;
178                         memcpy(rn->buf, tmpbuf + ORTP_AUDIO_HEADER_LEN,
179                                 ret - ORTP_AUDIO_HEADER_LEN);
180                         rn->loaded = ret - ORTP_AUDIO_HEADER_LEN;
181                         goto success;
182                 }
183                 if (header_len + ORTP_AUDIO_HEADER_LEN > ret) {
184                         ret = -E_INVALID_HEADER;
185                         goto err_out;
186                 }
187                 payload_len = ret - ORTP_AUDIO_HEADER_LEN - header_len;
188 //              PARA_INFO_LOG("len: %d header_len: %d, payload_len: %d, loaded: %d\n", ret,
189 //                      header_len, payload_len, rn->loaded);
190                 if (rn->loaded + payload_len > CHUNK_SIZE) {
191                         ret = -E_OVERRUN;
192                         goto err_out;
193                 }
194                 if (payload_len)
195                         memcpy(rn->buf + rn->loaded, tmpbuf
196                                 + (ret - payload_len), payload_len);
197                 rn->loaded += payload_len;
198                 goto success;
199         }
200 success:
201         freemsg(mp);
202         if (pord->c_bad) {
203                 pord->c_bad = 0;
204                 pord->next_chunk = now;
205         }
206         compute_next_chunk(&now, chunk_time, pord);
207         return 1;
208 err_out:
209         freemsg(mp);
210         return ret;
211 }
212
213 static void ortp_shutdown(void)
214 {
215 //      ortp_global_stats_display();
216         ortp_exit();
217 }
218
219 static void ortp_recv_close(struct receiver_node *rn)
220 {
221         struct private_ortp_recv_data *pord = rn->private_data;
222
223         rtp_session_destroy(pord->session);
224         free(rn->private_data);
225         free(rn->buf);
226 }
227
228 static void *ortp_recv_parse_config(int argc, char **argv)
229 {
230         int ret;
231
232         struct ortp_recv_args_info *tmp = para_calloc(sizeof(struct ortp_recv_args_info));
233
234         ret = ortp_recv_cmdline_parser(argc, argv, tmp)? -E_ORTP_SYNTAX : 1;
235         if (ret > 0)
236                 return tmp;
237         free(tmp);
238         return NULL;
239 }
240
241 static int ortp_recv_open(struct receiver_node *rn)
242 {
243         struct private_ortp_recv_data *pord;
244         struct ortp_recv_args_info *conf = rn->conf;
245
246         rn->buf = para_calloc(CHUNK_SIZE);
247
248         rn->private_data = para_calloc(sizeof(struct private_ortp_recv_data));
249         pord = rn->private_data;
250         pord->session = rtp_session_new(RTP_SESSION_RECVONLY);
251         PARA_NOTICE_LOG("receiving from %s:%d\n", conf->host_arg, conf->port_arg);
252         rtp_session_set_local_addr(pord->session, conf->host_arg, conf->port_arg);
253         rtp_session_set_payload_type(pord->session, PAYLOAD_AUDIO_CONTINUOUS);
254         if (conf->jitter_compensation_arg) {
255                 rtp_session_enable_adaptive_jitter_compensation(pord->session, TRUE);
256                 rtp_session_set_jitter_compensation(pord->session,
257                         conf->jitter_compensation_arg);
258         }
259         return 1;
260 }
261
262 /**
263  * the init function of the ortp receiver
264  *
265  * \param r pointer to the receiver struct to initialize
266  *
267  * Initialize all function pointers of \a r and call libortp's ortp_init().
268  */
269 void ortp_recv_init(struct receiver *r)
270 {
271         r->shutdown = ortp_shutdown;
272         r->open = ortp_recv_open;
273         r->close = ortp_recv_close;
274         r->pre_select = ortp_recv_pre_select;
275         r->post_select = ortp_recv_post_select;
276         r->parse_config = ortp_recv_parse_config;
277
278         ortp_init();
279 }