plc.d/functions

   1 # -*-Shell-script-*-
   2 # Common functions for PLC startup/shutdown scripts
   3 #
   4 # Mark Huang <mlhuang@cs.princeton.edu>
   5 # Copyright (C) 2006 The Trustees of Princeton University
   6 #
   7
   8 export PATH=/sbin:/bin:/usr/bin:/usr/sbin
   9
  10 # Source function library
  11 . /etc/init.d/functions
  12
  13 # Total number of errors
  14 ERRORS=0
  15
  16 # Count the exit status of the last command
  17 function check () {
  18     ERRORS=$(($ERRORS+$?))
  19 }
  20
  21 # Print status header
  22 function dialog () {
  23     echo -n "PLC: $*: " >&3
  24 }
  25
  26 # Print result
  27 function result () {
  28     if [ $ERRORS -eq 0 ] ; then
  29         success "$*" >&3
  30     else
  31         failure "$*" >&3
  32     fi
  33     echo >&3
  34 }
  35
  36 # Start up a program with a plc_ prefix
  37 function plc_daemon () {
  38     base=${1##*/}
  39
  40     # See if it's already running. Look *only* at the pid file.
  41     if [ -f /var/run/plc_${base}.pid ]; then
  42         local line p
  43         read line < /var/run/plc_${base}.pid
  44         for p in $line ; do
  45                 [ -z "${p//[0-9]/}" -a -d "/proc/$p" ] && pid="$pid $p"
  46         done
  47     fi
  48
  49     [ -n "${pid:-}" -a -z "${force:-}" ] && return
  50
  51     # And start it up.
  52     # Thierry -- June 18 2007
  53     # when invoking, e.g. service plc start httpd from an ssh connection
  54     # ssh stupidly hangs when everything is done
  55     # it turns out the forked ssh daemon exhibits the following stack at that point
  56     # (gdb) where
  57     # #0  0x001d6402 in __kernel_vsyscall ()
  58     # #1  0x003c2e7d in ___newselect_nocancel () from /lib/libc.so.6
  59     # #2  0x004387b4 in main () from /usr/sbin/sshd
  60     # So I figured the various file descriptors used were not properly closed
  61     (exec 3>&- 4>&- ; exec -a plc_${base} $*)
  62     ret=$?
  63
  64     if [ -f /var/run/${base}.pid ] ; then
  65         mv /var/run/${base}.pid /var/run/plc_${base}.pid
  66     fi
  67
  68     return $ret
  69 }
  70
  71 # Print IP address of hostname if resolvable
  72 function gethostbyname () {
  73     python -c 'import socket; import sys; print socket.gethostbyname(sys.argv[1])' $1 2>/dev/null
  74 }
  75
  76 # Forcefully make a symlink
  77 function symlink () {
  78     mkdir -p $(dirname $2)
  79     rm -f $2
  80     ln -s $1 $2
  81 }
  82
  83 # Argument(s) or stdin to lowercase stdout
  84 function lower () {
  85     if [ ${#*} -ge 1 ] ; then
  86         tr A-Z a-z <<<$*
  87     else
  88         tr A-Z a-z
  89     fi
  90 }
  91
  92 # Argument(s) or stdin to uppercase stdout
  93 function upper () {
  94     if [ ${#*} -ge 1 ] ; then
  95         tr a-z A-Z <<<$*
  96     else
  97         tr a-z A-Z
  98     fi
  99 }
 100
 101 # Regenerate configuration files
 102 function plc_reload () {
 103     force=$1
 104
 105     # Regenerate the main configuration file from default values
 106     # overlaid with site-specific and current values.
 107     # Thierry -- 2007-07-05 : values in plc_config.xml are *not* taken into account here
 108     files=(
 109         /etc/planetlab/default_config.xml
 110         /etc/planetlab/configs/site.xml
 111     )
 112
 113     for file in "${files[@]}" ; do
 114     if [ -n "$force" -o $file -nt /etc/planetlab/plc_config.xml ] ; then
 115         tmp=$(mktemp /tmp/plc_config.xml.XXXXXX)
 116         plc-config --xml "${files[@]}" >$tmp
 117         if [ $? -eq 0 ] ; then
 118         mv $tmp /etc/planetlab/plc_config.xml
 119         chmod 444 /etc/planetlab/plc_config.xml
 120         else
 121         echo "PLC: Warning: Invalid configuration file(s) detected"
 122         rm -f $tmp
 123         fi
 124         break
 125     fi
 126     done
 127
 128     # Convert configuration to various formats
 129     if [ -n "$force" -o /etc/planetlab/plc_config.xml -nt /etc/planetlab/plc_config ] ; then
 130     plc-config --shell >/etc/planetlab/plc_config
 131     fi
 132     if [ -n "$force" -o /etc/planetlab/plc_config.xml -nt /etc/planetlab/plc_config.py ] ; then
 133     plc-config --python >/etc/planetlab/plc_config.py
 134     fi
 135     if [ -n "$force" -o /etc/planetlab/plc_config.xml -nt /etc/planetlab/php/plc_config.php ] ; then
 136     mkdir -p /etc/planetlab/php
 137     plc-config --php >/etc/planetlab/php/plc_config.php
 138     fi
 139 }
 140
 141 # Make copies of stdout and stderr. The plc initscript redirects
 142 # stdout and stderr to a logfile if -v is not specified.
 143 [ ! -e /proc/self/fd/3 ] && exec 3>&1
 144 [ ! -e /proc/self/fd/4 ] && exec 4>&2