/*-----------------------------------------------------------------------
 *
 * PostgreSQL locale utilities
 *
 * Portions Copyright (c) 2002-2025, PostgreSQL Global Development Group
 *
 * src/backend/utils/adt/pg_locale.c
 *
 *-----------------------------------------------------------------------
 */

/*----------
 * Here is how the locale stuff is handled: LC_COLLATE and LC_CTYPE
 * are fixed at CREATE DATABASE time, stored in pg_database, and cannot
 * be changed. Thus, the effects of strcoll(), strxfrm(), isupper(),
 * toupper(), etc. are always in the same fixed locale.
 *
 * LC_MESSAGES is settable at run time and will take effect
 * immediately.
 *
 * The other categories, LC_MONETARY, LC_NUMERIC, and LC_TIME are
 * permanently set to "C", and then we use temporary locale_t
 * objects when we need to look up locale data based on the GUCs
 * of the same name.  Information is cached when the GUCs change.
 * The cached information is only used by the formatting functions
 * (to_char, etc.) and the money type.  For the user, this should all be
 * transparent.
 *----------
 */


#include "postgres.h"

#include <time.h>

#include "access/htup_details.h"
#include "catalog/pg_collation.h"
#include "catalog/pg_database.h"
#include "common/hashfn.h"
#include "common/string.h"
#include "mb/pg_wchar.h"
#include "miscadmin.h"
#include "utils/builtins.h"
#include "utils/formatting.h"
#include "utils/guc_hooks.h"
#include "utils/lsyscache.h"
#include "utils/memutils.h"
#include "utils/pg_locale.h"
#include "utils/relcache.h"
#include "utils/syscache.h"

#ifdef WIN32
#include <shlwapi.h>
#endif

/* Error triggered for locale-sensitive subroutines */
#define  PGLOCALE_SUPPORT_ERROR(provider) \
 elog(ERROR, "unsupported collprovider for %s: %c", __func__, provider)

/*
 * This should be large enough that most strings will fit, but small enough
 * that we feel comfortable putting it on the stack
 */
#define  TEXTBUFLEN   1024

#define  MAX_L10N_DATA  80

/* pg_locale_builtin.c */
extern pg_locale_t create_pg_locale_builtin(Oid collid, MemoryContext context);
extern char *get_collation_actual_version_builtin(const char *collcollate);

/* pg_locale_icu.c */
#ifdef USE_ICU
extern UCollator *pg_ucol_open(const char *loc_str);
extern char *get_collation_actual_version_icu(const char *collcollate);
#endif
extern pg_locale_t create_pg_locale_icu(Oid collid, MemoryContext context);

/* pg_locale_libc.c */
extern pg_locale_t create_pg_locale_libc(Oid collid, MemoryContext context);
extern char *get_collation_actual_version_libc(const char *collcollate);

extern size_t strlower_builtin(char *dst, size_t dstsize, const char *src,
          ssize_t srclen, pg_locale_t locale);
extern size_t strtitle_builtin(char *dst, size_t dstsize, const char *src,
          ssize_t srclen, pg_locale_t locale);
extern size_t strupper_builtin(char *dst, size_t dstsize, const char *src,
          ssize_t srclen, pg_locale_t locale);
extern size_t strfold_builtin(char *dst, size_t dstsize, const char *src,
         ssize_t srclen, pg_locale_t locale);

extern size_t strlower_icu(char *dst, size_t dstsize, const char *src,
         ssize_t srclen, pg_locale_t locale);
extern size_t strtitle_icu(char *dst, size_t dstsize, const char *src,
         ssize_t srclen, pg_locale_t locale);
extern size_t strupper_icu(char *dst, size_t dstsize, const char *src,
         ssize_t srclen, pg_locale_t locale);
extern size_t strfold_icu(char *dst, size_t dstsize, const char *src,
        ssize_t srclen, pg_locale_t locale);

extern size_t strlower_libc(char *dst, size_t dstsize, const char *src,
       ssize_t srclen, pg_locale_t locale);
extern size_t strtitle_libc(char *dst, size_t dstsize, const char *src,
       ssize_t srclen, pg_locale_t locale);
extern size_t strupper_libc(char *dst, size_t dstsize, const char *src,
       ssize_t srclen, pg_locale_t locale);

/* GUC settings */
char    *locale_messages;
char    *locale_monetary;
char    *locale_numeric;
char    *locale_time;

int   icu_validation_level = WARNING;

/*
 * lc_time localization cache.
 *
 * We use only the first 7 or 12 entries of these arrays.  The last array
 * element is left as NULL for the convenience of outside code that wants
 * to sequentially scan these arrays.
 */
char    *localized_abbrev_days[7 + 1];
char    *localized_full_days[7 + 1];
char    *localized_abbrev_months[12 + 1];
char    *localized_full_months[12 + 1];

/* is the databases's LC_CTYPE the C locale? */
bool  database_ctype_is_c = false;

static pg_locale_t default_locale = NULL;

/* indicates whether locale information cache is valid */
static bool CurrentLocaleConvValid = false;
static bool CurrentLCTimeValid = false;

static struct pg_locale_struct c_locale = {
 .provider = COLLPROVIDER_LIBC,
 .deterministic = true,
 .collate_is_c = true,
 .ctype_is_c = true,
};

/* Cache for collation-related knowledge */

typedef struct
{
 Oid   collid;   /* hash key: pg_collation OID */
 pg_locale_t locale;   /* locale_t struct, or 0 if not valid */

 /* needed for simplehash */
 uint32  hash;
 char  status;
} collation_cache_entry;

#define SH_PREFIX  collation_cache
#define SH_ELEMENT_TYPE collation_cache_entry
#define SH_KEY_TYPE  Oid
#define SH_KEY   collid
#define SH_HASH_KEY(tb, key)    murmurhash32((uint32) key)
#define SH_EQUAL(tb, a, b)  (a == b)
#define SH_GET_HASH(tb, a)  a->hash
#define SH_SCOPE  static inline
#define SH_STORE_HASH
#define SH_DECLARE
#define SH_DEFINE
#include "lib/simplehash.h"

static MemoryContext CollationCacheContext = NULL;
static collation_cache_hash *CollationCache = NULL;

/*
 * The collation cache is often accessed repeatedly for the same collation, so
 * remember the last one used.
 */
static Oid last_collation_cache_oid = InvalidOid;
static pg_locale_t last_collation_cache_locale = NULL;

#if defined(WIN32) && defined(LC_MESSAGES)
static char *IsoLocaleName(const char *);
#endif

/*
 * pg_perm_setlocale
 *
 * This wraps the libc function setlocale(), with two additions.  First, when
 * changing LC_CTYPE, update gettext's encoding for the current message
 * domain.  GNU gettext automatically tracks LC_CTYPE on most platforms, but
 * not on Windows.  Second, if the operation is successful, the corresponding
 * LC_XXX environment variable is set to match.  By setting the environment
 * variable, we ensure that any subsequent use of setlocale(..., "") will
 * preserve the settings made through this routine.  Of course, LC_ALL must
 * also be unset to fully ensure that, but that has to be done elsewhere after
 * all the individual LC_XXX variables have been set correctly.  (Thank you
 * Perl for making this kluge necessary.)
 */
