NetBSD/usr.bin/patch/util.c

487 lines
10 KiB
C
Raw Normal View History

2005-03-26 02:00:55 +03:00
/* $NetBSD: util.c,v 1.21 2005/03/25 23:00:55 skd Exp $ */
/*
* Copyright (c) 1988, Larry Wall
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following condition
* is met:
* 1. Redistributions of source code must retain the above copyright
* notice, this condition and the following disclaimer.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
1998-02-22 15:55:44 +03:00
#include <sys/cdefs.h>
#ifndef lint
2005-03-26 02:00:55 +03:00
__RCSID("$NetBSD: util.c,v 1.21 2005/03/25 23:00:55 skd Exp $");
#endif /* not lint */
#include <sys/param.h>
#include "EXTERN.h"
#include "common.h"
#include "INTERN.h"
#include "util.h"
#include "backupfile.h"
#include <stdarg.h>
1998-02-22 15:55:44 +03:00
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
2002-03-12 02:29:02 +03:00
/*
* Rename a file, copying it if necessary.
*/
int
move_file(char *from, char *to)
{
char bakname[MAXPATHLEN];
2002-03-12 02:29:02 +03:00
char *s;
size_t i;
2002-03-12 02:29:02 +03:00
int fromfd;
2005-03-26 02:00:55 +03:00
if ( check_only == TRUE )
return(0);
2002-03-12 02:29:02 +03:00
/* to stdout? */
2002-03-12 02:29:02 +03:00
if (strEQ(to, "-")) {
#ifdef DEBUGGING
2002-03-12 02:29:02 +03:00
if (debug & 4)
say("Moving %s to stdout.\n", from);
#endif
2002-03-12 02:29:02 +03:00
fromfd = open(from, 0);
if (fromfd < 0)
pfatal("internal error, can't reopen %s", from);
while ((i = read(fromfd, buf, sizeof buf)) > 0)
if (write(1, buf, i) != 1)
pfatal("write failed");
Close(fromfd);
return 0;
}
2002-03-12 02:29:02 +03:00
if (origprae) {
2003-07-30 12:51:04 +04:00
strlcpy(bakname, origprae, sizeof(bakname));
strlcat(bakname, to, sizeof(bakname));
2002-03-12 02:29:02 +03:00
} else {
#ifndef NODIR
2002-03-12 02:29:02 +03:00
char *backupname = find_backup_file_name(to);
2003-07-30 12:51:04 +04:00
strlcpy(bakname, backupname, sizeof(bakname));
2002-03-12 02:29:02 +03:00
free(backupname);
#else /* NODIR */
2003-07-30 12:51:04 +04:00
strlcpy(bakname, to, sizeof(bakname));
strlcat(bakname, simple_backup_suffix, sizeof(bakname));
#endif /* NODIR */
2002-03-12 02:29:02 +03:00
}
2002-03-12 02:29:02 +03:00
if (stat(to, &filestat) == 0) { /* output file exists */
dev_t to_device = filestat.st_dev;
ino_t to_inode = filestat.st_ino;
char *simplename = bakname;
2002-03-12 02:29:02 +03:00
for (s = bakname; *s; s++) {
if (*s == '/')
simplename = s + 1;
}
/*
* Find a backup name that is not the same file.
* Change the first lowercase char into uppercase;
* if that isn't sufficient, chop off the first char
* and try again.
*/
while (stat(bakname, &filestat) == 0 &&
to_device == filestat.st_dev &&
to_inode == filestat.st_ino) {
/* Skip initial non-lowercase chars. */
for (s = simplename;
*s && *s == toupper((unsigned char)*s);
2002-03-12 02:29:02 +03:00
s++)
;
if (*s)
*s = toupper((unsigned char)*s);
2002-03-12 02:29:02 +03:00
else
2003-07-30 12:51:55 +04:00
strcpy(simplename, simplename + 1);
2002-03-12 02:29:02 +03:00
}
while (unlink(bakname) >= 0)
; /* while() is for benefit of Eunice */
#ifdef DEBUGGING
2002-03-12 02:29:02 +03:00
if (debug & 4)
say("Moving %s to %s.\n", to, bakname);
#endif
2002-03-12 02:29:02 +03:00
if (link(to, bakname) < 0) {
/*
* Maybe `to' is a symlink into a different file
* system. Copying replaces the symlink with a file;
* using rename would be better.
*/
int tofd;
int bakfd;
bakfd = creat(bakname, 0666);
if (bakfd < 0) {
say("Can't backup %s, output is in %s: %s\n",
to, from, strerror(errno));
return -1;
}
tofd = open(to, 0);
if (tofd < 0)
pfatal("internal error, can't open %s", to);
while ((i = read(tofd, buf, sizeof buf)) > 0)
if (write(bakfd, buf, i) != i)
pfatal("write failed");
Close(tofd);
Close(bakfd);
}
while (unlink(to) >= 0) ;
}
#ifdef DEBUGGING
2002-03-12 02:29:02 +03:00
if (debug & 4)
say("Moving %s to %s.\n", from, to);
#endif
2002-03-12 02:29:02 +03:00
if (link(from, to) < 0) { /* different file system? */
int tofd;
tofd = creat(to, 0666);
if (tofd < 0) {
say("Can't create %s, output is in %s: %s\n",
to, from, strerror(errno));
return -1;
}
fromfd = open(from, 0);
if (fromfd < 0)
pfatal("internal error, can't reopen %s", from);
while ((i = read(fromfd, buf, sizeof buf)) > 0)
if (write(tofd, buf, i) != i)
pfatal("write failed");
Close(fromfd);
Close(tofd);
}
2002-03-12 02:29:02 +03:00
Unlink(from);
return 0;
}
2002-03-12 02:29:02 +03:00
/*
* Copy a file.
*/
void
copy_file(char *from, char *to)
{
2002-03-12 02:29:02 +03:00
int tofd;
int fromfd;
size_t i;
2002-03-12 02:29:02 +03:00
tofd = creat(to, 0666);
if (tofd < 0)
pfatal("can't create %s", to);
fromfd = open(from, 0);
if (fromfd < 0)
pfatal("internal error, can't reopen %s", from);
while ((i = read(fromfd, buf, sizeof buf)) > 0)
if (write(tofd, buf, i) != i)
pfatal("write to %s failed", to);
Close(fromfd);
Close(tofd);
}
/*
* malloc with result test.
*/
void *
xmalloc(size_t size)
{
void *p;
if ((p = malloc(size)) == NULL)
fatal("out of memory\n");
return p;
}
/*
* realloc with result test.
*/
void *
xrealloc(void *ptr, size_t size)
{
void *p;
if ((p = realloc(ptr, size)) == NULL)
fatal("out of memory\n");
return p;
}
2002-03-12 02:29:02 +03:00
/*
* strdup with result test.
2002-03-12 02:29:02 +03:00
*/
char *
xstrdup(const char *s)
{
char *p;
2002-03-12 02:29:02 +03:00
if ((p = strdup(s)) == NULL)
fatal("out of memory\n");
return p;
}
2002-03-12 02:29:02 +03:00
/*
* Vanilla terminal output.
2002-03-12 02:29:02 +03:00
*/
void
1998-02-22 15:55:44 +03:00
say(const char *pat, ...)
{
2002-03-12 02:29:02 +03:00
va_list ap;
va_start(ap, pat);
1998-02-22 15:55:44 +03:00
2002-03-12 02:29:02 +03:00
vfprintf(stderr, pat, ap);
va_end(ap);
Fflush(stderr);
}
2002-03-12 02:29:02 +03:00
/*
* Terminal output, pun intended.
*/
void /* very void */
1998-02-22 15:55:44 +03:00
fatal(const char *pat, ...)
{
2002-03-12 02:29:02 +03:00
va_list ap;
va_start(ap, pat);
1998-02-22 15:55:44 +03:00
2002-03-12 02:29:02 +03:00
fprintf(stderr, "patch: **** ");
vfprintf(stderr, pat, ap);
va_end(ap);
my_exit(1);
}
2002-03-12 02:29:02 +03:00
/*
* Say something from patch, something from the system, then silence...
*/
void /* very void */
1998-02-22 15:55:44 +03:00
pfatal(const char *pat, ...)
{
2002-03-12 02:29:02 +03:00
va_list ap;
int errnum = errno;
va_start(ap, pat);
1998-02-22 15:55:44 +03:00
2002-03-12 02:29:02 +03:00
fprintf(stderr, "patch: **** ");
vfprintf(stderr, pat, ap);
fprintf(stderr, ": %s\n", strerror(errnum));
va_end(ap);
my_exit(1);
}
2002-03-12 02:29:02 +03:00
/*
* Get a response from the user, somehow or other.
*/
void
1998-02-22 15:55:44 +03:00
ask(const char *pat, ...)
{
2002-03-12 02:29:02 +03:00
int ttyfd;
int r;
bool tty2 = isatty(2);
va_list ap;
va_start(ap, pat);
(void)vsprintf(buf, pat, ap);
va_end(ap);
Fflush(stderr);
write(2, buf, strlen(buf));
if (tty2) { /* might be redirected to a file */
r = read(2, buf, sizeof buf);
} else if (isatty(1)) { /* this may be new file output */
Fflush(stdout);
write(1, buf, strlen(buf));
r = read(1, buf, sizeof buf);
} else if ((ttyfd = open("/dev/tty", 2)) >= 0 && isatty(ttyfd)) {
2003-01-06 23:30:28 +03:00
/* might be deleted or unwritable */
2002-03-12 02:29:02 +03:00
write(ttyfd, buf, strlen(buf));
r = read(ttyfd, buf, sizeof buf);
Close(ttyfd);
} else if (isatty(0)) { /* this is probably patch input */
Fflush(stdin);
write(0, buf, strlen(buf));
r = read(0, buf, sizeof buf);
} else { /* no terminal at all--default it */
buf[0] = '\n';
r = 1;
}
if (r <= 0)
buf[0] = 0;
else
buf[r] = '\0';
if (!tty2)
say("%s", buf);
}
2002-03-12 02:29:02 +03:00
/*
* How to handle certain events when not in a critical region.
*/
void
set_signals(int reset)
{
2002-03-12 02:29:02 +03:00
static void (*hupval)(int);
static void (*intval)(int);
if (!reset) {
hupval = signal(SIGHUP, SIG_IGN);
if (hupval != SIG_IGN)
hupval = my_exit;
intval = signal(SIGINT, SIG_IGN);
if (intval != SIG_IGN)
intval = my_exit;
}
Signal(SIGHUP, hupval);
Signal(SIGINT, intval);
}
2002-03-12 02:29:02 +03:00
/*
* How to handle certain events when in a critical region.
*/
void
ignore_signals()
{
2002-03-12 02:29:02 +03:00
Signal(SIGHUP, SIG_IGN);
Signal(SIGINT, SIG_IGN);
}
2002-03-12 02:29:02 +03:00
/*
* Make sure we'll have the directories to create a file.
* If `striplast' is TRUE, ignore the last element of `filename'.
*/
void
makedirs(char *filename, bool striplast)
{
char tmpbuf[MAXPATHLEN];
2002-03-12 02:29:02 +03:00
char *s = tmpbuf;
char *dirv[MAXPATHLEN]; /* Point to the NULs between elements. */
2002-03-12 02:29:02 +03:00
int i;
int dirvp = 0; /* Number of finished entries in dirv. */
/*
* Copy `filename' into `tmpbuf' with a NUL instead of a slash
* between the directories.
*/
while (*filename) {
if (*filename == '/') {
filename++;
dirv[dirvp++] = s;
*s++ = '\0';
} else {
*s++ = *filename++;
}
}
2002-03-12 02:29:02 +03:00
*s = '\0';
dirv[dirvp] = s;
if (striplast)
dirvp--;
if (dirvp < 0)
return;
2003-07-30 12:51:04 +04:00
strlcpy(buf, "mkdir", sizeof(buf));
2002-03-12 02:29:02 +03:00
s = buf;
for (i = 0; i <= dirvp; i++) {
struct stat sbuf;
if (stat(tmpbuf, &sbuf) && errno == ENOENT) {
while (*s)
s++;
*s++ = ' ';
2003-07-30 12:51:04 +04:00
strlcpy(s, tmpbuf, sizeof(buf) - (s - buf));
2002-03-12 02:29:02 +03:00
}
*dirv[i] = '/';
}
2002-03-12 02:29:02 +03:00
if (s != buf)
system(buf);
}
2002-03-12 02:29:02 +03:00
/*
* Make filenames more reasonable.
*/
char *
fetchname(char *at, int strip_leading, int assume_exists)
{
2002-03-12 02:29:02 +03:00
char *fullname;
char *name;
char *t;
char tmpbuf[MAXPATHLEN];
2002-03-12 02:29:02 +03:00
int sleading = strip_leading;
if (!at)
return NULL;
while (isspace((unsigned char)*at))
at++;
#ifdef DEBUGGING
2002-03-12 02:29:02 +03:00
if (debug & 128)
say("fetchname %s %d %d\n", at, strip_leading, assume_exists);
#endif
2002-03-12 02:29:02 +03:00
filename_is_dev_null = FALSE;
if (strnEQ(at, "/dev/null", 9)) {
/* So files can be created by diffing against /dev/null. */
filename_is_dev_null = TRUE;
return NULL;
}
name = fullname = t = xstrdup(at);
2002-03-12 02:29:02 +03:00
/* Strip off up to `sleading' leading slashes and null terminate. */
for (; *t && !isspace((unsigned char)*t); t++)
if (*t == '/')
if (--sleading >= 0)
name = t + 1;
*t = '\0';
/*
* If no -p option was given (957 is the default value!),
* we were given a relative pathname,
* and the leading directories that we just stripped off all exist,
* put them back on.
*/
if (strip_leading == 957 && name != fullname && *fullname != '/') {
name[-1] = '\0';
if (stat(fullname, &filestat) == 0 &&
S_ISDIR(filestat.st_mode)) {
name[-1] = '/';
name = fullname;
}
}
2002-03-12 02:29:02 +03:00
name = xstrdup(name);
2002-03-12 02:29:02 +03:00
free(fullname);
if (stat(name, &filestat) && !assume_exists) {
char *filebase = basename(name);
size_t pathlen = filebase - name;
2002-03-12 02:29:02 +03:00
/* Put any leading path into `tmpbuf'. */
if (pathlen >= sizeof(tmpbuf))
return NULL;
2002-03-12 02:29:02 +03:00
strncpy(tmpbuf, name, pathlen);
tmpbuf[pathlen] = '\0';
2002-03-12 02:29:02 +03:00
#define try(f, a1, a2) \
2003-07-30 12:51:04 +04:00
(snprintf(tmpbuf + pathlen, sizeof(tmpbuf) - pathlen, f, a1, a2), \
stat(tmpbuf, &filestat) == 0)
2002-03-12 02:29:02 +03:00
#define try1(f, a1) \
2003-07-30 12:51:04 +04:00
(snprintf(tmpbuf + pathlen, sizeof(tmpbuf) - pathlen, f, a1), \
stat(tmpbuf, &filestat) == 0)
2002-03-12 02:29:02 +03:00
if (try("RCS/%s%s", filebase, RCSSUFFIX) ||
try1("RCS/%s" , filebase) ||
try( "%s%s", filebase, RCSSUFFIX) ||
try("SCCS/%s%s", SCCSPREFIX, filebase) ||
try( "%s%s", SCCSPREFIX, filebase))
return name;
free(name);
name = NULL;
}
return name;
}