commit 7c1be8ddd11a0ae7936cd5afc2c58c11735f88f3
author Kasun Athukorala <kasunath@google.com> Tue Jul 15 05:44:04 2025 +0000
committer Kasun Athukorala <kasunath@google.com> Tue Jul 15 17:22:15 2025 +0000
tree 11fed9af0e9c761a35671b949c65e30bcf0624bf
parent 6fd03fbf98ffe6d7713cc9fa1fe9ef8c34514163

Check dictionary sizes to avoid any heap overflows

Tested:
Unit tested

Signed-off-by: Kasun Athukorala <kasunath@google.com>
Change-Id: I34972b7f4daf0e818461ef2a4842966a247fc20e

test/bej_encoder_test.cpp

diff --git a/test/bej_encoder_test.cpp b/test/bej_encoder_test.cpp
index 413d4e1..113a9de 100644
--- a/test/bej_encoder_test.cpp
+++ b/test/bej_encoder_test.cpp
@@ -344,8 +344,11 @@
 
     BejDictionaries dictionaries = {
         .schemaDictionary = inputsOrErr->schemaDictionary,
+        .schemaDictionarySize = inputsOrErr->schemaDictionarySize,
         .annotationDictionary = inputsOrErr->annotationDictionary,
+        .annotationDictionarySize = inputsOrErr->annotationDictionarySize,
         .errorDictionary = inputsOrErr->errorDictionary,
+        .errorDictionarySize = inputsOrErr->errorDictionarySize,
     };
 
     std::vector<uint8_t> outputBuffer;
@@ -388,8 +391,11 @@
 
     BejDictionaries dictionaries = {
         .schemaDictionary = inputsOrErr->schemaDictionary,
+        .schemaDictionarySize = inputsOrErr->schemaDictionarySize,
         .annotationDictionary = inputsOrErr->annotationDictionary,
+        .annotationDictionarySize = inputsOrErr->annotationDictionarySize,
         .errorDictionary = inputsOrErr->errorDictionary,
+        .errorDictionarySize = inputsOrErr->errorDictionarySize,
     };
 
     libbej::BejEncoderJson encoder;