Support having multiple Unix-domain sockets per postmaster.
authorTom Lane <tgl@sss.pgh.pa.us>
Fri, 10 Aug 2012 21:26:44 +0000 (17:26 -0400)
committerTom Lane <tgl@sss.pgh.pa.us>
Fri, 10 Aug 2012 21:27:15 +0000 (17:27 -0400)
Replace unix_socket_directory with unix_socket_directories, which is a list
of socket directories, and adjust postmaster's code to allow zero or more
Unix-domain sockets to be created.

This is mostly a straightforward change, but since the Unix sockets ought
to be created after the TCP/IP sockets for safety reasons (better chance
of detecting a port number conflict), AddToDataDirLockFile needs to be
fixed to support out-of-order updates of data directory lockfile lines.
That's a change that had been foreseen to be necessary someday anyway.

Honza Horak, reviewed and revised by Tom Lane

17 files changed:
doc/src/sgml/client-auth.sgml
doc/src/sgml/config.sgml
doc/src/sgml/ref/postgres-ref.sgml
doc/src/sgml/runtime.sgml
src/backend/libpq/pqcomm.c
src/backend/postmaster/postmaster.c
src/backend/tcop/postgres.c
src/backend/utils/adt/varlena.c
src/backend/utils/init/miscinit.c
src/backend/utils/misc/guc.c
src/backend/utils/misc/postgresql.conf.sample
src/bin/initdb/initdb.c
src/bin/pg_ctl/pg_ctl.c
src/include/libpq/libpq.h
src/include/miscadmin.h
src/include/postmaster/postmaster.h
src/include/utils/builtins.h

index cfdb33a1fd6508aac31b0a87b8bb3ed73432c618..679c40a64ec64fbc0623e878954287a5e6081dc4 100644 (file)
@@ -838,7 +838,7 @@ omicron         bryanh                  guest1
     <varname>unix_socket_permissions</varname> (and possibly
     <varname>unix_socket_group</varname>) configuration parameters as
     described in <xref linkend="runtime-config-connection">.  Or you
-    could set the <varname>unix_socket_directory</varname>
+    could set the <varname>unix_socket_directories</varname>
     configuration parameter to place the socket file in a suitably
     restricted directory.
    </para>
index e30c5a0cf4e8f1d31c93907bd6894a8b38189005..7727ea7f89cf76e9fe7dbd0ced57f18dde387278 100644 (file)
@@ -445,17 +445,24 @@ SET ENABLE_SEQSCAN TO OFF;
       </listitem>
      </varlistentry>
 
-     <varlistentry id="guc-unix-socket-directory" xreflabel="unix_socket_directory">
-      <term><varname>unix_socket_directory</varname> (<type>string</type>)</term>
+     <varlistentry id="guc-unix-socket-directories" xreflabel="unix_socket_directories">
+      <term><varname>unix_socket_directories</varname> (<type>string</type>)</term>
       <indexterm>
-       <primary><varname>unix_socket_directory</> configuration parameter</primary>
+       <primary><varname>unix_socket_directories</> configuration parameter</primary>
       </indexterm>
       <listitem>
        <para>
-        Specifies the directory of the Unix-domain socket on which the
-        server is to listen for
-        connections from client applications.  The default is normally
-        <filename>/tmp</filename>, but can be changed at build time.
+        Specifies the directory of the Unix-domain socket(s) on which the
+        server is to listen for connections from client applications.
+        Multiple sockets can be created by listing multiple directories
+        separated by commas.  Whitespace between entries is
+        ignored; surround a directory name with double quotes if you need
+        to include whitespace or commas in the name.
+        An empty value
+        specifies not listening on any Unix-domain sockets, in which case
+        only TCP/IP sockets can be used to connect to the server.
+        The default value is normally
+        <filename>/tmp</filename>, but that can be changed at build time.
         This parameter can only be set at server start.
        </para>
 
@@ -464,8 +471,8 @@ SET ENABLE_SEQSCAN TO OFF;
         <literal>.s.PGSQL.<replaceable>nnnn</></literal> where
         <replaceable>nnnn</> is the server's port number, an ordinary file
         named <literal>.s.PGSQL.<replaceable>nnnn</>.lock</literal> will be
-        created in the <varname>unix_socket_directory</> directory.  Neither
-        file should ever be removed manually.
+        created in each of the <varname>unix_socket_directories</> directories.
+        Neither file should ever be removed manually.
        </para>
 
        <para>
@@ -482,8 +489,8 @@ SET ENABLE_SEQSCAN TO OFF;
       </indexterm>
       <listitem>
        <para>
