psqlscan_int.h File Reference

#include "fe_utils/psqlscan.h"

Include dependency graph for psqlscan_int.h:

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

Go to the source code of this file.

Data Structures
struct	StackElem
struct	PsqlScanStateData

Typedefs
typedef struct yy_buffer_state *	YY_BUFFER_STATE
typedef void *	yyscan_t
typedef struct StackElem	StackElem
typedef struct PsqlScanStateData	PsqlScanStateData

Functions
void	psqlscan_push_new_buffer (PsqlScanState state, const char *newstr, const char *varname)
void	psqlscan_pop_buffer_stack (PsqlScanState state)
void	psqlscan_select_top_buffer (PsqlScanState state)
bool	psqlscan_var_is_current_source (PsqlScanState state, const char *varname)
YY_BUFFER_STATE	psqlscan_prepare_buffer (PsqlScanState state, const char *txt, int len, char **txtcopy)
void	psqlscan_emit (PsqlScanState state, const char *txt, int len)
char *	psqlscan_extract_substring (PsqlScanState state, const char *txt, int len)
void	psqlscan_escape_variable (PsqlScanState state, const char *txt, int len, PsqlScanQuoteType quote)
void	psqlscan_test_variable (PsqlScanState state, const char *txt, int len)

Typedef Documentation

PsqlScanStateData

typedef struct PsqlScanStateData PsqlScanStateData

StackElem

typedef struct StackElem StackElem

YY_BUFFER_STATE

typedef struct yy_buffer_state* YY_BUFFER_STATE

Definition at line 54 of file psqlscan_int.h.

yyscan_t

typedef void* yyscan_t

Definition at line 55 of file psqlscan_int.h.

Function Documentation

psqlscan_emit()

void psqlscan_emit	(	PsqlScanState	state,
		const char *	txt,
		int	len
	)

Definition at line 1517 of file psqlscan.l.

{
    PQExpBuffer output_buf = state->output_buf;
 
    if (state->safe_encoding)
        appendBinaryPQExpBuffer(output_buf, txt, len);
    else
    {
        /* Gotta do it the hard way */
        const char *reference = state->refline;
        int         i;
 
        reference += (txt - state->curline);
 
        for (i = 0; i < len; i++)
        {
            char        ch = txt[i];
 
            if (ch == (char) 0xFF)
                ch = reference[i];
            appendPQExpBufferChar(output_buf, ch);
        }
    }
}

References appendBinaryPQExpBuffer(), appendPQExpBufferChar(), i, and len.

Referenced by psqlscan_escape_variable().

psqlscan_escape_variable()

void psqlscan_escape_variable	(	PsqlScanState	state,
		const char *	txt,
		int	len,
		PsqlScanQuoteType	quote
	)

Definition at line 1585 of file psqlscan.l.

{
    char       *varname;
    char       *value;
 
    /* Variable lookup. */
    varname = psqlscan_extract_substring(state, txt + 2, len - 3);
    if (state->callbacks->get_variable)
        value = state->callbacks->get_variable(varname, quote,
                                               state->cb_passthrough);
    else
        value = NULL;
    free(varname);
 
    if (value)
    {
        /* Emit the suitably-escaped value */
        appendPQExpBufferStr(state->output_buf, value);
        free(value);
    }
    else
    {
        /* Emit original token as-is */
        psqlscan_emit(state, txt, len);
    }
}

References appendPQExpBufferStr(), free, len, psqlscan_emit(), psqlscan_extract_substring(), and value.

psqlscan_extract_substring()

char * psqlscan_extract_substring	(	PsqlScanState	state,
		const char *	txt,
		int	len
	)

Definition at line 1549 of file psqlscan.l.

{
    char       *result = (char *) pg_malloc(len + 1);
 
    if (state->safe_encoding)
        memcpy(result, txt, len);
    else
    {
        /* Gotta do it the hard way */
        const char *reference = state->refline;
        int         i;
 
        reference += (txt - state->curline);
 
        for (i = 0; i < len; i++)
        {
            char        ch = txt[i];
 
            if (ch == (char) 0xFF)
                ch = reference[i];
            result[i] = ch;
        }
    }
    result[len] = '\0';
    return result;
}

References i, len, and pg_malloc().

Referenced by psqlscan_escape_variable(), and psqlscan_test_variable().

psqlscan_pop_buffer_stack()

void psqlscan_pop_buffer_stack

(

PsqlScanState

state

)

Definition at line 1413 of file psqlscan.l.

{
    StackElem  *stackelem = state->buffer_stack;
 
    state->buffer_stack = stackelem->next;
    yy_delete_buffer(stackelem->buf, state->scanner);
    free(stackelem->bufstring);
    if (stackelem->origstring)
        free(stackelem->origstring);
    if (stackelem->varname)
        free(stackelem->varname);
    free(stackelem);
}

