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 3779e5510a9eda7ea7ae988f140bfc40b003e63c..0e495ff87e19648c674941a47e80cd144839a568 100644 (file)
--- a/grep.c
+++ b/grep.c
@@ -1,18 +1,37 @@
+/* vi: set sw=4 ts=4: */
 /*
- * Copyright (c) 1999 by David I. Bell
- * Permission is granted to use, distribute, or modify this source,
- * provided that this copyright notice remains intact.
+ * Mini grep implementation for busybox
  *
- * The "grep" command, taken from sash.
- * This provides basic file searching.
  *
- * Permission to distribute this code under the GPL has been granted.
- * Modified for busybox by Erik Andersen <andersee@debian.org> <andersen@lineo.com>
+ * 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
+ * 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
+ *
  */
 
-#include "internal.h"
-#ifdef BB_GREP
+/*
+       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;
 
-const char grep_usage[] =
-"Search the input file(s) for lines matching the given pattern.\n"
-"\tI search stdin if no files are given.\n"
-"\tI can't grok full regular expressions.\n"
-"usage: grep [in] PATTERN [FILES]...\n"
-"\ti=ignore case, n=list line numbers\n";
+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);
 
+                       if (tellLine == TRUE)
+                               printf("%ld:", line);
 
-static BOOL    search
-       (const char * string, const char * word, BOOL ignoreCase);
+                       if (beQuiet == FALSE)
+                               fputs(haystack, stdout);
 
+                       match = TRUE;
+               }
+               free(haystack);
+       }
+}
 
-extern int 
-grep_main(struct FileInfo * unused, int argc, char ** argv)
-{
-       FILE *          fp;
-       const char *    word;
-       const char *    name;
-       const char *    cp;
-       BOOL            tellName;
-       BOOL            ignoreCase;
-       BOOL            tellLine;
-       long            line;
-       char            buf[BUF_SIZE];
-
-       ignoreCase = FALSE;
-       tellLine = FALSE;
 
-       argc--;
-       argv++;
-       if (argc < 1)
-       {
-               fprintf(stderr, "%s", grep_usage);
-               return 1;
+extern int grep_main(int argc, char **argv)
+{
+       FILE *fp;
+       char *needle;
+       char *fileName;
+       int tellName     = TRUE;
+       int ignoreCase   = FALSE;
+       int tellLine     = FALSE;
+       int invertSearch = FALSE;
+
+       if (argc < 1) {
+               usage(grep_usage);
        }
+       argv++;
 
-       if (**argv == '-')
-       {
-               argc--;
-               cp = *argv++;
-
-               while (*++cp) switch (*cp)
-               {
+       while (--argc >= 0 && *argv && (**argv == '-')) {
+               while (*++(*argv)) {
+                       switch (**argv) {
                        case 'i':
                                ignoreCase = TRUE;
                                break;
 
+                       case 'h':
+                               tellName = FALSE;
+                               break;
+
                        case 'n':
                                tellLine = TRUE;
                                break;
 
-                       default:
-                               fprintf(stderr, "Unknown option\n");
-                               return 1;
-               }
-       }
-
-       word = *argv++;
-       argc--;
-
-       tellName = (argc > 1);
-
-       while (argc-- > 0)
-       {
-               name = *argv++;
-
-               fp = fopen(name, "r");
-
-               if (fp == NULL)
-               {
-                       perror(name);
-
-                       continue;
-               }
-
-               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, word, ignoreCase))
-                       {
-                               if (tellName)
-                                       printf("%s: ", name);
+                       case 'q':
+                               beQuiet = TRUE;
+                               break;
 
-                               if (tellLine)
-                                       printf("%ld: ", line);
+                       case 'v':
+                               invertSearch = TRUE;
+                               break;
 
-                               fputs(buf, stdout);
+                       default:
+                               usage(grep_usage);
                        }
                }
-
-               if (ferror(fp))
-                       perror(name);
-
-               fclose(fp);
+               argv++;
        }
-       return 0;
-}
-
-
-/*
- * See if the specified word is found in the specified string.
- */
-static BOOL
-search(const char * string, const char * word, BOOL ignoreCase)
-{
-       const char *    cp1;
-       const char *    cp2;
-       int             len;
-       int             lowFirst;
-       int             ch1;
-       int             ch2;
 
-       len = strlen(word);
-
-       if (!ignoreCase)
-       {
-               while (TRUE)
-               {
-                       string = strchr(string, word[0]);
-
-                       if (string == NULL)
-                               return FALSE;
-
-                       if (memcmp(string, word, len) == 0)
-                               return TRUE;
-
-                       string++;
-               }
+       if (argc == 0 || *argv == NULL) {
+               fatalError(too_few_args, "grep");
        }
 
-       /*
-        * Here if we need to check case independence.
-        * Do the search by lower casing both strings.
-        */
-       lowFirst = *word;
-
-       if (isupper(lowFirst))
-               lowFirst = tolower(lowFirst);
-
-       while (TRUE)
-       {
-               while (*string && (*string != lowFirst) &&
-                       (!isupper(*string) || (tolower(*string) != lowFirst)))
-               {
-                       string++;
-               }
-
-               if (*string == '\0')
-                       return FALSE;
-
-               cp1 = string;
-               cp2 = word;
-
-               do
-               {
-                       if (*cp2 == '\0')
-                               return TRUE;
-
-                       ch1 = *cp1++;
-
-                       if (isupper(ch1))
-                               ch1 = tolower(ch1);
+       needle = *argv++;
+       argc--;
 
-                       ch2 = *cp2++;
+       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;
+                       }
 
-                       if (isupper(ch2))
-                               ch2 = tolower(ch2);
+                       do_grep(fp, needle, fileName, tellName, ignoreCase, tellLine, invertSearch);
 
+                       if (ferror(fp))
+                               perror(fileName);
+                       fclose(fp);
                }
-               while (ch1 == ch2);
-
-               string++;
        }
+       return(match);
 }
 
-#endif
+
 /* END CODE */