kernel32/tests: Call NtQuerySystemInformation(SystemProcessInformation) until it…

dlls/kernel32/tests/toolhelp.c

@@ -230,18 +230,19 @@ static DWORD WINAPI get_id_thread(void* curr_pid)
     ok(found == FALSE, "The thread order is not strictly consistent\n");
 
     /* Determine the order by NtQuerySystemInformation function */
-    pcs_buffer = NULL;
-    status = pNtQuerySystemInformation(SystemProcessInformation, pcs_buffer, buf_size, &buf_size);
-    ok(status == STATUS_INFO_LENGTH_MISMATCH, "Failed with %x\n", status);
-    if (status == STATUS_INFO_LENGTH_MISMATCH)
+
+    while ((status = NtQuerySystemInformation(SystemProcessInformation,
+            pcs_buffer, buf_size, &buf_size)) == STATUS_INFO_LENGTH_MISMATCH)
     {
-        pcs_buffer = HeapAlloc(GetProcessHeap(), 0, buf_size);
-        ok(pcs_buffer != NULL, "Unable to allocate space\n");
+        free(pcs_buffer);
+        pcs_buffer = malloc(buf_size);
+    }
+    ok(status == STATUS_SUCCESS, "got %#x\n", status);
     found = FALSE;
     matched_idx = -1;
 
-        status = NtQuerySystemInformation(SystemProcessInformation, pcs_buffer, buf_size, &buf_size);
-        do {
+    do
+    {
         spi = (SYSTEM_PROCESS_INFORMATION*)&pcs_buffer[pcs_offset];
         if (spi->UniqueProcessId == curr_pid)
         {
@@ -266,9 +267,7 @@ static DWORD WINAPI get_id_thread(void* curr_pid)
             thread_traversed[matched_idx++] = tid;
         }
     }
-	}
-    if (pcs_buffer)
-        HeapFree(GetProcessHeap(), 0, pcs_buffer);
+    free(pcs_buffer);
 
     ok(matched_idx > 0, "No thread id match found\n");