rtpdec.c 25.4 KB
Newer Older
1 2
/*
 * RTP input format
3
 * Copyright (c) 2002 Fabrice Bellard
4
 *
5
 * This file is part of Libav.
6
 *
7
 * Libav is free software; you can redistribute it and/or
8 9 10 11
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
12
 * Libav is distributed in the hope that it will be useful,
13 14 15 16 17
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
18
 * License along with Libav; if not, write to the Free Software
19 20
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */
21

22
#include "libavutil/mathematics.h"
23
#include "libavutil/avstring.h"
24
#include "libavutil/time.h"
25
#include "libavcodec/get_bits.h"
26 27
#include "avformat.h"
#include "mpegts.h"
28
#include "url.h"
29 30 31

#include "network.h"

32
#include "rtpdec.h"
33
#include "rtpdec_formats.h"
34 35 36 37 38 39 40 41 42

//#define DEBUG

/* TODO: - add RTCP statistics reporting (should be optional).

         - add support for h263/mpeg4 packetized output : IDEA: send a
         buffer to 'rtp_write_packet' contains all the packets for ONE
         frame. Each packet should have a four byte header containing
         the length in big endian format (same trick as
43
         'ffio_open_dyn_packet_buf')
44 45
*/

46
static RTPDynamicProtocolHandler ff_realmedia_mp3_dynamic_handler = {
47 48 49 50 51
    .enc_name           = "X-MP3-draft-00",
    .codec_type         = AVMEDIA_TYPE_AUDIO,
    .codec_id           = CODEC_ID_MP3ADU,
};

52
/* statistics functions */
53
static RTPDynamicProtocolHandler *RTPFirstDynamicPayloadHandler= NULL;
54

55
void ff_register_dynamic_payload_handler(RTPDynamicProtocolHandler *handler)
56 57 58 59 60 61 62
{
    handler->next= RTPFirstDynamicPayloadHandler;
    RTPFirstDynamicPayloadHandler= handler;
}

void av_register_rtp_dynamic_payload_handlers(void)
{
63 64
    ff_register_dynamic_payload_handler(&ff_mp4v_es_dynamic_handler);
    ff_register_dynamic_payload_handler(&ff_mpeg4_generic_dynamic_handler);
65 66
    ff_register_dynamic_payload_handler(&ff_amr_nb_dynamic_handler);
    ff_register_dynamic_payload_handler(&ff_amr_wb_dynamic_handler);
67 68
    ff_register_dynamic_payload_handler(&ff_h263_1998_dynamic_handler);
    ff_register_dynamic_payload_handler(&ff_h263_2000_dynamic_handler);
69
    ff_register_dynamic_payload_handler(&ff_h263_rfc2190_dynamic_handler);
70
    ff_register_dynamic_payload_handler(&ff_h264_dynamic_handler);
71
    ff_register_dynamic_payload_handler(&ff_ilbc_dynamic_handler);
72
    ff_register_dynamic_payload_handler(&ff_vorbis_dynamic_handler);
Josh Allmann's avatar
Josh Allmann committed
73
    ff_register_dynamic_payload_handler(&ff_theora_dynamic_handler);
Josh Allmann's avatar
Josh Allmann committed
74
    ff_register_dynamic_payload_handler(&ff_qdm2_dynamic_handler);
Josh Allmann's avatar
Josh Allmann committed
75
    ff_register_dynamic_payload_handler(&ff_svq3_dynamic_handler);
76
    ff_register_dynamic_payload_handler(&ff_mp4a_latm_dynamic_handler);
Josh Allmann's avatar
Josh Allmann committed
77
    ff_register_dynamic_payload_handler(&ff_vp8_dynamic_handler);
78
    ff_register_dynamic_payload_handler(&ff_qcelp_dynamic_handler);
79
    ff_register_dynamic_payload_handler(&ff_realmedia_mp3_dynamic_handler);
80 81 82

    ff_register_dynamic_payload_handler(&ff_ms_rtp_asf_pfv_handler);
    ff_register_dynamic_payload_handler(&ff_ms_rtp_asf_pfa_handler);
83 84 85 86 87

    ff_register_dynamic_payload_handler(&ff_qt_rtp_aud_handler);
    ff_register_dynamic_payload_handler(&ff_qt_rtp_vid_handler);
    ff_register_dynamic_payload_handler(&ff_quicktime_rtp_aud_handler);
    ff_register_dynamic_payload_handler(&ff_quicktime_rtp_vid_handler);
88 89 90 91 92

    ff_register_dynamic_payload_handler(&ff_g726_16_dynamic_handler);
    ff_register_dynamic_payload_handler(&ff_g726_24_dynamic_handler);
    ff_register_dynamic_payload_handler(&ff_g726_32_dynamic_handler);
    ff_register_dynamic_payload_handler(&ff_g726_40_dynamic_handler);
93 94
}

