blob: 08579aaf95228dee507691ba620975d673edd8c8 (
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
#!/bin/sh
#
# shutdown -- wrapper script to guard against accidental shutdowns
#
# Copyright © martin f. krafft <madduck@madduck.net>
# Released under the terms of the Artistic Licence 2.0
#
set -eu
ME=molly-guard
VERSION=0.4
SCRIPTSDIR="@cfgdir@/run.d"
CMD="${0##*/}"
EXEC="@REALPATH@/$CMD"
case "$CMD" in
halt|reboot|shutdown|poweroff|coldreboot|pm-hibernate|pm-suspend|pm-suspend-hybrid)
if [ ! -f $EXEC ]; then
echo "E: not a regular file: $EXEC" >&2
exit 4
fi
if [ ! -x $EXEC ]; then
echo "E: not an executable: $EXEC" >&2
exit 3
fi
;;
*)
echo "E: unsupported command: $CMD" >&2
exit 1
;;
esac
usage()
{
cat <<-_eousage
Usage: $ME [options] [-- script options]
(shielding $EXEC)
molly-guard's primary goal is to guard against accidental
shutdowns/reboots. $ME will run all scripts in $SCRIPTSDIR and only
invokes $EXEC if all scripts exited successfully.
Specifying --molly-guard-do-nothing as argument to the command will
make $ME echo the command it would execute rather than actually
executing it.
Options following the double hyphen will be passed unchanged to the
scripts.
Please see molly-guard(8) for more information.
The actual command's help output follows:
_eousage
}
CMDARGS=
SCRIPTARGS=
END_OF_ARGS=0
DO_NOTHING=0
for arg in "$@"; do
case "$arg" in
(*-molly-guard-do-nothing) DO_NOTHING=1;;
(*-help)
usage 2>&1
eval $EXEC --help 2>&1
exit 0
;;
--) END_OF_ARGS=1;;
*\"*)
echo 'E: cannot use double-quotes (") in arguments' >&2
exit 1
;;
*)
if [ $END_OF_ARGS -eq 0 ]; then
CMDARGS="${CMDARGS:+$CMDARGS }\"$arg\""
else
SCRIPTARGS="${SCRIPTARGS:+$SCRIPTARGS }--arg \"$arg\""
fi
;;
esac
done
do_real_cmd()
{
if [ $DO_NOTHING -eq 1 ]; then
echo "$ME: would run: $EXEC $CMDARGS"
exit 0
else
eval exec $EXEC "$CMDARGS"
fi
}
if [ $DO_NOTHING -eq 1 ]; then
echo "I: demo mode; $ME will not do anything due to --molly-guard-do-nothing." >&2
fi
if [ -n "${MOLLYGUARD_CMD:-}" ]; then
do_real_cmd
fi
MOLLYGUARD_CMD=$CMD; export MOLLYGUARD_CMD
MOLLYGUARD_DO_NOTHING=$DO_NOTHING; export MOLLYGUARD_DO_NOTHING
MOLLYGUARD_SETTINGS="@cfgdir@/rc"; export MOLLYGUARD_SETTINGS
# pass through certain commands
case "$CMD $CMDARGS" in
(*shutdown\ *-c*|*halt\ *-w*|*halt\ *-f*|*reboot\ *-f*)
# allow canceling shutdowns, only write wtmp and force immediate halt
echo "I: executing $CMD $CMDARGS regardless of check results." >&2
do_real_cmd
;;
esac
for script in $(run-parts --test $SCRIPTSDIR); do
ret=0
eval $script $SCRIPTARGS || ret=$?
if [ $ret -ne 0 ]; then
echo "W: aborting $CMD due to ${script##*/} exiting with code $ret." >&2
exit $ret
fi
done
do_real_cmd
|
