1 files changed, 99 insertions, 163 deletions

diff --git a/scan-tree.c b/scan-tree.c
index c120efe..780d405 100644
--- a/scan-tree.c
+++ b/scan-tree.c
@@ -1,269 +1,205 @@
 /* scan-tree.c
- *
- * Copyright (C) 2006-2014 cgit Development Team <cgit@lists.zx2c4.com>
+ * 
+ * Copyright (C) 2008-2009 Lars Hjemli
+ * Copyright (C) 2010 Jason A. Donenfeld <Jason@zx2c4.com>
  *
  * Licensed under GNU General Public License v2
  *   (see COPYING for full license text)
  */
 
 #include "cgit.h"
-#include "scan-tree.h"
 #include "configfile.h"
 #include "html.h"
-#include <config.h>
+
+#define MAX_PATH 4096
 
 /* return 1 if path contains a objects/ directory and a HEAD file */
 static int is_git_dir(const char *path)
 {
 	struct stat st;
-	struct strbuf pathbuf = STRBUF_INIT;
-	int result = 0;
+	static char buf[MAX_PATH];
 
-	strbuf_addf(&pathbuf, "%s/objects", path);
-	if (stat(pathbuf.buf, &st)) {
+	if (snprintf(buf, MAX_PATH, "%s/objects", path) >= MAX_PATH) {
+		fprintf(stderr, "Insanely long path: %s\n", path);
+		return 0;
+	}
+	if (stat(buf, &st)) {
 		if (errno != ENOENT)
 			fprintf(stderr, "Error checking path %s: %s (%d)\n",
 				path, strerror(errno), errno);
-		goto out;
+		return 0;
 	}
 	if (!S_ISDIR(st.st_mode))
-		goto out;
+		return 0;
 
-	strbuf_reset(&pathbuf);
-	strbuf_addf(&pathbuf, "%s/HEAD", path);
-	if (stat(pathbuf.buf, &st)) {
+	sprintf(buf, "%s/HEAD", path);
+	if (stat(buf, &st)) {
 		if (errno != ENOENT)
 			fprintf(stderr, "Error checking path %s: %s (%d)\n",
 				path, strerror(errno), errno);
-		goto out;
+		return 0;
 	}
 	if (!S_ISREG(st.st_mode))
-		goto out;
+		return 0;
 
-	result = 1;
-out:
-	strbuf_release(&pathbuf);
-	return result;
+	return 1;
 }
 
-static struct cgit_repo *repo;
+struct cgit_repo *repo;
+repo_config_fn config_fn;
+char *owner;
 
-static void scan_tree_repo_config(const char *name, const char *value)
+static void repo_config(const char *name, const char *value)
 {
-	cgit_repo_config(repo, name, value);
+	config_fn(repo, name, value);
 }
 
-static int gitconfig_config(const char *key, const char *value,
-		const __attribute__((unused)) struct config_context *ctx, void *cb)
+static int git_owner_config(const char *key, const char *value, void *cb)
 {
-	const char *name;
-
 	if (!strcmp(key, "gitweb.owner"))
-		cgit_repo_config(repo, "owner", value);
-	else if (!strcmp(key, "gitweb.description"))
-		cgit_repo_config(repo, "desc", value);
-	else if (!strcmp(key, "gitweb.category"))
-		cgit_repo_config(repo, "section", value);
-	else if (!strcmp(key, "gitweb.homepage"))
-		cgit_repo_config(repo, "homepage", value);
-	else if (skip_prefix(key, "cgit.", &name))
-		cgit_repo_config(repo, name, value);
-
+		owner = xstrdup(value);
 	return 0;
 }
 
-static char *xstrrchr(char *s, char *from, int c)
-{
-	while (from >= s && *from != c)
-		from--;
-	return from < s ? NULL : from;
-}
-
-static void add_repo(const char *base, struct strbuf *path)
+static void add_repo(const char *base, const char *path, repo_config_fn fn)
 {
 	struct stat st;
 	struct passwd *pwd;
-	size_t pathlen;
-	struct strbuf rel = STRBUF_INIT;
-	char *p, *slash;
-	int n;
+	char *p;
 	size_t size;
 
-	if (stat(path->buf, &st)) {
+	if (stat(path, &st)) {
 		fprintf(stderr, "Error accessing %s: %s (%d)\n",
-			path->buf, strerror(errno), errno);
+			path, strerror(errno), errno);
 		return;
 	}
-
-	strbuf_addch(path, '/');
-	pathlen = path->len;
-
-	if (ctx.cfg.strict_export) {
-		strbuf_addstr(path, ctx.cfg.strict_export);
-		if(stat(path->buf, &st))
-			return;
-		strbuf_setlen(path, pathlen);
-	}
-
-	strbuf_addstr(path, "noweb");
-	if (!stat(path->buf, &st))
+	if (!stat(fmt("%s/noweb", path), &st))
 		return;
-	strbuf_setlen(path, pathlen);
 
-	if (!starts_with(path->buf, base))
-		strbuf_addbuf(&rel, path);
+	owner = NULL;
+	if (ctx.cfg.enable_gitweb_owner)
+		git_config_from_file(git_owner_config, fmt("%s/config", path), NULL);
+	if (base == path)
+		p = fmt("%s", path);
 	else
-		strbuf_addstr(&rel, path->buf + strlen(base) + 1);
+		p = fmt("%s", path + strlen(base) + 1);
 
-	if (!strcmp(rel.buf + rel.len - 5, "/.git"))
-		strbuf_setlen(&rel, rel.len - 5);
-	else if (rel.len && rel.buf[rel.len - 1] == '/')
-		strbuf_setlen(&rel, rel.len - 1);
+	if (!strcmp(p + strlen(p) - 5, "/.git"))
+		p[strlen(p) - 5] = '\0';
 
-	repo = cgit_add_repo(rel.buf);
-	if (ctx.cfg.enable_git_config) {
-		strbuf_addstr(path, "config");
-		git_config_from_file(gitconfig_config, path->buf, NULL);
-		strbuf_setlen(path, pathlen);
-	}
-
-	if (ctx.cfg.remove_suffix) {
-		size_t urllen;
-		strip_suffix(repo->url, ".git", &urllen);
-		strip_suffix_mem(repo->url, &urllen, "/");
-		repo->url[urllen] = '\0';
-	}
-	repo->path = strdup_first_line(path->buf);
-	while (!repo->owner) {
+	repo = cgit_add_repo(xstrdup(p));
+	if (ctx.cfg.remove_suffix)
+		if ((p = strrchr(repo->url, '.')) && !strcmp(p, ".git"))
+			*p = '\0';
+	repo->name = repo->url;
+	repo->path = xstrdup(path);
+	while (!owner) {
 		if ((pwd = getpwuid(st.st_uid)) == NULL) {
 			fprintf(stderr, "Error reading owner-info for %s: %s (%d)\n",
-				path->buf, strerror(errno), errno);
+				path, strerror(errno), errno);
 			break;
 		}
 		if (pwd->pw_gecos)
 			if ((p = strchr(pwd->pw_gecos, ',')))
 				*p = '\0';
-		repo->owner = strdup_first_line(pwd->pw_gecos ? pwd->pw_gecos : pwd->pw_name);
+		owner = xstrdup(pwd->pw_gecos ? pwd->pw_gecos : pwd->pw_name);
 	}
+	repo->owner = owner;
 
-	if (repo->desc == cgit_default_repo_desc || !repo->desc) {
-		strbuf_addstr(path, "description");
-		if (!stat(path->buf, &st))
-			read_first_line(path->buf, &repo->desc, &size);
-		strbuf_setlen(path, pathlen);
-	}
+	p = fmt("%s/description", path);
+	if (!stat(p, &st))
+		readfile(p, &repo->desc, &size);
 
-	if (ctx.cfg.section_from_path) {
-		n = ctx.cfg.section_from_path;
-		if (n > 0) {
-			slash = rel.buf - 1;
-			while (slash && n && (slash = strchr(slash + 1, '/')))
-				n--;
-		} else {
-			slash = rel.buf + rel.len;
-			while (slash && n && (slash = xstrrchr(rel.buf, slash - 1, '/')))
-				n++;
-		}
-		if (slash && !n) {
-			*slash = '\0';
-			repo->section = strdup_first_line(rel.buf);
-			*slash = '/';
-			if (starts_with(repo->name, repo->section)) {
-				repo->name += strlen(repo->section);
-				if (*repo->name == '/')
-					repo->name++;
-			}
-		}
+	if (!repo->readme) {
+		p = fmt("%s/README.html", path);
+		if (!stat(p, &st))
+			repo->readme = "README.html";
 	}
 
-	strbuf_addstr(path, "cgitrc");
-	if (!stat(path->buf, &st))
-		parse_configfile(path->buf, &scan_tree_repo_config);
-
-	strbuf_release(&rel);
+	p = fmt("%s/cgitrc", path);
+	if (!stat(p, &st)) {
+		config_fn = fn;
+		parse_configfile(xstrdup(p), &repo_config);
+	}
 }
 
-static void scan_path(const char *base, const char *path)
+static void scan_path(const char *base, const char *path, repo_config_fn fn)
 {
-	DIR *dir = opendir(path);
+	DIR *dir;
 	struct dirent *ent;
-	struct strbuf pathbuf = STRBUF_INIT;
-	size_t pathlen = strlen(path);
+	char *buf;
 	struct stat st;
 
+	if (is_git_dir(path)) {
+		add_repo(base, path, fn);
+		return;
+	}
+	if (is_git_dir(fmt("%s/.git", path))) {
+		add_repo(base, fmt("%s/.git", path), fn);
+		return;
+	}
+	dir = opendir(path);
 	if (!dir) {
 		fprintf(stderr, "Error opening directory %s: %s (%d)\n",
 			path, strerror(errno), errno);
 		return;
 	}
-
-	strbuf_add(&pathbuf, path, strlen(path));
-	if (is_git_dir(pathbuf.buf)) {
-		add_repo(base, &pathbuf);
-		goto end;
-	}
-	strbuf_addstr(&pathbuf, "/.git");
-	if (is_git_dir(pathbuf.buf)) {
-		add_repo(base, &pathbuf);
-		goto end;
-	}
-	/*
-	 * Add one because we don't want to lose the trailing '/' when we
-	 * reset the length of pathbuf in the loop below.
-	 */
-	pathlen++;
-	while ((ent = readdir(dir)) != NULL) {
+	while((ent = readdir(dir)) != NULL) {
 		if (ent->d_name[0] == '.') {
 			if (ent->d_name[1] == '\0')
 				continue;
 			if (ent->d_name[1] == '.' && ent->d_name[2] == '\0')
 				continue;
-			if (!ctx.cfg.scan_hidden_path)
-				continue;
 		}
-		strbuf_setlen(&pathbuf, pathlen);
-		strbuf_addstr(&pathbuf, ent->d_name);
-		if (stat(pathbuf.buf, &st)) {
+		buf = malloc(strlen(path) + strlen(ent->d_name) + 2);
+		if (!buf) {
+			fprintf(stderr, "Alloc error on %s: %s (%d)\n",
+				path, strerror(errno), errno);
+			exit(1);
+		}
+		sprintf(buf, "%s/%s", path, ent->d_name);
+		if (stat(buf, &st)) {
 			fprintf(stderr, "Error checking path %s: %s (%d)\n",
-				pathbuf.buf, strerror(errno), errno);
+				buf, strerror(errno), errno);
+			free(buf);
 			continue;
 		}
 		if (S_ISDIR(st.st_mode))
-			scan_path(base, pathbuf.buf);
+			scan_path(base, buf, fn);
+		free(buf);
 	}
-end:
-	strbuf_release(&pathbuf);
 	closedir(dir);
 }
 
-void scan_projects(const char *path, const char *projectsfile)
+#define lastc(s) s[strlen(s) - 1]
+
+void scan_projects(const char *path, const char *projectsfile, repo_config_fn fn)
 {
-	struct strbuf line = STRBUF_INIT;
+	char line[MAX_PATH * 2], *z;
 	FILE *projects;
 	int err;
-
+	
 	projects = fopen(projectsfile, "r");
 	if (!projects) {
 		fprintf(stderr, "Error opening projectsfile %s: %s (%d)\n",
 			projectsfile, strerror(errno), errno);
-		return;
 	}
-	while (strbuf_getline(&line, projects) != EOF) {
-		if (!line.len)
-			continue;
-		strbuf_insert(&line, 0, "/", 1);
-		strbuf_insert(&line, 0, path, strlen(path));
-		scan_path(path, line.buf);
+	while (fgets(line, sizeof(line), projects) != NULL) {
+		for (z = &lastc(line);
+		     strlen(line) && strchr("\n\r", *z);
+		     z = &lastc(line))
+			*z = '\0';
+		if (strlen(line))
+			scan_path(path, fmt("%s/%s", path, line), fn);
 	}
 	if ((err = ferror(projects))) {
 		fprintf(stderr, "Error reading from projectsfile %s: %s (%d)\n",
 			projectsfile, strerror(err), err);
 	}
 	fclose(projects);
-	strbuf_release(&line);
 }
 
-void scan_tree(const char *path)
+void scan_tree(const char *path, repo_config_fn fn)
 {
-	scan_path(path, path);
+	scan_path(path, path, fn);
 }