tests/bigtest/buildzones.sh

#!/bin/bash
#
# Copyright (C) 2013, 2016  Internet Systems Consortium, Inc. ("ISC")
#
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

SYSTEMTESTTOP=..
. ../conf.sh

addr=127.127.0.0
ttl=300
named=${NAMED}
keygen=${KEYGEN}
dsfromkey=${DSFROMKEY}

nextaddr() {
    OLDIF="$IFS"
    IFS="${IFS}."
    set $1
    IFS="$OLDIFS"
    _a=$1 _b=$2 _c=$3 _d=$4
    _d=$(($_d + 1))
    case $_d in
    256) _c=$(($_c + 1)); _d=0;;
    esac
    case $_c in
    256) _b=$(($_b + 1)); _c=0;;
    esac
    echo $_a.$_b.$_c.$_d
}

parent() {
    OLDIF="$IFS"
    IFS="${IFS}."
    set $1
    IFS="$OLDIFS"
    shift
    while [ $# -ne 0 ]
    do
        printf %s ${1}
        shift
        printf %s ${1:+.}

    done
}

blackhole() {
    echo 'options {'
    echo '  port 5300;'
    echo "  listen-on { $1; };"
    echo "  query-source $1;"
    echo "  notify-source $1;"
    echo "  transfer-source $1;"
    echo '  key-directory "keys";'
    echo "  recursion ${2:-no};"
    echo '  pid-file "pids/'"${addr}"'.pid";'
    echo '  blackhole { 127.127.0.0; };'
    echo '};'
}

refuse() {
    echo 'options {'
    echo '  port 5300;'
    echo "  listen-on { $1; };"
    echo "  query-source $1;"
    echo "  notify-source $1;"
    echo "  transfer-source $1;"
    echo '  key-directory "keys";'
    echo "  recursion ${2:-no};"
    echo '  pid-file "pids/'"${addr}"'.pid";'
    echo '  allow-query { !127.127.0.0; any; };'
    echo '};'
}

options() {
    echo 'options {'
    echo '  port 5300;'
    echo "  listen-on { $1; };"
    echo "  query-source $1;"
    echo "  notify-source $1;"
    echo "  transfer-source $1;"
    echo '  key-directory "keys";'
    echo "  recursion ${2:-no};"
    echo '  pid-file "pids/'"${addr}"'.pid";'
    echo '};'
}

controls() {
       echo 'include "rndc.key";'
       echo "controls { inet $addr port 9953 allow { any; } keys { "rndc-key"; }; };"
}

delay() {
    _s=$1
    OLDIF="$IFS"
    IFS="${IFS}/"
    set ${2:-.}
    IFS="$OLDIFS"

    case $1 in
    .) _d=;;
    *) _d=$1;;
    esac
    case $_s in
    1) echo -T delay=${_d:-100};;
    2) echo -T delay=${2:-50};;
    3) echo -T delay=${3:-150};;
    4) echo -T delay=${4:-250};;
    5) echo -T delay=${5:-125};;
    6) echo -T delay=${6:-25};;
    7) echo -T delay=${7:-75};;
    8) echo -T delay=${8:-125};;
    9) echo -T delay=${9:-10};;
    10) echo -T delay=${10:-40};;
    11) echo -T delay=${11:-80};;
    12) echo -T delay=${12:-90};;
    *) echo -T delay=50;;
    esac
}

trusted-keys () {
    awk '$3 == "DNSKEY" {
            b = ""; for (i=7; i <= NF; i++) { b = b $i; };
        print "trusted-keys { \""$1"\"",$4,$5,$6,"\""b"\"; };" };'
}

signed-zone () {
    echo "zone "'"'"${1:-.}"'"'" {"
    echo "  type master;"
    echo "  file "'"'"master/${2}.db"'"'";"
    echo "  auto-dnssec maintain;"
    echo "  allow-update { any; };"
    echo "};"
}

unsigned-zone () {
    echo "zone "'"'"${1:-.}"'"'" {"
    echo "  type master;"
    echo "  file "'"'"master/${2}.db"'"'";"
    echo "};"
}

slave-zone () {
    echo "zone "'"'"${zone:-.}"'"'" {"
    echo "  type slave;"
    echo "  masters { ${master}; };"
    echo "};"
}

rm -rf servers master keys setup teardown run
mkdir -p servers
mkdir -p master
mkdir -p keys

echo "ifconfig lo0 $addr netmask 0xffffffff alias" >> setup
echo "ifconfig lo0 $addr -alias" >> teardown
controls $addr > named.conf
options $addr yes >> named.conf
echo 'zone "." { type hint; file "master/hint.db"; };' >> named.conf

