mirror of
https://github.com/postgres/postgres.git
synced 2026-06-15 13:09:51 -04:00
Tom Lane
f3164c0200
places --- that risks corrupting data structures, losing sync with the backend, etc. We now longjmp only from calls to readline, fgets, and fread, which we assume are coded to protect themselves against interrupts at undesirable times. This requires adding explicit tests for cancel_pressed in long-running loops, but on the whole it's far cleaner. Martijn van Oosterhout and Tom Lane.
161 lines
3.2 KiB
C
161 lines
3.2 KiB
C
/*-------------------------------------------------------------------------
|
|
*
|
|
* sprompt.c
|
|
* simple_prompt() routine
|
|
*
|
|
* Portions Copyright (c) 1996-2006, PostgreSQL Global Development Group
|
|
* Portions Copyright (c) 1994, Regents of the University of California
|
|
*
|
|
*
|
|
* IDENTIFICATION
|
|
* $PostgreSQL: pgsql/src/port/sprompt.c,v 1.17 2006/06/14 16:49:03 tgl Exp $
|
|
*
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
|
|
|
|
/*
|
|
* simple_prompt
|
|
*
|
|
* Generalized function especially intended for reading in usernames and
|
|
* password interactively. Reads from /dev/tty or stdin/stderr.
|
|
*
|
|
* prompt: The prompt to print
|
|
* maxlen: How many characters to accept
|
|
* echo: Set to false if you want to hide what is entered (for passwords)
|
|
*
|
|
* Returns a malloc()'ed string with the input (w/o trailing newline).
|
|
*/
|
|
#include "c.h"
|
|
|
|
#ifdef HAVE_TERMIOS_H
|
|
#include <termios.h>
|
|
#endif
|
|
|
|
extern char *simple_prompt(const char *prompt, int maxlen, bool echo);
|
|
|
|
char *
|
|
simple_prompt(const char *prompt, int maxlen, bool echo)
|
|
{
|
|
int length;
|
|
char *destination;
|
|
FILE *termin,
|
|
*termout;
|
|
|
|
#ifdef HAVE_TERMIOS_H
|
|
struct termios t_orig,
|
|
t;
|
|
#else
|
|
#ifdef WIN32
|
|
HANDLE t = NULL;
|
|
LPDWORD t_orig = NULL;
|
|
#endif
|
|
#endif
|
|
|
|
destination = (char *) malloc(maxlen + 1);
|
|
if (!destination)
|
|
return NULL;
|
|
|
|
/*
|
|
* Do not try to collapse these into one "w+" mode file. Doesn't work on
|
|
* some platforms (eg, HPUX 10.20).
|
|
*/
|
|
termin = fopen(DEVTTY, "r");
|
|
termout = fopen(DEVTTY, "w");
|
|
if (!termin || !termout
|
|
#ifdef WIN32
|
|
/* See DEVTTY comment for msys */
|
|
|| (getenv("OSTYPE") && strcmp(getenv("OSTYPE"), "msys") == 0)
|
|
#endif
|
|
)
|
|
{
|
|
if (termin)
|
|
fclose(termin);
|
|
if (termout)
|
|
fclose(termout);
|
|
termin = stdin;
|
|
termout = stderr;
|
|
}
|
|
|
|
#ifdef HAVE_TERMIOS_H
|
|
if (!echo)
|
|
{
|
|
tcgetattr(fileno(termin), &t);
|
|
t_orig = t;
|
|
t.c_lflag &= ~ECHO;
|
|
tcsetattr(fileno(termin), TCSAFLUSH, &t);
|
|
}
|
|
#else
|
|
#ifdef WIN32
|
|
if (!echo)
|
|
{
|
|
/* get a new handle to turn echo off */
|
|
t_orig = (LPDWORD) malloc(sizeof(DWORD));
|
|
t = GetStdHandle(STD_INPUT_HANDLE);
|
|
|
|
/* save the old configuration first */
|
|
GetConsoleMode(t, t_orig);
|
|
|
|
/* set to the new mode */
|
|
SetConsoleMode(t, ENABLE_LINE_INPUT | ENABLE_PROCESSED_INPUT);
|
|
}
|
|
#endif
|
|
#endif
|
|
|
|
if (prompt)
|
|
{
|
|
fputs(_(prompt), termout);
|
|
fflush(termout);
|
|
}
|
|
|
|
if (fgets(destination, maxlen + 1, termin) == NULL)
|
|
destination[0] = '\0';
|
|
|
|
length = strlen(destination);
|
|
if (length > 0 && destination[length - 1] != '\n')
|
|
{
|
|
/* eat rest of the line */
|
|
char buf[128];
|
|
int buflen;
|
|
|
|
do
|
|
{
|
|
if (fgets(buf, sizeof(buf), termin) == NULL)
|
|
break;
|
|
buflen = strlen(buf);
|
|
} while (buflen > 0 && buf[buflen - 1] != '\n');
|
|
}
|
|
|
|
if (length > 0 && destination[length - 1] == '\n')
|
|
/* remove trailing newline */
|
|
destination[length - 1] = '\0';
|
|
|
|
#ifdef HAVE_TERMIOS_H
|
|
if (!echo)
|
|
{
|
|
tcsetattr(fileno(termin), TCSAFLUSH, &t_orig);
|
|
fputs("\n", termout);
|
|
fflush(termout);
|
|
}
|
|
#else
|
|
#ifdef WIN32
|
|
if (!echo)
|
|
{
|
|
/* reset to the original console mode */
|
|
SetConsoleMode(t, *t_orig);
|
|
fputs("\n", termout);
|
|
fflush(termout);
|
|
free(t_orig);
|
|
}
|
|
#endif
|
|
#endif
|
|
|
|
if (termin != stdin)
|
|
{
|
|
fclose(termin);
|
|
fclose(termout);
|
|
}
|
|
|
|
return destination;
|
|
}
|