Ensure header hash is correctly persisted to context in table create …

…mode.
tpn · Jan 3, 2023 · ee91729 · ee91729
1 parent 2f8cb1b
commit ee91729
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/PerfectHash/PerfectHashContextTableCreate.c b/src/PerfectHash/PerfectHashContextTableCreate.c
@@ -848,6 +848,23 @@ Return Value:
     Context->TableCreateCsvFile = File;
     File->Vtbl->AddRef(File);
 
+    //
+    // Capture the header hash in the context.
+    //
+
+    Context->HexHeaderHash.Length = sizeof(Context->HexHeaderHashBuffer);
+    Context->HexHeaderHash.MaximumLength = Context->HexHeaderHash.Length;
+    ASSERT(Context->HexHeaderHash.Length >= HexHash.Length - 1);
+
+    ZeroMemory(&Context->HexHeaderHashBuffer, Context->HexHeaderHash.Length);
+    CopyMemory(Context->HexHeaderHashBuffer,
+               HexHash.Buffer + 1,
+               HexHash.Length - 1);
+
+    Context->HexHeaderHash.Buffer = (PCHAR)&Context->HexHeaderHashBuffer;
+    Context->HexHeaderHash.Length =
+        (USHORT)strlen(Context->HexHeaderHash.Buffer);
+
     //
     // We're done, finish up.
     //