From 8f80b52b8daec619516c0f9b37d791a88937a549 Mon Sep 17 00:00:00 2001
From: KiritoDv <kiritodev01@gmail.com>
Date: Wed, 31 Jan 2024 00:11:31 -0600
Subject: Fixed segment validation and added size fallback on decompressor

---
 src/utils/Decompressor.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'src/utils/Decompressor.cpp')

diff --git a/src/utils/Decompressor.cpp b/src/utils/Decompressor.cpp
index 4e89cb7..7de4338 100644
--- a/src/utils/Decompressor.cpp
+++ b/src/utils/Decompressor.cpp
@@ -63,7 +63,7 @@ DecompressedData Decompressor::AutoDecode(YAML::Node& node, std::vector<uint8_t>
 
     return {
         .root = nullptr,
-        .segment = { buffer.data() + offset, buffer.size() - offset }
+        .segment = { buffer.data() + offset, node["size"] ? node["size"].as<size_t>() : buffer.size() - offset }
     };
 }
 
-- 
cgit v1.2.3