op public repos

Blame

Date:: Fri Sep 23 06:01:40 2022 UTC
Message:: show file mode for new added files in work tree diffs Make work tree diffs of local changes on disk display the file mode of new added files like other versions of `got diff` currently do. ok stsp@
Actions:: History | Blob | Raw File
  1 e9ce266e 2022-03-07 op /*
  2 e9ce266e 2022-03-07 op  * Copyright 1986, Larry Wall
  3 f90b7a8c 2022-05-02 op  *
  4 e9ce266e 2022-03-07 op  * Redistribution and use in source and binary forms, with or without
  5 e9ce266e 2022-03-07 op  * modification, are permitted provided that the following condition is met:
  6 e9ce266e 2022-03-07 op  * 1. Redistributions of source code must retain the above copyright notice,
  7 e9ce266e 2022-03-07 op  * this condition and the following disclaimer.
  8 f90b7a8c 2022-05-02 op  *
  9 e9ce266e 2022-03-07 op  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND ANY
 10 e9ce266e 2022-03-07 op  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 11 e9ce266e 2022-03-07 op  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 12 e9ce266e 2022-03-07 op  * DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE FOR
 13 e9ce266e 2022-03-07 op  * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 14 e9ce266e 2022-03-07 op  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 15 e9ce266e 2022-03-07 op  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 16 e9ce266e 2022-03-07 op  * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 17 e9ce266e 2022-03-07 op  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 18 e9ce266e 2022-03-07 op  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 19 e9ce266e 2022-03-07 op  * SUCH DAMAGE.
 20 e9ce266e 2022-03-07 op  */
 21 e9ce266e 2022-03-07 op 
 22 e9ce266e 2022-03-07 op /*
 23 e9ce266e 2022-03-07 op  * Copyright (c) 2022 Omar Polo <op@openbsd.org>
 24 e9ce266e 2022-03-07 op  *
 25 e9ce266e 2022-03-07 op  * Permission to use, copy, modify, and distribute this software for any
 26 e9ce266e 2022-03-07 op  * purpose with or without fee is hereby granted, provided that the above
 27 e9ce266e 2022-03-07 op  * copyright notice and this permission notice appear in all copies.
 28 e9ce266e 2022-03-07 op  *
 29 e9ce266e 2022-03-07 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 30 e9ce266e 2022-03-07 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 31 e9ce266e 2022-03-07 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 32 e9ce266e 2022-03-07 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 33 e9ce266e 2022-03-07 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 34 e9ce266e 2022-03-07 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 35 e9ce266e 2022-03-07 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 36 e9ce266e 2022-03-07 op  */
 37 e9ce266e 2022-03-07 op 
 38 e9ce266e 2022-03-07 op #include <sys/types.h>
 39 e9ce266e 2022-03-07 op #include <sys/queue.h>
 40 e9ce266e 2022-03-07 op #include <sys/uio.h>
 41 e9ce266e 2022-03-07 op 
 42 e9ce266e 2022-03-07 op #include <ctype.h>
 43 e9ce266e 2022-03-07 op #include <limits.h>
 44 e9ce266e 2022-03-07 op #include <paths.h>
 45 e9ce266e 2022-03-07 op #include <sha1.h>
 46 e9ce266e 2022-03-07 op #include <stdint.h>
 47 e9ce266e 2022-03-07 op #include <stdio.h>
 48 e9ce266e 2022-03-07 op #include <stdlib.h>
 49 e9ce266e 2022-03-07 op #include <string.h>
 50 e9ce266e 2022-03-07 op #include <unistd.h>
 51 e9ce266e 2022-03-07 op #include <imsg.h>
 52 e9ce266e 2022-03-07 op 
 53 e9ce266e 2022-03-07 op #include "got_error.h"
 54 e9ce266e 2022-03-07 op #include "got_object.h"
 55 e9ce266e 2022-03-07 op 
 56 e9ce266e 2022-03-07 op #include "got_lib_delta.h"
 57 e9ce266e 2022-03-07 op #include "got_lib_object.h"
 58 e9ce266e 2022-03-07 op #include "got_lib_privsep.h"
 59 55e9459f 2022-06-19 op #include "got_lib_sha1.h"
 60 e9ce266e 2022-03-07 op 
 61 e9ce266e 2022-03-07 op struct imsgbuf ibuf;
 62 e9ce266e 2022-03-07 op 
 63 e9ce266e 2022-03-07 op static const struct got_error *
 64 d8b5af43 2022-06-19 op send_patch(const char *oldname, const char *newname, const char *commitid,
 65 611e5fc2 2022-09-21 mark     const char *blob, const int xbit, int git)
 66 e9ce266e 2022-03-07 op {
 67 e9ce266e 2022-03-07 op 	struct got_imsg_patch p;
 68 e9ce266e 2022-03-07 op 
 69 e9ce266e 2022-03-07 op 	memset(&p, 0, sizeof(p));
 70 e9ce266e 2022-03-07 op 
 71 9d6cabd5 2022-04-07 op 	if (oldname != NULL)
 72 e9ce266e 2022-03-07 op 		strlcpy(p.old, oldname, sizeof(p.old));
 73 7a30b5cb 2022-03-20 op 
 74 e9ce266e 2022-03-07 op 	if (newname != NULL)
 75 e9ce266e 2022-03-07 op 		strlcpy(p.new, newname, sizeof(p.new));
 76 e9ce266e 2022-03-07 op 
 77 497a5915 2022-06-27 op 	if (commitid != NULL)
 78 d8b5af43 2022-06-19 op 		strlcpy(p.cid, commitid, sizeof(p.cid));
 79 497a5915 2022-06-27 op 
 80 497a5915 2022-06-27 op 	if (blob != NULL)
 81 55e9459f 2022-06-19 op 		strlcpy(p.blob, blob, sizeof(p.blob));
 82 55e9459f 2022-06-19 op 
 83 611e5fc2 2022-09-21 mark 	p.xbit = xbit;
 84 9d6cabd5 2022-04-07 op 	p.git = git;
 85 55e9459f 2022-06-19 op 	if (imsg_compose(&ibuf, GOT_IMSG_PATCH, 0, 0, -1, &p, sizeof(p)) == -1)
 86 e9ce266e 2022-03-07 op 		return got_error_from_errno("imsg_compose GOT_IMSG_PATCH");
 87 e9ce266e 2022-03-07 op 	return NULL;
 88 e9ce266e 2022-03-07 op }
 89 e9ce266e 2022-03-07 op 
 90 e9ce266e 2022-03-07 op static const struct got_error *
 91 e9ce266e 2022-03-07 op send_patch_done(void)
 92 e9ce266e 2022-03-07 op {
 93 e9ce266e 2022-03-07 op 	if (imsg_compose(&ibuf, GOT_IMSG_PATCH_DONE, 0, 0, -1,
 94 e9ce266e 2022-03-07 op 	    NULL, 0) == -1)
 95 e9ce266e 2022-03-07 op 		return got_error_from_errno("imsg_compose GOT_IMSG_PATCH_EOF");
 96 08f44789 2022-07-07 op 	return got_privsep_flush_imsg(&ibuf);
 97 e9ce266e 2022-03-07 op }
 98 e9ce266e 2022-03-07 op 
 99 e9ce266e 2022-03-07 op /* based on fetchname from usr.bin/patch/util.c */