95 96 97 98 99 100
RTPDynamicProtocolHandler *ff_rtp_handler_find_by_name(const char *name,
                                                  enum AVMediaType codec_type)
{
    RTPDynamicProtocolHandler *handler;
    for (handler = RTPFirstDynamicPayloadHandler;
         handler; handler = handler->next)
101
        if (!av_strcasecmp(name, handler->enc_name) &&
102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118
            codec_type == handler->codec_type)
            return handler;
    return NULL;
}

RTPDynamicProtocolHandler *ff_rtp_handler_find_by_id(int id,
                                                enum AVMediaType codec_type)
{
    RTPDynamicProtocolHandler *handler;
    for (handler = RTPFirstDynamicPayloadHandler;
         handler; handler = handler->next)
        if (handler->static_payload_id && handler->static_payload_id == id &&
            codec_type == handler->codec_type)
            return handler;
    return NULL;
}

119 120
static int rtcp_parse_packet(RTPDemuxContext *s, const unsigned char *buf, int len)
{
121
    int payload_len;
122 123 124
    while (len >= 4) {
        payload_len = FFMIN(len, (AV_RB16(buf + 2) + 1) * 4);

125 126
        switch (buf[1]) {
        case RTCP_SR:
127
            if (payload_len < 20) {
128 129 130 131
                av_log(NULL, AV_LOG_ERROR, "Invalid length for RTCP SR packet\n");
                return AVERROR_INVALIDDATA;
            }

Josh Allmann's avatar
Josh Allmann committed
132 133
            s->last_rtcp_ntp_time = AV_RB64(buf + 8);
            s->last_rtcp_timestamp = AV_RB32(buf + 16);
134 135 136 137 138 139
            if (s->first_rtcp_ntp_time == AV_NOPTS_VALUE) {
                s->first_rtcp_ntp_time = s->last_rtcp_ntp_time;
                if (!s->base_timestamp)
                    s->base_timestamp = s->last_rtcp_timestamp;
                s->rtcp_ts_offset = s->last_rtcp_timestamp - s->base_timestamp;
            }
140 141

            break;
142 143
        case RTCP_BYE:
            return -RTCP_BYE;
144
        }
145 146 147

        buf += payload_len;
        len -= payload_len;
148
    }
149
    return -1;
150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227
}

#define RTP_SEQ_MOD (1<<16)

/**
* called on parse open packet
*/
static void rtp_init_statistics(RTPStatistics *s, uint16_t base_sequence) // called on parse open packet.
{
    memset(s, 0, sizeof(RTPStatistics));
    s->max_seq= base_sequence;
    s->probation= 1;
}

/**
* called whenever there is a large jump in sequence numbers, or when they get out of probation...
*/
static void rtp_init_sequence(RTPStatistics *s, uint16_t seq)
{
    s->max_seq= seq;
    s->cycles= 0;
    s->base_seq= seq -1;
    s->bad_seq= RTP_SEQ_MOD + 1;
    s->received= 0;
    s->expected_prior= 0;
    s->received_prior= 0;
    s->jitter= 0;
    s->transit= 0;
}

