Ability to checkpoint and the restore a myplc configuration. This will checkpoint...
[myplc.git] / guest.init
1 #!/bin/bash
2 #
3 # plc   Manages all PLC services on this machine
4 #
5 # chkconfig: 2345 5 99
6 #
7 # description:  Manages all PLC services on this machine
8 #
9 # $Id$
10 #
11
12 # Source function library and configuration
13 . /etc/plc.d/functions
14
15 # Verbosity
16 verbose=0
17
18 # All steps should be idempotent. This means that you should be able
19 # to run them multiple times without depending on anything previously
20 # being run. The idea is that when the configuration changes, "service
21 # plc restart" is called, all dependencies are fixed up, and
22 # everything just works.
23
24 ### NOTE.
25 # we want the resulting myplc to be able to easily skip
26 # some steps. e.g. the packages step takes ages if you install
27 # all rpms under the repository.
28 # We skip steps whose name contains a dot (.) or a tilde (~)
29 # this way the operations would just rename a step name e.g.
30 # cd /etc/plc.d
31 # mv packages packages.hide
32 #
33 # The drawback is, this stuff does not survive an rpm update
34 # but that's maybe a good thing, that all is done at first start
35 ###
36
37 steps=($(
38 for step in /etc/plc.d/* ; do
39     stepname=$(basename $step)
40     plainstepname=$(echo $stepname | sed -e 's,\.,,' -e 's,~,,')
41     if [ -f $step -a -x $step -a "$stepname" = "$plainstepname" ] ; then
42         priority=$(sed -ne 's/# priority: \(.*\)/\1/p' $step)
43         echo $priority $stepname
44     fi
45 done | sort -n | cut -d' ' -f2
46 ))
47 nsteps=${#steps[@]}
48
49 # Regenerate configuration files
50 reload ()
51 {
52     force=$1
53
54     # Regenerate the main configuration file from default values
55     # overlaid with site-specific and current values.
56     # Thierry -- 2007-07-05 : values in plc_config.xml are *not* taken into account here
57     files=(
58         /etc/planetlab/default_config.xml 
59         /etc/planetlab/configs/site.xml
60     )
61     for file in "${files[@]}" ; do
62         if [ -n "$force" -o $file -nt /etc/planetlab/plc_config.xml ] ; then
63             tmp=$(mktemp /tmp/plc_config.xml.XXXXXX)
64             plc-config --xml "${files[@]}" >$tmp
65             if [ $? -eq 0 ] ; then
66                 mv $tmp /etc/planetlab/plc_config.xml
67                 chmod 444 /etc/planetlab/plc_config.xml
68             else
69                 echo "PLC: Warning: Invalid configuration file(s) detected"
70                 rm -f $tmp
71             fi
72             break
73         fi
74     done
75
76     # Convert configuration to various formats
77     if [ -n "$force" -o /etc/planetlab/plc_config.xml -nt /etc/planetlab/plc_config ] ; then
78         plc-config --shell >/etc/planetlab/plc_config
79     fi
80     if [ -n "$force" -o /etc/planetlab/plc_config.xml -nt /etc/planetlab/php/plc_config.php ] ; then
81         mkdir -p /etc/planetlab/php
82         plc-config --php >/etc/planetlab/php/plc_config.php
83     fi
84 }
85
86 usage()
87 {
88     echo "Usage: $0 [OPTION]... [COMMAND] [STEP]..."
89     echo "      -v              Be verbose"
90     echo "      -h              This message"
91     echo
92     echo "Commands:"
93     echo "      start           Start all PLC subsystems"
94     echo "      stop            Stop all PLC subsystems"
95     echo "      reload          Regenerate configuration files"
96     echo "      restart         Restart all PLC subsystems"
97     echo "      steps           Displays ordered list of subsystems"
98     echo
99     echo "Steps:"
100     for step in "${steps[@]}" ; do
101         if [ -x /etc/plc.d/$step ] ; then
102             echo "      $(basename $step)"
103         fi
104     done
105     exit 1
106 }
107
108 # Get options
109 while getopts "vh" opt ; do
110     case $opt in
111         v)
112             verbose=1
113             ;;
114         h|*)
115             usage
116             ;;
117     esac
118 done
119
120 # Redirect stdout and stderr of each step to /var/log/boot.log
121 if [ $verbose -eq 0 ] ; then
122     touch /var/log/boot.log
123     chmod 600 /var/log/boot.log
124     exec 1>>/var/log/boot.log
125     exec 2>>/var/log/boot.log
126 fi
127
128 # Get command
129 shift $(($OPTIND - 1))
130 if [ -z "$1" ] ; then
131     usage
132 fi
133 command=$1
134
135 # Get step(s)
136 shift 1
137 if [ -z "$1" ] ; then
138     # Start or stop everything. Regenerate configuration first.
139     reload force
140 else
141     # Start or stop a particular step
142     steps=("$@")
143     nsteps=${#steps[@]}
144 fi
145
146 RETVAL=0
147
148 start ()
149 {
150     for step in "${steps[@]}" ; do
151         if [ -x /etc/plc.d/$step ] ; then
152             /etc/plc.d/$step start
153             # Steps may alter the configuration, may need to regenerate
154             reload
155         else
156             echo "PLC: $step: unrecognized step" >&4
157             exit 1
158         fi
159     done
160 }
161
162 stop ()
163 {
164     for i in $(seq 1 $nsteps) ; do
165         step=${steps[$(($nsteps - $i))]}
166         if [ -x /etc/plc.d/$step ] ; then
167             /etc/plc.d/$step stop
168             # Steps may alter the configuration, may need to regenerate
169             reload
170         else
171             echo "PLC: $step: unrecognized step" >&4
172             exit 1
173         fi
174     done
175 }
176
177 case "$command" in
178     start|stop)
179         $command
180         ;;
181
182     restart)
183         stop
184         start
185         ;;
186
187     reload)
188         reload force
189         ;;
190
191     checkpoint)
192         cpfile=$1
193         [ -z "$cpfile" ] && cpfile=/var/tmp/plc_checkpoint.$(date +"%Y-%m-%d-%H-%M-%S")
194         cpdir=$(mktemp -d tmp.XXXXXX)
195         cd $cpdir
196         mkdir -p ./etc/planetlab/
197         rsync -av /etc/planetlab ./etc/planetlab/
198         /etc/plc.d/db checkpoint ./etc/planetlab/plc_db.checkpoint ./etc/planetlab/plc_drupal.checkpoint
199         tar cjf $cpfile etc
200         cd -
201         rm -rf $cpdir
202         ;;
203
204     restore)
205         cpfile=$1
206         cpdir=$(mktemp -d tmp.XXXXXX)
207         cd $cpdir
208         tar xjf $cpfile
209         rsync -av ./etc/planetlab/ /etc/planetlab
210         /etc/plc.d/db restore ./etc/planetlab/plc_db.checkpoint ./etc/planetlab/plc_drupal.checkpoint
211         cd -
212         rm -rf $cpdir
213         ;;
214
215     steps)
216         echo "${steps[@]}" >&4
217         ;;
218
219     *)
220         usage >&3
221         ;;
222 esac
223
224 exit $RETVAL