1

#!/bin/bash

     2

# vim: set ft=sh :

     3

     4

set -ue

     5

     6

ROOT=$(pwd)

     7

     8

BIN=bin

     9

PROCESS_ZONE=$BIN/process-zone

    10

EXPAND_ZONE=$BIN/expand-zone

    11

UPDATE_SERIAL=$BIN/update-serial

    12

    13

SETTINGS=settings

    14

ZONES=zones

    15

SERIALS=serials

    16

    17

PROCESS_ARGS='--input-charset latin-1'

    18

    19

FORWARD_MX=mail

    20

REVERSE_ZONE=194.197.235

    21

REVERSE_DOMAIN=paivola.fi

    22

    23

NAMED_CHECKZONE=/usr/sbin/named-checkzone

    24

    25

## options

    26

IS_TTY=

    27

    28

LOG=y

    29

LOG_INFO=

    30

LOG_DEBUG=

    31

LOG_CMD=

    32

    33

UPDATE_FORCE=

    34

UPDATE_NOOP=

    35

UPDATE_DIFF=

    36

SERIAL_NOUPDATE=

    37

    38

function help_args {

    39

    local prog=$1

    40

    41

    cat <<END

    42

Usage: $prog [options]

    43

    44

General:

    45

    -h      display this help text

    46

    47

Logging:    

    48

    -q      quiet

    49

    -v      verbose

    50

    -D      debug

    51

    -C      debug commands

    52

    53

Updates:

    54

    -p      show changes

    55

    -F      force-updates without checking src mtime

    56

    -S      do not update serial

    57

    -n      no-op/mock-update; don't actually change anything; implies -Sp

    58

END

    59

}

    60

    61

function parse_args {

    62

    OPTIND=1

    63

    64

    while getopts 'hqvDCpFSn' opt "$@"; do

    65

        case $opt in

    66

            h)  

    67

                help_args $0

    68

                exit 0

    69

            ;;

    70

    71

            q)  LOG= ;;

    72

            v)  LOG_INFO=y ;;

    77

            C)  LOG_CMD=y ;;

    78

            p)  UPDATE_DIFF=y ;;

    79

            F)  UPDATE_FORCE=y ;;

    80

            S)  SERIAL_NOUPDATE=y ;;

    81

    82

            n)  

    83

                UPDATE_NOOP=y 

    84

                # implies -Sp

    85

                UPDATE_DIFF=y

    86

                SERIAL_NOUPDATE=y

    87

                ;;

    88

    89

    90

            ?)  

    91

                die 

    92

            ;;

    93

        esac

    94

    95

    done

    96

}

    97

    98

## lib

    99

function log_msg {

   100

    echo "$*" >&2

   101

}

   102

   103

function log_color {

   104

    local code=$1; shift

   105

   106

    if [ $IS_TTY ]; then

   107

        echo $'\e[0;'${code}'m'"$*"$'\e[00m' >&2

   108

    else

   109

        echo "$*" >&2

   110

    fi

   111

}

   112

   113

function log_error {

   114

    log_color 31 "$*"

   115

}

   116

   117

function log {

   118

    [ $LOG ] && log_msg "$*" || true

   119

}

   120

   121

function log_info {

   122

    [ $LOG_INFO ] && log_color 36 "  $*" || true

   123

}

   124

   125

function log_debug {

   126

    [ $LOG_DEBUG ] && log_color 32 "    $*" || true

   127

}

   128

   129

function log_cmd {

   130

    [ $LOG_CMD ] && log_color 35 "        \$ $*" || true

   131

}

   132

   133

# XXX: broken

   134

function log_stack {

   135

    local level=1

   136

   137

    while info=$(caller $level); do

   138

        echo $info | read line sub file

   139

   140

        log_msg "$file:$lineno $sub()"

   141

   142

        level=$(($level + 1))

   143

    done

   144

}

   145

   146

function fail {

   147

    func=$(caller 1 | cut -d ' ' -f 2)

   148

   149

    log_error "$func: $*"

   150

   151

    exit 2

   152

}

   153

   154

function die {

   155

    log_error "$*"

   156

    exit 1

   157

}

   158

   159

function cmd {

   160

    log_cmd "$@"

   161

   162

    "$@" || die "Failed"

   163

}

   164

   165

