#include <sys/types.h>
#include <sys/capsicum.h>
#include <sys/procdesc.h>
#include <sys/wait.h>
#include <assert.h>
#include <capsicum_helpers.h>
#include <ctype.h>
#include <err.h>
#include <fcntl.h>
#include <getopt.h>
#include <inttypes.h>
#include <limits.h>
#include <spawn.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
extern char **environ;
struct range {
int from;
int to;
};
enum difftype {
DIFF_NONE,
DIFF_TYPE1,
DIFF_TYPE2,
DIFF_TYPE3,
};
struct diff {
enum difftype type;
struct range old;
struct range new;
};
#define EFLAG_NONE 0
#define EFLAG_OVERLAP 1
#define EFLAG_NOOVERLAP 2
#define EFLAG_UNMERGED 3
static size_t szchanges;
static struct diff *d13;
static struct diff *d23;
static struct diff *de;
static char *overlap;
static int *de_delta;
static int overlapcnt;
static FILE *fp[3];
static int cline[3];
static int last[4];
static int Aflag, eflag, iflag, mflag, Tflag;
static int oflag;
static int strip_cr;
static char *f1mark, *f2mark, *f3mark;
static const char *oldmark = "<<<<<<<";
static const char *orgmark = "|||||||";
static const char *newmark = ">>>>>>>";
static const char *divider = "=======";
static bool duplicate(struct range *, struct range *);
static int edit(struct diff *, bool, int, enum difftype);
static char *getchange(FILE *);
static char *get_line(FILE *, size_t *);
static int readin(int fd, struct diff **);
static int skip(int, int, const char *);
static void change(int, struct range *, bool);
static void keep(int, struct range *);
static void merge(int, int);
static void prange(struct range *, bool);
static void repos(int);
static void separate(const char *);
static void edscript(int) __dead2;
static void Ascript(int) __dead2;
static void mergescript(int, int) __dead2;
static void increase(void);
static void usage(void);
static void printrange(FILE *, struct range *);
static const char diff3_version[] = "FreeBSD diff3 20260213";
enum {
DIFFPROG_OPT,
STRIPCR_OPT,
HELP_OPT,
VERSION_OPT
};
#define DIFF_PATH "/usr/bin/diff"
#define OPTIONS "3aAeEiL:mTxX"
static struct option longopts[] = {
{ "ed", no_argument, NULL, 'e' },
{ "show-overlap", no_argument, NULL, 'E' },
{ "overlap-only", no_argument, NULL, 'x' },
{ "initial-tab", no_argument, NULL, 'T' },
{ "text", no_argument, NULL, 'a' },
{ "strip-trailing-cr", no_argument, NULL, STRIPCR_OPT },
{ "show-all", no_argument, NULL, 'A' },
{ "easy-only", no_argument, NULL, '3' },
{ "merge", no_argument, NULL, 'm' },
{ "label", required_argument, NULL, 'L' },
{ "diff-program", required_argument, NULL, DIFFPROG_OPT },
{ "help", no_argument, NULL, HELP_OPT},
{ "version", no_argument, NULL, VERSION_OPT}
};
static void
usage(void)
{
fprintf(stderr, "usage: diff3 [-3aAeEimTxX] [-L label1] [-L label2] "
"[-L label3] file1 file2 file3\n");
}
static int
strtoi(char *str, char **end)
{
intmax_t num;
errno = 0;
num = strtoimax(str, end, 10);
if ((end != NULL && *end == str) ||
num < 0 || num > INT_MAX ||
errno == EINVAL || errno == ERANGE)
err(2, "error in diff output");
return (int)num;
}
static int
readin(int fd, struct diff **dd)
{
int a, b, c, d;
int i;
char kind, *p;
FILE *f;
f = fdopen(fd, "r");
if (f == NULL)
err(2, "fdopen");
for (i = 0; (p = getchange(f)) != NULL; i++) {
if ((size_t)i >= szchanges - 1)
increase();
a = b = strtoi(p, &p);
if (*p == ',')
b = strtoi(p + 1, &p);
kind = *p++;
c = d = strtoi(p, &p);
if (*p == ',')
d = strtoi(p + 1, &p);
if (*p != '\n')
errx(2, "error in diff output");
if (kind == 'a')
a++;
else if (kind == 'c')
;
else if (kind == 'd')
c++;
else
errx(2, "error in diff output");
b++;
d++;
if (b < a || d < c)
errx(2, "error in diff output");
(*dd)[i].old.from = a;
(*dd)[i].old.to = b;
(*dd)[i].new.from = c;
(*dd)[i].new.to = d;
if (i > 0) {
if ((*dd)[i].old.from < (*dd)[i - 1].old.to ||
(*dd)[i].new.from < (*dd)[i - 1].new.to)
errx(2, "diff output out of order");
}
}
if (i > 0) {
(*dd)[i].old.from = (*dd)[i].old.to = (*dd)[i - 1].old.to;
(*dd)[i].new.from = (*dd)[i].new.to = (*dd)[i - 1].new.to;
}
fclose(f);
return (i);
}
static int
diffexec(char **diffargv, int fd[])
{
posix_spawnattr_t sa;
posix_spawn_file_actions_t fa;
pid_t pid;
int pd, error;
if ((error = posix_spawnattr_init(&sa)) != 0)
errc(2, error, "posix_spawnattr_init");
if ((error = posix_spawn_file_actions_init(&fa)) != 0)
errc(2, error, "posix_spawn_file_actions_init");
posix_spawnattr_setprocdescp_np(&sa, &pd, 0);
posix_spawn_file_actions_adddup2(&fa, fd[1], STDOUT_FILENO);
error = posix_spawn(&pid, diffargv[0], &fa, &sa, diffargv, environ);
if (error != 0)
errc(2, error, "could not spawn diff");
posix_spawn_file_actions_destroy(&fa);
posix_spawnattr_destroy(&sa);
close(fd[1]);
return (pd);
}
static char *
getchange(FILE *b)
{
char *line;
while ((line = get_line(b, NULL)) != NULL) {
if (isdigit((unsigned char)line[0]))
return (line);
}
return (NULL);
}
static char *
get_line(FILE *b, size_t *n)
{
ssize_t len;
static char *buf = NULL;
static size_t bufsize = 0;
if ((len = getline(&buf, &bufsize, b)) < 0)
return (NULL);
if (strip_cr && len >= 2 && strcmp("\r\n", &(buf[len - 2])) == 0) {
buf[len - 2] = '\n';
buf[len - 1] = '\0';
len--;
}
if (n != NULL)
*n = len;
return (buf);
}
static void
merge(int m1, int m2)
{
struct diff *d1, *d2, *d3;
int j, t1, t2;
int f1f3delta;
bool dup = false;
d1 = d13;
d2 = d23;
j = 0;
f1f3delta = 0;
for (;;) {
t1 = (d1 < d13 + m1);
t2 = (d2 < d23 + m2);
if (!t1 && !t2)
break;
if (!t2 || (t1 && d1->new.to < d2->new.from)) {
if (eflag == EFLAG_NONE) {
separate("1");
change(1, &d1->old, false);
keep(2, &d1->new);
change(3, &d1->new, false);
} else if (mflag) {
j++;
de[j].type = DIFF_TYPE1;
de[j].old = d1->old;
de[j].new = d1->new;
overlap[j] = 0;
} else if (eflag == EFLAG_OVERLAP) {
j = edit(d2, dup, j, DIFF_TYPE1);
}
f1f3delta += (d1->old.to - d1->old.from) -
(d1->new.to - d1->new.from);
d1++;
continue;
}
if (!t1 || (t2 && d2->new.to < d1->new.from)) {
if (eflag == EFLAG_NONE) {
separate("2");
keep(1, &d2->new);
change(3, &d2->new, false);
change(2, &d2->old, false);
} else if (Aflag || mflag) {
if (eflag == EFLAG_UNMERGED) {
j = edit(d2, dup, j, DIFF_TYPE2);
de_delta[j] = f1f3delta;
}
}
d2++;
continue;
}
if (d1 + 1 < d13 + m1 && d1->new.to >= d1[1].new.from) {
d1[1].old.from = d1->old.from;
d1[1].new.from = d1->new.from;
d1++;
continue;
}
if (d2 + 1 < d23 + m2 && d2->new.to >= d2[1].new.from) {
d2[1].old.from = d2->old.from;
d2[1].new.from = d2->new.from;
d2++;
continue;
}
if (d1->new.from == d2->new.from && d1->new.to == d2->new.to) {
dup = duplicate(&d1->old, &d2->old);
if (eflag == EFLAG_NONE) {
separate(dup ? "3" : "");
change(1, &d1->old, dup);
change(2, &d2->old, false);
d3 = d1->old.to > d1->old.from ? d1 : d2;
change(3, &d3->new, false);
} else if (mflag) {
j++;
de[j].type = DIFF_TYPE3;
de[j].old = d1->old;
de[j].new = d1->new;
overlap[j] = !dup;
if (!dup)
overlapcnt++;
} else {
j = edit(d1, dup, j, DIFF_TYPE3);
}
dup = false;
f1f3delta += (d1->old.to - d1->old.from) -
(d1->new.to - d1->new.from);
d1++;
d2++;
continue;
}
if (d1->new.from < d2->new.from) {
d2->old.from -= d2->new.from - d1->new.from;
d2->new.from = d1->new.from;
} else if (d2->new.from < d1->new.from) {
d1->old.from -= d1->new.from - d2->new.from;
d1->new.from = d2->new.from;
}
if (d1->new.to > d2->new.to) {
d2->old.to += d1->new.to - d2->new.to;
d2->new.to = d1->new.to;
} else if (d2->new.to > d1->new.to) {
d1->old.to += d2->new.to - d1->new.to;
d1->new.to = d2->new.to;
}
}
if (mflag)
mergescript(j, f1f3delta);
else if (Aflag)
Ascript(j);
else if (eflag)
edscript(j);
}
static void
separate(const char *s)
{
printf("====%s\n", s);
}
static void
change(int i, struct range *rold, bool dup)
{
printf("%d:", i);
last[i] = rold->to;
prange(rold, false);
if (dup)
return;
i--;
skip(i, rold->from, NULL);
skip(i, rold->to, " ");
}
static void
prange(struct range *rold, bool delete)
{
if (rold->to <= rold->from)
printf("%da\n", rold->from - 1);
else {
printf("%d", rold->from);
if (rold->to > rold->from + 1)
printf(",%d", rold->to - 1);
if (delete)
printf("d\n");
else
printf("c\n");
}
}
static void
keep(int i, struct range *rnew)
{
int delta;
struct range trange;
delta = last[3] - last[i];
trange.from = rnew->from - delta;
trange.to = rnew->to - delta;
change(i, &trange, true);
}
static int
skip(int i, int from, const char *pr)
{
size_t j, n;
char *line;
for (n = 0; cline[i] < from - 1; n += j) {
if ((line = get_line(fp[i], &j)) == NULL)
errx(2, "logic error");
if (pr != NULL)
printf("%s%s", Tflag == 1 ? "\t" : pr, line);
cline[i]++;
}
return ((int) n);
}
static bool
duplicate(struct range *r1, struct range *r2)
{
int c, d;
int nchar;
int nline;
if (r1->to-r1->from != r2->to-r2->from)
return (0);
skip(0, r1->from, NULL);
skip(1, r2->from, NULL);
nchar = 0;
for (nline = 0; nline < r1->to - r1->from; nline++) {
do {
c = getc(fp[0]);
d = getc(fp[1]);
if (c == -1 && d == -1)
break;
if (c == -1 || d == -1)
errx(2, "logic error");
nchar++;
if (c != d) {
repos(nchar);
return (0);
}
} while (c != '\n');
}
repos(nchar);
return (1);
}
static void
repos(int nchar)
{
int i;
for (i = 0; i < 2; i++)
(void)fseek(fp[i], (long)-nchar, SEEK_CUR);
}
static int
edit(struct diff *diff, bool dup, int j, enum difftype difftype)
{
if (!(eflag == EFLAG_UNMERGED ||
(!dup && eflag == EFLAG_OVERLAP ) ||
(dup && eflag == EFLAG_NOOVERLAP))) {
return (j);
}
j++;
overlap[j] = !dup;
if (!dup)
overlapcnt++;
de[j].type = difftype;
de[j].old.from = diff->old.from;
de[j].old.to = diff->old.to;
de[j].new.from = diff->new.from;
de[j].new.to = diff->new.to;
return (j);
}
static void
printrange(FILE *p, struct range *r)
{
char *line = NULL;
size_t len = 0;
int i = 1;
if (r->from == r->to)
return;
if (r->from > r->to)
errx(2, "invalid print range");
fseek(p, 0L, SEEK_SET);
while (getline(&line, &len, p) > 0) {
if (i >= r->from)
printf("%s", line);
if (++i > r->to - 1)
break;
}
free(line);
}
static void
edscript(int n)
{
bool delete;
struct range *new, *old;
for (; n > 0; n--) {
new = &de[n].new;
old = &de[n].old;
delete = (new->from == new->to);
if (de[n].type == DIFF_TYPE1) {
if (delete)
printf("%dd\n", new->from - 1);
else if (old->from == new->from && old->to == new->to) {
printf("%dc\n", old->from);
printrange(fp[2], old);
printf(".\n");
}
continue;
} else {
if (!oflag || !overlap[n]) {
prange(old, delete);
} else {
printf("%da\n", old->to - 1);
printf("%s\n", divider);
}
printrange(fp[2], new);
if (!oflag || !overlap[n]) {
if (!delete)
printf(".\n");
} else {
printf("%s %s\n.\n", newmark, f3mark);
printf("%da\n%s %s\n.\n", old->from - 1,
oldmark, f1mark);
}
}
}
if (iflag)
printf("w\nq\n");
exit(oflag ? overlapcnt > 0 : 0);
}
static void
Ascript(int n)
{
int startmark;
bool deletenew;
bool deleteold;
struct range *new, *old;
for (; n > 0; n--) {
new = &de[n].new;
old = &de[n].old;
deletenew = (new->from == new->to);
deleteold = (old->from == old->to);
if (de[n].type == DIFF_TYPE2) {
if (!oflag || !overlap[n]) {
prange(old, deletenew);
printrange(fp[2], new);
} else {
startmark = new->to - 1 + de_delta[n];
printf("%da\n", startmark);
printf("%s %s\n", newmark, f3mark);
printf(".\n");
printf("%da\n", startmark -
(new->to - new->from));
printf("%s %s\n", oldmark, f2mark);
if (!deleteold)
printrange(fp[1], old);
printf("%s\n.\n", divider);
}
} else if (de[n].type == DIFF_TYPE3) {
startmark = old->to - 1;
if (!oflag || !overlap[n]) {
prange(old, deletenew);
printrange(fp[2], new);
} else {
printf("%da\n", startmark);
printf("%s %s\n", orgmark, f2mark);
if (deleteold) {
struct range r;
r.from = old->from-1;
r.to = new->to;
printrange(fp[1], &r);
} else
printrange(fp[1], old);
printf("%s\n", divider);
printrange(fp[2], new);
}
if (!oflag || !overlap[n]) {
if (!deletenew)
printf(".\n");
} else {
printf("%s %s\n.\n", newmark, f3mark);
printf("%da\n%s %s\n.\n",
startmark - (old->to - old->from),
oldmark, f1mark);
}
}
}
if (iflag)
printf("w\nq\n");
exit(overlapcnt > 0);
}
static void
mergescript(int i, int f1f3delta)
{
struct range r, *new, *old;
int n;
r.from = 1;
r.to = 1;
for (n = 1; n <= i; n++) {
new = &de[n].new;
old = &de[n].old;
if (de[n].type == DIFF_TYPE1)
r.to = old->to;
else if (de[n].type == DIFF_TYPE2)
r.to = new->from + de_delta[n];
else
r.to = old->from;
printrange(fp[0], &r);
switch (de[n].type) {
case DIFF_TYPE1:
break;
case DIFF_TYPE2:
printf("%s %s\n", oldmark, f2mark);
printrange(fp[1], old);
printf("%s\n", divider);
printrange(fp[2], new);
printf("%s %s\n", newmark, f3mark);
break;
case DIFF_TYPE3:
if (!oflag || !overlap[n]) {
printrange(fp[2], new);
} else {
printf("%s %s\n", oldmark, f1mark);
printrange(fp[0], old);
if (eflag != EFLAG_OVERLAP) {
printf("%s %s\n", orgmark, f2mark);
if (old->from == old->to) {
struct range or;
or.from = old->from - 1;
or.to = new->to;
printrange(fp[1], &or);
} else {
printrange(fp[1], old);
}
}
printf("%s\n", divider);
printrange(fp[2], new);
printf("%s %s\n", newmark, f3mark);
}
break;
default:
__assert_unreachable();
}
if (de[n].type == DIFF_TYPE2)
r.from = new->to + de_delta[n];
else
r.from = old->to;
}
r.from -= f1f3delta;
r.to = INT_MAX;
printrange(fp[2], &r);
exit(overlapcnt > 0);
}
static void
increase(void)
{
struct diff *p;
char *q;
int *s;
size_t newsz, incr;
newsz = szchanges == 0 ? 64 : 2 * szchanges;
incr = newsz - szchanges;
p = reallocarray(d13, newsz, sizeof(*p));
if (p == NULL)
err(2, NULL);
memset(p + szchanges, 0, incr * sizeof(*p));
d13 = p;
p = reallocarray(d23, newsz, sizeof(*p));
if (p == NULL)
err(2, NULL);
memset(p + szchanges, 0, incr * sizeof(*p));
d23 = p;
p = reallocarray(de, newsz, sizeof(*p));
if (p == NULL)
err(2, NULL);
memset(p + szchanges, 0, incr * sizeof(*p));
de = p;
q = reallocarray(overlap, newsz, 1);
if (q == NULL)
err(2, NULL);
memset(q + szchanges, 0, incr * 1);
overlap = q;
s = reallocarray(de_delta, newsz, sizeof(*s));
if (s == NULL)
err(2, NULL);
memset(s + szchanges, 0, incr * sizeof(*s));
de_delta = s;
szchanges = newsz;
}
static void
wait_and_check(int pd)
{
int status;
while (pdwait(pd, &status, WEXITED, NULL, NULL) == -1) {
if (errno != EINTR)
err(2, "pdwait");
}
close(pd);
if (WIFEXITED(status) && WEXITSTATUS(status) >= 2)
errx(2, "diff exited abnormally");
if (WIFSIGNALED(status))
errx(2, "diff killed by signal %d", WTERMSIG(status));
}
int
main(int argc, char **argv)
{
int ch, nblabels, m, n;
char *labels[] = { NULL, NULL, NULL };
const char *diffprog = DIFF_PATH;
char *file1, *file2, *file3;
char *diffargv[7];
int diffargc = 0;
int fd13[2], fd23[2];
int pd13, pd23;
cap_rights_t rights_ro;
nblabels = 0;
eflag = EFLAG_NONE;
oflag = 0;
diffargv[diffargc++] = __DECONST(char *, diffprog);
while ((ch = getopt_long(argc, argv, OPTIONS, longopts, NULL)) != -1) {
switch (ch) {
case '3':
eflag = EFLAG_NOOVERLAP;
break;
case 'a':
diffargv[diffargc++] = __DECONST(char *, "-a");
break;
case 'A':
Aflag = 1;
break;
case 'e':
eflag = EFLAG_UNMERGED;
break;
case 'E':
eflag = EFLAG_OVERLAP;
oflag = 1;
break;
case 'i':
iflag = 1;
break;
case 'L':
oflag = 1;
if (nblabels >= 3)
errx(2, "too many file label options");
labels[nblabels++] = optarg;
break;
case 'm':
Aflag = 1;
oflag = 1;
mflag = 1;
break;
case 'T':
Tflag = 1;
break;
case 'x':
eflag = EFLAG_OVERLAP;
break;
case 'X':
oflag = 1;
eflag = EFLAG_OVERLAP;
break;
case DIFFPROG_OPT:
diffargv[0] = optarg;
break;
case STRIPCR_OPT:
strip_cr = 1;
diffargv[diffargc++] = __DECONST(char *, "--strip-trailing-cr");
break;
case HELP_OPT:
usage();
exit(0);
case VERSION_OPT:
printf("%s\n", diff3_version);
exit(0);
}
}
argc -= optind;
argv += optind;
if (Aflag) {
if (eflag == EFLAG_NONE)
eflag = EFLAG_UNMERGED;
oflag = 1;
}
if (argc != 3) {
usage();
exit(2);
}
if (caph_limit_stdio() == -1)
err(2, "unable to limit stdio");
cap_rights_init(&rights_ro, CAP_READ, CAP_FSTAT, CAP_SEEK);
file1 = argv[0];
file2 = argv[1];
file3 = argv[2];
if (oflag) {
asprintf(&f1mark, "%s",
labels[0] != NULL ? labels[0] : file1);
if (f1mark == NULL)
err(2, "asprintf");
asprintf(&f2mark, "%s",
labels[1] != NULL ? labels[1] : file2);
if (f2mark == NULL)
err(2, "asprintf");
asprintf(&f3mark, "%s",
labels[2] != NULL ? labels[2] : file3);
if (f3mark == NULL)
err(2, "asprintf");
}
fp[0] = fopen(file1, "r");
if (fp[0] == NULL)
err(2, "Can't open %s", file1);
if (caph_rights_limit(fileno(fp[0]), &rights_ro) < 0)
err(2, "unable to limit rights on: %s", file1);
fp[1] = fopen(file2, "r");
if (fp[1] == NULL)
err(2, "Can't open %s", file2);
if (caph_rights_limit(fileno(fp[1]), &rights_ro) < 0)
err(2, "unable to limit rights on: %s", file2);
fp[2] = fopen(file3, "r");
if (fp[2] == NULL)
err(2, "Can't open %s", file3);
if (caph_rights_limit(fileno(fp[2]), &rights_ro) < 0)
err(2, "unable to limit rights on: %s", file3);
if (pipe2(fd13, O_CLOEXEC))
err(2, "pipe");
if (pipe2(fd23, O_CLOEXEC))
err(2, "pipe");
diffargv[diffargc] = file1;
diffargv[diffargc + 1] = file3;
diffargv[diffargc + 2] = NULL;
pd13 = diffexec(diffargv, fd13);
diffargv[diffargc] = file2;
pd23 = diffexec(diffargv, fd23);
caph_cache_catpages();
if (caph_enter() < 0)
err(2, "unable to enter capability mode");
increase();
m = readin(fd13[0], &d13);
n = readin(fd23[0], &d23);
wait_and_check(pd13);
wait_and_check(pd23);
merge(m, n);
exit(0);
}