diff mbox

[FFmpeg-devel,2/6] avcodec/loco: Check for end of input in pixel decode

Message ID 20190802234957.11098-2-michael@niedermayer.cc
State Accepted
Commit 8305a4509af2908d88bb623deb816fdaa8056c83
Headers show

Commit Message

Michael Niedermayer Aug. 2, 2019, 11:49 p.m. UTC
Fixes: Timeout (100sec -> 5sec)
Fixes: 15509/clusterfuzz-testcase-minimized-ffmpeg_AV_CODEC_ID_LOCO_fuzzer-5724297261219840

Found-by: continuous fuzzing process https://github.com/google/oss-fuzz/tree/master/projects/ffmpeg
Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>
---
 libavcodec/loco.c | 4 ++++
 1 file changed, 4 insertions(+)

Comments

Paul B Mahol Aug. 5, 2019, 3 p.m. UTC | #1
lgtm

On Sat, Aug 3, 2019 at 1:51 AM Michael Niedermayer <michael@niedermayer.cc>
wrote:

> Fixes: Timeout (100sec -> 5sec)
> Fixes:
> 15509/clusterfuzz-testcase-minimized-ffmpeg_AV_CODEC_ID_LOCO_fuzzer-5724297261219840
>
> Found-by: continuous fuzzing process
> https://github.com/google/oss-fuzz/tree/master/projects/ffmpeg
> Signed-off-by
> <https://github.com/google/oss-fuzz/tree/master/projects/ffmpegSigned-off-by>:
> Michael Niedermayer <michael@niedermayer.cc>
> ---
>  libavcodec/loco.c | 4 ++++
>  1 file changed, 4 insertions(+)
>
> diff --git a/libavcodec/loco.c b/libavcodec/loco.c
> index e8c62b8178..5fb414b411 100644
> --- a/libavcodec/loco.c
> +++ b/libavcodec/loco.c
> @@ -88,6 +88,8 @@ static inline int loco_get_rice(RICEContext *r)
>          loco_update_rice_param(r, 0);
>          return 0;
>      }
> +    if (get_bits_left(&r->gb) < 1)
> +        return INT_MIN;
>      v = get_ur_golomb_jpegls(&r->gb, loco_get_rice_param(r), INT_MAX, 0);
>      loco_update_rice_param(r, (v + 1) >> 1);
>      if (!v) {
> @@ -163,6 +165,8 @@ static int loco_decode_plane(LOCOContext *l, uint8_t
> *data, int width, int heigh
>          /* restore all other pixels */
>          for (i = 1; i < width; i++) {
>              val = loco_get_rice(&rc);
> +            if (val == INT_MIN)
> +                return -1;
>              data[i] = loco_predict(&data[i], stride) + val;
>          }
>          data += stride;
> --
> 2.22.0
>
> _______________________________________________
> ffmpeg-devel mailing list
> ffmpeg-devel@ffmpeg.org
> https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
>
> To unsubscribe, visit link above, or email
> ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe".
Michael Niedermayer Aug. 5, 2019, 3:09 p.m. UTC | #2
On Mon, Aug 05, 2019 at 05:00:08PM +0200, Paul B Mahol wrote:
> lgtm

will apply

thx

[...]
diff mbox

Patch

diff --git a/libavcodec/loco.c b/libavcodec/loco.c
index e8c62b8178..5fb414b411 100644
--- a/libavcodec/loco.c
+++ b/libavcodec/loco.c
@@ -88,6 +88,8 @@  static inline int loco_get_rice(RICEContext *r)
         loco_update_rice_param(r, 0);
         return 0;
     }
+    if (get_bits_left(&r->gb) < 1)
+        return INT_MIN;
     v = get_ur_golomb_jpegls(&r->gb, loco_get_rice_param(r), INT_MAX, 0);
     loco_update_rice_param(r, (v + 1) >> 1);
     if (!v) {
@@ -163,6 +165,8 @@  static int loco_decode_plane(LOCOContext *l, uint8_t *data, int width, int heigh
         /* restore all other pixels */
         for (i = 1; i < width; i++) {
             val = loco_get_rice(&rc);
+            if (val == INT_MIN)
+                return -1;
             data[i] = loco_predict(&data[i], stride) + val;
         }
         data += stride;