mirror of https://github.com/ginuerzh/gost
ginuerzh
6 years ago
8 changed files with 24 additions and 304 deletions
@ -1,160 +0,0 @@ |
|||
package gost |
|||
|
|||
import ( |
|||
"bufio" |
|||
"io" |
|||
"net" |
|||
"sync" |
|||
"time" |
|||
|
|||
"github.com/go-gost/log" |
|||
) |
|||
|
|||
// Host is a static mapping from hostname to IP.
|
|||
type Host struct { |
|||
IP net.IP |
|||
Hostname string |
|||
Aliases []string |
|||
} |
|||
|
|||
// NewHost creates a Host.
|
|||
func NewHost(ip net.IP, hostname string, aliases ...string) Host { |
|||
return Host{ |
|||
IP: ip, |
|||
Hostname: hostname, |
|||
Aliases: aliases, |
|||
} |
|||
} |
|||
|
|||
// Hosts is a static table lookup for hostnames.
|
|||
// For each host a single line should be present with the following information:
|
|||
// IP_address canonical_hostname [aliases...]
|
|||
// Fields of the entry are separated by any number of blanks and/or tab characters.
|
|||
// Text from a "#" character until the end of the line is a comment, and is ignored.
|
|||
type Hosts struct { |
|||
hosts []Host |
|||
period time.Duration |
|||
stopped chan struct{} |
|||
mux sync.RWMutex |
|||
} |
|||
|
|||
// NewHosts creates a Hosts with optional list of hosts.
|
|||
func NewHosts(hosts ...Host) *Hosts { |
|||
return &Hosts{ |
|||
hosts: hosts, |
|||
stopped: make(chan struct{}), |
|||
} |
|||
} |
|||
|
|||
// AddHost adds host(s) to the host table.
|
|||
func (h *Hosts) AddHost(host ...Host) { |
|||
h.mux.Lock() |
|||
defer h.mux.Unlock() |
|||
|
|||
h.hosts = append(h.hosts, host...) |
|||
} |
|||
|
|||
// Lookup searches the IP address corresponds to the given host from the host table.
|
|||
func (h *Hosts) Lookup(host string) (ip net.IP) { |
|||
if h == nil || host == "" { |
|||
return |
|||
} |
|||
|
|||
h.mux.RLock() |
|||
defer h.mux.RUnlock() |
|||
|
|||
for _, h := range h.hosts { |
|||
if h.Hostname == host { |
|||
ip = h.IP |
|||
break |
|||
} |
|||
for _, alias := range h.Aliases { |
|||
if alias == host { |
|||
ip = h.IP |
|||
break |
|||
} |
|||
} |
|||
} |
|||
if ip != nil && Debug { |
|||
log.Logf("[hosts] hit: %s %s", host, ip.String()) |
|||
} |
|||
return |
|||
} |
|||
|
|||
// Reload parses config from r, then live reloads the hosts.
|
|||
func (h *Hosts) Reload(r io.Reader) error { |
|||
var period time.Duration |
|||
var hosts []Host |
|||
|
|||
if r == nil || h.Stopped() { |
|||
return nil |
|||
} |
|||
|
|||
scanner := bufio.NewScanner(r) |
|||
for scanner.Scan() { |
|||
line := scanner.Text() |
|||
ss := splitLine(line) |
|||
if len(ss) < 2 { |
|||
continue // invalid lines are ignored
|
|||
} |
|||
|
|||
switch ss[0] { |
|||
case "reload": // reload option
|
|||
period, _ = time.ParseDuration(ss[1]) |
|||
default: |
|||
ip := net.ParseIP(ss[0]) |
|||
if ip == nil { |
|||
break // invalid IP addresses are ignored
|
|||
} |
|||
host := Host{ |
|||
IP: ip, |
|||
Hostname: ss[1], |
|||
} |
|||
if len(ss) > 2 { |
|||
host.Aliases = ss[2:] |
|||
} |
|||
hosts = append(hosts, host) |
|||
} |
|||
} |
|||
if err := scanner.Err(); err != nil { |
|||
return err |
|||
} |
|||
|
|||
h.mux.Lock() |
|||
h.period = period |
|||
h.hosts = hosts |
|||
h.mux.Unlock() |
|||
|
|||
return nil |
|||
} |
|||
|
|||
// Period returns the reload period
|
|||
func (h *Hosts) Period() time.Duration { |
|||
if h.Stopped() { |
|||
return -1 |
|||
} |
|||
|
|||
h.mux.RLock() |
|||
defer h.mux.RUnlock() |
|||
|
|||
return h.period |
|||
} |
|||
|
|||
// Stop stops reloading.
|
|||
func (h *Hosts) Stop() { |
|||
select { |
|||
case <-h.stopped: |
|||
default: |
|||
close(h.stopped) |
|||
} |
|||
} |
|||
|
|||
// Stopped checks whether the reloader is stopped.
|
|||
func (h *Hosts) Stopped() bool { |
|||
select { |
|||
case <-h.stopped: |
|||
return true |
|||
default: |
|||
return false |
|||
} |
|||
} |
|||
@ -1,130 +0,0 @@ |
|||
package gost |
|||
|
|||
import ( |
|||
"bytes" |
|||
"io" |
|||
"net" |
|||
"testing" |
|||
"time" |
|||
) |
|||
|
|||
var hostsLookupTests = []struct { |
|||
hosts []Host |
|||
host string |
|||
ip net.IP |
|||
}{ |
|||
{nil, "", nil}, |
|||
{nil, "example.com", nil}, |
|||
{[]Host{}, "", nil}, |
|||
{[]Host{}, "example.com", nil}, |
|||
{[]Host{NewHost(nil, "")}, "", nil}, |
|||
{[]Host{NewHost(nil, "example.com")}, "example.com", nil}, |
|||
{[]Host{NewHost(net.IPv4(192, 168, 1, 1), "")}, "", nil}, |
|||
{[]Host{NewHost(net.IPv4(192, 168, 1, 1), "example.com")}, "example.com", net.IPv4(192, 168, 1, 1)}, |
|||
{[]Host{NewHost(net.IPv4(192, 168, 1, 1), "example.com")}, "example", nil}, |
|||
{[]Host{NewHost(net.IPv4(192, 168, 1, 1), "example.com", "example", "examples")}, "example", net.IPv4(192, 168, 1, 1)}, |
|||
{[]Host{NewHost(net.IPv4(192, 168, 1, 1), "example.com", "example", "examples")}, "examples", net.IPv4(192, 168, 1, 1)}, |
|||
} |
|||
|
|||
func TestHostsLookup(t *testing.T) { |
|||
for i, tc := range hostsLookupTests { |
|||
hosts := NewHosts() |
|||
hosts.AddHost(tc.hosts...) |
|||
ip := hosts.Lookup(tc.host) |
|||
if !ip.Equal(tc.ip) { |
|||
t.Errorf("#%d test failed: lookup should be %s, got %s", i, tc.ip, ip) |
|||
} |
|||
} |
|||
} |
|||
|
|||
var HostsReloadTests = []struct { |
|||
r io.Reader |
|||
period time.Duration |
|||
host string |
|||
ip net.IP |
|||
stopped bool |
|||
}{ |
|||
{ |
|||
r: nil, |
|||
period: 0, |
|||
host: "", |
|||
ip: nil, |
|||
}, |
|||
{ |
|||
r: bytes.NewBufferString(""), |
|||
period: 0, |
|||
host: "example.com", |
|||
ip: nil, |
|||
}, |
|||
{ |
|||
r: bytes.NewBufferString("reload 10s"), |
|||
period: 10 * time.Second, |
|||
host: "example.com", |
|||
ip: nil, |
|||
}, |
|||
{ |
|||
r: bytes.NewBufferString("#reload 10s\ninvalid.ip.addr example.com"), |
|||
period: 0, |
|||
ip: nil, |
|||
}, |
|||
{ |
|||
r: bytes.NewBufferString("reload 10s\n192.168.1.1"), |
|||
period: 10 * time.Second, |
|||
host: "", |
|||
ip: nil, |
|||
}, |
|||
{ |
|||
r: bytes.NewBufferString("#reload 10s\n192.168.1.1 example.com"), |
|||
period: 0, |
|||
host: "example.com", |
|||
ip: net.IPv4(192, 168, 1, 1), |
|||
}, |
|||
{ |
|||
r: bytes.NewBufferString("#reload 10s\n#192.168.1.1 example.com"), |
|||
period: 0, |
|||
host: "example.com", |
|||
ip: nil, |
|||
stopped: true, |
|||
}, |
|||
{ |
|||
r: bytes.NewBufferString("#reload 10s\n192.168.1.1 example.com example examples"), |
|||
period: 0, |
|||
host: "example", |
|||
ip: net.IPv4(192, 168, 1, 1), |
|||
stopped: true, |
|||
}, |
|||
{ |
|||
r: bytes.NewBufferString("#reload 10s\n192.168.1.1 example.com example examples"), |
|||
period: 0, |
|||
host: "examples", |
|||
ip: net.IPv4(192, 168, 1, 1), |
|||
stopped: true, |
|||
}, |
|||
} |
|||
|
|||
func TestHostsReload(t *testing.T) { |
|||
for i, tc := range HostsReloadTests { |
|||
hosts := NewHosts() |
|||
if err := hosts.Reload(tc.r); err != nil { |
|||
t.Error(err) |
|||
} |
|||
if hosts.Period() != tc.period { |
|||
t.Errorf("#%d test failed: period value should be %v, got %v", |
|||
i, tc.period, hosts.Period()) |
|||
} |
|||
ip := hosts.Lookup(tc.host) |
|||
if !ip.Equal(tc.ip) { |
|||
t.Errorf("#%d test failed: lookup should be %s, got %s", i, tc.ip, ip) |
|||
} |
|||
if tc.stopped { |
|||
hosts.Stop() |
|||
if hosts.Period() >= 0 { |
|||
t.Errorf("period of the stopped reloader should be minus value") |
|||
} |
|||
} |
|||
if hosts.Stopped() != tc.stopped { |
|||
t.Errorf("#%d test failed: stopped value should be %v, got %v", |
|||
i, tc.stopped, hosts.Stopped()) |
|||
} |
|||
} |
|||
} |
|||
Loading…
Reference in new issue