Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commitdaf1e3a

Browse files
Jan WieckJan Wieck
Jan Wieck
authored and
Jan Wieck
committed
Added functions
 quote_ident(text) returns text quote_literal(text) returns textThese are handy to build up properly quoted query stringsfor the new PL/pgSQL EXECUTE functionality to submitdynamic DDL statements.Jan
1 parentd7f1e11 commitdaf1e3a

File tree

4 files changed

+304
-4
lines changed

4 files changed

+304
-4
lines changed

‎src/backend/utils/adt/Makefile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#
22
# Makefile for utils/adt
33
#
4-
# $Header: /cvsroot/pgsql/src/backend/utils/adt/Makefile,v 1.45 2000/08/31 16:10:45 petere Exp $
4+
# $Header: /cvsroot/pgsql/src/backend/utils/adt/Makefile,v 1.46 2000/09/05 20:25:12 wieck Exp $
55
#
66

77
subdir = src/backend/utils/adt
@@ -24,7 +24,7 @@ OBJS = acl.o arrayfuncs.o arrayutils.o bool.o cash.o char.o \
2424
tid.o timestamp.o varbit.o varchar.o varlena.o version.o\
2525
network.o mac.o inet_net_ntop.o inet_net_pton.o\
2626
ri_triggers.o pg_lzcompress.o pg_locale.o formatting.o\
27-
ascii.o
27+
ascii.o quote.o
2828

2929
all: SUBSYS.o
3030

‎src/backend/utils/adt/quote.c

