From: Joseph C. Lehner <joseph.c.lehner@gmail.com>
Date: Fri, 13 Jan 2017 10:22:57 +0000 (+0100)
Subject: Show exit prompt in Windows when run from console
X-Git-Tag: v0.9.10~7
X-Git-Url: https://git.librecmc.org/?a=commitdiff_plain;h=8bbfb543b5ffc9a6a8aaaa71f2843172b2f0fea9;p=oweals%2Fnmrpflash.git

Show exit prompt in Windows when run from console
---

diff --git a/main.c b/main.c
index 2830cf7..d17cb3a 100644
--- a/main.c
+++ b/main.c
@@ -104,6 +104,19 @@ void require_admin()
 
 	fprintf(stderr, "Warning: failed to check administrator privileges\n");
 }
+
+void show_exit_prompt()
+{
+	HWND win = GetConsoleWindow();
+	if (!win) {
+		return;
+	}
+
+	if (GetCurrentProcessId() == GetWindowThreadProcessId(win, NULL)) {
+		printf("\nPress any key to exit\n");
+		getch();
+	}
+}
 #else
 void require_admin()
 {
@@ -139,6 +152,8 @@ int main(int argc, char **argv)
 	char *windir = NULL;
 	WSADATA wsa;
 
+	atexit(&show_exit_prompt);
+
 	val = WSAStartup(MAKEWORD(2, 2), &wsa);
 	if (val != 0) {
 		win_perror2("WSAStartup", val);