while read zone servers nsfmt signed delay blackhole refuse flags
do
    i=1
    case "${zone}" in
    .) file=root zone=;;
    *) file="$zone";;
    esac
    if [ "${zone}" != "" ] ; then
        p=$(parent $zone)
        case "${p}" in
        "") p=root;;
        esac
    else
        p=hint
    fi
    #echo "zone='${zone}' parent='${p}'"
    addr=$(nextaddr $addr)
    ns=$(printf "$nsfmt" ${i} "${zone}")
    d=$(delay $i ${delay:-.})

    echo "${zone}. ${ttl} soa ${ns}. hostmaster.${zone}${zone:+.} 1 3600 1200 604800 1200" >> master/${file}.db
    echo "${zone}. ${ttl} ns ${ns}." >> master/${file}.db
    echo "${ns}. ${ttl} a ${addr}" >> master/${file}.db
    echo "${zone}. ${ttl} ns ${ns}." >> master/${p}.db
    echo "${ns}. ${ttl} a ${addr}" >> master/${p}.db
    if [ $signed = "S" ]; then
        kskkey=`${keygen} -K keys -f KSK ${zone:-.}`
        zskkey=`${keygen} -K keys ${zone:-.}`
        if [ "${zone}" != "" ] ; then
            ${dsfromkey} -T ${ttl} keys/${kskkey}.key >> master/${p}.db
        else
            trusted-keys <  keys/${kskkey}.key >> named.conf
        fi
    fi
    echo "ifconfig lo0 $addr netmask 0xffffffff alias" >> setup
    echo "ifconfig lo0 $addr -alias" >> teardown
    echo "${named} -D bigtest -c servers/${addr}.conf $d $flags" >> run
    options ${addr} > servers/${addr}.conf
    case ${signed} in
    S) signed-zone ${zone:-.} ${file} >> servers/${addr}.conf;;
    P) unsigned-zone ${zone:-.} ${file} >> servers/${addr}.conf;;
    *) echo ${signed}; exit 1;;
    esac

    # slave servers
    while [ $i -lt $servers ]
    do
        master=$addr
        i=$(($i + 1))
        ns=$(printf "$nsfmt" ${i} "${zone}")
        d=$(delay $i ${delay:-.})
        addr=$(nextaddr $addr)
        echo "${zone}. ${ttl} ns ${ns}." >> master/${file}.db
        echo "${ns}. ${ttl} a ${addr}" >> master/${file}.db
        echo "${zone}. ${ttl} ns ${ns}." >> master/${p}.db
        echo "${ns}. ${ttl} a ${addr}" >> master/${p}.db
        echo "ifconfig lo0 $addr netmask 0xffffffff alias" >> setup
        echo "ifconfig lo0 $addr -alias" >> teardown
        echo "${named} -D bigtest -c servers/${addr}.conf $d $flags" >> run
        if [ $i = ${refuse:-.} ]
        then
            refuse $addr > servers/${addr}.conf
        elif [ $i = ${blackhole:-.} ]
        then
            blackhole $addr > servers/${addr}.conf
        else
            options $addr > servers/${addr}.conf
        fi
        slave-zone ${zone:-.} ${master} >> servers/${addr}.conf
    done
    if [ "${zone}" != "" ] ; then
        echo "www.${zone}. ${ttl} a 127.0.0.1" >> master/${file}.db
        echo "www.${zone}. ${ttl} aaaa ::1" >> master/${file}.db
        echo "${zone}. ${ttl} mx 10 mail.${zone}." >> master/${file}.db
        echo "mail.${zone}. ${ttl} a 127.0.0.1" >> master/${file}.db
        echo "mail.${zone}. ${ttl} aaaa ::1" >> master/${file}.db
        echo "*.big.${zone}. ${ttl} txt (" >> master/${file}.db
        i=0
        while [ $i -lt 150 ]
        do
            echo "1234567890" >> master/${file}.db
            i=$(($i + 1))
        done
        echo ")" >> master/${file}.db
        echo "*.medium.${zone}. ${ttl} txt (" >> master/${file}.db
        i=0
        while [ $i -lt 120 ]
        do
            echo "1234567890" >> master/${file}.db
            i=$(($i + 1))
        done
        echo ")" >> master/${file}.db
        echo "*.medium.${zone}. ${ttl} txt (" >> master/${file}.db
        i=0
        while [ $i -lt 120 ]
        do
            echo "1234567890" >> master/${file}.db
            i=$(($i + 1))
        done
        echo ")" >> master/${file}.db
    fi
done