rtpdec: Pass the sequence number to depacketizers
[ffmpeg.git] / libavformat / rtpdec_vp8.c
1 /*
2  * RTP VP8 Depacketizer
3  * Copyright (c) 2010 Josh Allmann
4  * Copyright (c) 2012 Martin Storsjo
5  *
6  * This file is part of Libav.
7  *
8  * Libav is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * Libav is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with Libav; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22
23 /**
24  * @file
25  * @brief RTP support for the VP8 payload
26  * @author Josh Allmann <joshua.allmann@gmail.com>
27  * @see http://tools.ietf.org/html/draft-ietf-payload-vp8-05
28  */
29
30 #include "libavcodec/bytestream.h"
31
32 #include "rtpdec_formats.h"
33
34 struct PayloadContext {
35     AVIOContext *data;
36     uint32_t     timestamp;
37 };
38
39 static int vp8_handle_packet(AVFormatContext *ctx, PayloadContext *vp8,
40                              AVStream *st, AVPacket *pkt, uint32_t *timestamp,
41                              const uint8_t *buf, int len, uint16_t seq,
42                              int flags)
43 {
44     int start_partition, end_packet;
45     int extended_bits, part_id;
46     int pictureid_present = 0, tl0picidx_present = 0, tid_present = 0,
47         keyidx_present = 0;
48
49     if (len < 1)
50         return AVERROR_INVALIDDATA;
51
52     extended_bits   = buf[0] & 0x80;
53     start_partition = buf[0] & 0x10;
54     part_id         = buf[0] & 0x0f;
55     end_packet      = flags & RTP_FLAG_MARKER;
56     buf++;
57     len--;
58     if (extended_bits) {
59         if (len < 1)
60             return AVERROR_INVALIDDATA;
61         pictureid_present = buf[0] & 0x80;
62         tl0picidx_present = buf[0] & 0x40;
63         tid_present       = buf[0] & 0x20;
64         keyidx_present    = buf[0] & 0x10;
65         buf++;
66         len--;
67     }
68     if (pictureid_present) {
69         int size;
70         if (len < 1)
71             return AVERROR_INVALIDDATA;
72         size = buf[0] & 0x80 ? 2 : 1;
73         buf += size;
74         len -= size;
75     }
76     if (tl0picidx_present) {
77         // Ignoring temporal level zero index
78         buf++;
79         len--;
80     }
81     if (tid_present || keyidx_present) {
82         // Ignoring temporal layer index, layer sync bit and keyframe index
83         buf++;
84         len--;
85     }
86     if (len < 1)
87         return AVERROR_INVALIDDATA;
88
89     if (start_partition && part_id == 0) {
90         int res;
91         if (vp8->data) {
92             uint8_t *tmp;
93             avio_close_dyn_buf(vp8->data, &tmp);
94             av_free(tmp);
95             vp8->data = NULL;
96         }
97         if ((res = avio_open_dyn_buf(&vp8->data)) < 0)
98             return res;
99         vp8->timestamp = *timestamp;
100     }
101
102     if (!vp8->data || vp8->timestamp != *timestamp) {
103         av_log(ctx, AV_LOG_WARNING,
104                "Received no start marker; dropping frame\n");
105         return AVERROR(EAGAIN);
106     }
107
108     avio_write(vp8->data, buf, len);
109
110     if (end_packet) {
111         int ret = ff_rtp_finalize_packet(pkt, &vp8->data, st->index);
112         if (ret < 0)
113             return ret;
114         return 0;
115     }
116
117     return AVERROR(EAGAIN);
118 }
119
120 static PayloadContext *vp8_new_context(void)
121 {
122     return av_mallocz(sizeof(PayloadContext));
123 }
124
125 static void vp8_free_context(PayloadContext *vp8)
126 {
127     if (vp8->data) {
128         uint8_t *tmp;
129         avio_close_dyn_buf(vp8->data, &tmp);
130         av_free(tmp);
131     }
132     av_free(vp8);
133 }
134
135 RTPDynamicProtocolHandler ff_vp8_dynamic_handler = {
136     .enc_name       = "VP8",
137     .codec_type     = AVMEDIA_TYPE_VIDEO,
138     .codec_id       = AV_CODEC_ID_VP8,
139     .alloc          = vp8_new_context,
140     .free           = vp8_free_context,
141     .parse_packet   = vp8_handle_packet,
142 };