References StackElem::buf, StackElem::bufstring, free, StackElem::next, StackElem::origstring, and StackElem::varname.

Referenced by psql_scan_finish().

psqlscan_prepare_buffer()

YY_BUFFER_STATE psqlscan_prepare_buffer	(	PsqlScanState	state,
		const char *	txt,
		int	len,
		char **	txtcopy
	)

Definition at line 1476 of file psqlscan.l.

{
    char       *newtxt;
 
    /* Flex wants two \0 characters after the actual data */
    newtxt = pg_malloc(len + 2);
    *txtcopy = newtxt;
    newtxt[len] = newtxt[len + 1] = YY_END_OF_BUFFER_CHAR;
 
    if (state->safe_encoding)
        memcpy(newtxt, txt, len);
    else
    {
        /* Gotta do it the hard way */
        int         i = 0;
 
        while (i < len)
        {
            int         thislen = PQmblen(txt + i, state->encoding);
 
            /* first byte should always be okay... */
            newtxt[i] = txt[i];
            i++;
            while (--thislen > 0 && i < len)
                newtxt[i++] = (char) 0xFF;
        }
    }
 
    return yy_scan_buffer(newtxt, len + 2, state->scanner);
}

References i, len, pg_malloc(), and PQmblen().

Referenced by psql_scan_setup(), and psqlscan_push_new_buffer().

psqlscan_push_new_buffer()

void psqlscan_push_new_buffer	(	PsqlScanState	state,
		const char *	newstr,
		const char *	varname
	)

Definition at line 1374 of file psqlscan.l.

{
    StackElem  *stackelem;
 
    stackelem = (StackElem *) pg_malloc(sizeof(StackElem));
 
    /*
     * In current usage, the passed varname points at the current flex input
     * buffer; we must copy it before calling psqlscan_prepare_buffer()
     * because that will change the buffer state.
     */
    stackelem->varname = varname ? pg_strdup(varname) : NULL;
 
    stackelem->buf = psqlscan_prepare_buffer(state, newstr, strlen(newstr),
                                             &stackelem->bufstring);
    state->curline = stackelem->bufstring;
    if (state->safe_encoding)
    {
        stackelem->origstring = NULL;
        state->refline = stackelem->bufstring;
    }
    else
    {
        stackelem->origstring = pg_strdup(newstr);
        state->refline = stackelem->origstring;
    }
    stackelem->next = state->buffer_stack;
    state->buffer_stack = stackelem;
}

References StackElem::buf, StackElem::bufstring, StackElem::next, StackElem::origstring, pg_malloc(), pg_strdup(), psqlscan_prepare_buffer(), and StackElem::varname.

psqlscan_select_top_buffer()

void psqlscan_select_top_buffer

(

PsqlScanState

state

)

Definition at line 1431 of file psqlscan.l.

{
    StackElem  *stackelem = state->buffer_stack;
 
    if (stackelem != NULL)
    {
        yy_switch_to_buffer(stackelem->buf, state->scanner);
        state->curline = stackelem->bufstring;
        state->refline = stackelem->origstring ? stackelem->origstring : stackelem->bufstring;
    }
    else
    {
        yy_switch_to_buffer(state->scanbufhandle, state->scanner);
        state->curline = state->scanbuf;
        state->refline = state->scanline;
    }
}

References StackElem::buf, StackElem::bufstring, and StackElem::origstring.

psqlscan_test_variable()

void psqlscan_test_variable	(	PsqlScanState	state,
		const char *	txt,
		int	len
	)

Definition at line 1614 of file psqlscan.l.

{
    char       *varname;
    char       *value;
 
    varname = psqlscan_extract_substring(state, txt + 3, len - 4);
    if (state->callbacks->get_variable)
        value = state->callbacks->get_variable(varname, PQUOTE_PLAIN,
                                               state->cb_passthrough);
    else
        value = NULL;
    free(varname);
 
    if (value != NULL)
    {
        appendPQExpBufferStr(state->output_buf, "TRUE");
        free(value);
    }
    else
    {
        appendPQExpBufferStr(state->output_buf, "FALSE");
    }
}

References appendPQExpBufferStr(), free, len, PQUOTE_PLAIN, psqlscan_extract_substring(), and value.

psqlscan_var_is_current_source()

bool psqlscan_var_is_current_source	(	PsqlScanState	state,
		const char *	varname
	)

Definition at line 1454 of file psqlscan.l.

{
    StackElem  *stackelem;
 
    for (stackelem = state->buffer_stack;
         stackelem != NULL;
         stackelem = stackelem->next)
    {
        if (stackelem->varname && strcmp(stackelem->varname, varname) == 0)
            return true;
    }
    return false;
}

References StackElem::next, and StackElem::varname.