1caf54c4fSMartin Matuska /*-
26c95142eSMartin Matuska  * Copyright (c) 2003-2011 Tim Kientzle
36c95142eSMartin Matuska  * Copyright (c) 2011-2012 Michihiro NAKAJIMA
4caf54c4fSMartin Matuska  * All rights reserved.
5caf54c4fSMartin Matuska  *
6caf54c4fSMartin Matuska  * Redistribution and use in source and binary forms, with or without
7caf54c4fSMartin Matuska  * modification, are permitted provided that the following conditions
8caf54c4fSMartin Matuska  * are met:
9caf54c4fSMartin Matuska  * 1. Redistributions of source code must retain the above copyright
10caf54c4fSMartin Matuska  *    notice, this list of conditions and the following disclaimer.
11caf54c4fSMartin Matuska  * 2. Redistributions in binary form must reproduce the above copyright
12caf54c4fSMartin Matuska  *    notice, this list of conditions and the following disclaimer in the
13caf54c4fSMartin Matuska  *    documentation and/or other materials provided with the distribution.
14caf54c4fSMartin Matuska  *
15caf54c4fSMartin Matuska  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
16caf54c4fSMartin Matuska  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17caf54c4fSMartin Matuska  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18caf54c4fSMartin Matuska  * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
19caf54c4fSMartin Matuska  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20caf54c4fSMartin Matuska  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21caf54c4fSMartin Matuska  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22caf54c4fSMartin Matuska  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23caf54c4fSMartin Matuska  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24caf54c4fSMartin Matuska  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25caf54c4fSMartin Matuska  */
26caf54c4fSMartin Matuska 
27caf54c4fSMartin Matuska #include "archive_platform.h"
286c22d9efSMartin Matuska __FBSDID("$FreeBSD$");
29caf54c4fSMartin Matuska 
30caf54c4fSMartin Matuska /*
31caf54c4fSMartin Matuska  * Basic resizable string support, to simplify manipulating arbitrary-sized
32caf54c4fSMartin Matuska  * strings while minimizing heap activity.
336c95142eSMartin Matuska  *
346c95142eSMartin Matuska  * In particular, the buffer used by a string object is only grown, it
356c95142eSMartin Matuska  * never shrinks, so you can clear and reuse the same string object
366c95142eSMartin Matuska  * without incurring additional memory allocations.
37caf54c4fSMartin Matuska  */
38caf54c4fSMartin Matuska 
396c95142eSMartin Matuska #ifdef HAVE_ERRNO_H
406c95142eSMartin Matuska #include <errno.h>
416c95142eSMartin Matuska #endif
426c95142eSMartin Matuska #ifdef HAVE_ICONV_H
436c95142eSMartin Matuska #include <iconv.h>
446c95142eSMartin Matuska #endif
456c95142eSMartin Matuska #ifdef HAVE_LANGINFO_H
466c95142eSMartin Matuska #include <langinfo.h>
476c95142eSMartin Matuska #endif
486c95142eSMartin Matuska #ifdef HAVE_LOCALCHARSET_H
496c95142eSMartin Matuska #include <localcharset.h>
506c95142eSMartin Matuska #endif
51caf54c4fSMartin Matuska #ifdef HAVE_STDLIB_H
52caf54c4fSMartin Matuska #include <stdlib.h>
53caf54c4fSMartin Matuska #endif
54caf54c4fSMartin Matuska #ifdef HAVE_STRING_H
55caf54c4fSMartin Matuska #include <string.h>
56caf54c4fSMartin Matuska #endif
57caf54c4fSMartin Matuska #ifdef HAVE_WCHAR_H
58caf54c4fSMartin Matuska #include <wchar.h>
59caf54c4fSMartin Matuska #endif
60caf54c4fSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
61caf54c4fSMartin Matuska #include <windows.h>
626c95142eSMartin Matuska #include <locale.h>
636c95142eSMartin Matuska #endif
64caf54c4fSMartin Matuska 
656c95142eSMartin Matuska #include "archive_endian.h"
66caf54c4fSMartin Matuska #include "archive_private.h"
67caf54c4fSMartin Matuska #include "archive_string.h"
686c95142eSMartin Matuska #include "archive_string_composition.h"
69caf54c4fSMartin Matuska 
706c95142eSMartin Matuska #if !defined(HAVE_WMEMCPY) && !defined(wmemcpy)
716c95142eSMartin Matuska #define wmemcpy(a,b,i)  (wchar_t *)memcpy((a), (b), (i) * sizeof(wchar_t))
726c95142eSMartin Matuska #endif
736c95142eSMartin Matuska 
74cdf63a70SMartin Matuska #if !defined(HAVE_WMEMMOVE) && !defined(wmemmove)
75cdf63a70SMartin Matuska #define wmemmove(a,b,i)  (wchar_t *)memmove((a), (b), (i) * sizeof(wchar_t))
76cdf63a70SMartin Matuska #endif
77cdf63a70SMartin Matuska 
786c95142eSMartin Matuska struct archive_string_conv {
796c95142eSMartin Matuska 	struct archive_string_conv	*next;
806c95142eSMartin Matuska 	char				*from_charset;
816c95142eSMartin Matuska 	char				*to_charset;
826c95142eSMartin Matuska 	unsigned			 from_cp;
836c95142eSMartin Matuska 	unsigned			 to_cp;
846c95142eSMartin Matuska 	/* Set 1 if from_charset and to_charset are the same. */
856c95142eSMartin Matuska 	int				 same;
866c95142eSMartin Matuska 	int				 flag;
876c95142eSMartin Matuska #define SCONV_TO_CHARSET	1	/* MBS is being converted to specified
886c95142eSMartin Matuska 					 * charset. */
896c95142eSMartin Matuska #define SCONV_FROM_CHARSET	(1<<1)	/* MBS is being converted from
906c95142eSMartin Matuska 					 * specified charset. */
916c95142eSMartin Matuska #define SCONV_BEST_EFFORT 	(1<<2)	/* Copy at least ASCII code. */
926c95142eSMartin Matuska #define SCONV_WIN_CP	 	(1<<3)	/* Use Windows API for converting
936c95142eSMartin Matuska 					 * MBS. */
946c95142eSMartin Matuska #define SCONV_UTF8_LIBARCHIVE_2 (1<<4)	/* Incorrect UTF-8 made by libarchive
956c95142eSMartin Matuska 					 * 2.x in the wrong assumption. */
966c95142eSMartin Matuska #define SCONV_NORMALIZATION_C	(1<<6)	/* Need normalization to be Form C.
976c95142eSMartin Matuska 					 * Before UTF-8 characters are actually
986c95142eSMartin Matuska 					 * processed. */
996c95142eSMartin Matuska #define SCONV_NORMALIZATION_D	(1<<7)	/* Need normalization to be Form D.
1006c95142eSMartin Matuska 					 * Before UTF-8 characters are actually
1016c95142eSMartin Matuska 					 * processed.
1026c95142eSMartin Matuska 					 * Currently this only for MAC OS X. */
1036c95142eSMartin Matuska #define SCONV_TO_UTF8		(1<<8)	/* "to charset" side is UTF-8. */
1046c95142eSMartin Matuska #define SCONV_FROM_UTF8		(1<<9)	/* "from charset" side is UTF-8. */
1056c95142eSMartin Matuska #define SCONV_TO_UTF16BE 	(1<<10)	/* "to charset" side is UTF-16BE. */
1066c95142eSMartin Matuska #define SCONV_FROM_UTF16BE 	(1<<11)	/* "from charset" side is UTF-16BE. */
1076c95142eSMartin Matuska #define SCONV_TO_UTF16LE 	(1<<12)	/* "to charset" side is UTF-16LE. */
1086c95142eSMartin Matuska #define SCONV_FROM_UTF16LE 	(1<<13)	/* "from charset" side is UTF-16LE. */
1096c95142eSMartin Matuska #define SCONV_TO_UTF16		(SCONV_TO_UTF16BE | SCONV_TO_UTF16LE)
1106c95142eSMartin Matuska #define SCONV_FROM_UTF16	(SCONV_FROM_UTF16BE | SCONV_FROM_UTF16LE)
1116c95142eSMartin Matuska 
1126c95142eSMartin Matuska #if HAVE_ICONV
1136c95142eSMartin Matuska 	iconv_t				 cd;
1146c95142eSMartin Matuska 	iconv_t				 cd_w;/* Use at archive_mstring on
1156c95142eSMartin Matuska 				 	       * Windows. */
1166c95142eSMartin Matuska #endif
1176c95142eSMartin Matuska 	/* A temporary buffer for normalization. */
1186c95142eSMartin Matuska 	struct archive_string		 utftmp;
1196c95142eSMartin Matuska 	int (*converter[2])(struct archive_string *, const void *, size_t,
1206c95142eSMartin Matuska 	    struct archive_string_conv *);
1216c95142eSMartin Matuska 	int				 nconverter;
1226c95142eSMartin Matuska };
1236c95142eSMartin Matuska 
1246c95142eSMartin Matuska #define CP_C_LOCALE	0	/* "C" locale only for this file. */
1256c95142eSMartin Matuska #define CP_UTF16LE	1200
1266c95142eSMartin Matuska #define CP_UTF16BE	1201
1276c95142eSMartin Matuska 
1286c95142eSMartin Matuska #define IS_HIGH_SURROGATE_LA(uc) ((uc) >= 0xD800 && (uc) <= 0xDBFF)
1296c95142eSMartin Matuska #define IS_LOW_SURROGATE_LA(uc)	 ((uc) >= 0xDC00 && (uc) <= 0xDFFF)
1306c95142eSMartin Matuska #define IS_SURROGATE_PAIR_LA(uc) ((uc) >= 0xD800 && (uc) <= 0xDFFF)
1316c95142eSMartin Matuska #define UNICODE_MAX		0x10FFFF
1326c95142eSMartin Matuska #define UNICODE_R_CHAR		0xFFFD	/* Replacement character. */
1336c95142eSMartin Matuska /* Set U+FFFD(Replacement character) in UTF-8. */
134cdf63a70SMartin Matuska static const char utf8_replacement_char[] = {0xef, 0xbf, 0xbd};
1356c95142eSMartin Matuska 
1366c95142eSMartin Matuska static struct archive_string_conv *find_sconv_object(struct archive *,
1376c95142eSMartin Matuska 	const char *, const char *);
1386c95142eSMartin Matuska static void add_sconv_object(struct archive *, struct archive_string_conv *);
1396c95142eSMartin Matuska static struct archive_string_conv *create_sconv_object(const char *,
1406c95142eSMartin Matuska 	const char *, unsigned, int);
1416c95142eSMartin Matuska static void free_sconv_object(struct archive_string_conv *);
1426c95142eSMartin Matuska static struct archive_string_conv *get_sconv_object(struct archive *,
1436c95142eSMartin Matuska 	const char *, const char *, int);
1446c95142eSMartin Matuska static unsigned make_codepage_from_charset(const char *);
1456c95142eSMartin Matuska static unsigned get_current_codepage(void);
1466c95142eSMartin Matuska static unsigned get_current_oemcp(void);
1476c95142eSMartin Matuska static size_t mbsnbytes(const void *, size_t);
1486c95142eSMartin Matuska static size_t utf16nbytes(const void *, size_t);
1496c95142eSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
1506c95142eSMartin Matuska static int archive_wstring_append_from_mbs_in_codepage(
1516c95142eSMartin Matuska     struct archive_wstring *, const char *, size_t,
1526c95142eSMartin Matuska     struct archive_string_conv *);
1536c95142eSMartin Matuska static int archive_string_append_from_wcs_in_codepage(struct archive_string *,
1546c95142eSMartin Matuska     const wchar_t *, size_t, struct archive_string_conv *);
1556c95142eSMartin Matuska static int is_big_endian(void);
1566c95142eSMartin Matuska static int strncat_in_codepage(struct archive_string *, const void *,
1576c95142eSMartin Matuska     size_t, struct archive_string_conv *);
158fd082e96SMartin Matuska static int win_strncat_from_utf16be(struct archive_string *, const void *,
159fd082e96SMartin Matuska     size_t, struct archive_string_conv *);
160fd082e96SMartin Matuska static int win_strncat_from_utf16le(struct archive_string *, const void *,
161fd082e96SMartin Matuska     size_t, struct archive_string_conv *);
162fd082e96SMartin Matuska static int win_strncat_to_utf16be(struct archive_string *, const void *,
163fd082e96SMartin Matuska     size_t, struct archive_string_conv *);
164fd082e96SMartin Matuska static int win_strncat_to_utf16le(struct archive_string *, const void *,
165fd082e96SMartin Matuska     size_t, struct archive_string_conv *);
1666c95142eSMartin Matuska #endif
167fd082e96SMartin Matuska static int best_effort_strncat_from_utf16be(struct archive_string *,
168fd082e96SMartin Matuska     const void *, size_t, struct archive_string_conv *);
169fd082e96SMartin Matuska static int best_effort_strncat_from_utf16le(struct archive_string *,
170fd082e96SMartin Matuska     const void *, size_t, struct archive_string_conv *);
171fd082e96SMartin Matuska static int best_effort_strncat_to_utf16be(struct archive_string *,
172fd082e96SMartin Matuska     const void *, size_t, struct archive_string_conv *);
173fd082e96SMartin Matuska static int best_effort_strncat_to_utf16le(struct archive_string *,
174fd082e96SMartin Matuska     const void *, size_t, struct archive_string_conv *);
1756c95142eSMartin Matuska #if defined(HAVE_ICONV)
1766c95142eSMartin Matuska static int iconv_strncat_in_locale(struct archive_string *, const void *,
1776c95142eSMartin Matuska     size_t, struct archive_string_conv *);
1786c95142eSMartin Matuska #endif
179fd082e96SMartin Matuska static int best_effort_strncat_in_locale(struct archive_string *,
180fd082e96SMartin Matuska     const void *, size_t, struct archive_string_conv *);
1816c95142eSMartin Matuska static int _utf8_to_unicode(uint32_t *, const char *, size_t);
1826c95142eSMartin Matuska static int utf8_to_unicode(uint32_t *, const char *, size_t);
1836c95142eSMartin Matuska static inline uint32_t combine_surrogate_pair(uint32_t, uint32_t);
1846c95142eSMartin Matuska static int cesu8_to_unicode(uint32_t *, const char *, size_t);
1856c95142eSMartin Matuska static size_t unicode_to_utf8(char *, size_t, uint32_t);
1866c95142eSMartin Matuska static int utf16_to_unicode(uint32_t *, const char *, size_t, int);
1876c95142eSMartin Matuska static size_t unicode_to_utf16be(char *, size_t, uint32_t);
1886c95142eSMartin Matuska static size_t unicode_to_utf16le(char *, size_t, uint32_t);
1896c95142eSMartin Matuska static int strncat_from_utf8_libarchive2(struct archive_string *,
1906c95142eSMartin Matuska     const void *, size_t, struct archive_string_conv *);
1916c95142eSMartin Matuska static int strncat_from_utf8_to_utf8(struct archive_string *, const void *,
1926c95142eSMartin Matuska     size_t, struct archive_string_conv *);
1936c95142eSMartin Matuska static int archive_string_normalize_C(struct archive_string *, const void *,
1946c95142eSMartin Matuska     size_t, struct archive_string_conv *);
1956c95142eSMartin Matuska static int archive_string_normalize_D(struct archive_string *, const void *,
1966c95142eSMartin Matuska     size_t, struct archive_string_conv *);
1976c95142eSMartin Matuska static int archive_string_append_unicode(struct archive_string *,
1986c95142eSMartin Matuska     const void *, size_t, struct archive_string_conv *);
1996c95142eSMartin Matuska 
2006c95142eSMartin Matuska static struct archive_string *
2016c95142eSMartin Matuska archive_string_append(struct archive_string *as, const char *p, size_t s)
202caf54c4fSMartin Matuska {
2036c95142eSMartin Matuska 	if (archive_string_ensure(as, as->length + s + 1) == NULL)
2046c95142eSMartin Matuska 		return (NULL);
205cdf63a70SMartin Matuska 	memmove(as->s + as->length, p, s);
206caf54c4fSMartin Matuska 	as->length += s;
2076c95142eSMartin Matuska 	as->s[as->length] = 0;
2086c95142eSMartin Matuska 	return (as);
2096c95142eSMartin Matuska }
2106c95142eSMartin Matuska 
2116c95142eSMartin Matuska static struct archive_wstring *
2126c95142eSMartin Matuska archive_wstring_append(struct archive_wstring *as, const wchar_t *p, size_t s)
2136c95142eSMartin Matuska {
2146c95142eSMartin Matuska 	if (archive_wstring_ensure(as, as->length + s + 1) == NULL)
2156c95142eSMartin Matuska 		return (NULL);
216cdf63a70SMartin Matuska 	wmemmove(as->s + as->length, p, s);
2176c95142eSMartin Matuska 	as->length += s;
2186c95142eSMartin Matuska 	as->s[as->length] = 0;
219caf54c4fSMartin Matuska 	return (as);
220caf54c4fSMartin Matuska }
221caf54c4fSMartin Matuska 
222caf54c4fSMartin Matuska void
2236c95142eSMartin Matuska archive_string_concat(struct archive_string *dest, struct archive_string *src)
224caf54c4fSMartin Matuska {
2256c95142eSMartin Matuska 	if (archive_string_append(dest, src->s, src->length) == NULL)
226caf54c4fSMartin Matuska 		__archive_errx(1, "Out of memory");
227caf54c4fSMartin Matuska }
228caf54c4fSMartin Matuska 
229caf54c4fSMartin Matuska void
230fd082e96SMartin Matuska archive_wstring_concat(struct archive_wstring *dest,
231fd082e96SMartin Matuska     struct archive_wstring *src)
232caf54c4fSMartin Matuska {
2336c95142eSMartin Matuska 	if (archive_wstring_append(dest, src->s, src->length) == NULL)
234caf54c4fSMartin Matuska 		__archive_errx(1, "Out of memory");
235caf54c4fSMartin Matuska }
236caf54c4fSMartin Matuska 
237caf54c4fSMartin Matuska void
2386c95142eSMartin Matuska archive_string_free(struct archive_string *as)
239caf54c4fSMartin Matuska {
240caf54c4fSMartin Matuska 	as->length = 0;
241caf54c4fSMartin Matuska 	as->buffer_length = 0;
242caf54c4fSMartin Matuska 	free(as->s);
243caf54c4fSMartin Matuska 	as->s = NULL;
244caf54c4fSMartin Matuska }
2456c95142eSMartin Matuska 
2466c95142eSMartin Matuska void
2476c95142eSMartin Matuska archive_wstring_free(struct archive_wstring *as)
2486c95142eSMartin Matuska {
2496c95142eSMartin Matuska 	as->length = 0;
2506c95142eSMartin Matuska 	as->buffer_length = 0;
2516c95142eSMartin Matuska 	free(as->s);
2526c95142eSMartin Matuska 	as->s = NULL;
2536c95142eSMartin Matuska }
2546c95142eSMartin Matuska 
2556c95142eSMartin Matuska struct archive_wstring *
2566c95142eSMartin Matuska archive_wstring_ensure(struct archive_wstring *as, size_t s)
2576c95142eSMartin Matuska {
2586c95142eSMartin Matuska 	return (struct archive_wstring *)
2596c95142eSMartin Matuska 		archive_string_ensure((struct archive_string *)as,
2606c95142eSMartin Matuska 					s * sizeof(wchar_t));
261caf54c4fSMartin Matuska }
262caf54c4fSMartin Matuska 
263caf54c4fSMartin Matuska /* Returns NULL on any allocation failure. */
264caf54c4fSMartin Matuska struct archive_string *
2656c95142eSMartin Matuska archive_string_ensure(struct archive_string *as, size_t s)
266caf54c4fSMartin Matuska {
2676c95142eSMartin Matuska 	char *p;
2686c95142eSMartin Matuska 	size_t new_length;
2696c95142eSMartin Matuska 
270caf54c4fSMartin Matuska 	/* If buffer is already big enough, don't reallocate. */
271caf54c4fSMartin Matuska 	if (as->s && (s <= as->buffer_length))
272caf54c4fSMartin Matuska 		return (as);
273caf54c4fSMartin Matuska 
274caf54c4fSMartin Matuska 	/*
275caf54c4fSMartin Matuska 	 * Growing the buffer at least exponentially ensures that
276caf54c4fSMartin Matuska 	 * append operations are always linear in the number of
277caf54c4fSMartin Matuska 	 * characters appended.  Using a smaller growth rate for
278caf54c4fSMartin Matuska 	 * larger buffers reduces memory waste somewhat at the cost of
279caf54c4fSMartin Matuska 	 * a larger constant factor.
280caf54c4fSMartin Matuska 	 */
281caf54c4fSMartin Matuska 	if (as->buffer_length < 32)
282caf54c4fSMartin Matuska 		/* Start with a minimum 32-character buffer. */
2836c95142eSMartin Matuska 		new_length = 32;
284caf54c4fSMartin Matuska 	else if (as->buffer_length < 8192)
285caf54c4fSMartin Matuska 		/* Buffers under 8k are doubled for speed. */
2866c95142eSMartin Matuska 		new_length = as->buffer_length + as->buffer_length;
287caf54c4fSMartin Matuska 	else {
288caf54c4fSMartin Matuska 		/* Buffers 8k and over grow by at least 25% each time. */
2896c95142eSMartin Matuska 		new_length = as->buffer_length + as->buffer_length / 4;
2906c95142eSMartin Matuska 		/* Be safe: If size wraps, fail. */
2916c95142eSMartin Matuska 		if (new_length < as->buffer_length) {
2926c95142eSMartin Matuska 			/* On failure, wipe the string and return NULL. */
2936c95142eSMartin Matuska 			archive_string_free(as);
2946c95142eSMartin Matuska 			errno = ENOMEM;/* Make sure errno has ENOMEM. */
295caf54c4fSMartin Matuska 			return (NULL);
296caf54c4fSMartin Matuska 		}
297caf54c4fSMartin Matuska 	}
298caf54c4fSMartin Matuska 	/*
299caf54c4fSMartin Matuska 	 * The computation above is a lower limit to how much we'll
300caf54c4fSMartin Matuska 	 * grow the buffer.  In any case, we have to grow it enough to
301caf54c4fSMartin Matuska 	 * hold the request.
302caf54c4fSMartin Matuska 	 */
3036c95142eSMartin Matuska 	if (new_length < s)
3046c95142eSMartin Matuska 		new_length = s;
305caf54c4fSMartin Matuska 	/* Now we can reallocate the buffer. */
3066c95142eSMartin Matuska 	p = (char *)realloc(as->s, new_length);
3076c95142eSMartin Matuska 	if (p == NULL) {
3086c95142eSMartin Matuska 		/* On failure, wipe the string and return NULL. */
3096c95142eSMartin Matuska 		archive_string_free(as);
3106c95142eSMartin Matuska 		errno = ENOMEM;/* Make sure errno has ENOMEM. */
311caf54c4fSMartin Matuska 		return (NULL);
3126c95142eSMartin Matuska 	}
3136c95142eSMartin Matuska 
3146c95142eSMartin Matuska 	as->s = p;
3156c95142eSMartin Matuska 	as->buffer_length = new_length;
316caf54c4fSMartin Matuska 	return (as);
317caf54c4fSMartin Matuska }
318caf54c4fSMartin Matuska 
3196c95142eSMartin Matuska /*
3206c95142eSMartin Matuska  * TODO: See if there's a way to avoid scanning
3216c95142eSMartin Matuska  * the source string twice.  Then test to see
3226c95142eSMartin Matuska  * if it actually helps (remember that we're almost
3236c95142eSMartin Matuska  * always called with pretty short arguments, so
3246c95142eSMartin Matuska  * such an optimization might not help).
3256c95142eSMartin Matuska  */
326caf54c4fSMartin Matuska struct archive_string *
3276c95142eSMartin Matuska archive_strncat(struct archive_string *as, const void *_p, size_t n)
328caf54c4fSMartin Matuska {
329caf54c4fSMartin Matuska 	size_t s;
330caf54c4fSMartin Matuska 	const char *p, *pp;
331caf54c4fSMartin Matuska 
332caf54c4fSMartin Matuska 	p = (const char *)_p;
333caf54c4fSMartin Matuska 
334caf54c4fSMartin Matuska 	/* Like strlen(p), except won't examine positions beyond p[n]. */
335caf54c4fSMartin Matuska 	s = 0;
336caf54c4fSMartin Matuska 	pp = p;
337caf54c4fSMartin Matuska 	while (s < n && *pp) {
338caf54c4fSMartin Matuska 		pp++;
339caf54c4fSMartin Matuska 		s++;
340caf54c4fSMartin Matuska 	}
3416c95142eSMartin Matuska 	if ((as = archive_string_append(as, p, s)) == NULL)
342caf54c4fSMartin Matuska 		__archive_errx(1, "Out of memory");
3436c95142eSMartin Matuska 	return (as);
344caf54c4fSMartin Matuska }
3456c95142eSMartin Matuska 
3466c95142eSMartin Matuska struct archive_wstring *
3476c95142eSMartin Matuska archive_wstrncat(struct archive_wstring *as, const wchar_t *p, size_t n)
3486c95142eSMartin Matuska {
3496c95142eSMartin Matuska 	size_t s;
3506c95142eSMartin Matuska 	const wchar_t *pp;
3516c95142eSMartin Matuska 
3526c95142eSMartin Matuska 	/* Like strlen(p), except won't examine positions beyond p[n]. */
3536c95142eSMartin Matuska 	s = 0;
3546c95142eSMartin Matuska 	pp = p;
3556c95142eSMartin Matuska 	while (s < n && *pp) {
3566c95142eSMartin Matuska 		pp++;
3576c95142eSMartin Matuska 		s++;
358caf54c4fSMartin Matuska 	}
3596c95142eSMartin Matuska 	if ((as = archive_wstring_append(as, p, s)) == NULL)
3606c95142eSMartin Matuska 		__archive_errx(1, "Out of memory");
3616c95142eSMartin Matuska 	return (as);
362caf54c4fSMartin Matuska }
3636c95142eSMartin Matuska 
3646c95142eSMartin Matuska struct archive_string *
3656c95142eSMartin Matuska archive_strcat(struct archive_string *as, const void *p)
3666c95142eSMartin Matuska {
3676c95142eSMartin Matuska 	/* strcat is just strncat without an effective limit.
3686c95142eSMartin Matuska 	 * Assert that we'll never get called with a source
3696c95142eSMartin Matuska 	 * string over 16MB.
3706c95142eSMartin Matuska 	 * TODO: Review all uses of strcat in the source
3716c95142eSMartin Matuska 	 * and try to replace them with strncat().
3726c95142eSMartin Matuska 	 */
3736c95142eSMartin Matuska 	return archive_strncat(as, p, 0x1000000);
374caf54c4fSMartin Matuska }
3756c95142eSMartin Matuska 
3766c95142eSMartin Matuska struct archive_wstring *
3776c95142eSMartin Matuska archive_wstrcat(struct archive_wstring *as, const wchar_t *p)
3786c95142eSMartin Matuska {
3796c95142eSMartin Matuska 	/* Ditto. */
3806c95142eSMartin Matuska 	return archive_wstrncat(as, p, 0x1000000);
381caf54c4fSMartin Matuska }
3826c95142eSMartin Matuska 
3836c95142eSMartin Matuska struct archive_string *
3846c95142eSMartin Matuska archive_strappend_char(struct archive_string *as, char c)
3856c95142eSMartin Matuska {
3866c95142eSMartin Matuska 	if ((as = archive_string_append(as, &c, 1)) == NULL)
3876c95142eSMartin Matuska 		__archive_errx(1, "Out of memory");
3886c95142eSMartin Matuska 	return (as);
3896c95142eSMartin Matuska }
3906c95142eSMartin Matuska 
3916c95142eSMartin Matuska struct archive_wstring *
3926c95142eSMartin Matuska archive_wstrappend_wchar(struct archive_wstring *as, wchar_t c)
3936c95142eSMartin Matuska {
3946c95142eSMartin Matuska 	if ((as = archive_wstring_append(as, &c, 1)) == NULL)
3956c95142eSMartin Matuska 		__archive_errx(1, "Out of memory");
3966c95142eSMartin Matuska 	return (as);
3976c95142eSMartin Matuska }
3986c95142eSMartin Matuska 
3996c95142eSMartin Matuska /*
4006c95142eSMartin Matuska  * Get the "current character set" name to use with iconv.
4016c95142eSMartin Matuska  * On FreeBSD, the empty character set name "" chooses
4026c95142eSMartin Matuska  * the correct character encoding for the current locale,
4036c95142eSMartin Matuska  * so this isn't necessary.
4046c95142eSMartin Matuska  * But iconv on Mac OS 10.6 doesn't seem to handle this correctly;
4056c95142eSMartin Matuska  * on that system, we have to explicitly call nl_langinfo()
4066c95142eSMartin Matuska  * to get the right name.  Not sure about other platforms.
4076c95142eSMartin Matuska  *
4086c95142eSMartin Matuska  * NOTE: GNU libiconv does not recognize the character-set name
4096c95142eSMartin Matuska  * which some platform nl_langinfo(CODESET) returns, so we should
4106c95142eSMartin Matuska  * use locale_charset() instead of nl_langinfo(CODESET) for GNU libiconv.
4116c95142eSMartin Matuska  */
4126c95142eSMartin Matuska static const char *
4136c95142eSMartin Matuska default_iconv_charset(const char *charset) {
4146c95142eSMartin Matuska 	if (charset != NULL && charset[0] != '\0')
4156c95142eSMartin Matuska 		return charset;
4166c95142eSMartin Matuska #if HAVE_LOCALE_CHARSET && !defined(__APPLE__)
4176c95142eSMartin Matuska 	/* locale_charset() is broken on Mac OS */
4186c95142eSMartin Matuska 	return locale_charset();
4196c95142eSMartin Matuska #elif HAVE_NL_LANGINFO
4206c95142eSMartin Matuska 	return nl_langinfo(CODESET);
4216c95142eSMartin Matuska #else
4226c95142eSMartin Matuska 	return "";
4236c95142eSMartin Matuska #endif
424caf54c4fSMartin Matuska }
425caf54c4fSMartin Matuska 
426caf54c4fSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
427caf54c4fSMartin Matuska 
428caf54c4fSMartin Matuska /*
4296c95142eSMartin Matuska  * Convert MBS to WCS.
4306c95142eSMartin Matuska  * Note: returns -1 if conversion fails.
4316c95142eSMartin Matuska  */
4326c95142eSMartin Matuska int
4336c95142eSMartin Matuska archive_wstring_append_from_mbs(struct archive_wstring *dest,
4346c95142eSMartin Matuska     const char *p, size_t len)
4356c95142eSMartin Matuska {
436fd082e96SMartin Matuska 	return archive_wstring_append_from_mbs_in_codepage(dest, p, len, NULL);
4376c95142eSMartin Matuska }
4386c95142eSMartin Matuska 
4396c95142eSMartin Matuska static int
4406c95142eSMartin Matuska archive_wstring_append_from_mbs_in_codepage(struct archive_wstring *dest,
4416c95142eSMartin Matuska     const char *s, size_t length, struct archive_string_conv *sc)
4426c95142eSMartin Matuska {
4436c95142eSMartin Matuska 	int count, ret = 0;
4446c95142eSMartin Matuska 	UINT from_cp;
4456c95142eSMartin Matuska 
4466c95142eSMartin Matuska 	if (sc != NULL)
4476c95142eSMartin Matuska 		from_cp = sc->from_cp;
4486c95142eSMartin Matuska 	else
4496c95142eSMartin Matuska 		from_cp = get_current_codepage();
4506c95142eSMartin Matuska 
4516c95142eSMartin Matuska 	if (from_cp == CP_C_LOCALE) {
4526c95142eSMartin Matuska 		/*
4536c95142eSMartin Matuska 		 * "C" locale special process.
4546c95142eSMartin Matuska 		 */
4556c95142eSMartin Matuska 		wchar_t *ws;
4566c95142eSMartin Matuska 		const unsigned char *mp;
4576c95142eSMartin Matuska 
4586c95142eSMartin Matuska 		if (NULL == archive_wstring_ensure(dest,
4596c95142eSMartin Matuska 		    dest->length + length + 1))
4606c95142eSMartin Matuska 			return (-1);
4616c95142eSMartin Matuska 
4626c95142eSMartin Matuska 		ws = dest->s + dest->length;
4636c95142eSMartin Matuska 		mp = (const unsigned char *)s;
4646c95142eSMartin Matuska 		count = 0;
4656c95142eSMartin Matuska 		while (count < (int)length && *mp) {
4666c95142eSMartin Matuska 			*ws++ = (wchar_t)*mp++;
4676c95142eSMartin Matuska 			count++;
4686c95142eSMartin Matuska 		}
469fd082e96SMartin Matuska 	} else if (sc != NULL &&
470fd082e96SMartin Matuska 	    (sc->flag & (SCONV_NORMALIZATION_C | SCONV_NORMALIZATION_D))) {
4716c95142eSMartin Matuska 		/*
4726c95142eSMartin Matuska 		 * Normalize UTF-8 and UTF-16BE and convert it directly
4736c95142eSMartin Matuska 		 * to UTF-16 as wchar_t.
4746c95142eSMartin Matuska 		 */
4756c95142eSMartin Matuska 		struct archive_string u16;
4766c95142eSMartin Matuska 		int saved_flag = sc->flag;/* save current flag. */
4776c95142eSMartin Matuska 
4786c95142eSMartin Matuska 		if (is_big_endian())
4796c95142eSMartin Matuska 			sc->flag |= SCONV_TO_UTF16BE;
4806c95142eSMartin Matuska 		else
4816c95142eSMartin Matuska 			sc->flag |= SCONV_TO_UTF16LE;
4826c95142eSMartin Matuska 
4836c95142eSMartin Matuska 		if (sc->flag & SCONV_FROM_UTF16) {
4846c95142eSMartin Matuska 			/*
4856c95142eSMartin Matuska 			 *  UTF-16BE/LE NFD ===> UTF-16 NFC
486fd082e96SMartin Matuska 			 *  UTF-16BE/LE NFC ===> UTF-16 NFD
4876c95142eSMartin Matuska 			 */
488acc60b03SMartin Matuska 			count = (int)utf16nbytes(s, length);
4896c95142eSMartin Matuska 		} else {
4906c95142eSMartin Matuska 			/*
4916c95142eSMartin Matuska 			 *  UTF-8 NFD ===> UTF-16 NFC
492fd082e96SMartin Matuska 			 *  UTF-8 NFC ===> UTF-16 NFD
4936c95142eSMartin Matuska 			 */
494acc60b03SMartin Matuska 			count = (int)mbsnbytes(s, length);
4956c95142eSMartin Matuska 		}
4966c95142eSMartin Matuska 		u16.s = (char *)dest->s;
4976c95142eSMartin Matuska 		u16.length = dest->length << 1;;
4986c95142eSMartin Matuska 		u16.buffer_length = dest->buffer_length;
499fd082e96SMartin Matuska 		if (sc->flag & SCONV_NORMALIZATION_C)
5006c95142eSMartin Matuska 			ret = archive_string_normalize_C(&u16, s, count, sc);
501fd082e96SMartin Matuska 		else
502fd082e96SMartin Matuska 			ret = archive_string_normalize_D(&u16, s, count, sc);
5036c95142eSMartin Matuska 		dest->s = (wchar_t *)u16.s;
5046c95142eSMartin Matuska 		dest->length = u16.length >> 1;
5056c95142eSMartin Matuska 		dest->buffer_length = u16.buffer_length;
5066c95142eSMartin Matuska 		sc->flag = saved_flag;/* restore the saved flag. */
5076c95142eSMartin Matuska 		return (ret);
5086c95142eSMartin Matuska 	} else if (sc != NULL && (sc->flag & SCONV_FROM_UTF16)) {
509acc60b03SMartin Matuska 		count = (int)utf16nbytes(s, length);
5106c95142eSMartin Matuska 		count >>= 1; /* to be WCS length */
5116c95142eSMartin Matuska 		/* Allocate memory for WCS. */
5126c95142eSMartin Matuska 		if (NULL == archive_wstring_ensure(dest,
5136c95142eSMartin Matuska 		    dest->length + count + 1))
5146c95142eSMartin Matuska 			return (-1);
515fd082e96SMartin Matuska 		wmemcpy(dest->s + dest->length, (const wchar_t *)s, count);
5166c95142eSMartin Matuska 		if ((sc->flag & SCONV_FROM_UTF16BE) && !is_big_endian()) {
5176c95142eSMartin Matuska 			uint16_t *u16 = (uint16_t *)(dest->s + dest->length);
5186c95142eSMartin Matuska 			int b;
5196c95142eSMartin Matuska 			for (b = 0; b < count; b++) {
5206c95142eSMartin Matuska 				uint16_t val = archive_le16dec(u16+b);
5216c95142eSMartin Matuska 				archive_be16enc(u16+b, val);
5226c95142eSMartin Matuska 			}
5236c95142eSMartin Matuska 		} else if ((sc->flag & SCONV_FROM_UTF16LE) && is_big_endian()) {
5246c95142eSMartin Matuska 			uint16_t *u16 = (uint16_t *)(dest->s + dest->length);
5256c95142eSMartin Matuska 			int b;
5266c95142eSMartin Matuska 			for (b = 0; b < count; b++) {
5276c95142eSMartin Matuska 				uint16_t val = archive_be16dec(u16+b);
5286c95142eSMartin Matuska 				archive_le16enc(u16+b, val);
5296c95142eSMartin Matuska 			}
5306c95142eSMartin Matuska 		}
5316c95142eSMartin Matuska 	} else {
5326c95142eSMartin Matuska 		DWORD mbflag;
533fd082e96SMartin Matuska 		size_t buffsize;
5346c95142eSMartin Matuska 
5356c95142eSMartin Matuska 		if (sc == NULL)
5366c95142eSMartin Matuska 			mbflag = 0;
5376c95142eSMartin Matuska 		else if (sc->flag & SCONV_FROM_CHARSET) {
5386c95142eSMartin Matuska 			/* Do not trust the length which comes from
5396c95142eSMartin Matuska 			 * an archive file. */
5406c95142eSMartin Matuska 			length = mbsnbytes(s, length);
5416c95142eSMartin Matuska 			mbflag = 0;
5426c95142eSMartin Matuska 		} else
5436c95142eSMartin Matuska 			mbflag = MB_PRECOMPOSED;
5446c95142eSMartin Matuska 
545fd082e96SMartin Matuska 		buffsize = dest->length + length + 1;
546fd082e96SMartin Matuska 		do {
5476c95142eSMartin Matuska 			/* Allocate memory for WCS. */
548fd082e96SMartin Matuska 			if (NULL == archive_wstring_ensure(dest, buffsize))
5496c95142eSMartin Matuska 				return (-1);
5506c95142eSMartin Matuska 			/* Convert MBS to WCS. */
5516c95142eSMartin Matuska 			count = MultiByteToWideChar(from_cp,
552acc60b03SMartin Matuska 			    mbflag, s, (int)length, dest->s + dest->length,
553acc60b03SMartin Matuska 			    (int)(dest->buffer_length >> 1) -1);
554fd082e96SMartin Matuska 			if (count == 0 &&
555fd082e96SMartin Matuska 			    GetLastError() == ERROR_INSUFFICIENT_BUFFER) {
556fd082e96SMartin Matuska 				/* Expand the WCS buffer. */
557fd082e96SMartin Matuska 				buffsize = dest->buffer_length << 1;
558fd082e96SMartin Matuska 				continue;
559fd082e96SMartin Matuska 			}
560fd082e96SMartin Matuska 			if (count == 0 && length != 0)
5616c95142eSMartin Matuska 				ret = -1;
562ae5876eaSMartin Matuska 			break;
563ae5876eaSMartin Matuska 		} while (1);
5646c95142eSMartin Matuska 	}
5656c95142eSMartin Matuska 	dest->length += count;
5666c95142eSMartin Matuska 	dest->s[dest->length] = L'\0';
5676c95142eSMartin Matuska 	return (ret);
5686c95142eSMartin Matuska }
5696c95142eSMartin Matuska 
5706c95142eSMartin Matuska #else
5716c95142eSMartin Matuska 
5726c95142eSMartin Matuska /*
5736c95142eSMartin Matuska  * Convert MBS to WCS.
5746c95142eSMartin Matuska  * Note: returns -1 if conversion fails.
5756c95142eSMartin Matuska  */
5766c95142eSMartin Matuska int
5776c95142eSMartin Matuska archive_wstring_append_from_mbs(struct archive_wstring *dest,
5786c95142eSMartin Matuska     const char *p, size_t len)
5796c95142eSMartin Matuska {
5806c95142eSMartin Matuska 	size_t r;
581fd082e96SMartin Matuska 	int ret_val = 0;
5826c95142eSMartin Matuska 	/*
5836c95142eSMartin Matuska 	 * No single byte will be more than one wide character,
5846c95142eSMartin Matuska 	 * so this length estimate will always be big enough.
5856c95142eSMartin Matuska 	 */
5866c95142eSMartin Matuska 	size_t wcs_length = len;
5876c95142eSMartin Matuska 	size_t mbs_length = len;
5886c95142eSMartin Matuska 	const char *mbs = p;
5896c95142eSMartin Matuska 	wchar_t *wcs;
5906c95142eSMartin Matuska #if HAVE_MBRTOWC
5916c95142eSMartin Matuska 	mbstate_t shift_state;
5926c95142eSMartin Matuska 
5936c95142eSMartin Matuska 	memset(&shift_state, 0, sizeof(shift_state));
5946c95142eSMartin Matuska #endif
5956c95142eSMartin Matuska 	if (NULL == archive_wstring_ensure(dest, dest->length + wcs_length + 1))
596fd082e96SMartin Matuska 		return (-1);
5976c95142eSMartin Matuska 	wcs = dest->s + dest->length;
5986c95142eSMartin Matuska 	/*
5996c95142eSMartin Matuska 	 * We cannot use mbsrtowcs/mbstowcs here because those may convert
6006c95142eSMartin Matuska 	 * extra MBS when strlen(p) > len and one wide character consis of
6016c95142eSMartin Matuska 	 * multi bytes.
6026c95142eSMartin Matuska 	 */
603fd082e96SMartin Matuska 	while (*mbs && mbs_length > 0) {
604fd082e96SMartin Matuska 		if (wcs_length == 0) {
605fd082e96SMartin Matuska 			dest->length = wcs - dest->s;
606fd082e96SMartin Matuska 			dest->s[dest->length] = L'\0';
607fd082e96SMartin Matuska 			wcs_length = mbs_length;
608fd082e96SMartin Matuska 			if (NULL == archive_wstring_ensure(dest,
609fd082e96SMartin Matuska 			    dest->length + wcs_length + 1))
610fd082e96SMartin Matuska 				return (-1);
611fd082e96SMartin Matuska 			wcs = dest->s + dest->length;
612fd082e96SMartin Matuska 		}
6136c95142eSMartin Matuska #if HAVE_MBRTOWC
6146c95142eSMartin Matuska 		r = mbrtowc(wcs, mbs, wcs_length, &shift_state);
6156c95142eSMartin Matuska #else
6166c95142eSMartin Matuska 		r = mbtowc(wcs, mbs, wcs_length);
6176c95142eSMartin Matuska #endif
6186c95142eSMartin Matuska 		if (r == (size_t)-1 || r == (size_t)-2) {
619fd082e96SMartin Matuska 			ret_val = -1;
620fd082e96SMartin Matuska 			if (errno == EILSEQ) {
621fd082e96SMartin Matuska 				++mbs;
622fd082e96SMartin Matuska 				--mbs_length;
623fd082e96SMartin Matuska 				continue;
624fd082e96SMartin Matuska 			} else
625fd082e96SMartin Matuska 				break;
6266c95142eSMartin Matuska 		}
6276c95142eSMartin Matuska 		if (r == 0 || r > mbs_length)
6286c95142eSMartin Matuska 			break;
6296c95142eSMartin Matuska 		wcs++;
6306c95142eSMartin Matuska 		wcs_length--;
6316c95142eSMartin Matuska 		mbs += r;
6326c95142eSMartin Matuska 		mbs_length -= r;
6336c95142eSMartin Matuska 	}
6346c95142eSMartin Matuska 	dest->length = wcs - dest->s;
6356c95142eSMartin Matuska 	dest->s[dest->length] = L'\0';
636fd082e96SMartin Matuska 	return (ret_val);
6376c95142eSMartin Matuska }
6386c95142eSMartin Matuska 
6396c95142eSMartin Matuska #endif
6406c95142eSMartin Matuska 
6416c95142eSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
6426c95142eSMartin Matuska 
6436c95142eSMartin Matuska /*
6446c95142eSMartin Matuska  * WCS ==> MBS.
6456c95142eSMartin Matuska  * Note: returns -1 if conversion fails.
646caf54c4fSMartin Matuska  *
647caf54c4fSMartin Matuska  * Win32 builds use WideCharToMultiByte from the Windows API.
648caf54c4fSMartin Matuska  * (Maybe Cygwin should too?  WideCharToMultiByte will know a
649caf54c4fSMartin Matuska  * lot more about local character encodings than the wcrtomb()
650caf54c4fSMartin Matuska  * wrapper is going to know.)
651caf54c4fSMartin Matuska  */
6526c95142eSMartin Matuska int
6536c95142eSMartin Matuska archive_string_append_from_wcs(struct archive_string *as,
6546c95142eSMartin Matuska     const wchar_t *w, size_t len)
655caf54c4fSMartin Matuska {
656fd082e96SMartin Matuska 	return archive_string_append_from_wcs_in_codepage(as, w, len, NULL);
6576c95142eSMartin Matuska }
658caf54c4fSMartin Matuska 
6596c95142eSMartin Matuska static int
6606c95142eSMartin Matuska archive_string_append_from_wcs_in_codepage(struct archive_string *as,
6616c95142eSMartin Matuska     const wchar_t *ws, size_t len, struct archive_string_conv *sc)
6626c95142eSMartin Matuska {
6636c95142eSMartin Matuska 	BOOL defchar_used, *dp;
6646c95142eSMartin Matuska 	int count, ret = 0;
6656c95142eSMartin Matuska 	UINT to_cp;
6666c95142eSMartin Matuska 	int wslen = (int)len;
6676c95142eSMartin Matuska 
6686c95142eSMartin Matuska 	if (sc != NULL)
6696c95142eSMartin Matuska 		to_cp = sc->to_cp;
6706c95142eSMartin Matuska 	else
6716c95142eSMartin Matuska 		to_cp = get_current_codepage();
6726c95142eSMartin Matuska 
6736c95142eSMartin Matuska 	if (to_cp == CP_C_LOCALE) {
6746c95142eSMartin Matuska 		/*
6756c95142eSMartin Matuska 		 * "C" locale special process.
676caf54c4fSMartin Matuska 		 */
6776c95142eSMartin Matuska 		const wchar_t *wp = ws;
6786c95142eSMartin Matuska 		char *p;
6796c95142eSMartin Matuska 
6806c95142eSMartin Matuska 		if (NULL == archive_string_ensure(as,
6816c95142eSMartin Matuska 		    as->length + wslen +1))
6826c95142eSMartin Matuska 			return (-1);
6836c95142eSMartin Matuska 		p = as->s + as->length;
6846c95142eSMartin Matuska 		count = 0;
6856c95142eSMartin Matuska 		defchar_used = 0;
6866c95142eSMartin Matuska 		while (count < wslen && *wp) {
6876c95142eSMartin Matuska 			if (*wp > 255) {
6886c95142eSMartin Matuska 				*p++ = '?';
6896c95142eSMartin Matuska 				wp++;
6906c95142eSMartin Matuska 				defchar_used = 1;
6916c95142eSMartin Matuska 			} else
6926c95142eSMartin Matuska 				*p++ = (char)*wp++;
6936c95142eSMartin Matuska 			count++;
694caf54c4fSMartin Matuska 		}
6956c95142eSMartin Matuska 	} else if (sc != NULL && (sc->flag & SCONV_TO_UTF16)) {
6966c95142eSMartin Matuska 		uint16_t *u16;
6976c95142eSMartin Matuska 
6986c95142eSMartin Matuska 		if (NULL ==
6996c95142eSMartin Matuska 		    archive_string_ensure(as, as->length + len * 2 + 2))
7006c95142eSMartin Matuska 			return (-1);
7016c95142eSMartin Matuska 		u16 = (uint16_t *)(as->s + as->length);
7026c95142eSMartin Matuska 		count = 0;
7036c95142eSMartin Matuska 		defchar_used = 0;
7046c95142eSMartin Matuska 		if (sc->flag & SCONV_TO_UTF16BE) {
7056c95142eSMartin Matuska 			while (count < (int)len && *ws) {
7066c95142eSMartin Matuska 				archive_be16enc(u16+count, *ws);
7076c95142eSMartin Matuska 				ws++;
7086c95142eSMartin Matuska 				count++;
7096c95142eSMartin Matuska 			}
7106c95142eSMartin Matuska 		} else {
7116c95142eSMartin Matuska 			while (count < (int)len && *ws) {
7126c95142eSMartin Matuska 				archive_le16enc(u16+count, *ws);
7136c95142eSMartin Matuska 				ws++;
7146c95142eSMartin Matuska 				count++;
7156c95142eSMartin Matuska 			}
7166c95142eSMartin Matuska 		}
7176c95142eSMartin Matuska 		count <<= 1; /* to be byte size */
7186c95142eSMartin Matuska 	} else {
7196c95142eSMartin Matuska 		/* Make sure the MBS buffer has plenty to set. */
7206c95142eSMartin Matuska 		if (NULL ==
7216c95142eSMartin Matuska 		    archive_string_ensure(as, as->length + len * 2 + 1))
7226c95142eSMartin Matuska 			return (-1);
7236c95142eSMartin Matuska 		do {
7246c95142eSMartin Matuska 			defchar_used = 0;
7256c95142eSMartin Matuska 			if (to_cp == CP_UTF8 || sc == NULL)
7266c95142eSMartin Matuska 				dp = NULL;
7276c95142eSMartin Matuska 			else
7286c95142eSMartin Matuska 				dp = &defchar_used;
7296c95142eSMartin Matuska 			count = WideCharToMultiByte(to_cp, 0, ws, wslen,
730acc60b03SMartin Matuska 			    as->s + as->length, (int)as->buffer_length-1, NULL, dp);
7316c95142eSMartin Matuska 			if (count == 0 &&
7326c95142eSMartin Matuska 			    GetLastError() == ERROR_INSUFFICIENT_BUFFER) {
7336c95142eSMartin Matuska 				/* Expand the MBS buffer and retry. */
7346c95142eSMartin Matuska 				if (NULL == archive_string_ensure(as,
7356c95142eSMartin Matuska 					as->buffer_length + len))
7366c95142eSMartin Matuska 					return (-1);
7376c95142eSMartin Matuska 				continue;
7386c95142eSMartin Matuska 			}
7396c95142eSMartin Matuska 			if (count == 0)
7406c95142eSMartin Matuska 				ret = -1;
741cdf63a70SMartin Matuska 			break;
742cdf63a70SMartin Matuska 		} while (1);
7436c95142eSMartin Matuska 	}
7446c95142eSMartin Matuska 	as->length += count;
7456c95142eSMartin Matuska 	as->s[as->length] = '\0';
7466c95142eSMartin Matuska 	return (defchar_used?-1:ret);
747caf54c4fSMartin Matuska }
748caf54c4fSMartin Matuska 
7496c95142eSMartin Matuska #elif defined(HAVE_WCTOMB) || defined(HAVE_WCRTOMB)
7506c95142eSMartin Matuska 
7516c95142eSMartin Matuska /*
7526c95142eSMartin Matuska  * Translates a wide character string into current locale character set
7536c95142eSMartin Matuska  * and appends to the archive_string.  Note: returns -1 if conversion
7546c95142eSMartin Matuska  * fails.
7556c95142eSMartin Matuska  */
7566c95142eSMartin Matuska int
7576c95142eSMartin Matuska archive_string_append_from_wcs(struct archive_string *as,
7586c95142eSMartin Matuska     const wchar_t *w, size_t len)
7596c95142eSMartin Matuska {
760caf54c4fSMartin Matuska 	/* We cannot use the standard wcstombs() here because it
761caf54c4fSMartin Matuska 	 * cannot tell us how big the output buffer should be.  So
762caf54c4fSMartin Matuska 	 * I've built a loop around wcrtomb() or wctomb() that
763caf54c4fSMartin Matuska 	 * converts a character at a time and resizes the string as
764caf54c4fSMartin Matuska 	 * needed.  We prefer wcrtomb() when it's available because
765caf54c4fSMartin Matuska 	 * it's thread-safe. */
7666c95142eSMartin Matuska 	int n, ret_val = 0;
767caf54c4fSMartin Matuska 	char *p;
7686c95142eSMartin Matuska 	char *end;
769caf54c4fSMartin Matuska #if HAVE_WCRTOMB
770caf54c4fSMartin Matuska 	mbstate_t shift_state;
771caf54c4fSMartin Matuska 
772caf54c4fSMartin Matuska 	memset(&shift_state, 0, sizeof(shift_state));
773caf54c4fSMartin Matuska #else
774caf54c4fSMartin Matuska 	/* Clear the shift state before starting. */
775caf54c4fSMartin Matuska 	wctomb(NULL, L'\0');
776caf54c4fSMartin Matuska #endif
777caf54c4fSMartin Matuska 	/*
7786c95142eSMartin Matuska 	 * Allocate buffer for MBS.
7796c95142eSMartin Matuska 	 * We need this allocation here since it is possible that
7806c95142eSMartin Matuska 	 * as->s is still NULL.
781caf54c4fSMartin Matuska 	 */
7826c95142eSMartin Matuska 	if (archive_string_ensure(as, as->length + len + 1) == NULL)
783fd082e96SMartin Matuska 		return (-1);
7846c95142eSMartin Matuska 
7856c95142eSMartin Matuska 	p = as->s + as->length;
7866c95142eSMartin Matuska 	end = as->s + as->buffer_length - MB_CUR_MAX -1;
7876c95142eSMartin Matuska 	while (*w != L'\0' && len > 0) {
7886c95142eSMartin Matuska 		if (p >= end) {
7896c95142eSMartin Matuska 			as->length = p - as->s;
7906c95142eSMartin Matuska 			as->s[as->length] = '\0';
7916c95142eSMartin Matuska 			/* Re-allocate buffer for MBS. */
7926c95142eSMartin Matuska 			if (archive_string_ensure(as,
7936c95142eSMartin Matuska 			    as->length + len * 2 + 1) == NULL)
794fd082e96SMartin Matuska 				return (-1);
7956c95142eSMartin Matuska 			p = as->s + as->length;
7966c95142eSMartin Matuska 			end = as->s + as->buffer_length - MB_CUR_MAX -1;
797caf54c4fSMartin Matuska 		}
798caf54c4fSMartin Matuska #if HAVE_WCRTOMB
799caf54c4fSMartin Matuska 		n = wcrtomb(p, *w++, &shift_state);
800caf54c4fSMartin Matuska #else
801caf54c4fSMartin Matuska 		n = wctomb(p, *w++);
802caf54c4fSMartin Matuska #endif
8036c95142eSMartin Matuska 		if (n == -1) {
8046c95142eSMartin Matuska 			if (errno == EILSEQ) {
8056c95142eSMartin Matuska 				/* Skip an illegal wide char. */
8066c95142eSMartin Matuska 				*p++ = '?';
8076c95142eSMartin Matuska 				ret_val = -1;
8086c95142eSMartin Matuska 			} else {
8096c95142eSMartin Matuska 				ret_val = -1;
8106c95142eSMartin Matuska 				break;
811caf54c4fSMartin Matuska 			}
8126c95142eSMartin Matuska 		} else
8136c95142eSMartin Matuska 			p += n;
8146c95142eSMartin Matuska 		len--;
8156c95142eSMartin Matuska 	}
8166c95142eSMartin Matuska 	as->length = p - as->s;
8176c95142eSMartin Matuska 	as->s[as->length] = '\0';
8186c95142eSMartin Matuska 	return (ret_val);
8196c95142eSMartin Matuska }
8206c95142eSMartin Matuska 
8216c95142eSMartin Matuska #else /* HAVE_WCTOMB || HAVE_WCRTOMB */
8226c95142eSMartin Matuska 
8236c95142eSMartin Matuska /*
8246c95142eSMartin Matuska  * TODO: Test if __STDC_ISO_10646__ is defined.
8256c95142eSMartin Matuska  * Non-Windows uses ISO C wcrtomb() or wctomb() to perform the conversion
8266c95142eSMartin Matuska  * one character at a time.  If a non-Windows platform doesn't have
8276c95142eSMartin Matuska  * either of these, fall back to the built-in UTF8 conversion.
8286c95142eSMartin Matuska  */
8296c95142eSMartin Matuska int
8306c95142eSMartin Matuska archive_string_append_from_wcs(struct archive_string *as,
8316c95142eSMartin Matuska     const wchar_t *w, size_t len)
8326c95142eSMartin Matuska {
8336c95142eSMartin Matuska 	(void)as;/* UNUSED */
8346c95142eSMartin Matuska 	(void)w;/* UNUSED */
8356c95142eSMartin Matuska 	(void)len;/* UNUSED */
836fd082e96SMartin Matuska 	errno = ENOSYS;
8376c95142eSMartin Matuska 	return (-1);
8386c95142eSMartin Matuska }
8396c95142eSMartin Matuska 
8406c95142eSMartin Matuska #endif /* HAVE_WCTOMB || HAVE_WCRTOMB */
8416c95142eSMartin Matuska 
8426c95142eSMartin Matuska /*
8436c95142eSMartin Matuska  * Find a string conversion object by a pair of 'from' charset name
8446c95142eSMartin Matuska  * and 'to' charset name from an archive object.
8456c95142eSMartin Matuska  * Return NULL if not found.
8466c95142eSMartin Matuska  */
8476c95142eSMartin Matuska static struct archive_string_conv *
8486c95142eSMartin Matuska find_sconv_object(struct archive *a, const char *fc, const char *tc)
8496c95142eSMartin Matuska {
8506c95142eSMartin Matuska 	struct archive_string_conv *sc;
8516c95142eSMartin Matuska 
8526c95142eSMartin Matuska 	if (a == NULL)
8536c95142eSMartin Matuska 		return (NULL);
8546c95142eSMartin Matuska 
8556c95142eSMartin Matuska 	for (sc = a->sconv; sc != NULL; sc = sc->next) {
8566c95142eSMartin Matuska 		if (strcmp(sc->from_charset, fc) == 0 &&
8576c95142eSMartin Matuska 		    strcmp(sc->to_charset, tc) == 0)
8586c95142eSMartin Matuska 			break;
8596c95142eSMartin Matuska 	}
8606c95142eSMartin Matuska 	return (sc);
8616c95142eSMartin Matuska }
8626c95142eSMartin Matuska 
8636c95142eSMartin Matuska /*
8646c95142eSMartin Matuska  * Register a string object to an archive object.
8656c95142eSMartin Matuska  */
8666c95142eSMartin Matuska static void
8676c95142eSMartin Matuska add_sconv_object(struct archive *a, struct archive_string_conv *sc)
8686c95142eSMartin Matuska {
8696c95142eSMartin Matuska 	struct archive_string_conv **psc;
8706c95142eSMartin Matuska 
8716c95142eSMartin Matuska 	/* Add a new sconv to sconv list. */
8726c95142eSMartin Matuska 	psc = &(a->sconv);
8736c95142eSMartin Matuska 	while (*psc != NULL)
8746c95142eSMartin Matuska 		psc = &((*psc)->next);
8756c95142eSMartin Matuska 	*psc = sc;
8766c95142eSMartin Matuska }
8776c95142eSMartin Matuska 
8786c95142eSMartin Matuska static void
8796c95142eSMartin Matuska add_converter(struct archive_string_conv *sc, int (*converter)
8806c95142eSMartin Matuska     (struct archive_string *, const void *, size_t,
8816c95142eSMartin Matuska      struct archive_string_conv *))
8826c95142eSMartin Matuska {
8836c95142eSMartin Matuska 	if (sc == NULL || sc->nconverter >= 2)
8846c95142eSMartin Matuska 		__archive_errx(1, "Programing error");
8856c95142eSMartin Matuska 	sc->converter[sc->nconverter++] = converter;
8866c95142eSMartin Matuska }
8876c95142eSMartin Matuska 
8886c95142eSMartin Matuska static void
8896c95142eSMartin Matuska setup_converter(struct archive_string_conv *sc)
8906c95142eSMartin Matuska {
8916c95142eSMartin Matuska 
8926c95142eSMartin Matuska 	/* Reset. */
8936c95142eSMartin Matuska 	sc->nconverter = 0;
8946c95142eSMartin Matuska 
8956c95142eSMartin Matuska 	/*
8966c95142eSMartin Matuska 	 * Perform special sequence for the incorrect UTF-8 filenames
8976c95142eSMartin Matuska 	 * made by libarchive2.x.
8986c95142eSMartin Matuska 	 */
8996c95142eSMartin Matuska 	if (sc->flag & SCONV_UTF8_LIBARCHIVE_2) {
9006c95142eSMartin Matuska 		add_converter(sc, strncat_from_utf8_libarchive2);
9016c95142eSMartin Matuska 		return;
9026c95142eSMartin Matuska 	}
9036c95142eSMartin Matuska 
9046c95142eSMartin Matuska 	/*
9056c95142eSMartin Matuska 	 * Convert a string to UTF-16BE/LE.
9066c95142eSMartin Matuska 	 */
9076c95142eSMartin Matuska 	if (sc->flag & SCONV_TO_UTF16) {
9086c95142eSMartin Matuska 		/*
9096c95142eSMartin Matuska 		 * If the current locale is UTF-8, we can translate
9106c95142eSMartin Matuska 		 * a UTF-8 string into a UTF-16BE string.
9116c95142eSMartin Matuska 		 */
9126c95142eSMartin Matuska 		if (sc->flag & SCONV_FROM_UTF8) {
9136c95142eSMartin Matuska 			add_converter(sc, archive_string_append_unicode);
9146c95142eSMartin Matuska 			return;
9156c95142eSMartin Matuska 		}
9166c95142eSMartin Matuska 
9176c95142eSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
9186c95142eSMartin Matuska 		if (sc->flag & SCONV_WIN_CP) {
9196c95142eSMartin Matuska 			if (sc->flag & SCONV_TO_UTF16BE)
9206c95142eSMartin Matuska 				add_converter(sc, win_strncat_to_utf16be);
9216c95142eSMartin Matuska 			else
9226c95142eSMartin Matuska 				add_converter(sc, win_strncat_to_utf16le);
9236c95142eSMartin Matuska 			return;
9246c95142eSMartin Matuska 		}
9256c95142eSMartin Matuska #endif
9266c95142eSMartin Matuska 
9276c95142eSMartin Matuska #if defined(HAVE_ICONV)
9286c95142eSMartin Matuska 		if (sc->cd != (iconv_t)-1) {
9296c95142eSMartin Matuska 			add_converter(sc, iconv_strncat_in_locale);
9306c95142eSMartin Matuska 			return;
9316c95142eSMartin Matuska 		}
9326c95142eSMartin Matuska #endif
9336c95142eSMartin Matuska 
9346c95142eSMartin Matuska 		if (sc->flag & SCONV_BEST_EFFORT) {
9356c95142eSMartin Matuska 			if (sc->flag & SCONV_TO_UTF16BE)
936fd082e96SMartin Matuska 				add_converter(sc,
937fd082e96SMartin Matuska 					best_effort_strncat_to_utf16be);
9386c95142eSMartin Matuska 			else
939fd082e96SMartin Matuska 				add_converter(sc,
940fd082e96SMartin Matuska 					best_effort_strncat_to_utf16le);
9416c95142eSMartin Matuska 		} else
9426c95142eSMartin Matuska 			/* Make sure we have no converter. */
9436c95142eSMartin Matuska 			sc->nconverter = 0;
9446c95142eSMartin Matuska 		return;
9456c95142eSMartin Matuska 	}
9466c95142eSMartin Matuska 
9476c95142eSMartin Matuska 	/*
9486c95142eSMartin Matuska 	 * Convert a string from UTF-16BE/LE.
9496c95142eSMartin Matuska 	 */
9506c95142eSMartin Matuska 	if (sc->flag & SCONV_FROM_UTF16) {
9516c95142eSMartin Matuska 		/*
9526c95142eSMartin Matuska 		 * At least we should normalize a UTF-16BE string.
9536c95142eSMartin Matuska 		 */
9546c95142eSMartin Matuska 		if (sc->flag & SCONV_NORMALIZATION_D)
9556c95142eSMartin Matuska 			add_converter(sc,archive_string_normalize_D);
956fd082e96SMartin Matuska 		else if (sc->flag & SCONV_NORMALIZATION_C)
9576c95142eSMartin Matuska 			add_converter(sc, archive_string_normalize_C);
9586c95142eSMartin Matuska 
9596c95142eSMartin Matuska 		if (sc->flag & SCONV_TO_UTF8) {
9606c95142eSMartin Matuska 			/*
9616c95142eSMartin Matuska 			 * If the current locale is UTF-8, we can translate
9626c95142eSMartin Matuska 			 * a UTF-16BE/LE string into a UTF-8 string directly.
9636c95142eSMartin Matuska 			 */
9646c95142eSMartin Matuska 			if (!(sc->flag &
9656c95142eSMartin Matuska 			    (SCONV_NORMALIZATION_D |SCONV_NORMALIZATION_C)))
9666c95142eSMartin Matuska 				add_converter(sc,
9676c95142eSMartin Matuska 				    archive_string_append_unicode);
9686c95142eSMartin Matuska 			return;
9696c95142eSMartin Matuska 		}
9706c95142eSMartin Matuska 
9716c95142eSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
9726c95142eSMartin Matuska 		if (sc->flag & SCONV_WIN_CP) {
9736c95142eSMartin Matuska 			if (sc->flag & SCONV_FROM_UTF16BE)
9746c95142eSMartin Matuska 				add_converter(sc, win_strncat_from_utf16be);
9756c95142eSMartin Matuska 			else
9766c95142eSMartin Matuska 				add_converter(sc, win_strncat_from_utf16le);
9776c95142eSMartin Matuska 			return;
9786c95142eSMartin Matuska 		}
9796c95142eSMartin Matuska #endif
9806c95142eSMartin Matuska 
9816c95142eSMartin Matuska #if defined(HAVE_ICONV)
9826c95142eSMartin Matuska 		if (sc->cd != (iconv_t)-1) {
9836c95142eSMartin Matuska 			add_converter(sc, iconv_strncat_in_locale);
9846c95142eSMartin Matuska 			return;
9856c95142eSMartin Matuska 		}
9866c95142eSMartin Matuska #endif
9876c95142eSMartin Matuska 
9886c95142eSMartin Matuska 		if ((sc->flag & (SCONV_BEST_EFFORT | SCONV_FROM_UTF16BE))
9896c95142eSMartin Matuska 		    == (SCONV_BEST_EFFORT | SCONV_FROM_UTF16BE))
9906c95142eSMartin Matuska 			add_converter(sc, best_effort_strncat_from_utf16be);
9916c95142eSMartin Matuska 		else if ((sc->flag & (SCONV_BEST_EFFORT | SCONV_FROM_UTF16LE))
9926c95142eSMartin Matuska 		    == (SCONV_BEST_EFFORT | SCONV_FROM_UTF16LE))
9936c95142eSMartin Matuska 			add_converter(sc, best_effort_strncat_from_utf16le);
9946c95142eSMartin Matuska 		else
9956c95142eSMartin Matuska 			/* Make sure we have no converter. */
9966c95142eSMartin Matuska 			sc->nconverter = 0;
9976c95142eSMartin Matuska 		return;
9986c95142eSMartin Matuska 	}
9996c95142eSMartin Matuska 
10006c95142eSMartin Matuska 	if (sc->flag & SCONV_FROM_UTF8) {
10016c95142eSMartin Matuska 		/*
10026c95142eSMartin Matuska 		 * At least we should normalize a UTF-8 string.
10036c95142eSMartin Matuska 		 */
10046c95142eSMartin Matuska 		if (sc->flag & SCONV_NORMALIZATION_D)
10056c95142eSMartin Matuska 			add_converter(sc,archive_string_normalize_D);
1006fd082e96SMartin Matuska 		else if (sc->flag & SCONV_NORMALIZATION_C)
10076c95142eSMartin Matuska 			add_converter(sc, archive_string_normalize_C);
10086c95142eSMartin Matuska 
10096c95142eSMartin Matuska 		/*
10106c95142eSMartin Matuska 		 * Copy UTF-8 string with a check of CESU-8.
10116c95142eSMartin Matuska 		 * Apparently, iconv does not check surrogate pairs in UTF-8
10126c95142eSMartin Matuska 		 * when both from-charset and to-charset are UTF-8, and then
10136c95142eSMartin Matuska 		 * we use our UTF-8 copy code.
10146c95142eSMartin Matuska 		 */
10156c95142eSMartin Matuska 		if (sc->flag & SCONV_TO_UTF8) {
10166c95142eSMartin Matuska 			/*
10176c95142eSMartin Matuska 			 * If the current locale is UTF-8, we can translate
10186c95142eSMartin Matuska 			 * a UTF-16BE string into a UTF-8 string directly.
10196c95142eSMartin Matuska 			 */
10206c95142eSMartin Matuska 			if (!(sc->flag &
10216c95142eSMartin Matuska 			    (SCONV_NORMALIZATION_D |SCONV_NORMALIZATION_C)))
10226c95142eSMartin Matuska 				add_converter(sc, strncat_from_utf8_to_utf8);
10236c95142eSMartin Matuska 			return;
10246c95142eSMartin Matuska 		}
10256c95142eSMartin Matuska 	}
10266c95142eSMartin Matuska 
10276c95142eSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
10286c95142eSMartin Matuska 	/*
10296c95142eSMartin Matuska 	 * On Windows we can use Windows API for a string conversion.
10306c95142eSMartin Matuska 	 */
10316c95142eSMartin Matuska 	if (sc->flag & SCONV_WIN_CP) {
10326c95142eSMartin Matuska 		add_converter(sc, strncat_in_codepage);
10336c95142eSMartin Matuska 		return;
10346c95142eSMartin Matuska 	}
10356c95142eSMartin Matuska #endif
10366c95142eSMartin Matuska 
10376c95142eSMartin Matuska #if HAVE_ICONV
10386c95142eSMartin Matuska 	if (sc->cd != (iconv_t)-1) {
10396c95142eSMartin Matuska 		add_converter(sc, iconv_strncat_in_locale);
1040fd082e96SMartin Matuska 		/*
1041fd082e96SMartin Matuska 		 * iconv generally does not support UTF-8-MAC and so
1042fd082e96SMartin Matuska 		 * we have to the output of iconv from NFC to NFD if
1043fd082e96SMartin Matuska 		 * need.
1044fd082e96SMartin Matuska 		 */
1045fd082e96SMartin Matuska 		if ((sc->flag & SCONV_FROM_CHARSET) &&
1046fd082e96SMartin Matuska 		    (sc->flag & SCONV_TO_UTF8)) {
1047fd082e96SMartin Matuska 			if (sc->flag & SCONV_NORMALIZATION_D)
1048fd082e96SMartin Matuska 				add_converter(sc, archive_string_normalize_D);
1049fd082e96SMartin Matuska 		}
10506c95142eSMartin Matuska 		return;
10516c95142eSMartin Matuska 	}
10526c95142eSMartin Matuska #endif
10536c95142eSMartin Matuska 
10546c95142eSMartin Matuska 	/*
10556c95142eSMartin Matuska 	 * Try conversion in the best effort or no conversion.
10566c95142eSMartin Matuska 	 */
10576c95142eSMartin Matuska 	if ((sc->flag & SCONV_BEST_EFFORT) || sc->same)
10586c95142eSMartin Matuska 		add_converter(sc, best_effort_strncat_in_locale);
10596c95142eSMartin Matuska 	else
10606c95142eSMartin Matuska 		/* Make sure we have no converter. */
10616c95142eSMartin Matuska 		sc->nconverter = 0;
10626c95142eSMartin Matuska }
10636c95142eSMartin Matuska 
10646c95142eSMartin Matuska /*
10656c95142eSMartin Matuska  * Return canonicalized charset-name but this supports just UTF-8, UTF-16BE
10666c95142eSMartin Matuska  * and CP932 which are referenced in create_sconv_object().
10676c95142eSMartin Matuska  */
10686c95142eSMartin Matuska static const char *
10696c95142eSMartin Matuska canonical_charset_name(const char *charset)
10706c95142eSMartin Matuska {
10716c95142eSMartin Matuska 	char cs[16];
10726c95142eSMartin Matuska 	char *p;
10736c95142eSMartin Matuska 	const char *s;
10746c95142eSMartin Matuska 
10756c95142eSMartin Matuska 	if (charset == NULL || charset[0] == '\0'
10766c95142eSMartin Matuska 	    || strlen(charset) > 15)
10776c95142eSMartin Matuska 		return (charset);
10786c95142eSMartin Matuska 
10796c95142eSMartin Matuska 	/* Copy name to uppercase. */
10806c95142eSMartin Matuska 	p = cs;
10816c95142eSMartin Matuska 	s = charset;
10826c95142eSMartin Matuska 	while (*s) {
10836c95142eSMartin Matuska 		char c = *s++;
10846c95142eSMartin Matuska 		if (c >= 'a' && c <= 'z')
10856c95142eSMartin Matuska 			c -= 'a' - 'A';
10866c95142eSMartin Matuska 		*p++ = c;
10876c95142eSMartin Matuska 	}
10886c95142eSMartin Matuska 	*p++ = '\0';
10896c95142eSMartin Matuska 
10906c95142eSMartin Matuska 	if (strcmp(cs, "UTF-8") == 0 ||
10916c95142eSMartin Matuska 	    strcmp(cs, "UTF8") == 0)
10926c95142eSMartin Matuska 		return ("UTF-8");
10936c95142eSMartin Matuska 	if (strcmp(cs, "UTF-16BE") == 0 ||
10946c95142eSMartin Matuska 	    strcmp(cs, "UTF16BE") == 0)
10956c95142eSMartin Matuska 		return ("UTF-16BE");
10966c95142eSMartin Matuska 	if (strcmp(cs, "UTF-16LE") == 0 ||
10976c95142eSMartin Matuska 	    strcmp(cs, "UTF16LE") == 0)
10986c95142eSMartin Matuska 		return ("UTF-16LE");
10996c95142eSMartin Matuska 	if (strcmp(cs, "CP932") == 0)
11006c95142eSMartin Matuska 		return ("CP932");
11016c95142eSMartin Matuska 	return (charset);
11026c95142eSMartin Matuska }
11036c95142eSMartin Matuska 
11046c95142eSMartin Matuska /*
11056c95142eSMartin Matuska  * Create a string conversion object.
11066c95142eSMartin Matuska  */
11076c95142eSMartin Matuska static struct archive_string_conv *
11086c95142eSMartin Matuska create_sconv_object(const char *fc, const char *tc,
11096c95142eSMartin Matuska     unsigned current_codepage, int flag)
11106c95142eSMartin Matuska {
11116c95142eSMartin Matuska 	struct archive_string_conv *sc;
11126c95142eSMartin Matuska 
11136c95142eSMartin Matuska 	sc = calloc(1, sizeof(*sc));
11146c95142eSMartin Matuska 	if (sc == NULL)
11156c95142eSMartin Matuska 		return (NULL);
11166c95142eSMartin Matuska 	sc->next = NULL;
11176c95142eSMartin Matuska 	sc->from_charset = strdup(fc);
11186c95142eSMartin Matuska 	if (sc->from_charset == NULL) {
11196c95142eSMartin Matuska 		free(sc);
11206c95142eSMartin Matuska 		return (NULL);
11216c95142eSMartin Matuska 	}
11226c95142eSMartin Matuska 	sc->to_charset = strdup(tc);
11236c95142eSMartin Matuska 	if (sc->to_charset == NULL) {
11246c95142eSMartin Matuska 		free(sc->from_charset);
1125acc60b03SMartin Matuska 		free(sc);
11266c95142eSMartin Matuska 		return (NULL);
11276c95142eSMartin Matuska 	}
11286c95142eSMartin Matuska 	archive_string_init(&sc->utftmp);
11296c95142eSMartin Matuska 
11306c95142eSMartin Matuska 	if (flag & SCONV_TO_CHARSET) {
11316c95142eSMartin Matuska 		/*
11326c95142eSMartin Matuska 		 * Convert characters from the current locale charset to
11336c95142eSMartin Matuska 		 * a specified charset.
11346c95142eSMartin Matuska 		 */
11356c95142eSMartin Matuska 		sc->from_cp = current_codepage;
11366c95142eSMartin Matuska 		sc->to_cp = make_codepage_from_charset(tc);
11376c95142eSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
11386c95142eSMartin Matuska 		if (IsValidCodePage(sc->to_cp))
11396c95142eSMartin Matuska 			flag |= SCONV_WIN_CP;
11406c95142eSMartin Matuska #endif
11416c95142eSMartin Matuska 	} else if (flag & SCONV_FROM_CHARSET) {
11426c95142eSMartin Matuska 		/*
11436c95142eSMartin Matuska 		 * Convert characters from a specified charset to
11446c95142eSMartin Matuska 		 * the current locale charset.
11456c95142eSMartin Matuska 		 */
11466c95142eSMartin Matuska 		sc->to_cp = current_codepage;
11476c95142eSMartin Matuska 		sc->from_cp = make_codepage_from_charset(fc);
11486c95142eSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
11496c95142eSMartin Matuska 		if (IsValidCodePage(sc->from_cp))
11506c95142eSMartin Matuska 			flag |= SCONV_WIN_CP;
1151caf54c4fSMartin Matuska #endif
1152caf54c4fSMartin Matuska 	}
1153caf54c4fSMartin Matuska 
11546c95142eSMartin Matuska 	/*
11556c95142eSMartin Matuska 	 * Check if "from charset" and "to charset" are the same.
11566c95142eSMartin Matuska 	 */
11576c95142eSMartin Matuska 	if (strcmp(fc, tc) == 0 ||
11586c95142eSMartin Matuska 	    (sc->from_cp != (unsigned)-1 && sc->from_cp == sc->to_cp))
11596c95142eSMartin Matuska 		sc->same = 1;
11606c95142eSMartin Matuska 	else
11616c95142eSMartin Matuska 		sc->same = 0;
11626c95142eSMartin Matuska 
11636c95142eSMartin Matuska 	/*
11646c95142eSMartin Matuska 	 * Mark if "from charset" or "to charset" are UTF-8 or UTF-16BE/LE.
11656c95142eSMartin Matuska 	 */
11666c95142eSMartin Matuska 	if (strcmp(tc, "UTF-8") == 0)
11676c95142eSMartin Matuska 		flag |= SCONV_TO_UTF8;
11686c95142eSMartin Matuska 	else if (strcmp(tc, "UTF-16BE") == 0)
11696c95142eSMartin Matuska 		flag |= SCONV_TO_UTF16BE;
11706c95142eSMartin Matuska 	else if (strcmp(tc, "UTF-16LE") == 0)
11716c95142eSMartin Matuska 		flag |= SCONV_TO_UTF16LE;
11726c95142eSMartin Matuska 	if (strcmp(fc, "UTF-8") == 0)
11736c95142eSMartin Matuska 		flag |= SCONV_FROM_UTF8;
11746c95142eSMartin Matuska 	else if (strcmp(fc, "UTF-16BE") == 0)
11756c95142eSMartin Matuska 		flag |= SCONV_FROM_UTF16BE;
11766c95142eSMartin Matuska 	else if (strcmp(fc, "UTF-16LE") == 0)
11776c95142eSMartin Matuska 		flag |= SCONV_FROM_UTF16LE;
11786c95142eSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
11796c95142eSMartin Matuska 	if (sc->to_cp == CP_UTF8)
11806c95142eSMartin Matuska 		flag |= SCONV_TO_UTF8;
11816c95142eSMartin Matuska 	else if (sc->to_cp == CP_UTF16BE)
11826c95142eSMartin Matuska 		flag |= SCONV_TO_UTF16BE | SCONV_WIN_CP;
11836c95142eSMartin Matuska 	else if (sc->to_cp == CP_UTF16LE)
11846c95142eSMartin Matuska 		flag |= SCONV_TO_UTF16LE | SCONV_WIN_CP;
11856c95142eSMartin Matuska 	if (sc->from_cp == CP_UTF8)
11866c95142eSMartin Matuska 		flag |= SCONV_FROM_UTF8;
11876c95142eSMartin Matuska 	else if (sc->from_cp == CP_UTF16BE)
11886c95142eSMartin Matuska 		flag |= SCONV_FROM_UTF16BE | SCONV_WIN_CP;
11896c95142eSMartin Matuska 	else if (sc->from_cp == CP_UTF16LE)
11906c95142eSMartin Matuska 		flag |= SCONV_FROM_UTF16LE | SCONV_WIN_CP;
11916c95142eSMartin Matuska #endif
11926c95142eSMartin Matuska 
11936c95142eSMartin Matuska 	/*
11946c95142eSMartin Matuska 	 * Set a flag for Unicode NFD. Usually iconv cannot correctly
11956c95142eSMartin Matuska 	 * handle it. So we have to translate NFD characters to NFC ones
11966c95142eSMartin Matuska 	 * ourselves before iconv handles. Another reason is to prevent
11976c95142eSMartin Matuska 	 * that the same sight of two filenames, one is NFC and other
11986c95142eSMartin Matuska 	 * is NFD, would be in its directory.
11996c95142eSMartin Matuska 	 * On Mac OS X, although its filesystem layer automatically
12006c95142eSMartin Matuska 	 * convert filenames to NFD, it would be useful for filename
12016c95142eSMartin Matuska 	 * comparing to find out the same filenames that we normalize
12026c95142eSMartin Matuska 	 * that to be NFD ourselves.
12036c95142eSMartin Matuska 	 */
12046c95142eSMartin Matuska 	if ((flag & SCONV_FROM_CHARSET) &&
12056c95142eSMartin Matuska 	    (flag & (SCONV_FROM_UTF16 | SCONV_FROM_UTF8))) {
12066c95142eSMartin Matuska #if defined(__APPLE__)
1207fd082e96SMartin Matuska 		if (flag & SCONV_TO_UTF8)
12086c95142eSMartin Matuska 			flag |= SCONV_NORMALIZATION_D;
1209fd082e96SMartin Matuska 		else
12106c95142eSMartin Matuska #endif
12116c95142eSMartin Matuska 			flag |= SCONV_NORMALIZATION_C;
12126c95142eSMartin Matuska 	}
1213fd082e96SMartin Matuska #if defined(__APPLE__)
1214fd082e96SMartin Matuska 	/*
1215fd082e96SMartin Matuska 	 * In case writing an archive file, make sure that a filename
1216fd082e96SMartin Matuska 	 * going to be passed to iconv is a Unicode NFC string since
1217fd082e96SMartin Matuska 	 * a filename in HFS Plus filesystem is a Unicode NFD one and
1218fd082e96SMartin Matuska 	 * iconv cannot handle it with "UTF-8" charset. It is simpler
1219fd082e96SMartin Matuska 	 * than a use of "UTF-8-MAC" charset.
1220fd082e96SMartin Matuska 	 */
1221fd082e96SMartin Matuska 	if ((flag & SCONV_TO_CHARSET) &&
1222fd082e96SMartin Matuska 	    (flag & (SCONV_FROM_UTF16 | SCONV_FROM_UTF8)) &&
1223fd082e96SMartin Matuska 	    !(flag & (SCONV_TO_UTF16 | SCONV_TO_UTF8)))
1224fd082e96SMartin Matuska 		flag |= SCONV_NORMALIZATION_C;
1225fd082e96SMartin Matuska 	/*
1226fd082e96SMartin Matuska 	 * In case reading an archive file. make sure that a filename
1227fd082e96SMartin Matuska 	 * will be passed to users is a Unicode NFD string in order to
1228fd082e96SMartin Matuska 	 * correctly compare the filename with other one which comes
1229fd082e96SMartin Matuska 	 * from HFS Plus filesystem.
1230fd082e96SMartin Matuska 	 */
1231fd082e96SMartin Matuska 	if ((flag & SCONV_FROM_CHARSET) &&
1232fd082e96SMartin Matuska 	   !(flag & (SCONV_FROM_UTF16 | SCONV_FROM_UTF8)) &&
1233fd082e96SMartin Matuska 	    (flag & SCONV_TO_UTF8))
1234fd082e96SMartin Matuska 		flag |= SCONV_NORMALIZATION_D;
1235fd082e96SMartin Matuska #endif
12366c95142eSMartin Matuska 
12376c95142eSMartin Matuska #if defined(HAVE_ICONV)
12386c95142eSMartin Matuska 	sc->cd_w = (iconv_t)-1;
12396c95142eSMartin Matuska 	/*
12406c95142eSMartin Matuska 	 * Create an iconv object.
12416c95142eSMartin Matuska 	 */
12426c95142eSMartin Matuska 	if (((flag & (SCONV_TO_UTF8 | SCONV_TO_UTF16)) &&
12436c95142eSMartin Matuska 	    (flag & (SCONV_FROM_UTF8 | SCONV_FROM_UTF16))) ||
12446c95142eSMartin Matuska 	    (flag & SCONV_WIN_CP)) {
12456c95142eSMartin Matuska 		/* This case we won't use iconv. */
12466c95142eSMartin Matuska 		sc->cd = (iconv_t)-1;
12476c95142eSMartin Matuska 	} else {
12486c95142eSMartin Matuska 		sc->cd = iconv_open(tc, fc);
12496c95142eSMartin Matuska 		if (sc->cd == (iconv_t)-1 && (sc->flag & SCONV_BEST_EFFORT)) {
12506c95142eSMartin Matuska 			/*
12516c95142eSMartin Matuska 			 * Unfortunaly, all of iconv implements do support
12526c95142eSMartin Matuska 			 * "CP932" character-set, so we should use "SJIS"
12536c95142eSMartin Matuska 			 * instead if iconv_open failed.
12546c95142eSMartin Matuska 			 */
12556c95142eSMartin Matuska 			if (strcmp(tc, "CP932") == 0)
12566c95142eSMartin Matuska 				sc->cd = iconv_open("SJIS", fc);
12576c95142eSMartin Matuska 			else if (strcmp(fc, "CP932") == 0)
12586c95142eSMartin Matuska 				sc->cd = iconv_open(tc, "SJIS");
12596c95142eSMartin Matuska 		}
12606c95142eSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
12616c95142eSMartin Matuska 		/*
12626c95142eSMartin Matuska 		 * archive_mstring on Windows directly convert multi-bytes
12636c95142eSMartin Matuska 		 * into archive_wstring in order not to depend on locale
12646c95142eSMartin Matuska 		 * so that you can do a I18N programing. This will be
12656c95142eSMartin Matuska 		 * used only in archive_mstring_copy_mbs_len_l so far.
12666c95142eSMartin Matuska 		 */
12676c95142eSMartin Matuska 		if (flag & SCONV_FROM_CHARSET) {
12686c95142eSMartin Matuska 			sc->cd_w = iconv_open("UTF-8", fc);
12696c95142eSMartin Matuska 			if (sc->cd_w == (iconv_t)-1 &&
12706c95142eSMartin Matuska 			    (sc->flag & SCONV_BEST_EFFORT)) {
12716c95142eSMartin Matuska 				if (strcmp(fc, "CP932") == 0)
12726c95142eSMartin Matuska 					sc->cd_w = iconv_open("UTF-8", "SJIS");
12736c95142eSMartin Matuska 			}
12746c95142eSMartin Matuska 		}
1275caf54c4fSMartin Matuska #endif /* _WIN32 && !__CYGWIN__ */
12766c95142eSMartin Matuska 	}
12776c95142eSMartin Matuska #endif	/* HAVE_ICONV */
12786c95142eSMartin Matuska 
12796c95142eSMartin Matuska 	sc->flag = flag;
12806c95142eSMartin Matuska 
12816c95142eSMartin Matuska 	/*
12826c95142eSMartin Matuska 	 * Set up converters.
12836c95142eSMartin Matuska 	 */
12846c95142eSMartin Matuska 	setup_converter(sc);
12856c95142eSMartin Matuska 
12866c95142eSMartin Matuska 	return (sc);
12876c95142eSMartin Matuska }
12886c95142eSMartin Matuska 
12896c95142eSMartin Matuska /*
12906c95142eSMartin Matuska  * Free a string conversion object.
12916c95142eSMartin Matuska  */
12926c95142eSMartin Matuska static void
12936c95142eSMartin Matuska free_sconv_object(struct archive_string_conv *sc)
12946c95142eSMartin Matuska {
12956c95142eSMartin Matuska 	free(sc->from_charset);
12966c95142eSMartin Matuska 	free(sc->to_charset);
12976c95142eSMartin Matuska 	archive_string_free(&sc->utftmp);
12986c95142eSMartin Matuska #if HAVE_ICONV
12996c95142eSMartin Matuska 	if (sc->cd != (iconv_t)-1)
13006c95142eSMartin Matuska 		iconv_close(sc->cd);
13016c95142eSMartin Matuska 	if (sc->cd_w != (iconv_t)-1)
13026c95142eSMartin Matuska 		iconv_close(sc->cd_w);
13036c95142eSMartin Matuska #endif
13046c95142eSMartin Matuska 	free(sc);
13056c95142eSMartin Matuska }
13066c95142eSMartin Matuska 
13076c95142eSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
13086c95142eSMartin Matuska static unsigned
13096c95142eSMartin Matuska my_atoi(const char *p)
13106c95142eSMartin Matuska {
13116c95142eSMartin Matuska 	unsigned cp;
13126c95142eSMartin Matuska 
13136c95142eSMartin Matuska 	cp = 0;
13146c95142eSMartin Matuska 	while (*p) {
13156c95142eSMartin Matuska 		if (*p >= '0' && *p <= '9')
13166c95142eSMartin Matuska 			cp = cp * 10 + (*p - '0');
13176c95142eSMartin Matuska 		else
13186c95142eSMartin Matuska 			return (-1);
13196c95142eSMartin Matuska 		p++;
13206c95142eSMartin Matuska 	}
13216c95142eSMartin Matuska 	return (cp);
13226c95142eSMartin Matuska }
13236c95142eSMartin Matuska 
13246c95142eSMartin Matuska /*
13256c95142eSMartin Matuska  * Translate Charset name (as used by iconv) into CodePage (as used by Windows)
13266c95142eSMartin Matuska  * Return -1 if failed.
13276c95142eSMartin Matuska  *
13286c95142eSMartin Matuska  * Note: This translation code may be insufficient.
13296c95142eSMartin Matuska  */
13306c95142eSMartin Matuska static struct charset {
13316c95142eSMartin Matuska 	const char *name;
13326c95142eSMartin Matuska 	unsigned cp;
13336c95142eSMartin Matuska } charsets[] = {
13346c95142eSMartin Matuska 	/* MUST BE SORTED! */
13356c95142eSMartin Matuska 	{"ASCII", 1252},
13366c95142eSMartin Matuska 	{"ASMO-708", 708},
13376c95142eSMartin Matuska 	{"BIG5", 950},
13386c95142eSMartin Matuska 	{"CHINESE", 936},
13396c95142eSMartin Matuska 	{"CP367", 1252},
13406c95142eSMartin Matuska 	{"CP819", 1252},
13416c95142eSMartin Matuska 	{"CP1025", 21025},
13426c95142eSMartin Matuska 	{"DOS-720", 720},
13436c95142eSMartin Matuska 	{"DOS-862", 862},
13446c95142eSMartin Matuska 	{"EUC-CN", 51936},
13456c95142eSMartin Matuska 	{"EUC-JP", 51932},
13466c95142eSMartin Matuska 	{"EUC-KR", 949},
13476c95142eSMartin Matuska 	{"EUCCN", 51936},
13486c95142eSMartin Matuska 	{"EUCJP", 51932},
13496c95142eSMartin Matuska 	{"EUCKR", 949},
13506c95142eSMartin Matuska 	{"GB18030", 54936},
13516c95142eSMartin Matuska 	{"GB2312", 936},
13526c95142eSMartin Matuska 	{"HEBREW", 1255},
13536c95142eSMartin Matuska 	{"HZ-GB-2312", 52936},
13546c95142eSMartin Matuska 	{"IBM273", 20273},
13556c95142eSMartin Matuska 	{"IBM277", 20277},
13566c95142eSMartin Matuska 	{"IBM278", 20278},
13576c95142eSMartin Matuska 	{"IBM280", 20280},
13586c95142eSMartin Matuska 	{"IBM284", 20284},
13596c95142eSMartin Matuska 	{"IBM285", 20285},
13606c95142eSMartin Matuska 	{"IBM290", 20290},
13616c95142eSMartin Matuska 	{"IBM297", 20297},
13626c95142eSMartin Matuska 	{"IBM367", 1252},
13636c95142eSMartin Matuska 	{"IBM420", 20420},
13646c95142eSMartin Matuska 	{"IBM423", 20423},
13656c95142eSMartin Matuska 	{"IBM424", 20424},
13666c95142eSMartin Matuska 	{"IBM819", 1252},
13676c95142eSMartin Matuska 	{"IBM871", 20871},
13686c95142eSMartin Matuska 	{"IBM880", 20880},
13696c95142eSMartin Matuska 	{"IBM905", 20905},
13706c95142eSMartin Matuska 	{"IBM924", 20924},
13716c95142eSMartin Matuska 	{"ISO-8859-1", 28591},
13726c95142eSMartin Matuska 	{"ISO-8859-13", 28603},
13736c95142eSMartin Matuska 	{"ISO-8859-15", 28605},
13746c95142eSMartin Matuska 	{"ISO-8859-2", 28592},
13756c95142eSMartin Matuska 	{"ISO-8859-3", 28593},
13766c95142eSMartin Matuska 	{"ISO-8859-4", 28594},
13776c95142eSMartin Matuska 	{"ISO-8859-5", 28595},
13786c95142eSMartin Matuska 	{"ISO-8859-6", 28596},
13796c95142eSMartin Matuska 	{"ISO-8859-7", 28597},
13806c95142eSMartin Matuska 	{"ISO-8859-8", 28598},
13816c95142eSMartin Matuska 	{"ISO-8859-9", 28599},
13826c95142eSMartin Matuska 	{"ISO8859-1", 28591},
13836c95142eSMartin Matuska 	{"ISO8859-13", 28603},
13846c95142eSMartin Matuska 	{"ISO8859-15", 28605},
13856c95142eSMartin Matuska 	{"ISO8859-2", 28592},
13866c95142eSMartin Matuska 	{"ISO8859-3", 28593},
13876c95142eSMartin Matuska 	{"ISO8859-4", 28594},
13886c95142eSMartin Matuska 	{"ISO8859-5", 28595},
13896c95142eSMartin Matuska 	{"ISO8859-6", 28596},
13906c95142eSMartin Matuska 	{"ISO8859-7", 28597},
13916c95142eSMartin Matuska 	{"ISO8859-8", 28598},
13926c95142eSMartin Matuska 	{"ISO8859-9", 28599},
13936c95142eSMartin Matuska 	{"JOHAB", 1361},
13946c95142eSMartin Matuska 	{"KOI8-R", 20866},
13956c95142eSMartin Matuska 	{"KOI8-U", 21866},
13966c95142eSMartin Matuska 	{"KS_C_5601-1987", 949},
13976c95142eSMartin Matuska 	{"LATIN1", 1252},
13986c95142eSMartin Matuska 	{"LATIN2", 28592},
13996c95142eSMartin Matuska 	{"MACINTOSH", 10000},
14006c95142eSMartin Matuska 	{"SHIFT-JIS", 932},
14016c95142eSMartin Matuska 	{"SHIFT_JIS", 932},
14026c95142eSMartin Matuska 	{"SJIS", 932},
14036c95142eSMartin Matuska 	{"US", 1252},
14046c95142eSMartin Matuska 	{"US-ASCII", 1252},
14056c95142eSMartin Matuska 	{"UTF-16", 1200},
14066c95142eSMartin Matuska 	{"UTF-16BE", 1201},
14076c95142eSMartin Matuska 	{"UTF-16LE", 1200},
14086c95142eSMartin Matuska 	{"UTF-8", CP_UTF8},
14096c95142eSMartin Matuska 	{"X-EUROPA", 29001},
14106c95142eSMartin Matuska 	{"X-MAC-ARABIC", 10004},
14116c95142eSMartin Matuska 	{"X-MAC-CE", 10029},
14126c95142eSMartin Matuska 	{"X-MAC-CHINESEIMP", 10008},
14136c95142eSMartin Matuska 	{"X-MAC-CHINESETRAD", 10002},
14146c95142eSMartin Matuska 	{"X-MAC-CROATIAN", 10082},
14156c95142eSMartin Matuska 	{"X-MAC-CYRILLIC", 10007},
14166c95142eSMartin Matuska 	{"X-MAC-GREEK", 10006},
14176c95142eSMartin Matuska 	{"X-MAC-HEBREW", 10005},
14186c95142eSMartin Matuska 	{"X-MAC-ICELANDIC", 10079},
14196c95142eSMartin Matuska 	{"X-MAC-JAPANESE", 10001},
14206c95142eSMartin Matuska 	{"X-MAC-KOREAN", 10003},
14216c95142eSMartin Matuska 	{"X-MAC-ROMANIAN", 10010},
14226c95142eSMartin Matuska 	{"X-MAC-THAI", 10021},
14236c95142eSMartin Matuska 	{"X-MAC-TURKISH", 10081},
14246c95142eSMartin Matuska 	{"X-MAC-UKRAINIAN", 10017},
14256c95142eSMartin Matuska };
14266c95142eSMartin Matuska static unsigned
14276c95142eSMartin Matuska make_codepage_from_charset(const char *charset)
14286c95142eSMartin Matuska {
14296c95142eSMartin Matuska 	char cs[16];
14306c95142eSMartin Matuska 	char *p;
14316c95142eSMartin Matuska 	unsigned cp;
14326c95142eSMartin Matuska 	int a, b;
14336c95142eSMartin Matuska 
14346c95142eSMartin Matuska 	if (charset == NULL || strlen(charset) > 15)
14356c95142eSMartin Matuska 		return -1;
14366c95142eSMartin Matuska 
14376c95142eSMartin Matuska 	/* Copy name to uppercase. */
14386c95142eSMartin Matuska 	p = cs;
14396c95142eSMartin Matuska 	while (*charset) {
14406c95142eSMartin Matuska 		char c = *charset++;
14416c95142eSMartin Matuska 		if (c >= 'a' && c <= 'z')
14426c95142eSMartin Matuska 			c -= 'a' - 'A';
14436c95142eSMartin Matuska 		*p++ = c;
14446c95142eSMartin Matuska 	}
14456c95142eSMartin Matuska 	*p++ = '\0';
14466c95142eSMartin Matuska 	cp = -1;
14476c95142eSMartin Matuska 
14486c95142eSMartin Matuska 	/* Look it up in the table first, so that we can easily
14496c95142eSMartin Matuska 	 * override CP367, which we map to 1252 instead of 367. */
14506c95142eSMartin Matuska 	a = 0;
14516c95142eSMartin Matuska 	b = sizeof(charsets)/sizeof(charsets[0]);
14526c95142eSMartin Matuska 	while (b > a) {
14536c95142eSMartin Matuska 		int c = (b + a) / 2;
14546c95142eSMartin Matuska 		int r = strcmp(charsets[c].name, cs);
14556c95142eSMartin Matuska 		if (r < 0)
14566c95142eSMartin Matuska 			a = c + 1;
14576c95142eSMartin Matuska 		else if (r > 0)
14586c95142eSMartin Matuska 			b = c;
14596c95142eSMartin Matuska 		else
14606c95142eSMartin Matuska 			return charsets[c].cp;
14616c95142eSMartin Matuska 	}
14626c95142eSMartin Matuska 
14636c95142eSMartin Matuska 	/* If it's not in the table, try to parse it. */
14646c95142eSMartin Matuska 	switch (*cs) {
14656c95142eSMartin Matuska 	case 'C':
14666c95142eSMartin Matuska 		if (cs[1] == 'P' && cs[2] >= '0' && cs[2] <= '9') {
14676c95142eSMartin Matuska 			cp = my_atoi(cs + 2);
14686c95142eSMartin Matuska 		} else if (strcmp(cs, "CP_ACP") == 0)
14696c95142eSMartin Matuska 			cp = get_current_codepage();
14706c95142eSMartin Matuska 		else if (strcmp(cs, "CP_OEMCP") == 0)
14716c95142eSMartin Matuska 			cp = get_current_oemcp();
14726c95142eSMartin Matuska 		break;
14736c95142eSMartin Matuska 	case 'I':
14746c95142eSMartin Matuska 		if (cs[1] == 'B' && cs[2] == 'M' &&
14756c95142eSMartin Matuska 		    cs[3] >= '0' && cs[3] <= '9') {
14766c95142eSMartin Matuska 			cp = my_atoi(cs + 3);
14776c95142eSMartin Matuska 		}
14786c95142eSMartin Matuska 		break;
14796c95142eSMartin Matuska 	case 'W':
14806c95142eSMartin Matuska 		if (strncmp(cs, "WINDOWS-", 8) == 0) {
14816c95142eSMartin Matuska 			cp = my_atoi(cs + 8);
14826c95142eSMartin Matuska 			if (cp != 874 && (cp < 1250 || cp > 1258))
14836c95142eSMartin Matuska 				cp = -1;/* This may invalid code. */
14846c95142eSMartin Matuska 		}
14856c95142eSMartin Matuska 		break;
14866c95142eSMartin Matuska 	}
14876c95142eSMartin Matuska 	return (cp);
14886c95142eSMartin Matuska }
14896c95142eSMartin Matuska 
14906c95142eSMartin Matuska /*
14916c95142eSMartin Matuska  * Return ANSI Code Page of current locale set by setlocale().
14926c95142eSMartin Matuska  */
14936c95142eSMartin Matuska static unsigned
14946c95142eSMartin Matuska get_current_codepage(void)
14956c95142eSMartin Matuska {
14966c95142eSMartin Matuska 	char *locale, *p;
14976c95142eSMartin Matuska 	unsigned cp;
14986c95142eSMartin Matuska 
14996c95142eSMartin Matuska 	locale = setlocale(LC_CTYPE, NULL);
15006c95142eSMartin Matuska 	if (locale == NULL)
15016c95142eSMartin Matuska 		return (GetACP());
15026c95142eSMartin Matuska 	if (locale[0] == 'C' && locale[1] == '\0')
15036c95142eSMartin Matuska 		return (CP_C_LOCALE);
15046c95142eSMartin Matuska 	p = strrchr(locale, '.');
15056c95142eSMartin Matuska 	if (p == NULL)
15066c95142eSMartin Matuska 		return (GetACP());
15076c95142eSMartin Matuska 	cp = my_atoi(p+1);
15086c95142eSMartin Matuska 	if (cp <= 0)
15096c95142eSMartin Matuska 		return (GetACP());
15106c95142eSMartin Matuska 	return (cp);
15116c95142eSMartin Matuska }
15126c95142eSMartin Matuska 
15136c95142eSMartin Matuska /*
15146c95142eSMartin Matuska  * Translation table between Locale Name and ACP/OEMCP.
15156c95142eSMartin Matuska  */
15166c95142eSMartin Matuska static struct {
15176c95142eSMartin Matuska 	unsigned acp;
15186c95142eSMartin Matuska 	unsigned ocp;
15196c95142eSMartin Matuska 	const char *locale;
15206c95142eSMartin Matuska } acp_ocp_map[] = {
15216c95142eSMartin Matuska 	{  950,  950, "Chinese_Taiwan" },
15226c95142eSMartin Matuska 	{  936,  936, "Chinese_People's Republic of China" },
15236c95142eSMartin Matuska 	{  950,  950, "Chinese_Taiwan" },
15246c95142eSMartin Matuska 	{ 1250,  852, "Czech_Czech Republic" },
15256c95142eSMartin Matuska 	{ 1252,  850, "Danish_Denmark" },
15266c95142eSMartin Matuska 	{ 1252,  850, "Dutch_Netherlands" },
15276c95142eSMartin Matuska 	{ 1252,  850, "Dutch_Belgium" },
15286c95142eSMartin Matuska 	{ 1252,  437, "English_United States" },
15296c95142eSMartin Matuska 	{ 1252,  850, "English_Australia" },
15306c95142eSMartin Matuska 	{ 1252,  850, "English_Canada" },
15316c95142eSMartin Matuska 	{ 1252,  850, "English_New Zealand" },
15326c95142eSMartin Matuska 	{ 1252,  850, "English_United Kingdom" },
15336c95142eSMartin Matuska 	{ 1252,  437, "English_United States" },
15346c95142eSMartin Matuska 	{ 1252,  850, "Finnish_Finland" },
15356c95142eSMartin Matuska 	{ 1252,  850, "French_France" },
15366c95142eSMartin Matuska 	{ 1252,  850, "French_Belgium" },
15376c95142eSMartin Matuska 	{ 1252,  850, "French_Canada" },
15386c95142eSMartin Matuska 	{ 1252,  850, "French_Switzerland" },
15396c95142eSMartin Matuska 	{ 1252,  850, "German_Germany" },
15406c95142eSMartin Matuska 	{ 1252,  850, "German_Austria" },
15416c95142eSMartin Matuska 	{ 1252,  850, "German_Switzerland" },
15426c95142eSMartin Matuska 	{ 1253,  737, "Greek_Greece" },
15436c95142eSMartin Matuska 	{ 1250,  852, "Hungarian_Hungary" },
15446c95142eSMartin Matuska 	{ 1252,  850, "Icelandic_Iceland" },
15456c95142eSMartin Matuska 	{ 1252,  850, "Italian_Italy" },
15466c95142eSMartin Matuska 	{ 1252,  850, "Italian_Switzerland" },
15476c95142eSMartin Matuska 	{  932,  932, "Japanese_Japan" },
15486c95142eSMartin Matuska 	{  949,  949, "Korean_Korea" },
15496c95142eSMartin Matuska 	{ 1252,  850, "Norwegian (BokmOl)_Norway" },
15506c95142eSMartin Matuska 	{ 1252,  850, "Norwegian (BokmOl)_Norway" },
15516c95142eSMartin Matuska 	{ 1252,  850, "Norwegian-Nynorsk_Norway" },
15526c95142eSMartin Matuska 	{ 1250,  852, "Polish_Poland" },
15536c95142eSMartin Matuska 	{ 1252,  850, "Portuguese_Portugal" },
15546c95142eSMartin Matuska 	{ 1252,  850, "Portuguese_Brazil" },
15556c95142eSMartin Matuska 	{ 1251,  866, "Russian_Russia" },
15566c95142eSMartin Matuska 	{ 1250,  852, "Slovak_Slovakia" },
15576c95142eSMartin Matuska 	{ 1252,  850, "Spanish_Spain" },
15586c95142eSMartin Matuska 	{ 1252,  850, "Spanish_Mexico" },
15596c95142eSMartin Matuska 	{ 1252,  850, "Spanish_Spain" },
15606c95142eSMartin Matuska 	{ 1252,  850, "Swedish_Sweden" },
15616c95142eSMartin Matuska 	{ 1254,  857, "Turkish_Turkey" },
15626c95142eSMartin Matuska 	{ 0, 0, NULL}
15636c95142eSMartin Matuska };
15646c95142eSMartin Matuska 
15656c95142eSMartin Matuska /*
15666c95142eSMartin Matuska  * Return OEM Code Page of current locale set by setlocale().
15676c95142eSMartin Matuska  */
15686c95142eSMartin Matuska static unsigned
15696c95142eSMartin Matuska get_current_oemcp(void)
15706c95142eSMartin Matuska {
15716c95142eSMartin Matuska 	int i;
15726c95142eSMartin Matuska 	char *locale, *p;
15736c95142eSMartin Matuska 	size_t len;
15746c95142eSMartin Matuska 
15756c95142eSMartin Matuska 	locale = setlocale(LC_CTYPE, NULL);
15766c95142eSMartin Matuska 	if (locale == NULL)
15776c95142eSMartin Matuska 		return (GetOEMCP());
15786c95142eSMartin Matuska 	if (locale[0] == 'C' && locale[1] == '\0')
15796c95142eSMartin Matuska 		return (CP_C_LOCALE);
15806c95142eSMartin Matuska 
15816c95142eSMartin Matuska 	p = strrchr(locale, '.');
15826c95142eSMartin Matuska 	if (p == NULL)
15836c95142eSMartin Matuska 		return (GetOEMCP());
15846c95142eSMartin Matuska 	len = p - locale;
15856c95142eSMartin Matuska 	for (i = 0; acp_ocp_map[i].acp; i++) {
15866c95142eSMartin Matuska 		if (strncmp(acp_ocp_map[i].locale, locale, len) == 0)
15876c95142eSMartin Matuska 			return (acp_ocp_map[i].ocp);
15886c95142eSMartin Matuska 	}
15896c95142eSMartin Matuska 	return (GetOEMCP());
15906c95142eSMartin Matuska }
15916c95142eSMartin Matuska #else
15926c95142eSMartin Matuska 
15936c95142eSMartin Matuska /*
15946c95142eSMartin Matuska  * POSIX platform does not use CodePage.
15956c95142eSMartin Matuska  */
15966c95142eSMartin Matuska 
15976c95142eSMartin Matuska static unsigned
15986c95142eSMartin Matuska get_current_codepage(void)
15996c95142eSMartin Matuska {
16006c95142eSMartin Matuska 	return (-1);/* Unknown */
16016c95142eSMartin Matuska }
16026c95142eSMartin Matuska static unsigned
16036c95142eSMartin Matuska make_codepage_from_charset(const char *charset)
16046c95142eSMartin Matuska {
16056c95142eSMartin Matuska 	(void)charset; /* UNUSED */
16066c95142eSMartin Matuska 	return (-1);/* Unknown */
16076c95142eSMartin Matuska }
16086c95142eSMartin Matuska static unsigned
16096c95142eSMartin Matuska get_current_oemcp(void)
16106c95142eSMartin Matuska {
16116c95142eSMartin Matuska 	return (-1);/* Unknown */
16126c95142eSMartin Matuska }
16136c95142eSMartin Matuska 
16146c95142eSMartin Matuska #endif /* defined(_WIN32) && !defined(__CYGWIN__) */
16156c95142eSMartin Matuska 
16166c95142eSMartin Matuska /*
16176c95142eSMartin Matuska  * Return a string conversion object.
16186c95142eSMartin Matuska  */
16196c95142eSMartin Matuska static struct archive_string_conv *
16206c95142eSMartin Matuska get_sconv_object(struct archive *a, const char *fc, const char *tc, int flag)
16216c95142eSMartin Matuska {
16226c95142eSMartin Matuska 	struct archive_string_conv *sc;
16236c95142eSMartin Matuska 	unsigned current_codepage;
16246c95142eSMartin Matuska 
16256c95142eSMartin Matuska 	/* Check if we have made the sconv object. */
16266c95142eSMartin Matuska 	sc = find_sconv_object(a, fc, tc);
16276c95142eSMartin Matuska 	if (sc != NULL)
16286c95142eSMartin Matuska 		return (sc);
16296c95142eSMartin Matuska 
16306c95142eSMartin Matuska 	if (a == NULL)
16316c95142eSMartin Matuska 		current_codepage = get_current_codepage();
16326c95142eSMartin Matuska 	else
16336c95142eSMartin Matuska 		current_codepage = a->current_codepage;
16346c95142eSMartin Matuska 
16356c95142eSMartin Matuska 	sc = create_sconv_object(canonical_charset_name(fc),
16366c95142eSMartin Matuska 	    canonical_charset_name(tc), current_codepage, flag);
16376c95142eSMartin Matuska 	if (sc == NULL) {
16386c95142eSMartin Matuska 		if (a != NULL)
16396c95142eSMartin Matuska 			archive_set_error(a, ENOMEM,
16406c95142eSMartin Matuska 			    "Could not allocate memory for "
16416c95142eSMartin Matuska 			    "a string conversion object");
16426c95142eSMartin Matuska 		return (NULL);
16436c95142eSMartin Matuska 	}
16446c95142eSMartin Matuska 
16456c95142eSMartin Matuska 	/*
16466c95142eSMartin Matuska 	 * If there is no converter for current string conversion object,
16476c95142eSMartin Matuska 	 * we cannot handle this conversion.
16486c95142eSMartin Matuska 	 */
16496c95142eSMartin Matuska 	if (sc->nconverter == 0) {
16506c95142eSMartin Matuska 		if (a != NULL) {
16516c95142eSMartin Matuska #if HAVE_ICONV
16526c95142eSMartin Matuska 			archive_set_error(a, ARCHIVE_ERRNO_MISC,
16536c95142eSMartin Matuska 			    "iconv_open failed : Cannot handle ``%s''",
16546c95142eSMartin Matuska 			    (flag & SCONV_TO_CHARSET)?tc:fc);
16556c95142eSMartin Matuska #else
16566c95142eSMartin Matuska 			archive_set_error(a, ARCHIVE_ERRNO_MISC,
16576c95142eSMartin Matuska 			    "A character-set conversion not fully supported "
16586c95142eSMartin Matuska 			    "on this platform");
16596c95142eSMartin Matuska #endif
16606c95142eSMartin Matuska 		}
16616c95142eSMartin Matuska 		/* Failed; free a sconv object. */
16626c95142eSMartin Matuska 		free_sconv_object(sc);
16636c95142eSMartin Matuska 		return (NULL);
16646c95142eSMartin Matuska 	}
16656c95142eSMartin Matuska 
16666c95142eSMartin Matuska 	/*
16676c95142eSMartin Matuska 	 * Success!
16686c95142eSMartin Matuska 	 */
16696c95142eSMartin Matuska 	if (a != NULL)
16706c95142eSMartin Matuska 		add_sconv_object(a, sc);
16716c95142eSMartin Matuska 	return (sc);
16726c95142eSMartin Matuska }
16736c95142eSMartin Matuska 
16746c95142eSMartin Matuska static const char *
16756c95142eSMartin Matuska get_current_charset(struct archive *a)
16766c95142eSMartin Matuska {
16776c95142eSMartin Matuska 	const char *cur_charset;
16786c95142eSMartin Matuska 
16796c95142eSMartin Matuska 	if (a == NULL)
16806c95142eSMartin Matuska 		cur_charset = default_iconv_charset("");
16816c95142eSMartin Matuska 	else {
16826c95142eSMartin Matuska 		cur_charset = default_iconv_charset(a->current_code);
16836c95142eSMartin Matuska 		if (a->current_code == NULL) {
16846c95142eSMartin Matuska 			a->current_code = strdup(cur_charset);
16856c95142eSMartin Matuska 			a->current_codepage = get_current_codepage();
16866c95142eSMartin Matuska 			a->current_oemcp = get_current_oemcp();
16876c95142eSMartin Matuska 		}
16886c95142eSMartin Matuska 	}
16896c95142eSMartin Matuska 	return (cur_charset);
16906c95142eSMartin Matuska }
16916c95142eSMartin Matuska 
16926c95142eSMartin Matuska /*
16936c95142eSMartin Matuska  * Make and Return a string conversion object.
16946c95142eSMartin Matuska  * Return NULL if the platform does not support the specified conversion
16956c95142eSMartin Matuska  * and best_effort is 0.
16966c95142eSMartin Matuska  * If best_effort is set, A string conversion object must be returned
16976c95142eSMartin Matuska  * unless memory allocation for the object fails, but the conversion
16986c95142eSMartin Matuska  * might fail when non-ASCII code is found.
16996c95142eSMartin Matuska  */
17006c95142eSMartin Matuska struct archive_string_conv *
17016c95142eSMartin Matuska archive_string_conversion_to_charset(struct archive *a, const char *charset,
17026c95142eSMartin Matuska     int best_effort)
17036c95142eSMartin Matuska {
17046c95142eSMartin Matuska 	int flag = SCONV_TO_CHARSET;
17056c95142eSMartin Matuska 
17066c95142eSMartin Matuska 	if (best_effort)
17076c95142eSMartin Matuska 		flag |= SCONV_BEST_EFFORT;
17086c95142eSMartin Matuska 	return (get_sconv_object(a, get_current_charset(a), charset, flag));
17096c95142eSMartin Matuska }
17106c95142eSMartin Matuska 
17116c95142eSMartin Matuska struct archive_string_conv *
17126c95142eSMartin Matuska archive_string_conversion_from_charset(struct archive *a, const char *charset,
17136c95142eSMartin Matuska     int best_effort)
17146c95142eSMartin Matuska {
17156c95142eSMartin Matuska 	int flag = SCONV_FROM_CHARSET;
17166c95142eSMartin Matuska 
17176c95142eSMartin Matuska 	if (best_effort)
17186c95142eSMartin Matuska 		flag |= SCONV_BEST_EFFORT;
17196c95142eSMartin Matuska 	return (get_sconv_object(a, charset, get_current_charset(a), flag));
17206c95142eSMartin Matuska }
17216c95142eSMartin Matuska 
17226c95142eSMartin Matuska /*
17236c95142eSMartin Matuska  * archive_string_default_conversion_*_archive() are provided for Windows
17246c95142eSMartin Matuska  * platform because other archiver application use CP_OEMCP for
17256c95142eSMartin Matuska  * MultiByteToWideChar() and WideCharToMultiByte() for the filenames
17266c95142eSMartin Matuska  * in tar or zip files. But mbstowcs/wcstombs(CRT) usually use CP_ACP
17276c95142eSMartin Matuska  * unless you use setlocale(LC_ALL, ".OCP")(specify CP_OEMCP).
17286c95142eSMartin Matuska  * So we should make a string conversion between CP_ACP and CP_OEMCP
17296c95142eSMartin Matuska  * for compatibillty.
17306c95142eSMartin Matuska  */
17316c95142eSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
17326c95142eSMartin Matuska struct archive_string_conv *
17336c95142eSMartin Matuska archive_string_default_conversion_for_read(struct archive *a)
17346c95142eSMartin Matuska {
17356c95142eSMartin Matuska 	const char *cur_charset = get_current_charset(a);
17366c95142eSMartin Matuska 	char oemcp[16];
17376c95142eSMartin Matuska 
17386c95142eSMartin Matuska 	/* NOTE: a check of cur_charset is unneeded but we need
17396c95142eSMartin Matuska 	 * that get_current_charset() has been surely called at
17406c95142eSMartin Matuska 	 * this time whatever C compiler optimized. */
17416c95142eSMartin Matuska 	if (cur_charset != NULL &&
17426c95142eSMartin Matuska 	    (a->current_codepage == CP_C_LOCALE ||
17436c95142eSMartin Matuska 	     a->current_codepage == a->current_oemcp))
17446c95142eSMartin Matuska 		return (NULL);/* no conversion. */
17456c95142eSMartin Matuska 
17466c95142eSMartin Matuska 	_snprintf(oemcp, sizeof(oemcp)-1, "CP%d", a->current_oemcp);
17476c95142eSMartin Matuska 	/* Make sure a null termination must be set. */
17486c95142eSMartin Matuska 	oemcp[sizeof(oemcp)-1] = '\0';
17496c95142eSMartin Matuska 	return (get_sconv_object(a, oemcp, cur_charset,
17506c95142eSMartin Matuska 	    SCONV_FROM_CHARSET));
17516c95142eSMartin Matuska }
17526c95142eSMartin Matuska 
17536c95142eSMartin Matuska struct archive_string_conv *
17546c95142eSMartin Matuska archive_string_default_conversion_for_write(struct archive *a)
17556c95142eSMartin Matuska {
17566c95142eSMartin Matuska 	const char *cur_charset = get_current_charset(a);
17576c95142eSMartin Matuska 	char oemcp[16];
17586c95142eSMartin Matuska 
17596c95142eSMartin Matuska 	/* NOTE: a check of cur_charset is unneeded but we need
17606c95142eSMartin Matuska 	 * that get_current_charset() has been surely called at
17616c95142eSMartin Matuska 	 * this time whatever C compiler optimized. */
17626c95142eSMartin Matuska 	if (cur_charset != NULL &&
17636c95142eSMartin Matuska 	    (a->current_codepage == CP_C_LOCALE ||
17646c95142eSMartin Matuska 	     a->current_codepage == a->current_oemcp))
17656c95142eSMartin Matuska 		return (NULL);/* no conversion. */
17666c95142eSMartin Matuska 
17676c95142eSMartin Matuska 	_snprintf(oemcp, sizeof(oemcp)-1, "CP%d", a->current_oemcp);
17686c95142eSMartin Matuska 	/* Make sure a null termination must be set. */
17696c95142eSMartin Matuska 	oemcp[sizeof(oemcp)-1] = '\0';
17706c95142eSMartin Matuska 	return (get_sconv_object(a, cur_charset, oemcp,
17716c95142eSMartin Matuska 	    SCONV_TO_CHARSET));
17726c95142eSMartin Matuska }
17736c95142eSMartin Matuska #else
17746c95142eSMartin Matuska struct archive_string_conv *
17756c95142eSMartin Matuska archive_string_default_conversion_for_read(struct archive *a)
17766c95142eSMartin Matuska {
17776c95142eSMartin Matuska 	(void)a; /* UNUSED */
17786c95142eSMartin Matuska 	return (NULL);
17796c95142eSMartin Matuska }
17806c95142eSMartin Matuska 
17816c95142eSMartin Matuska struct archive_string_conv *
17826c95142eSMartin Matuska archive_string_default_conversion_for_write(struct archive *a)
17836c95142eSMartin Matuska {
17846c95142eSMartin Matuska 	(void)a; /* UNUSED */
17856c95142eSMartin Matuska 	return (NULL);
17866c95142eSMartin Matuska }
17876c95142eSMartin Matuska #endif
17886c95142eSMartin Matuska 
17896c95142eSMartin Matuska /*
17906c95142eSMartin Matuska  * Dispose of all character conversion objects in the archive object.
17916c95142eSMartin Matuska  */
17926c95142eSMartin Matuska void
17936c95142eSMartin Matuska archive_string_conversion_free(struct archive *a)
17946c95142eSMartin Matuska {
17956c95142eSMartin Matuska 	struct archive_string_conv *sc;
17966c95142eSMartin Matuska 	struct archive_string_conv *sc_next;
17976c95142eSMartin Matuska 
17986c95142eSMartin Matuska 	for (sc = a->sconv; sc != NULL; sc = sc_next) {
17996c95142eSMartin Matuska 		sc_next = sc->next;
18006c95142eSMartin Matuska 		free_sconv_object(sc);
18016c95142eSMartin Matuska 	}
18026c95142eSMartin Matuska 	a->sconv = NULL;
18036c95142eSMartin Matuska 	free(a->current_code);
18046c95142eSMartin Matuska 	a->current_code = NULL;
18056c95142eSMartin Matuska }
18066c95142eSMartin Matuska 
18076c95142eSMartin Matuska /*
18086c95142eSMartin Matuska  * Return a conversion charset name.
18096c95142eSMartin Matuska  */
18106c95142eSMartin Matuska const char *
18116c95142eSMartin Matuska archive_string_conversion_charset_name(struct archive_string_conv *sc)
18126c95142eSMartin Matuska {
18136c95142eSMartin Matuska 	if (sc->flag & SCONV_TO_CHARSET)
18146c95142eSMartin Matuska 		return (sc->to_charset);
18156c95142eSMartin Matuska 	else
18166c95142eSMartin Matuska 		return (sc->from_charset);
18176c95142eSMartin Matuska }
18186c95142eSMartin Matuska 
18196c95142eSMartin Matuska /*
18206c95142eSMartin Matuska  * Change the behavior of a string conversion.
18216c95142eSMartin Matuska  */
18226c95142eSMartin Matuska void
18236c95142eSMartin Matuska archive_string_conversion_set_opt(struct archive_string_conv *sc, int opt)
18246c95142eSMartin Matuska {
18256c95142eSMartin Matuska 	switch (opt) {
18266c95142eSMartin Matuska 	/*
18276c95142eSMartin Matuska 	 * A filename in UTF-8 was made with libarchive 2.x in a wrong
18286c95142eSMartin Matuska 	 * assumption that wchar_t was Unicode.
18296c95142eSMartin Matuska 	 * This option enables simulating the assumption in order to read
18306c95142eSMartin Matuska 	 * that filname correctly.
18316c95142eSMartin Matuska 	 */
18326c95142eSMartin Matuska 	case SCONV_SET_OPT_UTF8_LIBARCHIVE2X:
18336c95142eSMartin Matuska #if (defined(_WIN32) && !defined(__CYGWIN__)) \
18346c95142eSMartin Matuska 	 || defined(__STDC_ISO_10646__) || defined(__APPLE__)
18356c95142eSMartin Matuska 		/*
18366c95142eSMartin Matuska 		 * Nothing to do for it since wchar_t on these platforms
18376c95142eSMartin Matuska 		 * is really Unicode.
18386c95142eSMartin Matuska 		 */
18396c95142eSMartin Matuska 		(void)sc; /* UNUSED */
18406c95142eSMartin Matuska #else
18416c95142eSMartin Matuska 		if ((sc->flag & SCONV_UTF8_LIBARCHIVE_2) == 0) {
18426c95142eSMartin Matuska 			sc->flag |= SCONV_UTF8_LIBARCHIVE_2;
1843fd082e96SMartin Matuska 			/* Set up string converters. */
18446c95142eSMartin Matuska 			setup_converter(sc);
18456c95142eSMartin Matuska 		}
18466c95142eSMartin Matuska #endif
18476c95142eSMartin Matuska 		break;
1848fd082e96SMartin Matuska 	case SCONV_SET_OPT_NORMALIZATION_C:
1849fd082e96SMartin Matuska 		if ((sc->flag & SCONV_NORMALIZATION_C) == 0) {
1850fd082e96SMartin Matuska 			sc->flag |= SCONV_NORMALIZATION_C;
1851fd082e96SMartin Matuska 			sc->flag &= ~SCONV_NORMALIZATION_D;
1852fd082e96SMartin Matuska 			/* Set up string converters. */
1853fd082e96SMartin Matuska 			setup_converter(sc);
1854fd082e96SMartin Matuska 		}
1855fd082e96SMartin Matuska 		break;
1856fd082e96SMartin Matuska 	case SCONV_SET_OPT_NORMALIZATION_D:
1857fd082e96SMartin Matuska #if defined(HAVE_ICONV)
1858fd082e96SMartin Matuska 		/*
1859fd082e96SMartin Matuska 		 * If iconv will take the string, do not change the
1860fd082e96SMartin Matuska 		 * setting of the normalization.
1861fd082e96SMartin Matuska 		 */
1862fd082e96SMartin Matuska 		if (!(sc->flag & SCONV_WIN_CP) &&
1863fd082e96SMartin Matuska 		     (sc->flag & (SCONV_FROM_UTF16 | SCONV_FROM_UTF8)) &&
1864fd082e96SMartin Matuska 		    !(sc->flag & (SCONV_TO_UTF16 | SCONV_TO_UTF8)))
1865fd082e96SMartin Matuska 			break;
1866fd082e96SMartin Matuska #endif
1867fd082e96SMartin Matuska 		if ((sc->flag & SCONV_NORMALIZATION_D) == 0) {
1868fd082e96SMartin Matuska 			sc->flag |= SCONV_NORMALIZATION_D;
1869fd082e96SMartin Matuska 			sc->flag &= ~SCONV_NORMALIZATION_C;
1870fd082e96SMartin Matuska 			/* Set up string converters. */
1871fd082e96SMartin Matuska 			setup_converter(sc);
1872fd082e96SMartin Matuska 		}
1873fd082e96SMartin Matuska 		break;
18746c95142eSMartin Matuska 	default:
18756c95142eSMartin Matuska 		break;
18766c95142eSMartin Matuska 	}
18776c95142eSMartin Matuska }
18786c95142eSMartin Matuska 
18796c95142eSMartin Matuska /*
18806c95142eSMartin Matuska  *
18816c95142eSMartin Matuska  * Copy one archive_string to another in locale conversion.
18826c95142eSMartin Matuska  *
1883fd082e96SMartin Matuska  *	archive_strncat_l();
1884fd082e96SMartin Matuska  *	archive_strncpy_l();
18856c95142eSMartin Matuska  *
18866c95142eSMartin Matuska  */
18876c95142eSMartin Matuska 
18886c95142eSMartin Matuska static size_t
18896c95142eSMartin Matuska mbsnbytes(const void *_p, size_t n)
18906c95142eSMartin Matuska {
18916c95142eSMartin Matuska 	size_t s;
18926c95142eSMartin Matuska 	const char *p, *pp;
18936c95142eSMartin Matuska 
18946c95142eSMartin Matuska 	if (_p == NULL)
18956c95142eSMartin Matuska 		return (0);
18966c95142eSMartin Matuska 	p = (const char *)_p;
18976c95142eSMartin Matuska 
18986c95142eSMartin Matuska 	/* Like strlen(p), except won't examine positions beyond p[n]. */
18996c95142eSMartin Matuska 	s = 0;
19006c95142eSMartin Matuska 	pp = p;
19016c95142eSMartin Matuska 	while (s < n && *pp) {
19026c95142eSMartin Matuska 		pp++;
19036c95142eSMartin Matuska 		s++;
19046c95142eSMartin Matuska 	}
19056c95142eSMartin Matuska 	return (s);
19066c95142eSMartin Matuska }
19076c95142eSMartin Matuska 
19086c95142eSMartin Matuska static size_t
19096c95142eSMartin Matuska utf16nbytes(const void *_p, size_t n)
19106c95142eSMartin Matuska {
19116c95142eSMartin Matuska 	size_t s;
19126c95142eSMartin Matuska 	const char *p, *pp;
19136c95142eSMartin Matuska 
19146c95142eSMartin Matuska 	if (_p == NULL)
19156c95142eSMartin Matuska 		return (0);
19166c95142eSMartin Matuska 	p = (const char *)_p;
19176c95142eSMartin Matuska 
19186c95142eSMartin Matuska 	/* Like strlen(p), except won't examine positions beyond p[n]. */
19196c95142eSMartin Matuska 	s = 0;
19206c95142eSMartin Matuska 	pp = p;
19216c95142eSMartin Matuska 	n >>= 1;
19226c95142eSMartin Matuska 	while (s < n && (pp[0] || pp[1])) {
19236c95142eSMartin Matuska 		pp += 2;
19246c95142eSMartin Matuska 		s++;
19256c95142eSMartin Matuska 	}
19266c95142eSMartin Matuska 	return (s<<1);
19276c95142eSMartin Matuska }
19286c95142eSMartin Matuska 
19296c95142eSMartin Matuska int
1930fd082e96SMartin Matuska archive_strncpy_l(struct archive_string *as, const void *_p, size_t n,
19316c95142eSMartin Matuska     struct archive_string_conv *sc)
19326c95142eSMartin Matuska {
19336c95142eSMartin Matuska 	as->length = 0;
1934fd082e96SMartin Matuska 	return (archive_strncat_l(as, _p, n, sc));
19356c95142eSMartin Matuska }
19366c95142eSMartin Matuska 
19376c95142eSMartin Matuska int
1938fd082e96SMartin Matuska archive_strncat_l(struct archive_string *as, const void *_p, size_t n,
19396c95142eSMartin Matuska     struct archive_string_conv *sc)
19406c95142eSMartin Matuska {
19416c95142eSMartin Matuska 	const void *s;
19426c95142eSMartin Matuska 	size_t length;
19436c95142eSMartin Matuska 	int i, r = 0, r2;
19446c95142eSMartin Matuska 
19456c95142eSMartin Matuska 	/* We must allocate memory even if there is no data for conversion
19466c95142eSMartin Matuska 	 * or copy. This simulates archive_string_append behavior. */
19476c95142eSMartin Matuska 	if (_p == NULL || n == 0) {
19486c95142eSMartin Matuska 		int tn = 1;
19496c95142eSMartin Matuska 		if (sc != NULL && (sc->flag & SCONV_TO_UTF16))
19506c95142eSMartin Matuska 			tn = 2;
19516c95142eSMartin Matuska 		if (archive_string_ensure(as, as->length + tn) == NULL)
19526c95142eSMartin Matuska 			return (-1);
19536c95142eSMartin Matuska 		as->s[as->length] = 0;
19546c95142eSMartin Matuska 		if (tn == 2)
19556c95142eSMartin Matuska 			as->s[as->length+1] = 0;
19566c95142eSMartin Matuska 		return (0);
19576c95142eSMartin Matuska 	}
19586c95142eSMartin Matuska 
19596c95142eSMartin Matuska 	/*
19606c95142eSMartin Matuska 	 * If sc is NULL, we just make a copy.
19616c95142eSMartin Matuska 	 */
19626c95142eSMartin Matuska 	if (sc == NULL) {
19636c95142eSMartin Matuska 		length = mbsnbytes(_p, n);
19646c95142eSMartin Matuska 		if (archive_string_append(as, _p, length) == NULL)
19656c95142eSMartin Matuska 			return (-1);/* No memory */
19666c95142eSMartin Matuska 		return (0);
19676c95142eSMartin Matuska 	}
19686c95142eSMartin Matuska 
19696c95142eSMartin Matuska 	if (sc->flag & SCONV_FROM_UTF16)
19706c95142eSMartin Matuska 		length = utf16nbytes(_p, n);
19716c95142eSMartin Matuska 	else
19726c95142eSMartin Matuska 		length = mbsnbytes(_p, n);
19736c95142eSMartin Matuska 	s = _p;
19746c95142eSMartin Matuska 	i = 0;
19756c95142eSMartin Matuska 	if (sc->nconverter > 1) {
19766c95142eSMartin Matuska 		sc->utftmp.length = 0;
19776c95142eSMartin Matuska 		r2 = sc->converter[0](&(sc->utftmp), s, length, sc);
19786c95142eSMartin Matuska 		if (r2 != 0 && errno == ENOMEM)
19796c95142eSMartin Matuska 			return (r2);
19806c95142eSMartin Matuska 		if (r > r2)
19816c95142eSMartin Matuska 			r = r2;
19826c95142eSMartin Matuska 		s = sc->utftmp.s;
19836c95142eSMartin Matuska 		length = sc->utftmp.length;
19846c95142eSMartin Matuska 		++i;
19856c95142eSMartin Matuska 	}
19866c95142eSMartin Matuska 	r2 = sc->converter[i](as, s, length, sc);
19876c95142eSMartin Matuska 	if (r > r2)
19886c95142eSMartin Matuska 		r = r2;
19896c95142eSMartin Matuska 	return (r);
19906c95142eSMartin Matuska }
19916c95142eSMartin Matuska 
19926c95142eSMartin Matuska #if HAVE_ICONV
19936c95142eSMartin Matuska 
19946c95142eSMartin Matuska /*
19956c95142eSMartin Matuska  * Return -1 if conversion failes.
19966c95142eSMartin Matuska  */
19976c95142eSMartin Matuska static int
19986c95142eSMartin Matuska iconv_strncat_in_locale(struct archive_string *as, const void *_p,
19996c95142eSMartin Matuska     size_t length, struct archive_string_conv *sc)
20006c95142eSMartin Matuska {
2001acc60b03SMartin Matuska 	ICONV_CONST char *itp;
20026c95142eSMartin Matuska 	size_t remaining;
20036c95142eSMartin Matuska 	iconv_t cd;
20046c95142eSMartin Matuska 	char *outp;
20056c95142eSMartin Matuska 	size_t avail, bs;
20066c95142eSMartin Matuska 	int return_value = 0; /* success */
20076c95142eSMartin Matuska 	int to_size, from_size;
20086c95142eSMartin Matuska 
20096c95142eSMartin Matuska 	if (sc->flag & SCONV_TO_UTF16)
20106c95142eSMartin Matuska 		to_size = 2;
20116c95142eSMartin Matuska 	else
20126c95142eSMartin Matuska 		to_size = 1;
20136c95142eSMartin Matuska 	if (sc->flag & SCONV_FROM_UTF16)
20146c95142eSMartin Matuska 		from_size = 2;
20156c95142eSMartin Matuska 	else
20166c95142eSMartin Matuska 		from_size = 1;
20176c95142eSMartin Matuska 
20186c95142eSMartin Matuska 	if (archive_string_ensure(as, as->length + length*2+to_size) == NULL)
20196c95142eSMartin Matuska 		return (-1);
20206c95142eSMartin Matuska 
20216c95142eSMartin Matuska 	cd = sc->cd;
2022acc60b03SMartin Matuska 	itp = (char *)(uintptr_t)_p;
20236c95142eSMartin Matuska 	remaining = length;
20246c95142eSMartin Matuska 	outp = as->s + as->length;
20256c95142eSMartin Matuska 	avail = as->buffer_length - as->length - to_size;
20266c95142eSMartin Matuska 	while (remaining >= (size_t)from_size) {
2027acc60b03SMartin Matuska 		size_t result = iconv(cd, &itp, &remaining, &outp, &avail);
20286c95142eSMartin Matuska 
20296c95142eSMartin Matuska 		if (result != (size_t)-1)
20306c95142eSMartin Matuska 			break; /* Conversion completed. */
20316c95142eSMartin Matuska 
20326c95142eSMartin Matuska 		if (errno == EILSEQ || errno == EINVAL) {
20336c95142eSMartin Matuska 			/*
20346c95142eSMartin Matuska 		 	 * If an output charset is UTF-8 or UTF-16BE/LE,
20356c95142eSMartin Matuska 			 * unknown character should be U+FFFD
20366c95142eSMartin Matuska 			 * (replacement character).
20376c95142eSMartin Matuska 			 */
20386c95142eSMartin Matuska 			if (sc->flag & (SCONV_TO_UTF8 | SCONV_TO_UTF16)) {
20396c95142eSMartin Matuska 				size_t rbytes;
20406c95142eSMartin Matuska 				if (sc->flag & SCONV_TO_UTF8)
2041cdf63a70SMartin Matuska 					rbytes = sizeof(utf8_replacement_char);
20426c95142eSMartin Matuska 				else
20436c95142eSMartin Matuska 					rbytes = 2;
20446c95142eSMartin Matuska 
20456c95142eSMartin Matuska 				if (avail < rbytes) {
20466c95142eSMartin Matuska 					as->length = outp - as->s;
20476c95142eSMartin Matuska 					bs = as->buffer_length +
20486c95142eSMartin Matuska 					    (remaining * to_size) + rbytes;
20496c95142eSMartin Matuska 					if (NULL ==
20506c95142eSMartin Matuska 					    archive_string_ensure(as, bs))
20516c95142eSMartin Matuska 						return (-1);
20526c95142eSMartin Matuska 					outp = as->s + as->length;
20536c95142eSMartin Matuska 					avail = as->buffer_length
20546c95142eSMartin Matuska 					    - as->length - to_size;
20556c95142eSMartin Matuska 				}
20566c95142eSMartin Matuska 				if (sc->flag & SCONV_TO_UTF8)
2057cdf63a70SMartin Matuska 					memcpy(outp, utf8_replacement_char, sizeof(utf8_replacement_char));
20586c95142eSMartin Matuska 				else if (sc->flag & SCONV_TO_UTF16BE)
20596c95142eSMartin Matuska 					archive_be16enc(outp, UNICODE_R_CHAR);
20606c95142eSMartin Matuska 				else
20616c95142eSMartin Matuska 					archive_le16enc(outp, UNICODE_R_CHAR);
20626c95142eSMartin Matuska 				outp += rbytes;
20636c95142eSMartin Matuska 				avail -= rbytes;
20646c95142eSMartin Matuska 			} else {
20656c95142eSMartin Matuska 				/* Skip the illegal input bytes. */
20666c95142eSMartin Matuska 				*outp++ = '?';
20676c95142eSMartin Matuska 				avail--;
20686c95142eSMartin Matuska 			}
2069acc60b03SMartin Matuska 			itp += from_size;
20706c95142eSMartin Matuska 			remaining -= from_size;
20716c95142eSMartin Matuska 			return_value = -1; /* failure */
20726c95142eSMartin Matuska 		} else {
20736c95142eSMartin Matuska 			/* E2BIG no output buffer,
20746c95142eSMartin Matuska 			 * Increase an output buffer.  */
20756c95142eSMartin Matuska 			as->length = outp - as->s;
20766c95142eSMartin Matuska 			bs = as->buffer_length + remaining * 2;
20776c95142eSMartin Matuska 			if (NULL == archive_string_ensure(as, bs))
20786c95142eSMartin Matuska 				return (-1);
20796c95142eSMartin Matuska 			outp = as->s + as->length;
20806c95142eSMartin Matuska 			avail = as->buffer_length - as->length - to_size;
20816c95142eSMartin Matuska 		}
20826c95142eSMartin Matuska 	}
20836c95142eSMartin Matuska 	as->length = outp - as->s;
20846c95142eSMartin Matuska 	as->s[as->length] = 0;
20856c95142eSMartin Matuska 	if (to_size == 2)
20866c95142eSMartin Matuska 		as->s[as->length+1] = 0;
20876c95142eSMartin Matuska 	return (return_value);
20886c95142eSMartin Matuska }
20896c95142eSMartin Matuska 
20906c95142eSMartin Matuska #endif /* HAVE_ICONV */
20916c95142eSMartin Matuska 
20926c95142eSMartin Matuska 
20936c95142eSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
20946c95142eSMartin Matuska 
20956c95142eSMartin Matuska /*
20966c95142eSMartin Matuska  * Translate a string from a some CodePage to an another CodePage by
20976c95142eSMartin Matuska  * Windows APIs, and copy the result. Return -1 if conversion failes.
20986c95142eSMartin Matuska  */
20996c95142eSMartin Matuska static int
21006c95142eSMartin Matuska strncat_in_codepage(struct archive_string *as,
21016c95142eSMartin Matuska     const void *_p, size_t length, struct archive_string_conv *sc)
21026c95142eSMartin Matuska {
21036c95142eSMartin Matuska 	const char *s = (const char *)_p;
21046c95142eSMartin Matuska 	struct archive_wstring aws;
21056c95142eSMartin Matuska 	size_t l;
21066c95142eSMartin Matuska 	int r, saved_flag;
21076c95142eSMartin Matuska 
21086c95142eSMartin Matuska 	archive_string_init(&aws);
21096c95142eSMartin Matuska 	saved_flag = sc->flag;
21106c95142eSMartin Matuska 	sc->flag &= ~(SCONV_NORMALIZATION_D | SCONV_NORMALIZATION_C);
21116c95142eSMartin Matuska 	r = archive_wstring_append_from_mbs_in_codepage(&aws, s, length, sc);
21126c95142eSMartin Matuska 	sc->flag = saved_flag;
21136c95142eSMartin Matuska 	if (r != 0) {
21146c95142eSMartin Matuska 		archive_wstring_free(&aws);
21156c95142eSMartin Matuska 		if (errno != ENOMEM)
21166c95142eSMartin Matuska 			archive_string_append(as, s, length);
21176c95142eSMartin Matuska 		return (-1);
21186c95142eSMartin Matuska 	}
21196c95142eSMartin Matuska 
21206c95142eSMartin Matuska 	l = as->length;
21216c95142eSMartin Matuska 	r = archive_string_append_from_wcs_in_codepage(
21226c95142eSMartin Matuska 	    as, aws.s, aws.length, sc);
21236c95142eSMartin Matuska 	if (r != 0 && errno != ENOMEM && l == as->length)
21246c95142eSMartin Matuska 		archive_string_append(as, s, length);
21256c95142eSMartin Matuska 	archive_wstring_free(&aws);
21266c95142eSMartin Matuska 	return (r);
21276c95142eSMartin Matuska }
21286c95142eSMartin Matuska 
21296c95142eSMartin Matuska /*
21306c95142eSMartin Matuska  * Test whether MBS ==> WCS is okay.
21316c95142eSMartin Matuska  */
21326c95142eSMartin Matuska static int
21336c95142eSMartin Matuska invalid_mbs(const void *_p, size_t n, struct archive_string_conv *sc)
21346c95142eSMartin Matuska {
21356c95142eSMartin Matuska 	const char *p = (const char *)_p;
21366c95142eSMartin Matuska 	unsigned codepage;
21376c95142eSMartin Matuska 	DWORD mbflag = MB_ERR_INVALID_CHARS;
21386c95142eSMartin Matuska 
21396c95142eSMartin Matuska 	if (sc->flag & SCONV_FROM_CHARSET)
21406c95142eSMartin Matuska 		codepage = sc->to_cp;
21416c95142eSMartin Matuska 	else
21426c95142eSMartin Matuska 		codepage = sc->from_cp;
21436c95142eSMartin Matuska 
21446c95142eSMartin Matuska 	if (codepage == CP_C_LOCALE)
21456c95142eSMartin Matuska 		return (0);
21466c95142eSMartin Matuska 	if (codepage != CP_UTF8)
21476c95142eSMartin Matuska 		mbflag |= MB_PRECOMPOSED;
21486c95142eSMartin Matuska 
2149acc60b03SMartin Matuska 	if (MultiByteToWideChar(codepage, mbflag, p, (int)n, NULL, 0) == 0)
21506c95142eSMartin Matuska 		return (-1); /* Invalid */
21516c95142eSMartin Matuska 	return (0); /* Okay */
21526c95142eSMartin Matuska }
21536c95142eSMartin Matuska 
21546c95142eSMartin Matuska #else
21556c95142eSMartin Matuska 
21566c95142eSMartin Matuska /*
21576c95142eSMartin Matuska  * Test whether MBS ==> WCS is okay.
21586c95142eSMartin Matuska  */
21596c95142eSMartin Matuska static int
21606c95142eSMartin Matuska invalid_mbs(const void *_p, size_t n, struct archive_string_conv *sc)
21616c95142eSMartin Matuska {
21626c95142eSMartin Matuska 	const char *p = (const char *)_p;
21636c95142eSMartin Matuska 	size_t r;
21646c95142eSMartin Matuska 
21656c95142eSMartin Matuska #if HAVE_MBRTOWC
21666c95142eSMartin Matuska 	mbstate_t shift_state;
21676c95142eSMartin Matuska 
21686c95142eSMartin Matuska 	memset(&shift_state, 0, sizeof(shift_state));
21696c95142eSMartin Matuska #else
21706c95142eSMartin Matuska 	/* Clear the shift state before starting. */
21716c95142eSMartin Matuska 	mbtowc(NULL, NULL, 0);
21726c95142eSMartin Matuska #endif
21736c95142eSMartin Matuska 	while (n) {
21746c95142eSMartin Matuska 		wchar_t wc;
21756c95142eSMartin Matuska 
21766c95142eSMartin Matuska #if HAVE_MBRTOWC
21776c95142eSMartin Matuska 		r = mbrtowc(&wc, p, n, &shift_state);
21786c95142eSMartin Matuska #else
21796c95142eSMartin Matuska 		r = mbtowc(&wc, p, n);
21806c95142eSMartin Matuska #endif
21816c95142eSMartin Matuska 		if (r == (size_t)-1 || r == (size_t)-2)
21826c95142eSMartin Matuska 			return (-1);/* Invalid. */
21836c95142eSMartin Matuska 		if (r == 0)
21846c95142eSMartin Matuska 			break;
21856c95142eSMartin Matuska 		p += r;
21866c95142eSMartin Matuska 		n -= r;
21876c95142eSMartin Matuska 	}
2188fd082e96SMartin Matuska 	(void)sc; /* UNUSED */
21896c95142eSMartin Matuska 	return (0); /* All Okey. */
21906c95142eSMartin Matuska }
21916c95142eSMartin Matuska 
21926c95142eSMartin Matuska #endif /* defined(_WIN32) && !defined(__CYGWIN__) */
21936c95142eSMartin Matuska 
21946c95142eSMartin Matuska /*
21956c95142eSMartin Matuska  * Basically returns -1 because we cannot make a conversion of charset
21966c95142eSMartin Matuska  * without iconv but in some cases this would return 0.
21976c95142eSMartin Matuska  * Returns 0 if all copied characters are ASCII.
21986c95142eSMartin Matuska  * Returns 0 if both from-locale and to-locale are the same and those
21996c95142eSMartin Matuska  * can be WCS with no error.
22006c95142eSMartin Matuska  */
22016c95142eSMartin Matuska static int
22026c95142eSMartin Matuska best_effort_strncat_in_locale(struct archive_string *as, const void *_p,
22036c95142eSMartin Matuska     size_t length, struct archive_string_conv *sc)
22046c95142eSMartin Matuska {
22056c95142eSMartin Matuska 	size_t remaining;
2206acc60b03SMartin Matuska 	const uint8_t *itp;
22076c95142eSMartin Matuska 	int return_value = 0; /* success */
22086c95142eSMartin Matuska 
22096c95142eSMartin Matuska 	/*
22106c95142eSMartin Matuska 	 * If both from-locale and to-locale is the same, this makes a copy.
22116c95142eSMartin Matuska 	 * And then this checks all copied MBS can be WCS if so returns 0.
22126c95142eSMartin Matuska 	 */
22136c95142eSMartin Matuska 	if (sc->same) {
22146c95142eSMartin Matuska 		if (archive_string_append(as, _p, length) == NULL)
22156c95142eSMartin Matuska 			return (-1);/* No memory */
22166c95142eSMartin Matuska 		return (invalid_mbs(_p, length, sc));
22176c95142eSMartin Matuska 	}
22186c95142eSMartin Matuska 
22196c95142eSMartin Matuska 	/*
22206c95142eSMartin Matuska 	 * If a character is ASCII, this just copies it. If not, this
22216c95142eSMartin Matuska 	 * assigns '?' charater instead but in UTF-8 locale this assigns
22226c95142eSMartin Matuska 	 * byte sequence 0xEF 0xBD 0xBD, which are code point U+FFFD,
22236c95142eSMartin Matuska 	 * a Replacement Character in Unicode.
22246c95142eSMartin Matuska 	 */
22256c95142eSMartin Matuska 
22266c95142eSMartin Matuska 	remaining = length;
2227acc60b03SMartin Matuska 	itp = (const uint8_t *)_p;
2228acc60b03SMartin Matuska 	while (*itp && remaining > 0) {
2229cdf63a70SMartin Matuska 		if (*itp > 127) {
2230cdf63a70SMartin Matuska 			// Non-ASCII: Substitute with suitable replacement
2231cdf63a70SMartin Matuska 			if (sc->flag & SCONV_TO_UTF8) {
2232cdf63a70SMartin Matuska 				if (archive_string_append(as, utf8_replacement_char, sizeof(utf8_replacement_char)) == NULL) {
2233cdf63a70SMartin Matuska 					__archive_errx(1, "Out of memory");
22346c95142eSMartin Matuska 				}
2235cdf63a70SMartin Matuska 			} else {
2236cdf63a70SMartin Matuska 				archive_strappend_char(as, '?');
2237cdf63a70SMartin Matuska 			}
22386c95142eSMartin Matuska 			return_value = -1;
22396c95142eSMartin Matuska 		} else {
2240cdf63a70SMartin Matuska 			archive_strappend_char(as, *itp);
22416c95142eSMartin Matuska 		}
2242cdf63a70SMartin Matuska 		++itp;
22436c95142eSMartin Matuska 	}
22446c95142eSMartin Matuska 	return (return_value);
22456c95142eSMartin Matuska }
22466c95142eSMartin Matuska 
22476c95142eSMartin Matuska 
22486c95142eSMartin Matuska /*
22496c95142eSMartin Matuska  * Unicode conversion functions.
22506c95142eSMartin Matuska  *   - UTF-8 <===> UTF-8 in removing surrogate pairs.
22516c95142eSMartin Matuska  *   - UTF-8 NFD ===> UTF-8 NFC in removing surrogate pairs.
22526c95142eSMartin Matuska  *   - UTF-8 made by libarchive 2.x ===> UTF-8.
22536c95142eSMartin Matuska  *   - UTF-16BE <===> UTF-8.
22546c95142eSMartin Matuska  *
22556c95142eSMartin Matuska  */
22566c95142eSMartin Matuska 
22576c95142eSMartin Matuska /*
22586c95142eSMartin Matuska  * Utility to convert a single UTF-8 sequence.
22596c95142eSMartin Matuska  *
22606c95142eSMartin Matuska  * Usually return used bytes, return used byte in negative value when
22616c95142eSMartin Matuska  * a unicode character is replaced with U+FFFD.
22626c95142eSMartin Matuska  * See also http://unicode.org/review/pr-121.html Public Review Issue #121
22636c95142eSMartin Matuska  * Recommended Practice for Replacement Characters.
22646c95142eSMartin Matuska  */
22656c95142eSMartin Matuska static int
22666c95142eSMartin Matuska _utf8_to_unicode(uint32_t *pwc, const char *s, size_t n)
22676c95142eSMartin Matuska {
22686c95142eSMartin Matuska 	static const char utf8_count[256] = {
22696c95142eSMartin Matuska 		 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 00 - 0F */
22706c95142eSMartin Matuska 		 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 10 - 1F */
22716c95142eSMartin Matuska 		 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 20 - 2F */
22726c95142eSMartin Matuska 		 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 30 - 3F */
22736c95142eSMartin Matuska 		 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 40 - 4F */
22746c95142eSMartin Matuska 		 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 50 - 5F */
22756c95142eSMartin Matuska 		 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 60 - 6F */
22766c95142eSMartin Matuska 		 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 70 - 7F */
22776c95142eSMartin Matuska 		 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,/* 80 - 8F */
22786c95142eSMartin Matuska 		 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,/* 90 - 9F */
22796c95142eSMartin Matuska 		 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,/* A0 - AF */
22806c95142eSMartin Matuska 		 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,/* B0 - BF */
22816c95142eSMartin Matuska 		 0, 0, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,/* C0 - CF */
22826c95142eSMartin Matuska 		 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,/* D0 - DF */
22836c95142eSMartin Matuska 		 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,/* E0 - EF */
22846c95142eSMartin Matuska 		 4, 4, 4, 4, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 /* F0 - FF */
22856c95142eSMartin Matuska 	};
22866c95142eSMartin Matuska 	int ch, i;
22876c95142eSMartin Matuska 	int cnt;
22886c95142eSMartin Matuska 	uint32_t wc;
22896c95142eSMartin Matuska 
22906c95142eSMartin Matuska 	/* Sanity check. */
22916c95142eSMartin Matuska 	if (n == 0)
22926c95142eSMartin Matuska 		return (0);
22936c95142eSMartin Matuska 	/*
22946c95142eSMartin Matuska 	 * Decode 1-4 bytes depending on the value of the first byte.
22956c95142eSMartin Matuska 	 */
22966c95142eSMartin Matuska 	ch = (unsigned char)*s;
22976c95142eSMartin Matuska 	if (ch == 0)
22986c95142eSMartin Matuska 		return (0); /* Standard:  return 0 for end-of-string. */
22996c95142eSMartin Matuska 	cnt = utf8_count[ch];
23006c95142eSMartin Matuska 
2301a2e802b7SMartin Matuska 	/* Invalid sequence or there are not plenty bytes. */
23026c95142eSMartin Matuska 	if ((int)n < cnt) {
2303acc60b03SMartin Matuska 		cnt = (int)n;
23046c95142eSMartin Matuska 		for (i = 1; i < cnt; i++) {
23056c95142eSMartin Matuska 			if ((s[i] & 0xc0) != 0x80) {
23066c95142eSMartin Matuska 				cnt = i;
23076c95142eSMartin Matuska 				break;
23086c95142eSMartin Matuska 			}
23096c95142eSMartin Matuska 		}
23106c95142eSMartin Matuska 		goto invalid_sequence;
23116c95142eSMartin Matuska 	}
23126c95142eSMartin Matuska 
23136c95142eSMartin Matuska 	/* Make a Unicode code point from a single UTF-8 sequence. */
23146c95142eSMartin Matuska 	switch (cnt) {
23156c95142eSMartin Matuska 	case 1:	/* 1 byte sequence. */
23166c95142eSMartin Matuska 		*pwc = ch & 0x7f;
23176c95142eSMartin Matuska 		return (cnt);
23186c95142eSMartin Matuska 	case 2:	/* 2 bytes sequence. */
23196c95142eSMartin Matuska 		if ((s[1] & 0xc0) != 0x80) {
23206c95142eSMartin Matuska 			cnt = 1;
23216c95142eSMartin Matuska 			goto invalid_sequence;
23226c95142eSMartin Matuska 		}
23236c95142eSMartin Matuska 		*pwc = ((ch & 0x1f) << 6) | (s[1] & 0x3f);
23246c95142eSMartin Matuska 		return (cnt);
23256c95142eSMartin Matuska 	case 3:	/* 3 bytes sequence. */
23266c95142eSMartin Matuska 		if ((s[1] & 0xc0) != 0x80) {
23276c95142eSMartin Matuska 			cnt = 1;
23286c95142eSMartin Matuska 			goto invalid_sequence;
23296c95142eSMartin Matuska 		}
23306c95142eSMartin Matuska 		if ((s[2] & 0xc0) != 0x80) {
23316c95142eSMartin Matuska 			cnt = 2;
23326c95142eSMartin Matuska 			goto invalid_sequence;
23336c95142eSMartin Matuska 		}
23346c95142eSMartin Matuska 		wc = ((ch & 0x0f) << 12)
23356c95142eSMartin Matuska 		    | ((s[1] & 0x3f) << 6)
23366c95142eSMartin Matuska 		    | (s[2] & 0x3f);
23376c95142eSMartin Matuska 		if (wc < 0x800)
23386c95142eSMartin Matuska 			goto invalid_sequence;/* Overlong sequence. */
23396c95142eSMartin Matuska 		break;
23406c95142eSMartin Matuska 	case 4:	/* 4 bytes sequence. */
23416c95142eSMartin Matuska 		if ((s[1] & 0xc0) != 0x80) {
23426c95142eSMartin Matuska 			cnt = 1;
23436c95142eSMartin Matuska 			goto invalid_sequence;
23446c95142eSMartin Matuska 		}
23456c95142eSMartin Matuska 		if ((s[2] & 0xc0) != 0x80) {
23466c95142eSMartin Matuska 			cnt = 2;
23476c95142eSMartin Matuska 			goto invalid_sequence;
23486c95142eSMartin Matuska 		}
23496c95142eSMartin Matuska 		if ((s[3] & 0xc0) != 0x80) {
23506c95142eSMartin Matuska 			cnt = 3;
23516c95142eSMartin Matuska 			goto invalid_sequence;
23526c95142eSMartin Matuska 		}
23536c95142eSMartin Matuska 		wc = ((ch & 0x07) << 18)
23546c95142eSMartin Matuska 		    | ((s[1] & 0x3f) << 12)
23556c95142eSMartin Matuska 		    | ((s[2] & 0x3f) << 6)
23566c95142eSMartin Matuska 		    | (s[3] & 0x3f);
23576c95142eSMartin Matuska 		if (wc < 0x10000)
23586c95142eSMartin Matuska 			goto invalid_sequence;/* Overlong sequence. */
23596c95142eSMartin Matuska 		break;
23606c95142eSMartin Matuska 	default: /* Others are all invalid sequence. */
23616c95142eSMartin Matuska 		if (ch == 0xc0 || ch == 0xc1)
23626c95142eSMartin Matuska 			cnt = 2;
23636c95142eSMartin Matuska 		else if (ch >= 0xf5 && ch <= 0xf7)
23646c95142eSMartin Matuska 			cnt = 4;
23656c95142eSMartin Matuska 		else if (ch >= 0xf8 && ch <= 0xfb)
23666c95142eSMartin Matuska 			cnt = 5;
23676c95142eSMartin Matuska 		else if (ch == 0xfc || ch == 0xfd)
23686c95142eSMartin Matuska 			cnt = 6;
23696c95142eSMartin Matuska 		else
23706c95142eSMartin Matuska 			cnt = 1;
23716c95142eSMartin Matuska 		if ((int)n < cnt)
2372acc60b03SMartin Matuska 			cnt = (int)n;
23736c95142eSMartin Matuska 		for (i = 1; i < cnt; i++) {
23746c95142eSMartin Matuska 			if ((s[i] & 0xc0) != 0x80) {
23756c95142eSMartin Matuska 				cnt = i;
23766c95142eSMartin Matuska 				break;
23776c95142eSMartin Matuska 			}
23786c95142eSMartin Matuska 		}
23796c95142eSMartin Matuska 		goto invalid_sequence;
23806c95142eSMartin Matuska 	}
23816c95142eSMartin Matuska 
2382a2e802b7SMartin Matuska 	/* The code point larger than 0x10FFFF is not legal
23836c95142eSMartin Matuska 	 * Unicode values. */
23846c95142eSMartin Matuska 	if (wc > UNICODE_MAX)
23856c95142eSMartin Matuska 		goto invalid_sequence;
23866c95142eSMartin Matuska 	/* Correctly gets a Unicode, returns used bytes. */
23876c95142eSMartin Matuska 	*pwc = wc;
23886c95142eSMartin Matuska 	return (cnt);
23896c95142eSMartin Matuska invalid_sequence:
23906c95142eSMartin Matuska 	*pwc = UNICODE_R_CHAR;/* set the Replacement Character instead. */
23916c95142eSMartin Matuska 	return (cnt * -1);
23926c95142eSMartin Matuska }
23936c95142eSMartin Matuska 
23946c95142eSMartin Matuska static int
23956c95142eSMartin Matuska utf8_to_unicode(uint32_t *pwc, const char *s, size_t n)
23966c95142eSMartin Matuska {
23976c95142eSMartin Matuska 	int cnt;
23986c95142eSMartin Matuska 
23996c95142eSMartin Matuska 	cnt = _utf8_to_unicode(pwc, s, n);
2400a2e802b7SMartin Matuska 	/* Any of Surrogate pair is not legal Unicode values. */
24016c95142eSMartin Matuska 	if (cnt == 3 && IS_SURROGATE_PAIR_LA(*pwc))
24026c95142eSMartin Matuska 		return (-3);
24036c95142eSMartin Matuska 	return (cnt);
24046c95142eSMartin Matuska }
24056c95142eSMartin Matuska 
24066c95142eSMartin Matuska static inline uint32_t
24076c95142eSMartin Matuska combine_surrogate_pair(uint32_t uc, uint32_t uc2)
24086c95142eSMartin Matuska {
24096c95142eSMartin Matuska 	uc -= 0xD800;
24106c95142eSMartin Matuska 	uc *= 0x400;
24116c95142eSMartin Matuska 	uc += uc2 - 0xDC00;
24126c95142eSMartin Matuska 	uc += 0x10000;
24136c95142eSMartin Matuska 	return (uc);
24146c95142eSMartin Matuska }
24156c95142eSMartin Matuska 
24166c95142eSMartin Matuska /*
24176c95142eSMartin Matuska  * Convert a single UTF-8/CESU-8 sequence to a Unicode code point in
24186c95142eSMartin Matuska  * removing surrogate pairs.
24196c95142eSMartin Matuska  *
24206c95142eSMartin Matuska  * CESU-8: The Compatibility Encoding Scheme for UTF-16.
24216c95142eSMartin Matuska  *
24226c95142eSMartin Matuska  * Usually return used bytes, return used byte in negative value when
24236c95142eSMartin Matuska  * a unicode character is replaced with U+FFFD.
24246c95142eSMartin Matuska  */
24256c95142eSMartin Matuska static int
24266c95142eSMartin Matuska cesu8_to_unicode(uint32_t *pwc, const char *s, size_t n)
24276c95142eSMartin Matuska {
2428acc60b03SMartin Matuska 	uint32_t wc = 0;
24296c95142eSMartin Matuska 	int cnt;
24306c95142eSMartin Matuska 
24316c95142eSMartin Matuska 	cnt = _utf8_to_unicode(&wc, s, n);
24326c95142eSMartin Matuska 	if (cnt == 3 && IS_HIGH_SURROGATE_LA(wc)) {
2433acc60b03SMartin Matuska 		uint32_t wc2 = 0;
24346c95142eSMartin Matuska 		if (n - 3 < 3) {
24356c95142eSMartin Matuska 			/* Invalid byte sequence. */
24366c95142eSMartin Matuska 			goto invalid_sequence;
24376c95142eSMartin Matuska 		}
24386c95142eSMartin Matuska 		cnt = _utf8_to_unicode(&wc2, s+3, n-3);
24396c95142eSMartin Matuska 		if (cnt != 3 || !IS_LOW_SURROGATE_LA(wc2)) {
24406c95142eSMartin Matuska 			/* Invalid byte sequence. */
24416c95142eSMartin Matuska 			goto invalid_sequence;
24426c95142eSMartin Matuska 		}
24436c95142eSMartin Matuska 		wc = combine_surrogate_pair(wc, wc2);
24446c95142eSMartin Matuska 		cnt = 6;
24456c95142eSMartin Matuska 	} else if (cnt == 3 && IS_LOW_SURROGATE_LA(wc)) {
24466c95142eSMartin Matuska 		/* Invalid byte sequence. */
24476c95142eSMartin Matuska 		goto invalid_sequence;
24486c95142eSMartin Matuska 	}
24496c95142eSMartin Matuska 	*pwc = wc;
24506c95142eSMartin Matuska 	return (cnt);
24516c95142eSMartin Matuska invalid_sequence:
24526c95142eSMartin Matuska 	*pwc = UNICODE_R_CHAR;/* set the Replacement Character instead. */
24536c95142eSMartin Matuska 	if (cnt > 0)
24546c95142eSMartin Matuska 		cnt *= -1;
24556c95142eSMartin Matuska 	return (cnt);
24566c95142eSMartin Matuska }
24576c95142eSMartin Matuska 
24586c95142eSMartin Matuska /*
24596c95142eSMartin Matuska  * Convert a Unicode code point to a single UTF-8 sequence.
24606c95142eSMartin Matuska  *
2461a2e802b7SMartin Matuska  * NOTE:This function does not check if the Unicode is legal or not.
24626c95142eSMartin Matuska  * Please you definitely check it before calling this.
24636c95142eSMartin Matuska  */
24646c95142eSMartin Matuska static size_t
24656c95142eSMartin Matuska unicode_to_utf8(char *p, size_t remaining, uint32_t uc)
24666c95142eSMartin Matuska {
24676c95142eSMartin Matuska 	char *_p = p;
24686c95142eSMartin Matuska 
2469cdf63a70SMartin Matuska 	/* Invalid Unicode char maps to Replacement character */
2470cdf63a70SMartin Matuska 	if (uc > UNICODE_MAX)
2471cdf63a70SMartin Matuska 		uc = UNICODE_R_CHAR;
24726c95142eSMartin Matuska 	/* Translate code point to UTF8 */
24736c95142eSMartin Matuska 	if (uc <= 0x7f) {
24746c95142eSMartin Matuska 		if (remaining == 0)
24756c95142eSMartin Matuska 			return (0);
24766c95142eSMartin Matuska 		*p++ = (char)uc;
24776c95142eSMartin Matuska 	} else if (uc <= 0x7ff) {
24786c95142eSMartin Matuska 		if (remaining < 2)
24796c95142eSMartin Matuska 			return (0);
24806c95142eSMartin Matuska 		*p++ = 0xc0 | ((uc >> 6) & 0x1f);
24816c95142eSMartin Matuska 		*p++ = 0x80 | (uc & 0x3f);
24826c95142eSMartin Matuska 	} else if (uc <= 0xffff) {
24836c95142eSMartin Matuska 		if (remaining < 3)
24846c95142eSMartin Matuska 			return (0);
24856c95142eSMartin Matuska 		*p++ = 0xe0 | ((uc >> 12) & 0x0f);
24866c95142eSMartin Matuska 		*p++ = 0x80 | ((uc >> 6) & 0x3f);
24876c95142eSMartin Matuska 		*p++ = 0x80 | (uc & 0x3f);
2488cdf63a70SMartin Matuska 	} else {
24896c95142eSMartin Matuska 		if (remaining < 4)
24906c95142eSMartin Matuska 			return (0);
24916c95142eSMartin Matuska 		*p++ = 0xf0 | ((uc >> 18) & 0x07);
24926c95142eSMartin Matuska 		*p++ = 0x80 | ((uc >> 12) & 0x3f);
24936c95142eSMartin Matuska 		*p++ = 0x80 | ((uc >> 6) & 0x3f);
24946c95142eSMartin Matuska 		*p++ = 0x80 | (uc & 0x3f);
24956c95142eSMartin Matuska 	}
24966c95142eSMartin Matuska 	return (p - _p);
24976c95142eSMartin Matuska }
24986c95142eSMartin Matuska 
24996c95142eSMartin Matuska static int
25006c95142eSMartin Matuska utf16be_to_unicode(uint32_t *pwc, const char *s, size_t n)
25016c95142eSMartin Matuska {
25026c95142eSMartin Matuska 	return (utf16_to_unicode(pwc, s, n, 1));
25036c95142eSMartin Matuska }
25046c95142eSMartin Matuska 
25056c95142eSMartin Matuska static int
25066c95142eSMartin Matuska utf16le_to_unicode(uint32_t *pwc, const char *s, size_t n)
25076c95142eSMartin Matuska {
25086c95142eSMartin Matuska 	return (utf16_to_unicode(pwc, s, n, 0));
25096c95142eSMartin Matuska }
25106c95142eSMartin Matuska 
25116c95142eSMartin Matuska static int
25126c95142eSMartin Matuska utf16_to_unicode(uint32_t *pwc, const char *s, size_t n, int be)
25136c95142eSMartin Matuska {
25146c95142eSMartin Matuska 	const char *utf16 = s;
25156c95142eSMartin Matuska 	unsigned uc;
25166c95142eSMartin Matuska 
25176c95142eSMartin Matuska 	if (n == 0)
25186c95142eSMartin Matuska 		return (0);
25196c95142eSMartin Matuska 	if (n == 1) {
25206c95142eSMartin Matuska 		/* set the Replacement Character instead. */
25216c95142eSMartin Matuska 		*pwc = UNICODE_R_CHAR;
25226c95142eSMartin Matuska 		return (-1);
25236c95142eSMartin Matuska 	}
25246c95142eSMartin Matuska 
25256c95142eSMartin Matuska 	if (be)
25266c95142eSMartin Matuska 		uc = archive_be16dec(utf16);
25276c95142eSMartin Matuska 	else
25286c95142eSMartin Matuska 		uc = archive_le16dec(utf16);
25296c95142eSMartin Matuska 	utf16 += 2;
25306c95142eSMartin Matuska 
25316c95142eSMartin Matuska 	/* If this is a surrogate pair, assemble the full code point.*/
25326c95142eSMartin Matuska 	if (IS_HIGH_SURROGATE_LA(uc)) {
25336c95142eSMartin Matuska 		unsigned uc2;
25346c95142eSMartin Matuska 
25356c95142eSMartin Matuska 		if (n >= 4) {
25366c95142eSMartin Matuska 			if (be)
25376c95142eSMartin Matuska 				uc2 = archive_be16dec(utf16);
25386c95142eSMartin Matuska 			else
25396c95142eSMartin Matuska 				uc2 = archive_le16dec(utf16);
25406c95142eSMartin Matuska 		} else
25416c95142eSMartin Matuska 			uc2 = 0;
25426c95142eSMartin Matuska 		if (IS_LOW_SURROGATE_LA(uc2)) {
25436c95142eSMartin Matuska 			uc = combine_surrogate_pair(uc, uc2);
25446c95142eSMartin Matuska 			utf16 += 2;
25456c95142eSMartin Matuska 		} else {
25466c95142eSMartin Matuska 	 		/* Undescribed code point should be U+FFFD
25476c95142eSMartin Matuska 		 	* (replacement character). */
25486c95142eSMartin Matuska 			*pwc = UNICODE_R_CHAR;
25496c95142eSMartin Matuska 			return (-2);
25506c95142eSMartin Matuska 		}
25516c95142eSMartin Matuska 	}
25526c95142eSMartin Matuska 
25536c95142eSMartin Matuska 	/*
25546c95142eSMartin Matuska 	 * Surrogate pair values(0xd800 through 0xdfff) are only
25556c95142eSMartin Matuska 	 * used by UTF-16, so, after above culculation, the code
25566c95142eSMartin Matuska 	 * must not be surrogate values, and Unicode has no codes
2557a2e802b7SMartin Matuska 	 * larger than 0x10ffff. Thus, those are not legal Unicode
25586c95142eSMartin Matuska 	 * values.
25596c95142eSMartin Matuska 	 */
25606c95142eSMartin Matuska 	if (IS_SURROGATE_PAIR_LA(uc) || uc > UNICODE_MAX) {
25616c95142eSMartin Matuska 	 	/* Undescribed code point should be U+FFFD
25626c95142eSMartin Matuska 	 	* (replacement character). */
25636c95142eSMartin Matuska 		*pwc = UNICODE_R_CHAR;
25646c95142eSMartin Matuska 		return (((int)(utf16 - s)) * -1);
25656c95142eSMartin Matuska 	}
25666c95142eSMartin Matuska 	*pwc = uc;
25676c95142eSMartin Matuska 	return ((int)(utf16 - s));
25686c95142eSMartin Matuska }
25696c95142eSMartin Matuska 
25706c95142eSMartin Matuska static size_t
25716c95142eSMartin Matuska unicode_to_utf16be(char *p, size_t remaining, uint32_t uc)
25726c95142eSMartin Matuska {
25736c95142eSMartin Matuska 	char *utf16 = p;
25746c95142eSMartin Matuska 
25756c95142eSMartin Matuska 	if (uc > 0xffff) {
25766c95142eSMartin Matuska 		/* We have a code point that won't fit into a
25776c95142eSMartin Matuska 		 * wchar_t; convert it to a surrogate pair. */
25786c95142eSMartin Matuska 		if (remaining < 4)
25796c95142eSMartin Matuska 			return (0);
25806c95142eSMartin Matuska 		uc -= 0x10000;
25816c95142eSMartin Matuska 		archive_be16enc(utf16, ((uc >> 10) & 0x3ff) + 0xD800);
25826c95142eSMartin Matuska 		archive_be16enc(utf16+2, (uc & 0x3ff) + 0xDC00);
25836c95142eSMartin Matuska 		return (4);
25846c95142eSMartin Matuska 	} else {
25856c95142eSMartin Matuska 		if (remaining < 2)
25866c95142eSMartin Matuska 			return (0);
25876c95142eSMartin Matuska 		archive_be16enc(utf16, uc);
25886c95142eSMartin Matuska 		return (2);
25896c95142eSMartin Matuska 	}
25906c95142eSMartin Matuska }
25916c95142eSMartin Matuska 
25926c95142eSMartin Matuska static size_t
25936c95142eSMartin Matuska unicode_to_utf16le(char *p, size_t remaining, uint32_t uc)
25946c95142eSMartin Matuska {
25956c95142eSMartin Matuska 	char *utf16 = p;
25966c95142eSMartin Matuska 
25976c95142eSMartin Matuska 	if (uc > 0xffff) {
25986c95142eSMartin Matuska 		/* We have a code point that won't fit into a
25996c95142eSMartin Matuska 		 * wchar_t; convert it to a surrogate pair. */
26006c95142eSMartin Matuska 		if (remaining < 4)
26016c95142eSMartin Matuska 			return (0);
26026c95142eSMartin Matuska 		uc -= 0x10000;
26036c95142eSMartin Matuska 		archive_le16enc(utf16, ((uc >> 10) & 0x3ff) + 0xD800);
26046c95142eSMartin Matuska 		archive_le16enc(utf16+2, (uc & 0x3ff) + 0xDC00);
26056c95142eSMartin Matuska 		return (4);
26066c95142eSMartin Matuska 	} else {
26076c95142eSMartin Matuska 		if (remaining < 2)
26086c95142eSMartin Matuska 			return (0);
26096c95142eSMartin Matuska 		archive_le16enc(utf16, uc);
26106c95142eSMartin Matuska 		return (2);
26116c95142eSMartin Matuska 	}
26126c95142eSMartin Matuska }
26136c95142eSMartin Matuska 
26146c95142eSMartin Matuska /*
26156c95142eSMartin Matuska  * Copy UTF-8 string in checking surrogate pair.
26166c95142eSMartin Matuska  * If any surrogate pair are found, it would be canonicalized.
26176c95142eSMartin Matuska  */
26186c95142eSMartin Matuska static int
2619fd082e96SMartin Matuska strncat_from_utf8_to_utf8(struct archive_string *as, const void *_p,
2620fd082e96SMartin Matuska     size_t len, struct archive_string_conv *sc)
26216c95142eSMartin Matuska {
26226c95142eSMartin Matuska 	const char *s;
26236c95142eSMartin Matuska 	char *p, *endp;
26246c95142eSMartin Matuska 	int n, ret = 0;
26256c95142eSMartin Matuska 
26266c95142eSMartin Matuska 	(void)sc; /* UNUSED */
26276c95142eSMartin Matuska 
26286c95142eSMartin Matuska 	if (archive_string_ensure(as, as->length + len + 1) == NULL)
26296c95142eSMartin Matuska 		return (-1);
26306c95142eSMartin Matuska 
26316c95142eSMartin Matuska 	s = (const char *)_p;
26326c95142eSMartin Matuska 	p = as->s + as->length;
26336c95142eSMartin Matuska 	endp = as->s + as->buffer_length -1;
26346c95142eSMartin Matuska 	do {
26356c95142eSMartin Matuska 		uint32_t uc;
26366c95142eSMartin Matuska 		const char *ss = s;
26376c95142eSMartin Matuska 		size_t w;
26386c95142eSMartin Matuska 
26396c95142eSMartin Matuska 		/*
26406c95142eSMartin Matuska 		 * Forward byte sequence until a conversion of that is needed.
26416c95142eSMartin Matuska 		 */
26426c95142eSMartin Matuska 		while ((n = utf8_to_unicode(&uc, s, len)) > 0) {
26436c95142eSMartin Matuska 			s += n;
26446c95142eSMartin Matuska 			len -= n;
26456c95142eSMartin Matuska 		}
26466c95142eSMartin Matuska 		if (ss < s) {
26476c95142eSMartin Matuska 			if (p + (s - ss) > endp) {
26486c95142eSMartin Matuska 				as->length = p - as->s;
26496c95142eSMartin Matuska 				if (archive_string_ensure(as,
26506c95142eSMartin Matuska 				    as->buffer_length + len + 1) == NULL)
26516c95142eSMartin Matuska 					return (-1);
26526c95142eSMartin Matuska 				p = as->s + as->length;
26536c95142eSMartin Matuska 				endp = as->s + as->buffer_length -1;
26546c95142eSMartin Matuska 			}
26556c95142eSMartin Matuska 
26566c95142eSMartin Matuska 			memcpy(p, ss, s - ss);
26576c95142eSMartin Matuska 			p += s - ss;
26586c95142eSMartin Matuska 		}
26596c95142eSMartin Matuska 
26606c95142eSMartin Matuska 		/*
26616c95142eSMartin Matuska 		 * If n is negative, current byte sequence needs a replacement.
26626c95142eSMartin Matuska 		 */
26636c95142eSMartin Matuska 		if (n < 0) {
26646c95142eSMartin Matuska 			if (n == -3 && IS_SURROGATE_PAIR_LA(uc)) {
26656c95142eSMartin Matuska 				/* Current byte sequence may be CESU-8. */
26666c95142eSMartin Matuska 				n = cesu8_to_unicode(&uc, s, len);
26676c95142eSMartin Matuska 			}
26686c95142eSMartin Matuska 			if (n < 0) {
26696c95142eSMartin Matuska 				ret = -1;
26706c95142eSMartin Matuska 				n *= -1;/* Use a replaced unicode character. */
26716c95142eSMartin Matuska 			}
26726c95142eSMartin Matuska 
26736c95142eSMartin Matuska 			/* Rebuild UTF-8 byte sequence. */
26746c95142eSMartin Matuska 			while ((w = unicode_to_utf8(p, endp - p, uc)) == 0) {
26756c95142eSMartin Matuska 				as->length = p - as->s;
26766c95142eSMartin Matuska 				if (archive_string_ensure(as,
26776c95142eSMartin Matuska 				    as->buffer_length + len + 1) == NULL)
26786c95142eSMartin Matuska 					return (-1);
26796c95142eSMartin Matuska 				p = as->s + as->length;
26806c95142eSMartin Matuska 				endp = as->s + as->buffer_length -1;
26816c95142eSMartin Matuska 			}
26826c95142eSMartin Matuska 			p += w;
26836c95142eSMartin Matuska 			s += n;
26846c95142eSMartin Matuska 			len -= n;
26856c95142eSMartin Matuska 		}
26866c95142eSMartin Matuska 	} while (n > 0);
26876c95142eSMartin Matuska 	as->length = p - as->s;
26886c95142eSMartin Matuska 	as->s[as->length] = '\0';
26896c95142eSMartin Matuska 	return (ret);
26906c95142eSMartin Matuska }
26916c95142eSMartin Matuska 
26926c95142eSMartin Matuska static int
26936c95142eSMartin Matuska archive_string_append_unicode(struct archive_string *as, const void *_p,
26946c95142eSMartin Matuska     size_t len, struct archive_string_conv *sc)
26956c95142eSMartin Matuska {
26966c95142eSMartin Matuska 	const char *s;
26976c95142eSMartin Matuska 	char *p, *endp;
26986c95142eSMartin Matuska 	uint32_t uc;
26996c95142eSMartin Matuska 	size_t w;
27006c95142eSMartin Matuska 	int n, ret = 0, ts, tm;
27016c95142eSMartin Matuska 	int (*parse)(uint32_t *, const char *, size_t);
27026c95142eSMartin Matuska 	size_t (*unparse)(char *, size_t, uint32_t);
27036c95142eSMartin Matuska 
27046c95142eSMartin Matuska 	if (sc->flag & SCONV_TO_UTF16BE) {
27056c95142eSMartin Matuska 		unparse = unicode_to_utf16be;
27066c95142eSMartin Matuska 		ts = 2;
27076c95142eSMartin Matuska 	} else if (sc->flag & SCONV_TO_UTF16LE) {
27086c95142eSMartin Matuska 		unparse = unicode_to_utf16le;
27096c95142eSMartin Matuska 		ts = 2;
27106c95142eSMartin Matuska 	} else if (sc->flag & SCONV_TO_UTF8) {
27116c95142eSMartin Matuska 		unparse = unicode_to_utf8;
27126c95142eSMartin Matuska 		ts = 1;
27136c95142eSMartin Matuska 	} else {
27146c95142eSMartin Matuska 		/*
27156c95142eSMartin Matuska 		 * This case is going to be converted to another
27166c95142eSMartin Matuska 		 * character-set through iconv.
27176c95142eSMartin Matuska 		 */
27186c95142eSMartin Matuska 		if (sc->flag & SCONV_FROM_UTF16BE) {
27196c95142eSMartin Matuska 			unparse = unicode_to_utf16be;
27206c95142eSMartin Matuska 			ts = 2;
27216c95142eSMartin Matuska 		} else if (sc->flag & SCONV_FROM_UTF16LE) {
27226c95142eSMartin Matuska 			unparse = unicode_to_utf16le;
27236c95142eSMartin Matuska 			ts = 2;
27246c95142eSMartin Matuska 		} else {
27256c95142eSMartin Matuska 			unparse = unicode_to_utf8;
27266c95142eSMartin Matuska 			ts = 1;
27276c95142eSMartin Matuska 		}
27286c95142eSMartin Matuska 	}
27296c95142eSMartin Matuska 
27306c95142eSMartin Matuska 	if (sc->flag & SCONV_FROM_UTF16BE) {
27316c95142eSMartin Matuska 		parse = utf16be_to_unicode;
27326c95142eSMartin Matuska 		tm = 1;
27336c95142eSMartin Matuska 	} else if (sc->flag & SCONV_FROM_UTF16LE) {
27346c95142eSMartin Matuska 		parse = utf16le_to_unicode;
27356c95142eSMartin Matuska 		tm = 1;
27366c95142eSMartin Matuska 	} else {
27376c95142eSMartin Matuska 		parse = cesu8_to_unicode;
27386c95142eSMartin Matuska 		tm = ts;
27396c95142eSMartin Matuska 	}
27406c95142eSMartin Matuska 
27416c95142eSMartin Matuska 	if (archive_string_ensure(as, as->length + len * tm + ts) == NULL)
27426c95142eSMartin Matuska 		return (-1);
27436c95142eSMartin Matuska 
27446c95142eSMartin Matuska 	s = (const char *)_p;
27456c95142eSMartin Matuska 	p = as->s + as->length;
27466c95142eSMartin Matuska 	endp = as->s + as->buffer_length - ts;
27476c95142eSMartin Matuska 	while ((n = parse(&uc, s, len)) != 0) {
27486c95142eSMartin Matuska 		if (n < 0) {
27496c95142eSMartin Matuska 			/* Use a replaced unicode character. */
27506c95142eSMartin Matuska 			n *= -1;
27516c95142eSMartin Matuska 			ret = -1;
27526c95142eSMartin Matuska 		}
27536c95142eSMartin Matuska 		s += n;
27546c95142eSMartin Matuska 		len -= n;
27556c95142eSMartin Matuska 		while ((w = unparse(p, endp - p, uc)) == 0) {
27566c95142eSMartin Matuska 			/* There is not enough output buffer so
27576c95142eSMartin Matuska 			 * we have to expand it. */
27586c95142eSMartin Matuska 			as->length = p - as->s;
27596c95142eSMartin Matuska 			if (archive_string_ensure(as,
27606c95142eSMartin Matuska 			    as->buffer_length + len * tm + ts) == NULL)
27616c95142eSMartin Matuska 				return (-1);
27626c95142eSMartin Matuska 			p = as->s + as->length;
27636c95142eSMartin Matuska 			endp = as->s + as->buffer_length - ts;
27646c95142eSMartin Matuska 		}
27656c95142eSMartin Matuska 		p += w;
27666c95142eSMartin Matuska 	}
27676c95142eSMartin Matuska 	as->length = p - as->s;
27686c95142eSMartin Matuska 	as->s[as->length] = '\0';
27696c95142eSMartin Matuska 	if (ts == 2)
27706c95142eSMartin Matuska 		as->s[as->length+1] = '\0';
27716c95142eSMartin Matuska 	return (ret);
27726c95142eSMartin Matuska }
27736c95142eSMartin Matuska 
27746c95142eSMartin Matuska /*
27756c95142eSMartin Matuska  * Following Constants for Hangul compositions this information comes from
27766c95142eSMartin Matuska  * Unicode Standard Annex #15  http://unicode.org/reports/tr15/
27776c95142eSMartin Matuska  */
27786c95142eSMartin Matuska #define HC_SBASE	0xAC00
27796c95142eSMartin Matuska #define HC_LBASE	0x1100
27806c95142eSMartin Matuska #define HC_VBASE	0x1161
27816c95142eSMartin Matuska #define HC_TBASE	0x11A7
27826c95142eSMartin Matuska #define HC_LCOUNT	19
27836c95142eSMartin Matuska #define HC_VCOUNT	21
27846c95142eSMartin Matuska #define HC_TCOUNT	28
27856c95142eSMartin Matuska #define HC_NCOUNT	(HC_VCOUNT * HC_TCOUNT)
27866c95142eSMartin Matuska #define HC_SCOUNT	(HC_LCOUNT * HC_NCOUNT)
27876c95142eSMartin Matuska 
27886c95142eSMartin Matuska static uint32_t
27896c95142eSMartin Matuska get_nfc(uint32_t uc, uint32_t uc2)
27906c95142eSMartin Matuska {
27916c95142eSMartin Matuska 	int t, b;
27926c95142eSMartin Matuska 
27936c95142eSMartin Matuska 	t = 0;
27946c95142eSMartin Matuska 	b = sizeof(u_composition_table)/sizeof(u_composition_table[0]) -1;
27956c95142eSMartin Matuska 	while (b >= t) {
27966c95142eSMartin Matuska 		int m = (t + b) / 2;
27976c95142eSMartin Matuska 		if (u_composition_table[m].cp1 < uc)
27986c95142eSMartin Matuska 			t = m + 1;
27996c95142eSMartin Matuska 		else if (u_composition_table[m].cp1 > uc)
28006c95142eSMartin Matuska 			b = m - 1;
28016c95142eSMartin Matuska 		else if (u_composition_table[m].cp2 < uc2)
28026c95142eSMartin Matuska 			t = m + 1;
28036c95142eSMartin Matuska 		else if (u_composition_table[m].cp2 > uc2)
28046c95142eSMartin Matuska 			b = m - 1;
28056c95142eSMartin Matuska 		else
28066c95142eSMartin Matuska 			return (u_composition_table[m].nfc);
28076c95142eSMartin Matuska 	}
28086c95142eSMartin Matuska 	return (0);
28096c95142eSMartin Matuska }
28106c95142eSMartin Matuska 
28116c95142eSMartin Matuska #define FDC_MAX 10	/* The maximum number of Following Decomposable
28126c95142eSMartin Matuska 			 * Characters. */
28136c95142eSMartin Matuska 
28146c95142eSMartin Matuska /*
28156c95142eSMartin Matuska  * Update first code point.
28166c95142eSMartin Matuska  */
28176c95142eSMartin Matuska #define UPDATE_UC(new_uc)	do {		\
28186c95142eSMartin Matuska 	uc = new_uc;				\
28196c95142eSMartin Matuska 	ucptr = NULL;				\
28206c95142eSMartin Matuska } while (0)
28216c95142eSMartin Matuska 
28226c95142eSMartin Matuska /*
28236c95142eSMartin Matuska  * Replace first code point with second code point.
28246c95142eSMartin Matuska  */
28256c95142eSMartin Matuska #define REPLACE_UC_WITH_UC2() do {		\
28266c95142eSMartin Matuska 	uc = uc2;				\
28276c95142eSMartin Matuska 	ucptr = uc2ptr;				\
28286c95142eSMartin Matuska 	n = n2;					\
28296c95142eSMartin Matuska } while (0)
28306c95142eSMartin Matuska 
28316c95142eSMartin Matuska #define EXPAND_BUFFER() do {			\
28326c95142eSMartin Matuska 	as->length = p - as->s;			\
28336c95142eSMartin Matuska 	if (archive_string_ensure(as,		\
28346c95142eSMartin Matuska 	    as->buffer_length + len * tm + ts) == NULL)\
28356c95142eSMartin Matuska 		return (-1);			\
28366c95142eSMartin Matuska 	p = as->s + as->length;			\
28376c95142eSMartin Matuska 	endp = as->s + as->buffer_length - ts;	\
28386c95142eSMartin Matuska } while (0)
28396c95142eSMartin Matuska 
28406c95142eSMartin Matuska #define UNPARSE(p, endp, uc)	do {		\
28416c95142eSMartin Matuska 	while ((w = unparse(p, (endp) - (p), uc)) == 0) {\
28426c95142eSMartin Matuska 		EXPAND_BUFFER();		\
28436c95142eSMartin Matuska 	}					\
28446c95142eSMartin Matuska 	p += w;					\
28456c95142eSMartin Matuska } while (0)
28466c95142eSMartin Matuska 
28476c95142eSMartin Matuska /*
28486c95142eSMartin Matuska  * Write first code point.
28496c95142eSMartin Matuska  * If the code point has not be changed from its original code,
28506c95142eSMartin Matuska  * this just copies it from its original buffer pointer.
28516c95142eSMartin Matuska  * If not, this converts it to UTF-8 byte sequence and copies it.
28526c95142eSMartin Matuska  */
28536c95142eSMartin Matuska #define WRITE_UC()	do {			\
28546c95142eSMartin Matuska 	if (ucptr) {				\
28556c95142eSMartin Matuska 		if (p + n > endp)		\
28566c95142eSMartin Matuska 			EXPAND_BUFFER();	\
28576c95142eSMartin Matuska 		switch (n) {			\
28586c95142eSMartin Matuska 		case 4:				\
28596c95142eSMartin Matuska 			*p++ = *ucptr++;	\
28606c95142eSMartin Matuska 			/* FALL THROUGH */	\
28616c95142eSMartin Matuska 		case 3:				\
28626c95142eSMartin Matuska 			*p++ = *ucptr++;	\
28636c95142eSMartin Matuska 			/* FALL THROUGH */	\
28646c95142eSMartin Matuska 		case 2:				\
28656c95142eSMartin Matuska 			*p++ = *ucptr++;	\
28666c95142eSMartin Matuska 			/* FALL THROUGH */	\
28676c95142eSMartin Matuska 		case 1:				\
28686c95142eSMartin Matuska 			*p++ = *ucptr;		\
28696c95142eSMartin Matuska 			break;			\
28706c95142eSMartin Matuska 		}				\
28716c95142eSMartin Matuska 		ucptr = NULL;			\
28726c95142eSMartin Matuska 	} else {				\
28736c95142eSMartin Matuska 		UNPARSE(p, endp, uc);		\
28746c95142eSMartin Matuska 	}					\
28756c95142eSMartin Matuska } while (0)
28766c95142eSMartin Matuska 
28776c95142eSMartin Matuska /*
28786c95142eSMartin Matuska  * Collect following decomposable code points.
28796c95142eSMartin Matuska  */
28806c95142eSMartin Matuska #define COLLECT_CPS(start)	do {		\
28816c95142eSMartin Matuska 	int _i;					\
28826c95142eSMartin Matuska 	for (_i = start; _i < FDC_MAX ; _i++) {	\
28836c95142eSMartin Matuska 		nx = parse(&ucx[_i], s, len);	\
28846c95142eSMartin Matuska 		if (nx <= 0)			\
28856c95142eSMartin Matuska 			break;			\
28866c95142eSMartin Matuska 		cx = CCC(ucx[_i]);		\
28876c95142eSMartin Matuska 		if (cl >= cx && cl != 228 && cx != 228)\
28886c95142eSMartin Matuska 			break;			\
28896c95142eSMartin Matuska 		s += nx;			\
28906c95142eSMartin Matuska 		len -= nx;			\
28916c95142eSMartin Matuska 		cl = cx;			\
28926c95142eSMartin Matuska 		ccx[_i] = cx;			\
28936c95142eSMartin Matuska 	}					\
28946c95142eSMartin Matuska 	if (_i >= FDC_MAX) {			\
28956c95142eSMartin Matuska 		ret = -1;			\
28966c95142eSMartin Matuska 		ucx_size = FDC_MAX;		\
28976c95142eSMartin Matuska 	} else					\
28986c95142eSMartin Matuska 		ucx_size = _i;			\
28996c95142eSMartin Matuska } while (0)
29006c95142eSMartin Matuska 
29016c95142eSMartin Matuska /*
29026c95142eSMartin Matuska  * Normalize UTF-8/UTF-16BE characters to Form C and copy the result.
29036c95142eSMartin Matuska  *
29046c95142eSMartin Matuska  * TODO: Convert composition exclusions,which are never converted
29056c95142eSMartin Matuska  * from NFC,NFD,NFKC and NFKD, to Form C.
29066c95142eSMartin Matuska  */
29076c95142eSMartin Matuska static int
29086c95142eSMartin Matuska archive_string_normalize_C(struct archive_string *as, const void *_p,
29096c95142eSMartin Matuska     size_t len, struct archive_string_conv *sc)
29106c95142eSMartin Matuska {
29116c95142eSMartin Matuska 	const char *s = (const char *)_p;
29126c95142eSMartin Matuska 	char *p, *endp;
29136c95142eSMartin Matuska 	uint32_t uc, uc2;
29146c95142eSMartin Matuska 	size_t w;
29156c95142eSMartin Matuska 	int always_replace, n, n2, ret = 0, spair, ts, tm;
29166c95142eSMartin Matuska 	int (*parse)(uint32_t *, const char *, size_t);
29176c95142eSMartin Matuska 	size_t (*unparse)(char *, size_t, uint32_t);
29186c95142eSMartin Matuska 
29196c95142eSMartin Matuska 	always_replace = 1;
29206c95142eSMartin Matuska 	ts = 1;/* text size. */
29216c95142eSMartin Matuska 	if (sc->flag & SCONV_TO_UTF16BE) {
29226c95142eSMartin Matuska 		unparse = unicode_to_utf16be;
29236c95142eSMartin Matuska 		ts = 2;
29246c95142eSMartin Matuska 		if (sc->flag & SCONV_FROM_UTF16BE)
29256c95142eSMartin Matuska 			always_replace = 0;
29266c95142eSMartin Matuska 	} else if (sc->flag & SCONV_TO_UTF16LE) {
29276c95142eSMartin Matuska 		unparse = unicode_to_utf16le;
29286c95142eSMartin Matuska 		ts = 2;
29296c95142eSMartin Matuska 		if (sc->flag & SCONV_FROM_UTF16LE)
29306c95142eSMartin Matuska 			always_replace = 0;
29316c95142eSMartin Matuska 	} else if (sc->flag & SCONV_TO_UTF8) {
29326c95142eSMartin Matuska 		unparse = unicode_to_utf8;
29336c95142eSMartin Matuska 		if (sc->flag & SCONV_FROM_UTF8)
29346c95142eSMartin Matuska 			always_replace = 0;
29356c95142eSMartin Matuska 	} else {
29366c95142eSMartin Matuska 		/*
29376c95142eSMartin Matuska 		 * This case is going to be converted to another
29386c95142eSMartin Matuska 		 * character-set through iconv.
29396c95142eSMartin Matuska 		 */
29406c95142eSMartin Matuska 		always_replace = 0;
29416c95142eSMartin Matuska 		if (sc->flag & SCONV_FROM_UTF16BE) {
29426c95142eSMartin Matuska 			unparse = unicode_to_utf16be;
29436c95142eSMartin Matuska 			ts = 2;
29446c95142eSMartin Matuska 		} else if (sc->flag & SCONV_FROM_UTF16LE) {
29456c95142eSMartin Matuska 			unparse = unicode_to_utf16le;
29466c95142eSMartin Matuska 			ts = 2;
29476c95142eSMartin Matuska 		} else {
29486c95142eSMartin Matuska 			unparse = unicode_to_utf8;
29496c95142eSMartin Matuska 		}
29506c95142eSMartin Matuska 	}
29516c95142eSMartin Matuska 
29526c95142eSMartin Matuska 	if (sc->flag & SCONV_FROM_UTF16BE) {
29536c95142eSMartin Matuska 		parse = utf16be_to_unicode;
29546c95142eSMartin Matuska 		tm = 1;
29556c95142eSMartin Matuska 		spair = 4;/* surrogate pair size in UTF-16. */
29566c95142eSMartin Matuska 	} else if (sc->flag & SCONV_FROM_UTF16LE) {
29576c95142eSMartin Matuska 		parse = utf16le_to_unicode;
29586c95142eSMartin Matuska 		tm = 1;
29596c95142eSMartin Matuska 		spair = 4;/* surrogate pair size in UTF-16. */
29606c95142eSMartin Matuska 	} else {
29616c95142eSMartin Matuska 		parse = cesu8_to_unicode;
29626c95142eSMartin Matuska 		tm = ts;
29636c95142eSMartin Matuska 		spair = 6;/* surrogate pair size in UTF-8. */
29646c95142eSMartin Matuska 	}
29656c95142eSMartin Matuska 
29666c95142eSMartin Matuska 	if (archive_string_ensure(as, as->length + len * tm + ts) == NULL)
29676c95142eSMartin Matuska 		return (-1);
29686c95142eSMartin Matuska 
29696c95142eSMartin Matuska 	p = as->s + as->length;
29706c95142eSMartin Matuska 	endp = as->s + as->buffer_length - ts;
29716c95142eSMartin Matuska 	while ((n = parse(&uc, s, len)) != 0) {
29726c95142eSMartin Matuska 		const char *ucptr, *uc2ptr;
29736c95142eSMartin Matuska 
29746c95142eSMartin Matuska 		if (n < 0) {
29756c95142eSMartin Matuska 			/* Use a replaced unicode character. */
29766c95142eSMartin Matuska 			UNPARSE(p, endp, uc);
29776c95142eSMartin Matuska 			s += n*-1;
29786c95142eSMartin Matuska 			len -= n*-1;
29796c95142eSMartin Matuska 			ret = -1;
29806c95142eSMartin Matuska 			continue;
29816c95142eSMartin Matuska 		} else if (n == spair || always_replace)
29826c95142eSMartin Matuska 			/* uc is converted from a surrogate pair.
29836c95142eSMartin Matuska 			 * this should be treated as a changed code. */
29846c95142eSMartin Matuska 			ucptr = NULL;
29856c95142eSMartin Matuska 		else
29866c95142eSMartin Matuska 			ucptr = s;
29876c95142eSMartin Matuska 		s += n;
29886c95142eSMartin Matuska 		len -= n;
29896c95142eSMartin Matuska 
29906c95142eSMartin Matuska 		/* Read second code point. */
29916c95142eSMartin Matuska 		while ((n2 = parse(&uc2, s, len)) > 0) {
29926c95142eSMartin Matuska 			uint32_t ucx[FDC_MAX];
29936c95142eSMartin Matuska 			int ccx[FDC_MAX];
29946c95142eSMartin Matuska 			int cl, cx, i, nx, ucx_size;
29956c95142eSMartin Matuska 			int LIndex,SIndex;
29966c95142eSMartin Matuska 			uint32_t nfc;
29976c95142eSMartin Matuska 
29986c95142eSMartin Matuska 			if (n2 == spair || always_replace)
29996c95142eSMartin Matuska 				/* uc2 is converted from a surrogate pair.
30006c95142eSMartin Matuska 			 	 * this should be treated as a changed code. */
30016c95142eSMartin Matuska 				uc2ptr = NULL;
30026c95142eSMartin Matuska 			else
30036c95142eSMartin Matuska 				uc2ptr = s;
30046c95142eSMartin Matuska 			s += n2;
30056c95142eSMartin Matuska 			len -= n2;
30066c95142eSMartin Matuska 
30076c95142eSMartin Matuska 			/*
30086c95142eSMartin Matuska 			 * If current second code point is out of decomposable
30096c95142eSMartin Matuska 			 * code points, finding compositions is unneeded.
30106c95142eSMartin Matuska 			 */
30116c95142eSMartin Matuska 			if (!IS_DECOMPOSABLE_BLOCK(uc2)) {
30126c95142eSMartin Matuska 				WRITE_UC();
30136c95142eSMartin Matuska 				REPLACE_UC_WITH_UC2();
30146c95142eSMartin Matuska 				continue;
30156c95142eSMartin Matuska 			}
30166c95142eSMartin Matuska 
30176c95142eSMartin Matuska 			/*
30186c95142eSMartin Matuska 			 * Try to combine current code points.
30196c95142eSMartin Matuska 			 */
30206c95142eSMartin Matuska 			/*
30216c95142eSMartin Matuska 			 * We have to combine Hangul characters according to
30226c95142eSMartin Matuska 			 * http://uniicode.org/reports/tr15/#Hangul
30236c95142eSMartin Matuska 			 */
30246c95142eSMartin Matuska 			if (0 <= (LIndex = uc - HC_LBASE) &&
30256c95142eSMartin Matuska 			    LIndex < HC_LCOUNT) {
30266c95142eSMartin Matuska 				/*
30276c95142eSMartin Matuska 				 * Hangul Composition.
30286c95142eSMartin Matuska 				 * 1. Two current code points are L and V.
30296c95142eSMartin Matuska 				 */
30306c95142eSMartin Matuska 				int VIndex = uc2 - HC_VBASE;
30316c95142eSMartin Matuska 				if (0 <= VIndex && VIndex < HC_VCOUNT) {
30326c95142eSMartin Matuska 					/* Make syllable of form LV. */
30336c95142eSMartin Matuska 					UPDATE_UC(HC_SBASE +
30346c95142eSMartin Matuska 					    (LIndex * HC_VCOUNT + VIndex) *
30356c95142eSMartin Matuska 					     HC_TCOUNT);
30366c95142eSMartin Matuska 				} else {
30376c95142eSMartin Matuska 					WRITE_UC();
30386c95142eSMartin Matuska 					REPLACE_UC_WITH_UC2();
30396c95142eSMartin Matuska 				}
30406c95142eSMartin Matuska 				continue;
30416c95142eSMartin Matuska 			} else if (0 <= (SIndex = uc - HC_SBASE) &&
30426c95142eSMartin Matuska 			    SIndex < HC_SCOUNT && (SIndex % HC_TCOUNT) == 0) {
30436c95142eSMartin Matuska 				/*
30446c95142eSMartin Matuska 				 * Hangul Composition.
30456c95142eSMartin Matuska 				 * 2. Two current code points are LV and T.
30466c95142eSMartin Matuska 				 */
30476c95142eSMartin Matuska 				int TIndex = uc2 - HC_TBASE;
30486c95142eSMartin Matuska 				if (0 < TIndex && TIndex < HC_TCOUNT) {
30496c95142eSMartin Matuska 					/* Make syllable of form LVT. */
30506c95142eSMartin Matuska 					UPDATE_UC(uc + TIndex);
30516c95142eSMartin Matuska 				} else {
30526c95142eSMartin Matuska 					WRITE_UC();
30536c95142eSMartin Matuska 					REPLACE_UC_WITH_UC2();
30546c95142eSMartin Matuska 				}
30556c95142eSMartin Matuska 				continue;
30566c95142eSMartin Matuska 			} else if ((nfc = get_nfc(uc, uc2)) != 0) {
30576c95142eSMartin Matuska 				/* A composition to current code points
30586c95142eSMartin Matuska 				 * is found. */
30596c95142eSMartin Matuska 				UPDATE_UC(nfc);
30606c95142eSMartin Matuska 				continue;
30616c95142eSMartin Matuska 			} else if ((cl = CCC(uc2)) == 0) {
30626c95142eSMartin Matuska 				/* Clearly 'uc2' the second code point is not
30636c95142eSMartin Matuska 				 * a decomposable code. */
30646c95142eSMartin Matuska 				WRITE_UC();
30656c95142eSMartin Matuska 				REPLACE_UC_WITH_UC2();
30666c95142eSMartin Matuska 				continue;
30676c95142eSMartin Matuska 			}
30686c95142eSMartin Matuska 
30696c95142eSMartin Matuska 			/*
30706c95142eSMartin Matuska 			 * Collect following decomposable code points.
30716c95142eSMartin Matuska 			 */
30726c95142eSMartin Matuska 			cx = 0;
30736c95142eSMartin Matuska 			ucx[0] = uc2;
30746c95142eSMartin Matuska 			ccx[0] = cl;
30756c95142eSMartin Matuska 			COLLECT_CPS(1);
30766c95142eSMartin Matuska 
30776c95142eSMartin Matuska 			/*
30786c95142eSMartin Matuska 			 * Find a composed code in the collected code points.
30796c95142eSMartin Matuska 			 */
30806c95142eSMartin Matuska 			i = 1;
30816c95142eSMartin Matuska 			while (i < ucx_size) {
30826c95142eSMartin Matuska 				int j;
30836c95142eSMartin Matuska 
30846c95142eSMartin Matuska 				if ((nfc = get_nfc(uc, ucx[i])) == 0) {
30856c95142eSMartin Matuska 					i++;
30866c95142eSMartin Matuska 					continue;
30876c95142eSMartin Matuska 				}
30886c95142eSMartin Matuska 
30896c95142eSMartin Matuska 				/*
30906c95142eSMartin Matuska 				 * nfc is composed of uc and ucx[i].
30916c95142eSMartin Matuska 				 */
30926c95142eSMartin Matuska 				UPDATE_UC(nfc);
30936c95142eSMartin Matuska 
30946c95142eSMartin Matuska 				/*
30956c95142eSMartin Matuska 				 * Remove ucx[i] by shifting
30966c95142eSMartin Matuska 				 * following code points.
30976c95142eSMartin Matuska 				 */
30986c95142eSMartin Matuska 				for (j = i; j+1 < ucx_size; j++) {
30996c95142eSMartin Matuska 					ucx[j] = ucx[j+1];
31006c95142eSMartin Matuska 					ccx[j] = ccx[j+1];
31016c95142eSMartin Matuska 				}
31026c95142eSMartin Matuska 				ucx_size --;
31036c95142eSMartin Matuska 
31046c95142eSMartin Matuska 				/*
31056c95142eSMartin Matuska 				 * Collect following code points blocked
31066c95142eSMartin Matuska 				 * by ucx[i] the removed code point.
31076c95142eSMartin Matuska 				 */
31086c95142eSMartin Matuska 				if (ucx_size > 0 && i == ucx_size &&
31096c95142eSMartin Matuska 				    nx > 0 && cx == cl) {
31106c95142eSMartin Matuska 					cl =  ccx[ucx_size-1];
31116c95142eSMartin Matuska 					COLLECT_CPS(ucx_size);
31126c95142eSMartin Matuska 				}
31136c95142eSMartin Matuska 				/*
31146c95142eSMartin Matuska 				 * Restart finding a composed code with
31156c95142eSMartin Matuska 				 * the updated uc from the top of the
31166c95142eSMartin Matuska 				 * collected code points.
31176c95142eSMartin Matuska 				 */
31186c95142eSMartin Matuska 				i = 0;
31196c95142eSMartin Matuska 			}
31206c95142eSMartin Matuska 
31216c95142eSMartin Matuska 			/*
31226c95142eSMartin Matuska 			 * Apparently the current code points are not
31236c95142eSMartin Matuska 			 * decomposed characters or already composed.
31246c95142eSMartin Matuska 			 */
31256c95142eSMartin Matuska 			WRITE_UC();
31266c95142eSMartin Matuska 			for (i = 0; i < ucx_size; i++)
31276c95142eSMartin Matuska 				UNPARSE(p, endp, ucx[i]);
31286c95142eSMartin Matuska 
31296c95142eSMartin Matuska 			/*
31306c95142eSMartin Matuska 			 * Flush out remaining canonical combining characters.
31316c95142eSMartin Matuska 			 */
31326c95142eSMartin Matuska 			if (nx > 0 && cx == cl && len > 0) {
31336c95142eSMartin Matuska 				while ((nx = parse(&ucx[0], s, len))
31346c95142eSMartin Matuska 				    > 0) {
31356c95142eSMartin Matuska 					cx = CCC(ucx[0]);
31366c95142eSMartin Matuska 					if (cl > cx)
31376c95142eSMartin Matuska 						break;
31386c95142eSMartin Matuska 					s += nx;
31396c95142eSMartin Matuska 					len -= nx;
31406c95142eSMartin Matuska 					cl = cx;
31416c95142eSMartin Matuska 					UNPARSE(p, endp, ucx[0]);
31426c95142eSMartin Matuska 				}
31436c95142eSMartin Matuska 			}
31446c95142eSMartin Matuska 			break;
31456c95142eSMartin Matuska 		}
31466c95142eSMartin Matuska 		if (n2 < 0) {
31476c95142eSMartin Matuska 			WRITE_UC();
31486c95142eSMartin Matuska 			/* Use a replaced unicode character. */
31496c95142eSMartin Matuska 			UNPARSE(p, endp, uc2);
31506c95142eSMartin Matuska 			s += n2*-1;
31516c95142eSMartin Matuska 			len -= n2*-1;
31526c95142eSMartin Matuska 			ret = -1;
31536c95142eSMartin Matuska 			continue;
31546c95142eSMartin Matuska 		} else if (n2 == 0) {
31556c95142eSMartin Matuska 			WRITE_UC();
31566c95142eSMartin Matuska 			break;
31576c95142eSMartin Matuska 		}
31586c95142eSMartin Matuska 	}
31596c95142eSMartin Matuska 	as->length = p - as->s;
31606c95142eSMartin Matuska 	as->s[as->length] = '\0';
31616c95142eSMartin Matuska 	if (ts == 2)
31626c95142eSMartin Matuska 		as->s[as->length+1] = '\0';
31636c95142eSMartin Matuska 	return (ret);
31646c95142eSMartin Matuska }
31656c95142eSMartin Matuska 
3166fd082e96SMartin Matuska static int
3167fd082e96SMartin Matuska get_nfd(uint32_t *cp1, uint32_t *cp2, uint32_t uc)
3168fd082e96SMartin Matuska {
3169fd082e96SMartin Matuska 	int t, b;
3170fd082e96SMartin Matuska 
3171fd082e96SMartin Matuska 	/*
3172fd082e96SMartin Matuska 	 * These are not converted to NFD on Mac OS.
3173fd082e96SMartin Matuska 	 */
3174fd082e96SMartin Matuska 	if ((uc >= 0x2000 && uc <= 0x2FFF) ||
3175fd082e96SMartin Matuska 	    (uc >= 0xF900 && uc <= 0xFAFF) ||
3176fd082e96SMartin Matuska 	    (uc >= 0x2F800 && uc <= 0x2FAFF))
3177fd082e96SMartin Matuska 		return (0);
3178fd082e96SMartin Matuska 	/*
3179fd082e96SMartin Matuska 	 * Those code points are not converted to NFD on Mac OS.
3180fd082e96SMartin Matuska 	 * I do not know the reason because it is undocumented.
3181fd082e96SMartin Matuska 	 *   NFC        NFD
3182fd082e96SMartin Matuska 	 *   1109A  ==> 11099 110BA
3183fd082e96SMartin Matuska 	 *   1109C  ==> 1109B 110BA
3184fd082e96SMartin Matuska 	 *   110AB  ==> 110A5 110BA
3185fd082e96SMartin Matuska 	 */
3186fd082e96SMartin Matuska 	if (uc == 0x1109A || uc == 0x1109C || uc == 0x110AB)
3187fd082e96SMartin Matuska 		return (0);
3188fd082e96SMartin Matuska 
3189fd082e96SMartin Matuska 	t = 0;
3190fd082e96SMartin Matuska 	b = sizeof(u_decomposition_table)/sizeof(u_decomposition_table[0]) -1;
3191fd082e96SMartin Matuska 	while (b >= t) {
3192fd082e96SMartin Matuska 		int m = (t + b) / 2;
3193fd082e96SMartin Matuska 		if (u_decomposition_table[m].nfc < uc)
3194fd082e96SMartin Matuska 			t = m + 1;
3195fd082e96SMartin Matuska 		else if (u_decomposition_table[m].nfc > uc)
3196fd082e96SMartin Matuska 			b = m - 1;
3197fd082e96SMartin Matuska 		else {
3198fd082e96SMartin Matuska 			*cp1 = u_decomposition_table[m].cp1;
3199fd082e96SMartin Matuska 			*cp2 = u_decomposition_table[m].cp2;
3200fd082e96SMartin Matuska 			return (1);
3201fd082e96SMartin Matuska 		}
3202fd082e96SMartin Matuska 	}
3203fd082e96SMartin Matuska 	return (0);
3204fd082e96SMartin Matuska }
3205fd082e96SMartin Matuska 
3206fd082e96SMartin Matuska #define REPLACE_UC_WITH(cp) do {		\
3207fd082e96SMartin Matuska 	uc = cp;				\
3208fd082e96SMartin Matuska 	ucptr = NULL;				\
3209fd082e96SMartin Matuska } while (0)
32106c95142eSMartin Matuska 
32116c95142eSMartin Matuska /*
32126c95142eSMartin Matuska  * Normalize UTF-8 characters to Form D and copy the result.
32136c95142eSMartin Matuska  */
32146c95142eSMartin Matuska static int
32156c95142eSMartin Matuska archive_string_normalize_D(struct archive_string *as, const void *_p,
32166c95142eSMartin Matuska     size_t len, struct archive_string_conv *sc)
32176c95142eSMartin Matuska {
3218fd082e96SMartin Matuska 	const char *s = (const char *)_p;
3219fd082e96SMartin Matuska 	char *p, *endp;
3220fd082e96SMartin Matuska 	uint32_t uc, uc2;
3221fd082e96SMartin Matuska 	size_t w;
3222fd082e96SMartin Matuska 	int always_replace, n, n2, ret = 0, spair, ts, tm;
3223fd082e96SMartin Matuska 	int (*parse)(uint32_t *, const char *, size_t);
3224fd082e96SMartin Matuska 	size_t (*unparse)(char *, size_t, uint32_t);
32256c95142eSMartin Matuska 
3226fd082e96SMartin Matuska 	always_replace = 1;
3227fd082e96SMartin Matuska 	ts = 1;/* text size. */
3228fd082e96SMartin Matuska 	if (sc->flag & SCONV_TO_UTF16BE) {
3229fd082e96SMartin Matuska 		unparse = unicode_to_utf16be;
3230fd082e96SMartin Matuska 		ts = 2;
3231fd082e96SMartin Matuska 		if (sc->flag & SCONV_FROM_UTF16BE)
3232fd082e96SMartin Matuska 			always_replace = 0;
3233fd082e96SMartin Matuska 	} else if (sc->flag & SCONV_TO_UTF16LE) {
3234fd082e96SMartin Matuska 		unparse = unicode_to_utf16le;
3235fd082e96SMartin Matuska 		ts = 2;
3236fd082e96SMartin Matuska 		if (sc->flag & SCONV_FROM_UTF16LE)
3237fd082e96SMartin Matuska 			always_replace = 0;
3238fd082e96SMartin Matuska 	} else if (sc->flag & SCONV_TO_UTF8) {
3239fd082e96SMartin Matuska 		unparse = unicode_to_utf8;
3240fd082e96SMartin Matuska 		if (sc->flag & SCONV_FROM_UTF8)
3241fd082e96SMartin Matuska 			always_replace = 0;
3242fd082e96SMartin Matuska 	} else {
32436c95142eSMartin Matuska 		/*
32446c95142eSMartin Matuska 		 * This case is going to be converted to another
32456c95142eSMartin Matuska 		 * character-set through iconv.
32466c95142eSMartin Matuska 		 */
3247fd082e96SMartin Matuska 		always_replace = 0;
3248fd082e96SMartin Matuska 		if (sc->flag & SCONV_FROM_UTF16BE) {
3249fd082e96SMartin Matuska 			unparse = unicode_to_utf16be;
3250fd082e96SMartin Matuska 			ts = 2;
3251fd082e96SMartin Matuska 		} else if (sc->flag & SCONV_FROM_UTF16LE) {
3252fd082e96SMartin Matuska 			unparse = unicode_to_utf16le;
3253fd082e96SMartin Matuska 			ts = 2;
3254fd082e96SMartin Matuska 		} else {
3255fd082e96SMartin Matuska 			unparse = unicode_to_utf8;
32566c95142eSMartin Matuska 		}
3257fd082e96SMartin Matuska 	}
32586c95142eSMartin Matuska 
3259fd082e96SMartin Matuska 	if (sc->flag & SCONV_FROM_UTF16BE) {
3260fd082e96SMartin Matuska 		parse = utf16be_to_unicode;
3261fd082e96SMartin Matuska 		tm = 1;
3262fd082e96SMartin Matuska 		spair = 4;/* surrogate pair size in UTF-16. */
3263fd082e96SMartin Matuska 	} else if (sc->flag & SCONV_FROM_UTF16LE) {
3264fd082e96SMartin Matuska 		parse = utf16le_to_unicode;
3265fd082e96SMartin Matuska 		tm = 1;
3266fd082e96SMartin Matuska 		spair = 4;/* surrogate pair size in UTF-16. */
3267fd082e96SMartin Matuska 	} else {
3268fd082e96SMartin Matuska 		parse = cesu8_to_unicode;
3269fd082e96SMartin Matuska 		tm = ts;
3270fd082e96SMartin Matuska 		spair = 6;/* surrogate pair size in UTF-8. */
3271fd082e96SMartin Matuska 	}
3272fd082e96SMartin Matuska 
3273fd082e96SMartin Matuska 	if (archive_string_ensure(as, as->length + len * tm + ts) == NULL)
32746c95142eSMartin Matuska 		return (-1);
3275fd082e96SMartin Matuska 
3276fd082e96SMartin Matuska 	p = as->s + as->length;
3277fd082e96SMartin Matuska 	endp = as->s + as->buffer_length - ts;
3278fd082e96SMartin Matuska 	while ((n = parse(&uc, s, len)) != 0) {
3279fd082e96SMartin Matuska 		const char *ucptr;
3280fd082e96SMartin Matuska 		uint32_t cp1, cp2;
3281fd082e96SMartin Matuska 		int SIndex;
3282fd082e96SMartin Matuska 		struct {
3283fd082e96SMartin Matuska 			uint32_t uc;
3284fd082e96SMartin Matuska 			int ccc;
3285fd082e96SMartin Matuska 		} fdc[FDC_MAX];
3286fd082e96SMartin Matuska 		int fdi, fdj;
3287fd082e96SMartin Matuska 		int ccc;
3288fd082e96SMartin Matuska 
3289fd082e96SMartin Matuska check_first_code:
3290fd082e96SMartin Matuska 		if (n < 0) {
3291fd082e96SMartin Matuska 			/* Use a replaced unicode character. */
3292fd082e96SMartin Matuska 			UNPARSE(p, endp, uc);
3293fd082e96SMartin Matuska 			s += n*-1;
3294fd082e96SMartin Matuska 			len -= n*-1;
3295fd082e96SMartin Matuska 			ret = -1;
3296fd082e96SMartin Matuska 			continue;
3297fd082e96SMartin Matuska 		} else if (n == spair || always_replace)
3298fd082e96SMartin Matuska 			/* uc is converted from a surrogate pair.
3299fd082e96SMartin Matuska 			 * this should be treated as a changed code. */
3300fd082e96SMartin Matuska 			ucptr = NULL;
3301fd082e96SMartin Matuska 		else
3302fd082e96SMartin Matuska 			ucptr = s;
3303fd082e96SMartin Matuska 		s += n;
3304fd082e96SMartin Matuska 		len -= n;
3305fd082e96SMartin Matuska 
3306fd082e96SMartin Matuska 		/* Hangul Decomposition. */
3307fd082e96SMartin Matuska 		if ((SIndex = uc - HC_SBASE) >= 0 && SIndex < HC_SCOUNT) {
3308fd082e96SMartin Matuska 			int L = HC_LBASE + SIndex / HC_NCOUNT;
3309fd082e96SMartin Matuska 			int V = HC_VBASE + (SIndex % HC_NCOUNT) / HC_TCOUNT;
3310fd082e96SMartin Matuska 			int T = HC_TBASE + SIndex % HC_TCOUNT;
3311fd082e96SMartin Matuska 
3312fd082e96SMartin Matuska 			REPLACE_UC_WITH(L);
3313fd082e96SMartin Matuska 			WRITE_UC();
3314fd082e96SMartin Matuska 			REPLACE_UC_WITH(V);
3315fd082e96SMartin Matuska 			WRITE_UC();
3316fd082e96SMartin Matuska 			if (T != HC_TBASE) {
3317fd082e96SMartin Matuska 				REPLACE_UC_WITH(T);
3318fd082e96SMartin Matuska 				WRITE_UC();
3319fd082e96SMartin Matuska 			}
3320fd082e96SMartin Matuska 			continue;
3321fd082e96SMartin Matuska 		}
3322fd082e96SMartin Matuska 		if (IS_DECOMPOSABLE_BLOCK(uc) && CCC(uc) != 0) {
3323fd082e96SMartin Matuska 			WRITE_UC();
3324fd082e96SMartin Matuska 			continue;
33256c95142eSMartin Matuska 		}
33266c95142eSMartin Matuska 
3327fd082e96SMartin Matuska 		fdi = 0;
3328fd082e96SMartin Matuska 		while (get_nfd(&cp1, &cp2, uc) && fdi < FDC_MAX) {
3329fd082e96SMartin Matuska 			int k;
3330fd082e96SMartin Matuska 
3331fd082e96SMartin Matuska 			for (k = fdi; k > 0; k--)
3332fd082e96SMartin Matuska 				fdc[k] = fdc[k-1];
3333fd082e96SMartin Matuska 			fdc[0].ccc = CCC(cp2);
3334fd082e96SMartin Matuska 			fdc[0].uc = cp2;
3335fd082e96SMartin Matuska 			fdi++;
3336fd082e96SMartin Matuska 			REPLACE_UC_WITH(cp1);
3337fd082e96SMartin Matuska 		}
3338fd082e96SMartin Matuska 
3339fd082e96SMartin Matuska 		/* Read following code points. */
3340fd082e96SMartin Matuska 		while ((n2 = parse(&uc2, s, len)) > 0 &&
3341fd082e96SMartin Matuska 		    (ccc = CCC(uc2)) != 0 && fdi < FDC_MAX) {
3342fd082e96SMartin Matuska 			int j, k;
3343fd082e96SMartin Matuska 
3344fd082e96SMartin Matuska 			s += n2;
3345fd082e96SMartin Matuska 			len -= n2;
3346fd082e96SMartin Matuska 			for (j = 0; j < fdi; j++) {
3347fd082e96SMartin Matuska 				if (fdc[j].ccc > ccc)
3348fd082e96SMartin Matuska 					break;
3349fd082e96SMartin Matuska 			}
3350fd082e96SMartin Matuska 			if (j < fdi) {
3351fd082e96SMartin Matuska 				for (k = fdi; k > j; k--)
3352fd082e96SMartin Matuska 					fdc[k] = fdc[k-1];
3353fd082e96SMartin Matuska 				fdc[j].ccc = ccc;
3354fd082e96SMartin Matuska 				fdc[j].uc = uc2;
3355fd082e96SMartin Matuska 			} else {
3356fd082e96SMartin Matuska 				fdc[fdi].ccc = ccc;
3357fd082e96SMartin Matuska 				fdc[fdi].uc = uc2;
3358fd082e96SMartin Matuska 			}
3359fd082e96SMartin Matuska 			fdi++;
3360fd082e96SMartin Matuska 		}
3361fd082e96SMartin Matuska 
3362fd082e96SMartin Matuska 		WRITE_UC();
3363fd082e96SMartin Matuska 		for (fdj = 0; fdj < fdi; fdj++) {
3364fd082e96SMartin Matuska 			REPLACE_UC_WITH(fdc[fdj].uc);
3365fd082e96SMartin Matuska 			WRITE_UC();
3366fd082e96SMartin Matuska 		}
3367fd082e96SMartin Matuska 
3368fd082e96SMartin Matuska 		if (n2 == 0)
3369fd082e96SMartin Matuska 			break;
3370fd082e96SMartin Matuska 		REPLACE_UC_WITH(uc2);
3371fd082e96SMartin Matuska 		n = n2;
3372fd082e96SMartin Matuska 		goto check_first_code;
3373fd082e96SMartin Matuska 	}
3374fd082e96SMartin Matuska 	as->length = p - as->s;
3375fd082e96SMartin Matuska 	as->s[as->length] = '\0';
3376fd082e96SMartin Matuska 	if (ts == 2)
3377fd082e96SMartin Matuska 		as->s[as->length+1] = '\0';
3378fd082e96SMartin Matuska 	return (ret);
3379fd082e96SMartin Matuska }
33806c95142eSMartin Matuska 
33816c95142eSMartin Matuska /*
33826c95142eSMartin Matuska  * libarchive 2.x made incorrect UTF-8 strings in the wrong assumption
33836c95142eSMartin Matuska  * that WCS is Unicode. It is true for several platforms but some are false.
33846c95142eSMartin Matuska  * And then people who did not use UTF-8 locale on the non Unicode WCS
33856c95142eSMartin Matuska  * platform and made a tar file with libarchive(mostly bsdtar) 2.x. Those
33866c95142eSMartin Matuska  * now cannot get right filename from libarchive 3.x and later since we
33876c95142eSMartin Matuska  * fixed the wrong assumption and it is incompatible to older its versions.
33886c95142eSMartin Matuska  * So we provide special option, "compat-2x.x", for resolving it.
33896c95142eSMartin Matuska  * That option enable the string conversion of libarchive 2.x.
33906c95142eSMartin Matuska  *
33916c95142eSMartin Matuska  * Translates the wrong UTF-8 string made by libarchive 2.x into current
33926c95142eSMartin Matuska  * locale character set and appends to the archive_string.
33936c95142eSMartin Matuska  * Note: returns -1 if conversion fails.
33946c95142eSMartin Matuska  */
33956c95142eSMartin Matuska static int
33966c95142eSMartin Matuska strncat_from_utf8_libarchive2(struct archive_string *as,
33976c95142eSMartin Matuska     const void *_p, size_t len, struct archive_string_conv *sc)
33986c95142eSMartin Matuska {
33996c95142eSMartin Matuska 	const char *s;
34006c95142eSMartin Matuska 	int n;
34016c95142eSMartin Matuska 	char *p;
34026c95142eSMartin Matuska 	char *end;
34036c95142eSMartin Matuska 	uint32_t unicode;
34046c95142eSMartin Matuska #if HAVE_WCRTOMB
34056c95142eSMartin Matuska 	mbstate_t shift_state;
34066c95142eSMartin Matuska 
34076c95142eSMartin Matuska 	memset(&shift_state, 0, sizeof(shift_state));
34086c95142eSMartin Matuska #else
34096c95142eSMartin Matuska 	/* Clear the shift state before starting. */
34106c95142eSMartin Matuska 	wctomb(NULL, L'\0');
34116c95142eSMartin Matuska #endif
34126c95142eSMartin Matuska 	(void)sc; /* UNUSED */
34136c95142eSMartin Matuska 	/*
34146c95142eSMartin Matuska 	 * Allocate buffer for MBS.
34156c95142eSMartin Matuska 	 * We need this allocation here since it is possible that
34166c95142eSMartin Matuska 	 * as->s is still NULL.
34176c95142eSMartin Matuska 	 */
34186c95142eSMartin Matuska 	if (archive_string_ensure(as, as->length + len + 1) == NULL)
34196c95142eSMartin Matuska 		return (-1);
34206c95142eSMartin Matuska 
34216c95142eSMartin Matuska 	s = (const char *)_p;
34226c95142eSMartin Matuska 	p = as->s + as->length;
34236c95142eSMartin Matuska 	end = as->s + as->buffer_length - MB_CUR_MAX -1;
34246c95142eSMartin Matuska 	while ((n = _utf8_to_unicode(&unicode, s, len)) != 0) {
34256c95142eSMartin Matuska 		wchar_t wc;
34266c95142eSMartin Matuska 
34276c95142eSMartin Matuska 		if (p >= end) {
34286c95142eSMartin Matuska 			as->length = p - as->s;
34296c95142eSMartin Matuska 			/* Re-allocate buffer for MBS. */
34306c95142eSMartin Matuska 			if (archive_string_ensure(as,
34316c95142eSMartin Matuska 			    as->length + len * 2 + 1) == NULL)
34326c95142eSMartin Matuska 				return (-1);
34336c95142eSMartin Matuska 			p = as->s + as->length;
34346c95142eSMartin Matuska 			end = as->s + as->buffer_length - MB_CUR_MAX -1;
34356c95142eSMartin Matuska 		}
34366c95142eSMartin Matuska 
34376c95142eSMartin Matuska 		/*
34386c95142eSMartin Matuska 		 * As libarchie 2.x, translates the UTF-8 characters into
34396c95142eSMartin Matuska 		 * wide-characters in the assumption that WCS is Unicode.
34406c95142eSMartin Matuska 		 */
34416c95142eSMartin Matuska 		if (n < 0) {
34426c95142eSMartin Matuska 			n *= -1;
34436c95142eSMartin Matuska 			wc = L'?';
34446c95142eSMartin Matuska 		} else
34456c95142eSMartin Matuska 			wc = (wchar_t)unicode;
34466c95142eSMartin Matuska 
34476c95142eSMartin Matuska 		s += n;
34486c95142eSMartin Matuska 		len -= n;
34496c95142eSMartin Matuska 		/*
34506c95142eSMartin Matuska 		 * Translates the wide-character into the current locale MBS.
34516c95142eSMartin Matuska 		 */
34526c95142eSMartin Matuska #if HAVE_WCRTOMB
3453acc60b03SMartin Matuska 		n = (int)wcrtomb(p, wc, &shift_state);
34546c95142eSMartin Matuska #else
3455acc60b03SMartin Matuska 		n = (int)wctomb(p, wc);
34566c95142eSMartin Matuska #endif
34576c95142eSMartin Matuska 		if (n == -1)
34586c95142eSMartin Matuska 			return (-1);
34596c95142eSMartin Matuska 		p += n;
34606c95142eSMartin Matuska 	}
34616c95142eSMartin Matuska 	as->length = p - as->s;
34626c95142eSMartin Matuska 	as->s[as->length] = '\0';
34636c95142eSMartin Matuska 	return (0);
34646c95142eSMartin Matuska }
34656c95142eSMartin Matuska 
34666c95142eSMartin Matuska 
34676c95142eSMartin Matuska /*
34686c95142eSMartin Matuska  * Conversion functions between current locale dependent MBS and UTF-16BE.
34696c95142eSMartin Matuska  *   strncat_from_utf16be() : UTF-16BE --> MBS
34706c95142eSMartin Matuska  *   strncat_to_utf16be()   : MBS --> UTF16BE
34716c95142eSMartin Matuska  */
34726c95142eSMartin Matuska 
34736c95142eSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
34746c95142eSMartin Matuska 
34756c95142eSMartin Matuska /*
34766c95142eSMartin Matuska  * Convert a UTF-16BE/LE string to current locale and copy the result.
34776c95142eSMartin Matuska  * Return -1 if conversion failes.
34786c95142eSMartin Matuska  */
34796c95142eSMartin Matuska static int
34806c95142eSMartin Matuska win_strncat_from_utf16(struct archive_string *as, const void *_p, size_t bytes,
34816c95142eSMartin Matuska     struct archive_string_conv *sc, int be)
34826c95142eSMartin Matuska {
34836c95142eSMartin Matuska 	struct archive_string tmp;
34846c95142eSMartin Matuska 	const char *u16;
34856c95142eSMartin Matuska 	int ll;
34866c95142eSMartin Matuska 	BOOL defchar;
34876c95142eSMartin Matuska 	char *mbs;
34886c95142eSMartin Matuska 	size_t mbs_size, b;
34896c95142eSMartin Matuska 	int ret = 0;
34906c95142eSMartin Matuska 
34916c95142eSMartin Matuska 	bytes &= ~1;
34926c95142eSMartin Matuska 	if (archive_string_ensure(as, as->length + bytes +1) == NULL)
34936c95142eSMartin Matuska 		return (-1);
34946c95142eSMartin Matuska 
34956c95142eSMartin Matuska 	mbs = as->s + as->length;
34966c95142eSMartin Matuska 	mbs_size = as->buffer_length - as->length -1;
34976c95142eSMartin Matuska 
34986c95142eSMartin Matuska 	if (sc->to_cp == CP_C_LOCALE) {
34996c95142eSMartin Matuska 		/*
35006c95142eSMartin Matuska 		 * "C" locale special process.
35016c95142eSMartin Matuska 		 */
35026c95142eSMartin Matuska 		u16 = _p;
35036c95142eSMartin Matuska 		ll = 0;
35046c95142eSMartin Matuska 		for (b = 0; b < bytes; b += 2) {
35056c95142eSMartin Matuska 			uint16_t val;
35066c95142eSMartin Matuska 			if (be)
35076c95142eSMartin Matuska 				val = archive_be16dec(u16+b);
35086c95142eSMartin Matuska 			else
35096c95142eSMartin Matuska 				val = archive_le16dec(u16+b);
35106c95142eSMartin Matuska 			if (val > 255) {
35116c95142eSMartin Matuska 				*mbs++ = '?';
35126c95142eSMartin Matuska 				ret = -1;
35136c95142eSMartin Matuska 			} else
35146c95142eSMartin Matuska 				*mbs++ = (char)(val&0xff);
35156c95142eSMartin Matuska 			ll++;
35166c95142eSMartin Matuska 		}
35176c95142eSMartin Matuska 		as->length += ll;
35186c95142eSMartin Matuska 		as->s[as->length] = '\0';
35196c95142eSMartin Matuska 		return (ret);
35206c95142eSMartin Matuska 	}
35216c95142eSMartin Matuska 
35226c95142eSMartin Matuska 	archive_string_init(&tmp);
35236c95142eSMartin Matuska 	if (be) {
35246c95142eSMartin Matuska 		if (is_big_endian()) {
35256c95142eSMartin Matuska 			u16 = _p;
35266c95142eSMartin Matuska 		} else {
35276c95142eSMartin Matuska 			if (archive_string_ensure(&tmp, bytes+2) == NULL)
35286c95142eSMartin Matuska 				return (-1);
35296c95142eSMartin Matuska 			memcpy(tmp.s, _p, bytes);
35306c95142eSMartin Matuska 			for (b = 0; b < bytes; b += 2) {
35316c95142eSMartin Matuska 				uint16_t val = archive_be16dec(tmp.s+b);
35326c95142eSMartin Matuska 				archive_le16enc(tmp.s+b, val);
35336c95142eSMartin Matuska 			}
35346c95142eSMartin Matuska 			u16 = tmp.s;
35356c95142eSMartin Matuska 		}
35366c95142eSMartin Matuska 	} else {
35376c95142eSMartin Matuska 		if (!is_big_endian()) {
35386c95142eSMartin Matuska 			u16 = _p;
35396c95142eSMartin Matuska 		} else {
35406c95142eSMartin Matuska 			if (archive_string_ensure(&tmp, bytes+2) == NULL)
35416c95142eSMartin Matuska 				return (-1);
35426c95142eSMartin Matuska 			memcpy(tmp.s, _p, bytes);
35436c95142eSMartin Matuska 			for (b = 0; b < bytes; b += 2) {
35446c95142eSMartin Matuska 				uint16_t val = archive_le16dec(tmp.s+b);
35456c95142eSMartin Matuska 				archive_be16enc(tmp.s+b, val);
35466c95142eSMartin Matuska 			}
35476c95142eSMartin Matuska 			u16 = tmp.s;
35486c95142eSMartin Matuska 		}
35496c95142eSMartin Matuska 	}
35506c95142eSMartin Matuska 
35516c95142eSMartin Matuska 	do {
35526c95142eSMartin Matuska 		defchar = 0;
35536c95142eSMartin Matuska 		ll = WideCharToMultiByte(sc->to_cp, 0,
3554acc60b03SMartin Matuska 		    (LPCWSTR)u16, (int)bytes>>1, mbs, (int)mbs_size,
35556c95142eSMartin Matuska 			NULL, &defchar);
3556ae5876eaSMartin Matuska 		/* Exit loop if we succeeded */
3557ae5876eaSMartin Matuska 		if (ll != 0 ||
3558ae5876eaSMartin Matuska 		    GetLastError() != ERROR_INSUFFICIENT_BUFFER) {
3559ae5876eaSMartin Matuska 			break;
3560ae5876eaSMartin Matuska 		}
3561ae5876eaSMartin Matuska 		/* Else expand buffer and loop to try again. */
35626c95142eSMartin Matuska 		ll = WideCharToMultiByte(sc->to_cp, 0,
3563acc60b03SMartin Matuska 		    (LPCWSTR)u16, (int)bytes, NULL, 0, NULL, NULL);
35646c95142eSMartin Matuska 		if (archive_string_ensure(as, ll +1) == NULL)
35656c95142eSMartin Matuska 			return (-1);
35666c95142eSMartin Matuska 		mbs = as->s + as->length;
35676c95142eSMartin Matuska 		mbs_size = as->buffer_length - as->length -1;
3568ae5876eaSMartin Matuska 	} while (1);
35696c95142eSMartin Matuska 	archive_string_free(&tmp);
35706c95142eSMartin Matuska 	as->length += ll;
35716c95142eSMartin Matuska 	as->s[as->length] = '\0';
35726c95142eSMartin Matuska 	if (ll == 0 || defchar)
35736c95142eSMartin Matuska 		ret = -1;
35746c95142eSMartin Matuska 	return (ret);
35756c95142eSMartin Matuska }
35766c95142eSMartin Matuska 
35776c95142eSMartin Matuska static int
3578fd082e96SMartin Matuska win_strncat_from_utf16be(struct archive_string *as, const void *_p,
3579fd082e96SMartin Matuska     size_t bytes, struct archive_string_conv *sc)
35806c95142eSMartin Matuska {
35816c95142eSMartin Matuska 	return (win_strncat_from_utf16(as, _p, bytes, sc, 1));
35826c95142eSMartin Matuska }
35836c95142eSMartin Matuska 
35846c95142eSMartin Matuska static int
3585fd082e96SMartin Matuska win_strncat_from_utf16le(struct archive_string *as, const void *_p,
3586fd082e96SMartin Matuska     size_t bytes, struct archive_string_conv *sc)
35876c95142eSMartin Matuska {
35886c95142eSMartin Matuska 	return (win_strncat_from_utf16(as, _p, bytes, sc, 0));
35896c95142eSMartin Matuska }
35906c95142eSMartin Matuska 
35916c95142eSMartin Matuska static int
35926c95142eSMartin Matuska is_big_endian(void)
35936c95142eSMartin Matuska {
35946c95142eSMartin Matuska 	uint16_t d = 1;
35956c95142eSMartin Matuska 
35966c95142eSMartin Matuska 	return (archive_be16dec(&d) == 1);
35976c95142eSMartin Matuska }
35986c95142eSMartin Matuska 
35996c95142eSMartin Matuska /*
36006c95142eSMartin Matuska  * Convert a current locale string to UTF-16BE/LE and copy the result.
36016c95142eSMartin Matuska  * Return -1 if conversion failes.
36026c95142eSMartin Matuska  */
36036c95142eSMartin Matuska static int
3604fd082e96SMartin Matuska win_strncat_to_utf16(struct archive_string *as16, const void *_p,
3605fd082e96SMartin Matuska     size_t length, struct archive_string_conv *sc, int bigendian)
36066c95142eSMartin Matuska {
36076c95142eSMartin Matuska 	const char *s = (const char *)_p;
36086c95142eSMartin Matuska 	char *u16;
36096c95142eSMartin Matuska 	size_t count, avail;
36106c95142eSMartin Matuska 
36116c95142eSMartin Matuska 	if (archive_string_ensure(as16,
36126c95142eSMartin Matuska 	    as16->length + (length + 1) * 2) == NULL)
36136c95142eSMartin Matuska 		return (-1);
36146c95142eSMartin Matuska 
36156c95142eSMartin Matuska 	u16 = as16->s + as16->length;
36166c95142eSMartin Matuska 	avail = as16->buffer_length - 2;
36176c95142eSMartin Matuska 	if (sc->from_cp == CP_C_LOCALE) {
36186c95142eSMartin Matuska 		/*
36196c95142eSMartin Matuska 		 * "C" locale special process.
36206c95142eSMartin Matuska 		 */
36216c95142eSMartin Matuska 		count = 0;
36226c95142eSMartin Matuska 		while (count < length && *s) {
36236c95142eSMartin Matuska 			if (bigendian)
36246c95142eSMartin Matuska 				archive_be16enc(u16, *s);
36256c95142eSMartin Matuska 			else
36266c95142eSMartin Matuska 				archive_le16enc(u16, *s);
36276c95142eSMartin Matuska 			u16 += 2;
36286c95142eSMartin Matuska 			s++;
36296c95142eSMartin Matuska 			count++;
36306c95142eSMartin Matuska 		}
36316c95142eSMartin Matuska 		as16->length += count << 1;
36326c95142eSMartin Matuska 		as16->s[as16->length] = 0;
36336c95142eSMartin Matuska 		as16->s[as16->length+1] = 0;
36346c95142eSMartin Matuska 		return (0);
36356c95142eSMartin Matuska 	}
36366c95142eSMartin Matuska 	do {
36376c95142eSMartin Matuska 		count = MultiByteToWideChar(sc->from_cp,
3638acc60b03SMartin Matuska 		    MB_PRECOMPOSED, s, (int)length, (LPWSTR)u16, (int)avail>>1);
3639ae5876eaSMartin Matuska 		/* Exit loop if we succeeded */
3640ae5876eaSMartin Matuska 		if (count != 0 ||
3641ae5876eaSMartin Matuska 		    GetLastError() != ERROR_INSUFFICIENT_BUFFER) {
3642ae5876eaSMartin Matuska 			break;
3643ae5876eaSMartin Matuska 		}
3644ae5876eaSMartin Matuska 		/* Expand buffer and try again */
36456c95142eSMartin Matuska 		count = MultiByteToWideChar(sc->from_cp,
3646acc60b03SMartin Matuska 		    MB_PRECOMPOSED, s, (int)length, NULL, 0);
36476c95142eSMartin Matuska 		if (archive_string_ensure(as16, (count +1) * 2)
36486c95142eSMartin Matuska 		    == NULL)
36496c95142eSMartin Matuska 			return (-1);
36506c95142eSMartin Matuska 		u16 = as16->s + as16->length;
36516c95142eSMartin Matuska 		avail = as16->buffer_length - 2;
3652ae5876eaSMartin Matuska 	} while (1);
36536c95142eSMartin Matuska 	as16->length += count * 2;
36546c95142eSMartin Matuska 	as16->s[as16->length] = 0;
36556c95142eSMartin Matuska 	as16->s[as16->length+1] = 0;
36566c95142eSMartin Matuska 	if (count == 0)
36576c95142eSMartin Matuska 		return (-1);
36586c95142eSMartin Matuska 
36596c95142eSMartin Matuska 	if (is_big_endian()) {
36606c95142eSMartin Matuska 		if (!bigendian) {
36616c95142eSMartin Matuska 			while (count > 0) {
36626c95142eSMartin Matuska 				uint16_t v = archive_be16dec(u16);
36636c95142eSMartin Matuska 				archive_le16enc(u16, v);
36646c95142eSMartin Matuska 				u16 += 2;
36656c95142eSMartin Matuska 				count--;
36666c95142eSMartin Matuska 			}
36676c95142eSMartin Matuska 		}
36686c95142eSMartin Matuska 	} else {
36696c95142eSMartin Matuska 		if (bigendian) {
36706c95142eSMartin Matuska 			while (count > 0) {
36716c95142eSMartin Matuska 				uint16_t v = archive_le16dec(u16);
36726c95142eSMartin Matuska 				archive_be16enc(u16, v);
36736c95142eSMartin Matuska 				u16 += 2;
36746c95142eSMartin Matuska 				count--;
36756c95142eSMartin Matuska 			}
36766c95142eSMartin Matuska 		}
36776c95142eSMartin Matuska 	}
36786c95142eSMartin Matuska 	return (0);
36796c95142eSMartin Matuska }
36806c95142eSMartin Matuska 
36816c95142eSMartin Matuska static int
3682fd082e96SMartin Matuska win_strncat_to_utf16be(struct archive_string *as16, const void *_p,
3683fd082e96SMartin Matuska     size_t length, struct archive_string_conv *sc)
36846c95142eSMartin Matuska {
36856c95142eSMartin Matuska 	return (win_strncat_to_utf16(as16, _p, length, sc, 1));
36866c95142eSMartin Matuska }
36876c95142eSMartin Matuska 
36886c95142eSMartin Matuska static int
3689fd082e96SMartin Matuska win_strncat_to_utf16le(struct archive_string *as16, const void *_p,
3690fd082e96SMartin Matuska     size_t length, struct archive_string_conv *sc)
36916c95142eSMartin Matuska {
36926c95142eSMartin Matuska 	return (win_strncat_to_utf16(as16, _p, length, sc, 0));
36936c95142eSMartin Matuska }
36946c95142eSMartin Matuska 
36956c95142eSMartin Matuska #endif /* _WIN32 && !__CYGWIN__ */
36966c95142eSMartin Matuska 
36976c95142eSMartin Matuska /*
36986c95142eSMartin Matuska  * Do the best effort for conversions.
36996c95142eSMartin Matuska  * We cannot handle UTF-16BE character-set without such iconv,
37006c95142eSMartin Matuska  * but there is a chance if a string consists just ASCII code or
37016c95142eSMartin Matuska  * a current locale is UTF-8.
37026c95142eSMartin Matuska  */
37036c95142eSMartin Matuska 
37046c95142eSMartin Matuska /*
37056c95142eSMartin Matuska  * Convert a UTF-16BE string to current locale and copy the result.
37066c95142eSMartin Matuska  * Return -1 if conversion failes.
37076c95142eSMartin Matuska  */
37086c95142eSMartin Matuska static int
37096c95142eSMartin Matuska best_effort_strncat_from_utf16(struct archive_string *as, const void *_p,
37106c95142eSMartin Matuska     size_t bytes, struct archive_string_conv *sc, int be)
37116c95142eSMartin Matuska {
37126c95142eSMartin Matuska 	const char *utf16 = (const char *)_p;
37136c95142eSMartin Matuska 	char *mbs;
37146c95142eSMartin Matuska 	uint32_t uc;
37156c95142eSMartin Matuska 	int n, ret;
37166c95142eSMartin Matuska 
37176c95142eSMartin Matuska 	(void)sc; /* UNUSED */
37186c95142eSMartin Matuska 	/*
37196c95142eSMartin Matuska 	 * Other case, we should do the best effort.
37206c95142eSMartin Matuska 	 * If all character are ASCII(<0x7f), we can convert it.
37216c95142eSMartin Matuska 	 * if not , we set a alternative character and return -1.
37226c95142eSMartin Matuska 	 */
37236c95142eSMartin Matuska 	ret = 0;
37246c95142eSMartin Matuska 	if (archive_string_ensure(as, as->length + bytes +1) == NULL)
37256c95142eSMartin Matuska 		return (-1);
37266c95142eSMartin Matuska 	mbs = as->s + as->length;
37276c95142eSMartin Matuska 
37286c95142eSMartin Matuska 	while ((n = utf16_to_unicode(&uc, utf16, bytes, be)) != 0) {
37296c95142eSMartin Matuska 		if (n < 0) {
37306c95142eSMartin Matuska 			n *= -1;
37316c95142eSMartin Matuska 			ret =  -1;
37326c95142eSMartin Matuska 		}
37336c95142eSMartin Matuska 		bytes -= n;
37346c95142eSMartin Matuska 		utf16 += n;
37356c95142eSMartin Matuska 
37366c95142eSMartin Matuska 		if (uc > 127) {
37376c95142eSMartin Matuska 			/* We cannot handle it. */
37386c95142eSMartin Matuska 			*mbs++ = '?';
37396c95142eSMartin Matuska 			ret =  -1;
37406c95142eSMartin Matuska 		} else
37416c95142eSMartin Matuska 			*mbs++ = (char)uc;
37426c95142eSMartin Matuska 	}
37436c95142eSMartin Matuska 	as->length = mbs - as->s;
37446c95142eSMartin Matuska 	as->s[as->length] = '\0';
37456c95142eSMartin Matuska 	return (ret);
37466c95142eSMartin Matuska }
37476c95142eSMartin Matuska 
37486c95142eSMartin Matuska static int
37496c95142eSMartin Matuska best_effort_strncat_from_utf16be(struct archive_string *as, const void *_p,
37506c95142eSMartin Matuska     size_t bytes, struct archive_string_conv *sc)
37516c95142eSMartin Matuska {
37526c95142eSMartin Matuska 	return (best_effort_strncat_from_utf16(as, _p, bytes, sc, 1));
37536c95142eSMartin Matuska }
37546c95142eSMartin Matuska 
37556c95142eSMartin Matuska static int
37566c95142eSMartin Matuska best_effort_strncat_from_utf16le(struct archive_string *as, const void *_p,
37576c95142eSMartin Matuska     size_t bytes, struct archive_string_conv *sc)
37586c95142eSMartin Matuska {
37596c95142eSMartin Matuska 	return (best_effort_strncat_from_utf16(as, _p, bytes, sc, 0));
37606c95142eSMartin Matuska }
37616c95142eSMartin Matuska 
37626c95142eSMartin Matuska /*
37636c95142eSMartin Matuska  * Convert a current locale string to UTF-16BE/LE and copy the result.
37646c95142eSMartin Matuska  * Return -1 if conversion failes.
37656c95142eSMartin Matuska  */
37666c95142eSMartin Matuska static int
37676c95142eSMartin Matuska best_effort_strncat_to_utf16(struct archive_string *as16, const void *_p,
37686c95142eSMartin Matuska     size_t length, struct archive_string_conv *sc, int bigendian)
37696c95142eSMartin Matuska {
37706c95142eSMartin Matuska 	const char *s = (const char *)_p;
37716c95142eSMartin Matuska 	char *utf16;
37726c95142eSMartin Matuska 	size_t remaining;
37736c95142eSMartin Matuska 	int ret;
37746c95142eSMartin Matuska 
37756c95142eSMartin Matuska 	(void)sc; /* UNUSED */
37766c95142eSMartin Matuska 	/*
37776c95142eSMartin Matuska 	 * Other case, we should do the best effort.
37786c95142eSMartin Matuska 	 * If all character are ASCII(<0x7f), we can convert it.
37796c95142eSMartin Matuska 	 * if not , we set a alternative character and return -1.
37806c95142eSMartin Matuska 	 */
37816c95142eSMartin Matuska 	ret = 0;
37826c95142eSMartin Matuska 	remaining = length;
37836c95142eSMartin Matuska 
37846c95142eSMartin Matuska 	if (archive_string_ensure(as16,
37856c95142eSMartin Matuska 	    as16->length + (length + 1) * 2) == NULL)
37866c95142eSMartin Matuska 		return (-1);
37876c95142eSMartin Matuska 
37886c95142eSMartin Matuska 	utf16 = as16->s + as16->length;
37896c95142eSMartin Matuska 	while (remaining--) {
37906c95142eSMartin Matuska 		unsigned c = *s++;
37916c95142eSMartin Matuska 		if (c > 127) {
37926c95142eSMartin Matuska 			/* We cannot handle it. */
37936c95142eSMartin Matuska 			c = UNICODE_R_CHAR;
37946c95142eSMartin Matuska 			ret = -1;
37956c95142eSMartin Matuska 		}
37966c95142eSMartin Matuska 		if (bigendian)
37976c95142eSMartin Matuska 			archive_be16enc(utf16, c);
37986c95142eSMartin Matuska 		else
37996c95142eSMartin Matuska 			archive_le16enc(utf16, c);
38006c95142eSMartin Matuska 		utf16 += 2;
38016c95142eSMartin Matuska 	}
38026c95142eSMartin Matuska 	as16->length = utf16 - as16->s;
38036c95142eSMartin Matuska 	as16->s[as16->length] = 0;
38046c95142eSMartin Matuska 	as16->s[as16->length+1] = 0;
38056c95142eSMartin Matuska 	return (ret);
38066c95142eSMartin Matuska }
38076c95142eSMartin Matuska 
38086c95142eSMartin Matuska static int
38096c95142eSMartin Matuska best_effort_strncat_to_utf16be(struct archive_string *as16, const void *_p,
38106c95142eSMartin Matuska     size_t length, struct archive_string_conv *sc)
38116c95142eSMartin Matuska {
38126c95142eSMartin Matuska 	return (best_effort_strncat_to_utf16(as16, _p, length, sc, 1));
38136c95142eSMartin Matuska }
38146c95142eSMartin Matuska 
38156c95142eSMartin Matuska static int
38166c95142eSMartin Matuska best_effort_strncat_to_utf16le(struct archive_string *as16, const void *_p,
38176c95142eSMartin Matuska     size_t length, struct archive_string_conv *sc)
38186c95142eSMartin Matuska {
38196c95142eSMartin Matuska 	return (best_effort_strncat_to_utf16(as16, _p, length, sc, 0));
38206c95142eSMartin Matuska }
38216c95142eSMartin Matuska 
38226c95142eSMartin Matuska 
38236c95142eSMartin Matuska /*
38246c95142eSMartin Matuska  * Multistring operations.
38256c95142eSMartin Matuska  */
38266c95142eSMartin Matuska 
38276c95142eSMartin Matuska void
38286c95142eSMartin Matuska archive_mstring_clean(struct archive_mstring *aes)
38296c95142eSMartin Matuska {
38306c95142eSMartin Matuska 	archive_wstring_free(&(aes->aes_wcs));
38316c95142eSMartin Matuska 	archive_string_free(&(aes->aes_mbs));
38326c95142eSMartin Matuska 	archive_string_free(&(aes->aes_utf8));
38336c95142eSMartin Matuska 	archive_string_free(&(aes->aes_mbs_in_locale));
38346c95142eSMartin Matuska 	aes->aes_set = 0;
38356c95142eSMartin Matuska }
38366c95142eSMartin Matuska 
38376c95142eSMartin Matuska void
38386c95142eSMartin Matuska archive_mstring_copy(struct archive_mstring *dest, struct archive_mstring *src)
38396c95142eSMartin Matuska {
38406c95142eSMartin Matuska 	dest->aes_set = src->aes_set;
38416c95142eSMartin Matuska 	archive_string_copy(&(dest->aes_mbs), &(src->aes_mbs));
38426c95142eSMartin Matuska 	archive_string_copy(&(dest->aes_utf8), &(src->aes_utf8));
38436c95142eSMartin Matuska 	archive_wstring_copy(&(dest->aes_wcs), &(src->aes_wcs));
38446c95142eSMartin Matuska }
38456c95142eSMartin Matuska 
38466c95142eSMartin Matuska int
38476c95142eSMartin Matuska archive_mstring_get_utf8(struct archive *a, struct archive_mstring *aes,
38486c95142eSMartin Matuska   const char **p)
38496c95142eSMartin Matuska {
38506c95142eSMartin Matuska 	struct archive_string_conv *sc;
38516c95142eSMartin Matuska 	int r;
38526c95142eSMartin Matuska 
38536c95142eSMartin Matuska 	/* If we already have a UTF8 form, return that immediately. */
38546c95142eSMartin Matuska 	if (aes->aes_set & AES_SET_UTF8) {
38556c95142eSMartin Matuska 		*p = aes->aes_utf8.s;
38566c95142eSMartin Matuska 		return (0);
38576c95142eSMartin Matuska 	}
38586c95142eSMartin Matuska 
38596c95142eSMartin Matuska 	*p = NULL;
38606c95142eSMartin Matuska 	if (aes->aes_set & AES_SET_MBS) {
38616c95142eSMartin Matuska 		sc = archive_string_conversion_to_charset(a, "UTF-8", 1);
38626c95142eSMartin Matuska 		if (sc == NULL)
38636c95142eSMartin Matuska 			return (-1);/* Couldn't allocate memory for sc. */
3864cdf63a70SMartin Matuska 		r = archive_strncpy_l(&(aes->aes_utf8), aes->aes_mbs.s,
38656c95142eSMartin Matuska 		    aes->aes_mbs.length, sc);
38666c95142eSMartin Matuska 		if (a == NULL)
38676c95142eSMartin Matuska 			free_sconv_object(sc);
38686c95142eSMartin Matuska 		if (r == 0) {
38696c95142eSMartin Matuska 			aes->aes_set |= AES_SET_UTF8;
38706c95142eSMartin Matuska 			*p = aes->aes_utf8.s;
38716c95142eSMartin Matuska 			return (0);/* success. */
38726c95142eSMartin Matuska 		} else
38736c95142eSMartin Matuska 			return (-1);/* failure. */
38746c95142eSMartin Matuska 	}
38756c95142eSMartin Matuska 	return (0);/* success. */
38766c95142eSMartin Matuska }
38776c95142eSMartin Matuska 
38786c95142eSMartin Matuska int
38796c95142eSMartin Matuska archive_mstring_get_mbs(struct archive *a, struct archive_mstring *aes,
38806c95142eSMartin Matuska     const char **p)
38816c95142eSMartin Matuska {
38826c95142eSMartin Matuska 	int r, ret = 0;
38836c95142eSMartin Matuska 
38846c95142eSMartin Matuska 	(void)a; /* UNUSED */
38856c95142eSMartin Matuska 	/* If we already have an MBS form, return that immediately. */
38866c95142eSMartin Matuska 	if (aes->aes_set & AES_SET_MBS) {
38876c95142eSMartin Matuska 		*p = aes->aes_mbs.s;
38886c95142eSMartin Matuska 		return (ret);
38896c95142eSMartin Matuska 	}
38906c95142eSMartin Matuska 
38916c95142eSMartin Matuska 	*p = NULL;
38926c95142eSMartin Matuska 	/* If there's a WCS form, try converting with the native locale. */
38936c95142eSMartin Matuska 	if (aes->aes_set & AES_SET_WCS) {
38946c95142eSMartin Matuska 		archive_string_empty(&(aes->aes_mbs));
38956c95142eSMartin Matuska 		r = archive_string_append_from_wcs(&(aes->aes_mbs),
38966c95142eSMartin Matuska 		    aes->aes_wcs.s, aes->aes_wcs.length);
38976c95142eSMartin Matuska 		*p = aes->aes_mbs.s;
38986c95142eSMartin Matuska 		if (r == 0) {
38996c95142eSMartin Matuska 			aes->aes_set |= AES_SET_MBS;
39006c95142eSMartin Matuska 			return (ret);
39016c95142eSMartin Matuska 		} else
39026c95142eSMartin Matuska 			ret = -1;
39036c95142eSMartin Matuska 	}
39046c95142eSMartin Matuska 
39056c95142eSMartin Matuska 	/*
39066c95142eSMartin Matuska 	 * Only a UTF-8 form cannot avail because its conversion already
39076c95142eSMartin Matuska 	 * failed at archive_mstring_update_utf8().
39086c95142eSMartin Matuska 	 */
39096c95142eSMartin Matuska 	return (ret);
39106c95142eSMartin Matuska }
39116c95142eSMartin Matuska 
39126c95142eSMartin Matuska int
39136c95142eSMartin Matuska archive_mstring_get_wcs(struct archive *a, struct archive_mstring *aes,
39146c95142eSMartin Matuska     const wchar_t **wp)
39156c95142eSMartin Matuska {
39166c95142eSMartin Matuska 	int r, ret = 0;
39176c95142eSMartin Matuska 
39186c95142eSMartin Matuska 	(void)a;/* UNUSED */
39196c95142eSMartin Matuska 	/* Return WCS form if we already have it. */
39206c95142eSMartin Matuska 	if (aes->aes_set & AES_SET_WCS) {
39216c95142eSMartin Matuska 		*wp = aes->aes_wcs.s;
39226c95142eSMartin Matuska 		return (ret);
39236c95142eSMartin Matuska 	}
39246c95142eSMartin Matuska 
39256c95142eSMartin Matuska 	*wp = NULL;
39266c95142eSMartin Matuska 	/* Try converting MBS to WCS using native locale. */
39276c95142eSMartin Matuska 	if (aes->aes_set & AES_SET_MBS) {
39286c95142eSMartin Matuska 		archive_wstring_empty(&(aes->aes_wcs));
39296c95142eSMartin Matuska 		r = archive_wstring_append_from_mbs(&(aes->aes_wcs),
39306c95142eSMartin Matuska 		    aes->aes_mbs.s, aes->aes_mbs.length);
39316c95142eSMartin Matuska 		if (r == 0) {
39326c95142eSMartin Matuska 			aes->aes_set |= AES_SET_WCS;
39336c95142eSMartin Matuska 			*wp = aes->aes_wcs.s;
39346c95142eSMartin Matuska 		} else
39356c95142eSMartin Matuska 			ret = -1;/* failure. */
39366c95142eSMartin Matuska 	}
39376c95142eSMartin Matuska 	return (ret);
39386c95142eSMartin Matuska }
39396c95142eSMartin Matuska 
39406c95142eSMartin Matuska int
39416c95142eSMartin Matuska archive_mstring_get_mbs_l(struct archive_mstring *aes,
39426c95142eSMartin Matuska     const char **p, size_t *length, struct archive_string_conv *sc)
39436c95142eSMartin Matuska {
39446c95142eSMartin Matuska 	int r, ret = 0;
39456c95142eSMartin Matuska 
39466c95142eSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
39476c95142eSMartin Matuska 	/*
39486c95142eSMartin Matuska 	 * Internationalization programing on Windows must use Wide
39496c95142eSMartin Matuska 	 * characters because Windows platform cannot make locale UTF-8.
39506c95142eSMartin Matuska 	 */
39516c95142eSMartin Matuska 	if (sc != NULL && (aes->aes_set & AES_SET_WCS) != 0) {
39526c95142eSMartin Matuska 		archive_string_empty(&(aes->aes_mbs_in_locale));
39536c95142eSMartin Matuska 		r = archive_string_append_from_wcs_in_codepage(
39546c95142eSMartin Matuska 		    &(aes->aes_mbs_in_locale), aes->aes_wcs.s,
39556c95142eSMartin Matuska 		    aes->aes_wcs.length, sc);
39566c95142eSMartin Matuska 		if (r == 0) {
39576c95142eSMartin Matuska 			*p = aes->aes_mbs_in_locale.s;
39586c95142eSMartin Matuska 			if (length != NULL)
39596c95142eSMartin Matuska 				*length = aes->aes_mbs_in_locale.length;
39606c95142eSMartin Matuska 			return (0);
39616c95142eSMartin Matuska 		} else if (errno == ENOMEM)
39626c95142eSMartin Matuska 			return (-1);
39636c95142eSMartin Matuska 		else
39646c95142eSMartin Matuska 			ret = -1;
39656c95142eSMartin Matuska 	}
39666c95142eSMartin Matuska #endif
39676c95142eSMartin Matuska 
39686c95142eSMartin Matuska 	/* If there is not an MBS form but is a WCS form, try converting
39696c95142eSMartin Matuska 	 * with the native locale to be used for translating it to specified
39706c95142eSMartin Matuska 	 * character-set. */
39716c95142eSMartin Matuska 	if ((aes->aes_set & AES_SET_MBS) == 0 &&
39726c95142eSMartin Matuska 	    (aes->aes_set & AES_SET_WCS) != 0) {
39736c95142eSMartin Matuska 		archive_string_empty(&(aes->aes_mbs));
39746c95142eSMartin Matuska 		r = archive_string_append_from_wcs(&(aes->aes_mbs),
39756c95142eSMartin Matuska 		    aes->aes_wcs.s, aes->aes_wcs.length);
39766c95142eSMartin Matuska 		if (r == 0)
39776c95142eSMartin Matuska 			aes->aes_set |= AES_SET_MBS;
39786c95142eSMartin Matuska 		else if (errno == ENOMEM)
39796c95142eSMartin Matuska 			return (-1);
39806c95142eSMartin Matuska 		else
39816c95142eSMartin Matuska 			ret = -1;
39826c95142eSMartin Matuska 	}
39836c95142eSMartin Matuska 	/* If we already have an MBS form, use it to be translated to
39846c95142eSMartin Matuska 	 * specified character-set. */
39856c95142eSMartin Matuska 	if (aes->aes_set & AES_SET_MBS) {
39866c95142eSMartin Matuska 		if (sc == NULL) {
39876c95142eSMartin Matuska 			/* Conversion is unneeded. */
39886c95142eSMartin Matuska 			*p = aes->aes_mbs.s;
39896c95142eSMartin Matuska 			if (length != NULL)
39906c95142eSMartin Matuska 				*length = aes->aes_mbs.length;
39916c95142eSMartin Matuska 			return (0);
39926c95142eSMartin Matuska 		}
3993fd082e96SMartin Matuska 		ret = archive_strncpy_l(&(aes->aes_mbs_in_locale),
39946c95142eSMartin Matuska 		    aes->aes_mbs.s, aes->aes_mbs.length, sc);
39956c95142eSMartin Matuska 		*p = aes->aes_mbs_in_locale.s;
39966c95142eSMartin Matuska 		if (length != NULL)
39976c95142eSMartin Matuska 			*length = aes->aes_mbs_in_locale.length;
39986c95142eSMartin Matuska 	} else {
39996c95142eSMartin Matuska 		*p = NULL;
40006c95142eSMartin Matuska 		if (length != NULL)
40016c95142eSMartin Matuska 			*length = 0;
40026c95142eSMartin Matuska 	}
40036c95142eSMartin Matuska 	return (ret);
40046c95142eSMartin Matuska }
40056c95142eSMartin Matuska 
40066c95142eSMartin Matuska int
40076c95142eSMartin Matuska archive_mstring_copy_mbs(struct archive_mstring *aes, const char *mbs)
40086c95142eSMartin Matuska {
40096c95142eSMartin Matuska 	if (mbs == NULL) {
40106c95142eSMartin Matuska 		aes->aes_set = 0;
40116c95142eSMartin Matuska 		return (0);
40126c95142eSMartin Matuska 	}
40136c95142eSMartin Matuska 	return (archive_mstring_copy_mbs_len(aes, mbs, strlen(mbs)));
40146c95142eSMartin Matuska }
40156c95142eSMartin Matuska 
40166c95142eSMartin Matuska int
40176c95142eSMartin Matuska archive_mstring_copy_mbs_len(struct archive_mstring *aes, const char *mbs,
40186c95142eSMartin Matuska     size_t len)
40196c95142eSMartin Matuska {
40206c95142eSMartin Matuska 	if (mbs == NULL) {
40216c95142eSMartin Matuska 		aes->aes_set = 0;
40226c95142eSMartin Matuska 		return (0);
40236c95142eSMartin Matuska 	}
40246c95142eSMartin Matuska 	aes->aes_set = AES_SET_MBS; /* Only MBS form is set now. */
40256c95142eSMartin Matuska 	archive_strncpy(&(aes->aes_mbs), mbs, len);
40266c95142eSMartin Matuska 	archive_string_empty(&(aes->aes_utf8));
40276c95142eSMartin Matuska 	archive_wstring_empty(&(aes->aes_wcs));
40286c95142eSMartin Matuska 	return (0);
40296c95142eSMartin Matuska }
40306c95142eSMartin Matuska 
40316c95142eSMartin Matuska int
40326c95142eSMartin Matuska archive_mstring_copy_wcs(struct archive_mstring *aes, const wchar_t *wcs)
40336c95142eSMartin Matuska {
4034fd082e96SMartin Matuska 	return archive_mstring_copy_wcs_len(aes, wcs,
4035fd082e96SMartin Matuska 				wcs == NULL ? 0 : wcslen(wcs));
40366c95142eSMartin Matuska }
40376c95142eSMartin Matuska 
40386c95142eSMartin Matuska int
4039cdf63a70SMartin Matuska archive_mstring_copy_utf8(struct archive_mstring *aes, const char *utf8)
4040cdf63a70SMartin Matuska {
4041cdf63a70SMartin Matuska   if (utf8 == NULL) {
4042cdf63a70SMartin Matuska     aes->aes_set = 0;
4043cdf63a70SMartin Matuska   }
4044cdf63a70SMartin Matuska   aes->aes_set = AES_SET_UTF8;
4045cdf63a70SMartin Matuska   archive_string_empty(&(aes->aes_mbs));
4046cdf63a70SMartin Matuska   archive_string_empty(&(aes->aes_wcs));
4047cdf63a70SMartin Matuska   archive_strncpy(&(aes->aes_utf8), utf8, strlen(utf8));
4048cdf63a70SMartin Matuska   return (int)strlen(utf8);
4049cdf63a70SMartin Matuska }
4050cdf63a70SMartin Matuska 
4051cdf63a70SMartin Matuska int
40526c95142eSMartin Matuska archive_mstring_copy_wcs_len(struct archive_mstring *aes, const wchar_t *wcs,
40536c95142eSMartin Matuska     size_t len)
40546c95142eSMartin Matuska {
40556c95142eSMartin Matuska 	if (wcs == NULL) {
40566c95142eSMartin Matuska 		aes->aes_set = 0;
40576c95142eSMartin Matuska 	}
40586c95142eSMartin Matuska 	aes->aes_set = AES_SET_WCS; /* Only WCS form set. */
40596c95142eSMartin Matuska 	archive_string_empty(&(aes->aes_mbs));
40606c95142eSMartin Matuska 	archive_string_empty(&(aes->aes_utf8));
40616c95142eSMartin Matuska 	archive_wstrncpy(&(aes->aes_wcs), wcs, len);
40626c95142eSMartin Matuska 	return (0);
40636c95142eSMartin Matuska }
40646c95142eSMartin Matuska 
40656c95142eSMartin Matuska int
40666c95142eSMartin Matuska archive_mstring_copy_mbs_len_l(struct archive_mstring *aes,
40676c95142eSMartin Matuska     const char *mbs, size_t len, struct archive_string_conv *sc)
40686c95142eSMartin Matuska {
40696c95142eSMartin Matuska 	int r;
40706c95142eSMartin Matuska 
40716c95142eSMartin Matuska 	if (mbs == NULL) {
40726c95142eSMartin Matuska 		aes->aes_set = 0;
40736c95142eSMartin Matuska 		return (0);
40746c95142eSMartin Matuska 	}
40756c95142eSMartin Matuska 	archive_string_empty(&(aes->aes_mbs));
40766c95142eSMartin Matuska 	archive_wstring_empty(&(aes->aes_wcs));
40776c95142eSMartin Matuska 	archive_string_empty(&(aes->aes_utf8));
40786c95142eSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__)
40796c95142eSMartin Matuska 	/*
40806c95142eSMartin Matuska 	 * Internationalization programing on Windows must use Wide
40816c95142eSMartin Matuska 	 * characters because Windows platform cannot make locale UTF-8.
40826c95142eSMartin Matuska 	 */
40836c95142eSMartin Matuska 	if (sc == NULL) {
40846c95142eSMartin Matuska 		if (archive_string_append(&(aes->aes_mbs),
40856c95142eSMartin Matuska 			mbs, mbsnbytes(mbs, len)) == NULL) {
40866c95142eSMartin Matuska 			aes->aes_set = 0;
40876c95142eSMartin Matuska 			r = -1;
40886c95142eSMartin Matuska 		} else {
40896c95142eSMartin Matuska 			aes->aes_set = AES_SET_MBS;
40906c95142eSMartin Matuska 			r = 0;
40916c95142eSMartin Matuska 		}
40926c95142eSMartin Matuska #if defined(HAVE_ICONV)
40936c95142eSMartin Matuska 	} else if (sc != NULL && sc->cd_w != (iconv_t)-1) {
40946c95142eSMartin Matuska 		/*
40956c95142eSMartin Matuska 		 * This case happens only when MultiByteToWideChar() cannot
40966c95142eSMartin Matuska 		 * handle sc->from_cp, and we have to iconv in order to
40976c95142eSMartin Matuska 		 * translate character-set to wchar_t,UTF-16.
40986c95142eSMartin Matuska 		 */
40996c95142eSMartin Matuska 		iconv_t cd = sc->cd;
41006c95142eSMartin Matuska 		unsigned from_cp;
41016c95142eSMartin Matuska 		int flag;
41026c95142eSMartin Matuska 
41036c95142eSMartin Matuska 		/*
41046c95142eSMartin Matuska 		 * Translate multi-bytes from some character-set to UTF-8.
41056c95142eSMartin Matuska 		 */
41066c95142eSMartin Matuska 		sc->cd = sc->cd_w;
4107fd082e96SMartin Matuska 		r = archive_strncpy_l(&(aes->aes_utf8), mbs, len, sc);
41086c95142eSMartin Matuska 		sc->cd = cd;
41096c95142eSMartin Matuska 		if (r != 0) {
41106c95142eSMartin Matuska 			aes->aes_set = 0;
41116c95142eSMartin Matuska 			return (r);
41126c95142eSMartin Matuska 		}
41136c95142eSMartin Matuska 		aes->aes_set = AES_SET_UTF8;
41146c95142eSMartin Matuska 
41156c95142eSMartin Matuska 		/*
41166c95142eSMartin Matuska 		 * Append the UTF-8 string into wstring.
41176c95142eSMartin Matuska 		 */
41186c95142eSMartin Matuska 		flag = sc->flag;
41196c95142eSMartin Matuska 		sc->flag &= ~(SCONV_NORMALIZATION_C
41206c95142eSMartin Matuska 				| SCONV_TO_UTF16| SCONV_FROM_UTF16);
41216c95142eSMartin Matuska 		from_cp = sc->from_cp;
41226c95142eSMartin Matuska 		sc->from_cp = CP_UTF8;
41236c95142eSMartin Matuska 		r = archive_wstring_append_from_mbs_in_codepage(&(aes->aes_wcs),
41246c95142eSMartin Matuska 			aes->aes_utf8.s, aes->aes_utf8.length, sc);
41256c95142eSMartin Matuska 		sc->flag = flag;
41266c95142eSMartin Matuska 		sc->from_cp = from_cp;
41276c95142eSMartin Matuska 		if (r == 0)
41286c95142eSMartin Matuska 			aes->aes_set |= AES_SET_WCS;
41296c95142eSMartin Matuska #endif
41306c95142eSMartin Matuska 	} else {
41316c95142eSMartin Matuska 		r = archive_wstring_append_from_mbs_in_codepage(
41326c95142eSMartin Matuska 		    &(aes->aes_wcs), mbs, len, sc);
41336c95142eSMartin Matuska 		if (r == 0)
41346c95142eSMartin Matuska 			aes->aes_set = AES_SET_WCS;
41356c95142eSMartin Matuska 		else
41366c95142eSMartin Matuska 			aes->aes_set = 0;
41376c95142eSMartin Matuska 	}
41386c95142eSMartin Matuska #else
4139fd082e96SMartin Matuska 	r = archive_strncpy_l(&(aes->aes_mbs), mbs, len, sc);
41406c95142eSMartin Matuska 	if (r == 0)
41416c95142eSMartin Matuska 		aes->aes_set = AES_SET_MBS; /* Only MBS form is set now. */
41426c95142eSMartin Matuska 	else
41436c95142eSMartin Matuska 		aes->aes_set = 0;
41446c95142eSMartin Matuska #endif
41456c95142eSMartin Matuska 	return (r);
41466c95142eSMartin Matuska }
41476c95142eSMartin Matuska 
41486c95142eSMartin Matuska /*
41496c95142eSMartin Matuska  * The 'update' form tries to proactively update all forms of
41506c95142eSMartin Matuska  * this string (WCS and MBS) and returns an error if any of
41516c95142eSMartin Matuska  * them fail.  This is used by the 'pax' handler, for instance,
41526c95142eSMartin Matuska  * to detect and report character-conversion failures early while
41536c95142eSMartin Matuska  * still allowing clients to get potentially useful values from
41546c95142eSMartin Matuska  * the more tolerant lazy conversions.  (get_mbs and get_wcs will
41556c95142eSMartin Matuska  * strive to give the user something useful, so you can get hopefully
41566c95142eSMartin Matuska  * usable values even if some of the character conversions are failing.)
41576c95142eSMartin Matuska  */
41586c95142eSMartin Matuska int
41596c95142eSMartin Matuska archive_mstring_update_utf8(struct archive *a, struct archive_mstring *aes,
41606c95142eSMartin Matuska     const char *utf8)
41616c95142eSMartin Matuska {
41626c95142eSMartin Matuska 	struct archive_string_conv *sc;
41636c95142eSMartin Matuska 	int r;
41646c95142eSMartin Matuska 
41656c95142eSMartin Matuska 	if (utf8 == NULL) {
41666c95142eSMartin Matuska 		aes->aes_set = 0;
41676c95142eSMartin Matuska 		return (0); /* Succeeded in clearing everything. */
41686c95142eSMartin Matuska 	}
41696c95142eSMartin Matuska 
41706c95142eSMartin Matuska 	/* Save the UTF8 string. */
41716c95142eSMartin Matuska 	archive_strcpy(&(aes->aes_utf8), utf8);
41726c95142eSMartin Matuska 
41736c95142eSMartin Matuska 	/* Empty the mbs and wcs strings. */
41746c95142eSMartin Matuska 	archive_string_empty(&(aes->aes_mbs));
41756c95142eSMartin Matuska 	archive_wstring_empty(&(aes->aes_wcs));
41766c95142eSMartin Matuska 
41776c95142eSMartin Matuska 	aes->aes_set = AES_SET_UTF8;	/* Only UTF8 is set now. */
41786c95142eSMartin Matuska 
41796c95142eSMartin Matuska 	/* Try converting UTF-8 to MBS, return false on failure. */
41806c95142eSMartin Matuska 	sc = archive_string_conversion_from_charset(a, "UTF-8", 1);
41816c95142eSMartin Matuska 	if (sc == NULL)
41826c95142eSMartin Matuska 		return (-1);/* Couldn't allocate memory for sc. */
4183fd082e96SMartin Matuska 	r = archive_strcpy_l(&(aes->aes_mbs), utf8, sc);
41846c95142eSMartin Matuska 	if (a == NULL)
41856c95142eSMartin Matuska 		free_sconv_object(sc);
41866c95142eSMartin Matuska 	if (r != 0)
41876c95142eSMartin Matuska 		return (-1);
41886c95142eSMartin Matuska 	aes->aes_set = AES_SET_UTF8 | AES_SET_MBS; /* Both UTF8 and MBS set. */
41896c95142eSMartin Matuska 
41906c95142eSMartin Matuska 	/* Try converting MBS to WCS, return false on failure. */
41916c95142eSMartin Matuska 	if (archive_wstring_append_from_mbs(&(aes->aes_wcs), aes->aes_mbs.s,
41926c95142eSMartin Matuska 	    aes->aes_mbs.length))
41936c95142eSMartin Matuska 		return (-1);
41946c95142eSMartin Matuska 	aes->aes_set = AES_SET_UTF8 | AES_SET_WCS | AES_SET_MBS;
41956c95142eSMartin Matuska 
41966c95142eSMartin Matuska 	/* All conversions succeeded. */
41976c95142eSMartin Matuska 	return (0);
41986c95142eSMartin Matuska }
4199