op public repos

Blame

Date:: Wed Jan 18 20:00:34 2023 UTC
Message:: gotd: fix "bad packfile with zero objects" error while creating branches Clients will send an empty pack file if they are only creating new references and have no objects to upload. Make gotd handle this and add a regression test which triggers the bug. Problem found by op@. The new regression test caught an unrelated issue where the client connection was left lingering after references had been updated, which made 'got send' followed by 'got clone -l' fail with the connection limit configured for the test suite (just one connection is allowed at a time). Fix this as well. ok op@
Actions:: History | Blob | Raw File
   1 13b2bc37 2022-10-23 stsp /*
   2 13b2bc37 2022-10-23 stsp  * Copyright (c) 2022 Stefan Sperling <stsp@openbsd.org>
   3 13b2bc37 2022-10-23 stsp  *
   4 13b2bc37 2022-10-23 stsp  * Permission to use, copy, modify, and distribute this software for any
   5 13b2bc37 2022-10-23 stsp  * purpose with or without fee is hereby granted, provided that the above
   6 13b2bc37 2022-10-23 stsp  * copyright notice and this permission notice appear in all copies.
   7 13b2bc37 2022-10-23 stsp  *
   8 13b2bc37 2022-10-23 stsp  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9 13b2bc37 2022-10-23 stsp  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10 13b2bc37 2022-10-23 stsp  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11 13b2bc37 2022-10-23 stsp  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12 13b2bc37 2022-10-23 stsp  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13 13b2bc37 2022-10-23 stsp  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14 13b2bc37 2022-10-23 stsp  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15 13b2bc37 2022-10-23 stsp  */
  16 13b2bc37 2022-10-23 stsp 
  17 13b2bc37 2022-10-23 stsp #include <sys/queue.h>
  18 13b2bc37 2022-10-23 stsp #include <sys/stat.h>
  19 13b2bc37 2022-10-23 stsp #include <sys/tree.h>
  20 13b2bc37 2022-10-23 stsp #include <sys/types.h>
  21 13b2bc37 2022-10-23 stsp 
  22 13b2bc37 2022-10-23 stsp #include <event.h>
  23 13b2bc37 2022-10-23 stsp #include <errno.h>
  24 13b2bc37 2022-10-23 stsp #include <imsg.h>
  25 13b2bc37 2022-10-23 stsp #include <signal.h>
  26 13b2bc37 2022-10-23 stsp #include <siphash.h>
  27 13b2bc37 2022-10-23 stsp #include <stdio.h>
  28 13b2bc37 2022-10-23 stsp #include <stdlib.h>
  29 13b2bc37 2022-10-23 stsp #include <string.h>
  30 13b2bc37 2022-10-23 stsp #include <limits.h>
  31 13b2bc37 2022-10-23 stsp #include <poll.h>
  32 13b2bc37 2022-10-23 stsp #include <sha1.h>
  33 13b2bc37 2022-10-23 stsp #include <unistd.h>
  34 13b2bc37 2022-10-23 stsp #include <zlib.h>
  35 13b2bc37 2022-10-23 stsp 
  36 13b2bc37 2022-10-23 stsp #include "buf.h"
  37 13b2bc37 2022-10-23 stsp 
  38 13b2bc37 2022-10-23 stsp #include "got_error.h"
  39 13b2bc37 2022-10-23 stsp #include "got_repository.h"
  40 13b2bc37 2022-10-23 stsp #include "got_object.h"
  41 13b2bc37 2022-10-23 stsp #include "got_reference.h"
  42 13b2bc37 2022-10-23 stsp #include "got_path.h"
  43 13b2bc37 2022-10-23 stsp 
  44 13b2bc37 2022-10-23 stsp #include "got_lib_delta.h"
  45 13b2bc37 2022-10-23 stsp #include "got_lib_delta_cache.h"
  46 13b2bc37 2022-10-23 stsp #include "got_lib_object.h"
  47 13b2bc37 2022-10-23 stsp #include "got_lib_object_cache.h"
  48 13b2bc37 2022-10-23 stsp #include "got_lib_ratelimit.h"
  49 13b2bc37 2022-10-23 stsp #include "got_lib_pack.h"
  50 13b2bc37 2022-10-23 stsp #include "got_lib_pack_index.h"
  51 13b2bc37 2022-10-23 stsp #include "got_lib_repository.h"
  52 13b2bc37 2022-10-23 stsp #include "got_lib_poll.h"
  53 13b2bc37 2022-10-23 stsp 
  54 13b2bc37 2022-10-23 stsp #include "got_lib_sha1.h" /* XXX temp include for debugging */
  55 13b2bc37 2022-10-23 stsp 
  56 13b2bc37 2022-10-23 stsp #include "log.h"
  57 13b2bc37 2022-10-23 stsp #include "gotd.h"
  58 13b2bc37 2022-10-23 stsp #include "repo_write.h"
  59 13b2bc37 2022-10-23 stsp 
  60 13b2bc37 2022-10-23 stsp #ifndef nitems
  61 13b2bc37 2022-10-23 stsp #define nitems(_a)	(sizeof((_a)) / sizeof((_a)[0]))
  62 13b2bc37 2022-10-23 stsp #endif
  63 13b2bc37 2022-10-23 stsp 
  64 13b2bc37 2022-10-23 stsp static struct repo_write {
  65 13b2bc37 2022-10-23 stsp 	pid_t pid;
  66 13b2bc37 2022-10-23 stsp 	const char *title;
  67 13b2bc37 2022-10-23 stsp 	struct got_repository *repo;
  68 13b2bc37 2022-10-23 stsp 	int *pack_fds;
  69 13b2bc37 2022-10-23 stsp 	int *temp_fds;
  70 ae7c1b78 2023-01-10 stsp 	int session_fd;
  71 ae7c1b78 2023-01-10 stsp 	struct gotd_imsgev session_iev;
  72 13b2bc37 2022-10-23 stsp } repo_write;
  73 13b2bc37 2022-10-23 stsp 
  74 13b2bc37 2022-10-23 stsp struct gotd_ref_update {
  75 13b2bc37 2022-10-23 stsp 	STAILQ_ENTRY(gotd_ref_update) entry;
  76 13b2bc37 2022-10-23 stsp 	struct got_reference *ref;
  77 13b2bc37 2022-10-23 stsp 	int ref_is_new;
  78 13b2bc37 2022-10-23 stsp 	struct got_object_id old_id;
  79 13b2bc37 2022-10-23 stsp 	struct got_object_id new_id;
  80 13b2bc37 2022-10-23 stsp };
  81 13b2bc37 2022-10-23 stsp STAILQ_HEAD(gotd_ref_updates, gotd_ref_update);
  82 13b2bc37 2022-10-23 stsp 
  83 1a52c9bf 2022-12-30 stsp static struct repo_write_client {
  84 13b2bc37 2022-10-23 stsp 	uint32_t			 id;
  85 13b2bc37 2022-10-23 stsp 	int				 fd;
  86 7fec5f4a 2022-10-28 stsp 	int				 pack_pipe;
  87 13b2bc37 2022-10-23 stsp 	struct got_pack			 pack;
  88 13b2bc37 2022-10-23 stsp 	uint8_t				 pack_sha1[SHA1_DIGEST_LENGTH];
  89 13b2bc37 2022-10-23 stsp 	int				 packidx_fd;
  90 13b2bc37 2022-10-23 stsp 	struct gotd_ref_updates		 ref_updates;
  91 13b2bc37 2022-10-23 stsp 	int				 nref_updates;
  92 0ff2c315 2023-01-18 stsp 	int				 nref_new;
  93 1a52c9bf 2022-12-30 stsp } repo_write_client;
  94 13b2bc37 2022-10-23 stsp 
  95 13b2bc37 2022-10-23 stsp static volatile sig_atomic_t sigint_received;
  96 13b2bc37 2022-10-23 stsp static volatile sig_atomic_t sigterm_received;
  97 13b2bc37 2022-10-23 stsp 
  98 13b2bc37 2022-10-23 stsp static void
  99 13b2bc37 2022-10-23 stsp catch_sigint(int signo)
 100 13b2bc37 2022-10-23 stsp {
 101 13b2bc37 2022-10-23 stsp 	sigint_received = 1;
 102 13b2bc37 2022-10-23 stsp }
 103 13b2bc37 2022-10-23 stsp 
 104 13b2bc37 2022-10-23 stsp static void
 105 13b2bc37 2022-10-23 stsp catch_sigterm(int signo)
 106 13b2bc37 2022-10-23 stsp {
 107 13b2bc37 2022-10-23 stsp 	sigterm_received = 1;
 108 13b2bc37 2022-10-23 stsp }
 109 13b2bc37 2022-10-23 stsp 
 110 13b2bc37 2022-10-23 stsp static const struct got_error *
 111 13b2bc37 2022-10-23 stsp check_cancelled(void *arg)
 112 13b2bc37 2022-10-23 stsp {
 113 13b2bc37 2022-10-23 stsp 	if (sigint_received || sigterm_received)
 114 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_CANCELLED);
 115 13b2bc37 2022-10-23 stsp 
 116 13b2bc37 2022-10-23 stsp 	return NULL;
 117 13b2bc37 2022-10-23 stsp }
 118 13b2bc37 2022-10-23 stsp 
 119 13b2bc37 2022-10-23 stsp static const struct got_error *
 120 13b2bc37 2022-10-23 stsp send_peeled_tag_ref(struct got_reference *ref, struct got_object *obj,
 121 13b2bc37 2022-10-23 stsp     struct imsgbuf *ibuf)
 122 13b2bc37 2022-10-23 stsp {
 123 13b2bc37 2022-10-23 stsp 	const struct got_error *err = NULL;
 124 13b2bc37 2022-10-23 stsp 	struct got_tag_object *tag;
 125 13b2bc37 2022-10-23 stsp 	size_t namelen, len;
 126 13b2bc37 2022-10-23 stsp 	char *peeled_refname = NULL;
 127 13b2bc37 2022-10-23 stsp 	struct got_object_id *id;
 128 13b2bc37 2022-10-23 stsp 	struct ibuf *wbuf;
 129 13b2bc37 2022-10-23 stsp 
 130 13b2bc37 2022-10-23 stsp 	err = got_object_tag_open(&tag, repo_write.repo, obj);
 131 13b2bc37 2022-10-23 stsp 	if (err)
 132 13b2bc37 2022-10-23 stsp 		return err;
 133 13b2bc37 2022-10-23 stsp 
 134 13b2bc37 2022-10-23 stsp 	if (asprintf(&peeled_refname, "%s^{}", got_ref_get_name(ref)) == -1) {
 135 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("asprintf");
 136 13b2bc37 2022-10-23 stsp 		goto done;
 137 13b2bc37 2022-10-23 stsp 	}
 138 13b2bc37 2022-10-23 stsp 
 139 13b2bc37 2022-10-23 stsp 	id = got_object_tag_get_object_id(tag);
 140 13b2bc37 2022-10-23 stsp 	namelen = strlen(peeled_refname);
 141 13b2bc37 2022-10-23 stsp 
 142 13b2bc37 2022-10-23 stsp 	len = sizeof(struct gotd_imsg_ref) + namelen;
 143 13b2bc37 2022-10-23 stsp 	if (len > MAX_IMSGSIZE - IMSG_HEADER_SIZE) {
 144 13b2bc37 2022-10-23 stsp 		err = got_error(GOT_ERR_NO_SPACE);
 145 13b2bc37 2022-10-23 stsp 		goto done;
 146 13b2bc37 2022-10-23 stsp 	}
 147 13b2bc37 2022-10-23 stsp 
 148 13b2bc37 2022-10-23 stsp 	wbuf = imsg_create(ibuf, GOTD_IMSG_REF, PROC_REPO_WRITE,
 149 13b2bc37 2022-10-23 stsp 	    repo_write.pid, len);
 150 13b2bc37 2022-10-23 stsp 	if (wbuf == NULL) {
 151 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("imsg_create REF");
 152 13b2bc37 2022-10-23 stsp 		goto done;
 153 13b2bc37 2022-10-23 stsp 	}
 154 13b2bc37 2022-10-23 stsp 
 155 13b2bc37 2022-10-23 stsp 	/* Keep in sync with struct gotd_imsg_ref definition. */
 156 13b2bc37 2022-10-23 stsp 	if (imsg_add(wbuf, id->sha1, SHA1_DIGEST_LENGTH) == -1) {
 157 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("imsg_add REF");
 158 13b2bc37 2022-10-23 stsp 		goto done;
 159 13b2bc37 2022-10-23 stsp 	}
 160 13b2bc37 2022-10-23 stsp 	if (imsg_add(wbuf, &namelen, sizeof(namelen)) == -1) {
 161 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("imsg_add REF");
 162 13b2bc37 2022-10-23 stsp 		goto done;
 163 13b2bc37 2022-10-23 stsp 	}
 164 13b2bc37 2022-10-23 stsp 	if (imsg_add(wbuf, peeled_refname, namelen) == -1) {
 165 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("imsg_add REF");
 166 13b2bc37 2022-10-23 stsp 		goto done;
 167 13b2bc37 2022-10-23 stsp 	}
 168 13b2bc37 2022-10-23 stsp 
 169 13b2bc37 2022-10-23 stsp 	wbuf->fd = -1;
 170 13b2bc37 2022-10-23 stsp 	imsg_close(ibuf, wbuf);
 171 13b2bc37 2022-10-23 stsp done:
 172 13b2bc37 2022-10-23 stsp 	got_object_tag_close(tag);
 173 13b2bc37 2022-10-23 stsp 	return err;
 174 13b2bc37 2022-10-23 stsp }
 175 13b2bc37 2022-10-23 stsp 
 176 13b2bc37 2022-10-23 stsp static const struct got_error *
 177 13b2bc37 2022-10-23 stsp send_ref(struct got_reference *ref, struct imsgbuf *ibuf)
 178 13b2bc37 2022-10-23 stsp {
 179 13b2bc37 2022-10-23 stsp 	const struct got_error *err;
 180 13b2bc37 2022-10-23 stsp 	const char *refname = got_ref_get_name(ref);
 181 13b2bc37 2022-10-23 stsp 	size_t namelen;
 182 13b2bc37 2022-10-23 stsp 	struct got_object_id *id = NULL;
 183 13b2bc37 2022-10-23 stsp 	struct got_object *obj = NULL;
 184 13b2bc37 2022-10-23 stsp 	size_t len;
 185 13b2bc37 2022-10-23 stsp 	struct ibuf *wbuf;
 186 13b2bc37 2022-10-23 stsp 
 187 13b2bc37 2022-10-23 stsp 	namelen = strlen(refname);
 188 13b2bc37 2022-10-23 stsp 
 189 13b2bc37 2022-10-23 stsp 	len = sizeof(struct gotd_imsg_ref) + namelen;
 190 13b2bc37 2022-10-23 stsp 	if (len > MAX_IMSGSIZE - IMSG_HEADER_SIZE)
 191 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_NO_SPACE);
 192 13b2bc37 2022-10-23 stsp 
 193 13b2bc37 2022-10-23 stsp 	err = got_ref_resolve(&id, repo_write.repo, ref);
 194 13b2bc37 2022-10-23 stsp 	if (err)
 195 13b2bc37 2022-10-23 stsp 		return err;
 196 13b2bc37 2022-10-23 stsp 
 197 13b2bc37 2022-10-23 stsp 	wbuf = imsg_create(ibuf, GOTD_IMSG_REF, PROC_REPO_WRITE,
 198 13b2bc37 2022-10-23 stsp 	    repo_write.pid, len);
 199 13b2bc37 2022-10-23 stsp 	if (wbuf == NULL) {
 200 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("imsg_create REF");
 201 13b2bc37 2022-10-23 stsp 		goto done;
 202 13b2bc37 2022-10-23 stsp 	}
 203 13b2bc37 2022-10-23 stsp 
 204 13b2bc37 2022-10-23 stsp 	/* Keep in sync with struct gotd_imsg_ref definition. */
 205 13b2bc37 2022-10-23 stsp 	if (imsg_add(wbuf, id->sha1, SHA1_DIGEST_LENGTH) == -1)
 206 13b2bc37 2022-10-23 stsp 		return got_error_from_errno("imsg_add REF");
 207 13b2bc37 2022-10-23 stsp 	if (imsg_add(wbuf, &namelen, sizeof(namelen)) == -1)
 208 13b2bc37 2022-10-23 stsp 		return got_error_from_errno("imsg_add REF");
 209 13b2bc37 2022-10-23 stsp 	if (imsg_add(wbuf, refname, namelen) == -1)
 210 13b2bc37 2022-10-23 stsp 		return got_error_from_errno("imsg_add REF");
 211 13b2bc37 2022-10-23 stsp 
 212 13b2bc37 2022-10-23 stsp 	wbuf->fd = -1;
 213 13b2bc37 2022-10-23 stsp 	imsg_close(ibuf, wbuf);
 214 13b2bc37 2022-10-23 stsp 
 215 13b2bc37 2022-10-23 stsp 	err = got_object_open(&obj, repo_write.repo, id);
 216 13b2bc37 2022-10-23 stsp 	if (err)
 217 13b2bc37 2022-10-23 stsp 		goto done;
 218 13b2bc37 2022-10-23 stsp 	if (obj->type == GOT_OBJ_TYPE_TAG)
 219 13b2bc37 2022-10-23 stsp 		err = send_peeled_tag_ref(ref, obj, ibuf);
 220 13b2bc37 2022-10-23 stsp done:
 221 13b2bc37 2022-10-23 stsp 	if (obj)
 222 13b2bc37 2022-10-23 stsp 		got_object_close(obj);
 223 13b2bc37 2022-10-23 stsp 	free(id);
 224 13b2bc37 2022-10-23 stsp 	return err;
 225 13b2bc37 2022-10-23 stsp }
 226 13b2bc37 2022-10-23 stsp 
 227 13b2bc37 2022-10-23 stsp static const struct got_error *
 228 1a52c9bf 2022-12-30 stsp list_refs(struct imsg *imsg)
 229 13b2bc37 2022-10-23 stsp {
 230 13b2bc37 2022-10-23 stsp 	const struct got_error *err;
 231 1a52c9bf 2022-12-30 stsp 	struct repo_write_client *client = &repo_write_client;
 232 13b2bc37 2022-10-23 stsp 	struct got_reflist_head refs;
 233 13b2bc37 2022-10-23 stsp 	struct got_reflist_entry *re;
 234 13b2bc37 2022-10-23 stsp 	struct gotd_imsg_list_refs_internal ireq;
 235 13b2bc37 2022-10-23 stsp 	size_t datalen;
 236 13b2bc37 2022-10-23 stsp 	struct gotd_imsg_reflist irefs;
 237 13b2bc37 2022-10-23 stsp 	struct imsgbuf ibuf;
 238 13b2bc37 2022-10-23 stsp 	int client_fd = imsg->fd;
 239 13b2bc37 2022-10-23 stsp 
 240 13b2bc37 2022-10-23 stsp 	TAILQ_INIT(&refs);
 241 13b2bc37 2022-10-23 stsp 
 242 13b2bc37 2022-10-23 stsp 	if (client_fd == -1)
 243 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_PRIVSEP_NO_FD);
 244 13b2bc37 2022-10-23 stsp 
 245 13b2bc37 2022-10-23 stsp 	datalen = imsg->hdr.len - IMSG_HEADER_SIZE;
 246 13b2bc37 2022-10-23 stsp 	if (datalen != sizeof(ireq))
 247 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_PRIVSEP_LEN);
 248 13b2bc37 2022-10-23 stsp 	memcpy(&ireq, imsg->data, sizeof(ireq));
 249 13b2bc37 2022-10-23 stsp 
 250 1a52c9bf 2022-12-30 stsp 	if (ireq.client_id == 0)
 251 1a52c9bf 2022-12-30 stsp 		return got_error(GOT_ERR_CLIENT_ID);
 252 1a52c9bf 2022-12-30 stsp 	if (client->id != 0) {
 253 1a52c9bf 2022-12-30 stsp 		return got_error_msg(GOT_ERR_CLIENT_ID,
 254 1a52c9bf 2022-12-30 stsp 		    "duplicate list-refs request");
 255 1a52c9bf 2022-12-30 stsp 	}
 256 1a52c9bf 2022-12-30 stsp 	client->id = ireq.client_id;
 257 1a52c9bf 2022-12-30 stsp 	client->fd = client_fd;
 258 1a52c9bf 2022-12-30 stsp 	client->pack_pipe = -1;
 259 1a52c9bf 2022-12-30 stsp 	client->packidx_fd = -1;
 260 1a52c9bf 2022-12-30 stsp 	client->nref_updates = 0;
 261 0ff2c315 2023-01-18 stsp 	client->nref_new = 0;
 262 13b2bc37 2022-10-23 stsp 
 263 13b2bc37 2022-10-23 stsp 	imsg_init(&ibuf, client_fd);
 264 13b2bc37 2022-10-23 stsp 
 265 13b2bc37 2022-10-23 stsp 	err = got_ref_list(&refs, repo_write.repo, "",
 266 13b2bc37 2022-10-23 stsp 	    got_ref_cmp_by_name, NULL);
 267 13b2bc37 2022-10-23 stsp 	if (err)
 268 13b2bc37 2022-10-23 stsp 		return err;
 269 13b2bc37 2022-10-23 stsp 
 270 13b2bc37 2022-10-23 stsp 	memset(&irefs, 0, sizeof(irefs));
 271 13b2bc37 2022-10-23 stsp 	TAILQ_FOREACH(re, &refs, entry) {
 272 13b2bc37 2022-10-23 stsp 		struct got_object_id *id;
 273 13b2bc37 2022-10-23 stsp 		int obj_type;
 274 13b2bc37 2022-10-23 stsp 
 275 13b2bc37 2022-10-23 stsp 		if (got_ref_is_symbolic(re->ref))
 276 13b2bc37 2022-10-23 stsp 			continue;
 277 13b2bc37 2022-10-23 stsp 
 278 13b2bc37 2022-10-23 stsp 		irefs.nrefs++;
 279 13b2bc37 2022-10-23 stsp 
 280 13b2bc37 2022-10-23 stsp 		/* Account for a peeled tag refs. */
 281 13b2bc37 2022-10-23 stsp 		err = got_ref_resolve(&id, repo_write.repo, re->ref);
 282 13b2bc37 2022-10-23 stsp 		if (err)
 283 13b2bc37 2022-10-23 stsp 			goto done;
 284 e26970cc 2023-01-10 op 		err = got_object_get_type(&obj_type, repo_write.repo, id);
 285 13b2bc37 2022-10-23 stsp 		free(id);
 286 13b2bc37 2022-10-23 stsp 		if (err)
 287 13b2bc37 2022-10-23 stsp 			goto done;
 288 13b2bc37 2022-10-23 stsp 		if (obj_type == GOT_OBJ_TYPE_TAG)
 289 13b2bc37 2022-10-23 stsp 			irefs.nrefs++;
 290 13b2bc37 2022-10-23 stsp 	}
 291 13b2bc37 2022-10-23 stsp 
 292 13b2bc37 2022-10-23 stsp 	if (imsg_compose(&ibuf, GOTD_IMSG_REFLIST, PROC_REPO_WRITE,
 293 13b2bc37 2022-10-23 stsp 	    repo_write.pid, -1, &irefs, sizeof(irefs)) == -1) {
 294 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("imsg_compose REFLIST");
 295 13b2bc37 2022-10-23 stsp 		goto done;
 296 13b2bc37 2022-10-23 stsp 	}
 297 13b2bc37 2022-10-23 stsp 
 298 13b2bc37 2022-10-23 stsp 	TAILQ_FOREACH(re, &refs, entry) {
 299 13b2bc37 2022-10-23 stsp 		if (got_ref_is_symbolic(re->ref))
 300 13b2bc37 2022-10-23 stsp 			continue;
 301 13b2bc37 2022-10-23 stsp 		err = send_ref(re->ref, &ibuf);
 302 13b2bc37 2022-10-23 stsp 		if (err)
 303 13b2bc37 2022-10-23 stsp 			goto done;
 304 13b2bc37 2022-10-23 stsp 	}
 305 13b2bc37 2022-10-23 stsp 
 306 13b2bc37 2022-10-23 stsp 	err = gotd_imsg_flush(&ibuf);
 307 13b2bc37 2022-10-23 stsp done:
 308 13b2bc37 2022-10-23 stsp 	got_ref_list_free(&refs);
 309 13b2bc37 2022-10-23 stsp 	imsg_clear(&ibuf);
 310 13b2bc37 2022-10-23 stsp 	return err;
 311 13b2bc37 2022-10-23 stsp }
 312 13b2bc37 2022-10-23 stsp 
 313 13b2bc37 2022-10-23 stsp static const struct got_error *
 314 13b2bc37 2022-10-23 stsp protect_ref_namespace(struct got_reference *ref, const char *namespace)
 315 13b2bc37 2022-10-23 stsp {
 316 13b2bc37 2022-10-23 stsp 	size_t len = strlen(namespace);
 317 13b2bc37 2022-10-23 stsp 
 318 13b2bc37 2022-10-23 stsp 	if (len < 5 || strncmp("refs/", namespace, 5) != 0 ||
 319 13b2bc37 2022-10-23 stsp 	    namespace[len -1] != '/') {
 320 13b2bc37 2022-10-23 stsp 		return got_error_fmt(GOT_ERR_BAD_REF_NAME,
 321 13b2bc37 2022-10-23 stsp 		    "reference namespace '%s'", namespace);
 322 13b2bc37 2022-10-23 stsp 	}
 323 13b2bc37 2022-10-23 stsp 
 324 13b2bc37 2022-10-23 stsp 	if (strncmp(namespace, got_ref_get_name(ref), len) == 0)
 325 13b2bc37 2022-10-23 stsp 		return got_error_fmt(GOT_ERR_REFS_PROTECTED, "%s", namespace);
 326 13b2bc37 2022-10-23 stsp 
 327 13b2bc37 2022-10-23 stsp 	return NULL;
 328 13b2bc37 2022-10-23 stsp }
 329 13b2bc37 2022-10-23 stsp 
 330 13b2bc37 2022-10-23 stsp static const struct got_error *
 331 1a52c9bf 2022-12-30 stsp recv_ref_update(struct imsg *imsg)
 332 13b2bc37 2022-10-23 stsp {
 333 13b2bc37 2022-10-23 stsp 	const struct got_error *err = NULL;
 334 1a52c9bf 2022-12-30 stsp 	struct repo_write_client *client = &repo_write_client;
 335 13b2bc37 2022-10-23 stsp 	struct gotd_imsg_ref_update iref;
 336 13b2bc37 2022-10-23 stsp 	size_t datalen;
 337 13b2bc37 2022-10-23 stsp 	char *refname = NULL;
 338 13b2bc37 2022-10-23 stsp 	struct got_reference *ref = NULL;
 339 13b2bc37 2022-10-23 stsp 	struct got_object_id *id = NULL;
 340 13b2bc37 2022-10-23 stsp 	struct imsgbuf ibuf;
 341 13b2bc37 2022-10-23 stsp 	struct gotd_ref_update *ref_update = NULL;
 342 13b2bc37 2022-10-23 stsp 
 343 13b2bc37 2022-10-23 stsp 	log_debug("ref-update received");
 344 13b2bc37 2022-10-23 stsp 
 345 13b2bc37 2022-10-23 stsp 	datalen = imsg->hdr.len - IMSG_HEADER_SIZE;
 346 13b2bc37 2022-10-23 stsp 	if (datalen < sizeof(iref))
 347 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_PRIVSEP_LEN);
 348 13b2bc37 2022-10-23 stsp 	memcpy(&iref, imsg->data, sizeof(iref));
 349 13b2bc37 2022-10-23 stsp 	if (datalen != sizeof(iref) + iref.name_len)
 350 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_PRIVSEP_LEN);
 351 13b2bc37 2022-10-23 stsp 
 352 1a52c9bf 2022-12-30 stsp 	imsg_init(&ibuf, client->fd);
 353 13b2bc37 2022-10-23 stsp 
 354 00b3e9ae 2023-01-11 op 	refname = strndup(imsg->data + sizeof(iref), iref.name_len);
 355 13b2bc37 2022-10-23 stsp 	if (refname == NULL)
 356 00b3e9ae 2023-01-11 op 		return got_error_from_errno("strndup");
 357 13b2bc37 2022-10-23 stsp 
 358 13b2bc37 2022-10-23 stsp 	ref_update = calloc(1, sizeof(*ref_update));
 359 13b2bc37 2022-10-23 stsp 	if (ref_update == NULL) {
 360 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("malloc");
 361 13b2bc37 2022-10-23 stsp 		goto done;
 362 13b2bc37 2022-10-23 stsp 	}
 363 13b2bc37 2022-10-23 stsp 
 364 13b2bc37 2022-10-23 stsp 	memcpy(ref_update->old_id.sha1, iref.old_id, SHA1_DIGEST_LENGTH);
 365 13b2bc37 2022-10-23 stsp 	memcpy(ref_update->new_id.sha1, iref.new_id, SHA1_DIGEST_LENGTH);
 366 13b2bc37 2022-10-23 stsp 
 367 13b2bc37 2022-10-23 stsp 	err = got_ref_open(&ref, repo_write.repo, refname, 0);
 368 13b2bc37 2022-10-23 stsp 	if (err) {
 369 13b2bc37 2022-10-23 stsp 		if (err->code != GOT_ERR_NOT_REF)
 370 13b2bc37 2022-10-23 stsp 			goto done;
 371 13b2bc37 2022-10-23 stsp 		err = got_ref_alloc(&ref, refname, &ref_update->new_id);
 372 13b2bc37 2022-10-23 stsp 		if (err)
 373 13b2bc37 2022-10-23 stsp 			goto done;
 374 13b2bc37 2022-10-23 stsp 		ref_update->ref_is_new = 1;
 375 0ff2c315 2023-01-18 stsp 		client->nref_new++;
 376 13b2bc37 2022-10-23 stsp 	}
 377 13b2bc37 2022-10-23 stsp 	if (got_ref_is_symbolic(ref)) {
 378 13b2bc37 2022-10-23 stsp 		err = got_error_fmt(GOT_ERR_BAD_REF_TYPE,
 379 13b2bc37 2022-10-23 stsp 		    "'%s' is a symbolic reference and cannot "
 380 13b2bc37 2022-10-23 stsp 		    "be updated", got_ref_get_name(ref));
 381 13b2bc37 2022-10-23 stsp 		goto done;
 382 13b2bc37 2022-10-23 stsp 	}
 383 13b2bc37 2022-10-23 stsp 	if (strncmp("refs/", got_ref_get_name(ref), 5) != 0) {
 384 13b2bc37 2022-10-23 stsp 		err = got_error_fmt(GOT_ERR_BAD_REF_NAME,
 385 13b2bc37 2022-10-23 stsp 		    "%s: does not begin with 'refs/'",
 386 13b2bc37 2022-10-23 stsp 		    got_ref_get_name(ref));
 387 13b2bc37 2022-10-23 stsp 		goto done;
 388 13b2bc37 2022-10-23 stsp 	}
 389 13b2bc37 2022-10-23 stsp 
 390 13b2bc37 2022-10-23 stsp 	err = protect_ref_namespace(ref, "refs/got/");
 391 13b2bc37 2022-10-23 stsp 	if (err)
 392 13b2bc37 2022-10-23 stsp 		goto done;
 393 13b2bc37 2022-10-23 stsp 	err = protect_ref_namespace(ref, "refs/remotes/");
 394 13b2bc37 2022-10-23 stsp 	if (err)
 395 13b2bc37 2022-10-23 stsp 		goto done;
 396 13b2bc37 2022-10-23 stsp 
 397 13b2bc37 2022-10-23 stsp 	if (!ref_update->ref_is_new) {
 398 13b2bc37 2022-10-23 stsp 		/*
 399 13b2bc37 2022-10-23 stsp 		 * Ensure the client's idea of this update is still valid.
 400 13b2bc37 2022-10-23 stsp 		 * At this point we can only return an error, to prevent
 401 13b2bc37 2022-10-23 stsp 		 * the client from uploading a pack file which will likely
 402 13b2bc37 2022-10-23 stsp 		 * have to be discarded.
 403 13b2bc37 2022-10-23 stsp 		 */
 404 13b2bc37 2022-10-23 stsp 		err = got_ref_resolve(&id, repo_write.repo, ref);
 405 13b2bc37 2022-10-23 stsp 		if (err)
 406 13b2bc37 2022-10-23 stsp 			goto done;
 407 13b2bc37 2022-10-23 stsp 
 408 13b2bc37 2022-10-23 stsp 		if (got_object_id_cmp(id, &ref_update->old_id) != 0) {
 409 13b2bc37 2022-10-23 stsp 			err = got_error_fmt(GOT_ERR_REF_BUSY,
 410 13b2bc37 2022-10-23 stsp 			    "%s has been modified by someone else "
 411 13b2bc37 2022-10-23 stsp 			    "while transaction was in progress",
 412 13b2bc37 2022-10-23 stsp 			    got_ref_get_name(ref));
 413 13b2bc37 2022-10-23 stsp 			goto done;
 414 13b2bc37 2022-10-23 stsp 		}
 415 13b2bc37 2022-10-23 stsp 	}
 416 13b2bc37 2022-10-23 stsp 
 417 13b2bc37 2022-10-23 stsp 	gotd_imsg_send_ack(&ref_update->new_id, &ibuf, PROC_REPO_WRITE,
 418 13b2bc37 2022-10-23 stsp 	    repo_write.pid);
 419 13b2bc37 2022-10-23 stsp 
 420 13b2bc37 2022-10-23 stsp 	ref_update->ref = ref;
 421 1a52c9bf 2022-12-30 stsp 	STAILQ_INSERT_HEAD(&client->ref_updates, ref_update, entry);
 422 1a52c9bf 2022-12-30 stsp 	client->nref_updates++;
 423 13b2bc37 2022-10-23 stsp 	ref = NULL;
 424 13b2bc37 2022-10-23 stsp 	ref_update = NULL;
 425 13b2bc37 2022-10-23 stsp done:
 426 13b2bc37 2022-10-23 stsp 	if (ref)
 427 13b2bc37 2022-10-23 stsp 		got_ref_close(ref);
 428 13b2bc37 2022-10-23 stsp 	free(ref_update);
 429 13b2bc37 2022-10-23 stsp 	free(refname);
 430 13b2bc37 2022-10-23 stsp 	free(id);
 431 13b2bc37 2022-10-23 stsp 	return err;
 432 13b2bc37 2022-10-23 stsp }
 433 13b2bc37 2022-10-23 stsp 
 434 13b2bc37 2022-10-23 stsp static const struct got_error *
 435 13b2bc37 2022-10-23 stsp pack_index_progress(void *arg, uint32_t nobj_total, uint32_t nobj_indexed,
 436 13b2bc37 2022-10-23 stsp     uint32_t nobj_loose, uint32_t nobj_resolved)
 437 13b2bc37 2022-10-23 stsp {
 438 13b2bc37 2022-10-23 stsp 	int p_indexed = 0, p_resolved = 0;
 439 13b2bc37 2022-10-23 stsp 	int nobj_delta = nobj_total - nobj_loose;
 440 13b2bc37 2022-10-23 stsp 
 441 13b2bc37 2022-10-23 stsp 	if (nobj_total > 0)
 442 13b2bc37 2022-10-23 stsp 		p_indexed = (nobj_indexed * 100) / nobj_total;
 443 13b2bc37 2022-10-23 stsp 
 444 13b2bc37 2022-10-23 stsp 	if (nobj_delta > 0)
 445 13b2bc37 2022-10-23 stsp 		p_resolved = (nobj_resolved * 100) / nobj_delta;
 446 13b2bc37 2022-10-23 stsp 
 447 13b2bc37 2022-10-23 stsp 	if (p_resolved > 0) {
 448 13b2bc37 2022-10-23 stsp 		log_debug("indexing %d objects %d%%; resolving %d deltas %d%%",
 449 13b2bc37 2022-10-23 stsp 		    nobj_total, p_indexed, nobj_delta, p_resolved);
 450 13b2bc37 2022-10-23 stsp 	} else
 451 13b2bc37 2022-10-23 stsp 		log_debug("indexing %d objects %d%%", nobj_total, p_indexed);
 452 13b2bc37 2022-10-23 stsp 
 453 13b2bc37 2022-10-23 stsp 	return NULL;
 454 13b2bc37 2022-10-23 stsp }
 455 13b2bc37 2022-10-23 stsp 
 456 13b2bc37 2022-10-23 stsp static const struct got_error *
 457 13b2bc37 2022-10-23 stsp read_more_pack_stream(int infd, BUF *buf, size_t minsize)
 458 13b2bc37 2022-10-23 stsp {
 459 13b2bc37 2022-10-23 stsp 	const struct got_error *err = NULL;
 460 13b2bc37 2022-10-23 stsp 	uint8_t readahead[65536];
 461 13b2bc37 2022-10-23 stsp 	size_t have, newlen;
 462 13b2bc37 2022-10-23 stsp 
 463 13b2bc37 2022-10-23 stsp 	err = got_poll_read_full(infd, &have,
 464 13b2bc37 2022-10-23 stsp 	    readahead, sizeof(readahead), minsize);
 465 13b2bc37 2022-10-23 stsp 	if (err)
 466 13b2bc37 2022-10-23 stsp 		return err;
 467 13b2bc37 2022-10-23 stsp 
 468 13b2bc37 2022-10-23 stsp 	err = buf_append(&newlen, buf, readahead, have);
 469 13b2bc37 2022-10-23 stsp 	if (err)
 470 13b2bc37 2022-10-23 stsp 		return err;
 471 e26970cc 2023-01-10 op 	return NULL;
 472 13b2bc37 2022-10-23 stsp }
 473 13b2bc37 2022-10-23 stsp 
 474 13b2bc37 2022-10-23 stsp static const struct got_error *
 475 13b2bc37 2022-10-23 stsp copy_object_type_and_size(uint8_t *type, uint64_t *size, int infd, int outfd,
 476 13b2bc37 2022-10-23 stsp     off_t *outsize, BUF *buf, size_t *buf_pos, SHA1_CTX *ctx)
 477 13b2bc37 2022-10-23 stsp {
 478 13b2bc37 2022-10-23 stsp 	const struct got_error *err = NULL;
 479 13b2bc37 2022-10-23 stsp 	uint8_t t = 0;
 480 13b2bc37 2022-10-23 stsp 	uint64_t s = 0;
 481 13b2bc37 2022-10-23 stsp 	uint8_t sizebuf[8];
 482 13b2bc37 2022-10-23 stsp 	size_t i = 0;
 483 13b2bc37 2022-10-23 stsp 	off_t obj_offset = *outsize;
 484 13b2bc37 2022-10-23 stsp 
 485 13b2bc37 2022-10-23 stsp 	do {
 486 13b2bc37 2022-10-23 stsp 		/* We do not support size values which don't fit in 64 bit. */
 487 13b2bc37 2022-10-23 stsp 		if (i > 9)
 488 13b2bc37 2022-10-23 stsp 			return got_error_fmt(GOT_ERR_OBJ_TOO_LARGE,
 489 22ec3416 2022-10-25 op 			    "packfile offset %lld", (long long)obj_offset);
 490 13b2bc37 2022-10-23 stsp 
 491 13b2bc37 2022-10-23 stsp 		if (buf_len(buf) - *buf_pos < sizeof(sizebuf[0])) {
 492 13b2bc37 2022-10-23 stsp 			err = read_more_pack_stream(infd, buf,
 493 13b2bc37 2022-10-23 stsp 			    sizeof(sizebuf[0]));
 494 13b2bc37 2022-10-23 stsp 			if (err)
 495 13b2bc37 2022-10-23 stsp 				return err;
 496 13b2bc37 2022-10-23 stsp 		}
 497 13b2bc37 2022-10-23 stsp 
 498 13b2bc37 2022-10-23 stsp 		sizebuf[i] = buf_getc(buf, *buf_pos);
 499 13b2bc37 2022-10-23 stsp 		*buf_pos += sizeof(sizebuf[i]);
 500 13b2bc37 2022-10-23 stsp 
 501 13b2bc37 2022-10-23 stsp 		if (i == 0) {
 502 13b2bc37 2022-10-23 stsp 			t = (sizebuf[i] & GOT_PACK_OBJ_SIZE0_TYPE_MASK) >>
 503 13b2bc37 2022-10-23 stsp 			    GOT_PACK_OBJ_SIZE0_TYPE_MASK_SHIFT;
 504 13b2bc37 2022-10-23 stsp 			s = (sizebuf[i] & GOT_PACK_OBJ_SIZE0_VAL_MASK);
 505 13b2bc37 2022-10-23 stsp 		} else {
 506 13b2bc37 2022-10-23 stsp 			size_t shift = 4 + 7 * (i - 1);
 507 13b2bc37 2022-10-23 stsp 			s |= ((sizebuf[i] & GOT_PACK_OBJ_SIZE_VAL_MASK) <<
 508 13b2bc37 2022-10-23 stsp 			    shift);
 509 13b2bc37 2022-10-23 stsp 		}
 510 13b2bc37 2022-10-23 stsp 		i++;
 511 13b2bc37 2022-10-23 stsp 	} while (sizebuf[i - 1] & GOT_PACK_OBJ_SIZE_MORE);
 512 13b2bc37 2022-10-23 stsp 
 513 13b2bc37 2022-10-23 stsp 	err = got_pack_hwrite(outfd, sizebuf, i, ctx);
 514 13b2bc37 2022-10-23 stsp 	if (err)
 515 13b2bc37 2022-10-23 stsp 		return err;
 516 13b2bc37 2022-10-23 stsp 	*outsize += i;
 517 13b2bc37 2022-10-23 stsp 
 518 13b2bc37 2022-10-23 stsp 	*type = t;
 519 13b2bc37 2022-10-23 stsp 	*size = s;
 520 13b2bc37 2022-10-23 stsp 	return NULL;
 521 13b2bc37 2022-10-23 stsp }
 522 13b2bc37 2022-10-23 stsp 
 523 13b2bc37 2022-10-23 stsp static const struct got_error *
 524 13b2bc37 2022-10-23 stsp copy_ref_delta(int infd, int outfd, off_t *outsize, BUF *buf, size_t *buf_pos,
 525 13b2bc37 2022-10-23 stsp     SHA1_CTX *ctx)
 526 13b2bc37 2022-10-23 stsp {
 527 13b2bc37 2022-10-23 stsp 	const struct got_error *err = NULL;
 528 13b2bc37 2022-10-23 stsp 	size_t remain = buf_len(buf) - *buf_pos;
 529 13b2bc37 2022-10-23 stsp 
 530 13b2bc37 2022-10-23 stsp 	if (remain < SHA1_DIGEST_LENGTH) {
 531 13b2bc37 2022-10-23 stsp 		err = read_more_pack_stream(infd, buf,
 532 13b2bc37 2022-10-23 stsp 		    SHA1_DIGEST_LENGTH - remain);
 533 13b2bc37 2022-10-23 stsp 		if (err)
 534 13b2bc37 2022-10-23 stsp 			return err;
 535 13b2bc37 2022-10-23 stsp 	}
 536 13b2bc37 2022-10-23 stsp 
 537 13b2bc37 2022-10-23 stsp 	err = got_pack_hwrite(outfd, buf_get(buf) + *buf_pos,
 538 13b2bc37 2022-10-23 stsp 	    SHA1_DIGEST_LENGTH, ctx);
 539 13b2bc37 2022-10-23 stsp 	if (err)
 540 13b2bc37 2022-10-23 stsp 		return err;
 541 13b2bc37 2022-10-23 stsp 
 542 13b2bc37 2022-10-23 stsp 	*buf_pos += SHA1_DIGEST_LENGTH;
 543 13b2bc37 2022-10-23 stsp 	return NULL;
 544 13b2bc37 2022-10-23 stsp }
 545 13b2bc37 2022-10-23 stsp 
 546 13b2bc37 2022-10-23 stsp static const struct got_error *
 547 13b2bc37 2022-10-23 stsp copy_offset_delta(int infd, int outfd, off_t *outsize, BUF *buf, size_t *buf_pos,
 548 13b2bc37 2022-10-23 stsp     SHA1_CTX *ctx)
 549 13b2bc37 2022-10-23 stsp {
 550 13b2bc37 2022-10-23 stsp 	const struct got_error *err = NULL;
 551 13b2bc37 2022-10-23 stsp 	uint64_t o = 0;
 552 13b2bc37 2022-10-23 stsp 	uint8_t offbuf[8];
 553 13b2bc37 2022-10-23 stsp 	size_t i = 0;
 554 13b2bc37 2022-10-23 stsp 	off_t obj_offset = *outsize;
 555 13b2bc37 2022-10-23 stsp 
 556 13b2bc37 2022-10-23 stsp 	do {
 557 13b2bc37 2022-10-23 stsp 		/* We do not support offset values which don't fit in 64 bit. */
 558 13b2bc37 2022-10-23 stsp 		if (i > 8)
 559 13b2bc37 2022-10-23 stsp 			return got_error_fmt(GOT_ERR_OBJ_TOO_LARGE,
 560 22ec3416 2022-10-25 op 			    "packfile offset %lld", (long long)obj_offset);
 561 13b2bc37 2022-10-23 stsp 
 562 13b2bc37 2022-10-23 stsp 		if (buf_len(buf) - *buf_pos < sizeof(offbuf[0])) {
 563 13b2bc37 2022-10-23 stsp 			err = read_more_pack_stream(infd, buf,
 564 13b2bc37 2022-10-23 stsp 			    sizeof(offbuf[0]));
 565 13b2bc37 2022-10-23 stsp 			if (err)
 566 13b2bc37 2022-10-23 stsp 				return err;
 567 13b2bc37 2022-10-23 stsp 		}
 568 13b2bc37 2022-10-23 stsp 
 569 13b2bc37 2022-10-23 stsp 		offbuf[i] = buf_getc(buf, *buf_pos);
 570 13b2bc37 2022-10-23 stsp 		*buf_pos += sizeof(offbuf[i]);
 571 13b2bc37 2022-10-23 stsp 
 572 13b2bc37 2022-10-23 stsp 		if (i == 0)
 573 13b2bc37 2022-10-23 stsp 			o = (offbuf[i] & GOT_PACK_OBJ_DELTA_OFF_VAL_MASK);
 574 13b2bc37 2022-10-23 stsp 		else {
 575 13b2bc37 2022-10-23 stsp 			o++;
 576 13b2bc37 2022-10-23 stsp 			o <<= 7;
 577 13b2bc37 2022-10-23 stsp 			o += (offbuf[i] & GOT_PACK_OBJ_DELTA_OFF_VAL_MASK);
 578 13b2bc37 2022-10-23 stsp 		}
 579 13b2bc37 2022-10-23 stsp 		i++;
 580 13b2bc37 2022-10-23 stsp 	} while (offbuf[i - 1] & GOT_PACK_OBJ_DELTA_OFF_MORE);
 581 13b2bc37 2022-10-23 stsp 
 582 13b2bc37 2022-10-23 stsp 	if (o < sizeof(struct got_packfile_hdr) || o > *outsize)
 583 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_PACK_OFFSET);
 584 13b2bc37 2022-10-23 stsp 
 585 13b2bc37 2022-10-23 stsp 	err = got_pack_hwrite(outfd, offbuf, i, ctx);
 586 13b2bc37 2022-10-23 stsp 	if (err)
 587 13b2bc37 2022-10-23 stsp 		return err;
 588 13b2bc37 2022-10-23 stsp 
 589 13b2bc37 2022-10-23 stsp 	*outsize += i;
 590 13b2bc37 2022-10-23 stsp 	return NULL;
 591 13b2bc37 2022-10-23 stsp }
 592 13b2bc37 2022-10-23 stsp 
 593 13b2bc37 2022-10-23 stsp static const struct got_error *
 594 13b2bc37 2022-10-23 stsp copy_zstream(int infd, int outfd, off_t *outsize, BUF *buf, size_t *buf_pos,
 595 13b2bc37 2022-10-23 stsp     SHA1_CTX *ctx)
 596 13b2bc37 2022-10-23 stsp {
 597 13b2bc37 2022-10-23 stsp 	const struct got_error *err = NULL;
 598 13b2bc37 2022-10-23 stsp 	z_stream z;
 599 13b2bc37 2022-10-23 stsp 	int zret;
 600 13b2bc37 2022-10-23 stsp 	char voidbuf[1024];
 601 13b2bc37 2022-10-23 stsp 	size_t consumed_total = 0;
 602 13b2bc37 2022-10-23 stsp 	off_t zstream_offset = *outsize;
 603 13b2bc37 2022-10-23 stsp 
 604 13b2bc37 2022-10-23 stsp 	memset(&z, 0, sizeof(z));
 605 13b2bc37 2022-10-23 stsp 
 606 13b2bc37 2022-10-23 stsp 	z.zalloc = Z_NULL;
 607 13b2bc37 2022-10-23 stsp 	z.zfree = Z_NULL;
 608 13b2bc37 2022-10-23 stsp 	zret = inflateInit(&z);
 609 13b2bc37 2022-10-23 stsp 	if (zret != Z_OK) {
 610 13b2bc37 2022-10-23 stsp 		if  (zret == Z_ERRNO)
 611 13b2bc37 2022-10-23 stsp 			return got_error_from_errno("inflateInit");
 612 13b2bc37 2022-10-23 stsp 		if  (zret == Z_MEM_ERROR) {
 613 13b2bc37 2022-10-23 stsp 			errno = ENOMEM;
 614 13b2bc37 2022-10-23 stsp 			return got_error_from_errno("inflateInit");
 615 13b2bc37 2022-10-23 stsp 		}
 616 13b2bc37 2022-10-23 stsp 		return got_error_msg(GOT_ERR_DECOMPRESSION,
 617 13b2bc37 2022-10-23 stsp 		    "inflateInit failed");
 618 13b2bc37 2022-10-23 stsp 	}
 619 13b2bc37 2022-10-23 stsp 
 620 13b2bc37 2022-10-23 stsp 	while (zret != Z_STREAM_END) {
 621 13b2bc37 2022-10-23 stsp 		size_t last_total_in, consumed;
 622 13b2bc37 2022-10-23 stsp 
 623 13b2bc37 2022-10-23 stsp 		/*
 624 13b2bc37 2022-10-23 stsp 		 * Decompress into the void. Object data will be parsed
 625 13b2bc37 2022-10-23 stsp 		 * later, when the pack file is indexed. For now, we just
 626 13b2bc37 2022-10-23 stsp 		 * want to locate the end of the compressed stream.
 627 13b2bc37 2022-10-23 stsp 		 */
 628 13b2bc37 2022-10-23 stsp 		while (zret != Z_STREAM_END && buf_len(buf) - *buf_pos > 0) {
 629 13b2bc37 2022-10-23 stsp 			last_total_in = z.total_in;
 630 13b2bc37 2022-10-23 stsp 			z.next_in = buf_get(buf) + *buf_pos;
 631 13b2bc37 2022-10-23 stsp 			z.avail_in = buf_len(buf) - *buf_pos;
 632 13b2bc37 2022-10-23 stsp 			z.next_out = voidbuf;
 633 13b2bc37 2022-10-23 stsp 			z.avail_out = sizeof(voidbuf);
 634 13b2bc37 2022-10-23 stsp 
 635 13b2bc37 2022-10-23 stsp 			zret = inflate(&z, Z_SYNC_FLUSH);
 636 13b2bc37 2022-10-23 stsp 			if (zret != Z_OK && zret != Z_BUF_ERROR &&
 637 13b2bc37 2022-10-23 stsp 			    zret != Z_STREAM_END) {
 638 13b2bc37 2022-10-23 stsp 				err = got_error_fmt(GOT_ERR_DECOMPRESSION,
 639 22ec3416 2022-10-25 op 				    "packfile offset %lld",
 640 22ec3416 2022-10-25 op 				    (long long)zstream_offset);
 641 13b2bc37 2022-10-23 stsp 				goto done;
 642 13b2bc37 2022-10-23 stsp 			}
 643 13b2bc37 2022-10-23 stsp 			consumed = z.total_in - last_total_in;
 644 13b2bc37 2022-10-23 stsp 
 645 13b2bc37 2022-10-23 stsp 			err = got_pack_hwrite(outfd, buf_get(buf) + *buf_pos,
 646 13b2bc37 2022-10-23 stsp 			    consumed, ctx);
 647 13b2bc37 2022-10-23 stsp 			if (err)
 648 13b2bc37 2022-10-23 stsp 				goto done;
 649 13b2bc37 2022-10-23 stsp 
 650 13b2bc37 2022-10-23 stsp 			err = buf_discard(buf, *buf_pos + consumed);
 651 13b2bc37 2022-10-23 stsp 			if (err)
 652 13b2bc37 2022-10-23 stsp 				goto done;
 653 13b2bc37 2022-10-23 stsp 			*buf_pos = 0;
 654 13b2bc37 2022-10-23 stsp 
 655 13b2bc37 2022-10-23 stsp 			consumed_total += consumed;
 656 13b2bc37 2022-10-23 stsp 		}
 657 13b2bc37 2022-10-23 stsp 
 658 13b2bc37 2022-10-23 stsp 		if (zret != Z_STREAM_END) {
 659 13b2bc37 2022-10-23 stsp 			err = read_more_pack_stream(infd, buf, 1);
 660 13b2bc37 2022-10-23 stsp 			if (err)
 661 13b2bc37 2022-10-23 stsp 				goto done;
 662 13b2bc37 2022-10-23 stsp 		}
 663 13b2bc37 2022-10-23 stsp 	}
 664 13b2bc37 2022-10-23 stsp 
 665 13b2bc37 2022-10-23 stsp 	if (err == NULL)
 666 13b2bc37 2022-10-23 stsp 		*outsize += consumed_total;
 667 13b2bc37 2022-10-23 stsp done:
 668 13b2bc37 2022-10-23 stsp 	inflateEnd(&z);
 669 13b2bc37 2022-10-23 stsp 	return err;
 670 13b2bc37 2022-10-23 stsp }
 671 13b2bc37 2022-10-23 stsp 
 672 13b2bc37 2022-10-23 stsp static const struct got_error *
 673 13b2bc37 2022-10-23 stsp validate_object_type(int obj_type)
 674 13b2bc37 2022-10-23 stsp {
 675 13b2bc37 2022-10-23 stsp 	switch (obj_type) {
 676 13b2bc37 2022-10-23 stsp 	case GOT_OBJ_TYPE_BLOB:
 677 13b2bc37 2022-10-23 stsp 	case GOT_OBJ_TYPE_COMMIT:
 678 13b2bc37 2022-10-23 stsp 	case GOT_OBJ_TYPE_TREE:
 679 13b2bc37 2022-10-23 stsp 	case GOT_OBJ_TYPE_TAG:
 680 13b2bc37 2022-10-23 stsp 	case GOT_OBJ_TYPE_REF_DELTA:
 681 13b2bc37 2022-10-23 stsp 	case GOT_OBJ_TYPE_OFFSET_DELTA:
 682 13b2bc37 2022-10-23 stsp 		return NULL;
 683 13b2bc37 2022-10-23 stsp 	default:
 684 13b2bc37 2022-10-23 stsp 		break;
 685 13b2bc37 2022-10-23 stsp 	}
 686 13b2bc37 2022-10-23 stsp 
 687 13b2bc37 2022-10-23 stsp 	return got_error(GOT_ERR_OBJ_TYPE);
 688 13b2bc37 2022-10-23 stsp }
 689 13b2bc37 2022-10-23 stsp 
 690 13b2bc37 2022-10-23 stsp static const struct got_error *
 691 0ff2c315 2023-01-18 stsp recv_packdata(off_t *outsize, uint32_t *nobj, uint8_t *sha1,
 692 0ff2c315 2023-01-18 stsp     int infd, int outfd)
 693 13b2bc37 2022-10-23 stsp {
 694 13b2bc37 2022-10-23 stsp 	const struct got_error *err;
 695 0ff2c315 2023-01-18 stsp 	struct repo_write_client *client = &repo_write_client;
 696 13b2bc37 2022-10-23 stsp 	struct got_packfile_hdr hdr;
 697 13b2bc37 2022-10-23 stsp 	size_t have;
 698 0ff2c315 2023-01-18 stsp 	uint32_t nhave = 0;
 699 13b2bc37 2022-10-23 stsp 	SHA1_CTX ctx;
 700 13b2bc37 2022-10-23 stsp 	uint8_t expected_sha1[SHA1_DIGEST_LENGTH];
 701 13b2bc37 2022-10-23 stsp 	char hex[SHA1_DIGEST_STRING_LENGTH];
 702 13b2bc37 2022-10-23 stsp 	BUF *buf = NULL;
 703 13b2bc37 2022-10-23 stsp 	size_t buf_pos = 0, remain;
 704 13b2bc37 2022-10-23 stsp 	ssize_t w;
 705 13b2bc37 2022-10-23 stsp 
 706 13b2bc37 2022-10-23 stsp 	*outsize = 0;
 707 0ff2c315 2023-01-18 stsp 	*nobj = 0;
 708 13b2bc37 2022-10-23 stsp 	SHA1Init(&ctx);
 709 13b2bc37 2022-10-23 stsp 
 710 13b2bc37 2022-10-23 stsp 	err = got_poll_read_full(infd, &have, &hdr, sizeof(hdr), sizeof(hdr));
 711 13b2bc37 2022-10-23 stsp 	if (err)
 712 13b2bc37 2022-10-23 stsp 		return err;
 713 13b2bc37 2022-10-23 stsp 	if (have != sizeof(hdr))
 714 13b2bc37 2022-10-23 stsp 		return got_error_msg(GOT_ERR_BAD_PACKFILE, "short pack file");
 715 13b2bc37 2022-10-23 stsp 	*outsize += have;
 716 13b2bc37 2022-10-23 stsp 
 717 13b2bc37 2022-10-23 stsp 	if (hdr.signature != htobe32(GOT_PACKFILE_SIGNATURE))
 718 13b2bc37 2022-10-23 stsp 		return got_error_msg(GOT_ERR_BAD_PACKFILE,
 719 13b2bc37 2022-10-23 stsp 		    "bad packfile signature");
 720 13b2bc37 2022-10-23 stsp 	if (hdr.version != htobe32(GOT_PACKFILE_VERSION))
 721 13b2bc37 2022-10-23 stsp 		return got_error_msg(GOT_ERR_BAD_PACKFILE,
 722 13b2bc37 2022-10-23 stsp 		    "bad packfile version");
 723 13b2bc37 2022-10-23 stsp 
 724 0ff2c315 2023-01-18 stsp 	*nobj = be32toh(hdr.nobjects);
 725 0ff2c315 2023-01-18 stsp 	if (*nobj == 0) {
 726 0ff2c315 2023-01-18 stsp 		/*
 727 0ff2c315 2023-01-18 stsp 		 * Clients which are creating new references only
 728 0ff2c315 2023-01-18 stsp 		 * will send us an empty pack file.
 729 0ff2c315 2023-01-18 stsp 		 */
 730 0ff2c315 2023-01-18 stsp 		if (client->nref_updates > 0 &&
 731 0ff2c315 2023-01-18 stsp 		    client->nref_updates == client->nref_new)
 732 0ff2c315 2023-01-18 stsp 			return NULL;
 733 0ff2c315 2023-01-18 stsp 
 734 13b2bc37 2022-10-23 stsp 		return got_error_msg(GOT_ERR_BAD_PACKFILE,
 735 13b2bc37 2022-10-23 stsp 		    "bad packfile with zero objects");
 736 0ff2c315 2023-01-18 stsp 	}
 737 13b2bc37 2022-10-23 stsp 
 738 0ff2c315 2023-01-18 stsp 	log_debug("expecting %d objects", *nobj);
 739 13b2bc37 2022-10-23 stsp 
 740 13b2bc37 2022-10-23 stsp 	err = got_pack_hwrite(outfd, &hdr, sizeof(hdr), &ctx);
 741 13b2bc37 2022-10-23 stsp 	if (err)
 742 13b2bc37 2022-10-23 stsp 		return err;
 743 13b2bc37 2022-10-23 stsp 
 744 13b2bc37 2022-10-23 stsp 	err = buf_alloc(&buf, 65536);
 745 13b2bc37 2022-10-23 stsp 	if (err)
 746 13b2bc37 2022-10-23 stsp 		return err;
 747 13b2bc37 2022-10-23 stsp 
 748 0ff2c315 2023-01-18 stsp 	while (nhave != *nobj) {
 749 13b2bc37 2022-10-23 stsp 		uint8_t obj_type;
 750 13b2bc37 2022-10-23 stsp 		uint64_t obj_size;
 751 13b2bc37 2022-10-23 stsp 
 752 13b2bc37 2022-10-23 stsp 		err = copy_object_type_and_size(&obj_type, &obj_size,
 753 13b2bc37 2022-10-23 stsp 		    infd, outfd, outsize, buf, &buf_pos, &ctx);
 754 13b2bc37 2022-10-23 stsp 		if (err)
 755 13b2bc37 2022-10-23 stsp 			goto done;
 756 13b2bc37 2022-10-23 stsp 
 757 13b2bc37 2022-10-23 stsp 		err = validate_object_type(obj_type);
 758 13b2bc37 2022-10-23 stsp 		if (err)
 759 13b2bc37 2022-10-23 stsp 			goto done;
 760 13b2bc37 2022-10-23 stsp 
 761 13b2bc37 2022-10-23 stsp 		if (obj_type == GOT_OBJ_TYPE_REF_DELTA) {
 762 13b2bc37 2022-10-23 stsp 			err = copy_ref_delta(infd, outfd, outsize,
 763 13b2bc37 2022-10-23 stsp 			    buf, &buf_pos, &ctx);
 764 13b2bc37 2022-10-23 stsp 			if (err)
 765 13b2bc37 2022-10-23 stsp 				goto done;
 766 13b2bc37 2022-10-23 stsp 		} else if (obj_type == GOT_OBJ_TYPE_OFFSET_DELTA) {
 767 13b2bc37 2022-10-23 stsp 			err = copy_offset_delta(infd, outfd, outsize,
 768 13b2bc37 2022-10-23 stsp 			    buf, &buf_pos, &ctx);
 769 13b2bc37 2022-10-23 stsp 			if (err)
 770 13b2bc37 2022-10-23 stsp 				goto done;
 771 13b2bc37 2022-10-23 stsp 		}
 772 13b2bc37 2022-10-23 stsp 
 773 13b2bc37 2022-10-23 stsp 		err = copy_zstream(infd, outfd, outsize, buf, &buf_pos, &ctx);
 774 13b2bc37 2022-10-23 stsp 		if (err)
 775 13b2bc37 2022-10-23 stsp 			goto done;
 776 13b2bc37 2022-10-23 stsp 
 777 13b2bc37 2022-10-23 stsp 		nhave++;
 778 13b2bc37 2022-10-23 stsp 	}
 779 13b2bc37 2022-10-23 stsp 
 780 0ff2c315 2023-01-18 stsp 	log_debug("received %u objects", *nobj);
 781 13b2bc37 2022-10-23 stsp 
 782 13b2bc37 2022-10-23 stsp 	SHA1Final(expected_sha1, &ctx);
 783 13b2bc37 2022-10-23 stsp 
 784 13b2bc37 2022-10-23 stsp 	remain = buf_len(buf) - buf_pos;
 785 13b2bc37 2022-10-23 stsp 	if (remain < SHA1_DIGEST_LENGTH) {
 786 13b2bc37 2022-10-23 stsp 		err = read_more_pack_stream(infd, buf,
 787 13b2bc37 2022-10-23 stsp 		    SHA1_DIGEST_LENGTH - remain);
 788 13b2bc37 2022-10-23 stsp 		if (err)
 789 13b2bc37 2022-10-23 stsp 			return err;
 790 13b2bc37 2022-10-23 stsp 	}
 791 13b2bc37 2022-10-23 stsp 
 792 13b2bc37 2022-10-23 stsp 	got_sha1_digest_to_str(expected_sha1, hex, sizeof(hex));
 793 13b2bc37 2022-10-23 stsp 	log_debug("expect SHA1: %s", hex);
 794 13b2bc37 2022-10-23 stsp 	got_sha1_digest_to_str(buf_get(buf) + buf_pos, hex, sizeof(hex));
 795 13b2bc37 2022-10-23 stsp 	log_debug("actual SHA1: %s", hex);
 796 13b2bc37 2022-10-23 stsp 
 797 13b2bc37 2022-10-23 stsp 	if (memcmp(buf_get(buf) + buf_pos, expected_sha1,
 798 13b2bc37 2022-10-23 stsp 	    SHA1_DIGEST_LENGTH) != 0) {
 799 13b2bc37 2022-10-23 stsp 		err = got_error(GOT_ERR_PACKFILE_CSUM);
 800 13b2bc37 2022-10-23 stsp 		goto done;
 801 13b2bc37 2022-10-23 stsp 	}
 802 13b2bc37 2022-10-23 stsp 
 803 13b2bc37 2022-10-23 stsp 	memcpy(sha1, expected_sha1, SHA1_DIGEST_LENGTH);
 804 13b2bc37 2022-10-23 stsp 
 805 13b2bc37 2022-10-23 stsp 	w = write(outfd, expected_sha1, SHA1_DIGEST_LENGTH);
 806 13b2bc37 2022-10-23 stsp 	if (w == -1) {
 807 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("write");
 808 13b2bc37 2022-10-23 stsp 		goto done;
 809 13b2bc37 2022-10-23 stsp 	}
 810 13b2bc37 2022-10-23 stsp 	if (w != SHA1_DIGEST_LENGTH) {
 811 13b2bc37 2022-10-23 stsp 		err = got_error(GOT_ERR_IO);
 812 13b2bc37 2022-10-23 stsp 		goto done;
 813 13b2bc37 2022-10-23 stsp 	}
 814 13b2bc37 2022-10-23 stsp 
 815 13b2bc37 2022-10-23 stsp 	*outsize += SHA1_DIGEST_LENGTH;
 816 13b2bc37 2022-10-23 stsp 
 817 13b2bc37 2022-10-23 stsp 	if (fsync(outfd) == -1) {
 818 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("fsync");
 819 13b2bc37 2022-10-23 stsp 		goto done;
 820 13b2bc37 2022-10-23 stsp 	}
 821 13b2bc37 2022-10-23 stsp 	if (lseek(outfd, 0L, SEEK_SET) == -1) {
 822 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("lseek");
 823 13b2bc37 2022-10-23 stsp 		goto done;
 824 13b2bc37 2022-10-23 stsp 	}
 825 13b2bc37 2022-10-23 stsp done:
 826 13b2bc37 2022-10-23 stsp 	buf_free(buf);
 827 13b2bc37 2022-10-23 stsp 	return err;
 828 13b2bc37 2022-10-23 stsp }
 829 13b2bc37 2022-10-23 stsp 
 830 13b2bc37 2022-10-23 stsp static const struct got_error *
 831 1a52c9bf 2022-12-30 stsp report_pack_status(const struct got_error *unpack_err)
 832 13b2bc37 2022-10-23 stsp {
 833 13b2bc37 2022-10-23 stsp 	const struct got_error *err = NULL;
 834 1a52c9bf 2022-12-30 stsp 	struct repo_write_client *client = &repo_write_client;
 835 13b2bc37 2022-10-23 stsp 	struct gotd_imsg_packfile_status istatus;
 836 13b2bc37 2022-10-23 stsp 	struct ibuf *wbuf;
 837 13b2bc37 2022-10-23 stsp 	struct imsgbuf ibuf;
 838 13b2bc37 2022-10-23 stsp 	const char *unpack_ok = "unpack ok\n";
 839 13b2bc37 2022-10-23 stsp 	size_t len;
 840 e26970cc 2023-01-10 op 
 841 13b2bc37 2022-10-23 stsp 	imsg_init(&ibuf, client->fd);
 842 13b2bc37 2022-10-23 stsp 
 843 13b2bc37 2022-10-23 stsp 	if (unpack_err)
 844 13b2bc37 2022-10-23 stsp 		istatus.reason_len = strlen(unpack_err->msg);
 845 13b2bc37 2022-10-23 stsp 	else
 846 13b2bc37 2022-10-23 stsp 		istatus.reason_len = strlen(unpack_ok);
 847 13b2bc37 2022-10-23 stsp 
 848 13b2bc37 2022-10-23 stsp 	len = sizeof(istatus) + istatus.reason_len;
 849 13b2bc37 2022-10-23 stsp 	wbuf = imsg_create(&ibuf, GOTD_IMSG_PACKFILE_STATUS, PROC_REPO_WRITE,
 850 13b2bc37 2022-10-23 stsp 	    repo_write.pid, len);
 851 13b2bc37 2022-10-23 stsp 	if (wbuf == NULL) {
 852 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("imsg_create PACKFILE_STATUS");
 853 13b2bc37 2022-10-23 stsp 		goto done;
 854 13b2bc37 2022-10-23 stsp 	}
 855 13b2bc37 2022-10-23 stsp 
 856 13b2bc37 2022-10-23 stsp 	if (imsg_add(wbuf, &istatus, sizeof(istatus)) == -1) {
 857 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("imsg_add PACKFILE_STATUS");
 858 13b2bc37 2022-10-23 stsp 		goto done;
 859 13b2bc37 2022-10-23 stsp 	}
 860 13b2bc37 2022-10-23 stsp 
 861 13b2bc37 2022-10-23 stsp 	if (imsg_add(wbuf, err ? err->msg : unpack_ok,
 862 13b2bc37 2022-10-23 stsp 	    istatus.reason_len) == -1) {
 863 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("imsg_add PACKFILE_STATUS");
 864 13b2bc37 2022-10-23 stsp 		goto done;
 865 13b2bc37 2022-10-23 stsp 	}
 866 13b2bc37 2022-10-23 stsp 
 867 13b2bc37 2022-10-23 stsp 	wbuf->fd = -1;
 868 13b2bc37 2022-10-23 stsp 	imsg_close(&ibuf, wbuf);
 869 13b2bc37 2022-10-23 stsp 
 870 13b2bc37 2022-10-23 stsp 	err = gotd_imsg_flush(&ibuf);
 871 13b2bc37 2022-10-23 stsp done:
 872 13b2bc37 2022-10-23 stsp 	imsg_clear(&ibuf);
 873 13b2bc37 2022-10-23 stsp 	return err;
 874 13b2bc37 2022-10-23 stsp }
 875 13b2bc37 2022-10-23 stsp 
 876 13b2bc37 2022-10-23 stsp static const struct got_error *
 877 0ff2c315 2023-01-18 stsp recv_packfile(int *have_packfile, struct imsg *imsg)
 878 13b2bc37 2022-10-23 stsp {
 879 13b2bc37 2022-10-23 stsp 	const struct got_error *err = NULL, *unpack_err;
 880 1a52c9bf 2022-12-30 stsp 	struct repo_write_client *client = &repo_write_client;
 881 13b2bc37 2022-10-23 stsp 	struct gotd_imsg_recv_packfile ireq;
 882 13b2bc37 2022-10-23 stsp 	FILE *tempfiles[3] = { NULL, NULL, NULL };
 883 13b2bc37 2022-10-23 stsp 	struct repo_tempfile {
 884 13b2bc37 2022-10-23 stsp 		int fd;
 885 13b2bc37 2022-10-23 stsp 		int idx;
 886 13b2bc37 2022-10-23 stsp 	} repo_tempfiles[3] = { { - 1, - 1 }, { - 1, - 1 }, { - 1, - 1 }, };
 887 13b2bc37 2022-10-23 stsp 	int i;
 888 13b2bc37 2022-10-23 stsp 	size_t datalen;
 889 13b2bc37 2022-10-23 stsp 	struct imsgbuf ibuf;
 890 13b2bc37 2022-10-23 stsp 	struct got_ratelimit rl;
 891 13b2bc37 2022-10-23 stsp 	struct got_pack *pack = NULL;
 892 13b2bc37 2022-10-23 stsp 	off_t pack_filesize = 0;
 893 0ff2c315 2023-01-18 stsp 	uint32_t nobj = 0;
 894 13b2bc37 2022-10-23 stsp 
 895 13b2bc37 2022-10-23 stsp 	log_debug("packfile request received");
 896 13b2bc37 2022-10-23 stsp 
 897 0ff2c315 2023-01-18 stsp 	*have_packfile = 0;
 898 13b2bc37 2022-10-23 stsp 	got_ratelimit_init(&rl, 2, 0);
 899 13b2bc37 2022-10-23 stsp 
 900 13b2bc37 2022-10-23 stsp 	datalen = imsg->hdr.len - IMSG_HEADER_SIZE;
 901 13b2bc37 2022-10-23 stsp 	if (datalen != sizeof(ireq))
 902 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_PRIVSEP_LEN);
 903 13b2bc37 2022-10-23 stsp 	memcpy(&ireq, imsg->data, sizeof(ireq));
 904 13b2bc37 2022-10-23 stsp 
 905 1a52c9bf 2022-12-30 stsp 	if (client->pack_pipe == -1 || client->packidx_fd == -1)
 906 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_PRIVSEP_NO_FD);
 907 13b2bc37 2022-10-23 stsp 
 908 1a52c9bf 2022-12-30 stsp 	imsg_init(&ibuf, client->fd);
 909 13b2bc37 2022-10-23 stsp 
 910 13b2bc37 2022-10-23 stsp 	if (imsg->fd == -1)
 911 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_PRIVSEP_NO_FD);
 912 13b2bc37 2022-10-23 stsp 
 913 1a52c9bf 2022-12-30 stsp 	pack = &client->pack;
 914 13b2bc37 2022-10-23 stsp 	memset(pack, 0, sizeof(*pack));
 915 13b2bc37 2022-10-23 stsp 	pack->fd = imsg->fd;
 916 13b2bc37 2022-10-23 stsp 	err = got_delta_cache_alloc(&pack->delta_cache);
 917 13b2bc37 2022-10-23 stsp 	if (err)
 918 13b2bc37 2022-10-23 stsp 		return err;
 919 13b2bc37 2022-10-23 stsp 
 920 13b2bc37 2022-10-23 stsp 	for (i = 0; i < nitems(repo_tempfiles); i++) {
 921 13b2bc37 2022-10-23 stsp 		struct repo_tempfile *t = &repo_tempfiles[i];
 922 13b2bc37 2022-10-23 stsp 		err = got_repo_temp_fds_get(&t->fd, &t->idx, repo_write.repo);
 923 13b2bc37 2022-10-23 stsp 		if (err)
 924 13b2bc37 2022-10-23 stsp 			goto done;
 925 13b2bc37 2022-10-23 stsp 	}
 926 13b2bc37 2022-10-23 stsp 
 927 13b2bc37 2022-10-23 stsp 	for (i = 0; i < nitems(tempfiles); i++) {
 928 13b2bc37 2022-10-23 stsp 		int fd = dup(repo_tempfiles[i].fd);
 929 13b2bc37 2022-10-23 stsp 		FILE *f;
 930 13b2bc37 2022-10-23 stsp 		if (fd == -1) {
 931 13b2bc37 2022-10-23 stsp 			err = got_error_from_errno("dup");
 932 13b2bc37 2022-10-23 stsp 			goto done;
 933 13b2bc37 2022-10-23 stsp 		}
 934 13b2bc37 2022-10-23 stsp 		f = fdopen(fd, "w+");
 935 13b2bc37 2022-10-23 stsp 		if (f == NULL) {
 936 13b2bc37 2022-10-23 stsp 			err = got_error_from_errno("dup");
 937 13b2bc37 2022-10-23 stsp 			close(fd);
 938 13b2bc37 2022-10-23 stsp 			goto done;
 939 13b2bc37 2022-10-23 stsp 		}
 940 13b2bc37 2022-10-23 stsp 		tempfiles[i] = f;
 941 13b2bc37 2022-10-23 stsp 	}
 942 13b2bc37 2022-10-23 stsp 
 943 13b2bc37 2022-10-23 stsp 	err = gotd_imsg_flush(&ibuf);
 944 13b2bc37 2022-10-23 stsp 	if (err)
 945 13b2bc37 2022-10-23 stsp 		goto done;
 946 13b2bc37 2022-10-23 stsp 
 947 13b2bc37 2022-10-23 stsp 	log_debug("receiving pack data");
 948 0ff2c315 2023-01-18 stsp 	unpack_err = recv_packdata(&pack_filesize, &nobj,
 949 0ff2c315 2023-01-18 stsp 	    client->pack_sha1, client->pack_pipe, pack->fd);
 950 13b2bc37 2022-10-23 stsp 	if (ireq.report_status) {
 951 1a52c9bf 2022-12-30 stsp 		err = report_pack_status(unpack_err);
 952 13b2bc37 2022-10-23 stsp 		if (err) {
 953 13b2bc37 2022-10-23 stsp 			/* Git clients hang up after sending the pack file. */
 954 13b2bc37 2022-10-23 stsp 			if (err->code == GOT_ERR_EOF)
 955 13b2bc37 2022-10-23 stsp 				err = NULL;
 956 13b2bc37 2022-10-23 stsp 		}
 957 13b2bc37 2022-10-23 stsp 	}
 958 13b2bc37 2022-10-23 stsp 	if (unpack_err)
 959 13b2bc37 2022-10-23 stsp 		err = unpack_err;
 960 13b2bc37 2022-10-23 stsp 	if (err)
 961 13b2bc37 2022-10-23 stsp 		goto done;
 962 13b2bc37 2022-10-23 stsp 
 963 13b2bc37 2022-10-23 stsp 	log_debug("pack data received");
 964 0ff2c315 2023-01-18 stsp 
 965 0ff2c315 2023-01-18 stsp 	/*
 966 0ff2c315 2023-01-18 stsp 	 * Clients which are creating new references only will
 967 0ff2c315 2023-01-18 stsp 	 * send us an empty pack file.
 968 0ff2c315 2023-01-18 stsp 	 */
 969 0ff2c315 2023-01-18 stsp 	if (nobj == 0 &&
 970 0ff2c315 2023-01-18 stsp 	    pack_filesize == sizeof(struct got_packfile_hdr) &&
 971 0ff2c315 2023-01-18 stsp 	    client->nref_updates > 0 &&
 972 0ff2c315 2023-01-18 stsp 	    client->nref_updates == client->nref_new)
 973 0ff2c315 2023-01-18 stsp 		goto done;
 974 13b2bc37 2022-10-23 stsp 
 975 13b2bc37 2022-10-23 stsp 	pack->filesize = pack_filesize;
 976 0ff2c315 2023-01-18 stsp 	*have_packfile = 1;
 977 13b2bc37 2022-10-23 stsp 
 978 ad4cc361 2022-10-27 op 	log_debug("begin indexing pack (%lld bytes in size)",
 979 ad4cc361 2022-10-27 op 	    (long long)pack->filesize);
 980 1a52c9bf 2022-12-30 stsp 	err = got_pack_index(pack, client->packidx_fd,
 981 1a52c9bf 2022-12-30 stsp 	    tempfiles[0], tempfiles[1], tempfiles[2], client->pack_sha1,
 982 13b2bc37 2022-10-23 stsp 	    pack_index_progress, NULL, &rl);
 983 13b2bc37 2022-10-23 stsp 	if (err)
 984 13b2bc37 2022-10-23 stsp 		goto done;
 985 13b2bc37 2022-10-23 stsp 	log_debug("done indexing pack");
 986 13b2bc37 2022-10-23 stsp 
 987 1a52c9bf 2022-12-30 stsp 	if (fsync(client->packidx_fd) == -1) {
 988 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("fsync");
 989 13b2bc37 2022-10-23 stsp 		goto done;
 990 13b2bc37 2022-10-23 stsp 	}
 991 1a52c9bf 2022-12-30 stsp 	if (lseek(client->packidx_fd, 0L, SEEK_SET) == -1)
 992 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("lseek");
 993 13b2bc37 2022-10-23 stsp done:
 994 1a52c9bf 2022-12-30 stsp 	if (close(client->pack_pipe) == -1 && err == NULL)
 995 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("close");
 996 1a52c9bf 2022-12-30 stsp 	client->pack_pipe = -1;
 997 13b2bc37 2022-10-23 stsp 	for (i = 0; i < nitems(repo_tempfiles); i++) {
 998 13b2bc37 2022-10-23 stsp 		struct repo_tempfile *t = &repo_tempfiles[i];
 999 13b2bc37 2022-10-23 stsp 		if (t->idx != -1)
1000 13b2bc37 2022-10-23 stsp 			got_repo_temp_fds_put(t->idx, repo_write.repo);
1001 13b2bc37 2022-10-23 stsp 	}
1002 13b2bc37 2022-10-23 stsp 	for (i = 0; i < nitems(tempfiles); i++) {
1003 13b2bc37 2022-10-23 stsp 		if (tempfiles[i] && fclose(tempfiles[i]) == EOF && err == NULL)
1004 13b2bc37 2022-10-23 stsp 			err = got_error_from_errno("fclose");
1005 13b2bc37 2022-10-23 stsp 	}
1006 13b2bc37 2022-10-23 stsp 	if (err)
1007 13b2bc37 2022-10-23 stsp 		got_pack_close(pack);
1008 13b2bc37 2022-10-23 stsp 	imsg_clear(&ibuf);
1009 13b2bc37 2022-10-23 stsp 	return err;
1010 13b2bc37 2022-10-23 stsp }
1011 13b2bc37 2022-10-23 stsp 
1012 13b2bc37 2022-10-23 stsp static const struct got_error *
1013 1a52c9bf 2022-12-30 stsp verify_packfile(void)
1014 13b2bc37 2022-10-23 stsp {
1015 13b2bc37 2022-10-23 stsp 	const struct got_error *err = NULL, *close_err;
1016 1a52c9bf 2022-12-30 stsp 	struct repo_write_client *client = &repo_write_client;
1017 13b2bc37 2022-10-23 stsp 	struct gotd_ref_update *ref_update;
1018 13b2bc37 2022-10-23 stsp 	struct got_packidx *packidx = NULL;
1019 13b2bc37 2022-10-23 stsp 	struct stat sb;
1020 13b2bc37 2022-10-23 stsp 	char *id_str = NULL;
1021 13b2bc37 2022-10-23 stsp 	int idx = -1;
1022 13b2bc37 2022-10-23 stsp 
1023 13b2bc37 2022-10-23 stsp 	if (STAILQ_EMPTY(&client->ref_updates)) {
1024 13b2bc37 2022-10-23 stsp 		return got_error_msg(GOT_ERR_BAD_REQUEST,
1025 13b2bc37 2022-10-23 stsp 		    "cannot verify pack file without any ref-updates");
1026 13b2bc37 2022-10-23 stsp 	}
1027 13b2bc37 2022-10-23 stsp 
1028 13b2bc37 2022-10-23 stsp 	if (client->pack.fd == -1) {
1029 13b2bc37 2022-10-23 stsp 		return got_error_msg(GOT_ERR_BAD_REQUEST,
1030 13b2bc37 2022-10-23 stsp 		    "invalid pack file handle during pack verification");
1031 13b2bc37 2022-10-23 stsp 	}
1032 13b2bc37 2022-10-23 stsp 	if (client->packidx_fd == -1) {
1033 13b2bc37 2022-10-23 stsp 		return got_error_msg(GOT_ERR_BAD_REQUEST,
1034 13b2bc37 2022-10-23 stsp 		    "invalid pack index handle during pack verification");
1035 13b2bc37 2022-10-23 stsp 	}
1036 13b2bc37 2022-10-23 stsp 
1037 13b2bc37 2022-10-23 stsp 	if (fstat(client->packidx_fd, &sb) == -1)
1038 13b2bc37 2022-10-23 stsp 		return got_error_from_errno("pack index fstat");
1039 13b2bc37 2022-10-23 stsp 
1040 13b2bc37 2022-10-23 stsp 	packidx = malloc(sizeof(*packidx));
1041 13b2bc37 2022-10-23 stsp 	memset(packidx, 0, sizeof(*packidx));
1042 13b2bc37 2022-10-23 stsp 	packidx->fd = client->packidx_fd;
1043 13b2bc37 2022-10-23 stsp 	client->packidx_fd = -1;
1044 13b2bc37 2022-10-23 stsp 	packidx->len = sb.st_size;
1045 e26970cc 2023-01-10 op 
1046 13b2bc37 2022-10-23 stsp 	err = got_packidx_init_hdr(packidx, 1, client->pack.filesize);
1047 13b2bc37 2022-10-23 stsp 	if (err)
1048 13b2bc37 2022-10-23 stsp 		return err;
1049 13b2bc37 2022-10-23 stsp 
1050 13b2bc37 2022-10-23 stsp 	STAILQ_FOREACH(ref_update, &client->ref_updates, entry) {
1051 13b2bc37 2022-10-23 stsp 		err = got_object_id_str(&id_str, &ref_update->new_id);
1052 13b2bc37 2022-10-23 stsp 		if (err)
1053 13b2bc37 2022-10-23 stsp 			goto done;
1054 13b2bc37 2022-10-23 stsp 
1055 13b2bc37 2022-10-23 stsp 		idx = got_packidx_get_object_idx(packidx, &ref_update->new_id);
1056 13b2bc37 2022-10-23 stsp 		if (idx == -1) {
1057 13b2bc37 2022-10-23 stsp 			err = got_error_fmt(GOT_ERR_BAD_PACKFILE,
1058 13b2bc37 2022-10-23 stsp 			    "advertised object %s is missing from pack file",
1059 13b2bc37 2022-10-23 stsp 			    id_str);
1060 13b2bc37 2022-10-23 stsp 			goto done;
1061 13b2bc37 2022-10-23 stsp 		}
1062 13b2bc37 2022-10-23 stsp 	}
1063 13b2bc37 2022-10-23 stsp 
1064 e26970cc 2023-01-10 op done:
1065 13b2bc37 2022-10-23 stsp 	close_err = got_packidx_close(packidx);
1066 13b2bc37 2022-10-23 stsp 	if (close_err && err == NULL)
1067 13b2bc37 2022-10-23 stsp 		err = close_err;
1068 13b2bc37 2022-10-23 stsp 	free(id_str);
1069 13b2bc37 2022-10-23 stsp 	return err;
1070 13b2bc37 2022-10-23 stsp }
1071 13b2bc37 2022-10-23 stsp 
1072 13b2bc37 2022-10-23 stsp static const struct got_error *
1073 1a52c9bf 2022-12-30 stsp install_packfile(struct gotd_imsgev *iev)
1074 13b2bc37 2022-10-23 stsp {
1075 1a52c9bf 2022-12-30 stsp 	struct repo_write_client *client = &repo_write_client;
1076 13b2bc37 2022-10-23 stsp 	struct gotd_imsg_packfile_install inst;
1077 13b2bc37 2022-10-23 stsp 	int ret;
1078 13b2bc37 2022-10-23 stsp 
1079 13b2bc37 2022-10-23 stsp 	memset(&inst, 0, sizeof(inst));
1080 13b2bc37 2022-10-23 stsp 	inst.client_id = client->id;
1081 13b2bc37 2022-10-23 stsp 	memcpy(inst.pack_sha1, client->pack_sha1, SHA1_DIGEST_LENGTH);
1082 13b2bc37 2022-10-23 stsp 
1083 13b2bc37 2022-10-23 stsp 	ret = gotd_imsg_compose_event(iev, GOTD_IMSG_PACKFILE_INSTALL,
1084 13b2bc37 2022-10-23 stsp 	    PROC_REPO_WRITE, -1, &inst, sizeof(inst));
1085 13b2bc37 2022-10-23 stsp 	if (ret == -1)
1086 13b2bc37 2022-10-23 stsp 		return got_error_from_errno("imsg_compose PACKFILE_INSTALL");
1087 13b2bc37 2022-10-23 stsp 
1088 13b2bc37 2022-10-23 stsp 	return NULL;
1089 13b2bc37 2022-10-23 stsp }
1090 13b2bc37 2022-10-23 stsp 
1091 13b2bc37 2022-10-23 stsp static const struct got_error *
1092 1a52c9bf 2022-12-30 stsp send_ref_updates_start(int nref_updates, struct gotd_imsgev *iev)
1093 13b2bc37 2022-10-23 stsp {
1094 1a52c9bf 2022-12-30 stsp 	struct repo_write_client *client = &repo_write_client;
1095 13b2bc37 2022-10-23 stsp 	struct gotd_imsg_ref_updates_start istart;
1096 13b2bc37 2022-10-23 stsp 	int ret;
1097 13b2bc37 2022-10-23 stsp 
1098 13b2bc37 2022-10-23 stsp 	memset(&istart, 0, sizeof(istart));
1099 13b2bc37 2022-10-23 stsp 	istart.nref_updates = nref_updates;
1100 13b2bc37 2022-10-23 stsp 	istart.client_id = client->id;
1101 13b2bc37 2022-10-23 stsp 
1102 13b2bc37 2022-10-23 stsp 	ret = gotd_imsg_compose_event(iev, GOTD_IMSG_REF_UPDATES_START,
1103 13b2bc37 2022-10-23 stsp 	    PROC_REPO_WRITE, -1, &istart, sizeof(istart));
1104 13b2bc37 2022-10-23 stsp 	if (ret == -1)
1105 13b2bc37 2022-10-23 stsp 		return got_error_from_errno("imsg_compose REF_UPDATES_START");
1106 13b2bc37 2022-10-23 stsp 
1107 13b2bc37 2022-10-23 stsp 	return NULL;
1108 13b2bc37 2022-10-23 stsp }
1109 13b2bc37 2022-10-23 stsp 
1110 13b2bc37 2022-10-23 stsp 
1111 13b2bc37 2022-10-23 stsp static const struct got_error *
1112 1a52c9bf 2022-12-30 stsp send_ref_update(struct gotd_ref_update *ref_update, struct gotd_imsgev *iev)
1113 13b2bc37 2022-10-23 stsp {
1114 1a52c9bf 2022-12-30 stsp 	struct repo_write_client *client = &repo_write_client;
1115 13b2bc37 2022-10-23 stsp 	struct gotd_imsg_ref_update iref;
1116 13b2bc37 2022-10-23 stsp 	const char *refname = got_ref_get_name(ref_update->ref);
1117 13b2bc37 2022-10-23 stsp 	struct ibuf *wbuf;
1118 13b2bc37 2022-10-23 stsp 	size_t len;
1119 13b2bc37 2022-10-23 stsp 
1120 13b2bc37 2022-10-23 stsp 	memset(&iref, 0, sizeof(iref));
1121 13b2bc37 2022-10-23 stsp 	memcpy(iref.old_id, ref_update->old_id.sha1, SHA1_DIGEST_LENGTH);
1122 13b2bc37 2022-10-23 stsp 	memcpy(iref.new_id, ref_update->new_id.sha1, SHA1_DIGEST_LENGTH);
1123 13b2bc37 2022-10-23 stsp 	iref.ref_is_new = ref_update->ref_is_new;
1124 13b2bc37 2022-10-23 stsp 	iref.client_id = client->id;
1125 13b2bc37 2022-10-23 stsp 	iref.name_len = strlen(refname);
1126 13b2bc37 2022-10-23 stsp 
1127 13b2bc37 2022-10-23 stsp 	len = sizeof(iref) + iref.name_len;
1128 13b2bc37 2022-10-23 stsp 	wbuf = imsg_create(&iev->ibuf, GOTD_IMSG_REF_UPDATE, PROC_REPO_WRITE,
1129 13b2bc37 2022-10-23 stsp 	    repo_write.pid, len);
1130 13b2bc37 2022-10-23 stsp 	if (wbuf == NULL)
1131 13b2bc37 2022-10-23 stsp 		return got_error_from_errno("imsg_create REF_UPDATE");
1132 13b2bc37 2022-10-23 stsp 
1133 13b2bc37 2022-10-23 stsp 	if (imsg_add(wbuf, &iref, sizeof(iref)) == -1)
1134 13b2bc37 2022-10-23 stsp 		return got_error_from_errno("imsg_add REF_UPDATE");
1135 13b2bc37 2022-10-23 stsp 	if (imsg_add(wbuf, refname, iref.name_len) == -1)
1136 13b2bc37 2022-10-23 stsp 		return got_error_from_errno("imsg_add REF_UPDATE");
1137 13b2bc37 2022-10-23 stsp 
1138 13b2bc37 2022-10-23 stsp 	wbuf->fd = -1;
1139 13b2bc37 2022-10-23 stsp 	imsg_close(&iev->ibuf, wbuf);
1140 13b2bc37 2022-10-23 stsp 
1141 13b2bc37 2022-10-23 stsp 	gotd_imsg_event_add(iev);
1142 13b2bc37 2022-10-23 stsp 	return NULL;
1143 13b2bc37 2022-10-23 stsp }
1144 13b2bc37 2022-10-23 stsp 
1145 13b2bc37 2022-10-23 stsp static const struct got_error *
1146 1a52c9bf 2022-12-30 stsp update_refs(struct gotd_imsgev *iev)
1147 13b2bc37 2022-10-23 stsp {
1148 13b2bc37 2022-10-23 stsp 	const struct got_error *err = NULL;
1149 1a52c9bf 2022-12-30 stsp 	struct repo_write_client *client = &repo_write_client;
1150 13b2bc37 2022-10-23 stsp 	struct gotd_ref_update *ref_update;
1151 13b2bc37 2022-10-23 stsp 
1152 1a52c9bf 2022-12-30 stsp 	err = send_ref_updates_start(client->nref_updates, iev);
1153 13b2bc37 2022-10-23 stsp 	if (err)
1154 13b2bc37 2022-10-23 stsp 		return err;
1155 13b2bc37 2022-10-23 stsp 
1156 13b2bc37 2022-10-23 stsp 	STAILQ_FOREACH(ref_update, &client->ref_updates, entry) {
1157 1a52c9bf 2022-12-30 stsp 		err = send_ref_update(ref_update, iev);
1158 13b2bc37 2022-10-23 stsp 		if (err)
1159 13b2bc37 2022-10-23 stsp 			goto done;
1160 13b2bc37 2022-10-23 stsp 	}
1161 13b2bc37 2022-10-23 stsp done:
1162 13b2bc37 2022-10-23 stsp 	return err;
1163 13b2bc37 2022-10-23 stsp }
1164 13b2bc37 2022-10-23 stsp 
1165 13b2bc37 2022-10-23 stsp static const struct got_error *
1166 13b2bc37 2022-10-23 stsp recv_disconnect(struct imsg *imsg)
1167 13b2bc37 2022-10-23 stsp {
1168 13b2bc37 2022-10-23 stsp 	const struct got_error *err = NULL;
1169 13b2bc37 2022-10-23 stsp 	struct gotd_imsg_disconnect idisconnect;
1170 13b2bc37 2022-10-23 stsp 	size_t datalen;
1171 1a52c9bf 2022-12-30 stsp 	int pack_pipe = -1, idxfd = -1;
1172 1a52c9bf 2022-12-30 stsp 	struct repo_write_client *client = &repo_write_client;
1173 13b2bc37 2022-10-23 stsp 
1174 13b2bc37 2022-10-23 stsp 	datalen = imsg->hdr.len - IMSG_HEADER_SIZE;
1175 13b2bc37 2022-10-23 stsp 	if (datalen != sizeof(idisconnect))
1176 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_PRIVSEP_LEN);
1177 13b2bc37 2022-10-23 stsp 	memcpy(&idisconnect, imsg->data, sizeof(idisconnect));
1178 13b2bc37 2022-10-23 stsp 
1179 13b2bc37 2022-10-23 stsp 	log_debug("client disconnecting");
1180 13b2bc37 2022-10-23 stsp 
1181 13b2bc37 2022-10-23 stsp 	while (!STAILQ_EMPTY(&client->ref_updates)) {
1182 13b2bc37 2022-10-23 stsp 		struct gotd_ref_update *ref_update;
1183 13b2bc37 2022-10-23 stsp 		ref_update = STAILQ_FIRST(&client->ref_updates);
1184 13b2bc37 2022-10-23 stsp 		STAILQ_REMOVE_HEAD(&client->ref_updates, entry);
1185 13b2bc37 2022-10-23 stsp 		got_ref_close(ref_update->ref);
1186 13b2bc37 2022-10-23 stsp 		free(ref_update);
1187 13b2bc37 2022-10-23 stsp 	}
1188 13b2bc37 2022-10-23 stsp 	err = got_pack_close(&client->pack);
1189 1a52c9bf 2022-12-30 stsp 	if (client->fd != -1 && close(client->fd) == -1)
1190 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("close");
1191 1a52c9bf 2022-12-30 stsp 	pack_pipe = client->pack_pipe;
1192 7fec5f4a 2022-10-28 stsp 	if (pack_pipe != -1 && close(pack_pipe) == -1 && err == NULL)
1193 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("close");
1194 1a52c9bf 2022-12-30 stsp 	idxfd = client->packidx_fd;
1195 13b2bc37 2022-10-23 stsp 	if (idxfd != -1 && close(idxfd) == -1 && err == NULL)
1196 13b2bc37 2022-10-23 stsp 		err = got_error_from_errno("close");
1197 13b2bc37 2022-10-23 stsp 	return err;
1198 13b2bc37 2022-10-23 stsp }
1199 13b2bc37 2022-10-23 stsp 
1200 13b2bc37 2022-10-23 stsp static const struct got_error *
1201 1a52c9bf 2022-12-30 stsp receive_pack_pipe(struct imsg *imsg, struct gotd_imsgev *iev)
1202 13b2bc37 2022-10-23 stsp {
1203 1a52c9bf 2022-12-30 stsp 	struct repo_write_client *client = &repo_write_client;
1204 13b2bc37 2022-10-23 stsp 	struct gotd_imsg_packfile_pipe ireq;
1205 13b2bc37 2022-10-23 stsp 	size_t datalen;
1206 13b2bc37 2022-10-23 stsp 
1207 13b2bc37 2022-10-23 stsp 	log_debug("receving pack pipe descriptor");
1208 13b2bc37 2022-10-23 stsp 
1209 13b2bc37 2022-10-23 stsp 	if (imsg->fd == -1)
1210 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_PRIVSEP_NO_FD);
1211 13b2bc37 2022-10-23 stsp 
1212 13b2bc37 2022-10-23 stsp 	datalen = imsg->hdr.len - IMSG_HEADER_SIZE;
1213 13b2bc37 2022-10-23 stsp 	if (datalen != sizeof(ireq))
1214 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_PRIVSEP_LEN);
1215 13b2bc37 2022-10-23 stsp 	memcpy(&ireq, imsg->data, sizeof(ireq));
1216 13b2bc37 2022-10-23 stsp 
1217 1a52c9bf 2022-12-30 stsp 	if (client->pack_pipe != -1)
1218 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_PRIVSEP_MSG);
1219 13b2bc37 2022-10-23 stsp 
1220 1a52c9bf 2022-12-30 stsp 	client->pack_pipe = imsg->fd;
1221 13b2bc37 2022-10-23 stsp 	return NULL;
1222 13b2bc37 2022-10-23 stsp }
1223 13b2bc37 2022-10-23 stsp 
1224 13b2bc37 2022-10-23 stsp static const struct got_error *
1225 1a52c9bf 2022-12-30 stsp receive_pack_idx(struct imsg *imsg, struct gotd_imsgev *iev)
1226 13b2bc37 2022-10-23 stsp {
1227 1a52c9bf 2022-12-30 stsp 	struct repo_write_client *client = &repo_write_client;
1228 13b2bc37 2022-10-23 stsp 	struct gotd_imsg_packidx_file ireq;
1229 13b2bc37 2022-10-23 stsp 	size_t datalen;
1230 13b2bc37 2022-10-23 stsp 
1231 13b2bc37 2022-10-23 stsp 	log_debug("receving pack index output file");
1232 13b2bc37 2022-10-23 stsp 
1233 13b2bc37 2022-10-23 stsp 	if (imsg->fd == -1)
1234 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_PRIVSEP_NO_FD);
1235 13b2bc37 2022-10-23 stsp 
1236 13b2bc37 2022-10-23 stsp 	datalen = imsg->hdr.len - IMSG_HEADER_SIZE;
1237 13b2bc37 2022-10-23 stsp 	if (datalen != sizeof(ireq))
1238 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_PRIVSEP_LEN);
1239 13b2bc37 2022-10-23 stsp 	memcpy(&ireq, imsg->data, sizeof(ireq));
1240 13b2bc37 2022-10-23 stsp 
1241 1a52c9bf 2022-12-30 stsp 	if (client->packidx_fd != -1)
1242 13b2bc37 2022-10-23 stsp 		return got_error(GOT_ERR_PRIVSEP_MSG);
1243 13b2bc37 2022-10-23 stsp 
1244 1a52c9bf 2022-12-30 stsp 	client->packidx_fd = imsg->fd;
1245 13b2bc37 2022-10-23 stsp 	return NULL;
1246 13b2bc37 2022-10-23 stsp }
1247 13b2bc37 2022-10-23 stsp 
1248 13b2bc37 2022-10-23 stsp static void
1249 ae7c1b78 2023-01-10 stsp repo_write_dispatch_session(int fd, short event, void *arg)
1250 13b2bc37 2022-10-23 stsp {
1251 13b2bc37 2022-10-23 stsp 	const struct got_error *err = NULL;
1252 13b2bc37 2022-10-23 stsp 	struct gotd_imsgev *iev = arg;
1253 13b2bc37 2022-10-23 stsp 	struct imsgbuf *ibuf = &iev->ibuf;
1254 13b2bc37 2022-10-23 stsp 	struct imsg imsg;
1255 1a52c9bf 2022-12-30 stsp 	struct repo_write_client *client = &repo_write_client;
1256 13b2bc37 2022-10-23 stsp 	ssize_t n;
1257 0ff2c315 2023-01-18 stsp 	int shut = 0, have_packfile = 0;
1258 13b2bc37 2022-10-23 stsp 
1259 13b2bc37 2022-10-23 stsp 	if (event & EV_READ) {
1260 13b2bc37 2022-10-23 stsp 		if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
1261 13b2bc37 2022-10-23 stsp 			fatal("imsg_read error");
1262 13b2bc37 2022-10-23 stsp 		if (n == 0)	/* Connection closed. */
1263 13b2bc37 2022-10-23 stsp 			shut = 1;
1264 13b2bc37 2022-10-23 stsp 	}
1265 13b2bc37 2022-10-23 stsp 
1266 13b2bc37 2022-10-23 stsp 	if (event & EV_WRITE) {
1267 13b2bc37 2022-10-23 stsp 		n = msgbuf_write(&ibuf->w);
1268 13b2bc37 2022-10-23 stsp 		if (n == -1 && errno != EAGAIN)
1269 13b2bc37 2022-10-23 stsp 			fatal("msgbuf_write");
1270 13b2bc37 2022-10-23 stsp 		if (n == 0)	/* Connection closed. */
1271 13b2bc37 2022-10-23 stsp 			shut = 1;
1272 13b2bc37 2022-10-23 stsp 	}
1273 13b2bc37 2022-10-23 stsp 
1274 13b2bc37 2022-10-23 stsp 	for (;;) {
1275 13b2bc37 2022-10-23 stsp 		if ((n = imsg_get(ibuf, &imsg)) == -1)
1276 13b2bc37 2022-10-23 stsp 			fatal("%s: imsg_get error", __func__);
1277 13b2bc37 2022-10-23 stsp 		if (n == 0)	/* No more messages. */
1278 13b2bc37 2022-10-23 stsp 			break;
1279 13b2bc37 2022-10-23 stsp 
1280 1a52c9bf 2022-12-30 stsp 		if (imsg.hdr.type != GOTD_IMSG_LIST_REFS_INTERNAL &&
1281 1a52c9bf 2022-12-30 stsp 		    client->id == 0) {
1282 1a52c9bf 2022-12-30 stsp 			err = got_error(GOT_ERR_PRIVSEP_MSG);
1283 1a52c9bf 2022-12-30 stsp 			break;
1284 1a52c9bf 2022-12-30 stsp 		}
1285 1a52c9bf 2022-12-30 stsp 
1286 13b2bc37 2022-10-23 stsp 		switch (imsg.hdr.type) {
1287 13b2bc37 2022-10-23 stsp 		case GOTD_IMSG_LIST_REFS_INTERNAL:
1288 1a52c9bf 2022-12-30 stsp 			err = list_refs(&imsg);
1289 13b2bc37 2022-10-23 stsp 			if (err)
1290 13b2bc37 2022-10-23 stsp 				log_warnx("%s: ls-refs: %s", repo_write.title,
1291 13b2bc37 2022-10-23 stsp 				    err->msg);
1292 13b2bc37 2022-10-23 stsp 			break;
1293 13b2bc37 2022-10-23 stsp 		case GOTD_IMSG_REF_UPDATE:
1294 1a52c9bf 2022-12-30 stsp 			err = recv_ref_update(&imsg);
1295 13b2bc37 2022-10-23 stsp 			if (err)
1296 13b2bc37 2022-10-23 stsp 				log_warnx("%s: ref-update: %s",
1297 13b2bc37 2022-10-23 stsp 				    repo_write.title, err->msg);
1298 13b2bc37 2022-10-23 stsp 			break;
1299 13b2bc37 2022-10-23 stsp 		case GOTD_IMSG_PACKFILE_PIPE:
1300 1a52c9bf 2022-12-30 stsp 			err = receive_pack_pipe(&imsg, iev);
1301 13b2bc37 2022-10-23 stsp 			if (err) {
1302 13b2bc37 2022-10-23 stsp 				log_warnx("%s: receiving pack pipe: %s",
1303 13b2bc37 2022-10-23 stsp 				    repo_write.title, err->msg);
1304 13b2bc37 2022-10-23 stsp 				break;
1305 13b2bc37 2022-10-23 stsp 			}
1306 13b2bc37 2022-10-23 stsp 			break;
1307 13b2bc37 2022-10-23 stsp 		case GOTD_IMSG_PACKIDX_FILE:
1308 1a52c9bf 2022-12-30 stsp 			err = receive_pack_idx(&imsg, iev);
1309 13b2bc37 2022-10-23 stsp 			if (err) {
1310 13b2bc37 2022-10-23 stsp 				log_warnx("%s: receiving pack index: %s",
1311 13b2bc37 2022-10-23 stsp 				    repo_write.title, err->msg);
1312 13b2bc37 2022-10-23 stsp 				break;
1313 13b2bc37 2022-10-23 stsp 			}
1314 13b2bc37 2022-10-23 stsp 			break;
1315 13b2bc37 2022-10-23 stsp 		case GOTD_IMSG_RECV_PACKFILE:
1316 0ff2c315 2023-01-18 stsp 			err = recv_packfile(&have_packfile, &imsg);
1317 13b2bc37 2022-10-23 stsp 			if (err) {
1318 13b2bc37 2022-10-23 stsp 				log_warnx("%s: receive packfile: %s",
1319 13b2bc37 2022-10-23 stsp 				    repo_write.title, err->msg);
1320 13b2bc37 2022-10-23 stsp 				break;
1321 13b2bc37 2022-10-23 stsp 			}
1322 0ff2c315 2023-01-18 stsp 			if (have_packfile) {
1323 0ff2c315 2023-01-18 stsp 				err = verify_packfile();
1324 0ff2c315 2023-01-18 stsp 				if (err) {
1325 0ff2c315 2023-01-18 stsp 					log_warnx("%s: verify packfile: %s",
1326 0ff2c315 2023-01-18 stsp 					    repo_write.title, err->msg);
1327 0ff2c315 2023-01-18 stsp 					break;
1328 0ff2c315 2023-01-18 stsp 				}
1329 0ff2c315 2023-01-18 stsp 				err = install_packfile(iev);
1330 0ff2c315 2023-01-18 stsp 				if (err) {
1331 0ff2c315 2023-01-18 stsp 					log_warnx("%s: install packfile: %s",
1332 0ff2c315 2023-01-18 stsp 					    repo_write.title, err->msg);
1333 0ff2c315 2023-01-18 stsp 					break;
1334 0ff2c315 2023-01-18 stsp 				}
1335 13b2bc37 2022-10-23 stsp 			}
1336 1a52c9bf 2022-12-30 stsp 			err = update_refs(iev);
1337 13b2bc37 2022-10-23 stsp 			if (err) {
1338 13b2bc37 2022-10-23 stsp 				log_warnx("%s: update refs: %s",
1339 13b2bc37 2022-10-23 stsp 				    repo_write.title, err->msg);
1340 13b2bc37 2022-10-23 stsp 			}
1341 13b2bc37 2022-10-23 stsp 			break;
1342 13b2bc37 2022-10-23 stsp 		case GOTD_IMSG_DISCONNECT:
1343 13b2bc37 2022-10-23 stsp 			err = recv_disconnect(&imsg);
1344 13b2bc37 2022-10-23 stsp 			if (err)
1345 13b2bc37 2022-10-23 stsp 				log_warnx("%s: disconnect: %s",
1346 13b2bc37 2022-10-23 stsp 				    repo_write.title, err->msg);
1347 ae7c1b78 2023-01-10 stsp 			shut = 1;
1348 ae7c1b78 2023-01-10 stsp 			break;
1349 ae7c1b78 2023-01-10 stsp 		default:
1350 ae7c1b78 2023-01-10 stsp 			log_debug("%s: unexpected imsg %d", repo_write.title,
1351 ae7c1b78 2023-01-10 stsp 			    imsg.hdr.type);
1352 ae7c1b78 2023-01-10 stsp 			break;
1353 ae7c1b78 2023-01-10 stsp 		}
1354 ae7c1b78 2023-01-10 stsp 
1355 ae7c1b78 2023-01-10 stsp 		imsg_free(&imsg);
1356 ae7c1b78 2023-01-10 stsp 	}
1357 ae7c1b78 2023-01-10 stsp 
1358 ae7c1b78 2023-01-10 stsp 	if (!shut && check_cancelled(NULL) == NULL) {
1359 ae7c1b78 2023-01-10 stsp 		if (err &&
1360 ae7c1b78 2023-01-10 stsp 		    gotd_imsg_send_error_event(iev, PROC_REPO_WRITE,
1361 ae7c1b78 2023-01-10 stsp 		        client->id, err) == -1) {
1362 ae7c1b78 2023-01-10 stsp 			log_warnx("could not send error to parent: %s",
1363 ae7c1b78 2023-01-10 stsp 			    err->msg);
1364 ae7c1b78 2023-01-10 stsp 		}
1365 ae7c1b78 2023-01-10 stsp 		gotd_imsg_event_add(iev);
1366 ae7c1b78 2023-01-10 stsp 	} else {
1367 ae7c1b78 2023-01-10 stsp 		/* This pipe is dead. Remove its event handler */
1368 ae7c1b78 2023-01-10 stsp 		event_del(&iev->ev);
1369 ae7c1b78 2023-01-10 stsp 		event_loopexit(NULL);
1370 ae7c1b78 2023-01-10 stsp 	}
1371 ae7c1b78 2023-01-10 stsp }
1372 ae7c1b78 2023-01-10 stsp 
1373 ae7c1b78 2023-01-10 stsp static const struct got_error *
1374 ae7c1b78 2023-01-10 stsp recv_connect(struct imsg *imsg)
1375 ae7c1b78 2023-01-10 stsp {
1376 ae7c1b78 2023-01-10 stsp 	struct gotd_imsgev *iev = &repo_write.session_iev;
1377 ae7c1b78 2023-01-10 stsp 	size_t datalen;
1378 ae7c1b78 2023-01-10 stsp 
1379 ae7c1b78 2023-01-10 stsp 	datalen = imsg->hdr.len - IMSG_HEADER_SIZE;
1380 ae7c1b78 2023-01-10 stsp 	if (datalen != 0)
1381 ae7c1b78 2023-01-10 stsp 		return got_error(GOT_ERR_PRIVSEP_LEN);
1382 ae7c1b78 2023-01-10 stsp 	if (imsg->fd == -1)
1383 ae7c1b78 2023-01-10 stsp 		return got_error(GOT_ERR_PRIVSEP_NO_FD);
1384 ae7c1b78 2023-01-10 stsp 
1385 ae7c1b78 2023-01-10 stsp 	if (repo_write.session_fd != -1)
1386 ae7c1b78 2023-01-10 stsp 		return got_error(GOT_ERR_PRIVSEP_MSG);
1387 ae7c1b78 2023-01-10 stsp 
1388 ae7c1b78 2023-01-10 stsp 	repo_write.session_fd = imsg->fd;
1389 ae7c1b78 2023-01-10 stsp 
1390 ae7c1b78 2023-01-10 stsp 	imsg_init(&iev->ibuf, repo_write.session_fd);
1391 ae7c1b78 2023-01-10 stsp 	iev->handler = repo_write_dispatch_session;
1392 ae7c1b78 2023-01-10 stsp 	iev->events = EV_READ;
1393 ae7c1b78 2023-01-10 stsp 	iev->handler_arg = NULL;
1394 ae7c1b78 2023-01-10 stsp 	event_set(&iev->ev, iev->ibuf.fd, EV_READ,
1395 ae7c1b78 2023-01-10 stsp 	    repo_write_dispatch_session, iev);
1396 ae7c1b78 2023-01-10 stsp 	gotd_imsg_event_add(iev);
1397 ae7c1b78 2023-01-10 stsp 
1398 ae7c1b78 2023-01-10 stsp 	return NULL;
1399 ae7c1b78 2023-01-10 stsp }
1400 ae7c1b78 2023-01-10 stsp 
1401 ae7c1b78 2023-01-10 stsp static void
1402 ae7c1b78 2023-01-10 stsp repo_write_dispatch(int fd, short event, void *arg)
1403 ae7c1b78 2023-01-10 stsp {
1404 ae7c1b78 2023-01-10 stsp 	const struct got_error *err = NULL;
1405 ae7c1b78 2023-01-10 stsp 	struct gotd_imsgev *iev = arg;
1406 ae7c1b78 2023-01-10 stsp 	struct imsgbuf *ibuf = &iev->ibuf;
1407 ae7c1b78 2023-01-10 stsp 	struct imsg imsg;
1408 ae7c1b78 2023-01-10 stsp 	ssize_t n;
1409 ae7c1b78 2023-01-10 stsp 	int shut = 0;
1410 ae7c1b78 2023-01-10 stsp 	struct repo_write_client *client = &repo_write_client;
1411 ae7c1b78 2023-01-10 stsp 
1412 ae7c1b78 2023-01-10 stsp 	if (event & EV_READ) {
1413 ae7c1b78 2023-01-10 stsp 		if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
1414 ae7c1b78 2023-01-10 stsp 			fatal("imsg_read error");
1415 ae7c1b78 2023-01-10 stsp 		if (n == 0)	/* Connection closed. */
1416 1a52c9bf 2022-12-30 stsp 			shut = 1;
1417 ae7c1b78 2023-01-10 stsp 	}
1418 ae7c1b78 2023-01-10 stsp 
1419 ae7c1b78 2023-01-10 stsp 	if (event & EV_WRITE) {
1420 ae7c1b78 2023-01-10 stsp 		n = msgbuf_write(&ibuf->w);
1421 ae7c1b78 2023-01-10 stsp 		if (n == -1 && errno != EAGAIN)
1422 ae7c1b78 2023-01-10 stsp 			fatal("msgbuf_write");
1423 ae7c1b78 2023-01-10 stsp 		if (n == 0)	/* Connection closed. */
1424 ae7c1b78 2023-01-10 stsp 			shut = 1;
1425 ae7c1b78 2023-01-10 stsp 	}
1426 ae7c1b78 2023-01-10 stsp 
1427 ae7c1b78 2023-01-10 stsp 	while (err == NULL && check_cancelled(NULL) == NULL) {
1428 ae7c1b78 2023-01-10 stsp 		if ((n = imsg_get(ibuf, &imsg)) == -1)
1429 ae7c1b78 2023-01-10 stsp 			fatal("%s: imsg_get", __func__);
1430 ae7c1b78 2023-01-10 stsp 		if (n == 0)	/* No more messages. */
1431 13b2bc37 2022-10-23 stsp 			break;
1432 ae7c1b78 2023-01-10 stsp 
1433 ae7c1b78 2023-01-10 stsp 		switch (imsg.hdr.type) {
1434 ae7c1b78 2023-01-10 stsp 		case GOTD_IMSG_CONNECT_REPO_CHILD:
1435 ae7c1b78 2023-01-10 stsp 			err = recv_connect(&imsg);
1436 ae7c1b78 2023-01-10 stsp 			break;
1437 13b2bc37 2022-10-23 stsp 		default:
1438 13b2bc37 2022-10-23 stsp 			log_debug("%s: unexpected imsg %d", repo_write.title,
1439 13b2bc37 2022-10-23 stsp 			    imsg.hdr.type);
1440 13b2bc37 2022-10-23 stsp 			break;
1441 13b2bc37 2022-10-23 stsp 		}
1442 13b2bc37 2022-10-23 stsp 
1443 13b2bc37 2022-10-23 stsp 		imsg_free(&imsg);
1444 13b2bc37 2022-10-23 stsp 	}
1445 13b2bc37 2022-10-23 stsp 
1446 13b2bc37 2022-10-23 stsp 	if (!shut && check_cancelled(NULL) == NULL) {
1447 13b2bc37 2022-10-23 stsp 		if (err &&
1448 13b2bc37 2022-10-23 stsp 		    gotd_imsg_send_error_event(iev, PROC_REPO_WRITE,
1449 1a52c9bf 2022-12-30 stsp 		        client->id, err) == -1) {
1450 13b2bc37 2022-10-23 stsp 			log_warnx("could not send error to parent: %s",
1451 13b2bc37 2022-10-23 stsp 			    err->msg);
1452 13b2bc37 2022-10-23 stsp 		}
1453 13b2bc37 2022-10-23 stsp 		gotd_imsg_event_add(iev);
1454 13b2bc37 2022-10-23 stsp 	} else {
1455 13b2bc37 2022-10-23 stsp 		/* This pipe is dead. Remove its event handler */
1456 13b2bc37 2022-10-23 stsp 		event_del(&iev->ev);
1457 13b2bc37 2022-10-23 stsp 		event_loopexit(NULL);
1458 13b2bc37 2022-10-23 stsp 	}
1459 13b2bc37 2022-10-23 stsp }
1460 13b2bc37 2022-10-23 stsp 
1461 13b2bc37 2022-10-23 stsp void
1462 eec68231 2022-12-14 stsp repo_write_main(const char *title, const char *repo_path,
1463 eec68231 2022-12-14 stsp     int *pack_fds, int *temp_fds)
1464 13b2bc37 2022-10-23 stsp {
1465 13b2bc37 2022-10-23 stsp 	const struct got_error *err = NULL;
1466 13b2bc37 2022-10-23 stsp 	struct gotd_imsgev iev;
1467 13b2bc37 2022-10-23 stsp 
1468 13b2bc37 2022-10-23 stsp 	repo_write.title = title;
1469 13b2bc37 2022-10-23 stsp 	repo_write.pid = getpid();
1470 13b2bc37 2022-10-23 stsp 	repo_write.pack_fds = pack_fds;
1471 13b2bc37 2022-10-23 stsp 	repo_write.temp_fds = temp_fds;
1472 ae7c1b78 2023-01-10 stsp 	repo_write.session_fd = -1;
1473 ae7c1b78 2023-01-10 stsp 	repo_write.session_iev.ibuf.fd = -1;
1474 13b2bc37 2022-10-23 stsp 
1475 1a52c9bf 2022-12-30 stsp 	STAILQ_INIT(&repo_write_client.ref_updates);
1476 13b2bc37 2022-10-23 stsp 
1477 eec68231 2022-12-14 stsp 	err = got_repo_open(&repo_write.repo, repo_path, NULL, pack_fds);
1478 13b2bc37 2022-10-23 stsp 	if (err)
1479 13b2bc37 2022-10-23 stsp 		goto done;
1480 13b2bc37 2022-10-23 stsp 	if (!got_repo_is_bare(repo_write.repo)) {
1481 13b2bc37 2022-10-23 stsp 		err = got_error_msg(GOT_ERR_NOT_GIT_REPO,
1482 13b2bc37 2022-10-23 stsp 		    "bare git repository required");
1483 13b2bc37 2022-10-23 stsp 		goto done;
1484 13b2bc37 2022-10-23 stsp 	}
1485 13b2bc37 2022-10-23 stsp 
1486 13b2bc37 2022-10-23 stsp 	got_repo_temp_fds_set(repo_write.repo, temp_fds);
1487 13b2bc37 2022-10-23 stsp 
1488 13b2bc37 2022-10-23 stsp 	signal(SIGINT, catch_sigint);
1489 13b2bc37 2022-10-23 stsp 	signal(SIGTERM, catch_sigterm);
1490 13b2bc37 2022-10-23 stsp 	signal(SIGPIPE, SIG_IGN);
1491 13b2bc37 2022-10-23 stsp 	signal(SIGHUP, SIG_IGN);
1492 13b2bc37 2022-10-23 stsp 
1493 8c6fc146 2022-11-17 stsp 	imsg_init(&iev.ibuf, GOTD_FILENO_MSG_PIPE);
1494 13b2bc37 2022-10-23 stsp 	iev.handler = repo_write_dispatch;
1495 13b2bc37 2022-10-23 stsp 	iev.events = EV_READ;
1496 13b2bc37 2022-10-23 stsp 	iev.handler_arg = NULL;
1497 13b2bc37 2022-10-23 stsp 	event_set(&iev.ev, iev.ibuf.fd, EV_READ, repo_write_dispatch, &iev);
1498 b50a2b46 2022-12-29 stsp 	if (gotd_imsg_compose_event(&iev, GOTD_IMSG_REPO_CHILD_READY,
1499 b50a2b46 2022-12-29 stsp 	    PROC_REPO_WRITE, -1, NULL, 0) == -1) {
1500 b50a2b46 2022-12-29 stsp 		err = got_error_from_errno("imsg compose REPO_CHILD_READY");
1501 13b2bc37 2022-10-23 stsp 		goto done;
1502 13b2bc37 2022-10-23 stsp 	}
1503 13b2bc37 2022-10-23 stsp 
1504 13b2bc37 2022-10-23 stsp 	event_dispatch();
1505 13b2bc37 2022-10-23 stsp done:
1506 13b2bc37 2022-10-23 stsp 	if (err)
1507 13b2bc37 2022-10-23 stsp 		log_warnx("%s: %s", title, err->msg);
1508 13b2bc37 2022-10-23 stsp 	repo_write_shutdown();
1509 13b2bc37 2022-10-23 stsp }
1510 13b2bc37 2022-10-23 stsp 
1511 13b2bc37 2022-10-23 stsp void
1512 13b2bc37 2022-10-23 stsp repo_write_shutdown(void)
1513 13b2bc37 2022-10-23 stsp {
1514 13b2bc37 2022-10-23 stsp 	log_debug("%s: shutting down", repo_write.title);
1515 13b2bc37 2022-10-23 stsp 	if (repo_write.repo)
1516 13b2bc37 2022-10-23 stsp 		got_repo_close(repo_write.repo);
1517 13b2bc37 2022-10-23 stsp 	got_repo_pack_fds_close(repo_write.pack_fds);
1518 8193d041 2022-10-30 stsp 	got_repo_temp_fds_close(repo_write.temp_fds);
1519 ae7c1b78 2023-01-10 stsp 	if (repo_write.session_fd != -1)
1520 ae7c1b78 2023-01-10 stsp 		close(repo_write.session_fd);
1521 13b2bc37 2022-10-23 stsp 	exit(0);
1522 13b2bc37 2022-10-23 stsp }