deps/libchdr/deps/zstd-1.5.5/programs/zstdgrep

   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
  25 grep=${GREP:-grep}
  26 zcat=${ZCAT:-zstdcat}
  27
  28 endofopts=0
  29 pattern_found=0
  30 grep_args=""
  31 hyphen=0
  32 silent=0
  33
  34 prog=${0##*/}
  35
  36 # handle being called 'zegrep' or 'zfgrep'
  37 case $prog in
  38     *egrep*) prog=zegrep; grep_args='-E';;
  39     *fgrep*) prog=zfgrep; grep_args='-F';;
  40     *)       prog=zstdgrep;;
  41 esac
  42
  43 # skip all options and pass them on to grep taking care of options
  44 # with arguments, and if -e was supplied
  45
  46 while [ "$#" -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
  91 done
  92
  93 # if no -e option was found, take next argument as grep-pattern
  94 if [ "${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
 104 fi
 105
 106 EXIT_CODE=0
 107 # call grep ...
 108 if [ "$#" -lt 1 ]; then
 109     # ... on stdin
 110     set -f # Disable file name generation (globbing).
 111     # shellcheck disable=SC2086
 112     "${zcat}" - | "${grep}" ${grep_args} -- "${pattern}" -
 113     EXIT_CODE=$?
 114     set +f
 115 else
 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}" -- "$1" | "${grep}" --label="${1}" ${grep_args} -- -
 125         else
 126             "${zcat}" -- "$1" | "${grep}" --label="${1}" ${grep_args} -- "${pattern}" -
 127         fi
 128         [ "$?" -ne 0 ] && EXIT_CODE=1
 129         shift
 130     done
 131     set +f
 132 fi
 133
 134 exit "${EXIT_CODE}"