remove dns cache size opt

This commit is contained in:
Jason
2019-08-16 13:06:20 +08:00
parent 065c66c82c
commit 2a7209a0a9
3 changed files with 6 additions and 5 deletions
-1
View File
@@ -49,7 +49,6 @@ type CmdArgs struct {
// FakeDNS
EnableFakeDNS *bool
DNSCacheSize *int
FakeIPRange *string
FakeDNSAddr *string
FakeDNSHosts *string
+1 -2
View File
@@ -11,7 +11,6 @@ import (
func init() {
args.EnableFakeDNS = flag.Bool("fakeDNS", false, "Enable fake DNS")
args.DNSCacheSize = flag.Int("dnsCacheSize", 100, "Size of DNS LRU Cache")
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'")
@@ -19,7 +18,7 @@ func init() {
addPostFlagsInitFn(func() {
if *args.EnableFakeDNS {
var err error
fakeDNS, err = fakedns.NewServer(*args.FakeIPRange, *args.FakeDNSHosts, *args.DNSCacheSize)
fakeDNS, err = fakedns.NewServer(*args.FakeIPRange, *args.FakeDNSHosts)
if err != nil {
log.Fatalf("Create fake DNS server failed: %v", err)
}
+5 -2
View File
@@ -13,6 +13,8 @@ import (
)
const (
dnsCacheSize int = 1000
dnsFakeTTL uint32 = 1
dnsDefaultTTL uint32 = 600
)
@@ -67,12 +69,13 @@ func (s *Server) IPToHost(ip net.IP) (string, bool) {
return s.p.LookBack(ip)
}
func NewServer(fakeIPRange, hosts string, size int) (*Server, error) {
func NewServer(fakeIPRange, hosts string) (*Server, error) {
_, ipnet, err := net.ParseCIDR(fakeIPRange)
if err != nil {
return nil, err
}
pool, err := fakeip.New(ipnet, size)
pool, err := fakeip.New(ipnet, dnsCacheSize)
if err != nil {
return nil, err
}