1 f8a36e22 2021-08-26 stsp /*
  2 f8a36e22 2021-08-26 stsp  * Copyright (c) 2019 Ori Bernstein <ori@openbsd.org>
  3 f8a36e22 2021-08-26 stsp  * Copyright (c) 2021 Stefan Sperling <stsp@openbsd.org>
  4 f8a36e22 2021-08-26 stsp  *
  5 f8a36e22 2021-08-26 stsp  * Permission to use, copy, modify, and distribute this software for any
  6 f8a36e22 2021-08-26 stsp  * purpose with or without fee is hereby granted, provided that the above
  7 f8a36e22 2021-08-26 stsp  * copyright notice and this permission notice appear in all copies.
  8 f8a36e22 2021-08-26 stsp  *
  9 f8a36e22 2021-08-26 stsp  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 10 f8a36e22 2021-08-26 stsp  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 11 f8a36e22 2021-08-26 stsp  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 12 f8a36e22 2021-08-26 stsp  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 13 f8a36e22 2021-08-26 stsp  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 14 f8a36e22 2021-08-26 stsp  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 15 f8a36e22 2021-08-26 stsp  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 16 f8a36e22 2021-08-26 stsp  */
 17 f8a36e22 2021-08-26 stsp 
 18 f8a36e22 2021-08-26 stsp #include <sys/types.h>
 19 f8a36e22 2021-08-26 stsp #include <sys/uio.h>
 20 f8a36e22 2021-08-26 stsp #include <sys/time.h>
 21 f8a36e22 2021-08-26 stsp #include <sys/stat.h>
 22 f8a36e22 2021-08-26 stsp 
 23 f8a36e22 2021-08-26 stsp #include <stdint.h>
 24 f8a36e22 2021-08-26 stsp #include <errno.h>
 25 f8a36e22 2021-08-26 stsp #include <limits.h>
 26 f8a36e22 2021-08-26 stsp #include <signal.h>
 27 f8a36e22 2021-08-26 stsp #include <stdio.h>
 28 f8a36e22 2021-08-26 stsp #include <stdlib.h>
 29 f8a36e22 2021-08-26 stsp #include <string.h>
 30 f8a36e22 2021-08-26 stsp #include <ctype.h>
 31 f8a36e22 2021-08-26 stsp #include <fcntl.h>
 32 f8a36e22 2021-08-26 stsp #include <unistd.h>
 33 f8a36e22 2021-08-26 stsp #include <zlib.h>
 34 f8a36e22 2021-08-26 stsp #include <err.h>
 35 f8a36e22 2021-08-26 stsp 
 36 f8a36e22 2021-08-26 stsp #include "got_error.h"
 37 f8a36e22 2021-08-26 stsp #include "got_object.h"
 38 f8a36e22 2021-08-26 stsp #include "got_path.h"
 39 f8a36e22 2021-08-26 stsp #include "got_version.h"
 40 f8a36e22 2021-08-26 stsp #include "got_fetch.h"
 41 f8a36e22 2021-08-26 stsp #include "got_reference.h"
 42 f8a36e22 2021-08-26 stsp 
 43 f8a36e22 2021-08-26 stsp #include "got_lib_sha1.h"
 44 f8a36e22 2021-08-26 stsp #include "got_lib_delta.h"
 45 f8a36e22 2021-08-26 stsp #include "got_lib_object.h"
 46 f8a36e22 2021-08-26 stsp #include "got_lib_object_parse.h"
 47 f8a36e22 2021-08-26 stsp #include "got_lib_privsep.h"
 48 f8a36e22 2021-08-26 stsp #include "got_lib_pack.h"
 49 f024663d 2021-09-05 stsp #include "got_lib_pkt.h"
 50 bd3d9e54 2021-09-05 stsp #include "got_lib_gitproto.h"
 51 b7e51f67 2022-02-23 thomas #include "got_lib_ratelimit.h"
 52 f8a36e22 2021-08-26 stsp 
 53 f8a36e22 2021-08-26 stsp #ifndef nitems
 54 f8a36e22 2021-08-26 stsp #define nitems(_a)	(sizeof((_a)) / sizeof((_a)[0]))
 55 f8a36e22 2021-08-26 stsp #endif
 56 f8a36e22 2021-08-26 stsp 
 57 f8a36e22 2021-08-26 stsp struct got_object *indexed;
 58 f8a36e22 2021-08-26 stsp static int chattygot;
 59 f8a36e22 2021-08-26 stsp 
 60 f8a36e22 2021-08-26 stsp static const struct got_capability got_capabilities[] = {
 61 f8a36e22 2021-08-26 stsp 	{ GOT_CAPA_AGENT, "got/" GOT_VERSION_STR },
 62 f8a36e22 2021-08-26 stsp 	{ GOT_CAPA_OFS_DELTA, NULL },
 63 f8a36e22 2021-08-26 stsp #if 0
 64 f8a36e22 2021-08-26 stsp 	{ GOT_CAPA_SIDE_BAND_64K, NULL },
 65 f8a36e22 2021-08-26 stsp #endif
 66 f8a36e22 2021-08-26 stsp 	{ GOT_CAPA_REPORT_STATUS, NULL },
 67 f8a36e22 2021-08-26 stsp 	{ GOT_CAPA_DELETE_REFS, NULL },
 68 f8a36e22 2021-08-26 stsp };
 69 f8a36e22 2021-08-26 stsp 
 70 f8a36e22 2021-08-26 stsp static const struct got_error *
 71 b7e51f67 2022-02-23 thomas send_upload_progress(struct imsgbuf *ibuf, off_t bytes,
 72 b7e51f67 2022-02-23 thomas     struct got_ratelimit *rl)
 73 f8a36e22 2021-08-26 stsp {
 74 b7e51f67 2022-02-23 thomas 	const struct got_error *err = NULL;
 75 b7e51f67 2022-02-23 thomas 	int elapsed = 0;
 76 b7e51f67 2022-02-23 thomas 
 77 b7e51f67 2022-02-23 thomas 	if (rl) {
 78 b7e51f67 2022-02-23 thomas 		err = got_ratelimit_check(&elapsed, rl);
 79 b7e51f67 2022-02-23 thomas 		if (err || !elapsed)
 80 b7e51f67 2022-02-23 thomas 			return err;
 81 b7e51f67 2022-02-23 thomas 	}
 82 b7e51f67 2022-02-23 thomas 
 83 f8a36e22 2021-08-26 stsp 	if (imsg_compose(ibuf, GOT_IMSG_SEND_UPLOAD_PROGRESS, 0, 0, -1,
 84 f8a36e22 2021-08-26 stsp 	    &bytes, sizeof(bytes)) == -1)
 85 f8a36e22 2021-08-26 stsp 		return got_error_from_errno(
 86 f8a36e22 2021-08-26 stsp 		    "imsg_compose SEND_UPLOAD_PROGRESS");
 87 f8a36e22 2021-08-26 stsp 
 88 f8a36e22 2021-08-26 stsp 	return got_privsep_flush_imsg(ibuf);
 89 f8a36e22 2021-08-26 stsp }
 90 f8a36e22 2021-08-26 stsp 
 91 f8a36e22 2021-08-26 stsp static const struct got_error *
 92 f8a36e22 2021-08-26 stsp send_pack_request(struct imsgbuf *ibuf)
 93 f8a36e22 2021-08-26 stsp {
 94 f8a36e22 2021-08-26 stsp 	if (imsg_compose(ibuf, GOT_IMSG_SEND_PACK_REQUEST, 0, 0, -1,
 95 f8a36e22 2021-08-26 stsp 	    NULL, 0) == -1)
 96 f8a36e22 2021-08-26 stsp 		return got_error_from_errno("imsg_compose SEND_PACK_REQUEST");
 97 f8a36e22 2021-08-26 stsp 	return got_privsep_flush_imsg(ibuf);
 98 f8a36e22 2021-08-26 stsp }
 99 f8a36e22 2021-08-26 stsp 
