bej_decoder: Check decoded string length
We need to check the bejString value lengths to prevent heap
buffer overflow
Tested:
Unit tested
Change-Id: Ie6a014fbffeb31f111bfbae331db197b3fb2f2ca
Signed-off-by: Kasun Athukorala <kasunath@google.com>
diff --git a/src/bej_decoder_json.cpp b/src/bej_decoder_json.cpp
index 9fceeba..95e241f 100644
--- a/src/bej_decoder_json.cpp
+++ b/src/bej_decoder_json.cpp
@@ -1,5 +1,9 @@
#include "bej_decoder_json.hpp"
+#include <string.h>
+
+#define MAX_BEJ_STRING_LEN 65536
+
namespace libbej
{
@@ -172,17 +176,29 @@
*
* @param[in] propertyName - a NULL terminated string.
* @param[in] value - a NULL terminated string.
+ * @param[in] length - length of the string.
* @param[in] dataPtr - pointing to a valid BejJsonParam struct.
* @return 0 if successful.
*/
static int callbackString(const char* propertyName, const char* value,
- void* dataPtr)
+ size_t length, void* dataPtr)
{
+ if ((length > MAX_BEJ_STRING_LEN) ||
+ (strnlen(value, length) != (length - 1)))
+ {
+ fprintf(stderr,
+ "Incorrect BEJ string length %zu or it exceeds maximum %u.\n",
+ (length - 1), MAX_BEJ_STRING_LEN);
+ return bejErrorInvalidSize;
+ }
struct BejJsonParam* params =
reinterpret_cast<struct BejJsonParam*>(dataPtr);
addPropertyNameToOutput(params, propertyName);
params->output->push_back('\"');
- params->output->append(value);
+ if (length > 0)
+ {
+ params->output->append(value, length - 1);
+ }
params->output->push_back('\"');
*params->isPrevAnnotated = false;
return 0;