Mercurial > libavcodec.hg
annotate ac3.c @ 5817:ced30500e2b1 libavcodec
prevent infinite loop and memcpy of negative amounts
fixes issue194
author | michael |
---|---|
date | Sat, 13 Oct 2007 12:25:31 +0000 |
parents | b24bcdd0ae86 |
children | 29635f7c3d9c |
rev | line source |
---|---|
0 | 1 /* |
4642 | 2 * Common code between AC3 encoder and decoder |
429 | 3 * Copyright (c) 2000 Fabrice Bellard. |
0 | 4 * |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3668
diff
changeset
|
5 * This file is part of FFmpeg. |
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3668
diff
changeset
|
6 * |
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3668
diff
changeset
|
7 * FFmpeg is free software; you can redistribute it and/or |
429 | 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:
3668
diff
changeset
|
10 * version 2.1 of the License, or (at your option) any later version. |
0 | 11 * |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3668
diff
changeset
|
12 * FFmpeg is distributed in the hope that it will be useful, |
0 | 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
429 | 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
15 * Lesser General Public License for more details. | |
0 | 16 * |
429 | 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:
3668
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:
2979
diff
changeset
|
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
0 | 20 */ |
1106 | 21 |
22 /** | |
4642 | 23 * @file ac3.c |
24 * Common code between AC3 encoder and decoder. | |
1106 | 25 */ |
782 | 26 |
4642 | 27 #include "avcodec.h" |
28 #include "ac3.h" | |
4648 | 29 #include "bitstream.h" |
0 | 30 |
4879 | 31 static uint8_t bndtab[51]; |
32 static uint8_t masktab[253]; | |
33 | |
4641 | 34 static inline int calc_lowcomp1(int a, int b0, int b1, int c) |
0 | 35 { |
36 if ((b0 + 256) == b1) { | |
4641 | 37 a = c; |
2967 | 38 } else if (b0 > b1) { |
4641 | 39 a = FFMAX(a - 64, 0); |
0 | 40 } |
41 return a; | |
42 } | |
43 | |
44 static inline int calc_lowcomp(int a, int b0, int b1, int bin) | |
45 { | |
46 if (bin < 7) { | |
4641 | 47 return calc_lowcomp1(a, b0, b1, 384); |
0 | 48 } else if (bin < 20) { |
4641 | 49 return calc_lowcomp1(a, b0, b1, 320); |
0 | 50 } else { |
4641 | 51 return FFMAX(a - 128, 0); |
0 | 52 } |
53 } | |
54 | |
4684
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
55 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
|
56 int16_t *bndpsd) |
0 | 57 { |
4684
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
58 int bin, i, j, k, end1, v; |
0 | 59 |
60 /* exponent mapping to PSD */ | |
61 for(bin=start;bin<end;bin++) { | |
62 psd[bin]=(3072 - (exp[bin] << 7)); | |
63 } | |
64 | |
65 /* PSD integration */ | |
66 j=start; | |
67 k=masktab[start]; | |
68 do { | |
69 v=psd[j]; | |
70 j++; | |
4641 | 71 end1 = FFMIN(bndtab[k+1], end); |
0 | 72 for(i=j;i<end1;i++) { |
73 /* logadd */ | |
4641 | 74 int adr = FFMIN(FFABS(v - psd[j]) >> 1, 255); |
4879 | 75 v = FFMAX(v, psd[j]) + ff_ac3_latab[adr]; |
0 | 76 j++; |
77 } | |
78 bndpsd[k]=v; | |
79 k++; | |
80 } while (end > bndtab[k]); | |
4684
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
81 } |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
82 |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
83 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
|
84 int start, int end, int fgain, int is_lfe, |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
85 int deltbae, int deltnseg, uint8_t *deltoffst, |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
86 uint8_t *deltlen, uint8_t *deltba, |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
87 int16_t *mask) |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
88 { |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
89 int16_t excite[50]; /* excitation */ |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
90 int bin, k; |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
91 int bndstrt, bndend, begin, end1, tmp; |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
92 int lowcomp, fastleak, slowleak; |
0 | 93 |
94 /* excitation function */ | |
95 bndstrt = masktab[start]; | |
96 bndend = masktab[end-1] + 1; | |
2967 | 97 |
782 | 98 if (bndstrt == 0) { |
99 lowcomp = 0; | |
4641 | 100 lowcomp = calc_lowcomp1(lowcomp, bndpsd[0], bndpsd[1], 384); |
4640 | 101 excite[0] = bndpsd[0] - fgain - lowcomp; |
4641 | 102 lowcomp = calc_lowcomp1(lowcomp, bndpsd[1], bndpsd[2], 384); |
4640 | 103 excite[1] = bndpsd[1] - fgain - lowcomp; |
104 begin = 7; | |
782 | 105 for (bin = 2; bin < 7; bin++) { |
106 if (!(is_lfe && bin == 6)) | |
4641 | 107 lowcomp = calc_lowcomp1(lowcomp, bndpsd[bin], bndpsd[bin+1], 384); |
4640 | 108 fastleak = bndpsd[bin] - fgain; |
109 slowleak = bndpsd[bin] - s->sgain; | |
110 excite[bin] = fastleak - lowcomp; | |
782 | 111 if (!(is_lfe && bin == 6)) { |
112 if (bndpsd[bin] <= bndpsd[bin+1]) { | |
4640 | 113 begin = bin + 1; |
114 break; | |
782 | 115 } |
116 } | |
117 } | |
2967 | 118 |
782 | 119 end1=bndend; |
120 if (end1 > 22) end1=22; | |
2967 | 121 |
782 | 122 for (bin = begin; bin < end1; bin++) { |
123 if (!(is_lfe && bin == 6)) | |
4640 | 124 lowcomp = calc_lowcomp(lowcomp, bndpsd[bin], bndpsd[bin+1], bin); |
2967 | 125 |
4641 | 126 fastleak = FFMAX(fastleak - s->fdecay, bndpsd[bin] - fgain); |
127 slowleak = FFMAX(slowleak - s->sdecay, bndpsd[bin] - s->sgain); | |
128 excite[bin] = FFMAX(fastleak - lowcomp, slowleak); | |
782 | 129 } |
130 begin = 22; | |
131 } else { | |
132 /* coupling channel */ | |
133 begin = bndstrt; | |
2967 | 134 |
782 | 135 fastleak = (s->cplfleak << 8) + 768; |
136 slowleak = (s->cplsleak << 8) + 768; | |
0 | 137 } |
138 | |
782 | 139 for (bin = begin; bin < bndend; bin++) { |
4641 | 140 fastleak = FFMAX(fastleak - s->fdecay, bndpsd[bin] - fgain); |
141 slowleak = FFMAX(slowleak - s->sdecay, bndpsd[bin] - s->sgain); | |
142 excite[bin] = FFMAX(fastleak, slowleak); | |
0 | 143 } |
144 | |
145 /* compute masking curve */ | |
146 | |
147 for (bin = bndstrt; bin < bndend; bin++) { | |
148 tmp = s->dbknee - bndpsd[bin]; | |
149 if (tmp > 0) { | |
4641 | 150 excite[bin] += tmp >> 2; |
0 | 151 } |
4879 | 152 mask[bin] = FFMAX(ff_ac3_hth[bin >> s->halfratecod][s->fscod], excite[bin]); |
0 | 153 } |
154 | |
782 | 155 /* delta bit allocation */ |
156 | |
5331 | 157 if (deltbae == DBA_REUSE || deltbae == DBA_NEW) { |
782 | 158 int band, seg, delta; |
4640 | 159 band = 0; |
782 | 160 for (seg = 0; seg < deltnseg; seg++) { |
4640 | 161 band += deltoffst[seg]; |
782 | 162 if (deltba[seg] >= 4) { |
163 delta = (deltba[seg] - 3) << 7; | |
164 } else { | |
165 delta = (deltba[seg] - 4) << 7; | |
166 } | |
167 for (k = 0; k < deltlen[seg]; k++) { | |
4640 | 168 mask[band] += delta; |
169 band++; | |
782 | 170 } |
171 } | |
172 } | |
4684
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
173 } |
782 | 174 |
4684
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
175 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
|
176 int snroffset, int floor, uint8_t *bap) |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
177 { |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
178 int i, j, k, end1, v, address; |
2967 | 179 |
4689
c7828f1ae244
fix handling of special case for lowest snroffset. regressions are unaffected.
jbr
parents:
4684
diff
changeset
|
180 /* special case, if snroffset is -960, set all bap's to zero */ |
c7828f1ae244
fix handling of special case for lowest snroffset. regressions are unaffected.
jbr
parents:
4684
diff
changeset
|
181 if(snroffset == -960) { |
c7828f1ae244
fix handling of special case for lowest snroffset. regressions are unaffected.
jbr
parents:
4684
diff
changeset
|
182 memset(bap, 0, 256); |
c7828f1ae244
fix handling of special case for lowest snroffset. regressions are unaffected.
jbr
parents:
4684
diff
changeset
|
183 return; |
c7828f1ae244
fix handling of special case for lowest snroffset. regressions are unaffected.
jbr
parents:
4684
diff
changeset
|
184 } |
c7828f1ae244
fix handling of special case for lowest snroffset. regressions are unaffected.
jbr
parents:
4684
diff
changeset
|
185 |
4640 | 186 i = start; |
187 j = masktab[start]; | |
0 | 188 do { |
4684
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
189 v = (FFMAX(mask[j] - snroffset - floor, 0) & 0x1FE0) + floor; |
4879 | 190 end1 = FFMIN(bndtab[j] + ff_ac3_bndsz[j], end); |
0 | 191 for (k = i; k < end1; k++) { |
4641 | 192 address = av_clip((psd[i] - v) >> 5, 0, 63); |
4879 | 193 bap[i] = ff_ac3_baptab[address]; |
0 | 194 i++; |
195 } | |
4640 | 196 } while (end > bndtab[j++]); |
0 | 197 } |
198 | |
4684
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
199 /* AC3 bit allocation. The algorithm is the one described in the AC3 |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
200 spec. */ |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
201 void ac3_parametric_bit_allocation(AC3BitAllocParameters *s, uint8_t *bap, |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
202 int8_t *exp, int start, int end, |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
203 int snroffset, int fgain, int is_lfe, |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
204 int deltbae,int deltnseg, |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
205 uint8_t *deltoffst, uint8_t *deltlen, |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
206 uint8_t *deltba) |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
207 { |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
208 int16_t psd[256]; /* scaled exponents */ |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
209 int16_t bndpsd[50]; /* interpolated exponents */ |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
210 int16_t mask[50]; /* masking value */ |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
211 |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
212 ff_ac3_bit_alloc_calc_psd(exp, start, end, psd, bndpsd); |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
213 |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
214 ff_ac3_bit_alloc_calc_mask(s, bndpsd, start, end, fgain, is_lfe, |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
215 deltbae, deltnseg, deltoffst, deltlen, deltba, |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
216 mask); |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
217 |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
218 ff_ac3_bit_alloc_calc_bap(mask, psd, start, end, snroffset, s->floor, bap); |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
219 } |
6ec0afffc572
split ac3_parametric_bit_allocation into 3 separate functions
jbr
parents:
4679
diff
changeset
|
220 |
4645
056127e5df89
remove redundancy in AC-3 parser by using common tables from ac3tab.h
jbr
parents:
4642
diff
changeset
|
221 /** |
056127e5df89
remove redundancy in AC-3 parser by using common tables from ac3tab.h
jbr
parents:
4642
diff
changeset
|
222 * Initializes some tables. |
056127e5df89
remove redundancy in AC-3 parser by using common tables from ac3tab.h
jbr
parents:
4642
diff
changeset
|
223 * note: This function must remain thread safe because it is called by the |
056127e5df89
remove redundancy in AC-3 parser by using common tables from ac3tab.h
jbr
parents:
4642
diff
changeset
|
224 * AVParser init code. |
056127e5df89
remove redundancy in AC-3 parser by using common tables from ac3tab.h
jbr
parents:
4642
diff
changeset
|
225 */ |
4197 | 226 void ac3_common_init(void) |
782 | 227 { |
228 int i, j, k, l, v; | |
229 /* compute bndtab and masktab from bandsz */ | |
230 k = 0; | |
231 l = 0; | |
232 for(i=0;i<50;i++) { | |
233 bndtab[i] = l; | |
4879 | 234 v = ff_ac3_bndsz[i]; |
782 | 235 for(j=0;j<v;j++) masktab[k++]=i; |
236 l += v; | |
237 } | |
3668
34d76180e5d0
Fix 2 bit allocation bugs. One fix enables using a higher bandwidth code. The other fixes an issue with floorcod=7.
jbr
parents:
3280
diff
changeset
|
238 bndtab[50] = l; |
782 | 239 } |