Allow to configure TTL

Author: buglloc

commands/ns.go

@@ -27,6 +27,8 @@ func init() {
 		"your zone name (e.g. 'buglloc.com')")
 	flags.String("ipv4", "127.0.0.1",
 		"default ipv4 address")
+	flags.Uint32("ttl", cfg.TTL,
+		"DNS records TTL")
 	flags.String("ipv6", "::1",
 		"default ipv6 address")
 	flags.String("upstream", "77.88.8.8:53",
@@ -58,5 +60,6 @@ func parseServerConfig(cmd *cobra.Command, args []string) error {
 	cfg.AllowProxy = !viper.GetBool("NoProxy")
 	cfg.UseDefault = viper.GetBool("UseDefault")
 	cfg.Upstream = viper.GetString("Upstream")
+	cfg.TTL = uint32(viper.GetInt("Ttl"))
 	return nil
 }

pkg/cfg/cfg.go

@@ -21,4 +21,5 @@ var (
 	// Enable "strict" mode
 	UseDefault bool
 	AllowProxy bool
+	TTL        uint32 = 0
 )

pkg/handlers/cname.go

@@ -3,6 +3,8 @@ package handlers
 import (
 	"github.com/buglloc/simplelog"
 	"github.com/miekg/dns"
+
+	"github.com/buglloc/rip/pkg/cfg"
 )
 
 func CnameHandler(question dns.Question, name string, l log.Logger) (rrs []dns.RR) {
@@ -12,7 +14,7 @@ func CnameHandler(question dns.Question, name string, l log.Logger) (rrs []dns.R
 			Name:   question.Name,
 			Rrtype: dns.TypeCNAME,
 			Class:  dns.ClassINET,
-			Ttl:    0,
+			Ttl:    cfg.TTL,
 		},
 		Target: dns.Fqdn(subName),
 	}}

pkg/handlers/common.go

@@ -80,7 +80,7 @@ func createIpRR(question dns.Question, ip net.IP) (rr dns.RR) {
 		Name:   question.Name,
 		Rrtype: question.Qtype,
 		Class:  dns.ClassINET,
-		Ttl:    0,
+		Ttl:    cfg.TTL,
 	}
 
 	if question.Qtype == dns.TypeA {