annotate adtsenc.c @ 6400:6928d93146e6 libavformat

asfcrypt: fix unaligned read in ff_asfcrypt_dec()
author mru
date Mon, 23 Aug 2010 15:38:57 +0000
parents 79b5e6fa86c9
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
1 /*
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
2 * ADTS muxer.
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
3 * Copyright (c) 2006 Baptiste Coudurier <baptiste.coudurier@smartjog.com>
2218
a477001ad010 update my email address
mru
parents: 2210
diff changeset
4 * Mans Rullgard <mans@mansr.com>
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
5 *
1358
0899bfe4105c Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 1169
diff changeset
6 * This file is part of FFmpeg.
0899bfe4105c Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 1169
diff changeset
7 *
0899bfe4105c Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 1169
diff changeset
8 * FFmpeg is free software; you can redistribute it and/or
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
9 * modify it under the terms of the GNU Lesser General Public
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
10 * License as published by the Free Software Foundation; either
1358
0899bfe4105c Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 1169
diff changeset
11 * version 2.1 of the License, or (at your option) any later version.
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
12 *
1358
0899bfe4105c Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 1169
diff changeset
13 * FFmpeg is distributed in the hope that it will be useful,
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
16 * Lesser General Public License for more details.
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
17 *
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
18 * You should have received a copy of the GNU Lesser General Public
1358
0899bfe4105c Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 1169
diff changeset
19 * License along with FFmpeg; if not, write to the Free Software
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
21 */
3286
6f61c3b36632 Use full path for #includes from another directory.
diego
parents: 2771
diff changeset
22
4872
304a0ea063f0 Rename bitstream.h to get_bits.h.
stefano
parents: 4864
diff changeset
23 #include "libavcodec/get_bits.h"
4864
7aa7c5853bb6 Split bitstream.h, put the bitstream writer stuff in the new file
stefano
parents: 4666
diff changeset
24 #include "libavcodec/put_bits.h"
5079
cc9c5e6b8ccf Use av_log_missing_feature in the ADTS encoder. (Pseudo reverts r19173).
alexc
parents: 5025
diff changeset
25 #include "libavcodec/avcodec.h"
6205
1327f1008940 Allow remuxing of explicitly signalled AAC files into ADTS.
alexc
parents: 5787
diff changeset
26 #include "libavcodec/mpeg4audio.h"
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
27 #include "avformat.h"
5787
667736f463b3 In mpegts muxer, write adts header if aac bitstream does not contain it
bcoudurier
parents: 5079
diff changeset
28 #include "adts.h"
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
29
5787
667736f463b3 In mpegts muxer, write adts header if aac bitstream does not contain it
bcoudurier
parents: 5079
diff changeset
30 int ff_adts_decode_extradata(AVFormatContext *s, ADTSContext *adts, uint8_t *buf, int size)
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
31 {
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
32 GetBitContext gb;
4934
7e5d070c8f2e Add PCE support to the ADTS muxer.
alexc
parents: 4872
diff changeset
33 PutBitContext pb;
6205
1327f1008940 Allow remuxing of explicitly signalled AAC files into ADTS.
alexc
parents: 5787
diff changeset
34 MPEG4AudioConfig m4ac;
1327f1008940 Allow remuxing of explicitly signalled AAC files into ADTS.
alexc
parents: 5787
diff changeset
35 int off;
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
36
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
37 init_get_bits(&gb, buf, size * 8);
6205
1327f1008940 Allow remuxing of explicitly signalled AAC files into ADTS.
alexc
parents: 5787
diff changeset
38 off = ff_mpeg4audio_get_config(&m4ac, buf, size);
1327f1008940 Allow remuxing of explicitly signalled AAC files into ADTS.
alexc
parents: 5787
diff changeset
39 if (off < 0)
1327f1008940 Allow remuxing of explicitly signalled AAC files into ADTS.
alexc
parents: 5787
diff changeset
40 return off;
1327f1008940 Allow remuxing of explicitly signalled AAC files into ADTS.
alexc
parents: 5787
diff changeset
41 skip_bits_long(&gb, off);
1327f1008940 Allow remuxing of explicitly signalled AAC files into ADTS.
alexc
parents: 5787
diff changeset
42 adts->objecttype = m4ac.object_type - 1;
1327f1008940 Allow remuxing of explicitly signalled AAC files into ADTS.
alexc
parents: 5787
diff changeset
43 adts->sample_rate_index = m4ac.sampling_index;
1327f1008940 Allow remuxing of explicitly signalled AAC files into ADTS.
alexc
parents: 5787
diff changeset
44 adts->channel_conf = m4ac.chan_config;
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
45
4666
bf7e0b55d9ec ADTS: Increased protection against writing illegal/nonsense files.
alexc
parents: 4482
diff changeset
46 if (adts->objecttype > 3U) {
4468
d23db4a449ed Report the illegal audio object type, not it offset by 1.
alexc
parents: 4467
diff changeset
47 av_log(s, AV_LOG_ERROR, "MPEG-4 AOT %d is not allowed in ADTS\n", adts->objecttype+1);
4467
1dae64d3a6ff ADTS Muxer: Refuse to write illegal ADTS files by checking validity of header
superdump
parents: 3424
diff changeset
48 return -1;
1dae64d3a6ff ADTS Muxer: Refuse to write illegal ADTS files by checking validity of header
superdump
parents: 3424
diff changeset
49 }
1dae64d3a6ff ADTS Muxer: Refuse to write illegal ADTS files by checking validity of header
superdump
parents: 3424
diff changeset
50 if (adts->sample_rate_index == 15) {
1dae64d3a6ff ADTS Muxer: Refuse to write illegal ADTS files by checking validity of header
superdump
parents: 3424
diff changeset
51 av_log(s, AV_LOG_ERROR, "Escape sample rate index illegal in ADTS\n");
1dae64d3a6ff ADTS Muxer: Refuse to write illegal ADTS files by checking validity of header
superdump
parents: 3424
diff changeset
52 return -1;
1dae64d3a6ff ADTS Muxer: Refuse to write illegal ADTS files by checking validity of header
superdump
parents: 3424
diff changeset
53 }
4666
bf7e0b55d9ec ADTS: Increased protection against writing illegal/nonsense files.
alexc
parents: 4482
diff changeset
54 if (get_bits(&gb, 1)) {
bf7e0b55d9ec ADTS: Increased protection against writing illegal/nonsense files.
alexc
parents: 4482
diff changeset
55 av_log(s, AV_LOG_ERROR, "960/120 MDCT window is not allowed in ADTS\n");
bf7e0b55d9ec ADTS: Increased protection against writing illegal/nonsense files.
alexc
parents: 4482
diff changeset
56 return -1;
bf7e0b55d9ec ADTS: Increased protection against writing illegal/nonsense files.
alexc
parents: 4482
diff changeset
57 }
bf7e0b55d9ec ADTS: Increased protection against writing illegal/nonsense files.
alexc
parents: 4482
diff changeset
58 if (get_bits(&gb, 1)) {
bf7e0b55d9ec ADTS: Increased protection against writing illegal/nonsense files.
alexc
parents: 4482
diff changeset
59 av_log(s, AV_LOG_ERROR, "Scalable configurations are not allowed in ADTS\n");
bf7e0b55d9ec ADTS: Increased protection against writing illegal/nonsense files.
alexc
parents: 4482
diff changeset
60 return -1;
bf7e0b55d9ec ADTS: Increased protection against writing illegal/nonsense files.
alexc
parents: 4482
diff changeset
61 }
4934
7e5d070c8f2e Add PCE support to the ADTS muxer.
alexc
parents: 4872
diff changeset
62 if (!adts->channel_conf) {
7e5d070c8f2e Add PCE support to the ADTS muxer.
alexc
parents: 4872
diff changeset
63 init_put_bits(&pb, adts->pce_data, MAX_PCE_SIZE);
7e5d070c8f2e Add PCE support to the ADTS muxer.
alexc
parents: 4872
diff changeset
64
7e5d070c8f2e Add PCE support to the ADTS muxer.
alexc
parents: 4872
diff changeset
65 put_bits(&pb, 3, 5); //ID_PCE
7e5d070c8f2e Add PCE support to the ADTS muxer.
alexc
parents: 4872
diff changeset
66 adts->pce_size = (ff_copy_pce_data(&pb, &gb) + 3) / 8;
7e5d070c8f2e Add PCE support to the ADTS muxer.
alexc
parents: 4872
diff changeset
67 flush_put_bits(&pb);
7e5d070c8f2e Add PCE support to the ADTS muxer.
alexc
parents: 4872
diff changeset
68 }
4467
1dae64d3a6ff ADTS Muxer: Refuse to write illegal ADTS files by checking validity of header
superdump
parents: 3424
diff changeset
69
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
70 adts->write_adts = 1;
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
71
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
72 return 0;
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
73 }
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
74
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
75 static int adts_write_header(AVFormatContext *s)
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
76 {
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
77 ADTSContext *adts = s->priv_data;
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
78 AVCodecContext *avc = s->streams[0]->codec;
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
79
6242
79b5e6fa86c9 Cosmetics: whitespace
alexc
parents: 6205
diff changeset
80 if (avc->extradata_size > 0 &&
5787
667736f463b3 In mpegts muxer, write adts header if aac bitstream does not contain it
bcoudurier
parents: 5079
diff changeset
81 ff_adts_decode_extradata(s, adts, avc->extradata, avc->extradata_size) < 0)
4467
1dae64d3a6ff ADTS Muxer: Refuse to write illegal ADTS files by checking validity of header
superdump
parents: 3424
diff changeset
82 return -1;
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
83
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
84 return 0;
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
85 }
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
86
5787
667736f463b3 In mpegts muxer, write adts header if aac bitstream does not contain it
bcoudurier
parents: 5079
diff changeset
87 int ff_adts_write_frame_header(ADTSContext *ctx,
667736f463b3 In mpegts muxer, write adts header if aac bitstream does not contain it
bcoudurier
parents: 5079
diff changeset
88 uint8_t *buf, int size, int pce_size)
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
89 {
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
90 PutBitContext pb;
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
91
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
92 init_put_bits(&pb, buf, ADTS_HEADER_SIZE);
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
93
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
94 /* adts_fixed_header */
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
95 put_bits(&pb, 12, 0xfff); /* syncword */
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
96 put_bits(&pb, 1, 0); /* ID */
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
97 put_bits(&pb, 2, 0); /* layer */
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
98 put_bits(&pb, 1, 1); /* protection_absent */
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
99 put_bits(&pb, 2, ctx->objecttype); /* profile_objecttype */
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
100 put_bits(&pb, 4, ctx->sample_rate_index);
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
101 put_bits(&pb, 1, 0); /* private_bit */
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
102 put_bits(&pb, 3, ctx->channel_conf); /* channel_configuration */
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
103 put_bits(&pb, 1, 0); /* original_copy */
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
104 put_bits(&pb, 1, 0); /* home */
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
105
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
106 /* adts_variable_header */
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
107 put_bits(&pb, 1, 0); /* copyright_identification_bit */
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
108 put_bits(&pb, 1, 0); /* copyright_identification_start */
5787
667736f463b3 In mpegts muxer, write adts header if aac bitstream does not contain it
bcoudurier
parents: 5079
diff changeset
109 put_bits(&pb, 13, ADTS_HEADER_SIZE + size + pce_size); /* aac_frame_length */
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
110 put_bits(&pb, 11, 0x7ff); /* adts_buffer_fullness */
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
111 put_bits(&pb, 2, 0); /* number_of_raw_data_blocks_in_frame */
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
112
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
113 flush_put_bits(&pb);
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
114
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
115 return 0;
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
116 }
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
117
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
118 static int adts_write_packet(AVFormatContext *s, AVPacket *pkt)
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
119 {
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
120 ADTSContext *adts = s->priv_data;
2771
d52c718e83f9 Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents: 2218
diff changeset
121 ByteIOContext *pb = s->pb;
5787
667736f463b3 In mpegts muxer, write adts header if aac bitstream does not contain it
bcoudurier
parents: 5079
diff changeset
122 uint8_t buf[ADTS_HEADER_SIZE];
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
123
1043
09a392aa8b5a skip 0byte packets
bcoudurier
parents: 1041
diff changeset
124 if (!pkt->size)
09a392aa8b5a skip 0byte packets
bcoudurier
parents: 1041
diff changeset
125 return 0;
6242
79b5e6fa86c9 Cosmetics: whitespace
alexc
parents: 6205
diff changeset
126 if (adts->write_adts) {
5787
667736f463b3 In mpegts muxer, write adts header if aac bitstream does not contain it
bcoudurier
parents: 5079
diff changeset
127 ff_adts_write_frame_header(adts, buf, pkt->size, adts->pce_size);
667736f463b3 In mpegts muxer, write adts header if aac bitstream does not contain it
bcoudurier
parents: 5079
diff changeset
128 put_buffer(pb, buf, ADTS_HEADER_SIZE);
6242
79b5e6fa86c9 Cosmetics: whitespace
alexc
parents: 6205
diff changeset
129 if (adts->pce_size) {
5787
667736f463b3 In mpegts muxer, write adts header if aac bitstream does not contain it
bcoudurier
parents: 5079
diff changeset
130 put_buffer(pb, adts->pce_data, adts->pce_size);
667736f463b3 In mpegts muxer, write adts header if aac bitstream does not contain it
bcoudurier
parents: 5079
diff changeset
131 adts->pce_size = 0;
667736f463b3 In mpegts muxer, write adts header if aac bitstream does not contain it
bcoudurier
parents: 5079
diff changeset
132 }
667736f463b3 In mpegts muxer, write adts header if aac bitstream does not contain it
bcoudurier
parents: 5079
diff changeset
133 }
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
134 put_buffer(pb, pkt->data, pkt->size);
1041
ae2ea8f5a07f flush packets
bcoudurier
parents: 962
diff changeset
135 put_flush_packet(pb);
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
136
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
137 return 0;
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
138 }
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
139
1169
d18cc9a1fd02 allow individual selection of muxers and demuxers
mru
parents: 1167
diff changeset
140 AVOutputFormat adts_muxer = {
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
141 "adts",
3424
7a0230981402 Make long_names in lavf/lavdev optional depending on CONFIG_SMALL.
diego
parents: 3286
diff changeset
142 NULL_IF_CONFIG_SMALL("ADTS AAC"),
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
143 "audio/aac",
5003
e1749e19c60b Add the .adts file extension to the ADTS muxer.
alexc
parents: 4934
diff changeset
144 "aac,adts",
939
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
145 sizeof(ADTSContext),
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
146 CODEC_ID_AAC,
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
147 CODEC_ID_NONE,
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
148 adts_write_header,
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
149 adts_write_packet,
61959072be81 generate adts frame headers if required
mru
parents:
diff changeset
150 };