/**
* returns 1 if we should handle this packet.
*/
static int rtp_valid_packet_in_sequence(RTPStatistics *s, uint16_t seq)
{
    uint16_t udelta= seq - s->max_seq;
    const int MAX_DROPOUT= 3000;
    const int MAX_MISORDER = 100;
    const int MIN_SEQUENTIAL = 2;

    /* source not valid until MIN_SEQUENTIAL packets with sequence seq. numbers have been received */
    if(s->probation)
    {
        if(seq==s->max_seq + 1) {
            s->probation--;
            s->max_seq= seq;
            if(s->probation==0) {
                rtp_init_sequence(s, seq);
                s->received++;
                return 1;
            }
        } else {
            s->probation= MIN_SEQUENTIAL - 1;
            s->max_seq = seq;
        }
    } else if (udelta < MAX_DROPOUT) {
        // in order, with permissible gap
        if(seq < s->max_seq) {
            //sequence number wrapped; count antother 64k cycles
            s->cycles += RTP_SEQ_MOD;
        }
        s->max_seq= seq;
    } else if (udelta <= RTP_SEQ_MOD - MAX_MISORDER) {
        // sequence made a large jump...
        if(seq==s->bad_seq) {
            // two sequential packets-- assume that the other side restarted without telling us; just resync.
            rtp_init_sequence(s, seq);
        } else {
            s->bad_seq= (seq + 1) & (RTP_SEQ_MOD-1);
            return 0;
        }
    } else {
        // duplicate or reordered packet...
    }
    s->received++;
    return 1;
}

228
int ff_rtp_check_and_send_back_rr(RTPDemuxContext *s, int count)
229
{
230
    AVIOContext *pb;
231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256
    uint8_t *buf;
    int len;
    int rtcp_bytes;
    RTPStatistics *stats= &s->statistics;
    uint32_t lost;
    uint32_t extended_max;
    uint32_t expected_interval;
    uint32_t received_interval;
    uint32_t lost_interval;
    uint32_t expected;
    uint32_t fraction;
    uint64_t ntp_time= s->last_rtcp_ntp_time; // TODO: Get local ntp time?

    if (!s->rtp_ctx || (count < 1))
        return -1;

    /* TODO: I think this is way too often; RFC 1889 has algorithm for this */
    /* XXX: mpeg pts hardcoded. RTCP send every 0.5 seconds */
    s->octet_count += count;
    rtcp_bytes = ((s->octet_count - s->last_octet_count) * RTCP_TX_RATIO_NUM) /
        RTCP_TX_RATIO_DEN;
    rtcp_bytes /= 50; // mmu_man: that's enough for me... VLC sends much less btw !?
    if (rtcp_bytes < 28)
        return -1;
    s->last_octet_count = s->octet_count;

257
    if (avio_open_dyn_buf(&pb) < 0)
258 259 260
        return -1;

    // Receiver Report
261 262 263
    avio_w8(pb, (RTP_VERSION << 6) + 1); /* 1 report block */
    avio_w8(pb, RTCP_RR);
    avio_wb16(pb, 7); /* length in words - 1 */
264
    // our own SSRC: we use the server's SSRC + 1 to avoid conflicts
265 266
    avio_wb32(pb, s->ssrc + 1);
    avio_wb32(pb, s->ssrc); // server SSRC
267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282
    // some placeholders we should really fill...
    // RFC 1889/p64
    extended_max= stats->cycles + stats->max_seq;
    expected= extended_max - stats->base_seq + 1;
    lost= expected - stats->received;
    lost= FFMIN(lost, 0xffffff); // clamp it since it's only 24 bits...
    expected_interval= expected - stats->expected_prior;
    stats->expected_prior= expected;
    received_interval= stats->received - stats->received_prior;
    stats->received_prior= stats->received;
    lost_interval= expected_interval - received_interval;
    if (expected_interval==0 || lost_interval<=0) fraction= 0;
    else fraction = (lost_interval<<8)/expected_interval;

    fraction= (fraction<<24) | lost;

283 284 285
    avio_wb32(pb, fraction); /* 8 bits of fraction, 24 bits of total packets lost */
    avio_wb32(pb, extended_max); /* max sequence received */
    avio_wb32(pb, stats->jitter>>4); /* jitter */
286 287 288

    if(s->last_rtcp_ntp_time==AV_NOPTS_VALUE)
    {
289 290
        avio_wb32(pb, 0); /* last SR timestamp */
        avio_wb32(pb, 0); /* delay since last SR */
291 292 293 294
    } else {
        uint32_t middle_32_bits= s->last_rtcp_ntp_time>>16; // this is valid, right? do we need to handle 64 bit values special?
        uint32_t delay_since_last= ntp_time - s->last_rtcp_ntp_time;

295 296
        avio_wb32(pb, middle_32_bits); /* last SR timestamp */
        avio_wb32(pb, delay_since_last); /* delay since last SR */
297 298 299
    }

    // CNAME
300 301
    avio_w8(pb, (RTP_VERSION << 6) + 1); /* 1 report block */
    avio_w8(pb, RTCP_SDES);
302
    len = strlen(s->hostname);
303
    avio_wb16(pb, (6 + len + 3) / 4); /* length in words - 1 */
304
    avio_wb32(pb, s->ssrc + 1);
305 306 307
    avio_w8(pb, 0x01);
    avio_w8(pb, len);
    avio_write(pb, s->hostname, len);
308 309
    // padding
    for (len = (6 + len) % 4; len % 4; len++) {
310
        avio_w8(pb, 0);
311 312
    }

313
    avio_flush(pb);
314
    len = avio_close_dyn_buf(pb, &buf);
315
    if ((len > 0) && buf) {
316
        int av_unused result;
Luca Barbato's avatar
Luca Barbato committed
317
        av_dlog(s->ic, "sending %d bytes of RR\n", len);
318 319
        result= ffurl_write(s->rtp_ctx, buf, len);
        av_dlog(s->ic, "result from ffurl_write: %d\n", result);
320 321 322 323 324
        av_free(buf);
    }
    return 0;
}

