]> git.pond.sub.org Git - empserver/blob - src/client/getpass.c
client: Factor set_echo_if_tty() out of get_password()
[empserver] / src / client / getpass.c
1 /*
2  *  Empire - A multi-player, client/server Internet based war game.
3  *  Copyright (C) 1986-2020, Dave Pare, Jeff Bailey, Thomas Ruschak,
4  *                Ken Stevens, Steve McClure, Markus Armbruster
5  *
6  *  Empire is free software: you can redistribute it and/or modify
7  *  it under the terms of the GNU General Public License as published by
8  *  the Free Software Foundation, either version 3 of the License, or
9  *  (at your option) any later version.
10  *
11  *  This program is distributed in the hope that it will be useful,
12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  *  GNU General Public License for more details.
15  *
16  *  You should have received a copy of the GNU General Public License
17  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
18  *
19  *  ---
20  *
21  *  See files README, COPYING and CREDITS in the root of the source
22  *  tree for related information and legal notices.  It is expected
23  *  that future projects/authors will amend these files as needed.
24  *
25  *  ---
26  *
27  *  getpass.c: Get a password
28  *
29  *  Known contributors to this file:
30  *     Markus Armbruster, 2009-2020
31  */
32
33 #include <config.h>
34
35 #ifdef HAVE_GETPASS
36 #include <unistd.h>
37 #else
38 #include <string.h>
39 #ifdef _WIN32
40 #include <windows.h>
41 #endif
42 #endif
43 #include "misc.h"
44
45 #ifndef HAVE_GETPASS
46 static int
47 set_echo_if_tty(int on)
48 {
49 #ifdef _WIN32
50     DWORD mode;
51     HANDLE input_handle = GetStdHandle(STD_INPUT_HANDLE);
52
53     if (!GetConsoleMode(input_handle, &mode))
54         return 0;
55
56     if (on)
57         mode |= ENABLE_ECHO_INPUT;
58     else
59         mode &= ~ENABLE_ECHO_INPUT;
60
61     if (!SetConsoleMode(input_handle, mode))
62         return -1;
63     return 1;
64 #else
65     return 0;
66 #endif
67 }
68 #endif  /* !HAVE_GETPASS */
69
70 char *
71 get_password(const char *prompt)
72 {
73 #ifdef HAVE_GETPASS
74     return getpass(prompt);
75 #else
76     static char buf[128];
77     char *p;
78     size_t len;
79     int echo_set;
80
81     echo_set = set_echo_if_tty(0);
82     if (echo_set <= 0)
83         printf("Note: your input is echoed to the screen\n");
84
85     printf("%s", prompt);
86     fflush(stdout);
87     p = fgets(buf, sizeof(buf), stdin);
88
89     if (echo_set > 0)
90         set_echo_if_tty(1);
91
92     if (!p)
93         return NULL;
94     len = strlen(p);
95     if (p[len - 1] == '\n')
96         p[len - 1] = 0;
97     return p;
98 #endif  /* !HAVE_GETPASS */
99 }