do_for_each_reflog(): return early on error
[gitweb.git] / refs.c
diff --git a/refs.c b/refs.c
index 37ea557f3e843548d8418c51ec283aa05d4a4c14..8c39bd76e5c9044ca867c91898b208eb82028f17 100644 (file)
--- a/refs.c
+++ b/refs.c
@@ -276,6 +276,27 @@ static struct ref_entry *search_ref_dir(struct ref_dir *dir, const char *refname
        return *r;
 }
 
+/*
+ * Search for a directory entry directly within dir (without
+ * recursing).  Sort dir if necessary.  subdirname must be a directory
+ * name (i.e., end in '/').  If mkdir is set, then create the
+ * directory if it is missing; otherwise, return NULL if the desired
+ * directory cannot be found.
+ */
+static struct ref_entry *search_for_subdir(struct ref_dir *dir,
+                                          const char *subdirname, int mkdir)
+{
+       struct ref_entry *entry = search_ref_dir(dir, subdirname);
+       if (!entry) {
+               if (!mkdir)
+                       return NULL;
+               entry = create_dir_entry(subdirname);
+               add_entry_to_dir(dir, entry);
+       }
+       assert(entry->flag & REF_DIR);
+       return entry;
+}
+
 /*
  * If refname is a reference name, find the ref_dir within the dir
  * tree that should hold refname.  If refname is a directory name
@@ -294,17 +315,10 @@ static struct ref_dir *find_containing_dir(struct ref_dir *dir,
        for (slash = strchr(refname_copy, '/'); slash; slash = strchr(slash + 1, '/')) {
                char tmp = slash[1];
                slash[1] = '\0';
-               entry = search_ref_dir(dir, refname_copy);
-               if (!entry) {
-                       if (!mkdir) {
-                               dir = NULL;
-                               break;
-                       }
-                       entry = create_dir_entry(refname_copy);
-                       add_entry_to_dir(dir, entry);
-               }
+               entry = search_for_subdir(dir, refname_copy, mkdir);
                slash[1] = tmp;
-               assert(entry->flag & REF_DIR);
+               if (!entry)
+                       break;
                dir = &entry->u.subdir;
        }
 
@@ -749,13 +763,18 @@ void add_packed_ref(const char *refname, const unsigned char *sha1)
                        create_ref_entry(refname, sha1, REF_ISPACKED, 1));
 }
 
+/*
+ * Read the loose references for refs from the namespace dirname.
+ * dirname must end with '/'.  dir must be the directory entry
+ * corresponding to dirname.
+ */
 static void get_ref_dir(struct ref_cache *refs, const char *dirname,
                        struct ref_dir *dir)
 {
        DIR *d;
        const char *path;
        struct dirent *de;
-       int dirnamelen;
+       int dirnamelen = strlen(dirname);
        struct strbuf refname;
 
        if (*refs->name)
@@ -767,13 +786,8 @@ static void get_ref_dir(struct ref_cache *refs, const char *dirname,
        if (!d)
                return;
 
-       dirnamelen = strlen(dirname);
        strbuf_init(&refname, dirnamelen + 257);
        strbuf_add(&refname, dirname, dirnamelen);
-       if (dirnamelen && dirname[dirnamelen-1] != '/') {
-               strbuf_addch(&refname, '/');
-               dirnamelen++;
-       }
 
        while ((de = readdir(d)) != NULL) {
                unsigned char sha1[20];
@@ -792,7 +806,9 @@ static void get_ref_dir(struct ref_cache *refs, const char *dirname,
                if (stat(refdir, &st) < 0) {
                        ; /* silently ignore */
                } else if (S_ISDIR(st.st_mode)) {
-                       get_ref_dir(refs, refname.buf, dir);
+                       strbuf_addch(&refname, '/');
+                       get_ref_dir(refs, refname.buf,
+                                   &search_for_subdir(dir, refname.buf, 1)->u.subdir);
                } else {
                        if (*refs->name) {
                                hashclr(sha1);
@@ -805,7 +821,8 @@ static void get_ref_dir(struct ref_cache *refs, const char *dirname,
                                hashclr(sha1);
                                flag |= REF_ISBROKEN;
                        }
-                       add_ref(dir, create_ref_entry(refname.buf, sha1, flag, 1));
+                       add_entry_to_dir(dir,
+                                        create_ref_entry(refname.buf, sha1, flag, 1));
                }
                strbuf_setlen(&refname, dirnamelen);
        }
@@ -816,7 +833,8 @@ static void get_ref_dir(struct ref_cache *refs, const char *dirname,
 static struct ref_dir *get_loose_refs(struct ref_cache *refs)
 {
        if (!refs->did_loose) {
-               get_ref_dir(refs, "refs", &refs->loose);
+               get_ref_dir(refs, "refs/",
+                           &search_for_subdir(&refs->loose, "refs/", 1)->u.subdir);
                refs->did_loose = 1;
        }
        return &refs->loose;
@@ -2228,47 +2246,47 @@ static int do_for_each_reflog(const char *base, each_ref_fn fn, void *cb_data)
 {
        DIR *d = opendir(git_path("logs/%s", base));
        int retval = 0;
+       struct dirent *de;
+       int baselen;
+       char *log;
 
-       if (d) {
-               struct dirent *de;
-               int baselen = strlen(base);
-               char *log = xmalloc(baselen + 257);
+       if (!d)
+               return *base ? errno : 0;
 
-               memcpy(log, base, baselen);
-               if (baselen && base[baselen-1] != '/')
-                       log[baselen++] = '/';
+       baselen = strlen(base);
+       log = xmalloc(baselen + 257);
+       memcpy(log, base, baselen);
+       if (baselen && base[baselen-1] != '/')
+               log[baselen++] = '/';
 
-               while ((de = readdir(d)) != NULL) {
-                       struct stat st;
-                       int namelen;
+       while ((de = readdir(d)) != NULL) {
+               struct stat st;
+               int namelen;
 
-                       if (de->d_name[0] == '.')
-                               continue;
-                       namelen = strlen(de->d_name);
-                       if (namelen > 255)
-                               continue;
-                       if (has_extension(de->d_name, ".lock"))
-                               continue;
-                       memcpy(log + baselen, de->d_name, namelen+1);
-                       if (stat(git_path("logs/%s", log), &st) < 0)
-                               continue;
-                       if (S_ISDIR(st.st_mode)) {
-                               retval = do_for_each_reflog(log, fn, cb_data);
-                       } else {
-                               unsigned char sha1[20];
-                               if (read_ref_full(log, sha1, 0, NULL))
-                                       retval = error("bad ref for %s", log);
-                               else
-                                       retval = fn(log, sha1, 0, cb_data);
-                       }
-                       if (retval)
-                               break;
+               if (de->d_name[0] == '.')
+                       continue;
+               namelen = strlen(de->d_name);
+               if (namelen > 255)
+                       continue;
+               if (has_extension(de->d_name, ".lock"))
+                       continue;
+               memcpy(log + baselen, de->d_name, namelen+1);
+               if (stat(git_path("logs/%s", log), &st) < 0)
+                       continue;
+               if (S_ISDIR(st.st_mode)) {
+                       retval = do_for_each_reflog(log, fn, cb_data);
+               } else {
+                       unsigned char sha1[20];
+                       if (read_ref_full(log, sha1, 0, NULL))
+                               retval = error("bad ref for %s", log);
+                       else
+                               retval = fn(log, sha1, 0, cb_data);
                }
-               free(log);
-               closedir(d);
+               if (retval)
+                       break;
        }
-       else if (*base)
-               return errno;
+       free(log);
+       closedir(d);
        return retval;
 }