source: alternc-slavedns/trunk/slavedns @ 1675

#!/bin/sh
# Synchronize a dns server with alternc's remote server : 
# the configuration files contains definitions for remote masters.

CONFDIR=/etc/slavedns
CACHEDIR=/var/cache/slavedns
BINDDIR=/etc/bind/slavedns
BINDINCLUDE=/etc/bind/slavedns.conf
WGETRC=${HOME}/.wgetrc
WGET=wget
WGETFLAGS=""

NAMED="/etc/init.d/bind restart" 

TIMEOUT=5

CreateBindConf() {
    CFILE="$1"
    rm -f ${BINDDIR}/${CFILE}
    while read domain
    do
      cat >> ${BINDDIR}/${CFILE}.$$ <<EOF
zone "$domain" {
    type slave;
    allow-query { any; };
    file "$domain";
    masters { ${MASTERIP}; };
};
EOF
    done < ${CACHEDIR}/${CFILE}
    mv ${BINDDIR}/${CFILE}.$$ ${BINDDIR}/${CFILE}
    INCLUDE_STR="include \"${BINDDIR}/${CFILE}\";"

    grep -q "${INCLUDE_STR}" ${BINDINCLUDE} || echo ${INCLUDE_STR} >>${BINDINCLUDE}
}

SetWgetPass() {
    USER="$1"
    PASS="$2"
    if [ -e ${WGETRC} ]; then
        mv ${WGETRC} ${WGETRC}.$$
    fi
    touch ${WGETRC}
    chmod og-r ${WGETRC}
    cat >> ${WGETRC} <<EOF
http_user = ${USER}
http_passwd = ${PASS}
EOF
}

ResetWgetConf() {
    mv -f ${WGETRC}.$$ ${WGETRC} 2>/dev/null || rm -f ${WGETRC}
}

# Main procedure : parse each config file and download the raw slave list.
# if something changed in a list, call CreateBindConf $i 

cd $CONFDIR

RELOAD=""

# source defaults
. ./slavedns.conf

for conf in `ls | grep -v slavedns.conf | grep -v '~$' `
do
  # HOST LOGIN PASSWORD MASTERIP
  URL=""
  MASTERIP=""
  LOGIN=""
  PASSWORD=""
  . ./${conf}
  if [ -z "$URL" ]
  then
      if [ -z "$PROTOCOL" ]
      then
          if [ "YES" = "$SSL" ]
          then
              PROTOCOL=https
          else
              PROTOCOL=http
          fi
      fi
      URL=${PROTOCOL}://${HOST}/admin/domlist.php
  fi

  if [ -z "$URL" -a -z "$HOST" -o -z "$LOGIN" -o -z "$PASSWORD" -o -z "$MASTERIP" ]
  then
      echo "error in the config file '${conf}'" >&2
  else
      touch ${CACHEDIR}/${conf}
      rm -f ${CACHEDIR}/${conf}.temp
      SetWgetPass ${LOGIN} ${PASSWORD}
      ${WGET} ${URL} ${WGETFLAGS} -O ${CACHEDIR}/${conf}.temp -t 1 -T ${TIMEOUT} 2>/dev/null
      ResetWgetConf
      if [ -s "${CACHEDIR}/${conf}.temp" ]
      then
          # If the slave file has changed, synchronize it.
          if  [ "$1" != "-f" ] && cmp ${CACHEDIR}/${conf}.temp ${CACHEDIR}/${conf} > /dev/null
          then
              echo "no change found for '${conf}'"
          else
              echo "change detected for '${conf}', applying"
              mv -f ${CACHEDIR}/${conf}.temp ${CACHEDIR}/${conf}
              # Now parse the slave file and send it to /etc/bind/slavedns
              CreateBindConf ${conf} 
              RELOAD="YES"
          fi
      else
          echo "error: downloaded file for '${conf}' has zero size" >&2
      fi
  fi
done     # Main loop on config files

if [ "YES" = "$RELOAD" ]
then
    ${NAMED} >/dev/null
fi