325
void ff_rtp_send_punch_packets(URLContext* rtp_handle)
326
{
327
    AVIOContext *pb;
328 329 330 331
    uint8_t *buf;
    int len;

    /* Send a small RTP packet */
332
    if (avio_open_dyn_buf(&pb) < 0)
333 334
        return;

335 336 337 338 339
    avio_w8(pb, (RTP_VERSION << 6));
    avio_w8(pb, 0); /* Payload type */
    avio_wb16(pb, 0); /* Seq */
    avio_wb32(pb, 0); /* Timestamp */
    avio_wb32(pb, 0); /* SSRC */
340

341
    avio_flush(pb);
342
    len = avio_close_dyn_buf(pb, &buf);
343
    if ((len > 0) && buf)
344
        ffurl_write(rtp_handle, buf, len);
345 346 347
    av_free(buf);

    /* Send a minimal RTCP RR */
348
    if (avio_open_dyn_buf(&pb) < 0)
349 350
        return;

351 352 353 354
    avio_w8(pb, (RTP_VERSION << 6));
    avio_w8(pb, RTCP_RR); /* receiver report */
    avio_wb16(pb, 1); /* length in words - 1 */
    avio_wb32(pb, 0); /* our own SSRC */
355

356
    avio_flush(pb);
357
    len = avio_close_dyn_buf(pb, &buf);
358
    if ((len > 0) && buf)
359
        ffurl_write(rtp_handle, buf, len);
360 361 362 363
    av_free(buf);
}


364 365 366 367 368
/**
 * open a new RTP parse context for stream 'st'. 'st' can be NULL for
 * MPEG2TS streams to indicate that they should be demuxed inside the
 * rtp demux (otherwise CODEC_ID_MPEG2TS packets are returned)
 */
