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