100 f8a36e22 2021-08-26 stsp static const struct got_error *
101 f8a36e22 2021-08-26 stsp send_done(struct imsgbuf *ibuf)
102 f8a36e22 2021-08-26 stsp {
103 f8a36e22 2021-08-26 stsp 	if (imsg_compose(ibuf, GOT_IMSG_SEND_DONE, 0, 0, -1, NULL, 0) == -1)
104 f8a36e22 2021-08-26 stsp 		return got_error_from_errno("imsg_compose SEND_DONE");
105 f8a36e22 2021-08-26 stsp 	return got_privsep_flush_imsg(ibuf);
106 f8a36e22 2021-08-26 stsp }
107 f8a36e22 2021-08-26 stsp 
108 f8a36e22 2021-08-26 stsp static const struct got_error *
109 f8a36e22 2021-08-26 stsp recv_packfd(int *packfd, struct imsgbuf *ibuf)
110 f8a36e22 2021-08-26 stsp {
111 f8a36e22 2021-08-26 stsp 	const struct got_error *err;
112 f8a36e22 2021-08-26 stsp 	struct imsg imsg;
113 f8a36e22 2021-08-26 stsp 
114 f8a36e22 2021-08-26 stsp 	*packfd = -1;
115 f8a36e22 2021-08-26 stsp 
116 f8a36e22 2021-08-26 stsp 	err = got_privsep_recv_imsg(&imsg, ibuf, 0);
117 f8a36e22 2021-08-26 stsp 	if (err)
118 f8a36e22 2021-08-26 stsp 		return err;
119 f8a36e22 2021-08-26 stsp 		
120 f8a36e22 2021-08-26 stsp 	if (imsg.hdr.type == GOT_IMSG_STOP) {
121 f8a36e22 2021-08-26 stsp 		err = got_error(GOT_ERR_CANCELLED);
122 f8a36e22 2021-08-26 stsp 		goto done;
123 f8a36e22 2021-08-26 stsp 	}
124 f8a36e22 2021-08-26 stsp 
125 f8a36e22 2021-08-26 stsp 	if (imsg.hdr.type != GOT_IMSG_SEND_PACKFD) {
126 f8a36e22 2021-08-26 stsp 		err = got_error(GOT_ERR_PRIVSEP_MSG);
127 f8a36e22 2021-08-26 stsp 		goto done;
128 f8a36e22 2021-08-26 stsp 	}
129 f8a36e22 2021-08-26 stsp 
130 f8a36e22 2021-08-26 stsp 	if (imsg.hdr.len - IMSG_HEADER_SIZE != 0) {
131 f8a36e22 2021-08-26 stsp 		err = got_error(GOT_ERR_PRIVSEP_LEN);
132 f8a36e22 2021-08-26 stsp 		goto done;
133 f8a36e22 2021-08-26 stsp 	}
134 f8a36e22 2021-08-26 stsp 
135 f8a36e22 2021-08-26 stsp 	*packfd = imsg.fd;
136 f8a36e22 2021-08-26 stsp done:
137 f8a36e22 2021-08-26 stsp 	imsg_free(&imsg);
138 f8a36e22 2021-08-26 stsp 	return err;
139 f8a36e22 2021-08-26 stsp }
140 f8a36e22 2021-08-26 stsp 
141 f8a36e22 2021-08-26 stsp static const struct got_error *
142 f8a36e22 2021-08-26 stsp send_pack_file(int sendfd, int packfd, struct imsgbuf *ibuf)
143 f8a36e22 2021-08-26 stsp {
144 f8a36e22 2021-08-26 stsp 	const struct got_error *err;
145 f8a36e22 2021-08-26 stsp 	unsigned char buf[8192];
146 f8a36e22 2021-08-26 stsp 	ssize_t r, w;
147 f8a36e22 2021-08-26 stsp 	off_t wtotal = 0;
148 b7e51f67 2022-02-23 thomas 	struct got_ratelimit rl;
149 f8a36e22 2021-08-26 stsp 
150 f8a36e22 2021-08-26 stsp 	if (lseek(packfd, 0L, SEEK_SET) == -1)
151 f8a36e22 2021-08-26 stsp 		return got_error_from_errno("lseek");
152 f8a36e22 2021-08-26 stsp 
153 b7e51f67 2022-02-23 thomas 	got_ratelimit_init(&rl, 0, 500);
154 b7e51f67 2022-02-23 thomas 
155 f8a36e22 2021-08-26 stsp 	for (;;) {
156 f8a36e22 2021-08-26 stsp 		r = read(packfd, buf, sizeof(buf));
157 f8a36e22 2021-08-26 stsp 		if (r == -1)
158 f8a36e22 2021-08-26 stsp 			return got_error_from_errno("read");
159 f8a36e22 2021-08-26 stsp 		if (r == 0)
160 f8a36e22 2021-08-26 stsp 			break;
161 f8a36e22 2021-08-26 stsp 		w = write(sendfd, buf, r);
162 f8a36e22 2021-08-26 stsp 		if (w == -1)
163 f8a36e22 2021-08-26 stsp 			return got_error_from_errno("write");
164 f8a36e22 2021-08-26 stsp 		if (w != r)
165 f8a36e22 2021-08-26 stsp 			return got_error(GOT_ERR_IO);
166 f8a36e22 2021-08-26 stsp 		wtotal += w;
167 b7e51f67 2022-02-23 thomas 		err = send_upload_progress(ibuf, wtotal, &rl);
168 f8a36e22 2021-08-26 stsp 		if (err)
169 f8a36e22 2021-08-26 stsp 			return err;
170 f8a36e22 2021-08-26 stsp 	}
171 f8a36e22 2021-08-26 stsp 
172 b7e51f67 2022-02-23 thomas 	return send_upload_progress(ibuf, wtotal, NULL);
173 f8a36e22 2021-08-26 stsp }
174 f8a36e22 2021-08-26 stsp 
175 f8a36e22 2021-08-26 stsp static const struct got_error *
176 f8a36e22 2021-08-26 stsp send_error(const char *buf, size_t len)
177 f8a36e22 2021-08-26 stsp {
178 f8a36e22 2021-08-26 stsp 	static char msg[1024];
179 f8a36e22 2021-08-26 stsp 	size_t i;
180 f8a36e22 2021-08-26 stsp 
181 f8a36e22 2021-08-26 stsp 	for (i = 0; i < len && i < sizeof(msg) - 1; i++) {
182 f8a36e22 2021-08-26 stsp 		if (!isprint(buf[i]))
183 f8a36e22 2021-08-26 stsp 			return got_error_msg(GOT_ERR_BAD_PACKET,
184 f8a36e22 2021-08-26 stsp 			    "non-printable error message received from server");
185 f8a36e22 2021-08-26 stsp 		msg[i] = buf[i];
186 f8a36e22 2021-08-26 stsp 	}
187 f8a36e22 2021-08-26 stsp 	msg[i] = '\0';
188 f8a36e22 2021-08-26 stsp 	return got_error_msg(GOT_ERR_SEND_FAILED, msg);
189 f8a36e22 2021-08-26 stsp }
190 f8a36e22 2021-08-26 stsp 
191 f8a36e22 2021-08-26 stsp static const struct got_error *
192 f8a36e22 2021-08-26 stsp send_their_ref(struct imsgbuf *ibuf, struct got_object_id *refid,
193 f8a36e22 2021-08-26 stsp     const char *refname)
194 f8a36e22 2021-08-26 stsp {
195 f8a36e22 2021-08-26 stsp 	struct ibuf *wbuf;
196 f8a36e22 2021-08-26 stsp 	size_t len, reflen = strlen(refname);
197 f8a36e22 2021-08-26 stsp 
198 f8a36e22 2021-08-26 stsp 	len = sizeof(struct got_imsg_send_remote_ref) + reflen;
199 f8a36e22 2021-08-26 stsp 	if (len >= MAX_IMSGSIZE - IMSG_HEADER_SIZE)
200 f8a36e22 2021-08-26 stsp 		return got_error(GOT_ERR_NO_SPACE);
201 f8a36e22 2021-08-26 stsp 
202 f8a36e22 2021-08-26 stsp 	wbuf = imsg_create(ibuf, GOT_IMSG_SEND_REMOTE_REF, 0, 0, len);
203 f8a36e22 2021-08-26 stsp 	if (wbuf == NULL)
204 f8a36e22 2021-08-26 stsp 		return got_error_from_errno("imsg_create SEND_REMOTE_REF");
205 f8a36e22 2021-08-26 stsp 
206 f8a36e22 2021-08-26 stsp 	/* Keep in sync with struct got_imsg_send_remote_ref definition! */
207 e9f1a409 2022-05-19 thomas 	if (imsg_add(wbuf, refid->sha1, SHA1_DIGEST_LENGTH) == -1)
208 e9f1a409 2022-05-19 thomas 		return got_error_from_errno("imsg_add SEND_REMOTE_REF");
209 e9f1a409 2022-05-19 thomas 	if (imsg_add(wbuf, &reflen, sizeof(reflen)) == -1)
210 e9f1a409 2022-05-19 thomas 		return got_error_from_errno("imsg_add SEND_REMOTE_REF");
211 e9f1a409 2022-05-19 thomas 	if (imsg_add(wbuf, refname, reflen) == -1)
212 e9f1a409 2022-05-19 thomas 		return got_error_from_errno("imsg_add SEND_REMOTE_REF");
213 f8a36e22 2021-08-26 stsp 
214 f8a36e22 2021-08-26 stsp 	wbuf->fd = -1;
215 f8a36e22 2021-08-26 stsp 	imsg_close(ibuf, wbuf);
216 f8a36e22 2021-08-26 stsp 	return got_privsep_flush_imsg(ibuf);
217 f8a36e22 2021-08-26 stsp }
218 f8a36e22 2021-08-26 stsp 
219 f8a36e22 2021-08-26 stsp static const struct got_error *
220 f8a36e22 2021-08-26 stsp send_ref_status(struct imsgbuf *ibuf, const char *refname, int success,
221 f8a36e22 2021-08-26 stsp     struct got_pathlist_head *refs, struct got_pathlist_head *delete_refs)
222 f8a36e22 2021-08-26 stsp {
223 f8a36e22 2021-08-26 stsp 	struct ibuf *wbuf;
224 f8a36e22 2021-08-26 stsp 	size_t len, reflen = strlen(refname);
225 f8a36e22 2021-08-26 stsp 	struct got_pathlist_entry *pe;
226 f8a36e22 2021-08-26 stsp 	int ref_valid = 0;
227 f8a36e22 2021-08-26 stsp 	char *eol;
228 f8a36e22 2021-08-26 stsp 
229 f8a36e22 2021-08-26 stsp 	eol = strchr(refname, '\n');
230 f8a36e22 2021-08-26 stsp 	if (eol == NULL) {
231 f8a36e22 2021-08-26 stsp 		return got_error_msg(GOT_ERR_BAD_PACKET,
232 f8a36e22 2021-08-26 stsp 		    "unexpected message from server");
233 f8a36e22 2021-08-26 stsp 	}
234 f8a36e22 2021-08-26 stsp 	*eol = '\0';
235 f8a36e22 2021-08-26 stsp 
236 f8a36e22 2021-08-26 stsp 	TAILQ_FOREACH(pe, refs, entry) {
237 f8a36e22 2021-08-26 stsp 		if (strcmp(refname, pe->path) == 0) {
238 f8a36e22 2021-08-26 stsp 			ref_valid = 1;
239 f8a36e22 2021-08-26 stsp 			break;
240 f8a36e22 2021-08-26 stsp 		}
241 f8a36e22 2021-08-26 stsp 	}
242 f8a36e22 2021-08-26 stsp 	if (!ref_valid) {
243 f8a36e22 2021-08-26 stsp 		TAILQ_FOREACH(pe, delete_refs, entry) {
244 f8a36e22 2021-08-26 stsp 			if (strcmp(refname, pe->path) == 0) {
245 f8a36e22 2021-08-26 stsp 				ref_valid = 1;
246 f8a36e22 2021-08-26 stsp 				break;
247 f8a36e22 2021-08-26 stsp 			}
248 f8a36e22 2021-08-26 stsp 		}
249 f8a36e22 2021-08-26 stsp 	}
250 f8a36e22 2021-08-26 stsp 	if (!ref_valid) {
251 f8a36e22 2021-08-26 stsp 		return got_error_msg(GOT_ERR_BAD_PACKET,
252 f8a36e22 2021-08-26 stsp 		    "unexpected message from server");
253 f8a36e22 2021-08-26 stsp 	}
254 f8a36e22 2021-08-26 stsp 
255 f8a36e22 2021-08-26 stsp 	len = sizeof(struct got_imsg_send_ref_status) + reflen;
256 f8a36e22 2021-08-26 stsp 	if (len >= MAX_IMSGSIZE - IMSG_HEADER_SIZE)
257 f8a36e22 2021-08-26 stsp 		return got_error(GOT_ERR_NO_SPACE);
258 f8a36e22 2021-08-26 stsp 
259 f8a36e22 2021-08-26 stsp 	wbuf = imsg_create(ibuf, GOT_IMSG_SEND_REF_STATUS,
260 f8a36e22 2021-08-26 stsp 	    0, 0, len);
261 f8a36e22 2021-08-26 stsp 	if (wbuf == NULL)
262 f8a36e22 2021-08-26 stsp 		return got_error_from_errno("imsg_create SEND_REF_STATUS");
263 f8a36e22 2021-08-26 stsp 
264 f8a36e22 2021-08-26 stsp 	/* Keep in sync with struct got_imsg_send_ref_status definition! */
265 e9f1a409 2022-05-19 thomas 	if (imsg_add(wbuf, &success, sizeof(success)) == -1)
266 e9f1a409 2022-05-19 thomas 		return got_error_from_errno("imsg_add SEND_REF_STATUS");
267 e9f1a409 2022-05-19 thomas 	if (imsg_add(wbuf, &reflen, sizeof(reflen)) == -1)
268 e9f1a409 2022-05-19 thomas 		return got_error_from_errno("imsg_add SEND_REF_STATUS");
269 e9f1a409 2022-05-19 thomas 	if (imsg_add(wbuf, refname, reflen) == -1)
270 e9f1a409 2022-05-19 thomas 		return got_error_from_errno("imsg_add SEND_REF_STATUS");
271 f8a36e22 2021-08-26 stsp 
272 f8a36e22 2021-08-26 stsp 	wbuf->fd = -1;
273 f8a36e22 2021-08-26 stsp 	imsg_close(ibuf, wbuf);
274 f8a36e22 2021-08-26 stsp 	return got_privsep_flush_imsg(ibuf);
275 f8a36e22 2021-08-26 stsp }
276 f8a36e22 2021-08-26 stsp 
277 f8a36e22 2021-08-26 stsp static const struct got_error *
278 f8a36e22 2021-08-26 stsp describe_refchange(int *n, int *sent_my_capabilites,
279 f8a36e22 2021-08-26 stsp     const char *my_capabilities, char *buf, size_t bufsize,
280 f8a36e22 2021-08-26 stsp     const char *refname, const char *old_hashstr, const char *new_hashstr)
281 f8a36e22 2021-08-26 stsp {
282 f8a36e22 2021-08-26 stsp 	*n = snprintf(buf, bufsize, "%s %s %s",
283 f8a36e22 2021-08-26 stsp 	    old_hashstr, new_hashstr, refname);
284 f8a36e22 2021-08-26 stsp 	if (*n >= bufsize)
285 f8a36e22 2021-08-26 stsp 		return got_error(GOT_ERR_NO_SPACE);
286 f8a36e22 2021-08-26 stsp 
287 f8a36e22 2021-08-26 stsp 	/*
288 f8a36e22 2021-08-26 stsp 	 * We must announce our capabilities along with the first
289 f8a36e22 2021-08-26 stsp 	 * reference. Unfortunately, the protocol requires an embedded
290 f8a36e22 2021-08-26 stsp 	 * NUL as a separator between reference name and capabilities,
291 f8a36e22 2021-08-26 stsp 	 * which we have to deal with here.
292 f8a36e22 2021-08-26 stsp 	 * It also requires a linefeed for terminating packet data.
293 f8a36e22 2021-08-26 stsp 	 */
294 f8a36e22 2021-08-26 stsp 	if (!*sent_my_capabilites && my_capabilities != NULL) {
295 f8a36e22 2021-08-26 stsp 		int m;
296 f8a36e22 2021-08-26 stsp 		if (*n >= bufsize - 1)
297 f8a36e22 2021-08-26 stsp 			return got_error(GOT_ERR_NO_SPACE);
298 f8a36e22 2021-08-26 stsp 		m = snprintf(buf + *n + 1, /* offset after '\0' */
299 f8a36e22 2021-08-26 stsp 		    bufsize - (*n + 1), "%s\n", my_capabilities);
300 f8a36e22 2021-08-26 stsp 		if (*n + m >= bufsize)
301 f8a36e22 2021-08-26 stsp 			return got_error(GOT_ERR_NO_SPACE);
302 f8a36e22 2021-08-26 stsp 		*n += m;
303 f8a36e22 2021-08-26 stsp 		*sent_my_capabilites = 1;
304 f8a36e22 2021-08-26 stsp 	} else {
305 f8a36e22 2021-08-26 stsp 		*n = strlcat(buf, "\n", bufsize);
306 f8a36e22 2021-08-26 stsp 		if (*n >= bufsize)
307 f8a36e22 2021-08-26 stsp 			return got_error(GOT_ERR_NO_SPACE);
308 f8a36e22 2021-08-26 stsp 	}
309 f8a36e22 2021-08-26 stsp 
310 f8a36e22 2021-08-26 stsp 	return NULL;
311 f8a36e22 2021-08-26 stsp }
312 f8a36e22 2021-08-26 stsp 
313 f8a36e22 2021-08-26 stsp static const struct got_error *
314 f8a36e22 2021-08-26 stsp send_pack(int fd, struct got_pathlist_head *refs,
315 f8a36e22 2021-08-26 stsp     struct got_pathlist_head *delete_refs, struct imsgbuf *ibuf)
316 f8a36e22 2021-08-26 stsp {
317 f8a36e22 2021-08-26 stsp 	const struct got_error *err = NULL;
318 77d7d3bb 2021-09-05 stsp 	char buf[GOT_PKT_MAX];
319 fa8129f7 2022-03-22 thomas 	const unsigned char zero_id[SHA1_DIGEST_LENGTH] = { 0 };
320 f8a36e22 2021-08-26 stsp 	char old_hashstr[SHA1_DIGEST_STRING_LENGTH];
321 f8a36e22 2021-08-26 stsp 	char new_hashstr[SHA1_DIGEST_STRING_LENGTH];
322 f8a36e22 2021-08-26 stsp 	struct got_pathlist_head their_refs;
323 f8a36e22 2021-08-26 stsp 	int is_firstpkt = 1;
324 f8a36e22 2021-08-26 stsp 	int n, nsent = 0;
325 f8a36e22 2021-08-26 stsp 	int packfd = -1;
326 f8a36e22 2021-08-26 stsp 	char *id_str = NULL, *refname = NULL;
327 f8a36e22 2021-08-26 stsp 	struct got_object_id *id = NULL;
328 f8a36e22 2021-08-26 stsp 	char *server_capabilities = NULL, *my_capabilities = NULL;
329 f8a36e22 2021-08-26 stsp 	struct got_pathlist_entry *pe;
330 f8a36e22 2021-08-26 stsp 	int sent_my_capabilites = 0;
331 f8a36e22 2021-08-26 stsp 
332 f8a36e22 2021-08-26 stsp 	TAILQ_INIT(&their_refs);
333 f8a36e22 2021-08-26 stsp 
334 f8a36e22 2021-08-26 stsp 	if (TAILQ_EMPTY(refs) && TAILQ_EMPTY(delete_refs))
335 f8a36e22 2021-08-26 stsp 		return got_error(GOT_ERR_SEND_EMPTY);
336 f8a36e22 2021-08-26 stsp 
337 f8a36e22 2021-08-26 stsp 	while (1) {
338 f024663d 2021-09-05 stsp 		err = got_pkt_readpkt(&n, fd, buf, sizeof(buf), chattygot);
339 f8a36e22 2021-08-26 stsp 		if (err)
340 f8a36e22 2021-08-26 stsp 			goto done;
341 f8a36e22 2021-08-26 stsp 		if (n == 0)
342 f8a36e22 2021-08-26 stsp 			break;
343 f8a36e22 2021-08-26 stsp 		if (n >= 4 && strncmp(buf, "ERR ", 4) == 0) {
344 f8a36e22 2021-08-26 stsp 			err = send_error(&buf[4], n - 4);
345 f8a36e22 2021-08-26 stsp 			goto done;
346 f8a36e22 2021-08-26 stsp 		}
347 35add24a 2021-10-08 thomas 		free(id_str);
348 35add24a 2021-10-08 thomas 		free(refname);
349 bd3d9e54 2021-09-05 stsp 		err = got_gitproto_parse_refline(&id_str, &refname,
350 bd3d9e54 2021-09-05 stsp 		    &server_capabilities, buf, n);
351 f8a36e22 2021-08-26 stsp 		if (err)
352 f8a36e22 2021-08-26 stsp 			goto done;
353 f8a36e22 2021-08-26 stsp 		if (is_firstpkt) {
354 f8a36e22 2021-08-26 stsp 			if (chattygot && server_capabilities[0] != '\0')
355 f8a36e22 2021-08-26 stsp 				fprintf(stderr, "%s: server capabilities: %s\n",
356 f8a36e22 2021-08-26 stsp 				    getprogname(), server_capabilities);
357 bd3d9e54 2021-09-05 stsp 			err = got_gitproto_match_capabilities(&my_capabilities,
358 bd3d9e54 2021-09-05 stsp 			    NULL, server_capabilities, got_capabilities,
359 bd3d9e54 2021-09-05 stsp 			    nitems(got_capabilities));
360 f8a36e22 2021-08-26 stsp 			if (err)
361 f8a36e22 2021-08-26 stsp 				goto done;
362 f8a36e22 2021-08-26 stsp 			if (chattygot)
363 f8a36e22 2021-08-26 stsp 				fprintf(stderr, "%s: my capabilities:%s\n",
364 f8a36e22 2021-08-26 stsp 				    getprogname(), my_capabilities);
365 f8a36e22 2021-08-26 stsp 			is_firstpkt = 0;
366 f8a36e22 2021-08-26 stsp 		}
367 f8a36e22 2021-08-26 stsp 		if (strstr(refname, "^{}")) {
368 f8a36e22 2021-08-26 stsp 			if (chattygot) {
369 f8a36e22 2021-08-26 stsp 				fprintf(stderr, "%s: ignoring %s\n",
370 f8a36e22 2021-08-26 stsp 				    getprogname(), refname);
371 f8a36e22 2021-08-26 stsp 			}
372 f8a36e22 2021-08-26 stsp 			continue;
373 f8a36e22 2021-08-26 stsp 		}
374 f8a36e22 2021-08-26 stsp 
375 f8a36e22 2021-08-26 stsp 		id = malloc(sizeof(*id));
376 f8a36e22 2021-08-26 stsp 		if (id == NULL) {
377 f8a36e22 2021-08-26 stsp 			err = got_error_from_errno("malloc");
378 f8a36e22 2021-08-26 stsp 			goto done;
379 f8a36e22 2021-08-26 stsp 		}
380 f8a36e22 2021-08-26 stsp 		if (!got_parse_sha1_digest(id->sha1, id_str)) {
381 f8a36e22 2021-08-26 stsp 			err = got_error(GOT_ERR_BAD_OBJ_ID_STR);
382 f8a36e22 2021-08-26 stsp 			goto done;
383 f8a36e22 2021-08-26 stsp 		}
384 f8a36e22 2021-08-26 stsp 		err = send_their_ref(ibuf, id, refname);
385 f8a36e22 2021-08-26 stsp 		if (err)
386 f8a36e22 2021-08-26 stsp 			goto done;
387 f8a36e22 2021-08-26 stsp 
388 f8a36e22 2021-08-26 stsp 		err = got_pathlist_append(&their_refs, refname, id);
389 f8a36e22 2021-08-26 stsp 		if (chattygot)
390 f8a36e22 2021-08-26 stsp 			fprintf(stderr, "%s: remote has %s %s\n",
391 f8a36e22 2021-08-26 stsp 			    getprogname(), refname, id_str);
392 f8a36e22 2021-08-26 stsp 		free(id_str);
393 f8a36e22 2021-08-26 stsp 		id_str = NULL;
394 f8a36e22 2021-08-26 stsp 		refname = NULL; /* do not free; owned by their_refs */
395 f8a36e22 2021-08-26 stsp 		id = NULL; /* do not free; owned by their_refs */
396 f8a36e22 2021-08-26 stsp 	}
397 f8a36e22 2021-08-26 stsp 
398 f8a36e22 2021-08-26 stsp 	if (!TAILQ_EMPTY(delete_refs)) {
399 f8a36e22 2021-08-26 stsp 		if (my_capabilities == NULL ||
400 f8a36e22 2021-08-26 stsp 		    strstr(my_capabilities, GOT_CAPA_DELETE_REFS) == NULL) {
401 f8a36e22 2021-08-26 stsp 			err = got_error(GOT_ERR_CAPA_DELETE_REFS);
402 f8a36e22 2021-08-26 stsp 			goto done;
403 f8a36e22 2021-08-26 stsp 		}
404 f8a36e22 2021-08-26 stsp 	}
405 f8a36e22 2021-08-26 stsp 
406 f8a36e22 2021-08-26 stsp 	TAILQ_FOREACH(pe, delete_refs, entry) {
407 f8a36e22 2021-08-26 stsp 		const char *refname = pe->path;
408 f8a36e22 2021-08-26 stsp 		struct got_pathlist_entry *their_pe;
409 f8a36e22 2021-08-26 stsp 		struct got_object_id *their_id = NULL;
410 f8a36e22 2021-08-26 stsp 
411 f8a36e22 2021-08-26 stsp 		TAILQ_FOREACH(their_pe, &their_refs, entry) {
412 f8a36e22 2021-08-26 stsp 			const char *their_refname = their_pe->path;
413 f8a36e22 2021-08-26 stsp 			if (got_path_cmp(refname, their_refname,
414 f8a36e22 2021-08-26 stsp 			    strlen(refname), strlen(their_refname)) == 0) {
415 f8a36e22 2021-08-26 stsp 				their_id = their_pe->data;
416 f8a36e22 2021-08-26 stsp 				break;
417 f8a36e22 2021-08-26 stsp 			}
418 f8a36e22 2021-08-26 stsp 		}
419 f8a36e22 2021-08-26 stsp 		if (their_id == NULL) {
420 f8a36e22 2021-08-26 stsp 			err = got_error_fmt(GOT_ERR_NOT_REF,
421 f8a36e22 2021-08-26 stsp 			    "%s does not exist in remote repository",
422 f8a36e22 2021-08-26 stsp 			    refname);
423 f8a36e22 2021-08-26 stsp 			goto done;
424 f8a36e22 2021-08-26 stsp 		}
425 f8a36e22 2021-08-26 stsp 
426 f8a36e22 2021-08-26 stsp 		got_sha1_digest_to_str(their_id->sha1, old_hashstr,
427 f8a36e22 2021-08-26 stsp 		    sizeof(old_hashstr));
428 f8a36e22 2021-08-26 stsp 		got_sha1_digest_to_str(zero_id, new_hashstr,
429 f8a36e22 2021-08-26 stsp 		    sizeof(new_hashstr));
430 f8a36e22 2021-08-26 stsp 		err = describe_refchange(&n, &sent_my_capabilites,
431 f8a36e22 2021-08-26 stsp 		    my_capabilities, buf, sizeof(buf), refname,
432 f8a36e22 2021-08-26 stsp 		    old_hashstr, new_hashstr);
433 f8a36e22 2021-08-26 stsp 		if (err)
434 f8a36e22 2021-08-26 stsp 			goto done;
435 f024663d 2021-09-05 stsp 		err = got_pkt_writepkt(fd, buf, n, chattygot);
436 f8a36e22 2021-08-26 stsp 		if (err)
437 f8a36e22 2021-08-26 stsp 			goto done;
438 f8a36e22 2021-08-26 stsp 		if (chattygot) {
439 f8a36e22 2021-08-26 stsp 			fprintf(stderr, "%s: deleting %s %s\n",
440 f8a36e22 2021-08-26 stsp 			    getprogname(), refname, old_hashstr);
441 f8a36e22 2021-08-26 stsp 		}
442 f8a36e22 2021-08-26 stsp 		nsent++;
443 f8a36e22 2021-08-26 stsp 	}
444 f8a36e22 2021-08-26 stsp 
445 f8a36e22 2021-08-26 stsp 	TAILQ_FOREACH(pe, refs, entry) {
446 f8a36e22 2021-08-26 stsp 		const char *refname = pe->path;
447 f8a36e22 2021-08-26 stsp 		struct got_object_id *id = pe->data;
448 f8a36e22 2021-08-26 stsp 		struct got_object_id *their_id = NULL;
449 f8a36e22 2021-08-26 stsp 		struct got_pathlist_entry *their_pe;
450 f8a36e22 2021-08-26 stsp 
451 f8a36e22 2021-08-26 stsp 		TAILQ_FOREACH(their_pe, &their_refs, entry) {
452 f8a36e22 2021-08-26 stsp 			const char *their_refname = their_pe->path;
453 f8a36e22 2021-08-26 stsp 			if (got_path_cmp(refname, their_refname,
454 f8a36e22 2021-08-26 stsp 			    strlen(refname), strlen(their_refname)) == 0) {
455 f8a36e22 2021-08-26 stsp 				their_id = their_pe->data;
456 f8a36e22 2021-08-26 stsp 				break;
457 f8a36e22 2021-08-26 stsp 			}
458 f8a36e22 2021-08-26 stsp 		}
459 f8a36e22 2021-08-26 stsp 		if (their_id) {
460 f8a36e22 2021-08-26 stsp 			if (got_object_id_cmp(id, their_id) == 0) {
461 f8a36e22 2021-08-26 stsp 				if (chattygot) {
462 f8a36e22 2021-08-26 stsp 					fprintf(stderr,
463 f8a36e22 2021-08-26 stsp 					    "%s: no change for %s\n",
464 f8a36e22 2021-08-26 stsp 					    getprogname(), refname);
465 f8a36e22 2021-08-26 stsp 				}
466 f8a36e22 2021-08-26 stsp 				continue;
467 f8a36e22 2021-08-26 stsp 			}
468 f8a36e22 2021-08-26 stsp 			got_sha1_digest_to_str(their_id->sha1, old_hashstr,
469 f8a36e22 2021-08-26 stsp 			    sizeof(old_hashstr));
470 f8a36e22 2021-08-26 stsp 		} else {
471 f8a36e22 2021-08-26 stsp 			got_sha1_digest_to_str(zero_id, old_hashstr,
472 f8a36e22 2021-08-26 stsp 			    sizeof(old_hashstr));
473 f8a36e22 2021-08-26 stsp 		}
474 f8a36e22 2021-08-26 stsp 		got_sha1_digest_to_str(id->sha1, new_hashstr,
475 f8a36e22 2021-08-26 stsp 		    sizeof(new_hashstr));
476 f8a36e22 2021-08-26 stsp 		err = describe_refchange(&n, &sent_my_capabilites,
477 f8a36e22 2021-08-26 stsp 		    my_capabilities, buf, sizeof(buf), refname,
478 f8a36e22 2021-08-26 stsp 		    old_hashstr, new_hashstr);
479 f8a36e22 2021-08-26 stsp 		if (err)
480 f8a36e22 2021-08-26 stsp 			goto done;
481 f024663d 2021-09-05 stsp 		err = got_pkt_writepkt(fd, buf, n, chattygot);
482 f8a36e22 2021-08-26 stsp 		if (err)
483 f8a36e22 2021-08-26 stsp 			goto done;
484 f8a36e22 2021-08-26 stsp 		if (chattygot) {
485 f8a36e22 2021-08-26 stsp 			if (their_id) {
486 f8a36e22 2021-08-26 stsp 				fprintf(stderr, "%s: updating %s %s -> %s\n",
487 f8a36e22 2021-08-26 stsp 				    getprogname(), refname, old_hashstr,
488 f8a36e22 2021-08-26 stsp 				    new_hashstr);
489 f8a36e22 2021-08-26 stsp 			} else {
490 f8a36e22 2021-08-26 stsp 				fprintf(stderr, "%s: creating %s %s\n",
491 f8a36e22 2021-08-26 stsp 				    getprogname(), refname, new_hashstr);
492 f8a36e22 2021-08-26 stsp 			}
493 f8a36e22 2021-08-26 stsp 		}
494 f8a36e22 2021-08-26 stsp 		nsent++;
495 f8a36e22 2021-08-26 stsp 	}
496 f024663d 2021-09-05 stsp 	err = got_pkt_flushpkt(fd, chattygot);
497 f8a36e22 2021-08-26 stsp 	if (err)
498 f8a36e22 2021-08-26 stsp 		goto done;
499 f8a36e22 2021-08-26 stsp 
500 f8a36e22 2021-08-26 stsp 	err = send_pack_request(ibuf);
501 f8a36e22 2021-08-26 stsp 	if (err)
502 f8a36e22 2021-08-26 stsp 		goto done;
503 f8a36e22 2021-08-26 stsp 
504 f8a36e22 2021-08-26 stsp 	err = recv_packfd(&packfd, ibuf);
505 f8a36e22 2021-08-26 stsp 	if (err)
506 f8a36e22 2021-08-26 stsp 		goto done;
507 f8a36e22 2021-08-26 stsp 
508 27b75514 2021-08-28 stsp 	if (packfd != -1) {
509 27b75514 2021-08-28 stsp 		err = send_pack_file(fd, packfd, ibuf);
510 27b75514 2021-08-28 stsp 		if (err)
511 27b75514 2021-08-28 stsp 			goto done;
512 27b75514 2021-08-28 stsp 	}
513 f8a36e22 2021-08-26 stsp 
514 f024663d 2021-09-05 stsp 	err = got_pkt_readpkt(&n, fd, buf, sizeof(buf), chattygot);
515 f8a36e22 2021-08-26 stsp 	if (err)
516 f8a36e22 2021-08-26 stsp 		goto done;
517 f8a36e22 2021-08-26 stsp 	if (n >= 4 && strncmp(buf, "ERR ", 4) == 0) {
518 f8a36e22 2021-08-26 stsp 		err = send_error(&buf[4], n - 4);
519 f8a36e22 2021-08-26 stsp 		goto done;
520 f8a36e22 2021-08-26 stsp 	} else if (n < 10 || strncmp(buf, "unpack ok\n", 10) != 0) {
521 f8a36e22 2021-08-26 stsp 		err = got_error_msg(GOT_ERR_BAD_PACKET,
522 f8a36e22 2021-08-26 stsp 		    "unexpected message from server");
523 f8a36e22 2021-08-26 stsp 		goto done;
524 f8a36e22 2021-08-26 stsp 	}
525 f8a36e22 2021-08-26 stsp 
526 f8a36e22 2021-08-26 stsp 	while (nsent > 0) {
527 f024663d 2021-09-05 stsp 		err = got_pkt_readpkt(&n, fd, buf, sizeof(buf), chattygot);
528 f8a36e22 2021-08-26 stsp 		if (err)
529 f8a36e22 2021-08-26 stsp 			goto done;
530 f8a36e22 2021-08-26 stsp 		if (n < 3) {
531 f8a36e22 2021-08-26 stsp 			err = got_error_msg(GOT_ERR_BAD_PACKET,
532 f8a36e22 2021-08-26 stsp 			    "unexpected message from server");
533 f8a36e22 2021-08-26 stsp 			goto done;
534 f8a36e22 2021-08-26 stsp 		} else if (strncmp(buf, "ok ", 3) == 0) {
535 f8a36e22 2021-08-26 stsp 			err = send_ref_status(ibuf, buf + 3, 1,
536 abc59930 2021-09-05 naddy 			    refs, delete_refs);
537 f8a36e22 2021-08-26 stsp 			if (err)
538 f8a36e22 2021-08-26 stsp 				goto done;
539 f8a36e22 2021-08-26 stsp 		} else if (strncmp(buf, "ng ", 3) == 0) {
540 f8a36e22 2021-08-26 stsp 			err = send_ref_status(ibuf, buf + 3, 0,
541 f8a36e22 2021-08-26 stsp 			    refs, delete_refs);
542 f8a36e22 2021-08-26 stsp 			if (err)
543 f8a36e22 2021-08-26 stsp 				goto done;
544 f8a36e22 2021-08-26 stsp 		} else {
545 f8a36e22 2021-08-26 stsp 			err = got_error_msg(GOT_ERR_BAD_PACKET,
546 f8a36e22 2021-08-26 stsp 			    "unexpected message from server");
547 f8a36e22 2021-08-26 stsp 			goto done;
548 f8a36e22 2021-08-26 stsp 		}
549 f8a36e22 2021-08-26 stsp 		nsent--;
550 f8a36e22 2021-08-26 stsp 	}
551 f8a36e22 2021-08-26 stsp 
552 f8a36e22 2021-08-26 stsp 	err = send_done(ibuf);
553 f8a36e22 2021-08-26 stsp done:
554 f8a36e22 2021-08-26 stsp 	TAILQ_FOREACH(pe, &their_refs, entry) {
555 f8a36e22 2021-08-26 stsp 		free((void *)pe->path);
556 f8a36e22 2021-08-26 stsp 		free(pe->data);
557 f8a36e22 2021-08-26 stsp 	}
558 f8a36e22 2021-08-26 stsp 	got_pathlist_free(&their_refs);
559 f8a36e22 2021-08-26 stsp 	free(id_str);
560 f8a36e22 2021-08-26 stsp 	free(id);
561 f8a36e22 2021-08-26 stsp 	free(refname);
562 f8a36e22 2021-08-26 stsp 	free(server_capabilities);
563 f8a36e22 2021-08-26 stsp 	return err;
564 f8a36e22 2021-08-26 stsp }
565 f8a36e22 2021-08-26 stsp 
566 f8a36e22 2021-08-26 stsp int
567 f8a36e22 2021-08-26 stsp main(int argc, char **argv)
568 f8a36e22 2021-08-26 stsp {
569 f8a36e22 2021-08-26 stsp 	const struct got_error *err = NULL;
570 01bb5a15 2021-09-25 thomas.ad 	int sendfd;
571 f8a36e22 2021-08-26 stsp 	struct imsgbuf ibuf;
572 f8a36e22 2021-08-26 stsp 	struct imsg imsg;
573 f8a36e22 2021-08-26 stsp 	struct got_pathlist_head refs;
574 f8a36e22 2021-08-26 stsp 	struct got_pathlist_head delete_refs;
575 f8a36e22 2021-08-26 stsp 	struct got_pathlist_entry *pe;
576 f8a36e22 2021-08-26 stsp 	struct got_imsg_send_request send_req;
577 f8a36e22 2021-08-26 stsp 	struct got_imsg_send_ref href;
578 01bb5a15 2021-09-25 thomas.ad 	size_t datalen, i;
579 f8a36e22 2021-08-26 stsp #if 0
580 f8a36e22 2021-08-26 stsp 	static int attached;
581 f8a36e22 2021-08-26 stsp 	while (!attached)
582 f8a36e22 2021-08-26 stsp 		sleep (1);
583 f8a36e22 2021-08-26 stsp #endif
584 f8a36e22 2021-08-26 stsp 
585 f8a36e22 2021-08-26 stsp 	TAILQ_INIT(&refs);
586 f8a36e22 2021-08-26 stsp 	TAILQ_INIT(&delete_refs);
587 f8a36e22 2021-08-26 stsp 
588 f8a36e22 2021-08-26 stsp 	imsg_init(&ibuf, GOT_IMSG_FD_CHILD);
589 f8a36e22 2021-08-26 stsp #ifndef PROFILE
590 f8a36e22 2021-08-26 stsp 	/* revoke access to most system calls */
591 f8a36e22 2021-08-26 stsp 	if (pledge("stdio recvfd", NULL) == -1) {
592 f8a36e22 2021-08-26 stsp 		err = got_error_from_errno("pledge");
593 f8a36e22 2021-08-26 stsp 		got_privsep_send_error(&ibuf, err);
594 f8a36e22 2021-08-26 stsp 		return 1;
595 f8a36e22 2021-08-26 stsp 	}
596 97799ccd 2022-02-06 thomas 
597 97799ccd 2022-02-06 thomas 	/* revoke fs access */
598 97799ccd 2022-02-06 thomas 	if (landlock_no_fs() == -1) {
599 97799ccd 2022-02-06 thomas 		err = got_error_from_errno("landlock_no_fs");
600 97799ccd 2022-02-06 thomas 		got_privsep_send_error(&ibuf, err);
601 97799ccd 2022-02-06 thomas 		return 1;
602 97799ccd 2022-02-06 thomas 	}
603 5d120ea8 2022-06-23 op 	if (cap_enter() == -1) {
604 5d120ea8 2022-06-23 op 		err = got_error_from_errno("cap_enter");
605 5d120ea8 2022-06-23 op 		got_privsep_send_error(&ibuf, err);
606 5d120ea8 2022-06-23 op 		return 1;
607 5d120ea8 2022-06-23 op 	}
608 f8a36e22 2021-08-26 stsp #endif
609 f8a36e22 2021-08-26 stsp 	if ((err = got_privsep_recv_imsg(&imsg, &ibuf, 0)) != 0) {
610 f8a36e22 2021-08-26 stsp 		if (err->code == GOT_ERR_PRIVSEP_PIPE)
611 f8a36e22 2021-08-26 stsp 			err = NULL;
612 f8a36e22 2021-08-26 stsp 		goto done;
613 f8a36e22 2021-08-26 stsp 	}
614 f8a36e22 2021-08-26 stsp 	if (imsg.hdr.type == GOT_IMSG_STOP)
615 f8a36e22 2021-08-26 stsp 		goto done;
616 f8a36e22 2021-08-26 stsp 	if (imsg.hdr.type != GOT_IMSG_SEND_REQUEST) {
617 f8a36e22 2021-08-26 stsp 		err = got_error(GOT_ERR_PRIVSEP_MSG);
618 f8a36e22 2021-08-26 stsp 		goto done;
619 f8a36e22 2021-08-26 stsp 	}
620 f8a36e22 2021-08-26 stsp 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
621 f8a36e22 2021-08-26 stsp 	if (datalen < sizeof(send_req)) {
622 f8a36e22 2021-08-26 stsp 		err = got_error(GOT_ERR_PRIVSEP_LEN);
623 f8a36e22 2021-08-26 stsp 		goto done;
624 f8a36e22 2021-08-26 stsp 	}
625 f8a36e22 2021-08-26 stsp 	memcpy(&send_req, imsg.data, sizeof(send_req));
626 f8a36e22 2021-08-26 stsp 	sendfd = imsg.fd;
627 f8a36e22 2021-08-26 stsp 	imsg_free(&imsg);
628 f8a36e22 2021-08-26 stsp 
629 f8a36e22 2021-08-26 stsp 	if (send_req.verbosity > 0)
630 f8a36e22 2021-08-26 stsp 		chattygot += send_req.verbosity;
631 f8a36e22 2021-08-26 stsp 
632 f8a36e22 2021-08-26 stsp 	for (i = 0; i < send_req.nrefs; i++) {
633 f8a36e22 2021-08-26 stsp 		struct got_object_id *id;
634 f8a36e22 2021-08-26 stsp 		char *refname;
635 f8a36e22 2021-08-26 stsp 
636 f8a36e22 2021-08-26 stsp 		if ((err = got_privsep_recv_imsg(&imsg, &ibuf, 0)) != 0) {
637 f8a36e22 2021-08-26 stsp 			if (err->code == GOT_ERR_PRIVSEP_PIPE)
638 f8a36e22 2021-08-26 stsp 				err = NULL;
639 f8a36e22 2021-08-26 stsp 			goto done;
640 f8a36e22 2021-08-26 stsp 		}
641 f8a36e22 2021-08-26 stsp 		if (imsg.hdr.type == GOT_IMSG_STOP)
642 f8a36e22 2021-08-26 stsp 			goto done;
643 f8a36e22 2021-08-26 stsp 		if (imsg.hdr.type != GOT_IMSG_SEND_REF) {
644 f8a36e22 2021-08-26 stsp 			err = got_error(GOT_ERR_PRIVSEP_MSG);
645 f8a36e22 2021-08-26 stsp 			goto done;
646 f8a36e22 2021-08-26 stsp 		}
647 f8a36e22 2021-08-26 stsp 		datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
648 f8a36e22 2021-08-26 stsp 		if (datalen < sizeof(href)) {
649 f8a36e22 2021-08-26 stsp 			err = got_error(GOT_ERR_PRIVSEP_LEN);
650 f8a36e22 2021-08-26 stsp 			goto done;
651 f8a36e22 2021-08-26 stsp 		}
652 f8a36e22 2021-08-26 stsp 		memcpy(&href, imsg.data, sizeof(href));
653 f8a36e22 2021-08-26 stsp 		if (datalen - sizeof(href) < href.name_len) {
654 f8a36e22 2021-08-26 stsp 			err = got_error(GOT_ERR_PRIVSEP_LEN);
655 f8a36e22 2021-08-26 stsp 			goto done;
656 f8a36e22 2021-08-26 stsp 		}
657 f8a36e22 2021-08-26 stsp 		refname = malloc(href.name_len + 1);
658 f8a36e22 2021-08-26 stsp 		if (refname == NULL) {
659 f8a36e22 2021-08-26 stsp 			err = got_error_from_errno("malloc");
660 f8a36e22 2021-08-26 stsp 			goto done;
661 f8a36e22 2021-08-26 stsp 		}
662 f8a36e22 2021-08-26 stsp 		memcpy(refname, imsg.data + sizeof(href), href.name_len);
663 f8a36e22 2021-08-26 stsp 		refname[href.name_len] = '\0';
664 f8a36e22 2021-08-26 stsp 
665 f8a36e22 2021-08-26 stsp 		/*
666 f8a36e22 2021-08-26 stsp 		 * Prevent sending of references that won't make any
667 f8a36e22 2021-08-26 stsp 		 * sense outside the local repository's context.
668 f8a36e22 2021-08-26 stsp 		 */
669 f8a36e22 2021-08-26 stsp 		if (strncmp(refname, "refs/got/", 9) == 0 ||
670 f8a36e22 2021-08-26 stsp 		    strncmp(refname, "refs/remotes/", 13) == 0) {
671 f8a36e22 2021-08-26 stsp 			err = got_error_fmt(GOT_ERR_SEND_BAD_REF,
672 f8a36e22 2021-08-26 stsp 			    "%s", refname);
673 f8a36e22 2021-08-26 stsp 			goto done;
674 f8a36e22 2021-08-26 stsp 		}
675 f8a36e22 2021-08-26 stsp 
676 f8a36e22 2021-08-26 stsp 		id = malloc(sizeof(*id));
677 f8a36e22 2021-08-26 stsp 		if (id == NULL) {
678 f8a36e22 2021-08-26 stsp 			free(refname);
679 f8a36e22 2021-08-26 stsp 			err = got_error_from_errno("malloc");
680 f8a36e22 2021-08-26 stsp 			goto done;
681 f8a36e22 2021-08-26 stsp 		}
682 f8a36e22 2021-08-26 stsp 		memcpy(id->sha1, href.id, SHA1_DIGEST_LENGTH);
683 f8a36e22 2021-08-26 stsp 		if (href.delete)
684 f8a36e22 2021-08-26 stsp 			err = got_pathlist_append(&delete_refs, refname, id);
685 f8a36e22 2021-08-26 stsp 		else
686 f8a36e22 2021-08-26 stsp 			err = got_pathlist_append(&refs, refname, id);
687 f8a36e22 2021-08-26 stsp 		if (err) {
688 f8a36e22 2021-08-26 stsp 			free(refname);
689 f8a36e22 2021-08-26 stsp 			free(id);
690 f8a36e22 2021-08-26 stsp 			goto done;
691 f8a36e22 2021-08-26 stsp 		}
692 f8a36e22 2021-08-26 stsp 
693 f8a36e22 2021-08-26 stsp 		imsg_free(&imsg);
694 f8a36e22 2021-08-26 stsp 	}
695 f8a36e22 2021-08-26 stsp 
696 f8a36e22 2021-08-26 stsp 	err = send_pack(sendfd, &refs, &delete_refs, &ibuf);
697 f8a36e22 2021-08-26 stsp done:
698 f8a36e22 2021-08-26 stsp 	TAILQ_FOREACH(pe, &refs, entry) {
699 f8a36e22 2021-08-26 stsp 		free((char *)pe->path);
700 f8a36e22 2021-08-26 stsp 		free(pe->data);
701 f8a36e22 2021-08-26 stsp 	}
702 f8a36e22 2021-08-26 stsp 	got_pathlist_free(&refs);
703 f8a36e22 2021-08-26 stsp 	TAILQ_FOREACH(pe, &delete_refs, entry) {
704 f8a36e22 2021-08-26 stsp 		free((char *)pe->path);
705 f8a36e22 2021-08-26 stsp 		free(pe->data);
706 f8a36e22 2021-08-26 stsp 	}
707 f8a36e22 2021-08-26 stsp 	got_pathlist_free(&delete_refs);
708 f8a36e22 2021-08-26 stsp 	if (sendfd != -1 && close(sendfd) == -1 && err == NULL)
709 f8a36e22 2021-08-26 stsp 		err = got_error_from_errno("close");
710 f8a36e22 2021-08-26 stsp 	if (err != NULL && err->code != GOT_ERR_CANCELLED)  {
711 f8a36e22 2021-08-26 stsp 		fprintf(stderr, "%s: %s\n", getprogname(), err->msg);
712 f8a36e22 2021-08-26 stsp 		got_privsep_send_error(&ibuf, err);
713 f8a36e22 2021-08-26 stsp 	}
714 f8a36e22 2021-08-26 stsp 
715 f8a36e22 2021-08-26 stsp 	exit(0);
716 f8a36e22 2021-08-26 stsp }