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 <getopt.h>
  30 #include <errno.h>
  31 #include <stdlib.h>
  32
  33 static int print_count;
  34 static int print_uniq = 1;
  35 static int print_duplicates = 1;
  36
  37 static void print_line(char *line, int count, FILE *fp)
  38 {
  39         if ((print_duplicates && count > 1) || (print_uniq && count == 1)) {
  40                 if (print_count)
  41                         fprintf(fp, "%7d\t%s", count, line);
  42                 else
  43                         fputs(line, fp);
  44         }
  45 }
  46
  47 int uniq_main(int argc, char **argv)
  48 {
  49         FILE *in = stdin, *out = stdout;
  50         char *lastline = NULL, *input;
  51         int opt, count = 0;
  52
  53         /* parse argv[] */
  54         while ((opt = getopt(argc, argv, "cdu")) > 0) {
  55                 switch (opt) {
  56                         case 'c':
  57                                 print_count = 1;
  58                                 break;
  59                         case 'd':
  60                                 print_duplicates = 1;
  61                                 print_uniq = 0;
  62                                 break;
  63                         case 'u':
  64                                 print_duplicates = 0;
  65                                 print_uniq = 1;
  66                                 break;
  67                 }
  68         }
  69
  70         if (argv[optind] != NULL) {
  71                 in = xfopen(argv[optind], "r");
  72                 if (argv[optind+1] != NULL)
  73                         out = xfopen(argv[optind+1], "w");
  74         }
  75
  76         while ((input = get_line_from_file(in)) != NULL) {
  77                 if (lastline == NULL || strcmp(input, lastline) != 0) {
  78                         print_line(lastline, count, out);
  79                         free(lastline);
  80                         lastline = input;
  81                         count = 0;
  82                 }
  83                 count++;
  84         }
  85         print_line(lastline, count, out);
  86         free(lastline);
  87
  88         return EXIT_SUCCESS;
  89 }