In pg_upgrade on Windows, check if the directory is writable by actually
authorBruce Momjian <bruce@momjian.us>
Sun, 24 Jul 2011 05:42:45 +0000 (01:42 -0400)
committerBruce Momjian <bruce@momjian.us>
Sun, 24 Jul 2011 05:43:57 +0000 (01:43 -0400)
creating and removing a file because access() doesn't work on that
platform.

Backpatch to 9.1 where this check was added.

contrib/pg_upgrade/exec.c

index a76c06e302344a5bb92215e73e27523a85b5d519..3493696f92f3c237786ebd7d455541a5aba0541d 100644 (file)
@@ -16,6 +16,9 @@
 static void check_data_dir(const char *pg_data);
 static void check_bin_dir(ClusterInfo *cluster);
 static void validate_exec(const char *dir, const char *cmdName);
+#ifdef WIN32
+static int win32_check_directory_write_permissions(void);
+#endif
 
 
 /*
@@ -97,17 +100,11 @@ verify_directories(void)
 
    prep_status("Checking current, bin, and data directories");
 
-   if (access(".", R_OK | W_OK
 #ifndef WIN32
-
-   /*
-    * Do a directory execute check only on Unix because execute permission on
-    * NTFS means "can execute scripts", which we don't care about. Also, X_OK
-    * is not defined in the Windows API.
-    */
-              | X_OK
+   if (access(".", R_OK | W_OK | X_OK) != 0)
+#else
+   if (win32_check_directory_write_permissions() != 0)
 #endif
-              ) != 0)
        pg_log(PG_FATAL,
          "You must have read and write access in the current directory.\n");
 
@@ -119,6 +116,32 @@ verify_directories(void)
 }
 
 
+#ifdef WIN32
+/*
+ * win32_check_directory_write_permissions()
+ *
+ * access() on WIN32 can't check directory permissions, so we have to
+ * optionally create, then delete a file to check.
+ *     http://msdn.microsoft.com/en-us/library/1w06ktdy%28v=vs.80%29.aspx
+ */
+static int
+win32_check_directory_write_permissions(void)
+{
+   int fd;
+
+   /*
+    *  We open a file we would normally create anyway.  We do this even in
+    *  'check' mode, which isn't ideal, but this is the best we can do.
+    */ 
+   if ((fd = open(GLOBALS_DUMP_FILE, O_RDWR | O_CREAT, S_IRUSR | S_IWUSR)) < 0)
+       return -1;
+   close(fd);
+
+   return unlink(GLOBALS_DUMP_FILE);
+}
+#endif
+
+
 /*
  * check_data_dir()
  *