Rename environment variables.

[git.git] / init-db.c

diff --git a/init-db.c b/init-db.c
index bc65086..157b92b 100644 (file)
--- a/init-db.c
+++ b/init-db.c
@@ -5,52 +5,43 @@
  */
 #include "cache.h"
 
+void safe_create_dir(const char *dir)
+{
+       if (mkdir(dir, 0755) < 0) {
+               if (errno != EEXIST) {
+                       perror(dir);
+                       exit(1);
+               }
+       }
+}
+
+/*
+ * If you want to, you can share the DB area with any number of branches.
+ * That has advantages: you can save space by sharing all the SHA1 objects.
+ * On the other hand, it might just make lookup slower and messier. You
+ * be the judge.  The default case is to have one DB per managed directory.
+ */
 int main(int argc, char **argv)
 {
-       char *sha1_dir = getenv(DB_ENVIRONMENT), *path;
-       int len, i, fd;
+       const char *sha1_dir;
+       char *path;
+       int len, i;
 
-       if (mkdir(".dircache", 0700) < 0) {
-               perror("unable to create .dircache");
-               exit(1);
-       }
+       safe_create_dir(".git");
 
-       /*
-        * If you want to, you can share the DB area with any number of branches.
-        * That has advantages: you can save space by sharing all the SHA1 objects.
-        * On the other hand, it might just make lookup slower and messier. You
-        * be the judge.
-        */
-       sha1_dir = getenv(DB_ENVIRONMENT);
-       if (sha1_dir) {
-               struct stat st;
-               if (!stat(sha1_dir, &st) < 0 && S_ISDIR(st.st_mode))
-                       return;
-               fprintf(stderr, "DB_ENVIRONMENT set to bad directory %s: ", sha1_dir);
+       sha1_dir = gitenv(DB_ENVIRONMENT);
+       if (!sha1_dir) {
+               sha1_dir = DEFAULT_DB_ENVIRONMENT;
+               fprintf(stderr, "defaulting to local storage area\n");
        }
-
-       /*
-        * The default case is to have a DB per managed directory. 
-        */
-       sha1_dir = DEFAULT_DB_ENVIRONMENT;
-       fprintf(stderr, "defaulting to private storage area\n");
        len = strlen(sha1_dir);
-       if (mkdir(sha1_dir, 0700) < 0) {
-               if (errno != EEXIST) {
-                       perror(sha1_dir);
-                       exit(1);
-               }
-       }
-       path = malloc(len + 40);
+       path = xmalloc(len + 40);
        memcpy(path, sha1_dir, len);
+
+       safe_create_dir(sha1_dir);
        for (i = 0; i < 256; i++) {
                sprintf(path+len, "/%02x", i);
-               if (mkdir(path, 0700) < 0) {
-                       if (errno != EEXIST) {
-                               perror(path);
-                               exit(1);
-                       }
-               }
+               safe_create_dir(path);
        }
        return 0;
 }