Statistics
| Branch: | Tag: | Revision:

root / daemons / daemon-util.in @ e37f47d3

History | View | Annotate | Download (6.7 kB)

1 f154a7a3 Michael Hanselmann
#!/bin/bash
2 f154a7a3 Michael Hanselmann
#
3 f154a7a3 Michael Hanselmann
4 5ad68a23 Michael Hanselmann
# Copyright (C) 2009, 2011, 2012 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 6e3bf290 Michael Hanselmann
@SHELL_ENV_INIT@
24 6e3bf290 Michael Hanselmann
25 6e3bf290 Michael Hanselmann
readonly defaults_file="$SYSCONFDIR/default/ganeti"
26 f154a7a3 Michael Hanselmann
27 5d8acbdd Michael Hanselmann
# This is a list of all daemons and the order in which they're started. The
28 5d8acbdd Michael Hanselmann
# order is important as there are dependencies between them. On shutdown,
29 5d8acbdd Michael Hanselmann
# they're stopped in reverse order.
30 5d8acbdd Michael Hanselmann
DAEMONS=(
31 5d8acbdd Michael Hanselmann
  ganeti-noded
32 5d8acbdd Michael Hanselmann
  ganeti-masterd
33 5d8acbdd Michael Hanselmann
  ganeti-rapi
34 5d8acbdd Michael Hanselmann
  )
35 5d8acbdd Michael Hanselmann
36 bb514a22 Michael Hanselmann
_confd_enabled() {
37 bb514a22 Michael Hanselmann
  [[ "@CUSTOM_ENABLE_CONFD@" == True ]]
38 bb514a22 Michael Hanselmann
}
39 bb514a22 Michael Hanselmann
40 bb514a22 Michael Hanselmann
if _confd_enabled; then
41 c4e5d11e Iustin Pop
  DAEMONS+=( ganeti-confd )
42 c4e5d11e Iustin Pop
fi
43 c4e5d11e Iustin Pop
44 f154a7a3 Michael Hanselmann
NODED_ARGS=
45 f154a7a3 Michael Hanselmann
MASTERD_ARGS=
46 f154a7a3 Michael Hanselmann
CONFD_ARGS=
47 f154a7a3 Michael Hanselmann
RAPI_ARGS=
48 f154a7a3 Michael Hanselmann
49 f154a7a3 Michael Hanselmann
# Read defaults file if it exists
50 f154a7a3 Michael Hanselmann
if [[ -s $defaults_file ]]; then
51 f154a7a3 Michael Hanselmann
  . $defaults_file
52 f154a7a3 Michael Hanselmann
fi
53 f154a7a3 Michael Hanselmann
54 07c7c94a Jun Futagawa
# Meant to facilitate use utilities in /etc/rc.d/init.d/functions in case
55 07c7c94a Jun Futagawa
# start-stop-daemon is not available.
56 07c7c94a Jun Futagawa
_ignore_error() {
57 07c7c94a Jun Futagawa
  eval "$@" || :
58 07c7c94a Jun Futagawa
}
59 07c7c94a Jun Futagawa
60 f154a7a3 Michael Hanselmann
_daemon_pidfile() {
61 6e3bf290 Michael Hanselmann
  echo "$RUN_DIR/$1.pid"
62 f154a7a3 Michael Hanselmann
}
63 f154a7a3 Michael Hanselmann
64 a34d8cc0 Michael Hanselmann
_daemon_executable() {
65 a34d8cc0 Michael Hanselmann
  echo "@PREFIX@/sbin/$1"
66 a34d8cc0 Michael Hanselmann
}
67 a34d8cc0 Michael Hanselmann
68 cbccd9ca René Nussbaumer
_daemon_usergroup() {
69 cbccd9ca René Nussbaumer
  case "$1" in
70 cbccd9ca René Nussbaumer
    masterd)
71 cbccd9ca René Nussbaumer
      echo "@GNTMASTERUSER@:@GNTMASTERDGROUP@"
72 cbccd9ca René Nussbaumer
      ;;
73 cbccd9ca René Nussbaumer
    confd)
74 cbccd9ca René Nussbaumer
      echo "@GNTCONFDUSER@:@GNTCONFDGROUP@"
75 cbccd9ca René Nussbaumer
      ;;
76 cbccd9ca René Nussbaumer
    rapi)
77 cbccd9ca René Nussbaumer
      echo "@GNTRAPIUSER@:@GNTRAPIGROUP@"
78 cbccd9ca René Nussbaumer
      ;;
79 cbccd9ca René Nussbaumer
    noded)
