Home | History | Annotate | Download | only in control
      1 #!/bin/sh
      2 
      3 ################################################################################
      4 ##                                                                            ##
      5 ##   Copyright (c) 2010 Mohamed Naufal Basheer                                ##
      6 ##                                                                            ##
      7 ##   This program is free software;  you can redistribute it and/or modify    ##
      8 ##   it under the terms of the GNU General Public License as published by     ##
      9 ##   the Free Software Foundation; either version 2 of the License, or        ##
     10 ##   (at your option) any later version.                                      ##
     11 ##                                                                            ##
     12 ##   This program is distributed in the hope that it will be useful,          ##
     13 ##   but WITHOUT ANY WARRANTY;  without even the implied warranty of          ##
     14 ##   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See                ##
     15 ##   the GNU General Public License for more details.                         ##
     16 ##                                                                            ##
     17 ##   You should have received a copy of the GNU General Public License        ##
     18 ##   along with this program;  if not, write to the Free Software             ##
     19 ##   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA  ##
     20 ##                                                                            ##
     21 ##                                                                            ##
     22 ##   File:    memcg_control_test.sh                                           ##
     23 ##                                                                            ##
     24 ##   Purpose: Implement various memory controller tests                       ##
     25 ##                                                                            ##
     26 ##   Author:  Mohamed Naufal Basheer <naufal11 (at] gmail.com>                     ##
     27 ##                                                                            ##
     28 ################################################################################
     29 
     30 if [ "x$(grep -w memory /proc/cgroups | cut -f4)" != "x1" ]; then
     31 	echo "WARNING:"
     32 	echo "Either kernel does not support memory resource controller or feature not enabled"
     33 	echo "Skipping all memcg_control testcases...."
     34 	exit 0
     35 fi
     36 
     37 export TCID="memcg_control"
     38 export TST_TOTAL=1
     39 export TST_COUNT=0
     40 
     41 export TMP=${TMP:-/tmp}
     42 cd $TMP
     43 
     44 PAGE_SIZE=$(getconf PAGESIZE)
     45 
     46 TOT_MEM_LIMIT=$PAGE_SIZE
     47 ACTIVE_MEM_LIMIT=$PAGE_SIZE
     48 PROC_MEM=$((PAGE_SIZE * 2))
     49 
     50 TST_PATH=$PWD
     51 STATUS_PIPE="$TMP/status_pipe"
     52 
     53 PASS=0
     54 FAIL=1
     55 
     56 # Check if the test process is killed on crossing boundary
     57 test_proc_kill()
     58 {
     59 	cd $TMP
     60 	mem_process -m $PROC_MEM &
     61 	cd $OLDPWD
     62 	sleep 1
     63 	echo $! > tasks
     64 
     65 	#Instruct the test process to start acquiring memory
     66 	echo m > $STATUS_PIPE
     67 	sleep 5
     68 
     69 	#Check if killed
     70 	ps -p $! > /dev/null 2> /dev/null
     71 	if [ $? -eq 0 ]; then
     72 		echo m > $STATUS_PIPE
     73 		echo x > $STATUS_PIPE
     74 	else
     75 		: $((KILLED_CNT += 1))
     76 	fi
     77 }
     78 
     79 # Validate the memory usage limit imposed by the hierarchically topmost group
     80 testcase_1()
     81 {
     82 	TST_COUNT=1
     83 	tst_resm TINFO "Test #1: Checking if the memory usage limit imposed by the topmost group is enforced"
     84 
     85 	echo "$ACTIVE_MEM_LIMIT" > $TST_PATH/mnt/$TST_NUM/memory.limit_in_bytes
     86 	echo "$TOT_MEM_LIMIT" > $TST_PATH/mnt/$TST_NUM/memory.memsw.limit_in_bytes
     87 
     88 	mkdir sub
     89 	(cd sub
     90 	KILLED_CNT=0
     91 	test_proc_kill
     92 
     93 	if [ $PROC_MEM -gt $TOT_MEM_LIMIT ] && [ $KILLED_CNT -eq 0 ]; then
     94 		result $FAIL "Test #1: failed"
     95 	else
     96 		result $PASS "Test #1: passed"
     97 	fi)
     98 	rmdir sub
     99 }
    100 
    101 # Record the test results
    102 #
    103 # $1: Result of the test case, $PASS or $FAIL
    104 # $2: Output information
    105 result()
    106 {
    107 	RES=$1
    108 	INFO=$2
    109 
    110 	if [ $RES -eq $PASS ]; then
    111 		tst_resm TPASS "$INFO"
    112 	else
    113 		: $((FAILED_CNT += 1))
    114 		tst_resm TFAIL "$INFO"
    115 	fi
    116 }
    117 
    118 cleanup()
    119 {
    120 	if [ -e $TST_PATH/mnt ]; then
    121 		umount $TST_PATH/mnt 2> /dev/null
    122 		rm -rf $TST_PATH/mnt
    123 	fi
    124 }
    125 
    126 do_mount()
    127 {
    128 	cleanup
    129 
    130 	mkdir $TST_PATH/mnt
    131 	mount -t cgroup -o memory cgroup $TST_PATH/mnt 2> /dev/null
    132 	if [ $? -ne 0 ]; then
    133 		tst_brkm TBROK NULL "Mounting cgroup to temp dir failed"
    134 		rmdir $TST_PATH/mnt
    135 		exit 1
    136 	fi
    137 }
    138 
    139 do_mount
    140 
    141 echo 1 > mnt/memory.use_hierarchy 2> /dev/null
    142 
    143 FAILED_CNT=0
    144 
    145 TST_NUM=1
    146 while [ $TST_NUM -le $TST_TOTAL ]; do
    147 	mkdir $TST_PATH/mnt/$TST_NUM
    148 	(cd $TST_PATH/mnt/$TST_NUM && testcase_$TST_NUM)
    149 	rmdir $TST_PATH/mnt/$TST_NUM
    150 	: $((TST_NUM += 1))
    151 done
    152 
    153 echo 0 > mnt/memory.use_hierarchy 2> /dev/null
    154 
    155 cleanup
    156 
    157 if [ "$FAILED_CNT" -ne 0 ]; then
    158 	tst_resm TFAIL "memcg_control: failed"
    159 	exit 1
    160 else
    161 	tst_resm TPASS "memcg_control: passed"
    162 	exit 0
    163 fi
    164