char *
pg_perm_setlocale(int category, const char *locale)
{
 char    *result;
 const char *envvar;

#ifndef WIN32
 result = setlocale(category, locale);
#else

 /*
  * On Windows, setlocale(LC_MESSAGES) does not work, so just assume that
  * the given value is good and set it in the environment variables. We
  * must ignore attempts to set to "", which means "keep using the old
  * environment value".
 */
#ifdef LC_MESSAGES
 if (category == LC_MESSAGES)
 {
  result = (char *) locale;
  if (locale == NULL || locale[0] == '\0')
   return result;
 }
 else
#endif
  result = setlocale(category, locale);
#endif       /* WIN32 */

 if (result == NULL)
  return result;   /* fall out immediately on failure */

 /*
  * Use the right encoding in translated messages.  Under ENABLE_NLS, let
  * pg_bind_textdomain_codeset() figure it out.  Under !ENABLE_NLS, message
  * format strings are ASCII, but database-encoding strings may enter the
  * message via %s.  This makes the overall message encoding equal to the
  * database encoding.
 */
 if (category == LC_CTYPE)
 {
  static char save_lc_ctype[LOCALE_NAME_BUFLEN];

  /* copy setlocale() return value before callee invokes it again */
  strlcpy(save_lc_ctype, result, sizeof(save_lc_ctype));
  result = save_lc_ctype;

#ifdef ENABLE_NLS
  SetMessageEncoding(pg_bind_textdomain_codeset(textdomain(NULL)));
#else
  SetMessageEncoding(GetDatabaseEncoding());
#endif
 }

 switch (category)
 {
  case LC_COLLATE:
   envvar = "LC_COLLATE";
   break;
  case LC_CTYPE:
   envvar = "LC_CTYPE";
   break;
#ifdef LC_MESSAGES
  case LC_MESSAGES:
   envvar = "LC_MESSAGES";
#ifdef WIN32
   result = IsoLocaleName(locale);
   if (result == NULL)
    result = (char *) locale;
   elog(DEBUG3, "IsoLocaleName() executed; locale: \"%s\"", result);
#endif       /* WIN32 */
   break;
#endif       /* LC_MESSAGES */
  case LC_MONETARY:
   envvar = "LC_MONETARY";
   break;
  case LC_NUMERIC:
   envvar = "LC_NUMERIC";
   break;
  case LC_TIME:
   envvar = "LC_TIME";
   break;
  default:
   elog(FATAL, "unrecognized LC category: %d", category);
   return NULL;  /* keep compiler quiet */
 }

 if (setenv(envvar, result, 1) != 0)
  return NULL;

 return result;
}


/*
 * Is the locale name valid for the locale category?
 *
 * If successful, and canonname isn't NULL, a palloc'd copy of the locale's
 * canonical name is stored there.  This is especially useful for figuring out
 * what locale name "" means (ie, the server environment value).  (Actually,
 * it seems that on most implementations that's the only thing it's good for;
 * we could wish that setlocale gave back a canonically spelled version of
 * the locale name, but typically it doesn't.)
 */
bool
check_locale(int category, const char *locale, char **canonname)
{
 char    *save;
 char    *res;

 /* Don't let Windows' non-ASCII locale names in. */
 if (!pg_is_ascii(locale))
 {
  ereport(WARNING,
    (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
     errmsg("locale name \"%s\" contains non-ASCII characters",
      locale)));
  return false;
 }

 if (canonname)
  *canonname = NULL;  /* in case of failure */

 save = setlocale(category, NULL);
 if (!save)
  return false;   /* won't happen, we hope */

 /* save may be pointing at a modifiable scratch variable, see above. */
 save = pstrdup(save);

 /* set the locale with setlocale, to see if it accepts it. */
 res = setlocale(category, locale);

 /* save canonical name if requested. */
 if (res && canonname)
  *canonname = pstrdup(res);

 /* restore old value. */
 if (!setlocale(category, save))
  elog(WARNING, "failed to restore old locale \"%s\"", save);
 pfree(save);

 /* Don't let Windows' non-ASCII locale names out. */
 if (canonname && *canonname && !pg_is_ascii(*canonname))
 {
  ereport(WARNING,
    (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
     errmsg("locale name \"%s\" contains non-ASCII characters",
      *canonname)));
  pfree(*canonname);
  *canonname = NULL;
  return false;
 }

 return (res != NULL);
}


/*
 * GUC check/assign hooks
 *
 * For most locale categories, the assign hook doesn't actually set the locale
 * permanently, just reset flags so that the next use will cache the
 * appropriate values.  (See explanation at the top of this file.)
 *
 * Note: we accept value = "" as selecting the postmaster's environment
 * value, whatever it was (so long as the environment setting is legal).
 * This will have been locked down by an earlier call to pg_perm_setlocale.
 */
bool
check_locale_monetary(char **newval, void **extra, GucSource source)
{
 return check_locale(LC_MONETARY, *newval, NULL);
}

void
assign_locale_monetary(const char *newval, void *extra)
{
 CurrentLocaleConvValid = false;
}

bool
check_locale_numeric(char **newval, void **extra, GucSource source)
{
 return check_locale(LC_NUMERIC, *newval, NULL);
}

void
assign_locale_numeric(const char *newval, void *extra)
{
 CurrentLocaleConvValid = false;
}

bool
check_locale_time(char **newval, void **extra, GucSource source)
{
 return check_locale(LC_TIME, *newval, NULL);
}

void
assign_locale_time(const char *newval, void *extra)
{
 CurrentLCTimeValid = false;
}

/*
 * We allow LC_MESSAGES to actually be set globally.
 *
 * Note: we normally disallow value = "" because it wouldn't have consistent
 * semantics (it'd effectively just use the previous value).  However, this
 * is the value passed for PGC_S_DEFAULT, so don't complain in that case,
 * not even if the attempted setting fails due to invalid environment value.
 * The idea there is just to accept the environment setting *if possible*
 * during startup, until we can read the proper value from postgresql.conf.
 */
bool
check_locale_messages(char **newval, void **extra, GucSource source)
{
 if (**newval == '\0')
 {
  if (source == PGC_S_DEFAULT)
   return true;
  else
   return false;
 }

 /*
  * LC_MESSAGES category does not exist everywhere, but accept it anyway
  *
  * On Windows, we can't even check the value, so accept blindly
 */
#if defined(LC_MESSAGES) && !defined(WIN32)
 return check_locale(LC_MESSAGES, *newval, NULL);
#else
 return true;
#endif
}

void
assign_locale_messages(const char *newval, void *extra)
{
 /*
  * LC_MESSAGES category does not exist everywhere, but accept it anyway.
  * We ignore failure, as per comment above.
 */
#ifdef LC_MESSAGES
 (void) pg_perm_setlocale(LC_MESSAGES, newval);
#endif
}


