mirror of
https://github.com/xjasonlyu/tun2socks.git
synced 2025-10-15 21:30:50 +08:00
39 lines
1.0 KiB
Go
39 lines
1.0 KiB
Go
// +build fakeDNS
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
|
|
"github.com/xjasonlyu/tun2socks/component/fakedns"
|
|
"github.com/xjasonlyu/tun2socks/log"
|
|
)
|
|
|
|
func init() {
|
|
args.EnableFakeDNS = flag.Bool("fakeDNS", false, "Enable fake DNS")
|
|
args.FakeDNSAddr = flag.String("fakeDNSAddr", ":53", "Listen address of fake DNS")
|
|
args.FakeIPRange = flag.String("fakeIPRange", "198.18.0.0/15", "Fake IP CIDR range for DNS")
|
|
args.FakeDNSHosts = flag.String("fakeDNSHosts", "", "DNS hosts mapping, e.g. 'example.com=1.1.1.1,example.net=2.2.2.2'")
|
|
|
|
addPostFlagsInitFn(func() {
|
|
if *args.EnableFakeDNS {
|
|
var err error
|
|
fakeDNS, err = fakedns.NewServer(*args.FakeIPRange, *args.FakeDNSHosts)
|
|
if err != nil {
|
|
log.Fatalf("Create fake DNS server failed: %v", err)
|
|
}
|
|
|
|
// Set fakeDNS variables
|
|
fakedns.ServeAddr = *args.FakeDNSAddr
|
|
|
|
// Start fakeDNS server
|
|
if err := fakeDNS.Start(); err != nil {
|
|
log.Fatalf("Start fake DNS server failed: %v", err)
|
|
}
|
|
log.Infof("Fake DNS serving at %v", fakedns.ServeAddr)
|
|
} else {
|
|
fakeDNS = nil
|
|
}
|
|
})
|
|
}
|