coreutils/cmp.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini cmp implementation for busybox
   4  *
   5  * Copyright (C) 2000,2001 by Matt Kraai <kraai@alumni.carnegiemellon.edu>
   6  *
   7  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
   8  */
   9
  10 /* BB_AUDIT SUSv3 (virtually) compliant -- uses nicer GNU format for -l. */
  11 /* http://www.opengroup.org/onlinepubs/007904975/utilities/cmp.html */
  12
  13 /* Mar 16, 2003      Manuel Novoa III   (mjn3@codepoet.org)
  14  *
  15  * Original version majorly reworked for SUSv3 compliance, bug fixes, and
  16  * size optimizations.  Changes include:
  17  * 1) Now correctly distinguishes between errors and actual file differences.
  18  * 2) Proper handling of '-' args.
  19  * 3) Actual error checking of i/o.
  20  * 4) Accept SUSv3 -l option.  Note that we use the slightly nicer gnu format
  21  *    in the '-l' case.
  22  */
  23
  24 #include "busybox.h"
  25
  26 static FILE *cmp_xfopen_input(const char * const filename)
  27 {
  28         FILE *fp;
  29
  30         if ((fp = bb_wfopen_input(filename)) != NULL) {
  31                 return fp;
  32         }
  33
  34         exit(xfunc_error_retval);       /* We already output an error message. */
  35 }
  36
  37 static const char fmt_eof[] = "cmp: EOF on %s\n";
  38 static const char fmt_differ[] = "%s %s differ: char %d, line %d\n";
  39 // This fmt_l_opt uses gnu-isms.  SUSv3 would be "%.0s%.0s%d %o %o\n"
  40 static const char fmt_l_opt[] = "%.0s%.0s%d %3o %3o\n";
  41
  42 static const char opt_chars[] = "sl";
  43 #define CMP_OPT_s (1<<0)
  44 #define CMP_OPT_l (1<<1)
  45
  46 int cmp_main(int argc, char **argv)
  47 {
  48         FILE *fp1, *fp2, *outfile = stdout;
  49         const char *filename1, *filename2 = "-";
  50         const char *fmt;
  51         int c1, c2, char_pos = 0, line_pos = 1;
  52         unsigned opt;
  53         int retval = 0;
  54
  55         xfunc_error_retval = 2; /* 1 is returned if files are different. */
  56
  57         opt = getopt32(argc, argv, opt_chars);
  58
  59         if (((opt & (CMP_OPT_s|CMP_OPT_l)) == (CMP_OPT_s|CMP_OPT_l))
  60                         || (((unsigned int)(--argc - optind)) > 1))
  61                 bb_show_usage();
  62
  63         fp1 = cmp_xfopen_input(filename1 = *(argv += optind));
  64
  65         if (*++argv) {
  66                 filename2 = *argv;
  67         }
  68         fp2 = cmp_xfopen_input(filename2);
  69
  70         if (fp1 == fp2) {                       /* Paranioa check... stdin == stdin? */
  71                 /* Note that we don't bother reading stdin.  Neither does gnu wc.
  72                  * But perhaps we should, so that other apps down the chain don't
  73                  * get the input.  Consider 'echo hello | (cmp - - && cat -)'.
  74                  */
  75                 return 0;
  76         }
  77
  78         if (opt & CMP_OPT_l)
  79                 fmt = fmt_l_opt;
  80         else
  81                 fmt = fmt_differ;
  82
  83         do {
  84                 c1 = getc(fp1);
  85                 c2 = getc(fp2);
  86                 ++char_pos;
  87                 if (c1 != c2) {                 /* Remember: a read error may have occurred. */
  88                         retval = 1;             /* But assume the files are different for now. */
  89                         if (c2 == EOF) {
  90                                 /* We know that fp1 isn't at EOF or in an error state.  But to
  91                                  * save space below, things are setup to expect an EOF in fp1
  92                                  * if an EOF occurred.  So, swap things around.
  93                                  */
  94                                 fp1 = fp2;
  95                                 filename1 = filename2;
  96                                 c1 = c2;
  97                         }
  98                         if (c1 == EOF) {
  99                                 xferror(fp1, filename1);
 100                                 fmt = fmt_eof;  /* Well, no error, so it must really be EOF. */
 101                                 outfile = stderr;
 102                                 /* There may have been output to stdout (option -l), so
 103                                  * make sure we fflush before writing to stderr. */
 104                                 xfflush_stdout();
 105                         }
 106                         if (!(opt & CMP_OPT_s)) {
 107                                 if (opt & CMP_OPT_l) {
 108                                         line_pos = c1;  /* line_pos is unused in the -l case. */
 109                                 }
 110                                 bb_fprintf(outfile, fmt, filename1, filename2, char_pos, line_pos, c2);
 111                                 if (opt) {      /* This must be -l since not -s. */
 112                                         /* If we encountered an EOF,
 113                                          * the while check will catch it. */
 114                                         continue;
 115                                 }
 116                         }
 117                         break;
 118                 }
 119                 if (c1 == '\n') {
 120                         ++line_pos;
 121                 }
 122         } while (c1 != EOF);
 123
 124         xferror(fp1, filename1);
 125         xferror(fp2, filename2);
 126
 127         bb_fflush_stdout_and_exit(retval);
 128 }