Commit e268a352 authored by Janne Grunau's avatar Janne Grunau

mjpegdec: parse RSTn to prevent skipping other data in mjpeg_decode_scan

Check explicitly if enough bits are left to prevent an infinite loop
when the bitstream buffer is not followed by zero-padding.

Based on patches by Michael Niedermayer <michaelni@gmx.at>.
parent 5e5cde27
...@@ -881,14 +881,22 @@ static int mjpeg_decode_scan(MJpegDecodeContext *s, int nb_components, int Ah, i ...@@ -881,14 +881,22 @@ static int mjpeg_decode_scan(MJpegDecodeContext *s, int nb_components, int Ah, i
} }
} }
if (s->restart_interval && show_bits(&s->gb, 8) == 0xFF){ /* skip RSTn */ if (s->restart_interval) {
--s->restart_count; s->restart_count--;
align_get_bits(&s->gb); i = 8 + ((-get_bits_count(&s->gb)) & 7);
while(show_bits(&s->gb, 8) == 0xFF) /* skip RSTn */
skip_bits(&s->gb, 8); if (show_bits(&s->gb, i) == (1 << i) - 1) {
skip_bits(&s->gb, 8); int pos = get_bits_count(&s->gb);
for (i=0; i<nb_components; i++) /* reset dc */ align_get_bits(&s->gb);
s->last_dc[i] = 1024; while (get_bits_left(&s->gb) >= 8 && show_bits(&s->gb, 8) == 0xFF)
skip_bits(&s->gb, 8);
if ((get_bits(&s->gb, 8) & 0xF8) == 0xD0) {
for (i = 0; i < nb_components; i++) /* reset dc */
s->last_dc[i] = 1024;
} else {
skip_bits_long(&s->gb, pos - get_bits_count(&s->gb));
}
}
} }
} }
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment