From 0e5efe5ab4942c9583d58d7d31ad88df2cc93eff Mon Sep 17 00:00:00 2001 From: Brian Behlendorf Date: Thu, 8 Jan 2009 16:19:38 -0800 Subject: [PATCH] Update helper scripts --- scripts/check.sh | 4 +- scripts/load-zfs.sh | 55 ------------ scripts/unload-zfs.sh | 53 ------------ scripts/zfs.sh | 197 ++++++++++++++++++++++++++++++++++++++++++ 4 files changed, 199 insertions(+), 110 deletions(-) delete mode 100755 scripts/load-zfs.sh delete mode 100755 scripts/unload-zfs.sh create mode 100755 scripts/zfs.sh diff --git a/scripts/check.sh b/scripts/check.sh index bf93ee564e..e4b0e8e06f 100755 --- a/scripts/check.sh +++ b/scripts/check.sh @@ -11,7 +11,7 @@ if [ $(id -u) != 0 ]; then die "Must run as root" fi -./load-zfs.sh || die "" -./unload-zfs.sh || die "" +./zfs.sh -v || die "" +./zfs.sh -vu || die "" exit 0 diff --git a/scripts/load-zfs.sh b/scripts/load-zfs.sh deleted file mode 100755 index 94a8ac2cda..0000000000 --- a/scripts/load-zfs.sh +++ /dev/null @@ -1,55 +0,0 @@ -#!/bin/bash - -prog=load-zfs.sh -. ../.script-config - -spl_options=$1 -zpool_options=$2 - -spl_module=${SPLBUILD}/modules/spl/spl.ko -zlib_module=/lib/modules/${KERNELSRCVER}/kernel/lib/zlib_deflate/zlib_deflate.ko -zavl_module=${ZFSBUILD}/lib/libavl/zavl.ko -znvpair_module=${ZFSBUILD}/lib/libnvpair/znvpair.ko -zcommon_module=${ZFSBUILD}/lib/libzcommon/zcommon.ko -zpool_module=${ZFSBUILD}/lib/libzpool/zpool.ko -zctl_module=${ZFSBUILD}/lib/libdmu-ctl/zctl.ko -zpios_module=${ZFSBUILD}/lib/libzpios/zpios.ko - -die() { - echo "${prog}: $1" >&2 - exit 1 -} - -load_module() { - echo "Loading $1" - /sbin/insmod $* || die "Failed to load $1" -} - -if [ $(id -u) != 0 ]; then - die "Must run as root" -fi - -if /sbin/lsmod | egrep -q "^spl|^zavl|^znvpair|^zcommon|^zlib_deflate|^zpool"; then - die "Must start with modules unloaded" -fi - -if [ ! -f ${zavl_module} ] || - [ ! -f ${znvpair_module} ] || - [ ! -f ${zcommon_module} ] || - [ ! -f ${zpool_module} ]; then - die "Source tree must be built, run 'make'" -fi - -load_module ${spl_module} ${spl_options} -load_module ${zlib_module} -load_module ${zavl_module} -load_module ${znvpair_module} -load_module ${zcommon_module} -load_module ${zpool_module} ${zpool_options} -load_module ${zctl_module} -load_module ${zpios_module} - -sleep 1 -echo "Successfully loaded ZFS module stack" - -exit 0 diff --git a/scripts/unload-zfs.sh b/scripts/unload-zfs.sh deleted file mode 100755 index 58ed137e1a..0000000000 --- a/scripts/unload-zfs.sh +++ /dev/null @@ -1,53 +0,0 @@ -#!/bin/bash - -prog=unload-zfs.sh -. ../.script-config - -spl_module=${SPLBUILD}/modules/spl/spl.ko -zlib_module=/lib/modules/${KERNELSRCVER}/kernel/lib/zlib_deflate/zlib_deflate.ko -zavl_module=${ZFSBUILD}/lib/libavl/zavl.ko -znvpair_module=${ZFSBUILD}/lib/libnvpair/znvpair.ko -zcommon_module=${ZFSBUILD}/lib/libzcommon/zcommon.ko -zpool_module=${ZFSBUILD}/lib/libzpool/zpool.ko -zctl_module=${ZFSBUILD}/lib/libdmu-ctl/zctl.ko -zpios_module=${ZFSBUILD}/lib/libzpios/zpios.ko - -die() { - echo "${prog}: $1" >&2 - exit 1 -} - -unload_module() { - echo "Unloading $1" - /sbin/rmmod $1 || die "Failed to unload $1" -} - -if [ $(id -u) != 0 ]; then - die "Must run as root" -fi - -unload_module ${zpios_module} -unload_module ${zctl_module} -unload_module ${zpool_module} -unload_module ${zcommon_module} -unload_module ${znvpair_module} -unload_module ${zavl_module} -unload_module ${zlib_module} - -# Set DUMP=1 to generate debug logs on unload -if [ -n "${DUMP}" ]; then - sysctl -w kernel.spl.debug.dump=1 - # This is racy, I don't like it, but for a helper script it will do. - SPL_LOG=`dmesg | tail -n 1 | cut -f5 -d' '` - ${SPLBUILD}/cmd/spl ${SPL_LOG} >${SPL_LOG}.log - echo - echo "Dumped debug log: ${SPL_LOG}.log" - tail -n1 ${SPL_LOG}.log - echo -fi - -unload_module ${spl_module} - -echo "Successfully unloaded ZFS module stack" - -exit 0 diff --git a/scripts/zfs.sh b/scripts/zfs.sh new file mode 100755 index 0000000000..80e86f8bc1 --- /dev/null +++ b/scripts/zfs.sh @@ -0,0 +1,197 @@ +#!/bin/bash + +prog=load-zfs.sh +. ../.script-config + +KMOD=/lib/modules/${KERNELSRCVER}/kernel +kernel_modules=( \ + $KMOD/lib/zlib_deflate/zlib_deflate.ko \ +) + +spl_modules=( \ + ${SPLBUILD}/modules/spl/spl.ko \ +) + +zfs_modules=( \ + ${MODDIR}/avl/zavl.ko \ + ${MODDIR}/nvpair/znvpair.ko \ + ${MODDIR}/unicode/zunicode.ko \ + ${MODDIR}/zcommon/zcommon.ko \ + ${MODDIR}/zfs/zpool.ko \ +) + +test_modules=( \ + ${MODDIR}/zpios/zpios.ko \ +) + +modules=( \ + ${kernel_modules[*]} \ + ${spl_modules[*]} \ + ${zfs_modules[*]} \ + ${test_modules[*]} \ +) + +die() { + echo -e "${prog}: $1" >&2 + exit 1 +} + +usage() { +cat << EOF +usage: $0 [hvu] [module-options] + +OPTIONS: + -h Show this message + -v Verbose + -u Unload modules + -d Save debug log on unload + +MODULE-OPTIONS: Must be of the frm module="options", for example: + +$0 zpool="zfs_prefetch_disable=1" +$0 zpool="zfs_prefetch_disable=1 zfs_mdcomp_disable=1" +$0 spl="spl_debug_mask=0" + +EOF +} + +check_modules() { + loaded_modules=() + missing_modules=() + + for module in ${modules[*]}; do + name=`basename $module .ko` + + if /sbin/lsmod | egrep -q "^${name}"; then + loaded_modules=(${name} ${loaded_modules[*]}) + fi + + if [ ! -f $module ]; then + missing_modules=("\t${module}\n" ${missing_modules[*]}) + fi + done + + if [ ${#loaded_modules[*]} -gt 0 ]; then + error="The following modules must be unloaded, manually run:\n" + die "${error}'/sbin/rmmod ${loaded_modules[*]}'" + fi + + if [ ${#missing_modules[*]} -gt 0 ]; then + error="The following modules can not be found," + error="${error} ensure your source trees are built:\n" + die "${error}${missing_modules[*]}" + fi +} + +spl_dump_log() { + sysctl -w kernel.spl.debug.dump=1 &>/dev/null + name=`dmesg | tail -n 1 | cut -f5 -d' '` + ${SPLBUILD}/cmd/spl ${name} >${name}.log + echo + echo "Dumped debug log: ${name}.log" + tail -n1 ${name}.log + echo +} + +load_module() { + name=`basename $module .ko` + + if [ ${verbose} ]; then + echo "Loading $name ($@)" + fi + + /sbin/insmod $* || die "Failed to load $1" +} + +load_modules() { + + for module in ${modules[*]}; do + name=`basename $module .ko` + value= + + for opt in "$@"; do + opt_name=`echo $opt | cut -f1 -d'='` + + if [ ${name} = ${opt_name} ]; then + value=`echo $opt | cut -f2- -d'='` + fi + done + + load_module ${module} ${value} + done + + if [ ${verbose} ]; then + echo "Successfully loaded ZFS module stack" + fi +} + +unload_module() { + name=`basename $module .ko` + + if [ ${verbose} ]; then + echo "Unloading $name ($@)" + fi + + /sbin/rmmod $name || die "Failed to unload $name" +} + +unload_modules() { + modules_reverse=( $(echo ${modules[@]} | + awk '{for (i=NF;i>=1;i--) printf $i" "} END{print ""}') ) + + for module in ${modules_reverse[*]}; do + name=`basename $module .ko` + + if /sbin/lsmod | egrep -q "^${name}"; then + + if [ "${dump_log}" -a ${name} = "spl" ]; then + spl_dump_log + fi + + unload_module ${module} + fi + done + + if [ ${verbose} ]; then + echo "Successfully unloaded ZFS module stack" + fi +} + +if [ $(id -u) != 0 ]; then + die "Must run as root" +fi + +verbose= +unload= +dump_log= + +while getopts 'hvud' OPTION; do + case $OPTION in + h) + usage + exit 1 + ;; + v) + verbose=1 + ;; + u) + unload=1 + ;; + d) + dump_log=1 + ;; + ?) + usage + exit + ;; + esac +done + +if [ ${unload} ]; then + unload_modules +else + check_modules + load_modules "$@" +fi + +exit 0