/
setup.go
98 lines (85 loc) · 1.92 KB
/
setup.go
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
98
package ubios
import (
"bytes"
"fmt"
"net"
"os"
"os/exec"
"github.com/nextdns/nextdns/config"
"github.com/nextdns/nextdns/router/internal"
)
type Router struct {
DNSMasqPath string
ListenPort string
ClientReporting bool
}
func isUnifi() bool {
if st, _ := os.Stat("/data/unifi"); st != nil && st.IsDir() {
return true
}
if err := exec.Command("ubnt-device-info", "firmware").Run(); err == nil {
return true
}
return false
}
func New() (*Router, bool) {
if !isUnifi() {
return nil, false
}
return &Router{
DNSMasqPath: "/run/dnsmasq.conf.d/nextdns.conf",
ListenPort: "5342",
}, true
}
func (r *Router) String() string {
return "ubios"
}
func (r *Router) Configure(c *config.Config) error {
if dnsFilterEnabled() {
return fmt.Errorf(`UDM "Content Filtering" feature is enabled. Please disable it to use NextDNS`)
}
c.Listens = []string{net.JoinHostPort("localhost", r.ListenPort)}
r.ClientReporting = c.ReportClientInfo
if c.CacheSize == "0" || c.CacheSize == "" {
// Make sure we setup a non-0 cache as we disable dnsmasq cache
c.CacheSize = "10MB"
}
return nil
}
func (r *Router) Setup() error {
return r.setupDNSMasq()
}
func (r *Router) Restore() error {
if err := os.Remove(r.DNSMasqPath); err != nil {
return err
}
return killDNSMasq()
}
func (r *Router) setupDNSMasq() error {
if err := internal.WriteTemplate(r.DNSMasqPath, tmpl, r, 0644); err != nil {
return err
}
return killDNSMasq()
}
func dnsFilterEnabled() bool {
_, err := os.Stat("/run/dnsfilter/dnsfilter")
return err == nil
}
func killDNSMasq() error {
b, err := os.ReadFile("/run/dnsmasq.pid")
if err != nil {
return err
}
pid := string(bytes.TrimSpace(b))
if err := exec.Command("kill", pid).Run(); err != nil {
return fmt.Errorf("dnsmasq kill: %v", err)
}
return nil
}
var tmpl = `# Configuration generated by NextDNS
no-resolv
server=127.0.0.1#{{.ListenPort}}
add-mac
add-subnet=32,128
max-cache-ttl=0
`