1#!/bin/bash 2# SPDX-License-Identifier: GPL-2.0 3 4SYSFS= 5 6prerequisite() 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# 36hotpluggable_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 48hotpluggable_offline_memory() 49{ 50 hotpluggable_memory offline 51} 52 53hotpluggable_online_memory() 54{ 55 hotpluggable_memory online 56} 57 58memory_is_online() 59{ 60 grep -q online $SYSFS/devices/system/memory/memory$1/state 61} 62 63memory_is_offline() 64{ 65 grep -q offline $SYSFS/devices/system/memory/memory$1/state 66} 67 68online_memory() 69{ 70 echo online > $SYSFS/devices/system/memory/memory$1/state 71} 72 73offline_memory() 74{ 75 echo offline > $SYSFS/devices/system/memory/memory$1/state 76} 77 78online_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 92online_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 106offline_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 120offline_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 134error=-12 135priority=0 136ratio=10 137retval=0 138 139while 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 159done 160 161if ! [ "$error" -ge -4095 -a "$error" -lt 0 ]; then 162 echo "error code must be -4095 <= errno < 0" >&2 163 exit 1 164fi 165 166prerequisite 167 168echo "Test scope: $ratio% hotplug memory" 169 170# 171# Online all hot-pluggable memory 172# 173hotpluggable_num=`hotpluggable_offline_memory | wc -l` 174echo -e "\t online all hot-pluggable memory in offline state:" 175if [ "$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 182else 183 echo -e "\t\t SKIPPED - no hot-pluggable memory in offline state" 184fi 185 186# 187# Offline $ratio percent of hot-pluggable memory 188# 189hotpluggable_num=`hotpluggable_online_memory | wc -l` 190target=`echo "a=$hotpluggable_num*$ratio; if ( a%100 ) a/100+1 else a/100" | bc` 191echo -e "\t offline $ratio% hot-pluggable memory in online state" 192echo -e "\t trying to offline $target out of $hotpluggable_num memory block(s):" 193for 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 200done 201if [ "$target" -gt 0 ]; then 202 retval=1 203 echo -e "\t\t FAILED - unable to offline some memory blocks, device busy?" 204fi 205 206# 207# Online all hot-pluggable memory again 208# 209hotpluggable_num=`hotpluggable_offline_memory | wc -l` 210echo -e "\t online all hot-pluggable memory in offline state:" 211if [ "$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 218else 219 echo -e "\t\t SKIPPED - no hot-pluggable memory in offline state" 220fi 221 222# 223# Test with memory notifier error injection 224# 225 226DEBUGFS=`mount -t debugfs | head -1 | awk '{ print $3 }'` 227NOTIFIER_ERR_INJECT_DIR=$DEBUGFS/notifier-error-inject/memory 228 229prerequisite_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 247echo -e "\t Test with memory notifier error injection" 248prerequisite_extra 249 250# 251# Offline $ratio percent of hot-pluggable memory 252# 253echo 0 > $NOTIFIER_ERR_INJECT_DIR/actions/MEM_GOING_OFFLINE/error 254for memory in `hotpluggable_online_memory`; do 255 if [ $((RANDOM % 100)) -lt $ratio ]; then 256 offline_memory_expect_success $memory 257 fi 258done 259 260# 261# Test memory hot-add error handling (offline => online) 262# 263echo $error > $NOTIFIER_ERR_INJECT_DIR/actions/MEM_GOING_ONLINE/error 264for memory in `hotpluggable_offline_memory`; do 265 online_memory_expect_fail $memory 266done 267 268# 269# Online all hot-pluggable memory 270# 271echo 0 > $NOTIFIER_ERR_INJECT_DIR/actions/MEM_GOING_ONLINE/error 272for memory in `hotpluggable_offline_memory`; do 273 online_memory_expect_success $memory 274done 275 276# 277# Test memory hot-remove error handling (online => offline) 278# 279echo $error > $NOTIFIER_ERR_INJECT_DIR/actions/MEM_GOING_OFFLINE/error 280for memory in `hotpluggable_online_memory`; do 281 offline_memory_expect_fail $memory 282done 283 284echo 0 > $NOTIFIER_ERR_INJECT_DIR/actions/MEM_GOING_OFFLINE/error 285/sbin/modprobe -q -r memory-notifier-error-inject 286 287exit $retval 288