Blame


1 8e359fa0 2022-10-13 stsp /*
2 8e359fa0 2022-10-13 stsp * Copyright (c) 2018, 2019, 2022 Stefan Sperling <stsp@openbsd.org>
3 8e359fa0 2022-10-13 stsp *
4 8e359fa0 2022-10-13 stsp * Permission to use, copy, modify, and distribute this software for any
5 8e359fa0 2022-10-13 stsp * purpose with or without fee is hereby granted, provided that the above
6 8e359fa0 2022-10-13 stsp * copyright notice and this permission notice appear in all copies.
7 8e359fa0 2022-10-13 stsp *
8 8e359fa0 2022-10-13 stsp * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 8e359fa0 2022-10-13 stsp * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 8e359fa0 2022-10-13 stsp * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 8e359fa0 2022-10-13 stsp * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 8e359fa0 2022-10-13 stsp * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 8e359fa0 2022-10-13 stsp * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 8e359fa0 2022-10-13 stsp * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15 8e359fa0 2022-10-13 stsp */
16 8e359fa0 2022-10-13 stsp
17 8e359fa0 2022-10-13 stsp #include <sys/mman.h>
18 8e359fa0 2022-10-13 stsp #include <sys/queue.h>
19 8e359fa0 2022-10-13 stsp #include <sys/types.h>
20 8e359fa0 2022-10-13 stsp #include <sys/tree.h>
21 8e359fa0 2022-10-13 stsp #include <sys/stat.h>
22 8e359fa0 2022-10-13 stsp #include <sys/socket.h>
23 8e359fa0 2022-10-13 stsp #include <sys/uio.h>
24 8e359fa0 2022-10-13 stsp
25 8e359fa0 2022-10-13 stsp #include <errno.h>
26 8e359fa0 2022-10-13 stsp #include <imsg.h>
27 8e359fa0 2022-10-13 stsp #include <stdio.h>
28 8e359fa0 2022-10-13 stsp #include <stdint.h>
29 8e359fa0 2022-10-13 stsp #include <stdlib.h>
30 8e359fa0 2022-10-13 stsp #include <string.h>
31 8e359fa0 2022-10-13 stsp #include <sha1.h>
32 5822e79e 2023-02-23 op #include <sha2.h>
33 8e359fa0 2022-10-13 stsp #include <limits.h>
34 8e359fa0 2022-10-13 stsp #include <unistd.h>
35 8e359fa0 2022-10-13 stsp
36 8e359fa0 2022-10-13 stsp #include "got_error.h"
37 8e359fa0 2022-10-13 stsp #include "got_object.h"
38 8e359fa0 2022-10-13 stsp #include "got_repository.h"
39 8e359fa0 2022-10-13 stsp #include "got_opentemp.h"
40 8e359fa0 2022-10-13 stsp #include "got_path.h"
41 8e359fa0 2022-10-13 stsp
42 8e359fa0 2022-10-13 stsp #include "got_lib_delta.h"
43 8e359fa0 2022-10-13 stsp #include "got_lib_object.h"
44 8e359fa0 2022-10-13 stsp #include "got_lib_privsep.h"
45 8e359fa0 2022-10-13 stsp #include "got_lib_object_cache.h"
46 8e359fa0 2022-10-13 stsp #include "got_lib_pack.h"
47 8e359fa0 2022-10-13 stsp #include "got_lib_repository.h"
48 8e359fa0 2022-10-13 stsp
49 8e359fa0 2022-10-13 stsp static const struct got_error *
50 8e359fa0 2022-10-13 stsp request_packed_object(struct got_object **obj, struct got_pack *pack, int idx,
51 8e359fa0 2022-10-13 stsp struct got_object_id *id)
52 8e359fa0 2022-10-13 stsp {
53 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
54 8e359fa0 2022-10-13 stsp struct imsgbuf *ibuf = pack->privsep_child->ibuf;
55 8e359fa0 2022-10-13 stsp
56 8e359fa0 2022-10-13 stsp err = got_privsep_send_packed_obj_req(ibuf, idx, id);
57 8e359fa0 2022-10-13 stsp if (err)
58 8e359fa0 2022-10-13 stsp return err;
59 8e359fa0 2022-10-13 stsp
60 8e359fa0 2022-10-13 stsp err = got_privsep_recv_obj(obj, ibuf);
61 8e359fa0 2022-10-13 stsp if (err)
62 8e359fa0 2022-10-13 stsp return err;
63 8e359fa0 2022-10-13 stsp
64 8e359fa0 2022-10-13 stsp memcpy(&(*obj)->id, id, sizeof((*obj)->id));
65 8e359fa0 2022-10-13 stsp
66 8e359fa0 2022-10-13 stsp return NULL;
67 8e359fa0 2022-10-13 stsp }
68 8e359fa0 2022-10-13 stsp
69 8e359fa0 2022-10-13 stsp /* Create temporary files used during delta application. */
70 8e359fa0 2022-10-13 stsp static const struct got_error *
71 8e359fa0 2022-10-13 stsp pack_child_send_tempfiles(struct imsgbuf *ibuf, struct got_pack *pack)
72 8e359fa0 2022-10-13 stsp {
73 8e359fa0 2022-10-13 stsp const struct got_error *err;
74 8e359fa0 2022-10-13 stsp int basefd = -1, accumfd = -1;
75 8e359fa0 2022-10-13 stsp
76 8e359fa0 2022-10-13 stsp /*
77 8e359fa0 2022-10-13 stsp * For performance reasons, the child will keep reusing the
78 8e359fa0 2022-10-13 stsp * same temporary files during every object request.
79 8e359fa0 2022-10-13 stsp * Opening and closing new files for every object request is
80 8e359fa0 2022-10-13 stsp * too expensive during operations such as 'gotadmin pack'.
81 8e359fa0 2022-10-13 stsp */
82 8e359fa0 2022-10-13 stsp if (pack->child_has_tempfiles)
83 8e359fa0 2022-10-13 stsp return NULL;
84 8e359fa0 2022-10-13 stsp
85 8e359fa0 2022-10-13 stsp basefd = dup(pack->basefd);
86 8e359fa0 2022-10-13 stsp if (basefd == -1)
87 8e359fa0 2022-10-13 stsp return got_error_from_errno("dup");
88 8e359fa0 2022-10-13 stsp
89 8e359fa0 2022-10-13 stsp accumfd = dup(pack->accumfd);
90 8e359fa0 2022-10-13 stsp if (accumfd == -1) {
91 8e359fa0 2022-10-13 stsp err = got_error_from_errno("dup");
92 8e359fa0 2022-10-13 stsp goto done;
93 8e359fa0 2022-10-13 stsp }
94 8e359fa0 2022-10-13 stsp
95 8e359fa0 2022-10-13 stsp err = got_privsep_send_tmpfd(ibuf, basefd);
96 8e359fa0 2022-10-13 stsp if (err)
97 8e359fa0 2022-10-13 stsp goto done;
98 8e359fa0 2022-10-13 stsp
99 8e359fa0 2022-10-13 stsp err = got_privsep_send_tmpfd(ibuf, accumfd);
100 8e359fa0 2022-10-13 stsp done:
101 8e359fa0 2022-10-13 stsp if (err) {
102 8e359fa0 2022-10-13 stsp if (basefd != -1)
103 8e359fa0 2022-10-13 stsp close(basefd);
104 8e359fa0 2022-10-13 stsp if (accumfd != -1)
105 8e359fa0 2022-10-13 stsp close(accumfd);
106 8e359fa0 2022-10-13 stsp } else
107 8e359fa0 2022-10-13 stsp pack->child_has_tempfiles = 1;
108 c33c3763 2023-02-03 mark return err;
109 8e359fa0 2022-10-13 stsp }
110 8e359fa0 2022-10-13 stsp
111 8e359fa0 2022-10-13 stsp static const struct got_error *
112 8e359fa0 2022-10-13 stsp request_packed_object_raw(uint8_t **outbuf, off_t *size, size_t *hdrlen,
113 8e359fa0 2022-10-13 stsp int outfd, struct got_pack *pack, int idx, struct got_object_id *id)
114 8e359fa0 2022-10-13 stsp {
115 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
116 8e359fa0 2022-10-13 stsp struct imsgbuf *ibuf = pack->privsep_child->ibuf;
117 8e359fa0 2022-10-13 stsp int outfd_child;
118 8e359fa0 2022-10-13 stsp
119 8e359fa0 2022-10-13 stsp err = pack_child_send_tempfiles(ibuf, pack);
120 8e359fa0 2022-10-13 stsp if (err)
121 8e359fa0 2022-10-13 stsp return err;
122 8e359fa0 2022-10-13 stsp
123 8e359fa0 2022-10-13 stsp outfd_child = dup(outfd);
124 8e359fa0 2022-10-13 stsp if (outfd_child == -1)
125 8e359fa0 2022-10-13 stsp return got_error_from_errno("dup");
126 8e359fa0 2022-10-13 stsp
127 8e359fa0 2022-10-13 stsp err = got_privsep_send_packed_raw_obj_req(ibuf, idx, id);
128 8e359fa0 2022-10-13 stsp if (err) {
129 8e359fa0 2022-10-13 stsp close(outfd_child);
130 8e359fa0 2022-10-13 stsp return err;
131 8e359fa0 2022-10-13 stsp }
132 8e359fa0 2022-10-13 stsp
133 8e359fa0 2022-10-13 stsp err = got_privsep_send_raw_obj_outfd(ibuf, outfd_child);
134 8e359fa0 2022-10-13 stsp if (err)
135 8e359fa0 2022-10-13 stsp return err;
136 8e359fa0 2022-10-13 stsp
137 8e359fa0 2022-10-13 stsp err = got_privsep_recv_raw_obj(outbuf, size, hdrlen, ibuf);
138 8e359fa0 2022-10-13 stsp if (err)
139 8e359fa0 2022-10-13 stsp return err;
140 8e359fa0 2022-10-13 stsp
141 8e359fa0 2022-10-13 stsp return NULL;
142 8e359fa0 2022-10-13 stsp }
143 8e359fa0 2022-10-13 stsp
144 8e359fa0 2022-10-13 stsp static const struct got_error *
145 8e359fa0 2022-10-13 stsp read_packed_object_privsep(struct got_object **obj,
146 8e359fa0 2022-10-13 stsp struct got_repository *repo, struct got_pack *pack,
147 8e359fa0 2022-10-13 stsp struct got_packidx *packidx, int idx, struct got_object_id *id)
148 8e359fa0 2022-10-13 stsp {
149 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
150 8e359fa0 2022-10-13 stsp
151 8e359fa0 2022-10-13 stsp if (pack->privsep_child == NULL) {
152 8e359fa0 2022-10-13 stsp err = got_pack_start_privsep_child(pack, packidx);
153 8e359fa0 2022-10-13 stsp if (err)
154 8e359fa0 2022-10-13 stsp return err;
155 8e359fa0 2022-10-13 stsp }
156 8e359fa0 2022-10-13 stsp
157 8e359fa0 2022-10-13 stsp return request_packed_object(obj, pack, idx, id);
158 8e359fa0 2022-10-13 stsp }
159 8e359fa0 2022-10-13 stsp
160 8e359fa0 2022-10-13 stsp static const struct got_error *
161 8e359fa0 2022-10-13 stsp read_packed_object_raw_privsep(uint8_t **outbuf, off_t *size, size_t *hdrlen,
162 8e359fa0 2022-10-13 stsp int outfd, struct got_pack *pack, struct got_packidx *packidx, int idx,
163 8e359fa0 2022-10-13 stsp struct got_object_id *id)
164 8e359fa0 2022-10-13 stsp {
165 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
166 8e359fa0 2022-10-13 stsp
167 8e359fa0 2022-10-13 stsp if (pack->privsep_child == NULL) {
168 8e359fa0 2022-10-13 stsp err = got_pack_start_privsep_child(pack, packidx);
169 8e359fa0 2022-10-13 stsp if (err)
170 8e359fa0 2022-10-13 stsp return err;
171 8e359fa0 2022-10-13 stsp }
172 8e359fa0 2022-10-13 stsp
173 8e359fa0 2022-10-13 stsp return request_packed_object_raw(outbuf, size, hdrlen, outfd, pack,
174 8e359fa0 2022-10-13 stsp idx, id);
175 8e359fa0 2022-10-13 stsp }
176 8e359fa0 2022-10-13 stsp
177 8e359fa0 2022-10-13 stsp const struct got_error *
178 8e359fa0 2022-10-13 stsp got_object_open_packed(struct got_object **obj, struct got_object_id *id,
179 8e359fa0 2022-10-13 stsp struct got_repository *repo)
180 8e359fa0 2022-10-13 stsp {
181 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
182 8e359fa0 2022-10-13 stsp struct got_pack *pack = NULL;
183 8e359fa0 2022-10-13 stsp struct got_packidx *packidx = NULL;
184 8e359fa0 2022-10-13 stsp int idx;
185 8e359fa0 2022-10-13 stsp char *path_packfile;
186 8e359fa0 2022-10-13 stsp
187 8e359fa0 2022-10-13 stsp err = got_repo_search_packidx(&packidx, &idx, repo, id);
188 8e359fa0 2022-10-13 stsp if (err)
189 8e359fa0 2022-10-13 stsp return err;
190 8e359fa0 2022-10-13 stsp
191 8e359fa0 2022-10-13 stsp err = got_packidx_get_packfile_path(&path_packfile,
192 8e359fa0 2022-10-13 stsp packidx->path_packidx);
193 8e359fa0 2022-10-13 stsp if (err)
194 8e359fa0 2022-10-13 stsp return err;
195 8e359fa0 2022-10-13 stsp
196 8e359fa0 2022-10-13 stsp pack = got_repo_get_cached_pack(repo, path_packfile);
197 8e359fa0 2022-10-13 stsp if (pack == NULL) {
198 8e359fa0 2022-10-13 stsp err = got_repo_cache_pack(&pack, repo, path_packfile, packidx);
199 8e359fa0 2022-10-13 stsp if (err)
200 8e359fa0 2022-10-13 stsp goto done;
201 8e359fa0 2022-10-13 stsp }
202 8e359fa0 2022-10-13 stsp
203 8e359fa0 2022-10-13 stsp err = read_packed_object_privsep(obj, repo, pack, packidx, idx, id);
204 8e359fa0 2022-10-13 stsp if (err)
205 8e359fa0 2022-10-13 stsp goto done;
206 8e359fa0 2022-10-13 stsp done:
207 8e359fa0 2022-10-13 stsp free(path_packfile);
208 8e359fa0 2022-10-13 stsp return err;
209 8e359fa0 2022-10-13 stsp }
210 8e359fa0 2022-10-13 stsp
211 8e359fa0 2022-10-13 stsp const struct got_error *
212 8e359fa0 2022-10-13 stsp got_object_open_from_packfile(struct got_object **obj, struct got_object_id *id,
213 8e359fa0 2022-10-13 stsp struct got_pack *pack, struct got_packidx *packidx, int obj_idx,
214 8e359fa0 2022-10-13 stsp struct got_repository *repo)
215 8e359fa0 2022-10-13 stsp {
216 8e359fa0 2022-10-13 stsp return read_packed_object_privsep(obj, repo, pack, packidx,
217 8e359fa0 2022-10-13 stsp obj_idx, id);
218 8e359fa0 2022-10-13 stsp }
219 8e359fa0 2022-10-13 stsp
220 8e359fa0 2022-10-13 stsp const struct got_error *
221 8e359fa0 2022-10-13 stsp got_object_read_raw_delta(uint64_t *base_size, uint64_t *result_size,
222 8e359fa0 2022-10-13 stsp off_t *delta_size, off_t *delta_compressed_size, off_t *delta_offset,
223 8e359fa0 2022-10-13 stsp off_t *delta_out_offset, struct got_object_id **base_id, int delta_cache_fd,
224 8e359fa0 2022-10-13 stsp struct got_packidx *packidx, int obj_idx, struct got_object_id *id,
225 8e359fa0 2022-10-13 stsp struct got_repository *repo)
226 8e359fa0 2022-10-13 stsp {
227 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
228 8e359fa0 2022-10-13 stsp struct got_pack *pack = NULL;
229 8e359fa0 2022-10-13 stsp char *path_packfile;
230 8e359fa0 2022-10-13 stsp
231 8e359fa0 2022-10-13 stsp *base_size = 0;
232 8e359fa0 2022-10-13 stsp *result_size = 0;
233 8e359fa0 2022-10-13 stsp *delta_size = 0;
234 8e359fa0 2022-10-13 stsp *delta_compressed_size = 0;
235 8e359fa0 2022-10-13 stsp *delta_offset = 0;
236 8e359fa0 2022-10-13 stsp *delta_out_offset = 0;
237 8e359fa0 2022-10-13 stsp
238 8e359fa0 2022-10-13 stsp err = got_packidx_get_packfile_path(&path_packfile,
239 8e359fa0 2022-10-13 stsp packidx->path_packidx);
240 8e359fa0 2022-10-13 stsp if (err)
241 8e359fa0 2022-10-13 stsp return err;
242 8e359fa0 2022-10-13 stsp
243 8e359fa0 2022-10-13 stsp pack = got_repo_get_cached_pack(repo, path_packfile);
244 8e359fa0 2022-10-13 stsp if (pack == NULL) {
245 8e359fa0 2022-10-13 stsp err = got_repo_cache_pack(&pack, repo, path_packfile, packidx);
246 8e359fa0 2022-10-13 stsp if (err)
247 8e359fa0 2022-10-13 stsp return err;
248 8e359fa0 2022-10-13 stsp }
249 8e359fa0 2022-10-13 stsp
250 8e359fa0 2022-10-13 stsp if (pack->privsep_child == NULL) {
251 8e359fa0 2022-10-13 stsp err = got_pack_start_privsep_child(pack, packidx);
252 8e359fa0 2022-10-13 stsp if (err)
253 8e359fa0 2022-10-13 stsp return err;
254 8e359fa0 2022-10-13 stsp }
255 8e359fa0 2022-10-13 stsp
256 8e359fa0 2022-10-13 stsp if (!pack->child_has_delta_outfd) {
257 8e359fa0 2022-10-13 stsp int outfd_child;
258 8e359fa0 2022-10-13 stsp outfd_child = dup(delta_cache_fd);
259 8e359fa0 2022-10-13 stsp if (outfd_child == -1)
260 8e359fa0 2022-10-13 stsp return got_error_from_errno("dup");
261 8e359fa0 2022-10-13 stsp err = got_privsep_send_raw_delta_outfd(
262 8e359fa0 2022-10-13 stsp pack->privsep_child->ibuf, outfd_child);
263 8e359fa0 2022-10-13 stsp if (err)
264 8e359fa0 2022-10-13 stsp return err;
265 8e359fa0 2022-10-13 stsp pack->child_has_delta_outfd = 1;
266 8e359fa0 2022-10-13 stsp }
267 8e359fa0 2022-10-13 stsp
268 8e359fa0 2022-10-13 stsp err = got_privsep_send_raw_delta_req(pack->privsep_child->ibuf,
269 8e359fa0 2022-10-13 stsp obj_idx, id);
270 8e359fa0 2022-10-13 stsp if (err)
271 8e359fa0 2022-10-13 stsp return err;
272 8e359fa0 2022-10-13 stsp
273 8e359fa0 2022-10-13 stsp return got_privsep_recv_raw_delta(base_size, result_size, delta_size,
274 8e359fa0 2022-10-13 stsp delta_compressed_size, delta_offset, delta_out_offset, base_id,
275 8e359fa0 2022-10-13 stsp pack->privsep_child->ibuf);
276 8e359fa0 2022-10-13 stsp }
277 8e359fa0 2022-10-13 stsp
278 8e359fa0 2022-10-13 stsp static const struct got_error *
279 8e359fa0 2022-10-13 stsp request_object(struct got_object **obj, struct got_object_id *id,
280 8e359fa0 2022-10-13 stsp struct got_repository *repo, int fd)
281 8e359fa0 2022-10-13 stsp {
282 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
283 8e359fa0 2022-10-13 stsp struct imsgbuf *ibuf;
284 8e359fa0 2022-10-13 stsp
285 8e359fa0 2022-10-13 stsp ibuf = repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_OBJECT].ibuf;
286 8e359fa0 2022-10-13 stsp
287 8e359fa0 2022-10-13 stsp err = got_privsep_send_obj_req(ibuf, fd, id);
288 8e359fa0 2022-10-13 stsp if (err)
289 8e359fa0 2022-10-13 stsp return err;
290 8e359fa0 2022-10-13 stsp
291 8e359fa0 2022-10-13 stsp return got_privsep_recv_obj(obj, ibuf);
292 8e359fa0 2022-10-13 stsp }
293 8e359fa0 2022-10-13 stsp
294 8e359fa0 2022-10-13 stsp static const struct got_error *
295 8e359fa0 2022-10-13 stsp request_raw_object(uint8_t **outbuf, off_t *size, size_t *hdrlen, int outfd,
296 8e359fa0 2022-10-13 stsp struct got_object_id *id, struct got_repository *repo, int infd)
297 8e359fa0 2022-10-13 stsp {
298 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
299 8e359fa0 2022-10-13 stsp struct imsgbuf *ibuf;
300 8e359fa0 2022-10-13 stsp int outfd_child;
301 8e359fa0 2022-10-13 stsp
302 8e359fa0 2022-10-13 stsp ibuf = repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_OBJECT].ibuf;
303 8e359fa0 2022-10-13 stsp
304 8e359fa0 2022-10-13 stsp outfd_child = dup(outfd);
305 8e359fa0 2022-10-13 stsp if (outfd_child == -1)
306 8e359fa0 2022-10-13 stsp return got_error_from_errno("dup");
307 8e359fa0 2022-10-13 stsp
308 8e359fa0 2022-10-13 stsp err = got_privsep_send_raw_obj_req(ibuf, infd, id);
309 8e359fa0 2022-10-13 stsp if (err)
310 8e359fa0 2022-10-13 stsp return err;
311 8e359fa0 2022-10-13 stsp
312 8e359fa0 2022-10-13 stsp err = got_privsep_send_raw_obj_outfd(ibuf, outfd_child);
313 8e359fa0 2022-10-13 stsp if (err)
314 8e359fa0 2022-10-13 stsp return err;
315 8e359fa0 2022-10-13 stsp
316 8e359fa0 2022-10-13 stsp return got_privsep_recv_raw_obj(outbuf, size, hdrlen, ibuf);
317 8e359fa0 2022-10-13 stsp }
318 8e359fa0 2022-10-13 stsp
319 8e359fa0 2022-10-13 stsp static const struct got_error *
320 1d9e43b0 2022-10-13 stsp start_child(struct got_repository *repo, int type)
321 8e359fa0 2022-10-13 stsp {
322 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
323 8e359fa0 2022-10-13 stsp int imsg_fds[2];
324 8e359fa0 2022-10-13 stsp pid_t pid;
325 8e359fa0 2022-10-13 stsp struct imsgbuf *ibuf;
326 1d9e43b0 2022-10-13 stsp const char *prog_path;
327 1d9e43b0 2022-10-13 stsp
328 1d9e43b0 2022-10-13 stsp switch (type) {
329 1d9e43b0 2022-10-13 stsp case GOT_REPO_PRIVSEP_CHILD_OBJECT:
330 1d9e43b0 2022-10-13 stsp prog_path = GOT_PATH_PROG_READ_OBJECT;
331 1d9e43b0 2022-10-13 stsp break;
332 1d9e43b0 2022-10-13 stsp case GOT_REPO_PRIVSEP_CHILD_TREE:
333 1d9e43b0 2022-10-13 stsp prog_path = GOT_PATH_PROG_READ_TREE;
334 1d9e43b0 2022-10-13 stsp break;
335 1d9e43b0 2022-10-13 stsp case GOT_REPO_PRIVSEP_CHILD_COMMIT:
336 1d9e43b0 2022-10-13 stsp prog_path = GOT_PATH_PROG_READ_COMMIT;
337 1d9e43b0 2022-10-13 stsp break;
338 1d9e43b0 2022-10-13 stsp case GOT_REPO_PRIVSEP_CHILD_BLOB:
339 1d9e43b0 2022-10-13 stsp prog_path = GOT_PATH_PROG_READ_BLOB;
340 1d9e43b0 2022-10-13 stsp break;
341 1d9e43b0 2022-10-13 stsp case GOT_REPO_PRIVSEP_CHILD_TAG:
342 1d9e43b0 2022-10-13 stsp prog_path = GOT_PATH_PROG_READ_TAG;
343 1d9e43b0 2022-10-13 stsp break;
344 1d9e43b0 2022-10-13 stsp default:
345 1d9e43b0 2022-10-13 stsp return got_error(GOT_ERR_OBJ_TYPE);
346 1d9e43b0 2022-10-13 stsp }
347 8e359fa0 2022-10-13 stsp
348 8e359fa0 2022-10-13 stsp ibuf = calloc(1, sizeof(*ibuf));
349 8e359fa0 2022-10-13 stsp if (ibuf == NULL)
350 8e359fa0 2022-10-13 stsp return got_error_from_errno("calloc");
351 8e359fa0 2022-10-13 stsp
352 8e359fa0 2022-10-13 stsp if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, imsg_fds) == -1) {
353 8e359fa0 2022-10-13 stsp err = got_error_from_errno("socketpair");
354 8e359fa0 2022-10-13 stsp free(ibuf);
355 8e359fa0 2022-10-13 stsp return err;
356 8e359fa0 2022-10-13 stsp }
357 8e359fa0 2022-10-13 stsp
358 8e359fa0 2022-10-13 stsp pid = fork();
359 8e359fa0 2022-10-13 stsp if (pid == -1) {
360 8e359fa0 2022-10-13 stsp err = got_error_from_errno("fork");
361 8e359fa0 2022-10-13 stsp free(ibuf);
362 8e359fa0 2022-10-13 stsp return err;
363 8e359fa0 2022-10-13 stsp }
364 8e359fa0 2022-10-13 stsp else if (pid == 0) {
365 1d9e43b0 2022-10-13 stsp got_privsep_exec_child(imsg_fds, prog_path, repo->path);
366 8e359fa0 2022-10-13 stsp /* not reached */
367 8e359fa0 2022-10-13 stsp }
368 8e359fa0 2022-10-13 stsp
369 8e359fa0 2022-10-13 stsp if (close(imsg_fds[1]) == -1) {
370 8e359fa0 2022-10-13 stsp err = got_error_from_errno("close");
371 8e359fa0 2022-10-13 stsp free(ibuf);
372 8e359fa0 2022-10-13 stsp return err;
373 8e359fa0 2022-10-13 stsp }
374 8e359fa0 2022-10-13 stsp
375 1d9e43b0 2022-10-13 stsp repo->privsep_children[type].imsg_fd = imsg_fds[0];
376 1d9e43b0 2022-10-13 stsp repo->privsep_children[type].pid = pid;
377 8e359fa0 2022-10-13 stsp imsg_init(ibuf, imsg_fds[0]);
378 1d9e43b0 2022-10-13 stsp repo->privsep_children[type].ibuf = ibuf;
379 8e359fa0 2022-10-13 stsp
380 8e359fa0 2022-10-13 stsp return NULL;
381 8e359fa0 2022-10-13 stsp }
382 8e359fa0 2022-10-13 stsp
383 8e359fa0 2022-10-13 stsp const struct got_error *
384 8e359fa0 2022-10-13 stsp got_object_read_header_privsep(struct got_object **obj,
385 8e359fa0 2022-10-13 stsp struct got_object_id *id, struct got_repository *repo, int obj_fd)
386 8e359fa0 2022-10-13 stsp {
387 8e359fa0 2022-10-13 stsp const struct got_error *err;
388 8e359fa0 2022-10-13 stsp
389 8e359fa0 2022-10-13 stsp if (repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_OBJECT].imsg_fd != -1)
390 8e359fa0 2022-10-13 stsp return request_object(obj, id, repo, obj_fd);
391 8e359fa0 2022-10-13 stsp
392 1d9e43b0 2022-10-13 stsp err = start_child(repo, GOT_REPO_PRIVSEP_CHILD_OBJECT);
393 1d9e43b0 2022-10-13 stsp if (err)
394 8e359fa0 2022-10-13 stsp return err;
395 8e359fa0 2022-10-13 stsp
396 8e359fa0 2022-10-13 stsp return request_object(obj, id, repo, obj_fd);
397 8e359fa0 2022-10-13 stsp }
398 8e359fa0 2022-10-13 stsp
399 8e359fa0 2022-10-13 stsp static const struct got_error *
400 8e359fa0 2022-10-13 stsp read_object_raw_privsep(uint8_t **outbuf, off_t *size, size_t *hdrlen,
401 8e359fa0 2022-10-13 stsp int outfd, struct got_object_id *id, struct got_repository *repo,
402 8e359fa0 2022-10-13 stsp int obj_fd)
403 8e359fa0 2022-10-13 stsp {
404 8e359fa0 2022-10-13 stsp const struct got_error *err;
405 8e359fa0 2022-10-13 stsp
406 8e359fa0 2022-10-13 stsp if (repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_OBJECT].imsg_fd != -1)
407 8e359fa0 2022-10-13 stsp return request_raw_object(outbuf, size, hdrlen, outfd, id,
408 8e359fa0 2022-10-13 stsp repo, obj_fd);
409 8e359fa0 2022-10-13 stsp
410 1d9e43b0 2022-10-13 stsp err = start_child(repo, GOT_REPO_PRIVSEP_CHILD_OBJECT);
411 8e359fa0 2022-10-13 stsp if (err)
412 8e359fa0 2022-10-13 stsp return err;
413 8e359fa0 2022-10-13 stsp
414 8e359fa0 2022-10-13 stsp return request_raw_object(outbuf, size, hdrlen, outfd, id, repo,
415 8e359fa0 2022-10-13 stsp obj_fd);
416 8e359fa0 2022-10-13 stsp }
417 8e359fa0 2022-10-13 stsp
418 8e359fa0 2022-10-13 stsp const struct got_error *
419 8e359fa0 2022-10-13 stsp got_object_open(struct got_object **obj, struct got_repository *repo,
420 8e359fa0 2022-10-13 stsp struct got_object_id *id)
421 8e359fa0 2022-10-13 stsp {
422 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
423 8e359fa0 2022-10-13 stsp int fd;
424 8e359fa0 2022-10-13 stsp
425 8e359fa0 2022-10-13 stsp *obj = got_repo_get_cached_object(repo, id);
426 8e359fa0 2022-10-13 stsp if (*obj != NULL) {
427 8e359fa0 2022-10-13 stsp (*obj)->refcnt++;
428 8e359fa0 2022-10-13 stsp return NULL;
429 8e359fa0 2022-10-13 stsp }
430 8e359fa0 2022-10-13 stsp
431 8e359fa0 2022-10-13 stsp err = got_object_open_packed(obj, id, repo);
432 8e359fa0 2022-10-13 stsp if (err && err->code != GOT_ERR_NO_OBJ)
433 8e359fa0 2022-10-13 stsp return err;
434 8e359fa0 2022-10-13 stsp if (*obj) {
435 8e359fa0 2022-10-13 stsp (*obj)->refcnt++;
436 8e359fa0 2022-10-13 stsp return got_repo_cache_object(repo, id, *obj);
437 8e359fa0 2022-10-13 stsp }
438 8e359fa0 2022-10-13 stsp
439 8e359fa0 2022-10-13 stsp err = got_object_open_loose_fd(&fd, id, repo);
440 8e359fa0 2022-10-13 stsp if (err) {
441 8e359fa0 2022-10-13 stsp if (err->code == GOT_ERR_ERRNO && errno == ENOENT)
442 8e359fa0 2022-10-13 stsp err = got_error_no_obj(id);
443 8e359fa0 2022-10-13 stsp return err;
444 8e359fa0 2022-10-13 stsp }
445 8e359fa0 2022-10-13 stsp
446 8e359fa0 2022-10-13 stsp err = got_object_read_header_privsep(obj, id, repo, fd);
447 8e359fa0 2022-10-13 stsp if (err)
448 8e359fa0 2022-10-13 stsp return err;
449 8e359fa0 2022-10-13 stsp
450 3e0381db 2023-02-05 op memcpy(&(*obj)->id, id, sizeof(*id));
451 8e359fa0 2022-10-13 stsp
452 8e359fa0 2022-10-13 stsp (*obj)->refcnt++;
453 8e359fa0 2022-10-13 stsp return got_repo_cache_object(repo, id, *obj);
454 8e359fa0 2022-10-13 stsp }
455 8e359fa0 2022-10-13 stsp
456 8e359fa0 2022-10-13 stsp /* *outfd must be initialized to -1 by caller */
457 8e359fa0 2022-10-13 stsp const struct got_error *
458 8e359fa0 2022-10-13 stsp got_object_raw_open(struct got_raw_object **obj, int *outfd,
459 8e359fa0 2022-10-13 stsp struct got_repository *repo, struct got_object_id *id)
460 8e359fa0 2022-10-13 stsp {
461 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
462 8e359fa0 2022-10-13 stsp struct got_packidx *packidx = NULL;
463 8e359fa0 2022-10-13 stsp int idx;
464 8e359fa0 2022-10-13 stsp uint8_t *outbuf = NULL;
465 8e359fa0 2022-10-13 stsp off_t size = 0;
466 8e359fa0 2022-10-13 stsp size_t hdrlen = 0;
467 8e359fa0 2022-10-13 stsp char *path_packfile = NULL;
468 8e359fa0 2022-10-13 stsp
469 8e359fa0 2022-10-13 stsp *obj = got_repo_get_cached_raw_object(repo, id);
470 8e359fa0 2022-10-13 stsp if (*obj != NULL) {
471 8e359fa0 2022-10-13 stsp (*obj)->refcnt++;
472 8e359fa0 2022-10-13 stsp return NULL;
473 8e359fa0 2022-10-13 stsp }
474 8e359fa0 2022-10-13 stsp
475 8e359fa0 2022-10-13 stsp if (*outfd == -1) {
476 8e359fa0 2022-10-13 stsp *outfd = got_opentempfd();
477 8e359fa0 2022-10-13 stsp if (*outfd == -1)
478 8e359fa0 2022-10-13 stsp return got_error_from_errno("got_opentempfd");
479 8e359fa0 2022-10-13 stsp }
480 8e359fa0 2022-10-13 stsp
481 8e359fa0 2022-10-13 stsp err = got_repo_search_packidx(&packidx, &idx, repo, id);
482 8e359fa0 2022-10-13 stsp if (err == NULL) {
483 8e359fa0 2022-10-13 stsp struct got_pack *pack = NULL;
484 8e359fa0 2022-10-13 stsp
485 8e359fa0 2022-10-13 stsp err = got_packidx_get_packfile_path(&path_packfile,
486 8e359fa0 2022-10-13 stsp packidx->path_packidx);
487 8e359fa0 2022-10-13 stsp if (err)
488 8e359fa0 2022-10-13 stsp goto done;
489 8e359fa0 2022-10-13 stsp
490 8e359fa0 2022-10-13 stsp pack = got_repo_get_cached_pack(repo, path_packfile);
491 8e359fa0 2022-10-13 stsp if (pack == NULL) {
492 8e359fa0 2022-10-13 stsp err = got_repo_cache_pack(&pack, repo, path_packfile,
493 8e359fa0 2022-10-13 stsp packidx);
494 8e359fa0 2022-10-13 stsp if (err)
495 8e359fa0 2022-10-13 stsp goto done;
496 8e359fa0 2022-10-13 stsp }
497 8e359fa0 2022-10-13 stsp err = read_packed_object_raw_privsep(&outbuf, &size, &hdrlen,
498 8e359fa0 2022-10-13 stsp *outfd, pack, packidx, idx, id);
499 8e359fa0 2022-10-13 stsp if (err)
500 8e359fa0 2022-10-13 stsp goto done;
501 8e359fa0 2022-10-13 stsp } else if (err->code == GOT_ERR_NO_OBJ) {
502 8e359fa0 2022-10-13 stsp int fd;
503 8e359fa0 2022-10-13 stsp
504 8e359fa0 2022-10-13 stsp err = got_object_open_loose_fd(&fd, id, repo);
505 8e359fa0 2022-10-13 stsp if (err)
506 8e359fa0 2022-10-13 stsp goto done;
507 8e359fa0 2022-10-13 stsp err = read_object_raw_privsep(&outbuf, &size, &hdrlen, *outfd,
508 8e359fa0 2022-10-13 stsp id, repo, fd);
509 8e359fa0 2022-10-13 stsp if (err)
510 8e359fa0 2022-10-13 stsp goto done;
511 8e359fa0 2022-10-13 stsp }
512 8e359fa0 2022-10-13 stsp
513 60c140ae 2023-01-09 stsp err = got_object_raw_alloc(obj, outbuf, outfd,
514 60c140ae 2023-01-09 stsp GOT_DELTA_RESULT_SIZE_CACHED_MAX, hdrlen, size);
515 13b2bc37 2022-10-23 stsp if (err)
516 8e359fa0 2022-10-13 stsp goto done;
517 8e359fa0 2022-10-13 stsp
518 8e359fa0 2022-10-13 stsp err = got_repo_cache_raw_object(repo, id, *obj);
519 8e359fa0 2022-10-13 stsp done:
520 8e359fa0 2022-10-13 stsp free(path_packfile);
521 8e359fa0 2022-10-13 stsp if (err) {
522 8e359fa0 2022-10-13 stsp if (*obj) {
523 8e359fa0 2022-10-13 stsp got_object_raw_close(*obj);
524 8e359fa0 2022-10-13 stsp *obj = NULL;
525 8e359fa0 2022-10-13 stsp }
526 8e359fa0 2022-10-13 stsp free(outbuf);
527 e3ad2c64 2022-11-03 stsp }
528 8e359fa0 2022-10-13 stsp return err;
529 8e359fa0 2022-10-13 stsp }
530 8e359fa0 2022-10-13 stsp
531 8e359fa0 2022-10-13 stsp static const struct got_error *
532 8e359fa0 2022-10-13 stsp request_packed_commit(struct got_commit_object **commit, struct got_pack *pack,
533 8e359fa0 2022-10-13 stsp int pack_idx, struct got_object_id *id)
534 8e359fa0 2022-10-13 stsp {
535 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
536 8e359fa0 2022-10-13 stsp
537 8e359fa0 2022-10-13 stsp err = got_privsep_send_commit_req(pack->privsep_child->ibuf, -1, id,
538 8e359fa0 2022-10-13 stsp pack_idx);
539 8e359fa0 2022-10-13 stsp if (err)
540 8e359fa0 2022-10-13 stsp return err;
541 8e359fa0 2022-10-13 stsp
542 8e359fa0 2022-10-13 stsp err = got_privsep_recv_commit(commit, pack->privsep_child->ibuf);
543 8e359fa0 2022-10-13 stsp if (err)
544 8e359fa0 2022-10-13 stsp return err;
545 8e359fa0 2022-10-13 stsp
546 8e359fa0 2022-10-13 stsp (*commit)->flags |= GOT_COMMIT_FLAG_PACKED;
547 8e359fa0 2022-10-13 stsp return NULL;
548 8e359fa0 2022-10-13 stsp }
549 8e359fa0 2022-10-13 stsp
550 8e359fa0 2022-10-13 stsp static const struct got_error *
551 8e359fa0 2022-10-13 stsp read_packed_commit_privsep(struct got_commit_object **commit,
552 8e359fa0 2022-10-13 stsp struct got_pack *pack, struct got_packidx *packidx, int idx,
553 8e359fa0 2022-10-13 stsp struct got_object_id *id)
554 8e359fa0 2022-10-13 stsp {
555 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
556 8e359fa0 2022-10-13 stsp
557 8e359fa0 2022-10-13 stsp if (pack->privsep_child)
558 8e359fa0 2022-10-13 stsp return request_packed_commit(commit, pack, idx, id);
559 8e359fa0 2022-10-13 stsp
560 8e359fa0 2022-10-13 stsp err = got_pack_start_privsep_child(pack, packidx);
561 8e359fa0 2022-10-13 stsp if (err)
562 8e359fa0 2022-10-13 stsp return err;
563 8e359fa0 2022-10-13 stsp
564 8e359fa0 2022-10-13 stsp return request_packed_commit(commit, pack, idx, id);
565 8e359fa0 2022-10-13 stsp }
566 8e359fa0 2022-10-13 stsp
567 8e359fa0 2022-10-13 stsp static const struct got_error *
568 8e359fa0 2022-10-13 stsp request_commit(struct got_commit_object **commit, struct got_repository *repo,
569 8e359fa0 2022-10-13 stsp int fd, struct got_object_id *id)
570 8e359fa0 2022-10-13 stsp {
571 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
572 8e359fa0 2022-10-13 stsp struct imsgbuf *ibuf;
573 8e359fa0 2022-10-13 stsp
574 8e359fa0 2022-10-13 stsp ibuf = repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_COMMIT].ibuf;
575 8e359fa0 2022-10-13 stsp
576 8e359fa0 2022-10-13 stsp err = got_privsep_send_commit_req(ibuf, fd, id, -1);
577 8e359fa0 2022-10-13 stsp if (err)
578 8e359fa0 2022-10-13 stsp return err;
579 8e359fa0 2022-10-13 stsp
580 8e359fa0 2022-10-13 stsp return got_privsep_recv_commit(commit, ibuf);
581 8e359fa0 2022-10-13 stsp }
582 8e359fa0 2022-10-13 stsp
583 8e359fa0 2022-10-13 stsp static const struct got_error *
584 8e359fa0 2022-10-13 stsp read_commit_privsep(struct got_commit_object **commit, int obj_fd,
585 8e359fa0 2022-10-13 stsp struct got_object_id *id, struct got_repository *repo)
586 8e359fa0 2022-10-13 stsp {
587 8e359fa0 2022-10-13 stsp const struct got_error *err;
588 8e359fa0 2022-10-13 stsp
589 8e359fa0 2022-10-13 stsp if (repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_COMMIT].imsg_fd != -1)
590 8e359fa0 2022-10-13 stsp return request_commit(commit, repo, obj_fd, id);
591 8e359fa0 2022-10-13 stsp
592 1d9e43b0 2022-10-13 stsp err = start_child(repo, GOT_REPO_PRIVSEP_CHILD_COMMIT);
593 1d9e43b0 2022-10-13 stsp if (err)
594 8e359fa0 2022-10-13 stsp return err;
595 8e359fa0 2022-10-13 stsp
596 8e359fa0 2022-10-13 stsp return request_commit(commit, repo, obj_fd, id);
597 8e359fa0 2022-10-13 stsp }
598 8e359fa0 2022-10-13 stsp
599 8e359fa0 2022-10-13 stsp static const struct got_error *
600 8e359fa0 2022-10-13 stsp open_commit(struct got_commit_object **commit,
601 8e359fa0 2022-10-13 stsp struct got_repository *repo, struct got_object_id *id, int check_cache)
602 8e359fa0 2022-10-13 stsp {
603 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
604 8e359fa0 2022-10-13 stsp struct got_packidx *packidx = NULL;
605 8e359fa0 2022-10-13 stsp int idx;
606 8e359fa0 2022-10-13 stsp char *path_packfile = NULL;
607 8e359fa0 2022-10-13 stsp
608 8e359fa0 2022-10-13 stsp if (check_cache) {
609 8e359fa0 2022-10-13 stsp *commit = got_repo_get_cached_commit(repo, id);
610 8e359fa0 2022-10-13 stsp if (*commit != NULL) {
611 8e359fa0 2022-10-13 stsp (*commit)->refcnt++;
612 8e359fa0 2022-10-13 stsp return NULL;
613 8e359fa0 2022-10-13 stsp }
614 8e359fa0 2022-10-13 stsp } else
615 8e359fa0 2022-10-13 stsp *commit = NULL;
616 8e359fa0 2022-10-13 stsp
617 8e359fa0 2022-10-13 stsp err = got_repo_search_packidx(&packidx, &idx, repo, id);
618 8e359fa0 2022-10-13 stsp if (err == NULL) {
619 8e359fa0 2022-10-13 stsp struct got_pack *pack = NULL;
620 8e359fa0 2022-10-13 stsp
621 8e359fa0 2022-10-13 stsp err = got_packidx_get_packfile_path(&path_packfile,
622 8e359fa0 2022-10-13 stsp packidx->path_packidx);
623 8e359fa0 2022-10-13 stsp if (err)
624 8e359fa0 2022-10-13 stsp return err;
625 8e359fa0 2022-10-13 stsp
626 8e359fa0 2022-10-13 stsp pack = got_repo_get_cached_pack(repo, path_packfile);
627 8e359fa0 2022-10-13 stsp if (pack == NULL) {
628 8e359fa0 2022-10-13 stsp err = got_repo_cache_pack(&pack, repo, path_packfile,
629 8e359fa0 2022-10-13 stsp packidx);
630 8e359fa0 2022-10-13 stsp if (err)
631 8e359fa0 2022-10-13 stsp goto done;
632 8e359fa0 2022-10-13 stsp }
633 8e359fa0 2022-10-13 stsp err = read_packed_commit_privsep(commit, pack,
634 8e359fa0 2022-10-13 stsp packidx, idx, id);
635 8e359fa0 2022-10-13 stsp } else if (err->code == GOT_ERR_NO_OBJ) {
636 8e359fa0 2022-10-13 stsp int fd;
637 8e359fa0 2022-10-13 stsp
638 8e359fa0 2022-10-13 stsp err = got_object_open_loose_fd(&fd, id, repo);
639 8e359fa0 2022-10-13 stsp if (err)
640 8e359fa0 2022-10-13 stsp return err;
641 8e359fa0 2022-10-13 stsp err = read_commit_privsep(commit, fd, id, repo);
642 8e359fa0 2022-10-13 stsp }
643 8e359fa0 2022-10-13 stsp
644 8e359fa0 2022-10-13 stsp if (err == NULL) {
645 8e359fa0 2022-10-13 stsp (*commit)->refcnt++;
646 8e359fa0 2022-10-13 stsp err = got_repo_cache_commit(repo, id, *commit);
647 8e359fa0 2022-10-13 stsp }
648 8e359fa0 2022-10-13 stsp done:
649 8e359fa0 2022-10-13 stsp free(path_packfile);
650 8e359fa0 2022-10-13 stsp return err;
651 8e359fa0 2022-10-13 stsp }
652 8e359fa0 2022-10-13 stsp
653 8e359fa0 2022-10-13 stsp const struct got_error *
654 8e359fa0 2022-10-13 stsp got_object_open_as_commit(struct got_commit_object **commit,
655 8e359fa0 2022-10-13 stsp struct got_repository *repo, struct got_object_id *id)
656 8e359fa0 2022-10-13 stsp {
657 8e359fa0 2022-10-13 stsp *commit = got_repo_get_cached_commit(repo, id);
658 8e359fa0 2022-10-13 stsp if (*commit != NULL) {
659 8e359fa0 2022-10-13 stsp (*commit)->refcnt++;
660 8e359fa0 2022-10-13 stsp return NULL;
661 8e359fa0 2022-10-13 stsp }
662 8e359fa0 2022-10-13 stsp
663 8e359fa0 2022-10-13 stsp return open_commit(commit, repo, id, 0);
664 8e359fa0 2022-10-13 stsp }
665 8e359fa0 2022-10-13 stsp
666 8e359fa0 2022-10-13 stsp const struct got_error *
667 8e359fa0 2022-10-13 stsp got_object_commit_open(struct got_commit_object **commit,
668 8e359fa0 2022-10-13 stsp struct got_repository *repo, struct got_object *obj)
669 8e359fa0 2022-10-13 stsp {
670 8e359fa0 2022-10-13 stsp return open_commit(commit, repo, got_object_get_id(obj), 1);
671 8e359fa0 2022-10-13 stsp }
672 8e359fa0 2022-10-13 stsp
673 8e359fa0 2022-10-13 stsp static const struct got_error *
674 8e359fa0 2022-10-13 stsp request_packed_tree(struct got_tree_object **tree, struct got_pack *pack,
675 8e359fa0 2022-10-13 stsp int pack_idx, struct got_object_id *id)
676 8e359fa0 2022-10-13 stsp {
677 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
678 8e359fa0 2022-10-13 stsp
679 8e359fa0 2022-10-13 stsp err = got_privsep_send_tree_req(pack->privsep_child->ibuf, -1, id,
680 8e359fa0 2022-10-13 stsp pack_idx);
681 8e359fa0 2022-10-13 stsp if (err)
682 8e359fa0 2022-10-13 stsp return err;
683 8e359fa0 2022-10-13 stsp
684 8e359fa0 2022-10-13 stsp return got_privsep_recv_tree(tree, pack->privsep_child->ibuf);
685 8e359fa0 2022-10-13 stsp }
686 8e359fa0 2022-10-13 stsp
687 8e359fa0 2022-10-13 stsp static const struct got_error *
688 8e359fa0 2022-10-13 stsp read_packed_tree_privsep(struct got_tree_object **tree,
689 8e359fa0 2022-10-13 stsp struct got_pack *pack, struct got_packidx *packidx, int idx,
690 8e359fa0 2022-10-13 stsp struct got_object_id *id)
691 8e359fa0 2022-10-13 stsp {
692 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
693 8e359fa0 2022-10-13 stsp
694 8e359fa0 2022-10-13 stsp if (pack->privsep_child)
695 8e359fa0 2022-10-13 stsp return request_packed_tree(tree, pack, idx, id);
696 8e359fa0 2022-10-13 stsp
697 8e359fa0 2022-10-13 stsp err = got_pack_start_privsep_child(pack, packidx);
698 8e359fa0 2022-10-13 stsp if (err)
699 8e359fa0 2022-10-13 stsp return err;
700 8e359fa0 2022-10-13 stsp
701 8e359fa0 2022-10-13 stsp return request_packed_tree(tree, pack, idx, id);
702 8e359fa0 2022-10-13 stsp }
703 8e359fa0 2022-10-13 stsp
704 8e359fa0 2022-10-13 stsp static const struct got_error *
705 8e359fa0 2022-10-13 stsp request_tree(struct got_tree_object **tree, struct got_repository *repo,
706 8e359fa0 2022-10-13 stsp int fd, struct got_object_id *id)
707 8e359fa0 2022-10-13 stsp {
708 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
709 8e359fa0 2022-10-13 stsp struct imsgbuf *ibuf;
710 8e359fa0 2022-10-13 stsp
711 8e359fa0 2022-10-13 stsp ibuf = repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_TREE].ibuf;
712 8e359fa0 2022-10-13 stsp
713 8e359fa0 2022-10-13 stsp err = got_privsep_send_tree_req(ibuf, fd, id, -1);
714 8e359fa0 2022-10-13 stsp if (err)
715 8e359fa0 2022-10-13 stsp return err;
716 8e359fa0 2022-10-13 stsp
717 8e359fa0 2022-10-13 stsp return got_privsep_recv_tree(tree, ibuf);
718 8e359fa0 2022-10-13 stsp }
719 8e359fa0 2022-10-13 stsp
720 8e359fa0 2022-10-13 stsp static const struct got_error *
721 8e359fa0 2022-10-13 stsp read_tree_privsep(struct got_tree_object **tree, int obj_fd,
722 8e359fa0 2022-10-13 stsp struct got_object_id *id, struct got_repository *repo)
723 8e359fa0 2022-10-13 stsp {
724 8e359fa0 2022-10-13 stsp const struct got_error *err;
725 8e359fa0 2022-10-13 stsp
726 8e359fa0 2022-10-13 stsp if (repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_TREE].imsg_fd != -1)
727 8e359fa0 2022-10-13 stsp return request_tree(tree, repo, obj_fd, id);
728 8e359fa0 2022-10-13 stsp
729 1d9e43b0 2022-10-13 stsp err = start_child(repo, GOT_REPO_PRIVSEP_CHILD_TREE);
730 1d9e43b0 2022-10-13 stsp if (err)
731 8e359fa0 2022-10-13 stsp return err;
732 8e359fa0 2022-10-13 stsp
733 8e359fa0 2022-10-13 stsp return request_tree(tree, repo, obj_fd, id);
734 8e359fa0 2022-10-13 stsp }
735 8e359fa0 2022-10-13 stsp
736 8e359fa0 2022-10-13 stsp static const struct got_error *
737 8e359fa0 2022-10-13 stsp open_tree(struct got_tree_object **tree, struct got_repository *repo,
738 8e359fa0 2022-10-13 stsp struct got_object_id *id, int check_cache)
739 8e359fa0 2022-10-13 stsp {
740 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
741 8e359fa0 2022-10-13 stsp struct got_packidx *packidx = NULL;
742 8e359fa0 2022-10-13 stsp int idx;
743 8e359fa0 2022-10-13 stsp char *path_packfile = NULL;
744 8e359fa0 2022-10-13 stsp
745 8e359fa0 2022-10-13 stsp if (check_cache) {
746 8e359fa0 2022-10-13 stsp *tree = got_repo_get_cached_tree(repo, id);
747 8e359fa0 2022-10-13 stsp if (*tree != NULL) {
748 8e359fa0 2022-10-13 stsp (*tree)->refcnt++;
749 8e359fa0 2022-10-13 stsp return NULL;
750 8e359fa0 2022-10-13 stsp }
751 8e359fa0 2022-10-13 stsp } else
752 8e359fa0 2022-10-13 stsp *tree = NULL;
753 8e359fa0 2022-10-13 stsp
754 8e359fa0 2022-10-13 stsp err = got_repo_search_packidx(&packidx, &idx, repo, id);
755 8e359fa0 2022-10-13 stsp if (err == NULL) {
756 8e359fa0 2022-10-13 stsp struct got_pack *pack = NULL;
757 8e359fa0 2022-10-13 stsp
758 8e359fa0 2022-10-13 stsp err = got_packidx_get_packfile_path(&path_packfile,
759 8e359fa0 2022-10-13 stsp packidx->path_packidx);
760 8e359fa0 2022-10-13 stsp if (err)
761 8e359fa0 2022-10-13 stsp return err;
762 8e359fa0 2022-10-13 stsp
763 8e359fa0 2022-10-13 stsp pack = got_repo_get_cached_pack(repo, path_packfile);
764 8e359fa0 2022-10-13 stsp if (pack == NULL) {
765 8e359fa0 2022-10-13 stsp err = got_repo_cache_pack(&pack, repo, path_packfile,
766 8e359fa0 2022-10-13 stsp packidx);
767 8e359fa0 2022-10-13 stsp if (err)
768 8e359fa0 2022-10-13 stsp goto done;
769 8e359fa0 2022-10-13 stsp }
770 8e359fa0 2022-10-13 stsp err = read_packed_tree_privsep(tree, pack,
771 8e359fa0 2022-10-13 stsp packidx, idx, id);
772 8e359fa0 2022-10-13 stsp } else if (err->code == GOT_ERR_NO_OBJ) {
773 8e359fa0 2022-10-13 stsp int fd;
774 8e359fa0 2022-10-13 stsp
775 8e359fa0 2022-10-13 stsp err = got_object_open_loose_fd(&fd, id, repo);
776 8e359fa0 2022-10-13 stsp if (err)
777 8e359fa0 2022-10-13 stsp return err;
778 8e359fa0 2022-10-13 stsp err = read_tree_privsep(tree, fd, id, repo);
779 8e359fa0 2022-10-13 stsp }
780 8e359fa0 2022-10-13 stsp
781 8e359fa0 2022-10-13 stsp if (err == NULL) {
782 8e359fa0 2022-10-13 stsp (*tree)->refcnt++;
783 8e359fa0 2022-10-13 stsp err = got_repo_cache_tree(repo, id, *tree);
784 8e359fa0 2022-10-13 stsp }
785 8e359fa0 2022-10-13 stsp done:
786 8e359fa0 2022-10-13 stsp free(path_packfile);
787 8e359fa0 2022-10-13 stsp return err;
788 8e359fa0 2022-10-13 stsp }
789 8e359fa0 2022-10-13 stsp
790 8e359fa0 2022-10-13 stsp const struct got_error *
791 8e359fa0 2022-10-13 stsp got_object_open_as_tree(struct got_tree_object **tree,
792 8e359fa0 2022-10-13 stsp struct got_repository *repo, struct got_object_id *id)
793 8e359fa0 2022-10-13 stsp {
794 8e359fa0 2022-10-13 stsp *tree = got_repo_get_cached_tree(repo, id);
795 8e359fa0 2022-10-13 stsp if (*tree != NULL) {
796 8e359fa0 2022-10-13 stsp (*tree)->refcnt++;
797 8e359fa0 2022-10-13 stsp return NULL;
798 8e359fa0 2022-10-13 stsp }
799 8e359fa0 2022-10-13 stsp
800 8e359fa0 2022-10-13 stsp return open_tree(tree, repo, id, 0);
801 8e359fa0 2022-10-13 stsp }
802 8e359fa0 2022-10-13 stsp
803 8e359fa0 2022-10-13 stsp const struct got_error *
804 8e359fa0 2022-10-13 stsp got_object_tree_open(struct got_tree_object **tree,
805 8e359fa0 2022-10-13 stsp struct got_repository *repo, struct got_object *obj)
806 8e359fa0 2022-10-13 stsp {
807 8e359fa0 2022-10-13 stsp return open_tree(tree, repo, got_object_get_id(obj), 1);
808 8e359fa0 2022-10-13 stsp }
809 8e359fa0 2022-10-13 stsp
810 8e359fa0 2022-10-13 stsp static const struct got_error *
811 8e359fa0 2022-10-13 stsp request_packed_blob(uint8_t **outbuf, size_t *size, size_t *hdrlen, int outfd,
812 8e359fa0 2022-10-13 stsp struct got_pack *pack, struct got_packidx *packidx, int idx,
813 8e359fa0 2022-10-13 stsp struct got_object_id *id)
814 8e359fa0 2022-10-13 stsp {
815 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
816 8e359fa0 2022-10-13 stsp struct imsgbuf *ibuf = pack->privsep_child->ibuf;
817 8e359fa0 2022-10-13 stsp int outfd_child;
818 8e359fa0 2022-10-13 stsp
819 8e359fa0 2022-10-13 stsp err = pack_child_send_tempfiles(ibuf, pack);
820 8e359fa0 2022-10-13 stsp if (err)
821 8e359fa0 2022-10-13 stsp return err;
822 8e359fa0 2022-10-13 stsp
823 8e359fa0 2022-10-13 stsp outfd_child = dup(outfd);
824 8e359fa0 2022-10-13 stsp if (outfd_child == -1)
825 8e359fa0 2022-10-13 stsp return got_error_from_errno("dup");
826 8e359fa0 2022-10-13 stsp
827 8e359fa0 2022-10-13 stsp err = got_privsep_send_blob_req(pack->privsep_child->ibuf, -1, id, idx);
828 8e359fa0 2022-10-13 stsp if (err)
829 8e359fa0 2022-10-13 stsp return err;
830 8e359fa0 2022-10-13 stsp
831 8e359fa0 2022-10-13 stsp err = got_privsep_send_blob_outfd(pack->privsep_child->ibuf,
832 8e359fa0 2022-10-13 stsp outfd_child);
833 8e359fa0 2022-10-13 stsp if (err) {
834 8e359fa0 2022-10-13 stsp return err;
835 8e359fa0 2022-10-13 stsp }
836 8e359fa0 2022-10-13 stsp
837 8e359fa0 2022-10-13 stsp err = got_privsep_recv_blob(outbuf, size, hdrlen,
838 8e359fa0 2022-10-13 stsp pack->privsep_child->ibuf);
839 8e359fa0 2022-10-13 stsp if (err)
840 8e359fa0 2022-10-13 stsp return err;
841 8e359fa0 2022-10-13 stsp
842 8e359fa0 2022-10-13 stsp if (lseek(outfd, SEEK_SET, 0) == -1)
843 8e359fa0 2022-10-13 stsp err = got_error_from_errno("lseek");
844 8e359fa0 2022-10-13 stsp
845 8e359fa0 2022-10-13 stsp return err;
846 8e359fa0 2022-10-13 stsp }
847 8e359fa0 2022-10-13 stsp
848 8e359fa0 2022-10-13 stsp static const struct got_error *
849 8e359fa0 2022-10-13 stsp read_packed_blob_privsep(uint8_t **outbuf, size_t *size, size_t *hdrlen,
850 8e359fa0 2022-10-13 stsp int outfd, struct got_pack *pack, struct got_packidx *packidx, int idx,
851 8e359fa0 2022-10-13 stsp struct got_object_id *id)
852 8e359fa0 2022-10-13 stsp {
853 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
854 8e359fa0 2022-10-13 stsp
855 8e359fa0 2022-10-13 stsp if (pack->privsep_child == NULL) {
856 8e359fa0 2022-10-13 stsp err = got_pack_start_privsep_child(pack, packidx);
857 8e359fa0 2022-10-13 stsp if (err)
858 8e359fa0 2022-10-13 stsp return err;
859 8e359fa0 2022-10-13 stsp }
860 8e359fa0 2022-10-13 stsp
861 8e359fa0 2022-10-13 stsp return request_packed_blob(outbuf, size, hdrlen, outfd, pack, packidx,
862 8e359fa0 2022-10-13 stsp idx, id);
863 8e359fa0 2022-10-13 stsp }
864 8e359fa0 2022-10-13 stsp
865 8e359fa0 2022-10-13 stsp static const struct got_error *
866 8e359fa0 2022-10-13 stsp request_blob(uint8_t **outbuf, size_t *size, size_t *hdrlen, int outfd,
867 8e359fa0 2022-10-13 stsp int infd, struct got_object_id *id, struct imsgbuf *ibuf)
868 8e359fa0 2022-10-13 stsp {
869 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
870 8e359fa0 2022-10-13 stsp int outfd_child;
871 8e359fa0 2022-10-13 stsp
872 8e359fa0 2022-10-13 stsp outfd_child = dup(outfd);
873 8e359fa0 2022-10-13 stsp if (outfd_child == -1)
874 8e359fa0 2022-10-13 stsp return got_error_from_errno("dup");
875 8e359fa0 2022-10-13 stsp
876 8e359fa0 2022-10-13 stsp err = got_privsep_send_blob_req(ibuf, infd, id, -1);
877 8e359fa0 2022-10-13 stsp if (err)
878 8e359fa0 2022-10-13 stsp return err;
879 8e359fa0 2022-10-13 stsp
880 8e359fa0 2022-10-13 stsp err = got_privsep_send_blob_outfd(ibuf, outfd_child);
881 8e359fa0 2022-10-13 stsp if (err)
882 8e359fa0 2022-10-13 stsp return err;
883 8e359fa0 2022-10-13 stsp
884 8e359fa0 2022-10-13 stsp err = got_privsep_recv_blob(outbuf, size, hdrlen, ibuf);
885 8e359fa0 2022-10-13 stsp if (err)
886 8e359fa0 2022-10-13 stsp return err;
887 8e359fa0 2022-10-13 stsp
888 8e359fa0 2022-10-13 stsp if (lseek(outfd, SEEK_SET, 0) == -1)
889 8e359fa0 2022-10-13 stsp return got_error_from_errno("lseek");
890 8e359fa0 2022-10-13 stsp
891 8e359fa0 2022-10-13 stsp return err;
892 8e359fa0 2022-10-13 stsp }
893 8e359fa0 2022-10-13 stsp
894 8e359fa0 2022-10-13 stsp static const struct got_error *
895 8e359fa0 2022-10-13 stsp read_blob_privsep(uint8_t **outbuf, size_t *size, size_t *hdrlen,
896 8e359fa0 2022-10-13 stsp int outfd, int infd, struct got_object_id *id, struct got_repository *repo)
897 8e359fa0 2022-10-13 stsp {
898 8e359fa0 2022-10-13 stsp const struct got_error *err;
899 8e359fa0 2022-10-13 stsp struct imsgbuf *ibuf;
900 8e359fa0 2022-10-13 stsp
901 8e359fa0 2022-10-13 stsp if (repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_BLOB].imsg_fd != -1) {
902 8e359fa0 2022-10-13 stsp ibuf = repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_BLOB].ibuf;
903 8e359fa0 2022-10-13 stsp return request_blob(outbuf, size, hdrlen, outfd, infd, id,
904 8e359fa0 2022-10-13 stsp ibuf);
905 8e359fa0 2022-10-13 stsp }
906 8e359fa0 2022-10-13 stsp
907 1d9e43b0 2022-10-13 stsp err = start_child(repo, GOT_REPO_PRIVSEP_CHILD_BLOB);
908 1d9e43b0 2022-10-13 stsp if (err)
909 8e359fa0 2022-10-13 stsp return err;
910 8e359fa0 2022-10-13 stsp
911 1d9e43b0 2022-10-13 stsp ibuf = repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_BLOB].ibuf;
912 8e359fa0 2022-10-13 stsp return request_blob(outbuf, size, hdrlen, outfd, infd, id, ibuf);
913 8e359fa0 2022-10-13 stsp }
914 8e359fa0 2022-10-13 stsp
915 8e359fa0 2022-10-13 stsp static const struct got_error *
916 8e359fa0 2022-10-13 stsp open_blob(struct got_blob_object **blob, struct got_repository *repo,
917 8e359fa0 2022-10-13 stsp struct got_object_id *id, size_t blocksize, int outfd)
918 8e359fa0 2022-10-13 stsp {
919 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
920 8e359fa0 2022-10-13 stsp struct got_packidx *packidx = NULL;
921 8e359fa0 2022-10-13 stsp int idx, dfd = -1;
922 8e359fa0 2022-10-13 stsp char *path_packfile = NULL;
923 8e359fa0 2022-10-13 stsp uint8_t *outbuf;
924 8e359fa0 2022-10-13 stsp size_t size, hdrlen;
925 8e359fa0 2022-10-13 stsp struct stat sb;
926 8e359fa0 2022-10-13 stsp
927 8e359fa0 2022-10-13 stsp *blob = calloc(1, sizeof(**blob));
928 8e359fa0 2022-10-13 stsp if (*blob == NULL)
929 8e359fa0 2022-10-13 stsp return got_error_from_errno("calloc");
930 8e359fa0 2022-10-13 stsp
931 8e359fa0 2022-10-13 stsp (*blob)->read_buf = malloc(blocksize);
932 8e359fa0 2022-10-13 stsp if ((*blob)->read_buf == NULL) {
933 8e359fa0 2022-10-13 stsp err = got_error_from_errno("malloc");
934 8e359fa0 2022-10-13 stsp goto done;
935 8e359fa0 2022-10-13 stsp }
936 8e359fa0 2022-10-13 stsp
937 8e359fa0 2022-10-13 stsp if (ftruncate(outfd, 0L) == -1) {
938 8e359fa0 2022-10-13 stsp err = got_error_from_errno("ftruncate");
939 8e359fa0 2022-10-13 stsp goto done;
940 8e359fa0 2022-10-13 stsp }
941 8e359fa0 2022-10-13 stsp if (lseek(outfd, SEEK_SET, 0) == -1) {
942 8e359fa0 2022-10-13 stsp err = got_error_from_errno("lseek");
943 8e359fa0 2022-10-13 stsp goto done;
944 8e359fa0 2022-10-13 stsp }
945 8e359fa0 2022-10-13 stsp
946 8e359fa0 2022-10-13 stsp err = got_repo_search_packidx(&packidx, &idx, repo, id);
947 8e359fa0 2022-10-13 stsp if (err == NULL) {
948 8e359fa0 2022-10-13 stsp struct got_pack *pack = NULL;
949 8e359fa0 2022-10-13 stsp
950 8e359fa0 2022-10-13 stsp err = got_packidx_get_packfile_path(&path_packfile,
951 8e359fa0 2022-10-13 stsp packidx->path_packidx);
952 8e359fa0 2022-10-13 stsp if (err)
953 8e359fa0 2022-10-13 stsp goto done;
954 8e359fa0 2022-10-13 stsp
955 8e359fa0 2022-10-13 stsp pack = got_repo_get_cached_pack(repo, path_packfile);
956 8e359fa0 2022-10-13 stsp if (pack == NULL) {
957 8e359fa0 2022-10-13 stsp err = got_repo_cache_pack(&pack, repo, path_packfile,
958 8e359fa0 2022-10-13 stsp packidx);
959 8e359fa0 2022-10-13 stsp if (err)
960 8e359fa0 2022-10-13 stsp goto done;
961 8e359fa0 2022-10-13 stsp }
962 8e359fa0 2022-10-13 stsp err = read_packed_blob_privsep(&outbuf, &size, &hdrlen, outfd,
963 8e359fa0 2022-10-13 stsp pack, packidx, idx, id);
964 8e359fa0 2022-10-13 stsp } else if (err->code == GOT_ERR_NO_OBJ) {
965 8e359fa0 2022-10-13 stsp int infd;
966 8e359fa0 2022-10-13 stsp
967 8e359fa0 2022-10-13 stsp err = got_object_open_loose_fd(&infd, id, repo);
968 8e359fa0 2022-10-13 stsp if (err)
969 8e359fa0 2022-10-13 stsp goto done;
970 8e359fa0 2022-10-13 stsp err = read_blob_privsep(&outbuf, &size, &hdrlen, outfd, infd,
971 8e359fa0 2022-10-13 stsp id, repo);
972 8e359fa0 2022-10-13 stsp }
973 8e359fa0 2022-10-13 stsp if (err)
974 8e359fa0 2022-10-13 stsp goto done;
975 8e359fa0 2022-10-13 stsp
976 8e359fa0 2022-10-13 stsp if (hdrlen > size) {
977 8e359fa0 2022-10-13 stsp err = got_error(GOT_ERR_BAD_OBJ_HDR);
978 8e359fa0 2022-10-13 stsp goto done;
979 8e359fa0 2022-10-13 stsp }
980 8e359fa0 2022-10-13 stsp
981 8e359fa0 2022-10-13 stsp if (outbuf) {
982 8e359fa0 2022-10-13 stsp (*blob)->f = fmemopen(outbuf, size, "rb");
983 8e359fa0 2022-10-13 stsp if ((*blob)->f == NULL) {
984 8e359fa0 2022-10-13 stsp err = got_error_from_errno("fmemopen");
985 8e359fa0 2022-10-13 stsp free(outbuf);
986 8e359fa0 2022-10-13 stsp goto done;
987 8e359fa0 2022-10-13 stsp }
988 8e359fa0 2022-10-13 stsp (*blob)->data = outbuf;
989 8e359fa0 2022-10-13 stsp } else {
990 8e359fa0 2022-10-13 stsp if (fstat(outfd, &sb) == -1) {
991 8e359fa0 2022-10-13 stsp err = got_error_from_errno("fstat");
992 8e359fa0 2022-10-13 stsp goto done;
993 8e359fa0 2022-10-13 stsp }
994 8e359fa0 2022-10-13 stsp
995 8e359fa0 2022-10-13 stsp if (sb.st_size != size) {
996 8e359fa0 2022-10-13 stsp err = got_error(GOT_ERR_PRIVSEP_LEN);
997 8e359fa0 2022-10-13 stsp goto done;
998 8e359fa0 2022-10-13 stsp }
999 8e359fa0 2022-10-13 stsp
1000 8e359fa0 2022-10-13 stsp dfd = dup(outfd);
1001 8e359fa0 2022-10-13 stsp if (dfd == -1) {
1002 8e359fa0 2022-10-13 stsp err = got_error_from_errno("dup");
1003 8e359fa0 2022-10-13 stsp goto done;
1004 8e359fa0 2022-10-13 stsp }
1005 8e359fa0 2022-10-13 stsp
1006 8e359fa0 2022-10-13 stsp (*blob)->f = fdopen(dfd, "rb");
1007 8e359fa0 2022-10-13 stsp if ((*blob)->f == NULL) {
1008 8e359fa0 2022-10-13 stsp err = got_error_from_errno("fdopen");
1009 8e359fa0 2022-10-13 stsp close(dfd);
1010 8e359fa0 2022-10-13 stsp dfd = -1;
1011 8e359fa0 2022-10-13 stsp goto done;
1012 8e359fa0 2022-10-13 stsp }
1013 8e359fa0 2022-10-13 stsp }
1014 8e359fa0 2022-10-13 stsp
1015 8e359fa0 2022-10-13 stsp (*blob)->hdrlen = hdrlen;
1016 8e359fa0 2022-10-13 stsp (*blob)->blocksize = blocksize;
1017 08175cbb 2023-02-10 op memcpy(&(*blob)->id, id, sizeof(*id));
1018 8e359fa0 2022-10-13 stsp
1019 8e359fa0 2022-10-13 stsp done:
1020 8e359fa0 2022-10-13 stsp free(path_packfile);
1021 8e359fa0 2022-10-13 stsp if (err) {
1022 8e359fa0 2022-10-13 stsp if (*blob) {
1023 8e359fa0 2022-10-13 stsp got_object_blob_close(*blob);
1024 8e359fa0 2022-10-13 stsp *blob = NULL;
1025 8e359fa0 2022-10-13 stsp }
1026 8e359fa0 2022-10-13 stsp }
1027 8e359fa0 2022-10-13 stsp return err;
1028 8e359fa0 2022-10-13 stsp }
1029 8e359fa0 2022-10-13 stsp
1030 8e359fa0 2022-10-13 stsp const struct got_error *
1031 8e359fa0 2022-10-13 stsp got_object_open_as_blob(struct got_blob_object **blob,
1032 8e359fa0 2022-10-13 stsp struct got_repository *repo, struct got_object_id *id, size_t blocksize,
1033 8e359fa0 2022-10-13 stsp int outfd)
1034 8e359fa0 2022-10-13 stsp {
1035 8e359fa0 2022-10-13 stsp return open_blob(blob, repo, id, blocksize, outfd);
1036 8e359fa0 2022-10-13 stsp }
1037 8e359fa0 2022-10-13 stsp
1038 8e359fa0 2022-10-13 stsp const struct got_error *
1039 8e359fa0 2022-10-13 stsp got_object_blob_open(struct got_blob_object **blob,
1040 8e359fa0 2022-10-13 stsp struct got_repository *repo, struct got_object *obj, size_t blocksize,
1041 8e359fa0 2022-10-13 stsp int outfd)
1042 8e359fa0 2022-10-13 stsp {
1043 8e359fa0 2022-10-13 stsp return open_blob(blob, repo, got_object_get_id(obj), blocksize, outfd);
1044 8e359fa0 2022-10-13 stsp }
1045 8e359fa0 2022-10-13 stsp
1046 8e359fa0 2022-10-13 stsp static const struct got_error *
1047 8e359fa0 2022-10-13 stsp request_packed_tag(struct got_tag_object **tag, struct got_pack *pack,
1048 8e359fa0 2022-10-13 stsp int pack_idx, struct got_object_id *id)
1049 8e359fa0 2022-10-13 stsp {
1050 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
1051 8e359fa0 2022-10-13 stsp
1052 8e359fa0 2022-10-13 stsp err = got_privsep_send_tag_req(pack->privsep_child->ibuf, -1, id,
1053 8e359fa0 2022-10-13 stsp pack_idx);
1054 8e359fa0 2022-10-13 stsp if (err)
1055 8e359fa0 2022-10-13 stsp return err;
1056 8e359fa0 2022-10-13 stsp
1057 8e359fa0 2022-10-13 stsp return got_privsep_recv_tag(tag, pack->privsep_child->ibuf);
1058 8e359fa0 2022-10-13 stsp }
1059 8e359fa0 2022-10-13 stsp
1060 8e359fa0 2022-10-13 stsp static const struct got_error *
1061 8e359fa0 2022-10-13 stsp read_packed_tag_privsep(struct got_tag_object **tag,
1062 8e359fa0 2022-10-13 stsp struct got_pack *pack, struct got_packidx *packidx, int idx,
1063 8e359fa0 2022-10-13 stsp struct got_object_id *id)
1064 8e359fa0 2022-10-13 stsp {
1065 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
1066 8e359fa0 2022-10-13 stsp
1067 8e359fa0 2022-10-13 stsp if (pack->privsep_child)
1068 8e359fa0 2022-10-13 stsp return request_packed_tag(tag, pack, idx, id);
1069 8e359fa0 2022-10-13 stsp
1070 8e359fa0 2022-10-13 stsp err = got_pack_start_privsep_child(pack, packidx);
1071 8e359fa0 2022-10-13 stsp if (err)
1072 8e359fa0 2022-10-13 stsp return err;
1073 8e359fa0 2022-10-13 stsp
1074 8e359fa0 2022-10-13 stsp return request_packed_tag(tag, pack, idx, id);
1075 8e359fa0 2022-10-13 stsp }
1076 8e359fa0 2022-10-13 stsp
1077 8e359fa0 2022-10-13 stsp static const struct got_error *
1078 8e359fa0 2022-10-13 stsp request_tag(struct got_tag_object **tag, struct got_repository *repo,
1079 8e359fa0 2022-10-13 stsp int fd, struct got_object_id *id)
1080 8e359fa0 2022-10-13 stsp {
1081 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
1082 8e359fa0 2022-10-13 stsp struct imsgbuf *ibuf;
1083 8e359fa0 2022-10-13 stsp
1084 8e359fa0 2022-10-13 stsp ibuf = repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_TAG].ibuf;
1085 8e359fa0 2022-10-13 stsp
1086 8e359fa0 2022-10-13 stsp err = got_privsep_send_tag_req(ibuf, fd, id, -1);
1087 8e359fa0 2022-10-13 stsp if (err)
1088 8e359fa0 2022-10-13 stsp return err;
1089 8e359fa0 2022-10-13 stsp
1090 8e359fa0 2022-10-13 stsp return got_privsep_recv_tag(tag, ibuf);
1091 8e359fa0 2022-10-13 stsp }
1092 8e359fa0 2022-10-13 stsp
1093 8e359fa0 2022-10-13 stsp static const struct got_error *
1094 8e359fa0 2022-10-13 stsp read_tag_privsep(struct got_tag_object **tag, int obj_fd,
1095 8e359fa0 2022-10-13 stsp struct got_object_id *id, struct got_repository *repo)
1096 8e359fa0 2022-10-13 stsp {
1097 8e359fa0 2022-10-13 stsp const struct got_error *err;
1098 8e359fa0 2022-10-13 stsp
1099 8e359fa0 2022-10-13 stsp if (repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_TAG].imsg_fd != -1)
1100 8e359fa0 2022-10-13 stsp return request_tag(tag, repo, obj_fd, id);
1101 8e359fa0 2022-10-13 stsp
1102 1d9e43b0 2022-10-13 stsp err = start_child(repo, GOT_REPO_PRIVSEP_CHILD_TAG);
1103 1d9e43b0 2022-10-13 stsp if (err)
1104 8e359fa0 2022-10-13 stsp return err;
1105 8e359fa0 2022-10-13 stsp
1106 8e359fa0 2022-10-13 stsp return request_tag(tag, repo, obj_fd, id);
1107 8e359fa0 2022-10-13 stsp }
1108 8e359fa0 2022-10-13 stsp
1109 8e359fa0 2022-10-13 stsp static const struct got_error *
1110 8e359fa0 2022-10-13 stsp open_tag(struct got_tag_object **tag, struct got_repository *repo,
1111 8e359fa0 2022-10-13 stsp struct got_object_id *id, int check_cache)
1112 8e359fa0 2022-10-13 stsp {
1113 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
1114 8e359fa0 2022-10-13 stsp struct got_packidx *packidx = NULL;
1115 8e359fa0 2022-10-13 stsp int idx;
1116 8e359fa0 2022-10-13 stsp char *path_packfile = NULL;
1117 8e359fa0 2022-10-13 stsp struct got_object *obj = NULL;
1118 8e359fa0 2022-10-13 stsp int obj_type = GOT_OBJ_TYPE_ANY;
1119 8e359fa0 2022-10-13 stsp
1120 8e359fa0 2022-10-13 stsp if (check_cache) {
1121 8e359fa0 2022-10-13 stsp *tag = got_repo_get_cached_tag(repo, id);
1122 8e359fa0 2022-10-13 stsp if (*tag != NULL) {
1123 8e359fa0 2022-10-13 stsp (*tag)->refcnt++;
1124 8e359fa0 2022-10-13 stsp return NULL;
1125 8e359fa0 2022-10-13 stsp }
1126 8e359fa0 2022-10-13 stsp } else
1127 8e359fa0 2022-10-13 stsp *tag = NULL;
1128 8e359fa0 2022-10-13 stsp
1129 8e359fa0 2022-10-13 stsp err = got_repo_search_packidx(&packidx, &idx, repo, id);
1130 8e359fa0 2022-10-13 stsp if (err == NULL) {
1131 8e359fa0 2022-10-13 stsp struct got_pack *pack = NULL;
1132 8e359fa0 2022-10-13 stsp
1133 8e359fa0 2022-10-13 stsp err = got_packidx_get_packfile_path(&path_packfile,
1134 8e359fa0 2022-10-13 stsp packidx->path_packidx);
1135 8e359fa0 2022-10-13 stsp if (err)
1136 8e359fa0 2022-10-13 stsp return err;
1137 8e359fa0 2022-10-13 stsp
1138 8e359fa0 2022-10-13 stsp pack = got_repo_get_cached_pack(repo, path_packfile);
1139 8e359fa0 2022-10-13 stsp if (pack == NULL) {
1140 8e359fa0 2022-10-13 stsp err = got_repo_cache_pack(&pack, repo, path_packfile,
1141 8e359fa0 2022-10-13 stsp packidx);
1142 8e359fa0 2022-10-13 stsp if (err)
1143 8e359fa0 2022-10-13 stsp goto done;
1144 8e359fa0 2022-10-13 stsp }
1145 8e359fa0 2022-10-13 stsp
1146 8e359fa0 2022-10-13 stsp /* Beware of "lightweight" tags: Check object type first. */
1147 8e359fa0 2022-10-13 stsp err = read_packed_object_privsep(&obj, repo, pack, packidx,
1148 8e359fa0 2022-10-13 stsp idx, id);
1149 8e359fa0 2022-10-13 stsp if (err)
1150 8e359fa0 2022-10-13 stsp goto done;
1151 8e359fa0 2022-10-13 stsp obj_type = obj->type;
1152 8e359fa0 2022-10-13 stsp got_object_close(obj);
1153 8e359fa0 2022-10-13 stsp if (obj_type != GOT_OBJ_TYPE_TAG) {
1154 8e359fa0 2022-10-13 stsp err = got_error(GOT_ERR_OBJ_TYPE);
1155 8e359fa0 2022-10-13 stsp goto done;
1156 8e359fa0 2022-10-13 stsp }
1157 8e359fa0 2022-10-13 stsp err = read_packed_tag_privsep(tag, pack, packidx, idx, id);
1158 8e359fa0 2022-10-13 stsp } else if (err->code == GOT_ERR_NO_OBJ) {
1159 8e359fa0 2022-10-13 stsp int fd;
1160 8e359fa0 2022-10-13 stsp
1161 8e359fa0 2022-10-13 stsp err = got_object_open_loose_fd(&fd, id, repo);
1162 8e359fa0 2022-10-13 stsp if (err)
1163 8e359fa0 2022-10-13 stsp return err;
1164 8e359fa0 2022-10-13 stsp err = got_object_read_header_privsep(&obj, id, repo, fd);
1165 8e359fa0 2022-10-13 stsp if (err)
1166 8e359fa0 2022-10-13 stsp return err;
1167 8e359fa0 2022-10-13 stsp obj_type = obj->type;
1168 8e359fa0 2022-10-13 stsp got_object_close(obj);
1169 8e359fa0 2022-10-13 stsp if (obj_type != GOT_OBJ_TYPE_TAG)
1170 8e359fa0 2022-10-13 stsp return got_error(GOT_ERR_OBJ_TYPE);
1171 8e359fa0 2022-10-13 stsp
1172 8e359fa0 2022-10-13 stsp err = got_object_open_loose_fd(&fd, id, repo);
1173 8e359fa0 2022-10-13 stsp if (err)
1174 8e359fa0 2022-10-13 stsp return err;
1175 8e359fa0 2022-10-13 stsp err = read_tag_privsep(tag, fd, id, repo);
1176 8e359fa0 2022-10-13 stsp }
1177 8e359fa0 2022-10-13 stsp
1178 8e359fa0 2022-10-13 stsp if (err == NULL) {
1179 8e359fa0 2022-10-13 stsp (*tag)->refcnt++;
1180 8e359fa0 2022-10-13 stsp err = got_repo_cache_tag(repo, id, *tag);
1181 8e359fa0 2022-10-13 stsp }
1182 8e359fa0 2022-10-13 stsp done:
1183 8e359fa0 2022-10-13 stsp free(path_packfile);
1184 8e359fa0 2022-10-13 stsp return err;
1185 8e359fa0 2022-10-13 stsp }
1186 8e359fa0 2022-10-13 stsp
1187 8e359fa0 2022-10-13 stsp const struct got_error *
1188 8e359fa0 2022-10-13 stsp got_object_open_as_tag(struct got_tag_object **tag,
1189 8e359fa0 2022-10-13 stsp struct got_repository *repo, struct got_object_id *id)
1190 8e359fa0 2022-10-13 stsp {
1191 8e359fa0 2022-10-13 stsp *tag = got_repo_get_cached_tag(repo, id);
1192 8e359fa0 2022-10-13 stsp if (*tag != NULL) {
1193 8e359fa0 2022-10-13 stsp (*tag)->refcnt++;
1194 8e359fa0 2022-10-13 stsp return NULL;
1195 8e359fa0 2022-10-13 stsp }
1196 8e359fa0 2022-10-13 stsp
1197 8e359fa0 2022-10-13 stsp return open_tag(tag, repo, id, 0);
1198 8e359fa0 2022-10-13 stsp }
1199 8e359fa0 2022-10-13 stsp
1200 8e359fa0 2022-10-13 stsp const struct got_error *
1201 8e359fa0 2022-10-13 stsp got_object_tag_open(struct got_tag_object **tag,
1202 8e359fa0 2022-10-13 stsp struct got_repository *repo, struct got_object *obj)
1203 8e359fa0 2022-10-13 stsp {
1204 8e359fa0 2022-10-13 stsp return open_tag(tag, repo, got_object_get_id(obj), 1);
1205 8e359fa0 2022-10-13 stsp }
1206 8e359fa0 2022-10-13 stsp
1207 8e359fa0 2022-10-13 stsp const struct got_error *
1208 8e359fa0 2022-10-13 stsp got_traverse_packed_commits(struct got_object_id_queue *traversed_commits,
1209 8e359fa0 2022-10-13 stsp struct got_object_id *commit_id, const char *path,
1210 8e359fa0 2022-10-13 stsp struct got_repository *repo)
1211 8e359fa0 2022-10-13 stsp {
1212 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
1213 8e359fa0 2022-10-13 stsp struct got_pack *pack = NULL;
1214 8e359fa0 2022-10-13 stsp struct got_packidx *packidx = NULL;
1215 8e359fa0 2022-10-13 stsp char *path_packfile = NULL;
1216 8e359fa0 2022-10-13 stsp struct got_commit_object *changed_commit = NULL;
1217 8e359fa0 2022-10-13 stsp struct got_object_id *changed_commit_id = NULL;
1218 8e359fa0 2022-10-13 stsp int idx;
1219 8e359fa0 2022-10-13 stsp
1220 8e359fa0 2022-10-13 stsp err = got_repo_search_packidx(&packidx, &idx, repo, commit_id);
1221 8e359fa0 2022-10-13 stsp if (err) {
1222 8e359fa0 2022-10-13 stsp if (err->code != GOT_ERR_NO_OBJ)
1223 8e359fa0 2022-10-13 stsp return err;
1224 8e359fa0 2022-10-13 stsp return NULL;
1225 8e359fa0 2022-10-13 stsp }
1226 8e359fa0 2022-10-13 stsp
1227 8e359fa0 2022-10-13 stsp err = got_packidx_get_packfile_path(&path_packfile,
1228 8e359fa0 2022-10-13 stsp packidx->path_packidx);
1229 8e359fa0 2022-10-13 stsp if (err)
1230 8e359fa0 2022-10-13 stsp return err;
1231 8e359fa0 2022-10-13 stsp
1232 8e359fa0 2022-10-13 stsp pack = got_repo_get_cached_pack(repo, path_packfile);
1233 8e359fa0 2022-10-13 stsp if (pack == NULL) {
1234 8e359fa0 2022-10-13 stsp err = got_repo_cache_pack(&pack, repo, path_packfile, packidx);
1235 8e359fa0 2022-10-13 stsp if (err)
1236 8e359fa0 2022-10-13 stsp goto done;
1237 8e359fa0 2022-10-13 stsp }
1238 8e359fa0 2022-10-13 stsp
1239 8e359fa0 2022-10-13 stsp if (pack->privsep_child == NULL) {
1240 8e359fa0 2022-10-13 stsp err = got_pack_start_privsep_child(pack, packidx);
1241 8e359fa0 2022-10-13 stsp if (err)
1242 8e359fa0 2022-10-13 stsp goto done;
1243 8e359fa0 2022-10-13 stsp }
1244 8e359fa0 2022-10-13 stsp
1245 8e359fa0 2022-10-13 stsp err = got_privsep_send_commit_traversal_request(
1246 8e359fa0 2022-10-13 stsp pack->privsep_child->ibuf, commit_id, idx, path);
1247 8e359fa0 2022-10-13 stsp if (err)
1248 8e359fa0 2022-10-13 stsp goto done;
1249 8e359fa0 2022-10-13 stsp
1250 8e359fa0 2022-10-13 stsp err = got_privsep_recv_traversed_commits(&changed_commit,
1251 8e359fa0 2022-10-13 stsp &changed_commit_id, traversed_commits, pack->privsep_child->ibuf);
1252 8e359fa0 2022-10-13 stsp if (err)
1253 8e359fa0 2022-10-13 stsp goto done;
1254 8e359fa0 2022-10-13 stsp
1255 8e359fa0 2022-10-13 stsp if (changed_commit) {
1256 8e359fa0 2022-10-13 stsp /*
1257 8e359fa0 2022-10-13 stsp * Cache the commit in which the path was changed.
1258 8e359fa0 2022-10-13 stsp * This commit might be opened again soon.
1259 8e359fa0 2022-10-13 stsp */
1260 8e359fa0 2022-10-13 stsp changed_commit->refcnt++;
1261 8e359fa0 2022-10-13 stsp err = got_repo_cache_commit(repo, changed_commit_id,
1262 8e359fa0 2022-10-13 stsp changed_commit);
1263 8e359fa0 2022-10-13 stsp got_object_commit_close(changed_commit);
1264 8e359fa0 2022-10-13 stsp }
1265 8e359fa0 2022-10-13 stsp done:
1266 8e359fa0 2022-10-13 stsp free(path_packfile);
1267 8e359fa0 2022-10-13 stsp free(changed_commit_id);
1268 8e359fa0 2022-10-13 stsp return err;
1269 8e359fa0 2022-10-13 stsp }
1270 8e359fa0 2022-10-13 stsp
1271 8e359fa0 2022-10-13 stsp const struct got_error *
1272 8e359fa0 2022-10-13 stsp got_object_enumerate(int *found_all_objects,
1273 8e359fa0 2022-10-13 stsp got_object_enumerate_commit_cb cb_commit,
1274 8e359fa0 2022-10-13 stsp got_object_enumerate_tree_cb cb_tree, void *cb_arg,
1275 8e359fa0 2022-10-13 stsp struct got_object_id **ours, int nours,
1276 8e359fa0 2022-10-13 stsp struct got_object_id **theirs, int ntheirs,
1277 8e359fa0 2022-10-13 stsp struct got_packidx *packidx, struct got_repository *repo)
1278 8e359fa0 2022-10-13 stsp {
1279 8e359fa0 2022-10-13 stsp const struct got_error *err = NULL;
1280 8e359fa0 2022-10-13 stsp struct got_pack *pack;
1281 8e359fa0 2022-10-13 stsp char *path_packfile = NULL;
1282 8e359fa0 2022-10-13 stsp
1283 8e359fa0 2022-10-13 stsp err = got_packidx_get_packfile_path(&path_packfile,
1284 8e359fa0 2022-10-13 stsp packidx->path_packidx);
1285 8e359fa0 2022-10-13 stsp if (err)
1286 8e359fa0 2022-10-13 stsp return err;
1287 8e359fa0 2022-10-13 stsp
1288 8e359fa0 2022-10-13 stsp pack = got_repo_get_cached_pack(repo, path_packfile);
1289 8e359fa0 2022-10-13 stsp if (pack == NULL) {
1290 8e359fa0 2022-10-13 stsp err = got_repo_cache_pack(&pack, repo, path_packfile, packidx);
1291 8e359fa0 2022-10-13 stsp if (err)
1292 8e359fa0 2022-10-13 stsp goto done;
1293 8e359fa0 2022-10-13 stsp }
1294 8e359fa0 2022-10-13 stsp
1295 8e359fa0 2022-10-13 stsp if (pack->privsep_child == NULL) {
1296 8e359fa0 2022-10-13 stsp err = got_pack_start_privsep_child(pack, packidx);
1297 8e359fa0 2022-10-13 stsp if (err)
1298 8e359fa0 2022-10-13 stsp goto done;
1299 8e359fa0 2022-10-13 stsp }
1300 8e359fa0 2022-10-13 stsp
1301 8e359fa0 2022-10-13 stsp err = got_privsep_send_object_enumeration_request(
1302 8e359fa0 2022-10-13 stsp pack->privsep_child->ibuf);
1303 8e359fa0 2022-10-13 stsp if (err)
1304 8e359fa0 2022-10-13 stsp goto done;
1305 8e359fa0 2022-10-13 stsp
1306 8e359fa0 2022-10-13 stsp err = got_privsep_send_object_idlist(pack->privsep_child->ibuf,
1307 8e359fa0 2022-10-13 stsp ours, nours);
1308 8e359fa0 2022-10-13 stsp if (err)
1309 8e359fa0 2022-10-13 stsp goto done;
1310 8e359fa0 2022-10-13 stsp err = got_privsep_send_object_idlist_done(pack->privsep_child->ibuf);
1311 8e359fa0 2022-10-13 stsp if (err)
1312 8e359fa0 2022-10-13 stsp goto done;
1313 8e359fa0 2022-10-13 stsp
1314 8e359fa0 2022-10-13 stsp err = got_privsep_send_object_idlist(pack->privsep_child->ibuf,
1315 8e359fa0 2022-10-13 stsp theirs, ntheirs);
1316 8e359fa0 2022-10-13 stsp if (err)
1317 8e359fa0 2022-10-13 stsp goto done;
1318 8e359fa0 2022-10-13 stsp err = got_privsep_send_object_idlist_done(pack->privsep_child->ibuf);
1319 8e359fa0 2022-10-13 stsp if (err)
1320 8e359fa0 2022-10-13 stsp goto done;
1321 8e359fa0 2022-10-13 stsp
1322 8e359fa0 2022-10-13 stsp err = got_privsep_recv_enumerated_objects(found_all_objects,
1323 8e359fa0 2022-10-13 stsp pack->privsep_child->ibuf, cb_commit, cb_tree, cb_arg, repo);
1324 8e359fa0 2022-10-13 stsp done:
1325 8e359fa0 2022-10-13 stsp free(path_packfile);
1326 8e359fa0 2022-10-13 stsp return err;
1327 8e359fa0 2022-10-13 stsp }