summaryrefslogtreecommitdiff
path: root/etc/setup.d/15killprocs
blob: 3b679f51a90e7eaba0a4355107aad36dea96097f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#!/bin/sh
# Copyright © 2007       Kees Cook <kees@outflux.net>
# Copyright © 2007-2009  Roger Leigh <rleigh@debian.org>
#
# schroot is free software: you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# schroot is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see
# <http://www.gnu.org/licenses/>.
#
#####################################################################

set -e

. "$SETUP_DATA_DIR/common-data"
. "$SETUP_DATA_DIR/common-functions"

if [ -f "$CHROOT_SCRIPT_CONFIG" ]; then
    . "$CHROOT_SCRIPT_CONFIG"
elif [ "$STATUS" = "ok" ]; then
    fatal "script-config file '$CHROOT_SCRIPT_CONFIG' does not exist"
fi

# Wrapper around kill command. Turns errors into
# warnings when running in verbose mode, otherwise
# it ignores them.
# args: parameters for kill
kill_proc()
{
    if ! kill "$@" 2>/dev/null; then
        info "kill $@ failed: process already terminated?"
    fi
}

# Kill all processes that were run from within the chroot environment
# $1: mount base location
do_kill_all()
{
    if [ -z "$1" ]; then
	fatal "No path for finding stray processes: not reaping processes in chroot"
    fi

    info "Killing processes run inside $1"
    ls /proc | egrep '^[[:digit:]]+$' |
    while read pid; do
	# Check if process root are the same device/inode as chroot
	# root (for efficiency)
        if [ /proc/"$pid"/root -ef "$1" ]; then
	    # Check if process and chroot root are the same (may be
	    # different even if device/inode match).
	    root=$(readlink /proc/"$pid"/root || true)
	    if [ "$root" = "$1" ]; then
		exe=$(readlink /proc/"$pid"/exe || true)
		info "Killing left-over pid $pid (${exe##$1})"
		info "  Sending SIGTERM to pid $pid"

		kill_proc -TERM "$pid"

		count=0
		max=5
		while [ -d /proc/"$pid" ]; do
                    count=$(( $count + 1 ))
                    info "  Waiting for pid $pid to shut down... ($count/$max)"
                    sleep 1
                # Wait for $max seconds for process to die before -9'ing it
                    if [ "$count" -eq "$max" ]; then
			info "  Sending SIGKILL to pid $pid"
			kill_proc -KILL "$pid"
			sleep 1
			break
                    fi
		done
            fi
	fi
    done
}

if [ $STAGE = "setup-recover" ] || [ $STAGE = "setup-stop" ]; then
    do_kill_all "$CHROOT_MOUNT_LOCATION"
fi