initial networking setup
This commit is contained in:
parent
3008eec5b5
commit
14476852d4
4 changed files with 583 additions and 40 deletions
|
@ -5,6 +5,16 @@ log_level = "TRACE"
|
||||||
|
|
||||||
plugin "hello-driver" {
|
plugin "hello-driver" {
|
||||||
config {
|
config {
|
||||||
shell = "bash"
|
bridge = "mybridge"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
plugin "raw_exec" {
|
||||||
|
config {
|
||||||
|
enabled = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
consul {
|
||||||
|
address = "127.0.0.1:8500"
|
||||||
|
}
|
||||||
|
|
|
@ -3,15 +3,33 @@
|
||||||
|
|
||||||
job "example" {
|
job "example" {
|
||||||
datacenters = ["dc1"]
|
datacenters = ["dc1"]
|
||||||
type = "batch"
|
type = "service"
|
||||||
|
|
||||||
group "example" {
|
group "example" {
|
||||||
|
network {
|
||||||
|
# mode = "driver"
|
||||||
|
port "http" {
|
||||||
|
to = 8080
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
task "hello-world" {
|
task "hello-world" {
|
||||||
driver = "hello-world-example"
|
driver = "hello-world-example"
|
||||||
|
|
||||||
config {
|
config {
|
||||||
greeting = "hello"
|
unikernel = "/tmp/hello-key.hvt"
|
||||||
}
|
# network "device1" {
|
||||||
|
# ports = ["http", ...]
|
||||||
|
# ip_arg = "--monitoring"
|
||||||
|
# gateway_arg = "--xxx"
|
||||||
|
# }
|
||||||
|
# args = [...]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
service {
|
||||||
|
port = "http"
|
||||||
|
name = "hello-service"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
583
hello/driver.go
583
hello/driver.go
|
@ -5,12 +5,17 @@ package hello
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"crypto/md5"
|
||||||
|
// "encoding/base64"
|
||||||
|
"encoding/json"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"regexp"
|
"regexp"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/hashicorp/consul-template/signals"
|
"github.com/hashicorp/consul-template/signals"
|
||||||
|
@ -42,6 +47,15 @@ const (
|
||||||
// this is used to allow modification and migration of the task schema
|
// this is used to allow modification and migration of the task schema
|
||||||
// used by the plugin
|
// used by the plugin
|
||||||
taskHandleVersion = 1
|
taskHandleVersion = 1
|
||||||
|
|
||||||
|
// TODO: should these be made configurable?
|
||||||
|
tapDeviceBaseName = "miragetap"
|
||||||
|
iptablesChainBase = "MIRAGE"
|
||||||
|
|
||||||
|
// TODO: make this configurable
|
||||||
|
// addresses in the bridge are <bridgeRangeHi>.<bridgeRangeLo>.0.0/16
|
||||||
|
bridgeRangeHi = 10
|
||||||
|
bridgeRangeLo = 99
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
|
@ -71,9 +85,9 @@ var (
|
||||||
// shell = "fish"
|
// shell = "fish"
|
||||||
// }
|
// }
|
||||||
// }
|
// }
|
||||||
"shell": hclspec.NewDefault(
|
"bridge": hclspec.NewDefault(
|
||||||
hclspec.NewAttr("shell", "string", false),
|
hclspec.NewAttr("bridge", "string", false),
|
||||||
hclspec.NewLiteral(`"bash"`),
|
hclspec.NewLiteral(`"mirage"`),
|
||||||
),
|
),
|
||||||
})
|
})
|
||||||
|
|
||||||
|
@ -99,9 +113,9 @@ var (
|
||||||
// }
|
// }
|
||||||
// }
|
// }
|
||||||
// }
|
// }
|
||||||
"greeting": hclspec.NewDefault(
|
"unikernel": hclspec.NewDefault(
|
||||||
hclspec.NewAttr("greeting", "string", false),
|
hclspec.NewAttr("unikernel", "string", true),
|
||||||
hclspec.NewLiteral(`"Hello, World!"`),
|
hclspec.NewLiteral(`""`),
|
||||||
),
|
),
|
||||||
})
|
})
|
||||||
|
|
||||||
|
@ -115,6 +129,11 @@ var (
|
||||||
// https://godoc.org/github.com/hashicorp/nomad/plugins/drivers#Capabilities
|
// https://godoc.org/github.com/hashicorp/nomad/plugins/drivers#Capabilities
|
||||||
SendSignals: true,
|
SendSignals: true,
|
||||||
Exec: false,
|
Exec: false,
|
||||||
|
FSIsolation: drivers.FSIsolationNone,
|
||||||
|
NetIsolationModes: []drivers.NetIsolationMode{
|
||||||
|
drivers.NetIsolationModeTask,
|
||||||
|
},
|
||||||
|
MountConfigs: drivers.MountConfigSupportNone,
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -125,7 +144,7 @@ type Config struct {
|
||||||
// This struct is the decoded version of the schema defined in the
|
// This struct is the decoded version of the schema defined in the
|
||||||
// configSpec variable above. It's used to convert the HCL configuration
|
// configSpec variable above. It's used to convert the HCL configuration
|
||||||
// passed by the Nomad agent into Go contructs.
|
// passed by the Nomad agent into Go contructs.
|
||||||
Shell string `codec:"shell"`
|
Bridge string `codec:"bridge"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// TaskConfig contains configuration information for a task that runs with
|
// TaskConfig contains configuration information for a task that runs with
|
||||||
|
@ -136,7 +155,7 @@ type TaskConfig struct {
|
||||||
// This struct is the decoded version of the schema defined in the
|
// This struct is the decoded version of the schema defined in the
|
||||||
// taskConfigSpec variable above. It's used to convert the string
|
// taskConfigSpec variable above. It's used to convert the string
|
||||||
// configuration for the task into Go contructs.
|
// configuration for the task into Go contructs.
|
||||||
Greeting string `codec:"greeting"`
|
Unikernel string `codec:"unikernel"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// TaskState is the runtime state which is encoded in the handle returned to
|
// TaskState is the runtime state which is encoded in the handle returned to
|
||||||
|
@ -157,6 +176,19 @@ type TaskState struct {
|
||||||
// in-memory representation of the running tasks using the RecoverTask()
|
// in-memory representation of the running tasks using the RecoverTask()
|
||||||
// method below.
|
// method below.
|
||||||
Pid int
|
Pid int
|
||||||
|
NetDevices map[string]NetDeviceConfig
|
||||||
|
}
|
||||||
|
|
||||||
|
type NetDeviceConfig struct {
|
||||||
|
TapDevice string
|
||||||
|
Ip string
|
||||||
|
Mac string
|
||||||
|
Ports []NetDevicePortConfig
|
||||||
|
}
|
||||||
|
|
||||||
|
type NetDevicePortConfig struct {
|
||||||
|
From int // port on the host
|
||||||
|
To int // port inside the bridge
|
||||||
}
|
}
|
||||||
|
|
||||||
// HelloDriverPlugin is an example driver plugin. When provisioned in a job,
|
// HelloDriverPlugin is an example driver plugin. When provisioned in a job,
|
||||||
|
@ -187,6 +219,18 @@ type HelloDriverPlugin struct {
|
||||||
logger hclog.Logger
|
logger hclog.Logger
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// A unikernel manifest. Obtained by deserializing JSON returned by solo5-elftool.
|
||||||
|
type unikernelManifest struct {
|
||||||
|
Type string `json:"type"`
|
||||||
|
Version int `json:"version"`
|
||||||
|
Devices []unikernelDevice `json:"devices"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type unikernelDevice struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Type string `json:"type"` // BLOCK_BASIC, NET_BASIC
|
||||||
|
}
|
||||||
|
|
||||||
// NewPlugin returns a new example driver plugin
|
// NewPlugin returns a new example driver plugin
|
||||||
func NewPlugin(logger hclog.Logger) drivers.DriverPlugin {
|
func NewPlugin(logger hclog.Logger) drivers.DriverPlugin {
|
||||||
ctx, cancel := context.WithCancel(context.Background())
|
ctx, cancel := context.WithCancel(context.Background())
|
||||||
|
@ -212,6 +256,22 @@ func (d *HelloDriverPlugin) ConfigSchema() (*hclspec.Spec, error) {
|
||||||
return configSpec, nil
|
return configSpec, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func gatewayIp() string {
|
||||||
|
return fmt.Sprintf("%d.%d.0.1", bridgeRangeHi, bridgeRangeLo)
|
||||||
|
}
|
||||||
|
|
||||||
|
func broadcastIp() string {
|
||||||
|
return fmt.Sprintf("%d.%d.0.255", bridgeRangeHi, bridgeRangeLo)
|
||||||
|
}
|
||||||
|
|
||||||
|
func iptablesChainOut() string {
|
||||||
|
return iptablesChainBase + "_OUT"
|
||||||
|
}
|
||||||
|
|
||||||
|
func iptablesChainIn() string {
|
||||||
|
return iptablesChainBase + "_IN"
|
||||||
|
}
|
||||||
|
|
||||||
// SetConfig is called by the client to pass the configuration for the plugin.
|
// SetConfig is called by the client to pass the configuration for the plugin.
|
||||||
func (d *HelloDriverPlugin) SetConfig(cfg *base.Config) error {
|
func (d *HelloDriverPlugin) SetConfig(cfg *base.Config) error {
|
||||||
var config Config
|
var config Config
|
||||||
|
@ -233,10 +293,10 @@ func (d *HelloDriverPlugin) SetConfig(cfg *base.Config) error {
|
||||||
//
|
//
|
||||||
// In the example below we check if the shell specified by the user is
|
// In the example below we check if the shell specified by the user is
|
||||||
// supported by the plugin.
|
// supported by the plugin.
|
||||||
shell := d.config.Shell
|
// shell := d.config.Shell
|
||||||
if shell != "bash" && shell != "fish" {
|
// if shell != "bash" && shell != "fish" {
|
||||||
return fmt.Errorf("invalid shell %s", d.config.Shell)
|
// return fmt.Errorf("invalid shell %s", d.config.Shell)
|
||||||
}
|
// }
|
||||||
|
|
||||||
// Save the Nomad agent configuration
|
// Save the Nomad agent configuration
|
||||||
if cfg.AgentConfig != nil {
|
if cfg.AgentConfig != nil {
|
||||||
|
@ -248,6 +308,40 @@ func (d *HelloDriverPlugin) SetConfig(cfg *base.Config) error {
|
||||||
// Here you can use the config values to initialize any resources that are
|
// Here you can use the config values to initialize any resources that are
|
||||||
// shared by all tasks that use this driver, such as a daemon process.
|
// shared by all tasks that use this driver, such as a daemon process.
|
||||||
|
|
||||||
|
// create the bridge and iptables chains if they do not exist
|
||||||
|
// TODO: handle errors
|
||||||
|
|
||||||
|
// echo "1" > /proc/sys/net/ipv4/ip_forward
|
||||||
|
// ip link add name mybridge type bridge
|
||||||
|
// ip link set dev mybridge up
|
||||||
|
// ip address add 10.0.0.1/24 broadcast 10.0.0.255 dev mybridge
|
||||||
|
exec.Command("sh", "-c", `echo 1 > /proc/sys/net/ipv4/ip_forward`).Run()
|
||||||
|
exec.Command("ip", "link", "add", "name", d.config.Bridge, "type", "bridge").Run()
|
||||||
|
exec.Command("ip", "link", "set", "dev", d.config.Bridge, "up").Run()
|
||||||
|
// TODO: remove broadcast?
|
||||||
|
exec.Command("ip", "address", "add", gatewayIp()+"/16", "broadcast",
|
||||||
|
broadcastIp(), "dev", d.config.Bridge).Run()
|
||||||
|
|
||||||
|
// iptables -t nat -N MIRAGE_OUT
|
||||||
|
// iptables -t nat -A POSTROUTING -j MIRAGE_OUT
|
||||||
|
// + a rule for each unikernel...
|
||||||
|
exec.Command("iptables", "-t", "nat", "-N", iptablesChainOut()).Run()
|
||||||
|
// FIXME: this one gets added unconditionally even if it already exists
|
||||||
|
exec.Command("iptables", "-t", "nat", "-A", "POSTROUTING", "-j", iptablesChainOut()).Run()
|
||||||
|
|
||||||
|
// iptables -t nat -N MIRAGE_IN
|
||||||
|
// iptables -t nat -A PREROUTING -m addrtype --dst-type LOCAL -j MIRAGE_IN
|
||||||
|
// + a rule for each unikernel...
|
||||||
|
exec.Command("iptables", "-t", "nat", "-N", iptablesChainIn()).Run()
|
||||||
|
// FIXME: this one gets added unconditionally even if it already exists
|
||||||
|
exec.Command("iptables", "-t", "nat", "-A", "PREROUTING",
|
||||||
|
"-m", "addrtype", "--dst-type", "LOCAL",
|
||||||
|
"-j", iptablesChainIn()).Run()
|
||||||
|
|
||||||
|
// TODO:
|
||||||
|
// iptables -t nat -I OUTPUT -o lo -j MIRAGE_IN
|
||||||
|
// sysctl -w net.ipv4.conf.all.route_localnet=1
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -311,34 +405,392 @@ func (d *HelloDriverPlugin) buildFingerprint() *drivers.Fingerprint {
|
||||||
// the node in which the plugin is running (specific library availability,
|
// the node in which the plugin is running (specific library availability,
|
||||||
// installed versions of a software etc.). These attributes can then be
|
// installed versions of a software etc.). These attributes can then be
|
||||||
// used by an operator to set job constrains.
|
// used by an operator to set job constrains.
|
||||||
//
|
|
||||||
// In the example below we check if the shell specified by the user exists
|
|
||||||
// in the node.
|
|
||||||
shell := d.config.Shell
|
|
||||||
|
|
||||||
cmd := exec.Command("which", shell)
|
// FIXME: is this the right place to check this?
|
||||||
if err := cmd.Run(); err != nil {
|
// do we need to check for the binaries existence every N seconds?
|
||||||
|
// should we do these checks in SetConfig() instead?
|
||||||
|
|
||||||
|
// check that solo5-hvt exists
|
||||||
|
hvtCmd := exec.Command("which", "solo5-hvt")
|
||||||
|
hvtPath, hvtErr := hvtCmd.Output()
|
||||||
|
if hvtErr != nil {
|
||||||
return &drivers.Fingerprint{
|
return &drivers.Fingerprint{
|
||||||
Health: drivers.HealthStateUndetected,
|
Health: drivers.HealthStateUndetected,
|
||||||
HealthDescription: fmt.Sprintf("shell %s not found", shell),
|
HealthDescription: "solo5-hvt not found",
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// We also set the shell and its version as attributes
|
// check that solo5-elftool exists
|
||||||
cmd = exec.Command(shell, "--version")
|
elftoolCmd := exec.Command("which", "solo5-elftool")
|
||||||
if out, err := cmd.Output(); err != nil {
|
elftoolPath, elftoolErr := elftoolCmd.Output()
|
||||||
d.logger.Warn("failed to find shell version: %v", err)
|
if elftoolErr != nil {
|
||||||
} else {
|
return &drivers.Fingerprint{
|
||||||
re := regexp.MustCompile("[0-9]\\.[0-9]\\.[0-9]")
|
Health: drivers.HealthStateUndetected,
|
||||||
version := re.FindString(string(out))
|
HealthDescription: fmt.Sprintf("solo5-elftool not found"),
|
||||||
|
|
||||||
fp.Attributes["driver.hello.shell_version"] = structs.NewStringAttribute(version)
|
|
||||||
fp.Attributes["driver.hello.shell"] = structs.NewStringAttribute(shell)
|
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// We set their paths as attributes
|
||||||
|
fp.Attributes["driver.hello.solo5-hvt"] = structs.NewStringAttribute(string(hvtPath))
|
||||||
|
fp.Attributes["driver.hello.solo5-elftool"] = structs.NewStringAttribute(string(elftoolPath))
|
||||||
|
|
||||||
return fp
|
return fp
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func getFreshTapDeviceName() (string, error) {
|
||||||
|
cmd := exec.Command("ip", "tuntap", "show")
|
||||||
|
if err := cmd.Run(); err != nil {
|
||||||
|
return "", fmt.Errorf("running '%s' failed: %v", cmd.String(), err)
|
||||||
|
}
|
||||||
|
tuntapOut, _ := cmd.Output()
|
||||||
|
|
||||||
|
var tuntapLines []string
|
||||||
|
if string(tuntapOut) == "" {
|
||||||
|
tuntapLines = []string{}
|
||||||
|
} else {
|
||||||
|
tuntapLines = strings.Split(string(tuntapOut), "\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
maxId := 0
|
||||||
|
for _, line := range(tuntapLines) {
|
||||||
|
tapName, _, found := strings.Cut(line, ":")
|
||||||
|
if !found {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
tapIdStr, found := strings.CutPrefix(tapName, tapDeviceBaseName)
|
||||||
|
if !found {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
tapId, err := strconv.Atoi(tapIdStr)
|
||||||
|
if err != nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if tapId > maxId {
|
||||||
|
maxId = tapId
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return fmt.Sprintf("%s%d", tapDeviceBaseName, maxId+1), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func destroyTapDevice(name string) {
|
||||||
|
exec.Command("ip", "tuntap", "del", name, "mode", "tap").Run()
|
||||||
|
}
|
||||||
|
|
||||||
|
func createTapDevice(name string, bridge string) error {
|
||||||
|
// ip tuntap add <tap> mode tap
|
||||||
|
cmd := exec.Command("ip", "tuntap", "add", name, "mode", "tap")
|
||||||
|
if err := cmd.Run(); err != nil {
|
||||||
|
return fmt.Errorf("'%s' failed: %v", cmd.String(), err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ip link set dev <tap> up
|
||||||
|
cmd = exec.Command("ip", "link", "set", "dev", name, "up")
|
||||||
|
if err := cmd.Run(); err != nil {
|
||||||
|
destroyTapDevice(name)
|
||||||
|
return fmt.Errorf("'%s' failed: %v", cmd.String(), err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ip link set dev <tap> master <bridge>
|
||||||
|
cmd = exec.Command("ip", "link", "set", "dev", name, "master", bridge)
|
||||||
|
if err := cmd.Run(); err != nil {
|
||||||
|
destroyTapDevice(name)
|
||||||
|
return fmt.Errorf("'%s' failed: %v", cmd.String(), err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func getMacAddr(unikernel string, bridge string) string {
|
||||||
|
// deterministic mac address computation
|
||||||
|
// follow Albatross and use the VEB Kombinat Robotron prefix
|
||||||
|
// (thanks again, friends!)
|
||||||
|
ours := md5.Sum([]byte(bridge + unikernel))
|
||||||
|
return fmt.Sprintf(
|
||||||
|
"%x:%x:%x:%x:%x:%x",
|
||||||
|
0x00, 0x80, 0x41,
|
||||||
|
ours[0], ours[1], ours[2],
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
func nextIp(ip [2]int) ([2]int, error) {
|
||||||
|
nip := [2]int{ip[0], ip[1]}
|
||||||
|
// FIX: 255.255 is for broadcast
|
||||||
|
// .255 is for broadcast
|
||||||
|
if nip[1] < 255 {
|
||||||
|
nip[1]++
|
||||||
|
return nip, nil
|
||||||
|
}
|
||||||
|
nip[1] = 0
|
||||||
|
nip[0]++
|
||||||
|
|
||||||
|
if nip[0] < 256 {
|
||||||
|
return nip, nil
|
||||||
|
}
|
||||||
|
return [2]int{}, fmt.Errorf("nextIp")
|
||||||
|
}
|
||||||
|
|
||||||
|
func getFreshIp() (string, error) {
|
||||||
|
// iptables -t nat -L MIRAGE_OUT
|
||||||
|
// then grep for: XX.XX.XX.XX to find existing ips
|
||||||
|
// (where the first two bytes correspond to the range of our bridge)
|
||||||
|
cmd := exec.Command("iptables", "-t", "nat", "-L", iptablesChainOut())
|
||||||
|
if err := cmd.Run(); err != nil {
|
||||||
|
return "", fmt.Errorf("'%s' failed: %v", cmd.String(), err)
|
||||||
|
}
|
||||||
|
chainStr, _ := cmd.Output()
|
||||||
|
chainLines := strings.Split(string(chainStr), "\n")
|
||||||
|
|
||||||
|
r, _ := regexp.Compile(
|
||||||
|
strconv.Itoa(bridgeRangeHi) + "." +
|
||||||
|
strconv.Itoa(bridgeRangeLo) + `.(\d+).(\d+)`)
|
||||||
|
|
||||||
|
usedIps := make(map[[2]int]bool)
|
||||||
|
for i := 2; i < len(chainLines); i++ {
|
||||||
|
line := chainLines[i]
|
||||||
|
matches := r.FindStringSubmatch(line)
|
||||||
|
b1, _ := strconv.Atoi(matches[1])
|
||||||
|
b2, _ := strconv.Atoi(matches[2])
|
||||||
|
usedIps[[2]int{b1, b2}] = true
|
||||||
|
}
|
||||||
|
|
||||||
|
// XX.XX.0.1 is the gateway; start at XX.XX.0.2
|
||||||
|
ip, err := [2]int{0, 2}, error(nil)
|
||||||
|
for err == nil {
|
||||||
|
if !usedIps[ip] {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
ip, err = nextIp(ip)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("no available ip found")
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("%d.%d.%d.%d", bridgeRangeHi, bridgeRangeLo, ip[0], ip[1]), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func getNetDevicesPorts(
|
||||||
|
netDevicesNames []string,
|
||||||
|
taskCfg TaskConfig,
|
||||||
|
cfg *drivers.TaskConfig,
|
||||||
|
) map[string][]NetDevicePortConfig {
|
||||||
|
ports := make(map[string][]NetDevicePortConfig)
|
||||||
|
|
||||||
|
if len(netDevicesNames) == 0 {
|
||||||
|
return ports
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: extract port<->device assignation from taskCfg
|
||||||
|
// for now, assume that this config is empty, and implement the default
|
||||||
|
// strategy
|
||||||
|
|
||||||
|
for _, name := range netDevicesNames {
|
||||||
|
ports[name] = []NetDevicePortConfig{}
|
||||||
|
}
|
||||||
|
|
||||||
|
var defaultDevice string
|
||||||
|
if _, found := ports["service"]; found {
|
||||||
|
// use 'service' as the default device name when it exists
|
||||||
|
defaultDevice = "service"
|
||||||
|
} else {
|
||||||
|
// otherwise use the first one in the list
|
||||||
|
defaultDevice = netDevicesNames[0]
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, portMapping := range *cfg.Resources.Ports {
|
||||||
|
// only the default case for now
|
||||||
|
ports[defaultDevice] = append(ports[defaultDevice],
|
||||||
|
NetDevicePortConfig{
|
||||||
|
From: portMapping.Value,
|
||||||
|
To: portMapping.To,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return ports
|
||||||
|
}
|
||||||
|
|
||||||
|
// fields are "" if not specified
|
||||||
|
type NetDeviceArgs struct {
|
||||||
|
Ip string
|
||||||
|
Gateway string
|
||||||
|
}
|
||||||
|
|
||||||
|
func getNetDevicesArgs(
|
||||||
|
netDevicesNames []string,
|
||||||
|
taskCfg TaskConfig,
|
||||||
|
) map[string]NetDeviceArgs {
|
||||||
|
args := make(map[string]NetDeviceArgs)
|
||||||
|
// TODO: extract the flag<->device assignation from taskCfg
|
||||||
|
// for now, always assign --ipv4 to service and fail if there is another device
|
||||||
|
for _, name := range netDevicesNames {
|
||||||
|
if name == "service" {
|
||||||
|
args["service"] = NetDeviceArgs{
|
||||||
|
Ip: "--ipv4",
|
||||||
|
Gateway: "--ipv4-gateway",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return args
|
||||||
|
}
|
||||||
|
|
||||||
|
func rollbackIptablesRules(rules [][]string) {
|
||||||
|
for j := len(rules) - 1; j >= 0; j-- {
|
||||||
|
args := append([]string{"-t", "nat", "-D"}, rules[j]...)
|
||||||
|
exec.Command("iptables", args...).Run()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// run a list of iptables commands of the form "iptables -t nat -A ..." as a
|
||||||
|
// single transaction: if one rule fails, we roll back earlier rules before
|
||||||
|
// returning an error
|
||||||
|
func execIptablesRules(rules [][]string) error {
|
||||||
|
for i, rule := range rules {
|
||||||
|
args := append([]string{"-t", "nat", "-A"}, rule...)
|
||||||
|
cmd := exec.Command("iptables", args...)
|
||||||
|
if err := cmd.Run(); err != nil {
|
||||||
|
rollbackIptablesRules(rules[0:i])
|
||||||
|
return fmt.Errorf("'%s' failed: %v", cmd.String(), err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func netDeviceIptablesRules(ip string, ports []NetDevicePortConfig) [][]string {
|
||||||
|
// iptables rules to talk to the outside and setup incoming port redirections
|
||||||
|
rr := [][]string{}
|
||||||
|
addrule := func(args ...string) { rr = append(rr, args) }
|
||||||
|
|
||||||
|
// allow the ip to talk to the outside
|
||||||
|
// iptables -t nat -A MIRAGE_OUT -s <ip>/32 -j MASQUERADE
|
||||||
|
addrule(iptablesChainOut(), "-s", ip+"/32", "-j", "MASQUERADE")
|
||||||
|
|
||||||
|
// NAT port redirections
|
||||||
|
// iptables -t nat -A MIRAGE_IN ! -s <ip>/32 -p tcp -m tcp --dport 53
|
||||||
|
// -j DNAT --to-destination <ip>:53
|
||||||
|
// iptables -t nat -A MIRAGE_IN ! -s <ip>/32 -p udp -m udp --dport 53
|
||||||
|
// -j DNAT --to-destination <ip>:53
|
||||||
|
for _, port := range ports {
|
||||||
|
addrule(iptablesChainIn(), "!", "-s", ip+"/32", "-p", "tcp",
|
||||||
|
"--dport", strconv.Itoa(port.From), /* TODO:check */
|
||||||
|
"-j", "DNAT", "--to-destination", ip+":"+strconv.Itoa(port.To))
|
||||||
|
addrule(iptablesChainIn(), "!", "-s", ip+"/32", "-p", "udp",
|
||||||
|
"--dport", strconv.Itoa(port.From), /* TODO:check from/to */
|
||||||
|
"-j", "DNAT", "--to-destination", ip+":"+strconv.Itoa(port.To))
|
||||||
|
}
|
||||||
|
return rr
|
||||||
|
}
|
||||||
|
|
||||||
|
func setupNetDevice(
|
||||||
|
deviceName string,
|
||||||
|
unikernel string,
|
||||||
|
bridge string,
|
||||||
|
ports []NetDevicePortConfig,
|
||||||
|
logger hclog.Logger,
|
||||||
|
) (NetDeviceConfig, error) {
|
||||||
|
// list existing tap interfaces, and get a fresh tap interface number
|
||||||
|
tapDevice, err := getFreshTapDeviceName()
|
||||||
|
if err != nil {
|
||||||
|
return NetDeviceConfig{}, fmt.Errorf("could not get a fresh tap device name: %v", err)
|
||||||
|
}
|
||||||
|
logger.Debug("got tap device name", tapDevice)
|
||||||
|
|
||||||
|
// create new tap device
|
||||||
|
if err := createTapDevice(tapDevice, bridge); err != nil {
|
||||||
|
return NetDeviceConfig{}, fmt.Errorf("could not create tap device: %v", err)
|
||||||
|
}
|
||||||
|
logger.Debug("successfully created tap device")
|
||||||
|
|
||||||
|
// get a mac address for the unikernel
|
||||||
|
macAddr := getMacAddr(unikernel, bridge)
|
||||||
|
logger.Debug("mac address", macAddr)
|
||||||
|
|
||||||
|
// find an unused ip address
|
||||||
|
ipAddr, err := getFreshIp()
|
||||||
|
if err != nil {
|
||||||
|
destroyTapDevice(tapDevice)
|
||||||
|
return NetDeviceConfig{}, fmt.Errorf("could not get a fresh ip address: %v", err)
|
||||||
|
}
|
||||||
|
logger.Debug("got ip address", ipAddr)
|
||||||
|
|
||||||
|
// execute iptables rules
|
||||||
|
if err := execIptablesRules(netDeviceIptablesRules(ipAddr, ports)); err != nil {
|
||||||
|
destroyTapDevice(tapDevice)
|
||||||
|
return NetDeviceConfig{}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return NetDeviceConfig{
|
||||||
|
TapDevice: tapDevice,
|
||||||
|
Ip: ipAddr,
|
||||||
|
Mac: macAddr,
|
||||||
|
Ports: ports,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func destroyNetDevice(device NetDeviceConfig) {
|
||||||
|
rollbackIptablesRules(netDeviceIptablesRules(device.Ip, device.Ports))
|
||||||
|
destroyTapDevice(device.TapDevice)
|
||||||
|
}
|
||||||
|
|
||||||
|
func destroyNetDevices(devices map[string]NetDeviceConfig) {
|
||||||
|
for _, d := range devices {
|
||||||
|
destroyNetDevice(d)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func setupNetDevices(
|
||||||
|
netDevicesPorts map[string][]NetDevicePortConfig,
|
||||||
|
unikernel string,
|
||||||
|
bridge string,
|
||||||
|
logger hclog.Logger,
|
||||||
|
) (map[string]NetDeviceConfig, error) {
|
||||||
|
netDevices := make(map[string]NetDeviceConfig)
|
||||||
|
for deviceName, devicePorts := range netDevicesPorts {
|
||||||
|
deviceCfg, err :=
|
||||||
|
setupNetDevice(deviceName, unikernel, bridge, devicePorts, logger)
|
||||||
|
if err != nil {
|
||||||
|
destroyNetDevices(netDevices)
|
||||||
|
return nil,
|
||||||
|
fmt.Errorf("failed to setup net device %s: %v",
|
||||||
|
deviceName, err)
|
||||||
|
}
|
||||||
|
netDevices[deviceName] = deviceCfg
|
||||||
|
}
|
||||||
|
return netDevices, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: also return the list of block device names, when we handle block
|
||||||
|
// devices
|
||||||
|
func readUnikernelManifest(
|
||||||
|
unikernel string,
|
||||||
|
logger hclog.Logger,
|
||||||
|
) ([]string /* net devices */, error) {
|
||||||
|
manifestCmd := exec.Command("solo5-elftool", "query-manifest", unikernel)
|
||||||
|
manifestStr, manifestErr := manifestCmd.Output()
|
||||||
|
if manifestErr != nil {
|
||||||
|
return nil, fmt.Errorf("failed to query unikernel manifest: %v", manifestErr)
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.Debug("unikernel manifest", manifestStr)
|
||||||
|
|
||||||
|
var manifest unikernelManifest
|
||||||
|
if err := json.Unmarshal(manifestStr, &manifest); err != nil {
|
||||||
|
return nil, fmt.Errorf("could not parse manifest: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
netDevicesNames := []string{}
|
||||||
|
for _, d := range manifest.Devices {
|
||||||
|
switch d.Type {
|
||||||
|
case "NET_BASIC":
|
||||||
|
netDevicesNames = append(netDevicesNames, d.Name)
|
||||||
|
// case "BLOCK_BASIC":
|
||||||
|
// TODO
|
||||||
|
default:
|
||||||
|
return nil, fmt.Errorf("unhandled device type: %s", d.Type)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return netDevicesNames, nil
|
||||||
|
}
|
||||||
|
|
||||||
// StartTask returns a task handle and a driver network if necessary.
|
// StartTask returns a task handle and a driver network if necessary.
|
||||||
func (d *HelloDriverPlugin) StartTask(cfg *drivers.TaskConfig) (*drivers.TaskHandle, *drivers.DriverNetwork, error) {
|
func (d *HelloDriverPlugin) StartTask(cfg *drivers.TaskConfig) (*drivers.TaskHandle, *drivers.DriverNetwork, error) {
|
||||||
if _, ok := d.tasks.Get(cfg.ID); ok {
|
if _, ok := d.tasks.Get(cfg.ID); ok {
|
||||||
|
@ -375,28 +827,80 @@ func (d *HelloDriverPlugin) StartTask(cfg *drivers.TaskConfig) (*drivers.TaskHan
|
||||||
LogLevel: "debug",
|
LogLevel: "debug",
|
||||||
}
|
}
|
||||||
|
|
||||||
exec, pluginClient, err := executor.CreateExecutor(d.logger, d.nomadConfig, executorConfig)
|
logger := d.logger.With("task_name", handle.Config.Name, "alloc_id", handle.Config.AllocID)
|
||||||
|
|
||||||
|
netDevicesNames, err := readUnikernelManifest(driverConfig.Unikernel, logger)
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
netDevicesPorts := getNetDevicesPorts(netDevicesNames, driverConfig, cfg)
|
||||||
|
netDevicesArgs := getNetDevicesArgs(netDevicesNames, driverConfig)
|
||||||
|
netDevices, err :=
|
||||||
|
setupNetDevices(netDevicesPorts, driverConfig.Unikernel,
|
||||||
|
d.config.Bridge, logger)
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
solo5Args := []string{}
|
||||||
|
unikernelArgs := []string{}
|
||||||
|
// --net:<service from manifest>=<tapDevice>
|
||||||
|
// --net-mac:<service from manifest>=<macAddr>
|
||||||
|
for name, device := range netDevices {
|
||||||
|
solo5Args = append(solo5Args,
|
||||||
|
fmt.Sprintf("--net:%s=%s", name, device.TapDevice))
|
||||||
|
solo5Args = append(solo5Args,
|
||||||
|
fmt.Sprintf("--net-mac:%s=%s", name, device.Mac))
|
||||||
|
}
|
||||||
|
// --ipv4=<ipAddr>/16 --ipv4-gateway=<gatewayIp()> (by default)
|
||||||
|
// + user-specified equivalent arguments for additional net devices
|
||||||
|
for name, arg := range netDevicesArgs {
|
||||||
|
if arg.Ip != "" {
|
||||||
|
unikernelArgs = append(unikernelArgs,
|
||||||
|
fmt.Sprintf("%s=%s/16", arg.Ip, netDevices[name].Ip))
|
||||||
|
}
|
||||||
|
if arg.Gateway != "" {
|
||||||
|
unikernelArgs = append(unikernelArgs,
|
||||||
|
fmt.Sprintf("%s=%s", arg.Gateway, gatewayIp()))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: collect additional arguments from the config file
|
||||||
|
|
||||||
|
execu, pluginClient, err := executor.CreateExecutor(logger, d.nomadConfig, executorConfig)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, nil, fmt.Errorf("failed to create executor: %v", err)
|
return nil, nil, fmt.Errorf("failed to create executor: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
echoCmd := fmt.Sprintf(`echo "%s"`, driverConfig.Greeting)
|
cfgJson, _ := json.MarshalIndent(cfg, "", " ")
|
||||||
|
// cfgStr := base64.StdEncoding.EncodeToString(cfgJson)
|
||||||
|
logger.Debug("task config", string(cfgJson))
|
||||||
|
|
||||||
|
netCfgStr, _ := exec.Command("ip", "a").Output()
|
||||||
|
logger.Debug("host network config", string(netCfgStr))
|
||||||
|
|
||||||
|
args := append(solo5Args, []string{"--", driverConfig.Unikernel}...)
|
||||||
|
args = append(args, unikernelArgs...)
|
||||||
|
|
||||||
|
logger.Debug("solo5 command", exec.Command("solo5-hvt", args...).String())
|
||||||
|
|
||||||
execCmd := &executor.ExecCommand{
|
execCmd := &executor.ExecCommand{
|
||||||
Cmd: d.config.Shell,
|
Cmd: "solo5-hvt",
|
||||||
Args: []string{"-c", echoCmd},
|
Args: args,
|
||||||
StdoutPath: cfg.StdoutPath,
|
StdoutPath: cfg.StdoutPath,
|
||||||
StderrPath: cfg.StderrPath,
|
StderrPath: cfg.StderrPath,
|
||||||
}
|
}
|
||||||
|
|
||||||
ps, err := exec.Launch(execCmd)
|
ps, err := execu.Launch(execCmd)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
pluginClient.Kill()
|
pluginClient.Kill()
|
||||||
return nil, nil, fmt.Errorf("failed to launch command with executor: %v", err)
|
return nil, nil, fmt.Errorf("failed to launch command with executor: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
h := &taskHandle{
|
h := &taskHandle{
|
||||||
exec: exec,
|
exec: execu,
|
||||||
pid: ps.Pid,
|
pid: ps.Pid,
|
||||||
|
netDevices: netDevices,
|
||||||
pluginClient: pluginClient,
|
pluginClient: pluginClient,
|
||||||
taskConfig: cfg,
|
taskConfig: cfg,
|
||||||
procState: drivers.TaskStateRunning,
|
procState: drivers.TaskStateRunning,
|
||||||
|
@ -407,11 +911,13 @@ func (d *HelloDriverPlugin) StartTask(cfg *drivers.TaskConfig) (*drivers.TaskHan
|
||||||
driverState := TaskState{
|
driverState := TaskState{
|
||||||
ReattachConfig: structs.ReattachConfigFromGoPlugin(pluginClient.ReattachConfig()),
|
ReattachConfig: structs.ReattachConfigFromGoPlugin(pluginClient.ReattachConfig()),
|
||||||
Pid: ps.Pid,
|
Pid: ps.Pid,
|
||||||
|
NetDevices: netDevices,
|
||||||
TaskConfig: cfg,
|
TaskConfig: cfg,
|
||||||
StartedAt: h.startedAt,
|
StartedAt: h.startedAt,
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := handle.SetDriverState(&driverState); err != nil {
|
if err := handle.SetDriverState(&driverState); err != nil {
|
||||||
|
d.logger.Error("failed to start task, error setting driver state", "error", err)
|
||||||
return nil, nil, fmt.Errorf("failed to set driver state: %v", err)
|
return nil, nil, fmt.Errorf("failed to set driver state: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -460,6 +966,7 @@ func (d *HelloDriverPlugin) RecoverTask(handle *drivers.TaskHandle) error {
|
||||||
h := &taskHandle{
|
h := &taskHandle{
|
||||||
exec: execImpl,
|
exec: execImpl,
|
||||||
pid: taskState.Pid,
|
pid: taskState.Pid,
|
||||||
|
netDevices: taskState.NetDevices,
|
||||||
pluginClient: pluginClient,
|
pluginClient: pluginClient,
|
||||||
taskConfig: taskState.TaskConfig,
|
taskConfig: taskState.TaskConfig,
|
||||||
procState: drivers.TaskStateRunning,
|
procState: drivers.TaskStateRunning,
|
||||||
|
@ -575,6 +1082,12 @@ func (d *HelloDriverPlugin) DestroyTask(taskID string, force bool) error {
|
||||||
handle.pluginClient.Kill()
|
handle.pluginClient.Kill()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// remove the relevant iptables rules and the tap device
|
||||||
|
destroyNetDevices(handle.netDevices)
|
||||||
|
|
||||||
|
// TODO: destroy the bridge and the iptables chains if no one is using them
|
||||||
|
// anymore
|
||||||
|
|
||||||
d.tasks.Delete(taskID)
|
d.tasks.Delete(taskID)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
|
@ -33,6 +33,7 @@ type taskHandle struct {
|
||||||
|
|
||||||
// TODO: add any extra relevant information about the task.
|
// TODO: add any extra relevant information about the task.
|
||||||
pid int
|
pid int
|
||||||
|
netDevices map[string]NetDeviceConfig
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *taskHandle) TaskStatus() *drivers.TaskStatus {
|
func (h *taskHandle) TaskStatus() *drivers.TaskStatus {
|
||||||
|
@ -48,6 +49,7 @@ func (h *taskHandle) TaskStatus() *drivers.TaskStatus {
|
||||||
ExitResult: h.exitResult,
|
ExitResult: h.exitResult,
|
||||||
DriverAttributes: map[string]string{
|
DriverAttributes: map[string]string{
|
||||||
"pid": strconv.Itoa(h.pid),
|
"pid": strconv.Itoa(h.pid),
|
||||||
|
// TODO: emit more attributes?
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue