diff options
author | Marco Nelissen <marcone@google.com> | 2019-01-23 16:10:48 -0800 |
---|---|---|
committer | Marco Nelissen <marcone@google.com> | 2019-01-23 16:12:20 -0800 |
commit | b5e041da6bfe79b5ee5fb48374cd753df6d18015 (patch) | |
tree | b78174105e19a0ad3cc6d26f26401a72294b49a8 /Tremolo | |
parent | 8340cf450a9a0b43fb5e7a47d3aa78670f1ee515 (diff) | |
download | tremolo-b5e041da6bfe79b5ee5fb48374cd753df6d18015.tar.gz |
Add some error/overflow checks in codebook handling
Bug: 119120561
Test: manual
Change-Id: I2476ef29d81a8335fdcbdbbaa175a575138d4039
Diffstat (limited to 'Tremolo')
-rw-r--r-- | Tremolo/codebook.c | 17 |
1 files changed, 12 insertions, 5 deletions
diff --git a/Tremolo/codebook.c b/Tremolo/codebook.c index 467640d..a06302d 100644 --- a/Tremolo/codebook.c +++ b/Tremolo/codebook.c @@ -376,14 +376,21 @@ long _book_maptype1_quantvals(codebook *b){ long acc=1; long acc1=1; int i; - for(i=0;i<b->dim;i++){ - acc*=vals; - acc1*=vals+1; + for (i = 0; i < b->dim; i++) { + if (acc > b->entries / vals) { + break; + } + acc *= vals; + if (acc1 > LONG_MAX / (vals + 1)) { + acc1 = LONG_MAX; + } else { + acc1 *= (vals + 1); + } } - if(acc<=b->entries && acc1>b->entries){ + if (i >= b->dim && acc <= b->entries && acc1 > b->entries) { return(vals); }else{ - if(acc>b->entries){ + if (i < b->dim || acc > b->entries) { vals--; }else{ vals++; |