-        Sets the owning group of the Unix-domain socket.  (The owning
-        user of the socket is always the user that starts the
+        Sets the owning group of the Unix-domain socket(s).  (The owning
+        user of the sockets is always the user that starts the
         server.)  In combination with the parameter
         <varname>unix_socket_permissions</varname> this can be used as
         an additional access control mechanism for Unix-domain connections.
@@ -506,7 +513,7 @@ SET ENABLE_SEQSCAN TO OFF;
       </indexterm>
       <listitem>
        <para>
-        Sets the access permissions of the Unix-domain socket.  Unix-domain
+        Sets the access permissions of the Unix-domain socket(s).  Unix-domain
         sockets use the usual Unix file system permission set.
         The parameter value is expected to be a numeric mode
         specified in the format accepted by the
@@ -1852,7 +1859,7 @@ SET ENABLE_SEQSCAN TO OFF;
         <varname>commit_delay</varname> behaved differently and was much
         less effective: it affected only commits, rather than all WAL flushes,
         and waited for the entire configured delay even if the WAL flush
-        was completed sooner.  Beginning in <productname>PostgreSQL</> 9.3, 
+        was completed sooner.  Beginning in <productname>PostgreSQL</> 9.3,
         the first process that becomes ready to flush waits for the configured
         interval, while subsequent processes wait only until the leader
         completes the flush.  The default <varname>commit_delay</> is zero
@@ -6556,7 +6563,7 @@ LOG:  CleanUpLock: deleting: lock(0xb7acd844) id(24688,24696,0,0,0,1)
        </row>
        <row>
         <entry><option>-k <replaceable>x</replaceable></option></entry>
-        <entry><literal>unix_socket_directory = <replaceable>x</replaceable></></entry>
+        <entry><literal>unix_socket_directories = <replaceable>x</replaceable></></entry>
        </row>
        <row>
         <entry><option>-l</option></entry>
index 4e5cd02589c8ebdcdfb68afdb06a492f24e06de6..a1f36e1836cc040ff2e862da0696429e766d81f5 100644 (file)
@@ -254,8 +254,14 @@ PostgreSQL documentation
        <para>
         Specifies the directory of the Unix-domain socket on which
         <command>postgres</command> is to listen for
-        connections from client applications.  The default is normally
-        <filename>/tmp</filename>, but can be changed at build time.
+        connections from client applications.  The value can also be a
+        comma-separated list of directories.  An empty value
+        specifies not listening on any Unix-domain sockets, in which case
+        only TCP/IP sockets can be used to connect to the server.
+        The default value is normally
+        <filename>/tmp</filename>, but that can be changed at build time.
+        Specifying this option is equivalent to setting the <xref
+        linkend="guc-unix-socket-directories"> configuration parameter.
        </para>
       </listitem>
      </varlistentry>
index 87177989d76f2140ccd34175472c87869b66c320..9cc9d4260a9f69063b14034a6c4a88f740a007d3 100644 (file)
@@ -1718,7 +1718,7 @@ pg_dumpall -p 5432 | psql -d postgres -p 5433
   <para>
    The simplest way to prevent spoofing for <literal>local</>
    connections is to use a Unix domain socket directory (<xref
-   linkend="guc-unix-socket-directory">) that has write permission only
+   linkend="guc-unix-socket-directories">) that has write permission only
    for a trusted local user.  This prevents a malicious user from creating
    their own socket file in that directory.  If you are concerned that
    some applications might still reference <filename>/tmp</> for the
index 5272811cc0af190803666cd32dc7ac7167c12c77..5e86987f221fee01c5049f925492e0f9c441d372 100644 (file)
@@ -42,7 +42,7 @@
  *     StreamServerPort    - Open postmaster's server port
  *     StreamConnection    - Create new connection with client
  *     StreamClose         - Close a client/backend connection
- *     TouchSocketFile     - Protect socket file against /tmp cleaners
+ *     TouchSocketFiles    - Protect socket files against /tmp cleaners
  *     pq_init         - initialize libpq at backend startup
  *     pq_comm_reset   - reset libpq during error recovery
  *     pq_close        - shutdown libpq at backend exit
@@ -103,8 +103,8 @@ int         Unix_socket_permissions;
 char      *Unix_socket_group;
 
 
-/* Where the Unix socket file is */
-static char sock_path[MAXPGPATH];
+/* Where the Unix socket files are (list of palloc'd strings) */
+static List *sock_paths = NIL;
 
 
 /*
@@ -140,8 +140,8 @@ static int  internal_flush(void);
 static void pq_set_nonblocking(bool nonblocking);
 
 #ifdef HAVE_UNIX_SOCKETS
-static int Lock_AF_UNIX(unsigned short portNumber, char *unixSocketName);
-static int Setup_AF_UNIX(void);
+static int Lock_AF_UNIX(char *unixSocketDir, char *unixSocketPath);
+static int Setup_AF_UNIX(char *sock_path);
 #endif   /* HAVE_UNIX_SOCKETS */
 
 
