Arrange to set the LC_XXX environment variables to match our locale setup.

Back-patch of previous fix in HEAD for plperl-vs-locale issue.

diff --git a/src/backend/access/transam/xlog.c b/src/backend/access/transam/xlog.c
index 8b086b78f3cafe0e846726f484e46a971dbc1ae9..28016c35a1ee006f9a8c721f68f06fd8b2f4fc96 100644 (file)
--- a/src/backend/access/transam/xlog.c
+++ b/src/backend/access/transam/xlog.c
@@ -7,7 +7,7 @@
  * Portions Copyright (c) 1996-2005, PostgreSQL Global Development Group
  * Portions Copyright (c) 1994, Regents of the University of California
  *
- * $PostgreSQL: pgsql/src/backend/access/transam/xlog.c,v 1.180.4.3 2005/05/31 19:10:39 tgl Exp $
+ * $PostgreSQL: pgsql/src/backend/access/transam/xlog.c,v 1.180.4.4 2006/01/05 00:55:06 tgl Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -41,6 +41,7 @@
 #include "storage/spin.h"
 #include "utils/builtins.h"
 #include "utils/guc.h"
+#include "utils/pg_locale.h"
 #include "utils/relcache.h"
 
 
@@ -3347,14 +3348,14 @@ ReadControlFile(void)
              " but the server was compiled with LOCALE_NAME_BUFLEN %d.",
                           ControlFile->localeBuflen, LOCALE_NAME_BUFLEN),
             errhint("It looks like you need to recompile or initdb.")));
-   if (setlocale(LC_COLLATE, ControlFile->lc_collate) == NULL)
+   if (pg_perm_setlocale(LC_COLLATE, ControlFile->lc_collate) == NULL)
        ereport(FATAL,
        (errmsg("database files are incompatible with operating system"),
         errdetail("The database cluster was initialized with LC_COLLATE \"%s\","
                   " which is not recognized by setlocale().",
                   ControlFile->lc_collate),
         errhint("It looks like you need to initdb or install locale support.")));
