plc.d/postgresql

   1 #!/bin/bash
   2 #
   3 # priority: 700
   4 #
   5 # Manage the PostgreSQL database server
   6 #
   7 # Mark Huang <mlhuang@cs.princeton.edu>
   8 # Copyright (C) 2006 The Trustees of Princeton University
   9 #
  10
  11 # Source function library and configuration
  12 . /etc/plc.d/functions
  13 . /etc/planetlab/plc_config
  14 local_config=/etc/planetlab/configs/site.xml
  15
  16 # Be verbose
  17 set -x
  18
  19 # Default locations
  20 PGDATA=/var/lib/pgsql/data
  21 postgresql_conf=$PGDATA/postgresql.conf
  22 pghba_conf=$PGDATA/pg_hba.conf
  23 postgresql_sysconfig=/etc/sysconfig/pgsql/postgresql
  24
  25 # Export so that we do not have to specify -p to psql invocations
  26 export PGPORT=$PLC_DB_PORT
  27
  28 # can't trust the return of service postgresql start / nor status
  29 function postgresql_check () {
  30
  31     # wait until postmaster is up and running - or 10s max
  32     if systemctl status postgresql >& /dev/null; then
  33         # The only way we can be sure is if we can access it
  34         for i in $(seq 1 10) ; do
  35             # Must do this as the postgres user initially (before we
  36             # fix pg_hba.conf to passwordless localhost access).
  37             su -c 'psql -U postgres -c "" template1' postgres && return 0
  38             sleep 1
  39         done
  40     fi
  41
  42     return 1
  43 }
  44
  45 case "$1" in
  46     start)
  47         if [ "$PLC_DB_ENABLED" != "1" ] ; then
  48             exit 0
  49         fi
  50
  51         MESSAGE=$"Starting PostgreSQL server"
  52         dialog "$MESSAGE"
  53
  54         ######## sysconfig
  55 # xxx on f16, the systemd init script won't read /etc/sysconfig/pgsql/postgresql any more
  56 # need to find out how to perform this configuration, if still needed
  57         # Set data directory and redirect startup output to /var/log/pgsql
  58         mkdir -p $(dirname $postgresql_sysconfig)
  59         touch $postgresql_sysconfig
  60         tmp=${postgresql_sysconfig}.new
  61         # remove any previous definitions and write ours
  62         ( egrep -v '^(PGDATA=|PGLOG=|PGPORT=)' $postgresql_sysconfig
  63             echo "PGDATA=$PGDATA"
  64             echo "PGLOG=/var/log/pgsql"
  65             echo "PGPORT=$PLC_DB_PORT"
  66         ) > $tmp ; mv -f $tmp $postgresql_sysconfig
  67
  68         ######## /var/lib/pgsql/data
  69         # Fix ownership of /var/lib/pgsql (rpm installation may have changed it)
  70         chown -R -H postgres:postgres $(dirname $PGDATA)
  71
  72         # PostgreSQL must be started at least once to bootstrap
  73         # /var/lib/pgsql/data
  74         if [ ! -f $postgresql_conf ] ; then
  75 # fedora 16 uses systemd
  76 # http://docs.fedoraproject.org/en-US/Fedora/16/html/Release_Notes/sect-Release_Notes-Changes_for_Sysadmin.html
  77             if type postgresql-setup >& /dev/null ; then
  78                 postgresql-setup initdb || :
  79                 check
  80             else
  81                 sudo postgresql-setup --initdb --unit postgresql &> /dev/null || postgresql :
  82                 check
  83             fi
  84         fi
  85
  86         ######## /var/lib/pgsql/data/postgresql.conf
  87         # Enable DB server. drop Postgresql<=7.x
  88         # PostgreSQL >=8.0 defines listen_addresses
  89         # listen on a specific IP + localhost, more robust when run within a vserver
  90         sed -i -e '/^listen_addresses/d' $postgresql_conf
  91         echo "listen_addresses = '${PLC_DB_HOST},localhost'" >> $postgresql_conf
  92         # tweak timezone to be 'UTC'
  93         sed -i -e '/^timezone=/d' $postgresql_conf
  94         echo "timezone='UTC'" >> $postgresql_conf
  95
  96         ######## /var/lib/pgsql/data/pg_hba.conf
  97         # Disable access to MyPLC and drupal DBs from all hosts
  98         sed -i -e '/^\(host\|local\)/d' $pghba_conf
  99
 100         # Enable passwordless localhost access
 101         echo "local all all trust" >>$pghba_conf
 102
 103         # Enable access from the API, boot, and web servers
 104         PLC_API_IP=$(gethostbyname $PLC_API_HOST)
 105         PLC_BOOT_IP=$(gethostbyname $PLC_BOOT_HOST)
 106         PLC_WWW_IP=$(gethostbyname $PLC_WWW_HOST)
 107         ip_failure=0
 108         if [ -z "$PLC_API_IP" ] ; then
 109             MESSAGE=$"PLC_API_IP is not set"
 110             dialog "$MESSAGE"
 111             ip_failure=1
 112         fi
 113         if [ -z "$PLC_BOOT_IP" ] ; then
 114             MESSAGE=$"PLC_BOOT_IP is not set"
 115             dialog "$MESSAGE"
 116             ip_failure=1
 117         fi
 118         if [ -z "$PLC_WWW_IP" ] ; then
 119             MESSAGE=$"PLC_WWW_IP is not set"
 120             dialog "$MESSAGE"
 121             ip_failure=1
 122         fi
 123         if [ $ip_failure -eq 1 ] ; then
 124             /bin/false
 125             check
 126         fi
 127
 128         (
 129             echo "host $PLC_DB_NAME $PLC_DB_USER 127.0.0.1/32 password"
 130             echo "host $PLC_DB_NAME $PLC_DB_USER $PLC_API_IP/32 password"
 131             echo "host $PLC_DB_NAME $PLC_DB_USER $PLC_BOOT_IP/32 password"
 132             echo "host $PLC_DB_NAME $PLC_DB_USER $PLC_WWW_IP/32 password"
 133             # Drupal also uses PostgreSQL
 134             echo "host drupal $PLC_DB_USER 127.0.0.1/32 password"
 135             echo "host drupal $PLC_DB_USER $PLC_WWW_IP/32 password"
 136         ) >>$pghba_conf
 137
 138         # Append site-specific access rules
 139         for file in $pghba_conf.d/*.conf ; do
 140             cat "$file" >>$pghba_conf
 141         done
 142
 143         # Fix ownership (sed -i changes it)
 144         chown postgres:postgres $postgresql_conf $pghba_conf
 145
 146         ######## Start up the server - ignore retcod and check this our way
 147         (exec 3>&- 4>&- ; systemctl start postgresql)
 148         postgresql_check
 149         check
 150
 151         ######## Create/update the unprivileged database user and password
 152         if [ -z "$PLC_DB_PASSWORD" ] ; then
 153             PLC_DB_PASSWORD=$(uuidgen)
 154             plc-config --category=plc_db --variable=password --value="$PLC_DB_PASSWORD" --save=$local_config $local_config
 155         #service plc reload
 156         plc_reload force
 157         fi
 158         if ! psql -U $PLC_DB_USER -c "" template1 >/dev/null 2>&1 ; then
 159             psql -U postgres -c "CREATE USER $PLC_DB_USER PASSWORD '$PLC_DB_PASSWORD'" template1
 160         else
 161             psql -U postgres -c "ALTER USER $PLC_DB_USER WITH PASSWORD '$PLC_DB_PASSWORD'" template1
 162         fi
 163         check
 164
 165         ######## Create the databases if necessary
 166         if ! psql -U $PLC_DB_USER -c "" $PLC_DB_NAME >/dev/null 2>&1 ; then
 167             createdb -U postgres --template=template0 --encoding=UNICODE --owner=$PLC_DB_USER $PLC_DB_NAME
 168             psql -U $PLC_DB_USER -f /usr/share/plc_api/$PLC_DB_NAME.sql $PLC_DB_NAME
 169         fi
 170         check
 171         if ! psql -U $PLC_DB_USER -c "" drupal >/dev/null 2>&1 ; then
 172             createdb -U postgres --template=template0 --encoding=UNICODE --owner=$PLC_DB_USER drupal
 173             psql -U $PLC_DB_USER -f /var/www/html/database/database.pgsql drupal
 174         fi
 175
 176         result "$MESSAGE"
 177         ;;
 178
 179     stop)
 180         MESSAGE=$"Stopping PostgreSQL server"
 181         dialog "$MESSAGE"
 182
 183         # Drop the current user in case the username changes
 184         psql -U postgres -c "DROP USER $PLC_DB_USER" template1
 185
 186         # WARNING: If the DB name changes, the old DB will be left
 187         # intact and a new one will be created. If it changes
 188         # back, the old DB will not be re-created.
 189
 190         # Shut down the server
 191         systemctl stop postgresql
 192
 193         # /etc/init.d/postgresql fails if it is not running
 194         [ "$PLC_DB_ENABLED" = 1 ] && check
 195
 196         result "$MESSAGE"
 197         ;;
 198 esac
 199
 200 exit $ERRORS