292 lines
4.6 KiB
C
292 lines
4.6 KiB
C
|
/*-------------------------------------------------------------------------
|
||
|
*
|
||
|
* quote.c
|
||
|
* Functions for quoting identifiers and literals
|
||
|
*
|
||
|
* Portions Copyright (c) 2000, PostgreSQL, Inc
|
||
|
*
|
||
|
*
|
||
|
* IDENTIFICATION
|
||
|
* $Header: /cvsroot/pgsql/src/backend/utils/adt/quote.c,v 1.1 2000/09/05 20:25:13 wieck Exp $
|
||
|
*
|
||
|
*-------------------------------------------------------------------------
|
||
|
*/
|
||
|
#include <ctype.h>
|
||
|
|
||
|
#include "postgres.h"
|
||
|
|
||
|
#include "mb/pg_wchar.h"
|
||
|
#include "utils/builtins.h"
|
||
|
|
||
|
|
||
|
static bool quote_ident_required(text *iptr);
|
||
|
static text *do_quote_ident(text *iptr);
|
||
|
static text *do_quote_literal(text *iptr);
|
||
|
|
||
|
|
||
|
/*
|
||
|
* quote_ident -
|
||
|
* returns a properly quoted identifier
|
||
|
*/
|
||
|
Datum
|
||
|
quote_ident(PG_FUNCTION_ARGS)
|
||
|
{
|
||
|
text *t = PG_GETARG_TEXT_P(0);
|
||
|
text *result;
|
||
|
|
||
|
if (quote_ident_required(t))
|
||
|
{
|
||
|
result = do_quote_ident(t);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
result = (text *)palloc(VARSIZE(t));
|
||
|
memcpy(result, t, VARSIZE(t));
|
||
|
}
|
||
|
|
||
|
PG_FREE_IF_COPY(t, 0);
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* quote_literal -
|
||
|
* returns a properly quoted literal
|
||
|
*/
|
||
|
Datum
|
||
|
quote_literal(PG_FUNCTION_ARGS)
|
||
|
{
|
||
|
text *t = PG_GETARG_TEXT_P(0);
|
||
|
text *result;
|
||
|
|
||
|
result = do_quote_literal(t);
|
||
|
|
||
|
PG_FREE_IF_COPY(t, 0);
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
|
||
|
/*
|
||
|
* MULTIBYTE dependant internal functions follow
|
||
|
*
|
||
|
*/
|
||
|
|
||
|
|
||
|
#ifndef MULTIBYTE
|
||
|
|
||
|
/* Check if a given identifier needs quoting */
|
||
|
static bool
|
||
|
quote_ident_required(text *iptr)
|
||
|
{
|
||
|
char *cp;
|
||
|
char *ep;
|
||
|
|
||
|
cp = VARDATA(iptr);
|
||
|
ep = VARDATA(iptr) + VARSIZE(iptr) - VARHDRSZ;
|
||
|
|
||
|
if (cp >= ep)
|
||
|
return true;
|
||
|
|
||
|
if (!(*cp == '_' || (*cp >= 'a' && *cp <= 'z')))
|
||
|
return true;
|
||
|
|
||
|
while((++cp) < ep)
|
||
|
{
|
||
|
if (*cp >= 'a' && *cp <= 'z') continue;
|
||
|
if (*cp >= '0' && *cp <= '9') continue;
|
||
|
if (*cp == '_') continue;
|
||
|
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
/* Return a properly quoted identifier */
|
||
|
static text *
|
||
|
do_quote_ident(text *iptr)
|
||
|
{
|
||
|
text *result;
|
||
|
char *cp1;
|
||
|
char *cp2;
|
||
|
int len;
|
||
|
|
||
|
len = VARSIZE(iptr) - VARHDRSZ;
|
||
|
result = (text *)palloc(len * 2 + VARHDRSZ + 2);
|
||
|
|
||
|
cp1 = VARDATA(iptr);
|
||
|
cp2 = VARDATA(result);
|
||
|
|
||
|
*cp2++ = '"';
|
||
|
while(len-- > 0)
|
||
|
{
|
||
|
if (*cp1 == '"')
|
||
|
*cp2++ = '"';
|
||
|
if (*cp1 == '\\')
|
||
|
*cp2++ = '\\';
|
||
|
*cp2++ = *cp1++;
|
||
|
}
|
||
|
*cp2++ = '"';
|
||
|
|
||
|
VARATT_SIZEP(result) = cp2 - ((char *)result);
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
/* Return a properly quoted literal value */
|
||
|
static text *
|
||
|
do_quote_literal(text *lptr)
|
||
|
{
|
||
|
text *result;
|
||
|
char *cp1;
|
||
|
char *cp2;
|
||
|
int len;
|
||
|
|
||
|
len = VARSIZE(lptr) - VARHDRSZ;
|
||
|
result = (text *)palloc(len * 2 + VARHDRSZ + 2);
|
||
|
|
||
|
cp1 = VARDATA(lptr);
|
||
|
cp2 = VARDATA(result);
|
||
|
|
||
|
*cp2++ = '\'';
|
||
|
while(len-- > 0)
|
||
|
{
|
||
|
if (*cp1 == '\'')
|
||
|
*cp2++ = '\'';
|
||
|
if (*cp1 == '\\')
|
||
|
*cp2++ = '\\';
|
||
|
*cp2++ = *cp1++;
|
||
|
}
|
||
|
*cp2++ = '\'';
|
||
|
|
||
|
VARATT_SIZEP(result) = cp2 - ((char *)result);
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
#else
|
||
|
|
||
|
/* Check if a given identifier needs quoting (MULTIBYTE version) */
|
||
|
static bool
|
||
|
quote_ident_required(text *iptr)
|
||
|
{
|
||
|
char *cp;
|
||
|
char *ep;
|
||
|
|
||
|
cp = VARDATA(iptr);
|
||
|
ep = VARDATA(iptr) + VARSIZE(iptr) - VARHDRSZ;
|
||
|
|
||
|
if (cp >= ep)
|
||
|
return true;
|
||
|
|
||
|
if(pg_mblen(cp) != 1)
|
||
|
return true;
|
||
|
if (!(*cp == '_' || (*cp >= 'a' && *cp <= 'z')))
|
||
|
return true;
|
||
|
|
||
|
while((++cp) < ep)
|
||
|
{
|
||
|
if (pg_mblen(cp) != 1)
|
||
|
return true;
|
||
|
|
||
|
if (*cp >= 'a' && *cp <= 'z') continue;
|
||
|
if (*cp >= '0' && *cp <= '9') continue;
|
||
|
if (*cp == '_') continue;
|
||
|
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
/* Return a properly quoted identifier (MULTIBYTE version) */
|
||
|
static text *
|
||
|
do_quote_ident(text *iptr)
|
||
|
{
|
||
|
text *result;
|
||
|
char *cp1;
|
||
|
char *cp2;
|
||
|
int len;
|
||
|
int wl;
|
||
|
|
||
|
len = VARSIZE(iptr) - VARHDRSZ;
|
||
|
result = (text *)palloc(len * 2 + VARHDRSZ + 2);
|
||
|
|
||
|
cp1 = VARDATA(iptr);
|
||
|
cp2 = VARDATA(result);
|
||
|
|
||
|
*cp2++ = '"';
|
||
|
while(len > 0)
|
||
|
{
|
||
|
if ((wl = pg_mblen(cp1)) != 1)
|
||
|
{
|
||
|
len -= wl;
|
||
|
|
||
|
while(wl-- > 0)
|
||
|
*cp2++ = *cp1++;
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
if (*cp1 == '"')
|
||
|
*cp2++ = '"';
|
||
|
if (*cp1 == '\\')
|
||
|
*cp2++ = '\\';
|
||
|
*cp2++ = *cp1++;
|
||
|
|
||
|
len--;
|
||
|
}
|
||
|
*cp2++ = '"';
|
||
|
|
||
|
VARATT_SIZEP(result) = cp2 - ((char *)result);
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
/* Return a properly quoted literal value (MULTIBYTE version) */
|
||
|
static text *
|
||
|
do_quote_literal(text *lptr)
|
||
|
{
|
||
|
text *result;
|
||
|
char *cp1;
|
||
|
char *cp2;
|
||
|
int len;
|
||
|
int wl;
|
||
|
|
||
|
len = VARSIZE(lptr) - VARHDRSZ;
|
||
|
result = (text *)palloc(len * 2 + VARHDRSZ + 2);
|
||
|
|
||
|
cp1 = VARDATA(lptr);
|
||
|
cp2 = VARDATA(result);
|
||
|
|
||
|
*cp2++ = '\'';
|
||
|
while(len > 0)
|
||
|
{
|
||
|
if ((wl = pg_mblen(cp1)) != 1)
|
||
|
{
|
||
|
len -= wl;
|
||
|
|
||
|
while(wl-- > 0)
|
||
|
*cp2++ = *cp1++;
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
if (*cp1 == '\'')
|
||
|
*cp2++ = '\'';
|
||
|
if (*cp1 == '\\')
|
||
|
*cp2++ = '\\';
|
||
|
*cp2++ = *cp1++;
|
||
|
|
||
|
len--;
|
||
|
}
|
||
|
*cp2++ = '\'';
|
||
|
|
||
|
VARATT_SIZEP(result) = cp2 - ((char *)result);
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
#endif
|
||
|
|
||
|
|