-   if (setlocale(LC_CTYPE, ControlFile->lc_ctype) == NULL)
+   if (pg_perm_setlocale(LC_CTYPE, ControlFile->lc_ctype) == NULL)
        ereport(FATAL,
        (errmsg("database files are incompatible with operating system"),
         errdetail("The database cluster was initialized with LC_CTYPE \"%s\","

diff --git a/src/backend/main/main.c b/src/backend/main/main.c
index e4f112b42160d38f91e822b9ef188c21aa23e17f..abe229c0e15094f999d24279733c7f6529a57474 100644 (file)
--- a/src/backend/main/main.c
+++ b/src/backend/main/main.c
@@ -13,13 +13,12 @@
  *
  *
  * IDENTIFICATION
- *   $PostgreSQL: pgsql/src/backend/main/main.c,v 1.94 2004/12/31 21:59:53 pgsql Exp $
+ *   $PostgreSQL: pgsql/src/backend/main/main.c,v 1.94.4.1 2006/01/05 00:55:07 tgl Exp $
  *
  *-------------------------------------------------------------------------
  */
 #include "postgres.h"
 
-#include <errno.h>
 #include <pwd.h>
 #include <unistd.h>
 
@@ -40,6 +39,7 @@
 #include "postmaster/postmaster.h"
 #include "tcop/tcopprot.h"
 #include "utils/help_config.h"
+#include "utils/pg_locale.h"
 #include "utils/ps_status.h"
 #ifdef WIN32
 #include "libpq/pqsignal.h"
@@ -167,30 +167,37 @@ main(int argc, char *argv[])
     */
 
    if ((env_locale = getenv("LC_COLLATE")) != NULL)
-       setlocale(LC_COLLATE, env_locale);
+       pg_perm_setlocale(LC_COLLATE, env_locale);
    else
-       setlocale(LC_COLLATE, "");
+       pg_perm_setlocale(LC_COLLATE, "");
 
    if ((env_locale = getenv("LC_CTYPE")) != NULL)
-       setlocale(LC_CTYPE, env_locale);
+       pg_perm_setlocale(LC_CTYPE, env_locale);
    else
-       setlocale(LC_CTYPE, "");
+       pg_perm_setlocale(LC_CTYPE, "");
 #else
-   setlocale(LC_COLLATE, "");
-   setlocale(LC_CTYPE, "");
+   pg_perm_setlocale(LC_COLLATE, "");
+   pg_perm_setlocale(LC_CTYPE, "");
 #endif
 
 #ifdef LC_MESSAGES
-   setlocale(LC_MESSAGES, "");
+   pg_perm_setlocale(LC_MESSAGES, "");
 #endif
 
    /*
     * We keep these set to "C" always, except transiently in pg_locale.c;
     * see that file for explanations.
     */
-   setlocale(LC_MONETARY, "C");
-   setlocale(LC_NUMERIC, "C");
-   setlocale(LC_TIME, "C");
+   pg_perm_setlocale(LC_MONETARY, "C");
+   pg_perm_setlocale(LC_NUMERIC, "C");
+   pg_perm_setlocale(LC_TIME, "C");
+
+   /*
+    * Now that we have absorbed as much as we wish to from the locale
+    * environment, remove any LC_ALL setting, so that the environment
+    * variables installed by pg_perm_setlocale have force.
+    */
+   unsetenv("LC_ALL");
 
    /*
     * Skip permission checks if we're just trying to do --help or

diff --git a/src/backend/utils/adt/pg_locale.c b/src/backend/utils/adt/pg_locale.c
index f2d5f2c5cef146df1f05bc3d84379c008e2b0924..d65d0957d6ae98db201465bd0c458143422e14ac 100644 (file)
--- a/src/backend/utils/adt/pg_locale.c
+++ b/src/backend/utils/adt/pg_locale.c
@@ -4,7 +4,7 @@
  *
  * Portions Copyright (c) 2002-2005, PostgreSQL Global Development Group
  *
- * $PostgreSQL: pgsql/src/backend/utils/adt/pg_locale.c,v 1.30.4.1 2005/03/16 00:03:02 momjian Exp $
+ * $PostgreSQL: pgsql/src/backend/utils/adt/pg_locale.c,v 1.30.4.2 2006/01/05 00:55:07 tgl Exp $
  *
  *-----------------------------------------------------------------------
  */
@@ -50,13 +50,10 @@
 
 #include <locale.h>
 
+#include "catalog/pg_control.h"
 #include "utils/pg_locale.h"
 
 
-/* indicated whether locale information cache is valid */
-static bool CurrentLocaleConvValid = false;
-
-
 /* GUC storage area */
 
 char      *locale_messages;
@@ -64,6 +61,120 @@ char       *locale_monetary;
 char      *locale_numeric;
 char      *locale_time;
 
+/* indicates whether locale information cache is valid */
+static bool CurrentLocaleConvValid = false;
+
+/* Environment variable storage area */
+
+#define LC_ENV_BUFSIZE (LOCALE_NAME_BUFLEN + 20)
+
+static char lc_collate_envbuf[LC_ENV_BUFSIZE];
+static char lc_ctype_envbuf[LC_ENV_BUFSIZE];
+#ifdef LC_MESSAGES
+static char lc_messages_envbuf[LC_ENV_BUFSIZE];
+#endif
+static char lc_monetary_envbuf[LC_ENV_BUFSIZE];
+static char lc_numeric_envbuf[LC_ENV_BUFSIZE];
+static char lc_time_envbuf[LC_ENV_BUFSIZE];
+
+
+/*
+ * pg_perm_setlocale
+ *
+ * This is identical to the libc function setlocale(), with the addition
+ * that 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;
+   char   *envbuf;
+
+#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 */
+
+   switch (category)
+   {
+       case LC_COLLATE:
+           envvar = "LC_COLLATE";
+           envbuf = lc_collate_envbuf;
+           break;
+       case LC_CTYPE:
+           envvar = "LC_CTYPE";
+           envbuf = lc_ctype_envbuf;
+           break;
+#ifdef LC_MESSAGES
+       case LC_MESSAGES:
+           envvar = "LC_MESSAGES";
+           envbuf = lc_messages_envbuf;
+           break;
+#endif
+       case LC_MONETARY:
+           envvar = "LC_MONETARY";
+           envbuf = lc_monetary_envbuf;
+           break;
+       case LC_NUMERIC:
+           envvar = "LC_NUMERIC";
+           envbuf = lc_numeric_envbuf;
+           break;
+       case LC_TIME:
+           envvar = "LC_TIME";
+           envbuf = lc_time_envbuf;
+           break;
+       default:
+           elog(FATAL, "unrecognized LC category: %d", category);
+           envvar = NULL;      /* keep compiler quiet */
+           envbuf = NULL;
+           break;
+   }
+
+   snprintf(envbuf, LC_ENV_BUFSIZE-1, "%s=%s", envvar, result);
+
+#ifndef WIN32
+   if (putenv(envbuf))
+       return NULL;
+#else
+   /*
+    * On Windows, we need to modify both the process environment and the
+    * cached version in msvcrt
+    */
+   if (!SetEnvironmentVariable(envvar, result))
+       return NULL;
+   if (_putenv(envbuf))
+       return NULL;
+#endif
+
+   return result;
+}
+
 
 /* GUC assign hooks */
 
@@ -123,49 +234,25 @@ locale_time_assign(const char *value, bool doit, GucSource source)
 const char *
 locale_messages_assign(const char *value, bool doit, GucSource source)
 {
-#ifndef WIN32
    /*
     * LC_MESSAGES category does not exist everywhere, but accept it
     * anyway
+    *
+    * On Windows, we can't even check the value, so the non-doit case
+    * is a no-op
     */
 #ifdef LC_MESSAGES
    if (doit)
    {
-       if (!setlocale(LC_MESSAGES, value))
+       if (!pg_perm_setlocale(LC_MESSAGES, value))
            return NULL;
    }
+#ifndef WIN32
    else
        value = locale_xxx_assign(LC_MESSAGES, value, false, source);
+#endif   /* WIN32 */
 #endif   /* LC_MESSAGES */
    return value;
-
-#else /* WIN32 */
-
-   /*
-    * Win32 does not have working setlocale() for LC_MESSAGES. We can only
-    * use environment variables to change it (per gettext FAQ).  This
-    * means we can't actually check the supplied value, so always assume
-    * it's good.  Also, ignore attempts to set to "", which really means
-    * "keep using the old value".  (Actually it means "use the environment
-    * value", but we are too lazy to try to implement that exactly.)
-    */
-   if (doit && value[0])
-   {
-       /*
-        * We need to modify both the process environment and the cached
-        * version in msvcrt
-        */
-       static char env[128];
-
-       if (!SetEnvironmentVariable("LC_MESSAGES", value))
-           return NULL;
-
-       snprintf(env, sizeof(env)-1, "LC_MESSAGES=%s", value);
-       if (_putenv(env))
-           return NULL;
-   }
-   return value;
-#endif /* WIN32 */
 }
 
 

diff --git a/src/include/utils/pg_locale.h b/src/include/utils/pg_locale.h
index 47d6b38b30fa8599d2de370e3a4a510f81b56e34..577ba124851fd0fa00d95755db17a2ccea748845 100644 (file)
--- a/src/include/utils/pg_locale.h
+++ b/src/include/utils/pg_locale.h
@@ -2,7 +2,7 @@
  *
  * PostgreSQL locale utilities
  *
- * $PostgreSQL: pgsql/src/include/utils/pg_locale.h,v 1.19.4.1 2005/03/16 00:03:02 momjian Exp $
+ * $PostgreSQL: pgsql/src/include/utils/pg_locale.h,v 1.19.4.2 2006/01/05 00:55:07 tgl Exp $
  *
  * Copyright (c) 2002-2005, PostgreSQL Global Development Group
  *
@@ -31,6 +31,8 @@ extern const char *locale_numeric_assign(const char *value,
 extern const char *locale_time_assign(const char *value,
                   bool doit, GucSource source);
 
+extern char *pg_perm_setlocale(int category, const char *locale);
+
 extern bool lc_collate_is_c(void);
 extern bool lc_ctype_is_c(void);