michael@0: /*
michael@0:  * Copyright (c) 2007 2008
michael@0:  * Francois Dumont
michael@0:  *
michael@0:  * This material is provided "as is", with absolutely no warranty expressed
michael@0:  * or implied. Any use is at your own risk.
michael@0:  *
michael@0:  * Permission to use or copy this software for any purpose is hereby granted
michael@0:  * without fee, provided the above notices are retained on all copies.
michael@0:  * Permission to modify the code and to distribute modified code is granted,
michael@0:  * provided the above notices are retained, and a notice that the code was
michael@0:  * modified is included with the above copyright notice.
michael@0:  *
michael@0:  */
michael@0: 
michael@0: #if defined (_STLP_USE_SAFE_STRING_FUNCTIONS)
michael@0: #  define _STLP_WCSNCPY(D, DS, S, C) wcsncpy_s(D, DS, S, C)
michael@0: #else
michael@0: #  define _STLP_WCSNCPY(D, DS, S, C) wcsncpy(D, S, C)
michael@0: #endif
michael@0: 
michael@0: static const wchar_t* __wtrue_name = L"true";
michael@0: static const wchar_t* __wfalse_name = L"false";
michael@0: 
michael@0: typedef struct _Locale_codecvt {
michael@0:   _Locale_lcid_t lc;
michael@0:   UINT cp;
michael@0:   unsigned char cleads[256 / CHAR_BIT];
michael@0:   unsigned char max_char_size;
michael@0:   DWORD mbtowc_flags;
michael@0:   DWORD wctomb_flags;
michael@0: } _Locale_codecvt_t;
michael@0: 
michael@0: /* Ctype */
michael@0: _Locale_mask_t _WLocale_ctype(_Locale_ctype_t* ltype, wint_t c,
michael@0:                               _Locale_mask_t which_bits) {
michael@0:   wchar_t buf[2];
michael@0:   WORD out[2];
michael@0:   buf[0] = c; buf[1] = 0;
michael@0:   GetStringTypeW(CT_CTYPE1, buf, -1, out);
michael@0:   _STLP_MARK_PARAMETER_AS_UNUSED(ltype)
michael@0:   return (_Locale_mask_t)(MapCtypeMask(out[0]) & which_bits);
michael@0: }
michael@0: 
michael@0: wint_t _WLocale_tolower(_Locale_ctype_t* ltype, wint_t c) {
michael@0:   wchar_t in_c = c;
michael@0:   wchar_t res;
michael@0: 
michael@0:   LCMapStringW(ltype->lc.id, LCMAP_LOWERCASE, &in_c, 1, &res, 1);
michael@0:   return res;
michael@0: }
michael@0: 
michael@0: wint_t _WLocale_toupper(_Locale_ctype_t* ltype, wint_t c) {
michael@0:   wchar_t in_c = c;
michael@0:   wchar_t res;
michael@0: 
michael@0:   LCMapStringW(ltype->lc.id, LCMAP_UPPERCASE, &in_c, 1, &res, 1);
michael@0:   return res;
michael@0: }
michael@0: 
michael@0: _Locale_codecvt_t* _Locale_codecvt_create(const char * name, _Locale_lcid_t* lc_hint, int *__err_code) {
michael@0:   char cp_name[MAX_CP_LEN + 1];
michael@0:   unsigned char *ptr;
michael@0:   CPINFO CPInfo;
michael@0:   int i;
michael@0: 
michael@0:   _Locale_codecvt_t *lcodecvt = (_Locale_codecvt_t*)malloc(sizeof(_Locale_codecvt_t));
michael@0: 
michael@0:   if (!lcodecvt) { *__err_code = _STLP_LOC_NO_MEMORY; return lcodecvt; }
michael@0:   memset(lcodecvt, 0, sizeof(_Locale_codecvt_t));
michael@0: 
michael@0:   if (__GetLCIDFromName(name, &lcodecvt->lc.id, cp_name, lc_hint) == -1)
michael@0:   { free(lcodecvt); *__err_code = _STLP_LOC_UNKNOWN_NAME; return NULL; }
michael@0: 
michael@0:   lcodecvt->cp = atoi(cp_name);
michael@0:   if (!GetCPInfo(lcodecvt->cp, &CPInfo)) { free(lcodecvt); return NULL; }
michael@0: 
michael@0:   if (lcodecvt->cp != CP_UTF7 && lcodecvt->cp != CP_UTF8) {
michael@0:     lcodecvt->mbtowc_flags = MB_PRECOMPOSED;
michael@0:     lcodecvt->wctomb_flags = WC_COMPOSITECHECK | WC_SEPCHARS;
michael@0:   }
michael@0:   lcodecvt->max_char_size = CPInfo.MaxCharSize;
michael@0: 
michael@0:   if (CPInfo.MaxCharSize > 1) {
michael@0:     for (ptr = (unsigned char*)CPInfo.LeadByte; *ptr && *(ptr + 1); ptr += 2)
michael@0:       for (i = *ptr; i <= *(ptr + 1); ++i) lcodecvt->cleads[i / CHAR_BIT] |= (0x01 << i % CHAR_BIT);
michael@0:   }
michael@0: 
michael@0:   return lcodecvt;
michael@0: }
michael@0: 
michael@0: char const* _Locale_codecvt_name(const _Locale_codecvt_t* lcodecvt, char* buf) {
michael@0:   char cp_buf[MAX_CP_LEN + 1];
michael@0:   my_ltoa(lcodecvt->cp, cp_buf);
michael@0:   return __GetLocaleName(lcodecvt->lc.id, cp_buf, buf);
michael@0: }
michael@0: 
michael@0: void _Locale_codecvt_destroy(_Locale_codecvt_t* lcodecvt) {
michael@0:   if (!lcodecvt) return;
michael@0: 
michael@0:   free(lcodecvt);
michael@0: }
michael@0: 
michael@0: int _WLocale_mb_cur_max (_Locale_codecvt_t * lcodecvt)
michael@0: { return lcodecvt->max_char_size; }
michael@0: 
michael@0: int _WLocale_mb_cur_min (_Locale_codecvt_t *lcodecvt) {
michael@0:   _STLP_MARK_PARAMETER_AS_UNUSED(lcodecvt)
michael@0:   return 1;
michael@0: }
michael@0: 
michael@0: int _WLocale_is_stateless (_Locale_codecvt_t * lcodecvt)
michael@0: { return (lcodecvt->max_char_size == 1) ? 1 : 0; }
michael@0: 
michael@0: static int __isleadbyte(int i, unsigned char *ctable) {
michael@0:   unsigned char c = (unsigned char)i;
michael@0:   return (ctable[c / CHAR_BIT] & (0x01 << c % CHAR_BIT));
michael@0: }
michael@0: 
michael@0: static int __mbtowc(_Locale_codecvt_t *l, wchar_t *dst, const char *from, unsigned int count) {
michael@0:   int result;
michael@0: 
michael@0:   if (l->cp == CP_UTF7 || l->cp == CP_UTF8) {
michael@0:     result = MultiByteToWideChar(l->cp, l->mbtowc_flags, from, count, dst, 1);
michael@0:     if (result == 0) {
michael@0:       switch (GetLastError()) {
michael@0:         case ERROR_NO_UNICODE_TRANSLATION:
michael@0:           return -2;
michael@0:         default:
michael@0:           return -1;
michael@0:       }
michael@0:     }
michael@0:   }
michael@0:   else {
michael@0:     if (count == 1 && __isleadbyte(*from, l->cleads)) return (size_t)-2;
michael@0:     result = MultiByteToWideChar(l->cp, l->mbtowc_flags, from, count, dst, 1);
michael@0:     if (result == 0) return -1;
michael@0:   }
michael@0: 
michael@0:   return result;
michael@0: }
michael@0: 
michael@0: size_t _WLocale_mbtowc(_Locale_codecvt_t *lcodecvt, wchar_t *to,
michael@0:                        const char *from, size_t n, mbstate_t *shift_state) {
michael@0:   int result;
michael@0:   _STLP_MARK_PARAMETER_AS_UNUSED(shift_state)
michael@0:   if (lcodecvt->max_char_size == 1) { /* Single byte encoding. */
michael@0:     result = MultiByteToWideChar(lcodecvt->cp, lcodecvt->mbtowc_flags, from, 1, to, 1);
michael@0:     if (result == 0) return (size_t)-1;
michael@0:     return result;
michael@0:   }
michael@0:   else { /* Multi byte encoding. */
michael@0:     int retval;
michael@0:     unsigned int count = 1;
michael@0:     while (n--) {
michael@0:       retval = __mbtowc(lcodecvt, to, from, count);
michael@0:       if (retval == -2)
michael@0:       { if (++count > ((unsigned int)lcodecvt->max_char_size)) return (size_t)-1; }
michael@0:       else if (retval == -1)
michael@0:       { return (size_t)-1; }
michael@0:       else
michael@0:       { return count; }
michael@0:     }
michael@0:     return (size_t)-2;
michael@0:   }
michael@0: }
michael@0: 
michael@0: size_t _WLocale_wctomb(_Locale_codecvt_t *lcodecvt, char *to, size_t n,
michael@0:                        const wchar_t c, mbstate_t *shift_state) {
michael@0:   int size = WideCharToMultiByte(lcodecvt->cp, lcodecvt->wctomb_flags, &c, 1, NULL, 0, NULL, NULL);
michael@0: 
michael@0:   if (!size) return (size_t)-1;
michael@0:   if ((size_t)size > n) return (size_t)-2;
michael@0: 
michael@0:   if (n > INT_MAX)
michael@0:     /* Limiting the output buf size to INT_MAX seems like reasonable to transform a single wchar_t. */
michael@0:     n = INT_MAX;
michael@0: 
michael@0:   WideCharToMultiByte(lcodecvt->cp,  lcodecvt->wctomb_flags, &c, 1, to, (int)n, NULL, NULL);
michael@0: 
michael@0:   _STLP_MARK_PARAMETER_AS_UNUSED(shift_state)
michael@0:   return (size_t)size;
michael@0: }
michael@0: 
michael@0: size_t _WLocale_unshift(_Locale_codecvt_t *lcodecvt, mbstate_t *st,
michael@0:                         char *buf, size_t n, char **next) {
michael@0:   /* _WLocale_wctomb do not even touch to st, there is nothing to unshift in this localization implementation. */
michael@0:   _STLP_MARK_PARAMETER_AS_UNUSED(lcodecvt)
michael@0:   _STLP_MARK_PARAMETER_AS_UNUSED(st)
michael@0:   _STLP_MARK_PARAMETER_AS_UNUSED(&n)
michael@0:   *next = buf;
michael@0:   return 0;
michael@0: }
michael@0: 
michael@0: /* Collate */
michael@0: /* This function takes care of the potential size_t DWORD different size. */
michael@0: static int _WLocale_strcmp_aux(_Locale_collate_t* lcol,
michael@0:                                const wchar_t* s1, size_t n1,
michael@0:                                const wchar_t* s2, size_t n2) {
michael@0:   int result = CSTR_EQUAL;
michael@0:   while (n1 > 0 || n2 > 0) {
michael@0:     DWORD size1 = trim_size_t_to_DWORD(n1);
michael@0:     DWORD size2 = trim_size_t_to_DWORD(n2);
michael@0:     result = CompareStringW(lcol->lc.id, 0, s1, size1, s2, size2);
michael@0:     if (result != CSTR_EQUAL)
michael@0:       break;
michael@0:     n1 -= size1;
michael@0:     n2 -= size2;
michael@0:   }
michael@0:   return result;
michael@0: }
michael@0: 
michael@0: int _WLocale_strcmp(_Locale_collate_t* lcol,
michael@0:                     const wchar_t* s1, size_t n1,
michael@0:                     const wchar_t* s2, size_t n2) {
michael@0:   int result;
michael@0:   result = _WLocale_strcmp_aux(lcol, s1, n1, s2, n2);
michael@0:   return (result == CSTR_EQUAL) ? 0 : (result == CSTR_LESS_THAN) ? -1 : 1;
michael@0: }
michael@0: 
michael@0: size_t _WLocale_strxfrm(_Locale_collate_t* lcol,
michael@0:                         wchar_t* dst, size_t dst_size,
michael@0:                         const wchar_t* src, size_t src_size) {
michael@0:   int result, i;
michael@0: 
michael@0:   /* see _Locale_strxfrm: */
michael@0:   if (src_size > INT_MAX) {
michael@0:     if (dst != 0) {
michael@0:       _STLP_WCSNCPY(dst, dst_size, src, src_size);
michael@0:     }
michael@0:     return src_size;
michael@0:   }
michael@0:   if (dst_size > INT_MAX) {
michael@0:     dst_size = INT_MAX;
michael@0:   }
michael@0:   result = LCMapStringW(lcol->lc.id, LCMAP_SORTKEY, src, (int)src_size, dst, (int)dst_size);
michael@0:   if (result != 0 && dst != 0) {
michael@0:     for (i = result - 1; i >= 0; --i) {
michael@0:       dst[i] = ((unsigned char*)dst)[i];
michael@0:     }
michael@0:   }
michael@0:   return result != 0 ? result - 1 : 0;
michael@0: }
michael@0: 
michael@0: /* Numeric */
michael@0: wchar_t _WLocale_decimal_point(_Locale_numeric_t* lnum) {
michael@0:   wchar_t buf[4];
michael@0:   GetLocaleInfoW(lnum->lc.id, LOCALE_SDECIMAL, buf, 4);
michael@0:   return buf[0];
michael@0: }
michael@0: 
michael@0: wchar_t _WLocale_thousands_sep(_Locale_numeric_t* lnum) {
michael@0:   wchar_t buf[4];
michael@0:   GetLocaleInfoW(lnum->lc.id, LOCALE_STHOUSAND, buf, 4);
michael@0:   return buf[0];
michael@0: }
michael@0: 
michael@0: const wchar_t * _WLocale_true(_Locale_numeric_t* lnum, wchar_t* buf, size_t bufSize) {
michael@0:   _STLP_MARK_PARAMETER_AS_UNUSED(lnum)
michael@0:   _STLP_MARK_PARAMETER_AS_UNUSED(buf)
michael@0:   _STLP_MARK_PARAMETER_AS_UNUSED(&bufSize)
michael@0:   return __wtrue_name;
michael@0: }
michael@0: 
michael@0: const wchar_t * _WLocale_false(_Locale_numeric_t* lnum, wchar_t* buf, size_t bufSize) {
michael@0:   _STLP_MARK_PARAMETER_AS_UNUSED(lnum)
michael@0:   _STLP_MARK_PARAMETER_AS_UNUSED(buf)
michael@0:   _STLP_MARK_PARAMETER_AS_UNUSED(&bufSize)
michael@0:   return __wfalse_name;
michael@0: }
michael@0: 
michael@0: /* Monetary */
michael@0: const wchar_t* _WLocale_int_curr_symbol(_Locale_monetary_t * lmon, wchar_t* buf, size_t bufSize)
michael@0: { GetLocaleInfoW(lmon->lc.id, LOCALE_SINTLSYMBOL, buf, (int)bufSize); return buf; }
michael@0: 
michael@0: const wchar_t* _WLocale_currency_symbol(_Locale_monetary_t * lmon, wchar_t* buf, size_t bufSize)
michael@0: { GetLocaleInfoW(lmon->lc.id, LOCALE_SCURRENCY, buf, (int)bufSize); return buf; }
michael@0: 
michael@0: wchar_t _WLocale_mon_decimal_point(_Locale_monetary_t * lmon)
michael@0: { return lmon->decimal_point[0]; }
michael@0: 
michael@0: wchar_t _WLocale_mon_thousands_sep(_Locale_monetary_t * lmon)
michael@0: { return lmon->thousands_sep[0]; }
michael@0: 
michael@0: const wchar_t* _WLocale_positive_sign(_Locale_monetary_t * lmon, wchar_t* buf, size_t bufSize)
michael@0: { GetLocaleInfoW(lmon->lc.id, LOCALE_SPOSITIVESIGN, buf, (int)bufSize); return buf; }
michael@0: 
michael@0: const wchar_t* _WLocale_negative_sign(_Locale_monetary_t * lmon, wchar_t* buf, size_t bufSize)
michael@0: { GetLocaleInfoW(lmon->lc.id, LOCALE_SNEGATIVESIGN, buf, (int)bufSize); return buf; }
michael@0: 
michael@0: /* Time */
michael@0: const wchar_t * _WLocale_full_monthname(_Locale_time_t * ltime, int month,
michael@0:                                         wchar_t* buf, size_t bufSize)
michael@0: { GetLocaleInfoW(ltime->lc.id, LOCALE_SMONTHNAME1 + month, buf, (int)bufSize); return buf; }
michael@0: 
michael@0: const wchar_t * _WLocale_abbrev_monthname(_Locale_time_t * ltime, int month,
michael@0:                                           wchar_t* buf, size_t bufSize)
michael@0: { GetLocaleInfoW(ltime->lc.id, LOCALE_SABBREVMONTHNAME1 + month, buf, (int)bufSize); return buf; }
michael@0: 
michael@0: const wchar_t * _WLocale_full_dayofweek(_Locale_time_t * ltime, int day,
michael@0:                                         wchar_t* buf, size_t bufSize)
michael@0: { GetLocaleInfoW(ltime->lc.id, LOCALE_SDAYNAME1 + day, buf, (int)bufSize); return buf; }
michael@0: 
michael@0: const wchar_t * _WLocale_abbrev_dayofweek(_Locale_time_t * ltime, int day,
michael@0:                                           wchar_t* buf, size_t bufSize)
michael@0: { GetLocaleInfoW(ltime->lc.id, LOCALE_SABBREVDAYNAME1 + day, buf, (int)bufSize); return buf; }
michael@0: 
michael@0: const wchar_t* _WLocale_am_str(_Locale_time_t* ltime,
michael@0:                                wchar_t* buf, size_t bufSize)
michael@0: { GetLocaleInfoW(ltime->lc.id, LOCALE_S1159, buf, (int)bufSize); return buf; }
michael@0: 
michael@0: const wchar_t* _WLocale_pm_str(_Locale_time_t* ltime,
michael@0:                                wchar_t* buf, size_t bufSize)
michael@0: { GetLocaleInfoW(ltime->lc.id, LOCALE_S2359, buf, (int)bufSize); return buf; }