1 /** $MirOS: src/usr.bin/patch/util.c,v 1.3 2005/11/23 18:04:15 tg Exp $ */
2 /* $OpenBSD: util.c,v 1.31 2005/06/20 07:14:06 otto Exp $ */
3
4 /*
5 * patch - a program to apply diffs to original files
6 *
7 * Copyright 1986, Larry Wall
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following condition is met:
11 * 1. Redistributions of source code must retain the above copyright notice,
12 * this condition and the following disclaimer.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND ANY
15 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
16 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
17 * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE FOR
18 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
20 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
21 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 *
26 * -C option added in 1998, original code by Marc Espie, based on FreeBSD
27 * behaviour
28 */
29
30 #include <sys/param.h>
31 #include <sys/stat.h>
32
33 #include <ctype.h>
34 #include <errno.h>
35 #include <fcntl.h>
36 #include <libgen.h>
37 #include <paths.h>
38 #include <signal.h>
39 #include <stdarg.h>
40 #include <stdlib.h>
41 #include <stdio.h>
42 #include <string.h>
43 #include <unistd.h>
44
45 #include "common.h"
46 #include "util.h"
47 #include "backupfile.h"
48 #include "pathnames.h"
49
50 __RCSID("$MirOS: src/usr.bin/patch/util.c,v 1.3 2005/11/23 18:04:15 tg Exp $");
51
52 /* Rename a file, copying it if necessary. */
53
54 int
move_file(const char * from,const char * to)55 move_file(const char *from, const char *to)
56 {
57 int fromfd;
58 ssize_t i;
59
60 /* to stdout? */
61
62 if (strEQ(to, "-")) {
63 #ifdef DEBUGGING
64 if (debug & 4)
65 say("Moving %s to stdout.\n", from);
66 #endif
67 fromfd = open(from, O_RDONLY);
68 if (fromfd < 0)
69 pfatal("internal error, can't reopen %s", from);
70 while ((i = read(fromfd, buf, sizeof buf)) > 0)
71 if (write(STDOUT_FILENO, buf, i) != i)
72 pfatal("write failed");
73 close(fromfd);
74 return 0;
75 }
76 if (backup_file(to) < 0) {
77 say("Can't backup %s, output is in %s: %s\n", to, from,
78 strerror(errno));
79 return -1;
80 }
81 #ifdef DEBUGGING
82 if (debug & 4)
83 say("Moving %s to %s.\n", from, to);
84 #endif
85 if (rename(from, to) < 0) {
86 if (errno != EXDEV || copy_file(from, to) < 0) {
87 say("Can't create %s, output is in %s: %s\n",
88 to, from, strerror(errno));
89 return -1;
90 }
91 }
92 return 0;
93 }
94
95 /* Backup the original file. */
96
97 int
backup_file(const char * orig)98 backup_file(const char *orig)
99 {
100 struct stat filestat;
101 char bakname[MAXPATHLEN], *s, *simplename;
102 dev_t orig_device;
103 ino_t orig_inode;
104
105 if (backup_type == none || stat(orig, &filestat) != 0)
106 return 0; /* nothing to do */
107 orig_device = filestat.st_dev;
108 orig_inode = filestat.st_ino;
109
110 if (origprae) {
111 if (strlcpy(bakname, origprae, sizeof(bakname)) >= sizeof(bakname) ||
112 strlcat(bakname, orig, sizeof(bakname)) >= sizeof(bakname))
113 fatal("filename %s too long for buffer\n", origprae);
114 } else {
115 if ((s = find_backup_file_name(orig)) == NULL)
116 fatal("out of memory\n");
117 if (strlcpy(bakname, s, sizeof(bakname)) >= sizeof(bakname))
118 fatal("filename %s too long for buffer\n", s);
119 free(s);
120 }
121
122 if ((simplename = strrchr(bakname, '/')) != NULL)
123 simplename = simplename + 1;
124 else
125 simplename = bakname;
126
127 /*
128 * Find a backup name that is not the same file. Change the
129 * first lowercase char into uppercase; if that isn't
130 * sufficient, chop off the first char and try again.
131 */
132 while (stat(bakname, &filestat) == 0 &&
133 orig_device == filestat.st_dev && orig_inode == filestat.st_ino) {
134 /* Skip initial non-lowercase chars. */
135 for (s = simplename; *s && !islower(*s); s++)
136 ;
137 if (*s)
138 *s = toupper(*s);
139 else
140 memmove(simplename, simplename + 1,
141 strlen(simplename + 1) + 1);
142 }
143 #ifdef DEBUGGING
144 if (debug & 4)
145 say("Moving %s to %s.\n", orig, bakname);
146 #endif
147 if (rename(orig, bakname) < 0) {
148 if (errno != EXDEV || copy_file(orig, bakname) < 0)
149 return -1;
150 }
151 return 0;
152 }
153
154 /*
155 * Copy a file.
156 */
157 int
copy_file(const char * from,const char * to)158 copy_file(const char *from, const char *to)
159 {
160 int tofd, fromfd;
161 ssize_t i;
162
163 tofd = open(to, O_CREAT|O_TRUNC|O_WRONLY, 0666);
164 if (tofd < 0)
165 return -1;
166 fromfd = open(from, O_RDONLY, 0);
167 if (fromfd < 0)
168 pfatal("internal error, can't reopen %s", from);
169 while ((i = read(fromfd, buf, sizeof buf)) > 0)
170 if (write(tofd, buf, i) != i)
171 pfatal("write to %s failed", to);
172 close(fromfd);
173 close(tofd);
174 return 0;
175 }
176
177 /*
178 * Allocate a unique area for a string.
179 */
180 char *
savestr(const char * s)181 savestr(const char *s)
182 {
183 char *rv;
184
185 if (!s)
186 s = "Oops";
187 rv = strdup(s);
188 if (rv == NULL) {
189 if (using_plan_a)
190 out_of_mem = true;
191 else
192 fatal("out of memory\n");
193 }
194 return rv;
195 }
196
197 /*
198 * Vanilla terminal output (buffered).
199 */
200 void
say(const char * fmt,...)201 say(const char *fmt, ...)
202 {
203 va_list ap;
204
205 va_start(ap, fmt);
206 vfprintf(stderr, fmt, ap);
207 va_end(ap);
208 fflush(stderr);
209 }
210
211 /*
212 * Terminal output, pun intended.
213 */
214 void
fatal(const char * fmt,...)215 fatal(const char *fmt, ...)
216 {
217 va_list ap;
218
219 va_start(ap, fmt);
220 fprintf(stderr, "patch: **** ");
221 vfprintf(stderr, fmt, ap);
222 va_end(ap);
223 my_exit(2);
224 }
225
226 /*
227 * Say something from patch, something from the system, then silence . . .
228 */
229 void
pfatal(const char * fmt,...)230 pfatal(const char *fmt, ...)
231 {
232 va_list ap;
233 int errnum = errno;
234
235 fprintf(stderr, "patch: **** ");
236 va_start(ap, fmt);
237 vfprintf(stderr, fmt, ap);
238 va_end(ap);
239 fprintf(stderr, ": %s\n", strerror(errnum));
240 my_exit(2);
241 }
242
243 /*
244 * Get a response from the user via /dev/tty
245 */
246 void
ask(const char * fmt,...)247 ask(const char *fmt, ...)
248 {
249 va_list ap;
250 ssize_t nr;
251 static int ttyfd = -1;
252
253 va_start(ap, fmt);
254 vfprintf(stdout, fmt, ap);
255 va_end(ap);
256 fflush(stdout);
257 if (ttyfd < 0)
258 ttyfd = open(_PATH_TTY, O_RDONLY);
259 if (ttyfd >= 0) {
260 if ((nr = read(ttyfd, buf, sizeof(buf))) > 0 &&
261 buf[nr - 1] == '\n')
262 buf[nr - 1] = '\0';
263 }
264 if (ttyfd < 0 || nr <= 0) {
265 /* no tty or error reading, pretend user entered 'return' */
266 putchar('\n');
267 buf[0] = '\0';
268 }
269 }
270
271 /*
272 * How to handle certain events when not in a critical region.
273 */
274 void
set_signals(int reset)275 set_signals(int reset)
276 {
277 static sig_t hupval, intval;
278
279 if (!reset) {
280 hupval = signal(SIGHUP, SIG_IGN);
281 if (hupval != SIG_IGN)
282 hupval = (sig_t) my_exit;
283 intval = signal(SIGINT, SIG_IGN);
284 if (intval != SIG_IGN)
285 intval = (sig_t) my_exit;
286 }
287 signal(SIGHUP, hupval);
288 signal(SIGINT, intval);
289 }
290
291 /*
292 * How to handle certain events when in a critical region.
293 */
294 void
ignore_signals(void)295 ignore_signals(void)
296 {
297 signal(SIGHUP, SIG_IGN);
298 signal(SIGINT, SIG_IGN);
299 }
300
301 /*
302 * Make sure we'll have the directories to create a file. If `striplast' is
303 * true, ignore the last element of `filename'.
304 */
305
306 void
makedirs(const char * filename,bool striplast)307 makedirs(const char *filename, bool striplast)
308 {
309 char *tmpbuf;
310
311 if ((tmpbuf = strdup(filename)) == NULL)
312 fatal("out of memory\n");
313
314 if (striplast) {
315 char *s = strrchr(tmpbuf, '/');
316 if (s == NULL)
317 return; /* nothing to be done */
318 *s = '\0';
319 }
320 if (mkpath(tmpbuf) != 0)
321 pfatal("creation of %s failed", tmpbuf);
322 free(tmpbuf);
323 }
324
325 /*
326 * Make filenames more reasonable.
327 */
328 char *
fetchname(const char * at,bool * exists,int strip_leading)329 fetchname(const char *at, bool *exists, int strip_leading)
330 {
331 char *fullname, *name, *t;
332 int sleading, tab;
333 struct stat filestat;
334
335 if (at == NULL || *at == '\0')
336 return NULL;
337 while (isspace(*at))
338 at++;
339 #ifdef DEBUGGING
340 if (debug & 128)
341 say("fetchname %s %d\n", at, strip_leading);
342 #endif
343 /* So files can be created by diffing against /dev/null. */
344 if (strnEQ(at, _PATH_DEVNULL, sizeof(_PATH_DEVNULL) - 1))
345 return NULL;
346 name = fullname = t = savestr(at);
347
348 tab = strchr(t, '\t') != NULL;
349 /* Strip off up to `strip_leading' path components and NUL terminate. */
350 for (sleading = strip_leading; *t != '\0' && ((tab && *t != '\t') ||
351 !isspace(*t)); t++) {
352 if (t[0] == '/' && t[1] != '/' && t[1] != '\0')
353 if (--sleading >= 0)
354 name = t + 1;
355 }
356 *t = '\0';
357
358 /*
359 * If no -p option was given (957 is the default value!), we were
360 * given a relative pathname, and the leading directories that we
361 * just stripped off all exist, put them back on.
362 */
363 if (strip_leading == 957 && name != fullname && *fullname != '/') {
364 name[-1] = '\0';
365 if (stat(fullname, &filestat) == 0 && S_ISDIR(filestat.st_mode)) {
366 name[-1] = '/';
367 name = fullname;
368 }
369 }
370 name = savestr(name);
371 free(fullname);
372
373 *exists = stat(name, &filestat) == 0;
374 return name;
375 }
376
377 /*
378 * Takes the name returned by fetchname and looks in RCS/SCCS directories
379 * for a checked in version.
380 */
381 char *
checked_in(char * file)382 checked_in(char *file)
383 {
384 char *filebase, *filedir, tmpbuf[MAXPATHLEN];
385 struct stat filestat;
386
387 filebase = basename(file);
388 filedir = dirname(file);
389
390 #define try(f, a1, a2, a3) \
391 (snprintf(tmpbuf, sizeof tmpbuf, f, a1, a2, a3), stat(tmpbuf, &filestat) == 0)
392
393 if (try("%s/RCS/%s%s", filedir, filebase, RCSSUFFIX) ||
394 try("%s/RCS/%s%s", filedir, filebase, "") ||
395 try("%s/%s%s", filedir, filebase, RCSSUFFIX) ||
396 try("%s/SCCS/%s%s", filedir, SCCSPREFIX, filebase) ||
397 try("%s/%s%s", filedir, SCCSPREFIX, filebase))
398 return file;
399
400 return NULL;
401 }
402
403 void
version(void)404 version(void)
405 {
406 fprintf(stderr, "Patch version 2.0-12u8-OpenBSD\n");
407 my_exit(EXIT_SUCCESS);
408 }
409
410 /*
411 * Exit with cleanup.
412 */
413 void
my_exit(int status)414 my_exit(int status)
415 {
416 unlink(TMPINNAME);
417 if (!toutkeep)
418 unlink(TMPOUTNAME);
419 if (!trejkeep)
420 unlink(TMPREJNAME);
421 unlink(TMPPATNAME);
422 exit(status);
423 }
424