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