pg_get_line.c File Reference

#include "postgres.h"
#include <setjmp.h>
#include "common/string.h"
#include "lib/stringinfo.h"

Include dependency graph for pg_get_line.c:

Go to the source code of this file.

Functions
char *	pg_get_line (FILE *stream, PromptInterruptContext *prompt_ctx)
bool	pg_get_line_buf (FILE *stream, StringInfo buf)
bool	pg_get_line_append (FILE *stream, StringInfo buf, PromptInterruptContext *prompt_ctx)

Function Documentation

pg_get_line()

char* pg_get_line	(	FILE *	stream,
		PromptInterruptContext *	prompt_ctx
	)

Definition at line 59 of file pg_get_line.c.

{
    StringInfoData buf;
 
    initStringInfo(&buf);
 
    if (!pg_get_line_append(stream, &buf, prompt_ctx))
    {
        /* ensure that free() doesn't mess up errno */
        int         save_errno = errno;
 
        pfree(buf.data);
        errno = save_errno;
        return NULL;
    }
 
    return buf.data;
}

References buf, initStringInfo(), pfree(), and pg_get_line_append().

Referenced by get_su_pwd(), pipe_read_line(), and simple_prompt_extended().

pg_get_line_append()

bool pg_get_line_append	(	FILE *	stream,
		StringInfo	buf,
		PromptInterruptContext *	prompt_ctx
	)

Definition at line 124 of file pg_get_line.c.

{
    int         orig_len = buf->len;
 
    if (prompt_ctx && sigsetjmp(*((sigjmp_buf *) prompt_ctx->jmpbuf), 1) != 0)
    {
        /* Got here with longjmp */
        prompt_ctx->canceled = true;
        /* Discard any data we collected before detecting error */
        buf->len = orig_len;
        buf->data[orig_len] = '\0';
        return false;
    }
 
    /* Loop until newline or EOF/error */
    for (;;)
    {
        char       *res;
 
        /* Enable longjmp while waiting for input */
        if (prompt_ctx)
            *(prompt_ctx->enabled) = true;
 
        /* Read some data, appending it to whatever we already have */
        res = fgets(buf->data + buf->len, buf->maxlen - buf->len, stream);
 
        /* Disable longjmp again, then break if fgets failed */
        if (prompt_ctx)
            *(prompt_ctx->enabled) = false;
 
        if (res == NULL)
            break;
 
        /* Got data, so update buf->len */
        buf->len += strlen(buf->data + buf->len);
 
        /* Done if we have collected a newline */
        if (buf->len > orig_len && buf->data[buf->len - 1] == '\n')
            return true;
 
        /* Make some more room in the buffer, and loop to read more data */
        enlargeStringInfo(buf, 128);
    }
 
    /* Check for I/O errors and EOF */
    if (ferror(stream) || buf->len == orig_len)
    {
        /* Discard any data we collected before detecting error */
        buf->len = orig_len;
        buf->data[orig_len] = '\0';
        return false;
    }
 
    /* No newline at EOF, but we did collect some data */
    return true;
}

References buf, PromptInterruptContext::canceled, PromptInterruptContext::enabled, enlargeStringInfo(), PromptInterruptContext::jmpbuf, and res.

Referenced by pg_get_line(), pg_get_line_buf(), and tokenize_auth_file().

pg_get_line_buf()

bool pg_get_line_buf	(	FILE *	stream,
		StringInfo	buf
	)

Definition at line 95 of file pg_get_line.c.

{
    /* We just need to drop any data from the previous call */
    resetStringInfo(buf);
    return pg_get_line_append(stream, buf, NULL);
}

References buf, pg_get_line_append(), and resetStringInfo().

Referenced by ecpg_filter_source(), ecpg_filter_stderr(), filter_read_item(), read_quoted_string(), readfile(), SortTocFromFile(), and tsearch_readline().