/*
 * Frees the malloced content of a struct lconv.  (But not the struct
 * itself.)  It's important that this not throw elog(ERROR).
 */
static void
free_struct_lconv(struct lconv *s)
{
 free(s->decimal_point);
 free(s->thousands_sep);
 free(s->grouping);
 free(s->int_curr_symbol);
 free(s->currency_symbol);
 free(s->mon_decimal_point);
 free(s->mon_thousands_sep);
 free(s->mon_grouping);
 free(s->positive_sign);
 free(s->negative_sign);
}

/*
 * Check that all fields of a struct lconv (or at least, the ones we care
 * about) are non-NULL.  The field list must match free_struct_lconv().
 */
static bool
struct_lconv_is_valid(struct lconv *s)
{
 if (s->decimal_point == NULL)
  return false;
 if (s->thousands_sep == NULL)
  return false;
 if (s->grouping == NULL)
  return false;
 if (s->int_curr_symbol == NULL)
  return false;
 if (s->currency_symbol == NULL)
  return false;
 if (s->mon_decimal_point == NULL)
  return false;
 if (s->mon_thousands_sep == NULL)
  return false;
 if (s->mon_grouping == NULL)
  return false;
 if (s->positive_sign == NULL)
  return false;
 if (s->negative_sign == NULL)
  return false;
 return true;
}


/*
 * Convert the strdup'd string at *str from the specified encoding to the
 * database encoding.
 */
static void
db_encoding_convert(int encoding, char **str)
{
 char    *pstr;
 char    *mstr;

 /* convert the string to the database encoding */
 pstr = pg_any_to_server(*str, strlen(*str), encoding);
 if (pstr == *str)
  return;     /* no conversion happened */

 /* need it malloc'd not palloc'd */
 mstr = strdup(pstr);
 if (mstr == NULL)
  ereport(ERROR,
    (errcode(ERRCODE_OUT_OF_MEMORY),
     errmsg("out of memory")));

 /* replace old string */
 free(*str);
 *str = mstr;

 pfree(pstr);
}


/*
 * Return the POSIX lconv struct (contains number/money formatting
 * information) with locale information for all categories.
 */
struct lconv *
PGLC_localeconv(void)
{
 static struct lconv CurrentLocaleConv;
 static bool CurrentLocaleConvAllocated = false;
 struct lconv *extlconv;
 struct lconv tmp;
 struct lconv worklconv = {0};

 /* Did we do it already? */
 if (CurrentLocaleConvValid)
  return &CurrentLocaleConv;

 /* Free any already-allocated storage */
 if (CurrentLocaleConvAllocated)
 {
  free_struct_lconv(&CurrentLocaleConv);
  CurrentLocaleConvAllocated = false;
 }

 /*
  * Use thread-safe method of obtaining a copy of lconv from the operating
  * system.
 */
 if (pg_localeconv_r(locale_monetary,
      locale_numeric,
      &tmp) != 0)
  elog(ERROR,
    "could not get lconv for LC_MONETARY = \"%s\", LC_NUMERIC = \"%s\": %m",
    locale_monetary, locale_numeric);

 /* Must copy data now so we can re-encode it. */
 extlconv = &tmp;
 worklconv.decimal_point = strdup(extlconv->decimal_point);
 worklconv.thousands_sep = strdup(extlconv->thousands_sep);
 worklconv.grouping = strdup(extlconv->grouping);
 worklconv.int_curr_symbol = strdup(extlconv->int_curr_symbol);
 worklconv.currency_symbol = strdup(extlconv->currency_symbol);
 worklconv.mon_decimal_point = strdup(extlconv->mon_decimal_point);
 worklconv.mon_thousands_sep = strdup(extlconv->mon_thousands_sep);
 worklconv.mon_grouping = strdup(extlconv->mon_grouping);
 worklconv.positive_sign = strdup(extlconv->positive_sign);
 worklconv.negative_sign = strdup(extlconv->negative_sign);
 /* Copy scalar fields as well */
 worklconv.int_frac_digits = extlconv->int_frac_digits;
 worklconv.frac_digits = extlconv->frac_digits;
 worklconv.p_cs_precedes = extlconv->p_cs_precedes;
 worklconv.p_sep_by_space = extlconv->p_sep_by_space;
 worklconv.n_cs_precedes = extlconv->n_cs_precedes;
 worklconv.n_sep_by_space = extlconv->n_sep_by_space;
 worklconv.p_sign_posn = extlconv->p_sign_posn;
 worklconv.n_sign_posn = extlconv->n_sign_posn;

 /* Free the contents of the object populated by pg_localeconv_r(). */
 pg_localeconv_free(&tmp);

 /* If any of the preceding strdup calls failed, complain now. */
 if (!struct_lconv_is_valid(&worklconv))
  ereport(ERROR,
    (errcode(ERRCODE_OUT_OF_MEMORY),
     errmsg("out of memory")));

 PG_TRY();
 {
  int   encoding;

  /*
   * Now we must perform encoding conversion from whatever's associated
   * with the locales into the database encoding.  If we can't identify
   * the encoding implied by LC_NUMERIC or LC_MONETARY (ie we get -1),
   * use PG_SQL_ASCII, which will result in just validating that the
   * strings are OK in the database encoding.
 */
  encoding = pg_get_encoding_from_locale(locale_numeric, true);
  if (encoding < 0)
   encoding = PG_SQL_ASCII;

  db_encoding_convert(encoding, &worklconv.decimal_point);
  db_encoding_convert(encoding, &worklconv.thousands_sep);
  /* grouping is not text and does not require conversion */

  encoding = pg_get_encoding_from_locale(locale_monetary, true);
  if (encoding < 0)
   encoding = PG_SQL_ASCII;

  db_encoding_convert(encoding, &worklconv.int_curr_symbol);
  db_encoding_convert(encoding, &worklconv.currency_symbol);
  db_encoding_convert(encoding, &worklconv.mon_decimal_point);
  db_encoding_convert(encoding, &worklconv.mon_thousands_sep);
  /* mon_grouping is not text and does not require conversion */
  db_encoding_convert(encoding, &worklconv.positive_sign);
  db_encoding_convert(encoding, &worklconv.negative_sign);
 }
 PG_CATCH();
 {
  free_struct_lconv(&worklconv);
  PG_RE_THROW();
 }
 PG_END_TRY();

 /*
  * Everything is good, so save the results.
 */
 CurrentLocaleConv = worklconv;
 CurrentLocaleConvAllocated = true;
 CurrentLocaleConvValid = true;
 return &CurrentLocaleConv;
}