369
RTPDemuxContext *ff_rtp_parse_open(AVFormatContext *s1, AVStream *st, URLContext *rtpc, int payload_type, int queue_size)
370 371 372 373 374 375 376 377
{
    RTPDemuxContext *s;

    s = av_mallocz(sizeof(RTPDemuxContext));
    if (!s)
        return NULL;
    s->payload_type = payload_type;
    s->last_rtcp_ntp_time = AV_NOPTS_VALUE;
378
    s->first_rtcp_ntp_time = AV_NOPTS_VALUE;
379 380
    s->ic = s1;
    s->st = st;
381
    s->queue_size = queue_size;
382 383
    rtp_init_statistics(&s->statistics, 0); // do we know the initial sequence from sdp?
    if (!strcmp(ff_rtp_enc_name(payload_type), "MP2T")) {
384
        s->ts = ff_mpegts_parse_open(s->ic);
385 386 387 388
        if (s->ts == NULL) {
            av_free(s);
            return NULL;
        }
389
    } else if (st) {
390 391 392 393 394 395
        switch(st->codec->codec_id) {
        case CODEC_ID_MPEG1VIDEO:
        case CODEC_ID_MPEG2VIDEO:
        case CODEC_ID_MP2:
        case CODEC_ID_MP3:
        case CODEC_ID_MPEG4:
396
        case CODEC_ID_H263:
397 398 399
        case CODEC_ID_H264:
            st->need_parsing = AVSTREAM_PARSE_FULL;
            break;
400 401 402
        case CODEC_ID_VORBIS:
            st->need_parsing = AVSTREAM_PARSE_HEADERS;
            break;
403 404 405 406 407 408
        case CODEC_ID_ADPCM_G722:
            /* According to RFC 3551, the stream clock rate is 8000
             * even if the sample rate is 16000. */
            if (st->codec->sample_rate == 8000)
                st->codec->sample_rate = 16000;
            break;
409 410 411 412 413 414 415 416 417 418
        default:
            break;
        }
    }
    // needed to send back RTCP RR in RTSP sessions
    s->rtp_ctx = rtpc;
    gethostname(s->hostname, sizeof(s->hostname));
    return s;
}

419
void
420 421
ff_rtp_parse_set_dynamic_protocol(RTPDemuxContext *s, PayloadContext *ctx,
                                  RTPDynamicProtocolHandler *handler)
422 423 424 425 426
{
    s->dynamic_protocol_context = ctx;
    s->parse_packet = handler->parse_packet;
}

427 428 429 430 431
/**
 * This was the second switch in rtp_parse packet.  Normalizes time, if required, sets stream_index, etc.
 */
static void finalize_packet(RTPDemuxContext *s, AVPacket *pkt, uint32_t timestamp)
{
432 433
    if (pkt->pts != AV_NOPTS_VALUE || pkt->dts != AV_NOPTS_VALUE)
        return; /* Timestamp already set by depacketizer */
John Brooks's avatar
John Brooks committed
434 435 436
    if (timestamp == RTP_NOTS_VALUE)
        return;

437
    if (s->last_rtcp_ntp_time != AV_NOPTS_VALUE && s->ic->nb_streams > 1) {
Luca Abeni's avatar
Luca Abeni committed
438 439 440 441 442 443
        int64_t addend;
        int delta_timestamp;

        /* compute pts from timestamp with received ntp_time */
        delta_timestamp = timestamp - s->last_rtcp_timestamp;
        /* convert to the PTS timebase */
444
        addend = av_rescale(s->last_rtcp_ntp_time - s->first_rtcp_ntp_time, s->st->time_base.den, (uint64_t)s->st->time_base.num << 32);
445 446 447
        pkt->pts = s->range_start_offset + s->rtcp_ts_offset + addend +
                   delta_timestamp;
        return;
Luca Abeni's avatar
Luca Abeni committed
448
    }
John Brooks's avatar
John Brooks committed
449

450 451
    if (!s->base_timestamp)
        s->base_timestamp = timestamp;
452 453 454 455 456 457 458
    /* assume that the difference is INT32_MIN < x < INT32_MAX, but allow the first timestamp to exceed INT32_MAX */
    if (!s->timestamp)
        s->unwrapped_timestamp += timestamp;
    else
        s->unwrapped_timestamp += (int32_t)(timestamp - s->timestamp);
    s->timestamp = timestamp;
    pkt->pts = s->unwrapped_timestamp + s->range_start_offset - s->base_timestamp;
459 460
}

