wasm memzero

Signed-off-by: turuslan <[email protected]>
qdrvm · Jul 17, 2024 · 4cd26aa · 4cd26aa
1 parent 2d8d4d8
commit 4cd26aa
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/core/runtime/common/module_instance.cpp b/core/runtime/common/module_instance.cpp
@@ -6,6 +6,8 @@
 
 #include "runtime/module_instance.hpp"
 
+#include <openssl/mem.h>
+
 #include "common/int_serialization.hpp"
 #include "runtime/memory_provider.hpp"
 #include "runtime/trie_storage_provider.hpp"
@@ -59,6 +61,9 @@ namespace kagome::runtime {
           "__heap_base too low, allocations will overwrite wasm data segments");
     }
 
+    auto memory_size = memory.memory()->size();
+    OPENSSL_cleanse(memory.view(0, memory_size).value().data(), memory_size);
+
     forDataSegment([&](auto offset, auto segment) {
       memory.storeBuffer(offset, segment);
     });

diff --git a/core/runtime/memory.hpp b/core/runtime/memory.hpp
@@ -121,6 +121,10 @@ namespace kagome::runtime {
       return PtrSize{ptr, static_cast<WasmSize>(v.size())}.combine();
     }
 
+    auto &memory() const {
+      return handle_;
+    }
+
    private:
     std::shared_ptr<MemoryHandle> handle_;
     std::unique_ptr<MemoryAllocator> allocator_;