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 d1cda826 2017-11-06 stsp
24 a1fd68d8 2018-01-12 stsp #include <errno.h>
25 2178c42e 2018-04-22 stsp #include <fcntl.h>
26 d71d75ad 2017-11-05 stsp #include <stdio.h>
27 ab9a70b2 2017-11-06 stsp #include <stdlib.h>
28 ab9a70b2 2017-11-06 stsp #include <string.h>
29 2178c42e 2018-04-22 stsp #include <stdint.h>
30 d71d75ad 2017-11-05 stsp #include <sha1.h>
31 ab9a70b2 2017-11-06 stsp #include <zlib.h>
32 ab9a70b2 2017-11-06 stsp #include <ctype.h>
33 ab9a70b2 2017-11-06 stsp #include <limits.h>
34 2178c42e 2018-04-22 stsp #include <imsg.h>
35 788c352e 2018-06-16 stsp #include <time.h>
36 d71d75ad 2017-11-05 stsp
37 ab9a70b2 2017-11-06 stsp #include "got_error.h"
38 d71d75ad 2017-11-05 stsp #include "got_object.h"
39 ab9a70b2 2017-11-06 stsp #include "got_repository.h"
40 511a516b 2018-05-19 stsp #include "got_opentemp.h"
41 d71d75ad 2017-11-05 stsp
42 718b3ab0 2018-03-17 stsp #include "got_lib_sha1.h"
43 718b3ab0 2018-03-17 stsp #include "got_lib_delta.h"
44 718b3ab0 2018-03-17 stsp #include "got_lib_pack.h"
45 2178c42e 2018-04-22 stsp #include "got_lib_path.h"
46 63581804 2018-07-09 stsp #include "got_lib_inflate.h"
47 718b3ab0 2018-03-17 stsp #include "got_lib_object.h"
48 2178c42e 2018-04-22 stsp #include "got_lib_privsep.h"
49 ad242220 2018-09-08 stsp #include "got_lib_object_parse.h"
50 7bb0daa1 2018-06-21 stsp #include "got_lib_repository.h"
51 1411938b 2018-02-12 stsp
52 ab9a70b2 2017-11-06 stsp #ifndef MIN
53 ab9a70b2 2017-11-06 stsp #define MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
54 ab9a70b2 2017-11-06 stsp #endif
55 ab9a70b2 2017-11-06 stsp
56 ef0981d5 2018-02-12 stsp const struct got_error *
57 ef0981d5 2018-02-12 stsp got_object_id_str(char **outbuf, struct got_object_id *id)
58 d71d75ad 2017-11-05 stsp {
59 ef0981d5 2018-02-12 stsp static const size_t len = SHA1_DIGEST_STRING_LENGTH;
60 ef0981d5 2018-02-12 stsp
61 062ebb78 2018-07-12 stsp *outbuf = malloc(len);
62 ef0981d5 2018-02-12 stsp if (*outbuf == NULL)
63 0a585a0d 2018-03-17 stsp return got_error_from_errno();
64 ef0981d5 2018-02-12 stsp
65 ef0981d5 2018-02-12 stsp if (got_sha1_digest_to_str(id->sha1, *outbuf, len) == NULL) {
66 ef0981d5 2018-02-12 stsp free(*outbuf);
67 ef0981d5 2018-02-12 stsp *outbuf = NULL;
68 ef0981d5 2018-02-12 stsp return got_error(GOT_ERR_BAD_OBJ_ID_STR);
69 ef0981d5 2018-02-12 stsp }
70 ef0981d5 2018-02-12 stsp
71 ef0981d5 2018-02-12 stsp return NULL;
72 59ece79d 2018-02-12 stsp }
73 59ece79d 2018-02-12 stsp
74 a1fd68d8 2018-01-12 stsp int
75 a1fd68d8 2018-01-12 stsp got_object_id_cmp(struct got_object_id *id1, struct got_object_id *id2)
76 a1fd68d8 2018-01-12 stsp {
77 a1fd68d8 2018-01-12 stsp return memcmp(id1->sha1, id2->sha1, SHA1_DIGEST_LENGTH);
78 8bf5b3c9 2018-03-17 stsp }
79 8bf5b3c9 2018-03-17 stsp
80 8bf5b3c9 2018-03-17 stsp struct got_object_id *
81 8bf5b3c9 2018-03-17 stsp got_object_id_dup(struct got_object_id *id1)
82 8bf5b3c9 2018-03-17 stsp {
83 8bf5b3c9 2018-03-17 stsp struct got_object_id *id2;
84 8bf5b3c9 2018-03-17 stsp
85 8bf5b3c9 2018-03-17 stsp id2 = malloc(sizeof(*id2));
86 8bf5b3c9 2018-03-17 stsp if (id2 == NULL)
87 8bf5b3c9 2018-03-17 stsp return NULL;
88 8bf5b3c9 2018-03-17 stsp memcpy(id2, id1, sizeof(*id2));
89 8bf5b3c9 2018-03-17 stsp return id2;
90 3235492e 2018-04-01 stsp }
91 3235492e 2018-04-01 stsp
92 3235492e 2018-04-01 stsp struct got_object_id *
93 3235492e 2018-04-01 stsp got_object_get_id(struct got_object *obj)
94 3235492e 2018-04-01 stsp {
95 3235492e 2018-04-01 stsp return got_object_id_dup(&obj->id);
96 bacc9935 2018-05-20 stsp }
97 bacc9935 2018-05-20 stsp
98 bacc9935 2018-05-20 stsp const struct got_error *
99 bacc9935 2018-05-20 stsp got_object_get_id_str(char **outbuf, struct got_object *obj)
100 bacc9935 2018-05-20 stsp {
101 bacc9935 2018-05-20 stsp return got_object_id_str(outbuf, &obj->id);
102 a1fd68d8 2018-01-12 stsp }
103 d71d75ad 2017-11-05 stsp
104 b107e67f 2018-01-19 stsp int
105 b107e67f 2018-01-19 stsp got_object_get_type(struct got_object *obj)
106 a1fd68d8 2018-01-12 stsp {
107 b107e67f 2018-01-19 stsp switch (obj->type) {
108 a1fd68d8 2018-01-12 stsp case GOT_OBJ_TYPE_COMMIT:
109 a1fd68d8 2018-01-12 stsp case GOT_OBJ_TYPE_TREE:
110 a1fd68d8 2018-01-12 stsp case GOT_OBJ_TYPE_BLOB:
111 b107e67f 2018-01-19 stsp case GOT_OBJ_TYPE_TAG:
112 b107e67f 2018-01-19 stsp return obj->type;
113 96f5e8b3 2018-01-23 stsp default:
114 96f5e8b3 2018-01-23 stsp abort();
115 96f5e8b3 2018-01-23 stsp break;
116 d71d75ad 2017-11-05 stsp }
117 d71d75ad 2017-11-05 stsp
118 96f5e8b3 2018-01-23 stsp /* not reached */
119 96f5e8b3 2018-01-23 stsp return 0;
120 d71d75ad 2017-11-05 stsp }
121 ab9a70b2 2017-11-06 stsp
122 ab9a70b2 2017-11-06 stsp static const struct got_error *
123 4558fcd4 2018-01-14 stsp object_path(char **path, struct got_object_id *id, struct got_repository *repo)
124 ab9a70b2 2017-11-06 stsp {
125 ab9a70b2 2017-11-06 stsp const struct got_error *err = NULL;
126 7a132809 2018-07-23 stsp char *hex = NULL;
127 d1cda826 2017-11-06 stsp char *path_objects = got_repo_get_path_objects(repo);
128 e6b1056e 2018-04-22 stsp
129 e6b1056e 2018-04-22 stsp *path = NULL;
130 ab9a70b2 2017-11-06 stsp
131 ab9a70b2 2017-11-06 stsp if (path_objects == NULL)
132 0a585a0d 2018-03-17 stsp return got_error_from_errno();
133 ab9a70b2 2017-11-06 stsp
134 ef0981d5 2018-02-12 stsp err = got_object_id_str(&hex, id);
135 ef0981d5 2018-02-12 stsp if (err)
136 7a132809 2018-07-23 stsp goto done;
137 ab9a70b2 2017-11-06 stsp
138 d1cda826 2017-11-06 stsp if (asprintf(path, "%s/%.2x/%s", path_objects,
139 d1cda826 2017-11-06 stsp id->sha1[0], hex + 2) == -1)
140 0a585a0d 2018-03-17 stsp err = got_error_from_errno();
141 ab9a70b2 2017-11-06 stsp
142 7a132809 2018-07-23 stsp done:
143 ef0981d5 2018-02-12 stsp free(hex);
144 d1cda826 2017-11-06 stsp free(path_objects);
145 d1cda826 2017-11-06 stsp return err;
146 d1cda826 2017-11-06 stsp }
147 d1cda826 2017-11-06 stsp
148 4ee4114f 2018-01-23 stsp static const struct got_error *
149 d5003b79 2018-04-22 stsp open_loose_object(int *fd, struct got_object *obj, struct got_repository *repo)
150 d1cda826 2017-11-06 stsp {
151 d1cda826 2017-11-06 stsp const struct got_error *err = NULL;
152 a1fd68d8 2018-01-12 stsp char *path;
153 6c00b545 2018-01-17 stsp
154 6c00b545 2018-01-17 stsp err = object_path(&path, &obj->id, repo);
155 d1cda826 2017-11-06 stsp if (err)
156 d1cda826 2017-11-06 stsp return err;
157 d5003b79 2018-04-22 stsp *fd = open(path, O_RDONLY | O_NOFOLLOW, GOT_DEFAULT_FILE_MODE);
158 d5003b79 2018-04-22 stsp if (*fd == -1) {
159 6c00b545 2018-01-17 stsp err = got_error_from_errno();
160 6c00b545 2018-01-17 stsp goto done;
161 a1fd68d8 2018-01-12 stsp }
162 4558fcd4 2018-01-14 stsp done:
163 4558fcd4 2018-01-14 stsp free(path);
164 4558fcd4 2018-01-14 stsp return err;
165 4558fcd4 2018-01-14 stsp }
166 a1fd68d8 2018-01-12 stsp
167 4558fcd4 2018-01-14 stsp const struct got_error *
168 4558fcd4 2018-01-14 stsp got_object_open(struct got_object **obj, struct got_repository *repo,
169 4558fcd4 2018-01-14 stsp struct got_object_id *id)
170 4558fcd4 2018-01-14 stsp {
171 6c00b545 2018-01-17 stsp const struct got_error *err = NULL;
172 6c00b545 2018-01-17 stsp char *path;
173 2178c42e 2018-04-22 stsp int fd;
174 7bb0daa1 2018-06-21 stsp
175 7bb0daa1 2018-06-21 stsp *obj = got_repo_get_cached_object(repo, id);
176 7bb0daa1 2018-06-21 stsp if (*obj != NULL) {
177 7bb0daa1 2018-06-21 stsp (*obj)->refcnt++;
178 7bb0daa1 2018-06-21 stsp return NULL;
179 7bb0daa1 2018-06-21 stsp }
180 4558fcd4 2018-01-14 stsp
181 6c00b545 2018-01-17 stsp err = object_path(&path, id, repo);
182 4558fcd4 2018-01-14 stsp if (err)
183 4558fcd4 2018-01-14 stsp return err;
184 4558fcd4 2018-01-14 stsp
185 2178c42e 2018-04-22 stsp fd = open(path, O_RDONLY | O_NOFOLLOW, GOT_DEFAULT_FILE_MODE);
186 2178c42e 2018-04-22 stsp if (fd == -1) {
187 6c00b545 2018-01-17 stsp if (errno != ENOENT) {
188 6c00b545 2018-01-17 stsp err = got_error_from_errno();
189 6c00b545 2018-01-17 stsp goto done;
190 6c00b545 2018-01-17 stsp }
191 6c00b545 2018-01-17 stsp err = got_packfile_open_object(obj, id, repo);
192 6c00b545 2018-01-17 stsp if (err)
193 6c00b545 2018-01-17 stsp goto done;
194 6c00b545 2018-01-17 stsp if (*obj == NULL)
195 6c00b545 2018-01-17 stsp err = got_error(GOT_ERR_NO_OBJ);
196 6c00b545 2018-01-17 stsp } else {
197 ad242220 2018-09-08 stsp err = got_object_read_header_privsep(obj, repo, fd);
198 6c00b545 2018-01-17 stsp if (err)
199 6c00b545 2018-01-17 stsp goto done;
200 ab9a70b2 2017-11-06 stsp memcpy((*obj)->id.sha1, id->sha1, SHA1_DIGEST_LENGTH);
201 6c00b545 2018-01-17 stsp }
202 7bb0daa1 2018-06-21 stsp
203 7bb0daa1 2018-06-21 stsp if (err == NULL) {
204 7bb0daa1 2018-06-21 stsp (*obj)->refcnt++;
205 7bb0daa1 2018-06-21 stsp err = got_repo_cache_object(repo, id, *obj);
206 7bb0daa1 2018-06-21 stsp }
207 6c00b545 2018-01-17 stsp done:
208 6c00b545 2018-01-17 stsp free(path);
209 2178c42e 2018-04-22 stsp if (fd != -1)
210 2178c42e 2018-04-22 stsp close(fd);
211 ab9a70b2 2017-11-06 stsp return err;
212 6c00b545 2018-01-17 stsp
213 ab9a70b2 2017-11-06 stsp }
214 ab9a70b2 2017-11-06 stsp
215 6dfa2fd3 2018-02-12 stsp const struct got_error *
216 6dfa2fd3 2018-02-12 stsp got_object_open_by_id_str(struct got_object **obj, struct got_repository *repo,
217 6dfa2fd3 2018-02-12 stsp const char *id_str)
218 6dfa2fd3 2018-02-12 stsp {
219 6dfa2fd3 2018-02-12 stsp struct got_object_id id;
220 6dfa2fd3 2018-02-12 stsp
221 6dfa2fd3 2018-02-12 stsp if (!got_parse_sha1_digest(id.sha1, id_str))
222 6dfa2fd3 2018-02-12 stsp return got_error(GOT_ERR_BAD_OBJ_ID_STR);
223 6dfa2fd3 2018-02-12 stsp
224 6dfa2fd3 2018-02-12 stsp return got_object_open(obj, repo, &id);
225 bff6ca00 2018-04-23 stsp }
226 bff6ca00 2018-04-23 stsp
227 bff6ca00 2018-04-23 stsp const struct got_error *
228 be6a1b5a 2018-06-11 stsp got_object_open_as_commit(struct got_commit_object **commit,
229 be6a1b5a 2018-06-11 stsp struct got_repository *repo, struct got_object_id *id)
230 be6a1b5a 2018-06-11 stsp {
231 be6a1b5a 2018-06-11 stsp const struct got_error *err;
232 be6a1b5a 2018-06-11 stsp struct got_object *obj;
233 835e0dbd 2018-06-21 stsp
234 835e0dbd 2018-06-21 stsp *commit = NULL;
235 be6a1b5a 2018-06-11 stsp
236 be6a1b5a 2018-06-11 stsp err = got_object_open(&obj, repo, id);
237 be6a1b5a 2018-06-11 stsp if (err)
238 be6a1b5a 2018-06-11 stsp return err;
239 be6a1b5a 2018-06-11 stsp if (got_object_get_type(obj) != GOT_OBJ_TYPE_COMMIT) {
240 be6a1b5a 2018-06-11 stsp err = got_error(GOT_ERR_OBJ_TYPE);
241 be6a1b5a 2018-06-11 stsp goto done;
242 be6a1b5a 2018-06-11 stsp }
243 be6a1b5a 2018-06-11 stsp
244 be6a1b5a 2018-06-11 stsp err = got_object_commit_open(commit, repo, obj);
245 be6a1b5a 2018-06-11 stsp done:
246 be6a1b5a 2018-06-11 stsp got_object_close(obj);
247 be6a1b5a 2018-06-11 stsp return err;
248 be6a1b5a 2018-06-11 stsp }
249 be6a1b5a 2018-06-11 stsp
250 be6a1b5a 2018-06-11 stsp const struct got_error *
251 dbc6a6b6 2018-07-12 stsp got_object_qid_alloc(struct got_object_qid **qid, struct got_object_id *id)
252 dbc6a6b6 2018-07-12 stsp {
253 dbc6a6b6 2018-07-12 stsp const struct got_error *err = NULL;
254 dbc6a6b6 2018-07-12 stsp
255 dbc6a6b6 2018-07-12 stsp *qid = calloc(1, sizeof(**qid));
256 dbc6a6b6 2018-07-12 stsp if (*qid == NULL)
257 dbc6a6b6 2018-07-12 stsp return got_error_from_errno();
258 dbc6a6b6 2018-07-12 stsp
259 dbc6a6b6 2018-07-12 stsp (*qid)->id = got_object_id_dup(id);
260 dbc6a6b6 2018-07-12 stsp if ((*qid)->id == NULL) {
261 dbc6a6b6 2018-07-12 stsp err = got_error_from_errno();
262 fa2f6902 2018-07-23 stsp got_object_qid_free(*qid);
263 dbc6a6b6 2018-07-12 stsp *qid = NULL;
264 dbc6a6b6 2018-07-12 stsp return err;
265 dbc6a6b6 2018-07-12 stsp }
266 dbc6a6b6 2018-07-12 stsp
267 dbc6a6b6 2018-07-12 stsp return NULL;
268 dbc6a6b6 2018-07-12 stsp }
269 dbc6a6b6 2018-07-12 stsp
270 d1cda826 2017-11-06 stsp const struct got_error *
271 d1cda826 2017-11-06 stsp got_object_commit_open(struct got_commit_object **commit,
272 d1cda826 2017-11-06 stsp struct got_repository *repo, struct got_object *obj)
273 d1cda826 2017-11-06 stsp {
274 d1cda826 2017-11-06 stsp const struct got_error *err = NULL;
275 1943de01 2018-06-22 stsp
276 1943de01 2018-06-22 stsp *commit = got_repo_get_cached_commit(repo, &obj->id);
277 1943de01 2018-06-22 stsp if (*commit != NULL) {
278 1943de01 2018-06-22 stsp (*commit)->refcnt++;
279 1943de01 2018-06-22 stsp return NULL;
280 1943de01 2018-06-22 stsp }
281 d1cda826 2017-11-06 stsp
282 d1cda826 2017-11-06 stsp if (obj->type != GOT_OBJ_TYPE_COMMIT)
283 d1cda826 2017-11-06 stsp return got_error(GOT_ERR_OBJ_TYPE);
284 d1cda826 2017-11-06 stsp
285 ea35256b 2018-03-16 stsp if (obj->flags & GOT_OBJ_FLAG_PACKED) {
286 ea35256b 2018-03-16 stsp uint8_t *buf;
287 ea35256b 2018-03-16 stsp size_t len;
288 ea35256b 2018-03-16 stsp err = got_packfile_extract_object_to_mem(&buf, &len, obj, repo);
289 ea35256b 2018-03-16 stsp if (err)
290 ea35256b 2018-03-16 stsp return err;
291 b29656e2 2018-03-16 stsp obj->size = len;
292 a440fac0 2018-09-06 stsp err = got_object_parse_commit(commit, buf, len);
293 ea35256b 2018-03-16 stsp free(buf);
294 ea35256b 2018-03-16 stsp } else {
295 d5003b79 2018-04-22 stsp int fd;
296 d5003b79 2018-04-22 stsp err = open_loose_object(&fd, obj, repo);
297 ea35256b 2018-03-16 stsp if (err)
298 ea35256b 2018-03-16 stsp return err;
299 ad242220 2018-09-08 stsp err = got_object_read_commit_privsep(commit, obj, fd, repo);
300 bff6ca00 2018-04-23 stsp close(fd);
301 ea35256b 2018-03-16 stsp }
302 1943de01 2018-06-22 stsp
303 1943de01 2018-06-22 stsp if (err == NULL) {
304 1943de01 2018-06-22 stsp (*commit)->refcnt++;
305 1943de01 2018-06-22 stsp err = got_repo_cache_commit(repo, &obj->id, *commit);
306 1943de01 2018-06-22 stsp }
307 1943de01 2018-06-22 stsp
308 d1cda826 2017-11-06 stsp return err;
309 d1cda826 2017-11-06 stsp }
310 d1cda826 2017-11-06 stsp
311 0ffeb3c2 2017-11-26 stsp const struct got_error *
312 0ffeb3c2 2017-11-26 stsp got_object_tree_open(struct got_tree_object **tree,
313 0ffeb3c2 2017-11-26 stsp struct got_repository *repo, struct got_object *obj)
314 0ffeb3c2 2017-11-26 stsp {
315 0ffeb3c2 2017-11-26 stsp const struct got_error *err = NULL;
316 f6be5c30 2018-06-22 stsp
317 f6be5c30 2018-06-22 stsp *tree = got_repo_get_cached_tree(repo, &obj->id);
318 f6be5c30 2018-06-22 stsp if (*tree != NULL) {
319 f6be5c30 2018-06-22 stsp (*tree)->refcnt++;
320 f6be5c30 2018-06-22 stsp return NULL;
321 f6be5c30 2018-06-22 stsp }
322 0ffeb3c2 2017-11-26 stsp
323 0ffeb3c2 2017-11-26 stsp if (obj->type != GOT_OBJ_TYPE_TREE)
324 0ffeb3c2 2017-11-26 stsp return got_error(GOT_ERR_OBJ_TYPE);
325 0ffeb3c2 2017-11-26 stsp
326 e0ab43e7 2018-03-16 stsp if (obj->flags & GOT_OBJ_FLAG_PACKED) {
327 e0ab43e7 2018-03-16 stsp uint8_t *buf;
328 e0ab43e7 2018-03-16 stsp size_t len;
329 e0ab43e7 2018-03-16 stsp err = got_packfile_extract_object_to_mem(&buf, &len, obj, repo);
330 e0ab43e7 2018-03-16 stsp if (err)
331 e0ab43e7 2018-03-16 stsp return err;
332 b29656e2 2018-03-16 stsp obj->size = len;
333 a440fac0 2018-09-06 stsp err = got_object_parse_tree(tree, buf, len);
334 e0ab43e7 2018-03-16 stsp free(buf);
335 e0ab43e7 2018-03-16 stsp } else {
336 d5003b79 2018-04-22 stsp int fd;
337 d5003b79 2018-04-22 stsp err = open_loose_object(&fd, obj, repo);
338 e0ab43e7 2018-03-16 stsp if (err)
339 e0ab43e7 2018-03-16 stsp return err;
340 ad242220 2018-09-08 stsp err = got_object_read_tree_privsep(tree, obj, fd, repo);
341 d5003b79 2018-04-22 stsp close(fd);
342 776d4d29 2018-06-17 stsp }
343 f6be5c30 2018-06-22 stsp
344 f6be5c30 2018-06-22 stsp if (err == NULL) {
345 f6be5c30 2018-06-22 stsp (*tree)->refcnt++;
346 f6be5c30 2018-06-22 stsp err = got_repo_cache_tree(repo, &obj->id, *tree);
347 f6be5c30 2018-06-22 stsp }
348 f6be5c30 2018-06-22 stsp
349 776d4d29 2018-06-17 stsp return err;
350 776d4d29 2018-06-17 stsp }
351 776d4d29 2018-06-17 stsp
352 776d4d29 2018-06-17 stsp const struct got_error *
353 776d4d29 2018-06-17 stsp got_object_open_as_tree(struct got_tree_object **tree,
354 776d4d29 2018-06-17 stsp struct got_repository *repo, struct got_object_id *id)
355 776d4d29 2018-06-17 stsp {
356 776d4d29 2018-06-17 stsp const struct got_error *err;
357 776d4d29 2018-06-17 stsp struct got_object *obj;
358 835e0dbd 2018-06-21 stsp
359 835e0dbd 2018-06-21 stsp *tree = NULL;
360 776d4d29 2018-06-17 stsp
361 776d4d29 2018-06-17 stsp err = got_object_open(&obj, repo, id);
362 776d4d29 2018-06-17 stsp if (err)
363 776d4d29 2018-06-17 stsp return err;
364 1cbc02b6 2018-06-21 stsp if (got_object_get_type(obj) != GOT_OBJ_TYPE_TREE) {
365 776d4d29 2018-06-17 stsp err = got_error(GOT_ERR_OBJ_TYPE);
366 776d4d29 2018-06-17 stsp goto done;
367 e0ab43e7 2018-03-16 stsp }
368 776d4d29 2018-06-17 stsp
369 776d4d29 2018-06-17 stsp err = got_object_tree_open(tree, repo, obj);
370 776d4d29 2018-06-17 stsp done:
371 776d4d29 2018-06-17 stsp got_object_close(obj);
372 0ffeb3c2 2017-11-26 stsp return err;
373 57efb1af 2018-04-24 stsp }
374 ff6b18f8 2018-04-24 stsp
375 883f0469 2018-06-23 stsp const struct got_tree_entries *
376 883f0469 2018-06-23 stsp got_object_tree_get_entries(struct got_tree_object *tree)
377 883f0469 2018-06-23 stsp {
378 883f0469 2018-06-23 stsp return &tree->entries;
379 883f0469 2018-06-23 stsp }
380 883f0469 2018-06-23 stsp
381 68482ea3 2017-11-27 stsp const struct got_error *
382 68482ea3 2017-11-27 stsp got_object_blob_open(struct got_blob_object **blob,
383 68482ea3 2017-11-27 stsp struct got_repository *repo, struct got_object *obj, size_t blocksize)
384 68482ea3 2017-11-27 stsp {
385 68482ea3 2017-11-27 stsp const struct got_error *err = NULL;
386 68482ea3 2017-11-27 stsp
387 68482ea3 2017-11-27 stsp if (obj->type != GOT_OBJ_TYPE_BLOB)
388 68482ea3 2017-11-27 stsp return got_error(GOT_ERR_OBJ_TYPE);
389 68482ea3 2017-11-27 stsp
390 7d283eee 2017-11-29 stsp if (blocksize < obj->hdrlen)
391 7d283eee 2017-11-29 stsp return got_error(GOT_ERR_NO_SPACE);
392 7d283eee 2017-11-29 stsp
393 68482ea3 2017-11-27 stsp *blob = calloc(1, sizeof(**blob));
394 4558fcd4 2018-01-14 stsp if (*blob == NULL)
395 0a585a0d 2018-03-17 stsp return got_error_from_errno();
396 68482ea3 2017-11-27 stsp
397 062ebb78 2018-07-12 stsp (*blob)->read_buf = malloc(blocksize);
398 15c8b0e6 2018-04-24 stsp if ((*blob)->read_buf == NULL) {
399 15c8b0e6 2018-04-24 stsp err = got_error_from_errno();
400 c7254d79 2018-04-24 stsp goto done;
401 15c8b0e6 2018-04-24 stsp }
402 eb651edf 2018-02-11 stsp if (obj->flags & GOT_OBJ_FLAG_PACKED) {
403 eb651edf 2018-02-11 stsp err = got_packfile_extract_object(&((*blob)->f), obj, repo);
404 c7254d79 2018-04-24 stsp if (err)
405 c7254d79 2018-04-24 stsp goto done;
406 eb651edf 2018-02-11 stsp } else {
407 3aca5731 2018-04-24 stsp int infd, outfd;
408 2967a784 2018-04-24 stsp size_t size;
409 2967a784 2018-04-24 stsp struct stat sb;
410 c7254d79 2018-04-24 stsp
411 3aca5731 2018-04-24 stsp err = open_loose_object(&infd, obj, repo);
412 c7254d79 2018-04-24 stsp if (err)
413 c7254d79 2018-04-24 stsp goto done;
414 c7254d79 2018-04-24 stsp
415 3aca5731 2018-04-24 stsp
416 3aca5731 2018-04-24 stsp outfd = got_opentempfd();
417 3aca5731 2018-04-24 stsp if (outfd == -1) {
418 3aca5731 2018-04-24 stsp err = got_error_from_errno();
419 3aca5731 2018-04-24 stsp close(infd);
420 3aca5731 2018-04-24 stsp goto done;
421 3aca5731 2018-04-24 stsp }
422 3aca5731 2018-04-24 stsp
423 ad242220 2018-09-08 stsp err = got_object_read_blob_privsep(&size, outfd, infd, repo);
424 3aca5731 2018-04-24 stsp close(infd);
425 57efb1af 2018-04-24 stsp if (err)
426 c7254d79 2018-04-24 stsp goto done;
427 3aca5731 2018-04-24 stsp
428 2967a784 2018-04-24 stsp if (size != obj->hdrlen + obj->size) {
429 1a6b3ab7 2018-04-24 stsp err = got_error(GOT_ERR_PRIVSEP_LEN);
430 2967a784 2018-04-24 stsp close(outfd);
431 2967a784 2018-04-24 stsp goto done;
432 2967a784 2018-04-24 stsp }
433 2967a784 2018-04-24 stsp
434 2967a784 2018-04-24 stsp if (fstat(outfd, &sb) == -1) {
435 2967a784 2018-04-24 stsp err = got_error_from_errno();
436 2967a784 2018-04-24 stsp close(outfd);
437 2967a784 2018-04-24 stsp goto done;
438 2967a784 2018-04-24 stsp }
439 2967a784 2018-04-24 stsp
440 2967a784 2018-04-24 stsp if (sb.st_size != size) {
441 2967a784 2018-04-24 stsp err = got_error(GOT_ERR_PRIVSEP_LEN);
442 2967a784 2018-04-24 stsp close(outfd);
443 2967a784 2018-04-24 stsp goto done;
444 2967a784 2018-04-24 stsp }
445 2967a784 2018-04-24 stsp
446 3aca5731 2018-04-24 stsp (*blob)->f = fdopen(outfd, "rb");
447 3aca5731 2018-04-24 stsp if ((*blob)->f == NULL) {
448 3aca5731 2018-04-24 stsp err = got_error_from_errno();
449 3aca5731 2018-04-24 stsp close(outfd);
450 3aca5731 2018-04-24 stsp goto done;
451 3aca5731 2018-04-24 stsp }
452 68482ea3 2017-11-27 stsp }
453 68482ea3 2017-11-27 stsp
454 7d283eee 2017-11-29 stsp (*blob)->hdrlen = obj->hdrlen;
455 eb651edf 2018-02-11 stsp (*blob)->blocksize = blocksize;
456 f78b0693 2017-11-29 stsp memcpy(&(*blob)->id.sha1, obj->id.sha1, SHA1_DIGEST_LENGTH);
457 7d283eee 2017-11-29 stsp
458 c7254d79 2018-04-24 stsp done:
459 c7254d79 2018-04-24 stsp if (err && *blob) {
460 c7254d79 2018-04-24 stsp if ((*blob)->f)
461 c7254d79 2018-04-24 stsp fclose((*blob)->f);
462 c7254d79 2018-04-24 stsp free((*blob)->read_buf);
463 c7254d79 2018-04-24 stsp free(*blob);
464 c7254d79 2018-04-24 stsp *blob = NULL;
465 a19581a2 2018-06-21 stsp }
466 a19581a2 2018-06-21 stsp return err;
467 a19581a2 2018-06-21 stsp }
468 a19581a2 2018-06-21 stsp
469 a19581a2 2018-06-21 stsp const struct got_error *
470 a19581a2 2018-06-21 stsp got_object_open_as_blob(struct got_blob_object **blob,
471 a19581a2 2018-06-21 stsp struct got_repository *repo, struct got_object_id *id,
472 a19581a2 2018-06-21 stsp size_t blocksize)
473 a19581a2 2018-06-21 stsp {
474 a19581a2 2018-06-21 stsp const struct got_error *err;
475 a19581a2 2018-06-21 stsp struct got_object *obj;
476 835e0dbd 2018-06-21 stsp
477 835e0dbd 2018-06-21 stsp *blob = NULL;
478 a19581a2 2018-06-21 stsp
479 a19581a2 2018-06-21 stsp err = got_object_open(&obj, repo, id);
480 a19581a2 2018-06-21 stsp if (err)
481 a19581a2 2018-06-21 stsp return err;
482 a19581a2 2018-06-21 stsp if (got_object_get_type(obj) != GOT_OBJ_TYPE_BLOB) {
483 a19581a2 2018-06-21 stsp err = got_error(GOT_ERR_OBJ_TYPE);
484 a19581a2 2018-06-21 stsp goto done;
485 c7254d79 2018-04-24 stsp }
486 a19581a2 2018-06-21 stsp
487 a19581a2 2018-06-21 stsp err = got_object_blob_open(blob, repo, obj, blocksize);
488 a19581a2 2018-06-21 stsp done:
489 a19581a2 2018-06-21 stsp got_object_close(obj);
490 68482ea3 2017-11-27 stsp return err;
491 0ffeb3c2 2017-11-26 stsp }
492 68482ea3 2017-11-27 stsp
493 68482ea3 2017-11-27 stsp void
494 68482ea3 2017-11-27 stsp got_object_blob_close(struct got_blob_object *blob)
495 68482ea3 2017-11-27 stsp {
496 15c8b0e6 2018-04-24 stsp free(blob->read_buf);
497 68482ea3 2017-11-27 stsp fclose(blob->f);
498 68482ea3 2017-11-27 stsp free(blob);
499 f934cf2c 2018-02-12 stsp }
500 f934cf2c 2018-02-12 stsp
501 f934cf2c 2018-02-12 stsp char *
502 f934cf2c 2018-02-12 stsp got_object_blob_id_str(struct got_blob_object *blob, char *buf, size_t size)
503 f934cf2c 2018-02-12 stsp {
504 f934cf2c 2018-02-12 stsp return got_sha1_digest_to_str(blob->id.sha1, buf, size);
505 f934cf2c 2018-02-12 stsp }
506 f934cf2c 2018-02-12 stsp
507 f934cf2c 2018-02-12 stsp size_t
508 f934cf2c 2018-02-12 stsp got_object_blob_get_hdrlen(struct got_blob_object *blob)
509 f934cf2c 2018-02-12 stsp {
510 f934cf2c 2018-02-12 stsp return blob->hdrlen;
511 68482ea3 2017-11-27 stsp }
512 68482ea3 2017-11-27 stsp
513 f934cf2c 2018-02-12 stsp const uint8_t *
514 f934cf2c 2018-02-12 stsp got_object_blob_get_read_buf(struct got_blob_object *blob)
515 f934cf2c 2018-02-12 stsp {
516 f934cf2c 2018-02-12 stsp return blob->read_buf;
517 f934cf2c 2018-02-12 stsp }
518 f934cf2c 2018-02-12 stsp
519 68482ea3 2017-11-27 stsp const struct got_error *
520 eb651edf 2018-02-11 stsp got_object_blob_read_block(size_t *outlenp, struct got_blob_object *blob)
521 68482ea3 2017-11-27 stsp {
522 eb651edf 2018-02-11 stsp size_t n;
523 eb651edf 2018-02-11 stsp
524 eb651edf 2018-02-11 stsp n = fread(blob->read_buf, 1, blob->blocksize, blob->f);
525 eb651edf 2018-02-11 stsp if (n == 0 && ferror(blob->f))
526 eb651edf 2018-02-11 stsp return got_ferror(blob->f, GOT_ERR_IO);
527 eb651edf 2018-02-11 stsp *outlenp = n;
528 35e9ba5d 2018-06-21 stsp return NULL;
529 35e9ba5d 2018-06-21 stsp }
530 35e9ba5d 2018-06-21 stsp
531 35e9ba5d 2018-06-21 stsp const struct got_error *
532 84451b3e 2018-07-10 stsp got_object_blob_dump_to_file(size_t *total_len, size_t *nlines,
533 84451b3e 2018-07-10 stsp FILE *outfile, struct got_blob_object *blob)
534 35e9ba5d 2018-06-21 stsp {
535 35e9ba5d 2018-06-21 stsp const struct got_error *err = NULL;
536 35e9ba5d 2018-06-21 stsp size_t len, hdrlen;
537 84451b3e 2018-07-10 stsp const uint8_t *buf;
538 84451b3e 2018-07-10 stsp int i;
539 84451b3e 2018-07-10 stsp
540 84451b3e 2018-07-10 stsp if (total_len)
541 84451b3e 2018-07-10 stsp *total_len = 0;
542 84451b3e 2018-07-10 stsp if (nlines)
543 84451b3e 2018-07-10 stsp *nlines = 0;
544 35e9ba5d 2018-06-21 stsp
545 35e9ba5d 2018-06-21 stsp hdrlen = got_object_blob_get_hdrlen(blob);
546 35e9ba5d 2018-06-21 stsp do {
547 35e9ba5d 2018-06-21 stsp err = got_object_blob_read_block(&len, blob);
548 35e9ba5d 2018-06-21 stsp if (err)
549 35e9ba5d 2018-06-21 stsp return err;
550 35e9ba5d 2018-06-21 stsp if (len == 0)
551 35e9ba5d 2018-06-21 stsp break;
552 84451b3e 2018-07-10 stsp if (total_len)
553 84451b3e 2018-07-10 stsp *total_len += len;
554 84451b3e 2018-07-10 stsp buf = got_object_blob_get_read_buf(blob);
555 84451b3e 2018-07-10 stsp if (nlines) {
556 84451b3e 2018-07-10 stsp for (i = 0; i < len; i++) {
557 84451b3e 2018-07-10 stsp if (buf[i] == '\n')
558 84451b3e 2018-07-10 stsp (*nlines)++;
559 84451b3e 2018-07-10 stsp }
560 84451b3e 2018-07-10 stsp }
561 35e9ba5d 2018-06-21 stsp /* Skip blob object header first time around. */
562 84451b3e 2018-07-10 stsp fwrite(buf + hdrlen, len - hdrlen, 1, outfile);
563 35e9ba5d 2018-06-21 stsp hdrlen = 0;
564 35e9ba5d 2018-06-21 stsp } while (len != 0);
565 35e9ba5d 2018-06-21 stsp
566 35e9ba5d 2018-06-21 stsp fflush(outfile);
567 35e9ba5d 2018-06-21 stsp rewind(outfile);
568 35e9ba5d 2018-06-21 stsp
569 776d4d29 2018-06-17 stsp return NULL;
570 776d4d29 2018-06-17 stsp }
571 776d4d29 2018-06-17 stsp
572 776d4d29 2018-06-17 stsp static struct got_tree_entry *
573 776d4d29 2018-06-17 stsp find_entry_by_name(struct got_tree_object *tree, const char *name)
574 776d4d29 2018-06-17 stsp {
575 776d4d29 2018-06-17 stsp struct got_tree_entry *te;
576 776d4d29 2018-06-17 stsp
577 883f0469 2018-06-23 stsp SIMPLEQ_FOREACH(te, &tree->entries.head, entry) {
578 776d4d29 2018-06-17 stsp if (strcmp(te->name, name) == 0)
579 776d4d29 2018-06-17 stsp return te;
580 776d4d29 2018-06-17 stsp }
581 eb651edf 2018-02-11 stsp return NULL;
582 776d4d29 2018-06-17 stsp }
583 776d4d29 2018-06-17 stsp
584 776d4d29 2018-06-17 stsp const struct got_error *
585 776d4d29 2018-06-17 stsp got_object_open_by_path(struct got_object **obj, struct got_repository *repo,
586 776d4d29 2018-06-17 stsp struct got_object_id *commit_id, const char *path)
587 776d4d29 2018-06-17 stsp {
588 776d4d29 2018-06-17 stsp const struct got_error *err = NULL;
589 776d4d29 2018-06-17 stsp struct got_commit_object *commit = NULL;
590 776d4d29 2018-06-17 stsp struct got_tree_object *tree = NULL;
591 db37e2c0 2018-06-21 stsp struct got_tree_entry *te = NULL;
592 197aa481 2018-06-21 stsp char *seg, *s, *s0 = NULL;
593 67606321 2018-06-21 stsp size_t len = strlen(path);
594 776d4d29 2018-06-17 stsp
595 776d4d29 2018-06-17 stsp *obj = NULL;
596 776d4d29 2018-06-17 stsp
597 776d4d29 2018-06-17 stsp /* We are expecting an absolute in-repository path. */
598 776d4d29 2018-06-17 stsp if (path[0] != '/')
599 776d4d29 2018-06-17 stsp return got_error(GOT_ERR_NOT_ABSPATH);
600 776d4d29 2018-06-17 stsp
601 776d4d29 2018-06-17 stsp err = got_object_open_as_commit(&commit, repo, commit_id);
602 776d4d29 2018-06-17 stsp if (err)
603 776d4d29 2018-06-17 stsp goto done;
604 776d4d29 2018-06-17 stsp
605 776d4d29 2018-06-17 stsp /* Handle opening of root of commit's tree. */
606 776d4d29 2018-06-17 stsp if (path[1] == '\0') {
607 776d4d29 2018-06-17 stsp err = got_object_open(obj, repo, commit->tree_id);
608 ca008b32 2018-07-22 stsp goto done;
609 776d4d29 2018-06-17 stsp }
610 776d4d29 2018-06-17 stsp
611 776d4d29 2018-06-17 stsp err = got_object_open_as_tree(&tree, repo, commit->tree_id);
612 776d4d29 2018-06-17 stsp if (err)
613 776d4d29 2018-06-17 stsp goto done;
614 776d4d29 2018-06-17 stsp
615 197aa481 2018-06-21 stsp s0 = strdup(path);
616 197aa481 2018-06-21 stsp if (s0 == NULL) {
617 776d4d29 2018-06-17 stsp err = got_error_from_errno();
618 776d4d29 2018-06-17 stsp goto done;
619 776d4d29 2018-06-17 stsp }
620 67606321 2018-06-21 stsp err = got_canonpath(path, s0, len + 1);
621 776d4d29 2018-06-17 stsp if (err)
622 776d4d29 2018-06-17 stsp goto done;
623 776d4d29 2018-06-17 stsp
624 197aa481 2018-06-21 stsp s = s0;
625 776d4d29 2018-06-17 stsp s++; /* skip leading '/' */
626 67606321 2018-06-21 stsp len--;
627 776d4d29 2018-06-17 stsp seg = s;
628 67606321 2018-06-21 stsp while (len > 0) {
629 776d4d29 2018-06-17 stsp struct got_tree_object *next_tree;
630 776d4d29 2018-06-17 stsp
631 776d4d29 2018-06-17 stsp if (*s != '/') {
632 776d4d29 2018-06-17 stsp s++;
633 67606321 2018-06-21 stsp len--;
634 00530cfb 2018-06-21 stsp if (*s)
635 00530cfb 2018-06-21 stsp continue;
636 776d4d29 2018-06-17 stsp }
637 776d4d29 2018-06-17 stsp
638 776d4d29 2018-06-17 stsp /* end of path segment */
639 776d4d29 2018-06-17 stsp *s = '\0';
640 776d4d29 2018-06-17 stsp
641 db37e2c0 2018-06-21 stsp te = find_entry_by_name(tree, seg);
642 db37e2c0 2018-06-21 stsp if (te == NULL) {
643 776d4d29 2018-06-17 stsp err = got_error(GOT_ERR_NO_OBJ);
644 776d4d29 2018-06-17 stsp goto done;
645 776d4d29 2018-06-17 stsp }
646 776d4d29 2018-06-17 stsp
647 67606321 2018-06-21 stsp if (len == 0)
648 67606321 2018-06-21 stsp break;
649 67606321 2018-06-21 stsp
650 776d4d29 2018-06-17 stsp seg = s + 1;
651 776d4d29 2018-06-17 stsp s++;
652 67606321 2018-06-21 stsp len--;
653 776d4d29 2018-06-17 stsp if (*s) {
654 776d4d29 2018-06-17 stsp err = got_object_open_as_tree(&next_tree, repo,
655 db37e2c0 2018-06-21 stsp te->id);
656 db37e2c0 2018-06-21 stsp te = NULL;
657 776d4d29 2018-06-17 stsp if (err)
658 776d4d29 2018-06-17 stsp goto done;
659 776d4d29 2018-06-17 stsp got_object_tree_close(tree);
660 776d4d29 2018-06-17 stsp tree = next_tree;
661 776d4d29 2018-06-17 stsp }
662 776d4d29 2018-06-17 stsp }
663 776d4d29 2018-06-17 stsp
664 ca008b32 2018-07-22 stsp if (te)
665 db37e2c0 2018-06-21 stsp err = got_object_open(obj, repo, te->id);
666 ca008b32 2018-07-22 stsp else
667 776d4d29 2018-06-17 stsp err = got_error(GOT_ERR_NO_OBJ);
668 776d4d29 2018-06-17 stsp done:
669 197aa481 2018-06-21 stsp free(s0);
670 776d4d29 2018-06-17 stsp if (commit)
671 776d4d29 2018-06-17 stsp got_object_commit_close(commit);
672 776d4d29 2018-06-17 stsp if (tree)
673 776d4d29 2018-06-17 stsp got_object_tree_close(tree);
674 776d4d29 2018-06-17 stsp return err;
675 68482ea3 2017-11-27 stsp }