Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 1 | /* |
| 2 | * _codecs_cn.c: Codecs collection for Mainland Chinese encodings |
| 3 | * |
| 4 | * Written by Hye-Shik Chang <perky@FreeBSD.org> |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 5 | */ |
| 6 | |
| 7 | #include "cjkcodecs.h" |
| 8 | #include "mappings_cn.h" |
| 9 | |
Hye-Shik Chang | c5c57e6 | 2005-12-12 11:48:32 +0000 | [diff] [blame] | 10 | /** |
| 11 | * hz is predefined as 100 on AIX. So we undefine it to avoid |
| 12 | * conflict against hz codec's. |
| 13 | */ |
| 14 | #ifdef _AIX |
| 15 | #undef hz |
| 16 | #endif |
| 17 | |
Serhiy Storchaka | d3faf43 | 2015-01-18 11:28:37 +0200 | [diff] [blame] | 18 | /* GBK and GB2312 map differently in few code points that are listed below: |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 19 | * |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 20 | * gb2312 gbk |
| 21 | * A1A4 U+30FB KATAKANA MIDDLE DOT U+00B7 MIDDLE DOT |
| 22 | * A1AA U+2015 HORIZONTAL BAR U+2014 EM DASH |
| 23 | * A844 undefined U+2015 HORIZONTAL BAR |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 24 | */ |
| 25 | |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 26 | #define GBK_DECODE(dc1, dc2, writer) \ |
| 27 | if ((dc1) == 0xa1 && (dc2) == 0xaa) { \ |
| 28 | OUTCHAR(0x2014); \ |
| 29 | } \ |
| 30 | else if ((dc1) == 0xa8 && (dc2) == 0x44) { \ |
| 31 | OUTCHAR(0x2015); \ |
| 32 | } \ |
| 33 | else if ((dc1) == 0xa1 && (dc2) == 0xa4) { \ |
| 34 | OUTCHAR(0x00b7); \ |
| 35 | } \ |
| 36 | else if (TRYMAP_DEC(gb2312, decoded, dc1 ^ 0x80, dc2 ^ 0x80)) { \ |
| 37 | OUTCHAR(decoded); \ |
| 38 | } \ |
| 39 | else if (TRYMAP_DEC(gbkext, decoded, dc1, dc2)) { \ |
| 40 | OUTCHAR(decoded); \ |
| 41 | } |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 42 | |
Victor Stinner | 28c63f7 | 2013-10-29 00:59:44 +0100 | [diff] [blame] | 43 | #define GBK_ENCODE(code, assi) \ |
| 44 | if ((code) == 0x2014) { \ |
| 45 | (assi) = 0xa1aa; \ |
| 46 | } else if ((code) == 0x2015) { \ |
| 47 | (assi) = 0xa844; \ |
| 48 | } else if ((code) == 0x00b7) { \ |
| 49 | (assi) = 0xa1a4; \ |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 50 | } else if ((code) != 0x30fb && TRYMAP_ENC(gbcommon, assi, code)) { \ |
Victor Stinner | 28c63f7 | 2013-10-29 00:59:44 +0100 | [diff] [blame] | 51 | ; \ |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 52 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 53 | |
| 54 | /* |
| 55 | * GB2312 codec |
| 56 | */ |
| 57 | |
| 58 | ENCODER(gb2312) |
| 59 | { |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 60 | while (*inpos < inlen) { |
| 61 | Py_UCS4 c = INCHAR1; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 62 | DBCHAR code; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 63 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 64 | if (c < 0x80) { |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 65 | WRITEBYTE1((unsigned char)c); |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 66 | NEXT(1, 1); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 67 | continue; |
| 68 | } |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 69 | |
| 70 | if (c > 0xFFFF) |
| 71 | return 1; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 72 | |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 73 | REQUIRE_OUTBUF(2); |
Victor Stinner | 146a2ed | 2013-10-29 00:09:41 +0100 | [diff] [blame] | 74 | if (TRYMAP_ENC(gbcommon, code, c)) |
| 75 | ; |
| 76 | else |
| 77 | return 1; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 78 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 79 | if (code & 0x8000) /* MSB set: GBK */ |
| 80 | return 1; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 81 | |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 82 | OUTBYTE1((code >> 8) | 0x80); |
| 83 | OUTBYTE2((code & 0xFF) | 0x80); |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 84 | NEXT(1, 2); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 85 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 86 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 87 | return 0; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 88 | } |
| 89 | |
| 90 | DECODER(gb2312) |
| 91 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 92 | while (inleft > 0) { |
| 93 | unsigned char c = **inbuf; |
Victor Stinner | 11bdf91 | 2013-10-28 23:18:39 +0100 | [diff] [blame] | 94 | Py_UCS4 decoded; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 95 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 96 | if (c < 0x80) { |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 97 | OUTCHAR(c); |
| 98 | NEXT_IN(1); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 99 | continue; |
| 100 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 101 | |
Victor Stinner | 28c63f7 | 2013-10-29 00:59:44 +0100 | [diff] [blame] | 102 | REQUIRE_INBUF(2); |
Victor Stinner | 11bdf91 | 2013-10-28 23:18:39 +0100 | [diff] [blame] | 103 | if (TRYMAP_DEC(gb2312, decoded, c ^ 0x80, INBYTE2 ^ 0x80)) { |
| 104 | OUTCHAR(decoded); |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 105 | NEXT_IN(2); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 106 | } |
Victor Stinner | 146a2ed | 2013-10-29 00:09:41 +0100 | [diff] [blame] | 107 | else |
| 108 | return 1; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 109 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 110 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 111 | return 0; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 112 | } |
| 113 | |
| 114 | |
| 115 | /* |
| 116 | * GBK codec |
| 117 | */ |
| 118 | |
| 119 | ENCODER(gbk) |
| 120 | { |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 121 | while (*inpos < inlen) { |
| 122 | Py_UCS4 c = INCHAR1; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 123 | DBCHAR code; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 124 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 125 | if (c < 0x80) { |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 126 | WRITEBYTE1((unsigned char)c); |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 127 | NEXT(1, 1); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 128 | continue; |
| 129 | } |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 130 | |
| 131 | if (c > 0xFFFF) |
| 132 | return 1; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 133 | |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 134 | REQUIRE_OUTBUF(2); |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 135 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 136 | GBK_ENCODE(c, code) |
Victor Stinner | 146a2ed | 2013-10-29 00:09:41 +0100 | [diff] [blame] | 137 | else |
| 138 | return 1; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 139 | |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 140 | OUTBYTE1((code >> 8) | 0x80); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 141 | if (code & 0x8000) |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 142 | OUTBYTE2((code & 0xFF)); /* MSB set: GBK */ |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 143 | else |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 144 | OUTBYTE2((code & 0xFF) | 0x80); /* MSB unset: GB2312 */ |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 145 | NEXT(1, 2); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 146 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 147 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 148 | return 0; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 149 | } |
| 150 | |
| 151 | DECODER(gbk) |
| 152 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 153 | while (inleft > 0) { |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 154 | unsigned char c = INBYTE1; |
Victor Stinner | 11bdf91 | 2013-10-28 23:18:39 +0100 | [diff] [blame] | 155 | Py_UCS4 decoded; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 156 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 157 | if (c < 0x80) { |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 158 | OUTCHAR(c); |
| 159 | NEXT_IN(1); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 160 | continue; |
| 161 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 162 | |
Victor Stinner | 28c63f7 | 2013-10-29 00:59:44 +0100 | [diff] [blame] | 163 | REQUIRE_INBUF(2); |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 164 | |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 165 | GBK_DECODE(c, INBYTE2, writer) |
Victor Stinner | 146a2ed | 2013-10-29 00:09:41 +0100 | [diff] [blame] | 166 | else |
| 167 | return 1; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 168 | |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 169 | NEXT_IN(2); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 170 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 171 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 172 | return 0; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 173 | } |
| 174 | |
| 175 | |
| 176 | /* |
| 177 | * GB18030 codec |
| 178 | */ |
| 179 | |
| 180 | ENCODER(gb18030) |
| 181 | { |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 182 | while (*inpos < inlen) { |
| 183 | Py_UCS4 c = INCHAR1; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 184 | DBCHAR code; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 185 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 186 | if (c < 0x80) { |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 187 | WRITEBYTE1(c); |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 188 | NEXT(1, 1); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 189 | continue; |
| 190 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 191 | |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 192 | if (c >= 0x10000) { |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 193 | Py_UCS4 tc = c - 0x10000; |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 194 | assert (c <= 0x10FFFF); |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 195 | |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 196 | REQUIRE_OUTBUF(4); |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 197 | |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 198 | OUTBYTE4((unsigned char)(tc % 10) + 0x30); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 199 | tc /= 10; |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 200 | OUTBYTE3((unsigned char)(tc % 126) + 0x81); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 201 | tc /= 126; |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 202 | OUTBYTE2((unsigned char)(tc % 10) + 0x30); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 203 | tc /= 10; |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 204 | OUTBYTE1((unsigned char)(tc + 0x90)); |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 205 | |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 206 | NEXT(1, 4); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 207 | continue; |
| 208 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 209 | |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 210 | REQUIRE_OUTBUF(2); |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 211 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 212 | GBK_ENCODE(c, code) |
Victor Stinner | 146a2ed | 2013-10-29 00:09:41 +0100 | [diff] [blame] | 213 | else if (TRYMAP_ENC(gb18030ext, code, c)) |
| 214 | ; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 215 | else { |
| 216 | const struct _gb18030_to_unibmp_ranges *utrrange; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 217 | |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 218 | REQUIRE_OUTBUF(4); |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 219 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 220 | for (utrrange = gb18030_to_unibmp_ranges; |
| 221 | utrrange->first != 0; |
| 222 | utrrange++) |
| 223 | if (utrrange->first <= c && |
| 224 | c <= utrrange->last) { |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 225 | Py_UCS4 tc; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 226 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 227 | tc = c - utrrange->first + |
| 228 | utrrange->base; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 229 | |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 230 | OUTBYTE4((unsigned char)(tc % 10) + 0x30); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 231 | tc /= 10; |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 232 | OUTBYTE3((unsigned char)(tc % 126) + 0x81); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 233 | tc /= 126; |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 234 | OUTBYTE2((unsigned char)(tc % 10) + 0x30); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 235 | tc /= 10; |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 236 | OUTBYTE1((unsigned char)tc + 0x81); |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 237 | |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 238 | NEXT(1, 4); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 239 | break; |
| 240 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 241 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 242 | if (utrrange->first == 0) |
| 243 | return 1; |
| 244 | continue; |
| 245 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 246 | |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 247 | OUTBYTE1((code >> 8) | 0x80); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 248 | if (code & 0x8000) |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 249 | OUTBYTE2((code & 0xFF)); /* MSB set: GBK or GB18030ext */ |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 250 | else |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 251 | OUTBYTE2((code & 0xFF) | 0x80); /* MSB unset: GB2312 */ |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 252 | |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 253 | NEXT(1, 2); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 254 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 255 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 256 | return 0; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 257 | } |
| 258 | |
| 259 | DECODER(gb18030) |
| 260 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 261 | while (inleft > 0) { |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 262 | unsigned char c = INBYTE1, c2; |
Victor Stinner | 11bdf91 | 2013-10-28 23:18:39 +0100 | [diff] [blame] | 263 | Py_UCS4 decoded; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 264 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 265 | if (c < 0x80) { |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 266 | OUTCHAR(c); |
| 267 | NEXT_IN(1); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 268 | continue; |
| 269 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 270 | |
Victor Stinner | 28c63f7 | 2013-10-29 00:59:44 +0100 | [diff] [blame] | 271 | REQUIRE_INBUF(2); |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 272 | |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 273 | c2 = INBYTE2; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 274 | if (c2 >= 0x30 && c2 <= 0x39) { /* 4 bytes seq */ |
| 275 | const struct _gb18030_to_unibmp_ranges *utr; |
| 276 | unsigned char c3, c4; |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 277 | Py_UCS4 lseq; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 278 | |
Victor Stinner | 28c63f7 | 2013-10-29 00:59:44 +0100 | [diff] [blame] | 279 | REQUIRE_INBUF(4); |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 280 | c3 = INBYTE3; |
| 281 | c4 = INBYTE4; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 282 | if (c < 0x81 || c3 < 0x81 || c4 < 0x30 || c4 > 0x39) |
Victor Stinner | 2cded9c | 2011-07-08 01:45:13 +0200 | [diff] [blame] | 283 | return 1; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 284 | c -= 0x81; c2 -= 0x30; |
| 285 | c3 -= 0x81; c4 -= 0x30; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 286 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 287 | if (c < 4) { /* U+0080 - U+FFFF */ |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 288 | lseq = ((Py_UCS4)c * 10 + c2) * 1260 + |
| 289 | (Py_UCS4)c3 * 10 + c4; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 290 | if (lseq < 39420) { |
| 291 | for (utr = gb18030_to_unibmp_ranges; |
| 292 | lseq >= (utr + 1)->base; |
| 293 | utr++) ; |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 294 | OUTCHAR(utr->first - utr->base + lseq); |
| 295 | NEXT_IN(4); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 296 | continue; |
| 297 | } |
| 298 | } |
| 299 | else if (c >= 15) { /* U+10000 - U+10FFFF */ |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 300 | lseq = 0x10000 + (((Py_UCS4)c-15) * 10 + c2) |
| 301 | * 1260 + (Py_UCS4)c3 * 10 + c4; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 302 | if (lseq <= 0x10FFFF) { |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 303 | OUTCHAR(lseq); |
| 304 | NEXT_IN(4); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 305 | continue; |
| 306 | } |
| 307 | } |
Victor Stinner | 2cded9c | 2011-07-08 01:45:13 +0200 | [diff] [blame] | 308 | return 1; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 309 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 310 | |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 311 | GBK_DECODE(c, c2, writer) |
Victor Stinner | 11bdf91 | 2013-10-28 23:18:39 +0100 | [diff] [blame] | 312 | else if (TRYMAP_DEC(gb18030ext, decoded, c, c2)) |
| 313 | OUTCHAR(decoded); |
Victor Stinner | 146a2ed | 2013-10-29 00:09:41 +0100 | [diff] [blame] | 314 | else |
| 315 | return 1; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 316 | |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 317 | NEXT_IN(2); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 318 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 319 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 320 | return 0; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 321 | } |
| 322 | |
| 323 | |
| 324 | /* |
| 325 | * HZ codec |
| 326 | */ |
| 327 | |
| 328 | ENCODER_INIT(hz) |
| 329 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 330 | state->i = 0; |
| 331 | return 0; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 332 | } |
| 333 | |
| 334 | ENCODER_RESET(hz) |
| 335 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 336 | if (state->i != 0) { |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 337 | WRITEBYTE2('~', '}'); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 338 | state->i = 0; |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 339 | NEXT_OUT(2); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 340 | } |
| 341 | return 0; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 342 | } |
| 343 | |
| 344 | ENCODER(hz) |
| 345 | { |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 346 | while (*inpos < inlen) { |
| 347 | Py_UCS4 c = INCHAR1; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 348 | DBCHAR code; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 349 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 350 | if (c < 0x80) { |
| 351 | if (state->i == 0) { |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 352 | WRITEBYTE1((unsigned char)c); |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 353 | NEXT(1, 1); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 354 | } |
| 355 | else { |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 356 | WRITEBYTE3('~', '}', (unsigned char)c); |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 357 | NEXT(1, 3); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 358 | state->i = 0; |
| 359 | } |
| 360 | continue; |
| 361 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 362 | |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 363 | if (c > 0xFFFF) |
| 364 | return 1; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 365 | |
Victor Stinner | 146a2ed | 2013-10-29 00:09:41 +0100 | [diff] [blame] | 366 | if (TRYMAP_ENC(gbcommon, code, c)) |
| 367 | ; |
| 368 | else |
| 369 | return 1; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 370 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 371 | if (code & 0x8000) /* MSB set: GBK */ |
| 372 | return 1; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 373 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 374 | if (state->i == 0) { |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 375 | WRITEBYTE4('~', '{', code >> 8, code & 0xff); |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 376 | NEXT(1, 4); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 377 | state->i = 1; |
| 378 | } |
| 379 | else { |
Victor Stinner | 14c9fea | 2013-10-29 00:19:27 +0100 | [diff] [blame] | 380 | WRITEBYTE2(code >> 8, code & 0xff); |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 381 | NEXT(1, 2); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 382 | } |
| 383 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 384 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 385 | return 0; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 386 | } |
| 387 | |
| 388 | DECODER_INIT(hz) |
| 389 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 390 | state->i = 0; |
| 391 | return 0; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 392 | } |
| 393 | |
| 394 | DECODER_RESET(hz) |
| 395 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 396 | state->i = 0; |
| 397 | return 0; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 398 | } |
| 399 | |
| 400 | DECODER(hz) |
| 401 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 402 | while (inleft > 0) { |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 403 | unsigned char c = INBYTE1; |
Victor Stinner | 11bdf91 | 2013-10-28 23:18:39 +0100 | [diff] [blame] | 404 | Py_UCS4 decoded; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 405 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 406 | if (c == '~') { |
Victor Stinner | d949126 | 2013-04-14 02:06:32 +0200 | [diff] [blame] | 407 | unsigned char c2 = INBYTE2; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 408 | |
Victor Stinner | 28c63f7 | 2013-10-29 00:59:44 +0100 | [diff] [blame] | 409 | REQUIRE_INBUF(2); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 410 | if (c2 == '~') { |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 411 | OUTCHAR('~'); |
| 412 | NEXT_IN(2); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 413 | continue; |
| 414 | } |
| 415 | else if (c2 == '{' && state->i == 0) |
| 416 | state->i = 1; /* set GB */ |
| 417 | else if (c2 == '}' && state->i == 1) |
| 418 | state->i = 0; /* set ASCII */ |
| 419 | else if (c2 == '\n') |
| 420 | ; /* line-continuation */ |
| 421 | else |
Victor Stinner | 2cded9c | 2011-07-08 01:45:13 +0200 | [diff] [blame] | 422 | return 1; |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 423 | NEXT_IN(2); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 424 | continue; |
| 425 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 426 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 427 | if (c & 0x80) |
| 428 | return 1; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 429 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 430 | if (state->i == 0) { /* ASCII mode */ |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 431 | OUTCHAR(c); |
| 432 | NEXT_IN(1); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 433 | } |
| 434 | else { /* GB mode */ |
Victor Stinner | 28c63f7 | 2013-10-29 00:59:44 +0100 | [diff] [blame] | 435 | REQUIRE_INBUF(2); |
Victor Stinner | 11bdf91 | 2013-10-28 23:18:39 +0100 | [diff] [blame] | 436 | if (TRYMAP_DEC(gb2312, decoded, c, INBYTE2)) { |
| 437 | OUTCHAR(decoded); |
Victor Stinner | a0dd021 | 2013-04-11 22:09:04 +0200 | [diff] [blame] | 438 | NEXT_IN(2); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 439 | } |
| 440 | else |
Victor Stinner | 2cded9c | 2011-07-08 01:45:13 +0200 | [diff] [blame] | 441 | return 1; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 442 | } |
| 443 | } |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 444 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 445 | return 0; |
Hye-Shik Chang | 2bb146f | 2004-07-18 03:06:29 +0000 | [diff] [blame] | 446 | } |
| 447 | |
| 448 | |
| 449 | BEGIN_MAPPINGS_LIST |
| 450 | MAPPING_DECONLY(gb2312) |
| 451 | MAPPING_DECONLY(gbkext) |
| 452 | MAPPING_ENCONLY(gbcommon) |
| 453 | MAPPING_ENCDEC(gb18030ext) |
| 454 | END_MAPPINGS_LIST |
| 455 | |
| 456 | BEGIN_CODECS_LIST |
| 457 | CODEC_STATELESS(gb2312) |
| 458 | CODEC_STATELESS(gbk) |
| 459 | CODEC_STATELESS(gb18030) |
| 460 | CODEC_STATEFUL(hz) |
| 461 | END_CODECS_LIST |
| 462 | |
| 463 | I_AM_A_MODULE_FOR(cn) |