function run_cmd {

   166

    local msg=$1; shift

   167

   168

    log_info "$msg... "

   169

   170

    cmd "$@"

   171

}

   172

   173

function indent () {

   174

    local indent=$1; shift

   175

   176

    "$@" | (

   177

        while read line; do

   178

            echo "$indent$line"

   179

        done

   180

    ) || exit $?

   181

}

   182

   183

## test

   184

[ -d $SETTINGS ] || die "Missing settings: $SETTINGS"

   185

[ -d $SERIALS ] || die "Missing serials: $SERIALS"

   186

[ -d $ZONES ] || die "Missing zones: $ZONES"

   187

   188

## functions

   189

function check_update {

   190

    # target

   191

    local dst=$1; shift

   192

   193

    log_debug "check_update: $dst"

   194

   195

    # need update?

   196

    local update=

   197

   198

    if [ ! -e $dst ] || [ $UPDATE_FORCE ]; then

   199

        log_debug "  update forced"

   200

        update=y

   201

    fi

   202

   203

    # check deps

   204

    for dep in "$@"; do

   205

        # don't bother checking if already figured out

   206

        [ $update ] && continue

   207

   208

        # check

   209

   210

        if [ $dst -ot $dep ]; then

   211

            log_debug "  changed: $dep"

   212

            update=y

   213

        fi

   214

    done

   215

   216

    [ ! $update ] && log_debug "  up-to-date"

   217

   218

    # return

   219

    [ $update ]

   220

}

   221

   222

function do_update {

   223

    local dst=$1; shift

   224

    local tmp=$dst.new

   225

   226

    log_debug "update: $dst"

   227

    cmd "$@" > $tmp

   228

   229

    # compare

   230

    if [ -e $dst ] && [ $UPDATE_DIFF ]; then

   231

        log_debug "  changes:"

   232

   233

        # terse

   234

        indent "        " diff --unified=1 $dst $tmp

   235

    fi

   236

   237

    if [ $UPDATE_NOOP ]; then

   238

        # cleanup

   239

        log_debug "  no-op"

   240

   241

        cmd rm $tmp

   242

    else

   243

        # commit

   244

        log_debug "  update"

   245

   246

        cmd mv $tmp $dst

   247

    fi

   248

}

   249

   250

function update {

   251

    local dst=$1; shift;

   252

   253

    local sep=

   254

    local dep=()

   255

    local cmd=()

   256

   257

    for arg in "$@"; do

   258

        if [ $arg == '--' ]; then

   259

            sep=y

   260

        fi

   261

   262

        if [ $sep ]; then

   263

            cmd=("${cmd[@]:-}" "$arg")

   264

        else

   265

            dep=("${dep[@]:-}" "$arg")

   266

        fi

   267

    done

   268

   269

    [ ! $sep ] && fail "Invalid args given: $@"

   270

   271

    check_update $dst "${dep[@]}" && do_update $dst "${cmd[@]}" || true

   272

}

   273

   275

function update_serial {

   276

    local serial=$1; shift

   277

   278

    local old=$(test -e $serial && cat $serial || echo '')

   279

   280

    log_info "Updating serial: $serial"

   281

   282

    cmd $UPDATE_SERIAL $* $serial

   283

   284

    local new=$(cat $serial)

   285

   286

    log_debug "  $old -> $new"

   287

}

   288

   290

    local zone=$1

   291

    local part=$2

   292

   293

    local name=$zone.zone.$part

   294

    local src=$SETTINGS/$name

   295

    local dst=$ZONES/$name

   296

   297

   298

    if check_update $dst $src; then

   299

        log_info "Copying zone $zone.$part..."

   300

   301

        do_update $dst cat $src

   302

    else

   303

        log_info "Copying zone $zone.$part: not changed"

   304

    fi

   305

}

   306

   307

function update_zone {

   308

    local zone=$1

   309

   310

    local name=$zone.zone

   311

   312

    local out=$ZONES/$name

   313

    local in=$SETTINGS/$zone.zone

   314

    local serial=$SERIALS/$zone.serial

   315

   316

    if check_update $out $in $serial; then

   317

        log_info "Generating $zone zone headers..." 

   318

   319

        do_update $out \

   320

            $EXPAND_ZONE $SETTINGS/$zone.zone   \

   321

                --serial $SERIALS/$zone.serial  \

   322

                --expand zones=$ROOT/$ZONES

   323

    else

   324

        log_info "Generating $zone zone headers: not changed" 

   325

    fi

   326

}

   327

   328

