grep.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini grep implementation for busybox
   4  *
   5  *
   6  * Copyright (C) 1999,2000 by Lineo, inc.
   7  * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  17  * General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  22  *
  23  */
  24
  25 /*
  26         18-Dec-1999     Konstantin Boldyshev <konst@voshod.com>
  27
  28         + -q option (be quiet)
  29         + exit code depending on grep result (TRUE or FALSE)
  30           (useful for scripts)
  31 */
  32
  33 #include "internal.h"
  34 #include "regexp.h"
  35 #include <stdio.h>
  36 #include <dirent.h>
  37 #include <errno.h>
  38 #include <fcntl.h>
  39 #include <signal.h>
  40 #include <time.h>
  41 #include <ctype.h>
  42
  43 static const char grep_usage[] =
  44         "grep [OPTIONS]... PATTERN [FILE]...\n\n"
  45         "Search for PATTERN in each FILE or standard input.\n\n"
  46         "OPTIONS:\n"
  47         "\t-h\tsuppress the prefixing filename on output\n"
  48         "\t-i\tignore case distinctions\n"
  49         "\t-n\tprint line number with output lines\n"
  50         "\t-q\tbe quiet. Returns 0 if result was found, 1 otherwise\n\n"
  51 #if defined BB_REGEXP
  52         "This version of grep matches full regular expresions.\n";
  53 #else
  54         "This version of grep matches strings (not regular expresions).\n";
  55 #endif
  56
  57 static int match = FALSE, beQuiet = FALSE;
  58
  59 static void do_grep(FILE * fp, char *needle, char *fileName, int tellName,
  60                                         int ignoreCase, int tellLine)
  61 {
  62         char *cp;
  63         long line = 0;
  64         char haystack[BUF_SIZE];
  65
  66         while (fgets(haystack, sizeof(haystack), fp)) {
  67                 line++;
  68                 cp = &haystack[strlen(haystack) - 1];
  69
  70                 if (*cp != '\n')
  71                         fprintf(stderr, "%s: Line too long\n", fileName);
  72
  73                 if (find_match(haystack, needle, ignoreCase) == TRUE) {
  74                         if (tellName == TRUE)
  75                                 printf("%s:", fileName);
  76
  77                         if (tellLine == TRUE)
  78                                 printf("%ld:", line);
  79
  80                         if (beQuiet == FALSE)
  81                                 fputs(haystack, stdout);
  82
  83                         match = TRUE;
  84                 }
  85         }
  86 }
  87
  88
  89 extern int grep_main(int argc, char **argv)
  90 {
  91         FILE *fp;
  92         char *cp;
  93         char *needle;
  94         char *fileName;
  95         int tellName = TRUE;
  96         int ignoreCase = TRUE;
  97         int tellLine = FALSE;
  98
  99
 100         ignoreCase = FALSE;
 101         tellLine = FALSE;
 102
 103         argc--;
 104         argv++;
 105         if (argc < 1) {
 106                 usage(grep_usage);
 107         }
 108
 109         if (**argv == '-') {
 110                 argc--;
 111                 cp = *argv++;
 112
 113                 while (*++cp)
 114                         switch (*cp) {
 115                         case 'i':
 116                                 ignoreCase = TRUE;
 117                                 break;
 118
 119                         case 'h':
 120                                 tellName = FALSE;
 121                                 break;
 122
 123                         case 'n':
 124                                 tellLine = TRUE;
 125                                 break;
 126
 127                         case 'q':
 128                                 beQuiet = TRUE;
 129                                 break;
 130
 131                         default:
 132                                 usage(grep_usage);
 133                         }
 134         }
 135
 136         needle = *argv++;
 137         argc--;
 138
 139         if (argc == 0) {
 140                 do_grep(stdin, needle, "stdin", FALSE, ignoreCase, tellLine);
 141         } else {
 142                 /* Never print the filename for just one file */
 143                 if (argc == 1)
 144                         tellName = FALSE;
 145                 while (argc-- > 0) {
 146                         fileName = *argv++;
 147
 148                         fp = fopen(fileName, "r");
 149                         if (fp == NULL) {
 150                                 perror(fileName);
 151                                 continue;
 152                         }
 153
 154                         do_grep(fp, needle, fileName, tellName, ignoreCase, tellLine);
 155
 156                         if (ferror(fp))
 157                                 perror(fileName);
 158                         fclose(fp);
 159                 }
 160         }
 161         exit(match);
 162 }
 163
 164
 165 /* END CODE */