Line data Source code
1 : /*-------------------------------------------------------------------------
2 : * oracle_compat.c
3 : * Oracle compatible functions.
4 : *
5 : * Copyright (c) 1996-2019, PostgreSQL Global Development Group
6 : *
7 : * Author: Edmund Mergl <E.Mergl@bawue.de>
8 : * Multibyte enhancement: Tatsuo Ishii <ishii@postgresql.org>
9 : *
10 : *
11 : * IDENTIFICATION
12 : * src/backend/utils/adt/oracle_compat.c
13 : *
14 : *-------------------------------------------------------------------------
15 : */
16 : #include "postgres.h"
17 :
18 : #include "common/int.h"
19 : #include "utils/builtins.h"
20 : #include "utils/formatting.h"
21 : #include "mb/pg_wchar.h"
22 :
23 :
24 : static text *dotrim(const char *string, int stringlen,
25 : const char *set, int setlen,
26 : bool doltrim, bool dortrim);
27 :
28 :
29 : /********************************************************************
30 : *
31 : * lower
32 : *
33 : * Syntax:
34 : *
35 : * text lower(text string)
36 : *
37 : * Purpose:
38 : *
39 : * Returns string, with all letters forced to lowercase.
40 : *
41 : ********************************************************************/
42 :
43 : Datum
44 10676 : lower(PG_FUNCTION_ARGS)
45 : {
46 10676 : text *in_string = PG_GETARG_TEXT_PP(0);
47 : char *out_string;
48 : text *result;
49 :
50 32028 : out_string = str_tolower(VARDATA_ANY(in_string),
51 21352 : VARSIZE_ANY_EXHDR(in_string),
52 : PG_GET_COLLATION());
53 10676 : result = cstring_to_text(out_string);
54 10676 : pfree(out_string);
55 :
56 10676 : PG_RETURN_TEXT_P(result);
57 : }
58 :
59 :
60 : /********************************************************************
61 : *
62 : * upper
63 : *
64 : * Syntax:
65 : *
66 : * text upper(text string)
67 : *
68 : * Purpose:
69 : *
70 : * Returns string, with all letters forced to uppercase.
71 : *
72 : ********************************************************************/
73 :
74 : Datum
75 448 : upper(PG_FUNCTION_ARGS)
76 : {
77 448 : text *in_string = PG_GETARG_TEXT_PP(0);
78 : char *out_string;
79 : text *result;
80 :
81 1344 : out_string = str_toupper(VARDATA_ANY(in_string),
82 896 : VARSIZE_ANY_EXHDR(in_string),
83 : PG_GET_COLLATION());
84 448 : result = cstring_to_text(out_string);
85 448 : pfree(out_string);
86 :
87 448 : PG_RETURN_TEXT_P(result);
88 : }
89 :
90 :
91 : /********************************************************************
92 : *
93 : * initcap
94 : *
95 : * Syntax:
96 : *
97 : * text initcap(text string)
98 : *
99 : * Purpose:
100 : *
101 : * Returns string, with first letter of each word in uppercase, all
102 : * other letters in lowercase. A word is defined as a sequence of
103 : * alphanumeric characters, delimited by non-alphanumeric
104 : * characters.
105 : *
106 : ********************************************************************/
107 :
108 : Datum
109 24 : initcap(PG_FUNCTION_ARGS)
110 : {
111 24 : text *in_string = PG_GETARG_TEXT_PP(0);
112 : char *out_string;
113 : text *result;
114 :
115 72 : out_string = str_initcap(VARDATA_ANY(in_string),
116 48 : VARSIZE_ANY_EXHDR(in_string),
117 : PG_GET_COLLATION());
118 24 : result = cstring_to_text(out_string);
119 24 : pfree(out_string);
120 :
121 24 : PG_RETURN_TEXT_P(result);
122 : }
123 :
124 :
125 : /********************************************************************
126 : *
127 : * lpad
128 : *
129 : * Syntax:
130 : *
131 : * text lpad(text string1, int4 len, text string2)
132 : *
133 : * Purpose:
134 : *
135 : * Returns string1, left-padded to length len with the sequence of
136 : * characters in string2. If len is less than the length of string1,
137 : * instead truncate (on the right) to len.
138 : *
139 : ********************************************************************/
140 :
141 : Datum
142 28 : lpad(PG_FUNCTION_ARGS)
143 : {
144 28 : text *string1 = PG_GETARG_TEXT_PP(0);
145 28 : int32 len = PG_GETARG_INT32(1);
146 28 : text *string2 = PG_GETARG_TEXT_PP(2);
147 : text *ret;
148 : char *ptr1,
149 : *ptr2,
150 : *ptr2start,
151 : *ptr2end,
152 : *ptr_ret;
153 : int m,
154 : s1len,
155 : s2len;
156 :
157 : int bytelen;
158 :
159 : /* Negative len is silently taken as zero */
160 28 : if (len < 0)
161 4 : len = 0;
162 :
163 28 : s1len = VARSIZE_ANY_EXHDR(string1);
164 28 : if (s1len < 0)
165 0 : s1len = 0; /* shouldn't happen */
166 :
167 28 : s2len = VARSIZE_ANY_EXHDR(string2);
168 28 : if (s2len < 0)
169 0 : s2len = 0; /* shouldn't happen */
170 :
171 28 : s1len = pg_mbstrlen_with_len(VARDATA_ANY(string1), s1len);
172 :
173 28 : if (s1len > len)
174 8 : s1len = len; /* truncate string1 to len chars */
175 :
176 28 : if (s2len <= 0)
177 4 : len = s1len; /* nothing to pad with, so don't pad */
178 :
179 28 : bytelen = pg_database_encoding_max_length() * len;
180 :
181 : /* check for integer overflow */
182 28 : if (len != 0 && bytelen / pg_database_encoding_max_length() != len)
183 0 : ereport(ERROR,
184 : (errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
185 : errmsg("requested length too large")));
186 :
187 28 : ret = (text *) palloc(VARHDRSZ + bytelen);
188 :
189 28 : m = len - s1len;
190 :
191 28 : ptr2 = ptr2start = VARDATA_ANY(string2);
192 28 : ptr2end = ptr2 + s2len;
193 28 : ptr_ret = VARDATA(ret);
194 :
195 104 : while (m--)
196 : {
197 48 : int mlen = pg_mblen(ptr2);
198 :
199 48 : memcpy(ptr_ret, ptr2, mlen);
200 48 : ptr_ret += mlen;
201 48 : ptr2 += mlen;
202 48 : if (ptr2 == ptr2end) /* wrap around at end of s2 */
203 28 : ptr2 = ptr2start;
204 : }
205 :
206 28 : ptr1 = VARDATA_ANY(string1);
207 :
208 104 : while (s1len--)
209 : {
210 48 : int mlen = pg_mblen(ptr1);
211 :
212 48 : memcpy(ptr_ret, ptr1, mlen);
213 48 : ptr_ret += mlen;
214 48 : ptr1 += mlen;
215 : }
216 :
217 28 : SET_VARSIZE(ret, ptr_ret - (char *) ret);
218 :
219 28 : PG_RETURN_TEXT_P(ret);
220 : }
221 :
222 :
223 : /********************************************************************
224 : *
225 : * rpad
226 : *
227 : * Syntax:
228 : *
229 : * text rpad(text string1, int4 len, text string2)
230 : *
231 : * Purpose:
232 : *
233 : * Returns string1, right-padded to length len with the sequence of
234 : * characters in string2. If len is less than the length of string1,
235 : * instead truncate (on the right) to len.
236 : *
237 : ********************************************************************/
238 :
239 : Datum
240 32 : rpad(PG_FUNCTION_ARGS)
241 : {
242 32 : text *string1 = PG_GETARG_TEXT_PP(0);
243 32 : int32 len = PG_GETARG_INT32(1);
244 32 : text *string2 = PG_GETARG_TEXT_PP(2);
245 : text *ret;
246 : char *ptr1,
247 : *ptr2,
248 : *ptr2start,
249 : *ptr2end,
250 : *ptr_ret;
251 : int m,
252 : s1len,
253 : s2len;
254 :
255 : int bytelen;
256 :
257 : /* Negative len is silently taken as zero */
258 32 : if (len < 0)
259 4 : len = 0;
260 :
261 32 : s1len = VARSIZE_ANY_EXHDR(string1);
262 32 : if (s1len < 0)
263 0 : s1len = 0; /* shouldn't happen */
264 :
265 32 : s2len = VARSIZE_ANY_EXHDR(string2);
266 32 : if (s2len < 0)
267 0 : s2len = 0; /* shouldn't happen */
268 :
269 32 : s1len = pg_mbstrlen_with_len(VARDATA_ANY(string1), s1len);
270 :
271 32 : if (s1len > len)
272 8 : s1len = len; /* truncate string1 to len chars */
273 :
274 32 : if (s2len <= 0)
275 4 : len = s1len; /* nothing to pad with, so don't pad */
276 :
277 32 : bytelen = pg_database_encoding_max_length() * len;
278 :
279 : /* Check for integer overflow */
280 32 : if (len != 0 && bytelen / pg_database_encoding_max_length() != len)
281 0 : ereport(ERROR,
282 : (errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
283 : errmsg("requested length too large")));
284 :
285 32 : ret = (text *) palloc(VARHDRSZ + bytelen);
286 32 : m = len - s1len;
287 :
288 32 : ptr1 = VARDATA_ANY(string1);
289 32 : ptr_ret = VARDATA(ret);
290 :
291 112 : while (s1len--)
292 : {
293 48 : int mlen = pg_mblen(ptr1);
294 :
295 48 : memcpy(ptr_ret, ptr1, mlen);
296 48 : ptr_ret += mlen;
297 48 : ptr1 += mlen;
298 : }
299 :
300 32 : ptr2 = ptr2start = VARDATA_ANY(string2);
301 32 : ptr2end = ptr2 + s2len;
302 :
303 1280112 : while (m--)
304 : {
305 1280048 : int mlen = pg_mblen(ptr2);
306 :
307 1280048 : memcpy(ptr_ret, ptr2, mlen);
308 1280048 : ptr_ret += mlen;
309 1280048 : ptr2 += mlen;
310 1280048 : if (ptr2 == ptr2end) /* wrap around at end of s2 */
311 1280028 : ptr2 = ptr2start;
312 : }
313 :
314 32 : SET_VARSIZE(ret, ptr_ret - (char *) ret);
315 :
316 32 : PG_RETURN_TEXT_P(ret);
317 : }
318 :
319 :
320 : /********************************************************************
321 : *
322 : * btrim
323 : *
324 : * Syntax:
325 : *
326 : * text btrim(text string, text set)
327 : *
328 : * Purpose:
329 : *
330 : * Returns string with characters removed from the front and back
331 : * up to the first character not in set.
332 : *
333 : ********************************************************************/
334 :
335 : Datum
336 18 : btrim(PG_FUNCTION_ARGS)
337 : {
338 18 : text *string = PG_GETARG_TEXT_PP(0);
339 18 : text *set = PG_GETARG_TEXT_PP(1);
340 : text *ret;
341 :
342 72 : ret = dotrim(VARDATA_ANY(string), VARSIZE_ANY_EXHDR(string),
343 72 : VARDATA_ANY(set), VARSIZE_ANY_EXHDR(set),
344 : true, true);
345 :
346 18 : PG_RETURN_TEXT_P(ret);
347 : }
348 :
349 : /********************************************************************
350 : *
351 : * btrim1 --- btrim with set fixed as ' '
352 : *
353 : ********************************************************************/
354 :
355 : Datum
356 376 : btrim1(PG_FUNCTION_ARGS)
357 : {
358 376 : text *string = PG_GETARG_TEXT_PP(0);
359 : text *ret;
360 :
361 376 : ret = dotrim(VARDATA_ANY(string), VARSIZE_ANY_EXHDR(string),
362 : " ", 1,
363 : true, true);
364 :
365 376 : PG_RETURN_TEXT_P(ret);
366 : }
367 :
368 : /*
369 : * Common implementation for btrim, ltrim, rtrim
370 : */
371 : static text *
372 30380 : dotrim(const char *string, int stringlen,
373 : const char *set, int setlen,
374 : bool doltrim, bool dortrim)
375 : {
376 : int i;
377 :
378 : /* Nothing to do if either string or set is empty */
379 30380 : if (stringlen > 0 && setlen > 0)
380 : {
381 30380 : if (pg_database_encoding_max_length() > 1)
382 : {
383 : /*
384 : * In the multibyte-encoding case, build arrays of pointers to
385 : * character starts, so that we can avoid inefficient checks in
386 : * the inner loops.
387 : */
388 : const char **stringchars;
389 : const char **setchars;
390 : int *stringmblen;
391 : int *setmblen;
392 : int stringnchars;
393 : int setnchars;
394 : int resultndx;
395 : int resultnchars;
396 : const char *p;
397 : int len;
398 : int mblen;
399 : const char *str_pos;
400 : int str_len;
401 :
402 30368 : stringchars = (const char **) palloc(stringlen * sizeof(char *));
403 30368 : stringmblen = (int *) palloc(stringlen * sizeof(int));
404 30368 : stringnchars = 0;
405 30368 : p = string;
406 30368 : len = stringlen;
407 458920 : while (len > 0)
408 : {
409 398184 : stringchars[stringnchars] = p;
410 398184 : stringmblen[stringnchars] = mblen = pg_mblen(p);
411 398184 : stringnchars++;
412 398184 : p += mblen;
413 398184 : len -= mblen;
414 : }
415 :
416 30368 : setchars = (const char **) palloc(setlen * sizeof(char *));
417 30368 : setmblen = (int *) palloc(setlen * sizeof(int));
418 30368 : setnchars = 0;
419 30368 : p = set;
420 30368 : len = setlen;
421 91130 : while (len > 0)
422 : {
423 30394 : setchars[setnchars] = p;
424 30394 : setmblen[setnchars] = mblen = pg_mblen(p);
425 30394 : setnchars++;
426 30394 : p += mblen;
427 30394 : len -= mblen;
428 : }
429 :
430 30368 : resultndx = 0; /* index in stringchars[] */
431 30368 : resultnchars = stringnchars;
432 :
433 30368 : if (doltrim)
434 : {
435 56780 : while (resultnchars > 0)
436 : {
437 37770 : str_pos = stringchars[resultndx];
438 37770 : str_len = stringmblen[resultndx];
439 56882 : for (i = 0; i < setnchars; i++)
440 : {
441 75744 : if (str_len == setmblen[i] &&
442 37872 : memcmp(str_pos, setchars[i], str_len) == 0)
443 18760 : break;
444 : }
445 37770 : if (i >= setnchars)
446 19010 : break; /* no match here */
447 18760 : string += str_len;
448 18760 : stringlen -= str_len;
449 18760 : resultndx++;
450 18760 : resultnchars--;
451 : }
452 : }
453 :
454 30368 : if (dortrim)
455 : {
456 253734 : while (resultnchars > 0)
457 : {
458 241040 : str_pos = stringchars[resultndx + resultnchars - 1];
459 241040 : str_len = stringmblen[resultndx + resultnchars - 1];
460 251868 : for (i = 0; i < setnchars; i++)
461 : {
462 482116 : if (str_len == setmblen[i] &&
463 241058 : memcmp(str_pos, setchars[i], str_len) == 0)
464 230230 : break;
465 : }
466 241040 : if (i >= setnchars)
467 10810 : break; /* no match here */
468 230230 : stringlen -= str_len;
469 230230 : resultnchars--;
470 : }
471 : }
472 :
473 30368 : pfree(stringchars);
474 30368 : pfree(stringmblen);
475 30368 : pfree(setchars);
476 30368 : pfree(setmblen);
477 : }
478 : else
479 : {
480 : /*
481 : * In the single-byte-encoding case, we don't need such overhead.
482 : */
483 12 : if (doltrim)
484 : {
485 0 : while (stringlen > 0)
486 : {
487 0 : char str_ch = *string;
488 :
489 0 : for (i = 0; i < setlen; i++)
490 : {
491 0 : if (str_ch == set[i])
492 0 : break;
493 : }
494 0 : if (i >= setlen)
495 0 : break; /* no match here */
496 0 : string++;
497 0 : stringlen--;
498 : }
499 : }
500 :
501 12 : if (dortrim)
502 : {
503 36 : while (stringlen > 0)
504 : {
505 24 : char str_ch = string[stringlen - 1];
506 :
507 36 : for (i = 0; i < setlen; i++)
508 : {
509 24 : if (str_ch == set[i])
510 12 : break;
511 : }
512 24 : if (i >= setlen)
513 12 : break; /* no match here */
514 12 : stringlen--;
515 : }
516 : }
517 : }
518 : }
519 :
520 : /* Return selected portion of string */
521 30380 : return cstring_to_text_with_len(string, stringlen);
522 : }
523 :
524 : /********************************************************************
525 : *
526 : * byteatrim
527 : *
528 : * Syntax:
529 : *
530 : * bytea byteatrim(byta string, bytea set)
531 : *
532 : * Purpose:
533 : *
534 : * Returns string with characters removed from the front and back
535 : * up to the first character not in set.
536 : *
537 : * Cloned from btrim and modified as required.
538 : ********************************************************************/
539 :
540 : Datum
541 16 : byteatrim(PG_FUNCTION_ARGS)
542 : {
543 16 : bytea *string = PG_GETARG_BYTEA_PP(0);
544 16 : bytea *set = PG_GETARG_BYTEA_PP(1);
545 : bytea *ret;
546 : char *ptr,
547 : *end,
548 : *ptr2,
549 : *ptr2start,
550 : *end2;
551 : int m,
552 : stringlen,
553 : setlen;
554 :
555 16 : stringlen = VARSIZE_ANY_EXHDR(string);
556 16 : setlen = VARSIZE_ANY_EXHDR(set);
557 :
558 16 : if (stringlen <= 0 || setlen <= 0)
559 8 : PG_RETURN_BYTEA_P(string);
560 :
561 8 : m = stringlen;
562 8 : ptr = VARDATA_ANY(string);
563 8 : end = ptr + stringlen - 1;
564 8 : ptr2start = VARDATA_ANY(set);
565 8 : end2 = ptr2start + setlen - 1;
566 :
567 24 : while (m > 0)
568 : {
569 16 : ptr2 = ptr2start;
570 40 : while (ptr2 <= end2)
571 : {
572 16 : if (*ptr == *ptr2)
573 8 : break;
574 8 : ++ptr2;
575 : }
576 16 : if (ptr2 > end2)
577 8 : break;
578 8 : ptr++;
579 8 : m--;
580 : }
581 :
582 24 : while (m > 0)
583 : {
584 16 : ptr2 = ptr2start;
585 40 : while (ptr2 <= end2)
586 : {
587 16 : if (*end == *ptr2)
588 8 : break;
589 8 : ++ptr2;
590 : }
591 16 : if (ptr2 > end2)
592 8 : break;
593 8 : end--;
594 8 : m--;
595 : }
596 :
597 8 : ret = (bytea *) palloc(VARHDRSZ + m);
598 8 : SET_VARSIZE(ret, VARHDRSZ + m);
599 8 : memcpy(VARDATA(ret), ptr, m);
600 :
601 8 : PG_RETURN_BYTEA_P(ret);
602 : }
603 :
604 : /********************************************************************
605 : *
606 : * ltrim
607 : *
608 : * Syntax:
609 : *
610 : * text ltrim(text string, text set)
611 : *
612 : * Purpose:
613 : *
614 : * Returns string with initial characters removed up to the first
615 : * character not in set.
616 : *
617 : ********************************************************************/
618 :
619 : Datum
620 18610 : ltrim(PG_FUNCTION_ARGS)
621 : {
622 18610 : text *string = PG_GETARG_TEXT_PP(0);
623 18610 : text *set = PG_GETARG_TEXT_PP(1);
624 : text *ret;
625 :
626 74440 : ret = dotrim(VARDATA_ANY(string), VARSIZE_ANY_EXHDR(string),
627 74440 : VARDATA_ANY(set), VARSIZE_ANY_EXHDR(set),
628 : true, false);
629 :
630 18610 : PG_RETURN_TEXT_P(ret);
631 : }
632 :
633 : /********************************************************************
634 : *
635 : * ltrim1 --- ltrim with set fixed as ' '
636 : *
637 : ********************************************************************/
638 :
639 : Datum
640 6 : ltrim1(PG_FUNCTION_ARGS)
641 : {
642 6 : text *string = PG_GETARG_TEXT_PP(0);
643 : text *ret;
644 :
645 6 : ret = dotrim(VARDATA_ANY(string), VARSIZE_ANY_EXHDR(string),
646 : " ", 1,
647 : true, false);
648 :
649 6 : PG_RETURN_TEXT_P(ret);
650 : }
651 :
652 : /********************************************************************
653 : *
654 : * rtrim
655 : *
656 : * Syntax:
657 : *
658 : * text rtrim(text string, text set)
659 : *
660 : * Purpose:
661 : *
662 : * Returns string with final characters removed after the last
663 : * character not in set.
664 : *
665 : ********************************************************************/
666 :
667 : Datum
668 380 : rtrim(PG_FUNCTION_ARGS)
669 : {
670 380 : text *string = PG_GETARG_TEXT_PP(0);
671 380 : text *set = PG_GETARG_TEXT_PP(1);
672 : text *ret;
673 :
674 1520 : ret = dotrim(VARDATA_ANY(string), VARSIZE_ANY_EXHDR(string),
675 1520 : VARDATA_ANY(set), VARSIZE_ANY_EXHDR(set),
676 : false, true);
677 :
678 380 : PG_RETURN_TEXT_P(ret);
679 : }
680 :
681 : /********************************************************************
682 : *
683 : * rtrim1 --- rtrim with set fixed as ' '
684 : *
685 : ********************************************************************/
686 :
687 : Datum
688 10990 : rtrim1(PG_FUNCTION_ARGS)
689 : {
690 10990 : text *string = PG_GETARG_TEXT_PP(0);
691 : text *ret;
692 :
693 10990 : ret = dotrim(VARDATA_ANY(string), VARSIZE_ANY_EXHDR(string),
694 : " ", 1,
695 : false, true);
696 :
697 10990 : PG_RETURN_TEXT_P(ret);
698 : }
699 :
700 :
701 : /********************************************************************
702 : *
703 : * translate
704 : *
705 : * Syntax:
706 : *
707 : * text translate(text string, text from, text to)
708 : *
709 : * Purpose:
710 : *
711 : * Returns string after replacing all occurrences of characters in from
712 : * with the corresponding character in to. If from is longer than to,
713 : * occurrences of the extra characters in from are deleted.
714 : * Improved by Edwin Ramirez <ramirez@doc.mssm.edu>.
715 : *
716 : ********************************************************************/
717 :
718 : Datum
719 24 : translate(PG_FUNCTION_ARGS)
720 : {
721 24 : text *string = PG_GETARG_TEXT_PP(0);
722 24 : text *from = PG_GETARG_TEXT_PP(1);
723 24 : text *to = PG_GETARG_TEXT_PP(2);
724 : text *result;
725 : char *from_ptr,
726 : *to_ptr;
727 : char *source,
728 : *target;
729 : int m,
730 : fromlen,
731 : tolen,
732 : retlen,
733 : i;
734 : int worst_len;
735 : int len;
736 : int source_len;
737 : int from_index;
738 :
739 24 : m = VARSIZE_ANY_EXHDR(string);
740 24 : if (m <= 0)
741 4 : PG_RETURN_TEXT_P(string);
742 20 : source = VARDATA_ANY(string);
743 :
744 20 : fromlen = VARSIZE_ANY_EXHDR(from);
745 20 : from_ptr = VARDATA_ANY(from);
746 20 : tolen = VARSIZE_ANY_EXHDR(to);
747 20 : to_ptr = VARDATA_ANY(to);
748 :
749 : /*
750 : * The worst-case expansion is to substitute a max-length character for a
751 : * single-byte character at each position of the string.
752 : */
753 20 : worst_len = pg_database_encoding_max_length() * m;
754 :
755 : /* check for integer overflow */
756 20 : if (worst_len / pg_database_encoding_max_length() != m)
757 0 : ereport(ERROR,
758 : (errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
759 : errmsg("requested length too large")));
760 :
761 20 : result = (text *) palloc(worst_len + VARHDRSZ);
762 20 : target = VARDATA(result);
763 20 : retlen = 0;
764 :
765 252 : while (m > 0)
766 : {
767 212 : source_len = pg_mblen(source);
768 212 : from_index = 0;
769 :
770 576 : for (i = 0; i < fromlen; i += len)
771 : {
772 404 : len = pg_mblen(&from_ptr[i]);
773 808 : if (len == source_len &&
774 404 : memcmp(source, &from_ptr[i], len) == 0)
775 40 : break;
776 :
777 364 : from_index++;
778 : }
779 212 : if (i < fromlen)
780 : {
781 : /* substitute */
782 40 : char *p = to_ptr;
783 :
784 60 : for (i = 0; i < from_index; i++)
785 : {
786 20 : p += pg_mblen(p);
787 20 : if (p >= (to_ptr + tolen))
788 0 : break;
789 : }
790 40 : if (p < (to_ptr + tolen))
791 : {
792 40 : len = pg_mblen(p);
793 40 : memcpy(target, p, len);
794 40 : target += len;
795 40 : retlen += len;
796 : }
797 :
798 : }
799 : else
800 : {
801 : /* no match, so copy */
802 172 : memcpy(target, source, source_len);
803 172 : target += source_len;
804 172 : retlen += source_len;
805 : }
806 :
807 212 : source += source_len;
808 212 : m -= source_len;
809 : }
810 :
811 20 : SET_VARSIZE(result, retlen + VARHDRSZ);
812 :
813 : /*
814 : * The function result is probably much bigger than needed, if we're using
815 : * a multibyte encoding, but it's not worth reallocating it; the result
816 : * probably won't live long anyway.
817 : */
818 :
819 20 : PG_RETURN_TEXT_P(result);
820 : }
821 :
822 : /********************************************************************
823 : *
824 : * ascii
825 : *
826 : * Syntax:
827 : *
828 : * int ascii(text string)
829 : *
830 : * Purpose:
831 : *
832 : * Returns the decimal representation of the first character from
833 : * string.
834 : * If the string is empty we return 0.
835 : * If the database encoding is UTF8, we return the Unicode codepoint.
836 : * If the database encoding is any other multi-byte encoding, we
837 : * return the value of the first byte if it is an ASCII character
838 : * (range 1 .. 127), or raise an error.
839 : * For all other encodings we return the value of the first byte,
840 : * (range 1..255).
841 : *
842 : ********************************************************************/
843 :
844 : Datum
845 44 : ascii(PG_FUNCTION_ARGS)
846 : {
847 44 : text *string = PG_GETARG_TEXT_PP(0);
848 44 : int encoding = GetDatabaseEncoding();
849 : unsigned char *data;
850 :
851 44 : if (VARSIZE_ANY_EXHDR(string) <= 0)
852 4 : PG_RETURN_INT32(0);
853 :
854 40 : data = (unsigned char *) VARDATA_ANY(string);
855 :
856 40 : if (encoding == PG_UTF8 && *data > 127)
857 : {
858 : /* return the code point for Unicode */
859 :
860 0 : int result = 0,
861 0 : tbytes = 0,
862 : i;
863 :
864 0 : if (*data >= 0xF0)
865 : {
866 0 : result = *data & 0x07;
867 0 : tbytes = 3;
868 : }
869 0 : else if (*data >= 0xE0)
870 : {
871 0 : result = *data & 0x0F;
872 0 : tbytes = 2;
873 : }
874 : else
875 : {
876 : Assert(*data > 0xC0);
877 0 : result = *data & 0x1f;
878 0 : tbytes = 1;
879 : }
880 :
881 : Assert(tbytes > 0);
882 :
883 0 : for (i = 1; i <= tbytes; i++)
884 : {
885 : Assert((data[i] & 0xC0) == 0x80);
886 0 : result = (result << 6) + (data[i] & 0x3f);
887 : }
888 :
889 0 : PG_RETURN_INT32(result);
890 : }
891 : else
892 : {
893 40 : if (pg_encoding_max_length(encoding) > 1 && *data > 127)
894 0 : ereport(ERROR,
895 : (errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
896 : errmsg("requested character too large")));
897 :
898 :
899 40 : PG_RETURN_INT32((int32) *data);
900 : }
901 : }
902 :
903 : /********************************************************************
904 : *
905 : * chr
906 : *
907 : * Syntax:
908 : *
909 : * text chr(int val)
910 : *
911 : * Purpose:
912 : *
913 : * Returns the character having the binary equivalent to val.
914 : *
915 : * For UTF8 we treat the argument as a Unicode code point.
916 : * For other multi-byte encodings we raise an error for arguments
917 : * outside the strict ASCII range (1..127).
918 : *
919 : * It's important that we don't ever return a value that is not valid
920 : * in the database encoding, so that this doesn't become a way for
921 : * invalid data to enter the database.
922 : *
923 : ********************************************************************/
924 :
925 : Datum
926 543214 : chr (PG_FUNCTION_ARGS)
927 : {
928 543214 : uint32 cvalue = PG_GETARG_UINT32(0);
929 : text *result;
930 543214 : int encoding = GetDatabaseEncoding();
931 :
932 543214 : if (encoding == PG_UTF8 && cvalue > 127)
933 0 : {
934 : /* for Unicode we treat the argument as a code point */
935 : int bytes;
936 : unsigned char *wch;
937 :
938 : /*
939 : * We only allow valid Unicode code points; per RFC3629 that stops at
940 : * U+10FFFF, even though 4-byte UTF8 sequences can hold values up to
941 : * U+1FFFFF.
942 : */
943 0 : if (cvalue > 0x0010ffff)
944 0 : ereport(ERROR,
945 : (errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
946 : errmsg("requested character too large for encoding: %d",
947 : cvalue)));
948 :
949 0 : if (cvalue > 0xffff)
950 0 : bytes = 4;
951 0 : else if (cvalue > 0x07ff)
952 0 : bytes = 3;
953 : else
954 0 : bytes = 2;
955 :
956 0 : result = (text *) palloc(VARHDRSZ + bytes);
957 0 : SET_VARSIZE(result, VARHDRSZ + bytes);
958 0 : wch = (unsigned char *) VARDATA(result);
959 :
960 0 : if (bytes == 2)
961 : {
962 0 : wch[0] = 0xC0 | ((cvalue >> 6) & 0x1F);
963 0 : wch[1] = 0x80 | (cvalue & 0x3F);
964 : }
965 0 : else if (bytes == 3)
966 : {
967 0 : wch[0] = 0xE0 | ((cvalue >> 12) & 0x0F);
968 0 : wch[1] = 0x80 | ((cvalue >> 6) & 0x3F);
969 0 : wch[2] = 0x80 | (cvalue & 0x3F);
970 : }
971 : else
972 : {
973 0 : wch[0] = 0xF0 | ((cvalue >> 18) & 0x07);
974 0 : wch[1] = 0x80 | ((cvalue >> 12) & 0x3F);
975 0 : wch[2] = 0x80 | ((cvalue >> 6) & 0x3F);
976 0 : wch[3] = 0x80 | (cvalue & 0x3F);
977 : }
978 :
979 : /*
980 : * The preceding range check isn't sufficient, because UTF8 excludes
981 : * Unicode "surrogate pair" codes. Make sure what we created is valid
982 : * UTF8.
983 : */
984 0 : if (!pg_utf8_islegal(wch, bytes))
985 0 : ereport(ERROR,
986 : (errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
987 : errmsg("requested character not valid for encoding: %d",
988 : cvalue)));
989 : }
990 : else
991 : {
992 : bool is_mb;
993 :
994 : /*
995 : * Error out on arguments that make no sense or that we can't validly
996 : * represent in the encoding.
997 : */
998 543214 : if (cvalue == 0)
999 4 : ereport(ERROR,
1000 : (errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
1001 : errmsg("null character not permitted")));
1002 :
1003 543210 : is_mb = pg_encoding_max_length(encoding) > 1;
1004 :
1005 543210 : if ((is_mb && (cvalue > 127)) || (!is_mb && (cvalue > 255)))
1006 0 : ereport(ERROR,
1007 : (errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
1008 : errmsg("requested character too large for encoding: %d",
1009 : cvalue)));
1010 :
1011 543210 : result = (text *) palloc(VARHDRSZ + 1);
1012 543210 : SET_VARSIZE(result, VARHDRSZ + 1);
1013 543210 : *VARDATA(result) = (char) cvalue;
1014 : }
1015 :
1016 543210 : PG_RETURN_TEXT_P(result);
1017 : }
1018 :
1019 : /********************************************************************
1020 : *
1021 : * repeat
1022 : *
1023 : * Syntax:
1024 : *
1025 : * text repeat(text string, int val)
1026 : *
1027 : * Purpose:
1028 : *
1029 : * Repeat string by val.
1030 : *
1031 : ********************************************************************/
1032 :
1033 : Datum
1034 8148 : repeat(PG_FUNCTION_ARGS)
1035 : {
1036 8148 : text *string = PG_GETARG_TEXT_PP(0);
1037 8148 : int32 count = PG_GETARG_INT32(1);
1038 : text *result;
1039 : int slen,
1040 : tlen;
1041 : int i;
1042 : char *cp,
1043 : *sp;
1044 :
1045 8148 : if (count < 0)
1046 4 : count = 0;
1047 :
1048 8148 : slen = VARSIZE_ANY_EXHDR(string);
1049 :
1050 16296 : if (unlikely(pg_mul_s32_overflow(count, slen, &tlen)) ||
1051 8148 : unlikely(pg_add_s32_overflow(tlen, VARHDRSZ, &tlen)))
1052 0 : ereport(ERROR,
1053 : (errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
1054 : errmsg("requested length too large")));
1055 :
1056 8148 : result = (text *) palloc(tlen);
1057 :
1058 8148 : SET_VARSIZE(result, tlen);
1059 8148 : cp = VARDATA(result);
1060 8148 : sp = VARDATA_ANY(string);
1061 3036480 : for (i = 0; i < count; i++)
1062 : {
1063 3028332 : memcpy(cp, sp, slen);
1064 3028332 : cp += slen;
1065 : }
1066 :
1067 8148 : PG_RETURN_TEXT_P(result);
1068 : }
|
