Test applets containing numbers in their name. Thanks to Larry Doolittle.

[oweals/busybox.git] / tee.c

diff --git a/tee.c b/tee.c
index 45128b568fccecdff4f8ab33ff1851fc3e0180d2..aa3098c6a580873618e52bdac65966d1482f46b1 100644 (file)
--- a/tee.c
+++ b/tee.c
@@ -1,9 +1,9 @@
+/* vi: set sw=4 ts=4: */
 /*
  * Mini tee implementation for busybox
  *
- *
- * Copyright (C) 1999 by Lineo, inc.
- * Written by John Beppu <beppu@lineo.com>
+ * Copyright (C) 1999,2000,2001 by Lineo, inc.
+ * Written 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
@@ -21,118 +21,48 @@
  *
  */
 
-#include "internal.h"
+#include "busybox.h"
+#include <getopt.h>
 #include <stdio.h>
 
-static const char tee_usage[] =
-"Usage: tee [OPTION]... [FILE]...\n"
-"Copy standard input to each FILE, and also to standard output.\n\n"
-"  -a,    append to the given FILEs, do not overwrite\n"
-"  -i,    ignore interrupt signals\n"
-"  -h,    this help message\n";
-
-/* FileList _______________________________________________________________ */
-
-#define FL_MAX 1024
-static FILE *FileList[FL_MAX];
-static int  FL_end;
-
-typedef void (FL_Function)(FILE *file, char c);
-    
-/* initialize FileList */
-static void
-FL_init()
-{
-    FL_end = 0;
-    FileList[0] = stdout;
-}
-
-/* add a file to FileList */
-static int
-FL_add(const char *filename, char *opt_open)
-{
-    FILE    *file;
-
-    file = fopen(filename, opt_open);
-    if (!file) { return 0; };
-    if (FL_end < FL_MAX) {
-       FileList[++FL_end] = file;
-    }
-    return 1;
-}
-
-/* apply a function to everything in FileList */
-static void
-FL_apply(FL_Function *f, char c)
-{
-    int i;
-    for (i = 0; i <= FL_end; i++) {
-       f(FileList[i], c);
-    }
-}
-
-/* ________________________________________________________________________ */
-
-/* FL_Function for writing to files*/
-static void
-tee_fwrite(FILE *file, char c)
-{
-    fputc(c, file);
-}
-
-/* FL_Function for closing files */
-static void
-tee_fclose(FILE *file, char c)
-{
-    fclose(file);
-}
-
-/* BusyBoxed tee(1) */
 int
 tee_main(int argc, char **argv)
 {
-    int            i;
-    char    c;
-    char    opt;
-    char    opt_fopen[2] = "w";
-
-    /* parse argv[] */
-    for (i = 1; i < argc; i++) {
-       if (argv[i][0] == '-') {
-           opt = argv[i][1];
-           switch (opt) {
-               case 'a':
-                   opt_fopen[0] = 'a';
-                   break;
-               case 'i':
-                   fprintf(stderr, "ingore interrupt not implemented\n");
-                   break;
-               case 'h':
-                   usage(tee_usage);
-                   break;
+       char *mode = "w";
+       int c, i, status = 0, nfiles = 0;
+       FILE **files;
+
+       while ((c = getopt(argc, argv, "a")) != EOF) {
+               switch (c) {
+               case 'a': 
+                       mode = "a";
+                       break;
                default:
-                   fprintf(stderr, "tee: invalid option -- %c\n", opt);
-                   usage(tee_usage);
-           }
-       } else {
-           break;
+                       usage(tee_usage);
+               }
        }
-    }
 
-    /* init FILE pointers */
-    FL_init();
-    for ( ; i < argc; i++) {
-       FL_add(argv[i], opt_fopen);
-    }
+       files = (FILE **)xmalloc(sizeof(FILE *) * (argc - optind + 1));
+       files[nfiles++] = stdout;
+       while (optind < argc) {
+               if ((files[nfiles++] = fopen(argv[optind++], mode)) == NULL) {
+                       nfiles--;
+                       perror_msg("%s", argv[optind-1]);
+                       status = 1;
+               }
+       }
 
-    /* read and redirect */
-    while ((c = (char) getchar()) && (!feof(stdin))) {
-       FL_apply(tee_fwrite, c);
-    }
+       while ((c = getchar()) != EOF)
+               for (i = 0; i < nfiles; i++)
+                       putc(c, files[i]);
 
-    /* clean up */
-    FL_apply(tee_fclose, 0);
-    exit(0);
+       return status;
 }
 
-/* $Id: tee.c,v 1.3 1999/12/10 07:41:03 beppu Exp $ */
+/*
+Local Variables:
+c-file-style: "linux"
+c-basic-offset: 4
+tab-width: 4
+End:
+*/