#ifdef WIN32
/*
 * On Windows, strftime() returns its output in encoding CP_ACP (the default
 * operating system codepage for the computer), which is likely different
 * from SERVER_ENCODING.  This is especially important in Japanese versions
 * of Windows which will use SJIS encoding, which we don't support as a
 * server encoding.
 *
 * So, instead of using strftime(), use wcsftime() to return the value in
 * wide characters (internally UTF16) and then convert to UTF8, which we
 * know how to handle directly.
 *
 * Note that this only affects the calls to strftime() in this file, which are
 * used to get the locale-aware strings. Other parts of the backend use
 * pg_strftime(), which isn't locale-aware and does not need to be replaced.
 */
static size_t
strftime_l_win32(char *dst, size_t dstlen,
     const char *format, const struct tm *tm, locale_t locale)
{
 size_t  len;
 wchar_t  wformat[8];  /* formats used below need 3 chars */
 wchar_t  wbuf[MAX_L10N_DATA];

 /*
  * Get a wchar_t version of the format string.  We only actually use
  * plain-ASCII formats in this file, so we can say that they're UTF8.
 */
 len = MultiByteToWideChar(CP_UTF8, 0, format, -1,
         wformat, lengthof(wformat));
 if (len == 0)
  elog(ERROR, "could not convert format string from UTF-8: error code %lu",
    GetLastError());

 len = _wcsftime_l(wbuf, MAX_L10N_DATA, wformat, tm, locale);
 if (len == 0)
 {
  /*
   * wcsftime failed, possibly because the result would not fit in
   * MAX_L10N_DATA.  Return 0 with the contents of dst unspecified.
 */
  return 0;
 }

 len = WideCharToMultiByte(CP_UTF8, 0, wbuf, len, dst, dstlen - 1,
         NULL, NULL);
 if (len == 0)
  elog(ERROR, "could not convert string to UTF-8: error code %lu",
    GetLastError());

 dst[len] = '\0';

 return len;
}

/* redefine strftime_l() */
#define strftime_l(a,b,c,d,e) strftime_l_win32(a,b,c,d,e)
#endif       /* WIN32 */

/*
 * Subroutine for cache_locale_time().
 * Convert the given string from encoding "encoding" to the database
 * encoding, and store the result at *dst, replacing any previous value.
 */
static void
cache_single_string(char **dst, const char *src, int encoding)
{
 char    *ptr;
 char    *olddst;

 /* Convert the string to the database encoding, or validate it's OK */
 ptr = pg_any_to_server(src, strlen(src), encoding);

 /* Store the string in long-lived storage, replacing any previous value */
 olddst = *dst;
 *dst = MemoryContextStrdup(TopMemoryContext, ptr);
 if (olddst)
  pfree(olddst);

 /* Might as well clean up any palloc'd conversion result, too */
 if (ptr != src)
  pfree(ptr);
}

/*
 * Update the lc_time localization cache variables if needed.
 */
void
cache_locale_time(void)
{
 char  buf[(2 * 7 + 2 * 12) * MAX_L10N_DATA];
 char    *bufptr;
 time_t  timenow;
 struct tm  *timeinfo;
 struct tm timeinfobuf;
 bool  strftimefail = false;
 int   encoding;
 int   i;
 locale_t locale;

 /* did we do this already? */
 if (CurrentLCTimeValid)
  return;

 elog(DEBUG3, "cache_locale_time() executed; locale: \"%s\"", locale_time);

 errno = ENOENT;
#ifdef WIN32
 locale = _create_locale(LC_ALL, locale_time);
 if (locale == (locale_t) 0)
  _dosmaperr(GetLastError());
#else
 locale = newlocale(LC_ALL_MASK, locale_time, (locale_t) 0);
#endif
 if (!locale)
  report_newlocale_failure(locale_time);

 /* We use times close to current time as data for strftime(). */
 timenow = time(NULL);
 timeinfo = gmtime_r(&timenow, &timeinfobuf);

 /* Store the strftime results in MAX_L10N_DATA-sized portions of buf[] */
 bufptr = buf;

 /*
  * MAX_L10N_DATA is sufficient buffer space for every known locale, and
  * POSIX defines no strftime() errors.  (Buffer space exhaustion is not an
  * error.)  An implementation might report errors (e.g. ENOMEM) by
  * returning 0 (or, less plausibly, a negative value) and setting errno.
  * Report errno just in case the implementation did that, but clear it in
  * advance of the calls so we don't emit a stale, unrelated errno.
 */
 errno = 0;

 /* localized days */
 for (i = 0; i < 7; i++)
 {
  timeinfo->tm_wday = i;
  if (strftime_l(bufptr, MAX_L10N_DATA, "%a", timeinfo, locale) <= 0)
   strftimefail = true;
  bufptr += MAX_L10N_DATA;
  if (strftime_l(bufptr, MAX_L10N_DATA, "%A", timeinfo, locale) <= 0)
   strftimefail = true;
  bufptr += MAX_L10N_DATA;
 }

 /* localized months */
 for (i = 0; i < 12; i++)
 {
  timeinfo->tm_mon = i;
  timeinfo->tm_mday = 1; /* make sure we don't have invalid date */
  if (strftime_l(bufptr, MAX_L10N_DATA, "%b", timeinfo, locale) <= 0)
   strftimefail = true;
  bufptr += MAX_L10N_DATA;
  if (strftime_l(bufptr, MAX_L10N_DATA, "%B", timeinfo, locale) <= 0)
   strftimefail = true;
  bufptr += MAX_L10N_DATA;
 }

#ifdef WIN32
 _free_locale(locale);
#else
 freelocale(locale);
#endif

 /*
  * At this point we've done our best to clean up, and can throw errors, or
  * call functions that might throw errors, with a clean conscience.
 */
 if (strftimefail)
  elog(ERROR, "strftime_l() failed");

#ifndef WIN32

 /*
  * As in PGLC_localeconv(), we must convert strftime()'s output from the
  * encoding implied by LC_TIME to the database encoding.  If we can't
  * identify the LC_TIME encoding, just perform encoding validation.
 */
 encoding = pg_get_encoding_from_locale(locale_time, true);
 if (encoding < 0)
  encoding = PG_SQL_ASCII;

#else

 /*
  * On Windows, strftime_win32() always returns UTF8 data, so convert from
  * that if necessary.
 */
 encoding = PG_UTF8;

#endif       /* WIN32 */

 bufptr = buf;

 /* localized days */
 for (i = 0; i < 7; i++)
 {
  cache_single_string(&localized_abbrev_days[i], bufptr, encoding);
  bufptr += MAX_L10N_DATA;
  cache_single_string(&localized_full_days[i], bufptr, encoding);
  bufptr += MAX_L10N_DATA;
 }
 localized_abbrev_days[7] = NULL;
 localized_full_days[7] = NULL;

 /* localized months */
 for (i = 0; i < 12; i++)
 {
  cache_single_string(&localized_abbrev_months[i], bufptr, encoding);
  bufptr += MAX_L10N_DATA;
  cache_single_string(&localized_full_months[i], bufptr, encoding);
  bufptr += MAX_L10N_DATA;
 }
 localized_abbrev_months[12] = NULL;
 localized_full_months[12] = NULL;

 CurrentLCTimeValid = true;
}


