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

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

reg/tests: Free the WCHAR memory buffer if CreateFile() fails.
e0a97d32 · Hugh McMaster · Alexandre Julliard · c618e994 · e0a97d32
Commit e0a97d32 authored Aug 23, 2017 by Hugh McMaster Committed by Alexandre Julliard Aug 24, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

reg.c programs/reg/tests/reg.c +4 -1

No files found.
--- 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());