root / daemons / daemon-util.in @ 49d43244
History | View | Annotate | Download (4.5 kB)
1 | f154a7a3 | Michael Hanselmann | #!/bin/bash |
---|---|---|---|
2 | f154a7a3 | Michael Hanselmann | # |
3 | f154a7a3 | Michael Hanselmann | |
4 | f154a7a3 | Michael Hanselmann | # Copyright (C) 2009 Google Inc. |
5 | f154a7a3 | Michael Hanselmann | # |
6 | f154a7a3 | Michael Hanselmann | # This program is free software; you can redistribute it and/or modify |
7 | f154a7a3 | Michael Hanselmann | # it under the terms of the GNU General Public License as published by |
8 | f154a7a3 | Michael Hanselmann | # the Free Software Foundation; either version 2 of the License, or |
9 | f154a7a3 | Michael Hanselmann | # (at your option) any later version. |
10 | f154a7a3 | Michael Hanselmann | # |
11 | f154a7a3 | Michael Hanselmann | # This program is distributed in the hope that it will be useful, but |
12 | f154a7a3 | Michael Hanselmann | # WITHOUT ANY WARRANTY; without even the implied warranty of |
13 | f154a7a3 | Michael Hanselmann | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
14 | f154a7a3 | Michael Hanselmann | # General Public License for more details. |
15 | f154a7a3 | Michael Hanselmann | # |
16 | f154a7a3 | Michael Hanselmann | # You should have received a copy of the GNU General Public License |
17 | f154a7a3 | Michael Hanselmann | # along with this program; if not, write to the Free Software |
18 | f154a7a3 | Michael Hanselmann | # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA |
19 | f154a7a3 | Michael Hanselmann | # 02110-1301, USA. |
20 | f154a7a3 | Michael Hanselmann | |
21 | f154a7a3 | Michael Hanselmann | set -e |
22 | f154a7a3 | Michael Hanselmann | |
23 | f154a7a3 | Michael Hanselmann | defaults_file=@SYSCONFDIR@/default/ganeti |
24 | f154a7a3 | Michael Hanselmann | |
25 | 5d8acbdd | Michael Hanselmann | # This is a list of all daemons and the order in which they're started. The |
26 | 5d8acbdd | Michael Hanselmann | # order is important as there are dependencies between them. On shutdown, |
27 | 5d8acbdd | Michael Hanselmann | # they're stopped in reverse order. |
28 | 5d8acbdd | Michael Hanselmann | DAEMONS=( |
29 | 5d8acbdd | Michael Hanselmann | ganeti-noded |
30 | 5d8acbdd | Michael Hanselmann | ganeti-masterd |
31 | 5d8acbdd | Michael Hanselmann | ganeti-rapi |
32 | 5d8acbdd | Michael Hanselmann | ganeti-confd |
33 | 5d8acbdd | Michael Hanselmann | ) |
34 | 5d8acbdd | Michael Hanselmann | |
35 | f154a7a3 | Michael Hanselmann | NODED_ARGS= |
36 | f154a7a3 | Michael Hanselmann | MASTERD_ARGS= |
37 | f154a7a3 | Michael Hanselmann | CONFD_ARGS= |
38 | f154a7a3 | Michael Hanselmann | RAPI_ARGS= |
39 | f154a7a3 | Michael Hanselmann | |
40 | f154a7a3 | Michael Hanselmann | # Read defaults file if it exists |
41 | f154a7a3 | Michael Hanselmann | if [[ -s $defaults_file ]]; then |
42 | f154a7a3 | Michael Hanselmann | . $defaults_file |
43 | f154a7a3 | Michael Hanselmann | fi |
44 | f154a7a3 | Michael Hanselmann | |
45 | f154a7a3 | Michael Hanselmann | _daemon_pidfile() { |
46 | f154a7a3 | Michael Hanselmann | echo "@LOCALSTATEDIR@/run/ganeti/$1.pid" |
47 | f154a7a3 | Michael Hanselmann | } |
48 | f154a7a3 | Michael Hanselmann | |
49 | d2baa21d | Michael Hanselmann | # Checks whether the local machine is part of a cluster |
50 | d2baa21d | Michael Hanselmann | check_config() { |
51 | d2baa21d | Michael Hanselmann | local server_pem=@LOCALSTATEDIR@/lib/ganeti/server.pem |
52 | d2baa21d | Michael Hanselmann | local fname |
53 | d2baa21d | Michael Hanselmann | |
54 | d2baa21d | Michael Hanselmann | for fname in $server_pem; do |
55 | d2baa21d | Michael Hanselmann | if [[ ! -f $fname ]]; then |
56 | d2baa21d | Michael Hanselmann | echo "Missing configuration file $fname" >&2 |
57 | d2baa21d | Michael Hanselmann | return 1 |
58 | d2baa21d | Michael Hanselmann | fi |
59 | d2baa21d | Michael Hanselmann | done |
60 | d2baa21d | Michael Hanselmann | |
61 | d2baa21d | Michael Hanselmann | return 0 |
62 | d2baa21d | Michael Hanselmann | } |
63 | d2baa21d | Michael Hanselmann | |
64 | 10c5c1c3 | Michael Hanselmann | # Checks the exit code of a daemon |
65 | 10c5c1c3 | Michael Hanselmann | check_exitcode() { |
66 | 10c5c1c3 | Michael Hanselmann | if [[ "$#" -lt 1 ]]; then |
67 | 10c5c1c3 | Michael Hanselmann | echo 'Missing exit code.' >&2 |
68 | 10c5c1c3 | Michael Hanselmann | return 1 |
69 | 10c5c1c3 | Michael Hanselmann | fi |
70 | 10c5c1c3 | Michael Hanselmann | |
71 | 10c5c1c3 | Michael Hanselmann | local rc="$1"; shift |
72 | 10c5c1c3 | Michael Hanselmann | |
73 | 10c5c1c3 | Michael Hanselmann | case "$rc" in |
74 | 10c5c1c3 | Michael Hanselmann | 0) ;; |
75 | 10c5c1c3 | Michael Hanselmann | 11) |
76 | 10c5c1c3 | Michael Hanselmann | echo "not master" |
77 | 10c5c1c3 | Michael Hanselmann | ;; |
78 | 10c5c1c3 | Michael Hanselmann | *) |
79 | 10c5c1c3 | Michael Hanselmann | echo "exit code $rc" |
80 | 10c5c1c3 | Michael Hanselmann | return 1 |
81 | 10c5c1c3 | Michael Hanselmann | ;; |
82 | 10c5c1c3 | Michael Hanselmann | esac |
83 | 10c5c1c3 | Michael Hanselmann | |
84 | 10c5c1c3 | Michael Hanselmann | return 0 |
85 | 10c5c1c3 | Michael Hanselmann | } |
86 | 10c5c1c3 | Michael Hanselmann | |
87 | 5d8acbdd | Michael Hanselmann | # Prints a list of all daemons in the order in which they should be started |
88 | 5d8acbdd | Michael Hanselmann | list_start_daemons() { |
89 | 5d8acbdd | Michael Hanselmann | local name |
90 | 5d8acbdd | Michael Hanselmann | |
91 | 5d8acbdd | Michael Hanselmann | for name in "${DAEMONS[@]}"; do |
92 | 5d8acbdd | Michael Hanselmann | echo "$name" |
93 | 5d8acbdd | Michael Hanselmann | done |
94 | 5d8acbdd | Michael Hanselmann | } |
95 | 5d8acbdd | Michael Hanselmann | |
96 | 5d8acbdd | Michael Hanselmann | # Prints a list of all daemons in the order in which they should be stopped |
97 | 5d8acbdd | Michael Hanselmann | list_stop_daemons() { |
98 | 5d8acbdd | Michael Hanselmann | list_start_daemons | tac |
99 | 5d8acbdd | Michael Hanselmann | } |
100 | 5d8acbdd | Michael Hanselmann | |
101 | 5d8acbdd | Michael Hanselmann | # Checks whether a daemon name is known |
102 | 5d8acbdd | Michael Hanselmann | is_daemon_name() { |
103 | 5d8acbdd | Michael Hanselmann | if [[ "$#" -lt 1 ]]; then |
104 | 5d8acbdd | Michael Hanselmann | echo 'Missing daemon name.' >&2 |
105 | 5d8acbdd | Michael Hanselmann | return 1 |
106 | 5d8acbdd | Michael Hanselmann | fi |
107 | 5d8acbdd | Michael Hanselmann | |
108 | 5d8acbdd | Michael Hanselmann | local name="$1"; shift |
109 | 5d8acbdd | Michael Hanselmann | |
110 | 5d8acbdd | Michael Hanselmann | for i in "${DAEMONS[@]}"; do |
111 | 5d8acbdd | Michael Hanselmann | if [[ "$i" == "$name" ]]; then |
112 | 5d8acbdd | Michael Hanselmann | return 0 |
113 | 5d8acbdd | Michael Hanselmann | fi |
114 | 5d8acbdd | Michael Hanselmann | done |
115 | 5d8acbdd | Michael Hanselmann | |
116 | 5d8acbdd | Michael Hanselmann | echo "Unknown daemon name '$name'" >&2 |
117 | 5d8acbdd | Michael Hanselmann | return 1 |
118 | 5d8acbdd | Michael Hanselmann | } |
119 | 5d8acbdd | Michael Hanselmann | |
120 | f154a7a3 | Michael Hanselmann | # Checks whether daemon is running |
121 | f154a7a3 | Michael Hanselmann | check() { |
122 | f154a7a3 | Michael Hanselmann | if [[ "$#" -lt 1 ]]; then |
123 | f154a7a3 | Michael Hanselmann | echo 'Missing daemon name.' >&2 |
124 | 49d43244 | Michael Hanselmann | return 1 |
125 | f154a7a3 | Michael Hanselmann | fi |
126 | f154a7a3 | Michael Hanselmann | |
127 | f154a7a3 | Michael Hanselmann | local name="$1"; shift |
128 | f154a7a3 | Michael Hanselmann | |
129 | f154a7a3 | Michael Hanselmann | start-stop-daemon --stop --signal 0 --quiet \ |
130 | f154a7a3 | Michael Hanselmann | --pidfile $(_daemon_pidfile $name) |
131 | f154a7a3 | Michael Hanselmann | } |
132 | f154a7a3 | Michael Hanselmann | |
133 | f154a7a3 | Michael Hanselmann | # Starts a daemon |
134 | f154a7a3 | Michael Hanselmann | start() { |
135 | f154a7a3 | Michael Hanselmann | if [[ "$#" -lt 1 ]]; then |
136 | f154a7a3 | Michael Hanselmann | echo 'Missing daemon name.' >&2 |
137 | 49d43244 | Michael Hanselmann | return 1 |
138 | f154a7a3 | Michael Hanselmann | fi |
139 | f154a7a3 | Michael Hanselmann | |
140 | f154a7a3 | Michael Hanselmann | local name="$1"; shift |
141 | f154a7a3 | Michael Hanselmann | |
142 | f154a7a3 | Michael Hanselmann | # Convert daemon name to uppercase after removing "ganeti-" prefix |
143 | 8645ff83 | Iustin Pop | local ucname=$(echo ${name#ganeti-} | tr a-z A-Z) |
144 | f154a7a3 | Michael Hanselmann | |
145 | f154a7a3 | Michael Hanselmann | # Read $<daemon>_ARGS and $EXTRA_<daemon>_ARGS |
146 | 03956d39 | Michael Hanselmann | eval local args="\"\$${ucname}_ARGS \$EXTRA_${ucname}_ARGS\"" |
147 | f154a7a3 | Michael Hanselmann | |
148 | f154a7a3 | Michael Hanselmann | start-stop-daemon --start --quiet --oknodo \ |
149 | f154a7a3 | Michael Hanselmann | --pidfile $(_daemon_pidfile $name) \ |
150 | f154a7a3 | Michael Hanselmann | --startas "@PREFIX@/sbin/$name" \ |
151 | f154a7a3 | Michael Hanselmann | -- $args "$@" |
152 | f154a7a3 | Michael Hanselmann | } |
153 | f154a7a3 | Michael Hanselmann | |
154 | f154a7a3 | Michael Hanselmann | # Stops a daemon |
155 | f154a7a3 | Michael Hanselmann | stop() { |
156 | f154a7a3 | Michael Hanselmann | if [[ "$#" -lt 1 ]]; then |
157 | f154a7a3 | Michael Hanselmann | echo 'Missing daemon name.' >&2 |
158 | 49d43244 | Michael Hanselmann | return 1 |
159 | f154a7a3 | Michael Hanselmann | fi |
160 | f154a7a3 | Michael Hanselmann | |
161 | f154a7a3 | Michael Hanselmann | local name="$1"; shift |
162 | f154a7a3 | Michael Hanselmann | |
163 | f154a7a3 | Michael Hanselmann | start-stop-daemon --stop --quiet --oknodo --retry 30 \ |
164 | f154a7a3 | Michael Hanselmann | --pidfile $(_daemon_pidfile $name) |
165 | f154a7a3 | Michael Hanselmann | } |
166 | f154a7a3 | Michael Hanselmann | |
167 | f154a7a3 | Michael Hanselmann | # Starts a daemon if it's not yet running |
168 | f154a7a3 | Michael Hanselmann | check_and_start() { |
169 | f154a7a3 | Michael Hanselmann | local name="$1" |
170 | f154a7a3 | Michael Hanselmann | |
171 | f154a7a3 | Michael Hanselmann | if ! check $name; then |
172 | f154a7a3 | Michael Hanselmann | start $name |
173 | f154a7a3 | Michael Hanselmann | fi |
174 | f154a7a3 | Michael Hanselmann | } |
175 | f154a7a3 | Michael Hanselmann | |
176 | f154a7a3 | Michael Hanselmann | # Starts the master role |
177 | f154a7a3 | Michael Hanselmann | start_master() { |
178 | f154a7a3 | Michael Hanselmann | start ganeti-masterd |
179 | f154a7a3 | Michael Hanselmann | start ganeti-rapi |
180 | f154a7a3 | Michael Hanselmann | } |
181 | f154a7a3 | Michael Hanselmann | |
182 | f154a7a3 | Michael Hanselmann | # Stops the master role |
183 | f154a7a3 | Michael Hanselmann | stop_master() { |
184 | f154a7a3 | Michael Hanselmann | stop ganeti-rapi |
185 | f154a7a3 | Michael Hanselmann | stop ganeti-masterd |
186 | f154a7a3 | Michael Hanselmann | } |
187 | f154a7a3 | Michael Hanselmann | |
188 | 2250afb3 | Michael Hanselmann | # Start all daemons |
189 | 2250afb3 | Michael Hanselmann | start_all() { |
190 | 2250afb3 | Michael Hanselmann | for i in $(list_start_daemons); do |
191 | 2250afb3 | Michael Hanselmann | local rc=0 |
192 | 2250afb3 | Michael Hanselmann | |
193 | 2250afb3 | Michael Hanselmann | # Try to start daemon |
194 | 2250afb3 | Michael Hanselmann | start $i || rc=$? |
195 | 2250afb3 | Michael Hanselmann | |
196 | 2250afb3 | Michael Hanselmann | if ! errmsg=$(check_exitcode $rc); then |
197 | 2250afb3 | Michael Hanselmann | echo "$errmsg" >&2 |
198 | 2250afb3 | Michael Hanselmann | return 1 |
199 | 2250afb3 | Michael Hanselmann | fi |
200 | 2250afb3 | Michael Hanselmann | done |
201 | 2250afb3 | Michael Hanselmann | |
202 | 2250afb3 | Michael Hanselmann | return 0 |
203 | 2250afb3 | Michael Hanselmann | } |
204 | 2250afb3 | Michael Hanselmann | |
205 | 2250afb3 | Michael Hanselmann | # Stop all daemons |
206 | 2250afb3 | Michael Hanselmann | stop_all() { |
207 | 2250afb3 | Michael Hanselmann | for i in $(list_stop_daemons); do |
208 | 2250afb3 | Michael Hanselmann | stop $i |
209 | 2250afb3 | Michael Hanselmann | done |
210 | 2250afb3 | Michael Hanselmann | } |
211 | 2250afb3 | Michael Hanselmann | |
212 | 7e1fac25 | Michael Hanselmann | # Reloads the SSH keys |
213 | 7e1fac25 | Michael Hanselmann | reload_ssh_keys() { |
214 | 7e1fac25 | Michael Hanselmann | @RPL_SSH_INITD_SCRIPT@ restart |
215 | 7e1fac25 | Michael Hanselmann | } |
216 | 7e1fac25 | Michael Hanselmann | |
217 | f154a7a3 | Michael Hanselmann | if [[ "$#" -lt 1 ]]; then |
218 | f154a7a3 | Michael Hanselmann | echo "Usage: $0 <action>" >&2 |
219 | f154a7a3 | Michael Hanselmann | exit 1 |
220 | f154a7a3 | Michael Hanselmann | fi |
221 | f154a7a3 | Michael Hanselmann | |
222 | f154a7a3 | Michael Hanselmann | orig_action=$1; shift |
223 | f154a7a3 | Michael Hanselmann | |
224 | 7289c16e | Michael Hanselmann | if [[ "$orig_action" == *_* ]]; then |
225 | 7289c16e | Michael Hanselmann | echo "Command must not contain underscores" >&2 |
226 | 7289c16e | Michael Hanselmann | exit 1 |
227 | 7289c16e | Michael Hanselmann | fi |
228 | 7289c16e | Michael Hanselmann | |
229 | f154a7a3 | Michael Hanselmann | # Replace all dashes (-) with underlines (_) |
230 | f154a7a3 | Michael Hanselmann | action=${orig_action//-/_} |
231 | f154a7a3 | Michael Hanselmann | |
232 | f154a7a3 | Michael Hanselmann | # Is it a known function? |
233 | f154a7a3 | Michael Hanselmann | if ! declare -F "$action" >/dev/null 2>&1; then |
234 | f154a7a3 | Michael Hanselmann | echo "Unknown command: $orig_action" >&2 |
235 | f154a7a3 | Michael Hanselmann | exit 1 |
236 | f154a7a3 | Michael Hanselmann | fi |
237 | f154a7a3 | Michael Hanselmann | |
238 | f154a7a3 | Michael Hanselmann | # Call handler function |
239 | f154a7a3 | Michael Hanselmann | $action "$@" |