#if defined(WIN32) && defined(LC_MESSAGES)
/*
 * Convert a Windows setlocale() argument to a Unix-style one.
 *
 * Regardless of platform, we install message catalogs under a Unix-style
 * LL[_CC][.ENCODING][@VARIANT] naming convention.  Only LC_MESSAGES settings
 * following that style will elicit localized interface strings.
 *
 * Before Visual Studio 2012 (msvcr110.dll), Windows setlocale() accepted "C"
 * (but not "c") and strings of the form <Language>[_<Country>][.<CodePage>],
 * case-insensitive.  setlocale() returns the fully-qualified form; for
 * example, setlocale("thaI") returns "Thai_Thailand.874".  Internally,
 * setlocale() and _create_locale() select a "locale identifier"[1] and store
 * it in an undocumented _locale_t field.  From that LCID, we can retrieve the
 * ISO 639 language and the ISO 3166 country.  Character encoding does not
 * matter, because the server and client encodings govern that.
 *
 * Windows Vista introduced the "locale name" concept[2], closely following
 * RFC 4646.  Locale identifiers are now deprecated.  Starting with Visual
 * Studio 2012, setlocale() accepts locale names in addition to the strings it
 * accepted historically.  It does not standardize them; setlocale("Th-tH")
 * returns "Th-tH".  setlocale(category, "") still returns a traditional
 * string.  Furthermore, msvcr110.dll changed the undocumented _locale_t
 * content to carry locale names instead of locale identifiers.
 *
 * Visual Studio 2015 should still be able to do the same as Visual Studio
 * 2012, but the declaration of locale_name is missing in _locale_t, causing
 * this code compilation to fail, hence this falls back instead on to
 * enumerating all system locales by using EnumSystemLocalesEx to find the
 * required locale name.  If the input argument is in Unix-style then we can
 * get ISO Locale name directly by using GetLocaleInfoEx() with LCType as
 * LOCALE_SNAME.
 *
 * This function returns a pointer to a static buffer bearing the converted
 * name or NULL if conversion fails.
 *
 * [1] https://docs.microsoft.com/en-us/windows/win32/intl/locale-identifiers
 * [2] https://docs.microsoft.com/en-us/windows/win32/intl/locale-names
 */

/*
 * Callback function for EnumSystemLocalesEx() in get_iso_localename().
 *
 * This function enumerates all system locales, searching for one that matches
 * an input with the format: <Language>[_<Country>], e.g.
 * English[_United States]
 *
 * The input is a three wchar_t array as an LPARAM. The first element is the
 * locale_name we want to match, the second element is an allocated buffer
 * where the Unix-style locale is copied if a match is found, and the third
 * element is the search status, 1 if a match was found, 0 otherwise.
 */
static BOOL CALLBACK
search_locale_enum(LPWSTR pStr, DWORD dwFlags, LPARAM lparam)
{
 wchar_t  test_locale[LOCALE_NAME_MAX_LENGTH];
 wchar_t   **argv;

 (void) (dwFlags);

 argv = (wchar_t **) lparam;
 *argv[2] = (wchar_t) 0;

 memset(test_locale, 0, sizeof(test_locale));

 /* Get the name of the <Language> in English */
 if (GetLocaleInfoEx(pStr, LOCALE_SENGLISHLANGUAGENAME,
      test_locale, LOCALE_NAME_MAX_LENGTH))
 {
  /*
   * If the enumerated locale does not have a hyphen ("en") OR the
   * locale_name input does not have an underscore ("English"), we only
   * need to compare the <Language> tags.
 */
  if (wcsrchr(pStr, '-') == NULL || wcsrchr(argv[0], '_') == NULL)
  {
   if (_wcsicmp(argv[0], test_locale) == 0)
   {
    wcscpy(argv[1], pStr);
    *argv[2] = (wchar_t) 1;
    return FALSE;
   }
  }

  /*
   * We have to compare a full <Language>_<Country> tag, so we append
   * the underscore and name of the country/region in English, e.g.
   * "English_United States".
 */
  else
  {
   size_t  len;

   wcscat(test_locale, L"_");
   len = wcslen(test_locale);
   if (GetLocaleInfoEx(pStr, LOCALE_SENGLISHCOUNTRYNAME,
        test_locale + len,
        LOCALE_NAME_MAX_LENGTH - len))
   {
    if (_wcsicmp(argv[0], test_locale) == 0)
    {
     wcscpy(argv[1], pStr);
     *argv[2] = (wchar_t) 1;
     return FALSE;
    }
   }
  }
 }

 return TRUE;
}

/*
 * This function converts a Windows locale name to an ISO formatted version
 * for Visual Studio 2015 or greater.
 *
 * Returns NULL, if no valid conversion was found.
 */
static char *
get_iso_localename(const char *winlocname)
{
 wchar_t  wc_locale_name[LOCALE_NAME_MAX_LENGTH];
 wchar_t  buffer[LOCALE_NAME_MAX_LENGTH];
 static char iso_lc_messages[LOCALE_NAME_MAX_LENGTH];
 const char *period;
 int   len;
 int   ret_val;

 /*
  * Valid locales have the following syntax:
  * <Language>[_<Country>[.<CodePage>]]
  *
  * GetLocaleInfoEx can only take locale name without code-page and for the
  * purpose of this API the code-page doesn't matter.
 */
 period = strchr(winlocname, '.');
 if (period != NULL)
  len = period - winlocname;
 else
  len = pg_mbstrlen(winlocname);

 memset(wc_locale_name, 0, sizeof(wc_locale_name));
 memset(buffer, 0, sizeof(buffer));
 MultiByteToWideChar(CP_ACP, 0, winlocname, len, wc_locale_name,
      LOCALE_NAME_MAX_LENGTH);

 /*
  * If the lc_messages is already a Unix-style string, we have a direct
  * match with LOCALE_SNAME, e.g. en-US, en_US.
 */
 ret_val = GetLocaleInfoEx(wc_locale_name, LOCALE_SNAME, (LPWSTR) &buffer,
         LOCALE_NAME_MAX_LENGTH);
 if (!ret_val)
 {
  /*
   * Search for a locale in the system that matches language and country
   * name.
 */
  wchar_t    *argv[3];

  argv[0] = wc_locale_name;
  argv[1] = buffer;
  argv[2] = (wchar_t *) &ret_val;
  EnumSystemLocalesEx(search_locale_enum, LOCALE_WINDOWS, (LPARAM) argv,
       NULL);
 }

 if (ret_val)
 {
  size_t  rc;
  char    *hyphen;

  /* Locale names use only ASCII, any conversion locale suffices. */
  rc = wchar2char(iso_lc_messages, buffer, sizeof(iso_lc_messages), NULL);
  if (rc == -1 || rc == sizeof(iso_lc_messages))
   return NULL;

  /*
   * Since the message catalogs sit on a case-insensitive filesystem, we
   * need not standardize letter case here.  So long as we do not ship
   * message catalogs for which it would matter, we also need not
   * translate the script/variant portion, e.g.  uz-Cyrl-UZ to
   * uz_UZ@cyrillic.  Simply replace the hyphen with an underscore.
 */
  hyphen = strchr(iso_lc_messages, '-');
  if (hyphen)
   *hyphen = '_';
  return iso_lc_messages;
 }

 return NULL;
}

