scripts/zfs.sh

   1 #!/bin/sh
   2 #
   3 # A simple script to load/unload the ZFS module stack.
   4 #
   5
   6 BASE_DIR=$(dirname "$0")
   7 SCRIPT_COMMON=common.sh
   8 if [ -f "${BASE_DIR}/${SCRIPT_COMMON}" ]; then
   9         . "${BASE_DIR}/${SCRIPT_COMMON}"
  10 else
  11         echo "Missing helper script ${SCRIPT_COMMON}" && exit 1
  12 fi
  13
  14 PROG=zfs.sh
  15 VERBOSE="no"
  16 UNLOAD="no"
  17
  18 ZED_PIDFILE=${ZED_PIDFILE:-/var/run/zed.pid}
  19 LDMOD=${LDMOD:-/sbin/modprobe}
  20
  21 KMOD_ZLIB_DEFLATE=${KMOD_ZLIB_DEFLATE:-zlib_deflate}
  22 KMOD_ZLIB_INFLATE=${KMOD_ZLIB_INFLATE:-zlib_inflate}
  23 KMOD_SPL=${KMOD_SPL:-spl}
  24 KMOD_SPLAT=${KMOD_SPLAT:-splat}
  25 KMOD_ZAVL=${KMOD_ZAVL:-zavl}
  26 KMOD_ZNVPAIR=${KMOD_ZNVPAIR:-znvpair}
  27 KMOD_ZUNICODE=${KMOD_ZUNICODE:-zunicode}
  28 KMOD_ZCOMMON=${KMOD_ZCOMMON:-zcommon}
  29 KMOD_ICP=${KMOD_ICP:-icp}
  30 KMOD_ZFS=${KMOD_ZFS:-zfs}
  31
  32
  33 usage() {
  34 cat << EOF
  35 USAGE:
  36 $0 [hvud] [module-options]
  37
  38 DESCRIPTION:
  39         Load/unload the ZFS module stack.
  40
  41 OPTIONS:
  42         -h      Show this message
  43         -v      Verbose
  44         -u      Unload modules
  45 EOF
  46 }
  47
  48 while getopts 'hvu' OPTION; do
  49         case $OPTION in
  50         h)
  51                 usage
  52                 exit 1
  53                 ;;
  54         v)
  55                 VERBOSE="yes"
  56                 ;;
  57         u)
  58                 UNLOAD="yes"
  59                 ;;
  60         ?)
  61                 usage
  62                 exit
  63                 ;;
  64         esac
  65 done
  66
  67 kill_zed() {
  68         if [ -f "$ZED_PIDFILE" ]; then
  69                 PID=$(cat "$ZED_PIDFILE")
  70                 kill "$PID"
  71         fi
  72 }
  73
  74 check_modules() {
  75         local LOADED_MODULES=""
  76         local MISSING_MODULES=""
  77
  78         for KMOD in $KMOD_SPL $KMOD_SPLAT $KMOD_ZAVL $KMOD_ZNVPAIR \
  79             $KMOD_ZUNICODE $KMOD_ZCOMMON $KMOD_ICP $KMOD_ZFS; do
  80                 NAME=$(basename "$KMOD" .ko)
  81
  82                 if lsmod | egrep -q "^${NAME}"; then
  83                         LOADED_MODULES="$LOADED_MODULES\t$NAME\n"
  84                 fi
  85
  86                 if ! modinfo "$KMOD" >/dev/null 2>&1; then
  87                         MISSING_MODULES="$MISSING_MODULES\t${KMOD}\n"
  88                 fi
  89         done
  90
  91         if [ -n "$LOADED_MODULES" ]; then
  92                 printf "Unload the kernel modules by running '%s -u':\n" "$PROG"
  93                 printf "%b" "$LOADED_MODULES"
  94                 exit 1
  95         fi
  96
  97         if [ -n "$MISSING_MODULES" ]; then
  98                 printf "The following kernel modules can not be found:\n"
  99                 printf "%b" "$MISSING_MODULES"
 100                 exit 1
 101         fi
 102
 103         return 0
 104 }
 105
 106 load_module() {
 107         local KMOD=$1
 108
 109         FILE=$(modinfo "$KMOD" | awk '/^filename:/ {print $2}')
 110         VERSION=$(modinfo "$KMOD" | awk '/^version:/ {print $2}')
 111
 112         if [ "$VERBOSE" = "yes" ]; then
 113                 echo "Loading: $FILE ($VERSION)"
 114         fi
 115
 116         $LDMOD "$KMOD" >/dev/null 2>&1
 117         if [ $? -ne 0 ]; then
 118                 echo "Failed to load $KMOD"
 119                 return 1
 120         fi
 121
 122         return 0
 123 }
 124
 125 load_modules() {
 126         mkdir -p /etc/zfs
 127
 128         if modinfo "$KMOD_ZLIB_DEFLATE" >/dev/null 2>&1; then
 129                 modprobe "$KMOD_ZLIB_DEFLATE" >/dev/null 2>&1
 130         fi
 131
 132         if modinfo "$KMOD_ZLIB_INFLATE">/dev/null 2>&1; then
 133                 modprobe "$KMOD_ZLIB_INFLATE" >/dev/null 2>&1
 134         fi
 135
 136         for KMOD in $KMOD_SPL $KMOD_SPLAT $KMOD_ZAVL $KMOD_ZNVPAIR \
 137             $KMOD_ZUNICODE $KMOD_ZCOMMON $KMOD_ICP $KMOD_ZFS; do
 138                 load_module "$KMOD" || return 1
 139         done
 140
 141         if [ "$VERBOSE" = "yes" ]; then
 142                 echo "Successfully loaded ZFS module stack"
 143         fi
 144
 145         return 0
 146 }
 147
 148 unload_module() {
 149         local KMOD=$1
 150
 151         NAME=$(basename "$KMOD" .ko)
 152         FILE=$(modinfo "$KMOD" | awk '/^filename:/ {print $2}')
 153         VERSION=$(modinfo "$KMOD" | awk '/^version:/ {print $2}')
 154
 155         if [ "$VERBOSE" = "yes" ]; then
 156                 echo "Unloading: $KMOD ($VERSION)"
 157         fi
 158
 159         rmmod "$NAME" || echo "Failed to unload $NAME"
 160
 161         return 0
 162 }
 163
 164 unload_modules() {
 165         for KMOD in $KMOD_ZFS $KMOD_ICP $KMOD_ZCOMMON $KMOD_ZUNICODE \
 166             $KMOD_ZNVPAIR  $KMOD_ZAVL $KMOD_SPLAT $KMOD_SPL; do
 167                 NAME=$(basename "$KMOD" .ko)
 168                 USE_COUNT=$(lsmod | egrep "^${NAME} " | awk '{print $3}')
 169
 170                 if [ "$USE_COUNT" = "0" ] ; then
 171                         unload_module "$KMOD" || return 1
 172                 fi
 173         done
 174
 175         if modinfo "$KMOD_ZLIB_DEFLATE" >/dev/null 2>&1; then
 176                 modprobe -r "$KMOD_ZLIB_DEFLATE" >/dev/null 2>&1
 177         fi
 178
 179         if modinfo "$KMOD_ZLIB_INFLATE">/dev/null 2>&1; then
 180                 modprobe -r "$KMOD_ZLIB_INFLATE" >/dev/null 2>&1
 181         fi
 182
 183         if [ "$VERBOSE" = "yes" ]; then
 184                 echo "Successfully unloaded ZFS module stack"
 185         fi
 186
 187         return 0
 188 }
 189
 190 stack_clear() {
 191         local STACK_MAX_SIZE=/sys/kernel/debug/tracing/stack_max_size
 192         local STACK_TRACER_ENABLED=/proc/sys/kernel/stack_tracer_enabled
 193
 194         if [ -e "$STACK_MAX_SIZE" ]; then
 195                 echo 1 >"$STACK_TRACER_ENABLED"
 196                 echo 0 >"$STACK_MAX_SIZE"
 197         fi
 198 }
 199
 200 stack_check() {
 201         local STACK_MAX_SIZE=/sys/kernel/debug/tracing/stack_max_size
 202         local STACK_TRACE=/sys/kernel/debug/tracing/stack_trace
 203         local STACK_LIMIT=7600
 204
 205         if [ -e "$STACK_MAX_SIZE" ]; then
 206                 STACK_SIZE=$(cat "$STACK_MAX_SIZE")
 207
 208                 if [ "$STACK_SIZE" -ge "$STACK_LIMIT" ]; then
 209                         echo
 210                         echo "Warning: max stack size $STACK_SIZE bytes"
 211                         cat "$STACK_TRACE"
 212                 fi
 213         fi
 214 }
 215
 216 if [ "$(id -u)" != 0 ]; then
 217         echo "Must run as root"
 218         exit 1
 219 fi
 220
 221 if [ "$UNLOAD" = "yes" ]; then
 222         kill_zed
 223         umount -t zfs -a
 224         stack_check
 225         unload_modules
 226 else
 227         stack_clear
 228         check_modules
 229         load_modules "$@"
 230         udevadm trigger
 231         udevadm settle
 232 fi
 233
 234 exit 0