Game of Trees Repos

Blame

Date:: Sat Jul 2 13:56:21 2022 UTC
Message:: refactor the patch parser Introduce a patch_start routine that finds the next "diff" header (if there is one); the idea is to persist some state (commit id and wether it's a "git diff") while processing the content of the diff. It's needed because in the case of 'got diff' some information like the commit id are only present once at the beginning. As a consequence, the patch parser becomes slightly more robust (concatenating diffs produced by different means shouldn't confuse it anymore) and drops the support for "old" got diffs, the ones previous the introduction of the "commit -/+" header. ok tracey@
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 d8b5af43 2022-06-19 op     const char *blob, 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 9d6cabd5 2022-04-07 op 	p.git = git;
 84 55e9459f 2022-06-19 op 	if (imsg_compose(&ibuf, GOT_IMSG_PATCH, 0, 0, -1, &p, sizeof(p)) == -1)
 85 e9ce266e 2022-03-07 op 		return got_error_from_errno("imsg_compose GOT_IMSG_PATCH");
 86 e9ce266e 2022-03-07 op 	return NULL;
 87 e9ce266e 2022-03-07 op }
 88 e9ce266e 2022-03-07 op 
 89 e9ce266e 2022-03-07 op static const struct got_error *
 90 e9ce266e 2022-03-07 op send_patch_done(void)
 91 e9ce266e 2022-03-07 op {
 92 e9ce266e 2022-03-07 op 	if (imsg_compose(&ibuf, GOT_IMSG_PATCH_DONE, 0, 0, -1,
 93 e9ce266e 2022-03-07 op 	    NULL, 0) == -1)
 94 e9ce266e 2022-03-07 op 		return got_error_from_errno("imsg_compose GOT_IMSG_PATCH_EOF");
 95 e9ce266e 2022-03-07 op 	if (imsg_flush(&ibuf) == -1)
 96 e9ce266e 2022-03-07 op 		return got_error_from_errno("imsg_flush");
 97 e9ce266e 2022-03-07 op 	return NULL;
 98 e9ce266e 2022-03-07 op }
 99 e9ce266e 2022-03-07 op 
100 e9ce266e 2022-03-07 op /* based on fetchname from usr.bin/patch/util.c */
101 e9ce266e 2022-03-07 op static const struct got_error *
102 9d6cabd5 2022-04-07 op filename(const char *at, char **name)
103 e9ce266e 2022-03-07 op {
104 9d6cabd5 2022-04-07 op 	char	*tmp, *t;
105 e9ce266e 2022-03-07 op 
106 e9ce266e 2022-03-07 op 	*name = NULL;
107 e9ce266e 2022-03-07 op 	if (*at == '\0')
108 e9ce266e 2022-03-07 op 		return NULL;
109 e9ce266e 2022-03-07 op 
110 e9ce266e 2022-03-07 op 	while (isspace((unsigned char)*at))
111 e9ce266e 2022-03-07 op 		at++;
112 e9ce266e 2022-03-07 op 
113 e9ce266e 2022-03-07 op 	/* files can be created or removed by diffing against /dev/null */
114 46ebad13 2022-03-17 op 	if (!strncmp(at, _PATH_DEVNULL, sizeof(_PATH_DEVNULL) - 1))
115 e9ce266e 2022-03-07 op 		return NULL;
116 e9ce266e 2022-03-07 op 
117 9d6cabd5 2022-04-07 op 	tmp = strdup(at);
118 9d6cabd5 2022-04-07 op 	if (tmp == NULL)
119 e9ce266e 2022-03-07 op 		return got_error_from_errno("strdup");
120 9d6cabd5 2022-04-07 op 	if ((t = strchr(tmp, '\t')) != NULL)
121 9d6cabd5 2022-04-07 op 		*t = '\0';
122 9d6cabd5 2022-04-07 op 	if ((t = strchr(tmp, '\n')) != NULL)
123 9d6cabd5 2022-04-07 op 		*t = '\0';
124 e9ce266e 2022-03-07 op 
125 9d6cabd5 2022-04-07 op 	*name = strdup(tmp);
126 9d6cabd5 2022-04-07 op 	free(tmp);
127 e9ce266e 2022-03-07 op 	if (*name == NULL)
128 e9ce266e 2022-03-07 op 		return got_error_from_errno("strdup");
129 e9ce266e 2022-03-07 op 	return NULL;
130 e9ce266e 2022-03-07 op }
131 e9ce266e 2022-03-07 op 
132 e9ce266e 2022-03-07 op static const struct got_error *
133 db0dfdd7 2022-06-27 op blobid(const char *line, char **blob, int git)
134 55e9459f 2022-06-19 op {
135 55e9459f 2022-06-19 op 	uint8_t digest[SHA1_DIGEST_LENGTH];
136 55e9459f 2022-06-19 op 	size_t len;
137 55e9459f 2022-06-19 op 
138 55e9459f 2022-06-19 op 	*blob = NULL;
139 55e9459f 2022-06-19 op 
140 55e9459f 2022-06-19 op 	len = strspn(line, "0123456789abcdefABCDEF");
141 55e9459f 2022-06-19 op 	if ((*blob = strndup(line, len)) == NULL)
142 55e9459f 2022-06-19 op 		return got_error_from_errno("strndup");
143 55e9459f 2022-06-19 op 
144 db0dfdd7 2022-06-27 op 	if (!git && !got_parse_sha1_digest(digest, *blob)) {
145 55e9459f 2022-06-19 op 		/* silently ignore invalid blob ids */
146 55e9459f 2022-06-19 op 		free(*blob);
147 55e9459f 2022-06-19 op 		*blob = NULL;
148 55e9459f 2022-06-19 op 	}
149 55e9459f 2022-06-19 op 	return NULL;
150 55e9459f 2022-06-19 op }
151 55e9459f 2022-06-19 op 
152 55e9459f 2022-06-19 op static const struct got_error *
153 acf749fc 2022-07-02 op patch_start(int *git, char **cid, FILE *fp)
154 e9ce266e 2022-03-07 op {
155 e9ce266e 2022-03-07 op 	const struct got_error *err = NULL;
156 e9ce266e 2022-03-07 op 	char	*line = NULL;
157 e9ce266e 2022-03-07 op 	size_t	 linesize = 0;
158 e9ce266e 2022-03-07 op 	ssize_t	 linelen;
159 acf749fc 2022-07-02 op 
160 acf749fc 2022-07-02 op 	*git = 0;
161 acf749fc 2022-07-02 op 
162 acf749fc 2022-07-02 op 	while ((linelen = getline(&line, &linesize, fp)) != -1) {
163 acf749fc 2022-07-02 op 		if (!strncmp(line, "diff --git ", 11)) {
164 acf749fc 2022-07-02 op 			*git = 1;
165 acf749fc 2022-07-02 op 			free(*cid);
166 acf749fc 2022-07-02 op 			*cid = NULL;
167 acf749fc 2022-07-02 op 			break;
168 acf749fc 2022-07-02 op 		} else if (!strncmp(line, "diff ", 5)) {
169 acf749fc 2022-07-02 op 			*git = 0;
170 acf749fc 2022-07-02 op 			free(*cid);
171 acf749fc 2022-07-02 op 			*cid = NULL;
172 acf749fc 2022-07-02 op 		} else if (!strncmp(line, "commit - ", 9)) {
173 acf749fc 2022-07-02 op 			free(*cid);
174 acf749fc 2022-07-02 op 			err = blobid(line + 9, cid, *git);
175 acf749fc 2022-07-02 op 			if (err)
176 acf749fc 2022-07-02 op 				break;
177 acf749fc 2022-07-02 op 		} else if (!strncmp(line, "--- ", 4) ||
178 acf749fc 2022-07-02 op 		    !strncmp(line, "+++ ", 4) ||
179 acf749fc 2022-07-02 op 		    !strncmp(line, "blob - ", 7)) {
180 acf749fc 2022-07-02 op 			/* rewind to previous line */
181 acf749fc 2022-07-02 op 			if (fseeko(fp, -linelen, SEEK_CUR) == -1)
182 acf749fc 2022-07-02 op 				err = got_error_from_errno("fseeko");
183 acf749fc 2022-07-02 op 			break;
184 acf749fc 2022-07-02 op 		}
185 acf749fc 2022-07-02 op 	}
186 acf749fc 2022-07-02 op 
187 acf749fc 2022-07-02 op 	free(line);
188 acf749fc 2022-07-02 op 	if (ferror(fp) && err == NULL)
189 acf749fc 2022-07-02 op 		err = got_error_from_errno("getline");
190 acf749fc 2022-07-02 op 	if (feof(fp) && err == NULL)
191 acf749fc 2022-07-02 op 		err = got_error(GOT_ERR_NO_PATCH);
192 acf749fc 2022-07-02 op 	return err;
193 acf749fc 2022-07-02 op }
194 acf749fc 2022-07-02 op 
195 acf749fc 2022-07-02 op static const struct got_error *
196 acf749fc 2022-07-02 op find_diff(int *done, int *next, FILE *fp, int git, const char *commitid)
197 acf749fc 2022-07-02 op {
198 acf749fc 2022-07-02 op 	const struct got_error *err = NULL;
199 acf749fc 2022-07-02 op 	char	*old = NULL, *new = NULL;
200 acf749fc 2022-07-02 op 	char	*blob = NULL;
201 acf749fc 2022-07-02 op 	char	*line = NULL;
202 acf749fc 2022-07-02 op 	size_t	 linesize = 0;
203 acf749fc 2022-07-02 op 	ssize_t	 linelen;
204 acf749fc 2022-07-02 op 	int	 create, rename = 0;
205 e9ce266e 2022-03-07 op 
206 acf749fc 2022-07-02 op 	*done = 0;
207 acf749fc 2022-07-02 op 	*next = 0;
208 e9ce266e 2022-03-07 op 	while ((linelen = getline(&line, &linesize, fp)) != -1) {
209 e9ce266e 2022-03-07 op 		/*
210 e9ce266e 2022-03-07 op 		 * Ignore the Index name like GNU and larry' patch,
211 e9ce266e 2022-03-07 op 		 * we don't have to follow POSIX.
212 e9ce266e 2022-03-07 op 		 */
213 e9ce266e 2022-03-07 op 
214 9d6cabd5 2022-04-07 op 		if (!strncmp(line, "--- ", 4)) {
215 e9ce266e 2022-03-07 op 			free(old);
216 9d6cabd5 2022-04-07 op 			err = filename(line+4, &old);
217 4379a9aa 2022-05-02 op 		} else if (rename && !strncmp(line, "rename from ", 12)) {
218 4379a9aa 2022-05-02 op 			free(old);
219 4379a9aa 2022-05-02 op 			err = filename(line+12, &old);
220 e9ce266e 2022-03-07 op 		} else if (!strncmp(line, "+++ ", 4)) {
221 e9ce266e 2022-03-07 op 			free(new);
222 9d6cabd5 2022-04-07 op 			err = filename(line+4, &new);
223 55e9459f 2022-06-19 op 		} else if (!git && !strncmp(line, "blob - ", 7)) {
224 55e9459f 2022-06-19 op 			free(blob);
225 db0dfdd7 2022-06-27 op 			err = blobid(line + 7, &blob, git);
226 4379a9aa 2022-05-02 op 		} else if (rename && !strncmp(line, "rename to ", 10)) {
227 4379a9aa 2022-05-02 op 			free(new);
228 4379a9aa 2022-05-02 op 			err = filename(line + 10, &new);
229 4379a9aa 2022-05-02 op 		} else if (git && !strncmp(line, "similarity index 100%", 21))
230 4379a9aa 2022-05-02 op 			rename = 1;
231 db0dfdd7 2022-06-27 op 		else if (git && !strncmp(line, "index ", 6)) {
232 db0dfdd7 2022-06-27 op 			free(blob);
233 db0dfdd7 2022-06-27 op 			err = blobid(line + 6, &blob, git);
234 acf749fc 2022-07-02 op 		} else if (!strncmp(line, "diff ", 5)) {
235 acf749fc 2022-07-02 op 			/* rewind to previous line */
236 acf749fc 2022-07-02 op 			if (fseeko(fp, -linelen, SEEK_CUR) == -1)
237 acf749fc 2022-07-02 op 				err = got_error_from_errno("fseeko");
238 acf749fc 2022-07-02 op 			*next = 1;
239 acf749fc 2022-07-02 op 			break;
240 55e9459f 2022-06-19 op 		}
241 e9ce266e 2022-03-07 op 
242 e9ce266e 2022-03-07 op 		if (err)
243 4379a9aa 2022-05-02 op 			break;
244 4379a9aa 2022-05-02 op 
245 4379a9aa 2022-05-02 op 		/*
246 4379a9aa 2022-05-02 op 		 * Git-style diffs with "similarity index 100%" don't
247 4379a9aa 2022-05-02 op 		 * have any hunks and ends with the "rename to foobar"
248 4379a9aa 2022-05-02 op 		 * line.
249 4379a9aa 2022-05-02 op 		 */
250 4379a9aa 2022-05-02 op 		if (rename && old != NULL && new != NULL) {
251 6b7665ac 2022-05-02 op 			*done = 1;
252 d8b5af43 2022-06-19 op 			err = send_patch(old, new, commitid,
253 d8b5af43 2022-06-19 op 			    blob, git);
254 e9ce266e 2022-03-07 op 			break;
255 4379a9aa 2022-05-02 op 		}
256 e9ce266e 2022-03-07 op 
257 e9ce266e 2022-03-07 op 		if (!strncmp(line, "@@ -", 4)) {
258 e9ce266e 2022-03-07 op 			create = !strncmp(line+4, "0,0", 3);
259 e9ce266e 2022-03-07 op 			if ((old == NULL && new == NULL) ||
260 e9ce266e 2022-03-07 op 			    (!create && old == NULL))
261 e9ce266e 2022-03-07 op 				err = got_error(GOT_ERR_PATCH_MALFORMED);
262 e9ce266e 2022-03-07 op 			else
263 d8b5af43 2022-06-19 op 				err = send_patch(old, new, commitid,
264 d8b5af43 2022-06-19 op 				    blob, git);
265 e9ce266e 2022-03-07 op 
266 e9ce266e 2022-03-07 op 			if (err)
267 e9ce266e 2022-03-07 op 				break;
268 e9ce266e 2022-03-07 op 
269 e9ce266e 2022-03-07 op 			/* rewind to previous line */
270 e45f7eba 2022-05-14 naddy 			if (fseeko(fp, -linelen, SEEK_CUR) == -1)
271 e45f7eba 2022-05-14 naddy 				err = got_error_from_errno("fseeko");
272 e9ce266e 2022-03-07 op 			break;
273 e9ce266e 2022-03-07 op 		}
274 e9ce266e 2022-03-07 op 	}
275 e9ce266e 2022-03-07 op 
276 423faaa6 2022-03-12 op 	free(old);
277 423faaa6 2022-03-12 op 	free(new);
278 55e9459f 2022-06-19 op 	free(blob);
279 e9ce266e 2022-03-07 op 	free(line);
280 e9ce266e 2022-03-07 op 	if (ferror(fp) && err == NULL)
281 e9ce266e 2022-03-07 op 		err = got_error_from_errno("getline");
282 e9ce266e 2022-03-07 op 	if (feof(fp) && err == NULL)
283 e9ce266e 2022-03-07 op 		err = got_error(GOT_ERR_NO_PATCH);
284 e9ce266e 2022-03-07 op 	return err;
285 e9ce266e 2022-03-07 op }
286 e9ce266e 2022-03-07 op 
287 e9ce266e 2022-03-07 op static const struct got_error *
288 35095610 2022-06-14 op strtolnum(char **str, int *n)
289 e9ce266e 2022-03-07 op {
290 e9ce266e 2022-03-07 op 	char		*p, c;
291 e9ce266e 2022-03-07 op 	const char	*errstr;
292 e9ce266e 2022-03-07 op 
293 e9ce266e 2022-03-07 op 	for (p = *str; isdigit((unsigned char)*p); ++p)
294 e9ce266e 2022-03-07 op 		/* nop */;
295 e9ce266e 2022-03-07 op 
296 e9ce266e 2022-03-07 op 	c = *p;
297 e9ce266e 2022-03-07 op 	*p = '\0';
298 e9ce266e 2022-03-07 op 
299 35095610 2022-06-14 op 	*n = strtonum(*str, 0, INT_MAX, &errstr);
300 e9ce266e 2022-03-07 op 	if (errstr != NULL)
301 e9ce266e 2022-03-07 op 		return got_error(GOT_ERR_PATCH_MALFORMED);
302 e9ce266e 2022-03-07 op 
303 e9ce266e 2022-03-07 op 	*p = c;
304 e9ce266e 2022-03-07 op 	*str = p;
305 e9ce266e 2022-03-07 op 	return NULL;
306 e9ce266e 2022-03-07 op }
307 e9ce266e 2022-03-07 op 
308 e9ce266e 2022-03-07 op static const struct got_error *
309 d75b9573 2022-05-02 op parse_hdr(char *s, int *done, struct got_imsg_patch_hunk *hdr)
310 e9ce266e 2022-03-07 op {
311 e9ce266e 2022-03-07 op 	static const struct got_error *err = NULL;
312 e9ce266e 2022-03-07 op 
313 e9ce266e 2022-03-07 op 	if (strncmp(s, "@@ -", 4)) {
314 d75b9573 2022-05-02 op 		*done = 1;
315 e9ce266e 2022-03-07 op 		return NULL;
316 e9ce266e 2022-03-07 op 	}
317 e9ce266e 2022-03-07 op 
318 e9ce266e 2022-03-07 op 	s += 4;
319 e9ce266e 2022-03-07 op 	if (!*s)
320 e9ce266e 2022-03-07 op 		return NULL;
321 e9ce266e 2022-03-07 op 	err = strtolnum(&s, &hdr->oldfrom);
322 e9ce266e 2022-03-07 op 	if (err)
323 e9ce266e 2022-03-07 op 		return err;
324 e9ce266e 2022-03-07 op 	if (*s == ',') {
325 e9ce266e 2022-03-07 op 		s++;
326 e9ce266e 2022-03-07 op 		err = strtolnum(&s, &hdr->oldlines);
327 e9ce266e 2022-03-07 op 		if (err)
328 e9ce266e 2022-03-07 op 			return err;
329 e9ce266e 2022-03-07 op 	} else
330 e9ce266e 2022-03-07 op 		hdr->oldlines = 1;
331 e9ce266e 2022-03-07 op 
332 e9ce266e 2022-03-07 op 	if (*s == ' ')
333 e9ce266e 2022-03-07 op 		s++;
334 e9ce266e 2022-03-07 op 
335 e9ce266e 2022-03-07 op 	if (*s != '+' || !*++s)
336 e9ce266e 2022-03-07 op 		return got_error(GOT_ERR_PATCH_MALFORMED);
337 e9ce266e 2022-03-07 op 	err = strtolnum(&s, &hdr->newfrom);
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->newlines);
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->newlines = 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 != '@')
352 e9ce266e 2022-03-07 op 		return got_error(GOT_ERR_PATCH_MALFORMED);
353 e9ce266e 2022-03-07 op 
354 35095610 2022-06-14 op 	if (hdr->oldfrom >= INT_MAX - hdr->oldlines ||
355 35095610 2022-06-14 op 	    hdr->newfrom >= INT_MAX - hdr->newlines ||
356 e9ce266e 2022-03-07 op 	    /* not so sure about this one */
357 35095610 2022-06-14 op 	    hdr->oldlines >= INT_MAX - hdr->newlines - 1 ||
358 be33dff7 2022-05-13 op 	    (hdr->oldlines == 0 && hdr->newlines == 0))
359 e9ce266e 2022-03-07 op 		return got_error(GOT_ERR_PATCH_MALFORMED);
360 e9ce266e 2022-03-07 op 
361 e9ce266e 2022-03-07 op 	if (hdr->oldlines == 0) {
362 e9ce266e 2022-03-07 op 		/* larry says to "do append rather than insert"; I don't
363 e9ce266e 2022-03-07 op 		 * quite get it, but i trust him.
364 e9ce266e 2022-03-07 op 		 */
365 e9ce266e 2022-03-07 op 		hdr->oldfrom++;
366 e9ce266e 2022-03-07 op 	}
367 e9ce266e 2022-03-07 op 
368 e9ce266e 2022-03-07 op 	if (imsg_compose(&ibuf, GOT_IMSG_PATCH_HUNK, 0, 0, -1,
369 e9ce266e 2022-03-07 op 	    hdr, sizeof(*hdr)) == -1)
370 e9ce266e 2022-03-07 op 		return got_error_from_errno(
371 e9ce266e 2022-03-07 op 		    "imsg_compose GOT_IMSG_PATCH_HUNK");
372 e9ce266e 2022-03-07 op 	return NULL;
373 e9ce266e 2022-03-07 op }
374 e9ce266e 2022-03-07 op 
375 e9ce266e 2022-03-07 op static const struct got_error *
376 e9ce266e 2022-03-07 op send_line(const char *line)
377 e9ce266e 2022-03-07 op {
378 e9ce266e 2022-03-07 op 	static const struct got_error *err = NULL;
379 e9ce266e 2022-03-07 op 	char *p = NULL;
380 e9ce266e 2022-03-07 op 
381 b3c57ab2 2022-03-22 op 	if (*line != '+' && *line != '-' && *line != ' ' && *line != '\\') {
382 e9ce266e 2022-03-07 op 		if (asprintf(&p, " %s", line) == -1)
383 e9ce266e 2022-03-07 op 			return got_error_from_errno("asprintf");
384 e9ce266e 2022-03-07 op 		line = p;
385 e9ce266e 2022-03-07 op 	}
386 e9ce266e 2022-03-07 op 
387 e9ce266e 2022-03-07 op 	if (imsg_compose(&ibuf, GOT_IMSG_PATCH_LINE, 0, 0, -1,
388 46ebad13 2022-03-17 op 	    line, strlen(line) + 1) == -1)
389 e9ce266e 2022-03-07 op 		err = got_error_from_errno(
390 e9ce266e 2022-03-07 op 		    "imsg_compose GOT_IMSG_PATCH_LINE");
391 e9ce266e 2022-03-07 op 
392 e9ce266e 2022-03-07 op 	free(p);
393 e9ce266e 2022-03-07 op 	return err;
394 e9ce266e 2022-03-07 op }
395 e9ce266e 2022-03-07 op 
396 e9ce266e 2022-03-07 op static const struct got_error *
397 b2832778 2022-04-23 op peek_special_line(FILE *fp)
398 b3c57ab2 2022-03-22 op {
399 b3c57ab2 2022-03-22 op 	const struct got_error *err;
400 e260f8ae 2022-03-22 stsp 	int ch;
401 b3c57ab2 2022-03-22 op 
402 b3c57ab2 2022-03-22 op 	ch = fgetc(fp);
403 b3c57ab2 2022-03-22 op 	if (ch != EOF && ch != '\\') {
404 b3c57ab2 2022-03-22 op 		ungetc(ch, fp);
405 b3c57ab2 2022-03-22 op 		return NULL;
406 b3c57ab2 2022-03-22 op 	}
407 b3c57ab2 2022-03-22 op 
408 b2832778 2022-04-23 op 	if (ch == '\\') {
409 b3c57ab2 2022-03-22 op 		err = send_line("\\");
410 b3c57ab2 2022-03-22 op 		if (err)
411 b3c57ab2 2022-03-22 op 			return err;
412 b3c57ab2 2022-03-22 op 	}
413 b3c57ab2 2022-03-22 op 
414 b3c57ab2 2022-03-22 op 	while (ch != EOF && ch != '\n')
415 b3c57ab2 2022-03-22 op 		ch = fgetc(fp);
416 b3c57ab2 2022-03-22 op 
417 b3c57ab2 2022-03-22 op 	if (ch != EOF || feof(fp))
418 b3c57ab2 2022-03-22 op 		return NULL;
419 b3c57ab2 2022-03-22 op 	return got_error(GOT_ERR_IO);
420 b3c57ab2 2022-03-22 op }
421 b3c57ab2 2022-03-22 op 
422 b3c57ab2 2022-03-22 op static const struct got_error *
423 d75b9573 2022-05-02 op parse_hunk(FILE *fp, int *done)
424 e9ce266e 2022-03-07 op {
425 e9ce266e 2022-03-07 op 	static const struct got_error *err = NULL;
426 e9ce266e 2022-03-07 op 	struct got_imsg_patch_hunk hdr;
427 e9ce266e 2022-03-07 op 	char	*line = NULL, ch;
428 e9ce266e 2022-03-07 op 	size_t	 linesize = 0;
429 e9ce266e 2022-03-07 op 	ssize_t	 linelen;
430 35095610 2022-06-14 op 	int	 leftold, leftnew;
431 e9ce266e 2022-03-07 op 
432 e9ce266e 2022-03-07 op 	linelen = getline(&line, &linesize, fp);
433 e9ce266e 2022-03-07 op 	if (linelen == -1) {
434 d75b9573 2022-05-02 op 		*done = 1;
435 e9ce266e 2022-03-07 op 		goto done;
436 e9ce266e 2022-03-07 op 	}
437 e9ce266e 2022-03-07 op 
438 d75b9573 2022-05-02 op 	err = parse_hdr(line, done, &hdr);
439 e9ce266e 2022-03-07 op 	if (err)
440 e9ce266e 2022-03-07 op 		goto done;
441 d75b9573 2022-05-02 op 	if (*done) {
442 e45f7eba 2022-05-14 naddy 		if (fseeko(fp, -linelen, SEEK_CUR) == -1)
443 e45f7eba 2022-05-14 naddy 			err = got_error_from_errno("fseeko");
444 e9ce266e 2022-03-07 op 		goto done;
445 e9ce266e 2022-03-07 op 	}
446 e9ce266e 2022-03-07 op 
447 e9ce266e 2022-03-07 op 	leftold = hdr.oldlines;
448 e9ce266e 2022-03-07 op 	leftnew = hdr.newlines;
449 e9ce266e 2022-03-07 op 
450 e9ce266e 2022-03-07 op 	while (leftold > 0 || leftnew > 0) {
451 e9ce266e 2022-03-07 op 		linelen = getline(&line, &linesize, fp);
452 e9ce266e 2022-03-07 op 		if (linelen == -1) {
453 e9ce266e 2022-03-07 op 			if (ferror(fp)) {
454 e9ce266e 2022-03-07 op 				err = got_error_from_errno("getline");
455 e9ce266e 2022-03-07 op 				goto done;
456 e9ce266e 2022-03-07 op 			}
457 e9ce266e 2022-03-07 op 
458 e9ce266e 2022-03-07 op 			/* trailing newlines may be chopped */
459 e9ce266e 2022-03-07 op 			if (leftold < 3 && leftnew < 3) {
460 d75b9573 2022-05-02 op 				*done = 1;
461 e9ce266e 2022-03-07 op 				break;
462 e9ce266e 2022-03-07 op 			}
463 e9ce266e 2022-03-07 op 
464 e9ce266e 2022-03-07 op 			err = got_error(GOT_ERR_PATCH_TRUNCATED);
465 e9ce266e 2022-03-07 op 			goto done;
466 e9ce266e 2022-03-07 op 		}
467 b3c57ab2 2022-03-22 op 		if (line[linelen - 1] == '\n')
468 b3c57ab2 2022-03-22 op 			line[linelen - 1] = '\0';
469 e9ce266e 2022-03-07 op 
470 e9ce266e 2022-03-07 op 		/* usr.bin/patch allows '=' as context char */
471 e9ce266e 2022-03-07 op 		if (*line == '=')
472 e9ce266e 2022-03-07 op 			*line = ' ';
473 e9ce266e 2022-03-07 op 
474 e9ce266e 2022-03-07 op 		ch = *line;
475 b3c57ab2 2022-03-22 op 		if (ch == '\t' || ch == '\0')
476 e9ce266e 2022-03-07 op 			ch = ' ';	/* the space got eaten */
477 e9ce266e 2022-03-07 op 
478 e9ce266e 2022-03-07 op 		switch (ch) {
479 e9ce266e 2022-03-07 op 		case '-':
480 e9ce266e 2022-03-07 op 			leftold--;
481 e9ce266e 2022-03-07 op 			break;
482 e9ce266e 2022-03-07 op 		case ' ':
483 e9ce266e 2022-03-07 op 			leftold--;
484 e9ce266e 2022-03-07 op 			leftnew--;
485 e9ce266e 2022-03-07 op 			break;
486 e9ce266e 2022-03-07 op 		case '+':
487 e9ce266e 2022-03-07 op 			leftnew--;
488 e9ce266e 2022-03-07 op 			break;
489 e9ce266e 2022-03-07 op 		default:
490 e9ce266e 2022-03-07 op 			err = got_error(GOT_ERR_PATCH_MALFORMED);
491 e9ce266e 2022-03-07 op 			goto done;
492 e9ce266e 2022-03-07 op 		}
493 e9ce266e 2022-03-07 op 
494 e9ce266e 2022-03-07 op 		if (leftold < 0 || leftnew < 0) {
495 e9ce266e 2022-03-07 op 			err = got_error(GOT_ERR_PATCH_MALFORMED);
496 e9ce266e 2022-03-07 op 			goto done;
497 e9ce266e 2022-03-07 op 		}
498 e9ce266e 2022-03-07 op 
499 e9ce266e 2022-03-07 op 		err = send_line(line);
500 e9ce266e 2022-03-07 op 		if (err)
501 e9ce266e 2022-03-07 op 			goto done;
502 b3c57ab2 2022-03-22 op 
503 b3c57ab2 2022-03-22 op 		if ((ch == '-' && leftold == 0) ||
504 b3c57ab2 2022-03-22 op 		    (ch == '+' && leftnew == 0)) {
505 b2832778 2022-04-23 op 			err = peek_special_line(fp);
506 b3c57ab2 2022-03-22 op 			if (err)
507 b3c57ab2 2022-03-22 op 				goto done;
508 b3c57ab2 2022-03-22 op 		}
509 e9ce266e 2022-03-07 op 	}
510 e9ce266e 2022-03-07 op 
511 e9ce266e 2022-03-07 op done:
512 e9ce266e 2022-03-07 op 	free(line);
513 e9ce266e 2022-03-07 op 	return err;
514 e9ce266e 2022-03-07 op }
515 e9ce266e 2022-03-07 op 
516 e9ce266e 2022-03-07 op static const struct got_error *
517 e9ce266e 2022-03-07 op read_patch(struct imsgbuf *ibuf, int fd)
518 e9ce266e 2022-03-07 op {
519 e9ce266e 2022-03-07 op 	const struct got_error *err = NULL;
520 e9ce266e 2022-03-07 op 	FILE *fp;
521 acf749fc 2022-07-02 op 	int git, patch_found = 0;
522 acf749fc 2022-07-02 op 	char *cid = NULL;
523 e9ce266e 2022-03-07 op 
524 e9ce266e 2022-03-07 op 	if ((fp = fdopen(fd, "r")) == NULL) {
525 e9ce266e 2022-03-07 op 		err = got_error_from_errno("fdopen");
526 e9ce266e 2022-03-07 op 		close(fd);
527 e9ce266e 2022-03-07 op 		return err;
528 e9ce266e 2022-03-07 op 	}
529 e9ce266e 2022-03-07 op 
530 acf749fc 2022-07-02 op 	while ((err = patch_start(&git, &cid, fp)) == NULL) {
531 acf749fc 2022-07-02 op 		int done, next;
532 4379a9aa 2022-05-02 op 
533 acf749fc 2022-07-02 op 		err = find_diff(&done, &next, fp, git, cid);
534 e9ce266e 2022-03-07 op 		if (err)
535 e9ce266e 2022-03-07 op 			goto done;
536 acf749fc 2022-07-02 op 		if (next)
537 acf749fc 2022-07-02 op 			continue;
538 e9ce266e 2022-03-07 op 
539 e9ce266e 2022-03-07 op 		patch_found = 1;
540 d75b9573 2022-05-02 op 
541 d75b9573 2022-05-02 op 		while (!done) {
542 d75b9573 2022-05-02 op 			err = parse_hunk(fp, &done);
543 e9ce266e 2022-03-07 op 			if (err)
544 e9ce266e 2022-03-07 op 				goto done;
545 e9ce266e 2022-03-07 op 		}
546 d75b9573 2022-05-02 op 
547 d75b9573 2022-05-02 op 		err = send_patch_done();
548 d75b9573 2022-05-02 op 		if (err)
549 d75b9573 2022-05-02 op 			goto done;
550 e9ce266e 2022-03-07 op 	}
551 e9ce266e 2022-03-07 op 
552 e9ce266e 2022-03-07 op done:
553 e9ce266e 2022-03-07 op 	fclose(fp);
554 acf749fc 2022-07-02 op 	free(cid);
555 e9ce266e 2022-03-07 op 
556 e9ce266e 2022-03-07 op 	/* ignore trailing gibberish */
557 e9ce266e 2022-03-07 op 	if (err != NULL && err->code == GOT_ERR_NO_PATCH && patch_found)
558 e9ce266e 2022-03-07 op 		err = NULL;
559 e9ce266e 2022-03-07 op 
560 e9ce266e 2022-03-07 op 	return err;
561 e9ce266e 2022-03-07 op }
562 e9ce266e 2022-03-07 op 
563 e9ce266e 2022-03-07 op int
564 e9ce266e 2022-03-07 op main(int argc, char **argv)
565 e9ce266e 2022-03-07 op {
566 e9ce266e 2022-03-07 op 	const struct got_error *err = NULL;
567 e9ce266e 2022-03-07 op 	struct imsg imsg;
568 e9ce266e 2022-03-07 op #if 0
569 e9ce266e 2022-03-07 op 	static int attached;
570 e9ce266e 2022-03-07 op 	while (!attached)
571 e9ce266e 2022-03-07 op 		sleep(1);
572 e9ce266e 2022-03-07 op #endif
573 e9ce266e 2022-03-07 op 
574 e9ce266e 2022-03-07 op 	imsg_init(&ibuf, GOT_IMSG_FD_CHILD);
575 e9ce266e 2022-03-07 op #ifndef PROFILE
576 e9ce266e 2022-03-07 op 	/* revoke access to most system calls */
577 e9ce266e 2022-03-07 op 	if (pledge("stdio recvfd", NULL) == -1) {
578 e9ce266e 2022-03-07 op 		err = got_error_from_errno("pledge");
579 e9ce266e 2022-03-07 op 		got_privsep_send_error(&ibuf, err);
580 e9ce266e 2022-03-07 op 		return 1;
581 e9ce266e 2022-03-07 op 	}
582 e9ce266e 2022-03-07 op #endif
583 e9ce266e 2022-03-07 op 
584 e9ce266e 2022-03-07 op 	err = got_privsep_recv_imsg(&imsg, &ibuf, 0);
585 e9ce266e 2022-03-07 op 	if (err)
586 e9ce266e 2022-03-07 op 		goto done;
587 e9ce266e 2022-03-07 op 	if (imsg.hdr.type != GOT_IMSG_PATCH_FILE || imsg.fd == -1) {
588 e9ce266e 2022-03-07 op 		err = got_error(GOT_ERR_PRIVSEP_MSG);
589 e9ce266e 2022-03-07 op 		goto done;
590 e9ce266e 2022-03-07 op 	}
591 e9ce266e 2022-03-07 op 
592 e9ce266e 2022-03-07 op 	err = read_patch(&ibuf, imsg.fd);
593 e9ce266e 2022-03-07 op 	if (err)
594 e9ce266e 2022-03-07 op 		goto done;
595 e9ce266e 2022-03-07 op 	if (imsg_compose(&ibuf, GOT_IMSG_PATCH_EOF, 0, 0, -1,
596 e9ce266e 2022-03-07 op 	    NULL, 0) == -1) {
597 e9ce266e 2022-03-07 op 		err = got_error_from_errno("imsg_compose GOT_IMSG_PATCH_EOF");
598 e9ce266e 2022-03-07 op 		goto done;
599 e9ce266e 2022-03-07 op 	}
600 e9ce266e 2022-03-07 op 	err = got_privsep_flush_imsg(&ibuf);
601 e9ce266e 2022-03-07 op done:
602 e9ce266e 2022-03-07 op 	imsg_free(&imsg);
603 e9ce266e 2022-03-07 op 	if (err != NULL) {
604 e9ce266e 2022-03-07 op 		got_privsep_send_error(&ibuf, err);
605 e9ce266e 2022-03-07 op 		err = NULL;
606 e9ce266e 2022-03-07 op 	}
607 e9ce266e 2022-03-07 op 	if (close(GOT_IMSG_FD_CHILD) == -1 && err == NULL)
608 e9ce266e 2022-03-07 op 		err = got_error_from_errno("close");
609 e9ce266e 2022-03-07 op 	if (err && err->code != GOT_ERR_PRIVSEP_PIPE)
610 e9ce266e 2022-03-07 op 		fprintf(stderr, "%s: %s\n", getprogname(), err->msg);
611 e9ce266e 2022-03-07 op 	return err ? 1 : 0;
612 e9ce266e 2022-03-07 op }