]> gitweb.ps.run Git - ps-cgit/blob - scan-tree.c
Fix crash when projectsfile cannot be opened
[ps-cgit] / scan-tree.c
1 /* scan-tree.c
2  * 
3  * Copyright (C) 2008-2009 Lars Hjemli
4  * Copyright (C) 2010 Jason A. Donenfeld <Jason@zx2c4.com>
5  *
6  * Licensed under GNU General Public License v2
7  *   (see COPYING for full license text)
8  */
9
10 #include "cgit.h"
11 #include "configfile.h"
12 #include "html.h"
13
14 #define MAX_PATH 4096
15
16 /* return 1 if path contains a objects/ directory and a HEAD file */
17 static int is_git_dir(const char *path)
18 {
19         struct stat st;
20         static char buf[MAX_PATH];
21
22         if (snprintf(buf, MAX_PATH, "%s/objects", path) >= MAX_PATH) {
23                 fprintf(stderr, "Insanely long path: %s\n", path);
24                 return 0;
25         }
26         if (stat(buf, &st)) {
27                 if (errno != ENOENT)
28                         fprintf(stderr, "Error checking path %s: %s (%d)\n",
29                                 path, strerror(errno), errno);
30                 return 0;
31         }
32         if (!S_ISDIR(st.st_mode))
33                 return 0;
34
35         sprintf(buf, "%s/HEAD", path);
36         if (stat(buf, &st)) {
37                 if (errno != ENOENT)
38                         fprintf(stderr, "Error checking path %s: %s (%d)\n",
39                                 path, strerror(errno), errno);
40                 return 0;
41         }
42         if (!S_ISREG(st.st_mode))
43                 return 0;
44
45         return 1;
46 }
47
48 struct cgit_repo *repo;
49 repo_config_fn config_fn;
50 char *owner;
51
52 static void repo_config(const char *name, const char *value)
53 {
54         config_fn(repo, name, value);
55 }
56
57 static int git_owner_config(const char *key, const char *value, void *cb)
58 {
59         if (!strcmp(key, "gitweb.owner"))
60                 owner = xstrdup(value);
61         return 0;
62 }
63
64 static char *xstrrchr(char *s, char *from, int c)
65 {
66         while (from >= s && *from != c)
67                 from--;
68         return from < s ? NULL : from;
69 }
70
71 static void add_repo(const char *base, const char *path, repo_config_fn fn)
72 {
73         struct stat st;
74         struct passwd *pwd;
75         char *rel, *p, *slash;
76         int n;
77         size_t size;
78
79         if (stat(path, &st)) {
80                 fprintf(stderr, "Error accessing %s: %s (%d)\n",
81                         path, strerror(errno), errno);
82                 return;
83         }
84
85         if (ctx.cfg.strict_export && stat(fmt("%s/%s", path, ctx.cfg.strict_export), &st))
86                 return;
87
88         if (!stat(fmt("%s/noweb", path), &st))
89                 return;
90
91         owner = NULL;
92         if (ctx.cfg.enable_gitweb_owner)
93                 git_config_from_file(git_owner_config, fmt("%s/config", path), NULL);
94         if (base == path)
95                 rel = xstrdup(fmt("%s", path));
96         else
97                 rel = xstrdup(fmt("%s", path + strlen(base) + 1));
98
99         if (!strcmp(rel + strlen(rel) - 5, "/.git"))
100                 rel[strlen(rel) - 5] = '\0';
101
102         repo = cgit_add_repo(rel);
103         if (ctx.cfg.remove_suffix)
104                 if ((p = strrchr(repo->url, '.')) && !strcmp(p, ".git"))
105                         *p = '\0';
106         repo->name = repo->url;
107         repo->path = xstrdup(path);
108         while (!owner) {
109                 if ((pwd = getpwuid(st.st_uid)) == NULL) {
110                         fprintf(stderr, "Error reading owner-info for %s: %s (%d)\n",
111                                 path, strerror(errno), errno);
112                         break;
113                 }
114                 if (pwd->pw_gecos)
115                         if ((p = strchr(pwd->pw_gecos, ',')))
116                                 *p = '\0';
117                 owner = xstrdup(pwd->pw_gecos ? pwd->pw_gecos : pwd->pw_name);
118         }
119         repo->owner = owner;
120
121         p = fmt("%s/description", path);
122         if (!stat(p, &st))
123                 readfile(p, &repo->desc, &size);
124
125         if (!repo->readme) {
126                 p = fmt("%s/README.html", path);
127                 if (!stat(p, &st))
128                         repo->readme = "README.html";
129         }
130         if (ctx.cfg.section_from_path) {
131                 n  = ctx.cfg.section_from_path;
132                 if (n > 0) {
133                         slash = rel;
134                         while (slash && n && (slash = strchr(slash, '/')))
135                                 n--;
136                 } else {
137                         slash = rel + strlen(rel);
138                         while (slash && n && (slash = xstrrchr(rel, slash, '/')))
139                                 n++;
140                 }
141                 if (slash && !n) {
142                         *slash = '\0';
143                         repo->section = xstrdup(rel);
144                         *slash = '/';
145                         if (!prefixcmp(repo->name, repo->section)) {
146                                 repo->name += strlen(repo->section);
147                                 if (*repo->name == '/')
148                                         repo->name++;
149                         }
150                 }
151         }
152
153         p = fmt("%s/cgitrc", path);
154         if (!stat(p, &st)) {
155                 config_fn = fn;
156                 parse_configfile(xstrdup(p), &repo_config);
157         }
158 }
159
160 static void scan_path(const char *base, const char *path, repo_config_fn fn)
161 {
162         DIR *dir = opendir(path);
163         struct dirent *ent;
164         char *buf;
165         struct stat st;
166
167         if (!dir) {
168                 fprintf(stderr, "Error opening directory %s: %s (%d)\n",
169                         path, strerror(errno), errno);
170                 return;
171         }
172         if (is_git_dir(path)) {
173                 add_repo(base, path, fn);
174                 goto end;
175         }
176         if (is_git_dir(fmt("%s/.git", path))) {
177                 add_repo(base, fmt("%s/.git", path), fn);
178                 goto end;
179         }
180         while((ent = readdir(dir)) != NULL) {
181                 if (ent->d_name[0] == '.') {
182                         if (ent->d_name[1] == '\0')
183                                 continue;
184                         if (ent->d_name[1] == '.' && ent->d_name[2] == '\0')
185                                 continue;
186                         if (!ctx.cfg.scan_hidden_path)
187                                 continue;
188                 }
189                 buf = malloc(strlen(path) + strlen(ent->d_name) + 2);
190                 if (!buf) {
191                         fprintf(stderr, "Alloc error on %s: %s (%d)\n",
192                                 path, strerror(errno), errno);
193                         exit(1);
194                 }
195                 sprintf(buf, "%s/%s", path, ent->d_name);
196                 if (stat(buf, &st)) {
197                         fprintf(stderr, "Error checking path %s: %s (%d)\n",
198                                 buf, strerror(errno), errno);
199                         free(buf);
200                         continue;
201                 }
202                 if (S_ISDIR(st.st_mode))
203                         scan_path(base, buf, fn);
204                 free(buf);
205         }
206 end:
207         closedir(dir);
208 }
209
210 #define lastc(s) s[strlen(s) - 1]
211
212 void scan_projects(const char *path, const char *projectsfile, repo_config_fn fn)
213 {
214         char line[MAX_PATH * 2], *z;
215         FILE *projects;
216         int err;
217         
218         projects = fopen(projectsfile, "r");
219         if (!projects) {
220                 fprintf(stderr, "Error opening projectsfile %s: %s (%d)\n",
221                         projectsfile, strerror(errno), errno);
222                 return;
223         }
224         while (fgets(line, sizeof(line), projects) != NULL) {
225                 for (z = &lastc(line);
226                      strlen(line) && strchr("\n\r", *z);
227                      z = &lastc(line))
228                         *z = '\0';
229                 if (strlen(line))
230                         scan_path(path, fmt("%s/%s", path, line), fn);
231         }
232         if ((err = ferror(projects))) {
233                 fprintf(stderr, "Error reading from projectsfile %s: %s (%d)\n",
234                         projectsfile, strerror(err), err);
235         }
236         fclose(projects);
237 }
238
239 void scan_tree(const char *path, repo_config_fn fn)
240 {
241         scan_path(path, path, fn);
242 }