01f17521a17e2255a0ddab6500c190be51171791
[musl] / src / locale / iconv.c
1 #include <iconv.h>
2 #include <errno.h>
3 #include <wchar.h>
4 #include <string.h>
5 #include <stdlib.h>
6 #include <limits.h>
7 #include <stdint.h>
8 #include "locale_impl.h"
9
10 #define UTF_32BE    0300
11 #define UTF_16LE    0301
12 #define UTF_16BE    0302
13 #define UTF_32LE    0303
14 #define UCS2BE      0304
15 #define UCS2LE      0305
16 #define WCHAR_T     0306
17 #define US_ASCII    0307
18 #define UTF_8       0310
19 #define EUC_JP      0320
20 #define SHIFT_JIS   0321
21 #define ISO2022_JP  0322
22 #define GB18030     0330
23 #define GBK         0331
24 #define GB2312      0332
25 #define BIG5        0340
26 #define EUC_KR      0350
27
28 /* Definitions of charmaps. Each charmap consists of:
29  * 1. Empty-string-terminated list of null-terminated aliases.
30  * 2. Special type code or number of elided quads of entries.
31  * 3. Character table (size determined by field 2), consisting
32  *    of 5 bytes for every 4 characters, interpreted as 10-bit
33  *    indices into the legacy_chars table. */
34
35 static const unsigned char charmaps[] =
36 "utf8\0char\0\0\310"
37 "wchart\0\0\306"
38 "ucs2\0ucs2be\0\0\304"
39 "ucs2le\0\0\305"
40 "utf16\0utf16be\0\0\302"
41 "utf16le\0\0\301"
42 "ucs4\0ucs4be\0utf32\0utf32be\0\0\300"
43 "ucs4le\0utf32le\0\0\303"
44 "ascii\0usascii\0iso646\0iso646us\0\0\307"
45 "eucjp\0\0\320"
46 "shiftjis\0sjis\0\0\321"
47 "iso2022jp\0\0\322"
48 "gb18030\0\0\330"
49 "gbk\0\0\331"
50 "gb2312\0\0\332"
51 "big5\0bigfive\0cp950\0big5hkscs\0\0\340"
52 "euckr\0ksc5601\0ksx1001\0cp949\0\0\350"
53 #include "codepages.h"
54 ;
55
56 /* Table of characters that appear in legacy 8-bit codepages,
57  * limited to 1024 slots (10 bit indices). The first 256 entries
58  * are elided since those characters are obviously all included. */
59 static const unsigned short legacy_chars[] = {
60 #include "legacychars.h"
61 };
62
63 static const unsigned short jis0208[84][94] = {
64 #include "jis0208.h"
65 };
66
67 static const unsigned short gb18030[126][190] = {
68 #include "gb18030.h"
69 };
70
71 static const unsigned short big5[89][157] = {
72 #include "big5.h"
73 };
74
75 static const unsigned short hkscs[] = {
76 #include "hkscs.h"
77 };
78
79 static const unsigned short ksc[93][94] = {
80 #include "ksc.h"
81 };
82
83 static int fuzzycmp(const unsigned char *a, const unsigned char *b)
84 {
85         for (; *a && *b; a++, b++) {
86                 while (*a && (*a|32U)-'a'>26 && *a-'0'>10U) a++;
87                 if ((*a|32U) != *b) return 1;
88         }
89         return *a != *b;
90 }
91
92 static size_t find_charmap(const void *name)
93 {
94         const unsigned char *s;
95         if (!*(char *)name) name=charmaps; /* "utf8" */
96         for (s=charmaps; *s; ) {
97                 if (!fuzzycmp(name, s)) {
98                         for (; *s; s+=strlen((void *)s)+1);
99                         return s+1-charmaps;
100                 }
101                 s += strlen((void *)s)+1;
102                 if (!*s) {
103                         if (s[1] > 0200) s+=2;
104                         else s+=2+(64U-s[1])*5;
105                 }
106         }
107         return -1;
108 }
109
110 struct stateful_cd {
111         iconv_t base_cd;
112         unsigned state;
113 };
114
115 static iconv_t combine_to_from(size_t t, size_t f)
116 {
117         return (void *)(f<<16 | t<<1 | 1);
118 }
119
120 static size_t extract_from(iconv_t cd)
121 {
122         return (size_t)cd >> 16;
123 }
124
125 static size_t extract_to(iconv_t cd)
126 {
127         return (size_t)cd >> 1 & 0x7fff;
128 }
129
130 iconv_t iconv_open(const char *to, const char *from)
131 {
132         size_t f, t;
133         struct stateful_cd *scd;
134
135         if ((t = find_charmap(to))==-1
136          || (f = find_charmap(from))==-1
137          || (charmaps[t] >= 0320)) {
138                 errno = EINVAL;
139                 return (iconv_t)-1;
140         }
141         iconv_t cd = combine_to_from(t, f);
142
143         switch (charmaps[f]) {
144         case ISO2022_JP:
145                 scd = malloc(sizeof *scd);
146                 if (!scd) return (iconv_t)-1;
147                 scd->base_cd = cd;
148                 scd->state = 0;
149                 cd = (iconv_t)scd;
150         }
151
152         return cd;
153 }
154
155 static unsigned get_16(const unsigned char *s, int e)
156 {
157         e &= 1;
158         return s[e]<<8 | s[1-e];
159 }
160
161 static void put_16(unsigned char *s, unsigned c, int e)
162 {
163         e &= 1;
164         s[e] = c>>8;
165         s[1-e] = c;
166 }
167
168 static unsigned get_32(const unsigned char *s, int e)
169 {
170         e &= 3;
171         return s[e]+0U<<24 | s[e^1]<<16 | s[e^2]<<8 | s[e^3];
172 }
173
174 static void put_32(unsigned char *s, unsigned c, int e)
175 {
176         e &= 3;
177         s[e^0] = c>>24;
178         s[e^1] = c>>16;
179         s[e^2] = c>>8;
180         s[e^3] = c;
181 }
182
183 /* Adapt as needed */
184 #define mbrtowc_utf8 mbrtowc
185 #define wctomb_utf8 wctomb
186
187 static unsigned legacy_map(const unsigned char *map, unsigned c)
188 {
189         if (c < 4*map[-1]) return c;
190         unsigned x = c - 4*map[-1];
191         x = map[x*5/4]>>2*x%8 | map[x*5/4+1]<<8-2*x%8 & 1023;
192         return x < 256 ? x : legacy_chars[x-256];
193 }
194
195 size_t iconv(iconv_t cd, char **restrict in, size_t *restrict inb, char **restrict out, size_t *restrict outb)
196 {
197         size_t x=0;
198         struct stateful_cd *scd=0;
199         if (!((size_t)cd & 1)) {
200                 scd = (void *)cd;
201                 cd = scd->base_cd;
202         }
203         unsigned to = extract_to(cd);
204         unsigned from = extract_from(cd);
205         const unsigned char *map = charmaps+from+1;
206         const unsigned char *tomap = charmaps+to+1;
207         mbstate_t st = {0};
208         wchar_t wc;
209         unsigned c, d;
210         size_t k, l;
211         int err;
212         unsigned char type = map[-1];
213         unsigned char totype = tomap[-1];
214         locale_t *ploc = &CURRENT_LOCALE, loc = *ploc;
215
216         if (!in || !*in || !*inb) return 0;
217
218         *ploc = UTF8_LOCALE;
219
220         for (; *inb; *in+=l, *inb-=l) {
221                 c = *(unsigned char *)*in;
222                 l = 1;
223
224                 switch (type) {
225                 case UTF_8:
226                         if (c < 128) break;
227                         l = mbrtowc_utf8(&wc, *in, *inb, &st);
228                         if (l == (size_t)-1) goto ilseq;
229                         if (l == (size_t)-2) goto starved;
230                         c = wc;
231                         break;
232                 case US_ASCII:
233                         if (c >= 128) goto ilseq;
234                         break;
235                 case WCHAR_T:
236                         l = sizeof(wchar_t);
237                         if (*inb < l) goto starved;
238                         c = *(wchar_t *)*in;
239                         if (0) {
240                 case UTF_32BE:
241                 case UTF_32LE:
242                         l = 4;
243                         if (*inb < 4) goto starved;
244                         c = get_32((void *)*in, type);
245                         }
246                         if (c-0xd800u < 0x800u || c >= 0x110000u) goto ilseq;
247                         break;
248                 case UCS2BE:
249                 case UCS2LE:
250                 case UTF_16BE:
251                 case UTF_16LE:
252                         l = 2;
253                         if (*inb < 2) goto starved;
254                         c = get_16((void *)*in, type);
255                         if ((unsigned)(c-0xdc00) < 0x400) goto ilseq;
256                         if ((unsigned)(c-0xd800) < 0x400) {
257                                 if (type-UCS2BE < 2U) goto ilseq;
258                                 l = 4;
259                                 if (*inb < 4) goto starved;
260                                 d = get_16((void *)(*in + 2), type);
261                                 if ((unsigned)(d-0xdc00) >= 0x400) goto ilseq;
262                                 c = ((c-0xd7c0)<<10) + (d-0xdc00);
263                         }
264                         break;
265                 case SHIFT_JIS:
266                         if (c < 128) break;
267                         if (c-0xa1 <= 0xdf-0xa1) {
268                                 c += 0xff61-0xa1;
269                                 break;
270                         }
271                         l = 2;
272                         if (*inb < 2) goto starved;
273                         d = *((unsigned char *)*in + 1);
274                         if (c-129 <= 159-129) c -= 129;
275                         else if (c-224 <= 239-224) c -= 193;
276                         else goto ilseq;
277                         c *= 2;
278                         if (d-64 <= 158-64) {
279                                 if (d==127) goto ilseq;
280                                 if (d>127) d--;
281                                 d -= 64;
282                         } else if (d-159 <= 252-159) {
283                                 c++;
284                                 d -= 159;
285                         }
286                         c = jis0208[c][d];
287                         if (!c) goto ilseq;
288                         break;
289                 case EUC_JP:
290                         if (c < 128) break;
291                         l = 2;
292                         if (*inb < 2) goto starved;
293                         d = *((unsigned char *)*in + 1);
294                         if (c==0x8e) {
295                                 c = d;
296                                 if (c-0xa1 > 0xdf-0xa1) goto ilseq;
297                                 c += 0xff61 - 0xa1;
298                                 break;
299                         }
300                         c -= 0xa1;
301                         d -= 0xa1;
302                         if (c >= 84 || d >= 94) goto ilseq;
303                         c = jis0208[c][d];
304                         if (!c) goto ilseq;
305                         break;
306                 case ISO2022_JP:
307                         if (c >= 128) goto ilseq;
308                         if (c == '\033') {
309                                 l = 3;
310                                 if (*inb < 3) goto starved;
311                                 c = *((unsigned char *)*in + 1);
312                                 d = *((unsigned char *)*in + 2);
313                                 if (c != '(' && c != '$') goto ilseq;
314                                 switch (128*(c=='$') + d) {
315                                 case 'B': scd->state=0; continue;
316                                 case 'J': scd->state=1; continue;
317                                 case 'I': scd->state=4; continue;
318                                 case 128+'@': scd->state=2; continue;
319                                 case 128+'B': scd->state=3; continue;
320                                 }
321                                 goto ilseq;
322                         }
323                         switch (scd->state) {
324                         case 1:
325                                 if (c=='\\') c = 0xa5;
326                                 if (c=='~') c = 0x203e;
327                                 break;
328                         case 2:
329                         case 3:
330                                 l = 2;
331                                 if (*inb < 2) goto starved;
332                                 d = *((unsigned char *)*in + 1);
333                                 c -= 0x21;
334                                 d -= 0x21;
335                                 if (c >= 84 || d >= 94) goto ilseq;
336                                 c = jis0208[c][d];
337                                 if (!c) goto ilseq;
338                                 break;
339                         case 4:
340                                 if (c-0x60 < 0x1f) goto ilseq;
341                                 if (c-0x21 < 0x5e) c += 0xff61-0x21;
342                                 break;
343                         }
344                         break;
345                 case GB2312:
346                         if (c < 128) break;
347                         if (c < 0xa1) goto ilseq;
348                 case GBK:
349                 case GB18030:
350                         if (c < 128) break;
351                         c -= 0x81;
352                         if (c >= 126) goto ilseq;
353                         l = 2;
354                         if (*inb < 2) goto starved;
355                         d = *((unsigned char *)*in + 1);
356                         if (d < 0xa1 && type == GB2312) goto ilseq;
357                         if (d-0x40>=191 || d==127) {
358                                 if (d-'0'>9 || type != GB18030)
359                                         goto ilseq;
360                                 l = 4;
361                                 if (*inb < 4) goto starved;
362                                 c = (10*c + d-'0') * 1260;
363                                 d = *((unsigned char *)*in + 2);
364                                 if (d-0x81>126) goto ilseq;
365                                 c += 10*(d-0x81);
366                                 d = *((unsigned char *)*in + 3);
367                                 if (d-'0'>9) goto ilseq;
368                                 c += d-'0';
369                                 c += 128;
370                                 for (d=0; d<=c; ) {
371                                         k = 0;
372                                         for (int i=0; i<126; i++)
373                                                 for (int j=0; j<190; j++)
374                                                         if (gb18030[i][j]-d <= c-d)
375                                                                 k++;
376                                         d = c+1;
377                                         c += k;
378                                 }
379                                 break;
380                         }
381                         d -= 0x40;
382                         if (d>63) d--;
383                         c = gb18030[c][d];
384                         break;
385                 case BIG5:
386                         if (c < 128) break;
387                         l = 2;
388                         if (*inb < 2) goto starved;
389                         d = *((unsigned char *)*in + 1);
390                         if (d-0x40>=0xff-0x40 || d-0x7f<0xa1-0x7f) goto ilseq;
391                         d -= 0x40;
392                         if (d > 0x3e) d -= 0x22;
393                         if (c-0xa1>=0xfa-0xa1) {
394                                 if (c-0x87>=0xff-0x87) goto ilseq;
395                                 if (c < 0xa1) c -= 0x87;
396                                 else c -= 0x87 + (0xfa-0xa1);
397                                 c = (hkscs[4867+(c*157+d)/16]>>(c*157+d)%16)%2<<17
398                                         | hkscs[c*157+d];
399                                 /* A few HKSCS characters map to pairs of UCS
400                                  * characters. These are mapped to surrogate
401                                  * range in the hkscs table then hard-coded
402                                  * here. Ugly, yes. */
403                                 if (c/256 == 0xdc) {
404                                         if (totype-0300U > 8) k = 2;
405                                         else k = "\10\4\4\10\4\4\10\2\4"[totype-0300];
406                                         if (k > *outb) goto toobig;
407                                         x += iconv(combine_to_from(to, 0),
408                                                 &(char *){"\303\212\314\204"
409                                                 "\303\212\314\214"
410                                                 "\303\252\314\204"
411                                                 "\303\252\314\214"
412                                                 +c%256}, &(size_t){4},
413                                                 out, outb);
414                                         continue;
415                                 }
416                                 if (!c) goto ilseq;
417                                 break;
418                         }
419                         c -= 0xa1;
420                         c = big5[c][d]|(c==0x27&&(d==0x3a||d==0x3c||d==0x42))<<17;
421                         if (!c) goto ilseq;
422                         break;
423                 case EUC_KR:
424                         if (c < 128) break;
425                         l = 2;
426                         if (*inb < 2) goto starved;
427                         d = *((unsigned char *)*in + 1);
428                         c -= 0xa1;
429                         d -= 0xa1;
430                         if (c >= 93 || d >= 94) {
431                                 c += (0xa1-0x81);
432                                 d += 0xa1;
433                                 if (c >= 93 || c>=0xc6-0x81 && d>0x52)
434                                         goto ilseq;
435                                 if (d-'A'<26) d = d-'A';
436                                 else if (d-'a'<26) d = d-'a'+26;
437                                 else if (d-0x81<0xff-0x81) d = d-0x81+52;
438                                 else goto ilseq;
439                                 if (c < 0x20) c = 178*c + d;
440                                 else c = 178*0x20 + 84*(c-0x20) + d;
441                                 c += 0xac00;
442                                 for (d=0xac00; d<=c; ) {
443                                         k = 0;
444                                         for (int i=0; i<93; i++)
445                                                 for (int j=0; j<94; j++)
446                                                         if (ksc[i][j]-d <= c-d)
447                                                                 k++;
448                                         d = c+1;
449                                         c += k;
450                                 }
451                                 break;
452                         }
453                         c = ksc[c][d];
454                         if (!c) goto ilseq;
455                         break;
456                 default:
457                         if (!c) break;
458                         c = legacy_map(map, c);
459                         if (!c) goto ilseq;
460                 }
461
462                 switch (totype) {
463                 case WCHAR_T:
464                         if (*outb < sizeof(wchar_t)) goto toobig;
465                         *(wchar_t *)*out = c;
466                         *out += sizeof(wchar_t);
467                         *outb -= sizeof(wchar_t);
468                         break;
469                 case UTF_8:
470                         if (*outb < 4) {
471                                 char tmp[4];
472                                 k = wctomb_utf8(tmp, c);
473                                 if (*outb < k) goto toobig;
474                                 memcpy(*out, tmp, k);
475                         } else k = wctomb_utf8(*out, c);
476                         *out += k;
477                         *outb -= k;
478                         break;
479                 case US_ASCII:
480                         if (c > 0x7f) subst: x++, c='*';
481                 default:
482                         if (*outb < 1) goto toobig;
483                         if (c<256 && c==legacy_map(tomap, c)) {
484                         revout:
485                                 *(*out)++ = c;
486                                 *outb -= 1;
487                                 break;
488                         }
489                         d = c;
490                         for (c=4*totype; c<256; c++) {
491                                 if (d == legacy_map(tomap, c)) {
492                                         goto revout;
493                                 }
494                         }
495                         goto subst;
496                 case UCS2BE:
497                 case UCS2LE:
498                 case UTF_16BE:
499                 case UTF_16LE:
500                         if (c < 0x10000 || type-UCS2BE < 2U) {
501                                 if (c >= 0x10000) c = 0xFFFD;
502                                 if (*outb < 2) goto toobig;
503                                 put_16((void *)*out, c, totype);
504                                 *out += 2;
505                                 *outb -= 2;
506                                 break;
507                         }
508                         if (*outb < 4) goto toobig;
509                         c -= 0x10000;
510                         put_16((void *)*out, (c>>10)|0xd800, totype);
511                         put_16((void *)(*out + 2), (c&0x3ff)|0xdc00, totype);
512                         *out += 4;
513                         *outb -= 4;
514                         break;
515                 case UTF_32BE:
516                 case UTF_32LE:
517                         if (*outb < 4) goto toobig;
518                         put_32((void *)*out, c, totype);
519                         *out += 4;
520                         *outb -= 4;
521                         break;
522                 }
523         }
524         *ploc = loc;
525         return x;
526 ilseq:
527         err = EILSEQ;
528         x = -1;
529         goto end;
530 toobig:
531         err = E2BIG;
532         x = -1;
533         goto end;
534 starved:
535         err = EINVAL;
536         x = -1;
537 end:
538         errno = err;
539         *ploc = loc;
540         return x;
541 }