1 /* $FreeBSD: head/lib/libiconv_modules/ZW/citrus_zw.c 281550 2015-04-15 09:09:20Z tijl $ */
2 /* $NetBSD: citrus_zw.c,v 1.4 2008/06/14 16:01:08 tnozaki Exp $ */
3
4 /*-
5 * Copyright (c)2004, 2006 Citrus Project,
6 * All rights reserved.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 * 1. Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in the
15 * documentation and/or other materials provided with the distribution.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 *
29 */
30
31 #include <sys/cdefs.h>
32 #include <sys/types.h>
33
34 #include <assert.h>
35 #include <errno.h>
36 #include <limits.h>
37 #include <stddef.h>
38 #include <stdio.h>
39 #include <stdint.h>
40 #include <stdlib.h>
41 #include <string.h>
42 #include <wchar.h>
43
44 #include "citrus_namespace.h"
45 #include "citrus_types.h"
46 #include "citrus_module.h"
47 #include "citrus_stdenc.h"
48 #include "citrus_zw.h"
49
50 /* ----------------------------------------------------------------------
51 * private stuffs used by templates
52 */
53
54 typedef struct {
55 int dummy;
56 } _ZWEncodingInfo;
57
58 typedef enum {
59 NONE, AMBIGIOUS, ASCII, GB2312
60 } _ZWCharset;
61
62 typedef struct {
63 _ZWCharset charset;
64 int chlen;
65 char ch[4];
66 } _ZWState;
67
68 #define _CEI_TO_EI(_cei_) (&(_cei_)->ei)
69 #define _CEI_TO_STATE(_cei_, _func_) (_cei_)->states.s_##_func_
70
71 #define _FUNCNAME(m) _citrus_ZW_##m
72 #define _ENCODING_INFO _ZWEncodingInfo
73 #define _ENCODING_STATE _ZWState
74 #define _ENCODING_MB_CUR_MAX(_ei_) MB_LEN_MAX
75 #define _ENCODING_IS_STATE_DEPENDENT 1
76 #define _STATE_NEEDS_EXPLICIT_INIT(_ps_) ((_ps_)->charset != NONE)
77
78 static __inline void
79 /*ARGSUSED*/
_citrus_ZW_init_state(_ZWEncodingInfo * __restrict ei __unused,_ZWState * __restrict psenc)80 _citrus_ZW_init_state(_ZWEncodingInfo * __restrict ei __unused,
81 _ZWState * __restrict psenc)
82 {
83
84 psenc->chlen = 0;
85 psenc->charset = NONE;
86 }
87
88 #if 0
89 static __inline void
90 /*ARGSUSED*/
91 _citrus_ZW_pack_state(_ZWEncodingInfo * __restrict ei __unused,
92 void *__restrict pspriv, const _ZWState * __restrict psenc)
93 {
94
95 memcpy(pspriv, (const void *)psenc, sizeof(*psenc));
96 }
97
98 static __inline void
99 /*ARGSUSED*/
100 _citrus_ZW_unpack_state(_ZWEncodingInfo * __restrict ei __unused,
101 _ZWState * __restrict psenc, const void * __restrict pspriv)
102 {
103
104 memcpy((void *)psenc, pspriv, sizeof(*psenc));
105 }
106 #endif
107
108 static int
_citrus_ZW_mbrtowc_priv(_ZWEncodingInfo * __restrict ei,wchar_t * __restrict pwc,char ** __restrict s,size_t n,_ZWState * __restrict psenc,size_t * __restrict nresult)109 _citrus_ZW_mbrtowc_priv(_ZWEncodingInfo * __restrict ei,
110 wchar_t * __restrict pwc, char **__restrict s, size_t n,
111 _ZWState * __restrict psenc, size_t * __restrict nresult)
112 {
113 char *s0;
114 wchar_t wc;
115 int ch, len;
116
117 if (*s == NULL) {
118 _citrus_ZW_init_state(ei, psenc);
119 *nresult = (size_t)_ENCODING_IS_STATE_DEPENDENT;
120 return (0);
121 }
122 s0 = *s;
123 len = 0;
124
125 #define STORE \
126 do { \
127 if (n-- < 1) { \
128 *nresult = (size_t)-2; \
129 *s = s0; \
130 return (0); \
131 } \
132 ch = (unsigned char)*s0++; \
133 if (len++ > MB_LEN_MAX || ch > 0x7F)\
134 goto ilseq; \
135 psenc->ch[psenc->chlen++] = ch; \
136 } while (/*CONSTCOND*/0)
137
138 loop:
139 switch (psenc->charset) {
140 case ASCII:
141 switch (psenc->chlen) {
142 case 0:
143 STORE;
144 switch (psenc->ch[0]) {
145 case '\0': case '\n':
146 psenc->charset = NONE;
147 }
148 /*FALLTHROUGH*/
149 case 1:
150 break;
151 default:
152 return (EINVAL);
153 }
154 ch = (unsigned char)psenc->ch[0];
155 if (ch > 0x7F)
156 goto ilseq;
157 wc = (wchar_t)ch;
158 psenc->chlen = 0;
159 break;
160 case NONE:
161 if (psenc->chlen != 0)
162 return (EINVAL);
163 STORE;
164 ch = (unsigned char)psenc->ch[0];
165 if (ch != 'z') {
166 if (ch != '\n' && ch != '\0')
167 psenc->charset = ASCII;
168 wc = (wchar_t)ch;
169 psenc->chlen = 0;
170 break;
171 }
172 psenc->charset = AMBIGIOUS;
173 psenc->chlen = 0;
174 /* FALLTHROUGH */
175 case AMBIGIOUS:
176 if (psenc->chlen != 0)
177 return (EINVAL);
178 STORE;
179 if (psenc->ch[0] != 'W') {
180 psenc->charset = ASCII;
181 wc = L'z';
182 break;
183 }
184 psenc->charset = GB2312;
185 psenc->chlen = 0;
186 /* FALLTHROUGH */
187 case GB2312:
188 switch (psenc->chlen) {
189 case 0:
190 STORE;
191 ch = (unsigned char)psenc->ch[0];
192 if (ch == '\0') {
193 psenc->charset = NONE;
194 wc = (wchar_t)ch;
195 psenc->chlen = 0;
196 break;
197 } else if (ch == '\n') {
198 psenc->charset = NONE;
199 psenc->chlen = 0;
200 goto loop;
201 }
202 /*FALLTHROUGH*/
203 case 1:
204 STORE;
205 if (psenc->ch[0] == ' ') {
206 ch = (unsigned char)psenc->ch[1];
207 wc = (wchar_t)ch;
208 psenc->chlen = 0;
209 break;
210 } else if (psenc->ch[0] == '#') {
211 ch = (unsigned char)psenc->ch[1];
212 if (ch == '\n') {
213 psenc->charset = NONE;
214 wc = (wchar_t)ch;
215 psenc->chlen = 0;
216 break;
217 } else if (ch == ' ') {
218 wc = (wchar_t)ch;
219 psenc->chlen = 0;
220 break;
221 }
222 }
223 ch = (unsigned char)psenc->ch[0];
224 if (ch < 0x21 || ch > 0x7E)
225 goto ilseq;
226 wc = (wchar_t)(ch << 8);
227 ch = (unsigned char)psenc->ch[1];
228 if (ch < 0x21 || ch > 0x7E) {
229 ilseq:
230 *nresult = (size_t)-1;
231 return (EILSEQ);
232 }
233 wc |= (wchar_t)ch;
234 psenc->chlen = 0;
235 break;
236 default:
237 return (EINVAL);
238 }
239 break;
240 default:
241 return (EINVAL);
242 }
243 if (pwc != NULL)
244 *pwc = wc;
245
246 *nresult = (size_t)(wc == 0 ? 0 : len);
247 *s = s0;
248
249 return (0);
250 }
251
252 static int
253 /*ARGSUSED*/
_citrus_ZW_wcrtomb_priv(_ZWEncodingInfo * __restrict ei __unused,char * __restrict s,size_t n,wchar_t wc,_ZWState * __restrict psenc,size_t * __restrict nresult)254 _citrus_ZW_wcrtomb_priv(_ZWEncodingInfo * __restrict ei __unused,
255 char *__restrict s, size_t n, wchar_t wc,
256 _ZWState * __restrict psenc, size_t * __restrict nresult)
257 {
258 int ch;
259
260 if (psenc->chlen != 0)
261 return (EINVAL);
262 if ((uint32_t)wc <= 0x7F) {
263 ch = (unsigned char)wc;
264 switch (psenc->charset) {
265 case NONE:
266 if (ch == '\0' || ch == '\n')
267 psenc->ch[psenc->chlen++] = ch;
268 else {
269 if (n < 4)
270 return (E2BIG);
271 n -= 4;
272 psenc->ch[psenc->chlen++] = 'z';
273 psenc->ch[psenc->chlen++] = 'W';
274 psenc->ch[psenc->chlen++] = ' ';
275 psenc->ch[psenc->chlen++] = ch;
276 psenc->charset = GB2312;
277 }
278 break;
279 case GB2312:
280 if (n < 2)
281 return (E2BIG);
282 n -= 2;
283 if (ch == '\0') {
284 psenc->ch[psenc->chlen++] = '\n';
285 psenc->ch[psenc->chlen++] = '\0';
286 psenc->charset = NONE;
287 } else if (ch == '\n') {
288 psenc->ch[psenc->chlen++] = '#';
289 psenc->ch[psenc->chlen++] = '\n';
290 psenc->charset = NONE;
291 } else {
292 psenc->ch[psenc->chlen++] = ' ';
293 psenc->ch[psenc->chlen++] = ch;
294 }
295 break;
296 default:
297 return (EINVAL);
298 }
299 } else if ((uint32_t)wc <= 0x7E7E) {
300 switch (psenc->charset) {
301 case NONE:
302 if (n < 2)
303 return (E2BIG);
304 n -= 2;
305 psenc->ch[psenc->chlen++] = 'z';
306 psenc->ch[psenc->chlen++] = 'W';
307 psenc->charset = GB2312;
308 /* FALLTHROUGH*/
309 case GB2312:
310 if (n < 2)
311 return (E2BIG);
312 n -= 2;
313 ch = (wc >> 8) & 0xFF;
314 if (ch < 0x21 || ch > 0x7E)
315 goto ilseq;
316 psenc->ch[psenc->chlen++] = ch;
317 ch = wc & 0xFF;
318 if (ch < 0x21 || ch > 0x7E)
319 goto ilseq;
320 psenc->ch[psenc->chlen++] = ch;
321 break;
322 default:
323 return (EINVAL);
324 }
325 } else {
326 ilseq:
327 *nresult = (size_t)-1;
328 return (EILSEQ);
329 }
330 memcpy(s, psenc->ch, psenc->chlen);
331 *nresult = psenc->chlen;
332 psenc->chlen = 0;
333
334 return (0);
335 }
336
337 static int
338 /*ARGSUSED*/
_citrus_ZW_put_state_reset(_ZWEncodingInfo * __restrict ei __unused,char * __restrict s,size_t n,_ZWState * __restrict psenc,size_t * __restrict nresult)339 _citrus_ZW_put_state_reset(_ZWEncodingInfo * __restrict ei __unused,
340 char * __restrict s, size_t n, _ZWState * __restrict psenc,
341 size_t * __restrict nresult)
342 {
343
344 if (psenc->chlen != 0)
345 return (EINVAL);
346 switch (psenc->charset) {
347 case GB2312:
348 if (n-- < 1)
349 return (E2BIG);
350 psenc->ch[psenc->chlen++] = '\n';
351 psenc->charset = NONE;
352 /*FALLTHROUGH*/
353 case NONE:
354 *nresult = psenc->chlen;
355 if (psenc->chlen > 0) {
356 memcpy(s, psenc->ch, psenc->chlen);
357 psenc->chlen = 0;
358 }
359 break;
360 default:
361 return (EINVAL);
362 }
363
364 return (0);
365 }
366
367 static __inline int
368 /*ARGSUSED*/
_citrus_ZW_stdenc_get_state_desc_generic(_ZWEncodingInfo * __restrict ei __unused,_ZWState * __restrict psenc,int * __restrict rstate)369 _citrus_ZW_stdenc_get_state_desc_generic(_ZWEncodingInfo * __restrict ei __unused,
370 _ZWState * __restrict psenc, int * __restrict rstate)
371 {
372
373 switch (psenc->charset) {
374 case NONE:
375 if (psenc->chlen != 0)
376 return (EINVAL);
377 *rstate = _STDENC_SDGEN_INITIAL;
378 break;
379 case AMBIGIOUS:
380 if (psenc->chlen != 0)
381 return (EINVAL);
382 *rstate = _STDENC_SDGEN_INCOMPLETE_SHIFT;
383 break;
384 case ASCII:
385 case GB2312:
386 switch (psenc->chlen) {
387 case 0:
388 *rstate = _STDENC_SDGEN_STABLE;
389 break;
390 case 1:
391 *rstate = (psenc->ch[0] == '#') ?
392 _STDENC_SDGEN_INCOMPLETE_SHIFT :
393 _STDENC_SDGEN_INCOMPLETE_CHAR;
394 break;
395 default:
396 return (EINVAL);
397 }
398 break;
399 default:
400 return (EINVAL);
401 }
402 return (0);
403 }
404
405 static __inline int
406 /*ARGSUSED*/
_citrus_ZW_stdenc_wctocs(_ZWEncodingInfo * __restrict ei __unused,_csid_t * __restrict csid,_index_t * __restrict idx,wchar_t wc)407 _citrus_ZW_stdenc_wctocs(_ZWEncodingInfo * __restrict ei __unused,
408 _csid_t * __restrict csid, _index_t * __restrict idx, wchar_t wc)
409 {
410
411 *csid = (_csid_t)(wc <= (wchar_t)0x7FU) ? 0 : 1;
412 *idx = (_index_t)wc;
413
414 return (0);
415 }
416
417 static __inline int
418 /*ARGSUSED*/
_citrus_ZW_stdenc_cstowc(_ZWEncodingInfo * __restrict ei __unused,wchar_t * __restrict wc,_csid_t csid,_index_t idx)419 _citrus_ZW_stdenc_cstowc(_ZWEncodingInfo * __restrict ei __unused,
420 wchar_t * __restrict wc, _csid_t csid, _index_t idx)
421 {
422
423 switch (csid) {
424 case 0: case 1:
425 break;
426 default:
427 return (EINVAL);
428 }
429 *wc = (wchar_t)idx;
430
431 return (0);
432 }
433
434 static void
435 /*ARGSUSED*/
_citrus_ZW_encoding_module_uninit(_ZWEncodingInfo * ei __unused)436 _citrus_ZW_encoding_module_uninit(_ZWEncodingInfo *ei __unused)
437 {
438
439 }
440
441 static int
442 /*ARGSUSED*/
_citrus_ZW_encoding_module_init(_ZWEncodingInfo * __restrict ei __unused,const void * __restrict var __unused,size_t lenvar __unused)443 _citrus_ZW_encoding_module_init(_ZWEncodingInfo * __restrict ei __unused,
444 const void *__restrict var __unused, size_t lenvar __unused)
445 {
446
447 return (0);
448 }
449
450 /* ----------------------------------------------------------------------
451 * public interface for stdenc
452 */
453
454 _CITRUS_STDENC_DECLS(ZW);
455 _CITRUS_STDENC_DEF_OPS(ZW);
456
457 #include "citrus_stdenc_template.h"
458