Fix some formatting

[oweals/busybox.git] / grep.c

diff --git a/grep.c b/grep.c
index a2c07c52371513e07ad41408eb9140f9230edbb3..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,7 +26,9 @@
 #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() */
@@ -35,24 +37,28 @@ extern int errno;  /* for use with strerror() */
 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)
@@ -60,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 */
@@ -74,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)
@@ -90,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++;
@@ -118,6 +140,9 @@ extern int grep_main(int argc, char **argv)
                        case 's':
                                suppress_err_msgs++;
                                break;
+                       case 'c':
+                               print_count_only++;
+                               break;
                }
        }
 
@@ -128,7 +153,7 @@ 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;
        xregcomp(&regex, argv[optind], reflags);
@@ -142,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++) {
@@ -150,8 +176,9 @@ extern int grep_main(int argc, char **argv)
                        file = fopen(cur_file, "r");
                        if (file == NULL) {
                                if (!suppress_err_msgs)
-                                       errorMsg("%s: %s\n", cur_file, strerror(errno));
-                       } else {
+                                       perror_msg("%s", cur_file);
+                       }
+                       else {
                                grep_file(file);
                                fclose(file);
                        }
@@ -160,7 +187,7 @@ extern int grep_main(int argc, char **argv)
 
        regfree(&regex);
 
-       if (nmatches == 0)
+       if (!matched)
                return 1;
 
        return 0;