1#!/bin/sh 2# 3# Copyright (c) 2003 Thomas Klausner. 4# 5# Redistribution and use in source and binary forms, with or without 6# modification, are permitted provided that the following conditions 7# are met: 8# 1. Redistributions of source code must retain the above copyright 9# notice, this list of conditions and the following disclaimer. 10# 2. Redistributions in binary form must reproduce the above copyright 11# notice, this list of conditions and the following disclaimer in the 12# documentation and/or other materials provided with the distribution. 13# 14# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 15# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 16# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 17# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 18# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 19# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 20# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 21# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 22# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 23# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 24 25grep=${GREP:-grep} 26zcat=${ZCAT:-zstdcat} 27 28endofopts=0 29pattern_found=0 30grep_args="" 31hyphen=0 32silent=0 33 34prog=${0##*/} 35 36# handle being called 'zegrep' or 'zfgrep' 37case $prog in 38 *egrep*) prog=zegrep; grep_args='-E';; 39 *fgrep*) prog=zfgrep; grep_args='-F';; 40 *) prog=zstdgrep;; 41esac 42 43# skip all options and pass them on to grep taking care of options 44# with arguments, and if -e was supplied 45 46while [ "$#" -gt 0 ] && [ "${endofopts}" -eq 0 ]; do 47 case "$1" in 48 # from GNU grep-2.5.1 -- keep in sync! 49 -[ABCDXdefm]) 50 if [ "$#" -lt 2 ]; then 51 printf '%s: missing argument for %s flag\n' "${prog}" "$1" >&2 52 exit 1 53 fi 54 case "$1" in 55 -e) 56 pattern="$2" 57 pattern_found=1 58 shift 2 59 break 60 ;; 61 -f) 62 pattern_found=2 63 ;; 64 *) 65 ;; 66 esac 67 grep_args="${grep_args} $1 $2" 68 shift 2 69 ;; 70 --) 71 shift 72 endofopts=1 73 ;; 74 -) 75 hyphen=1 76 shift 77 ;; 78 -h) 79 silent=1 80 shift 81 ;; 82 -*) 83 grep_args="${grep_args} $1" 84 shift 85 ;; 86 *) 87 # pattern to grep for 88 endofopts=1 89 ;; 90 esac 91done 92 93# if no -e option was found, take next argument as grep-pattern 94if [ "${pattern_found}" -lt 1 ]; then 95 if [ "$#" -ge 1 ]; then 96 pattern="$1" 97 shift 98 elif [ "${hyphen}" -gt 0 ]; then 99 pattern="-" 100 else 101 printf '%s: missing pattern\n' "${prog}" >&2 102 exit 1 103 fi 104fi 105 106EXIT_CODE=0 107# call grep ... 108if [ "$#" -lt 1 ]; then 109 # ... on stdin 110 set -f # Disable file name generation (globbing). 111 # shellcheck disable=SC2086 112 "${zcat}" -fq - | "${grep}" ${grep_args} -- "${pattern}" - 113 EXIT_CODE=$? 114 set +f 115else 116 # ... on all files given on the command line 117 if [ "${silent}" -lt 1 ] && [ "$#" -gt 1 ]; then 118 grep_args="-H ${grep_args}" 119 fi 120 set -f 121 while [ "$#" -gt 0 ]; do 122 # shellcheck disable=SC2086 123 if [ $pattern_found -eq 2 ]; then 124 "${zcat}" -fq -- "$1" | "${grep}" --label="${1}" ${grep_args} -- - 125 else 126 "${zcat}" -fq -- "$1" | "${grep}" --label="${1}" ${grep_args} -- "${pattern}" - 127 fi 128 [ "$?" -ne 0 ] && EXIT_CODE=1 129 shift 130 done 131 set +f 132fi 133 134exit "${EXIT_CODE}" 135