Blame


1 069bbb86 2022-03-07 thomas /*
2 069bbb86 2022-03-07 thomas * Copyright 1986, Larry Wall
3 069bbb86 2022-03-07 thomas *
4 069bbb86 2022-03-07 thomas * Redistribution and use in source and binary forms, with or without
5 069bbb86 2022-03-07 thomas * modification, are permitted provided that the following condition is met:
6 069bbb86 2022-03-07 thomas * 1. Redistributions of source code must retain the above copyright notice,
7 069bbb86 2022-03-07 thomas * this condition and the following disclaimer.
8 069bbb86 2022-03-07 thomas *
9 069bbb86 2022-03-07 thomas * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND ANY
10 069bbb86 2022-03-07 thomas * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
11 069bbb86 2022-03-07 thomas * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
12 069bbb86 2022-03-07 thomas * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE FOR
13 069bbb86 2022-03-07 thomas * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
14 069bbb86 2022-03-07 thomas * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
15 069bbb86 2022-03-07 thomas * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
16 069bbb86 2022-03-07 thomas * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
17 069bbb86 2022-03-07 thomas * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
18 069bbb86 2022-03-07 thomas * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
19 069bbb86 2022-03-07 thomas * SUCH DAMAGE.
20 069bbb86 2022-03-07 thomas */
21 069bbb86 2022-03-07 thomas
22 069bbb86 2022-03-07 thomas /*
23 069bbb86 2022-03-07 thomas * Copyright (c) 2022 Omar Polo <op@openbsd.org>
24 069bbb86 2022-03-07 thomas *
25 069bbb86 2022-03-07 thomas * Permission to use, copy, modify, and distribute this software for any
26 069bbb86 2022-03-07 thomas * purpose with or without fee is hereby granted, provided that the above
27 069bbb86 2022-03-07 thomas * copyright notice and this permission notice appear in all copies.
28 069bbb86 2022-03-07 thomas *
29 069bbb86 2022-03-07 thomas * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
30 069bbb86 2022-03-07 thomas * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
31 069bbb86 2022-03-07 thomas * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
32 069bbb86 2022-03-07 thomas * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
33 069bbb86 2022-03-07 thomas * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
34 069bbb86 2022-03-07 thomas * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
35 069bbb86 2022-03-07 thomas * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
36 069bbb86 2022-03-07 thomas */
37 069bbb86 2022-03-07 thomas
38 069bbb86 2022-03-07 thomas #include <sys/types.h>
39 069bbb86 2022-03-07 thomas #include <sys/uio.h>
40 069bbb86 2022-03-07 thomas
41 069bbb86 2022-03-07 thomas #include <ctype.h>
42 069bbb86 2022-03-07 thomas #include <limits.h>
43 069bbb86 2022-03-07 thomas #include <paths.h>
44 069bbb86 2022-03-07 thomas #include <stdint.h>
45 069bbb86 2022-03-07 thomas #include <stdio.h>
46 069bbb86 2022-03-07 thomas #include <stdlib.h>
47 069bbb86 2022-03-07 thomas #include <string.h>
48 069bbb86 2022-03-07 thomas #include <unistd.h>
49 069bbb86 2022-03-07 thomas
50 069bbb86 2022-03-07 thomas #include "got_error.h"
51 069bbb86 2022-03-07 thomas #include "got_object.h"
52 069bbb86 2022-03-07 thomas
53 973f3f6e 2022-03-07 thomas #include "got_compat.h"
54 973f3f6e 2022-03-07 thomas
55 069bbb86 2022-03-07 thomas #include "got_lib_delta.h"
56 069bbb86 2022-03-07 thomas #include "got_lib_object.h"
57 069bbb86 2022-03-07 thomas #include "got_lib_privsep.h"
58 069bbb86 2022-03-07 thomas
59 069bbb86 2022-03-07 thomas struct imsgbuf ibuf;
60 069bbb86 2022-03-07 thomas
61 069bbb86 2022-03-07 thomas static const struct got_error *
62 be53ddb1 2022-03-22 thomas send_patch(const char *oldname, const char *newname, int git)
63 069bbb86 2022-03-07 thomas {
64 069bbb86 2022-03-07 thomas struct got_imsg_patch p;
65 069bbb86 2022-03-07 thomas
66 069bbb86 2022-03-07 thomas memset(&p, 0, sizeof(p));
67 069bbb86 2022-03-07 thomas
68 be53ddb1 2022-03-22 thomas /*
69 be53ddb1 2022-03-22 thomas * Prefer the new name if it's not /dev/null and it's not
70 be53ddb1 2022-03-22 thomas * a git-style diff.
71 be53ddb1 2022-03-22 thomas */
72 be53ddb1 2022-03-22 thomas if (!git && newname != NULL && oldname != NULL)
73 be53ddb1 2022-03-22 thomas strlcpy(p.old, newname, sizeof(p.old));
74 be53ddb1 2022-03-22 thomas else if (oldname != NULL)
75 069bbb86 2022-03-07 thomas strlcpy(p.old, oldname, sizeof(p.old));
76 be53ddb1 2022-03-22 thomas
77 069bbb86 2022-03-07 thomas if (newname != NULL)
78 069bbb86 2022-03-07 thomas strlcpy(p.new, newname, sizeof(p.new));
79 069bbb86 2022-03-07 thomas
80 069bbb86 2022-03-07 thomas if (imsg_compose(&ibuf, GOT_IMSG_PATCH, 0, 0, -1,
81 069bbb86 2022-03-07 thomas &p, sizeof(p)) == -1)
82 069bbb86 2022-03-07 thomas return got_error_from_errno("imsg_compose GOT_IMSG_PATCH");
83 069bbb86 2022-03-07 thomas return NULL;
84 069bbb86 2022-03-07 thomas }
85 069bbb86 2022-03-07 thomas
86 069bbb86 2022-03-07 thomas static const struct got_error *
87 069bbb86 2022-03-07 thomas send_patch_done(void)
88 069bbb86 2022-03-07 thomas {
89 069bbb86 2022-03-07 thomas if (imsg_compose(&ibuf, GOT_IMSG_PATCH_DONE, 0, 0, -1,
90 069bbb86 2022-03-07 thomas NULL, 0) == -1)
91 069bbb86 2022-03-07 thomas return got_error_from_errno("imsg_compose GOT_IMSG_PATCH_EOF");
92 069bbb86 2022-03-07 thomas if (imsg_flush(&ibuf) == -1)
93 069bbb86 2022-03-07 thomas return got_error_from_errno("imsg_flush");
94 069bbb86 2022-03-07 thomas return NULL;
95 069bbb86 2022-03-07 thomas }
96 069bbb86 2022-03-07 thomas
97 069bbb86 2022-03-07 thomas /* based on fetchname from usr.bin/patch/util.c */
98 069bbb86 2022-03-07 thomas static const struct got_error *
99 069bbb86 2022-03-07 thomas filename(const char *at, char **name, int strip)
100 069bbb86 2022-03-07 thomas {
101 069bbb86 2022-03-07 thomas char *fullname, *t;
102 069bbb86 2022-03-07 thomas int l, tab;
103 069bbb86 2022-03-07 thomas
104 069bbb86 2022-03-07 thomas *name = NULL;
105 069bbb86 2022-03-07 thomas if (*at == '\0')
106 069bbb86 2022-03-07 thomas return NULL;
107 069bbb86 2022-03-07 thomas
108 069bbb86 2022-03-07 thomas while (isspace((unsigned char)*at))
109 069bbb86 2022-03-07 thomas at++;
110 069bbb86 2022-03-07 thomas
111 069bbb86 2022-03-07 thomas /* files can be created or removed by diffing against /dev/null */
112 17d6446a 2022-03-22 thomas if (!strncmp(at, _PATH_DEVNULL, sizeof(_PATH_DEVNULL) - 1))
113 069bbb86 2022-03-07 thomas return NULL;
114 069bbb86 2022-03-07 thomas
115 069bbb86 2022-03-07 thomas t = strdup(at);
116 069bbb86 2022-03-07 thomas if (t == NULL)
117 069bbb86 2022-03-07 thomas return got_error_from_errno("strdup");
118 069bbb86 2022-03-07 thomas *name = fullname = t;
119 069bbb86 2022-03-07 thomas tab = strchr(t, '\t') != NULL;
120 069bbb86 2022-03-07 thomas
121 069bbb86 2022-03-07 thomas /* strip off path components and NUL-terminate */
122 069bbb86 2022-03-07 thomas for (l = strip;
123 069bbb86 2022-03-07 thomas *t != '\0' && ((tab && *t != '\t') || !isspace((unsigned char)*t));
124 069bbb86 2022-03-07 thomas ++t) {
125 069bbb86 2022-03-07 thomas if (t[0] == '/' && t[1] != '/' && t[1] != '\0')
126 069bbb86 2022-03-07 thomas if (--l >= 0)
127 17d6446a 2022-03-22 thomas *name = t + 1;
128 069bbb86 2022-03-07 thomas }
129 069bbb86 2022-03-07 thomas *t = '\0';
130 069bbb86 2022-03-07 thomas
131 069bbb86 2022-03-07 thomas *name = strdup(*name);
132 069bbb86 2022-03-07 thomas free(fullname);
133 069bbb86 2022-03-07 thomas if (*name == NULL)
134 069bbb86 2022-03-07 thomas return got_error_from_errno("strdup");
135 069bbb86 2022-03-07 thomas return NULL;
136 069bbb86 2022-03-07 thomas }
137 069bbb86 2022-03-07 thomas
138 069bbb86 2022-03-07 thomas static const struct got_error *
139 069bbb86 2022-03-07 thomas find_patch(FILE *fp)
140 069bbb86 2022-03-07 thomas {
141 069bbb86 2022-03-07 thomas const struct got_error *err = NULL;
142 069bbb86 2022-03-07 thomas char *old = NULL, *new = NULL;
143 069bbb86 2022-03-07 thomas char *line = NULL;
144 069bbb86 2022-03-07 thomas size_t linesize = 0;
145 069bbb86 2022-03-07 thomas ssize_t linelen;
146 069bbb86 2022-03-07 thomas int create, git = 0;
147 069bbb86 2022-03-07 thomas
148 069bbb86 2022-03-07 thomas while ((linelen = getline(&line, &linesize, fp)) != -1) {
149 069bbb86 2022-03-07 thomas /*
150 069bbb86 2022-03-07 thomas * Ignore the Index name like GNU and larry' patch,
151 069bbb86 2022-03-07 thomas * we don't have to follow POSIX.
152 069bbb86 2022-03-07 thomas */
153 069bbb86 2022-03-07 thomas
154 069bbb86 2022-03-07 thomas if (git && !strncmp(line, "--- a/", 6)) {
155 069bbb86 2022-03-07 thomas free(old);
156 069bbb86 2022-03-07 thomas err = filename(line+6, &old, 0);
157 069bbb86 2022-03-07 thomas } else if (!strncmp(line, "--- ", 4)) {
158 069bbb86 2022-03-07 thomas free(old);
159 069bbb86 2022-03-07 thomas err = filename(line+4, &old, 0);
160 069bbb86 2022-03-07 thomas } else if (git && !strncmp(line, "+++ b/", 6)) {
161 069bbb86 2022-03-07 thomas free(new);
162 069bbb86 2022-03-07 thomas err = filename(line+6, &new, 0);
163 069bbb86 2022-03-07 thomas } else if (!strncmp(line, "+++ ", 4)) {
164 069bbb86 2022-03-07 thomas free(new);
165 069bbb86 2022-03-07 thomas err = filename(line+4, &new, 0);
166 069bbb86 2022-03-07 thomas } else if (!strncmp(line, "diff --git a/", 13))
167 069bbb86 2022-03-07 thomas git = 1;
168 069bbb86 2022-03-07 thomas
169 069bbb86 2022-03-07 thomas if (err)
170 069bbb86 2022-03-07 thomas break;
171 069bbb86 2022-03-07 thomas
172 069bbb86 2022-03-07 thomas if (!strncmp(line, "@@ -", 4)) {
173 069bbb86 2022-03-07 thomas create = !strncmp(line+4, "0,0", 3);
174 069bbb86 2022-03-07 thomas if ((old == NULL && new == NULL) ||
175 069bbb86 2022-03-07 thomas (!create && old == NULL))
176 069bbb86 2022-03-07 thomas err = got_error(GOT_ERR_PATCH_MALFORMED);
177 069bbb86 2022-03-07 thomas else
178 be53ddb1 2022-03-22 thomas err = send_patch(old, new, git);
179 069bbb86 2022-03-07 thomas
180 069bbb86 2022-03-07 thomas if (err)
181 069bbb86 2022-03-07 thomas break;
182 069bbb86 2022-03-07 thomas
183 069bbb86 2022-03-07 thomas /* rewind to previous line */
184 069bbb86 2022-03-07 thomas if (fseek(fp, linelen * -1, SEEK_CUR) == -1)
185 069bbb86 2022-03-07 thomas err = got_error_from_errno("fseek");
186 069bbb86 2022-03-07 thomas break;
187 069bbb86 2022-03-07 thomas }
188 069bbb86 2022-03-07 thomas }
189 069bbb86 2022-03-07 thomas
190 19f1a2ac 2022-03-13 thomas free(old);
191 19f1a2ac 2022-03-13 thomas free(new);
192 069bbb86 2022-03-07 thomas free(line);
193 069bbb86 2022-03-07 thomas if (ferror(fp) && err == NULL)
194 069bbb86 2022-03-07 thomas err = got_error_from_errno("getline");
195 069bbb86 2022-03-07 thomas if (feof(fp) && err == NULL)
196 069bbb86 2022-03-07 thomas err = got_error(GOT_ERR_NO_PATCH);
197 069bbb86 2022-03-07 thomas return err;
198 069bbb86 2022-03-07 thomas }
199 069bbb86 2022-03-07 thomas
200 069bbb86 2022-03-07 thomas static const struct got_error *
201 069bbb86 2022-03-07 thomas strtolnum(char **str, long *n)
202 069bbb86 2022-03-07 thomas {
203 069bbb86 2022-03-07 thomas char *p, c;
204 069bbb86 2022-03-07 thomas const char *errstr;
205 069bbb86 2022-03-07 thomas
206 069bbb86 2022-03-07 thomas for (p = *str; isdigit((unsigned char)*p); ++p)
207 069bbb86 2022-03-07 thomas /* nop */;
208 069bbb86 2022-03-07 thomas
209 069bbb86 2022-03-07 thomas c = *p;
210 069bbb86 2022-03-07 thomas *p = '\0';
211 069bbb86 2022-03-07 thomas
212 069bbb86 2022-03-07 thomas *n = strtonum(*str, 0, LONG_MAX, &errstr);
213 069bbb86 2022-03-07 thomas if (errstr != NULL)
214 069bbb86 2022-03-07 thomas return got_error(GOT_ERR_PATCH_MALFORMED);
215 069bbb86 2022-03-07 thomas
216 069bbb86 2022-03-07 thomas *p = c;
217 069bbb86 2022-03-07 thomas *str = p;
218 069bbb86 2022-03-07 thomas return NULL;
219 069bbb86 2022-03-07 thomas }
220 069bbb86 2022-03-07 thomas
221 069bbb86 2022-03-07 thomas static const struct got_error *
222 069bbb86 2022-03-07 thomas parse_hdr(char *s, int *ok, struct got_imsg_patch_hunk *hdr)
223 069bbb86 2022-03-07 thomas {
224 069bbb86 2022-03-07 thomas static const struct got_error *err = NULL;
225 069bbb86 2022-03-07 thomas
226 069bbb86 2022-03-07 thomas *ok = 1;
227 069bbb86 2022-03-07 thomas if (strncmp(s, "@@ -", 4)) {
228 069bbb86 2022-03-07 thomas *ok = 0;
229 069bbb86 2022-03-07 thomas return NULL;
230 069bbb86 2022-03-07 thomas }
231 069bbb86 2022-03-07 thomas
232 069bbb86 2022-03-07 thomas s += 4;
233 069bbb86 2022-03-07 thomas if (!*s)
234 069bbb86 2022-03-07 thomas return NULL;
235 069bbb86 2022-03-07 thomas err = strtolnum(&s, &hdr->oldfrom);
236 069bbb86 2022-03-07 thomas if (err)
237 069bbb86 2022-03-07 thomas return err;
238 069bbb86 2022-03-07 thomas if (*s == ',') {
239 069bbb86 2022-03-07 thomas s++;
240 069bbb86 2022-03-07 thomas err = strtolnum(&s, &hdr->oldlines);
241 069bbb86 2022-03-07 thomas if (err)
242 069bbb86 2022-03-07 thomas return err;
243 069bbb86 2022-03-07 thomas } else
244 069bbb86 2022-03-07 thomas hdr->oldlines = 1;
245 069bbb86 2022-03-07 thomas
246 069bbb86 2022-03-07 thomas if (*s == ' ')
247 069bbb86 2022-03-07 thomas s++;
248 069bbb86 2022-03-07 thomas
249 069bbb86 2022-03-07 thomas if (*s != '+' || !*++s)
250 069bbb86 2022-03-07 thomas return got_error(GOT_ERR_PATCH_MALFORMED);
251 069bbb86 2022-03-07 thomas err = strtolnum(&s, &hdr->newfrom);
252 069bbb86 2022-03-07 thomas if (err)
253 069bbb86 2022-03-07 thomas return err;
254 069bbb86 2022-03-07 thomas if (*s == ',') {
255 069bbb86 2022-03-07 thomas s++;
256 069bbb86 2022-03-07 thomas err = strtolnum(&s, &hdr->newlines);
257 069bbb86 2022-03-07 thomas if (err)
258 069bbb86 2022-03-07 thomas return err;
259 069bbb86 2022-03-07 thomas } else
260 069bbb86 2022-03-07 thomas hdr->newlines = 1;
261 069bbb86 2022-03-07 thomas
262 069bbb86 2022-03-07 thomas if (*s == ' ')
263 069bbb86 2022-03-07 thomas s++;
264 069bbb86 2022-03-07 thomas
265 069bbb86 2022-03-07 thomas if (*s != '@')
266 069bbb86 2022-03-07 thomas return got_error(GOT_ERR_PATCH_MALFORMED);
267 069bbb86 2022-03-07 thomas
268 069bbb86 2022-03-07 thomas if (hdr->oldfrom >= LONG_MAX - hdr->oldlines ||
269 069bbb86 2022-03-07 thomas hdr->newfrom >= LONG_MAX - hdr->newlines ||
270 069bbb86 2022-03-07 thomas /* not so sure about this one */
271 069bbb86 2022-03-07 thomas hdr->oldlines >= LONG_MAX - hdr->newlines - 1)
272 069bbb86 2022-03-07 thomas return got_error(GOT_ERR_PATCH_MALFORMED);
273 069bbb86 2022-03-07 thomas
274 069bbb86 2022-03-07 thomas if (hdr->oldlines == 0) {
275 069bbb86 2022-03-07 thomas /* larry says to "do append rather than insert"; I don't
276 069bbb86 2022-03-07 thomas * quite get it, but i trust him.
277 069bbb86 2022-03-07 thomas */
278 069bbb86 2022-03-07 thomas hdr->oldfrom++;
279 069bbb86 2022-03-07 thomas }
280 069bbb86 2022-03-07 thomas
281 069bbb86 2022-03-07 thomas if (imsg_compose(&ibuf, GOT_IMSG_PATCH_HUNK, 0, 0, -1,
282 069bbb86 2022-03-07 thomas hdr, sizeof(*hdr)) == -1)
283 069bbb86 2022-03-07 thomas return got_error_from_errno(
284 069bbb86 2022-03-07 thomas "imsg_compose GOT_IMSG_PATCH_HUNK");
285 069bbb86 2022-03-07 thomas return NULL;
286 069bbb86 2022-03-07 thomas }
287 069bbb86 2022-03-07 thomas
288 069bbb86 2022-03-07 thomas static const struct got_error *
289 069bbb86 2022-03-07 thomas send_line(const char *line)
290 069bbb86 2022-03-07 thomas {
291 069bbb86 2022-03-07 thomas static const struct got_error *err = NULL;
292 069bbb86 2022-03-07 thomas char *p = NULL;
293 069bbb86 2022-03-07 thomas
294 ff7f34d3 2022-03-22 thomas if (*line != '+' && *line != '-' && *line != ' ' && *line != '\\') {
295 069bbb86 2022-03-07 thomas if (asprintf(&p, " %s", line) == -1)
296 069bbb86 2022-03-07 thomas return got_error_from_errno("asprintf");
297 069bbb86 2022-03-07 thomas line = p;
298 069bbb86 2022-03-07 thomas }
299 069bbb86 2022-03-07 thomas
300 069bbb86 2022-03-07 thomas if (imsg_compose(&ibuf, GOT_IMSG_PATCH_LINE, 0, 0, -1,
301 17d6446a 2022-03-22 thomas line, strlen(line) + 1) == -1)
302 069bbb86 2022-03-07 thomas err = got_error_from_errno(
303 069bbb86 2022-03-07 thomas "imsg_compose GOT_IMSG_PATCH_LINE");
304 069bbb86 2022-03-07 thomas
305 069bbb86 2022-03-07 thomas free(p);
306 069bbb86 2022-03-07 thomas return err;
307 069bbb86 2022-03-07 thomas }
308 069bbb86 2022-03-07 thomas
309 069bbb86 2022-03-07 thomas static const struct got_error *
310 ff7f34d3 2022-03-22 thomas peek_special_line(FILE *fp, int send)
311 ff7f34d3 2022-03-22 thomas {
312 ff7f34d3 2022-03-22 thomas const struct got_error *err;
313 ff7f34d3 2022-03-22 thomas char ch;
314 ff7f34d3 2022-03-22 thomas
315 ff7f34d3 2022-03-22 thomas ch = fgetc(fp);
316 ff7f34d3 2022-03-22 thomas if (ch != EOF && ch != '\\') {
317 ff7f34d3 2022-03-22 thomas ungetc(ch, fp);
318 ff7f34d3 2022-03-22 thomas return NULL;
319 ff7f34d3 2022-03-22 thomas }
320 ff7f34d3 2022-03-22 thomas
321 ff7f34d3 2022-03-22 thomas if (ch == '\\' && send) {
322 ff7f34d3 2022-03-22 thomas err = send_line("\\");
323 ff7f34d3 2022-03-22 thomas if (err)
324 ff7f34d3 2022-03-22 thomas return err;
325 ff7f34d3 2022-03-22 thomas }
326 ff7f34d3 2022-03-22 thomas
327 ff7f34d3 2022-03-22 thomas while (ch != EOF && ch != '\n')
328 ff7f34d3 2022-03-22 thomas ch = fgetc(fp);
329 ff7f34d3 2022-03-22 thomas
330 ff7f34d3 2022-03-22 thomas if (ch != EOF || feof(fp))
331 ff7f34d3 2022-03-22 thomas return NULL;
332 ff7f34d3 2022-03-22 thomas return got_error(GOT_ERR_IO);
333 ff7f34d3 2022-03-22 thomas }
334 ff7f34d3 2022-03-22 thomas
335 ff7f34d3 2022-03-22 thomas static const struct got_error *
336 069bbb86 2022-03-07 thomas parse_hunk(FILE *fp, int *ok)
337 069bbb86 2022-03-07 thomas {
338 069bbb86 2022-03-07 thomas static const struct got_error *err = NULL;
339 069bbb86 2022-03-07 thomas struct got_imsg_patch_hunk hdr;
340 069bbb86 2022-03-07 thomas char *line = NULL, ch;
341 069bbb86 2022-03-07 thomas size_t linesize = 0;
342 069bbb86 2022-03-07 thomas ssize_t linelen;
343 069bbb86 2022-03-07 thomas long leftold, leftnew;
344 069bbb86 2022-03-07 thomas
345 069bbb86 2022-03-07 thomas linelen = getline(&line, &linesize, fp);
346 069bbb86 2022-03-07 thomas if (linelen == -1) {
347 069bbb86 2022-03-07 thomas *ok = 0;
348 069bbb86 2022-03-07 thomas goto done;
349 069bbb86 2022-03-07 thomas }
350 069bbb86 2022-03-07 thomas
351 069bbb86 2022-03-07 thomas err = parse_hdr(line, ok, &hdr);
352 069bbb86 2022-03-07 thomas if (err)
353 069bbb86 2022-03-07 thomas goto done;
354 069bbb86 2022-03-07 thomas if (!*ok) {
355 069bbb86 2022-03-07 thomas if (fseek(fp, linelen * -1, SEEK_CUR) == -1)
356 069bbb86 2022-03-07 thomas err = got_error_from_errno("fseek");
357 069bbb86 2022-03-07 thomas goto done;
358 069bbb86 2022-03-07 thomas }
359 069bbb86 2022-03-07 thomas
360 069bbb86 2022-03-07 thomas leftold = hdr.oldlines;
361 069bbb86 2022-03-07 thomas leftnew = hdr.newlines;
362 069bbb86 2022-03-07 thomas
363 069bbb86 2022-03-07 thomas while (leftold > 0 || leftnew > 0) {
364 069bbb86 2022-03-07 thomas linelen = getline(&line, &linesize, fp);
365 069bbb86 2022-03-07 thomas if (linelen == -1) {
366 069bbb86 2022-03-07 thomas if (ferror(fp)) {
367 069bbb86 2022-03-07 thomas err = got_error_from_errno("getline");
368 069bbb86 2022-03-07 thomas goto done;
369 069bbb86 2022-03-07 thomas }
370 069bbb86 2022-03-07 thomas
371 069bbb86 2022-03-07 thomas /* trailing newlines may be chopped */
372 069bbb86 2022-03-07 thomas if (leftold < 3 && leftnew < 3) {
373 069bbb86 2022-03-07 thomas *ok = 0;
374 069bbb86 2022-03-07 thomas break;
375 069bbb86 2022-03-07 thomas }
376 069bbb86 2022-03-07 thomas
377 069bbb86 2022-03-07 thomas err = got_error(GOT_ERR_PATCH_TRUNCATED);
378 069bbb86 2022-03-07 thomas goto done;
379 069bbb86 2022-03-07 thomas }
380 ff7f34d3 2022-03-22 thomas if (line[linelen - 1] == '\n')
381 ff7f34d3 2022-03-22 thomas line[linelen - 1] = '\0';
382 069bbb86 2022-03-07 thomas
383 069bbb86 2022-03-07 thomas /* usr.bin/patch allows '=' as context char */
384 069bbb86 2022-03-07 thomas if (*line == '=')
385 069bbb86 2022-03-07 thomas *line = ' ';
386 069bbb86 2022-03-07 thomas
387 069bbb86 2022-03-07 thomas ch = *line;
388 ff7f34d3 2022-03-22 thomas if (ch == '\t' || ch == '\0')
389 069bbb86 2022-03-07 thomas ch = ' '; /* the space got eaten */
390 069bbb86 2022-03-07 thomas
391 069bbb86 2022-03-07 thomas switch (ch) {
392 069bbb86 2022-03-07 thomas case '-':
393 069bbb86 2022-03-07 thomas leftold--;
394 069bbb86 2022-03-07 thomas break;
395 069bbb86 2022-03-07 thomas case ' ':
396 069bbb86 2022-03-07 thomas leftold--;
397 069bbb86 2022-03-07 thomas leftnew--;
398 069bbb86 2022-03-07 thomas break;
399 069bbb86 2022-03-07 thomas case '+':
400 069bbb86 2022-03-07 thomas leftnew--;
401 069bbb86 2022-03-07 thomas break;
402 069bbb86 2022-03-07 thomas default:
403 069bbb86 2022-03-07 thomas err = got_error(GOT_ERR_PATCH_MALFORMED);
404 069bbb86 2022-03-07 thomas goto done;
405 069bbb86 2022-03-07 thomas }
406 069bbb86 2022-03-07 thomas
407 069bbb86 2022-03-07 thomas if (leftold < 0 || leftnew < 0) {
408 069bbb86 2022-03-07 thomas err = got_error(GOT_ERR_PATCH_MALFORMED);
409 069bbb86 2022-03-07 thomas goto done;
410 069bbb86 2022-03-07 thomas }
411 069bbb86 2022-03-07 thomas
412 069bbb86 2022-03-07 thomas err = send_line(line);
413 069bbb86 2022-03-07 thomas if (err)
414 069bbb86 2022-03-07 thomas goto done;
415 ff7f34d3 2022-03-22 thomas
416 ff7f34d3 2022-03-22 thomas if ((ch == '-' && leftold == 0) ||
417 ff7f34d3 2022-03-22 thomas (ch == '+' && leftnew == 0)) {
418 ff7f34d3 2022-03-22 thomas err = peek_special_line(fp, ch == '+');
419 ff7f34d3 2022-03-22 thomas if (err)
420 ff7f34d3 2022-03-22 thomas goto done;
421 ff7f34d3 2022-03-22 thomas }
422 069bbb86 2022-03-07 thomas }
423 069bbb86 2022-03-07 thomas
424 069bbb86 2022-03-07 thomas done:
425 069bbb86 2022-03-07 thomas free(line);
426 069bbb86 2022-03-07 thomas return err;
427 069bbb86 2022-03-07 thomas }
428 069bbb86 2022-03-07 thomas
429 069bbb86 2022-03-07 thomas static const struct got_error *
430 069bbb86 2022-03-07 thomas read_patch(struct imsgbuf *ibuf, int fd)
431 069bbb86 2022-03-07 thomas {
432 069bbb86 2022-03-07 thomas const struct got_error *err = NULL;
433 069bbb86 2022-03-07 thomas FILE *fp;
434 069bbb86 2022-03-07 thomas int ok, patch_found = 0;
435 069bbb86 2022-03-07 thomas
436 069bbb86 2022-03-07 thomas if ((fp = fdopen(fd, "r")) == NULL) {
437 069bbb86 2022-03-07 thomas err = got_error_from_errno("fdopen");
438 069bbb86 2022-03-07 thomas close(fd);
439 069bbb86 2022-03-07 thomas return err;
440 069bbb86 2022-03-07 thomas }
441 069bbb86 2022-03-07 thomas
442 069bbb86 2022-03-07 thomas while (!feof(fp)) {
443 069bbb86 2022-03-07 thomas err = find_patch(fp);
444 069bbb86 2022-03-07 thomas if (err)
445 069bbb86 2022-03-07 thomas goto done;
446 069bbb86 2022-03-07 thomas
447 069bbb86 2022-03-07 thomas patch_found = 1;
448 069bbb86 2022-03-07 thomas for (;;) {
449 069bbb86 2022-03-07 thomas err = parse_hunk(fp, &ok);
450 069bbb86 2022-03-07 thomas if (err)
451 069bbb86 2022-03-07 thomas goto done;
452 069bbb86 2022-03-07 thomas if (!ok) {
453 069bbb86 2022-03-07 thomas err = send_patch_done();
454 069bbb86 2022-03-07 thomas if (err)
455 069bbb86 2022-03-07 thomas goto done;
456 069bbb86 2022-03-07 thomas break;
457 069bbb86 2022-03-07 thomas }
458 069bbb86 2022-03-07 thomas }
459 069bbb86 2022-03-07 thomas }
460 069bbb86 2022-03-07 thomas
461 069bbb86 2022-03-07 thomas done:
462 069bbb86 2022-03-07 thomas fclose(fp);
463 069bbb86 2022-03-07 thomas
464 069bbb86 2022-03-07 thomas /* ignore trailing gibberish */
465 069bbb86 2022-03-07 thomas if (err != NULL && err->code == GOT_ERR_NO_PATCH && patch_found)
466 069bbb86 2022-03-07 thomas err = NULL;
467 069bbb86 2022-03-07 thomas
468 069bbb86 2022-03-07 thomas return err;
469 069bbb86 2022-03-07 thomas }
470 069bbb86 2022-03-07 thomas
471 069bbb86 2022-03-07 thomas int
472 069bbb86 2022-03-07 thomas main(int argc, char **argv)
473 069bbb86 2022-03-07 thomas {
474 069bbb86 2022-03-07 thomas const struct got_error *err = NULL;
475 069bbb86 2022-03-07 thomas struct imsg imsg;
476 069bbb86 2022-03-07 thomas #if 0
477 069bbb86 2022-03-07 thomas static int attached;
478 069bbb86 2022-03-07 thomas while (!attached)
479 069bbb86 2022-03-07 thomas sleep(1);
480 069bbb86 2022-03-07 thomas #endif
481 069bbb86 2022-03-07 thomas
482 069bbb86 2022-03-07 thomas imsg_init(&ibuf, GOT_IMSG_FD_CHILD);
483 069bbb86 2022-03-07 thomas #ifndef PROFILE
484 069bbb86 2022-03-07 thomas /* revoke access to most system calls */
485 069bbb86 2022-03-07 thomas if (pledge("stdio recvfd", NULL) == -1) {
486 069bbb86 2022-03-07 thomas err = got_error_from_errno("pledge");
487 069bbb86 2022-03-07 thomas got_privsep_send_error(&ibuf, err);
488 069bbb86 2022-03-07 thomas return 1;
489 069bbb86 2022-03-07 thomas }
490 762ddcd8 2022-03-09 thomas
491 762ddcd8 2022-03-09 thomas /* revoke fs access */
492 762ddcd8 2022-03-09 thomas if (landlock_no_fs() == -1) {
493 762ddcd8 2022-03-09 thomas err = got_error_from_errno("landlock_no_fs");
494 762ddcd8 2022-03-09 thomas got_privsep_send_error(&ibuf, err);
495 762ddcd8 2022-03-09 thomas return 1;
496 762ddcd8 2022-03-09 thomas }
497 069bbb86 2022-03-07 thomas #endif
498 069bbb86 2022-03-07 thomas
499 069bbb86 2022-03-07 thomas err = got_privsep_recv_imsg(&imsg, &ibuf, 0);
500 069bbb86 2022-03-07 thomas if (err)
501 069bbb86 2022-03-07 thomas goto done;
502 069bbb86 2022-03-07 thomas if (imsg.hdr.type != GOT_IMSG_PATCH_FILE || imsg.fd == -1) {
503 069bbb86 2022-03-07 thomas err = got_error(GOT_ERR_PRIVSEP_MSG);
504 069bbb86 2022-03-07 thomas goto done;
505 069bbb86 2022-03-07 thomas }
506 069bbb86 2022-03-07 thomas
507 069bbb86 2022-03-07 thomas err = read_patch(&ibuf, imsg.fd);
508 069bbb86 2022-03-07 thomas if (err)
509 069bbb86 2022-03-07 thomas goto done;
510 069bbb86 2022-03-07 thomas if (imsg_compose(&ibuf, GOT_IMSG_PATCH_EOF, 0, 0, -1,
511 069bbb86 2022-03-07 thomas NULL, 0) == -1) {
512 069bbb86 2022-03-07 thomas err = got_error_from_errno("imsg_compose GOT_IMSG_PATCH_EOF");
513 069bbb86 2022-03-07 thomas goto done;
514 069bbb86 2022-03-07 thomas }
515 069bbb86 2022-03-07 thomas err = got_privsep_flush_imsg(&ibuf);
516 069bbb86 2022-03-07 thomas done:
517 069bbb86 2022-03-07 thomas imsg_free(&imsg);
518 069bbb86 2022-03-07 thomas if (err != NULL) {
519 069bbb86 2022-03-07 thomas got_privsep_send_error(&ibuf, err);
520 069bbb86 2022-03-07 thomas err = NULL;
521 069bbb86 2022-03-07 thomas }
522 069bbb86 2022-03-07 thomas if (close(GOT_IMSG_FD_CHILD) == -1 && err == NULL)
523 069bbb86 2022-03-07 thomas err = got_error_from_errno("close");
524 069bbb86 2022-03-07 thomas if (err && err->code != GOT_ERR_PRIVSEP_PIPE)
525 069bbb86 2022-03-07 thomas fprintf(stderr, "%s: %s\n", getprogname(), err->msg);
526 069bbb86 2022-03-07 thomas return err ? 1 : 0;
527 069bbb86 2022-03-07 thomas }