Blame


1 d71d75ad 2017-11-05 stsp /*
2 a1fd68d8 2018-01-12 stsp * Copyright (c) 2018 Stefan Sperling <stsp@openbsd.org>
3 d71d75ad 2017-11-05 stsp *
4 d71d75ad 2017-11-05 stsp * Permission to use, copy, modify, and distribute this software for any
5 d71d75ad 2017-11-05 stsp * purpose with or without fee is hereby granted, provided that the above
6 d71d75ad 2017-11-05 stsp * copyright notice and this permission notice appear in all copies.
7 d71d75ad 2017-11-05 stsp *
8 d71d75ad 2017-11-05 stsp * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 d71d75ad 2017-11-05 stsp * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 d71d75ad 2017-11-05 stsp * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 d71d75ad 2017-11-05 stsp * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 d71d75ad 2017-11-05 stsp * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 d71d75ad 2017-11-05 stsp * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 d71d75ad 2017-11-05 stsp * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15 d71d75ad 2017-11-05 stsp */
16 d71d75ad 2017-11-05 stsp
17 2178c42e 2018-04-22 stsp #include <sys/types.h>
18 0ffeb3c2 2017-11-26 stsp #include <sys/stat.h>
19 d1cda826 2017-11-06 stsp #include <sys/queue.h>
20 2178c42e 2018-04-22 stsp #include <sys/uio.h>
21 2178c42e 2018-04-22 stsp #include <sys/socket.h>
22 2178c42e 2018-04-22 stsp #include <sys/wait.h>
23 876c234b 2018-09-10 stsp #include <sys/syslimits.h>
24 d1cda826 2017-11-06 stsp
25 a1fd68d8 2018-01-12 stsp #include <errno.h>
26 2178c42e 2018-04-22 stsp #include <fcntl.h>
27 d71d75ad 2017-11-05 stsp #include <stdio.h>
28 ab9a70b2 2017-11-06 stsp #include <stdlib.h>
29 ab9a70b2 2017-11-06 stsp #include <string.h>
30 2178c42e 2018-04-22 stsp #include <stdint.h>
31 d71d75ad 2017-11-05 stsp #include <sha1.h>
32 ab9a70b2 2017-11-06 stsp #include <zlib.h>
33 ab9a70b2 2017-11-06 stsp #include <ctype.h>
34 ab9a70b2 2017-11-06 stsp #include <limits.h>
35 2178c42e 2018-04-22 stsp #include <imsg.h>
36 788c352e 2018-06-16 stsp #include <time.h>
37 d71d75ad 2017-11-05 stsp
38 ab9a70b2 2017-11-06 stsp #include "got_error.h"
39 d71d75ad 2017-11-05 stsp #include "got_object.h"
40 ab9a70b2 2017-11-06 stsp #include "got_repository.h"
41 511a516b 2018-05-19 stsp #include "got_opentemp.h"
42 d71d75ad 2017-11-05 stsp
43 718b3ab0 2018-03-17 stsp #include "got_lib_sha1.h"
44 718b3ab0 2018-03-17 stsp #include "got_lib_delta.h"
45 718b3ab0 2018-03-17 stsp #include "got_lib_pack.h"
46 2178c42e 2018-04-22 stsp #include "got_lib_path.h"
47 63581804 2018-07-09 stsp #include "got_lib_inflate.h"
48 718b3ab0 2018-03-17 stsp #include "got_lib_object.h"
49 2178c42e 2018-04-22 stsp #include "got_lib_privsep.h"
50 6bef87be 2018-09-11 stsp #include "got_lib_object_idcache.h"
51 6bef87be 2018-09-11 stsp #include "got_lib_object_cache.h"
52 ad242220 2018-09-08 stsp #include "got_lib_object_parse.h"
53 7bb0daa1 2018-06-21 stsp #include "got_lib_repository.h"
54 1411938b 2018-02-12 stsp
55 ab9a70b2 2017-11-06 stsp #ifndef MIN
56 ab9a70b2 2017-11-06 stsp #define MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
57 ab9a70b2 2017-11-06 stsp #endif
58 ef0981d5 2018-02-12 stsp
59 a1fd68d8 2018-01-12 stsp int
60 a1fd68d8 2018-01-12 stsp got_object_id_cmp(struct got_object_id *id1, struct got_object_id *id2)
61 a1fd68d8 2018-01-12 stsp {
62 a1fd68d8 2018-01-12 stsp return memcmp(id1->sha1, id2->sha1, SHA1_DIGEST_LENGTH);
63 8bf5b3c9 2018-03-17 stsp }
64 8bf5b3c9 2018-03-17 stsp
65 8bf5b3c9 2018-03-17 stsp struct got_object_id *
66 8bf5b3c9 2018-03-17 stsp got_object_id_dup(struct got_object_id *id1)
67 8bf5b3c9 2018-03-17 stsp {
68 8bf5b3c9 2018-03-17 stsp struct got_object_id *id2;
69 8bf5b3c9 2018-03-17 stsp
70 8bf5b3c9 2018-03-17 stsp id2 = malloc(sizeof(*id2));
71 8bf5b3c9 2018-03-17 stsp if (id2 == NULL)
72 8bf5b3c9 2018-03-17 stsp return NULL;
73 8bf5b3c9 2018-03-17 stsp memcpy(id2, id1, sizeof(*id2));
74 8bf5b3c9 2018-03-17 stsp return id2;
75 3235492e 2018-04-01 stsp }
76 3235492e 2018-04-01 stsp
77 3235492e 2018-04-01 stsp struct got_object_id *
78 3235492e 2018-04-01 stsp got_object_get_id(struct got_object *obj)
79 3235492e 2018-04-01 stsp {
80 6402fb3c 2018-09-15 stsp return &obj->id;
81 bacc9935 2018-05-20 stsp }
82 bacc9935 2018-05-20 stsp
83 bacc9935 2018-05-20 stsp const struct got_error *
84 bacc9935 2018-05-20 stsp got_object_get_id_str(char **outbuf, struct got_object *obj)
85 bacc9935 2018-05-20 stsp {
86 bacc9935 2018-05-20 stsp return got_object_id_str(outbuf, &obj->id);
87 a1fd68d8 2018-01-12 stsp }
88 d71d75ad 2017-11-05 stsp
89 b107e67f 2018-01-19 stsp int
90 b107e67f 2018-01-19 stsp got_object_get_type(struct got_object *obj)
91 a1fd68d8 2018-01-12 stsp {
92 b107e67f 2018-01-19 stsp switch (obj->type) {
93 a1fd68d8 2018-01-12 stsp case GOT_OBJ_TYPE_COMMIT:
94 a1fd68d8 2018-01-12 stsp case GOT_OBJ_TYPE_TREE:
95 a1fd68d8 2018-01-12 stsp case GOT_OBJ_TYPE_BLOB:
96 b107e67f 2018-01-19 stsp case GOT_OBJ_TYPE_TAG:
97 b107e67f 2018-01-19 stsp return obj->type;
98 96f5e8b3 2018-01-23 stsp default:
99 96f5e8b3 2018-01-23 stsp abort();
100 96f5e8b3 2018-01-23 stsp break;
101 d71d75ad 2017-11-05 stsp }
102 d71d75ad 2017-11-05 stsp
103 96f5e8b3 2018-01-23 stsp /* not reached */
104 96f5e8b3 2018-01-23 stsp return 0;
105 d71d75ad 2017-11-05 stsp }
106 ab9a70b2 2017-11-06 stsp
107 ab9a70b2 2017-11-06 stsp static const struct got_error *
108 4558fcd4 2018-01-14 stsp object_path(char **path, struct got_object_id *id, struct got_repository *repo)
109 ab9a70b2 2017-11-06 stsp {
110 ab9a70b2 2017-11-06 stsp const struct got_error *err = NULL;
111 7a132809 2018-07-23 stsp char *hex = NULL;
112 d1cda826 2017-11-06 stsp char *path_objects = got_repo_get_path_objects(repo);
113 e6b1056e 2018-04-22 stsp
114 e6b1056e 2018-04-22 stsp *path = NULL;
115 ab9a70b2 2017-11-06 stsp
116 ab9a70b2 2017-11-06 stsp if (path_objects == NULL)
117 0a585a0d 2018-03-17 stsp return got_error_from_errno();
118 ab9a70b2 2017-11-06 stsp
119 ef0981d5 2018-02-12 stsp err = got_object_id_str(&hex, id);
120 ef0981d5 2018-02-12 stsp if (err)
121 7a132809 2018-07-23 stsp goto done;
122 ab9a70b2 2017-11-06 stsp
123 d1cda826 2017-11-06 stsp if (asprintf(path, "%s/%.2x/%s", path_objects,
124 d1cda826 2017-11-06 stsp id->sha1[0], hex + 2) == -1)
125 0a585a0d 2018-03-17 stsp err = got_error_from_errno();
126 ab9a70b2 2017-11-06 stsp
127 7a132809 2018-07-23 stsp done:
128 ef0981d5 2018-02-12 stsp free(hex);
129 d1cda826 2017-11-06 stsp free(path_objects);
130 d1cda826 2017-11-06 stsp return err;
131 d1cda826 2017-11-06 stsp }
132 d1cda826 2017-11-06 stsp
133 4ee4114f 2018-01-23 stsp static const struct got_error *
134 d5003b79 2018-04-22 stsp open_loose_object(int *fd, struct got_object *obj, struct got_repository *repo)
135 d1cda826 2017-11-06 stsp {
136 d1cda826 2017-11-06 stsp const struct got_error *err = NULL;
137 a1fd68d8 2018-01-12 stsp char *path;
138 6c00b545 2018-01-17 stsp
139 6c00b545 2018-01-17 stsp err = object_path(&path, &obj->id, repo);
140 d1cda826 2017-11-06 stsp if (err)
141 d1cda826 2017-11-06 stsp return err;
142 d5003b79 2018-04-22 stsp *fd = open(path, O_RDONLY | O_NOFOLLOW, GOT_DEFAULT_FILE_MODE);
143 d5003b79 2018-04-22 stsp if (*fd == -1) {
144 6c00b545 2018-01-17 stsp err = got_error_from_errno();
145 6c00b545 2018-01-17 stsp goto done;
146 a1fd68d8 2018-01-12 stsp }
147 4558fcd4 2018-01-14 stsp done:
148 4558fcd4 2018-01-14 stsp free(path);
149 2090a03d 2018-09-09 stsp return err;
150 2090a03d 2018-09-09 stsp }
151 2090a03d 2018-09-09 stsp
152 2090a03d 2018-09-09 stsp static const struct got_error *
153 2090a03d 2018-09-09 stsp get_packfile_path(char **path_packfile, struct got_packidx *packidx)
154 2090a03d 2018-09-09 stsp {
155 2090a03d 2018-09-09 stsp size_t size;
156 2090a03d 2018-09-09 stsp
157 2090a03d 2018-09-09 stsp /* Packfile path contains ".pack" instead of ".idx", so add one byte. */
158 2090a03d 2018-09-09 stsp size = strlen(packidx->path_packidx) + 2;
159 2090a03d 2018-09-09 stsp if (size < GOT_PACKFILE_NAMELEN + 1)
160 2090a03d 2018-09-09 stsp return got_error(GOT_ERR_BAD_PATH);
161 2090a03d 2018-09-09 stsp
162 2090a03d 2018-09-09 stsp *path_packfile = calloc(size, sizeof(**path_packfile));
163 2090a03d 2018-09-09 stsp if (*path_packfile == NULL)
164 2090a03d 2018-09-09 stsp return got_error_from_errno();
165 2090a03d 2018-09-09 stsp
166 2090a03d 2018-09-09 stsp /* Copy up to and excluding ".idx". */
167 2090a03d 2018-09-09 stsp if (strlcpy(*path_packfile, packidx->path_packidx,
168 2090a03d 2018-09-09 stsp size - strlen(GOT_PACKIDX_SUFFIX) - 1) >= size)
169 2090a03d 2018-09-09 stsp return got_error(GOT_ERR_NO_SPACE);
170 2090a03d 2018-09-09 stsp
171 2090a03d 2018-09-09 stsp if (strlcat(*path_packfile, GOT_PACKFILE_SUFFIX, size) >= size)
172 2090a03d 2018-09-09 stsp return got_error(GOT_ERR_NO_SPACE);
173 2090a03d 2018-09-09 stsp
174 2090a03d 2018-09-09 stsp return NULL;
175 2090a03d 2018-09-09 stsp }
176 2090a03d 2018-09-09 stsp
177 2090a03d 2018-09-09 stsp static const struct got_error *
178 2090a03d 2018-09-09 stsp open_packed_object(struct got_object **obj, struct got_object_id *id,
179 2090a03d 2018-09-09 stsp struct got_repository *repo)
180 2090a03d 2018-09-09 stsp {
181 2090a03d 2018-09-09 stsp const struct got_error *err = NULL;
182 2090a03d 2018-09-09 stsp struct got_pack *pack = NULL;
183 2090a03d 2018-09-09 stsp struct got_packidx *packidx = NULL;
184 2090a03d 2018-09-09 stsp int idx;
185 2090a03d 2018-09-09 stsp char *path_packfile;
186 2090a03d 2018-09-09 stsp
187 2090a03d 2018-09-09 stsp err = got_repo_search_packidx(&packidx, &idx, repo, id);
188 2090a03d 2018-09-09 stsp if (err)
189 2090a03d 2018-09-09 stsp return err;
190 2090a03d 2018-09-09 stsp
191 2090a03d 2018-09-09 stsp err = get_packfile_path(&path_packfile, packidx);
192 2090a03d 2018-09-09 stsp if (err)
193 2090a03d 2018-09-09 stsp return err;
194 2090a03d 2018-09-09 stsp
195 2090a03d 2018-09-09 stsp pack = got_repo_get_cached_pack(repo, path_packfile);
196 2090a03d 2018-09-09 stsp if (pack == NULL) {
197 2090a03d 2018-09-09 stsp err = got_repo_cache_pack(&pack, repo, path_packfile, packidx);
198 2090a03d 2018-09-09 stsp if (err)
199 2090a03d 2018-09-09 stsp goto done;
200 2090a03d 2018-09-09 stsp }
201 2090a03d 2018-09-09 stsp
202 876c234b 2018-09-10 stsp err = got_object_packed_read_privsep(obj, repo, pack, packidx, idx, id);
203 2090a03d 2018-09-09 stsp if (err)
204 2090a03d 2018-09-09 stsp goto done;
205 2090a03d 2018-09-09 stsp
206 2090a03d 2018-09-09 stsp err = got_repo_cache_pack(NULL, repo, (*obj)->path_packfile, packidx);
207 2090a03d 2018-09-09 stsp done:
208 2090a03d 2018-09-09 stsp free(path_packfile);
209 4558fcd4 2018-01-14 stsp return err;
210 4558fcd4 2018-01-14 stsp }
211 a1fd68d8 2018-01-12 stsp
212 4558fcd4 2018-01-14 stsp const struct got_error *
213 4558fcd4 2018-01-14 stsp got_object_open(struct got_object **obj, struct got_repository *repo,
214 4558fcd4 2018-01-14 stsp struct got_object_id *id)
215 4558fcd4 2018-01-14 stsp {
216 6c00b545 2018-01-17 stsp const struct got_error *err = NULL;
217 6c00b545 2018-01-17 stsp char *path;
218 2178c42e 2018-04-22 stsp int fd;
219 7bb0daa1 2018-06-21 stsp
220 7bb0daa1 2018-06-21 stsp *obj = got_repo_get_cached_object(repo, id);
221 7bb0daa1 2018-06-21 stsp if (*obj != NULL) {
222 7bb0daa1 2018-06-21 stsp (*obj)->refcnt++;
223 7bb0daa1 2018-06-21 stsp return NULL;
224 7bb0daa1 2018-06-21 stsp }
225 4558fcd4 2018-01-14 stsp
226 59790a32 2018-09-15 stsp err = open_packed_object(obj, id, repo);
227 59790a32 2018-09-15 stsp if (err && err->code != GOT_ERR_NO_OBJ)
228 59790a32 2018-09-15 stsp return err;
229 59790a32 2018-09-15 stsp if (*obj) {
230 59790a32 2018-09-15 stsp (*obj)->refcnt++;
231 59790a32 2018-09-15 stsp return got_repo_cache_object(repo, id, *obj);
232 59790a32 2018-09-15 stsp }
233 59790a32 2018-09-15 stsp
234 6c00b545 2018-01-17 stsp err = object_path(&path, id, repo);
235 4558fcd4 2018-01-14 stsp if (err)
236 4558fcd4 2018-01-14 stsp return err;
237 4558fcd4 2018-01-14 stsp
238 2178c42e 2018-04-22 stsp fd = open(path, O_RDONLY | O_NOFOLLOW, GOT_DEFAULT_FILE_MODE);
239 2178c42e 2018-04-22 stsp if (fd == -1) {
240 59790a32 2018-09-15 stsp if (errno == ENOENT)
241 59790a32 2018-09-15 stsp err = got_error(GOT_ERR_NO_OBJ);
242 59790a32 2018-09-15 stsp else
243 6c00b545 2018-01-17 stsp err = got_error_from_errno();
244 59790a32 2018-09-15 stsp goto done;
245 6c00b545 2018-01-17 stsp } else {
246 ad242220 2018-09-08 stsp err = got_object_read_header_privsep(obj, repo, fd);
247 6c00b545 2018-01-17 stsp if (err)
248 6c00b545 2018-01-17 stsp goto done;
249 ab9a70b2 2017-11-06 stsp memcpy((*obj)->id.sha1, id->sha1, SHA1_DIGEST_LENGTH);
250 6c00b545 2018-01-17 stsp }
251 7bb0daa1 2018-06-21 stsp
252 59790a32 2018-09-15 stsp (*obj)->refcnt++;
253 59790a32 2018-09-15 stsp err = got_repo_cache_object(repo, id, *obj);
254 6c00b545 2018-01-17 stsp done:
255 6c00b545 2018-01-17 stsp free(path);
256 2178c42e 2018-04-22 stsp if (fd != -1)
257 2178c42e 2018-04-22 stsp close(fd);
258 ab9a70b2 2017-11-06 stsp return err;
259 6c00b545 2018-01-17 stsp
260 ab9a70b2 2017-11-06 stsp }
261 ab9a70b2 2017-11-06 stsp
262 6dfa2fd3 2018-02-12 stsp const struct got_error *
263 6dfa2fd3 2018-02-12 stsp got_object_open_by_id_str(struct got_object **obj, struct got_repository *repo,
264 6dfa2fd3 2018-02-12 stsp const char *id_str)
265 6dfa2fd3 2018-02-12 stsp {
266 6dfa2fd3 2018-02-12 stsp struct got_object_id id;
267 6dfa2fd3 2018-02-12 stsp
268 6dfa2fd3 2018-02-12 stsp if (!got_parse_sha1_digest(id.sha1, id_str))
269 6dfa2fd3 2018-02-12 stsp return got_error(GOT_ERR_BAD_OBJ_ID_STR);
270 6dfa2fd3 2018-02-12 stsp
271 6dfa2fd3 2018-02-12 stsp return got_object_open(obj, repo, &id);
272 434025f3 2018-09-16 stsp }
273 434025f3 2018-09-16 stsp
274 434025f3 2018-09-16 stsp static const struct got_error *
275 434025f3 2018-09-16 stsp open_commit(struct got_commit_object **commit,
276 434025f3 2018-09-16 stsp struct got_repository *repo, struct got_object *obj, int check_cache)
277 434025f3 2018-09-16 stsp {
278 434025f3 2018-09-16 stsp const struct got_error *err = NULL;
279 434025f3 2018-09-16 stsp
280 434025f3 2018-09-16 stsp if (check_cache) {
281 434025f3 2018-09-16 stsp *commit = got_repo_get_cached_commit(repo, &obj->id);
282 434025f3 2018-09-16 stsp if (*commit != NULL) {
283 434025f3 2018-09-16 stsp (*commit)->refcnt++;
284 434025f3 2018-09-16 stsp return NULL;
285 434025f3 2018-09-16 stsp }
286 434025f3 2018-09-16 stsp } else
287 434025f3 2018-09-16 stsp *commit = NULL;
288 434025f3 2018-09-16 stsp
289 434025f3 2018-09-16 stsp if (obj->type != GOT_OBJ_TYPE_COMMIT)
290 434025f3 2018-09-16 stsp return got_error(GOT_ERR_OBJ_TYPE);
291 434025f3 2018-09-16 stsp
292 434025f3 2018-09-16 stsp if (obj->flags & GOT_OBJ_FLAG_PACKED) {
293 434025f3 2018-09-16 stsp struct got_pack *pack;
294 434025f3 2018-09-16 stsp pack = got_repo_get_cached_pack(repo, obj->path_packfile);
295 434025f3 2018-09-16 stsp if (pack == NULL) {
296 434025f3 2018-09-16 stsp err = got_repo_cache_pack(&pack, repo,
297 434025f3 2018-09-16 stsp obj->path_packfile, NULL);
298 434025f3 2018-09-16 stsp if (err)
299 434025f3 2018-09-16 stsp return err;
300 434025f3 2018-09-16 stsp }
301 434025f3 2018-09-16 stsp err = got_object_read_packed_commit_privsep(commit, obj, pack);
302 434025f3 2018-09-16 stsp } else {
303 434025f3 2018-09-16 stsp int fd;
304 434025f3 2018-09-16 stsp err = open_loose_object(&fd, obj, repo);
305 434025f3 2018-09-16 stsp if (err)
306 434025f3 2018-09-16 stsp return err;
307 434025f3 2018-09-16 stsp err = got_object_read_commit_privsep(commit, obj, fd, repo);
308 434025f3 2018-09-16 stsp close(fd);
309 434025f3 2018-09-16 stsp }
310 434025f3 2018-09-16 stsp
311 434025f3 2018-09-16 stsp if (err == NULL) {
312 434025f3 2018-09-16 stsp (*commit)->refcnt++;
313 434025f3 2018-09-16 stsp err = got_repo_cache_commit(repo, &obj->id, *commit);
314 7762fe12 2018-11-05 stsp }
315 7762fe12 2018-11-05 stsp
316 7762fe12 2018-11-05 stsp return err;
317 7762fe12 2018-11-05 stsp }
318 7762fe12 2018-11-05 stsp
319 7762fe12 2018-11-05 stsp static const struct got_error *
320 7762fe12 2018-11-05 stsp open_mini_commit(struct got_commit_object_mini **commit,
321 7762fe12 2018-11-05 stsp struct got_repository *repo, struct got_object *obj)
322 7762fe12 2018-11-05 stsp {
323 7762fe12 2018-11-05 stsp const struct got_error *err = NULL;
324 7762fe12 2018-11-05 stsp
325 7762fe12 2018-11-05 stsp *commit = NULL;
326 7762fe12 2018-11-05 stsp
327 7762fe12 2018-11-05 stsp if (obj->type != GOT_OBJ_TYPE_COMMIT)
328 7762fe12 2018-11-05 stsp return got_error(GOT_ERR_OBJ_TYPE);
329 7762fe12 2018-11-05 stsp
330 7762fe12 2018-11-05 stsp if (obj->flags & GOT_OBJ_FLAG_PACKED) {
331 7762fe12 2018-11-05 stsp struct got_pack *pack;
332 7762fe12 2018-11-05 stsp pack = got_repo_get_cached_pack(repo, obj->path_packfile);
333 7762fe12 2018-11-05 stsp if (pack == NULL) {
334 7762fe12 2018-11-05 stsp err = got_repo_cache_pack(&pack, repo,
335 7762fe12 2018-11-05 stsp obj->path_packfile, NULL);
336 7762fe12 2018-11-05 stsp if (err)
337 7762fe12 2018-11-05 stsp return err;
338 7762fe12 2018-11-05 stsp }
339 7762fe12 2018-11-05 stsp err = got_object_read_packed_mini_commit_privsep(commit, obj,
340 7762fe12 2018-11-05 stsp pack);
341 7762fe12 2018-11-05 stsp } else {
342 7762fe12 2018-11-05 stsp int fd;
343 7762fe12 2018-11-05 stsp err = open_loose_object(&fd, obj, repo);
344 7762fe12 2018-11-05 stsp if (err)
345 7762fe12 2018-11-05 stsp return err;
346 7762fe12 2018-11-05 stsp err = got_object_read_mini_commit_privsep(commit, obj, fd,
347 7762fe12 2018-11-05 stsp repo);
348 7762fe12 2018-11-05 stsp close(fd);
349 434025f3 2018-09-16 stsp }
350 434025f3 2018-09-16 stsp
351 434025f3 2018-09-16 stsp return err;
352 bff6ca00 2018-04-23 stsp }
353 bff6ca00 2018-04-23 stsp
354 bff6ca00 2018-04-23 stsp const struct got_error *
355 be6a1b5a 2018-06-11 stsp got_object_open_as_commit(struct got_commit_object **commit,
356 be6a1b5a 2018-06-11 stsp struct got_repository *repo, struct got_object_id *id)
357 be6a1b5a 2018-06-11 stsp {
358 be6a1b5a 2018-06-11 stsp const struct got_error *err;
359 be6a1b5a 2018-06-11 stsp struct got_object *obj;
360 835e0dbd 2018-06-21 stsp
361 e8eb494a 2018-09-16 stsp *commit = got_repo_get_cached_commit(repo, id);
362 e8eb494a 2018-09-16 stsp if (*commit != NULL) {
363 e8eb494a 2018-09-16 stsp (*commit)->refcnt++;
364 e8eb494a 2018-09-16 stsp return NULL;
365 e8eb494a 2018-09-16 stsp }
366 be6a1b5a 2018-06-11 stsp
367 be6a1b5a 2018-06-11 stsp err = got_object_open(&obj, repo, id);
368 be6a1b5a 2018-06-11 stsp if (err)
369 be6a1b5a 2018-06-11 stsp return err;
370 be6a1b5a 2018-06-11 stsp if (got_object_get_type(obj) != GOT_OBJ_TYPE_COMMIT) {
371 be6a1b5a 2018-06-11 stsp err = got_error(GOT_ERR_OBJ_TYPE);
372 be6a1b5a 2018-06-11 stsp goto done;
373 be6a1b5a 2018-06-11 stsp }
374 be6a1b5a 2018-06-11 stsp
375 434025f3 2018-09-16 stsp err = open_commit(commit, repo, obj, 0);
376 be6a1b5a 2018-06-11 stsp done:
377 be6a1b5a 2018-06-11 stsp got_object_close(obj);
378 be6a1b5a 2018-06-11 stsp return err;
379 434025f3 2018-09-16 stsp }
380 434025f3 2018-09-16 stsp
381 434025f3 2018-09-16 stsp const struct got_error *
382 434025f3 2018-09-16 stsp got_object_commit_open(struct got_commit_object **commit,
383 434025f3 2018-09-16 stsp struct got_repository *repo, struct got_object *obj)
384 434025f3 2018-09-16 stsp {
385 434025f3 2018-09-16 stsp return open_commit(commit, repo, obj, 1);
386 7762fe12 2018-11-05 stsp }
387 7762fe12 2018-11-05 stsp
388 7762fe12 2018-11-05 stsp const struct got_error *
389 7762fe12 2018-11-05 stsp got_object_open_mini_commit(struct got_commit_object_mini **commit,
390 7762fe12 2018-11-05 stsp struct got_repository *repo, struct got_object_id *id)
391 7762fe12 2018-11-05 stsp {
392 7762fe12 2018-11-05 stsp const struct got_error *err;
393 7762fe12 2018-11-05 stsp struct got_object *obj;
394 7762fe12 2018-11-05 stsp
395 7762fe12 2018-11-05 stsp err = got_object_open(&obj, repo, id);
396 7762fe12 2018-11-05 stsp if (err)
397 7762fe12 2018-11-05 stsp return err;
398 7762fe12 2018-11-05 stsp if (got_object_get_type(obj) != GOT_OBJ_TYPE_COMMIT) {
399 7762fe12 2018-11-05 stsp err = got_error(GOT_ERR_OBJ_TYPE);
400 7762fe12 2018-11-05 stsp goto done;
401 7762fe12 2018-11-05 stsp }
402 7762fe12 2018-11-05 stsp
403 7762fe12 2018-11-05 stsp err = open_mini_commit(commit, repo, obj);
404 7762fe12 2018-11-05 stsp done:
405 7762fe12 2018-11-05 stsp got_object_close(obj);
406 7762fe12 2018-11-05 stsp return err;
407 be6a1b5a 2018-06-11 stsp }
408 be6a1b5a 2018-06-11 stsp
409 be6a1b5a 2018-06-11 stsp const struct got_error *
410 dbc6a6b6 2018-07-12 stsp got_object_qid_alloc(struct got_object_qid **qid, struct got_object_id *id)
411 dbc6a6b6 2018-07-12 stsp {
412 dbc6a6b6 2018-07-12 stsp const struct got_error *err = NULL;
413 dbc6a6b6 2018-07-12 stsp
414 dbc6a6b6 2018-07-12 stsp *qid = calloc(1, sizeof(**qid));
415 dbc6a6b6 2018-07-12 stsp if (*qid == NULL)
416 dbc6a6b6 2018-07-12 stsp return got_error_from_errno();
417 dbc6a6b6 2018-07-12 stsp
418 dbc6a6b6 2018-07-12 stsp (*qid)->id = got_object_id_dup(id);
419 dbc6a6b6 2018-07-12 stsp if ((*qid)->id == NULL) {
420 dbc6a6b6 2018-07-12 stsp err = got_error_from_errno();
421 fa2f6902 2018-07-23 stsp got_object_qid_free(*qid);
422 dbc6a6b6 2018-07-12 stsp *qid = NULL;
423 dbc6a6b6 2018-07-12 stsp return err;
424 dbc6a6b6 2018-07-12 stsp }
425 dbc6a6b6 2018-07-12 stsp
426 dbc6a6b6 2018-07-12 stsp return NULL;
427 7e212e3d 2018-09-09 stsp }
428 7e212e3d 2018-09-09 stsp
429 71eb0e7f 2018-09-16 stsp static const struct got_error *
430 71eb0e7f 2018-09-16 stsp open_tree(struct got_tree_object **tree,
431 71eb0e7f 2018-09-16 stsp struct got_repository *repo, struct got_object *obj, int check_cache)
432 0ffeb3c2 2017-11-26 stsp {
433 0ffeb3c2 2017-11-26 stsp const struct got_error *err = NULL;
434 f6be5c30 2018-06-22 stsp
435 71eb0e7f 2018-09-16 stsp if (check_cache) {
436 71eb0e7f 2018-09-16 stsp *tree = got_repo_get_cached_tree(repo, &obj->id);
437 71eb0e7f 2018-09-16 stsp if (*tree != NULL) {
438 71eb0e7f 2018-09-16 stsp (*tree)->refcnt++;
439 71eb0e7f 2018-09-16 stsp return NULL;
440 71eb0e7f 2018-09-16 stsp }
441 71eb0e7f 2018-09-16 stsp } else
442 71eb0e7f 2018-09-16 stsp *tree = NULL;
443 0ffeb3c2 2017-11-26 stsp
444 0ffeb3c2 2017-11-26 stsp if (obj->type != GOT_OBJ_TYPE_TREE)
445 0ffeb3c2 2017-11-26 stsp return got_error(GOT_ERR_OBJ_TYPE);
446 0ffeb3c2 2017-11-26 stsp
447 e0ab43e7 2018-03-16 stsp if (obj->flags & GOT_OBJ_FLAG_PACKED) {
448 e7885405 2018-09-10 stsp struct got_pack *pack;
449 e7885405 2018-09-10 stsp pack = got_repo_get_cached_pack(repo, obj->path_packfile);
450 e7885405 2018-09-10 stsp if (pack == NULL) {
451 e7885405 2018-09-10 stsp err = got_repo_cache_pack(&pack, repo,
452 e7885405 2018-09-10 stsp obj->path_packfile, NULL);
453 e7885405 2018-09-10 stsp if (err)
454 e7885405 2018-09-10 stsp return err;
455 e7885405 2018-09-10 stsp }
456 e7885405 2018-09-10 stsp err = got_object_read_packed_tree_privsep(tree, obj, pack);
457 e0ab43e7 2018-03-16 stsp } else {
458 d5003b79 2018-04-22 stsp int fd;
459 d5003b79 2018-04-22 stsp err = open_loose_object(&fd, obj, repo);
460 e0ab43e7 2018-03-16 stsp if (err)
461 e0ab43e7 2018-03-16 stsp return err;
462 ad242220 2018-09-08 stsp err = got_object_read_tree_privsep(tree, obj, fd, repo);
463 d5003b79 2018-04-22 stsp close(fd);
464 776d4d29 2018-06-17 stsp }
465 f6be5c30 2018-06-22 stsp
466 f6be5c30 2018-06-22 stsp if (err == NULL) {
467 f6be5c30 2018-06-22 stsp (*tree)->refcnt++;
468 f6be5c30 2018-06-22 stsp err = got_repo_cache_tree(repo, &obj->id, *tree);
469 f6be5c30 2018-06-22 stsp }
470 f6be5c30 2018-06-22 stsp
471 776d4d29 2018-06-17 stsp return err;
472 776d4d29 2018-06-17 stsp }
473 776d4d29 2018-06-17 stsp
474 776d4d29 2018-06-17 stsp const struct got_error *
475 776d4d29 2018-06-17 stsp got_object_open_as_tree(struct got_tree_object **tree,
476 776d4d29 2018-06-17 stsp struct got_repository *repo, struct got_object_id *id)
477 776d4d29 2018-06-17 stsp {
478 776d4d29 2018-06-17 stsp const struct got_error *err;
479 776d4d29 2018-06-17 stsp struct got_object *obj;
480 835e0dbd 2018-06-21 stsp
481 e8eb494a 2018-09-16 stsp *tree = got_repo_get_cached_tree(repo, id);
482 e8eb494a 2018-09-16 stsp if (*tree != NULL) {
483 e8eb494a 2018-09-16 stsp (*tree)->refcnt++;
484 e8eb494a 2018-09-16 stsp return NULL;
485 e8eb494a 2018-09-16 stsp }
486 776d4d29 2018-06-17 stsp
487 776d4d29 2018-06-17 stsp err = got_object_open(&obj, repo, id);
488 776d4d29 2018-06-17 stsp if (err)
489 776d4d29 2018-06-17 stsp return err;
490 1cbc02b6 2018-06-21 stsp if (got_object_get_type(obj) != GOT_OBJ_TYPE_TREE) {
491 776d4d29 2018-06-17 stsp err = got_error(GOT_ERR_OBJ_TYPE);
492 776d4d29 2018-06-17 stsp goto done;
493 e0ab43e7 2018-03-16 stsp }
494 776d4d29 2018-06-17 stsp
495 71eb0e7f 2018-09-16 stsp err = open_tree(tree, repo, obj, 0);
496 776d4d29 2018-06-17 stsp done:
497 776d4d29 2018-06-17 stsp got_object_close(obj);
498 0ffeb3c2 2017-11-26 stsp return err;
499 57efb1af 2018-04-24 stsp }
500 ff6b18f8 2018-04-24 stsp
501 71eb0e7f 2018-09-16 stsp const struct got_error *
502 71eb0e7f 2018-09-16 stsp got_object_tree_open(struct got_tree_object **tree,
503 71eb0e7f 2018-09-16 stsp struct got_repository *repo, struct got_object *obj)
504 71eb0e7f 2018-09-16 stsp {
505 71eb0e7f 2018-09-16 stsp return open_tree(tree, repo, obj, 1);
506 71eb0e7f 2018-09-16 stsp }
507 71eb0e7f 2018-09-16 stsp
508 883f0469 2018-06-23 stsp const struct got_tree_entries *
509 883f0469 2018-06-23 stsp got_object_tree_get_entries(struct got_tree_object *tree)
510 883f0469 2018-06-23 stsp {
511 883f0469 2018-06-23 stsp return &tree->entries;
512 883f0469 2018-06-23 stsp }
513 3840f4c9 2018-09-12 stsp
514 3840f4c9 2018-09-12 stsp static const struct got_error *
515 3840f4c9 2018-09-12 stsp read_packed_blob_privsep(size_t *size, int outfd, struct got_object *obj,
516 3840f4c9 2018-09-12 stsp struct got_pack *pack)
517 3840f4c9 2018-09-12 stsp {
518 3840f4c9 2018-09-12 stsp const struct got_error *err = NULL;
519 3840f4c9 2018-09-12 stsp int outfd_child;
520 3840f4c9 2018-09-12 stsp int basefd, accumfd; /* temporary files for delta application */
521 24140570 2018-09-09 stsp
522 3840f4c9 2018-09-12 stsp basefd = got_opentempfd();
523 3840f4c9 2018-09-12 stsp if (basefd == -1)
524 3840f4c9 2018-09-12 stsp return got_error_from_errno();
525 3840f4c9 2018-09-12 stsp accumfd = got_opentempfd();
526 3840f4c9 2018-09-12 stsp if (accumfd == -1)
527 3840f4c9 2018-09-12 stsp return got_error_from_errno();
528 3840f4c9 2018-09-12 stsp
529 3840f4c9 2018-09-12 stsp outfd_child = dup(outfd);
530 3840f4c9 2018-09-12 stsp if (outfd_child == -1)
531 3840f4c9 2018-09-12 stsp return got_error_from_errno();
532 3840f4c9 2018-09-12 stsp
533 3840f4c9 2018-09-12 stsp err = got_privsep_send_obj_req(pack->privsep_child->ibuf, -1, obj);
534 3840f4c9 2018-09-12 stsp if (err)
535 3840f4c9 2018-09-12 stsp return err;
536 3840f4c9 2018-09-12 stsp
537 3840f4c9 2018-09-12 stsp err = got_privsep_send_blob_outfd(pack->privsep_child->ibuf,
538 3840f4c9 2018-09-12 stsp outfd_child);
539 3840f4c9 2018-09-12 stsp if (err) {
540 3840f4c9 2018-09-12 stsp close(outfd_child);
541 3840f4c9 2018-09-12 stsp return err;
542 3840f4c9 2018-09-12 stsp }
543 3840f4c9 2018-09-12 stsp err = got_privsep_send_tmpfd(pack->privsep_child->ibuf,
544 3840f4c9 2018-09-12 stsp basefd);
545 3840f4c9 2018-09-12 stsp if (err) {
546 3840f4c9 2018-09-12 stsp close(basefd);
547 3840f4c9 2018-09-12 stsp close(accumfd);
548 3840f4c9 2018-09-12 stsp close(outfd_child);
549 3840f4c9 2018-09-12 stsp return err;
550 3840f4c9 2018-09-12 stsp }
551 3840f4c9 2018-09-12 stsp
552 3840f4c9 2018-09-12 stsp err = got_privsep_send_tmpfd(pack->privsep_child->ibuf,
553 3840f4c9 2018-09-12 stsp accumfd);
554 3840f4c9 2018-09-12 stsp if (err) {
555 3840f4c9 2018-09-12 stsp close(accumfd);
556 3840f4c9 2018-09-12 stsp close(outfd_child);
557 3840f4c9 2018-09-12 stsp return err;
558 3840f4c9 2018-09-12 stsp }
559 3840f4c9 2018-09-12 stsp
560 3840f4c9 2018-09-12 stsp err = got_privsep_recv_blob(size, pack->privsep_child->ibuf);
561 3840f4c9 2018-09-12 stsp if (err)
562 3840f4c9 2018-09-12 stsp return err;
563 3840f4c9 2018-09-12 stsp
564 3840f4c9 2018-09-12 stsp if (lseek(outfd, SEEK_SET, 0) == -1)
565 3840f4c9 2018-09-12 stsp err = got_error_from_errno();
566 3840f4c9 2018-09-12 stsp
567 3840f4c9 2018-09-12 stsp return err;
568 3840f4c9 2018-09-12 stsp }
569 3840f4c9 2018-09-12 stsp
570 68482ea3 2017-11-27 stsp const struct got_error *
571 68482ea3 2017-11-27 stsp got_object_blob_open(struct got_blob_object **blob,
572 68482ea3 2017-11-27 stsp struct got_repository *repo, struct got_object *obj, size_t blocksize)
573 68482ea3 2017-11-27 stsp {
574 68482ea3 2017-11-27 stsp const struct got_error *err = NULL;
575 55da3778 2018-09-10 stsp int outfd;
576 55da3778 2018-09-10 stsp size_t size;
577 55da3778 2018-09-10 stsp struct stat sb;
578 68482ea3 2017-11-27 stsp
579 68482ea3 2017-11-27 stsp if (obj->type != GOT_OBJ_TYPE_BLOB)
580 68482ea3 2017-11-27 stsp return got_error(GOT_ERR_OBJ_TYPE);
581 68482ea3 2017-11-27 stsp
582 7d283eee 2017-11-29 stsp if (blocksize < obj->hdrlen)
583 7d283eee 2017-11-29 stsp return got_error(GOT_ERR_NO_SPACE);
584 7d283eee 2017-11-29 stsp
585 68482ea3 2017-11-27 stsp *blob = calloc(1, sizeof(**blob));
586 4558fcd4 2018-01-14 stsp if (*blob == NULL)
587 0a585a0d 2018-03-17 stsp return got_error_from_errno();
588 68482ea3 2017-11-27 stsp
589 55da3778 2018-09-10 stsp outfd = got_opentempfd();
590 55da3778 2018-09-10 stsp if (outfd == -1)
591 55da3778 2018-09-10 stsp return got_error_from_errno();
592 55da3778 2018-09-10 stsp
593 062ebb78 2018-07-12 stsp (*blob)->read_buf = malloc(blocksize);
594 15c8b0e6 2018-04-24 stsp if ((*blob)->read_buf == NULL) {
595 15c8b0e6 2018-04-24 stsp err = got_error_from_errno();
596 c7254d79 2018-04-24 stsp goto done;
597 15c8b0e6 2018-04-24 stsp }
598 eb651edf 2018-02-11 stsp if (obj->flags & GOT_OBJ_FLAG_PACKED) {
599 55da3778 2018-09-10 stsp struct got_pack *pack;
600 55da3778 2018-09-10 stsp pack = got_repo_get_cached_pack(repo, obj->path_packfile);
601 55da3778 2018-09-10 stsp if (pack == NULL) {
602 55da3778 2018-09-10 stsp err = got_repo_cache_pack(&pack, repo,
603 55da3778 2018-09-10 stsp obj->path_packfile, NULL);
604 55da3778 2018-09-10 stsp if (err)
605 55da3778 2018-09-10 stsp goto done;
606 55da3778 2018-09-10 stsp }
607 3840f4c9 2018-09-12 stsp err = read_packed_blob_privsep(&size, outfd, obj, pack);
608 c7254d79 2018-04-24 stsp if (err)
609 c7254d79 2018-04-24 stsp goto done;
610 55da3778 2018-09-10 stsp obj->size = size;
611 eb651edf 2018-02-11 stsp } else {
612 55da3778 2018-09-10 stsp int infd;
613 c7254d79 2018-04-24 stsp
614 3aca5731 2018-04-24 stsp err = open_loose_object(&infd, obj, repo);
615 c7254d79 2018-04-24 stsp if (err)
616 c7254d79 2018-04-24 stsp goto done;
617 c7254d79 2018-04-24 stsp
618 ad242220 2018-09-08 stsp err = got_object_read_blob_privsep(&size, outfd, infd, repo);
619 3aca5731 2018-04-24 stsp close(infd);
620 57efb1af 2018-04-24 stsp if (err)
621 c7254d79 2018-04-24 stsp goto done;
622 3aca5731 2018-04-24 stsp
623 2967a784 2018-04-24 stsp if (size != obj->hdrlen + obj->size) {
624 1a6b3ab7 2018-04-24 stsp err = got_error(GOT_ERR_PRIVSEP_LEN);
625 2967a784 2018-04-24 stsp goto done;
626 2967a784 2018-04-24 stsp }
627 55da3778 2018-09-10 stsp }
628 2967a784 2018-04-24 stsp
629 55da3778 2018-09-10 stsp if (fstat(outfd, &sb) == -1) {
630 55da3778 2018-09-10 stsp err = got_error_from_errno();
631 55da3778 2018-09-10 stsp goto done;
632 55da3778 2018-09-10 stsp }
633 2967a784 2018-04-24 stsp
634 55da3778 2018-09-10 stsp if (sb.st_size != obj->hdrlen + obj->size) {
635 55da3778 2018-09-10 stsp err = got_error(GOT_ERR_PRIVSEP_LEN);
636 55da3778 2018-09-10 stsp goto done;
637 55da3778 2018-09-10 stsp }
638 2967a784 2018-04-24 stsp
639 55da3778 2018-09-10 stsp (*blob)->f = fdopen(outfd, "rb");
640 55da3778 2018-09-10 stsp if ((*blob)->f == NULL) {
641 55da3778 2018-09-10 stsp err = got_error_from_errno();
642 55da3778 2018-09-10 stsp close(outfd);
643 55da3778 2018-09-10 stsp goto done;
644 68482ea3 2017-11-27 stsp }
645 68482ea3 2017-11-27 stsp
646 7d283eee 2017-11-29 stsp (*blob)->hdrlen = obj->hdrlen;
647 eb651edf 2018-02-11 stsp (*blob)->blocksize = blocksize;
648 f78b0693 2017-11-29 stsp memcpy(&(*blob)->id.sha1, obj->id.sha1, SHA1_DIGEST_LENGTH);
649 7d283eee 2017-11-29 stsp
650 c7254d79 2018-04-24 stsp done:
651 55da3778 2018-09-10 stsp if (err) {
652 55da3778 2018-09-10 stsp if (*blob) {
653 55da3778 2018-09-10 stsp if ((*blob)->f)
654 55da3778 2018-09-10 stsp fclose((*blob)->f);
655 55da3778 2018-09-10 stsp free((*blob)->read_buf);
656 55da3778 2018-09-10 stsp free(*blob);
657 55da3778 2018-09-10 stsp *blob = NULL;
658 55da3778 2018-09-10 stsp } else if (outfd != -1)
659 55da3778 2018-09-10 stsp close(outfd);
660 a19581a2 2018-06-21 stsp }
661 a19581a2 2018-06-21 stsp return err;
662 a19581a2 2018-06-21 stsp }
663 a19581a2 2018-06-21 stsp
664 a19581a2 2018-06-21 stsp const struct got_error *
665 a19581a2 2018-06-21 stsp got_object_open_as_blob(struct got_blob_object **blob,
666 a19581a2 2018-06-21 stsp struct got_repository *repo, struct got_object_id *id,
667 a19581a2 2018-06-21 stsp size_t blocksize)
668 a19581a2 2018-06-21 stsp {
669 a19581a2 2018-06-21 stsp const struct got_error *err;
670 a19581a2 2018-06-21 stsp struct got_object *obj;
671 835e0dbd 2018-06-21 stsp
672 835e0dbd 2018-06-21 stsp *blob = NULL;
673 a19581a2 2018-06-21 stsp
674 a19581a2 2018-06-21 stsp err = got_object_open(&obj, repo, id);
675 a19581a2 2018-06-21 stsp if (err)
676 a19581a2 2018-06-21 stsp return err;
677 a19581a2 2018-06-21 stsp if (got_object_get_type(obj) != GOT_OBJ_TYPE_BLOB) {
678 a19581a2 2018-06-21 stsp err = got_error(GOT_ERR_OBJ_TYPE);
679 a19581a2 2018-06-21 stsp goto done;
680 c7254d79 2018-04-24 stsp }
681 a19581a2 2018-06-21 stsp
682 a19581a2 2018-06-21 stsp err = got_object_blob_open(blob, repo, obj, blocksize);
683 a19581a2 2018-06-21 stsp done:
684 a19581a2 2018-06-21 stsp got_object_close(obj);
685 68482ea3 2017-11-27 stsp return err;
686 0ffeb3c2 2017-11-26 stsp }
687 68482ea3 2017-11-27 stsp
688 68482ea3 2017-11-27 stsp void
689 68482ea3 2017-11-27 stsp got_object_blob_close(struct got_blob_object *blob)
690 68482ea3 2017-11-27 stsp {
691 15c8b0e6 2018-04-24 stsp free(blob->read_buf);
692 68482ea3 2017-11-27 stsp fclose(blob->f);
693 68482ea3 2017-11-27 stsp free(blob);
694 f934cf2c 2018-02-12 stsp }
695 f934cf2c 2018-02-12 stsp
696 f934cf2c 2018-02-12 stsp char *
697 f934cf2c 2018-02-12 stsp got_object_blob_id_str(struct got_blob_object *blob, char *buf, size_t size)
698 f934cf2c 2018-02-12 stsp {
699 f934cf2c 2018-02-12 stsp return got_sha1_digest_to_str(blob->id.sha1, buf, size);
700 f934cf2c 2018-02-12 stsp }
701 f934cf2c 2018-02-12 stsp
702 f934cf2c 2018-02-12 stsp size_t
703 f934cf2c 2018-02-12 stsp got_object_blob_get_hdrlen(struct got_blob_object *blob)
704 f934cf2c 2018-02-12 stsp {
705 f934cf2c 2018-02-12 stsp return blob->hdrlen;
706 68482ea3 2017-11-27 stsp }
707 68482ea3 2017-11-27 stsp
708 f934cf2c 2018-02-12 stsp const uint8_t *
709 f934cf2c 2018-02-12 stsp got_object_blob_get_read_buf(struct got_blob_object *blob)
710 f934cf2c 2018-02-12 stsp {
711 f934cf2c 2018-02-12 stsp return blob->read_buf;
712 f934cf2c 2018-02-12 stsp }
713 f934cf2c 2018-02-12 stsp
714 68482ea3 2017-11-27 stsp const struct got_error *
715 eb651edf 2018-02-11 stsp got_object_blob_read_block(size_t *outlenp, struct got_blob_object *blob)
716 68482ea3 2017-11-27 stsp {
717 eb651edf 2018-02-11 stsp size_t n;
718 eb651edf 2018-02-11 stsp
719 eb651edf 2018-02-11 stsp n = fread(blob->read_buf, 1, blob->blocksize, blob->f);
720 eb651edf 2018-02-11 stsp if (n == 0 && ferror(blob->f))
721 eb651edf 2018-02-11 stsp return got_ferror(blob->f, GOT_ERR_IO);
722 eb651edf 2018-02-11 stsp *outlenp = n;
723 35e9ba5d 2018-06-21 stsp return NULL;
724 35e9ba5d 2018-06-21 stsp }
725 35e9ba5d 2018-06-21 stsp
726 35e9ba5d 2018-06-21 stsp const struct got_error *
727 84451b3e 2018-07-10 stsp got_object_blob_dump_to_file(size_t *total_len, size_t *nlines,
728 84451b3e 2018-07-10 stsp FILE *outfile, struct got_blob_object *blob)
729 35e9ba5d 2018-06-21 stsp {
730 35e9ba5d 2018-06-21 stsp const struct got_error *err = NULL;
731 35e9ba5d 2018-06-21 stsp size_t len, hdrlen;
732 84451b3e 2018-07-10 stsp const uint8_t *buf;
733 84451b3e 2018-07-10 stsp int i;
734 84451b3e 2018-07-10 stsp
735 84451b3e 2018-07-10 stsp if (total_len)
736 84451b3e 2018-07-10 stsp *total_len = 0;
737 84451b3e 2018-07-10 stsp if (nlines)
738 84451b3e 2018-07-10 stsp *nlines = 0;
739 35e9ba5d 2018-06-21 stsp
740 35e9ba5d 2018-06-21 stsp hdrlen = got_object_blob_get_hdrlen(blob);
741 35e9ba5d 2018-06-21 stsp do {
742 35e9ba5d 2018-06-21 stsp err = got_object_blob_read_block(&len, blob);
743 35e9ba5d 2018-06-21 stsp if (err)
744 35e9ba5d 2018-06-21 stsp return err;
745 35e9ba5d 2018-06-21 stsp if (len == 0)
746 35e9ba5d 2018-06-21 stsp break;
747 84451b3e 2018-07-10 stsp if (total_len)
748 84451b3e 2018-07-10 stsp *total_len += len;
749 84451b3e 2018-07-10 stsp buf = got_object_blob_get_read_buf(blob);
750 84451b3e 2018-07-10 stsp if (nlines) {
751 84451b3e 2018-07-10 stsp for (i = 0; i < len; i++) {
752 84451b3e 2018-07-10 stsp if (buf[i] == '\n')
753 84451b3e 2018-07-10 stsp (*nlines)++;
754 84451b3e 2018-07-10 stsp }
755 84451b3e 2018-07-10 stsp }
756 35e9ba5d 2018-06-21 stsp /* Skip blob object header first time around. */
757 84451b3e 2018-07-10 stsp fwrite(buf + hdrlen, len - hdrlen, 1, outfile);
758 35e9ba5d 2018-06-21 stsp hdrlen = 0;
759 35e9ba5d 2018-06-21 stsp } while (len != 0);
760 35e9ba5d 2018-06-21 stsp
761 35e9ba5d 2018-06-21 stsp fflush(outfile);
762 35e9ba5d 2018-06-21 stsp rewind(outfile);
763 35e9ba5d 2018-06-21 stsp
764 776d4d29 2018-06-17 stsp return NULL;
765 776d4d29 2018-06-17 stsp }
766 776d4d29 2018-06-17 stsp
767 776d4d29 2018-06-17 stsp static struct got_tree_entry *
768 65a9bbe9 2018-09-15 stsp find_entry_by_name(struct got_tree_object *tree, const char *name, size_t len)
769 776d4d29 2018-06-17 stsp {
770 776d4d29 2018-06-17 stsp struct got_tree_entry *te;
771 776d4d29 2018-06-17 stsp
772 883f0469 2018-06-23 stsp SIMPLEQ_FOREACH(te, &tree->entries.head, entry) {
773 2673a8da 2018-09-19 stsp if (strncmp(te->name, name, len) == 0 && te->name[len] == '\0')
774 776d4d29 2018-06-17 stsp return te;
775 776d4d29 2018-06-17 stsp }
776 eb651edf 2018-02-11 stsp return NULL;
777 776d4d29 2018-06-17 stsp }
778 776d4d29 2018-06-17 stsp
779 776d4d29 2018-06-17 stsp const struct got_error *
780 27d434c2 2018-09-15 stsp got_object_id_by_path(struct got_object_id **id, struct got_repository *repo,
781 776d4d29 2018-06-17 stsp struct got_object_id *commit_id, const char *path)
782 776d4d29 2018-06-17 stsp {
783 776d4d29 2018-06-17 stsp const struct got_error *err = NULL;
784 776d4d29 2018-06-17 stsp struct got_commit_object *commit = NULL;
785 776d4d29 2018-06-17 stsp struct got_tree_object *tree = NULL;
786 db37e2c0 2018-06-21 stsp struct got_tree_entry *te = NULL;
787 65a9bbe9 2018-09-15 stsp const char *seg, *s;
788 65a9bbe9 2018-09-15 stsp size_t seglen, len = strlen(path);
789 776d4d29 2018-06-17 stsp
790 27d434c2 2018-09-15 stsp *id = NULL;
791 776d4d29 2018-06-17 stsp
792 776d4d29 2018-06-17 stsp /* We are expecting an absolute in-repository path. */
793 776d4d29 2018-06-17 stsp if (path[0] != '/')
794 776d4d29 2018-06-17 stsp return got_error(GOT_ERR_NOT_ABSPATH);
795 776d4d29 2018-06-17 stsp
796 776d4d29 2018-06-17 stsp err = got_object_open_as_commit(&commit, repo, commit_id);
797 776d4d29 2018-06-17 stsp if (err)
798 776d4d29 2018-06-17 stsp goto done;
799 776d4d29 2018-06-17 stsp
800 776d4d29 2018-06-17 stsp /* Handle opening of root of commit's tree. */
801 776d4d29 2018-06-17 stsp if (path[1] == '\0') {
802 27d434c2 2018-09-15 stsp *id = got_object_id_dup(commit->tree_id);
803 27d434c2 2018-09-15 stsp if (*id == NULL)
804 27d434c2 2018-09-15 stsp err = got_error_from_errno();
805 ca008b32 2018-07-22 stsp goto done;
806 776d4d29 2018-06-17 stsp }
807 776d4d29 2018-06-17 stsp
808 776d4d29 2018-06-17 stsp err = got_object_open_as_tree(&tree, repo, commit->tree_id);
809 776d4d29 2018-06-17 stsp if (err)
810 776d4d29 2018-06-17 stsp goto done;
811 776d4d29 2018-06-17 stsp
812 65a9bbe9 2018-09-15 stsp s = path;
813 776d4d29 2018-06-17 stsp s++; /* skip leading '/' */
814 67606321 2018-06-21 stsp len--;
815 776d4d29 2018-06-17 stsp seg = s;
816 65a9bbe9 2018-09-15 stsp seglen = 0;
817 67606321 2018-06-21 stsp while (len > 0) {
818 776d4d29 2018-06-17 stsp struct got_tree_object *next_tree;
819 776d4d29 2018-06-17 stsp
820 776d4d29 2018-06-17 stsp if (*s != '/') {
821 776d4d29 2018-06-17 stsp s++;
822 67606321 2018-06-21 stsp len--;
823 65a9bbe9 2018-09-15 stsp seglen++;
824 00530cfb 2018-06-21 stsp if (*s)
825 00530cfb 2018-06-21 stsp continue;
826 776d4d29 2018-06-17 stsp }
827 776d4d29 2018-06-17 stsp
828 65a9bbe9 2018-09-15 stsp te = find_entry_by_name(tree, seg, seglen);
829 db37e2c0 2018-06-21 stsp if (te == NULL) {
830 776d4d29 2018-06-17 stsp err = got_error(GOT_ERR_NO_OBJ);
831 776d4d29 2018-06-17 stsp goto done;
832 776d4d29 2018-06-17 stsp }
833 776d4d29 2018-06-17 stsp
834 67606321 2018-06-21 stsp if (len == 0)
835 67606321 2018-06-21 stsp break;
836 67606321 2018-06-21 stsp
837 776d4d29 2018-06-17 stsp seg = s + 1;
838 65a9bbe9 2018-09-15 stsp seglen = 0;
839 776d4d29 2018-06-17 stsp s++;
840 67606321 2018-06-21 stsp len--;
841 776d4d29 2018-06-17 stsp if (*s) {
842 776d4d29 2018-06-17 stsp err = got_object_open_as_tree(&next_tree, repo,
843 db37e2c0 2018-06-21 stsp te->id);
844 db37e2c0 2018-06-21 stsp te = NULL;
845 776d4d29 2018-06-17 stsp if (err)
846 776d4d29 2018-06-17 stsp goto done;
847 776d4d29 2018-06-17 stsp got_object_tree_close(tree);
848 776d4d29 2018-06-17 stsp tree = next_tree;
849 776d4d29 2018-06-17 stsp }
850 776d4d29 2018-06-17 stsp }
851 776d4d29 2018-06-17 stsp
852 27d434c2 2018-09-15 stsp if (te) {
853 27d434c2 2018-09-15 stsp *id = got_object_id_dup(te->id);
854 27d434c2 2018-09-15 stsp if (*id == NULL)
855 27d434c2 2018-09-15 stsp return got_error_from_errno();
856 27d434c2 2018-09-15 stsp } else
857 776d4d29 2018-06-17 stsp err = got_error(GOT_ERR_NO_OBJ);
858 776d4d29 2018-06-17 stsp done:
859 776d4d29 2018-06-17 stsp if (commit)
860 776d4d29 2018-06-17 stsp got_object_commit_close(commit);
861 776d4d29 2018-06-17 stsp if (tree)
862 776d4d29 2018-06-17 stsp got_object_tree_close(tree);
863 776d4d29 2018-06-17 stsp return err;
864 68482ea3 2017-11-27 stsp }
865 07862c20 2018-09-15 stsp
866 07862c20 2018-09-15 stsp const struct got_error *
867 07862c20 2018-09-15 stsp got_object_tree_path_changed(int *changed,
868 07862c20 2018-09-15 stsp struct got_tree_object *tree01, struct got_tree_object *tree02,
869 07862c20 2018-09-15 stsp const char *path, struct got_repository *repo)
870 07862c20 2018-09-15 stsp {
871 07862c20 2018-09-15 stsp const struct got_error *err = NULL;
872 07862c20 2018-09-15 stsp struct got_tree_object *tree1 = NULL, *tree2 = NULL;
873 07862c20 2018-09-15 stsp struct got_tree_entry *te1 = NULL, *te2 = NULL;
874 65a9bbe9 2018-09-15 stsp const char *seg, *s;
875 65a9bbe9 2018-09-15 stsp size_t seglen, len = strlen(path);
876 07862c20 2018-09-15 stsp
877 07862c20 2018-09-15 stsp *changed = 0;
878 07862c20 2018-09-15 stsp
879 07862c20 2018-09-15 stsp /* We are expecting an absolute in-repository path. */
880 07862c20 2018-09-15 stsp if (path[0] != '/')
881 07862c20 2018-09-15 stsp return got_error(GOT_ERR_NOT_ABSPATH);
882 07862c20 2018-09-15 stsp
883 07862c20 2018-09-15 stsp /* We not do support comparing the root path. */
884 07862c20 2018-09-15 stsp if (path[1] == '\0')
885 07862c20 2018-09-15 stsp return got_error(GOT_ERR_BAD_PATH);
886 07862c20 2018-09-15 stsp
887 07862c20 2018-09-15 stsp tree1 = tree01;
888 07862c20 2018-09-15 stsp tree2 = tree02;
889 65a9bbe9 2018-09-15 stsp s = path;
890 07862c20 2018-09-15 stsp s++; /* skip leading '/' */
891 07862c20 2018-09-15 stsp len--;
892 07862c20 2018-09-15 stsp seg = s;
893 65a9bbe9 2018-09-15 stsp seglen = 0;
894 07862c20 2018-09-15 stsp while (len > 0) {
895 07862c20 2018-09-15 stsp struct got_tree_object *next_tree1, *next_tree2;
896 07862c20 2018-09-15 stsp
897 07862c20 2018-09-15 stsp if (*s != '/') {
898 07862c20 2018-09-15 stsp s++;
899 07862c20 2018-09-15 stsp len--;
900 65a9bbe9 2018-09-15 stsp seglen++;
901 07862c20 2018-09-15 stsp if (*s)
902 07862c20 2018-09-15 stsp continue;
903 07862c20 2018-09-15 stsp }
904 07862c20 2018-09-15 stsp
905 65a9bbe9 2018-09-15 stsp te1 = find_entry_by_name(tree1, seg, seglen);
906 07862c20 2018-09-15 stsp if (te1 == NULL) {
907 07862c20 2018-09-15 stsp err = got_error(GOT_ERR_NO_OBJ);
908 07862c20 2018-09-15 stsp goto done;
909 07862c20 2018-09-15 stsp }
910 07862c20 2018-09-15 stsp
911 65a9bbe9 2018-09-15 stsp te2 = find_entry_by_name(tree2, seg, seglen);
912 07862c20 2018-09-15 stsp if (te2 == NULL) {
913 07862c20 2018-09-15 stsp *changed = 1;
914 07862c20 2018-09-15 stsp goto done;
915 07862c20 2018-09-15 stsp }
916 07862c20 2018-09-15 stsp
917 07862c20 2018-09-15 stsp if (te1->mode != te2->mode) {
918 07862c20 2018-09-15 stsp *changed = 1;
919 07862c20 2018-09-15 stsp goto done;
920 07862c20 2018-09-15 stsp }
921 07862c20 2018-09-15 stsp
922 07862c20 2018-09-15 stsp if (got_object_id_cmp(te1->id, te2->id) == 0) {
923 07862c20 2018-09-15 stsp *changed = 0;
924 07862c20 2018-09-15 stsp goto done;
925 07862c20 2018-09-15 stsp }
926 07862c20 2018-09-15 stsp
927 f970fa8a 2018-09-15 stsp if (len == 0) { /* final path element */
928 07862c20 2018-09-15 stsp *changed = 1;
929 07862c20 2018-09-15 stsp goto done;
930 07862c20 2018-09-15 stsp }
931 07862c20 2018-09-15 stsp
932 07862c20 2018-09-15 stsp seg = s + 1;
933 07862c20 2018-09-15 stsp s++;
934 07862c20 2018-09-15 stsp len--;
935 65a9bbe9 2018-09-15 stsp seglen = 0;
936 07862c20 2018-09-15 stsp if (*s) {
937 07862c20 2018-09-15 stsp err = got_object_open_as_tree(&next_tree1, repo,
938 07862c20 2018-09-15 stsp te1->id);
939 07862c20 2018-09-15 stsp te1 = NULL;
940 07862c20 2018-09-15 stsp if (err)
941 07862c20 2018-09-15 stsp goto done;
942 a31cea73 2018-09-15 stsp if (tree1 != tree01)
943 a31cea73 2018-09-15 stsp got_object_tree_close(tree1);
944 07862c20 2018-09-15 stsp tree1 = next_tree1;
945 07862c20 2018-09-15 stsp
946 07862c20 2018-09-15 stsp err = got_object_open_as_tree(&next_tree2, repo,
947 07862c20 2018-09-15 stsp te2->id);
948 07862c20 2018-09-15 stsp te2 = NULL;
949 07862c20 2018-09-15 stsp if (err)
950 07862c20 2018-09-15 stsp goto done;
951 a31cea73 2018-09-15 stsp if (tree2 != tree02)
952 a31cea73 2018-09-15 stsp got_object_tree_close(tree2);
953 07862c20 2018-09-15 stsp tree2 = next_tree2;
954 07862c20 2018-09-15 stsp }
955 07862c20 2018-09-15 stsp }
956 07862c20 2018-09-15 stsp done:
957 a31cea73 2018-09-15 stsp if (tree1 && tree1 != tree01)
958 07862c20 2018-09-15 stsp got_object_tree_close(tree1);
959 a31cea73 2018-09-15 stsp if (tree2 && tree2 != tree02)
960 07862c20 2018-09-15 stsp got_object_tree_close(tree2);
961 77880158 2018-11-04 stsp return err;
962 77880158 2018-11-04 stsp }
963 77880158 2018-11-04 stsp
964 77880158 2018-11-04 stsp static void
965 77880158 2018-11-04 stsp exec_privsep_child(int imsg_fds[2], const char *path, const char *repo_path)
966 77880158 2018-11-04 stsp {
967 77880158 2018-11-04 stsp close(imsg_fds[0]);
968 77880158 2018-11-04 stsp
969 77880158 2018-11-04 stsp if (dup2(imsg_fds[1], GOT_IMSG_FD_CHILD) == -1) {
970 77880158 2018-11-04 stsp fprintf(stderr, "%s: %s\n", getprogname(),
971 77880158 2018-11-04 stsp strerror(errno));
972 77880158 2018-11-04 stsp _exit(1);
973 77880158 2018-11-04 stsp }
974 77880158 2018-11-04 stsp if (closefrom(GOT_IMSG_FD_CHILD + 1) == -1) {
975 77880158 2018-11-04 stsp fprintf(stderr, "%s: %s\n", getprogname(),
976 77880158 2018-11-04 stsp strerror(errno));
977 77880158 2018-11-04 stsp _exit(1);
978 77880158 2018-11-04 stsp }
979 77880158 2018-11-04 stsp
980 77880158 2018-11-04 stsp if (execl(path, path, repo_path, (char *)NULL) == -1) {
981 77880158 2018-11-04 stsp fprintf(stderr, "%s: %s: %s\n", getprogname(), path,
982 77880158 2018-11-04 stsp strerror(errno));
983 77880158 2018-11-04 stsp _exit(1);
984 77880158 2018-11-04 stsp }
985 77880158 2018-11-04 stsp }
986 77880158 2018-11-04 stsp
987 77880158 2018-11-04 stsp static const struct got_error *
988 77880158 2018-11-04 stsp request_object(struct got_object **obj, struct got_repository *repo, int fd)
989 77880158 2018-11-04 stsp {
990 77880158 2018-11-04 stsp const struct got_error *err = NULL;
991 77880158 2018-11-04 stsp struct imsgbuf *ibuf;
992 77880158 2018-11-04 stsp
993 77880158 2018-11-04 stsp ibuf = repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_OBJECT].ibuf;
994 77880158 2018-11-04 stsp
995 77880158 2018-11-04 stsp err = got_privsep_send_obj_req(ibuf, fd, NULL);
996 77880158 2018-11-04 stsp if (err)
997 77880158 2018-11-04 stsp return err;
998 77880158 2018-11-04 stsp
999 77880158 2018-11-04 stsp return got_privsep_recv_obj(obj, ibuf);
1000 77880158 2018-11-04 stsp }
1001 77880158 2018-11-04 stsp
1002 77880158 2018-11-04 stsp const struct got_error *
1003 77880158 2018-11-04 stsp got_object_read_header_privsep(struct got_object **obj,
1004 77880158 2018-11-04 stsp struct got_repository *repo, int obj_fd)
1005 77880158 2018-11-04 stsp {
1006 77880158 2018-11-04 stsp int imsg_fds[2];
1007 77880158 2018-11-04 stsp pid_t pid;
1008 77880158 2018-11-04 stsp struct imsgbuf *ibuf;
1009 77880158 2018-11-04 stsp
1010 77880158 2018-11-04 stsp if (repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_OBJECT].imsg_fd != -1)
1011 77880158 2018-11-04 stsp return request_object(obj, repo, obj_fd);
1012 77880158 2018-11-04 stsp
1013 77880158 2018-11-04 stsp ibuf = calloc(1, sizeof(*ibuf));
1014 77880158 2018-11-04 stsp if (ibuf == NULL)
1015 77880158 2018-11-04 stsp return got_error_from_errno();
1016 77880158 2018-11-04 stsp
1017 77880158 2018-11-04 stsp if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, imsg_fds) == -1)
1018 77880158 2018-11-04 stsp return got_error_from_errno();
1019 77880158 2018-11-04 stsp
1020 77880158 2018-11-04 stsp pid = fork();
1021 77880158 2018-11-04 stsp if (pid == -1)
1022 77880158 2018-11-04 stsp return got_error_from_errno();
1023 77880158 2018-11-04 stsp else if (pid == 0) {
1024 77880158 2018-11-04 stsp exec_privsep_child(imsg_fds, GOT_PATH_PROG_READ_OBJECT,
1025 77880158 2018-11-04 stsp repo->path);
1026 77880158 2018-11-04 stsp /* not reached */
1027 77880158 2018-11-04 stsp }
1028 77880158 2018-11-04 stsp
1029 77880158 2018-11-04 stsp close(imsg_fds[1]);
1030 77880158 2018-11-04 stsp repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_OBJECT].imsg_fd =
1031 77880158 2018-11-04 stsp imsg_fds[0];
1032 77880158 2018-11-04 stsp repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_OBJECT].pid = pid;
1033 77880158 2018-11-04 stsp imsg_init(ibuf, imsg_fds[0]);
1034 77880158 2018-11-04 stsp repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_OBJECT].ibuf = ibuf;
1035 77880158 2018-11-04 stsp
1036 77880158 2018-11-04 stsp return request_object(obj, repo, obj_fd);
1037 77880158 2018-11-04 stsp }
1038 77880158 2018-11-04 stsp
1039 77880158 2018-11-04 stsp static const struct got_error *
1040 77880158 2018-11-04 stsp request_packed_object(struct got_object **obj, struct got_pack *pack, int idx,
1041 77880158 2018-11-04 stsp struct got_object_id *id)
1042 77880158 2018-11-04 stsp {
1043 77880158 2018-11-04 stsp const struct got_error *err = NULL;
1044 77880158 2018-11-04 stsp struct imsgbuf *ibuf = pack->privsep_child->ibuf;
1045 77880158 2018-11-04 stsp
1046 77880158 2018-11-04 stsp err = got_privsep_send_packed_obj_req(ibuf, idx, id);
1047 77880158 2018-11-04 stsp if (err)
1048 77880158 2018-11-04 stsp return err;
1049 77880158 2018-11-04 stsp
1050 77880158 2018-11-04 stsp err = got_privsep_recv_obj(obj, ibuf);
1051 77880158 2018-11-04 stsp if (err)
1052 77880158 2018-11-04 stsp return err;
1053 77880158 2018-11-04 stsp
1054 77880158 2018-11-04 stsp (*obj)->path_packfile = strdup(pack->path_packfile);
1055 77880158 2018-11-04 stsp if ((*obj)->path_packfile == NULL) {
1056 77880158 2018-11-04 stsp err = got_error_from_errno();
1057 77880158 2018-11-04 stsp return err;
1058 77880158 2018-11-04 stsp }
1059 77880158 2018-11-04 stsp memcpy(&(*obj)->id, id, sizeof((*obj)->id));
1060 77880158 2018-11-04 stsp
1061 77880158 2018-11-04 stsp return NULL;
1062 77880158 2018-11-04 stsp }
1063 77880158 2018-11-04 stsp
1064 77880158 2018-11-04 stsp const struct got_error *
1065 77880158 2018-11-04 stsp got_object_packed_read_privsep(struct got_object **obj,
1066 77880158 2018-11-04 stsp struct got_repository *repo, struct got_pack *pack,
1067 77880158 2018-11-04 stsp struct got_packidx *packidx, int idx, struct got_object_id *id)
1068 77880158 2018-11-04 stsp {
1069 77880158 2018-11-04 stsp const struct got_error *err = NULL;
1070 77880158 2018-11-04 stsp int imsg_fds[2];
1071 77880158 2018-11-04 stsp pid_t pid;
1072 77880158 2018-11-04 stsp struct imsgbuf *ibuf;
1073 77880158 2018-11-04 stsp
1074 77880158 2018-11-04 stsp if (pack->privsep_child)
1075 77880158 2018-11-04 stsp return request_packed_object(obj, pack, idx, id);
1076 77880158 2018-11-04 stsp
1077 77880158 2018-11-04 stsp ibuf = calloc(1, sizeof(*ibuf));
1078 77880158 2018-11-04 stsp if (ibuf == NULL)
1079 77880158 2018-11-04 stsp return got_error_from_errno();
1080 77880158 2018-11-04 stsp
1081 77880158 2018-11-04 stsp pack->privsep_child = calloc(1, sizeof(*pack->privsep_child));
1082 77880158 2018-11-04 stsp if (pack->privsep_child == NULL) {
1083 77880158 2018-11-04 stsp err = got_error_from_errno();
1084 77880158 2018-11-04 stsp free(ibuf);
1085 77880158 2018-11-04 stsp return err;
1086 77880158 2018-11-04 stsp }
1087 77880158 2018-11-04 stsp
1088 77880158 2018-11-04 stsp if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, imsg_fds) == -1) {
1089 77880158 2018-11-04 stsp err = got_error_from_errno();
1090 77880158 2018-11-04 stsp goto done;
1091 77880158 2018-11-04 stsp }
1092 77880158 2018-11-04 stsp
1093 77880158 2018-11-04 stsp pid = fork();
1094 77880158 2018-11-04 stsp if (pid == -1) {
1095 77880158 2018-11-04 stsp err = got_error_from_errno();
1096 77880158 2018-11-04 stsp goto done;
1097 77880158 2018-11-04 stsp } else if (pid == 0) {
1098 77880158 2018-11-04 stsp exec_privsep_child(imsg_fds, GOT_PATH_PROG_READ_PACK,
1099 77880158 2018-11-04 stsp pack->path_packfile);
1100 77880158 2018-11-04 stsp /* not reached */
1101 77880158 2018-11-04 stsp }
1102 77880158 2018-11-04 stsp
1103 77880158 2018-11-04 stsp close(imsg_fds[1]);
1104 77880158 2018-11-04 stsp pack->privsep_child->imsg_fd = imsg_fds[0];
1105 77880158 2018-11-04 stsp pack->privsep_child->pid = pid;
1106 77880158 2018-11-04 stsp imsg_init(ibuf, imsg_fds[0]);
1107 77880158 2018-11-04 stsp pack->privsep_child->ibuf = ibuf;
1108 77880158 2018-11-04 stsp
1109 77880158 2018-11-04 stsp err = got_privsep_init_pack_child(ibuf, pack, packidx);
1110 77880158 2018-11-04 stsp if (err) {
1111 77880158 2018-11-04 stsp const struct got_error *child_err;
1112 77880158 2018-11-04 stsp err = got_privsep_send_stop(pack->privsep_child->imsg_fd);
1113 77880158 2018-11-04 stsp child_err = got_privsep_wait_for_child(
1114 77880158 2018-11-04 stsp pack->privsep_child->pid);
1115 77880158 2018-11-04 stsp if (child_err && err == NULL)
1116 77880158 2018-11-04 stsp err = child_err;
1117 77880158 2018-11-04 stsp free(ibuf);
1118 77880158 2018-11-04 stsp free(pack->privsep_child);
1119 77880158 2018-11-04 stsp pack->privsep_child = NULL;
1120 77880158 2018-11-04 stsp return err;
1121 77880158 2018-11-04 stsp }
1122 77880158 2018-11-04 stsp
1123 77880158 2018-11-04 stsp done:
1124 77880158 2018-11-04 stsp if (err) {
1125 77880158 2018-11-04 stsp free(ibuf);
1126 77880158 2018-11-04 stsp free(pack->privsep_child);
1127 77880158 2018-11-04 stsp pack->privsep_child = NULL;
1128 77880158 2018-11-04 stsp } else
1129 77880158 2018-11-04 stsp err = request_packed_object(obj, pack, idx, id);
1130 07862c20 2018-09-15 stsp return err;
1131 07862c20 2018-09-15 stsp }
1132 77880158 2018-11-04 stsp
1133 77880158 2018-11-04 stsp static const struct got_error *
1134 77880158 2018-11-04 stsp request_commit(struct got_commit_object **commit, struct got_repository *repo,
1135 77880158 2018-11-04 stsp struct got_object *obj, int fd)
1136 77880158 2018-11-04 stsp {
1137 77880158 2018-11-04 stsp const struct got_error *err = NULL;
1138 77880158 2018-11-04 stsp struct imsgbuf *ibuf;
1139 77880158 2018-11-04 stsp
1140 77880158 2018-11-04 stsp ibuf = repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_COMMIT].ibuf;
1141 77880158 2018-11-04 stsp
1142 77880158 2018-11-04 stsp err = got_privsep_send_obj_req(ibuf, fd, obj);
1143 77880158 2018-11-04 stsp if (err)
1144 77880158 2018-11-04 stsp return err;
1145 77880158 2018-11-04 stsp
1146 77880158 2018-11-04 stsp return got_privsep_recv_commit(commit, ibuf);
1147 7762fe12 2018-11-05 stsp }
1148 7762fe12 2018-11-05 stsp
1149 7762fe12 2018-11-05 stsp static const struct got_error *
1150 7762fe12 2018-11-05 stsp request_mini_commit(struct got_commit_object_mini **commit,
1151 7762fe12 2018-11-05 stsp struct got_repository *repo, struct got_object *obj, int fd)
1152 7762fe12 2018-11-05 stsp {
1153 7762fe12 2018-11-05 stsp const struct got_error *err = NULL;
1154 7762fe12 2018-11-05 stsp struct imsgbuf *ibuf;
1155 7762fe12 2018-11-05 stsp
1156 7762fe12 2018-11-05 stsp ibuf = repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_COMMIT].ibuf;
1157 7762fe12 2018-11-05 stsp
1158 7762fe12 2018-11-05 stsp err = got_privsep_send_mini_commit_req(ibuf, fd, obj);
1159 7762fe12 2018-11-05 stsp if (err)
1160 7762fe12 2018-11-05 stsp return err;
1161 7762fe12 2018-11-05 stsp
1162 7762fe12 2018-11-05 stsp return got_privsep_recv_mini_commit(commit, ibuf);
1163 77880158 2018-11-04 stsp }
1164 77880158 2018-11-04 stsp
1165 77880158 2018-11-04 stsp const struct got_error *
1166 77880158 2018-11-04 stsp got_object_read_packed_commit_privsep(struct got_commit_object **commit,
1167 77880158 2018-11-04 stsp struct got_object *obj, struct got_pack *pack)
1168 77880158 2018-11-04 stsp {
1169 77880158 2018-11-04 stsp const struct got_error *err = NULL;
1170 77880158 2018-11-04 stsp
1171 77880158 2018-11-04 stsp err = got_privsep_send_obj_req(pack->privsep_child->ibuf, -1, obj);
1172 77880158 2018-11-04 stsp if (err)
1173 77880158 2018-11-04 stsp return err;
1174 77880158 2018-11-04 stsp
1175 77880158 2018-11-04 stsp return got_privsep_recv_commit(commit, pack->privsep_child->ibuf);
1176 77880158 2018-11-04 stsp }
1177 77880158 2018-11-04 stsp
1178 77880158 2018-11-04 stsp const struct got_error *
1179 7762fe12 2018-11-05 stsp got_object_read_packed_mini_commit_privsep(struct got_commit_object_mini **commit,
1180 7762fe12 2018-11-05 stsp struct got_object *obj, struct got_pack *pack)
1181 7762fe12 2018-11-05 stsp {
1182 7762fe12 2018-11-05 stsp const struct got_error *err = NULL;
1183 7762fe12 2018-11-05 stsp
1184 7762fe12 2018-11-05 stsp err = got_privsep_send_mini_commit_req(pack->privsep_child->ibuf, -1,
1185 7762fe12 2018-11-05 stsp obj);
1186 7762fe12 2018-11-05 stsp if (err)
1187 7762fe12 2018-11-05 stsp return err;
1188 7762fe12 2018-11-05 stsp
1189 7762fe12 2018-11-05 stsp return got_privsep_recv_mini_commit(commit, pack->privsep_child->ibuf);
1190 7762fe12 2018-11-05 stsp }
1191 7762fe12 2018-11-05 stsp
1192 7762fe12 2018-11-05 stsp const struct got_error *
1193 77880158 2018-11-04 stsp got_object_read_commit_privsep(struct got_commit_object **commit,
1194 77880158 2018-11-04 stsp struct got_object *obj, int obj_fd, struct got_repository *repo)
1195 77880158 2018-11-04 stsp {
1196 77880158 2018-11-04 stsp int imsg_fds[2];
1197 77880158 2018-11-04 stsp pid_t pid;
1198 77880158 2018-11-04 stsp struct imsgbuf *ibuf;
1199 77880158 2018-11-04 stsp
1200 77880158 2018-11-04 stsp if (repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_COMMIT].imsg_fd != -1)
1201 77880158 2018-11-04 stsp return request_commit(commit, repo, obj, obj_fd);
1202 77880158 2018-11-04 stsp
1203 77880158 2018-11-04 stsp ibuf = calloc(1, sizeof(*ibuf));
1204 77880158 2018-11-04 stsp if (ibuf == NULL)
1205 77880158 2018-11-04 stsp return got_error_from_errno();
1206 77880158 2018-11-04 stsp
1207 77880158 2018-11-04 stsp if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, imsg_fds) == -1)
1208 77880158 2018-11-04 stsp return got_error_from_errno();
1209 77880158 2018-11-04 stsp
1210 77880158 2018-11-04 stsp pid = fork();
1211 77880158 2018-11-04 stsp if (pid == -1)
1212 77880158 2018-11-04 stsp return got_error_from_errno();
1213 77880158 2018-11-04 stsp else if (pid == 0) {
1214 77880158 2018-11-04 stsp exec_privsep_child(imsg_fds, GOT_PATH_PROG_READ_COMMIT,
1215 77880158 2018-11-04 stsp repo->path);
1216 77880158 2018-11-04 stsp /* not reached */
1217 77880158 2018-11-04 stsp }
1218 77880158 2018-11-04 stsp
1219 77880158 2018-11-04 stsp close(imsg_fds[1]);
1220 77880158 2018-11-04 stsp repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_COMMIT].imsg_fd =
1221 77880158 2018-11-04 stsp imsg_fds[0];
1222 77880158 2018-11-04 stsp repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_COMMIT].pid = pid;
1223 77880158 2018-11-04 stsp imsg_init(ibuf, imsg_fds[0]);
1224 77880158 2018-11-04 stsp repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_COMMIT].ibuf = ibuf;
1225 77880158 2018-11-04 stsp
1226 77880158 2018-11-04 stsp return request_commit(commit, repo, obj, obj_fd);
1227 77880158 2018-11-04 stsp }
1228 77880158 2018-11-04 stsp
1229 7762fe12 2018-11-05 stsp const struct got_error *
1230 7762fe12 2018-11-05 stsp got_object_read_mini_commit_privsep(struct got_commit_object_mini **commit,
1231 7762fe12 2018-11-05 stsp struct got_object *obj, int obj_fd, struct got_repository *repo)
1232 7762fe12 2018-11-05 stsp {
1233 7762fe12 2018-11-05 stsp int imsg_fds[2];
1234 7762fe12 2018-11-05 stsp pid_t pid;
1235 7762fe12 2018-11-05 stsp struct imsgbuf *ibuf;
1236 7762fe12 2018-11-05 stsp
1237 7762fe12 2018-11-05 stsp if (repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_COMMIT].imsg_fd != -1)
1238 7762fe12 2018-11-05 stsp return request_mini_commit(commit, repo, obj, obj_fd);
1239 7762fe12 2018-11-05 stsp
1240 7762fe12 2018-11-05 stsp ibuf = calloc(1, sizeof(*ibuf));
1241 7762fe12 2018-11-05 stsp if (ibuf == NULL)
1242 7762fe12 2018-11-05 stsp return got_error_from_errno();
1243 7762fe12 2018-11-05 stsp
1244 7762fe12 2018-11-05 stsp if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, imsg_fds) == -1)
1245 7762fe12 2018-11-05 stsp return got_error_from_errno();
1246 7762fe12 2018-11-05 stsp
1247 7762fe12 2018-11-05 stsp pid = fork();
1248 7762fe12 2018-11-05 stsp if (pid == -1)
1249 7762fe12 2018-11-05 stsp return got_error_from_errno();
1250 7762fe12 2018-11-05 stsp else if (pid == 0) {
1251 7762fe12 2018-11-05 stsp exec_privsep_child(imsg_fds, GOT_PATH_PROG_READ_COMMIT,
1252 7762fe12 2018-11-05 stsp repo->path);
1253 7762fe12 2018-11-05 stsp /* not reached */
1254 7762fe12 2018-11-05 stsp }
1255 7762fe12 2018-11-05 stsp
1256 7762fe12 2018-11-05 stsp close(imsg_fds[1]);
1257 7762fe12 2018-11-05 stsp repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_COMMIT].imsg_fd =
1258 7762fe12 2018-11-05 stsp imsg_fds[0];
1259 7762fe12 2018-11-05 stsp repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_COMMIT].pid = pid;
1260 7762fe12 2018-11-05 stsp imsg_init(ibuf, imsg_fds[0]);
1261 7762fe12 2018-11-05 stsp repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_COMMIT].ibuf = ibuf;
1262 7762fe12 2018-11-05 stsp
1263 7762fe12 2018-11-05 stsp return request_mini_commit(commit, repo, obj, obj_fd);
1264 7762fe12 2018-11-05 stsp }
1265 7762fe12 2018-11-05 stsp
1266 77880158 2018-11-04 stsp static const struct got_error *
1267 77880158 2018-11-04 stsp request_tree(struct got_tree_object **tree, struct got_repository *repo,
1268 77880158 2018-11-04 stsp struct got_object *obj, int fd)
1269 77880158 2018-11-04 stsp {
1270 77880158 2018-11-04 stsp const struct got_error *err = NULL;
1271 77880158 2018-11-04 stsp struct imsgbuf *ibuf;
1272 77880158 2018-11-04 stsp
1273 77880158 2018-11-04 stsp ibuf = repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_TREE].ibuf;
1274 77880158 2018-11-04 stsp
1275 77880158 2018-11-04 stsp err = got_privsep_send_obj_req(ibuf, fd, obj);
1276 77880158 2018-11-04 stsp if (err)
1277 77880158 2018-11-04 stsp return err;
1278 77880158 2018-11-04 stsp
1279 77880158 2018-11-04 stsp return got_privsep_recv_tree(tree, ibuf);
1280 77880158 2018-11-04 stsp }
1281 77880158 2018-11-04 stsp
1282 77880158 2018-11-04 stsp const struct got_error *
1283 77880158 2018-11-04 stsp got_object_read_tree_privsep(struct got_tree_object **tree,
1284 77880158 2018-11-04 stsp struct got_object *obj, int obj_fd, struct got_repository *repo)
1285 77880158 2018-11-04 stsp {
1286 77880158 2018-11-04 stsp int imsg_fds[2];
1287 77880158 2018-11-04 stsp pid_t pid;
1288 77880158 2018-11-04 stsp struct imsgbuf *ibuf;
1289 77880158 2018-11-04 stsp
1290 77880158 2018-11-04 stsp if (repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_TREE].imsg_fd != -1)
1291 77880158 2018-11-04 stsp return request_tree(tree, repo, obj, obj_fd);
1292 77880158 2018-11-04 stsp
1293 77880158 2018-11-04 stsp ibuf = calloc(1, sizeof(*ibuf));
1294 77880158 2018-11-04 stsp if (ibuf == NULL)
1295 77880158 2018-11-04 stsp return got_error_from_errno();
1296 77880158 2018-11-04 stsp
1297 77880158 2018-11-04 stsp if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, imsg_fds) == -1)
1298 77880158 2018-11-04 stsp return got_error_from_errno();
1299 77880158 2018-11-04 stsp
1300 77880158 2018-11-04 stsp pid = fork();
1301 77880158 2018-11-04 stsp if (pid == -1)
1302 77880158 2018-11-04 stsp return got_error_from_errno();
1303 77880158 2018-11-04 stsp else if (pid == 0) {
1304 77880158 2018-11-04 stsp exec_privsep_child(imsg_fds, GOT_PATH_PROG_READ_TREE,
1305 77880158 2018-11-04 stsp repo->path);
1306 77880158 2018-11-04 stsp /* not reached */
1307 77880158 2018-11-04 stsp }
1308 77880158 2018-11-04 stsp
1309 77880158 2018-11-04 stsp close(imsg_fds[1]);
1310 77880158 2018-11-04 stsp
1311 77880158 2018-11-04 stsp repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_TREE].imsg_fd =
1312 77880158 2018-11-04 stsp imsg_fds[0];
1313 77880158 2018-11-04 stsp repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_TREE].pid = pid;
1314 77880158 2018-11-04 stsp imsg_init(ibuf, imsg_fds[0]);
1315 77880158 2018-11-04 stsp repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_TREE].ibuf = ibuf;
1316 77880158 2018-11-04 stsp
1317 77880158 2018-11-04 stsp
1318 77880158 2018-11-04 stsp return request_tree(tree, repo, obj, obj_fd);
1319 77880158 2018-11-04 stsp }
1320 77880158 2018-11-04 stsp
1321 77880158 2018-11-04 stsp const struct got_error *
1322 77880158 2018-11-04 stsp got_object_read_packed_tree_privsep(struct got_tree_object **tree,
1323 77880158 2018-11-04 stsp struct got_object *obj, struct got_pack *pack)
1324 77880158 2018-11-04 stsp {
1325 77880158 2018-11-04 stsp const struct got_error *err = NULL;
1326 77880158 2018-11-04 stsp
1327 77880158 2018-11-04 stsp err = got_privsep_send_obj_req(pack->privsep_child->ibuf, -1, obj);
1328 77880158 2018-11-04 stsp if (err)
1329 77880158 2018-11-04 stsp return err;
1330 77880158 2018-11-04 stsp
1331 77880158 2018-11-04 stsp return got_privsep_recv_tree(tree, pack->privsep_child->ibuf);
1332 77880158 2018-11-04 stsp }
1333 77880158 2018-11-04 stsp
1334 77880158 2018-11-04 stsp static const struct got_error *
1335 77880158 2018-11-04 stsp request_blob(size_t *size, int outfd, int infd, struct imsgbuf *ibuf)
1336 77880158 2018-11-04 stsp {
1337 77880158 2018-11-04 stsp const struct got_error *err = NULL;
1338 77880158 2018-11-04 stsp int outfd_child;
1339 77880158 2018-11-04 stsp
1340 77880158 2018-11-04 stsp outfd_child = dup(outfd);
1341 77880158 2018-11-04 stsp if (outfd_child == -1)
1342 77880158 2018-11-04 stsp return got_error_from_errno();
1343 77880158 2018-11-04 stsp
1344 77880158 2018-11-04 stsp err = got_privsep_send_blob_req(ibuf, infd);
1345 77880158 2018-11-04 stsp if (err)
1346 77880158 2018-11-04 stsp return err;
1347 77880158 2018-11-04 stsp
1348 77880158 2018-11-04 stsp err = got_privsep_send_blob_outfd(ibuf, outfd_child);
1349 77880158 2018-11-04 stsp if (err) {
1350 77880158 2018-11-04 stsp close(outfd_child);
1351 77880158 2018-11-04 stsp return err;
1352 77880158 2018-11-04 stsp }
1353 77880158 2018-11-04 stsp
1354 77880158 2018-11-04 stsp err = got_privsep_recv_blob(size, ibuf);
1355 77880158 2018-11-04 stsp if (err)
1356 77880158 2018-11-04 stsp return err;
1357 77880158 2018-11-04 stsp
1358 77880158 2018-11-04 stsp if (lseek(outfd, SEEK_SET, 0) == -1)
1359 77880158 2018-11-04 stsp return got_error_from_errno();
1360 77880158 2018-11-04 stsp
1361 77880158 2018-11-04 stsp return err;
1362 77880158 2018-11-04 stsp }
1363 77880158 2018-11-04 stsp
1364 77880158 2018-11-04 stsp const struct got_error *
1365 77880158 2018-11-04 stsp got_object_read_blob_privsep(size_t *size, int outfd, int infd,
1366 77880158 2018-11-04 stsp struct got_repository *repo)
1367 77880158 2018-11-04 stsp {
1368 77880158 2018-11-04 stsp int imsg_fds[2];
1369 77880158 2018-11-04 stsp pid_t pid;
1370 77880158 2018-11-04 stsp struct imsgbuf *ibuf;
1371 77880158 2018-11-04 stsp
1372 77880158 2018-11-04 stsp if (repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_BLOB].imsg_fd != -1) {
1373 77880158 2018-11-04 stsp ibuf = repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_BLOB].ibuf;
1374 77880158 2018-11-04 stsp return request_blob(size, outfd, infd, ibuf);
1375 77880158 2018-11-04 stsp }
1376 77880158 2018-11-04 stsp
1377 77880158 2018-11-04 stsp ibuf = calloc(1, sizeof(*ibuf));
1378 77880158 2018-11-04 stsp if (ibuf == NULL)
1379 77880158 2018-11-04 stsp return got_error_from_errno();
1380 77880158 2018-11-04 stsp
1381 77880158 2018-11-04 stsp if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, imsg_fds) == -1)
1382 77880158 2018-11-04 stsp return got_error_from_errno();
1383 77880158 2018-11-04 stsp
1384 77880158 2018-11-04 stsp pid = fork();
1385 77880158 2018-11-04 stsp if (pid == -1)
1386 77880158 2018-11-04 stsp return got_error_from_errno();
1387 77880158 2018-11-04 stsp else if (pid == 0) {
1388 77880158 2018-11-04 stsp exec_privsep_child(imsg_fds, GOT_PATH_PROG_READ_BLOB,
1389 77880158 2018-11-04 stsp repo->path);
1390 77880158 2018-11-04 stsp /* not reached */
1391 77880158 2018-11-04 stsp }
1392 77880158 2018-11-04 stsp
1393 77880158 2018-11-04 stsp close(imsg_fds[1]);
1394 77880158 2018-11-04 stsp repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_BLOB].imsg_fd =
1395 77880158 2018-11-04 stsp imsg_fds[0];
1396 77880158 2018-11-04 stsp repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_BLOB].pid = pid;
1397 77880158 2018-11-04 stsp imsg_init(ibuf, imsg_fds[0]);
1398 77880158 2018-11-04 stsp repo->privsep_children[GOT_REPO_PRIVSEP_CHILD_BLOB].ibuf = ibuf;
1399 77880158 2018-11-04 stsp
1400 77880158 2018-11-04 stsp return request_blob(size, outfd, infd, ibuf);
1401 77880158 2018-11-04 stsp }