home *** CD-ROM | disk | FTP | other *** search
/ PC-Online 1996 May / PCOnline_05_1996.bin / linux / source / a / txtutils / textutil.9 / textutil / textutils-1.9 / src / comm.c < prev    next >
Encoding:
C/C++ Source or Header  |  1993-10-23  |  6.4 KB  |  277 lines
  1. /* comm -- compare two sorted files line by line.
  2.    Copyright (C) 1986, 1990, 1991 Free Software Foundation, Inc.
  3.  
  4.    This program is free software; you can redistribute it and/or modify
  5.    it under the terms of the GNU General Public License as published by
  6.    the Free Software Foundation; either version 2, or (at your option)
  7.    any later version.
  8.  
  9.    This program is distributed in the hope that it will be useful,
  10.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.    GNU General Public License for more details.
  13.  
  14.    You should have received a copy of the GNU General Public License
  15.    along with this program; if not, write to the Free Software
  16.    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
  17.  
  18. /* Written by Richard Stallman and David MacKenzie. */
  19.  
  20. #ifdef HAVE_CONFIG_H
  21. #if defined (CONFIG_BROKETS)
  22. /* We use <config.h> instead of "config.h" so that a compilation
  23.    using -I. -I$srcdir will use ./config.h rather than $srcdir/config.h
  24.    (which it would do because it found this file in $srcdir).  */
  25. #include <config.h>
  26. #else
  27. #include "config.h"
  28. #endif
  29. #endif
  30.  
  31. #include <stdio.h>
  32. #include <getopt.h>
  33. #include <sys/types.h>
  34. #include "system.h"
  35. #include "linebuffer.h"
  36. #include "version.h"
  37.  
  38. #define min(x, y) ((x) < (y) ? (x) : (y))
  39.  
  40. /* The name this program was run with. */
  41. char *program_name;
  42.  
  43. /* If nonzero, print lines that are found only in file 1. */
  44. static int only_file_1;
  45.  
  46. /* If nonzero, print lines that are found only in file 2. */
  47. static int only_file_2;
  48.  
  49. /* If nonzero, print lines that are found in both files. */
  50. static int both;
  51.  
  52. /* If non-zero, display usage information and exit.  */
  53. static int show_help;
  54.  
  55. /* If non-zero, print the version on standard output then exit.  */
  56. static int show_version;
  57.  
  58. static struct option const long_options[] =
  59. {
  60.   {"help", no_argument, &show_help, 1},
  61.   {"version", no_argument, &show_version, 1},
  62.   {0, 0, 0, 0}
  63. };
  64.  
  65. void error ();
  66. static int compare_files ();
  67. static void writeline ();
  68. static void usage ();
  69.  
  70. void
  71. main (argc, argv)
  72.      int argc;
  73.      char *argv[];
  74. {
  75.   int c;
  76.  
  77.   program_name = argv[0];
  78.  
  79.   only_file_1 = 1;
  80.   only_file_2 = 1;
  81.   both = 1;
  82.  
  83.   while ((c = getopt_long (argc, argv, "123", long_options, (int *) 0)) != EOF)
  84.     switch (c)
  85.       {
  86.       case 0:
  87.     break;
  88.  
  89.       case '1':
  90.     only_file_1 = 0;
  91.     break;
  92.  
  93.       case '2':
  94.     only_file_2 = 0;
  95.     break;
  96.  
  97.       case '3':
  98.     both = 0;
  99.     break;
  100.  
  101.       default:
  102.     usage (1);
  103.       }
  104.  
  105.   if (show_version)
  106.     {
  107.       printf ("%s\n", version_string);
  108.       exit (0);
  109.     }
  110.  
  111.   if (show_help)
  112.     usage (0);
  113.  
  114.   if (optind + 2 != argc)
  115.     usage (1);
  116.  
  117.   exit (compare_files (argv + optind));
  118. }
  119.  
  120. /* Compare INFILES[0] and INFILES[1].
  121.    If either is "-", use the standard input for that file.
  122.    Assume that each input file is sorted;
  123.    merge them and output the result.
  124.    Return 0 if successful, 1 if any errors occur. */
  125.  
  126. static int
  127. compare_files (infiles)
  128.      char **infiles;
  129. {
  130.   /* For each file, we have one linebuffer in lb1.  */
  131.   struct linebuffer lb1[2];
  132.  
  133.   /* thisline[i] points to the linebuffer holding the next available line
  134.      in file i, or is NULL if there are no lines left in that file.  */
  135.   struct linebuffer *thisline[2];
  136.  
  137.   /* streams[i] holds the input stream for file i.  */
  138.   FILE *streams[2];
  139.  
  140.   int i, ret = 0;
  141.  
  142.   /* Initialize the storage. */
  143.   for (i = 0; i < 2; i++)
  144.     {
  145.       initbuffer (&lb1[i]);
  146.       thisline[i] = &lb1[i];
  147.       streams[i] = strcmp (infiles[i], "-")
  148.     ? fopen (infiles[i], "r") : stdin;
  149.       if (!streams[i])
  150.     {
  151.       error (0, errno, "%s", infiles[i]);
  152.       return 1;
  153.     }
  154.  
  155.       thisline[i] = readline (thisline[i], streams[i]);
  156.     }
  157.  
  158.   while (thisline[0] || thisline[1])
  159.     {
  160.       int order;
  161.  
  162.       /* Compare the next available lines of the two files.  */
  163.  
  164.       if (!thisline[0])
  165.     order = 1;
  166.       else if (!thisline[1])
  167.     order = -1;
  168.       else
  169.     {
  170.       /* Cannot use bcmp -- it only returns a boolean value. */
  171.       order = memcmp (thisline[0]->buffer, thisline[1]->buffer,
  172.               min (thisline[0]->length, thisline[1]->length));
  173.       if (order == 0)
  174.         order = thisline[0]->length - thisline[1]->length;
  175.     }
  176.  
  177.       /* Output the line that is lesser. */
  178.       if (order == 0)
  179.     writeline (thisline[1], stdout, 3);
  180.       else if (order > 0)
  181.     writeline (thisline[1], stdout, 2);
  182.       else
  183.     writeline (thisline[0], stdout, 1);
  184.  
  185.       /* Step the file the line came from.
  186.      If the files match, step both files.  */
  187.       if (order >= 0)
  188.     thisline[1] = readline (thisline[1], streams[1]);
  189.       if (order <= 0)
  190.     thisline[0] = readline (thisline[0], streams[0]);
  191.     }
  192.  
  193.   /* Free all storage and close all input streams. */
  194.   for (i = 0; i < 2; i++)
  195.     {
  196.       free (lb1[i].buffer);
  197.       if (ferror (streams[i]) || fclose (streams[i]) == EOF)
  198.     {
  199.       error (0, errno, "%s", infiles[i]);
  200.       ret = 1;
  201.     }
  202.     }
  203.   if (ferror (stdout) || fclose (stdout) == EOF)
  204.     {
  205.       error (0, errno, "write error");
  206.       ret = 1;
  207.     }
  208.   return ret;
  209. }
  210.  
  211. /* Output the line in linebuffer LINE to stream STREAM
  212.    provided the switches say it should be output.
  213.    CLASS is 1 for a line found only in file 1,
  214.    2 for a line only in file 2, 3 for a line in both. */
  215.  
  216. static void
  217. writeline (line, stream, class)
  218.      struct linebuffer *line;
  219.      FILE *stream;
  220.      int class;
  221. {
  222.   switch (class)
  223.     {
  224.     case 1:
  225.       if (!only_file_1)
  226.     return;
  227.       break;
  228.  
  229.     case 2:
  230.       if (!only_file_2)
  231.     return;
  232.       /* Skip the tab stop for case 1, if we are printing case 1.  */
  233.       if (only_file_1)
  234.     putc ('\t', stream);
  235.       break;
  236.  
  237.     case 3:
  238.       if (!both)
  239.     return;
  240.       /* Skip the tab stop for case 1, if we are printing case 1.  */
  241.       if (only_file_1)
  242.     putc ('\t', stream);
  243.       /* Skip the tab stop for case 2, if we are printing case 2.  */
  244.       if (only_file_2)
  245.     putc ('\t', stream);
  246.       break;
  247.     }
  248.  
  249.   fwrite (line->buffer, sizeof (char), line->length, stream);
  250.   putc ('\n', stream);
  251. }
  252.  
  253. static void
  254. usage (status)
  255.      int status;
  256. {
  257.   if (status != 0)
  258.     fprintf (stderr, "Try `%s --help' for more information.\n",
  259.          program_name);
  260.   else
  261.     {
  262.       printf ("\
  263. Usage: %s [OPTION]... LEFT_FILE RIGHT_FILE\n\
  264. ",
  265.           program_name);
  266.       printf ("\
  267. \n\
  268.   -1              suppress lines unique to left file\n\
  269.   -2              suppress lines unique to right file\n\
  270.   -3              suppress lines unique to both files\n\
  271.       --help      display this help and exit\n\
  272.       --version   output version information and exit\n\
  273. ");
  274.     }
  275.   exit (status);
  276. }
  277.