rpm2cpio: handle bz2 too; code shrink

[oweals/busybox.git] / coreutils / uniq.c

diff --git a/coreutils/uniq.c b/coreutils/uniq.c
index c0229aecb492c8d5e0a8ff7024e7d538b6cb2f27..126eaeef99dd672d243bfab68ce86fd993c79da0 100644 (file)
--- a/coreutils/uniq.c
+++ b/coreutils/uniq.c
@@ -1,87 +1,102 @@
 /* vi: set sw=4 ts=4: */
 /*
- * Mini uniq implementation for busybox
+ * uniq implementation for busybox
  *
+ * Copyright (C) 2005  Manuel Novoa III  <mjn3@codepoet.org>
  *
- * Copyright (C) 1999,2000 by Lineo, inc.
- * Written by John Beppu <beppu@lineo.com>
- * Rewritten by Matt Kraai <kraai@alumni.carnegiemellon.edu>
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
- *
+ * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
  */
 
-#include "busybox.h"
-#include <stdio.h>
-#include <string.h>
-#include <errno.h>
+/* BB_AUDIT SUSv3 compliant */
+/* http://www.opengroup.org/onlinepubs/007904975/utilities/uniq.html */
 
-static int print_count;
-static int print_uniq = 1;
-static int print_duplicates = 1;
+#include "libbb.h"
 
-static void print_line(char *line, int count, FILE *fp)
+static FILE *xgetoptfile_uniq_s(char **argv, int read0write2)
 {
-       if ((print_duplicates && count > 1) || (print_uniq && count == 1)) {
-               if (print_count)
-                       fprintf(fp, "%7d\t%s", count, line);
-               else
-                       fputs(line, fp);
+       const char *n;
+
+       n = *argv;
+       if (n != NULL) {
+               if ((*n != '-') || n[1]) {
+                       return xfopen(n, "r\0w" + read0write2);
+               }
        }
+       return (read0write2) ? stdout : stdin;
 }
 
-int uniq_main(int argc, char **argv)
+int uniq_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int uniq_main(int argc UNUSED_PARAM, char **argv)
 {
-       FILE *in = stdin, *out = stdout;
-       char *lastline = NULL, *input;
-       int opt, count = 0;
-
-       /* parse argv[] */
-       while ((opt = getopt(argc, argv, "cdu")) > 0) {
-               switch (opt) {
-                       case 'c':
-                               print_count = 1;
-                               break;
-                       case 'd':
-                               print_duplicates = 1;
-                               print_uniq = 0;
-                               break;
-                       case 'u':
-                               print_duplicates = 0;
-                               print_uniq = 1;
-                               break;
-               }
-       }
+       FILE *in, *out;
+       const char *s0, *e0, *s1, *e1, *input_filename;
+       unsigned long dups;
+       unsigned skip_fields, skip_chars, max_chars;
+       unsigned opt;
+       unsigned i;
+
+       enum {
+               OPT_c = 0x1,
+               OPT_d = 0x2,
+               OPT_u = 0x4,
+               OPT_f = 0x8,
+               OPT_s = 0x10,
+               OPT_w = 0x20,
+       };
+
+       skip_fields = skip_chars = 0;
+       max_chars = INT_MAX;
 
-       if (argv[optind] != NULL) {
-               in = xfopen(argv[optind], "r");
-               if (argv[optind+1] != NULL)
-                       out = xfopen(argv[optind+1], "w");
+       opt_complementary = "f+:s+:w+";
+       opt = getopt32(argv, "cduf:s:w:", &skip_fields, &skip_chars, &max_chars);
+       argv += optind;
+
+       input_filename = *argv;
+
+       in = xgetoptfile_uniq_s(argv, 0);
+       if (*argv) {
+               ++argv;
+       }
+       out = xgetoptfile_uniq_s(argv, 2);
+       if (*argv && argv[1]) {
+               bb_show_usage();
        }
 
-       while ((input = get_line_from_file(in)) != NULL) {
-               if (lastline == NULL || strcmp(input, lastline) != 0) {
-                       print_line(lastline, count, out);
-                       free(lastline);
-                       lastline = input;
-                       count = 0;
+       s1 = e1 = NULL; /* prime the pump */
+
+       do {
+               s0 = s1;
+               e0 = e1;
+               dups = 0;
+
+               /* gnu uniq ignores newlines */
+               while ((s1 = xmalloc_fgetline(in)) != NULL) {
+                       e1 = s1;
+                       for (i = skip_fields; i; i--) {
+                               e1 = skip_whitespace(e1);
+                               e1 = skip_non_whitespace(e1);
+                       }
+                       for (i = skip_chars; *e1 && i; i--) {
+                               ++e1;
+                       }
+
+                       if (!s0 || strncmp(e0, e1, max_chars)) {
+                               break;
+                       }
+
+                       ++dups;  /* note: testing for overflow seems excessive. */
                }
-               count++;
-       }
-       print_line(lastline, count, out);
-       free(lastline);
 
-       return EXIT_SUCCESS;
+               if (s0) {
+                       if (!(opt & (OPT_d << !!dups))) { /* (if dups, opt & OPT_e) */
+                               fprintf(out, "\0%ld " + (opt & 1), dups + 1); /* 1 == OPT_c */
+                               fprintf(out, "%s\n", s0);
+                       }
+                       free((void *)s0);
+               }
+       } while (s1);
+
+       die_if_ferror(in, input_filename);
+
+       fflush_stdout_and_exit(EXIT_SUCCESS);
 }