test/memcpy: Use BCryptGenRandom

(cherry picked from commit 79caf85)
Signed-off-by: LIU Hao <lh_mouse@126.com>

test/memory.c

@@ -10,14 +10,20 @@
 #include <stdio.h>
 #include <string.h>
 #include <winbase.h>
-#include <ntsecapi.h>
 
 static char comp[100];
 static char data[100];
 
 int
 main(void)
   {
+    HMODULE bcrypt = LoadLibraryW(L"bcrypt.dll");
+    assert(bcrypt);
+
+    typedef NTSTATUS __stdcall BCryptGenRandom_t(HANDLE, void*, ULONG, ULONG);
+    BCryptGenRandom_t* pBCryptGenRandom = (BCryptGenRandom_t*)(intptr_t) GetProcAddress(bcrypt, "BCryptGenRandom");
+    assert(pBCryptGenRandom);
+
     HMODULE msvcrt = LoadLibraryW(L"msvcrt.dll");
     assert(msvcrt);
 
@@ -34,7 +40,7 @@ main(void)
     assert(pmemcmp);
 
     // __MCF_mfill
-    RtlGenRandom(comp, sizeof(comp));
+    pBCryptGenRandom(NULL, comp, sizeof(comp), 2);
     pmemmove(data, comp, sizeof(comp));
 
     pmemset(comp + 20, 'b', 30);
@@ -49,7 +55,7 @@ main(void)
     assert(memcmp(comp, data, sizeof(comp)) == 0);
 
     // __MCF_mzero
-    RtlGenRandom(comp, sizeof(comp));
+    pBCryptGenRandom(NULL, comp, sizeof(comp), 2);
     pmemmove(data, comp, sizeof(comp));
 
     pmemset(comp + 20, 0, 30);
@@ -64,7 +70,7 @@ main(void)
     assert(memcmp(comp, data, sizeof(comp)) == 0);
 
     // __MCF_mcopy
-    RtlGenRandom(comp, sizeof(comp));
+    pBCryptGenRandom(NULL, comp, sizeof(comp), 2);
     pmemmove(data, comp, sizeof(comp));
 
     pmemmove(comp + 10, comp + 20, 40);
@@ -75,7 +81,7 @@ main(void)
     assert(memcmp(comp, data, sizeof(comp)) == 0);
 
     // __MCF_mcopy_backward
-    RtlGenRandom(comp, sizeof(comp));
+    pBCryptGenRandom(NULL, comp, sizeof(comp), 2);
     pmemmove(data, comp, sizeof(comp));
 
     pmemmove(comp + 20, comp + 10, 40);
@@ -86,7 +92,7 @@ main(void)
     assert(memcmp(comp, data, sizeof(comp)) == 0);
 
     // __MCF_mcomp (equal)
-    RtlGenRandom(comp, sizeof(comp));
+    pBCryptGenRandom(NULL, comp, sizeof(comp), 2);
     pmemmove(data, comp, sizeof(comp));
 
     assert(pmemcmp(comp, data, sizeof(comp)) == 0);