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