PArtial Changelog update. I'm still on vacation (I'm at a campground
[oweals/busybox.git] / uniq.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini uniq implementation for busybox
4  *
5  *
6  * Copyright (C) 1999,2000,2001 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 <stdio.h>
27 #include <string.h>
28 #include <getopt.h>
29 #include <errno.h>
30 #include <stdlib.h>
31 #include "busybox.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 }