xref: /freebsd/lib/libc/locale/ascii.c (revision 52267f7411adcc76ede961420e08c0e42f42d415)
1 /*-
2  * Copyright (c) 2002-2004 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  * 4. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  */
33 
34 #include <sys/cdefs.h>
35 __FBSDID("$FreeBSD$");
36 
37 #include <errno.h>
38 #include <limits.h>
39 #include <runetype.h>
40 #include <stddef.h>
41 #include <stdio.h>
42 #include <stdlib.h>
43 #include <string.h>
44 #include <wchar.h>
45 #include "mblocal.h"
46 
47 static size_t	_ascii_mbrtowc(wchar_t * __restrict, const char * __restrict,
48 		    size_t, mbstate_t * __restrict);
49 static int	_ascii_mbsinit(const mbstate_t *);
50 static size_t	_ascii_mbsnrtowcs(wchar_t * __restrict dst,
51 		    const char ** __restrict src, size_t nms, size_t len,
52 		    mbstate_t * __restrict ps __unused);
53 static size_t	_ascii_wcrtomb(char * __restrict, wchar_t,
54 		    mbstate_t * __restrict);
55 static size_t	_ascii_wcsnrtombs(char * __restrict, const wchar_t ** __restrict,
56 		    size_t, size_t, mbstate_t * __restrict);
57 
58 int
59 _ascii_init(_RuneLocale *rl)
60 {
61 
62 	__mbrtowc = _ascii_mbrtowc;
63 	__mbsinit = _ascii_mbsinit;
64 	__mbsnrtowcs = _ascii_mbsnrtowcs;
65 	__wcrtomb = _ascii_wcrtomb;
66 	__wcsnrtombs = _ascii_wcsnrtombs;
67 	_CurrentRuneLocale = rl;
68 	__mb_cur_max = 1;
69 	__mb_sb_limit = 128;
70 	return(0);
71 }
72 
73 static int
74 _ascii_mbsinit(const mbstate_t *ps __unused)
75 {
76 
77 	/*
78 	 * Encoding is not state dependent - we are always in the
79 	 * initial state.
80 	 */
81 	return (1);
82 }
83 
84 static size_t
85 _ascii_mbrtowc(wchar_t * __restrict pwc, const char * __restrict s, size_t n,
86     mbstate_t * __restrict ps __unused)
87 {
88 
89 	if (s == NULL)
90 		/* Reset to initial shift state (no-op) */
91 		return (0);
92 	if (n == 0)
93 		/* Incomplete multibyte sequence */
94 		return ((size_t)-2);
95 	if (*s & 0x80) {
96 		errno = EILSEQ;
97 		return ((size_t)-1);
98 	}
99 	if (pwc != NULL)
100 		*pwc = (unsigned char)*s;
101 	return (*s == '\0' ? 0 : 1);
102 }
103 
104 static size_t
105 _ascii_wcrtomb(char * __restrict s, wchar_t wc,
106     mbstate_t * __restrict ps __unused)
107 {
108 
109 	if (s == NULL)
110 		/* Reset to initial shift state (no-op) */
111 		return (1);
112 	if (wc < 0 || wc > 127) {
113 		errno = EILSEQ;
114 		return ((size_t)-1);
115 	}
116 	*s = (unsigned char)wc;
117 	return (1);
118 }
119 
120 static size_t
121 _ascii_mbsnrtowcs(wchar_t * __restrict dst, const char ** __restrict src,
122     size_t nms, size_t len, mbstate_t * __restrict ps __unused)
123 {
124 	const char *s;
125 	size_t nchr;
126 
127 	if (dst == NULL) {
128 		for (s = *src; nms > 0 && *s != '\0'; s++, nms--) {
129 			if (*s & 0x80) {
130 				errno = EILSEQ;
131 				return ((size_t)-1);
132 			}
133 		}
134 		return (s - *src);
135 	}
136 
137 	s = *src;
138 	nchr = 0;
139 	while (len-- > 0 && nms-- > 0) {
140 		if (*s & 0x80) {
141 			errno = EILSEQ;
142 			return ((size_t)-1);
143 		}
144 		if ((*dst++ = (unsigned char)*s++) == L'\0') {
145 			*src = NULL;
146 			return (nchr);
147 		}
148 		nchr++;
149 	}
150 	*src = s;
151 	return (nchr);
152 }
153 
154 static size_t
155 _ascii_wcsnrtombs(char * __restrict dst, const wchar_t ** __restrict src,
156     size_t nwc, size_t len, mbstate_t * __restrict ps __unused)
157 {
158 	const wchar_t *s;
159 	size_t nchr;
160 
161 	if (dst == NULL) {
162 		for (s = *src; nwc > 0 && *s != L'\0'; s++, nwc--) {
163 			if (*s < 0 || *s > 127) {
164 				errno = EILSEQ;
165 				return ((size_t)-1);
166 			}
167 		}
168 		return (s - *src);
169 	}
170 
171 	s = *src;
172 	nchr = 0;
173 	while (len-- > 0 && nwc-- > 0) {
174 		if (*s < 0 || *s > 127) {
175 			errno = EILSEQ;
176 			return ((size_t)-1);
177 		}
178 		if ((*dst++ = *s++) == '\0') {
179 			*src = NULL;
180 			return (nchr);
181 		}
182 		nchr++;
183 	}
184 	*src = s;
185 	return (nchr);
186 }
187 
188