461 462
static int rtp_parse_packet_internal(RTPDemuxContext *s, AVPacket *pkt,
                                     const uint8_t *buf, int len)
463 464
{
    unsigned int ssrc, h;
465
    int payload_type, seq, ret, flags = 0;
466
    int ext;
467 468 469 470
    AVStream *st;
    uint32_t timestamp;
    int rv= 0;

471
    ext = buf[0] & 0x10;
472
    payload_type = buf[1] & 0x7f;
473 474
    if (buf[1] & 0x80)
        flags |= RTP_FLAG_MARKER;
475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493
    seq  = AV_RB16(buf + 2);
    timestamp = AV_RB32(buf + 4);
    ssrc = AV_RB32(buf + 8);
    /* store the ssrc in the RTPDemuxContext */
    s->ssrc = ssrc;

    /* NOTE: we can handle only one payload type */
    if (s->payload_type != payload_type)
        return -1;

    st = s->st;
    // only do something with this if all the rtp checks pass...
    if(!rtp_valid_packet_in_sequence(&s->statistics, seq))
    {
        av_log(st?st->codec:NULL, AV_LOG_ERROR, "RTP: PT=%02x: bad cseq %04x expected=%04x\n",
               payload_type, seq, ((s->seq + 1) & 0xffff));
        return -1;
    }

494 495 496 497 498 499
    if (buf[0] & 0x20) {
        int padding = buf[len - 1];
        if (len >= 12 + padding)
            len -= padding;
    }

500 501 502 503
    s->seq = seq;
    len -= 12;
    buf += 12;

504 505 506 507 508 509 510 511 512 513 514 515 516 517 518
    /* RFC 3550 Section 5.3.1 RTP Header Extension handling */
    if (ext) {
        if (len < 4)
            return -1;
        /* calculate the header extension length (stored as number
         * of 32-bit words) */
        ext = (AV_RB16(buf + 2) + 1) << 2;

        if (len < ext)
            return -1;
        // skip past RTP header extension
        len -= ext;
        buf += ext;
    }

519 520
    if (!st) {
        /* specific MPEG2TS demux support */
521
        ret = ff_mpegts_parse_packet(s->ts, pkt, buf, len);
522 523 524
        /* The only error that can be returned from ff_mpegts_parse_packet
         * is "no more data to return from the provided buffer", so return
         * AVERROR(EAGAIN) for all errors */
525
        if (ret < 0)
526
            return AVERROR(EAGAIN);
527 528 529 530 531 532
        if (ret < len) {
            s->read_buf_size = len - ret;
            memcpy(s->buf, buf + ret, s->read_buf_size);
            s->read_buf_index = 0;
            return 1;
        }
533
        return 0;
534
    } else if (s->parse_packet) {
535
        rv = s->parse_packet(s->ic, s->dynamic_protocol_context,
536
                             s->st, pkt, &timestamp, buf, len, flags);
537 538 539 540
    } else {
        // at this point, the RTP header has been stripped;  This is ASSUMING that there is only 1 CSRC, which in't wise.
        switch(st->codec->codec_id) {
        case CODEC_ID_MP2:
541
        case CODEC_ID_MP3:
542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569
            /* better than nothing: skip mpeg audio RTP header */
            if (len <= 4)
                return -1;
            h = AV_RB32(buf);
            len -= 4;
            buf += 4;
            av_new_packet(pkt, len);
            memcpy(pkt->data, buf, len);
            break;
        case CODEC_ID_MPEG1VIDEO:
        case CODEC_ID_MPEG2VIDEO:
            /* better than nothing: skip mpeg video RTP header */
            if (len <= 4)
                return -1;
            h = AV_RB32(buf);
            buf += 4;
            len -= 4;
            if (h & (1 << 26)) {
                /* mpeg2 */
                if (len <= 4)
                    return -1;
                buf += 4;
                len -= 4;
            }
            av_new_packet(pkt, len);
            memcpy(pkt->data, buf, len);
            break;
        default:
570 571
            av_new_packet(pkt, len);
            memcpy(pkt->data, buf, len);
572 573
            break;
        }
574 575

        pkt->stream_index = st->index;
576
    }
577

Ronald S. Bultje's avatar
Ronald S. Bultje committed
578 579
    // now perform timestamp things....
    finalize_packet(s, pkt, timestamp);
580

581 582 583
    return rv;
}

