op public repos

Blob

Date:: Fri Feb 10 22:19:50 2023 UTC
Message:: gotwebd: abort tree and blob as well
Actions:: History | Blame | Raw File
1 /*
2  * Copyright (c) 2020-2022 Tracey Emery <tracey@traceyemery.net>
3  * Copyright (c) 2018, 2019 Stefan Sperling <stsp@openbsd.org>
4  *
5  * Permission to use, copy, modify, and distribute this software for any
6  * purpose with or without fee is hereby granted, provided that the above
7  * copyright notice and this permission notice appear in all copies.
8  *
9  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16  */
17 
18 #include <sys/queue.h>
19 #include <sys/socket.h>
20 #include <sys/stat.h>
21 
22 #include <event.h>
23 #include <imsg.h>
24 #include <sha1.h>
25 #include <stdlib.h>
26 #include <stdio.h>
27 #include <string.h>
28 #include <unistd.h>
29 
30 #include "got_error.h"
31 #include "got_object.h"
32 #include "got_reference.h"
33 #include "got_repository.h"
34 #include "got_path.h"
35 #include "got_cancel.h"
36 #include "got_diff.h"
37 #include "got_commit_graph.h"
38 #include "got_blame.h"
39 #include "got_privsep.h"
40 
41 #include "proc.h"
42 #include "gotwebd.h"
43 
44 static const struct got_error *got_init_repo_commit(struct repo_commit **);
45 static const struct got_error *got_init_repo_tag(struct repo_tag **);
46 static const struct got_error *got_get_repo_commit(struct request *,
47     struct repo_commit *, struct got_commit_object *, struct got_reflist_head *,
48     struct got_object_id *);
49 static const struct got_error *got_gotweb_dupfd(int *, int *);
50 static const struct got_error *got_gotweb_openfile(FILE **, int *, int *);
51 static const struct got_error *got_gotweb_blame_cb(void *, int, int,
52     struct got_commit_object *,struct got_object_id *);
53 
54 const struct got_error *
55 got_gotweb_flushfile(FILE *f, int fd)
56 {
57 	if (fseek(f, 0, SEEK_SET) == -1)
58 		return got_error_from_errno("fseek");
59 
60 	if (ftruncate(fd, 0) == -1)
61 		return got_error_from_errno("ftruncate");
62 
63 	if (fsync(fd) == -1)
64 		return got_error_from_errno("fsync");
65 
66 	if (f && fclose(f) == EOF)
67 		return got_error_from_errno("fclose");
68 
69 	if (fd != -1 && close(fd) != -1)
70 		return got_error_from_errno("close");
71 
72 	return NULL;
73 }
74 
75 static const struct got_error *
76 got_gotweb_openfile(FILE **f, int *priv_fd, int *fd)
77 {
78 	*fd = dup(*priv_fd);
79 	if (*fd == -1)
80 		return got_error_from_errno("dup");
81 
82 	*f = fdopen(*fd, "w+");
83 	if (*f == NULL) {
84 		close(*fd);
85 		return got_error(GOT_ERR_PRIVSEP_NO_FD);
86 	}
87 
88 	return NULL;
89 }
90 
91 static const struct got_error *
92 got_gotweb_dupfd(int *priv_fd, int *fd)
93 {
94 	*fd = dup(*priv_fd);
95 
96 	if (*fd == -1)
97 		return got_error_from_errno("dup");
98 
99 	return NULL;
100 }
101 
102 const struct got_error *
103 got_get_repo_owner(char **owner, struct request *c)
104 {
105 	struct server *srv = c->srv;
106 	struct transport *t = c->t;
107 	struct got_repository *repo = t->repo;
108 	const char *gitconfig_owner;
109 
110 	*owner = NULL;
111 
112 	if (srv->show_repo_owner == 0)
113 		return NULL;
114 
115 	gitconfig_owner = got_repo_get_gitconfig_owner(repo);
116 	if (gitconfig_owner) {
117 		*owner = strdup(gitconfig_owner);
118 		if (*owner == NULL)
119 			return got_error_from_errno("strdup");
120 	} else {
121 		*owner = strdup("");
122 		if (*owner == NULL)
123 			return got_error_from_errno("strdup");
124 	}
125 	return NULL;
126 }
127 
128 const struct got_error *
129 got_get_repo_age(time_t *repo_age, struct request *c, const char *refname)
130 {
131 	const struct got_error *error = NULL;
132 	struct server *srv = c->srv;
133 	struct transport *t = c->t;
134 	struct got_repository *repo = t->repo;
135 	struct got_commit_object *commit = NULL;
136 	struct got_reflist_head refs;
137 	struct got_reflist_entry *re;
138 	time_t committer_time = 0, cmp_time = 0;
139 
140 	TAILQ_INIT(&refs);
141 
142 	if (srv->show_repo_age == 0)
143 		return NULL;
144 
145 	error = got_ref_list(&refs, repo, "refs/heads",
146 	    got_ref_cmp_by_name, NULL);
147 	if (error)
148 		goto done;
149 
150 	/*
151 	 * Find the youngest branch tip in the repository, or the age of
152 	 * the a specific branch tip if a name was provided by the caller.
153 	 */
154 	TAILQ_FOREACH(re, &refs, entry) {
155 		struct got_object_id *id = NULL;
156 
157 		if (refname && strcmp(got_ref_get_name(re->ref), refname) != 0)
158 			continue;
159 
160 		error = got_ref_resolve(&id, repo, re->ref);
161 		if (error)
162 			goto done;
163 
164 		error = got_object_open_as_commit(&commit, repo, id);
165 		free(id);
166 		if (error)
167 			goto done;
168 
169 		committer_time =
170 		    got_object_commit_get_committer_time(commit);
171 		got_object_commit_close(commit);
172 		if (cmp_time < committer_time)
173 			cmp_time = committer_time;
174 
175 		if (refname)
176 			break;
177 	}
178 
179 	if (cmp_time != 0)
180 		*repo_age = cmp_time;
181 done:
182 	got_ref_list_free(&refs);
183 	return error;
184 }
185 
186 static const struct got_error *
187 got_get_repo_commit(struct request *c, struct repo_commit *repo_commit,
188     struct got_commit_object *commit, struct got_reflist_head *refs,
189     struct got_object_id *id)
190 {
191 	const struct got_error *error = NULL;
192 	struct got_reflist_entry *re;
193 	struct got_object_id *id2 = NULL;
194 	struct got_object_qid *parent_id;
195 	struct transport *t = c->t;
196 	struct querystring *qs = c->t->qs;
197 	char *commit_msg = NULL, *commit_msg0;
198 
199 	TAILQ_FOREACH(re, refs, entry) {
200 		char *s;
201 		const char *name;
202 		struct got_tag_object *tag = NULL;
203 		struct got_object_id *ref_id;
204 		int cmp;
205 
206 		if (got_ref_is_symbolic(re->ref))
207 			continue;
208 
209 		name = got_ref_get_name(re->ref);
210 		if (strncmp(name, "refs/", 5) == 0)
211 			name += 5;
212 		if (strncmp(name, "got/", 4) == 0)
213 			continue;
214 		if (strncmp(name, "heads/", 6) == 0)
215 			name += 6;
216 		if (strncmp(name, "remotes/", 8) == 0) {
217 			name += 8;
218 			if (strstr(name, "/" GOT_REF_HEAD) != NULL)
219 				continue;
220 		}
221 		error = got_ref_resolve(&ref_id, t->repo, re->ref);
222 		if (error)
223 			return error;
224 		if (strncmp(name, "tags/", 5) == 0) {
225 			error = got_object_open_as_tag(&tag, t->repo, ref_id);
226 			if (error) {
227 				if (error->code != GOT_ERR_OBJ_TYPE) {
228 					free(ref_id);
229 					continue;
230 				}
231 				/*
232 				 * Ref points at something other
233 				 * than a tag.
234 				 */
235 				error = NULL;
236 				tag = NULL;
237 			}
238 		}
239 		cmp = got_object_id_cmp(tag ?
240 		    got_object_tag_get_object_id(tag) : ref_id, id);
241 		free(ref_id);
242 		if (tag)
243 			got_object_tag_close(tag);
244 		if (cmp != 0)
245 			continue;
246 		s = repo_commit->refs_str;
247 		if (asprintf(&repo_commit->refs_str, "%s%s%s", s ? s : "",
248 		    s ? ", " : "", name) == -1) {
249 			error = got_error_from_errno("asprintf");
250 			free(s);
251 			repo_commit->refs_str = NULL;
252 			return error;
253 		}
254 		free(s);
255 	}
256 
257 	error = got_object_id_str(&repo_commit->commit_id, id);
258 	if (error)
259 		return error;
260 
261 	error = got_object_id_str(&repo_commit->tree_id,
262 	    got_object_commit_get_tree_id(commit));
263 	if (error)
264 		return error;
265 
266 	if (qs->action == DIFF) {
267 		parent_id = STAILQ_FIRST(
268 		    got_object_commit_get_parent_ids(commit));
269 		if (parent_id != NULL) {
270 			id2 = got_object_id_dup(&parent_id->id);
271 			error = got_object_id_str(&repo_commit->parent_id, id2);
272 			if (error)
273 				return error;
274 			free(id2);
275 		} else {
276 			repo_commit->parent_id = strdup("/dev/null");
277 			if (repo_commit->parent_id == NULL) {
278 				error = got_error_from_errno("strdup");
279 				return error;
280 			}
281 		}
282 	}
283 
284 	repo_commit->committer_time =
285 	    got_object_commit_get_committer_time(commit);
286 
287 	repo_commit->author =
288 	    strdup(got_object_commit_get_author(commit));
289 	if (repo_commit->author == NULL) {
290 		error = got_error_from_errno("strdup");
291 		return error;
292 	}
293 	repo_commit->committer =
294 	    strdup(got_object_commit_get_committer(commit));
295 	if (repo_commit->committer == NULL) {
296 		error = got_error_from_errno("strdup");
297 		return error;
298 	}
299 	error = got_object_commit_get_logmsg(&commit_msg0, commit);
300 	if (error)
301 		return error;
302 
303 	commit_msg = commit_msg0;
304 	while (*commit_msg == '\n')
305 		commit_msg++;
306 
307 	repo_commit->commit_msg = strdup(commit_msg);
308 	if (repo_commit->commit_msg == NULL)
309 		error = got_error_from_errno("strdup");
310 	free(commit_msg0);
311 	return error;
312 }
313 
314 const struct got_error *
315 got_get_repo_commits(struct request *c, int limit)
316 {
317 	const struct got_error *error = NULL;
318 	struct got_object_id *id = NULL;
319 	struct got_commit_graph *graph = NULL;
320 	struct got_commit_object *commit = NULL;
321 	struct got_reflist_head refs;
322 	struct got_reference *ref = NULL;
323 	struct repo_commit *repo_commit = NULL;
324 	struct server *srv = c->srv;
325 	struct transport *t = c->t;
326 	struct got_repository *repo = t->repo;
327 	struct querystring *qs = t->qs;
328 	struct repo_dir *repo_dir = t->repo_dir;
329 	char *in_repo_path = NULL, *repo_path = NULL, *file_path = NULL;
330 	int chk_next = 0, chk_multi = 0;
331 
332 	TAILQ_INIT(&refs);
333 
334 	if (qs->file != NULL && strlen(qs->file) > 0)
335 		if (asprintf(&file_path, "%s/%s", qs->folder ? qs->folder : "",
336 		    qs->file) == -1)
337 			return got_error_from_errno("asprintf");
338 
339 	if (asprintf(&repo_path, "%s/%s", srv->repos_path,
340 	    repo_dir->name) == -1) {
341 		error = got_error_from_errno("asprintf");
342 		goto done;
343 	}
344 
345 	if (qs->commit) {
346 		error = got_repo_match_object_id_prefix(&id, qs->commit,
347 		    GOT_OBJ_TYPE_COMMIT, repo);
348 		if (error)
349 			goto done;
350 	} else {
351 		error = got_ref_open(&ref, repo, qs->headref, 0);
352 		if (error)
353 			goto done;
354 
355 		error = got_ref_resolve(&id, repo, ref);
356 		if (error)
357 			goto done;
358 	}
359 
360 	error = got_repo_map_path(&in_repo_path, repo, repo_path);
361 	if (error)
362 		goto done;
363 
364 	error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
365 	if (error)
366 		goto done;
367 
368 	if (qs->file != NULL && strlen(qs->file) > 0) {
369 		error = got_commit_graph_open(&graph, file_path, 0);
370 		if (error)
371 			goto done;
372 	} else {
373 		error = got_commit_graph_open(&graph, in_repo_path, 0);
374 		if (error)
375 			goto done;
376 	}
377 
378 	error = got_commit_graph_iter_start(graph, id, repo, NULL, NULL);
379 	if (error)
380 		goto done;
381 
382 	for (;;) {
383 		struct got_object_id next_id;
384 
385 		error = got_commit_graph_iter_next(&next_id, graph, repo, NULL,
386 		    NULL);
387 		if (error) {
388 			if (error->code == GOT_ERR_ITER_COMPLETED)
389 				error = NULL;
390 			goto done;
391 		}
392 
393 		error = got_object_open_as_commit(&commit, repo, &next_id);
394 		if (error)
395 			goto done;
396 
397 		error = got_init_repo_commit(&repo_commit);
398 		if (error)
399 			goto done;
400 
401 		error = got_get_repo_commit(c, repo_commit, commit,
402 		    &refs, &next_id);
403 		if (error) {
404 			gotweb_free_repo_commit(repo_commit);
405 			goto done;
406 		}
407 
408 		TAILQ_INSERT_TAIL(&t->repo_commits, repo_commit, entry);
409 
410 		if (!chk_multi || limit != 1 ||
411 		    srv->max_commits_display == 1) {
412 			chk_multi = 1;
413 
414 			/*
415 			 * check for one more commit before breaking,
416 			 * so we know whether to navigate through briefs
417 			 * commits and summary
418 			 */
419 			if (chk_next && (qs->action == BRIEFS ||
420 			    qs->action == COMMITS || qs->action == SUMMARY)) {
421 				t->more_id = strdup(repo_commit->commit_id);
422 				if (t->more_id == NULL) {
423 					error = got_error_from_errno("strdup");
424 					goto done;
425 				}
426 				got_object_commit_close(commit);
427 				commit = NULL;
428 				TAILQ_REMOVE(&t->repo_commits, repo_commit,
429 				    entry);
430 				gotweb_free_repo_commit(repo_commit);
431 				goto done;
432 			}
433 		}
434 		if (error || (limit && --limit == 0)) {
435 			if (qs->file != NULL && *qs->file != '\0')
436 				if (chk_multi == 0)
437 					break;
438 			chk_next = 1;
439 		}
440 		if (commit) {
441 			got_object_commit_close(commit);
442 			commit = NULL;
443 		}
444 	}
445 done:
446 	if (ref)
447 		got_ref_close(ref);
448 	if (commit)
449 		got_object_commit_close(commit);
450 	if (graph)
451 		got_commit_graph_close(graph);
452 	got_ref_list_free(&refs);
453 	free(in_repo_path);
454 	free(file_path);
455 	free(repo_path);
456 	free(id);
457 	return error;
458 }
459 
460 const struct got_error *
461 got_get_repo_tags(struct request *c, int limit)
462 {
463 	const struct got_error *error = NULL;
464 	struct got_object_id *id = NULL;
465 	struct got_commit_object *commit = NULL;
466 	struct got_reflist_head refs;
467 	struct got_reference *ref;
468 	struct got_reflist_entry *re;
469 	struct server *srv = c->srv;
470 	struct transport *t = c->t;
471 	struct got_repository *repo = t->repo;
472 	struct querystring *qs = t->qs;
473 	struct repo_dir *repo_dir = t->repo_dir;
474 	struct got_tag_object *tag = NULL;
475 	struct repo_tag *rt = NULL, *trt = NULL;
476 	char *in_repo_path = NULL, *repo_path = NULL, *id_str = NULL;
477 	char *tag_commit = NULL, *tag_commit0 = NULL;
478 	char *commit_msg = NULL, *commit_msg0 = NULL;
479 	int chk_next = 0, chk_multi = 1, commit_found = 0, c_cnt = 0;
480 
481 	TAILQ_INIT(&refs);
482 
483 	if (asprintf(&repo_path, "%s/%s", srv->repos_path,
484 	    repo_dir->name) == -1)
485 		return got_error_from_errno("asprintf");
486 
487 	if (qs->commit == NULL && (qs->action == TAGS || qs->action == RSS)) {
488 		error = got_ref_open(&ref, repo, qs->headref, 0);
489 		if (error)
490 			goto err;
491 		error = got_ref_resolve(&id, repo, ref);
492 		got_ref_close(ref);
493 		if (error)
494 			goto err;
495 	} else if (qs->commit == NULL && qs->action == TAG) {
496 		error = got_error_msg(GOT_ERR_EOF, "commit id missing");
497 		goto err;
498 	} else {
499 		error = got_repo_match_object_id_prefix(&id, qs->commit,
500 		    GOT_OBJ_TYPE_COMMIT, repo);
501 		if (error)
502 			goto err;
503 	}
504 
505 	if (qs->action != SUMMARY && qs->action != TAGS) {
506 		error = got_object_open_as_commit(&commit, repo, id);
507 		if (error)
508 			goto err;
509 		error = got_object_commit_get_logmsg(&commit_msg0, commit);
510 		if (error)
511 			goto err;
512 		if (commit) {
513 			got_object_commit_close(commit);
514 			commit = NULL;
515 		}
516 	}
517 
518 	error = got_repo_map_path(&in_repo_path, repo, repo_path);
519 	if (error)
520 		goto err;
521 
522 	error = got_ref_list(&refs, repo, "refs/tags", got_ref_cmp_tags,
523 	   repo);
524 	if (error)
525 		goto err;
526 
527 	if (limit == 1)
528 		chk_multi = 0;
529 
530 	/*
531 	 * XXX: again, see previous message about caching
532 	 */
533 
534 	TAILQ_FOREACH(re, &refs, entry) {
535 		struct repo_tag *new_repo_tag = NULL;
536 		error = got_init_repo_tag(&new_repo_tag);
537 		if (error)
538 			goto err;
539 
540 		TAILQ_INSERT_TAIL(&t->repo_tags, new_repo_tag, entry);
541 
542 		new_repo_tag->tag_name = strdup(got_ref_get_name(re->ref));
543 		if (new_repo_tag->tag_name == NULL) {
544 			error = got_error_from_errno("strdup");
545 			goto err;
546 		}
547 
548 		free(id);
549 		id = NULL;
550 
551 		free(id_str);
552 		id_str = NULL;
553 
554 		error = got_ref_resolve(&id, repo, re->ref);
555 		if (error)
556 			goto done;
557 
558 		if (tag)
559 			got_object_tag_close(tag);
560 		error = got_object_open_as_tag(&tag, repo, id);
561 		if (error) {
562 			if (error->code != GOT_ERR_OBJ_TYPE)
563 				goto done;
564 			/* "lightweight" tag */
565 			error = got_object_open_as_commit(&commit, repo, id);
566 			if (error)
567 				goto done;
568 			new_repo_tag->tagger =
569 			    strdup(got_object_commit_get_committer(commit));
570 			if (new_repo_tag->tagger == NULL) {
571 				error = got_error_from_errno("strdup");
572 				goto err;
573 			}
574 			new_repo_tag->tagger_time =
575 			    got_object_commit_get_committer_time(commit);
576 			error = got_object_id_str(&id_str, id);
577 			if (error)
578 				goto err;
579 		} else {
580 			new_repo_tag->tagger =
581 			    strdup(got_object_tag_get_tagger(tag));
582 			if (new_repo_tag->tagger == NULL) {
583 				error = got_error_from_errno("strdup");
584 				goto err;
585 			}
586 			new_repo_tag->tagger_time =
587 			    got_object_tag_get_tagger_time(tag);
588 			error = got_object_id_str(&id_str,
589 			    got_object_tag_get_object_id(tag));
590 			if (error)
591 				goto err;
592 		}
593 
594 		new_repo_tag->commit_id = strdup(id_str);
595 		if (new_repo_tag->commit_id == NULL)
596 			goto err;
597 
598 		if (commit_found == 0 && qs->commit != NULL &&
599 		    strncmp(id_str, qs->commit, strlen(id_str)) != 0)
600 			continue;
601 		else
602 			commit_found = 1;
603 
604 		t->tag_count++;
605 
606 		/*
607 		 * check for one more commit before breaking,
608 		 * so we know whether to navigate through briefs
609 		 * commits and summary
610 		 */
611 		if (chk_next) {
612 			t->next_id = strdup(new_repo_tag->commit_id);
613 			if (t->next_id == NULL) {
614 				error = got_error_from_errno("strdup");
615 				goto err;
616 			}
617 			if (commit) {
618 				got_object_commit_close(commit);
619 				commit = NULL;
620 			}
621 			TAILQ_REMOVE(&t->repo_tags, new_repo_tag, entry);
622 			gotweb_free_repo_tag(new_repo_tag);
623 			goto done;
624 		}
625 
626 		if (commit) {
627 			error = got_object_commit_get_logmsg(&tag_commit0,
628 			    commit);
629 			if (error)
630 				goto err;
631 			got_object_commit_close(commit);
632 			commit = NULL;
633 		} else {
634 			tag_commit0 = strdup(got_object_tag_get_message(tag));
635 			if (tag_commit0 == NULL) {
636 				error = got_error_from_errno("strdup");
637 				goto err;
638 			}
639 		}
640 
641 		tag_commit = tag_commit0;
642 		while (*tag_commit == '\n')
643 			tag_commit++;
644 		new_repo_tag->tag_commit = strdup(tag_commit);
645 		if (new_repo_tag->tag_commit == NULL) {
646 			error = got_error_from_errno("strdup");
647 			free(tag_commit0);
648 			goto err;
649 		}
650 		free(tag_commit0);
651 
652 		if (qs->action != SUMMARY && qs->action != TAGS) {
653 			commit_msg = commit_msg0;
654 			while (*commit_msg == '\n')
655 				commit_msg++;
656 
657 			new_repo_tag->commit_msg = strdup(commit_msg);
658 			if (new_repo_tag->commit_msg == NULL) {
659 				error = got_error_from_errno("strdup");
660 				goto err;
661 			}
662 		}
663 
664 		if (limit && --limit == 0) {
665 			if (chk_multi == 0)
666 				break;
667 			chk_next = 1;
668 		}
669 	}
670 
671 done:
672 	/*
673 	 * we have tailq populated, so find previous commit id
674 	 * for navigation through briefs and commits
675 	 */
676 	if (t->tag_count == 0) {
677 		TAILQ_FOREACH_SAFE(rt, &t->repo_tags, entry, trt) {
678 			TAILQ_REMOVE(&t->repo_tags, rt, entry);
679 			gotweb_free_repo_tag(rt);
680 		}
681 	}
682 	if (t->tag_count > 0 && t->prev_id == NULL && qs->commit != NULL) {
683 		commit_found = 0;
684 		TAILQ_FOREACH_REVERSE(rt, &t->repo_tags, repo_tags_head,
685 		    entry) {
686 			if (commit_found == 0 && rt->commit_id != NULL &&
687 			    strcmp(qs->commit, rt->commit_id) != 0) {
688 				continue;
689 			} else
690 				commit_found = 1;
691 			if (c_cnt == srv->max_commits_display ||
692 			    rt == TAILQ_FIRST(&t->repo_tags)) {
693 				t->prev_id = strdup(rt->commit_id);
694 				if (t->prev_id == NULL)
695 					error = got_error_from_errno("strdup");
696 				break;
697 			}
698 			c_cnt++;
699 		}
700 	}
701 err:
702 	if (commit)
703 		got_object_commit_close(commit);
704 	if (tag)
705 		got_object_tag_close(tag);
706 	got_ref_list_free(&refs);
707 	free(commit_msg0);
708 	free(in_repo_path);
709 	free(repo_path);
710 	free(id);
711 	free(id_str);
712 	return error;
713 }
714 
715 int
716 got_output_repo_tree(struct request *c,
717     int (*cb)(struct template *, struct got_tree_entry *))
718 {
719 	const struct got_error *error = NULL;
720 	struct transport *t = c->t;
721 	struct got_commit_object *commit = NULL;
722 	struct got_repository *repo = t->repo;
723 	struct querystring *qs = t->qs;
724 	struct repo_commit *rc = NULL;
725 	struct got_object_id *tree_id = NULL, *commit_id = NULL;
726 	struct got_reflist_head refs;
727 	struct got_tree_object *tree = NULL;
728 	struct got_tree_entry *te;
729 	struct repo_dir *repo_dir = t->repo_dir;
730 	char *escaped_name = NULL, *path = NULL;
731 	int nentries, i;
732 
733 	TAILQ_INIT(&refs);
734 
735 	rc = TAILQ_FIRST(&t->repo_commits);
736 
737 	if (qs->folder != NULL) {
738 		path = strdup(qs->folder);
739 		if (path == NULL) {
740 			error = got_error_from_errno("strdup");
741 			goto done;
742 		}
743 	} else {
744 		error = got_repo_map_path(&path, repo, repo_dir->path);
745 		if (error)
746 			goto done;
747 	}
748 
749 	error = got_repo_match_object_id(&commit_id, NULL, rc->commit_id,
750 	    GOT_OBJ_TYPE_COMMIT, &refs, repo);
751 	if (error)
752 		goto done;
753 
754 	error = got_object_open_as_commit(&commit, repo, commit_id);
755 	if (error)
756 		goto done;
757 
758 	error = got_object_id_by_path(&tree_id, repo, commit, path);
759 	if (error)
760 		goto done;
761 
762 	error = got_object_open_as_tree(&tree, repo, tree_id);
763 	if (error)
764 		goto done;
765 
766 	nentries = got_object_tree_get_nentries(tree);
767 
768 	for (i = 0; i < nentries; i++) {
769 		te = got_object_tree_get_entry(tree, i);
770 		if (cb(c->tp, te) == -1) {
771 			error = got_error(GOT_ERR_CANCELLED);
772 			break;
773 		}
774 	}
775 done:
776 	free(escaped_name);
777 	free(path);
778 	got_ref_list_free(&refs);
779 	if (commit)
780 		got_object_commit_close(commit);
781 	if (tree)
782 		got_object_tree_close(tree);
783 	free(commit_id);
784 	free(tree_id);
785 	if (error) {
786 		log_warnx("%s: %s", __func__, error->msg);
787 		return -1;
788 	}
789 	return 0;
790 }
791 
792 const struct got_error *
793 got_open_blob_for_output(struct got_blob_object **blob, int *fd,
794     int *binary, struct request *c)
795 {
796 	const struct got_error *error = NULL;
797 	struct transport *t = c->t;
798 	struct got_repository *repo = t->repo;
799 	struct querystring *qs = c->t->qs;
800 	struct got_commit_object *commit = NULL;
801 	struct got_object_id *commit_id = NULL;
802 	struct got_reflist_head refs;
803 	char *path = NULL, *in_repo_path = NULL;
804 	int obj_type;
805 
806 	TAILQ_INIT(&refs);
807 
808 	*blob = NULL;
809 	*fd = -1;
810 	*binary = 0;
811 
812 	if (asprintf(&path, "%s%s%s", qs->folder ? qs->folder : "",
813 	    qs->folder ? "/" : "", qs->file) == -1) {
814 		error = got_error_from_errno("asprintf");
815 		goto done;
816 	}
817 
818 	error = got_repo_map_path(&in_repo_path, repo, path);
819 	if (error)
820 		goto done;
821 
822 	error = got_repo_match_object_id(&commit_id, NULL, qs->commit,
823 	    GOT_OBJ_TYPE_COMMIT, &refs, repo);
824 	if (error)
825 		goto done;
826 
827 	error = got_object_open_as_commit(&commit, repo, commit_id);
828 	if (error)
829 		goto done;
830 
831 	error = got_object_id_by_path(&commit_id, repo, commit, in_repo_path);
832 	if (error)
833 		goto done;
834 
835 	if (commit_id == NULL) {
836 		error = got_error(GOT_ERR_NO_OBJ);
837 		goto done;
838 	}
839 
840 	error = got_object_get_type(&obj_type, repo, commit_id);
841 	if (error)
842 		goto done;
843 
844 	if (obj_type != GOT_OBJ_TYPE_BLOB) {
845 		error = got_error(GOT_ERR_OBJ_TYPE);
846 		goto done;
847 	}
848 
849 	error = got_gotweb_dupfd(&c->priv_fd[BLOB_FD_1], fd);
850 	if (error)
851 		goto done;
852 
853 	error = got_object_open_as_blob(blob, repo, commit_id, BUF, *fd);
854 	if (error)
855 		goto done;
856 
857 	error = got_object_blob_is_binary(binary, *blob);
858 	if (error)
859 		goto done;
860 
861  done:
862 	if (commit)
863 		got_object_commit_close(commit);
864 
865 	if (error) {
866 		if (*fd != -1)
867 			close(*fd);
868 		if (*blob)
869 			got_object_blob_close(*blob);
870 		*fd = -1;
871 		*blob = NULL;
872 	}
873 
874 	free(in_repo_path);
875 	free(commit_id);
876 	free(path);
877 	return error;
878 }
879 
880 int
881 got_output_blob_by_lines(struct template *tp, struct got_blob_object *blob,
882     int (*cb)(struct template *, const char *, size_t))
883 {
884 	const struct got_error	*err;
885 	char			*line = NULL;
886 	size_t			 linesize = 0;
887 	size_t			 lineno = 0;
888 	ssize_t			 linelen = 0;
889 
890 	for (;;) {
891 		err = got_object_blob_getline(&line, &linelen, &linesize,
892 		    blob);
893 		if (err || linelen == -1)
894 			break;
895 		lineno++;
896 		if (cb(tp, line, lineno) == -1) {
897 			err = got_error(GOT_ERR_CANCELLED);
898 			break;
899 		}
900 	}
901 
902 	free(line);
903 
904 	if (err) {
905 		log_warnx("%s: got_object_blob_getline failed: %s",
906 		    __func__, err->msg);
907 		return -1;
908 	}
909 	return 0;
910 }
911 
912 struct blame_cb_args {
913 	struct blame_line *lines;
914 	int nlines;
915 	int nlines_prec;
916 	int lineno_cur;
917 	off_t *line_offsets;
918 	FILE *f;
919 	struct got_repository *repo;
920 	struct request *c;
921 	got_render_blame_line_cb cb;
922 };
923 
924 static const struct got_error *
925 got_gotweb_blame_cb(void *arg, int nlines, int lineno,
926     struct got_commit_object *commit, struct got_object_id *id)
927 {
928 	const struct got_error *err = NULL;
929 	struct blame_cb_args *a = arg;
930 	struct blame_line *bline;
931 	struct request *c = a->c;
932 	char *line = NULL;
933 	size_t linesize = 0;
934 	off_t offset;
935 	struct tm tm;
936 	time_t committer_time;
937 
938 	if (nlines != a->nlines ||
939 	    (lineno != -1 && lineno < 1) || lineno > a->nlines)
940 		return got_error(GOT_ERR_RANGE);
941 
942 	if (lineno == -1)
943 		return NULL; /* no change in this commit */
944 
945 	/* Annotate this line. */
946 	bline = &a->lines[lineno - 1];
947 	if (bline->annotated)
948 		return NULL;
949 	err = got_object_id_str(&bline->id_str, id);
950 	if (err)
951 		return err;
952 
953 	bline->committer = strdup(got_object_commit_get_committer(commit));
954 	if (bline->committer == NULL) {
955 		err = got_error_from_errno("strdup");
956 		goto done;
957 	}
958 
959 	committer_time = got_object_commit_get_committer_time(commit);
960 	if (gmtime_r(&committer_time, &tm) == NULL)
961 		return got_error_from_errno("gmtime_r");
962 	if (strftime(bline->datebuf, sizeof(bline->datebuf), "%G-%m-%d",
963 	    &tm) == 0) {
964 		err = got_error(GOT_ERR_NO_SPACE);
965 		goto done;
966 	}
967 	bline->annotated = 1;
968 
969 	/* Print lines annotated so far. */
970 	bline = &a->lines[a->lineno_cur - 1];
971 	if (!bline->annotated)
972 		goto done;
973 
974 	offset = a->line_offsets[a->lineno_cur - 1];
975 	if (fseeko(a->f, offset, SEEK_SET) == -1) {
976 		err = got_error_from_errno("fseeko");
977 		goto done;
978 	}
979 
980 	while (a->lineno_cur <= a->nlines && bline->annotated) {
981 		if (getline(&line, &linesize, a->f) == -1) {
982 			if (ferror(a->f))
983 				err = got_error_from_errno("getline");
984 			break;
985 		}
986 
987 		if (a->cb(c->tp, line, bline, a->nlines_prec,
988 		    a->lineno_cur) == -1) {
989 			err = got_error(GOT_ERR_CANCELLED);
990 			break;
991 		}
992 
993 		a->lineno_cur++;
994 		bline = &a->lines[a->lineno_cur - 1];
995 	}
996 done:
997 	free(line);
998 	return err;
999 }
1000 
1001 const struct got_error *
1002 got_output_file_blame(struct request *c, got_render_blame_line_cb cb)
1003 {
1004 	const struct got_error *error = NULL;
1005 	struct transport *t = c->t;
1006 	struct got_repository *repo = t->repo;
1007 	struct querystring *qs = c->t->qs;
1008 	struct got_object_id *obj_id = NULL, *commit_id = NULL;
1009 	struct got_commit_object *commit = NULL;
1010 	struct got_reflist_head refs;
1011 	struct got_blob_object *blob = NULL;
1012 	char *path = NULL, *in_repo_path = NULL;
1013 	struct blame_cb_args bca;
1014 	int i, obj_type, fd1 = -1, fd2 = -1, fd3 = -1, fd4 = -1, fd5 = -1;
1015 	int fd6 = -1;
1016 	off_t filesize;
1017 	FILE *f1 = NULL, *f2 = NULL;
1018 
1019 	TAILQ_INIT(&refs);
1020 	bca.f = NULL;
1021 	bca.lines = NULL;
1022 	bca.cb = cb;
1023 
1024 	if (asprintf(&path, "%s%s%s", qs->folder ? qs->folder : "",
1025 	    qs->folder ? "/" : "", qs->file) == -1) {
1026 		error = got_error_from_errno("asprintf");
1027 		goto done;
1028 	}
1029 
1030 	error = got_repo_map_path(&in_repo_path, repo, path);
1031 	if (error)
1032 		goto done;
1033 
1034 	error = got_repo_match_object_id(&commit_id, NULL, qs->commit,
1035 	    GOT_OBJ_TYPE_COMMIT, &refs, repo);
1036 	if (error)
1037 		goto done;
1038 
1039 	error = got_object_open_as_commit(&commit, repo, commit_id);
1040 	if (error)
1041 		goto done;
1042 
1043 	error = got_object_id_by_path(&obj_id, repo, commit, in_repo_path);
1044 	if (error)
1045 		goto done;
1046 
1047 	error = got_object_get_type(&obj_type, repo, obj_id);
1048 	if (error)
1049 		goto done;
1050 
1051 	if (obj_type != GOT_OBJ_TYPE_BLOB) {
1052 		error = got_error(GOT_ERR_OBJ_TYPE);
1053 		goto done;
1054 	}
1055 
1056 	error = got_gotweb_openfile(&bca.f, &c->priv_fd[BLAME_FD_1], &fd1);
1057 	if (error)
1058 		goto done;
1059 
1060 	error = got_gotweb_dupfd(&c->priv_fd[BLAME_FD_2], &fd2);
1061 	if (error)
1062 		goto done;
1063 
1064 	error = got_object_open_as_blob(&blob, repo, obj_id, BUF, fd2);
1065 	if (error)
1066 		goto done;
1067 
1068 	error = got_object_blob_dump_to_file(&filesize, &bca.nlines,
1069 	    &bca.line_offsets, bca.f, blob);
1070 	if (error || bca.nlines == 0)
1071 		goto done;
1072 
1073 	/* Don't include \n at EOF in the blame line count. */
1074 	if (bca.line_offsets[bca.nlines - 1] == filesize)
1075 		bca.nlines--;
1076 
1077 	bca.lines = calloc(bca.nlines, sizeof(*bca.lines));
1078 	if (bca.lines == NULL) {
1079 		error = got_error_from_errno("calloc");
1080 		goto done;
1081 	}
1082 	bca.lineno_cur = 1;
1083 	bca.nlines_prec = 0;
1084 	i = bca.nlines;
1085 	while (i > 0) {
1086 		i /= 10;
1087 		bca.nlines_prec++;
1088 	}
1089 	bca.repo = repo;
1090 	bca.c = c;
1091 
1092 	error = got_gotweb_dupfd(&c->priv_fd[BLAME_FD_3], &fd3);
1093 	if (error)
1094 		goto done;
1095 
1096 	error = got_gotweb_dupfd(&c->priv_fd[BLAME_FD_4], &fd4);
1097 	if (error)
1098 		goto done;
1099 
1100 	error = got_gotweb_openfile(&f1, &c->priv_fd[BLAME_FD_5], &fd5);
1101 	if (error)
1102 		goto done;
1103 
1104 	error = got_gotweb_openfile(&f2, &c->priv_fd[BLAME_FD_6], &fd6);
1105 	if (error)
1106 		goto done;
1107 
1108 	error = got_blame(in_repo_path, commit_id, repo,
1109 	    GOT_DIFF_ALGORITHM_MYERS, got_gotweb_blame_cb, &bca, NULL, NULL,
1110 	    fd3, fd4, f1, f2);
1111 
1112 done:
1113 	if (bca.lines) {
1114 		free(bca.line_offsets);
1115 		for (i = 0; i < bca.nlines; i++) {
1116 			struct blame_line *bline = &bca.lines[i];
1117 			free(bline->id_str);
1118 			free(bline->committer);
1119 		}
1120 	}
1121 	free(bca.lines);
1122 	if (fd2 != -1 && close(fd2) == -1 && error == NULL)
1123 		error = got_error_from_errno("close");
1124 	if (fd3 != -1 && close(fd3) == -1 && error == NULL)
1125 		error = got_error_from_errno("close");
1126 	if (fd4 != -1 && close(fd4) == -1 && error == NULL)
1127 		error = got_error_from_errno("close");
1128 	if (bca.f) {
1129 		const struct got_error *bca_err =
1130 		    got_gotweb_flushfile(bca.f, fd1);
1131 		if (error == NULL)
1132 			error = bca_err;
1133 	}
1134 	if (f1) {
1135 		const struct got_error *f1_err =
1136 		    got_gotweb_flushfile(f1, fd5);
1137 		if (error == NULL)
1138 			error = f1_err;
1139 	}
1140 	if (f2) {
1141 		const struct got_error *f2_err =
1142 		    got_gotweb_flushfile(f2, fd6);
1143 		if (error == NULL)
1144 			error = f2_err;
1145 	}
1146 	if (commit)
1147 		got_object_commit_close(commit);
1148 	if (blob)
1149 		got_object_blob_close(blob);
1150 	free(in_repo_path);
1151 	free(commit_id);
1152 	free(obj_id);
1153 	free(path);
1154 	got_ref_list_free(&refs);
1155 	return error;
1156 }
1157 
1158 const struct got_error *
1159 got_open_diff_for_output(FILE **fp, int *fd, struct request *c)
1160 {
1161 	const struct got_error *error = NULL;
1162 	struct transport *t = c->t;
1163 	struct got_repository *repo = t->repo;
1164 	struct repo_commit *rc = NULL;
1165 	struct got_object_id *id1 = NULL, *id2 = NULL;
1166 	struct got_reflist_head refs;
1167 	FILE *f1 = NULL, *f2 = NULL, *f3 = NULL;
1168 	int obj_type, fd1, fd2, fd3, fd4 = -1, fd5 = -1;
1169 
1170 	*fp = NULL;
1171 	*fd = -1;
1172 
1173 	TAILQ_INIT(&refs);
1174 
1175 	error = got_gotweb_openfile(&f1, &c->priv_fd[DIFF_FD_1], &fd1);
1176 	if (error)
1177 		return error;
1178 
1179 	error = got_gotweb_openfile(&f2, &c->priv_fd[DIFF_FD_2], &fd2);
1180 	if (error)
1181 		return error;
1182 
1183 	error = got_gotweb_openfile(&f3, &c->priv_fd[DIFF_FD_3], &fd3);
1184 	if (error)
1185 		return error;
1186 
1187 	rc = TAILQ_FIRST(&t->repo_commits);
1188 
1189 	if (rc->parent_id != NULL &&
1190 	    strncmp(rc->parent_id, "/dev/null", 9) != 0) {
1191 		error = got_repo_match_object_id(&id1, NULL,
1192 		    rc->parent_id, GOT_OBJ_TYPE_ANY,
1193 		    &refs, repo);
1194 		if (error)
1195 			goto done;
1196 	}
1197 
1198 	error = got_repo_match_object_id(&id2, NULL, rc->commit_id,
1199 	    GOT_OBJ_TYPE_ANY, &refs, repo);
1200 	if (error)
1201 		goto done;
1202 
1203 	error = got_object_get_type(&obj_type, repo, id2);
1204 	if (error)
1205 		goto done;
1206 
1207 	error = got_gotweb_dupfd(&c->priv_fd[DIFF_FD_4], &fd4);
1208 	if (error)
1209 		goto done;
1210 
1211 	error = got_gotweb_dupfd(&c->priv_fd[DIFF_FD_5], &fd5);
1212 	if (error)
1213 		goto done;
1214 
1215 	switch (obj_type) {
1216 	case GOT_OBJ_TYPE_BLOB:
1217 		error = got_diff_objects_as_blobs(NULL, NULL, f1, f2, fd4, fd5,
1218 		     id1, id2, NULL, NULL, GOT_DIFF_ALGORITHM_MYERS, 3, 0, 0,
1219 		     NULL, repo, f3);
1220 		break;
1221 	case GOT_OBJ_TYPE_TREE:
1222 		error = got_diff_objects_as_trees(NULL, NULL, f1, f2, fd4, fd5,
1223 		    id1, id2, NULL, "", "",  GOT_DIFF_ALGORITHM_MYERS, 3, 0, 0,
1224 		    NULL, repo, f3);
1225 		break;
1226 	case GOT_OBJ_TYPE_COMMIT:
1227 		error = got_diff_objects_as_commits(NULL, NULL, f1, f2, fd4,
1228 		    fd5, id1, id2, NULL,  GOT_DIFF_ALGORITHM_MYERS, 3, 0, 0,
1229 		    NULL, repo, f3);
1230 		break;
1231 	default:
1232 		error = got_error(GOT_ERR_OBJ_TYPE);
1233 	}
1234 	if (error)
1235 		goto done;
1236 
1237 	if (fseek(f1, 0, SEEK_SET) == -1) {
1238 		error = got_ferror(f1, GOT_ERR_IO);
1239 		goto done;
1240 	}
1241 
1242 	if (fseek(f2, 0, SEEK_SET) == -1) {
1243 		error = got_ferror(f2, GOT_ERR_IO);
1244 		goto done;
1245 	}
1246 
1247 	if (fseek(f3, 0, SEEK_SET) == -1) {
1248 		error = got_ferror(f3, GOT_ERR_IO);
1249 		goto done;
1250 	}
1251 
1252 	*fp = f3;
1253 	*fd = fd3;
1254 
1255 done:
1256 	if (fd4 != -1 && close(fd4) == -1 && error == NULL)
1257 		error = got_error_from_errno("close");
1258 	if (fd5 != -1 && close(fd5) == -1 && error == NULL)
1259 		error = got_error_from_errno("close");
1260 	if (f1) {
1261 		const struct got_error *f1_err =
1262 		    got_gotweb_flushfile(f1, fd1);
1263 		if (error == NULL)
1264 			error = f1_err;
1265 	}
1266 	if (f2) {
1267 		const struct got_error *f2_err =
1268 		    got_gotweb_flushfile(f2, fd2);
1269 		if (error == NULL)
1270 			error = f2_err;
1271 	}
1272 	if (error && f3) {
1273 		got_gotweb_flushfile(f3, fd3);
1274 		*fp = NULL;
1275 		*fd = -1;
1276 	}
1277 	got_ref_list_free(&refs);
1278 	free(id1);
1279 	free(id2);
1280 	return error;
1281 }
1282 
1283 static const struct got_error *
1284 got_init_repo_commit(struct repo_commit **rc)
1285 {
1286 	*rc = calloc(1, sizeof(**rc));
1287 	if (*rc == NULL)
1288 		return got_error_from_errno2("%s: calloc", __func__);
1289 
1290 	(*rc)->path = NULL;
1291 	(*rc)->refs_str = NULL;
1292 	(*rc)->commit_id = NULL;
1293 	(*rc)->committer = NULL;
1294 	(*rc)->author = NULL;
1295 	(*rc)->parent_id = NULL;
1296 	(*rc)->tree_id = NULL;
1297 	(*rc)->commit_msg = NULL;
1298 
1299 	return NULL;
1300 }
1301 
1302 static const struct got_error *
1303 got_init_repo_tag(struct repo_tag **rt)
1304 {
1305 	*rt = calloc(1, sizeof(**rt));
1306 	if (*rt == NULL)
1307 		return got_error_from_errno2("%s: calloc", __func__);
1308 
1309 	(*rt)->commit_id = NULL;
1310 	(*rt)->tag_name = NULL;
1311 	(*rt)->tag_commit = NULL;
1312 	(*rt)->commit_msg = NULL;
1313 	(*rt)->tagger = NULL;
1314 
1315 	return NULL;
1316 }