Age Owner Branch data TLA Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * GB18030 <--> UTF8
4 : : *
5 : : * Portions Copyright (c) 1996-2025, PostgreSQL Global Development Group
6 : : * Portions Copyright (c) 1994, Regents of the University of California
7 : : *
8 : : * IDENTIFICATION
9 : : * src/backend/utils/mb/conversion_procs/utf8_and_gb18030/utf8_and_gb18030.c
10 : : *
11 : : *-------------------------------------------------------------------------
12 : : */
13 : :
14 : : #include "postgres.h"
15 : : #include "fmgr.h"
16 : : #include "mb/pg_wchar.h"
17 : : #include "../../Unicode/gb18030_to_utf8.map"
18 : : #include "../../Unicode/utf8_to_gb18030.map"
19 : :
166 tgl@sss.pgh.pa.us 20 :CBC 6 : PG_MODULE_MAGIC_EXT(
21 : : .name = "utf8_and_gb18030",
22 : : .version = PG_VERSION
23 : : );
24 : :
8353 25 : 6 : PG_FUNCTION_INFO_V1(gb18030_to_utf8);
26 : 6 : PG_FUNCTION_INFO_V1(utf8_to_gb18030);
27 : :
28 : : /*
29 : : * Convert 4-byte GB18030 characters to and from a linear code space
30 : : *
31 : : * The first and third bytes can range from 0x81 to 0xfe (126 values),
32 : : * while the second and fourth bytes can range from 0x30 to 0x39 (10 values).
33 : : */
34 : : static inline uint32
3769 35 : 90 : gb_linear(uint32 gb)
36 : : {
37 : 90 : uint32 b0 = (gb & 0xff000000) >> 24;
38 : 90 : uint32 b1 = (gb & 0x00ff0000) >> 16;
39 : 90 : uint32 b2 = (gb & 0x0000ff00) >> 8;
40 : 90 : uint32 b3 = (gb & 0x000000ff);
41 : :
42 : 90 : return b0 * 12600 + b1 * 1260 + b2 * 10 + b3 -
43 : : (0x81 * 12600 + 0x30 * 1260 + 0x81 * 10 + 0x30);
44 : : }
45 : :
46 : : static inline uint32
47 : 36 : gb_unlinear(uint32 lin)
48 : : {
49 : 36 : uint32 r0 = 0x81 + lin / 12600;
50 : 36 : uint32 r1 = 0x30 + (lin / 1260) % 10;
51 : 36 : uint32 r2 = 0x81 + (lin / 10) % 126;
52 : 36 : uint32 r3 = 0x30 + lin % 10;
53 : :
54 : 36 : return (r0 << 24) | (r1 << 16) | (r2 << 8) | r3;
55 : : }
56 : :
57 : : /*
58 : : * Convert word-formatted UTF8 to and from Unicode code points
59 : : *
60 : : * Probably this should be somewhere else ...
61 : : */
62 : : static inline uint32
63 : 27 : unicode_to_utf8word(uint32 c)
64 : : {
65 : : uint32 word;
66 : :
67 [ - + ]: 27 : if (c <= 0x7F)
68 : : {
3769 tgl@sss.pgh.pa.us 69 :UBC 0 : word = c;
70 : : }
3769 tgl@sss.pgh.pa.us 71 [ - + ]:CBC 27 : else if (c <= 0x7FF)
72 : : {
3769 tgl@sss.pgh.pa.us 73 :UBC 0 : word = (0xC0 | ((c >> 6) & 0x1F)) << 8;
74 : 0 : word |= 0x80 | (c & 0x3F);
75 : : }
3769 tgl@sss.pgh.pa.us 76 [ + - ]:CBC 27 : else if (c <= 0xFFFF)
77 : : {
78 : 27 : word = (0xE0 | ((c >> 12) & 0x0F)) << 16;
79 : 27 : word |= (0x80 | ((c >> 6) & 0x3F)) << 8;
80 : 27 : word |= 0x80 | (c & 0x3F);
81 : : }
82 : : else
83 : : {
3769 tgl@sss.pgh.pa.us 84 :UBC 0 : word = (0xF0 | ((c >> 18) & 0x07)) << 24;
85 : 0 : word |= (0x80 | ((c >> 12) & 0x3F)) << 16;
86 : 0 : word |= (0x80 | ((c >> 6) & 0x3F)) << 8;
87 : 0 : word |= 0x80 | (c & 0x3F);
88 : : }
89 : :
3769 tgl@sss.pgh.pa.us 90 :CBC 27 : return word;
91 : : }
92 : :
93 : : static inline uint32
94 : 36 : utf8word_to_unicode(uint32 c)
95 : : {
96 : : uint32 ucs;
97 : :
98 [ - + ]: 36 : if (c <= 0x7F)
99 : : {
3769 tgl@sss.pgh.pa.us 100 :UBC 0 : ucs = c;
101 : : }
3769 tgl@sss.pgh.pa.us 102 [ - + ]:CBC 36 : else if (c <= 0xFFFF)
103 : : {
3769 tgl@sss.pgh.pa.us 104 :UBC 0 : ucs = ((c >> 8) & 0x1F) << 6;
105 : 0 : ucs |= c & 0x3F;
106 : : }
3769 tgl@sss.pgh.pa.us 107 [ + - ]:CBC 36 : else if (c <= 0xFFFFFF)
108 : : {
109 : 36 : ucs = ((c >> 16) & 0x0F) << 12;
110 : 36 : ucs |= ((c >> 8) & 0x3F) << 6;
111 : 36 : ucs |= c & 0x3F;
112 : : }
113 : : else
114 : : {
3769 tgl@sss.pgh.pa.us 115 :UBC 0 : ucs = ((c >> 24) & 0x07) << 18;
116 : 0 : ucs |= ((c >> 16) & 0x3F) << 12;
117 : 0 : ucs |= ((c >> 8) & 0x3F) << 6;
118 : 0 : ucs |= c & 0x3F;
119 : : }
120 : :
3769 tgl@sss.pgh.pa.us 121 :CBC 36 : return ucs;
122 : : }
123 : :
124 : : /*
125 : : * Perform mapping of GB18030 ranges to UTF8
126 : : *
127 : : * The ranges we need to convert are specified in gb-18030-2000.xml.
128 : : * All are ranges of 4-byte GB18030 codes.
129 : : */
130 : : static uint32
131 : 45 : conv_18030_to_utf8(uint32 code)
132 : : {
133 : : #define conv18030(minunicode, mincode, maxcode) \
134 : : if (code >= mincode && code <= maxcode) \
135 : : return unicode_to_utf8word(gb_linear(code) - gb_linear(mincode) + minunicode)
136 : :
137 [ + - - + ]: 45 : conv18030(0x0452, 0x8130D330, 0x8136A531);
138 [ + - - + ]: 45 : conv18030(0x2643, 0x8137A839, 0x8138FD38);
139 [ + - - + ]: 45 : conv18030(0x361B, 0x8230A633, 0x8230F237);
140 [ + - - + ]: 45 : conv18030(0x3CE1, 0x8231D438, 0x8232AF32);
141 [ + - - + ]: 45 : conv18030(0x4160, 0x8232C937, 0x8232F837);
142 [ + - - + ]: 45 : conv18030(0x44D7, 0x8233A339, 0x8233C931);
143 [ + - - + ]: 45 : conv18030(0x478E, 0x8233E838, 0x82349638);
144 [ + - - + ]: 45 : conv18030(0x49B8, 0x8234A131, 0x8234E733);
145 [ + - - + ]: 45 : conv18030(0x9FA6, 0x82358F33, 0x8336C738);
146 [ + - - + ]: 45 : conv18030(0xE865, 0x8336D030, 0x84308534);
147 [ + - + + ]: 45 : conv18030(0xFA2A, 0x84309C38, 0x84318537);
148 [ + - - + ]: 18 : conv18030(0xFFE6, 0x8431A234, 0x8431A439);
149 [ - + - - ]: 18 : conv18030(0x10000, 0x90308130, 0xE3329A35);
150 : : /* No mapping exists */
151 : 18 : return 0;
152 : : }
153 : :
154 : : /*
155 : : * Perform mapping of UTF8 ranges to GB18030
156 : : */
157 : : static uint32
158 : 36 : conv_utf8_to_18030(uint32 code)
159 : : {
160 : 36 : uint32 ucs = utf8word_to_unicode(code);
161 : :
162 : : #define convutf8(minunicode, maxunicode, mincode) \
163 : : if (ucs >= minunicode && ucs <= maxunicode) \
164 : : return gb_unlinear(ucs - minunicode + gb_linear(mincode))
165 : :
166 [ + - - + ]: 36 : convutf8(0x0452, 0x200F, 0x8130D330);
167 [ + - - + ]: 36 : convutf8(0x2643, 0x2E80, 0x8137A839);
168 [ + - - + ]: 36 : convutf8(0x361B, 0x3917, 0x8230A633);
169 [ + - - + ]: 36 : convutf8(0x3CE1, 0x4055, 0x8231D438);
170 [ + - - + ]: 36 : convutf8(0x4160, 0x4336, 0x8232C937);
171 [ + - - + ]: 36 : convutf8(0x44D7, 0x464B, 0x8233A339);
172 [ + - - + ]: 36 : convutf8(0x478E, 0x4946, 0x8233E838);
173 [ + - - + ]: 36 : convutf8(0x49B8, 0x4C76, 0x8234A131);
174 [ + - + + ]: 36 : convutf8(0x9FA6, 0xD7FF, 0x82358F33);
175 [ + - - + ]: 9 : convutf8(0xE865, 0xF92B, 0x8336D030);
176 [ + - + - ]: 9 : convutf8(0xFA2A, 0xFE2F, 0x84309C38);
3769 tgl@sss.pgh.pa.us 177 [ # # # # ]:UBC 0 : convutf8(0xFFE6, 0xFFFF, 0x8431A234);
178 [ # # # # ]: 0 : convutf8(0x10000, 0x10FFFF, 0x90308130);
179 : : /* No mapping exists */
180 : 0 : return 0;
181 : : }
182 : :
183 : : /* ----------
184 : : * conv_proc(
185 : : * INTEGER, -- source encoding id
186 : : * INTEGER, -- destination encoding id
187 : : * CSTRING, -- source string (null terminated C string)
188 : : * CSTRING, -- destination string (null terminated C string)
189 : : * INTEGER, -- source string length
190 : : * BOOL -- if true, don't throw an error if conversion fails
191 : : * ) returns INTEGER;
192 : : *
193 : : * Returns the number of bytes successfully converted.
194 : : * ----------
195 : : */
196 : : Datum
8455 ishii@postgresql.org 197 :CBC 138 : gb18030_to_utf8(PG_FUNCTION_ARGS)
198 : : {
7289 tgl@sss.pgh.pa.us 199 : 138 : unsigned char *src = (unsigned char *) PG_GETARG_CSTRING(2);
200 : 138 : unsigned char *dest = (unsigned char *) PG_GETARG_CSTRING(3);
8405 bruce@momjian.us 201 : 138 : int len = PG_GETARG_INT32(4);
1621 heikki.linnakangas@i 202 : 138 : bool noError = PG_GETARG_BOOL(5);
203 : : int converted;
204 : :
6066 tgl@sss.pgh.pa.us 205 : 138 : CHECK_ENCODING_CONVERSION_ARGS(PG_GB18030, PG_UTF8);
206 : :
1621 heikki.linnakangas@i 207 : 138 : converted = LocalToUtf(src, len, dest,
208 : : &gb18030_to_unicode_tree,
209 : : NULL, 0,
210 : : conv_18030_to_utf8,
211 : : PG_GB18030,
212 : : noError);
213 : :
214 : 84 : PG_RETURN_INT32(converted);
215 : : }
216 : :
217 : : Datum
8455 ishii@postgresql.org 218 : 165 : utf8_to_gb18030(PG_FUNCTION_ARGS)
219 : : {
7289 tgl@sss.pgh.pa.us 220 : 165 : unsigned char *src = (unsigned char *) PG_GETARG_CSTRING(2);
221 : 165 : unsigned char *dest = (unsigned char *) PG_GETARG_CSTRING(3);
8405 bruce@momjian.us 222 : 165 : int len = PG_GETARG_INT32(4);
1621 heikki.linnakangas@i 223 : 165 : bool noError = PG_GETARG_BOOL(5);
224 : : int converted;
225 : :
6066 tgl@sss.pgh.pa.us 226 : 165 : CHECK_ENCODING_CONVERSION_ARGS(PG_UTF8, PG_GB18030);
227 : :
1621 heikki.linnakangas@i 228 : 165 : converted = UtfToLocal(src, len, dest,
229 : : &gb18030_from_unicode_tree,
230 : : NULL, 0,
231 : : conv_utf8_to_18030,
232 : : PG_GB18030,
233 : : noError);
234 : :
235 : 120 : PG_RETURN_INT32(converted);
236 : : }
|