584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627
void ff_rtp_reset_packet_queue(RTPDemuxContext *s)
{
    while (s->queue) {
        RTPPacket *next = s->queue->next;
        av_free(s->queue->buf);
        av_free(s->queue);
        s->queue = next;
    }
    s->seq       = 0;
    s->queue_len = 0;
    s->prev_ret  = 0;
}

static void enqueue_packet(RTPDemuxContext *s, uint8_t *buf, int len)
{
    uint16_t seq = AV_RB16(buf + 2);
    RTPPacket *cur = s->queue, *prev = NULL, *packet;

    /* Find the correct place in the queue to insert the packet */
    while (cur) {
        int16_t diff = seq - cur->seq;
        if (diff < 0)
            break;
        prev = cur;
        cur = cur->next;
    }

    packet = av_mallocz(sizeof(*packet));
    if (!packet)
        return;
    packet->recvtime = av_gettime();
    packet->seq = seq;
    packet->len = len;
    packet->buf = buf;
    packet->next = cur;
    if (prev)
        prev->next = packet;
    else
        s->queue = packet;
    s->queue_len++;
}

static int has_next_packet(RTPDemuxContext *s)
{
628
    return s->queue && s->queue->seq == (uint16_t) (s->seq + 1);
629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654
}

int64_t ff_rtp_queued_packet_time(RTPDemuxContext *s)
{
    return s->queue ? s->queue->recvtime : 0;
}

static int rtp_parse_queued_packet(RTPDemuxContext *s, AVPacket *pkt)
{
    int rv;
    RTPPacket *next;

    if (s->queue_len <= 0)
        return -1;

    if (!has_next_packet(s))
        av_log(s->st ? s->st->codec : NULL, AV_LOG_WARNING,
               "RTP: missed %d packets\n", s->queue->seq - s->seq - 1);

    /* Parse the first packet in the queue, and dequeue it */
    rv = rtp_parse_packet_internal(s, pkt, s->queue->buf, s->queue->len);
    next = s->queue->next;
    av_free(s->queue->buf);
    av_free(s->queue);
    s->queue = next;
    s->queue_len--;
655
    return rv;
656 657
}

658
static int rtp_parse_one_packet(RTPDemuxContext *s, AVPacket *pkt,
659 660 661 662 663 664 665 666
                     uint8_t **bufptr, int len)
{
    uint8_t* buf = bufptr ? *bufptr : NULL;
    int ret, flags = 0;
    uint32_t timestamp;
    int rv= 0;

    if (!buf) {
667 668
        /* If parsing of the previous packet actually returned 0 or an error,
         * there's nothing more to be parsed from that packet, but we may have
669
         * indicated that we can return the next enqueued packet. */
670
        if (s->prev_ret <= 0)
671
            return rtp_parse_queued_packet(s, pkt);
672 673 674 675 676 677 678 679
        /* return the next packets, if any */
        if(s->st && s->parse_packet) {
            /* timestamp should be overwritten by parse_packet, if not,
             * the packet is left with pts == AV_NOPTS_VALUE */
            timestamp = RTP_NOTS_VALUE;
            rv= s->parse_packet(s->ic, s->dynamic_protocol_context,
                                s->st, pkt, &timestamp, NULL, 0, flags);
            finalize_packet(s, pkt, timestamp);
680
            return rv;
681 682
        } else {
            // TODO: Move to a dynamic packet handler (like above)
683
            if (s->read_buf_index >= s->read_buf_size)
684
                return AVERROR(EAGAIN);
685 686
            ret = ff_mpegts_parse_packet(s->ts, pkt, s->buf + s->read_buf_index,
                                      s->read_buf_size - s->read_buf_index);
687
            if (ret < 0)
688
                return AVERROR(EAGAIN);
689 690 691
            s->read_buf_index += ret;
            if (s->read_buf_index < s->read_buf_size)
                return 1;
692 693
            else
                return 0;
694 695 696 697 698 699 700 701
        }
    }

    if (len < 12)
        return -1;

    if ((buf[0] & 0xc0) != (RTP_VERSION << 6))
        return -1;
702
    if (RTP_PT_IS_RTCP(buf[1])) {
703 704 705
        return rtcp_parse_packet(s, buf, len);
    }

706
    if ((s->seq == 0 && !s->queue) || s->queue_size <= 1) {
707 708 709 710 711 712 713 714 715 716 717 718 719
        /* First packet, or no reordering */
        return rtp_parse_packet_internal(s, pkt, buf, len);
    } else {
        uint16_t seq = AV_RB16(buf + 2);
        int16_t diff = seq - s->seq;
        if (diff < 0) {
            /* Packet older than the previously emitted one, drop */
            av_log(s->st ? s->st->codec : NULL, AV_LOG_WARNING,
                   "RTP: dropping old packet received too late\n");
            return -1;
        } else if (diff <= 1) {
            /* Correct packet */
            rv = rtp_parse_packet_internal(s, pkt, buf, len);
720
            return rv;
721 722 723 724 725 726 727 728 729 730 731
        } else {
            /* Still missing some packet, enqueue this one. */
            enqueue_packet(s, buf, len);
            *bufptr = NULL;
            /* Return the first enqueued packet if the queue is full,
             * even if we're missing something */
            if (s->queue_len >= s->queue_size)
                return rtp_parse_queued_packet(s, pkt);
            return -1;
        }
    }
732 733
}