static char *
IsoLocaleName(const char *winlocname)
{
 static char iso_lc_messages[LOCALE_NAME_MAX_LENGTH];

 if (pg_strcasecmp("c", winlocname) == 0 ||
  pg_strcasecmp("posix", winlocname) == 0)
 {
  strcpy(iso_lc_messages, "C");
  return iso_lc_messages;
 }
 else
  return get_iso_localename(winlocname);
}

#endif       /* WIN32 && LC_MESSAGES */

/*
 * Create a new pg_locale_t struct for the given collation oid.
 */
static pg_locale_t
create_pg_locale(Oid collid, MemoryContext context)
{
 HeapTuple tp;
 Form_pg_collation collform;
 pg_locale_t result;
 Datum  datum;
 bool  isnull;

 tp = SearchSysCache1(COLLOID, ObjectIdGetDatum(collid));
 if (!HeapTupleIsValid(tp))
  elog(ERROR, "cache lookup failed for collation %u", collid);
 collform = (Form_pg_collation) GETSTRUCT(tp);

 if (collform->collprovider == COLLPROVIDER_BUILTIN)
  result = create_pg_locale_builtin(collid, context);
 else if (collform->collprovider == COLLPROVIDER_ICU)
  result = create_pg_locale_icu(collid, context);
 else if (collform->collprovider == COLLPROVIDER_LIBC)
  result = create_pg_locale_libc(collid, context);
 else
  /* shouldn't happen */
  PGLOCALE_SUPPORT_ERROR(collform->collprovider);

 result->is_default = false;

 Assert((result->collate_is_c && result->collate == NULL) ||
     (!result->collate_is_c && result->collate != NULL));

 datum = SysCacheGetAttr(COLLOID, tp, Anum_pg_collation_collversion,
       &isnull);
 if (!isnull)
 {
  char    *actual_versionstr;
  char    *collversionstr;

  collversionstr = TextDatumGetCString(datum);

  if (collform->collprovider == COLLPROVIDER_LIBC)
   datum = SysCacheGetAttrNotNull(COLLOID, tp, Anum_pg_collation_collcollate);
  else
   datum = SysCacheGetAttrNotNull(COLLOID, tp, Anum_pg_collation_colllocale);

  actual_versionstr = get_collation_actual_version(collform->collprovider,
               TextDatumGetCString(datum));
  if (!actual_versionstr)
  {
   /*
    * This could happen when specifying a version in CREATE COLLATION
    * but the provider does not support versioning, or manually
    * creating a mess in the catalogs.
 */
   ereport(ERROR,
     (errmsg("collation \"%s\" has no actual version, but a version was recorded",
       NameStr(collform->collname))));
  }

  if (strcmp(actual_versionstr, collversionstr) != 0)
   ereport(WARNING,
     (errmsg("collation \"%s\" has version mismatch",
       NameStr(collform->collname)),
      errdetail("The collation in the database was created using version %s, "
          "but the operating system provides version %s.",
          collversionstr, actual_versionstr),
      errhint("Rebuild all objects affected by this collation and run "
        "ALTER COLLATION %s REFRESH VERSION, "
        "or build PostgreSQL with the right library version.",
        quote_qualified_identifier(get_namespace_name(collform->collnamespace),
              NameStr(collform->collname)))));
 }

 ReleaseSysCache(tp);

 return result;
}

/*
 * Initialize default_locale with database locale settings.
 */
void
init_database_collation(void)
{
 HeapTuple tup;
 Form_pg_database dbform;
 pg_locale_t result;

 Assert(default_locale == NULL);

 /* Fetch our pg_database row normally, via syscache */
 tup = SearchSysCache1(DATABASEOID, ObjectIdGetDatum(MyDatabaseId));
 if (!HeapTupleIsValid(tup))
  elog(ERROR, "cache lookup failed for database %u", MyDatabaseId);
 dbform = (Form_pg_database) GETSTRUCT(tup);

 if (dbform->datlocprovider == COLLPROVIDER_BUILTIN)
  result = create_pg_locale_builtin(DEFAULT_COLLATION_OID,
            TopMemoryContext);
 else if (dbform->datlocprovider == COLLPROVIDER_ICU)
  result = create_pg_locale_icu(DEFAULT_COLLATION_OID,
           TopMemoryContext);
 else if (dbform->datlocprovider == COLLPROVIDER_LIBC)
  result = create_pg_locale_libc(DEFAULT_COLLATION_OID,
            TopMemoryContext);
 else
  /* shouldn't happen */
  PGLOCALE_SUPPORT_ERROR(dbform->datlocprovider);

 result->is_default = true;
 ReleaseSysCache(tup);

 default_locale = result;
}

/*
 * Create a pg_locale_t from a collation OID.  Results are cached for the
 * lifetime of the backend.  Thus, do not free the result with freelocale().
 *
 * For simplicity, we always generate COLLATE + CTYPE even though we
 * might only need one of them.  Since this is called only once per session,
 * it shouldn't cost much.
 */
pg_locale_t
pg_newlocale_from_collation(Oid collid)
{
 collation_cache_entry *cache_entry;
 bool  found;

 if (collid == DEFAULT_COLLATION_OID)
  return default_locale;

 /*
  * Some callers expect C_COLLATION_OID to succeed even without catalog
  * access.
 */
 if (collid == C_COLLATION_OID)
  return &c_locale;

 if (!OidIsValid(collid))
  elog(ERROR, "cache lookup failed for collation %u", collid);

 AssertCouldGetRelation();

 if (last_collation_cache_oid == collid)
  return last_collation_cache_locale;

 if (CollationCache == NULL)
 {
  CollationCacheContext = AllocSetContextCreate(TopMemoryContext,
               "collation cache",
               ALLOCSET_DEFAULT_SIZES);
  CollationCache = collation_cache_create(CollationCacheContext,
            16, NULL);
 }

 cache_entry = collation_cache_insert(CollationCache, collid, &found);
 if (!found)
 {
  /*
   * Make sure cache entry is marked invalid, in case we fail before
   * setting things.
 */
  cache_entry->locale = 0;
 }

 if (cache_entry->locale == 0)
 {
  cache_entry->locale = create_pg_locale(collid, CollationCacheContext);
 }

 last_collation_cache_oid = collid;
 last_collation_cache_locale = cache_entry->locale;

 return cache_entry->locale;
}

/*
 * Get provider-specific collation version string for the given collation from
 * the operating system/library.
 */
