Home | History | Annotate | Line # | Download | only in locale
      1 /*	$NetBSD: mbrtoc8.c,v 1.8 2024/08/21 18:36:11 rillig Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 2024 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  *
     16  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     17  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     18  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     19  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     20  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     21  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     22  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     23  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     24  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     25  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     26  * POSSIBILITY OF SUCH DAMAGE.
     27  */
     28 
     29 /*
     30  * mbrtoc8(&c8, s, n, ps)
     31  *
     32  *	Decode a Unicode scalar value from up to n bytes out of the
     33  *	multibyte string s, using multibyte encoding state ps, and
     34  *	store the next code unit in the UTF-8 representation of that
     35  *	scalar value at c8.
     36  *
     37  *	If the UTF-8 representation of that scalar value is multiple
     38  *	bytes long, mbrtoc8 will yield the leading byte in one call that
     39  *	consumes input, and will yield the trailing bytes in subsequent
     40  *	calls without consuming any input and returning (size_t)-3
     41  *	instead.
     42  *
     43  *	Return the number of bytes consumed on success, or:
     44  *
     45  *	- 0 if the code unit is NUL, or
     46  *	- (size_t)-3 if a trailing byte was returned without consuming
     47  *	  any additional input, or
     48  *	- (size_t)-2 if the input is incomplete, or
     49  *	- (size_t)-1 on error with errno set to EILSEQ.
     50  *
     51  *	In the case of incomplete input, the decoding state so far
     52  *	after processing s[0], s[1], ..., s[n - 1] is saved in ps, so
     53  *	subsequent calls to mbrtoc8 will pick up n bytes later into
     54  *	the input stream.
     55  *
     56  * References:
     57  *
     58  *	The Unicode Standard, Version 15.0 -- Core Specification, The
     59  *	Unicode Consortium, Sec. 3.8 `Surrogates', p. 118.
     60  *	https://www.unicode.org/versions/Unicode15.0.0/UnicodeStandard-15.0.pdf#page=144
     61  *	https://web.archive.org/web/20240718101254/https://www.unicode.org/versions/Unicode15.0.0/UnicodeStandard-15.0.pdf#page=144
     62  *
     63  *	The Unicode Standard, Version 15.0 -- Core Specification, The
     64  *	Unicode Consortium, Sec. 3.9 `Unicode Encoding Forms': UTF-8,
     65  *	p. 124.
     66  *	https://www.unicode.org/versions/Unicode15.0.0/UnicodeStandard-15.0.pdf#page=150
     67  *	https://web.archive.org/web/20240718101254/https://www.unicode.org/versions/Unicode15.0.0/UnicodeStandard-15.0.pdf#page=150
     68  *
     69  *	F. Yergeau, `UTF-8, a transformation format of ISO 10646',
     70  *	RFC 3629, Internet Engineering Task Force, November 2003.
     71  *	https://datatracker.ietf.org/doc/html/rfc3629
     72  */
     73 
     74 #include <sys/cdefs.h>
     75 __RCSID("$NetBSD: mbrtoc8.c,v 1.8 2024/08/21 18:36:11 rillig Exp $");
     76 
     77 #include "namespace.h"
     78 
     79 #include <assert.h>
     80 #include <errno.h>
     81 #include <locale.h>
     82 #include <stdalign.h>
     83 #include <stddef.h>
     84 #include <uchar.h>
     85 
     86 #include "mbrtoc32.h"
     87 #include "setlocale_local.h"
     88 
     89 struct mbrtoc8state {
     90 	char8_t		nleft;
     91 	char8_t		buf[3];
     92 	mbstate_t	mbs;
     93 };
     94 __CTASSERT(offsetof(struct mbrtoc8state, mbs) <= sizeof(mbstate_t));
     95 __CTASSERT(sizeof(struct mbrtoc32state) <= sizeof(mbstate_t) -
     96     offsetof(struct mbrtoc8state, mbs));
     97 __CTASSERT(alignof(struct mbrtoc8state) <= alignof(mbstate_t));
     98 
     99 #ifdef __weak_alias
    100 __weak_alias(mbrtoc8_l,_mbrtoc8_l)
    101 #endif
    102 
    103 size_t
    104 mbrtoc8(char8_t *restrict pc8, const char *restrict s, size_t n,
    105     mbstate_t *restrict ps)
    106 {
    107 
    108 	return mbrtoc8_l(pc8, s, n, ps, _current_locale());
    109 }
    110 
    111 size_t
    112 mbrtoc8_l(char8_t *restrict pc8, const char *restrict s, size_t n,
    113     mbstate_t *restrict ps, locale_t restrict loc)
    114 {
    115 	static mbstate_t psbuf;
    116 	struct mbrtoc8state *S;
    117 	char32_t c32;
    118 	size_t len;
    119 
    120 	/*
    121 	 * `If ps is a null pointer, each function uses its own
    122 	 *  internal mbstate_t object instead, which is initialized at
    123 	 *  program startup to the initial conversion state; the
    124 	 *  functions are not required to avoid data races with other
    125 	 *  calls to the same function in this case.  The
    126 	 *  implementation behaves as if no library function calls
    127 	 *  these functions with a null pointer for ps.'
    128 	 */
    129 	if (ps == NULL)
    130 		ps = &psbuf;
    131 
    132 	/*
    133 	 * `If s is a null pointer, the mbrtoc8 function is equivalent
    134 	 *  to the call:
    135 	 *
    136 	 *	mbrtoc8(NULL, "", 1, ps)
    137 	 *
    138 	 *  In this case, the values of the parameters pc8 and n are
    139 	 *  ignored.'
    140 	 */
    141 	if (s == NULL) {
    142 		pc8 = NULL;
    143 		s = "";
    144 		n = 1;
    145 	}
    146 
    147 	/*
    148 	 * Get the private conversion state.
    149 	 */
    150 	S = (struct mbrtoc8state *)(void *)ps;
    151 
    152 	/*
    153 	 * If there are pending trailing bytes, yield them and return
    154 	 * (size_t)-3 to indicate that no bytes of input were consumed.
    155 	 */
    156 	if (S->nleft) {
    157 		if (pc8)
    158 			*pc8 = S->buf[sizeof(S->buf) - S->nleft];
    159 		S->buf[sizeof(S->buf) - S->nleft] = 0; /* paranoia */
    160 		S->nleft--;
    161 		return (size_t)-3;
    162 	}
    163 
    164 	/*
    165 	 * Consume the next scalar value.  If no full scalar value can
    166 	 * be obtained, stop here.
    167 	 */
    168 	len = mbrtoc32_l(&c32, s, n, &S->mbs, loc);
    169 	switch (len) {
    170 	case 0:			/* NUL */
    171 		if (pc8)
    172 			*pc8 = 0;
    173 		return 0;
    174 	case (size_t)-2:	/* still incomplete after n bytes */
    175 	case (size_t)-1:	/* error */
    176 		return len;
    177 	default:		/* consumed len bytes of input */
    178 		break;
    179 	}
    180 
    181 	/*
    182 	 * We consumed a scalar value from the input.
    183 	 *
    184 	 * Encode it as UTF-8, yield the leading byte, and buffer the
    185 	 * trailing bytes to yield later.
    186 	 *
    187 	 * Table 3-6: UTF-8 Bit Distribution
    188 	 * Table 3-7: Well-Formed UTF-8 Byte Sequences
    189 	 */
    190 	switch (c32) {
    191 	case 0x00 ... 0x7f:
    192 		if (pc8)
    193 			*pc8 = c32;
    194 		_DIAGASSERT(S->nleft == 0);
    195 		break;
    196 	case 0x0080 ... 0x07ff:
    197 		if (pc8)
    198 			*pc8 = 0xc0 | __SHIFTOUT(c32, __BITS(10,6));
    199 		S->buf[2] = 0x80 | __SHIFTOUT(c32, __BITS(5,0));
    200 		S->nleft = 1;
    201 		break;
    202 	case 0x0800 ... 0xffff:
    203 		if (pc8)
    204 			*pc8 = 0xe0 | __SHIFTOUT(c32, __BITS(15,12));
    205 		S->buf[1] = 0x80 | __SHIFTOUT(c32, __BITS(11,6));
    206 		S->buf[2] = 0x80 | __SHIFTOUT(c32, __BITS(5,0));
    207 		S->nleft = 2;
    208 		break;
    209 	case 0x10000 ... 0x10ffff:
    210 		if (pc8)
    211 			*pc8 = 0xf0 | __SHIFTOUT(c32, __BITS(20,18));
    212 		S->buf[0] = 0x80 | __SHIFTOUT(c32, __BITS(17,12));
    213 		S->buf[1] = 0x80 | __SHIFTOUT(c32, __BITS(11,6));
    214 		S->buf[2] = 0x80 | __SHIFTOUT(c32, __BITS(5,0));
    215 		S->nleft = 3;
    216 		break;
    217 	default:
    218 		errno = EILSEQ;
    219 		return (size_t)-1;
    220 	}
    221 
    222 	/*
    223 	 * Return the number of bytes consumed from the input.
    224 	 */
    225 	return len;
    226 }
    227