1 /*- 2 * Copyright (c) 2002, 2003 Tim J. Robbins. All rights reserved. 3 * Copyright (c) 1993 4 * The Regents of the University of California. All rights reserved. 5 * 6 * This code is derived from software contributed to Berkeley by 7 * Paul Borman at Krystal Technologies. 8 * 9 * Redistribution and use in source and binary forms, with or without 10 * modification, are permitted provided that the following conditions 11 * are met: 12 * 1. Redistributions of source code must retain the above copyright 13 * notice, this list of conditions and the following disclaimer. 14 * 2. Redistributions in binary form must reproduce the above copyright 15 * notice, this list of conditions and the following disclaimer in the 16 * documentation and/or other materials provided with the distribution. 17 * 3. All advertising materials mentioning features or use of this software 18 * must display the following acknowledgement: 19 * This product includes software developed by the University of 20 * California, Berkeley and its contributors. 21 * 4. Neither the name of the University nor the names of its contributors 22 * may be used to endorse or promote products derived from this software 23 * without specific prior written permission. 24 * 25 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 26 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 27 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 28 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 29 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 30 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 31 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 32 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 33 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 34 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 35 * SUCH DAMAGE. 36 */ 37 38 #include <sys/cdefs.h> 39 __FBSDID("$FreeBSD$"); 40 41 #include <sys/types.h> 42 #include <runetype.h> 43 #include <stddef.h> 44 #include <stdio.h> 45 #include <stdlib.h> 46 #include <wchar.h> 47 48 extern size_t (*__mbrtowc)(wchar_t * __restrict, const char * __restrict, 49 size_t, mbstate_t * __restrict); 50 extern size_t (*__wcrtomb)(char * __restrict, wchar_t, mbstate_t * __restrict); 51 52 int _GBK_init(_RuneLocale *); 53 size_t _GBK_mbrtowc(wchar_t * __restrict, const char * __restrict, size_t, 54 mbstate_t * __restrict); 55 size_t _GBK_wcrtomb(char * __restrict, wchar_t, mbstate_t * __restrict); 56 57 int 58 _GBK_init(_RuneLocale *rl) 59 { 60 61 __mbrtowc = _GBK_mbrtowc; 62 __wcrtomb = _GBK_wcrtomb; 63 _CurrentRuneLocale = rl; 64 __mb_cur_max = 2; 65 return (0); 66 } 67 68 static __inline int 69 _gbk_check(u_int c) 70 { 71 72 c &= 0xff; 73 return ((c >= 0x80 && c <= 0xfe) ? 2 : 1); 74 } 75 76 size_t 77 _GBK_mbrtowc(wchar_t * __restrict pwc, const char * __restrict s, size_t n, 78 mbstate_t * __restrict ps __unused) 79 { 80 wchar_t wc; 81 int i, len; 82 83 if (s == NULL) 84 /* Reset to initial shift state (no-op) */ 85 return (0); 86 if (n == 0 || (size_t)(len = _gbk_check(*s)) > n) 87 /* Incomplete multibyte sequence */ 88 return ((size_t)-2); 89 wc = 0; 90 i = len; 91 while (i-- > 0) 92 wc = (wc << 8) | (unsigned char)*s++; 93 if (pwc != NULL) 94 *pwc = wc; 95 return (wc == L'\0' ? 0 : len); 96 } 97 98 size_t 99 _GBK_wcrtomb(char * __restrict s, wchar_t wc, 100 mbstate_t * __restrict ps __unused) 101 { 102 103 if (s == NULL) 104 /* Reset to initial shift state (no-op) */ 105 return (1); 106 if (wc & 0x8000) { 107 *s++ = (wc >> 8) & 0xff; 108 *s = wc & 0xff; 109 return (2); 110 } 111 *s = wc & 0xff; 112 return (1); 113 } 114