1 /*
2  * Copyright (c) 2023 Omar Polo <op@openbsd.org>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include <sys/queue.h>
18 #include <sys/types.h>
19 
20 #include <ctype.h>
21 #include <limits.h>
22 #include <sha1.h>
23 #include <sha2.h>
24 #include <stdint.h>
25 #include <stdio.h>
26 #include <stdlib.h>
27 #include <string.h>
28 #include <time.h>
29 
30 #include "got_error.h"
31 #include "got_cancel.h"
32 #include "got_reference.h"
33 #include "got_repository_admin.h" /* XXX for pack_progress */
34 #include "got_object.h"
35 #include "got_opentemp.h"
36 #include "got_repository_dump.h"
37 
38 #include "got_lib_delta.h"
39 #include "got_lib_object.h"
40 #include "got_lib_object_idset.h"
41 #include "got_lib_ratelimit.h"
42 #include "got_lib_pack_create.h"
43 
44 #define GIT_BUNDLE_SIGNATURE_V2 "# v2 git bundle"
45 
46 struct idvec {
47 	struct got_object_id	**ids;
48 	size_t			  len;
49 	size_t			  size;
50 };
51 
52 static const struct got_error *
53 idvec_push(struct idvec *v, struct got_object_id *id)
54 {
55 	size_t	 newsize;
56 	void	*t;
57 
58 	if (v->len == v->size) {
59 		newsize = v->size + 8;
60 		t = reallocarray(v->ids, newsize, sizeof(*v->ids));
61 		if (t == NULL)
62 			return got_error_from_errno("reallocarray");
63 		v->ids = t;
64 		v->size = newsize;
65 	}
66 
67 	v->ids[v->len++] = id;
68 	return NULL;
69 }
70 
71 static void
72 idvec_free(struct idvec *v)
73 {
74 	size_t i;
75 
76 	for (i = 0; i < v->len; ++i)
77 		free(v->ids[i]);
78 	free(v->ids);
79 }
80 
81 const struct got_error *
82 got_repo_dump(FILE *out, struct got_reflist_head *include_refs,
83     struct got_reflist_head *exclude_refs, struct got_repository *repo,
84     got_pack_progress_cb progress_cb, void *progress_arg,
85     got_cancel_cb cancel_cb, void *cancel_arg)
86 {
87 	const struct got_error *err = NULL;
88 	struct got_ratelimit rl;
89 	uint8_t packsha[SHA1_DIGEST_LENGTH];
90 	FILE *delta_cache = NULL;
91 	struct got_reflist_entry *e;
92 	struct got_object_id *id = NULL;
93 	struct got_commit_object *commit = NULL;
94 	struct idvec ours, theirs;
95 	char *nl, *s, *hex, *logmsg = NULL;
96 	const char *refname;
97 	int r;
98 
99 	got_ratelimit_init(&rl, 0, 500);
100 
101 	memset(&ours, 0, sizeof(ours));
102 	memset(&theirs, 0, sizeof(theirs));
103 
104 	r = fprintf(out, "%s\n", GIT_BUNDLE_SIGNATURE_V2);
105 	if (r != strlen(GIT_BUNDLE_SIGNATURE_V2) + 1)
106 		return got_ferror(out, GOT_ERR_IO);
107 
108 	TAILQ_FOREACH(e, exclude_refs, entry) {
109 		err = got_ref_resolve(&id, repo, e->ref);
110 		if (err)
111 			goto done;
112 
113 		idvec_push(&theirs, id);
114 		if (err)
115 			goto done;
116 
117 		err = got_object_open_as_commit(&commit, repo, id);
118 		if (err)
119 			goto done;
120 
121 		err = got_object_commit_get_logmsg(&logmsg, commit);
122 		if (err)
123 			goto done;
124 
125 		s = logmsg;
126 		while (isspace((unsigned char)*s))
127 			s++;
128 		nl = strchr(s, '\n');
129 		if (nl)
130 			*nl = '\0';
131 
132 		err = got_object_id_str(&hex, id);
133 		if (err)
134 			goto done;
135 		fprintf(out, "-%s %s\n", hex, s);
136 		free(hex);
137 
138 		got_object_commit_close(commit);
139 		commit = NULL;
140 
141 		free(logmsg);
142 		logmsg = NULL;
143 	}
144 
145 	TAILQ_FOREACH(e, include_refs, entry) {
146 		err = got_ref_resolve(&id, repo, e->ref);
147 		if (err)
148 			goto done;
149 
150 		err = idvec_push(&ours, id);
151 		if (err)
152 			goto done;
153 
154 		refname = got_ref_get_name(e->ref);
155 
156 		err = got_object_id_str(&hex, id);
157 		if (err)
158 			goto done;
159 		fprintf(out, "%s %s\n", hex, refname);
160 		free(hex);
161 	}
162 
163 	if (fputc('\n', out) == EOF || fflush(out) == EOF) {
164 		err = got_ferror(out, GOT_ERR_IO);
165 		goto done;
166 	}
167 
168 	delta_cache = got_opentemp();
169 	if (delta_cache == NULL) {
170 		err = got_error_from_errno("got_opentemp");
171 		goto done;
172 	}
173 
174 	err = got_pack_create(&packsha[0], fileno(out), delta_cache,
175 	    theirs.ids, theirs.len, ours.ids, ours.len,
176 	    repo, 0, 0, 0, progress_cb, progress_arg, &rl,
177 	    cancel_cb, cancel_arg);
178 
179  done:
180 	idvec_free(&ours);
181 	idvec_free(&theirs);
182 	if (commit)
183 		got_object_commit_close(commit);
184 	if (delta_cache && fclose(delta_cache) == EOF && err == NULL)
185 		err = got_error_from_errno("fclose");
186 	return err;
187 }