Fix the crash issue for 7-Zip's hash algorithms wrappers which mentio…

…ned #542. (Thanks to RuesanG.)
M2Team · Jan 7, 2025 · 3c78701 · 3c78701
1 parent a32520b
commit 3c78701
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/NanaZip.Core/Wrappers/Sha256Wrapper.cpp b/NanaZip.Core/Wrappers/Sha256Wrapper.cpp
@@ -54,6 +54,12 @@ void Sha256_InitState(
 void Sha256_Init(
     CSha256* p)
 {
+    if (p)
+    {
+        // Workaround for using the uninitialized memory.
+        std::memset(p, 0, sizeof(CSha256));
+    }
+
     ::Sha256_InitState(p);
 }
 

diff --git a/NanaZip.Core/Wrappers/Sha512Wrapper.cpp b/NanaZip.Core/Wrappers/Sha512Wrapper.cpp
@@ -67,6 +67,12 @@ void Sha512_Init(
     CSha512* p,
     unsigned digestSize)
 {
+    if (p)
+    {
+        // Workaround for using the uninitialized memory.
+        std::memset(p, 0, sizeof(CSha512));
+    }
+
     ::Sha512_InitState(p, digestSize);
 }