freebsd-dev/gnu/usr.bin/patch/inp.c
Garance A Drosehn d77254e807 Change parts of the source code to make it closer to style(9), mainly
changing indentation and some comments.  Main goal is not perfect style,
but just to reduce differences with NetBSD.  The object code is exactly
the same after this change as before it (except for assert() statements
which have moved).

Reviewed by:	/sbin/md5 on i386
Obtained from:	NetBSD
2002-05-21 00:44:18 +00:00

398 lines
9.7 KiB
C

/* $FreeBSD$
*
* $Log: inp.c,v $
* Revision 2.0.1.1 88/06/03 15:06:13 lwall
* patch10: made a little smarter about sccs files
*
* Revision 2.0 86/09/17 15:37:02 lwall
* Baseline for netwide release.
*
*/
#include "EXTERN.h"
#include "common.h"
#include "util.h"
#include "pch.h"
#include "INTERN.h"
#include "inp.h"
/* Input-file-with-indexable-lines abstract type. */
static long i_size; /* Size of the input file */
static char *i_womp; /* Plan a buffer for entire file */
static char **i_ptr; /* Pointers to lines in i_womp */
static int tifd = -1; /* Plan b virtual string array */
static char *tibuf[2]; /* Plan b buffers */
static LINENUM tiline[2] = {-1, -1}; /* 1st line in each buffer */
static LINENUM lines_per_buf; /* How many lines per buffer */
static int tireclen; /* Length of records in tmp file */
/*
* New patch--prepare to edit another file.
*/
void
re_input(void)
{
if (using_plan_a) {
i_size = 0;
#ifndef lint
if (i_ptr != Null(char**))
free((char *)i_ptr);
#endif
if (i_womp != Nullch)
free(i_womp);
i_womp = Nullch;
i_ptr = Null(char **);
} else {
using_plan_a = TRUE; /* maybe the next one is smaller */
Close(tifd);
tifd = -1;
free(tibuf[0]);
free(tibuf[1]);
tibuf[0] = tibuf[1] = Nullch;
tiline[0] = tiline[1] = -1;
tireclen = 0;
}
}
/*
* Constuct the line index, somehow or other.
*/
void
scan_input(char *filename)
{
if (!plan_a(filename))
plan_b(filename);
if (verbose) {
say3("Patching file %s using Plan %s...\n", filename,
(using_plan_a ? "A" : "B") );
}
}
/*
* Try keeping everything in memory.
*/
bool
plan_a(char *filename)
{
int ifd, statfailed;
Reg1 char *s;
Reg2 LINENUM iline;
char lbuf[MAXLINELEN];
int output_elsewhere = strcmp(filename, outname);
extern int check_patch;
statfailed = stat(filename, &filestat);
if (statfailed && ok_to_create_file) {
if (verbose)
say2("(Creating file %s...)\n",filename);
if (check_patch)
return TRUE;
makedirs(filename, TRUE);
close(creat(filename, 0666));
statfailed = stat(filename, &filestat);
}
if (statfailed && check_patch) {
fatal2("%s not found and in check_patch mode.", filename);
}
/*
* For nonexistent or read-only files, look for RCS or SCCS
* versions.
*/
if (statfailed ||
(! output_elsewhere &&
(/* No one can write to it. */
(filestat.st_mode & 0222) == 0 ||
/* I can't write to it. */
((filestat.st_mode & 0022) == 0 && filestat.st_uid != myuid)))) {
struct stat cstat;
char *cs = Nullch;
char *filebase;
int pathlen;
filebase = basename(filename);
pathlen = filebase - filename;
/*
* Put any leading path into `s'.
* Leave room in lbuf for the diff command.
*/
s = lbuf + 20;
strncpy(s, filename, pathlen);
#define try(f,a1,a2) (Sprintf(s + pathlen, f, a1, a2), stat(s, &cstat) == 0)
if ((try("RCS/%s%s", filebase, RCSSUFFIX) ||
try("RCS/%s%s", filebase, "") ||
try("%s%s", filebase, RCSSUFFIX)) &&
/*
* Check that RCS file is not working file.
* Some hosts don't report file name length errors.
*/
(statfailed ||
((filestat.st_dev ^ cstat.st_dev) |
(filestat.st_ino ^ cstat.st_ino)))) {
Sprintf(buf, output_elsewhere?CHECKOUT:CHECKOUT_LOCKED, filename);
Sprintf(lbuf, RCSDIFF, filename);
cs = "RCS";
} else if (try("SCCS/%s%s", SCCSPREFIX, filebase) ||
try("%s%s", SCCSPREFIX, filebase)) {
Sprintf(buf, output_elsewhere?GET:GET_LOCKED, s);
Sprintf(lbuf, SCCSDIFF, s, filename);
cs = "SCCS";
} else if (statfailed)
fatal2("can't find %s\n", filename);
/*
* else we can't write to it but it's not under a version
* control system, so just proceed.
*/
if (cs) {
if (!statfailed) {
if ((filestat.st_mode & 0222) != 0)
/* The owner can write to it. */
fatal3(
"file %s seems to be locked by somebody else under %s\n",
filename, cs);
/*
* It might be checked out unlocked. See if
* it's safe to check out the default version
* locked.
*/
if (verbose)
say3(
"Comparing file %s to default %s version...\n",
filename, cs);
if (system(lbuf))
fatal3(
"can't check out file %s: differs from default %s version\n",
filename, cs);
}
if (verbose)
say3("Checking out file %s from %s...\n", filename, cs);
if (system(buf) || stat(filename, &filestat))
fatal3("can't check out file %s from %s\n",
filename, cs);
}
}
filemode = filestat.st_mode;
if (!S_ISREG(filemode))
fatal2("%s is not a normal file--can't patch\n", filename);
i_size = filestat.st_size;
if (out_of_mem) {
set_hunkmax(); /* make sure dynamic arrays are allocated */
out_of_mem = FALSE;
return FALSE; /* force plan b because plan a bombed */
}
#ifdef lint
i_womp = Nullch;
#else
/*
* Lint says this may alloc less than i_size,
* but that's okay, I think.
*/
i_womp = malloc((MEM)(i_size + 2));
#endif
if (i_womp == Nullch)
return FALSE;
if ((ifd = open(filename, 0)) < 0)
pfatal2("can't open file %s", filename);
#ifndef lint
if (read(ifd, i_womp, (int)i_size) != i_size) {
/*
* This probably means i_size > 15 or 16 bits worth. At
* this point it doesn't matter if i_womp was undersized.
*/
Close(ifd);
free(i_womp);
return FALSE;
}
#endif
Close(ifd);
if (i_size && i_womp[i_size - 1] != '\n')
i_womp[i_size++] = '\n';
i_womp[i_size] = '\0';
/*
* Count the lines in the buffer so we know how many pointers we
* need.
*/
iline = 0;
for (s = i_womp; *s; s++) {
if (*s == '\n')
iline++;
}
#ifdef lint
i_ptr = Null(char**);
#else
i_ptr = (char **)malloc((MEM)((iline + 2) * sizeof(char *)));
#endif
if (i_ptr == Null(char **)) { /* shucks, it was a near thing */
free((char *)i_womp);
return FALSE;
}
/* Now scan the buffer and build pointer array. */
iline = 1;
i_ptr[iline] = i_womp;
for (s = i_womp; *s; s++) {
if (*s == '\n') {
/* These are NOT null terminated. */
i_ptr[++iline] = s + 1;
}
}
input_lines = iline - 1;
/* Now check for revision, if any. */
if (revision != Nullch) {
if (!rev_in_string(i_womp)) {
if (force) {
if (verbose)
say2(
"Warning: this file doesn't appear to be the %s version--patching anyway.\n",
revision);
} else if (batch) {
fatal2(
"this file doesn't appear to be the %s version--aborting.\n", revision);
} else {
(void) ask2(
"This file doesn't appear to be the %s version--patch anyway? [n] ",
revision);
if (*buf != 'y')
fatal1("aborted\n");
}
} else if (verbose)
say2("Good. This file appears to be the %s version.\n",
revision);
}
return TRUE; /* Plan a will work. */
}
/*
* Keep (virtually) nothing in memory.
*/
void
plan_b(char *filename)
{
Reg3 FILE *ifp;
Reg1 int i = 0;
Reg2 int maxlen = 1;
Reg4 bool found_revision = (revision == Nullch);
using_plan_a = FALSE;
if ((ifp = fopen(filename, "r")) == Nullfp)
pfatal2("can't open file %s", filename);
if ((tifd = creat(TMPINNAME, 0666)) < 0)
pfatal2("can't open file %s", TMPINNAME);
while (fgets(buf, sizeof buf, ifp) != Nullch) {
if (revision != Nullch && !found_revision && rev_in_string(buf))
found_revision = TRUE;
if ((i = strlen(buf)) > maxlen)
maxlen = i; /* Find longest line. */
}
if (revision != Nullch) {
if (!found_revision) {
if (force) {
if (verbose)
say2(
"Warning: this file doesn't appear to be the %s version--patching anyway.\n",
revision);
} else if (batch) {
fatal2(
"this file doesn't appear to be the %s version--aborting.\n", revision);
} else {
(void) ask2(
"This file doesn't appear to be the %s version--patch anyway? [n] ",
revision);
if (*buf != 'y')
fatal1("aborted\n");
}
} else if (verbose)
say2("Good. This file appears to be the %s version.\n",
revision);
}
Fseek(ifp, 0L, 0); /* Rewind file. */
lines_per_buf = BUFFERSIZE / maxlen;
tireclen = maxlen;
tibuf[0] = malloc((MEM)(BUFFERSIZE + 1));
tibuf[1] = malloc((MEM)(BUFFERSIZE + 1));
if (tibuf[1] == Nullch)
fatal1("out of memory\n");
for (i = 1; ; i++) {
if (! (i % lines_per_buf)) /* New block. */
if (write(tifd, tibuf[0], BUFFERSIZE) < BUFFERSIZE)
pfatal1("can't write temp file");
if (fgets(tibuf[0] + maxlen * (i%lines_per_buf),
maxlen + 1, ifp) == Nullch) {
input_lines = i - 1;
if (i % lines_per_buf)
if (write(tifd, tibuf[0], BUFFERSIZE) <
BUFFERSIZE)
pfatal1("can't write temp file");
break;
}
}
Fclose(ifp);
Close(tifd);
if ((tifd = open(TMPINNAME, 0)) < 0) {
pfatal2("can't reopen file %s", TMPINNAME);
}
}
/*
* Fetch a line from the input file, \n terminated, not necessarily \0.
*/
char *
ifetch(line,whichbuf)
Reg1 LINENUM line;
int whichbuf; /* ignored when file in memory */
{
if (line < 1 || line > input_lines)
return "";
if (using_plan_a)
return i_ptr[line];
else {
LINENUM offline = line % lines_per_buf;
LINENUM baseline = line - offline;
if (tiline[0] == baseline)
whichbuf = 0;
else if (tiline[1] == baseline)
whichbuf = 1;
else {
tiline[whichbuf] = baseline;
#ifndef lint /* complains of long accuracy */
Lseek(tifd, (long)baseline / lines_per_buf * BUFFERSIZE, 0);
#endif
if (read(tifd, tibuf[whichbuf], BUFFERSIZE) < 0)
pfatal2("error reading tmp file %s", TMPINNAME);
}
return tibuf[whichbuf] + (tireclen * offline);
}
}
/*
* True if the string argument contains the revision number we want.
*/
bool
rev_in_string(char *string)
{
Reg1 char *s;
Reg2 int patlen;
if (revision == Nullch)
return TRUE;
patlen = strlen(revision);
if (strnEQ(string,revision,patlen) &&
isspace((unsigned char)string[patlen]))
return TRUE;
for (s = string; *s; s++) {
if (isspace((unsigned char)*s) &&
strnEQ(s + 1, revision, patlen) &&
isspace((unsigned char)s[patlen + 1] )) {
return TRUE;
}
}
return FALSE;
}