char *
get_collation_actual_version(char collprovider, const char *collcollate)
{
 char    *collversion = NULL;

 if (collprovider == COLLPROVIDER_BUILTIN)
  collversion = get_collation_actual_version_builtin(collcollate);
#ifdef USE_ICU
 else if (collprovider == COLLPROVIDER_ICU)
  collversion = get_collation_actual_version_icu(collcollate);
#endif
 else if (collprovider == COLLPROVIDER_LIBC)
  collversion = get_collation_actual_version_libc(collcollate);

 return collversion;
}

size_t
pg_strlower(char *dst, size_t dstsize, const char *src, ssize_t srclen,
   pg_locale_t locale)
{
 if (locale->provider == COLLPROVIDER_BUILTIN)
  return strlower_builtin(dst, dstsize, src, srclen, locale);
#ifdef USE_ICU
 else if (locale->provider == COLLPROVIDER_ICU)
  return strlower_icu(dst, dstsize, src, srclen, locale);
#endif
 else if (locale->provider == COLLPROVIDER_LIBC)
  return strlower_libc(dst, dstsize, src, srclen, locale);
 else
  /* shouldn't happen */
  PGLOCALE_SUPPORT_ERROR(locale->provider);

 return 0;     /* keep compiler quiet */
}

size_t
pg_strtitle(char *dst, size_t dstsize, const char *src, ssize_t srclen,
   pg_locale_t locale)
{
 if (locale->provider == COLLPROVIDER_BUILTIN)
  return strtitle_builtin(dst, dstsize, src, srclen, locale);
#ifdef USE_ICU
 else if (locale->provider == COLLPROVIDER_ICU)
  return strtitle_icu(dst, dstsize, src, srclen, locale);
#endif
 else if (locale->provider == COLLPROVIDER_LIBC)
  return strtitle_libc(dst, dstsize, src, srclen, locale);
 else
  /* shouldn't happen */
  PGLOCALE_SUPPORT_ERROR(locale->provider);

 return 0;     /* keep compiler quiet */
}

size_t
pg_strupper(char *dst, size_t dstsize, const char *src, ssize_t srclen,
   pg_locale_t locale)
{
 if (locale->provider == COLLPROVIDER_BUILTIN)
  return strupper_builtin(dst, dstsize, src, srclen, locale);
#ifdef USE_ICU
 else if (locale->provider == COLLPROVIDER_ICU)
  return strupper_icu(dst, dstsize, src, srclen, locale);
#endif
 else if (locale->provider == COLLPROVIDER_LIBC)
  return strupper_libc(dst, dstsize, src, srclen, locale);
 else
  /* shouldn't happen */
  PGLOCALE_SUPPORT_ERROR(locale->provider);

 return 0;     /* keep compiler quiet */
}

size_t
pg_strfold(char *dst, size_t dstsize, const char *src, ssize_t srclen,
     pg_locale_t locale)
{
 if (locale->provider == COLLPROVIDER_BUILTIN)
  return strfold_builtin(dst, dstsize, src, srclen, locale);
#ifdef USE_ICU
 else if (locale->provider == COLLPROVIDER_ICU)
  return strfold_icu(dst, dstsize, src, srclen, locale);
#endif
 /* for libc, just use strlower */
 else if (locale->provider == COLLPROVIDER_LIBC)
  return strlower_libc(dst, dstsize, src, srclen, locale);
 else
  /* shouldn't happen */
  PGLOCALE_SUPPORT_ERROR(locale->provider);

 return 0;     /* keep compiler quiet */
}

/*
 * pg_strcoll
 *
 * Like pg_strncoll for NUL-terminated input strings.
 */
int
pg_strcoll(const char *arg1, const char *arg2, pg_locale_t locale)
{
 return locale->collate->strncoll(arg1, -1, arg2, -1, locale);
}

/*
 * pg_strncoll
 *
 * Call ucol_strcollUTF8(), ucol_strcoll(), strcoll_l() or wcscoll_l() as
 * appropriate for the given locale, platform, and database encoding. If the
 * locale is not specified, use the database collation.
 *
 * The input strings must be encoded in the database encoding. If an input
 * string is NUL-terminated, its length may be specified as -1.
 *
 * The caller is responsible for breaking ties if the collation is
 * deterministic; this maintains consistency with pg_strnxfrm(), which cannot
 * easily account for deterministic collations.
 */
int
pg_strncoll(const char *arg1, ssize_t len1, const char *arg2, ssize_t len2,
   pg_locale_t locale)
{
 return locale->collate->strncoll(arg1, len1, arg2, len2, locale);
}

/*
 * Return true if the collation provider supports pg_strxfrm() and
 * pg_strnxfrm(); otherwise false.
 *
 *
 * No similar problem is known for the ICU provider.
 */
bool
pg_strxfrm_enabled(pg_locale_t locale)
{
 /*
  * locale->collate->strnxfrm is still a required method, even if it may
  * have the wrong behavior, because the planner uses it for estimates in
  * some cases.
 */
 return locale->collate->strxfrm_is_safe;
}

/*
 * pg_strxfrm
 *
 * Like pg_strnxfrm for a NUL-terminated input string.
 */
size_t
pg_strxfrm(char *dest, const char *src, size_t destsize, pg_locale_t locale)
{
 return locale->collate->strnxfrm(dest, destsize, src, -1, locale);
}

/*
 * pg_strnxfrm
 *
 * Transforms 'src' to a nul-terminated string stored in 'dest' such that
 * ordinary strcmp() on transformed strings is equivalent to pg_strcoll() on
 * untransformed strings.
 *
 * The input string must be encoded in the database encoding. If the input
 * string is NUL-terminated, its length may be specified as -1. If 'destsize'
 * is zero, 'dest' may be NULL.
 *
 * Not all providers support pg_strnxfrm() safely. The caller should check
 * pg_strxfrm_enabled() first, otherwise this function may return wrong
 * results or an error.
 *
 * Returns the number of bytes needed (or more) to store the transformed
 * string, excluding the terminating nul byte. If the value returned is
 * 'destsize' or greater, the resulting contents of 'dest' are undefined.
 */
size_t
pg_strnxfrm(char *dest, size_t destsize, const char *src, ssize_t srclen,
   pg_locale_t locale)
{
 return locale->collate->strnxfrm(dest, destsize, src, srclen, locale);
}

/*
 * Return true if the collation provider supports pg_strxfrm_prefix() and
 * pg_strnxfrm_prefix(); otherwise false.
 */
bool
pg_strxfrm_prefix_enabled(pg_locale_t locale)
{
 return (locale->collate->strnxfrm_prefix != NULL);
}

/*
 * pg_strxfrm_prefix
 *
 * Like pg_strnxfrm_prefix for a NUL-terminated input string.
 */
size_t
pg_strxfrm_prefix(char *dest, const char *src, size_t destsize,
      pg_locale_t locale)
{
 return locale->collate->strnxfrm_prefix(dest, destsize, src, -1, locale);
}

