/* $OpenBSD: util.c,v 1.24 2004/02/04 18:38:52 millert Exp $ */ /*- * Copyright (c) 1999 James Howard and Dag-Erling Coïdan Smørgrav * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF * SUCH DAMAGE. */ #include #include #include #include #include #include #include #include #include #include #include #include #include "grep.h" /* * Process a file line by line... */ static int linesqueued; static int procline(str_t *l, int); static int grep_search(fastgrep_t *, unsigned char *, size_t, regmatch_t *pmatch); static int grep_cmp(const unsigned char *, const unsigned char *, size_t); static void grep_revstr(unsigned char *, int); int grep_tree(char **argv) { FTS *fts; FTSENT *p; int c, fts_flags; c = fts_flags = 0; if (Hflag) fts_flags = FTS_COMFOLLOW; if (Pflag) fts_flags = FTS_PHYSICAL; if (Sflag) fts_flags = FTS_LOGICAL; fts_flags |= FTS_NOSTAT | FTS_NOCHDIR; if (!(fts = fts_open(argv, fts_flags, NULL))) err(2, NULL); while ((p = fts_read(fts)) != NULL) { switch (p->fts_info) { case FTS_DNR: break; case FTS_ERR: errx(2, "%s: %s", p->fts_path, strerror(p->fts_errno)); break; case FTS_DP: break; default: c += procfile(p->fts_path); break; } } return c; } int procfile(char *fn) { str_t ln; file_t *f; int c, t, z, nottext; if (fn == NULL) { fn = "(standard input)"; f = grep_fdopen(STDIN_FILENO, "r"); } else { f = grep_open(fn, "r"); } if (f == NULL) { if (!sflag) warn("%s", fn); return 0; } nottext = grep_bin_file(f); if (nottext && binbehave == BIN_FILE_SKIP) { grep_close(f); return 0; } ln.file = fn; ln.line_no = 0; ln.len = 0; linesqueued = 0; ln.off = -1; if (Bflag > 0) initqueue(); for (c = 0; !(lflag && c);) { ln.off += ln.len + 1; if ((ln.dat = grep_fgetln(f, &ln.len)) == NULL) break; if (ln.len > 0 && ln.dat[ln.len - 1] == '\n') --ln.len; ln.line_no++; z = tail; if ((t = procline(&ln, nottext)) == 0 && Bflag > 0 && z == 0) { enqueue(&ln); linesqueued++; } c += t; } if (Bflag > 0) clearqueue(); grep_close(f); if (cflag) { if (!hflag) printf("%s:", ln.file); printf("%u\n", c); } if (lflag && c != 0) printf("%s\n", fn); if (Lflag && c == 0) printf("%s\n", fn); if (c && !cflag && !lflag && !Lflag && binbehave == BIN_FILE_BIN && nottext && !qflag) printf("Binary file %s matches\n", fn); return c; } /* * Process an individual line in a file. Return non-zero if it matches. */ #define isword(x) (isalnum(x) || (x) == '_') static int procline(str_t *l, int nottext) { regmatch_t pmatch; int c, i, r; if (matchall) { c = !vflag; goto print; } for (c = i = 0; i < patterns; i++) { if (fg_pattern[i].pattern) { r = grep_search(&fg_pattern[i], (unsigned char *)l->dat, l->len, &pmatch); } else { pmatch.rm_so = 0; pmatch.rm_eo = l->len; r = regexec(&r_pattern[i], l->dat, 1, &pmatch, eflags); } if (r == 0 && xflag) { if (pmatch.rm_so != 0 || pmatch.rm_eo != l->len) r = REG_NOMATCH; } if (r == 0) { c++; break; } } if (vflag) c = !c; print: if (c && binbehave == BIN_FILE_BIN && nottext) return c; /* Binary file */ if ((tail > 0 || c) && !cflag && !qflag) { if (c) { if (first > 0 && tail == 0 && (Bflag < linesqueued) && (Aflag || Bflag)) printf("--\n"); first = 1; tail = Aflag; if (Bflag > 0) printqueue(); linesqueued = 0; printline(l, ':'); } else { printline(l, '-'); tail--; } } return c; } /* * Returns: -1 on failure, 0 on success */ int fastcomp(fastgrep_t *fg, const char *pattern) { int i; int bol = 0; int eol = 0; int origPatternLen; int shiftPatternLen; int hasDot = 0; int firstHalfDot = -1; int firstLastHalfDot = -1; int lastHalfDot = 0; if (Fflag) { fg->pattern = NULL; return (-1); } /* Initialize. */ origPatternLen = fg->patternLen = strlen(pattern); fg->bol = 0; fg->eol = 0; fg->wmatch = 0; fg->reversedSearch = 0; /* Remove end-of-line character ('$'). */ if (pattern[fg->patternLen - 1] == '$') { eol++; fg->eol = 1; fg->patternLen--; boleol = 1; } /* Remove beginning-of-line character ('^'). */ if (pattern[0] == '^') { bol++; fg->bol = 1; fg->patternLen--; boleol = 1; } /* Remove enclosing [[:<:]] and [[:>:]] (word match). */ if (fg->patternLen >= 14 && strncmp(pattern + fg->bol, "[[:<:]]", 7) == 0 && strncmp(pattern + fg->bol + fg->patternLen - 7, "[[:>:]]", 7) == 0) { fg->patternLen -= 14; fg->wmatch = 7; } /* * Copy pattern minus '^' and '$' characters as well as word * match character classes at the beginning and ending of the * string respectively. */ fg->pattern = grep_malloc(fg->patternLen + 1); memcpy(fg->pattern, pattern + bol + fg->wmatch, fg->patternLen); fg->pattern[fg->patternLen] = '\0'; /* Look for ways to cheat...er...avoid the full regex engine. */ for (i = 0; i < fg->patternLen; i++) { /* Can still cheat? */ if ((isalnum(fg->pattern[i])) || isspace(fg->pattern[i]) || (fg->pattern[i] == '_') || (fg->pattern[i] == ',') || (fg->pattern[i] == '^') || (fg->pattern[i] == '$') || (fg->pattern[i] == '=') || (fg->pattern[i] == '-') || (fg->pattern[i] == ':') || (fg->pattern[i] == '/')) { /* As long as it is good, upper case it for later. */ if (iflag) fg->pattern[i] = toupper(fg->pattern[i]); } else if (fg->pattern[i] == '.') { hasDot = i; if (i < fg->patternLen / 2) { if (firstHalfDot < 0) /* Closest dot to the beginning */ firstHalfDot = i; } else { /* Closest dot to the end of the pattern. */ lastHalfDot = i; if (firstLastHalfDot < 0) firstLastHalfDot = i; } } else { /* Free memory and let others know this is empty. */ free(fg->pattern); fg->pattern = NULL; return (-1); } } /* * Determine if a reverse search would be faster based on the placement * of the dots. */ if ((!(lflag || cflag)) && ((!(bol || eol)) && ((lastHalfDot) && ((firstHalfDot < 0) || ((fg->patternLen - (lastHalfDot + 1)) < firstHalfDot))))) { fg->reversedSearch = 1; hasDot = fg->patternLen - (firstHalfDot < 0 ? firstLastHalfDot : firstHalfDot) - 1; grep_revstr(fg->pattern, fg->patternLen); } /* * Normal Quick Search would require a shift based on the position the * next character after the comparison is within the pattern. With * wildcards, the position of the last dot effects the maximum shift * distance. * The closer to the end the wild card is the slower the search. A * reverse version of this algorithm would be useful for wildcards near * the end of the string. * * Examples: * Pattern Max shift * ------- --------- * this 5 * .his 4 * t.is 3 * th.s 2 * thi. 1 */ /* Adjust the shift based on location of the last dot ('.'). */ shiftPatternLen = fg->patternLen - hasDot; /* Preprocess pattern. */ for (i = 0; i <= UCHAR_MAX; i++) fg->qsBc[i] = shiftPatternLen; for (i = hasDot + 1; i < fg->patternLen; i++) { fg->qsBc[fg->pattern[i]] = fg->patternLen - i; /* * If case is ignored, make the jump apply to both upper and * lower cased characters. As the pattern is stored in upper * case, apply the same to the lower case equivalents. */ if (iflag) fg->qsBc[tolower(fg->pattern[i])] = fg->patternLen - i; } /* * Put pattern back to normal after pre-processing to allow for easy * comparisons later. */ if (fg->reversedSearch) grep_revstr(fg->pattern, fg->patternLen); return (0); } #define wmatch(d, l, s, e) \ ((s == 0 || !isword(d[s-1])) && (e == l || !isword(d[e]))) static int grep_search(fastgrep_t *fg, unsigned char *data, size_t dataLen, regmatch_t *pmatch) { int j; int rtrnVal = REG_NOMATCH; pmatch->rm_so = -1; pmatch->rm_eo = -1; /* No point in going farther if we do not have enough data. */ if (dataLen < fg->patternLen) return (rtrnVal); /* Only try once at the beginning or ending of the line. */ if (fg->bol || fg->eol) { /* Simple text comparison. */ /* Verify data is >= pattern length before searching on it. */ if (dataLen >= fg->patternLen) { /* Determine where in data to start search at. */ if (fg->eol) j = dataLen - fg->patternLen; else j = 0; if (!((fg->bol && fg->eol) && (dataLen != fg->patternLen))) if (grep_cmp(fg->pattern, data + j, fg->patternLen) == -1) { pmatch->rm_so = j; pmatch->rm_eo = j + fg->patternLen; if (!fg->wmatch || wmatch(data, dataLen, pmatch->rm_so, pmatch->rm_eo)) rtrnVal = 0; } } } else if (fg->reversedSearch) { /* Quick Search algorithm. */ j = dataLen; do { if (grep_cmp(fg->pattern, data + j - fg->patternLen, fg->patternLen) == -1) { pmatch->rm_so = j - fg->patternLen; pmatch->rm_eo = j; if (!fg->wmatch || wmatch(data, dataLen, pmatch->rm_so, pmatch->rm_eo)) { rtrnVal = 0; break; } } /* Shift if within bounds, otherwise, we are done. */ if (j == fg->patternLen) break; j -= fg->qsBc[data[j - fg->patternLen - 1]]; } while (j >= fg->patternLen); } else { /* Quick Search algorithm. */ j = 0; do { if (grep_cmp(fg->pattern, data + j, fg->patternLen) == -1) { pmatch->rm_so = j; pmatch->rm_eo = j + fg->patternLen; if (!fg->wmatch || wmatch(data, dataLen, pmatch->rm_so, pmatch->rm_eo)) { rtrnVal = 0; break; } } /* Shift if within bounds, otherwise, we are done. */ if (j + fg->patternLen == dataLen) break; else j += fg->qsBc[data[j + fg->patternLen]]; } while (j <= (dataLen - fg->patternLen)); } return (rtrnVal); } void * grep_malloc(size_t size) { void *ptr; if ((ptr = malloc(size)) == NULL) err(2, "malloc"); return ptr; } void * grep_realloc(void *ptr, size_t size) { if ((ptr = realloc(ptr, size)) == NULL) err(2, "realloc"); return ptr; } /* * Returns: i >= 0 on failure (position that it failed) * -1 on success */ static int grep_cmp(const unsigned char *pattern, const unsigned char *data, size_t len) { int i; for (i = 0; i < len; i++) { if (((pattern[i] == data[i]) || (pattern[i] == '.')) || (iflag && pattern[i] == toupper(data[i]))) continue; return (i); } return (-1); } static void grep_revstr(unsigned char *str, int len) { int i; char c; for (i = 0; i < len / 2; i++) { c = str[i]; str[i] = str[len - i - 1]; str[len - i - 1] = c; } } void printline(str_t *line, int sep) { int n; n = 0; if (!hflag) { fputs(line->file, stdout); ++n; } if (nflag) { if (n) putchar(sep); printf("%d", line->line_no); ++n; } if (bflag) { if (n) putchar(sep); printf("%lld", (long long)line->off); ++n; } if (n) putchar(sep); fwrite(line->dat, line->len, 1, stdout); putchar('\n'); }