@@ -234,29 +234,43 @@ pq_close(int code, Datum arg)
 
 /* StreamDoUnlink()
  * Shutdown routine for backend connection
- * If a Unix socket is used for communication, explicitly close it.
+ * If any Unix sockets are used for communication, explicitly close them.
  */
 #ifdef HAVE_UNIX_SOCKETS
 static void
 StreamDoUnlink(int code, Datum arg)
 {
-   Assert(sock_path[0]);
-   unlink(sock_path);
+   ListCell   *l;
+
+   /* Loop through all created sockets... */
+   foreach(l, sock_paths)
+   {
+       char       *sock_path = (char *) lfirst(l);
+
+       unlink(sock_path);
+   }
+   /* Since we're about to exit, no need to reclaim storage */
+   sock_paths = NIL;
 }
 #endif   /* HAVE_UNIX_SOCKETS */
 
 /*
  * StreamServerPort -- open a "listening" port to accept connections.
  *
- * Successfully opened sockets are added to the ListenSocket[] array,
- * at the first position that isn't PGINVALID_SOCKET.
+ * family should be AF_UNIX or AF_UNSPEC; portNumber is the port number.
+ * For AF_UNIX ports, hostName should be NULL and unixSocketDir must be
+ * specified.  For TCP ports, hostName is either NULL for all interfaces or
+ * the interface to listen on, and unixSocketDir is ignored (can be NULL).
+ *
+ * Successfully opened sockets are added to the ListenSocket[] array (of
+ * length MaxListen), at the first position that isn't PGINVALID_SOCKET.
  *
  * RETURNS: STATUS_OK or STATUS_ERROR
  */
 
 int
 StreamServerPort(int family, char *hostName, unsigned short portNumber,
-                char *unixSocketName,
+                char *unixSocketDir,
                 pgsocket ListenSocket[], int MaxListen)
 {
    pgsocket    fd;
@@ -273,6 +287,9 @@ StreamServerPort(int family, char *hostName, unsigned short portNumber,
    int         listen_index = 0;
    int         added = 0;
 
+#ifdef HAVE_UNIX_SOCKETS
+   char        unixSocketPath[MAXPGPATH];
+#endif
 #if !defined(WIN32) || defined(IPV6_V6ONLY)
    int         one = 1;
 #endif
@@ -286,10 +303,14 @@ StreamServerPort(int family, char *hostName, unsigned short portNumber,
 #ifdef HAVE_UNIX_SOCKETS
    if (family == AF_UNIX)
    {
-       /* Lock_AF_UNIX will also fill in sock_path. */
-       if (Lock_AF_UNIX(portNumber, unixSocketName) != STATUS_OK)
+       /*
+        * Create unixSocketPath from portNumber and unixSocketDir and lock
+        * that file path
+        */
+       UNIXSOCK_PATH(unixSocketPath, portNumber, unixSocketDir);
+       if (Lock_AF_UNIX(unixSocketDir, unixSocketPath) != STATUS_OK)
            return STATUS_ERROR;
-       service = sock_path;
+       service = unixSocketPath;
    }
    else
 #endif   /* HAVE_UNIX_SOCKETS */
@@ -432,7 +453,7 @@ StreamServerPort(int family, char *hostName, unsigned short portNumber,
                     (IS_AF_UNIX(addr->ai_family)) ?
                  errhint("Is another postmaster already running on port %d?"
                          " If not, remove socket file \"%s\" and retry.",
-                         (int) portNumber, sock_path) :
+                         (int) portNumber, service) :
                  errhint("Is another postmaster already running on port %d?"
                          " If not, wait a few seconds and retry.",
                          (int) portNumber)));
@@ -443,7 +464,7 @@ StreamServerPort(int family, char *hostName, unsigned short portNumber,
 #ifdef HAVE_UNIX_SOCKETS
        if (addr->ai_family == AF_UNIX)
        {
-           if (Setup_AF_UNIX() != STATUS_OK)
+           if (Setup_AF_UNIX(service) != STATUS_OK)
            {
                closesocket(fd);
                break;
@@ -490,10 +511,8 @@ StreamServerPort(int family, char *hostName, unsigned short portNumber,
  * Lock_AF_UNIX -- configure unix socket file path
  */
 static int
-Lock_AF_UNIX(unsigned short portNumber, char *unixSocketName)
+Lock_AF_UNIX(char *unixSocketDir, char *unixSocketPath)
 {
-   UNIXSOCK_PATH(sock_path, portNumber, unixSocketName);
-
    /*
     * Grab an interlock file associated with the socket file.
     *
@@ -502,13 +521,23 @@ Lock_AF_UNIX(unsigned short portNumber, char *unixSocketName)
     * more portable, and second, it lets us remove any pre-existing socket
     * file without race conditions.
     */
-   CreateSocketLockFile(sock_path, true);
+   CreateSocketLockFile(unixSocketPath, true, unixSocketDir);
 
    /*
     * Once we have the interlock, we can safely delete any pre-existing
     * socket file to avoid failure at bind() time.
     */
-   unlink(sock_path);
+   unlink(unixSocketPath);
+
+   /*
+    * Arrange to unlink the socket file(s) at proc_exit.  If this is the
+    * first one, set up the on_proc_exit function to do it; then add this
+    * socket file to the list of files to unlink.
+    */
+   if (sock_paths == NIL)
+       on_proc_exit(StreamDoUnlink, 0);
+
+   sock_paths = lappend(sock_paths, pstrdup(unixSocketPath));
 
    return STATUS_OK;
 }
@@ -518,11 +547,8 @@ Lock_AF_UNIX(unsigned short portNumber, char *unixSocketName)
  * Setup_AF_UNIX -- configure unix socket permissions
  */
 static int
-Setup_AF_UNIX(void)
+Setup_AF_UNIX(char *sock_path)
 {
-   /* Arrange to unlink the socket file at exit */
-   on_proc_exit(StreamDoUnlink, 0);
-
    /*
     * Fix socket ownership/permission if requested.  Note we must do this
     * before we listen() to avoid a window where unwanted connections could
@@ -704,20 +730,24 @@ StreamClose(pgsocket sock)
 }
 
 /*
- * TouchSocketFile -- mark socket file as recently accessed
+ * TouchSocketFiles -- mark socket files as recently accessed
  *
  * This routine should be called every so often to ensure that the socket
- * file has a recent mod date (ordinary operations on sockets usually won't
- * change the mod date).  That saves it from being removed by
+ * files have a recent mod date (ordinary operations on sockets usually won't
+ * change the mod date).  That saves them from being removed by
  * overenthusiastic /tmp-directory-cleaner daemons.  (Another reason we should
  * never have put the socket file in /tmp...)
  */
 void
-TouchSocketFile(void)
+TouchSocketFiles(void)
 {
-   /* Do nothing if we did not create a socket... */
-   if (sock_path[0] != '\0')
+   ListCell   *l;
+
+   /* Loop through all created sockets... */
+   foreach(l, sock_paths)
    {
+       char       *sock_path = (char *) lfirst(l);
+
        /*
         * utime() is POSIX standard, utimes() is a common alternative. If we
         * have neither, there's no way to affect the mod or access time of
index fee699ea9ef2da504c5ce2b4d9b2284ea37eef06..9e26fb895f3f6c70eef42f69f4f4f283cb15ba65 100644 (file)
@@ -157,7 +157,9 @@ static Backend *ShmemBackendArray;
 
 /* The socket number we are listening for connections on */
 int            PostPortNumber;
-char      *UnixSocketDir;
+/* The directory names for Unix socket(s) */
+char      *Unix_socket_directories;
+/* The TCP listen address(es) */
 char      *ListenAddresses;
 
 /*
@@ -612,7 +614,7 @@ PostmasterMain(int argc, char *argv[])
                break;
 
            case 'k':
-               SetConfigOption("unix_socket_directory", optarg, PGC_POSTMASTER, PGC_S_ARGV);
+               SetConfigOption("unix_socket_directories", optarg, PGC_POSTMASTER, PGC_S_ARGV);
                break;
 
            case 'l':
@@ -856,7 +858,7 @@ PostmasterMain(int argc, char *argv[])
        /* Need a modifiable copy of ListenAddresses */
        rawstring = pstrdup(ListenAddresses);
 
-       /* Parse string into list of identifiers */
+       /* Parse string into list of hostnames */
        if (!SplitIdentifierString(rawstring, ',', &elemlist))
        {
            /* syntax error in list */
@@ -872,12 +874,12 @@ PostmasterMain(int argc, char *argv[])
            if (strcmp(curhost, "*") == 0)
                status = StreamServerPort(AF_UNSPEC, NULL,
                                          (unsigned short) PostPortNumber,
-                                         UnixSocketDir,
+                                         NULL,
                                          ListenSocket, MAXLISTEN);
            else
                status = StreamServerPort(AF_UNSPEC, curhost,
                                          (unsigned short) PostPortNumber,
-                                         UnixSocketDir,
+                                         NULL,
                                          ListenSocket, MAXLISTEN);
 
            if (status == STATUS_OK)
@@ -896,7 +898,7 @@ PostmasterMain(int argc, char *argv[])
                                curhost)));
        }
 
-       if (!success && list_length(elemlist))
+       if (!success && elemlist != NIL)
            ereport(FATAL,
                    (errmsg("could not create any TCP/IP sockets")));
 
@@ -943,13 +945,54 @@ PostmasterMain(int argc, char *argv[])
 #endif
 
 #ifdef HAVE_UNIX_SOCKETS
-   status = StreamServerPort(AF_UNIX, NULL,
-                             (unsigned short) PostPortNumber,
-                             UnixSocketDir,
-                             ListenSocket, MAXLISTEN);
-   if (status != STATUS_OK)
-       ereport(WARNING,
-               (errmsg("could not create Unix-domain socket")));
+   if (Unix_socket_directories)
+   {
+       char       *rawstring;
+       List       *elemlist;
+       ListCell   *l;
+       int         success = 0;
+
+       /* Need a modifiable copy of Unix_socket_directories */
+       rawstring = pstrdup(Unix_socket_directories);
+
+       /* Parse string into list of directories */
+       if (!SplitDirectoriesString(rawstring, ',', &elemlist))
+       {
+           /* syntax error in list */
+           ereport(FATAL,
+                   (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
+                    errmsg("invalid list syntax for \"unix_socket_directories\"")));
+       }
+
+       foreach(l, elemlist)
+       {
+           char       *socketdir = (char *) lfirst(l);
+
+           status = StreamServerPort(AF_UNIX, NULL,
+                                     (unsigned short) PostPortNumber,
+                                     socketdir,
+                                     ListenSocket, MAXLISTEN);
+
+           if (status == STATUS_OK)
+           {
+               success++;
+               /* record the first successful Unix socket in lockfile */
+               if (success == 1)
+                   AddToDataDirLockFile(LOCK_FILE_LINE_SOCKET_DIR, socketdir);
+           }
+           else
+               ereport(WARNING,
+                       (errmsg("could not create Unix-domain socket in directory \"%s\"",
+                               socketdir)));
+       }
+
+       if (!success && elemlist != NIL)
+           ereport(FATAL,
+                   (errmsg("could not create any Unix-domain sockets")));
+
+       list_free_deep(elemlist);
+       pfree(rawstring);
+   }
 #endif
 
    /*
@@ -1439,15 +1482,15 @@ ServerLoop(void)
        }
 
        /*
-        * Touch the socket and lock file every 58 minutes, to ensure that
+        * Touch Unix socket and lock files every 58 minutes, to ensure that
         * they are not removed by overzealous /tmp-cleaning tasks.  We assume
         * no one runs cleaners with cutoff times of less than an hour ...
         */
        now = time(NULL);
        if (now - last_touch_time >= 58 * SECS_PER_MINUTE)
        {
-           TouchSocketFile();
-           TouchSocketLockFile();
+           TouchSocketFiles();
+           TouchSocketLockFiles();
            last_touch_time = now;
        }
    }
index 37dfa18c1d0b793d096f662597b4b0a2687a1fd6..f1248a851bf90188da8d3a7e8b61ac99bf78ebbd 100644 (file)
@@ -3344,7 +3344,7 @@ process_postgres_switches(int argc, char *argv[], GucContext ctx)
                break;
 
            case 'k':
-               SetConfigOption("unix_socket_directory", optarg, ctx, gucsource);
+               SetConfigOption("unix_socket_directories", optarg, ctx, gucsource);
                break;
 
            case 'l':
index e1b57ba3fc71fa2cb3898503fa380ff4bc21861b..d9e6bc4338b06dd2d2a1a48ffaaeefe32f4f294d 100644 (file)
@@ -2446,6 +2446,116 @@ SplitIdentifierString(char *rawstring, char separator,
 }
 
 
+/*
+ * SplitDirectoriesString --- parse a string containing directory names
+ *
+ * This is similar to SplitIdentifierString, except that the parsing
+ * rules are meant to handle pathnames instead of identifiers: there is
+ * no downcasing, the max length is MAXPGPATH-1, and we apply
+ * canonicalize_path() to each extracted string.  Because of the last,
+ * the returned strings are separately palloc'd rather than being
+ * pointers into rawstring --- but we still scribble on rawstring.
+ *
+ * Inputs:
+ * rawstring: the input string; must be modifiable!
+ * separator: the separator punctuation expected between directories
+ *            (typically ',' or ';').  Whitespace may also appear around
+ *            directories.
+ * Outputs:
+ * namelist: filled with a palloc'd list of directory names.
+ *           Caller should list_free_deep() this even on error return.
+ *
+ * Returns TRUE if okay, FALSE if there is a syntax error in the string.
+ *
+ * Note that an empty string is considered okay here.
+ */
+bool
+SplitDirectoriesString(char *rawstring, char separator,
+                      List **namelist)
+{
+   char       *nextp = rawstring;
+   bool        done = false;
+
+   *namelist = NIL;
+
+   while (isspace((unsigned char) *nextp))
+       nextp++;                /* skip leading whitespace */
+
+   if (*nextp == '\0')
+       return true;            /* allow empty string */
+
+   /* At the top of the loop, we are at start of a new directory. */
+   do
+   {
+       char       *curname;
+       char       *endp;
+
+       if (*nextp == '\"')
+       {
+           /* Quoted name --- collapse quote-quote pairs */
+           curname = nextp + 1;
+           for (;;)
+           {
+               endp = strchr(nextp + 1, '\"');
+               if (endp == NULL)
+                   return false;       /* mismatched quotes */
+               if (endp[1] != '\"')
+                   break;      /* found end of quoted name */
+               /* Collapse adjacent quotes into one quote, and look again */
+               memmove(endp, endp + 1, strlen(endp));
+               nextp = endp;
+           }
+           /* endp now points at the terminating quote */
+           nextp = endp + 1;
+       }
+       else
+       {
+           /* Unquoted name --- extends to separator or whitespace */
+           curname = nextp;
+           while (*nextp && *nextp != separator &&
+                  !isspace((unsigned char) *nextp))
+               nextp++;
+           endp = nextp;
+           if (curname == nextp)
+               return false;   /* empty unquoted name not allowed */
+       }
+
+       while (isspace((unsigned char) *nextp))
+           nextp++;            /* skip trailing whitespace */
+
+       if (*nextp == separator)
+       {
+           nextp++;
+           while (isspace((unsigned char) *nextp))
+               nextp++;        /* skip leading whitespace for next */
+           /* we expect another name, so done remains false */
+       }
+       else if (*nextp == '\0')
+           done = true;
+       else
+           return false;       /* invalid syntax */
+
+       /* Now safe to overwrite separator with a null */
+       *endp = '\0';
+
+       /* Truncate path if it's overlength */
+       if (strlen(curname) >= MAXPGPATH)
+           curname[MAXPGPATH - 1] = '\0';
+
+       /*
+        * Finished isolating current name --- add it to list
+        */
+       curname = pstrdup(curname);
+       canonicalize_path(curname);
+       *namelist = lappend(*namelist, curname);
+
+       /* Loop back if we didn't reach end of string */
+   } while (!done);
+
+   return true;
+}
+
+
 /*****************************************************************************
  * Comparison Functions used for bytea
  *
index fb376a0d271730f2813bb1d1f2e15562e82ff865..775d71f56c57f4c954054653b454270427fc5395 100644 (file)
@@ -49,8 +49,8 @@
 
 ProcessingMode Mode = InitProcessing;
 
-/* Note: we rely on this to initialize as zeroes */
-static char socketLockFile[MAXPGPATH];
+/* List of lock files to be removed at proc exit */
+static List *lock_files = NIL;
 
 
 /* ----------------------------------------------------------------
@@ -628,7 +628,7 @@ GetUserNameFromId(Oid roleid)
  *             Interlock-file support
  *
  * These routines are used to create both a data-directory lockfile
- * ($DATADIR/postmaster.pid) and a Unix-socket-file lockfile ($SOCKFILE.lock).
+ * ($DATADIR/postmaster.pid) and Unix-socket-file lockfiles ($SOCKFILE.lock).
  * Both kinds of files contain the same info initially, although we can add
  * more information to a data-directory lockfile after it's created, using
  * AddToDataDirLockFile(). See miscadmin.h for documentation of the contents
@@ -640,32 +640,35 @@ GetUserNameFromId(Oid roleid)
  */
 
 /*
- * proc_exit callback to remove a lockfile.
+ * proc_exit callback to remove lockfiles.
  */
 static void
-UnlinkLockFile(int status, Datum filename)
+UnlinkLockFiles(int status, Datum arg)
 {
-   char       *fname = (char *) DatumGetPointer(filename);
+   ListCell   *l;
 
-   if (fname != NULL)
+   foreach(l, lock_files)
    {
-       if (unlink(fname) != 0)
-       {
-           /* Should we complain if the unlink fails? */
-       }
-       free(fname);
+       char       *curfile = (char *) lfirst(l);
+
+       unlink(curfile);
+       /* Should we complain if the unlink fails? */
    }
+   /* Since we're about to exit, no need to reclaim storage */
+   lock_files = NIL;
 }
 
 /*
  * Create a lockfile.
  *
- * filename is the name of the lockfile to create.
+ * filename is the path name of the lockfile to create.
  * amPostmaster is used to determine how to encode the output PID.
+ * socketDir is the Unix socket directory path to include (possibly empty).
  * isDDLock and refName are used to determine what error message to produce.
  */
 static void
 CreateLockFile(const char *filename, bool amPostmaster,
+              const char *socketDir,
               bool isDDLock, const char *refName)
 {
    int         fd;
@@ -891,12 +894,7 @@ CreateLockFile(const char *filename, bool amPostmaster,
             DataDir,
             (long) MyStartTime,
             PostPortNumber,
-#ifdef HAVE_UNIX_SOCKETS
-            (*UnixSocketDir != '\0') ? UnixSocketDir : DEFAULT_PGSOCKET_DIR
-#else
-            ""
-#endif
-       );
+            socketDir);
 
    errno = 0;
    if (write(fd, buffer, strlen(buffer)) != strlen(buffer))
@@ -934,9 +932,14 @@ CreateLockFile(const char *filename, bool amPostmaster,
    }
 
    /*
-    * Arrange for automatic removal of lockfile at proc_exit.
+    * Arrange to unlink the lock file(s) at proc_exit.  If this is the
+    * first one, set up the on_proc_exit function to do it; then add this
+    * lock file to the list of files to unlink.
     */
-   on_proc_exit(UnlinkLockFile, PointerGetDatum(strdup(filename)));
+   if (lock_files == NIL)
+       on_proc_exit(UnlinkLockFiles, 0);
+
+   lock_files = lappend(lock_files, pstrdup(filename));
 }
 
 /*
@@ -945,41 +948,50 @@ CreateLockFile(const char *filename, bool amPostmaster,
  * When this is called, we must have already switched the working
  * directory to DataDir, so we can just use a relative path.  This
  * helps ensure that we are locking the directory we should be.
+ *
+ * Note that the socket directory path line is initially written as empty.
+ * postmaster.c will rewrite it upon creating the first Unix socket.
  */
 void
 CreateDataDirLockFile(bool amPostmaster)
 {
-   CreateLockFile(DIRECTORY_LOCK_FILE, amPostmaster, true, DataDir);
+   CreateLockFile(DIRECTORY_LOCK_FILE, amPostmaster, "", true, DataDir);
 }
 
 /*
  * Create a lockfile for the specified Unix socket file.
  */
 void
-CreateSocketLockFile(const char *socketfile, bool amPostmaster)
+CreateSocketLockFile(const char *socketfile, bool amPostmaster,
+                    const char *socketDir)
 {
    char        lockfile[MAXPGPATH];
 
    snprintf(lockfile, sizeof(lockfile), "%s.lock", socketfile);
-   CreateLockFile(lockfile, amPostmaster, false, socketfile);
-   /* Save name of lockfile for TouchSocketLockFile */
-   strcpy(socketLockFile, lockfile);
+   CreateLockFile(lockfile, amPostmaster, socketDir, false, socketfile);
 }
 
 /*
- * TouchSocketLockFile -- mark socket lock file as recently accessed
+ * TouchSocketLockFiles -- mark socket lock files as recently accessed
  *
- * This routine should be called every so often to ensure that the lock file
- * has a recent mod or access date.  That saves it
+ * This routine should be called every so often to ensure that the socket
+ * lock files have a recent mod or access date.  That saves them
  * from being removed by overenthusiastic /tmp-directory-cleaner daemons.
  * (Another reason we should never have put the socket file in /tmp...)
  */
 void
-TouchSocketLockFile(void)
+TouchSocketLockFiles(void)
 {
-   /* Do nothing if we did not create a socket... */
-   if (socketLockFile[0] != '\0')
+   ListCell   *l;
+
+   foreach(l, lock_files)
    {
+       char       *socketLockFile = (char *) lfirst(l);
+
+       /* No need to touch the data directory lock file, we trust */
+       if (strcmp(socketLockFile, DIRECTORY_LOCK_FILE) == 0)
+           continue;
+
        /*
         * utime() is POSIX standard, utimes() is a common alternative; if we
         * have neither, fall back to actually reading the file (which only
@@ -1011,8 +1023,10 @@ TouchSocketLockFile(void)
  * Add (or replace) a line in the data directory lock file.
  * The given string should not include a trailing newline.
  *
- * Caution: this erases all following lines.  In current usage that is OK
- * because lines are added in order.  We could improve it if needed.
+ * Note: because we don't truncate the file, if we were to rewrite a line
+ * with less data than it had before, there would be garbage after the last
+ * line.  We don't ever actually do that, so not worth adding another kernel
+ * call to cover the possibility.
  */
 void
 AddToDataDirLockFile(int target_line, const char *str)
@@ -1020,8 +1034,10 @@ AddToDataDirLockFile(int target_line, const char *str)
    int         fd;
    int         len;
    int         lineno;
-   char       *ptr;
-   char        buffer[BLCKSZ];
+   char       *srcptr;
+   char       *destptr;
+   char        srcbuffer[BLCKSZ];
+   char        destbuffer[BLCKSZ];
 
    fd = open(DIRECTORY_LOCK_FILE, O_RDWR | PG_BINARY, 0);
    if (fd < 0)
@@ -1032,7 +1048,7 @@ AddToDataDirLockFile(int target_line, const char *str)
                        DIRECTORY_LOCK_FILE)));
        return;
    }
-   len = read(fd, buffer, sizeof(buffer) - 1);
+   len = read(fd, srcbuffer, sizeof(srcbuffer) - 1);
    if (len < 0)
    {
        ereport(LOG,
@@ -1042,36 +1058,50 @@ AddToDataDirLockFile(int target_line, const char *str)
        close(fd);
        return;
    }
-   buffer[len] = '\0';
+   srcbuffer[len] = '\0';
 
    /*
-    * Skip over lines we are not supposed to rewrite.
+    * Advance over lines we are not supposed to rewrite, then copy them
+    * to destbuffer.
     */
-   ptr = buffer;
+   srcptr = srcbuffer;
    for (lineno = 1; lineno < target_line; lineno++)
    {
-       if ((ptr = strchr(ptr, '\n')) == NULL)
+       if ((srcptr = strchr(srcptr, '\n')) == NULL)
        {
            elog(LOG, "bogus data in \"%s\"", DIRECTORY_LOCK_FILE);
            close(fd);
            return;
        }
-       ptr++;
+       srcptr++;
    }
+   memcpy(destbuffer, srcbuffer, srcptr - srcbuffer);
+   destptr = destbuffer + (srcptr - srcbuffer);
 
    /*
     * Write or rewrite the target line.
     */
-   snprintf(ptr, buffer + sizeof(buffer) - ptr, "%s\n", str);
+   snprintf(destptr, destbuffer + sizeof(destbuffer) - destptr, "%s\n", str);
+   destptr += strlen(destptr);
+
+   /*
+    * If there are more lines in the old file, append them to destbuffer.
+    */
+   if ((srcptr = strchr(srcptr, '\n')) != NULL)
+   {
+       srcptr++;
+       snprintf(destptr, destbuffer + sizeof(destbuffer) - destptr, "%s",
+                srcptr);
+   }
 
    /*
     * And rewrite the data.  Since we write in a single kernel call, this
     * update should appear atomic to onlookers.
     */
-   len = strlen(buffer);
+   len = strlen(destbuffer);
    errno = 0;
    if (lseek(fd, (off_t) 0, SEEK_SET) != 0 ||
-       (int) write(fd, buffer, len) != len)
+       (int) write(fd, destbuffer, len) != len)
    {
        /* if write didn't set errno, assume problem is no disk space */
        if (errno == 0)
index 087aaf9b0bb4f09c449c346f4ec7524f7c675472..80e5aa1816f272d526543c9c6267e5353cc6f401 100644 (file)
@@ -2895,14 +2895,18 @@ static struct config_string ConfigureNamesString[] =
    },
 
    {
-       {"unix_socket_directory", PGC_POSTMASTER, CONN_AUTH_SETTINGS,
-           gettext_noop("Sets the directory where the Unix-domain socket will be created."),
+       {"unix_socket_directories", PGC_POSTMASTER, CONN_AUTH_SETTINGS,
+           gettext_noop("Sets the directories where Unix-domain sockets will be created."),
            NULL,
            GUC_SUPERUSER_ONLY
        },
-       &UnixSocketDir,
+       &Unix_socket_directories,
+#ifdef HAVE_UNIX_SOCKETS
+       DEFAULT_PGSOCKET_DIR,
+#else
        "",
-       check_canonical_path, NULL, NULL
+#endif
+       NULL, NULL, NULL
    },
 
    {
index fa75d00f70872bbe7b48192861baf89ec236db98..c24afb043436e46fe0d959ec048aa8b0c7784f0e 100644 (file)
@@ -65,7 +65,8 @@
 # Note:  Increasing max_connections costs ~400 bytes of shared memory per
 # connection slot, plus lock space (see max_locks_per_transaction).
 #superuser_reserved_connections = 3    # (change requires restart)
-#unix_socket_directory = ''        # (change requires restart)
+#unix_socket_directories = '/tmp'  # comma-separated list of directories
+                   # (change requires restart)
 #unix_socket_group = ''            # (change requires restart)
 #unix_socket_permissions = 0777        # begin with 0 to use octal notation
                    # (change requires restart)
index 4292231d0d59f3caa31879e6c0b804587dbfa721..132ad0fa4cdb9b59f052f2d97494910c27b1f219 100644 (file)
@@ -1156,7 +1156,7 @@ static void
 setup_config(void)
 {
    char      **conflines;
-   char        repltok[TZ_STRLEN_MAX + 100];
+   char        repltok[MAXPGPATH];
    char        path[MAXPGPATH];
    const char *default_timezone;
 
@@ -1178,6 +1178,15 @@ setup_config(void)
                 n_buffers * (BLCKSZ / 1024));
    conflines = replace_token(conflines, "#shared_buffers = 32MB", repltok);
 
+#ifdef HAVE_UNIX_SOCKETS
+   snprintf(repltok, sizeof(repltok), "#unix_socket_directories = '%s'",
+            DEFAULT_PGSOCKET_DIR);
+#else
+   snprintf(repltok, sizeof(repltok), "#unix_socket_directories = ''");
+#endif
+   conflines = replace_token(conflines, "#unix_socket_directories = '/tmp'",
+                             repltok);
+
 #if DEF_PGPORT != 5432
    snprintf(repltok, sizeof(repltok), "#port = %d", DEF_PGPORT);
    conflines = replace_token(conflines, "#port = 5432", repltok);
index 72fc4c1abf6ae80b92aafea727c9f07d36d9efb3..af8d8b28e690ce37aa3a8565bcbe4f3e4ffa62a9 100644 (file)
@@ -521,7 +521,7 @@ test_postmaster_connection(bool do_checkpoint)
                        hostaddr = optlines[LOCK_FILE_LINE_LISTEN_ADDR - 1];
 
                        /*
-                        * While unix_socket_directory can accept relative
+                        * While unix_socket_directories can accept relative
                         * directories, libpq's host parameter must have a
                         * leading slash to indicate a socket directory.  So,
                         * ignore sockdir if it's relative, and try to use TCP
index 7083cd866b68ea906ab66b4f59a9e3b6dfd47885..41e52f2a7c68c9b9d4761d1b0e6e925c9cb7637b 100644 (file)
@@ -44,12 +44,12 @@ typedef struct
 /*
  * prototypes for functions in pqcomm.c
  */
-extern int StreamServerPort(int family, char *hostName,
-   unsigned short portNumber, char *unixSocketName, pgsocket ListenSocket[],
-                int MaxListen);
+extern int StreamServerPort(int family, char *hostName,
+                unsigned short portNumber, char *unixSocketDir,
+                pgsocket ListenSocket[], int MaxListen);
 extern int StreamConnection(pgsocket server_fd, Port *port);
 extern void StreamClose(pgsocket sock);
-extern void TouchSocketFile(void);
+extern void TouchSocketFiles(void);
 extern void pq_init(void);
 extern void pq_comm_reset(void);
 extern int pq_getbytes(char *s, size_t len);
index 8df2a28126a1f24e7a72513fb253da5c3f87773f..9f57989f8056344a239f9da6e99f6f83f717062e 100644 (file)
@@ -411,7 +411,7 @@ extern char *local_preload_libraries_string;
  *     2   data directory path
  *     3   postmaster start timestamp (time_t representation)
  *     4   port number
- *     5   socket directory path (empty on Windows)
+ *     5   first Unix socket directory path (empty if none)
  *     6   first listen_address (IP address or "*"; empty if no TCP port)
  *     7   shared memory key (not present on Windows)
  *
@@ -429,8 +429,9 @@ extern char *local_preload_libraries_string;
 #define LOCK_FILE_LINE_SHMEM_KEY   7
 
 extern void CreateDataDirLockFile(bool amPostmaster);
-extern void CreateSocketLockFile(const char *socketfile, bool amPostmaster);
-extern void TouchSocketLockFile(void);
+extern void CreateSocketLockFile(const char *socketfile, bool amPostmaster,
+                    const char *socketDir);
+extern void TouchSocketLockFiles(void);
 extern void AddToDataDirLockFile(int target_line, const char *str);
 extern void ValidatePgVersion(const char *path);
 extern void process_shared_preload_libraries(void);
index 7d01d3d6248610a0aecee9bcdbd85d4cb54ca8fa..0fe7ec26db7646472fca248dee6982dc5c595c6c 100644 (file)
@@ -19,7 +19,7 @@ extern int    ReservedBackends;
 extern int PostPortNumber;
 extern int Unix_socket_permissions;
 extern char *Unix_socket_group;
-extern char *UnixSocketDir;
+extern char *Unix_socket_directories;
 extern char *ListenAddresses;
 extern bool ClientAuthInProgress;
 extern int PreAuthDelay;
index f706ba705231e05891b0dc3b3dce1f97bf9e7487..c9c665dae09c08a726c04ed58591d63e3b1b3347 100644 (file)
@@ -756,6 +756,8 @@ extern int  varstr_cmp(char *arg1, int len1, char *arg2, int len2, Oid collid);
 extern List *textToQualifiedNameList(text *textval);
 extern bool SplitIdentifierString(char *rawstring, char separator,
                      List **namelist);
+extern bool SplitDirectoriesString(char *rawstring, char separator,
+                      List **namelist);
 extern Datum replace_text(PG_FUNCTION_ARGS);
 extern text *replace_text_regexp(text *src_text, void *regexp,
                    text *replace_text, bool glob);