80 cbccd9ca René Nussbaumer
      echo "@GNTNODEDUSER@:@GNTDAEMONSGROUP@"
81 cbccd9ca René Nussbaumer
      ;;
82 cbccd9ca René Nussbaumer
    *)
83 cbccd9ca René Nussbaumer
      echo "root:@GNTDAEMONSGROUP@"
84 cbccd9ca René Nussbaumer
      ;;
85 cbccd9ca René Nussbaumer
  esac
86 cbccd9ca René Nussbaumer
}
87 cbccd9ca René Nussbaumer
88 d2baa21d Michael Hanselmann
# Checks whether the local machine is part of a cluster
89 d2baa21d Michael Hanselmann
check_config() {
90 6e3bf290 Michael Hanselmann
  local server_pem=$DATA_DIR/server.pem
91 d2baa21d Michael Hanselmann
  local fname
92 d2baa21d Michael Hanselmann
93 d2baa21d Michael Hanselmann
  for fname in $server_pem; do
94 d2baa21d Michael Hanselmann
    if [[ ! -f $fname ]]; then
95 d2baa21d Michael Hanselmann
      echo "Missing configuration file $fname" >&2
96 d2baa21d Michael Hanselmann
      return 1
97 d2baa21d Michael Hanselmann
    fi
98 d2baa21d Michael Hanselmann
  done
99 d2baa21d Michael Hanselmann
100 d2baa21d Michael Hanselmann
  return 0
101 d2baa21d Michael Hanselmann
}
102 d2baa21d Michael Hanselmann
103 10c5c1c3 Michael Hanselmann
# Checks the exit code of a daemon
104 10c5c1c3 Michael Hanselmann
check_exitcode() {
105 10c5c1c3 Michael Hanselmann
  if [[ "$#" -lt 1 ]]; then
106 10c5c1c3 Michael Hanselmann
    echo 'Missing exit code.' >&2
107 10c5c1c3 Michael Hanselmann
    return 1
108 10c5c1c3 Michael Hanselmann
  fi
109 10c5c1c3 Michael Hanselmann
110 10c5c1c3 Michael Hanselmann
  local rc="$1"; shift
111 10c5c1c3 Michael Hanselmann
112 10c5c1c3 Michael Hanselmann
  case "$rc" in
113 10c5c1c3 Michael Hanselmann
    0) ;;
114 10c5c1c3 Michael Hanselmann
    11)
115 10c5c1c3 Michael Hanselmann
      echo "not master"
116 10c5c1c3 Michael Hanselmann
    ;;
117 10c5c1c3 Michael Hanselmann
    *)
118 10c5c1c3 Michael Hanselmann
      echo "exit code $rc"
119 10c5c1c3 Michael Hanselmann
      return 1
120 10c5c1c3 Michael Hanselmann
    ;;
121 10c5c1c3 Michael Hanselmann
  esac
122 10c5c1c3 Michael Hanselmann
123 10c5c1c3 Michael Hanselmann
  return 0
