Fix some formatting

[oweals/busybox.git] / grep.c

diff --git a/grep.c b/grep.c
index dec365f05be675baf00d34a01e36db85dc91dba0..e9ff91b3ee3d52dd8df100ac29e28c7a0aff5435 100644 (file)
--- a/grep.c
+++ b/grep.c
@@ -2,7 +2,7 @@
  * Mini grep implementation for busybox using libc regex.
  *
  * Copyright (C) 1999,2000 by Lineo, inc.
- * Written by Mark Whitley <markw@lineo.com>, <markw@enol.com>
+ * Written by Mark Whitley <markw@lineo.com>, <markw@codepoet.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
@@ -26,48 +26,39 @@
 #include <regex.h>
 #include <string.h> /* for strerror() */
 #include <errno.h>
-#include "internal.h"
+#include "busybox.h"
+
+extern void xregcomp(regex_t *preg, const char *regex, int cflags);
 
 extern int optind; /* in unistd.h */
 extern int errno;  /* for use with strerror() */
 
-static const char grep_usage[] =
-       "grep [-ihHnqvs] pattern [files...]\n"
-#ifndef BB_FEATURE_TRIVIAL_HELP
-       "\nSearch for PATTERN in each FILE or standard input.\n\n"
-       "Options:\n"
-       "\t-H\tprefix output lines with filename where match was found\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"
-       "\t-s\tsuppress file open/read error messages\n\n"
-#endif
-       ;
-
 /* options */
 static int ignore_case       = 0;
 static int print_filename    = 0;
 static int print_line_num    = 0;
+static int print_count_only  = 0;
 static int be_quiet          = 0;
 static int invert_search     = 0;
 static int suppress_err_msgs = 0;
 
 /* globals */
 static regex_t regex; /* storage space for compiled regular expression */
-static int nmatches = 0; /* keeps track of the number of matches */
+static int matched; /* keeps track of whether we ever matched */
 static char *cur_file = NULL; /* the current file we are reading */
 
 
 static void print_matched_line(char *line, int linenum)
 {
+       if (print_count_only)
+               return;
+
        if (print_filename)
                printf("%s:", cur_file);
        if (print_line_num)
                printf("%i:", linenum);
 
-       printf("%s", line);
+       puts(line);
 }
 
 static void grep_file(FILE *file)
@@ -75,8 +66,11 @@ static void grep_file(FILE *file)
        char *line = NULL;
        int ret;
        int linenum = 0;
+       int nmatches = 0;
 
        while ((line = get_line_from_file(file)) != NULL) {
+               if (line[strlen(line)-1] == '\n')
+                       line[strlen(line)-1] = '\0';
                linenum++;
                ret = regexec(&regex, line, 0, NULL, 0);
                if (ret == 0 && !invert_search) { /* match */
@@ -89,15 +83,32 @@ static void grep_file(FILE *file)
                        }
 
                        nmatches++;
-
                        print_matched_line(line, linenum);
 
-               } else if (ret == REG_NOMATCH && invert_search) {
+               }
+               else if (ret == REG_NOMATCH && invert_search) {
+                       if (be_quiet) {
+                               regfree(&regex);
+                               exit(0);
+                       }
+
+                       nmatches++;
                        print_matched_line(line, linenum);
                }
 
                free(line);
        }
+
+       /* special-case post processing */
+       if (print_count_only) {
+               if (print_filename)
+                       printf("%s:", cur_file);
+               printf("%i\n", nmatches);
+       }
+
+       /* record if we matched */
+       if (nmatches != 0)
+               matched = 1;
 }
 
 extern int grep_main(int argc, char **argv)
@@ -105,12 +116,8 @@ extern int grep_main(int argc, char **argv)
        int opt;
        int reflags;
 
-       /* do special-case option parsing */
-       if (argv[1] && (strcmp(argv[1], "--help") == 0))
-               usage(grep_usage);
-
        /* do normal option parsing */
-       while ((opt = getopt(argc, argv, "iHhnqvs")) > 0) {
+       while ((opt = getopt(argc, argv, "iHhnqvsc")) > 0) {
                switch (opt) {
                        case 'i':
                                ignore_case++;
@@ -133,6 +140,9 @@ extern int grep_main(int argc, char **argv)
                        case 's':
                                suppress_err_msgs++;
                                break;
+                       case 'c':
+                               print_count_only++;
+                               break;
                }
        }
 
@@ -143,11 +153,10 @@ extern int grep_main(int argc, char **argv)
        /* compile the regular expression
         * we're not going to mess with sub-expressions, and we need to
         * treat newlines right. */
-       reflags = REG_NOSUB | REG_NEWLINE; 
+       reflags = REG_NOSUB; 
        if (ignore_case)
                reflags |= REG_ICASE;
-       if (bb_regcomp(&regex, argv[optind], reflags) != 0)
-               exit(1);
+       xregcomp(&regex, argv[optind], reflags);
 
        /* argv[(optind+1)..(argc-1)] should be names of file to grep through. If
         * there is more than one file to grep, we will print the filenames */
@@ -158,7 +167,8 @@ extern int grep_main(int argc, char **argv)
         * stdin. Otherwise, we grep through all the files specified. */
        if (argv[optind+1] == NULL || (strcmp(argv[optind+1], "-") == 0)) {
                grep_file(stdin);
-       } else {
+       }
+       else {
                int i;
                FILE *file;
                for (i = optind + 1; i < argc; i++) {
@@ -166,8 +176,9 @@ extern int grep_main(int argc, char **argv)
                        file = fopen(cur_file, "r");
                        if (file == NULL) {
                                if (!suppress_err_msgs)
-                                       fprintf(stderr, "grep: %s: %s\n", cur_file, strerror(errno));
-                       } else {
+                                       perror_msg("%s", cur_file);
+                       }
+                       else {
                                grep_file(file);
                                fclose(file);
                        }
@@ -176,7 +187,7 @@ extern int grep_main(int argc, char **argv)
 
        regfree(&regex);
 
-       if (nmatches == 0)
+       if (!matched)
                return 1;
 
        return 0;