/*------------------------------------------------------------------------- * * quote.c * Functions for quoting identifiers and literals * * Portions Copyright (c) 2000-2008, PostgreSQL Global Development Group * * * IDENTIFICATION * $PostgreSQL: pgsql/src/backend/utils/adt/quote.c,v 1.23 2008/01/01 19:45:52 momjian Exp $ * *------------------------------------------------------------------------- */ #include "postgres.h" #include "utils/builtins.h" /* * quote_ident - * returns a properly quoted identifier */ Datum quote_ident(PG_FUNCTION_ARGS) { text *t = PG_GETARG_TEXT_P(0); text *result; const char *qstr; char *str; int len; /* We have to convert to a C string to use quote_identifier */ len = VARSIZE(t) - VARHDRSZ; str = (char *) palloc(len + 1); memcpy(str, VARDATA(t), len); str[len] = '\0'; qstr = quote_identifier(str); len = strlen(qstr); result = (text *) palloc(len + VARHDRSZ); SET_VARSIZE(result, len + VARHDRSZ); memcpy(VARDATA(result), qstr, len); PG_RETURN_TEXT_P(result); } /* * quote_literal - * returns a properly quoted literal * * NOTE: think not to make this function's behavior change with * standard_conforming_strings. We don't know where the result * literal will be used, and so we must generate a result that * will work with either setting. Take a look at what dblink * uses this for before thinking you know better. */ Datum quote_literal(PG_FUNCTION_ARGS) { text *t = PG_GETARG_TEXT_P(0); text *result; char *cp1; char *cp2; int len; len = VARSIZE(t) - VARHDRSZ; /* We make a worst-case result area; wasting a little space is OK */ result = (text *) palloc(len * 2 + 3 + VARHDRSZ); cp1 = VARDATA(t); cp2 = VARDATA(result); for (; len-- > 0; cp1++) { if (*cp1 == '\\') { *cp2++ = ESCAPE_STRING_SYNTAX; break; } } len = VARSIZE(t) - VARHDRSZ; cp1 = VARDATA(t); *cp2++ = '\''; while (len-- > 0) { if (SQL_STR_DOUBLE(*cp1, true)) *cp2++ = *cp1; *cp2++ = *cp1++; } *cp2++ = '\''; SET_VARSIZE(result, cp2 - ((char *) result)); PG_RETURN_TEXT_P(result); }
