rtp: Rename the open/close functions to alloc/free
[ffmpeg.git] / libavformat / rtpdec_mpeg4.c
1 /**
2  * Common code for the RTP depacketization of MPEG-4 formats.
3  * Copyright (c) 2010 Fabrice Bellard
4  *                    Romain Degez
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 MPEG4 / RTP Code
26  * @author Fabrice Bellard
27  * @author Romain Degez
28  */
29
30 #include "rtpdec_formats.h"
31 #include "internal.h"
32 #include "libavutil/avstring.h"
33 #include "libavcodec/get_bits.h"
34 #include <strings.h>
35
36 /** Structure listing useful vars to parse RTP packet payload*/
37 struct PayloadContext
38 {
39     int sizelength;
40     int indexlength;
41     int indexdeltalength;
42     int profile_level_id;
43     int streamtype;
44     int objecttype;
45     char *mode;
46
47     /** mpeg 4 AU headers */
48     struct AUHeaders {
49         int size;
50         int index;
51         int cts_flag;
52         int cts;
53         int dts_flag;
54         int dts;
55         int rap_flag;
56         int streamstate;
57     } *au_headers;
58     int au_headers_allocated;
59     int nb_au_headers;
60     int au_headers_length_bytes;
61     int cur_au_index;
62 };
63
64 typedef struct {
65     const char *str;
66     uint16_t    type;
67     uint32_t    offset;
68 } AttrNameMap;
69
70 /* All known fmtp parameters and the corresponding RTPAttrTypeEnum */
71 #define ATTR_NAME_TYPE_INT 0
72 #define ATTR_NAME_TYPE_STR 1
73 static const AttrNameMap attr_names[]=
74 {
75     { "SizeLength",       ATTR_NAME_TYPE_INT,
76       offsetof(PayloadContext, sizelength) },
77     { "IndexLength",      ATTR_NAME_TYPE_INT,
78       offsetof(PayloadContext, indexlength) },
79     { "IndexDeltaLength", ATTR_NAME_TYPE_INT,
80       offsetof(PayloadContext, indexdeltalength) },
81     { "profile-level-id", ATTR_NAME_TYPE_INT,
82       offsetof(PayloadContext, profile_level_id) },
83     { "StreamType",       ATTR_NAME_TYPE_INT,
84       offsetof(PayloadContext, streamtype) },
85     { "mode",             ATTR_NAME_TYPE_STR,
86       offsetof(PayloadContext, mode) },
87     { NULL, -1, -1 },
88 };
89
90 static PayloadContext *new_context(void)
91 {
92     return av_mallocz(sizeof(PayloadContext));
93 }
94
95 static void free_context(PayloadContext * data)
96 {
97     int i;
98     for (i = 0; i < data->nb_au_headers; i++) {
99          /* according to rtp_parse_mp4_au, we treat multiple
100           * au headers as one, so nb_au_headers is always 1.
101           * loop anyway in case this changes.
102           * (note: changes done carelessly might lead to a double free)
103           */
104        av_free(&data->au_headers[i]);
105     }
106     av_free(data->mode);
107     av_free(data);
108 }
109
110 static int parse_fmtp_config(AVCodecContext * codec, char *value)
111 {
112     /* decode the hexa encoded parameter */
113     int len = ff_hex_to_data(NULL, value);
114     av_free(codec->extradata);
115     codec->extradata = av_mallocz(len + FF_INPUT_BUFFER_PADDING_SIZE);
116     if (!codec->extradata)
117         return AVERROR(ENOMEM);
118     codec->extradata_size = len;
119     ff_hex_to_data(codec->extradata, value);
120     return 0;
121 }
122
123 static int rtp_parse_mp4_au(PayloadContext *data, const uint8_t *buf)
124 {
125     int au_headers_length, au_header_size, i;
126     GetBitContext getbitcontext;
127
128     /* decode the first 2 bytes where the AUHeader sections are stored
129        length in bits */
130     au_headers_length = AV_RB16(buf);
131
132     if (au_headers_length > RTP_MAX_PACKET_LENGTH)
133       return -1;
134
135     data->au_headers_length_bytes = (au_headers_length + 7) / 8;
136
137     /* skip AU headers length section (2 bytes) */
138     buf += 2;
139
140     init_get_bits(&getbitcontext, buf, data->au_headers_length_bytes * 8);
141
142     /* XXX: Wrong if optionnal additional sections are present (cts, dts etc...) */
143     au_header_size = data->sizelength + data->indexlength;
144     if (au_header_size <= 0 || (au_headers_length % au_header_size != 0))
145         return -1;
146
147     data->nb_au_headers = au_headers_length / au_header_size;
148     if (!data->au_headers || data->au_headers_allocated < data->nb_au_headers) {
149         av_free(data->au_headers);
150         data->au_headers = av_malloc(sizeof(struct AUHeaders) * data->nb_au_headers);
151         data->au_headers_allocated = data->nb_au_headers;
152     }
153
154     /* XXX: We handle multiple AU Section as only one (need to fix this for interleaving)
155        In my test, the FAAD decoder does not behave correctly when sending each AU one by one
156        but does when sending the whole as one big packet...  */
157     data->au_headers[0].size = 0;
158     data->au_headers[0].index = 0;
159     for (i = 0; i < data->nb_au_headers; ++i) {
160         data->au_headers[0].size += get_bits_long(&getbitcontext, data->sizelength);
161         data->au_headers[0].index = get_bits_long(&getbitcontext, data->indexlength);
162     }
163
164     data->nb_au_headers = 1;
165
166     return 0;
167 }
168
169
170 /* Follows RFC 3640 */
171 static int aac_parse_packet(AVFormatContext *ctx,
172                             PayloadContext *data,
173                             AVStream *st,
174                             AVPacket *pkt,
175                             uint32_t *timestamp,
176                             const uint8_t *buf, int len, int flags)
177 {
178     if (rtp_parse_mp4_au(data, buf))
179         return -1;
180
181     buf += data->au_headers_length_bytes + 2;
182     len -= data->au_headers_length_bytes + 2;
183
184     /* XXX: Fixme we only handle the case where rtp_parse_mp4_au define
185                     one au_header */
186     av_new_packet(pkt, data->au_headers[0].size);
187     memcpy(pkt->data, buf, data->au_headers[0].size);
188
189     pkt->stream_index = st->index;
190     return 0;
191 }
192
193 static int parse_fmtp(AVStream *stream, PayloadContext *data,
194                       char *attr, char *value)
195 {
196     AVCodecContext *codec = stream->codec;
197     int res, i;
198
199     if (!strcmp(attr, "config")) {
200         res = parse_fmtp_config(codec, value);
201
202         if (res < 0)
203             return res;
204     }
205
206     if (codec->codec_id == CODEC_ID_AAC) {
207         /* Looking for a known attribute */
208         for (i = 0; attr_names[i].str; ++i) {
209             if (!strcasecmp(attr, attr_names[i].str)) {
210                 if (attr_names[i].type == ATTR_NAME_TYPE_INT) {
211                     *(int *)((char *)data+
212                         attr_names[i].offset) = atoi(value);
213                 } else if (attr_names[i].type == ATTR_NAME_TYPE_STR)
214                     *(char **)((char *)data+
215                         attr_names[i].offset) = av_strdup(value);
216             }
217         }
218     }
219     return 0;
220 }
221
222 static int parse_sdp_line(AVFormatContext *s, int st_index,
223                           PayloadContext *data, const char *line)
224 {
225     const char *p;
226
227     if (av_strstart(line, "fmtp:", &p))
228         return ff_parse_fmtp(s->streams[st_index], data, p, parse_fmtp);
229
230     return 0;
231 }
232
233 RTPDynamicProtocolHandler ff_mp4v_es_dynamic_handler = {
234     .enc_name           = "MP4V-ES",
235     .codec_type         = AVMEDIA_TYPE_VIDEO,
236     .codec_id           = CODEC_ID_MPEG4,
237     .parse_sdp_a_line   = parse_sdp_line,
238     .alloc              = NULL,
239     .free               = NULL,
240     .parse_packet       = NULL
241 };
242
243 RTPDynamicProtocolHandler ff_mpeg4_generic_dynamic_handler = {
244     .enc_name           = "mpeg4-generic",
245     .codec_type         = AVMEDIA_TYPE_AUDIO,
246     .codec_id           = CODEC_ID_AAC,
247     .parse_sdp_a_line   = parse_sdp_line,
248     .alloc              = new_context,
249     .free               = free_context,
250     .parse_packet       = aac_parse_packet
251 };