add all files from Hong
This commit is contained in:
zhangsz
83
plugins/dns/plugin.go
Normal file
83
plugins/dns/plugin.go
Normal file
@@ -0,0 +1,83 @@
|
||||
// Copyright 2018-present the CoreDHCP Authors. All rights reserved
|
||||
// This source code is licensed under the MIT license found in the
|
||||
// LICENSE file in the root directory of this source tree.
|
||||
|
||||
package dns
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"net"
|
||||
|
||||
"github.com/coredhcp/coredhcp/handler"
|
||||
"github.com/coredhcp/coredhcp/logger"
|
||||
"github.com/coredhcp/coredhcp/plugins"
|
||||
"github.com/insomniacslk/dhcp/dhcpv4"
|
||||
"github.com/insomniacslk/dhcp/dhcpv6"
|
||||
)
|
||||
|
||||
var log = logger.GetLogger("plugins/dns")
|
||||
|
||||
// Plugin wraps the DNS plugin information.
|
||||
var Plugin = plugins.Plugin{
|
||||
Name: "dns",
|
||||
Setup6: setup6,
|
||||
Setup4: setup4,
|
||||
}
|
||||
|
||||
var (
|
||||
dnsServers6 []net.IP
|
||||
dnsServers4 []net.IP
|
||||
)
|
||||
|
||||
func setup6(args ...string) (handler.Handler6, error) {
|
||||
if len(args) < 1 {
|
||||
return nil, errors.New("need at least one DNS server")
|
||||
}
|
||||
for _, arg := range args {
|
||||
server := net.ParseIP(arg)
|
||||
if server.To16() == nil {
|
||||
return Handler6, errors.New("expected an DNS server address, got: " + arg)
|
||||
}
|
||||
dnsServers6 = append(dnsServers6, server)
|
||||
}
|
||||
log.Infof("loaded %d DNS servers.", len(dnsServers6))
|
||||
return Handler6, nil
|
||||
}
|
||||
|
||||
func setup4(args ...string) (handler.Handler4, error) {
|
||||
log.Printf("loaded plugin for DHCPv4.")
|
||||
if len(args) < 1 {
|
||||
return nil, errors.New("need at least one DNS server")
|
||||
}
|
||||
for _, arg := range args {
|
||||
DNSServer := net.ParseIP(arg)
|
||||
if DNSServer.To4() == nil {
|
||||
return Handler4, errors.New("expected an DNS server address, got: " + arg)
|
||||
}
|
||||
dnsServers4 = append(dnsServers4, DNSServer)
|
||||
}
|
||||
log.Infof("loaded %d DNS servers.", len(dnsServers4))
|
||||
return Handler4, nil
|
||||
}
|
||||
|
||||
// Handler6 handles DHCPv6 packets for the dns plugin
|
||||
func Handler6(req, resp dhcpv6.DHCPv6) (dhcpv6.DHCPv6, bool) {
|
||||
decap, err := req.GetInnerMessage()
|
||||
if err != nil {
|
||||
log.Errorf("Could not decapsulate relayed message, aborting: %v", err)
|
||||
return nil, true
|
||||
}
|
||||
|
||||
if decap.IsOptionRequested(dhcpv6.OptionDNSRecursiveNameServer) {
|
||||
resp.UpdateOption(dhcpv6.OptDNS(dnsServers6...))
|
||||
}
|
||||
return resp, false
|
||||
}
|
||||
|
||||
//Handler4 handles DHCPv4 packets for the dns plugin
|
||||
func Handler4(req, resp *dhcpv4.DHCPv4) (*dhcpv4.DHCPv4, bool) {
|
||||
if req.IsOptionRequested(dhcpv4.OptionDomainNameServer) {
|
||||
resp.Options.Update(dhcpv4.OptDNS(dnsServers4...))
|
||||
}
|
||||
return resp, false
|
||||
}
|
||||
149
plugins/dns/plugin_test.go
Normal file
149
plugins/dns/plugin_test.go
Normal file
@@ -0,0 +1,149 @@
|
||||
// Copyright 2018-present the CoreDHCP Authors. All rights reserved
|
||||
// This source code is licensed under the MIT license found in the
|
||||
// LICENSE file in the root directory of this source tree.
|
||||
|
||||
package dns
|
||||
|
||||
import (
|
||||
"net"
|
||||
"testing"
|
||||
|
||||
"github.com/insomniacslk/dhcp/dhcpv4"
|
||||
"github.com/insomniacslk/dhcp/dhcpv6"
|
||||
)
|
||||
|
||||
func TestAddServer6(t *testing.T) {
|
||||
req, err := dhcpv6.NewMessage()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
req.MessageType = dhcpv6.MessageTypeRequest
|
||||
req.AddOption(dhcpv6.OptRequestedOption(dhcpv6.OptionDNSRecursiveNameServer))
|
||||
|
||||
stub, err := dhcpv6.NewMessage()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
stub.MessageType = dhcpv6.MessageTypeReply
|
||||
|
||||
dnsServers6 = []net.IP{
|
||||
net.ParseIP("2001:db8::1"),
|
||||
net.ParseIP("2001:db8::3"),
|
||||
}
|
||||
|
||||
resp, stop := Handler6(req, stub)
|
||||
if resp == nil {
|
||||
t.Fatal("plugin did not return a message")
|
||||
}
|
||||
|
||||
if stop {
|
||||
t.Error("plugin interrupted processing")
|
||||
}
|
||||
opts := resp.GetOption(dhcpv6.OptionDNSRecursiveNameServer)
|
||||
if len(opts) != 1 {
|
||||
t.Fatalf("Expected 1 RDNSS option, got %d: %v", len(opts), opts)
|
||||
}
|
||||
foundServers := resp.(*dhcpv6.Message).Options.DNS()
|
||||
// XXX: is enforcing the order relevant here ?
|
||||
for i, srv := range foundServers {
|
||||
if !srv.Equal(dnsServers6[i]) {
|
||||
t.Errorf("Found server %s, expected %s", srv, dnsServers6[i])
|
||||
}
|
||||
}
|
||||
if len(foundServers) != len(dnsServers6) {
|
||||
t.Errorf("Found %d servers, expected %d", len(foundServers), len(dnsServers6))
|
||||
}
|
||||
}
|
||||
|
||||
func TestNotRequested6(t *testing.T) {
|
||||
req, err := dhcpv6.NewMessage()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
req.MessageType = dhcpv6.MessageTypeRequest
|
||||
req.AddOption(dhcpv6.OptRequestedOption())
|
||||
|
||||
stub, err := dhcpv6.NewMessage()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
stub.MessageType = dhcpv6.MessageTypeReply
|
||||
|
||||
dnsServers6 = []net.IP{
|
||||
net.ParseIP("2001:db8::1"),
|
||||
}
|
||||
|
||||
resp, stop := Handler6(req, stub)
|
||||
if resp == nil {
|
||||
t.Fatal("plugin did not return a message")
|
||||
}
|
||||
if stop {
|
||||
t.Error("plugin interrupted processing")
|
||||
}
|
||||
|
||||
opts := resp.GetOption(dhcpv6.OptionDNSRecursiveNameServer)
|
||||
if len(opts) != 0 {
|
||||
t.Errorf("RDNSS options were added when not requested: %v", opts)
|
||||
}
|
||||
}
|
||||
|
||||
func TestAddServer4(t *testing.T) {
|
||||
req, err := dhcpv4.NewDiscovery(net.HardwareAddr{0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
stub, err := dhcpv4.NewReplyFromRequest(req)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
dnsServers4 = []net.IP{
|
||||
net.ParseIP("192.0.2.1"),
|
||||
net.ParseIP("192.0.2.3"),
|
||||
}
|
||||
|
||||
resp, stop := Handler4(req, stub)
|
||||
if resp == nil {
|
||||
t.Fatal("plugin did not return a message")
|
||||
}
|
||||
if stop {
|
||||
t.Error("plugin interrupted processing")
|
||||
}
|
||||
servers := resp.DNS()
|
||||
for i, srv := range servers {
|
||||
if !srv.Equal(dnsServers4[i]) {
|
||||
t.Errorf("Found server %s, expected %s", srv, dnsServers4[i])
|
||||
}
|
||||
}
|
||||
if len(servers) != len(dnsServers4) {
|
||||
t.Errorf("Found %d servers, expected %d", len(servers), len(dnsServers4))
|
||||
}
|
||||
}
|
||||
|
||||
func TestNotRequested4(t *testing.T) {
|
||||
req, err := dhcpv4.NewDiscovery(net.HardwareAddr{0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
stub, err := dhcpv4.NewReplyFromRequest(req)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
dnsServers4 = []net.IP{
|
||||
net.ParseIP("192.0.2.1"),
|
||||
}
|
||||
req.UpdateOption(dhcpv4.OptParameterRequestList(dhcpv4.OptionBroadcastAddress))
|
||||
|
||||
resp, stop := Handler4(req, stub)
|
||||
if resp == nil {
|
||||
t.Fatal("plugin did not return a message")
|
||||
}
|
||||
if stop {
|
||||
t.Error("plugin interrupted processing")
|
||||
}
|
||||
servers := dhcpv4.GetIPs(dhcpv4.OptionDomainNameServer, resp.Options)
|
||||
if len(servers) != 0 {
|
||||
t.Errorf("Found %d DNS servers when explicitly not requested", len(servers))
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user