xref: /freebsd/tools/test/stress2/tools/kldload.sh (revision 8a272653)
1#!/bin/sh
2
3#
4# Copyright (c) 2015 EMC Corp.
5# All rights reserved.
6#
7# Redistribution and use in source and binary forms, with or without
8# modification, are permitted provided that the following conditions
9# are met:
10# 1. Redistributions of source code must retain the above copyright
11#    notice, this list of conditions and the following disclaimer.
12# 2. Redistributions in binary form must reproduce the above copyright
13#    notice, this list of conditions and the following disclaimer in the
14#    documentation and/or other materials provided with the distribution.
15#
16# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19# ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26# SUCH DAMAGE.
27#
28
29# Add support for kldload(8) of compressed modules.
30
31[ $# -ne 1 ] && echo "Usage: $0 <module>" && exit 1
32
33# If possible avoid using a temporary file due to kgdb.
34kldstat | grep -q $1 && exit 0
35kldload $1 2>/dev/null && exit 0
36
37module=$(basename $1 .ko)
38paths=`sysctl -n kern.module_path`
39
40: ${TMPDIR:=/tmp}
41tmpkld=$(mktemp $TMPDIR/$module.XXXXXX) || exit
42trap "rm -f $tmpkld" EXIT INT TERM
43IFS=';'
44for path in $paths; do
45	kld_base=$path/$module.ko
46	for ext in "" .gz .bz2 .xz; do
47		kld=$kld_base$ext
48		[ -s $kld ] || continue
49		case "$kld" in
50		*.gz)
51		    gzcat $kld
52		    ;;
53		*.bz2)
54		    bzcat $kld
55		    ;;
56		*.xz)
57		    xzcat $kld
58		    ;;
59		*)
60		    cat $kld
61		;;
62		esac > $tmpkld
63		[ $? -eq 0 ] && break
64		rm -f $tmpkld
65	done
66	[ -s $tmpkld ] && kldload $tmpkld; ec=$?
67	[ $ec -eq 0 ] && exit
68done
69
70echo "Could not kldload $kld_base*"
71exit $ec
72