124 10c5c1c3 Michael Hanselmann
}
125 10c5c1c3 Michael Hanselmann
126 8e2ed2e8 Michael Hanselmann
# Prints path to PID file for a daemon.
127 8e2ed2e8 Michael Hanselmann
daemon_pidfile() {
128 8e2ed2e8 Michael Hanselmann
  if [[ "$#" -lt 1 ]]; then
129 8e2ed2e8 Michael Hanselmann
    echo 'Missing daemon name.' >&2
130 8e2ed2e8 Michael Hanselmann
    return 1
131 8e2ed2e8 Michael Hanselmann
  fi
132 8e2ed2e8 Michael Hanselmann
133 8e2ed2e8 Michael Hanselmann
  local name="$1"; shift
134 8e2ed2e8 Michael Hanselmann
135 8e2ed2e8 Michael Hanselmann
  _daemon_pidfile $name
136 8e2ed2e8 Michael Hanselmann
}
137 8e2ed2e8 Michael Hanselmann
138 8e2ed2e8 Michael Hanselmann
# Prints path to daemon executable.
139 8e2ed2e8 Michael Hanselmann
daemon_executable() {
140 8e2ed2e8 Michael Hanselmann
  if [[ "$#" -lt 1 ]]; then
141 8e2ed2e8 Michael Hanselmann
    echo 'Missing daemon name.' >&2
142 8e2ed2e8 Michael Hanselmann
    return 1
143 8e2ed2e8 Michael Hanselmann
  fi
144 8e2ed2e8 Michael Hanselmann
145 8e2ed2e8 Michael Hanselmann
  local name="$1"; shift
146 8e2ed2e8 Michael Hanselmann
147 8e2ed2e8 Michael Hanselmann
  _daemon_executable $name
148 8e2ed2e8 Michael Hanselmann
}
149 8e2ed2e8 Michael Hanselmann
150 5d8acbdd Michael Hanselmann
# Prints a list of all daemons in the order in which they should be started
151 5d8acbdd Michael Hanselmann
list_start_daemons() {
152 5d8acbdd Michael Hanselmann
  local name
153 5d8acbdd Michael Hanselmann
154 5d8acbdd Michael Hanselmann
  for name in "${DAEMONS[@]}"; do
155 5d8acbdd Michael Hanselmann
    echo "$name"
156 5d8acbdd Michael Hanselmann
  done
157 5d8acbdd Michael Hanselmann
}
158 5d8acbdd Michael Hanselmann
159 5d8acbdd Michael Hanselmann
# Prints a list of all daemons in the order in which they should be stopped
160 5d8acbdd Michael Hanselmann
list_stop_daemons() {
161 5d8acbdd Michael Hanselmann
  list_start_daemons | tac
162 5d8acbdd Michael Hanselmann
}
163 5d8acbdd Michael Hanselmann
164 5d8acbdd Michael Hanselmann
# Checks whether a daemon name is known
165 5d8acbdd Michael Hanselmann
is_daemon_name() {
166 5d8acbdd Michael Hanselmann
  if [[ "$#" -lt 1 ]]; then
167 5d8acbdd Michael Hanselmann
    echo 'Missing daemon name.' >&2
168 5d8acbdd Michael Hanselmann
    return 1
169 5d8acbdd Michael Hanselmann
  fi
170 5d8acbdd Michael Hanselmann
171 5d8acbdd Michael Hanselmann
  local name="$1"; shift
172 5d8acbdd Michael Hanselmann
173 5d8acbdd Michael Hanselmann
  for i in "${DAEMONS[@]}"; do
174 5d8acbdd Michael Hanselmann
    if [[ "$i" == "$name" ]]; then
175 5d8acbdd Michael Hanselmann
      return 0
176 5d8acbdd Michael Hanselmann
    fi
177 5d8acbdd Michael Hanselmann
  done
178 5d8acbdd Michael Hanselmann
179 5d8acbdd Michael Hanselmann
  echo "Unknown daemon name '$name'" >&2
180 5d8acbdd Michael Hanselmann
  return 1
181 5d8acbdd Michael Hanselmann
}
182 5d8acbdd Michael Hanselmann
183 f154a7a3 Michael Hanselmann
# Checks whether daemon is running
184 f154a7a3 Michael Hanselmann
check() {
185 f154a7a3 Michael Hanselmann
  if [[ "$#" -lt 1 ]]; then
186 f154a7a3 Michael Hanselmann
    echo 'Missing daemon name.' >&2
187 49d43244 Michael Hanselmann
    return 1
188 f154a7a3 Michael Hanselmann
  fi
189 f154a7a3 Michael Hanselmann
190 f154a7a3 Michael Hanselmann
  local name="$1"; shift
191 07c7c94a Jun Futagawa
  local pidfile=$(_daemon_pidfile $name)
192 07c7c94a Jun Futagawa
  local daemonexec=$(_daemon_executable $name)
193 07c7c94a Jun Futagawa
194 07c7c94a Jun Futagawa
  if type -p start-stop-daemon >/dev/null; then
195 07c7c94a Jun Futagawa
    start-stop-daemon --stop --signal 0 --quiet \
196 07c7c94a Jun Futagawa
      --pidfile $pidfile
197 07c7c94a Jun Futagawa
  else
198 07c7c94a Jun Futagawa
    _ignore_error status \
199 07c7c94a Jun Futagawa
      -p $pidfile \
200 07c7c94a Jun Futagawa
      $daemonexec
201 07c7c94a Jun Futagawa
  fi
202 f154a7a3 Michael Hanselmann
}
203 f154a7a3 Michael Hanselmann
204 f154a7a3 Michael Hanselmann
# Starts a daemon
205 f154a7a3 Michael Hanselmann
start() {
206 f154a7a3 Michael Hanselmann
  if [[ "$#" -lt 1 ]]; then
207 f154a7a3 Michael Hanselmann
    echo 'Missing daemon name.' >&2
208 49d43244 Michael Hanselmann
    return 1
209 f154a7a3 Michael Hanselmann
  fi
210 f154a7a3 Michael Hanselmann
211 f154a7a3 Michael Hanselmann
  local name="$1"; shift
212 3c29e49d Iustin Pop
  # Convert daemon name to uppercase after removing "ganeti-" prefix
213 3c29e49d Iustin Pop
  local plain_name=${name#ganeti-}
214 3c29e49d Iustin Pop
  local ucname=$(tr a-z A-Z <<<$plain_name)
215 07c7c94a Jun Futagawa
  local pidfile=$(_daemon_pidfile $name)
216 07c7c94a Jun Futagawa
  local usergroup=$(_daemon_usergroup $plain_name)
217 07c7c94a Jun Futagawa
  local daemonexec=$(_daemon_executable $name)
218 f154a7a3 Michael Hanselmann
219 bb514a22 Michael Hanselmann
  if [[ "$name" == ganeti-confd ]] && ! _confd_enabled; then
220 c4e5d11e Iustin Pop
    echo 'ganeti-confd disabled at build time' >&2
221 c4e5d11e Iustin Pop
    return 1
222 c4e5d11e Iustin Pop
  fi
223 c4e5d11e Iustin Pop
224 f154a7a3 Michael Hanselmann
  # Read $<daemon>_ARGS and $EXTRA_<daemon>_ARGS
225 03956d39 Michael Hanselmann
  eval local args="\"\$${ucname}_ARGS \$EXTRA_${ucname}_ARGS\""
226 f154a7a3 Michael Hanselmann
227 46a8da3b René Nussbaumer
  @PKGLIBDIR@/ensure-dirs
228 46a8da3b René Nussbaumer
229 07c7c94a Jun Futagawa
  if type -p start-stop-daemon >/dev/null; then
230 07c7c94a Jun Futagawa
    start-stop-daemon --start --quiet --oknodo \
231 07c7c94a Jun Futagawa
      --pidfile $pidfile \
232 07c7c94a Jun Futagawa
      --startas $daemonexec \
233 07c7c94a Jun Futagawa
      --chuid $usergroup \
234 07c7c94a Jun Futagawa
      -- $args "$@"
235 07c7c94a Jun Futagawa
  else
236 07c7c94a Jun Futagawa
    # TODO: Find a way to start daemon with a group, until then the group must
237 07c7c94a Jun Futagawa
    # be removed
238 07c7c94a Jun Futagawa
    _ignore_error daemon \
239 07c7c94a Jun Futagawa
      --pidfile $pidfile \
240 07c7c94a Jun Futagawa
      --user ${usergroup%:*} \
241 07c7c94a Jun Futagawa
      $daemonexec $args "$@"
242 07c7c94a Jun Futagawa
  fi
243 f154a7a3 Michael Hanselmann
}
244 f154a7a3 Michael Hanselmann
245 f154a7a3 Michael Hanselmann
# Stops a daemon
246 f154a7a3 Michael Hanselmann
stop() {
247 f154a7a3 Michael Hanselmann
  if [[ "$#" -lt 1 ]]; then
248 f154a7a3 Michael Hanselmann
    echo 'Missing daemon name.' >&2
249 49d43244 Michael Hanselmann
    return 1
250 f154a7a3 Michael Hanselmann
  fi
251 f154a7a3 Michael Hanselmann
252 f154a7a3 Michael Hanselmann
  local name="$1"; shift
253 07c7c94a Jun Futagawa
  local pidfile=$(_daemon_pidfile $name)
254 f154a7a3 Michael Hanselmann
255 07c7c94a Jun Futagawa
  if type -p start-stop-daemon >/dev/null; then
256 07c7c94a Jun Futagawa
    start-stop-daemon --stop --quiet --oknodo --retry 30 \
257 07c7c94a Jun Futagawa
      --pidfile $pidfile
258 07c7c94a Jun Futagawa
  else
259 07c7c94a Jun Futagawa
    _ignore_error killproc -p $pidfile $name
260 07c7c94a Jun Futagawa
  fi
261 f154a7a3 Michael Hanselmann
}
262 f154a7a3 Michael Hanselmann
263 f154a7a3 Michael Hanselmann
# Starts a daemon if it's not yet running
264 f154a7a3 Michael Hanselmann
check_and_start() {
265 f154a7a3 Michael Hanselmann
  local name="$1"
266 f154a7a3 Michael Hanselmann
267 f154a7a3 Michael Hanselmann
  if ! check $name; then
268 f154a7a3 Michael Hanselmann
    start $name
269 f154a7a3 Michael Hanselmann
  fi
270 f154a7a3 Michael Hanselmann
}
271 f154a7a3 Michael Hanselmann
272 f154a7a3 Michael Hanselmann
# Starts the master role
273 f154a7a3 Michael Hanselmann
start_master() {
274 f154a7a3 Michael Hanselmann
  start ganeti-masterd
275 f154a7a3 Michael Hanselmann
  start ganeti-rapi
276 f154a7a3 Michael Hanselmann
}
277 f154a7a3 Michael Hanselmann
278 f154a7a3 Michael Hanselmann
# Stops the master role
279 f154a7a3 Michael Hanselmann
stop_master() {
280 f154a7a3 Michael Hanselmann
  stop ganeti-rapi
281 f154a7a3 Michael Hanselmann
  stop ganeti-masterd
282 f154a7a3 Michael Hanselmann
}
283 f154a7a3 Michael Hanselmann
284 2250afb3 Michael Hanselmann
# Start all daemons
285 2250afb3 Michael Hanselmann
start_all() {
286 2250afb3 Michael Hanselmann
  for i in $(list_start_daemons); do
287 2250afb3 Michael Hanselmann
    local rc=0
288 2250afb3 Michael Hanselmann
289 2250afb3 Michael Hanselmann
    # Try to start daemon
290 2250afb3 Michael Hanselmann
    start $i || rc=$?
291 2250afb3 Michael Hanselmann
292 2250afb3 Michael Hanselmann
    if ! errmsg=$(check_exitcode $rc); then
293 2250afb3 Michael Hanselmann
      echo "$errmsg" >&2
294 2250afb3 Michael Hanselmann
      return 1
295 2250afb3 Michael Hanselmann
    fi
296 2250afb3 Michael Hanselmann
  done
297 2250afb3 Michael Hanselmann
298 2250afb3 Michael Hanselmann
  return 0
299 2250afb3 Michael Hanselmann
}
300 2250afb3 Michael Hanselmann
301 2250afb3 Michael Hanselmann
# Stop all daemons
302 2250afb3 Michael Hanselmann
stop_all() {
303 2250afb3 Michael Hanselmann
  for i in $(list_stop_daemons); do
304 2250afb3 Michael Hanselmann
    stop $i
305 2250afb3 Michael Hanselmann
  done
306 2250afb3 Michael Hanselmann
}
307 2250afb3 Michael Hanselmann
308 7e1fac25 Michael Hanselmann
# Reloads the SSH keys
309 7e1fac25 Michael Hanselmann
reload_ssh_keys() {
310 7e1fac25 Michael Hanselmann
  @RPL_SSH_INITD_SCRIPT@ restart
311 7e1fac25 Michael Hanselmann
}
312 7e1fac25 Michael Hanselmann
313 07c7c94a Jun Futagawa
# Read @SYSCONFDIR@/rc.d/init.d/functions if start-stop-daemon not available
314 07c7c94a Jun Futagawa
if ! type -p start-stop-daemon >/dev/null && \
315 07c7c94a Jun Futagawa
   [[ -f @SYSCONFDIR@/rc.d/init.d/functions ]]; then
316 07c7c94a Jun Futagawa
  _ignore_error . @SYSCONFDIR@/rc.d/init.d/functions
317 07c7c94a Jun Futagawa
fi
318 07c7c94a Jun Futagawa
319 f154a7a3 Michael Hanselmann
if [[ "$#" -lt 1 ]]; then
320 f154a7a3 Michael Hanselmann
  echo "Usage: $0 <action>" >&2
321 f154a7a3 Michael Hanselmann
  exit 1
322 f154a7a3 Michael Hanselmann
fi
323 f154a7a3 Michael Hanselmann
324 f154a7a3 Michael Hanselmann
orig_action=$1; shift
325 f154a7a3 Michael Hanselmann
326 7289c16e Michael Hanselmann
if [[ "$orig_action" == *_* ]]; then
327 7289c16e Michael Hanselmann
  echo "Command must not contain underscores" >&2
328 7289c16e Michael Hanselmann
  exit 1
329 7289c16e Michael Hanselmann
fi
330 7289c16e Michael Hanselmann
331 f154a7a3 Michael Hanselmann
# Replace all dashes (-) with underlines (_)
332 f154a7a3 Michael Hanselmann
action=${orig_action//-/_}
333 f154a7a3 Michael Hanselmann
334 f154a7a3 Michael Hanselmann
# Is it a known function?
335 f154a7a3 Michael Hanselmann
if ! declare -F "$action" >/dev/null 2>&1; then
336 f154a7a3 Michael Hanselmann
  echo "Unknown command: $orig_action" >&2
337 f154a7a3 Michael Hanselmann
  exit 1
338 f154a7a3 Michael Hanselmann
fi
339 f154a7a3 Michael Hanselmann
340 f154a7a3 Michael Hanselmann
# Call handler function
341 f154a7a3 Michael Hanselmann
$action "$@"