tools/testing/selftests/memory-hotplug/mem-on-off-test.sh

   1 #!/bin/bash
   2 # SPDX-License-Identifier: GPL-2.0
   3
   4 SYSFS=
   5
   6 prerequisite()
   7 {
   8         msg="skip all tests:"
   9
  10         if [ $UID != 0 ]; then
  11                 echo $msg must be run as root >&2
  12                 exit 0
  13         fi
  14
  15         SYSFS=`mount -t sysfs | head -1 | awk '{ print $3 }'`
  16
  17         if [ ! -d "$SYSFS" ]; then
  18                 echo $msg sysfs is not mounted >&2
  19                 exit 0
  20         fi
  21
  22         if ! ls $SYSFS/devices/system/memory/memory* > /dev/null 2>&1; then
  23                 echo $msg memory hotplug is not supported >&2
  24                 exit 0
  25         fi
  26
  27         if ! grep -q 1 $SYSFS/devices/system/memory/memory*/removable; then
  28                 echo $msg no hot-pluggable memory >&2
  29                 exit 0
  30         fi
  31 }
  32
  33 #
  34 # list all hot-pluggable memory
  35 #
  36 hotpluggable_memory()
  37 {
  38         local state=${1:-.\*}
  39
  40         for memory in $SYSFS/devices/system/memory/memory*; do
  41                 if grep -q 1 $memory/removable &&
  42                    grep -q $state $memory/state; then
  43                         echo ${memory##/*/memory}
  44                 fi
  45         done
  46 }
  47
  48 hotpluggable_offline_memory()
  49 {
  50         hotpluggable_memory offline
  51 }
  52
  53 hotpluggable_online_memory()
  54 {
  55         hotpluggable_memory online
  56 }
  57
  58 memory_is_online()
  59 {
  60         grep -q online $SYSFS/devices/system/memory/memory$1/state
  61 }
  62
  63 memory_is_offline()
  64 {
  65         grep -q offline $SYSFS/devices/system/memory/memory$1/state
  66 }
  67
  68 online_memory()
  69 {
  70         echo online > $SYSFS/devices/system/memory/memory$1/state
  71 }
  72
  73 offline_memory()
  74 {
  75         echo offline > $SYSFS/devices/system/memory/memory$1/state
  76 }
  77
  78 online_memory_expect_success()
  79 {
  80         local memory=$1
  81
  82         if ! online_memory $memory; then
  83                 echo $FUNCNAME $memory: unexpected fail >&2
  84                 return 1
  85         elif ! memory_is_online $memory; then
  86                 echo $FUNCNAME $memory: unexpected offline >&2
  87                 return 1
  88         fi
  89         return 0
  90 }
  91
  92 online_memory_expect_fail()
  93 {
  94         local memory=$1
  95
  96         if online_memory $memory 2> /dev/null; then
  97                 echo $FUNCNAME $memory: unexpected success >&2
  98                 return 1
  99         elif ! memory_is_offline $memory; then
 100                 echo $FUNCNAME $memory: unexpected online >&2
 101                 return 1
 102         fi
 103         return 0
 104 }
 105
 106 offline_memory_expect_success()
 107 {
 108         local memory=$1
 109
 110         if ! offline_memory $memory; then
 111                 echo $FUNCNAME $memory: unexpected fail >&2
 112                 return 1
 113         elif ! memory_is_offline $memory; then
 114                 echo $FUNCNAME $memory: unexpected offline >&2
 115                 return 1
 116         fi
 117         return 0
 118 }
 119
 120 offline_memory_expect_fail()
 121 {
 122         local memory=$1
 123
 124         if offline_memory $memory 2> /dev/null; then
 125                 echo $FUNCNAME $memory: unexpected success >&2
 126                 return 1
 127         elif ! memory_is_online $memory; then
 128                 echo $FUNCNAME $memory: unexpected offline >&2
 129                 return 1
 130         fi
 131         return 0
 132 }
 133
 134 error=-12
 135 priority=0
 136 ratio=10
 137 retval=0
 138
 139 while getopts e:hp:r: opt; do
 140         case $opt in
 141         e)
 142                 error=$OPTARG
 143                 ;;
 144         h)
 145                 echo "Usage $0 [ -e errno ] [ -p notifier-priority ] [ -r percent-of-memory-to-offline ]"
 146                 exit
 147                 ;;
 148         p)
 149                 priority=$OPTARG
 150                 ;;
 151         r)
 152                 ratio=$OPTARG
 153                 if [ "$ratio" -gt 100 ] || [ "$ratio" -lt 0 ]; then
 154                         echo "The percentage should be an integer within 0~100 range"
 155                         exit 1
 156                 fi
 157                 ;;
 158         esac
 159 done
 160
 161 if ! [ "$error" -ge -4095 -a "$error" -lt 0 ]; then
 162         echo "error code must be -4095 <= errno < 0" >&2
 163         exit 1
 164 fi
 165
 166 prerequisite
 167
 168 echo "Test scope: $ratio% hotplug memory"
 169
 170 #
 171 # Online all hot-pluggable memory
 172 #
 173 hotpluggable_num=`hotpluggable_offline_memory | wc -l`
 174 echo -e "\t online all hot-pluggable memory in offline state:"
 175 if [ "$hotpluggable_num" -gt 0 ]; then
 176         for memory in `hotpluggable_offline_memory`; do
 177                 echo "offline->online memory$memory"
 178                 if ! online_memory_expect_success $memory; then
 179                         retval=1
 180                 fi
 181         done
 182 else
 183         echo -e "\t\t SKIPPED - no hot-pluggable memory in offline state"
 184 fi
 185
 186 #
 187 # Offline $ratio percent of hot-pluggable memory
 188 #
 189 hotpluggable_num=`hotpluggable_online_memory | wc -l`
 190 target=`echo "a=$hotpluggable_num*$ratio; if ( a%100 ) a/100+1 else a/100" | bc`
 191 echo -e "\t offline $ratio% hot-pluggable memory in online state"
 192 echo -e "\t trying to offline $target out of $hotpluggable_num memory block(s):"
 193 for memory in `hotpluggable_online_memory`; do
 194         if [ "$target" -gt 0 ]; then
 195                 echo "online->offline memory$memory"
 196                 if offline_memory_expect_success $memory; then
 197                         target=$(($target - 1))
 198                 fi
 199         fi
 200 done
 201 if [ "$target" -gt 0 ]; then
 202         retval=1
 203         echo -e "\t\t FAILED - unable to offline some memory blocks, device busy?"
 204 fi
 205
 206 #
 207 # Online all hot-pluggable memory again
 208 #
 209 hotpluggable_num=`hotpluggable_offline_memory | wc -l`
 210 echo -e "\t online all hot-pluggable memory in offline state:"
 211 if [ "$hotpluggable_num" -gt 0 ]; then
 212         for memory in `hotpluggable_offline_memory`; do
 213                 echo "offline->online memory$memory"
 214                 if ! online_memory_expect_success $memory; then
 215                         retval=1
 216                 fi
 217         done
 218 else
 219         echo -e "\t\t SKIPPED - no hot-pluggable memory in offline state"
 220 fi
 221
 222 #
 223 # Test with memory notifier error injection
 224 #
 225
 226 DEBUGFS=`mount -t debugfs | head -1 | awk '{ print $3 }'`
 227 NOTIFIER_ERR_INJECT_DIR=$DEBUGFS/notifier-error-inject/memory
 228
 229 prerequisite_extra()
 230 {
 231         msg="skip extra tests:"
 232
 233         /sbin/modprobe -q -r memory-notifier-error-inject
 234         /sbin/modprobe -q memory-notifier-error-inject priority=$priority
 235
 236         if [ ! -d "$DEBUGFS" ]; then
 237                 echo $msg debugfs is not mounted >&2
 238                 exit $retval
 239         fi
 240
 241         if [ ! -d $NOTIFIER_ERR_INJECT_DIR ]; then
 242                 echo $msg memory-notifier-error-inject module is not available >&2
 243                 exit $retval
 244         fi
 245 }
 246
 247 echo -e "\t Test with memory notifier error injection"
 248 prerequisite_extra
 249
 250 #
 251 # Offline $ratio percent of hot-pluggable memory
 252 #
 253 echo 0 > $NOTIFIER_ERR_INJECT_DIR/actions/MEM_GOING_OFFLINE/error
 254 for memory in `hotpluggable_online_memory`; do
 255         if [ $((RANDOM % 100)) -lt $ratio ]; then
 256                 offline_memory_expect_success $memory
 257         fi
 258 done
 259
 260 #
 261 # Test memory hot-add error handling (offline => online)
 262 #
 263 echo $error > $NOTIFIER_ERR_INJECT_DIR/actions/MEM_GOING_ONLINE/error
 264 for memory in `hotpluggable_offline_memory`; do
 265         online_memory_expect_fail $memory
 266 done
 267
 268 #
 269 # Online all hot-pluggable memory
 270 #
 271 echo 0 > $NOTIFIER_ERR_INJECT_DIR/actions/MEM_GOING_ONLINE/error
 272 for memory in `hotpluggable_offline_memory`; do
 273         online_memory_expect_success $memory
 274 done
 275
 276 #
 277 # Test memory hot-remove error handling (online => offline)
 278 #
 279 echo $error > $NOTIFIER_ERR_INJECT_DIR/actions/MEM_GOING_OFFLINE/error
 280 for memory in `hotpluggable_online_memory`; do
 281         offline_memory_expect_fail $memory
 282 done
 283
 284 echo 0 > $NOTIFIER_ERR_INJECT_DIR/actions/MEM_GOING_OFFLINE/error
 285 /sbin/modprobe -q -r memory-notifier-error-inject
 286
 287 exit $retval