4 # Provides a generic SFA wrapper based on the initial PlanetLab Implementation
6 # hopefully right after plc
7 # chkconfig: 2345 61 39
9 # description: An implementation of the SFA Architecture
13 # borrowed from postgresql
14 function debian_get_postgresql_versions () {
16 for v in `ls /usr/lib/postgresql/ 2>/dev/null`; do
17 if [ -x /usr/lib/postgresql/$v/bin/pg_ctl ] && [ ! -x /etc/init.d/postgresql-$v ]; then
21 if [[ ${#versions[*]} == "0" ]]; then
22 echo "E: Missing postgresql installation. Aborting."
25 if [[ ${#versions[*]} != "1" ]]; then
26 echo "E: Too many postgresql versions installed. Aborting."
33 if [ -f /etc/redhat-release ] ; then
34 # source function library
35 . /etc/init.d/functions
36 PGDATA=/var/lib/pgsql/data/
38 PGLOCK=/var/lock/subsys/postgresql
39 SFALOCK=/var/lock/subsys/sfa-start.pid
40 elif [ -f /etc/debian_version ] ; then
41 . /etc/init.d/functions.sfa
42 debian_get_postgresql_versions
43 PGDATA=/etc/postgresql/$pgver/main/
45 PGLOCK=/var/run/postgresql/$pgver-main.pid
46 SFALOCK=/var/run/sfa-start.pid
48 echo "initscript can only handle redhat/fedora or debian/ubuntu systems"
53 postgresql_conf=$PGDATA/postgresql.conf
54 pghba_conf=$PGDATA/pg_hba.conf
55 postgresql_sysconfig=/etc/sysconfig/pgsql
57 # SFA consolidated (merged) config file
58 sfa_whole_config=/etc/sfa/sfa_config
59 # SFA default config (read-only template)
60 sfa_default_config=/etc/sfa/default_config.xml
61 # SFA local (site-dependent) file
62 sfa_local_config=/etc/sfa/configs/site_config
63 sfa_local_config_xml=/etc/sfa/configs/site_config.xml
65 # Source sfa shell config if present
66 [ -f /etc/sfa/sfa_config.sh ] && . /etc/sfa/sfa_config.sh
68 # Export so that we do not have to specify -p to psql invocations
69 export PGPORT=$SFA_DB_PORT
72 # Total number of errors
75 # Count the exit status of the last command
78 ERRORS=$(($ERRORS+$?))
81 # can't trust the return of service postgresql start / nor status
82 function postgresql_check () {
84 # wait until postmaster is up and running - or 10s max
85 if status $PGWATCH >& /dev/null && [ -f $PGLOCK ] ; then
86 # The only way we can be sure is if we can access it
87 for i in $(seq 1 10) ; do
88 # Must do this as the postgres user initially (before we
89 # fix pg_hba.conf to passwordless localhost access).
90 su -c 'psql -U postgres -c "" template1' postgres && return 0
98 # use a single date of this script invocation for the dump_*_db functions.
99 DATE=$(date +"%Y-%m-%d-%H-%M-%S")
101 # Dumps the database - optional argument to specify filename suffix
102 function dump_sfa_db() {
103 if [ -n "$1" ] ; then suffix="-$1" ; else suffix="" ; fi
104 mkdir -p /usr/share/sfa/backups
105 dumpfile=/usr/share/sfa/backups/$(date +"${SFA_DB_NAME}.${DATE}${suffix}.sql")
106 pg_dump -U $SFA_DB_USER $SFA_DB_NAME > $dumpfile
107 echo "Saved sfa database in $dumpfile"
111 # Regenerate configuration files - almost verbatim from plc.init
115 # Regenerate the main configuration file from default values
116 # overlaid with site-specific and current values.
117 files=( $sfa_default_config $sfa_local_config )
118 for file in "${files[@]}" ; do
119 if [ -n "$force" -o $file -nt $sfa_whole_config ] ; then
120 tmp=$(mktemp /tmp/sfa_config.XXXXXX)
121 sfa-config --python "${files[@]}" >$tmp
122 if [ $? -eq 0 ] ; then
123 mv $tmp $sfa_whole_config
124 chmod 444 $sfa_whole_config
126 echo "SFA: Warning: Invalid configuration file(s) detected"
133 # Convert configuration to various formats
134 if [ -f $sfa_local_config_xml ] ; then
135 sfa-config --python $sfa_local_config_xml > $sfa_local_config
136 rm $sfa_local_config_xml
138 if [ -n "$force" -o $sfa_local_config -nt $sfa_whole_config ] ; then
139 sfa-config --python $sfa_default_config $sfa_local_config > $sfa_whole_config
141 if [ -n "$force" -o $sfa_whole_config -nt /etc/sfa/sfa_config.sh ] ; then
142 sfa-config --shell $sfa_default_config $sfa_local_config > /etc/sfa/sfa_config.sh
145 # [re]generate the sfa_component_config
146 # this is a server-side thing but produces a file that somehow needs to be pushed
147 # on the planetlab nodes; in the case where sfa and myplc run on different boxes
148 # (or there is no myplc at all) this should be turned off
149 # as the component manager is not operational yet we skip this for now
150 #gen-sfa-cm-config.py
152 # reload the shell version
153 [ -f /etc/sfa/sfa_config.sh ] && . /etc/sfa/sfa_config.sh
157 ### initialize DB (don't chkconfig postgresql on)
158 function db_start () {
161 [ "$SFA_DB_ENABLED" == 1 -o "$SFA_DB_ENABLED" == True ] || return
163 if ! rpm -q myplc >& /dev/null; then
165 ######## standalone deployment - no colocated myplc
168 # Set data directory and redirect startup output to /var/log/pgsql
169 mkdir -p $(dirname $postgresql_sysconfig)
170 # remove previous definitions
171 touch $postgresql_sysconfig
172 tmp=${postgresql_sysconfig}.new
173 ( egrep -v '^(PGDATA=|PGLOG=|PGPORT=)' $postgresql_sysconfig
174 echo "PGDATA=$PGDATA"
175 echo "PGLOG=/var/log/pgsql"
176 echo "PGPORT=$SFA_DB_PORT"
177 ) >> $tmp ; mv -f $tmp $postgresql_sysconfig
179 ######## /var/lib/pgsql/data
180 # Fix ownership (rpm installation may have changed it)
181 chown -R -H postgres:postgres $(dirname $PGDATA)
183 # PostgreSQL must be started at least once to bootstrap
184 # /var/lib/pgsql/data
185 if [ ! -f $postgresql_conf ] ; then
186 service postgresql initdb &> /dev/null || :
190 ######## /var/lib/pgsql/data/postgresql.conf
192 foo=$(python -c "import socket; print socket.gethostbyname(\"$SFA_REGISTRY_HOST\")") && registry_ip="$foo"
193 # Enable DB server. drop Postgresql<=7.x
194 # PostgreSQL >=8.0 defines listen_addresses
195 # listen on a specific IP + localhost, more robust when run within a vserver
196 sed -i -e '/^listen_addresses/d' $postgresql_conf
197 if [ -z "$registry_ip" ] ; then
198 echo "listen_addresses = 'localhost'" >> $postgresql_conf
200 echo "listen_addresses = '${registry_ip},localhost'" >> $postgresql_conf
202 # tweak timezone to be 'UTC'
203 sed -i -e '/^timezone=/d' $postgresql_conf
204 echo "timezone='UTC'" >> $postgresql_conf
206 ######## /var/lib/pgsql/data/pg_hba.conf
207 # Disable access to all DBs from all hosts
208 sed -i -e '/^\(host\|local\)/d' $pghba_conf
210 # Enable passwordless localhost access
211 echo "local all all trust" >>$pghba_conf
214 echo "host $SFA_DB_NAME $SFA_DB_USER 127.0.0.1/32 password"
215 [ -n "$registry_ip" ] && echo "host $SFA_DB_NAME $SFA_DB_USER ${registry_ip}/32 password"
218 if [ "$SFA_GENERIC_FLAVOUR" == "openstack" ] ; then
219 [ -n "$registry_ip" ] && echo "host nova nova ${registry_ip}/32 password" >> $pghba_conf
222 # Fix ownership (sed -i changes it)
223 chown postgres:postgres $postgresql_conf $pghba_conf
225 ######## compute a password if needed
226 if [ -z "$SFA_DB_PASSWORD" ] ; then
227 SFA_DB_PASSWORD=$(uuidgen)
228 sfa-config --category=sfa_db --variable=password --value="$SFA_DB_PASSWORD" --save=$sfa_local_config $sfa_local_config >& /dev/null
234 ######## we are colocated with a myplc
235 # no need to worry about the pgsql setup (see /etc/plc.d/postgresql)
236 # myplc enforces the password for its user
237 PLC_DB_USER=$(plc-config --category=plc_db --variable=user)
238 PLC_DB_PASSWORD=$(plc-config --category=plc_db --variable=password)
239 # store this as the SFA user/password
240 sfa-config --category=sfa_db --variable=user --value=$PLC_DB_USER --save=$sfa_local_config $sfa_local_config >& /dev/null
241 sfa-config --category=sfa_db --variable=password --value=$PLC_DB_PASSWORD --save=$sfa_local_config $sfa_local_config >& /dev/null
245 ######## Start up the server
246 # not too nice, but.. when co-located with myplc we'll let it start/stop postgresql
247 if [ ! -f /etc/myplc-release ] ; then
249 service postgresql start >& /dev/null
254 ######## make sure we have the user and db created
256 if ! psql -U $SFA_DB_USER -c "" template1 >/dev/null 2>&1 ; then
257 psql -U postgres -c "CREATE USER $SFA_DB_USER PASSWORD '$SFA_DB_PASSWORD'" template1 >& /dev/null
259 psql -U postgres -c "ALTER USER $SFA_DB_USER WITH PASSWORD '$SFA_DB_PASSWORD'" template1 >& /dev/null
264 if ! psql -U $SFA_DB_USER -c "" $SFA_DB_NAME >/dev/null 2>&1 ; then
265 createdb -U postgres --template=template0 --encoding=UNICODE --owner=$SFA_DB_USER $SFA_DB_NAME
269 # mention sfaadmin.py instead of just sfaadmin for people who do not install through rpm
270 sfaadmin.py reg sync_db
272 MESSAGE=$"SFA: Checking for PostgreSQL server"
274 [ "$ERRORS" == 0 ] && success "$MESSAGE" || failure "$MESSAGE" ; echo
278 function db_stop () {
281 [ "$SFA_DB_ENABLED" == 1 -o "$SFA_DB_ENABLED" == True ] || return
283 # not too nice, but.. when co-located with myplc we'll let it start/stop postgresql
284 if ! rpm -q myplc >& /dev/null ; then
285 service postgresql stop >& /dev/null
287 MESSAGE=$"Stopping PostgreSQL server"
289 [ "$ERRORS" == 0 ] && success "$MESSAGE" || failure "$MESSAGE" ; echo
298 # migrations are now handled in the code by sfa.storage.dbschema
301 action $"SFA: installing peer certs" daemon /usr/bin/sfa-start.py -t -d $OPTIONS
303 [ "$SFA_REGISTRY_ENABLED" == 1 -o "$SFA_REGISTRY_ENABLED" == True ] && action $"SFA: Registry" daemon /usr/bin/sfa-start.py -r -d $OPTIONS
305 [ "$SFA_AGGREGATE_ENABLED" == 1 -o "$SFA_AGGREGATE_ENABLED" == True ] && action $"SFA: Aggregate" daemon /usr/bin/sfa-start.py -a -d $OPTIONS
307 [ "$SFA_SM_ENABLED" == 1 -o "$SFA_SM_ENABLED" == True ] && action "SFA: SliceMgr" daemon /usr/bin/sfa-start.py -s -d $OPTIONS
309 [ "$SFA_FLASHPOLICY_ENABLED" == 1 ] && \
310 action "Flash Policy Server" daemon /usr/bin/sfa_flashpolicy.py --file="$SFA_FLASHPOLICY_CONFIG_FILE" --port=$SFA_FLASHPOLICY_PORT -d
317 action $"Shutting down SFA" killproc sfa-start.py
318 # a possible alternative reads; esp. as we remove lock manually below
319 # echo $"Shutting down SFA" ; pkill '^sfa-start'
330 reload) reload force ;;
331 restart) stop; start ;;
333 if [ -f $SFALOCK ]; then
340 # possible alternative for debian
341 # pids=$(pgrep '^sfa-start'); [ -n "$pids" ] && ps $pids
349 echo $"Usage: $0 {start|stop|reload|restart|condrestart|status|dbdump}"