op public repos

Blame

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