[BACK]Return to gzexe CVS log [TXT][DIR] Up to [local] / src / usr.bin / compress

Annotation of src/usr.bin/compress/gzexe, Revision 1.4

1.1       otto        1: #!/bin/sh -
                      2: #
1.4     ! otto        3: # $OpenBSD: gzexe,v 1.3 2003/08/05 18:22:17 deraadt Exp $
1.1       otto        4: #
                      5: #  Copyright (c) 2003 Otto Moerbeek <otto@drijf.net>
1.3       deraadt     6: #
1.1       otto        7: #  Permission to use, copy, modify, and distribute this software for any
                      8: #  purpose with or without fee is hereby granted, provided that the above
                      9: #  copyright notice and this permission notice appear in all copies.
1.3       deraadt    10: #
1.1       otto       11: #  THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
                     12: #  WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
                     13: #  MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
                     14: #  ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
                     15: #  WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
                     16: #  ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
                     17: #  OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
                     18: #
                     19:
                     20: # The number of lines plus one in the on-the-fly decompression script
1.2       otto       21: lines=19
1.1       otto       22:
                     23: # A simple string to recognize already compressed files
                     24: magic="# compressed by gzexe"
                     25:
                     26: # Write the decompression script to stdout
                     27: header () {
                     28:        typeset prog tmp
                     29:        # first section needs variable expansion, second not
                     30:        cat <<- EOF
                     31:                #!/bin/sh -
                     32:                $magic
                     33:                lines=$lines
                     34:        EOF
                     35:        cat <<- 'EOF'
                     36:                prog=`/usr/bin/basename "$0"`
                     37:                tmp=`/usr/bin/mktemp -d /tmp/gzexeXXXXXXXXXX` || {
                     38:                        /bin/echo "$prog: cannot create tmp dir"; exit 1
                     39:                }
                     40:                trap '/bin/rm -rf "$tmp"' 0
1.2       otto       41:                if /usr/bin/tail +$lines "$0" |
                     42:                    /usr/bin/gzip -dc > "$tmp/$prog" 2> /dev/null; then
1.1       otto       43:                        /bin/chmod u+x "$tmp/$prog"
                     44:                        "$tmp/$prog" ${1+"$@"}
                     45:                        ret=$?
                     46:                else
                     47:                        /bin/echo "$prog: cannot decompress $0"
                     48:                        ret=1
                     49:                fi
                     50:                exit $ret
                     51:        EOF
                     52: }
                     53:
                     54: # Test if a file is compressed by checking the magic line
                     55: compressed () {
                     56:        test "X`sed -n 2p "$1" 2> /dev/null`" = "X$magic"
                     57: }
                     58:
                     59: # Decompress a file
                     60: decompress () {
                     61:        tmp=`mktemp /tmp/gzexeXXXXXXXXXX` || {
                     62:                echo "$prog: cannot create tmp file"
                     63:                return 1
                     64:        }
                     65:        if ! cp "$1" "$tmp"; then
                     66:                echo "$prog: cannot copy $1 to $tmp"
                     67:                rm -f "$tmp"
                     68:                return 1
                     69:        fi
                     70:        if ! tail +$lines "$tmp" | gzip -vdc > "$1"; then
                     71:                echo "$prog: cannot decompress $1"
                     72:                cp "$tmp" "$1"
                     73:                rm -f "$tmp"
                     74:                return 1
                     75:        fi
                     76: }
                     77:
                     78: # Perform some sanity checks on the file
                     79: check () {
                     80:        if test ! -e "$1"; then
                     81:                echo "$prog: cannot compress non-existing file $1"
                     82:                return 1
                     83:        fi
                     84:
                     85:        if test ! -f "$1"; then
                     86:                echo "$prog: cannot compress non-regular file $1"
                     87:                return 1
                     88:        fi
                     89:
                     90:        case `basename "$1"` in
1.4     ! otto       91:                sh | mktemp | rm | echo | tail | gzip | chmod | basename)
1.1       otto       92:                        echo "$prog: cannot compress $1, I depend on it"
                     93:                        return 1
                     94:        esac
                     95:
                     96:        if test ! -x "$1"; then
                     97:                echo "$prog: cannot compress $1, it is not executable"
                     98:                return 1
                     99:        fi
                    100:
                    101:        if test -u "$1" -o -g "$1"; then
                    102:                echo "$prog: cannot compress $1, it has an s bit set"
                    103:                return 1
                    104:        fi
