plc.d/functions

   1 # -*-Shell-script-*-
   2 #
   3 # Common functions for PLC startup/shutdown scripts
   4 #
   5 # Mark Huang <mlhuang@cs.princeton.edu>
   6 # Copyright (C) 2006 The Trustees of Princeton University
   7 #
   8 # $Id: functions,v 1.1 2006/04/06 21:51:59 mlhuang Exp $
   9 #
  10
  11 export PATH=/sbin:/bin:/usr/bin:/usr/sbin
  12
  13 # Source function library
  14 . /etc/init.d/functions
  15
  16 # Source configuration
  17 . /etc/planetlab/plc_config
  18
  19 # Total number of errors
  20 ERRORS=0
  21
  22 # Count the exit status of the last command
  23 check ()
  24 {
  25     ERRORS=$(($ERRORS+$?))
  26 }
  27
  28 # Print status header
  29 dialog ()
  30 {
  31     echo -n "PLC: $*: " >&3
  32 }
  33
  34 # Print result
  35 result ()
  36 {
  37     if [ $ERRORS -eq 0 ] ; then
  38         success "$*" >&3
  39     else
  40         failure "$*" >&3
  41     fi
  42     echo >&3
  43 }
  44
  45 # Start up a program with a plc_ prefix
  46 plc_daemon ()
  47 {
  48     base=${1##*/}
  49
  50     # See if it's already running. Look *only* at the pid file.
  51     if [ -f /var/run/plc_${base}.pid ]; then
  52         local line p
  53         read line < /var/run/plc_${base}.pid
  54         for p in $line ; do
  55                 [ -z "${p//[0-9]/}" -a -d "/proc/$p" ] && pid="$pid $p"
  56         done
  57     fi
  58
  59     [ -n "${pid:-}" -a -z "${force:-}" ] && return
  60
  61     # And start it up.
  62     (exec -a plc_${base} $*)
  63     ret=$?
  64
  65     if [ -f /var/run/${base}.pid ] ; then
  66         mv /var/run/${base}.pid /var/run/plc_${base}.pid
  67     fi
  68
  69     return $ret
  70 }
  71
  72 # Print IP address of hostname if resolvable
  73 gethostbyname ()
  74 {
  75     perl -MSocket -e '($a,$b,$c,$d,@addrs) = gethostbyname($ARGV[0]); print inet_ntoa($addrs[0]) . "\n";' $1 2>/dev/null
  76 }
  77
  78 # Print the CNAME of an SSL certificate
  79 ssl_cname ()
  80 {
  81     openssl x509 -noout -in $1 -subject | \
  82         sed -n -e 's@.*/CN=\([^/]*\).*@\1@p'
  83 }
  84
  85 # Forcefully make a symlink
  86 symlink ()
  87 {
  88     mkdir -p $(dirname $2)
  89     rm -f $2
  90     ln -s $1 $2
  91 }
  92
  93 # Make copies of stdout and stderr. The plc initscript redirects
  94 # stdout and stderr to a logfile if -v is not specified.
  95 [ ! -e /dev/fd/3 ] && exec 3>&1
  96 [ ! -e /dev/fd/4 ] && exec 4>&2