From f24eb9e549f03df5d7f28c591d787ac8898ce5c9 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 | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/main.c b/main.c index 2830cf7..66d7d80 100644 --- a/main.c +++ b/main.c @@ -104,6 +104,20 @@ void require_admin() fprintf(stderr, "Warning: failed to check administrator privileges\n"); } + +void show_exit_prompt() +{ + DWORD pid; + HWND win = GetConsoleWindow(); + if (!win || !GetWindowThreadProcessId(win, &pid)) { + return; + } + + if (GetCurrentProcessId() == pid) { + printf("Press any key to exit\n"); + getch(); + } +} #else void require_admin() { @@ -139,6 +153,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