summaryrefslogtreecommitdiff
path: root/main.go
blob: b8242b7b13359e0cff8c5bf79bf8010643a18c71 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
package main

import (
	"github.com/jessevdk/go-flags"
	"log"
	"os"
	"time"
)

type Options struct {
	Verbose			bool	`long:"verbose" short:"v"`
	Watch			bool	`long:"watch" description:"Watch for interface changes"`

	Interface		string	`long:"interface" short:"i" value-name:"IFACE" description:"Use address from interface"`
	InterfaceFamily	Family	`long:"interface-family"`

	Server		string	`long:"server" value-name:"HOST[:PORT]"`
	Timeout		time.Duration `long:"timeout" value-name:"DURATION" default:"10s"`
	Retry		time.Duration `long:"retry" value-name:"DURATION" default:"30s"`
	TSIGName	string	`long:"tsig-name"`
	TSIGSecret	string	`long:"tsig-secret" env:"TSIG_SECRET"`
	TSIGAlgorithm TSIGAlgorithm `long:"tsig-algorithm" default:"hmac-sha1."`

	Zone		string	`long:"zone" description:"Zone to update"`
	TTL			time.Duration `long:"ttl" default:"60s"`

	Args		struct {
		Name		string	`description:"DNS Name to update"`
	} `positional-args:"yes"`
}

func main() {
	var options Options

	if _, err := flags.Parse(&options); err != nil {
		log.Fatalf("flags.Parse: %v", err)
		os.Exit(1)
	}

	var update = Update{
		ttl:	 int(options.TTL.Seconds()),
		timeout: options.Timeout,
		retry:   options.Retry,
		verbose: options.Verbose,
	}

	if err := update.Init(options.Args.Name, options.Zone, options.Server); err != nil {
		log.Fatalf("init: %v", err)
	}

	if options.TSIGSecret != "" {
		var name = options.TSIGName

		if name == "" {
			name = options.Args.Name
		}

		log.Printf("using TSIG: %v (algo=%v)", name, options.TSIGAlgorithm)

		update.InitTSIG(name, options.TSIGSecret, options.TSIGAlgorithm)
	}

	// addrs
	addrs, err := InterfaceAddrs(options.Interface, options.InterfaceFamily)
	if err != nil {
		log.Fatalf("addrs scan: %v", err)
	}

	// update
	update.Start()

	for {
		log.Printf("update...")

		if err := update.Update(addrs); err != nil {
			log.Fatalf("update: %v", err)
		}

		if !options.Watch {
			break
		}

		if err := addrs.Read(); err != nil {
			log.Fatalf("addrs read: %v", err)
		} else {
			log.Printf("addrs update...")
		}
	}

	log.Printf("wait...")

	if err := update.Done(); err != nil {
		log.Printf("update done: %v", err)
	} else {
		log.Printf("update done")
	}
}