#!/bin/bash # # priority: 850 # # Manage settings for the Zabbix installtion and # other monitor-related things # # Stephen Soltesz # Copyright (C) 2008 The Trustees of Princeton University # # $Id$ # # Source function library and configuration . /etc/plc.d/functions . /etc/planetlab/plc_config local_config=/etc/planetlab/configs/site.xml MONITORPATH=/usr/share/monitor-server # Be verbose set -x # Default locations PGDATA=/var/lib/pgsql/data postgresql_conf=$PGDATA/postgresql.conf pghba_conf=$PGDATA/pg_hba.conf # Export so that we do not have to specify -p to psql invocations export PGPORT=$PLC_DB_PORT ZABBIX_DB_USER="zabbixuser" ZABBIX_DB_NAME="zabbix" function check_pg_hba () { #### SETUP ACCESS to this user and database mkdir -p $PGDATA/pg_hba.conf.d ZABCONF=$PGDATA/pg_hba.conf.d/zabbix.conf if [ ! -f $ZABCONF ] ; then echo "host $ZABBIX_DB_NAME $ZABBIX_DB_USER 127.0.0.1/32 password" > $ZABCONF echo "host $ZABBIX_DB_NAME $ZABBIX_DB_USER $PLC_MONITOR_IP/32 password" >> $ZABCONF # NOTE: restart db to enable access by users granted above. service plc restart posgresql fi } function check_user_and_db() { CREATED= # confirm user is present or create it user_present=$( psql -U postgres -c "select * from pg_user;" -d template1 | grep $ZABBIX_DB_USER ) if [ -z $user_present ] ; then createuser --no-superuser --no-createdb --no-createrole --login --unencrypted --echo $ZABBIX_DB_USER -U postgres CREATED="true" fi # confirm database is present or create it db_present=$( psql -U postgres -c "select * from pg_database;" -d template1 | grep $ZABBIX_DB_NAME ) if [ -z $db_present ] ; then createdb --owner=$ZABBIX_DB_USER $ZABBIX_DB_NAME -U postgres CREATED="true" fi # Create/update the unprivileged database user and password if [ -z "$PLC_MONITOR_DBPASSWORD" ] ; then # Zabbix doesn't like plain uuidgen passwords PLC_MONITOR_DBPASSWORD=$( uuidgen | md5sum - | awk '{print $1}' ) plc-config --category=plc_monitor --variable=dbpassword --value="$PLC_MONITOR_DBPASSWORD" --save=$local_config $local_config service plc reload CREATED="true" fi if [ -n "$CREATED" ] ; then psql -d template1 -U postgres -c "ALTER USER $ZABBIX_DB_USER WITH PASSWORD '$PLC_MONITOR_DBPASSWORD';" fi } function if_present_load () { file=$1 if [ -f $file ] ; then psql -d $ZABBIX_DB_NAME -U $ZABBIX_DB_USER < $file fi } function check_schema_and_data() { schema_present=$( psql -U $ZABBIX_DB_USER $ZABBIX_DB_NAME -c "\d;" < /dev/null | grep hosts ) if [ -z $schema_present ] ; then echo "... initial import can take SEVERAL minutes. please wait ..." if_present_load "/usr/local/zabbix/misc/create/schema/postgresql.sql" if_present_load "/usr/local/zabbix/misc/create/data/data.sql" if_present_load "/usr/local/zabbix/misc/create/data/images_pgsql.sql" ## TODO: update ZABBIX Server entry, "update hosts set status=0, host='MyPLC Server' where hostid=10017" fi } function check_templates_and_import () { # LOG IN COOKIE_FILE=/tmp/cookiejar.txt TEMPLATES_DIR=/usr/share/monitor/zabbix/templates curl -s --cookie $COOKIE_FILE --cookie-jar $COOKIE_FILE \ --form "enter=Enter" \ --form "name=Admin" \ --form "password=zabbix" \ "http://${PLC_MONITOR_HOST}/zabbix/index.php?login=1" deleted=$( grep 'deleted' $COOKIE_FILE ) if [ -n "$deleted" ] ; then echo "Login to the zabbix web server failed!!!" return 1 fi for file in ${TEMPLATES_DIR}/*.xml ; do # 0 - update , 1 - skip, 0 - add echo "############### IMPORTING $file" >> /var/log/monitor.log curl -s --cookie $COOKIE_FILE --cookie-jar $COOKIE_FILE \ --form "config=1" \ --form "import_file=@${file}" \ --form "rules[host][exist]=0" \ --form "rules[host][missed]=0" \ --form "rules[template][exist]=0" \ --form "rules[template][missed]=1" \ --form "rules[item][exist]=0" \ --form "rules[item][missed]=0" \ --form "rules[trigger][exist]=0" \ --form "rules[trigger][missed]=0" \ --form "rules[graph][exist]=0" \ --form "rules[graph][missed]=0" \ --form "import=Import" \ "http://${PLC_MONITOR_HOST}/zabbix/exp_imp.php" >> /var/log/monitor.log done } function check_monitor_conf () { MONITOR_CONFIG=/etc/monitor.conf if [ -z "$PLC_MONITOR_IP" ] ; then PLC_MONITOR_IP=$( gethostbyname $PLC_MONITOR_HOST ) fi # Using plcsh add default, monitor user plcsh </dev/null AddPerson({'first_name' : 'Monitor', 'last_name' : 'Server', 'password' : '${PLC_MONITOR_DBPASSWORD}', 'email' : '${PLC_MONITOR_EMAIL}'}) AddRoleToPerson('admin', '${PLC_MONITOR_EMAIL}') AddPersonToSite('${PLC_MONITOR_EMAIL}', '${PLC_SLICE_PREFIX}') UpdatePerson('${PLC_MONITOR_EMAIL}', { 'enabled' : True, 'password' : '${PLC_MONITOR_DBPASSWORD}' }) EOF if [ ! -f ${MONITOR_CONFIG} ] ; then cat < ${MONITOR_CONFIG} [monitorconfig] RT_DB_HOST=rt.planet-lab.org RT_DB_USER= RT_DB_PASSWORD= RT_DB_NAME= # RT Web user account RT_WEB_SERVER=https://rt.planet-lab.org/ RT_WEB_TOOLS_PATH= RT_WEB_USER= RT_WEB_PASSWORD= RT_WEB_DEBUG=0 RT_QUEUE= # PLC admin account API_SERVER=https://${PLC_BOOT_HOST}/PLCAPI/ API_AUTH_USER=${PLC_MONITOR_EMAIL} API_AUTH_PASSWORD=${PLC_MONITOR_DBPASSWORD} # SERVER PATHS MONITOR_SCRIPT_ROOT=/usr/share/monitor-server MONITOR_DATA_ROOT=/var/lib/monitor-server MONITOR_ARCHIVE_ROOT=/var/lib/monitor-server/archive-pdb MONITOR_HOSTNAME=${PLC_MONITOR_HOST} MONITOR_IP=${PLC_MONITOR_IP} # used for debug mode email= # all messages will appear to be from this address from_email=${PLC_MONITOR_EMAIL} # a separate address for support messages support_email=${PLC_MAIL_SUPPORT_ADDRESS} # mailing list copied on all out-going messages cc_email= [monitordatabase] monitor_dburi=postgres://monitoruser:${PLC_MONITOR_DBPASSWORD}@localhost:5432/monitor zabbix_dburi=postgres://zabbixuser:${PLC_MONITOR_DBPASSWORD}@localhost:5432/zabbix cachetime=60 [commandline] echo=False debug=False mail=True bcc=False run=False checkopt=False squeeze=True policysavedb=True EOF fi } function check_zab_server () { ZABBIXCFG=/etc/zabbix TMP_FILE=`mktemp /tmp/zbxtmpXXXXXX` if [ -f ${ZABBIXCFG}/zabbix_server.conf ] ; then sed -e "s/#DBHost=.*/DBHost=$PLC_MONITOR_HOST/g" \ -e "s#DBName=.*#DBName=$ZABBIX_DB_NAME#g" \ -e "s#DBUser=.*#DBUser=$ZABBIX_DB_USER#g" \ -e "s#DBPassword=.*#DBPassword=$PLC_MONITOR_DBPASSWORD#g" \ ${ZABBIXCFG}/zabbix_server.conf > $TMP_FILE cat $TMP_FILE > ${ZABBIXCFG}/zabbix_server.conf fi service zabbix_server start rm -f $TMP_FILE } function check_zab_agentd () { ZABBIXCFG=/etc/zabbix TMP_FILE=`mktemp /tmp/zbxtmpXXXXXX` if [ -f ${ZABBIXCFG}/zabbix_agentd.conf ] ; then HOST=`hostname` sed -e "s#Server=.*#Server=$PLC_MONITOR_HOST#g" \ -e "s#Hostname=.*#Hostname=$HOST#g" \ ${ZABBIXCFG}/zabbix_agentd.conf > $TMP_FILE cat $TMP_FILE > ${ZABBIXCFG}/zabbix_agentd.conf fi service zabbix_agentd start rm -f $TMP_FILE } function check_zab_webconfig() { # SETUP zabbix gui configuration ZABBIX_WEB_CFG=/var/www/html/zabbix/conf/zabbix.conf.php if [ ! -f $ZABBIX_WEB_CFG ] ; then touch $ZABBIX_WEB_CFG cat < $ZABBIX_WEB_CFG EOF chmod 644 $ZABBIX_WEB_CFG fi } if [ "$PLC_MONITOR_ENABLED" != "1" ] ; then exit 0 fi case "$1" in start) MESSAGE=$"Bootstrap Monitoring" dialog "$MESSAGE" # DATABASE acces, creation, and data loading check_pg_hba check_user_and_db check_schema_and_data check_templates_and_import # WRITE default /etc/monitor.conf check_monitor_conf # START zabbix services. SETUP default config files. check_zab_server check_zab_agentd check_zab_webconfig result "$MESSAGE" ;; sync) MESSAGE=$"Syncing PLC db with Zabbix DB" dialog "$MESSAGE" # turn off zabbix server, etc. before writing to the db. service plc stop monitor $MONITORPATH/zabbix/zabbixsync.py --setupglobal &> /var/log/monitor-server # import any templates check_templates_and_import service plc start monitor result "$MESSAGE" ;; stop) MESSAGE=$"Stopping Monitor" dialog "$MESSAGE" service zabbix_server stop service zabbix_agentd stop # TODO: is there anything to stop? result "$MESSAGE" ;; esac exit $ERRORS