feedc0de/ArduinoJson
1
0
Fork 0
forked from bblanchon/ArduinoJson
Code Pull Requests Activity

MsgPackDeserializer: check extension allocation result

Browse Source
This commit is contained in:
Benoit Blanchon
2024-09-04 14:34:02 +02:00
parent 1f7a3f3174
commit 3b64197869
2 changed files with 51 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
extras/tests/MsgPackDeserializer/errors.cpp
View File

@ -199,3 +199,44 @@ TEST_CASE(
REQUIRE(err == DeserializationError::NoMemory); REQUIRE(err == DeserializationError::NoMemory);
} }
} }
TEST_CASE(
"deserializeMsgPack() returns NoMemory if extension allocation fails") {
JsonDocument doc(FailingAllocator::instance());
SECTION("uint32_t should pass") {
auto err = deserializeMsgPack(doc, "\xceXXXX");
REQUIRE(err == DeserializationError::Ok);
}
SECTION("uint64_t should fail") {
auto err = deserializeMsgPack(doc, "\xcfXXXXXXXX");
REQUIRE(err == DeserializationError::NoMemory);
}
SECTION("int32_t should pass") {
auto err = deserializeMsgPack(doc, "\xd2XXXX");
REQUIRE(err == DeserializationError::Ok);
}
SECTION("int64_t should fail") {
auto err = deserializeMsgPack(doc, "\xd3XXXXXXXX");
REQUIRE(err == DeserializationError::NoMemory);
}
SECTION("float should pass") {
auto err = deserializeMsgPack(doc, "\xcaXXXX");
REQUIRE(err == DeserializationError::Ok);
}
SECTION("double should fail") {
auto err = deserializeMsgPack(doc, "\xcbXXXXXXXX");
REQUIRE(err == DeserializationError::NoMemory);
}
}

16
src/ArduinoJson/MsgPack/MsgPackDeserializer.hpp
View File

@ -230,13 +230,16 @@ class MsgPackDeserializer {
if (isSigned) { if (isSigned) {
auto truncatedValue = static_cast<JsonInteger>(signedValue); auto truncatedValue = static_cast<JsonInteger>(signedValue);
if (truncatedValue == signedValue) if (truncatedValue == signedValue) {
variant->setInteger(truncatedValue, resources_); if (!variant->setInteger(truncatedValue, resources_))
return DeserializationError::NoMemory;
}
// else set null on overflow // else set null on overflow
} else { } else {
auto truncatedValue = static_cast<JsonUInt>(unsignedValue); auto truncatedValue = static_cast<JsonUInt>(unsignedValue);
if (truncatedValue == unsignedValue) if (truncatedValue == unsignedValue)
variant->setInteger(truncatedValue, resources_); if (!variant->setInteger(truncatedValue, resources_))
return DeserializationError::NoMemory;
// else set null on overflow // else set null on overflow
} }
@ -270,9 +273,10 @@ class MsgPackDeserializer {
return err; return err;
fixEndianness(value); fixEndianness(value);
variant->setFloat(value, resources_); if (variant->setFloat(value, resources_))
return DeserializationError::Ok;
return DeserializationError::Ok; else
return DeserializationError::NoMemory;
} }
template <typename T> template <typename T>