1 a440fac0 2018-09-06 stsp /*
  2 a440fac0 2018-09-06 stsp  * Copyright (c) 2018 Stefan Sperling <stsp@openbsd.org>
  3 a440fac0 2018-09-06 stsp  *
  4 a440fac0 2018-09-06 stsp  * Permission to use, copy, modify, and distribute this software for any
  5 a440fac0 2018-09-06 stsp  * purpose with or without fee is hereby granted, provided that the above
  6 a440fac0 2018-09-06 stsp  * copyright notice and this permission notice appear in all copies.
  7 a440fac0 2018-09-06 stsp  *
  8 a440fac0 2018-09-06 stsp  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 a440fac0 2018-09-06 stsp  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 a440fac0 2018-09-06 stsp  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 a440fac0 2018-09-06 stsp  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 a440fac0 2018-09-06 stsp  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 a440fac0 2018-09-06 stsp  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 a440fac0 2018-09-06 stsp  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 a440fac0 2018-09-06 stsp  */
 16 a440fac0 2018-09-06 stsp 
 17 a440fac0 2018-09-06 stsp #include <sys/types.h>
 18 a440fac0 2018-09-06 stsp #include <sys/stat.h>
 19 a440fac0 2018-09-06 stsp #include <sys/queue.h>
 20 a440fac0 2018-09-06 stsp #include <sys/uio.h>
 21 a440fac0 2018-09-06 stsp #include <sys/socket.h>
 22 a440fac0 2018-09-06 stsp #include <sys/wait.h>
 23 a440fac0 2018-09-06 stsp 
 24 a440fac0 2018-09-06 stsp #include <errno.h>
 25 a440fac0 2018-09-06 stsp #include <stdio.h>
 26 a440fac0 2018-09-06 stsp #include <stdlib.h>
 27 a440fac0 2018-09-06 stsp #include <string.h>
 28 a440fac0 2018-09-06 stsp #include <stdint.h>
 29 a440fac0 2018-09-06 stsp #include <sha1.h>
 30 a440fac0 2018-09-06 stsp #include <zlib.h>
 31 a440fac0 2018-09-06 stsp #include <ctype.h>
 32 a440fac0 2018-09-06 stsp #include <limits.h>
 33 a440fac0 2018-09-06 stsp #include <imsg.h>
 34 a440fac0 2018-09-06 stsp #include <time.h>
 35 a440fac0 2018-09-06 stsp 
 36 a440fac0 2018-09-06 stsp #include "got_error.h"
 37 a440fac0 2018-09-06 stsp #include "got_object.h"
 38 a440fac0 2018-09-06 stsp #include "got_repository.h"
 39 a440fac0 2018-09-06 stsp #include "got_opentemp.h"
 40 a440fac0 2018-09-06 stsp 
 41 a440fac0 2018-09-06 stsp #include "got_lib_sha1.h"
 42 a440fac0 2018-09-06 stsp #include "got_lib_delta.h"
 43 a440fac0 2018-09-06 stsp #include "got_lib_pack.h"
 44 a440fac0 2018-09-06 stsp #include "got_lib_inflate.h"
 45 a440fac0 2018-09-06 stsp #include "got_lib_object.h"
 46 a440fac0 2018-09-06 stsp #include "got_lib_privsep.h"
 47 a440fac0 2018-09-06 stsp 
 48 a440fac0 2018-09-06 stsp #ifndef nitems
 49 a440fac0 2018-09-06 stsp #define nitems(_a) (sizeof(_a) / sizeof((_a)[0]))
 50 a440fac0 2018-09-06 stsp #endif
 51 a440fac0 2018-09-06 stsp 
 52 a440fac0 2018-09-06 stsp #define GOT_OBJ_TAG_COMMIT	"commit"
 53 a440fac0 2018-09-06 stsp #define GOT_OBJ_TAG_TREE	"tree"
 54 a440fac0 2018-09-06 stsp #define GOT_OBJ_TAG_BLOB	"blob"
 55 a440fac0 2018-09-06 stsp 
 56 a440fac0 2018-09-06 stsp #define GOT_COMMIT_TAG_TREE		"tree "
 57 a440fac0 2018-09-06 stsp #define GOT_COMMIT_TAG_PARENT		"parent "
 58 a440fac0 2018-09-06 stsp #define GOT_COMMIT_TAG_AUTHOR		"author "
 59 a440fac0 2018-09-06 stsp #define GOT_COMMIT_TAG_COMMITTER	"committer "
 60 a440fac0 2018-09-06 stsp 
 61 03fa71c8 2018-09-06 stsp void
 62 03fa71c8 2018-09-06 stsp got_object_close(struct got_object *obj)
 63 03fa71c8 2018-09-06 stsp {
 64 03fa71c8 2018-09-06 stsp 	if (obj->refcnt > 0) {
 65 03fa71c8 2018-09-06 stsp 		obj->refcnt--;
 66 03fa71c8 2018-09-06 stsp 		if (obj->refcnt > 0)
 67 03fa71c8 2018-09-06 stsp 			return;
 68 03fa71c8 2018-09-06 stsp 	}
 69 03fa71c8 2018-09-06 stsp 
 70 03fa71c8 2018-09-06 stsp 	if (obj->flags & GOT_OBJ_FLAG_DELTIFIED) {
 71 03fa71c8 2018-09-06 stsp 		struct got_delta *delta;
 72 03fa71c8 2018-09-06 stsp 		while (!SIMPLEQ_EMPTY(&obj->deltas.entries)) {
 73 03fa71c8 2018-09-06 stsp 			delta = SIMPLEQ_FIRST(&obj->deltas.entries);
 74 03fa71c8 2018-09-06 stsp 			SIMPLEQ_REMOVE_HEAD(&obj->deltas.entries, entry);
 75 03fa71c8 2018-09-06 stsp 			got_delta_close(delta);
 76 03fa71c8 2018-09-06 stsp 		}
 77 03fa71c8 2018-09-06 stsp 	}
 78 03fa71c8 2018-09-06 stsp 	if (obj->flags & GOT_OBJ_FLAG_PACKED)
 79 03fa71c8 2018-09-06 stsp 		free(obj->path_packfile);
 80 03fa71c8 2018-09-06 stsp 	free(obj);
 81 03fa71c8 2018-09-06 stsp }
 82 03fa71c8 2018-09-06 stsp 
 83 03fa71c8 2018-09-06 stsp void
 84 03fa71c8 2018-09-06 stsp got_object_qid_free(struct got_object_qid *qid)
 85 03fa71c8 2018-09-06 stsp {
 86 03fa71c8 2018-09-06 stsp 	free(qid->id);
 87 03fa71c8 2018-09-06 stsp 	free(qid);
 88 03fa71c8 2018-09-06 stsp }
 89 03fa71c8 2018-09-06 stsp 
 90 a440fac0 2018-09-06 stsp static const struct got_error *
 91 a440fac0 2018-09-06 stsp parse_object_header(struct got_object **obj, char *buf, size_t len)
 92 a440fac0 2018-09-06 stsp {
 93 a440fac0 2018-09-06 stsp 	const char *obj_tags[] = {
 94 a440fac0 2018-09-06 stsp 		GOT_OBJ_TAG_COMMIT,
 95 a440fac0 2018-09-06 stsp 		GOT_OBJ_TAG_TREE,
 96 a440fac0 2018-09-06 stsp 		GOT_OBJ_TAG_BLOB
 97 a440fac0 2018-09-06 stsp 	};
 98 a440fac0 2018-09-06 stsp 	const int obj_types[] = {
 99 a440fac0 2018-09-06 stsp 		GOT_OBJ_TYPE_COMMIT,
100 a440fac0 2018-09-06 stsp 		GOT_OBJ_TYPE_TREE,
101 a440fac0 2018-09-06 stsp 		GOT_OBJ_TYPE_BLOB,
102 a440fac0 2018-09-06 stsp 	};
103 a440fac0 2018-09-06 stsp 	int type = 0;
104 a440fac0 2018-09-06 stsp 	size_t size = 0, hdrlen = 0;
105 a440fac0 2018-09-06 stsp 	int i;
106 a440fac0 2018-09-06 stsp 	char *p = strchr(buf, '\0');
107 a440fac0 2018-09-06 stsp 
108 a440fac0 2018-09-06 stsp 	if (p == NULL)
109 a440fac0 2018-09-06 stsp 		return got_error(GOT_ERR_BAD_OBJ_HDR);
110 a440fac0 2018-09-06 stsp 
111 a440fac0 2018-09-06 stsp 	hdrlen = strlen(buf) + 1 /* '\0' */;
112 a440fac0 2018-09-06 stsp 
113 a440fac0 2018-09-06 stsp 	for (i = 0; i < nitems(obj_tags); i++) {
114 a440fac0 2018-09-06 stsp 		const char *tag = obj_tags[i];
115 a440fac0 2018-09-06 stsp 		size_t tlen = strlen(tag);
116 a440fac0 2018-09-06 stsp 		const char *errstr;
117 a440fac0 2018-09-06 stsp 
118 a440fac0 2018-09-06 stsp 		if (strncmp(buf, tag, tlen) != 0)
119 a440fac0 2018-09-06 stsp 			continue;
120 a440fac0 2018-09-06 stsp 
121 a440fac0 2018-09-06 stsp 		type = obj_types[i];
122 a440fac0 2018-09-06 stsp 		if (len <= tlen)
123 a440fac0 2018-09-06 stsp 			return got_error(GOT_ERR_BAD_OBJ_HDR);
124 a440fac0 2018-09-06 stsp 		size = strtonum(buf + tlen, 0, LONG_MAX, &errstr);
125 a440fac0 2018-09-06 stsp 		if (errstr != NULL)
126 a440fac0 2018-09-06 stsp 			return got_error(GOT_ERR_BAD_OBJ_HDR);
127 a440fac0 2018-09-06 stsp 		break;
128 a440fac0 2018-09-06 stsp 	}
129 a440fac0 2018-09-06 stsp 
130 a440fac0 2018-09-06 stsp 	if (type == 0)
131 a440fac0 2018-09-06 stsp 		return got_error(GOT_ERR_BAD_OBJ_HDR);
132 a440fac0 2018-09-06 stsp 
133 a440fac0 2018-09-06 stsp 	*obj = calloc(1, sizeof(**obj));
134 a440fac0 2018-09-06 stsp 	if (*obj == NULL)
135 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
136 a440fac0 2018-09-06 stsp 	(*obj)->type = type;
137 a440fac0 2018-09-06 stsp 	(*obj)->hdrlen = hdrlen;
138 a440fac0 2018-09-06 stsp 	(*obj)->size = size;
139 a440fac0 2018-09-06 stsp 	return NULL;
140 a440fac0 2018-09-06 stsp }
141 a440fac0 2018-09-06 stsp 
142 a440fac0 2018-09-06 stsp static const struct got_error *
143 a440fac0 2018-09-06 stsp read_object_header(struct got_object **obj, int fd)
144 a440fac0 2018-09-06 stsp {
145 a440fac0 2018-09-06 stsp 	const struct got_error *err;
146 a440fac0 2018-09-06 stsp 	struct got_zstream_buf zb;
147 a440fac0 2018-09-06 stsp 	char *buf;
148 a440fac0 2018-09-06 stsp 	const size_t zbsize = 64;
149 a440fac0 2018-09-06 stsp 	size_t outlen, totlen;
150 a440fac0 2018-09-06 stsp 	int nbuf = 1;
151 a440fac0 2018-09-06 stsp 
152 a440fac0 2018-09-06 stsp 	buf = malloc(zbsize);
153 a440fac0 2018-09-06 stsp 	if (buf == NULL)
154 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
155 a440fac0 2018-09-06 stsp 
156 a440fac0 2018-09-06 stsp 	err = got_inflate_init(&zb, buf, zbsize);
157 a440fac0 2018-09-06 stsp 	if (err)
158 a440fac0 2018-09-06 stsp 		return err;
159 a440fac0 2018-09-06 stsp 
160 a440fac0 2018-09-06 stsp 	totlen = 0;
161 a440fac0 2018-09-06 stsp 	do {
162 a440fac0 2018-09-06 stsp 		err = got_inflate_read_fd(&zb, fd, &outlen);
163 a440fac0 2018-09-06 stsp 		if (err)
164 a440fac0 2018-09-06 stsp 			goto done;
165 a440fac0 2018-09-06 stsp 		if (outlen == 0)
166 a440fac0 2018-09-06 stsp 			break;
167 a440fac0 2018-09-06 stsp 		totlen += outlen;
168 a440fac0 2018-09-06 stsp 		if (strchr(zb.outbuf, '\0') == NULL) {
169 a440fac0 2018-09-06 stsp 			char *newbuf;
170 a440fac0 2018-09-06 stsp 			nbuf++;
171 a440fac0 2018-09-06 stsp 			newbuf = recallocarray(buf, nbuf - 1, nbuf, zbsize);
172 a440fac0 2018-09-06 stsp 			if (newbuf == NULL) {
173 a440fac0 2018-09-06 stsp 				err = got_error_from_errno();
174 a440fac0 2018-09-06 stsp 				goto done;
175 a440fac0 2018-09-06 stsp 			}
176 a440fac0 2018-09-06 stsp 			buf = newbuf;
177 a440fac0 2018-09-06 stsp 			zb.outbuf = newbuf + totlen;
178 a440fac0 2018-09-06 stsp 			zb.outlen = (nbuf * zbsize) - totlen;
179 a440fac0 2018-09-06 stsp 		}
180 a440fac0 2018-09-06 stsp 	} while (strchr(zb.outbuf, '\0') == NULL);
181 a440fac0 2018-09-06 stsp 
182 a440fac0 2018-09-06 stsp 	err = parse_object_header(obj, buf, totlen);
183 a440fac0 2018-09-06 stsp done:
184 a440fac0 2018-09-06 stsp 	free(buf);
185 a440fac0 2018-09-06 stsp 	got_inflate_end(&zb);
186 a440fac0 2018-09-06 stsp 	return err;
187 a440fac0 2018-09-06 stsp }
188 a440fac0 2018-09-06 stsp 
189 a440fac0 2018-09-06 stsp static void
190 a440fac0 2018-09-06 stsp read_object_header_privsep_child(int obj_fd, int imsg_fds[2])
191 a440fac0 2018-09-06 stsp {
192 a440fac0 2018-09-06 stsp 	const struct got_error *err = NULL;
193 a440fac0 2018-09-06 stsp 	struct got_object *obj = NULL;
194 a440fac0 2018-09-06 stsp 	struct imsgbuf ibuf;
195 a440fac0 2018-09-06 stsp 	int status = 0;
196 a440fac0 2018-09-06 stsp 
197 a440fac0 2018-09-06 stsp 	setproctitle("read object header");
198 a440fac0 2018-09-06 stsp 	close(imsg_fds[0]);
199 a440fac0 2018-09-06 stsp 	imsg_init(&ibuf, imsg_fds[1]);
200 a440fac0 2018-09-06 stsp 
201 a440fac0 2018-09-06 stsp 	/* revoke access to most system calls */
202 a440fac0 2018-09-06 stsp 	if (pledge("stdio", NULL) == -1) {
203 a440fac0 2018-09-06 stsp 		err = got_error_from_errno();
204 a440fac0 2018-09-06 stsp 		goto done;
205 a440fac0 2018-09-06 stsp 	}
206 a440fac0 2018-09-06 stsp 
207 a440fac0 2018-09-06 stsp 	err = read_object_header(&obj, obj_fd);
208 a440fac0 2018-09-06 stsp 	if (err)
209 a440fac0 2018-09-06 stsp 		goto done;
210 a440fac0 2018-09-06 stsp 
211 a440fac0 2018-09-06 stsp 	err = got_privsep_send_obj(&ibuf, obj, 0);
212 a440fac0 2018-09-06 stsp done:
213 a440fac0 2018-09-06 stsp 	if (obj)
214 a440fac0 2018-09-06 stsp 		got_object_close(obj);
215 a440fac0 2018-09-06 stsp 	if (err) {
216 a440fac0 2018-09-06 stsp 		got_privsep_send_error(&ibuf, err);
217 a440fac0 2018-09-06 stsp 		status = 1;
218 a440fac0 2018-09-06 stsp 	}
219 a440fac0 2018-09-06 stsp 	close(obj_fd);
220 a440fac0 2018-09-06 stsp 	imsg_clear(&ibuf);
221 a440fac0 2018-09-06 stsp 	close(imsg_fds[1]);
222 a440fac0 2018-09-06 stsp 	_exit(status);
223 a440fac0 2018-09-06 stsp }
224 a440fac0 2018-09-06 stsp 
225 a440fac0 2018-09-06 stsp static const struct got_error *
226 a440fac0 2018-09-06 stsp wait_for_child(pid_t pid)
227 a440fac0 2018-09-06 stsp {
228 a440fac0 2018-09-06 stsp 	int child_status;
229 a440fac0 2018-09-06 stsp 
230 a440fac0 2018-09-06 stsp 	waitpid(pid, &child_status, 0);
231 a440fac0 2018-09-06 stsp 
232 a440fac0 2018-09-06 stsp 	if (!WIFEXITED(child_status))
233 a440fac0 2018-09-06 stsp 		return got_error(GOT_ERR_PRIVSEP_DIED);
234 a440fac0 2018-09-06 stsp 
235 a440fac0 2018-09-06 stsp 	if (WEXITSTATUS(child_status) != 0)
236 a440fac0 2018-09-06 stsp 		return got_error(GOT_ERR_PRIVSEP_EXIT);
237 a440fac0 2018-09-06 stsp 
238 a440fac0 2018-09-06 stsp 	return NULL;
239 a440fac0 2018-09-06 stsp }
240 a440fac0 2018-09-06 stsp 
241 a440fac0 2018-09-06 stsp const struct got_error *
242 a440fac0 2018-09-06 stsp got_object_read_header_privsep(struct got_object **obj, int fd)
243 a440fac0 2018-09-06 stsp {
244 a440fac0 2018-09-06 stsp 	struct imsgbuf parent_ibuf;
245 a440fac0 2018-09-06 stsp 	int imsg_fds[2];
246 a440fac0 2018-09-06 stsp 	const struct got_error *err = NULL, *err_child = NULL;
247 a440fac0 2018-09-06 stsp 	pid_t pid;
248 a440fac0 2018-09-06 stsp 
249 a440fac0 2018-09-06 stsp 	if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, imsg_fds) == -1)
250 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
251 a440fac0 2018-09-06 stsp 
252 a440fac0 2018-09-06 stsp 	pid = fork();
253 a440fac0 2018-09-06 stsp 	if (pid == -1)
254 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
255 a440fac0 2018-09-06 stsp 	else if (pid == 0) {
256 a440fac0 2018-09-06 stsp 		read_object_header_privsep_child(fd, imsg_fds);
257 a440fac0 2018-09-06 stsp 		/* not reached */
258 a440fac0 2018-09-06 stsp 	}
259 a440fac0 2018-09-06 stsp 
260 a440fac0 2018-09-06 stsp 	close(imsg_fds[1]);
261 a440fac0 2018-09-06 stsp 	imsg_init(&parent_ibuf, imsg_fds[0]);
262 a440fac0 2018-09-06 stsp 	err = got_privsep_recv_obj(obj, &parent_ibuf);
263 a440fac0 2018-09-06 stsp 	imsg_clear(&parent_ibuf);
264 a440fac0 2018-09-06 stsp 	err_child = wait_for_child(pid);
265 a440fac0 2018-09-06 stsp 	close(imsg_fds[0]);
266 a440fac0 2018-09-06 stsp 	return err ? err : err_child;
267 a440fac0 2018-09-06 stsp }
268 a440fac0 2018-09-06 stsp 
269 a440fac0 2018-09-06 stsp struct got_commit_object *
270 a440fac0 2018-09-06 stsp got_object_commit_alloc_partial(void)
271 a440fac0 2018-09-06 stsp {
272 a440fac0 2018-09-06 stsp 	struct got_commit_object *commit;
273 a440fac0 2018-09-06 stsp 
274 a440fac0 2018-09-06 stsp 	commit = calloc(1, sizeof(*commit));
275 a440fac0 2018-09-06 stsp 	if (commit == NULL)
276 a440fac0 2018-09-06 stsp 		return NULL;
277 a440fac0 2018-09-06 stsp 	commit->tree_id = calloc(1, sizeof(*commit->tree_id));
278 a440fac0 2018-09-06 stsp 	if (commit->tree_id == NULL) {
279 a440fac0 2018-09-06 stsp 		free(commit);
280 a440fac0 2018-09-06 stsp 		return NULL;
281 a440fac0 2018-09-06 stsp 	}
282 a440fac0 2018-09-06 stsp 
283 a440fac0 2018-09-06 stsp 	SIMPLEQ_INIT(&commit->parent_ids);
284 a440fac0 2018-09-06 stsp 
285 a440fac0 2018-09-06 stsp 	return commit;
286 a440fac0 2018-09-06 stsp }
287 a440fac0 2018-09-06 stsp 
288 a440fac0 2018-09-06 stsp const struct got_error *
289 a440fac0 2018-09-06 stsp got_object_commit_add_parent(struct got_commit_object *commit,
290 a440fac0 2018-09-06 stsp     const char *id_str)
291 a440fac0 2018-09-06 stsp {
292 a440fac0 2018-09-06 stsp 	const struct got_error *err = NULL;
293 a440fac0 2018-09-06 stsp 	struct got_object_qid *qid;
294 a440fac0 2018-09-06 stsp 
295 a440fac0 2018-09-06 stsp 	qid = malloc(sizeof(*qid));
296 a440fac0 2018-09-06 stsp 	if (qid == NULL)
297 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
298 a440fac0 2018-09-06 stsp 
299 a440fac0 2018-09-06 stsp 	qid->id = malloc(sizeof(*qid->id));
300 a440fac0 2018-09-06 stsp 	if (qid->id == NULL) {
301 a440fac0 2018-09-06 stsp 		err = got_error_from_errno();
302 a440fac0 2018-09-06 stsp 		got_object_qid_free(qid);
303 a440fac0 2018-09-06 stsp 		return err;
304 a440fac0 2018-09-06 stsp 	}
305 a440fac0 2018-09-06 stsp 
306 a440fac0 2018-09-06 stsp 	if (!got_parse_sha1_digest(qid->id->sha1, id_str)) {
307 a440fac0 2018-09-06 stsp 		err = got_error(GOT_ERR_BAD_OBJ_DATA);
308 a440fac0 2018-09-06 stsp 		free(qid->id);
309 a440fac0 2018-09-06 stsp 		free(qid);
310 a440fac0 2018-09-06 stsp 		return err;
311 a440fac0 2018-09-06 stsp 	}
312 a440fac0 2018-09-06 stsp 
313 a440fac0 2018-09-06 stsp 	SIMPLEQ_INSERT_TAIL(&commit->parent_ids, qid, entry);
314 a440fac0 2018-09-06 stsp 	commit->nparents++;
315 a440fac0 2018-09-06 stsp 
316 a440fac0 2018-09-06 stsp 	return NULL;
317 a440fac0 2018-09-06 stsp }
318 a440fac0 2018-09-06 stsp 
319 a440fac0 2018-09-06 stsp static const struct got_error *
320 a440fac0 2018-09-06 stsp parse_gmtoff(time_t *gmtoff, const char *tzstr)
321 a440fac0 2018-09-06 stsp {
322 a440fac0 2018-09-06 stsp 	int sign = 1;
323 a440fac0 2018-09-06 stsp 	const char *p = tzstr;
324 a440fac0 2018-09-06 stsp 	time_t h, m;
325 a440fac0 2018-09-06 stsp 
326 a440fac0 2018-09-06 stsp 	*gmtoff = 0;
327 a440fac0 2018-09-06 stsp 
328 a440fac0 2018-09-06 stsp 	if (*p == '-')
329 a440fac0 2018-09-06 stsp 		sign = -1;
330 a440fac0 2018-09-06 stsp 	else if (*p != '+')
331 a440fac0 2018-09-06 stsp 		return got_error(GOT_ERR_BAD_OBJ_DATA);
332 a440fac0 2018-09-06 stsp 	p++;
333 a440fac0 2018-09-06 stsp 	if (!isdigit(*p) && !isdigit(*(p + 1)))
334 a440fac0 2018-09-06 stsp 		return got_error(GOT_ERR_BAD_OBJ_DATA);
335 a440fac0 2018-09-06 stsp 	h = (((*p - '0') * 10) + (*(p + 1) - '0'));
336 a440fac0 2018-09-06 stsp 
337 a440fac0 2018-09-06 stsp 	p += 2;
338 a440fac0 2018-09-06 stsp 	if (!isdigit(*p) && !isdigit(*(p + 1)))
339 a440fac0 2018-09-06 stsp 		return got_error(GOT_ERR_BAD_OBJ_DATA);
340 a440fac0 2018-09-06 stsp 	m = ((*p - '0') * 10) + (*(p + 1) - '0');
341 a440fac0 2018-09-06 stsp 
342 a440fac0 2018-09-06 stsp 	*gmtoff = (h * 60 * 60 + m * 60) * sign;
343 a440fac0 2018-09-06 stsp 	return NULL;
344 a440fac0 2018-09-06 stsp }
345 a440fac0 2018-09-06 stsp 
346 a440fac0 2018-09-06 stsp static const struct got_error *
347 a440fac0 2018-09-06 stsp parse_commit_time(struct tm *tm, char *committer)
348 a440fac0 2018-09-06 stsp {
349 a440fac0 2018-09-06 stsp 	const struct got_error *err = NULL;
350 a440fac0 2018-09-06 stsp 	const char *errstr;
351 a440fac0 2018-09-06 stsp 	char *space, *tzstr;
352 a440fac0 2018-09-06 stsp 	time_t gmtoff;
353 a440fac0 2018-09-06 stsp 	time_t time;
354 a440fac0 2018-09-06 stsp 
355 a440fac0 2018-09-06 stsp 	/* Parse and strip off trailing timezone indicator string. */
356 a440fac0 2018-09-06 stsp 	space = strrchr(committer, ' ');
357 a440fac0 2018-09-06 stsp 	if (space == NULL)
358 a440fac0 2018-09-06 stsp 		return got_error(GOT_ERR_BAD_OBJ_DATA);
359 a440fac0 2018-09-06 stsp 	tzstr = strdup(space + 1);
360 a440fac0 2018-09-06 stsp 	if (tzstr == NULL)
361 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
362 a440fac0 2018-09-06 stsp 	err = parse_gmtoff(&gmtoff, tzstr);
363 a440fac0 2018-09-06 stsp 	free(tzstr);
364 a440fac0 2018-09-06 stsp 	if (err)
365 a440fac0 2018-09-06 stsp 		return err;
366 a440fac0 2018-09-06 stsp 	*space = '\0';
367 a440fac0 2018-09-06 stsp 
368 a440fac0 2018-09-06 stsp 	/* Timestamp is separated from committer name + email by space. */
369 a440fac0 2018-09-06 stsp 	space = strrchr(committer, ' ');
370 a440fac0 2018-09-06 stsp 	if (space == NULL)
371 a440fac0 2018-09-06 stsp 		return got_error(GOT_ERR_BAD_OBJ_DATA);
372 a440fac0 2018-09-06 stsp 
373 a440fac0 2018-09-06 stsp 	/* Timestamp parsed here is expressed in comitter's local time. */
374 a440fac0 2018-09-06 stsp 	time = strtonum(space + 1, 0, INT64_MAX, &errstr);
375 a440fac0 2018-09-06 stsp 	if (errstr)
376 a440fac0 2018-09-06 stsp 		return got_error(GOT_ERR_BAD_OBJ_DATA);
377 a440fac0 2018-09-06 stsp 
378 a440fac0 2018-09-06 stsp 	/* Express the time stamp in UTC. */
379 a440fac0 2018-09-06 stsp 	memset(tm, 0, sizeof(*tm));
380 a440fac0 2018-09-06 stsp 	time -= gmtoff;
381 a440fac0 2018-09-06 stsp 	if (localtime_r(&time, tm) == NULL)
382 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
383 a440fac0 2018-09-06 stsp 	tm->tm_gmtoff = gmtoff;
384 a440fac0 2018-09-06 stsp 
385 a440fac0 2018-09-06 stsp 	/* Strip off parsed time information, leaving just author and email. */
386 a440fac0 2018-09-06 stsp 	*space = '\0';
387 a440fac0 2018-09-06 stsp 
388 a440fac0 2018-09-06 stsp 	return NULL;
389 a440fac0 2018-09-06 stsp }
390 a440fac0 2018-09-06 stsp 
391 03fa71c8 2018-09-06 stsp void
392 03fa71c8 2018-09-06 stsp got_object_commit_close(struct got_commit_object *commit)
393 03fa71c8 2018-09-06 stsp {
394 03fa71c8 2018-09-06 stsp 	struct got_object_qid *qid;
395 03fa71c8 2018-09-06 stsp 
396 03fa71c8 2018-09-06 stsp 	if (commit->refcnt > 0) {
397 03fa71c8 2018-09-06 stsp 		commit->refcnt--;
398 03fa71c8 2018-09-06 stsp 		if (commit->refcnt > 0)
399 03fa71c8 2018-09-06 stsp 			return;
400 03fa71c8 2018-09-06 stsp 	}
401 03fa71c8 2018-09-06 stsp 
402 03fa71c8 2018-09-06 stsp 	while (!SIMPLEQ_EMPTY(&commit->parent_ids)) {
403 03fa71c8 2018-09-06 stsp 		qid = SIMPLEQ_FIRST(&commit->parent_ids);
404 03fa71c8 2018-09-06 stsp 		SIMPLEQ_REMOVE_HEAD(&commit->parent_ids, entry);
405 03fa71c8 2018-09-06 stsp 		got_object_qid_free(qid);
406 03fa71c8 2018-09-06 stsp 	}
407 03fa71c8 2018-09-06 stsp 
408 03fa71c8 2018-09-06 stsp 	free(commit->tree_id);
409 03fa71c8 2018-09-06 stsp 	free(commit->author);
410 03fa71c8 2018-09-06 stsp 	free(commit->committer);
411 03fa71c8 2018-09-06 stsp 	free(commit->logmsg);
412 03fa71c8 2018-09-06 stsp 	free(commit);
413 03fa71c8 2018-09-06 stsp }
414 03fa71c8 2018-09-06 stsp 
415 a440fac0 2018-09-06 stsp const struct got_error *
416 a440fac0 2018-09-06 stsp got_object_parse_commit(struct got_commit_object **commit, char *buf, size_t len)
417 a440fac0 2018-09-06 stsp {
418 a440fac0 2018-09-06 stsp 	const struct got_error *err = NULL;
419 a440fac0 2018-09-06 stsp 	char *s = buf;
420 a440fac0 2018-09-06 stsp 	size_t tlen;
421 a440fac0 2018-09-06 stsp 	ssize_t remain = (ssize_t)len;
422 a440fac0 2018-09-06 stsp  
423 a440fac0 2018-09-06 stsp 	*commit = got_object_commit_alloc_partial();
424 a440fac0 2018-09-06 stsp 	if (*commit == NULL)
425 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
426 a440fac0 2018-09-06 stsp 
427 a440fac0 2018-09-06 stsp 	tlen = strlen(GOT_COMMIT_TAG_TREE);
428 a440fac0 2018-09-06 stsp 	if (strncmp(s, GOT_COMMIT_TAG_TREE, tlen) == 0) {
429 a440fac0 2018-09-06 stsp 		remain -= tlen;
430 a440fac0 2018-09-06 stsp 		if (remain < SHA1_DIGEST_STRING_LENGTH) {
431 a440fac0 2018-09-06 stsp 			err = got_error(GOT_ERR_BAD_OBJ_DATA);
432 a440fac0 2018-09-06 stsp 			goto done;
433 a440fac0 2018-09-06 stsp 		}
434 a440fac0 2018-09-06 stsp 		s += tlen;
435 a440fac0 2018-09-06 stsp 		if (!got_parse_sha1_digest((*commit)->tree_id->sha1, s)) {
436 a440fac0 2018-09-06 stsp 			err = got_error(GOT_ERR_BAD_OBJ_DATA);
437 a440fac0 2018-09-06 stsp 			goto done;
438 a440fac0 2018-09-06 stsp 		}
439 a440fac0 2018-09-06 stsp 		remain -= SHA1_DIGEST_STRING_LENGTH;
440 a440fac0 2018-09-06 stsp 		s += SHA1_DIGEST_STRING_LENGTH;
441 a440fac0 2018-09-06 stsp 	} else {
442 a440fac0 2018-09-06 stsp 		err = got_error(GOT_ERR_BAD_OBJ_DATA);
443 a440fac0 2018-09-06 stsp 		goto done;
444 a440fac0 2018-09-06 stsp 	}
445 a440fac0 2018-09-06 stsp 
446 a440fac0 2018-09-06 stsp 	tlen = strlen(GOT_COMMIT_TAG_PARENT);
447 a440fac0 2018-09-06 stsp 	while (strncmp(s, GOT_COMMIT_TAG_PARENT, tlen) == 0) {
448 a440fac0 2018-09-06 stsp 		remain -= tlen;
449 a440fac0 2018-09-06 stsp 		if (remain < SHA1_DIGEST_STRING_LENGTH) {
450 a440fac0 2018-09-06 stsp 			err = got_error(GOT_ERR_BAD_OBJ_DATA);
451 a440fac0 2018-09-06 stsp 			goto done;
452 a440fac0 2018-09-06 stsp 		}
453 a440fac0 2018-09-06 stsp 		s += tlen;
454 a440fac0 2018-09-06 stsp 		err = got_object_commit_add_parent(*commit, s);
455 a440fac0 2018-09-06 stsp 		if (err)
456 a440fac0 2018-09-06 stsp 			goto done;
457 a440fac0 2018-09-06 stsp 
458 a440fac0 2018-09-06 stsp 		remain -= SHA1_DIGEST_STRING_LENGTH;
459 a440fac0 2018-09-06 stsp 		s += SHA1_DIGEST_STRING_LENGTH;
460 a440fac0 2018-09-06 stsp 	}
461 a440fac0 2018-09-06 stsp 
462 a440fac0 2018-09-06 stsp 	tlen = strlen(GOT_COMMIT_TAG_AUTHOR);
463 a440fac0 2018-09-06 stsp 	if (strncmp(s, GOT_COMMIT_TAG_AUTHOR, tlen) == 0) {
464 a440fac0 2018-09-06 stsp 		char *p;
465 a440fac0 2018-09-06 stsp 		size_t slen;
466 a440fac0 2018-09-06 stsp 
467 a440fac0 2018-09-06 stsp 		remain -= tlen;
468 a440fac0 2018-09-06 stsp 		if (remain <= 0) {
469 a440fac0 2018-09-06 stsp 			err = got_error(GOT_ERR_BAD_OBJ_DATA);
470 a440fac0 2018-09-06 stsp 			goto done;
471 a440fac0 2018-09-06 stsp 		}
472 a440fac0 2018-09-06 stsp 		s += tlen;
473 a440fac0 2018-09-06 stsp 		p = strchr(s, '\n');
474 a440fac0 2018-09-06 stsp 		if (p == NULL) {
475 a440fac0 2018-09-06 stsp 			err = got_error(GOT_ERR_BAD_OBJ_DATA);
476 a440fac0 2018-09-06 stsp 			goto done;
477 a440fac0 2018-09-06 stsp 		}
478 a440fac0 2018-09-06 stsp 		*p = '\0';
479 a440fac0 2018-09-06 stsp 		slen = strlen(s);
480 a440fac0 2018-09-06 stsp 		err = parse_commit_time(&(*commit)->tm_author, s);
481 a440fac0 2018-09-06 stsp 		if (err)
482 a440fac0 2018-09-06 stsp 			goto done;
483 a440fac0 2018-09-06 stsp 		(*commit)->author = strdup(s);
484 a440fac0 2018-09-06 stsp 		if ((*commit)->author == NULL) {
485 a440fac0 2018-09-06 stsp 			err = got_error_from_errno();
486 a440fac0 2018-09-06 stsp 			goto done;
487 a440fac0 2018-09-06 stsp 		}
488 a440fac0 2018-09-06 stsp 		s += slen + 1;
489 a440fac0 2018-09-06 stsp 		remain -= slen + 1;
490 a440fac0 2018-09-06 stsp 	}
491 a440fac0 2018-09-06 stsp 
492 a440fac0 2018-09-06 stsp 	tlen = strlen(GOT_COMMIT_TAG_COMMITTER);
493 a440fac0 2018-09-06 stsp 	if (strncmp(s, GOT_COMMIT_TAG_COMMITTER, tlen) == 0) {
494 a440fac0 2018-09-06 stsp 		char *p;
495 a440fac0 2018-09-06 stsp 		size_t slen;
496 a440fac0 2018-09-06 stsp 
497 a440fac0 2018-09-06 stsp 		remain -= tlen;
498 a440fac0 2018-09-06 stsp 		if (remain <= 0) {
499 a440fac0 2018-09-06 stsp 			err = got_error(GOT_ERR_BAD_OBJ_DATA);
500 a440fac0 2018-09-06 stsp 			goto done;
501 a440fac0 2018-09-06 stsp 		}
502 a440fac0 2018-09-06 stsp 		s += tlen;
503 a440fac0 2018-09-06 stsp 		p = strchr(s, '\n');
504 a440fac0 2018-09-06 stsp 		if (p == NULL) {
505 a440fac0 2018-09-06 stsp 			err = got_error(GOT_ERR_BAD_OBJ_DATA);
506 a440fac0 2018-09-06 stsp 			goto done;
507 a440fac0 2018-09-06 stsp 		}
508 a440fac0 2018-09-06 stsp 		*p = '\0';
509 a440fac0 2018-09-06 stsp 		slen = strlen(s);
510 a440fac0 2018-09-06 stsp 		err = parse_commit_time(&(*commit)->tm_committer, s);
511 a440fac0 2018-09-06 stsp 		if (err)
512 a440fac0 2018-09-06 stsp 			goto done;
513 a440fac0 2018-09-06 stsp 		(*commit)->committer = strdup(s);
514 a440fac0 2018-09-06 stsp 		if ((*commit)->committer == NULL) {
515 a440fac0 2018-09-06 stsp 			err = got_error_from_errno();
516 a440fac0 2018-09-06 stsp 			goto done;
517 a440fac0 2018-09-06 stsp 		}
518 a440fac0 2018-09-06 stsp 		s += slen + 1;
519 a440fac0 2018-09-06 stsp 		remain -= slen + 1;
520 a440fac0 2018-09-06 stsp 	}
521 a440fac0 2018-09-06 stsp 
522 a440fac0 2018-09-06 stsp 	(*commit)->logmsg = strndup(s, remain);
523 a440fac0 2018-09-06 stsp 	if ((*commit)->logmsg == NULL) {
524 a440fac0 2018-09-06 stsp 		err = got_error_from_errno();
525 a440fac0 2018-09-06 stsp 		goto done;
526 a440fac0 2018-09-06 stsp 	}
527 a440fac0 2018-09-06 stsp done:
528 a440fac0 2018-09-06 stsp 	if (err) {
529 a440fac0 2018-09-06 stsp 		got_object_commit_close(*commit);
530 a440fac0 2018-09-06 stsp 		*commit = NULL;
531 a440fac0 2018-09-06 stsp 	}
532 a440fac0 2018-09-06 stsp 	return err;
533 a440fac0 2018-09-06 stsp }
534 a440fac0 2018-09-06 stsp 
535 03fa71c8 2018-09-06 stsp static void
536 03fa71c8 2018-09-06 stsp tree_entry_close(struct got_tree_entry *te)
537 a440fac0 2018-09-06 stsp {
538 a440fac0 2018-09-06 stsp 	free(te->id);
539 a440fac0 2018-09-06 stsp 	free(te->name);
540 a440fac0 2018-09-06 stsp 	free(te);
541 a440fac0 2018-09-06 stsp }
542 a440fac0 2018-09-06 stsp 
543 03fa71c8 2018-09-06 stsp void
544 03fa71c8 2018-09-06 stsp got_object_tree_close(struct got_tree_object *tree)
545 03fa71c8 2018-09-06 stsp {
546 03fa71c8 2018-09-06 stsp 	struct got_tree_entry *te;
547 03fa71c8 2018-09-06 stsp 
548 03fa71c8 2018-09-06 stsp 	if (tree->refcnt > 0) {
549 03fa71c8 2018-09-06 stsp 		tree->refcnt--;
550 03fa71c8 2018-09-06 stsp 		if (tree->refcnt > 0)
551 03fa71c8 2018-09-06 stsp 			return;
552 03fa71c8 2018-09-06 stsp 	}
553 03fa71c8 2018-09-06 stsp 
554 03fa71c8 2018-09-06 stsp 	while (!SIMPLEQ_EMPTY(&tree->entries.head)) {
555 03fa71c8 2018-09-06 stsp 		te = SIMPLEQ_FIRST(&tree->entries.head);
556 03fa71c8 2018-09-06 stsp 		SIMPLEQ_REMOVE_HEAD(&tree->entries.head, entry);
557 03fa71c8 2018-09-06 stsp 		tree_entry_close(te);
558 03fa71c8 2018-09-06 stsp 	}
559 03fa71c8 2018-09-06 stsp 
560 03fa71c8 2018-09-06 stsp 	free(tree);
561 03fa71c8 2018-09-06 stsp }
562 03fa71c8 2018-09-06 stsp 
563 a440fac0 2018-09-06 stsp struct got_tree_entry *
564 a440fac0 2018-09-06 stsp got_alloc_tree_entry_partial(void)
565 a440fac0 2018-09-06 stsp {
566 a440fac0 2018-09-06 stsp 	struct got_tree_entry *te;
567 a440fac0 2018-09-06 stsp 
568 a440fac0 2018-09-06 stsp 	te = calloc(1, sizeof(*te));
569 a440fac0 2018-09-06 stsp 	if (te == NULL)
570 a440fac0 2018-09-06 stsp 		return NULL;
571 a440fac0 2018-09-06 stsp 
572 a440fac0 2018-09-06 stsp 	te->id = calloc(1, sizeof(*te->id));
573 a440fac0 2018-09-06 stsp 	if (te->id == NULL) {
574 a440fac0 2018-09-06 stsp 		free(te);
575 a440fac0 2018-09-06 stsp 		te = NULL;
576 a440fac0 2018-09-06 stsp 	}
577 a440fac0 2018-09-06 stsp 	return te;
578 a440fac0 2018-09-06 stsp }
579 a440fac0 2018-09-06 stsp 
580 a440fac0 2018-09-06 stsp static const struct got_error *
581 a440fac0 2018-09-06 stsp parse_tree_entry(struct got_tree_entry **te, size_t *elen, char *buf,
582 a440fac0 2018-09-06 stsp     size_t maxlen)
583 a440fac0 2018-09-06 stsp {
584 a440fac0 2018-09-06 stsp 	char *p = buf, *space;
585 a440fac0 2018-09-06 stsp 	const struct got_error *err = NULL;
586 a440fac0 2018-09-06 stsp 
587 a440fac0 2018-09-06 stsp 	*te = got_alloc_tree_entry_partial();
588 a440fac0 2018-09-06 stsp 	if (*te == NULL)
589 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
590 a440fac0 2018-09-06 stsp 
591 a440fac0 2018-09-06 stsp 	*elen = strlen(buf) + 1;
592 a440fac0 2018-09-06 stsp 	if (*elen > maxlen) {
593 a440fac0 2018-09-06 stsp 		free(*te);
594 a440fac0 2018-09-06 stsp 		*te = NULL;
595 a440fac0 2018-09-06 stsp 		return got_error(GOT_ERR_BAD_OBJ_DATA);
596 a440fac0 2018-09-06 stsp 	}
597 a440fac0 2018-09-06 stsp 
598 a440fac0 2018-09-06 stsp 	space = strchr(buf, ' ');
599 a440fac0 2018-09-06 stsp 	if (space == NULL) {
600 a440fac0 2018-09-06 stsp 		err = got_error(GOT_ERR_BAD_OBJ_DATA);
601 a440fac0 2018-09-06 stsp 		free(*te);
602 a440fac0 2018-09-06 stsp 		*te = NULL;
603 a440fac0 2018-09-06 stsp 		return err;
604 a440fac0 2018-09-06 stsp 	}
605 a440fac0 2018-09-06 stsp 	while (*p != ' ') {
606 a440fac0 2018-09-06 stsp 		if (*p < '0' && *p > '7') {
607 a440fac0 2018-09-06 stsp 			err = got_error(GOT_ERR_BAD_OBJ_DATA);
608 a440fac0 2018-09-06 stsp 			goto done;
609 a440fac0 2018-09-06 stsp 		}
610 a440fac0 2018-09-06 stsp 		(*te)->mode <<= 3;
611 a440fac0 2018-09-06 stsp 		(*te)->mode |= *p - '0';
612 a440fac0 2018-09-06 stsp 		p++;
613 a440fac0 2018-09-06 stsp 	}
614 a440fac0 2018-09-06 stsp 
615 a440fac0 2018-09-06 stsp 	(*te)->name = strdup(space + 1);
616 a440fac0 2018-09-06 stsp 	if (*elen > maxlen || maxlen - *elen < SHA1_DIGEST_LENGTH) {
617 a440fac0 2018-09-06 stsp 		err = got_error(GOT_ERR_BAD_OBJ_DATA);
618 a440fac0 2018-09-06 stsp 		goto done;
619 a440fac0 2018-09-06 stsp 	}
620 a440fac0 2018-09-06 stsp 	buf += strlen(buf) + 1;
621 a440fac0 2018-09-06 stsp 	memcpy((*te)->id->sha1, buf, SHA1_DIGEST_LENGTH);
622 a440fac0 2018-09-06 stsp 	*elen += SHA1_DIGEST_LENGTH;
623 a440fac0 2018-09-06 stsp done:
624 a440fac0 2018-09-06 stsp 	if (err) {
625 03fa71c8 2018-09-06 stsp 		tree_entry_close(*te);
626 a440fac0 2018-09-06 stsp 		*te = NULL;
627 a440fac0 2018-09-06 stsp 	}
628 a440fac0 2018-09-06 stsp 	return err;
629 a440fac0 2018-09-06 stsp }
630 a440fac0 2018-09-06 stsp 
631 a440fac0 2018-09-06 stsp const struct got_error *
632 a440fac0 2018-09-06 stsp got_object_parse_tree(struct got_tree_object **tree, uint8_t *buf, size_t len)
633 a440fac0 2018-09-06 stsp {
634 a440fac0 2018-09-06 stsp 	const struct got_error *err;
635 a440fac0 2018-09-06 stsp 	size_t remain = len;
636 a440fac0 2018-09-06 stsp 
637 a440fac0 2018-09-06 stsp 	*tree = calloc(1, sizeof(**tree));
638 a440fac0 2018-09-06 stsp 	if (*tree == NULL)
639 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
640 a440fac0 2018-09-06 stsp 
641 a440fac0 2018-09-06 stsp 	SIMPLEQ_INIT(&(*tree)->entries.head);
642 a440fac0 2018-09-06 stsp 
643 a440fac0 2018-09-06 stsp 	while (remain > 0) {
644 a440fac0 2018-09-06 stsp 		struct got_tree_entry *te;
645 a440fac0 2018-09-06 stsp 		size_t elen;
646 a440fac0 2018-09-06 stsp 
647 a440fac0 2018-09-06 stsp 		err = parse_tree_entry(&te, &elen, buf, remain);
648 a440fac0 2018-09-06 stsp 		if (err)
649 a440fac0 2018-09-06 stsp 			return err;
650 a440fac0 2018-09-06 stsp 		(*tree)->entries.nentries++;
651 a440fac0 2018-09-06 stsp 		SIMPLEQ_INSERT_TAIL(&(*tree)->entries.head, te, entry);
652 a440fac0 2018-09-06 stsp 		buf += elen;
653 a440fac0 2018-09-06 stsp 		remain -= elen;
654 a440fac0 2018-09-06 stsp 	}
655 a440fac0 2018-09-06 stsp 
656 a440fac0 2018-09-06 stsp 	if (remain != 0) {
657 a440fac0 2018-09-06 stsp 		got_object_tree_close(*tree);
658 a440fac0 2018-09-06 stsp 		return got_error(GOT_ERR_BAD_OBJ_DATA);
659 a440fac0 2018-09-06 stsp 	}
660 a440fac0 2018-09-06 stsp 
661 a440fac0 2018-09-06 stsp 	return NULL;
662 a440fac0 2018-09-06 stsp }
663 a440fac0 2018-09-06 stsp 
664 a440fac0 2018-09-06 stsp static const struct got_error *
665 a440fac0 2018-09-06 stsp read_to_mem(uint8_t **outbuf, size_t *outlen, FILE *f)
666 a440fac0 2018-09-06 stsp {
667 a440fac0 2018-09-06 stsp 	const struct got_error *err = NULL;
668 a440fac0 2018-09-06 stsp 	static const size_t blocksize = 512;
669 a440fac0 2018-09-06 stsp 	size_t n, total, remain;
670 a440fac0 2018-09-06 stsp 	uint8_t *buf;
671 a440fac0 2018-09-06 stsp 
672 a440fac0 2018-09-06 stsp 	*outbuf = NULL;
673 a440fac0 2018-09-06 stsp 	*outlen = 0;
674 a440fac0 2018-09-06 stsp 
675 a440fac0 2018-09-06 stsp 	buf = malloc(blocksize);
676 a440fac0 2018-09-06 stsp 	if (buf == NULL)
677 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
678 a440fac0 2018-09-06 stsp 
679 a440fac0 2018-09-06 stsp 	remain = blocksize;
680 a440fac0 2018-09-06 stsp 	total = 0;
681 a440fac0 2018-09-06 stsp 	while (1) {
682 a440fac0 2018-09-06 stsp 		if (remain == 0) {
683 a440fac0 2018-09-06 stsp 			uint8_t *newbuf;
684 a440fac0 2018-09-06 stsp 			newbuf = reallocarray(buf, 1, total + blocksize);
685 a440fac0 2018-09-06 stsp 			if (newbuf == NULL) {
686 a440fac0 2018-09-06 stsp 				err = got_error_from_errno();
687 a440fac0 2018-09-06 stsp 				goto done;
688 a440fac0 2018-09-06 stsp 			}
689 a440fac0 2018-09-06 stsp 			buf = newbuf;
690 a440fac0 2018-09-06 stsp 			remain += blocksize;
691 a440fac0 2018-09-06 stsp 		}
692 a440fac0 2018-09-06 stsp 		n = fread(buf + total, 1, remain, f);
693 a440fac0 2018-09-06 stsp 		if (n == 0) {
694 a440fac0 2018-09-06 stsp 			if (ferror(f)) {
695 a440fac0 2018-09-06 stsp 				err = got_ferror(f, GOT_ERR_IO);
696 a440fac0 2018-09-06 stsp 				goto done;
697 a440fac0 2018-09-06 stsp 			}
698 a440fac0 2018-09-06 stsp 			break; /* EOF */
699 a440fac0 2018-09-06 stsp 		}
700 a440fac0 2018-09-06 stsp 		remain -= n;
701 a440fac0 2018-09-06 stsp 		total += n;
702 a440fac0 2018-09-06 stsp 	};
703 a440fac0 2018-09-06 stsp 
704 a440fac0 2018-09-06 stsp done:
705 a440fac0 2018-09-06 stsp 	if (err == NULL) {
706 a440fac0 2018-09-06 stsp 		*outbuf = buf;
707 a440fac0 2018-09-06 stsp 		*outlen = total;
708 a440fac0 2018-09-06 stsp 	} else
709 a440fac0 2018-09-06 stsp 		free(buf);
710 a440fac0 2018-09-06 stsp 	return err;
711 a440fac0 2018-09-06 stsp }
712 a440fac0 2018-09-06 stsp 
713 a440fac0 2018-09-06 stsp static const struct got_error *
714 a440fac0 2018-09-06 stsp read_commit_object(struct got_commit_object **commit, struct got_object *obj,
715 a440fac0 2018-09-06 stsp     FILE *f)
716 a440fac0 2018-09-06 stsp {
717 a440fac0 2018-09-06 stsp 	const struct got_error *err = NULL;
718 a440fac0 2018-09-06 stsp 	size_t len;
719 a440fac0 2018-09-06 stsp 	uint8_t *p;
720 a440fac0 2018-09-06 stsp 
721 a440fac0 2018-09-06 stsp 	if (obj->flags & GOT_OBJ_FLAG_PACKED)
722 a440fac0 2018-09-06 stsp 		err = read_to_mem(&p, &len, f);
723 a440fac0 2018-09-06 stsp 	else
724 a440fac0 2018-09-06 stsp 		err = got_inflate_to_mem(&p, &len, f);
725 a440fac0 2018-09-06 stsp 	if (err)
726 a440fac0 2018-09-06 stsp 		return err;
727 a440fac0 2018-09-06 stsp 
728 a440fac0 2018-09-06 stsp 	if (len < obj->hdrlen + obj->size) {
729 a440fac0 2018-09-06 stsp 		err = got_error(GOT_ERR_BAD_OBJ_DATA);
730 a440fac0 2018-09-06 stsp 		goto done;
731 a440fac0 2018-09-06 stsp 	}
732 a440fac0 2018-09-06 stsp 
733 a440fac0 2018-09-06 stsp 	/* Skip object header. */
734 a440fac0 2018-09-06 stsp 	len -= obj->hdrlen;
735 a440fac0 2018-09-06 stsp 	err = got_object_parse_commit(commit, p + obj->hdrlen, len);
736 a440fac0 2018-09-06 stsp 	free(p);
737 a440fac0 2018-09-06 stsp done:
738 a440fac0 2018-09-06 stsp 	return err;
739 a440fac0 2018-09-06 stsp }
740 a440fac0 2018-09-06 stsp 
741 a440fac0 2018-09-06 stsp static void
742 a440fac0 2018-09-06 stsp read_commit_object_privsep_child(struct got_object *obj, int obj_fd,
743 a440fac0 2018-09-06 stsp     int imsg_fds[2])
744 a440fac0 2018-09-06 stsp {
745 a440fac0 2018-09-06 stsp 	const struct got_error *err = NULL;
746 a440fac0 2018-09-06 stsp 	struct got_commit_object *commit = NULL;
747 a440fac0 2018-09-06 stsp 	struct imsgbuf ibuf;
748 a440fac0 2018-09-06 stsp 	FILE *f = NULL;
749 a440fac0 2018-09-06 stsp 	int status = 0;
750 a440fac0 2018-09-06 stsp 
751 a440fac0 2018-09-06 stsp 	setproctitle("read commit object");
752 a440fac0 2018-09-06 stsp 	close(imsg_fds[0]);
753 a440fac0 2018-09-06 stsp 	imsg_init(&ibuf, imsg_fds[1]);
754 a440fac0 2018-09-06 stsp 
755 a440fac0 2018-09-06 stsp 	/* revoke access to most system calls */
756 a440fac0 2018-09-06 stsp 	if (pledge("stdio", NULL) == -1) {
757 a440fac0 2018-09-06 stsp 		err = got_error_from_errno();
758 a440fac0 2018-09-06 stsp 		goto done;
759 a440fac0 2018-09-06 stsp 	}
760 a440fac0 2018-09-06 stsp 
761 a440fac0 2018-09-06 stsp 	f = fdopen(obj_fd, "rb");
762 a440fac0 2018-09-06 stsp 	if (f == NULL) {
763 a440fac0 2018-09-06 stsp 		err = got_error_from_errno();
764 a440fac0 2018-09-06 stsp 		close(obj_fd);
765 a440fac0 2018-09-06 stsp 		goto done;
766 a440fac0 2018-09-06 stsp 	}
767 a440fac0 2018-09-06 stsp 
768 a440fac0 2018-09-06 stsp 	err = read_commit_object(&commit, obj, f);
769 a440fac0 2018-09-06 stsp 	if (err)
770 a440fac0 2018-09-06 stsp 		goto done;
771 a440fac0 2018-09-06 stsp 
772 a440fac0 2018-09-06 stsp 	err = got_privsep_send_commit(&ibuf, commit);
773 a440fac0 2018-09-06 stsp done:
774 a440fac0 2018-09-06 stsp 	if (commit)
775 a440fac0 2018-09-06 stsp 		got_object_commit_close(commit);
776 a440fac0 2018-09-06 stsp 	if (err) {
777 a440fac0 2018-09-06 stsp 		got_privsep_send_error(&ibuf, err);
778 a440fac0 2018-09-06 stsp 		status = 1;
779 a440fac0 2018-09-06 stsp 	}
780 a440fac0 2018-09-06 stsp 	if (f)
781 a440fac0 2018-09-06 stsp 		fclose(f);
782 a440fac0 2018-09-06 stsp 	imsg_clear(&ibuf);
783 a440fac0 2018-09-06 stsp 	close(imsg_fds[1]);
784 a440fac0 2018-09-06 stsp 	_exit(status);
785 a440fac0 2018-09-06 stsp }
786 a440fac0 2018-09-06 stsp 
787 a440fac0 2018-09-06 stsp const struct got_error *
788 a440fac0 2018-09-06 stsp got_object_read_commit_privsep(struct got_commit_object **commit,
789 a440fac0 2018-09-06 stsp     struct got_object *obj, int fd)
790 a440fac0 2018-09-06 stsp {
791 a440fac0 2018-09-06 stsp 	const struct got_error *err = NULL, *err_child = NULL;
792 a440fac0 2018-09-06 stsp 	struct imsgbuf parent_ibuf;
793 a440fac0 2018-09-06 stsp 	int imsg_fds[2];
794 a440fac0 2018-09-06 stsp 	pid_t pid;
795 a440fac0 2018-09-06 stsp 
796 a440fac0 2018-09-06 stsp 	if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, imsg_fds) == -1)
797 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
798 a440fac0 2018-09-06 stsp 
799 a440fac0 2018-09-06 stsp 	pid = fork();
800 a440fac0 2018-09-06 stsp 	if (pid == -1)
801 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
802 a440fac0 2018-09-06 stsp 	else if (pid == 0) {
803 a440fac0 2018-09-06 stsp 		read_commit_object_privsep_child(obj, fd, imsg_fds);
804 a440fac0 2018-09-06 stsp 		/* not reached */
805 a440fac0 2018-09-06 stsp 	}
806 a440fac0 2018-09-06 stsp 
807 a440fac0 2018-09-06 stsp 	close(imsg_fds[1]);
808 a440fac0 2018-09-06 stsp 	imsg_init(&parent_ibuf, imsg_fds[0]);
809 a440fac0 2018-09-06 stsp 	err = got_privsep_recv_commit(commit, &parent_ibuf);
810 a440fac0 2018-09-06 stsp 	imsg_clear(&parent_ibuf);
811 a440fac0 2018-09-06 stsp 	err_child = wait_for_child(pid);
812 a440fac0 2018-09-06 stsp 	close(imsg_fds[0]);
813 a440fac0 2018-09-06 stsp 	return err ? err : err_child;
814 a440fac0 2018-09-06 stsp }
815 a440fac0 2018-09-06 stsp 
816 a440fac0 2018-09-06 stsp static const struct got_error *
817 a440fac0 2018-09-06 stsp read_tree_object(struct got_tree_object **tree, struct got_object *obj, FILE *f)
818 a440fac0 2018-09-06 stsp {
819 a440fac0 2018-09-06 stsp 	const struct got_error *err = NULL;
820 a440fac0 2018-09-06 stsp 	size_t len;
821 a440fac0 2018-09-06 stsp 	uint8_t *p;
822 a440fac0 2018-09-06 stsp 
823 a440fac0 2018-09-06 stsp 	if (obj->flags & GOT_OBJ_FLAG_PACKED)
824 a440fac0 2018-09-06 stsp 		err = read_to_mem(&p, &len, f);
825 a440fac0 2018-09-06 stsp 	else
826 a440fac0 2018-09-06 stsp 		err = got_inflate_to_mem(&p, &len, f);
827 a440fac0 2018-09-06 stsp 	if (err)
828 a440fac0 2018-09-06 stsp 		return err;
829 a440fac0 2018-09-06 stsp 
830 a440fac0 2018-09-06 stsp 	if (len < obj->hdrlen + obj->size) {
831 a440fac0 2018-09-06 stsp 		err = got_error(GOT_ERR_BAD_OBJ_DATA);
832 a440fac0 2018-09-06 stsp 		goto done;
833 a440fac0 2018-09-06 stsp 	}
834 a440fac0 2018-09-06 stsp 
835 a440fac0 2018-09-06 stsp 	/* Skip object header. */
836 a440fac0 2018-09-06 stsp 	len -= obj->hdrlen;
837 a440fac0 2018-09-06 stsp 	err = got_object_parse_tree(tree, p + obj->hdrlen, len);
838 a440fac0 2018-09-06 stsp 	free(p);
839 a440fac0 2018-09-06 stsp done:
840 a440fac0 2018-09-06 stsp 	return err;
841 a440fac0 2018-09-06 stsp }
842 a440fac0 2018-09-06 stsp 
843 a440fac0 2018-09-06 stsp static void
844 a440fac0 2018-09-06 stsp read_tree_object_privsep_child(struct got_object *obj, int obj_fd,
845 a440fac0 2018-09-06 stsp     int imsg_fds[2])
846 a440fac0 2018-09-06 stsp {
847 a440fac0 2018-09-06 stsp 	const struct got_error *err = NULL;
848 a440fac0 2018-09-06 stsp 	struct got_tree_object *tree = NULL;
849 a440fac0 2018-09-06 stsp 	struct imsgbuf ibuf;
850 a440fac0 2018-09-06 stsp 	FILE *f = NULL;
851 a440fac0 2018-09-06 stsp 	int status = 0;
852 a440fac0 2018-09-06 stsp 
853 a440fac0 2018-09-06 stsp 	setproctitle("read tree object");
854 a440fac0 2018-09-06 stsp 	close(imsg_fds[0]);
855 a440fac0 2018-09-06 stsp 	imsg_init(&ibuf, imsg_fds[1]);
856 a440fac0 2018-09-06 stsp 
857 a440fac0 2018-09-06 stsp 	/* revoke access to most system calls */
858 a440fac0 2018-09-06 stsp 	if (pledge("stdio", NULL) == -1) {
859 a440fac0 2018-09-06 stsp 		err = got_error_from_errno();
860 a440fac0 2018-09-06 stsp 		goto done;
861 a440fac0 2018-09-06 stsp 	}
862 a440fac0 2018-09-06 stsp 
863 a440fac0 2018-09-06 stsp 	f = fdopen(obj_fd, "rb");
864 a440fac0 2018-09-06 stsp 	if (f == NULL) {
865 a440fac0 2018-09-06 stsp 		err = got_error_from_errno();
866 a440fac0 2018-09-06 stsp 		close(obj_fd);
867 a440fac0 2018-09-06 stsp 		goto done;
868 a440fac0 2018-09-06 stsp 	}
869 a440fac0 2018-09-06 stsp 
870 a440fac0 2018-09-06 stsp 	err = read_tree_object(&tree, obj, f);
871 a440fac0 2018-09-06 stsp 	if (err)
872 a440fac0 2018-09-06 stsp 		goto done;
873 a440fac0 2018-09-06 stsp 
874 a440fac0 2018-09-06 stsp 	err = got_privsep_send_tree(&ibuf, tree);
875 a440fac0 2018-09-06 stsp done:
876 a440fac0 2018-09-06 stsp 	if (tree)
877 a440fac0 2018-09-06 stsp 		got_object_tree_close(tree);
878 a440fac0 2018-09-06 stsp 	if (err) {
879 a440fac0 2018-09-06 stsp 		got_privsep_send_error(&ibuf, err);
880 a440fac0 2018-09-06 stsp 		status = 1;
881 a440fac0 2018-09-06 stsp 	}
882 a440fac0 2018-09-06 stsp 	if (f)
883 a440fac0 2018-09-06 stsp 		fclose(f);
884 a440fac0 2018-09-06 stsp 	imsg_clear(&ibuf);
885 a440fac0 2018-09-06 stsp 	close(imsg_fds[1]);
886 a440fac0 2018-09-06 stsp 	_exit(status);
887 a440fac0 2018-09-06 stsp }
888 a440fac0 2018-09-06 stsp 
889 a440fac0 2018-09-06 stsp const struct got_error *
890 a440fac0 2018-09-06 stsp got_object_read_tree_privsep(struct got_tree_object **tree,
891 a440fac0 2018-09-06 stsp     struct got_object *obj, int fd)
892 a440fac0 2018-09-06 stsp {
893 a440fac0 2018-09-06 stsp 	const struct got_error *err = NULL, *err_child = NULL;
894 a440fac0 2018-09-06 stsp 	struct imsgbuf parent_ibuf;
895 a440fac0 2018-09-06 stsp 	int imsg_fds[2];
896 a440fac0 2018-09-06 stsp 	pid_t pid;
897 a440fac0 2018-09-06 stsp 
898 a440fac0 2018-09-06 stsp 	if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, imsg_fds) == -1)
899 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
900 a440fac0 2018-09-06 stsp 
901 a440fac0 2018-09-06 stsp 	pid = fork();
902 a440fac0 2018-09-06 stsp 	if (pid == -1)
903 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
904 a440fac0 2018-09-06 stsp 	else if (pid == 0) {
905 a440fac0 2018-09-06 stsp 		read_tree_object_privsep_child(obj, fd, imsg_fds);
906 a440fac0 2018-09-06 stsp 		/* not reached */
907 a440fac0 2018-09-06 stsp 	}
908 a440fac0 2018-09-06 stsp 
909 a440fac0 2018-09-06 stsp 	close(imsg_fds[1]);
910 a440fac0 2018-09-06 stsp 	imsg_init(&parent_ibuf, imsg_fds[0]);
911 a440fac0 2018-09-06 stsp 	err = got_privsep_recv_tree(tree, &parent_ibuf);
912 a440fac0 2018-09-06 stsp 	imsg_clear(&parent_ibuf);
913 a440fac0 2018-09-06 stsp 	err_child = wait_for_child(pid);
914 a440fac0 2018-09-06 stsp 	close(imsg_fds[0]);
915 a440fac0 2018-09-06 stsp 	return err ? err : err_child;
916 a440fac0 2018-09-06 stsp }
917 a440fac0 2018-09-06 stsp 
918 a440fac0 2018-09-06 stsp static const struct got_error *
919 a440fac0 2018-09-06 stsp read_blob_object_privsep_child(int outfd, int infd, int imsg_fds[2])
920 a440fac0 2018-09-06 stsp {
921 a440fac0 2018-09-06 stsp 	const struct got_error *err = NULL;
922 a440fac0 2018-09-06 stsp 	struct imsgbuf ibuf;
923 a440fac0 2018-09-06 stsp 	int status = 0;
924 a440fac0 2018-09-06 stsp 	size_t size;
925 a440fac0 2018-09-06 stsp 	FILE *infile = NULL;
926 a440fac0 2018-09-06 stsp 
927 a440fac0 2018-09-06 stsp 	setproctitle("read blob object");
928 a440fac0 2018-09-06 stsp 	close(imsg_fds[0]);
929 a440fac0 2018-09-06 stsp 	imsg_init(&ibuf, imsg_fds[1]);
930 a440fac0 2018-09-06 stsp 
931 a440fac0 2018-09-06 stsp 	/* revoke access to most system calls */
932 a440fac0 2018-09-06 stsp 	if (pledge("stdio", NULL) == -1) {
933 a440fac0 2018-09-06 stsp 		err = got_error_from_errno();
934 a440fac0 2018-09-06 stsp 		goto done;
935 a440fac0 2018-09-06 stsp 	}
936 a440fac0 2018-09-06 stsp 
937 a440fac0 2018-09-06 stsp 	infile = fdopen(infd, "rb");
938 a440fac0 2018-09-06 stsp 	if (infile == NULL) {
939 a440fac0 2018-09-06 stsp 		err = got_error_from_errno();
940 a440fac0 2018-09-06 stsp 		close(infd);
941 a440fac0 2018-09-06 stsp 		goto done;
942 a440fac0 2018-09-06 stsp 	}
943 a440fac0 2018-09-06 stsp 	err = got_inflate_to_fd(&size, infile, outfd);
944 a440fac0 2018-09-06 stsp 	fclose(infile);
945 a440fac0 2018-09-06 stsp 	if (err)
946 a440fac0 2018-09-06 stsp 		goto done;
947 a440fac0 2018-09-06 stsp 
948 a440fac0 2018-09-06 stsp 	err = got_privsep_send_blob(&ibuf, size);
949 a440fac0 2018-09-06 stsp done:
950 a440fac0 2018-09-06 stsp 	if (err) {
951 a440fac0 2018-09-06 stsp 		got_privsep_send_error(&ibuf, err);
952 a440fac0 2018-09-06 stsp 		status = 1;
953 a440fac0 2018-09-06 stsp 	}
954 a440fac0 2018-09-06 stsp 	close(outfd);
955 a440fac0 2018-09-06 stsp 	imsg_clear(&ibuf);
956 a440fac0 2018-09-06 stsp 	close(imsg_fds[1]);
957 a440fac0 2018-09-06 stsp 	_exit(status);
958 a440fac0 2018-09-06 stsp }
959 a440fac0 2018-09-06 stsp 
960 a440fac0 2018-09-06 stsp const struct got_error *
961 a440fac0 2018-09-06 stsp got_object_read_blob_privsep(size_t *size, int outfd, int infd)
962 a440fac0 2018-09-06 stsp {
963 a440fac0 2018-09-06 stsp 	struct imsgbuf parent_ibuf;
964 a440fac0 2018-09-06 stsp 	int imsg_fds[2];
965 a440fac0 2018-09-06 stsp 	const struct got_error *err = NULL, *err_child = NULL;
966 a440fac0 2018-09-06 stsp 	pid_t pid;
967 a440fac0 2018-09-06 stsp 
968 a440fac0 2018-09-06 stsp 	if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, imsg_fds) == -1)
969 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
970 a440fac0 2018-09-06 stsp 
971 a440fac0 2018-09-06 stsp 	pid = fork();
972 a440fac0 2018-09-06 stsp 	if (pid == -1)
973 a440fac0 2018-09-06 stsp 		return got_error_from_errno();
974 a440fac0 2018-09-06 stsp 	else if (pid == 0) {
975 a440fac0 2018-09-06 stsp 		read_blob_object_privsep_child(outfd, infd, imsg_fds);
976 a440fac0 2018-09-06 stsp 		/* not reached */
977 a440fac0 2018-09-06 stsp 	}
978 a440fac0 2018-09-06 stsp 
979 a440fac0 2018-09-06 stsp 	close(imsg_fds[1]);
980 a440fac0 2018-09-06 stsp 	imsg_init(&parent_ibuf, imsg_fds[0]);
981 a440fac0 2018-09-06 stsp 	err = got_privsep_recv_blob(size, &parent_ibuf);
982 a440fac0 2018-09-06 stsp 	imsg_clear(&parent_ibuf);
983 a440fac0 2018-09-06 stsp 	err_child = wait_for_child(pid);
984 a440fac0 2018-09-06 stsp 	close(imsg_fds[0]);
985 a440fac0 2018-09-06 stsp 	if (lseek(outfd, SEEK_SET, 0) == -1)
986 a440fac0 2018-09-06 stsp 		err = got_error_from_errno();
987 a440fac0 2018-09-06 stsp 	return err ? err : err_child;
988 a440fac0 2018-09-06 stsp }