1.4     ! otto      105:
        !           106:        # Build a list of files we should not compress.
        !           107:        # * files we need to decompress
        !           108:        CHECK_LIST="
        !           109:        /bin/chmod
        !           110:        /bin/echo
        !           111:        /bin/sh
        !           112:        /bin/rm
        !           113:        /usr/bin/basename
        !           114:        /usr/bin/gzip
        !           115:        /usr/bin/mktemp
        !           116:        /usr/bin/tail
        !           117:        "
        !           118:        # * files in /bin and /sbin (decompression fails if /usr/bin is not mounted)
        !           119:        # (You could skip these if /usr/bin is always mounted on the same mount point.)
        !           120:        CHECK_LIST="$CHECK_LIST
        !           121:        /bin/*
        !           122:        /sbin/*
        !           123:        "
        !           124:        # See if the program we are trying to compress is in the list.
        !           125:        # To avoid compressing hardlinked files (eg compress & gzip)
        !           126:        # we compare the device & inode.
        !           127:        PROG_STAT_INFO=`stat -f '%d %i' "$1"`
        !           128:        for CHECK in $CHECK_LIST; do
        !           129:                if test -f "$CHECK"; then
        !           130:                        CHECK_STAT_INFO=`stat -f '%d %i' "$CHECK"`
        !           131:                        if test "X$PROG_STAT_INFO" == "X$CHECK_STAT_INFO"; then
        !           132:                                echo "$prog: cannot compress $1, it is the same file as $CHECK"
        !           133:                                return 1
        !           134:                        fi
        !           135:                fi
        !           136:        done
1.1       otto      137: }
                    138:
                    139: # Compress a file
                    140: compress () {
                    141:        tmp=`mktemp /tmp/gzexeXXXXXXXXXX` || {
                    142:                echo "$prog: cannot create tmp file"
                    143:                return 1
                    144:        }
                    145:        if ! cp "$1" "$tmp"; then
                    146:                echo "$prog: cannot copy $1 to $tmp"
                    147:                rm -f "$tmp"
                    148:                return 1
                    149:        fi
                    150:        if ! cp "$1" "$1"~; then
                    151:                echo "$prog: cannot create backup copy $1~"
                    152:                rm -f "$1"~ "$tmp"
                    153:                return 1
                    154:        fi
                    155:
                    156:        # Use cp to overwrite the existing file preserving mode and owner
                    157:        # if possible. If the file is not writable, this will produce an
                    158:        # error.
                    159:
                    160:        if header "$1" > "$tmp" && gzip -vc "$1" >> "$tmp"; then
                    161:                if ! cp "$tmp" "$1"; then
                    162:                        echo "$prog: cannot copy $tmp to $1"
                    163:                        rm -f "$tmp"
                    164:                        return 1
                    165:                fi
                    166:        else
                    167:                echo "$prog: cannot compress $1"
                    168:                rm -f "$1"~ "$tmp"
                    169:                return 1
                    170:        fi
                    171: }
                    172:
                    173: # Is the -d flag specified?
                    174: dflag=
                    175:
                    176: # Return value
                    177: rc=0
                    178:
                    179: if test "X$1" = X-d; then
                    180:        dflag=1
                    181:        shift
                    182: fi
                    183:
                    184: prog=`basename "$0"`
                    185: USAGE="usage: $prog [-d] file ..."
                    186: if test $# -eq 0; then
                    187:        echo $USAGE
                    188:        exit 1
                    189: fi
                    190:
                    191: while test $# -ne 0; do
                    192:        if test $dflag; then
                    193:                if ! compressed "$1"; then
                    194:                        echo "$prog: $1 is not compressed"
                    195:                        rc=1;
                    196:                elif ! decompress "$1"; then
                    197:                        rc=$?
                    198:                fi
                    199:        else
                    200:                if compressed "$1"; then
                    201:                        echo "$prog: $1 is already compressed"
                    202:                        rc=1;
                    203:                elif ! check "$1" || ! compress "$1"; then
                    204:                        rc=$?
                    205:                fi
                    206:        fi
                    207:        shift
                    208: done
                    209: exit $rc