Fixed segfault with 'cut -f 1 -d:' and added 'cut -s' suport.
[oweals/busybox.git] / grep.c
diff --git a/grep.c b/grep.c
index de1c820a040d788bbde21be40dad9465fc019d4f..0e495ff87e19648c674941a47e80cd144839a568 100644 (file)
--- a/grep.c
+++ b/grep.c
@@ -1,7 +1,10 @@
+/* vi: set sw=4 ts=4: */
 /*
  * Mini grep implementation for busybox
  *
- * Copyright (C) 1998 by Erik Andersen <andersee@debian.org>
+ *
+ * Copyright (C) 1999,2000 by Lineo, inc.
+ * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
  *
  * 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
  *
  */
 
+/*
+       18-Dec-1999     Konstantin Boldyshev <konst@voshod.com>
+
+       + -q option (be quiet) 
+       + exit code depending on grep result (TRUE or FALSE)
+         (useful for scripts)
+*/
+
 #include "internal.h"
+#include "regexp.h"
 #include <stdio.h>
 #include <dirent.h>
 #include <errno.h>
 #include <signal.h>
 #include <time.h>
 #include <ctype.h>
+#define BB_DECLARE_EXTERN
+#define bb_need_too_few_args
+#include "messages.c"
+
+static const char grep_usage[] =
+       "grep [OPTIONS]... PATTERN [FILE]...\n"
+#ifndef BB_FEATURE_TRIVIAL_HELP
+       "\nSearch for PATTERN in each FILE or standard input.\n\n"
+       "OPTIONS:\n"
+       "\t-h\tsuppress the prefixing filename on output\n"
+       "\t-i\tignore case distinctions\n"
+       "\t-n\tprint line number with output lines\n"
+       "\t-q\tbe quiet. Returns 0 if result was found, 1 otherwise\n"
+       "\t-v\tselect non-matching lines\n\n"
+#if defined BB_REGEXP
+       "This version of grep matches full regular expressions.\n";
+#else
+       "This version of grep matches strings (not regular expressions).\n"
+#endif
+#endif
+       ;
+
+static int match = FALSE, beQuiet = FALSE;
+
+static void do_grep(FILE * fp, char *needle, char *fileName, int tellName,
+                                       int ignoreCase, int tellLine, int invertSearch)
+{
+       long line = 0;
+       char *haystack;
+       int  truth = !invertSearch;
 
+       while ((haystack = cstring_lineFromFile(fp))) {
+               line++;
+               if (find_match(haystack, needle, ignoreCase) == truth) {
+                       if (tellName == TRUE)
+                               printf("%s:", fileName);
 
-const char grep_usage[] =
-"grep [-ihn]... PATTERN [FILE]...\n"
-"Search for PATTERN in each FILE or standard input.\n\n"
-"\t-h\tsuppress the prefixing filename on output\n"
-"\t-i\tignore case distinctions\n"
-"\t-n\tprint line number with output lines\n\n"
-"This version of grep matches strings (not full regexps).\n";
-
+                       if (tellLine == TRUE)
+                               printf("%ld:", line);
 
-/*
- * See if the specified needle is found in the specified haystack.
- */
-static int search (const char *haystack, const char *needle, int ignoreCase)
-{
+                       if (beQuiet == FALSE)
+                               fputs(haystack, stdout);
 
-    if (ignoreCase == FALSE) {
-       haystack = strstr (haystack, needle);
-       if (haystack == NULL)
-           return FALSE;
-       return TRUE;
-    } else {
-       int i;
-       char needle1[BUF_SIZE];
-       char haystack1[BUF_SIZE];
-
-       strncpy( haystack1, haystack, sizeof(haystack1));
-       strncpy( needle1, needle, sizeof(needle1));
-       for( i=0; i<sizeof(haystack1) && haystack1[i]; i++)
-           haystack1[i]=tolower( haystack1[i]);
-       for( i=0; i<sizeof(needle1) && needle1[i]; i++)
-           needle1[i]=tolower( needle1[i]);
-       haystack = strstr (haystack1, needle1);
-       if (haystack == NULL)
-           return FALSE;
-       return TRUE;
-    }
+                       match = TRUE;
+               }
+               free(haystack);
+       }
 }
 
 
-extern int grep_main (int argc, char **argv)
+extern int grep_main(int argc, char **argv)
 {
-    FILE *fp;
-    const char *needle;
-    const char *name;
-    const char *cp;
-    int tellName=TRUE;
-    int ignoreCase=FALSE;
-    int tellLine=FALSE;
-    long line;
-    char buf[BUF_SIZE];
-
-    ignoreCase = FALSE;
-    tellLine = FALSE;
-
-    argc--;
-    argv++;
-    if (argc < 1) {
-       usage(grep_usage);
-    }
-
-    if (**argv == '-') {
-       argc--;
-       cp = *argv++;
-
-       while (*++cp)
-           switch (*cp) {
-           case 'i':
-               ignoreCase = TRUE;
-               break;
-
-           case 'h':
-               tellName = FALSE;
-               break;
-
-           case 'n':
-               tellLine = TRUE;
-               break;
-
-           default:
+       FILE *fp;
+       char *needle;
+       char *fileName;
+       int tellName     = TRUE;
+       int ignoreCase   = FALSE;
+       int tellLine     = FALSE;
+       int invertSearch = FALSE;
+
+       if (argc < 1) {
                usage(grep_usage);
-           }
-    }
-
-    needle = *argv++;
-    argc--;
-
-    while (argc-- > 0) {
-       name = *argv++;
-
-       fp = fopen (name, "r");
-       if (fp == NULL) {
-           perror (name);
-           continue;
+       }
+       argv++;
+
+       while (--argc >= 0 && *argv && (**argv == '-')) {
+               while (*++(*argv)) {
+                       switch (**argv) {
+                       case 'i':
+                               ignoreCase = TRUE;
+                               break;
+
+                       case 'h':
+                               tellName = FALSE;
+                               break;
+
+                       case 'n':
+                               tellLine = TRUE;
+                               break;
+
+                       case 'q':
+                               beQuiet = TRUE;
+                               break;
+
+                       case 'v':
+                               invertSearch = TRUE;
+                               break;
+
+                       default:
+                               usage(grep_usage);
+                       }
+               }
+               argv++;
        }
 
-       line = 0;
-
-       while (fgets (buf, sizeof (buf), fp)) {
-           line++;
-           cp = &buf[strlen (buf) - 1];
-
-           if (*cp != '\n')
-               fprintf (stderr, "%s: Line too long\n", name);
-
-           if (search (buf, needle, ignoreCase)==TRUE) {
-               if (tellName==TRUE)
-                   printf ("%s: ", name);
-
-               if (tellLine==TRUE)
-                   printf ("%ld: ", line);
-
-               fputs (buf, stdout);
-           }
+       if (argc == 0 || *argv == NULL) {
+               fatalError(too_few_args, "grep");
        }
 
-       if (ferror (fp))
-           perror (name);
+       needle = *argv++;
+       argc--;
 
-       fclose (fp);
-    }
-    exit( TRUE);
+       if (argc == 0) {
+               do_grep(stdin, needle, "stdin", FALSE, ignoreCase, tellLine, invertSearch);
+       } else {
+               /* Never print the filename for just one file */
+               if (argc == 1)
+                       tellName = FALSE;
+               while (argc-- > 0) {
+                       fileName = *argv++;
+
+                       fp = fopen(fileName, "r");
+                       if (fp == NULL) {
+                               perror(fileName);
+                               continue;
+                       }
+
+                       do_grep(fp, needle, fileName, tellName, ignoreCase, tellLine, invertSearch);
+
+                       if (ferror(fp))
+                               perror(fileName);
+                       fclose(fp);
+               }
+       }
+       return(match);
 }
 
 
 /* END CODE */
-
-