string.h File Reference

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
bool	pg_str_endswith (const char *str, const char *end)
int	strtoint (const char *pg_restrict str, char **pg_restrict endptr, int base)
void	pg_clean_ascii (char *str)
int	pg_strip_crlf (char *str)
char *	pg_get_line (FILE *stream)
bool	pg_get_line_buf (FILE *stream, struct StringInfoData *buf)
bool	pg_get_line_append (FILE *stream, struct StringInfoData *buf)
char *	simple_prompt (const char *prompt, bool echo)

Function Documentation

pg_clean_ascii()

void pg_clean_ascii

(

char *

str

)

Definition at line 82 of file string.c.

Referenced by check_application_name(), check_cluster_name(), and ProcessStartupPacket().

{
    /* Only allow clean ASCII chars in the string */
    char       *p;

    for (p = str; *p != '\0'; p++)
    {
        if (*p < 32 || *p > 126)
            *p = '?';
    }
}

pg_get_line()

char* pg_get_line

(

FILE *

stream

)

Definition at line 52 of file pg_get_line.c.

References buf, StringInfoData::data, initStringInfo(), pfree(), and pg_get_line_append().

Referenced by get_su_pwd(), and simple_prompt().

{
    StringInfoData buf;

    initStringInfo(&buf);

    if (!pg_get_line_append(stream, &buf))
    {
        /* ensure that free() doesn't mess up errno */
        int         save_errno = errno;

        pfree(buf.data);
        errno = save_errno;
        return NULL;
    }

    return buf.data;
}

pg_get_line_append()

bool pg_get_line_append	(	FILE *	stream,
		struct StringInfoData *	buf
	)

pg_get_line_buf()

bool pg_get_line_buf	(	FILE *	stream,
		struct StringInfoData *	buf
	)

pg_str_endswith()

bool pg_str_endswith	(	const char *	str,
		const char *	end
	)

Definition at line 31 of file string.c.

Referenced by process_source_file(), ReceiveTarAndUnpackCopyChunk(), and StartupReplicationSlots().

{
    size_t      slen = strlen(str);
    size_t      elen = strlen(end);

    /* can't be a postfix if longer */
    if (elen > slen)
        return false;

    /* compare the end of the strings */
    str += slen - elen;
    return strcmp(str, end) == 0;
}

pg_strip_crlf()

int pg_strip_crlf

(

char *

str

)

Definition at line 105 of file string.c.

Referenced by adjust_data_dir(), CheckDataVersion(), get_prompt(), get_sock_dir(), get_su_pwd(), getRestoreCommand(), passwordFromFile(), run_ssl_passphrase_command(), simple_prompt(), and tokenize_file().

{
    int         len = strlen(str);

    while (len > 0 && (str[len - 1] == '\n' ||
                       str[len - 1] == '\r'))
        str[--len] = '\0';

    return len;
}

simple_prompt()

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

Definition at line 38 of file sprompt.c.

References _, pg_get_line(), pg_strdup(), and pg_strip_crlf().

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

{
    char       *result;
    FILE       *termin,
               *termout;
#if defined(HAVE_TERMIOS_H)
    struct termios t_orig,
                t;
#elif defined(WIN32)
    HANDLE      t = NULL;
    DWORD       t_orig = 0;
#endif

#ifdef WIN32

    /*
     * A Windows console has an "input code page" and an "output code page";
     * these usually match each other, but they rarely match the "Windows ANSI
     * code page" defined at system boot and expected of "char *" arguments to
     * Windows API functions.  The Microsoft CRT write() implementation
     * automatically converts text between these code pages when writing to a
     * console.  To identify such file descriptors, it calls GetConsoleMode()
     * on the underlying HANDLE, which in turn requires GENERIC_READ access on
     * the HANDLE.  Opening termout in mode "w+" allows that detection to
     * succeed.  Otherwise, write() would not recognize the descriptor as a
     * console, and non-ASCII characters would display incorrectly.
     *
     * XXX fgets() still receives text in the console's input code page.  This
     * makes non-ASCII credentials unportable.
     *
     * Unintuitively, we also open termin in mode "w+", even though we only
     * read it; that's needed for SetConsoleMode() to succeed.
     */
    termin = fopen("CONIN$", "w+");
    termout = fopen("CONOUT$", "w+");
#else

    /*
     * Do not try to collapse these into one "w+" mode file. Doesn't work on
     * some platforms (eg, HPUX 10.20).
     */
    termin = fopen("/dev/tty", "r");
    termout = fopen("/dev/tty", "w");
#endif
    if (!termin || !termout
#ifdef WIN32

    /*
     * Direct console I/O does not work from the MSYS 1.0.10 console.  Writes
     * reach nowhere user-visible; reads block indefinitely.  XXX This affects
     * most Windows terminal environments, including rxvt, mintty, Cygwin
     * xterm, Cygwin sshd, and PowerShell ISE.  Switch to a more-generic test.
     */
        || (getenv("OSTYPE") && strcmp(getenv("OSTYPE"), "msys") == 0)
#endif
        )
    {
        if (termin)
            fclose(termin);
        if (termout)
            fclose(termout);
        termin = stdin;
        termout = stderr;
    }

    if (!echo)
    {
#if defined(HAVE_TERMIOS_H)
        /* disable echo via tcgetattr/tcsetattr */
        tcgetattr(fileno(termin), &t);
        t_orig = t;
        t.c_lflag &= ~ECHO;
        tcsetattr(fileno(termin), TCSAFLUSH, &t);
#elif defined(WIN32)
        /* need the file's HANDLE to turn echo off */
        t = (HANDLE) _get_osfhandle(_fileno(termin));

        /* save the old configuration first */
        GetConsoleMode(t, &t_orig);

        /* set to the new mode */
        SetConsoleMode(t, ENABLE_LINE_INPUT | ENABLE_PROCESSED_INPUT);
#endif
    }

    if (prompt)
    {
        fputs(_(prompt), termout);
        fflush(termout);
    }

    result = pg_get_line(termin);

    /* If we failed to read anything, just return an empty string */
    if (result == NULL)
        result = pg_strdup("");

    /* strip trailing newline, including \r in case we're on Windows */
    (void) pg_strip_crlf(result);

    if (!echo)
    {
        /* restore previous echo behavior, then echo \n */
#if defined(HAVE_TERMIOS_H)
        tcsetattr(fileno(termin), TCSAFLUSH, &t_orig);
        fputs("\n", termout);
        fflush(termout);
#elif defined(WIN32)
        SetConsoleMode(t, t_orig);
        fputs("\n", termout);
        fflush(termout);
#endif
    }

    if (termin != stdin)
    {
        fclose(termin);
        fclose(termout);
    }

    return result;
}

strtoint()

int strtoint	(	const char *pg_restrict	str,
		char **pg_restrict	endptr,
		int	base
	)

Definition at line 50 of file string.c.

References val.

Referenced by buildDefItem(), DecodeDateTime(), DecodeInterval(), DecodeNumber(), DecodeNumberField(), DecodeTime(), DecodeTimeOnly(), DecodeTimezone(), and nodeTokenType().

{
    long        val;

    val = strtol(str, endptr, base);
    if (val != (int) val)
        errno = ERANGE;
    return (int) val;
}