From 8bbfb543b5ffc9a6a8aaaa71f2843172b2f0fea9 Mon Sep 17 00:00:00 2001 From: "Joseph C. Lehner" Date: Fri, 13 Jan 2017 11:22:57 +0100 Subject: [PATCH] Show exit prompt in Windows when run from console --- main.c | 15 +++++++++++++++ 1 file changed, 15 insertions(+) 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); -- 2.25.1