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
  24 # Export so that we do not have to specify -p to psql invocations
  25 export PGPORT=$PLC_DB_PORT
  26
  27 # /etc/init.d/postgresql always returns 0, even on failure
  28 postgresql_start ()
  29 {
  30     # start() always returns 0
  31     (exec 3>&- 4>&- ; service postgresql start)
  32
  33     # status() will still return 0 even while still initializing
  34     if status postmaster && [ -f /var/lock/subsys/postgresql ] ; then
  35         # The only way we can be sure is if we can access it
  36         for i in $(seq 1 10) ; do
  37             # Must do this as the postgres user initially (before we
  38             # fix pg_hba.conf to passwordless localhost access).
  39             su -c 'psql -U postgres -c "" template1' postgres && return 0
  40             sleep 1
  41         done
  42     fi
  43
  44     return 1
  45 }
  46
  47 postgresql_init ()
  48 {
  49     service postgresql initdb &> /dev/null || :
  50     postgresql_start
  51 }
  52
  53 case "$1" in
  54     start)
  55         if [ "$PLC_DB_ENABLED" != "1" ] ; then
  56             exit 0
  57         fi
  58
  59         MESSAGE=$"Starting PostgreSQL server"
  60         dialog "$MESSAGE"
  61
  62         # Set data directory and redirect startup output to /var/log/pgsql
  63         mkdir -p /etc/sysconfig/pgsql
  64         (
  65             echo "PGDATA=$PGDATA"
  66             echo "PGLOG=/var/log/pgsql"
  67             echo "PGPORT=$PLC_DB_PORT"
  68         ) >>/etc/sysconfig/pgsql/postgresql
  69
  70         # Fix ownership (rpm installation may have changed it)
  71         chown -R -H postgres:postgres $(dirname $PGDATA)
  72
  73         # PostgreSQL must be started at least once to bootstrap
  74         # /var/lib/pgsql/data
  75         if [ ! -f $postgresql_conf ] ; then
  76             postgresql_init
  77             check
  78             service postgresql stop
  79             check
  80         fi
  81
  82         # Enable DB server. PostgreSQL >=8.0 defines listen_addresses,
  83         # PostgreSQL 7.x uses tcpip_socket.
  84         if grep -q listen_addresses $postgresql_conf ; then
  85             sed -i -e '/^listen_addresses/d' $postgresql_conf
  86             echo "listen_addresses = '*'" >> $postgresql_conf
  87             # tweak timezone to be 'UTC'
  88             sed -i -e '/^timezone=/d' $postgresql_conf
  89             echo "timezone='UTC'" >> $postgresql_conf
  90         else
  91             dialog "PostgreSQL <= 7.x - not supported"
  92             /bin/false
  93             check
  94         fi
  95
  96         # Disable access to all DBs from all hosts
  97         sed -i -e '/^\(host\|local\)/d' $pghba_conf
  98
  99         # Enable passwordless localhost access
 100         echo "local all all trust" >>$pghba_conf
 101
 102         # Enable access from the API, boot, and web servers
 103         PLC_API_IP=$(gethostbyname $PLC_API_HOST)
 104         PLC_BOOT_IP=$(gethostbyname $PLC_BOOT_HOST)
 105         PLC_WWW_IP=$(gethostbyname $PLC_WWW_HOST)
 106         ip_failure=0
 107         if [ -z "$PLC_API_IP" ] ; then
 108             MESSAGE=$"PLC_API_IP is not set"
 109             dialog "$MESSAGE"
 110             ip_failure=1
 111         fi
 112         if [ -z "$PLC_BOOT_IP" ] ; then
 113             MESSAGE=$"PLC_BOOT_IP is not set"
 114             dialog "$MESSAGE"
 115             ip_failure=1
 116         fi
 117         if [ -z "$PLC_WWW_IP" ] ; then
 118             MESSAGE=$"PLC_WWW_IP is not set"
 119             dialog "$MESSAGE"
 120             ip_failure=1
 121         fi
 122         if [ $ip_failure -eq 1 ] ; then
 123             /bin/false
 124             check
 125         fi
 126
 127         (
 128             echo "host $PLC_DB_NAME $PLC_DB_USER 127.0.0.1/32 password"
 129             echo "host $PLC_DB_NAME $PLC_DB_USER $PLC_API_IP/32 password"
 130             echo "host $PLC_DB_NAME $PLC_DB_USER $PLC_BOOT_IP/32 password"
 131             echo "host $PLC_DB_NAME $PLC_DB_USER $PLC_WWW_IP/32 password"
 132             # Drupal also uses PostgreSQL
 133             echo "host drupal $PLC_DB_USER 127.0.0.1/32 password"
 134             echo "host drupal $PLC_DB_USER $PLC_WWW_IP/32 password"
 135         ) >>$pghba_conf
 136
 137         # Append site-specific access rules
 138         for file in $pghba_conf.d/*.conf ; do
 139             cat "$file" >>$pghba_conf
 140         done
 141
 142         # Fix ownership (sed -i changes it)
 143         chown postgres:postgres $postgresql_conf $pghba_conf
 144
 145         # Start up the server
 146         postgresql_start
 147         check
 148
 149         # Create/update the unprivileged database user and password
 150         if [ -z "$PLC_DB_PASSWORD" ] ; then
 151             PLC_DB_PASSWORD=$(uuidgen)
 152             plc-config --category=plc_db --variable=password --value="$PLC_DB_PASSWORD" --save=$local_config $local_config
 153             service plc reload
 154         fi
 155         if ! psql -U $PLC_DB_USER -c "" template1 >/dev/null 2>&1 ; then
 156             psql -U postgres -c "CREATE USER $PLC_DB_USER PASSWORD '$PLC_DB_PASSWORD'" template1
 157         else
 158             psql -U postgres -c "ALTER USER $PLC_DB_USER WITH PASSWORD '$PLC_DB_PASSWORD'" template1
 159         fi
 160         check
 161
 162         # Create the databases if necessary
 163         if ! psql -U $PLC_DB_USER -c "" $PLC_DB_NAME >/dev/null 2>&1 ; then
 164             createdb -U postgres --template=template0 --encoding=UNICODE --owner=$PLC_DB_USER $PLC_DB_NAME
 165             psql -U $PLC_DB_USER -f /usr/share/plc_api/$PLC_DB_NAME.sql $PLC_DB_NAME
 166         fi
 167         check
 168         if ! psql -U $PLC_DB_USER -c "" drupal >/dev/null 2>&1 ; then
 169             createdb -U postgres --template=template0 --encoding=UNICODE --owner=$PLC_DB_USER drupal
 170             psql -U $PLC_DB_USER -f /var/www/html/database/database.pgsql drupal
 171         fi
 172         check
 173
 174         result "$MESSAGE"
 175         ;;
 176
 177     stop)
 178         MESSAGE=$"Stopping PostgreSQL server"
 179         dialog "$MESSAGE"
 180
 181         # Drop the current user in case the username changes
 182         psql -U postgres -c "DROP USER $PLC_DB_USER" template1
 183
 184         # WARNING: If the DB name changes, the old DB will be left
 185         # intact and a new one will be created. If it changes
 186         # back, the old DB will not be re-created.
 187
 188         # Shut down the server
 189         service postgresql stop
 190
 191         # /etc/init.d/postgresql fails if it is not running
 192         [ "$PLC_DB_ENABLED" = 1 ] && check
 193
 194         result "$MESSAGE"
 195         ;;
 196 esac
 197
 198 exit $ERRORS