X-Git-Url: https://git.sesse.net/?p=fjl;a=blobdiff_plain;f=driver.c;h=fcd9f7fb9aed2c1fa8629a2913ddb3b7df2117dc;hp=c5ae87e3b003cc2b060ccfe3fa1571ff817a2246;hb=bbe665e5ef6a103657f4d921ddd4b540f38daf3d;hpb=a1596fbb6a92217ff5eb57600b36a9d695ca1ae8

diff --git a/driver.c b/driver.c
index c5ae87e..fcd9f7f 100644
--- a/driver.c
+++ b/driver.c
@@ -116,18 +116,16 @@ void read_sof(struct byte_source* source, struct jpeg_image* image)
 
 void decode_ac_coefficients(const struct huffman_table* tbl, struct bit_source* bits, int16_t* coeff)
 {
-	for (unsigned i = 1; i < DCTSIZE2; ++i) {
+	for (unsigned i = 0; i < DCTSIZE2 - 1; ) {
 		possibly_refill(bits, DEHUF_AC_TABLE_BITS);
 		unsigned lookup = peek_bits(bits, DEHUF_AC_TABLE_BITS);
 		int code = tbl->ac_table_codes[lookup];
 
-		assert(length == AC_DEHUF_SLOW_PATH || (length > 0 && length <= DEHUF_AC_TABLE_BITS));
-
 		if (__builtin_expect(code == AC_DEHUF_SLOW_PATH, 0)) {
 			unsigned rs = read_huffman_symbol_no_refill(tbl, bits);
 			unsigned r = rs >> 4;
 			unsigned s = rs & 0xf;
-			i += r;
+			i += r + 1;
 			possibly_refill(bits, s);
 
 			if (rs == 0x00) {
@@ -143,9 +141,9 @@ void decode_ac_coefficients(const struct huffman_table* tbl, struct bit_source*
 
 			coeff[unzigzag[i]] = extend(read_bits(bits, s), s);
 		} else {
-			int length = tbl->ac_table_length[lookup];
+			unsigned length = tbl->ac_table_length[lookup];
 			int r = tbl->ac_table_skip[lookup];
-			assert(r >= 0);
+			assert(r >= 1);
 			i += r;
 			assert(bits->bits_available >= length);
 			read_bits(bits, length);