156 lines
3.6 KiB
C
156 lines
3.6 KiB
C
/* vi: set sw=4 ts=4: */
|
|
/*
|
|
* Mini comm implementation for busybox
|
|
*
|
|
* Copyright (C) 2005 by Robert Sullivan <cogito.ergo.cogito@gmail.com>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
|
|
* 02111-1307 USA
|
|
*
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include "busybox.h"
|
|
|
|
#define COMM_OPT_1 0x01
|
|
#define COMM_OPT_2 0x02
|
|
#define COMM_OPT_3 0x04
|
|
|
|
/* These three variables control behaviour if non-zero */
|
|
|
|
static int only_file_1;
|
|
static int only_file_2;
|
|
static int both;
|
|
|
|
/* writeline outputs the input given, appropriately aligned according to class */
|
|
static void writeline (char *line, int class) {
|
|
switch (class) {
|
|
case 1: if (!only_file_1)
|
|
return;
|
|
break;
|
|
case 2: if (!only_file_2)
|
|
return;
|
|
if (only_file_1)
|
|
putchar('\t');
|
|
break;
|
|
case 3: if (!both)
|
|
return;
|
|
if (only_file_1)
|
|
putchar('\t');
|
|
if (only_file_2)
|
|
putchar('\t');
|
|
break;
|
|
}
|
|
fputs(line, stdout);
|
|
}
|
|
|
|
/* This is the real core of the program - lines are compared here */
|
|
static int cmp_files(char **infiles) {
|
|
|
|
char thisline[2][100];
|
|
FILE *streams[2];
|
|
int i = 0;
|
|
|
|
for (i = 0; i < 2; i++) {
|
|
streams[i] = (strcmp(infiles[i], "=") == 0 ? stdin : fopen(infiles[i], "r"));
|
|
fgets(thisline[i], 100, streams[i]);
|
|
}
|
|
|
|
while (thisline[0] || thisline[1]) {
|
|
|
|
int order = 0;
|
|
int tl0_len = strlen(thisline[0]);
|
|
int tl1_len = strlen(thisline[1]);
|
|
if (!thisline[0])
|
|
order = 1;
|
|
else if (!thisline[1])
|
|
order = -1;
|
|
else {
|
|
order = memcmp(thisline[0], thisline[1], tl0_len < tl1_len ? tl0_len : tl1_len);
|
|
if (!order)
|
|
order = tl0_len < tl1_len ? -1 : tl0_len != tl1_len;
|
|
}
|
|
|
|
if ((order == 0) && (!feof(streams[0])) && (!feof(streams[1])))
|
|
writeline(thisline[1], 3);
|
|
else if ((order > 0) && (!feof(streams[1])))
|
|
writeline(thisline[1], 2);
|
|
else if ((order < 0) && (!feof(streams[0])))
|
|
writeline(thisline[0], 1);
|
|
|
|
if (feof(streams[0]) && feof(streams[1])) {
|
|
fclose(streams[0]);
|
|
fclose(streams[1]);
|
|
break;
|
|
}
|
|
else if (feof(streams[0])) {
|
|
|
|
while (!feof(streams[1])) {
|
|
if (order < 0)
|
|
writeline(thisline[1], 2);
|
|
fgets(thisline[1], 100, streams[1]);
|
|
}
|
|
fclose(streams[0]);
|
|
fclose(streams[1]);
|
|
break;
|
|
}
|
|
else if (feof(streams[1])) {
|
|
|
|
while (!feof(streams[0])) {
|
|
if (order > 0)
|
|
writeline(thisline[0], 1);
|
|
fgets(thisline[0], 100, streams[0]);
|
|
}
|
|
fclose(streams[0]);
|
|
fclose(streams[1]);
|
|
break;
|
|
}
|
|
else {
|
|
if (order >= 0)
|
|
fgets(thisline[1], 100, streams[1]);
|
|
if (order <= 0)
|
|
fgets(thisline[0], 100, streams[0]);
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int comm_main (int argc, char **argv) {
|
|
|
|
unsigned long opt;
|
|
only_file_1 = 1;
|
|
only_file_2 = 1;
|
|
both = 1;
|
|
|
|
opt = bb_getopt_ulflags(argc, argv, "123");
|
|
|
|
if ((opt & 0x80000000UL) || (optind == argc)) {
|
|
bb_show_usage();
|
|
}
|
|
|
|
if (opt & COMM_OPT_1)
|
|
only_file_1 = 0;
|
|
if (opt & COMM_OPT_2)
|
|
only_file_2 = 0;
|
|
if (opt & COMM_OPT_3)
|
|
both = 0;
|
|
|
|
exit(cmp_files(argv + optind) == 0 ? EXIT_SUCCESS : EXIT_FAILURE);
|
|
}
|