Merge branch 'master' of github.com:jclehner/nmrpflash

jclehner · Jan 13, 2017 · 495891f · 495891f
2 parents 8bbfb54 + f24eb9e
commit 495891f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/main.c b/main.c
@@ -107,13 +107,14 @@ void require_admin()
 
 void show_exit_prompt()
 {
+	DWORD pid;
 	HWND win = GetConsoleWindow();
-	if (!win) {
+	if (!win || !GetWindowThreadProcessId(win, &pid)) {
 		return;
 	}
 
-	if (GetCurrentProcessId() == GetWindowThreadProcessId(win, NULL)) {
-		printf("\nPress any key to exit\n");
+	if (GetCurrentProcessId() == pid) {
+		printf("Press any key to exit\n");
 		getch();
 	}
 }