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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
// 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.
//go:build integration
// +build integration
package main
import (
"fmt"
"log"
"net"
"runtime"
"github.com/insomniacslk/dhcp/dhcpv6"
"github.com/insomniacslk/dhcp/dhcpv6/client6"
"github.com/insomniacslk/dhcp/iana"
"github.com/vishvananda/netns"
"github.com/coredhcp/coredhcp/config"
"github.com/coredhcp/coredhcp/plugins"
"github.com/coredhcp/coredhcp/server"
// Plugins
"github.com/coredhcp/coredhcp/plugins/file"
"github.com/coredhcp/coredhcp/plugins/serverid"
)
var serverConfig = config.Config{
Server6: &config.ServerConfig{
Addresses: []net.UDPAddr{
{
IP: net.ParseIP("ff02::1:2"),
Port: dhcpv6.DefaultServerPort,
Zone: "cdhcp_srv",
},
},
Plugins: []config.PluginConfig{
{Name: "server_id", Args: []string{"LL", "11:22:33:44:55:66"}},
{Name: "file", Args: []string{"./leases-dhcpv6-test.txt"}},
},
},
}
// This function *must* be run in its own routine
// For now this assumes ns are created outside.
// TODO: dynamically create NS and interfaces directly in the test program
func runServer(readyCh chan<- struct{}, nsName string, desiredPlugins []*plugins.Plugin) {
runtime.LockOSThread()
defer runtime.UnlockOSThread()
ns, err := netns.GetFromName(nsName)
if err != nil {
log.Panicf("Netns `%s` not set up: %v", nsName, err)
}
if err := netns.Set(ns); err != nil {
log.Panicf("Failed to switch to netns `%s`: %v", nsName, err)
}
// register plugins
for _, pl := range desiredPlugins {
if err := plugins.RegisterPlugin(pl); err != nil {
log.Panicf("Failed to register plugin `%s`: %v", pl.Name, err)
}
}
// start DHCP server
srv, err := server.Start(&serverConfig)
if err != nil {
log.Panicf("Server could not start: %v", err)
}
readyCh <- struct{}{}
if err := srv.Wait(); err != nil {
log.Panicf("Server errored during run: %v", err)
}
}
// runInNs will execute the provided cmd in the namespace nsName.
// It returns the error status of the cmd. Errors in NS management will panic
func runClient6(nsName, iface string, modifiers ...dhcpv6.Modifier) error {
runtime.LockOSThread()
defer runtime.UnlockOSThread()
backupNS, err := netns.Get()
if err != nil {
panic("Could not save handle to original NS")
}
ns, err := netns.GetFromName(nsName)
if err != nil {
panic("netns not set up")
}
if err := netns.Set(ns); err != nil {
panic(fmt.Sprintf("Couldn't switch to test NS: %v", err))
}
client := client6.NewClient()
_, cErr := client.Exchange(iface, modifiers...)
if netns.Set(backupNS) != nil {
panic("couldn't switch back to original NS")
}
return cErr
}
// Create a server and run a DORA exchange with it
func main() {
readyCh := make(chan struct{}, 1)
go runServer(readyCh,
"coredhcp-direct-upper",
[]*plugins.Plugin{
&serverid.Plugin, &file.Plugin,
},
)
// wait for server to be ready before sending DHCP request
<-readyCh
mac, err := net.ParseMAC("de:ad:be:ef:00:00")
if err != nil {
panic(err)
}
err = runClient6(
"coredhcp-direct-lower", "cdhcp_cli",
dhcpv6.WithClientID(&dhcpv6.DUIDLL{
HWType: iana.HWTypeEthernet,
LinkLayerAddr: mac,
}),
)
if err != nil {
panic(err)
}
}
|
