coreutils/uniq.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini uniq implementation for busybox
   4  *
   5  *
   6  * Copyright (C) 1999,2000 by Lineo, inc.
   7  * Written by John Beppu <beppu@lineo.com>
   8  * Rewritten by Matt Kraai <kraai@alumni.carnegiemellon.edu>
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  18  * General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  23  *
  24  */
  25
  26 #include "busybox.h"
  27 #include <stdio.h>
  28 #include <string.h>
  29 #include <errno.h>
  30
  31 static int print_count;
  32 static int print_uniq = 1;
  33 static int print_duplicates = 1;
  34
  35 static void print_line(char *line, int count, FILE *fp)
  36 {
  37         if ((print_duplicates && count > 1) || (print_uniq && count == 1)) {
  38                 if (print_count)
  39                         fprintf(fp, "%7d\t%s", count, line);
  40                 else
  41                         fputs(line, fp);
  42         }
  43 }
  44
  45 int uniq_main(int argc, char **argv)
  46 {
  47         FILE *in = stdin, *out = stdout;
  48         char *lastline = NULL, *input;
  49         int opt, count = 0;
  50
  51         /* parse argv[] */
  52         while ((opt = getopt(argc, argv, "cdu")) > 0) {
  53                 switch (opt) {
  54                         case 'c':
  55                                 print_count = 1;
  56                                 break;
  57                         case 'd':
  58                                 print_duplicates = 1;
  59                                 print_uniq = 0;
  60                                 break;
  61                         case 'u':
  62                                 print_duplicates = 0;
  63                                 print_uniq = 1;
  64                                 break;
  65                 }
  66         }
  67
  68         if (argv[optind] != NULL) {
  69                 in = xfopen(argv[optind], "r");
  70                 if (argv[optind+1] != NULL)
  71                         out = xfopen(argv[optind+1], "w");
  72         }
  73
  74         while ((input = get_line_from_file(in)) != NULL) {
  75                 if (lastline == NULL || strcmp(input, lastline) != 0) {
  76                         print_line(lastline, count, out);
  77                         free(lastline);
  78                         lastline = input;
  79                         count = 0;
  80                 }
  81                 count++;
  82         }
  83         print_line(lastline, count, out);
  84         free(lastline);
  85
  86         return EXIT_SUCCESS;
  87 }