ortp_recv.c

   1 /*
   2  * Copyright (C) 2005-2007 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 "list.h"
  25 #include "sched.h"
  26 #include "recv.h"
  27 #include "ortp_recv.cmdline.h"
  28
  29 #include "error.h"
  30 #include "audiod.h"
  31 #include "string.h"
  32
  33 #define CHUNK_SIZE 128 * 1024
  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  * whether a header was received
  45  *
  46  * A flag indicating whether the ortp receiver already received a packet which
  47  * contains the audio file header.
  48  *
  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
  51  * header is received.
  52  */
  53 int have_header;
  54 /** the ortp session this instance of the receiver belongs to */
  55 RtpSession *session;
  56 /** the time the first data or header packet was received */
  57 struct timeval start;
  58 /** ETA of the next chunk */
  59 struct timeval next_chunk;
  60 /** number of consecutive bad chunks */
  61 int c_bad;
  62 /** the current timestamp which is passed to the receiving function of libortp */
  63 uint32_t timestamp;
  64 /** \a timestamp increases by this amount */
  65 uint32_t chunk_ts;
  66 };
  67
  68
  69 static int msg_to_buf(mblk_t *mp, char *buffer, int len)
  70 {
  71         mblk_t *m, *mprev;
  72         size_t mlen, rlen = len;
  73
  74         m = mp->b_cont;
  75         mprev = mp;
  76         while (m != NULL) {
  77                 mlen = m->b_wptr - m->b_rptr;
  78                 if (mlen <= rlen) {
  79                         mblk_t *consumed = m;
  80                         memcpy(buffer, m->b_rptr, mlen);
  81                         /* go to next mblk_t */
  82                         mprev->b_cont = m->b_cont;
  83                         m = m->b_cont;
  84                         consumed->b_cont = NULL;
  85                         freeb (consumed);
  86                         buffer += mlen;
  87                         rlen -= mlen;
  88                 } else {
  89                         memcpy (buffer, m->b_rptr, rlen);
  90                         m->b_rptr += rlen;
  91                         return len;
  92                 }
  93         }
  94         return len - rlen;
  95 }
  96
  97
  98 static void ortp_recv_pre_select(struct sched *s, struct task *t)
  99 {
 100         struct receiver_node *rn = t->private_data;
 101         struct private_ortp_recv_data *pord = rn->private_data;
 102         struct timeval tmp;
 103
 104         if (tv_diff(now, &pord->next_chunk, &tmp) >= 0) {
 105                 tmp.tv_sec = 0;
 106                 tmp.tv_usec = 1000;
 107         }
 108         if (tv_diff(&s->timeout, &tmp, NULL) > 0)
 109                 s->timeout = tmp;
 110         t->ret = 1;
 111 }
 112
 113 static void compute_next_chunk(unsigned chunk_time,
 114                 struct private_ortp_recv_data *pord)
 115 {
 116         struct timeval chunk_tv = {0, chunk_time};
 117         struct timeval tmp;
 118
 119         tv_add(&chunk_tv, &pord->next_chunk, &tmp);
 120         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);
 125 }
 126
 127 static void ortp_recv_post_select(__a_unused struct sched *s, struct task *t)
 128 {
 129         struct receiver_node *rn = t->private_data;
 130         struct private_ortp_recv_data *pord = rn->private_data;
 131         mblk_t *mp;
 132         int packet_type, stream_type;
 133         char tmpbuf[CHUNK_SIZE + 3];
 134         unsigned chunk_time;
 135         size_t packet_size;
 136
 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) {
 140                 rn->eof = 1;
 141                 return;
 142         }
 143         t->ret = 1;
 144         if (pord->start.tv_sec)
 145                 if (tv_diff(now, &pord->next_chunk, NULL) < 0)
 146                         return;
 147         mp = rtp_session_recvm_with_ts(pord->session, pord->timestamp);
 148         if (!mp) {
 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);
 152                 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)
 155                         return;
 156                 t->ret = 1;
 157                 tv_add(now, &min_delay, &pord->next_chunk);
 158                 return;
 159         }
 160         /* okay, we have a chunk of data */
 161         if (!pord->start.tv_sec)
 162                 pord->start = *now;
 163         t->ret = msg_to_buf(mp, tmpbuf, CHUNK_SIZE);
 164         if (t->ret < ORTP_AUDIO_HEADER_LEN) {
 165                 rn->eof = 1;
 166                 if (t->ret < 0)
 167                         t->ret = -E_MSG_TO_BUF;
 168                 else
 169                         t->ret = -E_ORTP_RECV_EOF;
 170                 goto err_out;
 171         }
 172         packet_size = t->ret;
 173         packet_type = READ_PACKET_TYPE(tmpbuf);
 174         stream_type = READ_STREAM_TYPE(tmpbuf);
 175         chunk_time = READ_CHUNK_TIME(tmpbuf);
 176         pord->chunk_ts = READ_CHUNK_TS(tmpbuf);
 177 //      PARA_INFO_LOG("packet type: %d, stream type: %d, chunk time: %u,"
 178 //              " chunk_ts: %u, loaded: %u, bad: %d\n", packet_type,
 179 //              stream_type, chunk_time, pord->chunk_ts, rn->loaded, pord->c_bad);
 180         switch (packet_type) {
 181         unsigned header_len, payload_len;
 182         case ORTP_EOF:
 183                 rn->eof = 1;
 184                 t->ret = -E_ORTP_RECV_EOF;
 185                 goto err_out;
 186         case ORTP_BOF:
 187                 PARA_INFO_LOG("bof (%zu)\n", packet_size);
 188                 pord->have_header = 1;
 189                 /* fall through */
 190         case ORTP_DATA:
 191                 if (!pord->have_header && stream_type)
 192                 /* can't use the data, wait for header */
 193                         goto success;
 194                 t->ret = -E_OVERRUN;
 195                 if (packet_size + rn->loaded >= CHUNK_SIZE + ORTP_AUDIO_HEADER_LEN)
 196                         goto err_out;
 197                 if (packet_size > ORTP_AUDIO_HEADER_LEN) {
 198                         memcpy(rn->buf + rn->loaded, tmpbuf + ORTP_AUDIO_HEADER_LEN,
 199                                 packet_size - ORTP_AUDIO_HEADER_LEN);
 200                         rn->loaded += packet_size - ORTP_AUDIO_HEADER_LEN;
 201                 }
 202                 goto success;
 203         case ORTP_HEADER:
 204                 header_len = READ_HEADER_LEN(tmpbuf);
 205                 PARA_DEBUG_LOG("header packet (%zu bytes), header len: %d\n",
 206                         packet_size, header_len);
 207                 if (!pord->have_header) {
 208                         pord->have_header = 1;
 209                         memcpy(rn->buf, tmpbuf + ORTP_AUDIO_HEADER_LEN,
 210                                 packet_size - ORTP_AUDIO_HEADER_LEN);
 211                         rn->loaded = packet_size - ORTP_AUDIO_HEADER_LEN;
 212                         goto success;
 213                 }
 214                 t->ret = -E_INVALID_HEADER;
 215                 if (header_len + ORTP_AUDIO_HEADER_LEN > packet_size)
 216                         goto err_out;
 217                 payload_len = packet_size - ORTP_AUDIO_HEADER_LEN - header_len;
 218                 t->ret = -E_OVERRUN;
 219                 if (rn->loaded + payload_len > CHUNK_SIZE)
 220                         goto err_out;
 221                 if (payload_len)
 222                         memcpy(rn->buf + rn->loaded, tmpbuf
 223                                 + (packet_size - payload_len), payload_len);
 224                 rn->loaded += payload_len;
 225                 goto success;
 226         }
 227 success:
 228         t->ret = 1;
 229         freemsg(mp);
 230         if (pord->c_bad) {
 231                 pord->c_bad = 0;
 232                 pord->next_chunk = *now;
 233         }
 234         compute_next_chunk(chunk_time, pord);
 235         return;
 236 err_out:
 237         rn->eof = 1;
 238         freemsg(mp);
 239 }
 240
 241 static void ortp_shutdown(void)
 242 {
 243 //      ortp_global_stats_display();
 244         ortp_exit();
 245 }
 246
 247 static void ortp_recv_close(struct receiver_node *rn)
 248 {
 249         struct private_ortp_recv_data *pord = rn->private_data;
 250
 251         rtp_session_destroy(pord->session);
 252         free(rn->private_data);
 253         free(rn->buf);
 254 }
 255
 256 static void *ortp_recv_parse_config(int argc, char **argv)
 257 {
 258         int ret;
 259
 260         struct ortp_recv_args_info *tmp =
 261                 para_calloc(sizeof(struct ortp_recv_args_info));
 262
 263         ret = ortp_recv_cmdline_parser(argc, argv, tmp)? -E_ORTP_SYNTAX : 1;
 264         if (ret > 0)
 265                 return tmp;
 266         free(tmp);
 267         return NULL;
 268 }
 269
 270 static int ortp_recv_open(struct receiver_node *rn)
 271 {
 272         struct private_ortp_recv_data *pord;
 273         struct ortp_recv_args_info *c = rn->conf;
 274
 275         rn->buf = para_calloc(CHUNK_SIZE);
 276
 277         rn->private_data = para_calloc(sizeof(struct private_ortp_recv_data));
 278         pord = rn->private_data;
 279         pord->session = rtp_session_new(RTP_SESSION_RECVONLY);
 280         PARA_NOTICE_LOG("receiving from %s:%d\n", c->host_arg, c->port_arg);
 281         rtp_session_set_local_addr(pord->session, c->host_arg, c->port_arg);
 282         rtp_session_set_remote_addr(pord->session, c->host_arg, c->port_arg);
 283         rtp_session_set_payload_type(pord->session, PAYLOAD_AUDIO_CONTINUOUS);
 284         if (c->jitter_compensation_arg) {
 285                 rtp_session_enable_adaptive_jitter_compensation(pord->session, TRUE);
 286                 rtp_session_set_jitter_compensation(pord->session,
 287                         c->jitter_compensation_arg);
 288         }
 289         return 1;
 290 }
 291
 292 /**
 293  * the init function of the ortp receiver
 294  *
 295  * \param r pointer to the receiver struct to initialize
 296  *
 297  * Initialize all function pointers of \a r and call libortp's ortp_init().
 298  */
 299 void ortp_recv_init(struct receiver *r)
 300 {
 301         r->shutdown = ortp_shutdown;
 302         r->open = ortp_recv_open;
 303         r->close = ortp_recv_close;
 304         r->pre_select = ortp_recv_pre_select;
 305         r->post_select = ortp_recv_post_select;
 306         r->parse_config = ortp_recv_parse_config;
 307
 308         ortp_init();
 309 }