Lines changed: 291 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,291 @@
1+
/*-------------------------------------------------------------------------
2+
*
3+
* quote.c
4+
* Functions for quoting identifiers and literals
5+
*
6+
* Portions Copyright (c) 2000, PostgreSQL, Inc
7+
*
8+
*
9+
* IDENTIFICATION
10+
* $Header: /cvsroot/pgsql/src/backend/utils/adt/quote.c,v 1.1 2000/09/05 20:25:13 wieck Exp $
11+
*
12+
*-------------------------------------------------------------------------
13+
*/
14+
#include<ctype.h>
15+
16+
#include"postgres.h"
17+
18+
#include"mb/pg_wchar.h"
19+
#include"utils/builtins.h"
20+
21+
22+
staticboolquote_ident_required(text*iptr);
23+
statictext*do_quote_ident(text*iptr);
24+
statictext*do_quote_literal(text*iptr);
25+
26+
27+
/*
28+
* quote_ident -
29+
* returns a properly quoted identifier
30+
*/
31+
Datum
32+
quote_ident(PG_FUNCTION_ARGS)
33+
{
34+
text*t=PG_GETARG_TEXT_P(0);
35+
text*result;
36+
37+
if (quote_ident_required(t))
38+
{
39+
result=do_quote_ident(t);
40+
}
41+
else
42+
{
43+
result= (text*)palloc(VARSIZE(t));
44+
memcpy(result,t,VARSIZE(t));
45+
}
46+
47+
PG_FREE_IF_COPY(t,0);
48+
49+
returnresult;
50+
}
51+
52+
/*
53+
* quote_literal -
54+
* returns a properly quoted literal
55+
*/
56+
Datum
57+
quote_literal(PG_FUNCTION_ARGS)
58+
{
59+
text*t=PG_GETARG_TEXT_P(0);
60+
text*result;
61+
62+
result=do_quote_literal(t);
63+
64+
PG_FREE_IF_COPY(t,0);
65+
66+
returnresult;
67+
}
68+
69+
70+
/*
71+
* MULTIBYTE dependant internal functions follow
72+
*
73+
*/
74+
75+
76+
#ifndefMULTIBYTE
77+
78+
/* Check if a given identifier needs quoting */
79+
staticbool
80+
quote_ident_required(text*iptr)
81+
{
82+
char*cp;
83+
char*ep;
84+
85+
cp=VARDATA(iptr);
86+
ep=VARDATA(iptr)+VARSIZE(iptr)-VARHDRSZ;
87+
88+
if (cp >=ep)
89+
return true;
90+
91+
if (!(*cp=='_'|| (*cp >='a'&&*cp <='z')))
92+
return true;
93+
94+
while((++cp)<ep)
95+
{
96+
if (*cp >='a'&&*cp <='z')continue;
97+
if (*cp >='0'&&*cp <='9')continue;
98+
if (*cp=='_')continue;
99+
100+
return true;
101+
}
102+
103+
return false;
104+
}
105+
106+
/* Return a properly quoted identifier */
107+
statictext*
108+
do_quote_ident(text*iptr)
109+
{
110+
text*result;
111+
char*cp1;
112+
char*cp2;
113+
intlen;
114+
115+
len=VARSIZE(iptr)-VARHDRSZ;
116+
result= (text*)palloc(len*2+VARHDRSZ+2);
117+
118+
cp1=VARDATA(iptr);
119+
cp2=VARDATA(result);
120+
121+
*cp2++='"';
122+
while(len-->0)
123+
{
124+
if (*cp1=='"')
125+
*cp2++='"';
126+
if (*cp1=='\\')
127+
*cp2++='\\';
128+
*cp2++=*cp1++;
129+
}
130+
*cp2++='"';
131+
132+
VARATT_SIZEP(result)=cp2- ((char*)result);
133+
134+
returnresult;
135+
}
136+
137+
/* Return a properly quoted literal value */
138+
statictext*
139+
do_quote_literal(text*lptr)
140+
{
141+
text*result;
142+
char*cp1;
143+
char*cp2;
144+
intlen;
145+
146+
len=VARSIZE(lptr)-VARHDRSZ;
147+
result= (text*)palloc(len*2+VARHDRSZ+2);
148+
149+
cp1=VARDATA(lptr);
150+
cp2=VARDATA(result);
151+
152+
*cp2++='\'';
153+
while(len-->0)
154+
{
155+
if (*cp1=='\'')
156+
*cp2++='\'';
157+
if (*cp1=='\\')
158+
*cp2++='\\';
159+
*cp2++=*cp1++;
160+
}
161+
*cp2++='\'';
162+
163+
VARATT_SIZEP(result)=cp2- ((char*)result);
164+
165+
returnresult;
166+
}
167+
168+
#else
169+
170+
/* Check if a given identifier needs quoting (MULTIBYTE version) */
171+
staticbool
172+
quote_ident_required(text*iptr)
173+
{
174+
char*cp;
175+
char*ep;
176+
177+
cp=VARDATA(iptr);
178+
ep=VARDATA(iptr)+VARSIZE(iptr)-VARHDRSZ;
179+
180+
if (cp >=ep)
181+
return true;
182+
183+
if(pg_mblen(cp)!=1)
184+
return true;
185+
if (!(*cp=='_'|| (*cp >='a'&&*cp <='z')))
186+
return true;
187+
188+
while((++cp)<ep)
189+
{
190+
if (pg_mblen(cp)!=1)
191+
return true;
192+
193+
if (*cp >='a'&&*cp <='z')continue;
194+
if (*cp >='0'&&*cp <='9')continue;
195+
if (*cp=='_')continue;
196+
197+
return true;
198+
}
199+
200+
return false;
201+
}
202+
203+
/* Return a properly quoted identifier (MULTIBYTE version) */
204+
statictext*
205+
do_quote_ident(text*iptr)
206+
{
207+
text*result;
208+
char*cp1;
209+
char*cp2;
210+
intlen;
211+
intwl;
212+
213+
len=VARSIZE(iptr)-VARHDRSZ;
214+
result= (text*)palloc(len*2+VARHDRSZ+2);
215+
216+
cp1=VARDATA(iptr);
217+
cp2=VARDATA(result);
218+
219+
*cp2++='"';
220+
while(len>0)
221+
{
222+
if ((wl=pg_mblen(cp1))!=1)
223+
{
224+
len-=wl;
225+
226+
while(wl-->0)
227+
*cp2++=*cp1++;
228+
continue;
229+
}
230+
231+
if (*cp1=='"')
232+
*cp2++='"';
233+
if (*cp1=='\\')
234+
*cp2++='\\';
235+
*cp2++=*cp1++;
236+
237+
len--;
238+
}
239+
*cp2++='"';
240+
241+
VARATT_SIZEP(result)=cp2- ((char*)result);
242+
243+
returnresult;
244+
}
245+
246+
/* Return a properly quoted literal value (MULTIBYTE version) */
247+
statictext*
248+
do_quote_literal(text*lptr)
249+
{
250+
text*result;
251+
char*cp1;
252+
char*cp2;
253+
intlen;
254+
intwl;
255+
256+
len=VARSIZE(lptr)-VARHDRSZ;
257+
result= (text*)palloc(len*2+VARHDRSZ+2);
258+
259+
cp1=VARDATA(lptr);
260+
cp2=VARDATA(result);
261+
262+
*cp2++='\'';
263+
while(len>0)
264+
{
265+
if ((wl=pg_mblen(cp1))!=1)
266+
{
267+
len-=wl;
268+
269+
while(wl-->0)
270+
*cp2++=*cp1++;
271+
continue;
272+
}
273+
274+
if (*cp1=='\'')
275+
*cp2++='\'';
276+
if (*cp1=='\\')
277+
*cp2++='\\';
278+
*cp2++=*cp1++;
279+
280+
len--;
281+
}
282+
*cp2++='\'';
283+
284+
VARATT_SIZEP(result)=cp2- ((char*)result);
285+
286+
returnresult;
287+
}
288+
289+
#endif
290+
291+

