From e1f5788fb0dfbe8003727802e931d1dbeaf68ae1 Mon Sep 17 00:00:00 2001
From: Evgenii Kliuchnikov <eustas@google.com>
Date: Fri, 14 Apr 2023 15:46:39 +0000
Subject: Fix internal buffer reset

PiperOrigin-RevId: 524301253
---
 c/dec/decode.c | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'c')

diff --git a/c/dec/decode.c b/c/dec/decode.c
index fe8af63..de46eb4 100644
--- a/c/dec/decode.c
+++ b/c/dec/decode.c
@@ -116,6 +116,10 @@ static BROTLI_NOINLINE BrotliDecoderResult SaveErrorCode(
     BrotliDecoderState* s, BrotliDecoderErrorCode e, size_t consumed_input) {
   s->error_code = (int)e;
   s->used_input += consumed_input;
+  if ((s->buffer_length != 0) && (s->br.next_in == s->br.last_in)) {
+    /* If internal buffer is depleted at last, reset it. */
+    s->buffer_length = 0;
+  }
   switch (e) {
     case BROTLI_DECODER_SUCCESS:
       return BROTLI_DECODER_RESULT_SUCCESS;
@@ -2351,6 +2355,7 @@ BrotliDecoderResult BrotliDecoderDecompressStream(
             /* Not enough data in buffer, but can take one more byte from
                input stream. */
             result = BROTLI_DECODER_SUCCESS;
+            BROTLI_DCHECK(s->buffer_length < 8);
             s->buffer.u8[s->buffer_length] = **next_in;
             s->buffer_length++;
             BrotliBitReaderSetInput(br, &s->buffer.u8[0], s->buffer_length);
-- 
cgit v1.1