nominatim/sprompt.c

   1 /*-------------------------------------------------------------------------
   2  *
   3  * sprompt.c
   4  *        simple_prompt() routine
   5  *
   6  * Portions Copyright (c) 1996-2006, PostgreSQL Global Development Group
   7  * Portions Copyright (c) 1994, Regents of the University of California
   8  *
   9  *
  10  * IDENTIFICATION
  11  *        $PostgreSQL: pgsql/src/port/sprompt.c,v 1.18 2006/10/04 00:30:14 momjian Exp $
  12  *
  13  *-------------------------------------------------------------------------
  14  *
  15  * PostgreSQL Database Management System
  16  * (formerly known as Postgres, then as Postgres95)
  17  *
  18  * Portions Copyright (c) 1996-2006, PostgreSQL Global Development Group
  19  *
  20  * Portions Copyright (c) 1994, The Regents of the University of California
  21  *
  22  * Permission to use, copy, modify, and distribute this software and its
  23  * documentation for any purpose, without fee, and without a written agreement
  24  * is hereby granted, provided that the above copyright notice and this
  25  * paragraph and the following two paragraphs appear in all copies.
  26  *
  27  * IN NO EVENT SHALL THE UNIVERSITY OF CALIFORNIA BE LIABLE TO ANY PARTY FOR
  28  * DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES, INCLUDING
  29  * LOST PROFITS, ARISING OUT OF THE USE OF THIS SOFTWARE AND ITS
  30  * DOCUMENTATION, EVEN IF THE UNIVERSITY OF CALIFORNIA HAS BEEN ADVISED OF THE
  31  * POSSIBILITY OF SUCH DAMAGE.
  32  *
  33  * THE UNIVERSITY OF CALIFORNIA SPECIFICALLY DISCLAIMS ANY WARRANTIES,
  34  * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
  35  * AND FITNESS FOR A PARTICULAR PURPOSE.  THE SOFTWARE PROVIDED HEREUNDER IS
  36  * ON AN "AS IS" BASIS, AND THE UNIVERSITY OF CALIFORNIA HAS NO OBLIGATIONS TO
  37  * PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
  38  *
  39  */
  40
  41
  42 /*
  43  * simple_prompt
  44  *
  45  * Generalized function especially intended for reading in usernames and
  46  * password interactively. Reads from /dev/tty or stdin/stderr.
  47  *
  48  * prompt:              The prompt to print
  49  * maxlen:              How many characters to accept
  50  * echo:                Set to false if you want to hide what is entered (for passwords)
  51  *
  52  * Returns a malloc()'ed string with the input (w/o trailing newline).
  53  */
  54
  55 #define DEVTTY "/dev/tty"
  56
  57 #include <stdio.h>
  58 #include <unistd.h>
  59 #include <stdlib.h>
  60 #include <string.h>
  61 #include <assert.h>
  62
  63 #include <libpq-fe.h>
  64
  65 #ifdef __MINGW_H
  66 # include <windows.h>
  67 #else
  68 # define HAVE_TERMIOS_H
  69 # include <termios.h>
  70 #endif
  71
  72 /*
  73 extern char *simple_prompt(const char *prompt, int maxlen, int echo);
  74 */
  75
  76 char *
  77 simple_prompt(const char *prompt, int maxlen, int echo)
  78 {
  79         int                     length;
  80         char       *destination;
  81         FILE       *termin,
  82                            *termout;
  83
  84 #ifdef HAVE_TERMIOS_H
  85         struct termios t_orig,
  86                                 t;
  87 #else
  88 #ifdef WIN32
  89         HANDLE          t = NULL;
  90         LPDWORD         t_orig = NULL;
  91 #endif
  92 #endif
  93
  94         destination = (char *) malloc(maxlen + 1);
  95         if (!destination)
  96                 return NULL;
  97
  98         /*
  99          * Do not try to collapse these into one "w+" mode file. Doesn't work on
 100          * some platforms (eg, HPUX 10.20).
 101          */
 102         termin = fopen(DEVTTY, "r");
 103         termout = fopen(DEVTTY, "w");
 104         if (!termin || !termout
 105 #ifdef WIN32
 106         /* See DEVTTY comment for msys */
 107                 || (getenv("OSTYPE") && strcmp(getenv("OSTYPE"), "msys") == 0)
 108 #endif
 109                 )
 110         {
 111                 if (termin)
 112                         fclose(termin);
 113                 if (termout)
 114                         fclose(termout);
 115                 termin = stdin;
 116                 termout = stderr;
 117         }
 118
 119 #ifdef HAVE_TERMIOS_H
 120         if (!echo)
 121         {
 122                 tcgetattr(fileno(termin), &t);
 123                 t_orig = t;
 124                 t.c_lflag &= ~ECHO;
 125                 tcsetattr(fileno(termin), TCSAFLUSH, &t);
 126         }
 127 #else
 128 #ifdef WIN32
 129         if (!echo)
 130         {
 131                 /* get a new handle to turn echo off */
 132                 t_orig = (LPDWORD) malloc(sizeof(DWORD));
 133                 t = GetStdHandle(STD_INPUT_HANDLE);
 134
 135                 /* save the old configuration first */
 136                 GetConsoleMode(t, t_orig);
 137
 138                 /* set to the new mode */
 139                 SetConsoleMode(t, ENABLE_LINE_INPUT | ENABLE_PROCESSED_INPUT);
 140         }
 141 #endif
 142 #endif
 143
 144         if (prompt)
 145         {
 146                 fputs(prompt, termout);
 147                 fflush(termout);
 148         }
 149
 150         if (fgets(destination, maxlen + 1, termin) == NULL)
 151                 destination[0] = '\0';
 152
 153         length = strlen(destination);
 154         if (length > 0 && destination[length - 1] != '\n')
 155         {
 156                 /* eat rest of the line */
 157                 char            buf[128];
 158                 int                     buflen;
 159
 160                 do
 161                 {
 162                         if (fgets(buf, sizeof(buf), termin) == NULL)
 163                                 break;
 164                         buflen = strlen(buf);
 165                 } while (buflen > 0 && buf[buflen - 1] != '\n');
 166         }
 167
 168         if (length > 0 && destination[length - 1] == '\n')
 169                 /* remove trailing newline */
 170                 destination[length - 1] = '\0';
 171
 172 #ifdef HAVE_TERMIOS_H
 173         if (!echo)
 174         {
 175                 tcsetattr(fileno(termin), TCSAFLUSH, &t_orig);
 176                 fputs("\n", termout);
 177                 fflush(termout);
 178         }
 179 #else
 180 #ifdef WIN32
 181         if (!echo)
 182         {
 183                 /* reset to the original console mode */
 184                 SetConsoleMode(t, *t_orig);
 185                 fputs("\n", termout);
 186                 fflush(termout);
 187                 free(t_orig);
 188         }
 189 #endif
 190 #endif
 191
 192         if (termin != stdin)
 193         {
 194                 fclose(termin);
 195                 fclose(termout);
 196         }
 197
 198         return destination;
 199 }