]> gitweb.ps.run Git - ps-cgit/blob - shared.c
Merge branch 'fh/mimetypes'
[ps-cgit] / shared.c
1 /* shared.c: global vars + some callback functions
2  *
3  * Copyright (C) 2006 Lars Hjemli
4  *
5  * Licensed under GNU General Public License v2
6  *   (see COPYING for full license text)
7  */
8
9 #include "cgit.h"
10 #include <stdio.h>
11
12 struct cgit_repolist cgit_repolist;
13 struct cgit_context ctx;
14
15 int chk_zero(int result, char *msg)
16 {
17         if (result != 0)
18                 die("%s: %s", msg, strerror(errno));
19         return result;
20 }
21
22 int chk_positive(int result, char *msg)
23 {
24         if (result <= 0)
25                 die("%s: %s", msg, strerror(errno));
26         return result;
27 }
28
29 int chk_non_negative(int result, char *msg)
30 {
31         if (result < 0)
32                 die("%s: %s",msg, strerror(errno));
33         return result;
34 }
35
36 struct cgit_repo *cgit_add_repo(const char *url)
37 {
38         struct cgit_repo *ret;
39
40         if (++cgit_repolist.count > cgit_repolist.length) {
41                 if (cgit_repolist.length == 0)
42                         cgit_repolist.length = 8;
43                 else
44                         cgit_repolist.length *= 2;
45                 cgit_repolist.repos = xrealloc(cgit_repolist.repos,
46                                                cgit_repolist.length *
47                                                sizeof(struct cgit_repo));
48         }
49
50         ret = &cgit_repolist.repos[cgit_repolist.count-1];
51         memset(ret, 0, sizeof(struct cgit_repo));
52         ret->url = trim_end(url, '/');
53         ret->name = ret->url;
54         ret->path = NULL;
55         ret->desc = "[no description]";
56         ret->owner = NULL;
57         ret->section = ctx.cfg.section;
58         ret->snapshots = ctx.cfg.snapshots;
59         ret->enable_commit_graph = ctx.cfg.enable_commit_graph;
60         ret->enable_log_filecount = ctx.cfg.enable_log_filecount;
61         ret->enable_log_linecount = ctx.cfg.enable_log_linecount;
62         ret->enable_remote_branches = ctx.cfg.enable_remote_branches;
63         ret->enable_subject_links = ctx.cfg.enable_subject_links;
64         ret->max_stats = ctx.cfg.max_stats;
65         ret->module_link = ctx.cfg.module_link;
66         ret->readme = ctx.cfg.readme;
67         ret->mtime = -1;
68         ret->about_filter = ctx.cfg.about_filter;
69         ret->commit_filter = ctx.cfg.commit_filter;
70         ret->source_filter = ctx.cfg.source_filter;
71         ret->clone_url = ctx.cfg.clone_url;
72         ret->submodules.strdup_strings = 1;
73         return ret;
74 }
75
76 struct cgit_repo *cgit_get_repoinfo(const char *url)
77 {
78         int i;
79         struct cgit_repo *repo;
80
81         for (i=0; i<cgit_repolist.count; i++) {
82                 repo = &cgit_repolist.repos[i];
83                 if (!strcmp(repo->url, url))
84                         return repo;
85         }
86         return NULL;
87 }
88
89 void *cgit_free_commitinfo(struct commitinfo *info)
90 {
91         free(info->author);
92         free(info->author_email);
93         free(info->committer);
94         free(info->committer_email);
95         free(info->subject);
96         free(info->msg);
97         free(info->msg_encoding);
98         free(info);
99         return NULL;
100 }
101
102 char *trim_end(const char *str, char c)
103 {
104         int len;
105
106         if (str == NULL)
107                 return NULL;
108         len = strlen(str);
109         while(len > 0 && str[len - 1] == c)
110                 len--;
111         if (len == 0)
112                 return NULL;
113         return xstrndup(str, len);
114 }
115
116 char *strlpart(char *txt, int maxlen)
117 {
118         char *result;
119
120         if (!txt)
121                 return txt;
122
123         if (strlen(txt) <= maxlen)
124                 return txt;
125         result = xmalloc(maxlen + 1);
126         memcpy(result, txt, maxlen - 3);
127         result[maxlen-1] = result[maxlen-2] = result[maxlen-3] = '.';
128         result[maxlen] = '\0';
129         return result;
130 }
131
132 char *strrpart(char *txt, int maxlen)
133 {
134         char *result;
135
136         if (!txt)
137                 return txt;
138
139         if (strlen(txt) <= maxlen)
140                 return txt;
141         result = xmalloc(maxlen + 1);
142         memcpy(result + 3, txt + strlen(txt) - maxlen + 4, maxlen - 3);
143         result[0] = result[1] = result[2] = '.';
144         return result;
145 }
146
147 void cgit_add_ref(struct reflist *list, struct refinfo *ref)
148 {
149         size_t size;
150
151         if (list->count >= list->alloc) {
152                 list->alloc += (list->alloc ? list->alloc : 4);
153                 size = list->alloc * sizeof(struct refinfo *);
154                 list->refs = xrealloc(list->refs, size);
155         }
156         list->refs[list->count++] = ref;
157 }
158
159 struct refinfo *cgit_mk_refinfo(const char *refname, const unsigned char *sha1)
160 {
161         struct refinfo *ref;
162
163         ref = xmalloc(sizeof (struct refinfo));
164         ref->refname = xstrdup(refname);
165         ref->object = parse_object(sha1);
166         switch (ref->object->type) {
167         case OBJ_TAG:
168                 ref->tag = cgit_parse_tag((struct tag *)ref->object);
169                 break;
170         case OBJ_COMMIT:
171                 ref->commit = cgit_parse_commit((struct commit *)ref->object);
172                 break;
173         }
174         return ref;
175 }
176
177 int cgit_refs_cb(const char *refname, const unsigned char *sha1, int flags,
178                   void *cb_data)
179 {
180         struct reflist *list = (struct reflist *)cb_data;
181         struct refinfo *info = cgit_mk_refinfo(refname, sha1);
182
183         if (info)
184                 cgit_add_ref(list, info);
185         return 0;
186 }
187
188 void cgit_diff_tree_cb(struct diff_queue_struct *q,
189                        struct diff_options *options, void *data)
190 {
191         int i;
192
193         for (i = 0; i < q->nr; i++) {
194                 if (q->queue[i]->status == 'U')
195                         continue;
196                 ((filepair_fn)data)(q->queue[i]);
197         }
198 }
199
200 static int load_mmfile(mmfile_t *file, const unsigned char *sha1)
201 {
202         enum object_type type;
203
204         if (is_null_sha1(sha1)) {
205                 file->ptr = (char *)"";
206                 file->size = 0;
207         } else {
208                 file->ptr = read_sha1_file(sha1, &type, 
209                                            (unsigned long *)&file->size);
210         }
211         return 1;
212 }
213
214 /*
215  * Receive diff-buffers from xdiff and concatenate them as
216  * needed across multiple callbacks.
217  *
218  * This is basically a copy of xdiff-interface.c/xdiff_outf(),
219  * ripped from git and modified to use globals instead of
220  * a special callback-struct.
221  */
222 char *diffbuf = NULL;
223 int buflen = 0;
224
225 int filediff_cb(void *priv, mmbuffer_t *mb, int nbuf)
226 {
227         int i;
228
229         for (i = 0; i < nbuf; i++) {
230                 if (mb[i].ptr[mb[i].size-1] != '\n') {
231                         /* Incomplete line */
232                         diffbuf = xrealloc(diffbuf, buflen + mb[i].size);
233                         memcpy(diffbuf + buflen, mb[i].ptr, mb[i].size);
234                         buflen += mb[i].size;
235                         continue;
236                 }
237
238                 /* we have a complete line */
239                 if (!diffbuf) {
240                         ((linediff_fn)priv)(mb[i].ptr, mb[i].size);
241                         continue;
242                 }
243                 diffbuf = xrealloc(diffbuf, buflen + mb[i].size);
244                 memcpy(diffbuf + buflen, mb[i].ptr, mb[i].size);
245                 ((linediff_fn)priv)(diffbuf, buflen + mb[i].size);
246                 free(diffbuf);
247                 diffbuf = NULL;
248                 buflen = 0;
249         }
250         if (diffbuf) {
251                 ((linediff_fn)priv)(diffbuf, buflen);
252                 free(diffbuf);
253                 diffbuf = NULL;
254                 buflen = 0;
255         }
256         return 0;
257 }
258
259 int cgit_diff_files(const unsigned char *old_sha1,
260                     const unsigned char *new_sha1, unsigned long *old_size,
261                     unsigned long *new_size, int *binary, int context,
262                     int ignorews, linediff_fn fn)
263 {
264         mmfile_t file1, file2;
265         xpparam_t diff_params;
266         xdemitconf_t emit_params;
267         xdemitcb_t emit_cb;
268
269         if (!load_mmfile(&file1, old_sha1) || !load_mmfile(&file2, new_sha1))
270                 return 1;
271
272         *old_size = file1.size;
273         *new_size = file2.size;
274
275         if ((file1.ptr && buffer_is_binary(file1.ptr, file1.size)) ||
276             (file2.ptr && buffer_is_binary(file2.ptr, file2.size))) {
277                 *binary = 1;
278                 if (file1.size)
279                         free(file1.ptr);
280                 if (file2.size)
281                         free(file2.ptr);
282                 return 0;
283         }
284
285         memset(&diff_params, 0, sizeof(diff_params));
286         memset(&emit_params, 0, sizeof(emit_params));
287         memset(&emit_cb, 0, sizeof(emit_cb));
288         diff_params.flags = XDF_NEED_MINIMAL;
289         if (ignorews)
290                 diff_params.flags |= XDF_IGNORE_WHITESPACE;
291         emit_params.ctxlen = context > 0 ? context : 3;
292         emit_params.flags = XDL_EMIT_FUNCNAMES;
293         emit_cb.outf = filediff_cb;
294         emit_cb.priv = fn;
295         xdl_diff(&file1, &file2, &diff_params, &emit_params, &emit_cb);
296         if (file1.size)
297                 free(file1.ptr);
298         if (file2.size)
299                 free(file2.ptr);
300         return 0;
301 }
302
303 void cgit_diff_tree(const unsigned char *old_sha1,
304                     const unsigned char *new_sha1,
305                     filepair_fn fn, const char *prefix, int ignorews)
306 {
307         struct diff_options opt;
308         int prefixlen;
309
310         diff_setup(&opt);
311         opt.output_format = DIFF_FORMAT_CALLBACK;
312         opt.detect_rename = 1;
313         opt.rename_limit = ctx.cfg.renamelimit;
314         DIFF_OPT_SET(&opt, RECURSIVE);
315         if (ignorews)
316                 DIFF_XDL_SET(&opt, IGNORE_WHITESPACE);
317         opt.format_callback = cgit_diff_tree_cb;
318         opt.format_callback_data = fn;
319         if (prefix) {
320                 opt.nr_paths = 1;
321                 opt.paths = &prefix;
322                 prefixlen = strlen(prefix);
323                 opt.pathlens = &prefixlen;
324         }
325         diff_setup_done(&opt);
326
327         if (old_sha1 && !is_null_sha1(old_sha1))
328                 diff_tree_sha1(old_sha1, new_sha1, "", &opt);
329         else
330                 diff_root_tree_sha1(new_sha1, "", &opt);
331         diffcore_std(&opt);
332         diff_flush(&opt);
333 }
334
335 void cgit_diff_commit(struct commit *commit, filepair_fn fn, const char *prefix)
336 {
337         unsigned char *old_sha1 = NULL;
338
339         if (commit->parents)
340                 old_sha1 = commit->parents->item->object.sha1;
341         cgit_diff_tree(old_sha1, commit->object.sha1, fn, prefix,
342                        ctx.qry.ignorews);
343 }
344
345 int cgit_parse_snapshots_mask(const char *str)
346 {
347         const struct cgit_snapshot_format *f;
348         static const char *delim = " \t,:/|;";
349         int tl, sl, rv = 0;
350
351         /* favor legacy setting */
352         if(atoi(str))
353                 return 1;
354         for(;;) {
355                 str += strspn(str,delim);
356                 tl = strcspn(str,delim);
357                 if (!tl)
358                         break;
359                 for (f = cgit_snapshot_formats; f->suffix; f++) {
360                         sl = strlen(f->suffix);
361                         if((tl == sl && !strncmp(f->suffix, str, tl)) ||
362                            (tl == sl-1 && !strncmp(f->suffix+1, str, tl-1))) {
363                                 rv |= f->bit;
364                                 break;
365                         }
366                 }
367                 str += tl;
368         }
369         return rv;
370 }
371
372 typedef struct {
373         char * name;
374         char * value;
375 } cgit_env_var;
376
377 void cgit_prepare_repo_env(struct cgit_repo * repo)
378 {
379         cgit_env_var env_vars[] = {
380                 { .name = "CGIT_REPO_URL", .value = repo->url },
381                 { .name = "CGIT_REPO_NAME", .value = repo->name },
382                 { .name = "CGIT_REPO_PATH", .value = repo->path },
383                 { .name = "CGIT_REPO_OWNER", .value = repo->owner },
384                 { .name = "CGIT_REPO_DEFBRANCH", .value = repo->defbranch },
385                 { .name = "CGIT_REPO_SECTION", .value = repo->section },
386                 { .name = "CGIT_REPO_CLONE_URL", .value = repo->clone_url }
387         };
388         int env_var_count = ARRAY_SIZE(env_vars);
389         cgit_env_var *p, *q;
390         static char *warn = "cgit warning: failed to set env: %s=%s\n";
391
392         p = env_vars;
393         q = p + env_var_count;
394         for (; p < q; p++)
395                 if (p->value && setenv(p->name, p->value, 1))
396                         fprintf(stderr, warn, p->name, p->value);
397 }
398
399 int cgit_open_filter(struct cgit_filter *filter)
400 {
401
402         filter->old_stdout = chk_positive(dup(STDOUT_FILENO),
403                 "Unable to duplicate STDOUT");
404         chk_zero(pipe(filter->pipe_fh), "Unable to create pipe to subprocess");
405         filter->pid = chk_non_negative(fork(), "Unable to create subprocess");
406         if (filter->pid == 0) {
407                 close(filter->pipe_fh[1]);
408                 chk_non_negative(dup2(filter->pipe_fh[0], STDIN_FILENO),
409                         "Unable to use pipe as STDIN");
410                 execvp(filter->cmd, filter->argv);
411                 die("Unable to exec subprocess %s: %s (%d)", filter->cmd,
412                         strerror(errno), errno);
413         }
414         close(filter->pipe_fh[0]);
415         chk_non_negative(dup2(filter->pipe_fh[1], STDOUT_FILENO),
416                 "Unable to use pipe as STDOUT");
417         close(filter->pipe_fh[1]);
418         return 0;
419 }
420
421 int cgit_close_filter(struct cgit_filter *filter)
422 {
423         chk_non_negative(dup2(filter->old_stdout, STDOUT_FILENO),
424                 "Unable to restore STDOUT");
425         close(filter->old_stdout);
426         if (filter->pid < 0)
427                 return 0;
428         waitpid(filter->pid, &filter->exitstatus, 0);
429         if (WIFEXITED(filter->exitstatus) && !WEXITSTATUS(filter->exitstatus))
430                 return 0;
431         die("Subprocess %s exited abnormally", filter->cmd);
432 }
433
434 /* Read the content of the specified file into a newly allocated buffer,
435  * zeroterminate the buffer and return 0 on success, errno otherwise.
436  */
437 int readfile(const char *path, char **buf, size_t *size)
438 {
439         int fd, e;
440         struct stat st;
441
442         fd = open(path, O_RDONLY);
443         if (fd == -1)
444                 return errno;
445         if (fstat(fd, &st)) {
446                 e = errno;
447                 close(fd);
448                 return e;
449         }
450         if (!S_ISREG(st.st_mode)) {
451                 close(fd);
452                 return EISDIR;
453         }
454         *buf = xmalloc(st.st_size + 1);
455         *size = read_in_full(fd, *buf, st.st_size);
456         e = errno;
457         (*buf)[*size] = '\0';
458         close(fd);
459         return (*size == st.st_size ? 0 : e);
460 }
461
462 int is_token_char(char c)
463 {
464         return isalnum(c) || c == '_';
465 }
466
467 /* Replace name with getenv(name), return pointer to zero-terminating char
468  */
469 char *expand_macro(char *name, int maxlength)
470 {
471         char *value;
472         int len;
473
474         len = 0;
475         value = getenv(name);
476         if (value) {
477                 len = strlen(value);
478                 if (len > maxlength)
479                         len = maxlength;
480                 strncpy(name, value, len);
481         }
482         return name + len;
483 }
484
485 #define EXPBUFSIZE (1024 * 8)
486
487 /* Replace all tokens prefixed by '$' in the specified text with the
488  * value of the named environment variable.
489  * NB: the return value is a static buffer, i.e. it must be strdup'd
490  * by the caller.
491  */
492 char *expand_macros(const char *txt)
493 {
494         static char result[EXPBUFSIZE];
495         char *p, *start;
496         int len;
497
498         p = result;
499         start = NULL;
500         while (p < result + EXPBUFSIZE - 1 && txt && *txt) {
501                 *p = *txt;
502                 if (start) {
503                         if (!is_token_char(*txt)) {
504                                 if (p - start > 0) {
505                                         *p = '\0';
506                                         len = result + EXPBUFSIZE - start - 1;
507                                         p = expand_macro(start, len) - 1;
508                                 }
509                                 start = NULL;
510                                 txt--;
511                         }
512                         p++;
513                         txt++;
514                         continue;
515                 }
516                 if (*txt == '$') {
517                         start = p;
518                         txt++;
519                         continue;
520                 }
521                 p++;
522                 txt++;
523         }
524         *p = '\0';
525         if (start && p - start > 0) {
526                 len = result + EXPBUFSIZE - start - 1;
527                 p = expand_macro(start, len);
528                 *p = '\0';
529         }
530         return result;
531 }