function update_zone_view {

   329

    local zone=$1

   330

    local view=$2

   331

   332

    local name=$view/$zone.zone

   333

   334

    local out=$ZONES/$name

   335

    local in=$SETTINGS/$zone.zone

   336

    local serial=$SERIALS/$zone.serial

   337

   338

    if check_update $out $in $serial; then

   339

        log_info "Generating $zone:$view zone headers..."

   340

   341

        do_update $out \

   342

            $EXPAND_ZONE $SETTINGS/$zone.zone   \

   343

                --serial $SERIALS/$zone.serial  \

   344

                --expand zones=$ROOT/$ZONES     \

   345

                --expand view=$view

   346

    else

   347

        log_info "Generating $zone:$view zone headers: not changed"

   348

    fi

   349

}

   350

   351

function update_hosts {

   352

    local dst=$1; shift

   353

    local src=$1; shift

   354

   355

   356

    if check_update $dst $src; then

   357

        log_info "Generating $dst..."

   358

   359

        do_update $dst $PROCESS_ZONE $PROCESS_ARGS $src "$@"

   360

    else

   361

        log_info "Generating $dst: not changed"

   362

    fi

   363

}

   364

   365

function check_hosts {

   366

    local hosts=$1; shift 1

   367

   368

    local cmd=($PROCESS_ZONE $PROCESS_ARGS $hosts --check-hosts "$@")

   369

   370

    if "${cmd[@]}" -q; then

   371

        log_info "Check $hosts: OK"

   372

    else

   373

        log_error "  Check $hosts: Failed"

   374

   375

        indent "    " "${cmd[@]}"

   376

   377

        exit 1

   378

    fi

   379

}

   380

   381

function check_zone {

   382

    local name=$1

   383

    local file=$2

   384

   385

    local cmd=($NAMED_CHECKZONE $name $file)

   386

   387

    # test

   388

    # XXX: checkzone is very specific about the order of arguments, -q must be first

   389

    if $NAMED_CHECKZONE -q $name $file; then

   390

        log_info "Check $file($name): OK"

   391

    else

   392

        log_error "  Check $file($name): Failed:"

   393

   394

        indent "    " "${cmd[@]}"

   395

   396

        exit 1

   397

    fi

   398

}

   399

   400

function main {

   401

    # test tty

   402

    [ -t 1 ] && IS_TTY=y

   403

   404

    parse_args "$@"

   405

   406

    log "Testing hosts..."

   407

        check_hosts     $SETTINGS/paivola.txt --check-exempt ufc

   408

   409

    log "Generating host zones..."

   410

        update_hosts    $ZONES/external/paivola.zone.hosts  $SETTINGS/paivola.txt --forward-zone

   411

        update_hosts    $ZONES/internal/paivola.zone.hosts  $SETTINGS/paivola.txt --forward-zone --forward-txt --forward-mx $FORWARD_MX

   412

        update_hosts    $ZONES/paivola-reverse.zone.hosts   $SETTINGS/paivola.txt --reverse-zone $REVERSE_ZONE --reverse-domain $REVERSE_DOMAIN

   413

   414

    log "Copying zone parts..."

   419

   420

    log "Updating serials..."

   421

   422

    if [ $SERIAL_NOUPDATE ]; then

   423

        log_info "Skipped"

   424

    else

   425

        update_serial   $SERIALS/paivola.serial

   426

        update_serial   $SERIALS/paivola-reverse.serial

   427

    fi

   428

   429

   430

    log "Updating zones headers..."

   431

        update_zone         paivola-reverse

   432

        update_zone_view    paivola             internal

   433

        update_zone_view    paivola             external

   434

   435

    log "Testing zones..."

   436

        check_zone          paivola.fi                  $ZONES/external/paivola.zone

   437

        check_zone          paivola.fi                  $ZONES/external/paivola.zone

   438

        check_zone          235.197.194.in-addr.arpa    $ZONES/paivola-reverse.zone

   439

   440

}

   441

   442

main "$@"