monitor-server.init

   1 #!/bin/bash
   2 #
   3 # priority: 850
   4 #
   5 # Manage settings for the Zabbix installtion and
   6 #       other monitor-related things
   7 #
   8 # Stephen Soltesz <soltesz@cs.princeton.edu>
   9 # Copyright (C) 2008 The Trustees of Princeton University
  10 #
  11 # $Id$
  12 #
  13
  14 # Source function library and configuration
  15 . /etc/plc.d/functions
  16 . /etc/planetlab/plc_config
  17 local_config=/etc/planetlab/configs/site.xml
  18
  19 # Be verbose
  20 set -x
  21
  22 # Default locations
  23 PGDATA=/var/lib/pgsql/data
  24 postgresql_conf=$PGDATA/postgresql.conf
  25 pghba_conf=$PGDATA/pg_hba.conf
  26
  27 # Export so that we do not have to specify -p to psql invocations
  28 export PGPORT=$PLC_DB_PORT
  29
  30
  31 # TODO:
  32 ##   setup database
  33 ###    import schema & data
  34 ###    tweak values
  35 ###    add zabbix to pg_hba.conf
  36 ##   setup zabbix.conf.php
  37 ##
  38 DB_USER="zabbixuser"
  39 DB_NAME="zabbix"
  40
  41 function check_user_and_db()
  42 {
  43     # confirm user is present or create it
  44     user_present=$( psql -U postgres -c "select * from pg_user;" -d template1 | grep $ZABBIX_DB_NAME )
  45     if [ -z $user_present ] ; then
  46         createuser --no-superuser --no-createdb --no-createrole --login --unencrypted --echo $ZABBIX_DB_NAME -U postgres
  47     fi
  48
  49     # confirm database is present or create it
  50     db_present=$( psql -U postgres -c "select * from pg_database;" -d template1 | grep $ZABBIX_DB_NAME )
  51     if [ -z $db_present ] ; then
  52         createdb --owner=$ZABBIX_DB_NAME $ZABBIX_DB_NAME -U postgres
  53     fi
  54
  55     # Create/update the unprivileged database user and password
  56     if [ -z "$PLC_MONITOR_DBPASSWORD" ] ; then
  57         PLC_MONITOR_DBPASSWORD=$(uuidgen)
  58         plc-config --category=plc_monitor --variable=dbpassword --value="$PLC_MONITOR_DBPASSWORD" --save=$local_config $local_config
  59         psql -d template1 -U postgres -c "ALTER USER $ZABBIX_DB_NAME WITH PASSWORD '$PLC_MONITOR_DBPASSWORD';"
  60         service plc reload
  61     fi
  62 }
  63
  64 function if_present_load ()
  65 {
  66     file=$1
  67     if [ -f $file ] ; then
  68         psql -d $ZABBIX_DB_NAME -U $ZABBIX_DB_USER < $file
  69     fi
  70 }
  71
  72 function check_schema_and_data()
  73 {
  74     schema_present=$( psql -U $ZABBIX_DB_USER $ZABBIX_DB_NAME -c "\d;" < /dev/null | grep hosts )
  75     if [ -z $schema_present ] ; then
  76         if_present_load "/usr/local/zabbix/misc/create/schema/postgresql.sql"
  77         if_present_load "/usr/local/zabbix/misc/create/data/data.sql"
  78         if_present_load "/usr/local/zabbix/misc/create/data/images_pgsql.sql"
  79         ## TODO: update ZABBIX Server entry, "update hosts set status=0, host='MyPLC Server' where hostid=10017"
  80     fi
  81 }
  82
  83
  84 case "$1" in
  85     start)
  86         if [ "$PLC_MONITOR_ENABLED" != "1" ] ; then
  87             exit 0
  88         fi
  89         MESSAGE=$"Bootstrap Monitoring"
  90         dialog "$MESSAGE"
  91
  92         check_user_and_db
  93         check_schema_and_data
  94
  95         mkdir -p /var/lib/pgsql/data/pg_hba.conf.d
  96         ZABCONF=/var/lib/pgsql/data/pg_hba.conf.d/zabbix.conf
  97         if [ ! -f $ZABCONF ] ; then
  98                 echo "host $ZABBIX_DB_NAME $ZABBIX_DB_USER 127.0.0.1/32 password"   > $ZAB
  99                 echo "host $ZABBIX_DB_NAME $ZABBIX_DB_USER $PLC_MONITOR_IP/32 password" >> $ZAB
 100         fi
 101
 102         # UPDATE /etc/zabbix/*.conf
 103         ZABBIXCFG=/etc/zabbix
 104         TMP_FILE=`mktemp /tmp/zbxtmpXXXXXX`
 105         # TODO:  How to know if I need to restart the services?
 106
 107         if [ -f ${ZABBIXCFG}/zabbix_server.conf ] ; then
 108                 sed -e "s/#DBHost=.*/DBHost=$PLC_MONITOR_HOST/g" \
 109                     -e "s#DBName=.*#DBName=$ZABBIX_DB_NAME#g" \
 110                     -e "s#DBUser=.*#DBUser=$ZABBIX_DB_USER#g" \
 111                     -e "s#DBPassword=.*#$PLC_MONITOR_DBPASSWORD#g" \
 112                     ${ZABBIXCFG}/zabbix_server.conf > $TMP_FILE
 113                 cat $TMP_FILE > ${ZABBIXCFG}/zabbix_server.conf
 114         fi
 115         if [ -f ${ZABBIXCFG}/zabbix_agentd.conf ] ; then
 116                 HOST=`hostname`
 117                 sed -e "s#Server=.*#Server=$PLC_MONITOR_HOST#g" \
 118                     -e "s#Hostname=.*#Hostname=$HOST#g" \
 119                     ${ZABBIXCFG}/zabbix_agentd.conf > $TMP_FILE
 120                 cat $TMP_FILE > ${ZABBIXCFG}/zabbix_agentd.conf
 121         fi
 122         service zabbix_server start
 123         service zabbix_agentd start
 124
 125         # SETUP zabbix gui configuration
 126         ZABBIX_WEB_CFG=/var/www/html/zabbix/conf/zabbix.conf.php
 127         if [ ! -f $ZABBIX_WEB_CFG ] ; then
 128                 touch  $ZABBIX_WEB_CFG
 129                 cat <<EOF > $ZABBIX_WEB_CFG
 130 <?php
 131 global \$DB;
 132
 133 \$DB["TYPE"]            = "POSTGRESQL";
 134 \$DB["SERVER"]          = "localhost";
 135 \$DB["PORT"]            = "0";
 136 \$DB["DATABASE"]                = "$ZABBIX_DB_NAME";
 137 \$DB["USER"]            = "$ZABBIX_DB_USER";
 138 \$DB["PASSWORD"]                = "$PLC_MONITOR_DBPASSWORD";
 139 \$ZBX_SERVER            = "$PLC_MONITOR_HOST";
 140 \$ZBX_SERVER_PORT       = "10051";
 141 \$IMAGE_FORMAT_DEFAULT  = IMAGE_FORMAT_PNG;
 142 ?>
 143 EOF
 144                 chmod 644 $ZABBIX_WEB_CFG
 145         fi
 146
 147         result "$MESSAGE"
 148         ;;
 149
 150     stop)
 151         MESSAGE=$"Stopping Monitor"
 152         dialog "$MESSAGE"
 153
 154         service zabbix_server stop
 155         service zabbix_agentd stop
 156         # TODO: is there anything to stop?
 157         result "$MESSAGE"
 158         ;;
 159 esac
 160
 161 exit $ERRORS