Mercurial > libavcodec.hg
annotate ac3.h @ 4733:507d08212e36 libavcodec
check input validity, this prevents a few variables from reachin odd values which might have lead to out of array writes and thus might have been exploitable
author | michael |
---|---|
date | Sun, 25 Mar 2007 23:37:38 +0000 |
parents | 6ec0afffc572 |
children | 40f3a7f2b1fd |
rev | line source |
---|---|
782 | 1 /* |
2 * Common code between AC3 encoder and decoder | |
3 * Copyright (c) 2000, 2001, 2002 Fabrice Bellard. | |
4 * | |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
5 * This file is part of FFmpeg. |
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
6 * |
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
7 * FFmpeg is free software; you can redistribute it and/or |
782 | 8 * modify it under the terms of the GNU Lesser General Public |
9 * License as published by the Free Software Foundation; either | |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
10 * version 2.1 of the License, or (at your option) any later version. |
782 | 11 * |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
12 * FFmpeg is distributed in the hope that it will be useful, |
782 | 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
15 * Lesser General Public License for more details. | |
16 * | |
17 * You should have received a copy of the GNU Lesser General Public | |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
18 * License along with FFmpeg; if not, write to the Free Software |
3036
0b546eab515d
Update licensing information: The FSF changed postal address.
diego
parents:
2967
diff
changeset
|
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
782 | 20 */ |
21 | |
1106 | 22 /** |
23 * @file ac3.h | |
24 * Common code between AC3 encoder and decoder. | |
25 */ | |
26 | |
782 | 27 #define AC3_MAX_CODED_FRAME_SIZE 3840 /* in bytes */ |
28 #define AC3_MAX_CHANNELS 6 /* including LFE channel */ | |
29 | |
30 #define NB_BLOCKS 6 /* number of PCM blocks inside an AC3 frame */ | |
31 #define AC3_FRAME_SIZE (NB_BLOCKS * 256) | |
32 | |
33 /* exponent encoding strategy */ | |
34 #define EXP_REUSE 0 | |
35 #define EXP_NEW 1 | |
36 | |
37 #define EXP_D15 1 | |
38 #define EXP_D25 2 | |
39 #define EXP_D45 3 | |
40 | |
41 typedef struct AC3BitAllocParameters { | |
42 int fscod; /* frequency */ | |
43 int halfratecod; | |
44 int sgain, sdecay, fdecay, dbknee, floor; | |
45 int cplfleak, cplsleak; | |
46 } AC3BitAllocParameters; | |
47 | |
4648 | 48 /** |
49 * @struct AC3HeaderInfo | |
50 * Coded AC-3 header values up to the lfeon element, plus derived values. | |
51 */ | |
52 typedef struct { | |
53 /** @defgroup coded Coded elements | |
54 * @{ | |
55 */ | |
56 uint16_t sync_word; | |
57 uint16_t crc1; | |
58 uint8_t fscod; | |
59 uint8_t frmsizecod; | |
60 uint8_t bsid; | |
61 uint8_t bsmod; | |
62 uint8_t acmod; | |
63 uint8_t cmixlev; | |
64 uint8_t surmixlev; | |
65 uint8_t dsurmod; | |
66 uint8_t lfeon; | |
67 /** @} */ | |
68 | |
69 /** @defgroup derived Derived values | |
70 * @{ | |
71 */ | |
72 uint8_t halfratecod; | |
73 uint16_t sample_rate; | |
74 uint32_t bit_rate; | |
75 uint8_t channels; | |
76 uint16_t frame_size; | |
77 /** @} */ | |
78 } AC3HeaderInfo; | |
79 | |
80 /** | |
81 * Parses AC-3 frame header. | |
82 * Parses the header up to the lfeon element, which is the first 52 or 54 bits | |
83 * depending on the audio coding mode. | |
84 * @param buf[in] Array containing the first 7 bytes of the frame. | |
85 * @param hdr[out] Pointer to struct where header info is written. | |
86 * @return Returns 0 on success, -1 if there is a sync word mismatch, | |
87 * -2 if the bsid (version) element is invalid, -3 if the fscod (sample rate) | |
88 * element is invalid, or -4 if the frmsizecod (bit rate) element is invalid. | |
89 */ | |
90 int ff_ac3_parse_header(const uint8_t buf[7], AC3HeaderInfo *hdr); | |
91 | |
4679 | 92 extern const uint16_t ff_ac3_frame_sizes[38][3]; |
4645
056127e5df89
remove redundancy in AC-3 parser by using common tables from ac3tab.h
jbr
parents:
4643
diff
changeset
|
93 extern const uint8_t ff_ac3_channels[8]; |
4643
1e175640dad3
Remove common code from AC-3 encoder and utilize ac3.c.
jbr
parents:
4197
diff
changeset
|
94 extern const uint16_t ff_ac3_freqs[3]; |
1e175640dad3
Remove common code from AC-3 encoder and utilize ac3.c.
jbr
parents:
4197
diff
changeset
|
95 extern const uint16_t ff_ac3_bitratetab[19]; |
1e175640dad3
Remove common code from AC-3 encoder and utilize ac3.c.
jbr
parents:
4197
diff
changeset
|
96 extern const int16_t ff_ac3_window[256]; |
1e175640dad3
Remove common code from AC-3 encoder and utilize ac3.c.
jbr
parents:
4197
diff
changeset
|
97 extern const uint8_t ff_sdecaytab[4]; |
1e175640dad3
Remove common code from AC-3 encoder and utilize ac3.c.
jbr
parents:
4197
diff
changeset
|
98 extern const uint8_t ff_fdecaytab[4]; |
1e175640dad3
Remove common code from AC-3 encoder and utilize ac3.c.
jbr
parents:
4197
diff
changeset
|
99 extern const uint16_t ff_sgaintab[4]; |
1e175640dad3
Remove common code from AC-3 encoder and utilize ac3.c.
jbr
parents:
4197
diff
changeset
|
100 extern const uint16_t ff_dbkneetab[4]; |
1e175640dad3
Remove common code from AC-3 encoder and utilize ac3.c.
jbr
parents:
4197
diff
changeset
|
101 extern const int16_t ff_floortab[8]; |
1e175640dad3
Remove common code from AC-3 encoder and utilize ac3.c.
jbr
parents:
4197
diff
changeset
|
102 extern const uint16_t ff_fgaintab[8]; |
782 | 103 |
4197 | 104 void ac3_common_init(void); |
4684
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
105 |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
106 /** |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
107 * Calculates the log power-spectral density of the input signal. |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
108 * This gives a rough estimate of signal power in the frequency domain by using |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
109 * the spectral envelope (exponents). The psd is also separately grouped |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
110 * into critical bands for use in the calculating the masking curve. |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
111 * 128 units in psd = -6 dB. The dbknee parameter in AC3BitAllocParameters |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
112 * determines the reference level. |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
113 * |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
114 * @param[in] exp frequency coefficient exponents |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
115 * @param[in] start starting bin location |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
116 * @param[in] end ending bin location |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
117 * @param[out] psd signal power for each frequency bin |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
118 * @param[out] bndpsd signal power for each critical band |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
119 */ |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
120 void ff_ac3_bit_alloc_calc_psd(int8_t *exp, int start, int end, int16_t *psd, |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
121 int16_t *bndpsd); |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
122 |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
123 /** |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
124 * Calculates the masking curve. |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
125 * First, the excitation is calculated using parameters in \p s and the signal |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
126 * power in each critical band. The excitation is compared with a predefined |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
127 * hearing threshold table to produce the masking curve. If delta bit |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
128 * allocation information is provided, it is used for adjusting the masking |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
129 * curve, usually to give a closer match to a better psychoacoustic model. |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
130 * |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
131 * @param[in] s adjustable bit allocation parameters |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
132 * @param[in] bndpsd signal power for each critical band |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
133 * @param[in] start starting bin location |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
134 * @param[in] end ending bin location |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
135 * @param[in] fgain fast gain (estimated signal-to-mask ratio) |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
136 * @param[in] is_lfe whether or not the channel being processed is the LFE |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
137 * @param[in] deltbae delta bit allocation exists (none, reuse, or new) |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
138 * @param[in] deltnseg number of delta segments |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
139 * @param[in] deltoffst location offsets for each segment |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
140 * @param[in] deltlen length of each segment |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
141 * @param[in] deltba delta bit allocation for each segment |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
142 * @param[out] mask calculated masking curve |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
143 */ |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
144 void ff_ac3_bit_alloc_calc_mask(AC3BitAllocParameters *s, int16_t *bndpsd, |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
145 int start, int end, int fgain, int is_lfe, |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
146 int deltbae, int deltnseg, uint8_t *deltoffst, |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
147 uint8_t *deltlen, uint8_t *deltba, |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
148 int16_t *mask); |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
149 |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
150 /** |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
151 * Calculates bit allocation pointers. |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
152 * The SNR is the difference between the masking curve and the signal. AC-3 |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
153 * uses this value for each frequency bin to allocate bits. The \p snroffset |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
154 * parameter is a global adjustment to the SNR for all bins. |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
155 * |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
156 * @param[in] mask masking curve |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
157 * @param[in] psd signal power for each frequency bin |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
158 * @param[in] start starting bin location |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
159 * @param[in] end ending bin location |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
160 * @param[in] snroffset SNR adjustment |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
161 * @param[in] floor noise floor |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
162 * @param[out] bap bit allocation pointers |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
163 */ |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
164 void ff_ac3_bit_alloc_calc_bap(int16_t *mask, int16_t *psd, int start, int end, |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
165 int snroffset, int floor, uint8_t *bap); |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
166 |
4197 | 167 void ac3_parametric_bit_allocation(AC3BitAllocParameters *s, uint8_t *bap, |
168 int8_t *exp, int start, int end, | |
169 int snroffset, int fgain, int is_lfe, | |
170 int deltbae,int deltnseg, | |
171 uint8_t *deltoffst, uint8_t *deltlen, uint8_t *deltba); |