comm: almost total rewrite; add testsuite

function                                             old     new   delta
writeline                                            128     125      -3
comm_main                                            540     239    -301
------------------------------------------------------------------------------
(add/remove: 0/0 grow/shrink: 0/2 up/down: 0/-304)           Total: -304 bytes
This commit is contained in:
Denis Vlasenko 2008-07-15 06:45:14 +00:00
parent 51f1b6c0e0
commit bb13079c8e
3 changed files with 61 additions and 56 deletions

View File

@ -14,8 +14,9 @@
#define COMM_OPT_3 (1 << 2)
/* writeline outputs the input given, appropriately aligned according to class */
static void writeline(char *line, int class, int flags)
static void writeline(char *line, int class)
{
int flags = option_mask32;
if (class == 0) {
if (flags & COMM_OPT_1)
return;
@ -32,81 +33,67 @@ static void writeline(char *line, int class, int flags)
if (!(flags & COMM_OPT_2))
putchar('\t');
}
fputs(line, stdout);
puts(line);
}
int comm_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int comm_main(int argc UNUSED_PARAM, char **argv)
{
#define LINE_LEN 100
#define BB_EOF_0 0x1
#define BB_EOF_1 0x2
char thisline[2][LINE_LEN];
FILE *streams[2];
char *thisline[2];
FILE *stream[2];
int i;
unsigned flags;
int order;
opt_complementary = "=2";
flags = getopt32(argv, "123");
getopt32(argv, "123");
argv += optind;
for (i = 0; i < 2; ++i) {
streams[i] = (argv[i][0] == '-' && !argv[i][1]) ? stdin : xfopen(argv[i], "r");
fgets(thisline[i], LINE_LEN, streams[i]);
stream[i] = xfopen_stdin(argv[i]);
}
/* This is the real core of the program - lines are compared here */
while (*thisline[0] || *thisline[1]) {
int order = 0;
i = 0;
if (feof(streams[0])) i |= BB_EOF_0;
if (feof(streams[1])) i |= BB_EOF_1;
if (!*thisline[0])
order = 1;
else if (!*thisline[1])
order = -1;
else {
int tl0_len, tl1_len;
tl0_len = strlen(thisline[0]);
tl1_len = strlen(thisline[1]);
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;
order = 0;
thisline[1] = thisline[0] = NULL;
while (1) {
if (order <= 0) {
free(thisline[0]);
thisline[0] = xmalloc_fgetline(stream[0]);
}
if (order >= 0) {
free(thisline[1]);
thisline[1] = xmalloc_fgetline(stream[1]);
}
if (order == 0 && !i)
writeline(thisline[1], 2, flags);
else if (order > 0 && !(i & BB_EOF_1))
writeline(thisline[1], 1, flags);
else if (order < 0 && !(i & BB_EOF_0))
writeline(thisline[0], 0, flags);
if (i & BB_EOF_0 & BB_EOF_1) {
i = !thisline[0] + (!thisline[1] << 1);
if (i)
break;
order = strcmp(thisline[0], thisline[1]);
} else if (i) {
i = (i & BB_EOF_0 ? 1 : 0);
while (!feof(streams[i])) {
if ((order < 0 && i) || (order > 0 && !i))
writeline(thisline[i], i, flags);
fgets(thisline[i], LINE_LEN, streams[i]);
}
break;
if (order >= 0)
writeline(thisline[1], order ? 1 : 2);
else
writeline(thisline[0], 0);
}
} else {
if (order >= 0)
fgets(thisline[1], LINE_LEN, streams[1]);
if (order <= 0)
fgets(thisline[0], LINE_LEN, streams[0]);
/* EOF at least on one of the streams */
i &= 1;
if (thisline[i]) {
/* stream[i] is not at EOF yet */
/* we did not print thisline[i] yet */
char *p = thisline[i];
writeline(p, i);
while (1) {
free(p);
p = xmalloc_fgetline(stream[i]);
if (!p)
break;
writeline(p, i);
}
}
if (ENABLE_FEATURE_CLEAN_UP) {
fclose(streams[0]);
fclose(streams[1]);
fclose(stream[0]);
fclose(stream[1]);
}
return EXIT_SUCCESS;

View File

@ -1,12 +1,11 @@
#!/bin/sh
# awk tests.
# Copyright 2007 by Denys Vlasenko <vda.linux@googlemail.com>
# Licensed under GPL v2, see file LICENSE for details.
. testing.sh
# testing "description" "arguments" "result" "infile" "stdin"
# testing "description" "command" "result" "infile" "stdin"
testing "awk -F case 0" "awk -F '[#]' '{ print NF }'" "" "" ""
testing "awk -F case 1" "awk -F '[#]' '{ print NF }'" "0\n" "" "\n"

19
testsuite/comm.tests Executable file
View File

@ -0,0 +1,19 @@
#!/bin/sh
# Copyright 2008 by Denys Vlasenko <vda.linux@googlemail.com>
# Licensed under GPL v2, see file LICENSE for details.
. testing.sh
# testing "description" "command" "result" "infile" "stdin"
testing "comm test 1" "comm input -" "\t123\n""456\n""abc\n""\tdef\n" "456\nabc\n" "123\ndef\n"
testing "comm test 2" "comm - input" "123\n""\t456\n""\tabc\n""def\n" "456\nabc\n" "123\ndef\n"
testing "comm test 3" "comm input -" "abc\n""\tdef\n""xyz\n" "abc\nxyz\n" "def\n"
testing "comm test 4" "comm - input" "\tabc\n""def\n""\txyz\n" "abc\nxyz\n" "def\n"
testing "comm test 5" "comm input -" "123\n""abc\n""\tdef\n" "123\nabc\n" "def\n"
testing "comm test 6" "comm - input" "\t123\n""\tabc\n""def\n" "123\nabc\n" "def\n"
testing "comm unterminated line 1" "comm input -" "abc\n""\tdef\n" "abc" "def"
testing "comm unterminated line 2" "comm - input" "\tabc\n""def\n" "abc" "def"
exit $FAILCOUNT