lib: base64 - Add more w_buf_len assertions to encoder.

diff --git a/src/lib/base64.c b/src/lib/base64.c
index 1970cde890ccbbd97c43a3c4545a256551a82918..9aec963b58d71f629605b8a9cee8b0debedd2e4e 100644 (file)
--- a/src/lib/base64.c
+++ b/src/lib/base64.c
@@ -381,6 +381,7 @@ bool base64_encode_finish(struct base64_encoder *enc, buffer_t *dest)
        case 0:
                break;
        case 1:
+               i_assert(w_buf_len < (sizeof(w_buf) - 3));
                w_buf[w_buf_len] = b64enc[enc->buf];
                w_buf_len ++;
                if (padding) {
@@ -390,6 +391,7 @@ bool base64_encode_finish(struct base64_encoder *enc, buffer_t *dest)
                }
                break;
        case 2:
+               i_assert(w_buf_len < (sizeof(w_buf) - 2));
                w_buf[w_buf_len] = b64enc[enc->buf];
                w_buf_len++;
                if (padding) {