734 735 736 737 738 739 740 741 742
/**
 * Parse an RTP or RTCP packet directly sent as a buffer.
 * @param s RTP parse context.
 * @param pkt returned packet
 * @param bufptr pointer to the input buffer or NULL to read the next packets
 * @param len buffer len
 * @return 0 if a packet is returned, 1 if a packet is returned and more can follow
 * (use buf as NULL to read the next). -1 if no packet (error or no more packet).
 */
743 744
int ff_rtp_parse_packet(RTPDemuxContext *s, AVPacket *pkt,
                        uint8_t **bufptr, int len)
745 746 747
{
    int rv = rtp_parse_one_packet(s, pkt, bufptr, len);
    s->prev_ret = rv;
748 749
    while (rv == AVERROR(EAGAIN) && has_next_packet(s))
        rv = rtp_parse_queued_packet(s, pkt);
750 751 752
    return rv ? rv : has_next_packet(s);
}

753
void ff_rtp_parse_close(RTPDemuxContext *s)
754
{
755
    ff_rtp_reset_packet_queue(s);
756
    if (!strcmp(ff_rtp_enc_name(s->payload_type), "MP2T")) {
757
        ff_mpegts_parse_close(s->ts);
758 759 760
    }
    av_free(s);
}
761 762 763 764 765 766 767

int ff_parse_fmtp(AVStream *stream, PayloadContext *data, const char *p,
                  int (*parse_fmtp)(AVStream *stream,
                                    PayloadContext *data,
                                    char *attr, char *value))
{
    char attr[256];
768
    char *value;
769
    int res;
770 771 772 773 774 775
    int value_size = strlen(p) + 1;

    if (!(value = av_malloc(value_size))) {
        av_log(stream, AV_LOG_ERROR, "Failed to allocate data for FMTP.");
        return AVERROR(ENOMEM);
    }
776 777 778 779 780 781 782 783

    // remove protocol identifier
    while (*p && *p == ' ') p++; // strip spaces
    while (*p && *p != ' ') p++; // eat protocol identifier
    while (*p && *p == ' ') p++; // strip trailing spaces

    while (ff_rtsp_next_attr_and_value(&p,
                                       attr, sizeof(attr),
784
                                       value, value_size)) {
785 786

        res = parse_fmtp(stream, data, attr, value);
787 788
        if (res < 0 && res != AVERROR_PATCHWELCOME) {
            av_free(value);
789
            return res;
790
        }
791
    }
792
    av_free(value);
793 794
    return 0;
}