/*
 * pg_strnxfrm_prefix
 *
 * Transforms 'src' to a byte sequence stored in 'dest' such that ordinary
 * memcmp() on the byte sequence is equivalent to pg_strncoll() on
 * untransformed strings. The result is not nul-terminated.
 *
 * The input string must be encoded in the database encoding. If the input
 * string is NUL-terminated, its length may be specified as -1.
 *
 * Not all providers support pg_strnxfrm_prefix() safely. The caller should
 * check pg_strxfrm_prefix_enabled() first, otherwise this function may return
 * wrong results or an error.
 *
 * If destsize is not large enough to hold the resulting byte sequence, stores
 * only the first destsize bytes in 'dest'. Returns the number of bytes
 * actually copied to 'dest'.
 */
size_t
pg_strnxfrm_prefix(char *dest, size_t destsize, const char *src,
       ssize_t srclen, pg_locale_t locale)
{
 return locale->collate->strnxfrm_prefix(dest, destsize, src, srclen, locale);
}

/*
 * Return required encoding ID for the given locale, or -1 if any encoding is
 * valid for the locale.
 */
int
builtin_locale_encoding(const char *locale)
{
 if (strcmp(locale, "C") == 0)
  return -1;
 else if (strcmp(locale, "C.UTF-8") == 0)
  return PG_UTF8;
 else if (strcmp(locale, "PG_UNICODE_FAST") == 0)
  return PG_UTF8;


 ereport(ERROR,
   (errcode(ERRCODE_WRONG_OBJECT_TYPE),
    errmsg("invalid locale name \"%s\" for builtin provider",
     locale)));

 return 0;     /* keep compiler quiet */
}


/*
 * Validate the locale and encoding combination, and return the canonical form
 * of the locale name.
 */
const char *
builtin_validate_locale(int encoding, const char *locale)
{
 const char *canonical_name = NULL;
 int   required_encoding;

 if (strcmp(locale, "C") == 0)
  canonical_name = "C";
 else if (strcmp(locale, "C.UTF-8") == 0 || strcmp(locale, "C.UTF8") == 0)
  canonical_name = "C.UTF-8";
 else if (strcmp(locale, "PG_UNICODE_FAST") == 0)
  canonical_name = "PG_UNICODE_FAST";

 if (!canonical_name)
  ereport(ERROR,
    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
     errmsg("invalid locale name \"%s\" for builtin provider",
      locale)));

 required_encoding = builtin_locale_encoding(canonical_name);
 if (required_encoding >= 0 && encoding != required_encoding)
  ereport(ERROR,
    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
     errmsg("encoding \"%s\" does not match locale \"%s\"",
      pg_encoding_to_char(encoding), locale)));

 return canonical_name;
}



/*
 * Return the BCP47 language tag representation of the requested locale.
 *
 * This function should be called before passing the string to ucol_open(),
 * because conversion to a language tag also performs "level 2
 * canonicalization". In addition to producing a consistent format, level 2
 * canonicalization is able to more accurately interpret different input
 * locale string formats, such as POSIX and .NET IDs.
 */
char *
icu_language_tag(const char *loc_str, int elevel)
{
#ifdef USE_ICU
 UErrorCode status;
 char    *langtag;
 size_t  buflen = 32; /* arbitrary starting buffer size */
 const bool strict = true;

 /*
  * A BCP47 language tag doesn't have a clearly-defined upper limit (cf.
  * RFC5646 section 4.4). Additionally, in older ICU versions,
  * uloc_toLanguageTag() doesn't always return the ultimate length on the
  * first call, necessitating a loop.
 */
 langtag = palloc(buflen);
 while (true)
 {
  status = U_ZERO_ERROR;
  uloc_toLanguageTag(loc_str, langtag, buflen, strict, &status);

  /* try again if the buffer is not large enough */
  if ((status == U_BUFFER_OVERFLOW_ERROR ||
    status == U_STRING_NOT_TERMINATED_WARNING) &&
   buflen < MaxAllocSize)
  {
   buflen = Min(buflen * 2, MaxAllocSize);
   langtag = repalloc(langtag, buflen);
   continue;
  }

  break;
 }

 if (U_FAILURE(status))
 {
  pfree(langtag);

  if (elevel > 0)
   ereport(elevel,
     (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
      errmsg("could not convert locale name \"%s\" to language tag: %s",
       loc_str, u_errorName(status))));
  return NULL;
 }

 return langtag;
#else       /* not USE_ICU */
 ereport(ERROR,
   (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
    errmsg("ICU is not supported in this build")));
 return NULL;    /* keep compiler quiet */
#endif       /* not USE_ICU */
}

/*
 * Perform best-effort check that the locale is a valid one.
 */
void
icu_validate_locale(const char *loc_str)
{
#ifdef USE_ICU
 UCollator  *collator;
 UErrorCode status;
 char  lang[ULOC_LANG_CAPACITY];
 bool  found = false;
 int   elevel = icu_validation_level;

 /* no validation */
 if (elevel < 0)
  return;

 /* downgrade to WARNING during pg_upgrade */
 if (IsBinaryUpgrade && elevel > WARNING)
  elevel = WARNING;

 /* validate that we can extract the language */
 status = U_ZERO_ERROR;
 uloc_getLanguage(loc_str, lang, ULOC_LANG_CAPACITY, &status);
 if (U_FAILURE(status) || status == U_STRING_NOT_TERMINATED_WARNING)
 {
  ereport(elevel,
    (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
     errmsg("could not get language from ICU locale \"%s\": %s",
      loc_str, u_errorName(status)),
     errhint("To disable ICU locale validation, set the parameter \"%s\" to \"%s\".",
       "icu_validation_level", "disabled")));
  return;
 }

 /* check for special language name */
 if (strcmp(lang, "") == 0 ||
  strcmp(lang, "root") == 0 || strcmp(lang, "und") == 0)
  found = true;

 /* search for matching language within ICU */
 for (int32_t i = 0; !found && i < uloc_countAvailable(); i++)
 {
  const char *otherloc = uloc_getAvailable(i);
  char  otherlang[ULOC_LANG_CAPACITY];

  status = U_ZERO_ERROR;
  uloc_getLanguage(otherloc, otherlang, ULOC_LANG_CAPACITY, &status);
  if (U_FAILURE(status) || status == U_STRING_NOT_TERMINATED_WARNING)
   continue;

  if (strcmp(lang, otherlang) == 0)
   found = true;
 }

 if (!found)
  ereport(elevel,
    (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
     errmsg("ICU locale \"%s\" has unknown language \"%s\"",
      loc_str, lang),
     errhint("To disable ICU locale validation, set the parameter \"%s\" to \"%s\".",
       "icu_validation_level", "disabled")));

 /* check that it can be opened */
 collator = pg_ucol_open(loc_str);
 ucol_close(collator);
#else       /* not USE_ICU */
 /* could get here if a collation was created by a build with ICU */
 ereport(ERROR,
   (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
    errmsg("ICU is not supported in this build")));
#endif       /* not USE_ICU */
}