git.pond.sub.org Git - empserver/blob - 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 #else
  42 #include <termios.h>
  43 #endif
  44 #endif
  45 #include "misc.h"
  46
  47 #ifndef HAVE_GETPASS
  48 static int
  49 set_echo_if_tty(int on)
  50 {
  51 #ifdef _WIN32
  52     DWORD mode;
  53     HANDLE input_handle = GetStdHandle(STD_INPUT_HANDLE);
  54
  55     if (!GetConsoleMode(input_handle, &mode))
  56         return 0;
  57
  58     if (on)
  59         mode |= ENABLE_ECHO_INPUT;
  60     else
  61         mode &= ~ENABLE_ECHO_INPUT;
  62
  63     if (!SetConsoleMode(input_handle, mode))
  64         return -1;
  65     return 1;
  66 #else
  67     struct termios tcattr;
  68
  69     if (tcgetattr(0, &tcattr) < 0)
  70        return 0;
  71
  72     if (on)
  73        tcattr.c_lflag |= ECHO;
  74     else
  75        tcattr.c_lflag &= ~ECHO;
  76
  77     if (tcsetattr(0, TCSAFLUSH, &tcattr) < 0)
  78        return -1;
  79     return 1;
  80 #endif
  81 }
  82 #endif  /* !HAVE_GETPASS */
  83
  84 char *
  85 get_password(const char *prompt)
  86 {
  87 #ifdef HAVE_GETPASS
  88     return getpass(prompt);
  89 #else
  90     static char buf[128];
  91     char *p;
  92     size_t len;
  93     int echo_set;
  94
  95     echo_set = set_echo_if_tty(0);
  96     if (echo_set <= 0)
  97         printf("Note: your input is echoed to the screen\n");
  98
  99     printf("%s", prompt);
 100     fflush(stdout);
 101     p = fgets(buf, sizeof(buf), stdin);
 102
 103     if (echo_set > 0)
 104         set_echo_if_tty(1);
 105
 106     if (!p)
 107         return NULL;
 108     len = strlen(p);
 109     if (p[len - 1] == '\n')
 110         p[len - 1] = 0;
 111     return p;
 112 #endif  /* !HAVE_GETPASS */
 113 }