98 lines
2.3 KiB
Go
98 lines
2.3 KiB
Go
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" value-name:"ipv4|ipv6|all"`
|
|
|
|
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" value-name:"FQDN"`
|
|
TSIGSecret string `long:"tsig-secret" value-name:"BASE-64" env:"TSIG_SECRET"`
|
|
TSIGAlgorithm TSIGAlgorithm `long:"tsig-algorithm" value-name:"hmac-{md5,sha1,sha256,sha512}" default:"hmac-sha1."`
|
|
|
|
Zone string `long:"zone" value-name:"FQDN" description:"Zone to update"`
|
|
TTL time.Duration `long:"ttl" value-name:"DURATION" 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")
|
|
}
|
|
}
|