100 e9ce266e 2022-03-07 op static const struct got_error *
101 9d6cabd5 2022-04-07 op filename(const char *at, char **name)
102 e9ce266e 2022-03-07 op {
103 9d6cabd5 2022-04-07 op 	char	*tmp, *t;
104 e9ce266e 2022-03-07 op 
105 e9ce266e 2022-03-07 op 	*name = NULL;
106 e9ce266e 2022-03-07 op 	if (*at == '\0')
107 e9ce266e 2022-03-07 op 		return NULL;
108 e9ce266e 2022-03-07 op 
109 e9ce266e 2022-03-07 op 	while (isspace((unsigned char)*at))
110 e9ce266e 2022-03-07 op 		at++;
111 e9ce266e 2022-03-07 op 
112 e9ce266e 2022-03-07 op 	/* files can be created or removed by diffing against /dev/null */
113 46ebad13 2022-03-17 op 	if (!strncmp(at, _PATH_DEVNULL, sizeof(_PATH_DEVNULL) - 1))
114 e9ce266e 2022-03-07 op 		return NULL;
115 e9ce266e 2022-03-07 op 
116 9d6cabd5 2022-04-07 op 	tmp = strdup(at);
117 9d6cabd5 2022-04-07 op 	if (tmp == NULL)
118 e9ce266e 2022-03-07 op 		return got_error_from_errno("strdup");
119 9d6cabd5 2022-04-07 op 	if ((t = strchr(tmp, '\t')) != NULL)
120 9d6cabd5 2022-04-07 op 		*t = '\0';
121 9d6cabd5 2022-04-07 op 	if ((t = strchr(tmp, '\n')) != NULL)
122 9d6cabd5 2022-04-07 op 		*t = '\0';
123 e9ce266e 2022-03-07 op 
124 9d6cabd5 2022-04-07 op 	*name = strdup(tmp);
125 9d6cabd5 2022-04-07 op 	free(tmp);
126 e9ce266e 2022-03-07 op 	if (*name == NULL)
127 e9ce266e 2022-03-07 op 		return got_error_from_errno("strdup");
128 e9ce266e 2022-03-07 op 	return NULL;
129 e9ce266e 2022-03-07 op }
130 e9ce266e 2022-03-07 op 
131 611e5fc2 2022-09-21 mark static int
132 611e5fc2 2022-09-21 mark filexbit(const char *line)
133 611e5fc2 2022-09-21 mark {
134 611e5fc2 2022-09-21 mark 	char *m;
135 611e5fc2 2022-09-21 mark 
136 611e5fc2 2022-09-21 mark 	m = strchr(line, '(');
137 611e5fc2 2022-09-21 mark 	if (m && !strncmp(m + 1, "mode ", 5))
138 611e5fc2 2022-09-21 mark 		return strncmp(m + 6, "755", 3) == 0;
139 611e5fc2 2022-09-21 mark 
140 611e5fc2 2022-09-21 mark 	return 0;
141 611e5fc2 2022-09-21 mark }
142 611e5fc2 2022-09-21 mark 
143 e9ce266e 2022-03-07 op static const struct got_error *
144 db0dfdd7 2022-06-27 op blobid(const char *line, char **blob, int git)
145 55e9459f 2022-06-19 op {
146 55e9459f 2022-06-19 op 	uint8_t digest[SHA1_DIGEST_LENGTH];
147 55e9459f 2022-06-19 op 	size_t len;
148 55e9459f 2022-06-19 op 
149 55e9459f 2022-06-19 op 	*blob = NULL;
150 55e9459f 2022-06-19 op 
151 55e9459f 2022-06-19 op 	len = strspn(line, "0123456789abcdefABCDEF");
152 55e9459f 2022-06-19 op 	if ((*blob = strndup(line, len)) == NULL)
153 55e9459f 2022-06-19 op 		return got_error_from_errno("strndup");
154 55e9459f 2022-06-19 op 
155 db0dfdd7 2022-06-27 op 	if (!git && !got_parse_sha1_digest(digest, *blob)) {
156 55e9459f 2022-06-19 op 		/* silently ignore invalid blob ids */
157 55e9459f 2022-06-19 op 		free(*blob);
158 55e9459f 2022-06-19 op 		*blob = NULL;
159 55e9459f 2022-06-19 op 	}
160 55e9459f 2022-06-19 op 	return NULL;
161 55e9459f 2022-06-19 op }
162 55e9459f 2022-06-19 op 
163 55e9459f 2022-06-19 op static const struct got_error *
164 acf749fc 2022-07-02 op patch_start(int *git, char **cid, FILE *fp)
165 e9ce266e 2022-03-07 op {
166 e9ce266e 2022-03-07 op 	const struct got_error *err = NULL;
167 e9ce266e 2022-03-07 op 	char	*line = NULL;
168 e9ce266e 2022-03-07 op 	size_t	 linesize = 0;
169 e9ce266e 2022-03-07 op 	ssize_t	 linelen;
170 acf749fc 2022-07-02 op 
171 acf749fc 2022-07-02 op 	*git = 0;
172 acf749fc 2022-07-02 op 
173 acf749fc 2022-07-02 op 	while ((linelen = getline(&line, &linesize, fp)) != -1) {
174 acf749fc 2022-07-02 op 		if (!strncmp(line, "diff --git ", 11)) {
175 acf749fc 2022-07-02 op 			*git = 1;
176 acf749fc 2022-07-02 op 			free(*cid);
177 acf749fc 2022-07-02 op 			*cid = NULL;
178 acf749fc 2022-07-02 op 			break;
179 acf749fc 2022-07-02 op 		} else if (!strncmp(line, "diff ", 5)) {
180 acf749fc 2022-07-02 op 			*git = 0;
181 acf749fc 2022-07-02 op 			free(*cid);
182 acf749fc 2022-07-02 op 			*cid = NULL;
183 acf749fc 2022-07-02 op 		} else if (!strncmp(line, "commit - ", 9)) {
184 acf749fc 2022-07-02 op 			free(*cid);
185 acf749fc 2022-07-02 op 			err = blobid(line + 9, cid, *git);
186 acf749fc 2022-07-02 op 			if (err)
187 acf749fc 2022-07-02 op 				break;
188 acf749fc 2022-07-02 op 		} else if (!strncmp(line, "--- ", 4) ||
189 acf749fc 2022-07-02 op 		    !strncmp(line, "+++ ", 4) ||
190 acf749fc 2022-07-02 op 		    !strncmp(line, "blob - ", 7)) {
191 acf749fc 2022-07-02 op 			/* rewind to previous line */
192 acf749fc 2022-07-02 op 			if (fseeko(fp, -linelen, SEEK_CUR) == -1)
193 acf749fc 2022-07-02 op 				err = got_error_from_errno("fseeko");
194 acf749fc 2022-07-02 op 			break;
195 acf749fc 2022-07-02 op 		}
196 acf749fc 2022-07-02 op 	}
197 acf749fc 2022-07-02 op 
198 acf749fc 2022-07-02 op 	free(line);
199 acf749fc 2022-07-02 op 	if (ferror(fp) && err == NULL)
200 acf749fc 2022-07-02 op 		err = got_error_from_errno("getline");
201 acf749fc 2022-07-02 op 	if (feof(fp) && err == NULL)
202 acf749fc 2022-07-02 op 		err = got_error(GOT_ERR_NO_PATCH);
203 acf749fc 2022-07-02 op 	return err;
204 acf749fc 2022-07-02 op }
205 acf749fc 2022-07-02 op 
206 acf749fc 2022-07-02 op static const struct got_error *
207 acf749fc 2022-07-02 op find_diff(int *done, int *next, FILE *fp, int git, const char *commitid)
208 acf749fc 2022-07-02 op {
209 acf749fc 2022-07-02 op 	const struct got_error *err = NULL;
210 acf749fc 2022-07-02 op 	char	*old = NULL, *new = NULL;
211 acf749fc 2022-07-02 op 	char	*blob = NULL;
212 acf749fc 2022-07-02 op 	char	*line = NULL;
213 acf749fc 2022-07-02 op 	size_t	 linesize = 0;
214 acf749fc 2022-07-02 op 	ssize_t	 linelen;
215 611e5fc2 2022-09-21 mark 	int	 create, rename = 0, xbit = 0;
216 e9ce266e 2022-03-07 op 
217 acf749fc 2022-07-02 op 	*done = 0;
218 acf749fc 2022-07-02 op 	*next = 0;
219 e9ce266e 2022-03-07 op 	while ((linelen = getline(&line, &linesize, fp)) != -1) {
220 e9ce266e 2022-03-07 op 		/*
221 e9ce266e 2022-03-07 op 		 * Ignore the Index name like GNU and larry' patch,
222 e9ce266e 2022-03-07 op 		 * we don't have to follow POSIX.
223 e9ce266e 2022-03-07 op 		 */
224 e9ce266e 2022-03-07 op 
225 9d6cabd5 2022-04-07 op 		if (!strncmp(line, "--- ", 4)) {
226 e9ce266e 2022-03-07 op 			free(old);
227 9d6cabd5 2022-04-07 op 			err = filename(line+4, &old);
228 4379a9aa 2022-05-02 op 		} else if (rename && !strncmp(line, "rename from ", 12)) {
229 4379a9aa 2022-05-02 op 			free(old);
230 4379a9aa 2022-05-02 op 			err = filename(line+12, &old);
231 e9ce266e 2022-03-07 op 		} else if (!strncmp(line, "+++ ", 4)) {
232 e9ce266e 2022-03-07 op 			free(new);
233 9d6cabd5 2022-04-07 op 			err = filename(line+4, &new);
234 c87842d5 2022-09-23 mark 		} else if (!strncmp(line, "blob + ", 7) ||
235 c87842d5 2022-09-23 mark 		    !strncmp(line, "file + ", 7)) {
236 611e5fc2 2022-09-21 mark 			xbit = filexbit(line);
237 55e9459f 2022-06-19 op 		} else if (!git && !strncmp(line, "blob - ", 7)) {
238 55e9459f 2022-06-19 op 			free(blob);
239 db0dfdd7 2022-06-27 op 			err = blobid(line + 7, &blob, git);
240 4379a9aa 2022-05-02 op 		} else if (rename && !strncmp(line, "rename to ", 10)) {
241 4379a9aa 2022-05-02 op 			free(new);
242 4379a9aa 2022-05-02 op 			err = filename(line + 10, &new);
243 4379a9aa 2022-05-02 op 		} else if (git && !strncmp(line, "similarity index 100%", 21))
244 4379a9aa 2022-05-02 op 			rename = 1;
245 611e5fc2 2022-09-21 mark 		else if (git && !strncmp(line, "new file mode 100", 17))
246 611e5fc2 2022-09-21 mark 			xbit = strncmp(line + 17, "755", 3) == 0;
247 db0dfdd7 2022-06-27 op 		else if (git && !strncmp(line, "index ", 6)) {
248 db0dfdd7 2022-06-27 op 			free(blob);
249 db0dfdd7 2022-06-27 op 			err = blobid(line + 6, &blob, git);
250 acf749fc 2022-07-02 op 		} else if (!strncmp(line, "diff ", 5)) {
251 acf749fc 2022-07-02 op 			/* rewind to previous line */
252 acf749fc 2022-07-02 op 			if (fseeko(fp, -linelen, SEEK_CUR) == -1)
253 acf749fc 2022-07-02 op 				err = got_error_from_errno("fseeko");
254 acf749fc 2022-07-02 op 			*next = 1;
255 acf749fc 2022-07-02 op 			break;
256 55e9459f 2022-06-19 op 		}
257 e9ce266e 2022-03-07 op 
258 e9ce266e 2022-03-07 op 		if (err)
259 4379a9aa 2022-05-02 op 			break;
260 4379a9aa 2022-05-02 op 
261 4379a9aa 2022-05-02 op 		/*
262 4379a9aa 2022-05-02 op 		 * Git-style diffs with "similarity index 100%" don't
263 4379a9aa 2022-05-02 op 		 * have any hunks and ends with the "rename to foobar"
264 4379a9aa 2022-05-02 op 		 * line.
265 4379a9aa 2022-05-02 op 		 */
266 4379a9aa 2022-05-02 op 		if (rename && old != NULL && new != NULL) {
267 6b7665ac 2022-05-02 op 			*done = 1;
268 d8b5af43 2022-06-19 op 			err = send_patch(old, new, commitid,
269 611e5fc2 2022-09-21 mark 			    blob, xbit, git);
270 e9ce266e 2022-03-07 op 			break;
271 4379a9aa 2022-05-02 op 		}
272 e9ce266e 2022-03-07 op 
273 e9ce266e 2022-03-07 op 		if (!strncmp(line, "@@ -", 4)) {
274 e9ce266e 2022-03-07 op 			create = !strncmp(line+4, "0,0", 3);
275 e9ce266e 2022-03-07 op 			if ((old == NULL && new == NULL) ||
276 e9ce266e 2022-03-07 op 			    (!create && old == NULL))
277 e9ce266e 2022-03-07 op 				err = got_error(GOT_ERR_PATCH_MALFORMED);
278 e9ce266e 2022-03-07 op 			else
279 d8b5af43 2022-06-19 op 				err = send_patch(old, new, commitid,
280 611e5fc2 2022-09-21 mark 				    blob, xbit, git);
281 e9ce266e 2022-03-07 op 
282 e9ce266e 2022-03-07 op 			if (err)
283 e9ce266e 2022-03-07 op 				break;
284 e9ce266e 2022-03-07 op 
285 e9ce266e 2022-03-07 op 			/* rewind to previous line */
286 e45f7eba 2022-05-14 naddy 			if (fseeko(fp, -linelen, SEEK_CUR) == -1)
287 e45f7eba 2022-05-14 naddy 				err = got_error_from_errno("fseeko");
288 e9ce266e 2022-03-07 op 			break;
289 e9ce266e 2022-03-07 op 		}
290 e9ce266e 2022-03-07 op 	}
291 e9ce266e 2022-03-07 op 
292 423faaa6 2022-03-12 op 	free(old);
293 423faaa6 2022-03-12 op 	free(new);
294 55e9459f 2022-06-19 op 	free(blob);
295 e9ce266e 2022-03-07 op 	free(line);
296 e9ce266e 2022-03-07 op 	if (ferror(fp) && err == NULL)
297 e9ce266e 2022-03-07 op 		err = got_error_from_errno("getline");
298 e9ce266e 2022-03-07 op 	if (feof(fp) && err == NULL)
299 e9ce266e 2022-03-07 op 		err = got_error(GOT_ERR_NO_PATCH);
300 e9ce266e 2022-03-07 op 	return err;
301 e9ce266e 2022-03-07 op }
302 e9ce266e 2022-03-07 op 
303 e9ce266e 2022-03-07 op static const struct got_error *
304 35095610 2022-06-14 op strtolnum(char **str, int *n)
305 e9ce266e 2022-03-07 op {
306 e9ce266e 2022-03-07 op 	char		*p, c;
307 e9ce266e 2022-03-07 op 	const char	*errstr;
308 e9ce266e 2022-03-07 op 
309 e9ce266e 2022-03-07 op 	for (p = *str; isdigit((unsigned char)*p); ++p)
310 e9ce266e 2022-03-07 op 		/* nop */;
311 e9ce266e 2022-03-07 op 
312 e9ce266e 2022-03-07 op 	c = *p;
313 e9ce266e 2022-03-07 op 	*p = '\0';
314 e9ce266e 2022-03-07 op 
315 35095610 2022-06-14 op 	*n = strtonum(*str, 0, INT_MAX, &errstr);
316 e9ce266e 2022-03-07 op 	if (errstr != NULL)
317 e9ce266e 2022-03-07 op 		return got_error(GOT_ERR_PATCH_MALFORMED);
318 e9ce266e 2022-03-07 op 
319 e9ce266e 2022-03-07 op 	*p = c;
320 e9ce266e 2022-03-07 op 	*str = p;
321 e9ce266e 2022-03-07 op 	return NULL;
322 e9ce266e 2022-03-07 op }
323 e9ce266e 2022-03-07 op 
324 e9ce266e 2022-03-07 op static const struct got_error *
325 d75b9573 2022-05-02 op parse_hdr(char *s, int *done, struct got_imsg_patch_hunk *hdr)
326 e9ce266e 2022-03-07 op {
327 e9ce266e 2022-03-07 op 	static const struct got_error *err = NULL;
328 e9ce266e 2022-03-07 op 
329 e9ce266e 2022-03-07 op 	if (strncmp(s, "@@ -", 4)) {
330 d75b9573 2022-05-02 op 		*done = 1;
331 e9ce266e 2022-03-07 op 		return NULL;
332 e9ce266e 2022-03-07 op 	}
333 e9ce266e 2022-03-07 op 
334 e9ce266e 2022-03-07 op 	s += 4;
335 e9ce266e 2022-03-07 op 	if (!*s)
336 e9ce266e 2022-03-07 op 		return NULL;
337 e9ce266e 2022-03-07 op 	err = strtolnum(&s, &hdr->oldfrom);
338 e9ce266e 2022-03-07 op 	if (err)
339 e9ce266e 2022-03-07 op 		return err;
340 e9ce266e 2022-03-07 op 	if (*s == ',') {
341 e9ce266e 2022-03-07 op 		s++;
342 e9ce266e 2022-03-07 op 		err = strtolnum(&s, &hdr->oldlines);
343 e9ce266e 2022-03-07 op 		if (err)
344 e9ce266e 2022-03-07 op 			return err;
345 e9ce266e 2022-03-07 op 	} else
346 e9ce266e 2022-03-07 op 		hdr->oldlines = 1;
347 e9ce266e 2022-03-07 op 
348 e9ce266e 2022-03-07 op 	if (*s == ' ')
349 e9ce266e 2022-03-07 op 		s++;
350 e9ce266e 2022-03-07 op 
351 e9ce266e 2022-03-07 op 	if (*s != '+' || !*++s)
352 e9ce266e 2022-03-07 op 		return got_error(GOT_ERR_PATCH_MALFORMED);
353 e9ce266e 2022-03-07 op 	err = strtolnum(&s, &hdr->newfrom);
354 e9ce266e 2022-03-07 op 	if (err)
355 e9ce266e 2022-03-07 op 		return err;
356 e9ce266e 2022-03-07 op 	if (*s == ',') {
357 e9ce266e 2022-03-07 op 		s++;
358 e9ce266e 2022-03-07 op 		err = strtolnum(&s, &hdr->newlines);
359 e9ce266e 2022-03-07 op 		if (err)
360 e9ce266e 2022-03-07 op 			return err;
361 e9ce266e 2022-03-07 op 	} else
362 e9ce266e 2022-03-07 op 		hdr->newlines = 1;
363 e9ce266e 2022-03-07 op 
364 e9ce266e 2022-03-07 op 	if (*s == ' ')
365 e9ce266e 2022-03-07 op 		s++;
366 e9ce266e 2022-03-07 op 
367 e9ce266e 2022-03-07 op 	if (*s != '@')
368 e9ce266e 2022-03-07 op 		return got_error(GOT_ERR_PATCH_MALFORMED);
369 e9ce266e 2022-03-07 op 
370 35095610 2022-06-14 op 	if (hdr->oldfrom >= INT_MAX - hdr->oldlines ||
371 35095610 2022-06-14 op 	    hdr->newfrom >= INT_MAX - hdr->newlines ||
372 e9ce266e 2022-03-07 op 	    /* not so sure about this one */
373 35095610 2022-06-14 op 	    hdr->oldlines >= INT_MAX - hdr->newlines - 1 ||
374 be33dff7 2022-05-13 op 	    (hdr->oldlines == 0 && hdr->newlines == 0))
375 e9ce266e 2022-03-07 op 		return got_error(GOT_ERR_PATCH_MALFORMED);
376 e9ce266e 2022-03-07 op 
377 e9ce266e 2022-03-07 op 	if (hdr->oldlines == 0) {
378 e9ce266e 2022-03-07 op 		/* larry says to "do append rather than insert"; I don't
379 e9ce266e 2022-03-07 op 		 * quite get it, but i trust him.
380 e9ce266e 2022-03-07 op 		 */
381 e9ce266e 2022-03-07 op 		hdr->oldfrom++;
382 e9ce266e 2022-03-07 op 	}
383 e9ce266e 2022-03-07 op 
384 e9ce266e 2022-03-07 op 	if (imsg_compose(&ibuf, GOT_IMSG_PATCH_HUNK, 0, 0, -1,
385 e9ce266e 2022-03-07 op 	    hdr, sizeof(*hdr)) == -1)
386 e9ce266e 2022-03-07 op 		return got_error_from_errno(
387 e9ce266e 2022-03-07 op 		    "imsg_compose GOT_IMSG_PATCH_HUNK");
388 e9ce266e 2022-03-07 op 	return NULL;
389 e9ce266e 2022-03-07 op }
390 e9ce266e 2022-03-07 op 
391 e9ce266e 2022-03-07 op static const struct got_error *
392 e9ce266e 2022-03-07 op send_line(const char *line)
393 e9ce266e 2022-03-07 op {
394 e9ce266e 2022-03-07 op 	static const struct got_error *err = NULL;
395 e9ce266e 2022-03-07 op 	char *p = NULL;
396 e9ce266e 2022-03-07 op 
397 b3c57ab2 2022-03-22 op 	if (*line != '+' && *line != '-' && *line != ' ' && *line != '\\') {
398 e9ce266e 2022-03-07 op 		if (asprintf(&p, " %s", line) == -1)
399 e9ce266e 2022-03-07 op 			return got_error_from_errno("asprintf");
400 e9ce266e 2022-03-07 op 		line = p;
401 e9ce266e 2022-03-07 op 	}
402 e9ce266e 2022-03-07 op 
403 e9ce266e 2022-03-07 op 	if (imsg_compose(&ibuf, GOT_IMSG_PATCH_LINE, 0, 0, -1,
404 46ebad13 2022-03-17 op 	    line, strlen(line) + 1) == -1)
405 e9ce266e 2022-03-07 op 		err = got_error_from_errno(
406 e9ce266e 2022-03-07 op 		    "imsg_compose GOT_IMSG_PATCH_LINE");
407 e9ce266e 2022-03-07 op 
408 e9ce266e 2022-03-07 op 	free(p);
409 e9ce266e 2022-03-07 op 	return err;
410 e9ce266e 2022-03-07 op }
411 e9ce266e 2022-03-07 op 
412 e9ce266e 2022-03-07 op static const struct got_error *
413 b2832778 2022-04-23 op peek_special_line(FILE *fp)
414 b3c57ab2 2022-03-22 op {
415 b3c57ab2 2022-03-22 op 	const struct got_error *err;
416 e260f8ae 2022-03-22 stsp 	int ch;
417 b3c57ab2 2022-03-22 op 
418 b3c57ab2 2022-03-22 op 	ch = fgetc(fp);
419 b3c57ab2 2022-03-22 op 	if (ch != EOF && ch != '\\') {
420 b3c57ab2 2022-03-22 op 		ungetc(ch, fp);
421 b3c57ab2 2022-03-22 op 		return NULL;
422 b3c57ab2 2022-03-22 op 	}
423 b3c57ab2 2022-03-22 op 
424 b2832778 2022-04-23 op 	if (ch == '\\') {
425 b3c57ab2 2022-03-22 op 		err = send_line("\\");
426 b3c57ab2 2022-03-22 op 		if (err)
427 b3c57ab2 2022-03-22 op 			return err;
428 b3c57ab2 2022-03-22 op 	}
429 b3c57ab2 2022-03-22 op 
430 b3c57ab2 2022-03-22 op 	while (ch != EOF && ch != '\n')
431 b3c57ab2 2022-03-22 op 		ch = fgetc(fp);
432 b3c57ab2 2022-03-22 op 
433 b3c57ab2 2022-03-22 op 	if (ch != EOF || feof(fp))
434 b3c57ab2 2022-03-22 op 		return NULL;
435 b3c57ab2 2022-03-22 op 	return got_error(GOT_ERR_IO);
436 b3c57ab2 2022-03-22 op }
437 b3c57ab2 2022-03-22 op 
438 b3c57ab2 2022-03-22 op static const struct got_error *
439 d75b9573 2022-05-02 op parse_hunk(FILE *fp, int *done)
440 e9ce266e 2022-03-07 op {
441 e9ce266e 2022-03-07 op 	static const struct got_error *err = NULL;
442 e9ce266e 2022-03-07 op 	struct got_imsg_patch_hunk hdr;
443 e9ce266e 2022-03-07 op 	char	*line = NULL, ch;
444 e9ce266e 2022-03-07 op 	size_t	 linesize = 0;
445 e9ce266e 2022-03-07 op 	ssize_t	 linelen;
446 35095610 2022-06-14 op 	int	 leftold, leftnew;
447 e9ce266e 2022-03-07 op 
448 e9ce266e 2022-03-07 op 	linelen = getline(&line, &linesize, fp);
449 e9ce266e 2022-03-07 op 	if (linelen == -1) {
450 d75b9573 2022-05-02 op 		*done = 1;
451 e9ce266e 2022-03-07 op 		goto done;
452 e9ce266e 2022-03-07 op 	}
453 e9ce266e 2022-03-07 op 
454 d75b9573 2022-05-02 op 	err = parse_hdr(line, done, &hdr);
455 e9ce266e 2022-03-07 op 	if (err)
456 e9ce266e 2022-03-07 op 		goto done;
457 d75b9573 2022-05-02 op 	if (*done) {
458 e45f7eba 2022-05-14 naddy 		if (fseeko(fp, -linelen, SEEK_CUR) == -1)
459 e45f7eba 2022-05-14 naddy 			err = got_error_from_errno("fseeko");
460 e9ce266e 2022-03-07 op 		goto done;
461 e9ce266e 2022-03-07 op 	}
462 e9ce266e 2022-03-07 op 
463 e9ce266e 2022-03-07 op 	leftold = hdr.oldlines;
464 e9ce266e 2022-03-07 op 	leftnew = hdr.newlines;
465 e9ce266e 2022-03-07 op 
466 e9ce266e 2022-03-07 op 	while (leftold > 0 || leftnew > 0) {
467 e9ce266e 2022-03-07 op 		linelen = getline(&line, &linesize, fp);
468 e9ce266e 2022-03-07 op 		if (linelen == -1) {
469 e9ce266e 2022-03-07 op 			if (ferror(fp)) {
470 e9ce266e 2022-03-07 op 				err = got_error_from_errno("getline");
471 e9ce266e 2022-03-07 op 				goto done;
472 e9ce266e 2022-03-07 op 			}
473 e9ce266e 2022-03-07 op 
474 e9ce266e 2022-03-07 op 			/* trailing newlines may be chopped */
475 e9ce266e 2022-03-07 op 			if (leftold < 3 && leftnew < 3) {
476 d75b9573 2022-05-02 op 				*done = 1;
477 e9ce266e 2022-03-07 op 				break;
478 e9ce266e 2022-03-07 op 			}
479 e9ce266e 2022-03-07 op 
480 e9ce266e 2022-03-07 op 			err = got_error(GOT_ERR_PATCH_TRUNCATED);
481 e9ce266e 2022-03-07 op 			goto done;
482 e9ce266e 2022-03-07 op 		}
483 b3c57ab2 2022-03-22 op 		if (line[linelen - 1] == '\n')
484 b3c57ab2 2022-03-22 op 			line[linelen - 1] = '\0';
485 e9ce266e 2022-03-07 op 
486 e9ce266e 2022-03-07 op 		/* usr.bin/patch allows '=' as context char */
487 e9ce266e 2022-03-07 op 		if (*line == '=')
488 e9ce266e 2022-03-07 op 			*line = ' ';
489 e9ce266e 2022-03-07 op 
490 e9ce266e 2022-03-07 op 		ch = *line;
491 b3c57ab2 2022-03-22 op 		if (ch == '\t' || ch == '\0')
492 e9ce266e 2022-03-07 op 			ch = ' ';	/* the space got eaten */
493 e9ce266e 2022-03-07 op 
494 e9ce266e 2022-03-07 op 		switch (ch) {
495 e9ce266e 2022-03-07 op 		case '-':
496 e9ce266e 2022-03-07 op 			leftold--;
497 e9ce266e 2022-03-07 op 			break;
498 e9ce266e 2022-03-07 op 		case ' ':
499 e9ce266e 2022-03-07 op 			leftold--;
500 e9ce266e 2022-03-07 op 			leftnew--;
501 e9ce266e 2022-03-07 op 			break;
502 e9ce266e 2022-03-07 op 		case '+':
503 e9ce266e 2022-03-07 op 			leftnew--;
504 e9ce266e 2022-03-07 op 			break;
505 e9ce266e 2022-03-07 op 		default:
506 e9ce266e 2022-03-07 op 			err = got_error(GOT_ERR_PATCH_MALFORMED);
507 e9ce266e 2022-03-07 op 			goto done;
508 e9ce266e 2022-03-07 op 		}
509 e9ce266e 2022-03-07 op 
510 e9ce266e 2022-03-07 op 		if (leftold < 0 || leftnew < 0) {
511 e9ce266e 2022-03-07 op 			err = got_error(GOT_ERR_PATCH_MALFORMED);
512 e9ce266e 2022-03-07 op 			goto done;
513 e9ce266e 2022-03-07 op 		}
514 e9ce266e 2022-03-07 op 
515 e9ce266e 2022-03-07 op 		err = send_line(line);
516 e9ce266e 2022-03-07 op 		if (err)
517 e9ce266e 2022-03-07 op 			goto done;
518 b3c57ab2 2022-03-22 op 
519 b3c57ab2 2022-03-22 op 		if ((ch == '-' && leftold == 0) ||
520 b3c57ab2 2022-03-22 op 		    (ch == '+' && leftnew == 0)) {
521 b2832778 2022-04-23 op 			err = peek_special_line(fp);
522 b3c57ab2 2022-03-22 op 			if (err)
523 b3c57ab2 2022-03-22 op 				goto done;
524 b3c57ab2 2022-03-22 op 		}
525 e9ce266e 2022-03-07 op 	}
526 e9ce266e 2022-03-07 op 
527 e9ce266e 2022-03-07 op done:
528 e9ce266e 2022-03-07 op 	free(line);
529 e9ce266e 2022-03-07 op 	return err;
530 e9ce266e 2022-03-07 op }
531 e9ce266e 2022-03-07 op 
532 e9ce266e 2022-03-07 op static const struct got_error *
533 e9ce266e 2022-03-07 op read_patch(struct imsgbuf *ibuf, int fd)
534 e9ce266e 2022-03-07 op {
535 e9ce266e 2022-03-07 op 	const struct got_error *err = NULL;
536 e9ce266e 2022-03-07 op 	FILE *fp;
537 acf749fc 2022-07-02 op 	int git, patch_found = 0;
538 acf749fc 2022-07-02 op 	char *cid = NULL;
539 e9ce266e 2022-03-07 op 
540 e9ce266e 2022-03-07 op 	if ((fp = fdopen(fd, "r")) == NULL) {
541 e9ce266e 2022-03-07 op 		err = got_error_from_errno("fdopen");
542 e9ce266e 2022-03-07 op 		close(fd);
543 e9ce266e 2022-03-07 op 		return err;
544 e9ce266e 2022-03-07 op 	}
545 e9ce266e 2022-03-07 op 
546 acf749fc 2022-07-02 op 	while ((err = patch_start(&git, &cid, fp)) == NULL) {
547 acf749fc 2022-07-02 op 		int done, next;
548 4379a9aa 2022-05-02 op 
549 acf749fc 2022-07-02 op 		err = find_diff(&done, &next, fp, git, cid);
550 e9ce266e 2022-03-07 op 		if (err)
551 e9ce266e 2022-03-07 op 			goto done;
552 acf749fc 2022-07-02 op 		if (next)
553 acf749fc 2022-07-02 op 			continue;
554 e9ce266e 2022-03-07 op 
555 e9ce266e 2022-03-07 op 		patch_found = 1;
556 d75b9573 2022-05-02 op 
557 d75b9573 2022-05-02 op 		while (!done) {
558 d75b9573 2022-05-02 op 			err = parse_hunk(fp, &done);
559 e9ce266e 2022-03-07 op 			if (err)
560 e9ce266e 2022-03-07 op 				goto done;
561 e9ce266e 2022-03-07 op 		}
562 d75b9573 2022-05-02 op 
563 d75b9573 2022-05-02 op 		err = send_patch_done();
564 d75b9573 2022-05-02 op 		if (err)
565 d75b9573 2022-05-02 op 			goto done;
566 e9ce266e 2022-03-07 op 	}
567 e9ce266e 2022-03-07 op 
568 e9ce266e 2022-03-07 op done:
569 e9ce266e 2022-03-07 op 	fclose(fp);
570 acf749fc 2022-07-02 op 	free(cid);
571 e9ce266e 2022-03-07 op 
572 e9ce266e 2022-03-07 op 	/* ignore trailing gibberish */
573 e9ce266e 2022-03-07 op 	if (err != NULL && err->code == GOT_ERR_NO_PATCH && patch_found)
574 e9ce266e 2022-03-07 op 		err = NULL;
575 e9ce266e 2022-03-07 op 
576 e9ce266e 2022-03-07 op 	return err;
577 e9ce266e 2022-03-07 op }
578 e9ce266e 2022-03-07 op 
579 e9ce266e 2022-03-07 op int
580 e9ce266e 2022-03-07 op main(int argc, char **argv)
581 e9ce266e 2022-03-07 op {
582 e9ce266e 2022-03-07 op 	const struct got_error *err = NULL;
583 e9ce266e 2022-03-07 op 	struct imsg imsg;
584 e9ce266e 2022-03-07 op #if 0
585 e9ce266e 2022-03-07 op 	static int attached;
586 e9ce266e 2022-03-07 op 	while (!attached)
587 e9ce266e 2022-03-07 op 		sleep(1);
588 e9ce266e 2022-03-07 op #endif
589 e9ce266e 2022-03-07 op 
590 e9ce266e 2022-03-07 op 	imsg_init(&ibuf, GOT_IMSG_FD_CHILD);
591 e9ce266e 2022-03-07 op #ifndef PROFILE
592 e9ce266e 2022-03-07 op 	/* revoke access to most system calls */
593 e9ce266e 2022-03-07 op 	if (pledge("stdio recvfd", NULL) == -1) {
594 e9ce266e 2022-03-07 op 		err = got_error_from_errno("pledge");
595 e9ce266e 2022-03-07 op 		got_privsep_send_error(&ibuf, err);
596 e9ce266e 2022-03-07 op 		return 1;
597 e9ce266e 2022-03-07 op 	}
598 e9ce266e 2022-03-07 op #endif
599 e9ce266e 2022-03-07 op 
600 e9ce266e 2022-03-07 op 	err = got_privsep_recv_imsg(&imsg, &ibuf, 0);
601 e9ce266e 2022-03-07 op 	if (err)
602 e9ce266e 2022-03-07 op 		goto done;
603 e9ce266e 2022-03-07 op 	if (imsg.hdr.type != GOT_IMSG_PATCH_FILE || imsg.fd == -1) {
604 e9ce266e 2022-03-07 op 		err = got_error(GOT_ERR_PRIVSEP_MSG);
605 e9ce266e 2022-03-07 op 		goto done;
606 e9ce266e 2022-03-07 op 	}
607 e9ce266e 2022-03-07 op 
608 e9ce266e 2022-03-07 op 	err = read_patch(&ibuf, imsg.fd);
609 e9ce266e 2022-03-07 op 	if (err)
610 e9ce266e 2022-03-07 op 		goto done;
611 e9ce266e 2022-03-07 op 	if (imsg_compose(&ibuf, GOT_IMSG_PATCH_EOF, 0, 0, -1,
612 e9ce266e 2022-03-07 op 	    NULL, 0) == -1) {
613 e9ce266e 2022-03-07 op 		err = got_error_from_errno("imsg_compose GOT_IMSG_PATCH_EOF");
614 e9ce266e 2022-03-07 op 		goto done;
615 e9ce266e 2022-03-07 op 	}
616 e9ce266e 2022-03-07 op 	err = got_privsep_flush_imsg(&ibuf);
617 e9ce266e 2022-03-07 op done:
618 e9ce266e 2022-03-07 op 	imsg_free(&imsg);
619 e9ce266e 2022-03-07 op 	if (err != NULL) {
620 e9ce266e 2022-03-07 op 		got_privsep_send_error(&ibuf, err);
621 e9ce266e 2022-03-07 op 		err = NULL;
622 e9ce266e 2022-03-07 op 	}
623 e9ce266e 2022-03-07 op 	if (close(GOT_IMSG_FD_CHILD) == -1 && err == NULL)
624 e9ce266e 2022-03-07 op 		err = got_error_from_errno("close");
625 e9ce266e 2022-03-07 op 	if (err && err->code != GOT_ERR_PRIVSEP_PIPE)
626 e9ce266e 2022-03-07 op 		fprintf(stderr, "%s: %s\n", getprogname(), err->msg);
627 e9ce266e 2022-03-07 op 	return err ? 1 : 0;
628 e9ce266e 2022-03-07 op }