pogmommy/nmrpflash
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Show exit prompt in Windows when run from console

Browse source
This commit is contained in:
Joseph C. Lehner 2017-01-13 11:22:57 +01:00
parent 6835881e80
commit f24eb9e549
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
main.c
View file

@ -104,6 +104,20 @@ void require_admin()
fprintf(stderr, "Warning: failed to check administrator privileges\n"); 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 #else
void require_admin() void require_admin()
{ {
@ -139,6 +153,8 @@ int main(int argc, char **argv)
char *windir = NULL; char *windir = NULL;
WSADATA wsa; WSADATA wsa;
atexit(&show_exit_prompt);
val = WSAStartup(MAKEWORD(2, 2), &wsa); val = WSAStartup(MAKEWORD(2, 2), &wsa);
if (val != 0) { if (val != 0) {
win_perror2("WSAStartup", val); win_perror2("WSAStartup", val);