1# 2# Copyright 2022 Mateusz Piotrowski <0mp@FreeBSD.org> 3# 4# SPDX-License-Identifier: BSD-2-Clause-FreeBSD 5# 6# Redistribution and use in source and binary forms, with or without 7# modification, are permitted provided that the following conditions 8# are met: 9# 1. Redistributions of source code must retain the above copyright 10# notice, this list of conditions and the following disclaimer. 11# 2. Redistributions in binary form must reproduce the above copyright 12# notice, this list of conditions and the following disclaimer in the 13# documentation and/or other materials provided with the distribution. 14# 15# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 16# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 18# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 19# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 21# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25# SUCH DAMAGE. 26# 27 28atf_test_case oomprotect_all 29oomprotect_all_head() 30{ 31 atf_set "descr" "Verify that \${name}_oomprotect=all protects " \ 32 "the command and all its current and future children" 33 atf_set "require.user" "root" # For protect(1). 34} 35 36oomprotect_all_body() 37{ 38 __name="$(atf_get ident)" 39 __pidfile="$(mktemp -t "${__name}.pid")" 40 __childpidfile="$(mktemp -t "${__name}.childpid")" 41 __script=$(mktemp -t "${__name}.script") 42 43 cat >> "$__script" <<-'LITERAL' 44 . /etc/rc.subr 45 name="$1" 46 pidfile="$2" 47 _childpidfile="$3" 48 _rc_arg="$4" 49 setvar "${name}_oomprotect" all 50 command="/usr/sbin/daemon" 51 command_args="-P $pidfile -p $_childpidfile -- /bin/sleep 5" 52 run_rc_command "$_rc_arg" 53 LITERAL 54 55 atf_check -s exit:0 -o inline:"Starting ${__name}.\n" -e empty \ 56 /bin/sh "$__script" "$__name" "$__pidfile" "$__childpidfile" onestart 57 atf_check -s exit:0 -o match:'^..1..... .......1$' -e empty \ 58 ps -p "$(cat "$__pidfile")" -ax -o flags,flags2 59 atf_check -s exit:0 -o match:'^..1..... .......1$' -e empty \ 60 ps -p "$(cat "$__childpidfile")" -ax -o flags,flags2 61 atf_check -s exit:0 -o ignore -e empty \ 62 /bin/sh "$__script" "$__name" "$__pidfile" "$__childpidfile" onestop 63} 64 65atf_test_case oomprotect_yes 66oomprotect_yes_head() 67{ 68 atf_set "descr" "Verify that \${name}_oomprotect=yes protects " \ 69 "the command but not its children" 70 atf_set "require.user" "root" # For protect(1). 71} 72 73oomprotect_yes_body() 74{ 75 __name="$(atf_get ident)" 76 __pidfile="$(mktemp -t "${__name}.pid")" 77 __script=$(mktemp -t "${__name}.script") 78 79 cat >> "$__script" <<-'LITERAL' 80 . /etc/rc.subr 81 name="$1" 82 pidfile="$2" 83 _rc_arg="$3" 84 setvar "${name}_oomprotect" yes 85 procname="/bin/sleep" 86 command="/usr/sbin/daemon" 87 command_args="-p $pidfile -- $procname 5" 88 run_rc_command "$_rc_arg" 89 LITERAL 90 91 atf_check -s exit:0 -o inline:"Starting ${__name}.\n" -e empty \ 92 /bin/sh "$__script" "$__name" "$__pidfile" onestart 93 atf_check -s exit:0 -o match:'^..1..... .......0$' -e empty \ 94 ps -p "$(cat "$__pidfile")" -ax -o flags,flags2 95 atf_check -s exit:0 -o ignore -e empty \ 96 /bin/sh "$__script" "$__name" "$__pidfile" onestop 97} 98 99atf_init_test_cases() 100{ 101 atf_add_test_case oomprotect_all 102 atf_add_test_case oomprotect_yes 103} 104