Changeset 3389 for trunk/src/kmk/kmkbuiltin/mkdir.c

Timestamp:

Jun 26, 2020, 7:16:26 PM (5 years ago)

Author:

bird

Message:

kmk: Avoid setting umask just to get it, store the current value in a global variable (g_fUMask). The umask(0777) call in cp.c raced other code (kmk_append) that created files and directories, leaving us with read-only files sometimes.

File:

• trunk/src/kmk/kmkbuiltin/mkdir.c (modified) (6 diffs)

trunk/src/kmk/kmkbuiltin/mkdir.c

@@ -49 +49 @@
 #include "err.h"
 #include <errno.h>
+#include <assert.h>
 #ifndef _MSC_VER
 # include <libgen.h>
@@ -80 +81 @@
 };
 
+extern mode_t g_fUMask;
 
 extern void * bsd_setmode(const char *p);
@@ -167 +169 @@
 
 #ifdef KMK_BUILTIN_STANDALONE
+mode_t g_fUMask;
 int main(int argc, char **argv, char **envp)
 {
     KMKBUILTINCTX Ctx = { "kmk_mkdir", NULL };
+    umask(g_fUMask = umask(0077));
     return kmk_builtin_mkdir(argc, argv, envp, &Ctx);
 }
@@ -240 +244 @@
                          * instead of doing chmod's.
                          */
-                        oumask = umask(0);
+#ifdef KMK_BUILTIN_STANDALONE
+                        oumask = umask(0077);
+#else
+                        oumask = g_fUMask;
+                        assert(oumask == umask(g_fUMask));
+#endif
                         numask = oumask & ~(S_IWUSR | S_IXUSR);
-                        (void)umask(numask);
+                        if (numask != oumask)
+                            (void)umask(numask);
                         first = 0;
                 }
-                if (last)
+                if (last && oumask != numask)
                         (void)umask(oumask);
                 if (mkdir(path, last ? omode : S_IRWXU | S_IRWXG | S_IRWXO) < 0) {
@@ -255 +265 @@
                                         retval = 1;
                                         break;
-                                } else if (!S_ISDIR(sb.st_mode)) {
+                                }
+                                if (!S_ISDIR(sb.st_mode)) {
                                         if (last)
                                                 errno = EEXIST;
@@ -274 +285 @@
                     *p = '/';
         }
-        if (!first && !last)
+        if (!first && !last && oumask != numask)
                 (void)umask(oumask);
         return (retval);