‎src/include/catalog/pg_proc.h

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
* Portions Copyright (c) 1996-2000, PostgreSQL, Inc
88
* Portions Copyright (c) 1994, Regents of the University of California
99
*
10-
* $Id: pg_proc.h,v 1.164 2000/08/24 03:29:08 tgl Exp $
10+
* $Id: pg_proc.h,v 1.165 2000/09/05 20:25:13 wieck Exp $
1111
*
1212
* NOTES
1313
* The script catalog/genbki.sh reads this file and generates .bki
@@ -2409,6 +2409,11 @@ DESCR("convert text to timestamp");
24092409
DATA(insertOID=1780 (to_datePGUID12ftft2f1082"25 25"10000100to_date- ));
24102410
DESCR("convert text to date");
24112411

2412+
DATA(insertOID=1282 (quote_identPGUID12fttt1f25"25"10000100quote_ident- ));
2413+
DESCR("quote an identifier for usage in a querystring");
2414+
DATA(insertOID=1283 (quote_literalPGUID12fttt1f25"25"10000100quote_literal- ));
2415+
DESCR("quote a literal for usage in a querystring");
2416+
24122417
DATA(insertOID=1798 (oidinPGUID12fttt1f26"0"10000100oidin- ));
24132418
DESCR("(internal)");
24142419
DATA(insertOID=1799 (oidoutPGUID12fttt1f23"0"10000100oidout- ));

‎src/include/utils/builtins.h

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
* Portions Copyright (c) 1996-2000, PostgreSQL, Inc
88
* Portions Copyright (c) 1994, Regents of the University of California
99
*
10-
* $Id: builtins.h,v 1.135 2000/08/25 18:05:53 tgl Exp $
10+
* $Id: builtins.h,v 1.136 2000/09/05 20:25:14 wieck Exp $
1111
*
1212
*-------------------------------------------------------------------------
1313
*/
@@ -585,4 +585,8 @@ extern Datum format_type(PG_FUNCTION_ARGS);
585585
externDatumoidvectortypes(PG_FUNCTION_ARGS);
586586
externint32type_maximum_size(Oidtype_oid,int32typemod);
587587

588+
/* quote.c */
589+
externDatumquote_ident(PG_FUNCTION_ARGS);
590+
externDatumquote_literal(PG_FUNCTION_ARGS);
591+
588592
#endif/* BUILTINS_H */

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp