Hugh McMaster : reg/tests: Free the WCHAR memory buffer if CreateFile() fails.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: e0a97d32e24528128b874641393107bac0591fd1
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=e0a97d32e24528128b874641393107bac0591fd1

Author: Hugh McMaster <hugh.mcmaster at outlook.com>
Date:   Wed Aug 23 12:31:56 2017 +0000

reg/tests: Free the WCHAR memory buffer if CreateFile() fails.

Signed-off-by: Hugh McMaster <hugh.mcmaster at outlook.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>

---

 programs/reg/tests/reg.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/programs/reg/tests/reg.c b/programs/reg/tests/reg.c
index 0a2fdbf..f6a7ce1 100644
--- a/programs/reg/tests/reg.c
+++ b/programs/reg/tests/reg.c
@@ -771,8 +771,11 @@ static BOOL test_import_wstr_(unsigned line, const char *file_contents, DWORD *r
     regfile = CreateFileA("test.reg", GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
                           FILE_ATTRIBUTE_NORMAL, NULL);
     lok(regfile != INVALID_HANDLE_VALUE, "Failed to create test.reg file\n");
-    if(regfile == INVALID_HANDLE_VALUE)
+    if (regfile == INVALID_HANDLE_VALUE)
+    {
+        HeapFree(GetProcessHeap(), 0, wstr);
         return FALSE;
+    }
 
     ret = WriteFile(regfile, wstr, memsize, &written, NULL);
     lok(ret, "WriteFile failed: %u\n", GetLastError());