PostgreSQL Source Code  git master
sprompt.c File Reference
#include "c.h"
#include "common/fe_memutils.h"
#include "common/string.h"
Include dependency graph for sprompt.c:

Go to the source code of this file.

Functions

char * simple_prompt (const char *prompt, bool echo)
 
char * simple_prompt_extended (const char *prompt, bool echo, PromptInterruptContext *prompt_ctx)
 

Function Documentation

◆ simple_prompt()

char* simple_prompt ( const char *  prompt,
bool  echo 
)

Definition at line 38 of file sprompt.c.

39 {
40  return simple_prompt_extended(prompt, echo, NULL);
41 }
char * simple_prompt_extended(const char *prompt, bool echo, PromptInterruptContext *prompt_ctx)
Definition: sprompt.c:53

References simple_prompt_extended().

Referenced by ConnectDatabase(), connectDatabase(), doConnect(), get_su_pwd(), GetConnection(), main(), sql_conn(), vacuumlo(), and yesno_prompt().

◆ simple_prompt_extended()

char* simple_prompt_extended ( const char *  prompt,
bool  echo,
PromptInterruptContext prompt_ctx 
)

Definition at line 53 of file sprompt.c.

55 {
56  char *result;
57  FILE *termin,
58  *termout;
59 #if defined(HAVE_TERMIOS_H)
60  struct termios t_orig,
61  t;
62 #elif defined(WIN32)
63  HANDLE t = NULL;
64  DWORD t_orig = 0;
65 #endif
66 
67 #ifdef WIN32
68 
69  /*
70  * A Windows console has an "input code page" and an "output code page";
71  * these usually match each other, but they rarely match the "Windows ANSI
72  * code page" defined at system boot and expected of "char *" arguments to
73  * Windows API functions. The Microsoft CRT write() implementation
74  * automatically converts text between these code pages when writing to a
75  * console. To identify such file descriptors, it calls GetConsoleMode()
76  * on the underlying HANDLE, which in turn requires GENERIC_READ access on
77  * the HANDLE. Opening termout in mode "w+" allows that detection to
78  * succeed. Otherwise, write() would not recognize the descriptor as a
79  * console, and non-ASCII characters would display incorrectly.
80  *
81  * XXX fgets() still receives text in the console's input code page. This
82  * makes non-ASCII credentials unportable.
83  *
84  * Unintuitively, we also open termin in mode "w+", even though we only
85  * read it; that's needed for SetConsoleMode() to succeed.
86  */
87  termin = fopen("CONIN$", "w+");
88  termout = fopen("CONOUT$", "w+");
89 #else
90 
91  /*
92  * Do not try to collapse these into one "w+" mode file. Doesn't work on
93  * some platforms (eg, HPUX 10.20).
94  */
95  termin = fopen("/dev/tty", "r");
96  termout = fopen("/dev/tty", "w");
97 #endif
98  if (!termin || !termout
99 #ifdef WIN32
100 
101  /*
102  * Direct console I/O does not work from the MSYS 1.0.10 console. Writes
103  * reach nowhere user-visible; reads block indefinitely. XXX This affects
104  * most Windows terminal environments, including rxvt, mintty, Cygwin
105  * xterm, Cygwin sshd, and PowerShell ISE. Switch to a more-generic test.
106  */
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  if (!echo)
120  {
121 #if defined(HAVE_TERMIOS_H)
122  /* disable echo via tcgetattr/tcsetattr */
123  tcgetattr(fileno(termin), &t);
124  t_orig = t;
125  t.c_lflag &= ~ECHO;
126  tcsetattr(fileno(termin), TCSAFLUSH, &t);
127 #elif defined(WIN32)
128  /* need the file's HANDLE to turn echo off */
129  t = (HANDLE) _get_osfhandle(_fileno(termin));
130 
131  /* save the old configuration first */
132  GetConsoleMode(t, &t_orig);
133 
134  /* set to the new mode */
135  SetConsoleMode(t, ENABLE_LINE_INPUT | ENABLE_PROCESSED_INPUT);
136 #endif
137  }
138 
139  if (prompt)
140  {
141  fputs(_(prompt), termout);
142  fflush(termout);
143  }
144 
145  result = pg_get_line(termin, prompt_ctx);
146 
147  /* If we failed to read anything, just return an empty string */
148  if (result == NULL)
149  result = pg_strdup("");
150 
151  /* strip trailing newline, including \r in case we're on Windows */
152  (void) pg_strip_crlf(result);
153 
154  if (!echo)
155  {
156  /* restore previous echo behavior, then echo \n */
157 #if defined(HAVE_TERMIOS_H)
158  tcsetattr(fileno(termin), TCSAFLUSH, &t_orig);
159  fputs("\n", termout);
160  fflush(termout);
161 #elif defined(WIN32)
162  SetConsoleMode(t, t_orig);
163  fputs("\n", termout);
164  fflush(termout);
165 #endif
166  }
167  else if (prompt_ctx && prompt_ctx->canceled)
168  {
169  /* also echo \n if prompt was canceled */
170  fputs("\n", termout);
171  fflush(termout);
172  }
173 
174  if (termin != stdin)
175  {
176  fclose(termin);
177  fclose(termout);
178  }
179 
180  return result;
181 }
#define _(x)
Definition: elog.c:90
char * pg_strdup(const char *in)
Definition: fe_memutils.c:85
static void const char fflush(stdout)
char * pg_get_line(FILE *stream, PromptInterruptContext *prompt_ctx)
Definition: pg_get_line.c:59
int pg_strip_crlf(char *str)
Definition: string.c:155

References _, PromptInterruptContext::canceled, fflush(), pg_get_line(), pg_strdup(), and pg_strip_crlf().

Referenced by exec_command_password(), exec_command_prompt(), prompt_for_password(), and simple_prompt().