mirror of
https://github.com/kata-containers/kata-containers.git
synced 2025-06-25 15:02:45 +00:00
clh: remove cli builder
Remove cli builder code as now that we use http client Signed-off-by: Bo Chen <chen.bo@intel.com> Signed-off-by: Jose Carlos Venegas Munoz <jose.carlos.venegas.munoz@intel.com>
This commit is contained in:
parent
f73723a23f
commit
9f15dd2da3
@ -816,7 +816,7 @@ func (clh *cloudHypervisor) LaunchClh() (string, int, error) {
|
||||
return "", -1, err
|
||||
}
|
||||
|
||||
args := []string{cscApisocket, clh.state.apiSocket}
|
||||
args := []string{cscAPIsocket, clh.state.apiSocket}
|
||||
if clh.config.Debug {
|
||||
|
||||
logfile, err := clh.logFilePath(clh.id)
|
||||
@ -867,298 +867,10 @@ const (
|
||||
)
|
||||
|
||||
const (
|
||||
cscApisocket string = "--api-socket"
|
||||
cscCmdline string = "--cmdline"
|
||||
cscConsole string = "--console"
|
||||
cscCpus string = "--cpus"
|
||||
cscDisk string = "--disk"
|
||||
cscFs string = "--fs"
|
||||
cscKernel string = "--kernel"
|
||||
cscAPIsocket string = "--api-socket"
|
||||
cscLogFile string = "--log-file"
|
||||
cscMemory string = "--memory"
|
||||
cscNet string = "--net"
|
||||
cscRng string = "--rng"
|
||||
cscSerial string = "--serial"
|
||||
cscVsock string = "--vsock"
|
||||
)
|
||||
|
||||
type CommandLineBuilder interface {
|
||||
AddKernelParameters(cmdline []Param)
|
||||
SetConsole(console *CLIConsole)
|
||||
SetCpus(cpus *CLICpus)
|
||||
SetDisk(disk *CLIDisk)
|
||||
SetFs(fs *CLIFs)
|
||||
SetKernel(kernel *CLIKernel)
|
||||
SetMemory(memory *CLIMemory)
|
||||
AddNet(net CLINet)
|
||||
SetRng(rng *CLIRng)
|
||||
SetSerial(serial *CLISerialConsole)
|
||||
SetVsock(vsock *CLIVsock)
|
||||
SetAPISocket(apiSocket *CLIAPISocket)
|
||||
SetLogFile(logFile *CLILogFile)
|
||||
GetCommandLine() (*CommandLine, error)
|
||||
}
|
||||
|
||||
type CLIOption interface {
|
||||
Build(cmdline *CommandLine)
|
||||
}
|
||||
|
||||
type CommandLine struct {
|
||||
args []string
|
||||
}
|
||||
|
||||
//**********************************
|
||||
// The (virtio) Console
|
||||
//**********************************
|
||||
type CLIConsole struct {
|
||||
consoleType string
|
||||
filePath string
|
||||
iommu bool
|
||||
}
|
||||
|
||||
func (o *CLIConsole) Build(cmdline *CommandLine) {
|
||||
|
||||
cmdline.args = append(cmdline.args, cscConsole)
|
||||
|
||||
consoleArg := ""
|
||||
if o.consoleType == cctFILE {
|
||||
consoleArg = o.consoleType + "=" + o.filePath
|
||||
if o.iommu {
|
||||
consoleArg += ",iommu=on"
|
||||
} else {
|
||||
consoleArg += ",iommu=off"
|
||||
}
|
||||
} else {
|
||||
consoleArg = o.consoleType
|
||||
}
|
||||
|
||||
cmdline.args = append(cmdline.args, consoleArg)
|
||||
}
|
||||
|
||||
//**********************************
|
||||
// The serial port
|
||||
//**********************************
|
||||
type CLISerialConsole struct {
|
||||
consoleType string
|
||||
filePath string
|
||||
}
|
||||
|
||||
func (o *CLISerialConsole) Build(cmdline *CommandLine) {
|
||||
|
||||
cmdline.args = append(cmdline.args, cscSerial)
|
||||
if o.consoleType == cctFILE {
|
||||
cmdline.args = append(cmdline.args, o.consoleType+"="+o.filePath)
|
||||
} else {
|
||||
cmdline.args = append(cmdline.args, o.consoleType)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
//**********************************
|
||||
// The API socket
|
||||
//**********************************
|
||||
type CLIAPISocket struct {
|
||||
socketPath string
|
||||
}
|
||||
|
||||
func (o *CLIAPISocket) Build(cmdline *CommandLine) {
|
||||
|
||||
cmdline.args = append(cmdline.args, cscApisocket)
|
||||
if o.socketPath != "" {
|
||||
cmdline.args = append(cmdline.args, o.socketPath)
|
||||
}
|
||||
}
|
||||
|
||||
//**********************************
|
||||
// The amount of memory in Mb
|
||||
//**********************************
|
||||
type CLIMemory struct {
|
||||
memorySize uint32
|
||||
backingFile string
|
||||
}
|
||||
|
||||
func (o *CLIMemory) Build(cmdline *CommandLine) {
|
||||
|
||||
cmdline.args = append(cmdline.args, cscMemory)
|
||||
if o.backingFile == "" {
|
||||
cmdline.args = append(cmdline.args, "size="+strconv.FormatUint(uint64(o.memorySize), 10)+"M")
|
||||
} else {
|
||||
cmdline.args = append(cmdline.args, "size="+strconv.FormatUint(uint64(o.memorySize), 10)+"M,file="+o.backingFile)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
//**********************************
|
||||
// The number of CPU's
|
||||
//**********************************
|
||||
type CLICpus struct {
|
||||
cpus uint32
|
||||
}
|
||||
|
||||
func (o *CLICpus) Build(cmdline *CommandLine) {
|
||||
|
||||
cmdline.args = append(cmdline.args, cscCpus)
|
||||
cmdline.args = append(cmdline.args, strconv.FormatUint(uint64(o.cpus), 10))
|
||||
|
||||
}
|
||||
|
||||
//**********************************
|
||||
// The Path to the kernel image
|
||||
//**********************************
|
||||
type CLIKernel struct {
|
||||
path string
|
||||
}
|
||||
|
||||
func (o *CLIKernel) Build(cmdline *CommandLine) {
|
||||
|
||||
cmdline.args = append(cmdline.args, cscKernel)
|
||||
cmdline.args = append(cmdline.args, o.path)
|
||||
|
||||
}
|
||||
|
||||
//****************************************
|
||||
// The Path to the root (boot) disk image
|
||||
//****************************************
|
||||
type CLIDisk struct {
|
||||
path string
|
||||
iommu bool
|
||||
}
|
||||
|
||||
func (o *CLIDisk) Build(cmdline *CommandLine) {
|
||||
|
||||
cmdline.args = append(cmdline.args, cscDisk)
|
||||
if o.iommu {
|
||||
cmdline.args = append(cmdline.args, "path="+o.path+",iommu=on")
|
||||
} else {
|
||||
cmdline.args = append(cmdline.args, "path="+o.path+",iommu=off")
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
//****************************************
|
||||
// The random device
|
||||
//****************************************
|
||||
type CLIRng struct {
|
||||
src string
|
||||
iommu bool
|
||||
}
|
||||
|
||||
func (o *CLIRng) Build(cmdline *CommandLine) {
|
||||
|
||||
cmdline.args = append(cmdline.args, cscRng)
|
||||
if o.iommu {
|
||||
cmdline.args = append(cmdline.args, "src="+o.src+",iommu=on")
|
||||
} else {
|
||||
cmdline.args = append(cmdline.args, "src="+o.src+",iommu=off")
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
//****************************************
|
||||
// The VSock socket
|
||||
//****************************************
|
||||
type CLIVsock struct {
|
||||
socketPath string
|
||||
cid uint32
|
||||
iommu bool
|
||||
}
|
||||
|
||||
func (o *CLIVsock) Build(cmdline *CommandLine) {
|
||||
|
||||
cmdline.args = append(cmdline.args, cscVsock)
|
||||
if o.iommu {
|
||||
cmdline.args = append(cmdline.args, "cid="+strconv.FormatUint(uint64(o.cid), 10)+",sock="+o.socketPath+",iommu=on")
|
||||
} else {
|
||||
cmdline.args = append(cmdline.args, "cid="+strconv.FormatUint(uint64(o.cid), 10)+",sock="+o.socketPath+",iommu=off")
|
||||
}
|
||||
}
|
||||
|
||||
//****************************************
|
||||
// The shard (virtio) file system
|
||||
//****************************************
|
||||
type CLIFs struct {
|
||||
tag string
|
||||
socketPath string
|
||||
queues uint32
|
||||
queueSize uint32
|
||||
dax bool
|
||||
cacheSize string
|
||||
}
|
||||
|
||||
func (o *CLIFs) Build(cmdline *CommandLine) {
|
||||
|
||||
cmdline.args = append(cmdline.args, cscFs)
|
||||
|
||||
fsarg := "tag=" + o.tag + ",sock=" + o.socketPath
|
||||
if o.dax {
|
||||
fsarg += ",dax=on"
|
||||
if o.cacheSize != "" {
|
||||
fsarg += ",cache_size=" + o.cacheSize
|
||||
}
|
||||
} else {
|
||||
fsarg += ",num_queues=" + strconv.FormatUint(uint64(o.queues), 10) + ",queue_size=" + strconv.FormatUint(uint64(o.queueSize), 10)
|
||||
}
|
||||
cmdline.args = append(cmdline.args, fsarg)
|
||||
}
|
||||
|
||||
//****************************************
|
||||
// The net (nic)
|
||||
//****************************************
|
||||
type CLINet struct {
|
||||
device string
|
||||
ip string
|
||||
mask string
|
||||
mac string
|
||||
iommu bool
|
||||
}
|
||||
|
||||
type CLINets struct {
|
||||
networks []CLINet
|
||||
}
|
||||
|
||||
func (o *CLINets) Build(cmdline *CommandLine) {
|
||||
|
||||
cmdline.args = append(cmdline.args, cscNet)
|
||||
|
||||
networks := ""
|
||||
netIndex := 1
|
||||
for _, net := range o.networks {
|
||||
cnet := "tap=tap" + strconv.FormatUint(uint64(netIndex), 10)
|
||||
if net.ip != "" && net.mask != "" {
|
||||
cnet += ",ip=" + net.ip + ",mask=" + net.mask
|
||||
}
|
||||
if net.mac != "" {
|
||||
cnet += ",mac=" + net.mac
|
||||
}
|
||||
|
||||
if net.iommu {
|
||||
cnet += ",iommu=on"
|
||||
}
|
||||
if netIndex > 1 {
|
||||
networks += "," + cnet
|
||||
} else {
|
||||
networks += cnet
|
||||
}
|
||||
netIndex++
|
||||
}
|
||||
cmdline.args = append(cmdline.args, networks)
|
||||
}
|
||||
|
||||
//****************************************
|
||||
// The log file
|
||||
//****************************************
|
||||
type CLILogFile struct {
|
||||
path string
|
||||
}
|
||||
|
||||
func (o *CLILogFile) Build(cmdline *CommandLine) {
|
||||
|
||||
if o.path != "" {
|
||||
cmdline.args = append(cmdline.args, cscLogFile)
|
||||
cmdline.args = append(cmdline.args, o.path)
|
||||
}
|
||||
}
|
||||
|
||||
//****************************************
|
||||
// The kernel command line
|
||||
//****************************************
|
||||
|
@ -4,688 +4,3 @@
|
||||
//
|
||||
|
||||
package virtcontainers
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/kata-containers/runtime/virtcontainers/device/config"
|
||||
"github.com/kata-containers/runtime/virtcontainers/store"
|
||||
"github.com/kata-containers/runtime/virtcontainers/types"
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
//
|
||||
// Cli helper functions
|
||||
//
|
||||
|
||||
func getCliOption(args []string, key string) (string, error) {
|
||||
for i, p := range args {
|
||||
if p == key && i < (len(args)-1) {
|
||||
return args[i+1], nil
|
||||
}
|
||||
}
|
||||
return "", errors.New("Key not found")
|
||||
}
|
||||
|
||||
func newClhConfig() HypervisorConfig {
|
||||
return HypervisorConfig{
|
||||
KernelPath: testClhKernelPath,
|
||||
ImagePath: testClhImagePath,
|
||||
HypervisorPath: testClhPath,
|
||||
NumVCPUs: defaultVCPUs,
|
||||
BlockDeviceDriver: config.VirtioBlock,
|
||||
MemorySize: defaultMemSzMiB,
|
||||
DefaultBridges: defaultBridges,
|
||||
DefaultMaxVCPUs: MaxClhVCPUs(),
|
||||
// Adding this here, as hypervisorconfig.valid()
|
||||
// forcefully adds it even when 9pfs is not supported
|
||||
Msize9p: defaultMsize9p,
|
||||
VirtioFSCache: virtioFsCacheAlways,
|
||||
}
|
||||
}
|
||||
|
||||
//
|
||||
// --cmdline <cmdline> Kernel command line
|
||||
//
|
||||
func TestClhCliKernelParameters(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
|
||||
expectedOut := "--cmdline foo=foo bar=bar"
|
||||
params := []Param{
|
||||
{
|
||||
Key: "foo",
|
||||
Value: "foo",
|
||||
},
|
||||
{
|
||||
Key: "bar",
|
||||
Value: "bar",
|
||||
},
|
||||
}
|
||||
|
||||
builder := &DefaultCLIBuilder{}
|
||||
builder.AddKernelParameters(params)
|
||||
director := &CommandLineDirector{}
|
||||
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), expectedOut)
|
||||
}
|
||||
|
||||
//
|
||||
// Control (virtio) console: <console> off|null|tty|file=/path/to/a/file,iommu=on|off [default: tty]
|
||||
//
|
||||
func TestClhCliConsole(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultFilePath := "/a/b/c"
|
||||
|
||||
builder.SetConsole(&CLIConsole{
|
||||
consoleType: cctOFF,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--console off")
|
||||
|
||||
builder.SetConsole(&CLIConsole{
|
||||
consoleType: cctNULL,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--console null")
|
||||
|
||||
builder.SetConsole(&CLIConsole{
|
||||
consoleType: cctTTY,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--console tty")
|
||||
|
||||
builder.SetConsole(&CLIConsole{
|
||||
consoleType: cctFILE,
|
||||
filePath: defaultFilePath,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--console file="+defaultFilePath+",iommu=off")
|
||||
|
||||
builder.SetConsole(&CLIConsole{
|
||||
consoleType: cctFILE,
|
||||
filePath: defaultFilePath,
|
||||
iommu: true,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--console file="+defaultFilePath+",iommu=on")
|
||||
|
||||
}
|
||||
|
||||
//
|
||||
// Control serial port: off|null|tty|file=/path/to/a/file [default: tty]
|
||||
//
|
||||
func TestClhCliSerial(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultFilePath := "/a/b/c"
|
||||
|
||||
builder.SetSerial(&CLISerialConsole{
|
||||
consoleType: cctOFF,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--serial off")
|
||||
|
||||
builder.SetSerial(&CLISerialConsole{
|
||||
consoleType: cctNULL,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--serial null")
|
||||
|
||||
builder.SetSerial(&CLISerialConsole{
|
||||
consoleType: cctTTY,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--serial tty")
|
||||
|
||||
builder.SetSerial(&CLISerialConsole{
|
||||
consoleType: cctFILE,
|
||||
filePath: defaultFilePath,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--serial file="+defaultFilePath)
|
||||
|
||||
}
|
||||
|
||||
//
|
||||
// --api-socket <api-socket> HTTP API socket path (UNIX domain socket). [default: /run/cloud-hypervisor.23605
|
||||
//
|
||||
func TestClhCliApiSocket(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultFilePath := "/a/b/c"
|
||||
|
||||
builder.SetAPISocket(&CLIAPISocket{
|
||||
socketPath: defaultFilePath,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--api-socket "+defaultFilePath)
|
||||
}
|
||||
|
||||
//
|
||||
// --cpus <cpus> Number of virtual CPUs [default: 1]
|
||||
//
|
||||
func TestClhCliCpus(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultCPUs := "4"
|
||||
|
||||
builder.SetCpus(&CLICpus{
|
||||
cpus: 4,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--cpus "+defaultCPUs)
|
||||
}
|
||||
|
||||
//
|
||||
// --disk <disk> Disk parameters "path=<disk_image_path>,iommu=on|off"
|
||||
//
|
||||
func TestClhCliDisk(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultDiskPath := "/a/b/c.img"
|
||||
|
||||
builder.SetDisk(&CLIDisk{
|
||||
path: defaultDiskPath,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--disk path="+defaultDiskPath+",iommu=off")
|
||||
|
||||
builder.SetDisk(&CLIDisk{
|
||||
path: defaultDiskPath,
|
||||
iommu: true,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--disk path="+defaultDiskPath+",iommu=on")
|
||||
}
|
||||
|
||||
//
|
||||
// --fs <fs> virtio-fs parameters
|
||||
// "tag=<tag_name>,sock=<socket_path>,num_queues=<number_of_queues>,queue_size=<size_of_each_queue>,dax=on|off,cache_size=<DAX
|
||||
// cache size: default 8Gib>"
|
||||
//
|
||||
func TestClhCliFs(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultFsPath := "/a/b/c"
|
||||
defaultFsQs := uint32(2)
|
||||
defaultFsQss := uint32(1024)
|
||||
defaultFsTag := "myTag"
|
||||
defaultFsCacheSize := "1Gib"
|
||||
|
||||
builder.SetFs(&CLIFs{
|
||||
tag: defaultFsTag,
|
||||
socketPath: defaultFsPath,
|
||||
queues: defaultFsQs,
|
||||
queueSize: defaultFsQss,
|
||||
dax: false,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")),
|
||||
"--fs tag="+defaultFsTag+
|
||||
",sock="+defaultFsPath+
|
||||
",num_queues="+strconv.FormatUint(uint64(defaultFsQs), 10)+
|
||||
",queue_size="+strconv.FormatUint(uint64(defaultFsQss), 10))
|
||||
|
||||
builder.SetFs(&CLIFs{
|
||||
tag: defaultFsTag,
|
||||
socketPath: defaultFsPath,
|
||||
dax: true,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")),
|
||||
"--fs tag="+defaultFsTag+
|
||||
",sock="+defaultFsPath+
|
||||
",dax=on")
|
||||
|
||||
builder.SetFs(&CLIFs{
|
||||
tag: defaultFsTag,
|
||||
socketPath: defaultFsPath,
|
||||
dax: true,
|
||||
cacheSize: defaultFsCacheSize,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")),
|
||||
"--fs tag="+defaultFsTag+
|
||||
",sock="+defaultFsPath+
|
||||
",dax=on,cache_size="+defaultFsCacheSize)
|
||||
|
||||
}
|
||||
|
||||
//
|
||||
// --disk <disk> Disk parameters "path=<disk_image_path>,iommu=on|off"
|
||||
//
|
||||
func TestClhCliKernel(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultKernel := "/a/b/vmlinuz"
|
||||
|
||||
builder.SetKernel(&CLIKernel{
|
||||
path: defaultKernel,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--kernel "+defaultKernel)
|
||||
|
||||
}
|
||||
|
||||
//
|
||||
// --log-file <log-file> Log file. Standard error is used if not specified
|
||||
//
|
||||
func TestClhCliLogFile(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultPath := "/a/b/clh.log"
|
||||
|
||||
builder.SetLogFile(&CLILogFile{
|
||||
path: defaultPath,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--log-file "+defaultPath)
|
||||
|
||||
}
|
||||
|
||||
//
|
||||
// --memory <memory> Memory parameters "size=<guest_memory_size>,file=<backing_file_path>"
|
||||
//
|
||||
func TestClhCliMemory(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultMemory := uint32(1024)
|
||||
defaultFile := "/a/b.shm"
|
||||
|
||||
builder.SetMemory(&CLIMemory{
|
||||
memorySize: defaultMemory,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--memory size="+strconv.FormatUint(uint64(defaultMemory), 10)+"M")
|
||||
|
||||
builder.SetMemory(&CLIMemory{
|
||||
memorySize: defaultMemory,
|
||||
backingFile: defaultFile,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")),
|
||||
"--memory size="+strconv.FormatUint(uint64(defaultMemory), 10)+"M"+
|
||||
",file="+defaultFile)
|
||||
|
||||
}
|
||||
|
||||
//
|
||||
// --net <net> Network parameters
|
||||
// "tap=<if_name>,ip=<ip_addr>,mask=<net_mask>,mac=<mac_addr>,iommu=on|off"
|
||||
//
|
||||
func TestClhCliNetwork(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultIfName := "tap1"
|
||||
defaultIP := "1.2.3.4"
|
||||
defaultNetMask := "255.255.255.0"
|
||||
defaultMac := "00:11:22:33:44:55"
|
||||
|
||||
builder.AddNet(CLINet{
|
||||
mac: defaultMac,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--net tap=tap1,mac="+defaultMac)
|
||||
|
||||
builder = &DefaultCLIBuilder{}
|
||||
builder.AddNet(CLINet{
|
||||
device: defaultIfName,
|
||||
mac: defaultMac,
|
||||
})
|
||||
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--net tap="+defaultIfName+",mac="+defaultMac)
|
||||
|
||||
builder = &DefaultCLIBuilder{}
|
||||
builder.AddNet(CLINet{
|
||||
device: defaultIfName,
|
||||
mac: defaultMac,
|
||||
ip: defaultIP,
|
||||
mask: defaultNetMask,
|
||||
})
|
||||
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--net tap="+defaultIfName+
|
||||
",ip="+defaultIP+
|
||||
",mask="+defaultNetMask+
|
||||
",mac="+defaultMac)
|
||||
|
||||
builder.AddNet(CLINet{
|
||||
mac: defaultMac,
|
||||
ip: defaultIP,
|
||||
mask: defaultNetMask,
|
||||
})
|
||||
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")),
|
||||
"--net tap="+defaultIfName+
|
||||
",ip="+defaultIP+
|
||||
",mask="+defaultNetMask+
|
||||
",mac="+defaultMac+
|
||||
",tap=tap2"+
|
||||
",ip="+defaultIP+
|
||||
",mask="+defaultNetMask+
|
||||
",mac="+defaultMac)
|
||||
}
|
||||
|
||||
//
|
||||
// --rng <rng> Random number generator parameters "src=<entropy_source_path>,iommu=on|off"
|
||||
//
|
||||
func TestClhCliRng(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultSource := "/a/b/c.random"
|
||||
|
||||
builder.SetRng(&CLIRng{
|
||||
src: defaultSource,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--rng src="+defaultSource+",iommu=off")
|
||||
|
||||
builder.SetRng(&CLIRng{
|
||||
src: defaultSource,
|
||||
iommu: true,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--rng src="+defaultSource+",iommu=on")
|
||||
}
|
||||
|
||||
//
|
||||
// --vsock <vsock> Virtio VSOCK parameters "cid=<context_id>,sock=<socket_path>,iommu=on|off"
|
||||
//
|
||||
func TestClhCliVsock(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultSocket := "/a/b/c.sock"
|
||||
defaultCid := uint32(12345)
|
||||
|
||||
builder.SetVsock(&CLIVsock{
|
||||
cid: defaultCid,
|
||||
socketPath: defaultSocket,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")),
|
||||
"--vsock cid="+strconv.FormatUint(uint64(defaultCid), 10)+
|
||||
",sock="+defaultSocket+
|
||||
",iommu=off")
|
||||
|
||||
builder.SetVsock(&CLIVsock{
|
||||
cid: defaultCid,
|
||||
socketPath: defaultSocket,
|
||||
iommu: true,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")),
|
||||
"--vsock cid="+strconv.FormatUint(uint64(defaultCid), 10)+
|
||||
",sock="+defaultSocket+
|
||||
",iommu=on")
|
||||
|
||||
}
|
||||
|
||||
func TestClhCreateSandbox(t *testing.T) {
|
||||
clhConfig := newClhConfig()
|
||||
clh := &cloudHypervisor{
|
||||
config: clhConfig,
|
||||
}
|
||||
assert := assert.New(t)
|
||||
|
||||
sandbox := &Sandbox{
|
||||
ctx: context.Background(),
|
||||
id: "testSandbox",
|
||||
config: &SandboxConfig{
|
||||
HypervisorConfig: clhConfig,
|
||||
},
|
||||
}
|
||||
|
||||
vcStore, err := store.NewVCSandboxStore(sandbox.ctx, sandbox.id)
|
||||
assert.NoError(err)
|
||||
|
||||
sandbox.store = vcStore
|
||||
|
||||
// Create the hypervisor fake binary
|
||||
testClhPath := filepath.Join(testDir, testHypervisor)
|
||||
_, err = os.Create(testClhPath)
|
||||
assert.NoError(err)
|
||||
|
||||
// Create parent dir path for hypervisor.json
|
||||
parentDir := store.SandboxConfigurationRootPath(sandbox.id)
|
||||
assert.NoError(os.MkdirAll(parentDir, store.DirMode))
|
||||
|
||||
err = clh.createSandbox(context.Background(), sandbox.id, NetworkNamespace{}, &sandbox.config.HypervisorConfig, sandbox.store, false)
|
||||
assert.NoError(err)
|
||||
assert.NoError(os.RemoveAll(parentDir))
|
||||
assert.Exactly(clhConfig, clh.config)
|
||||
}
|
||||
|
||||
func TestClhAddDeviceNet(t *testing.T) {
|
||||
defaultEndpointName := "tap1"
|
||||
defaultMac := "55:44:33:22:11:00"
|
||||
assert := assert.New(t)
|
||||
clh := &cloudHypervisor{
|
||||
ctx: context.Background(),
|
||||
cliBuilder: &DefaultCLIBuilder{},
|
||||
}
|
||||
|
||||
tep, _ := createTapNetworkEndpoint(0, defaultEndpointName)
|
||||
tep.TapInterface.TAPIface.HardAddr = defaultMac
|
||||
|
||||
err := clh.addDevice(Endpoint(tep), netDev)
|
||||
assert.NoError(err)
|
||||
|
||||
director := &CommandLineDirector{}
|
||||
|
||||
cli, err := director.Build(clh.cliBuilder)
|
||||
assert.NoError(err)
|
||||
assert.NotNil(cli)
|
||||
|
||||
netarg, err := getCliOption(cli.args, "--net")
|
||||
assert.NoError(err)
|
||||
assert.Equal(netarg, "tap="+defaultEndpointName+",mac="+defaultMac)
|
||||
|
||||
}
|
||||
|
||||
func TestClhAddDeviceVSock(t *testing.T) {
|
||||
defaultCid := uint64(12345)
|
||||
defaultUdsPath := "/a/b/c"
|
||||
defaultPort := uint32(1024)
|
||||
assert := assert.New(t)
|
||||
clh := &cloudHypervisor{
|
||||
ctx: context.Background(),
|
||||
cliBuilder: &DefaultCLIBuilder{},
|
||||
}
|
||||
|
||||
vsock := types.HybridVSock{
|
||||
UdsPath: defaultUdsPath,
|
||||
ContextID: defaultCid,
|
||||
Port: defaultPort,
|
||||
}
|
||||
|
||||
err := clh.addDevice(vsock, netDev)
|
||||
assert.NoError(err)
|
||||
|
||||
director := &CommandLineDirector{}
|
||||
|
||||
cli, err := director.Build(clh.cliBuilder)
|
||||
assert.NoError(err)
|
||||
assert.NotNil(cli)
|
||||
|
||||
netarg, err := getCliOption(cli.args, "--vsock")
|
||||
assert.NoError(err)
|
||||
assert.Equal(netarg, "cid="+strconv.FormatUint(defaultCid, 10)+",sock="+defaultUdsPath+",iommu=off")
|
||||
|
||||
}
|
||||
|
||||
func TestClhCapabilities(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
clh := &cloudHypervisor{
|
||||
ctx: context.Background(),
|
||||
cliBuilder: &DefaultCLIBuilder{},
|
||||
}
|
||||
|
||||
caps := clh.capabilities()
|
||||
|
||||
assert.False(caps.IsBlockDeviceHotplugSupported())
|
||||
assert.True(caps.IsFsSharingSupported())
|
||||
assert.False(caps.IsMultiQueueSupported())
|
||||
}
|
||||
|
||||
func TestClhGenerateSocket(t *testing.T) {
|
||||
defaultID := "123-456-99"
|
||||
defaultPort := uint32(1024)
|
||||
|
||||
assert := assert.New(t)
|
||||
clh := &cloudHypervisor{
|
||||
ctx: context.Background(),
|
||||
cliBuilder: &DefaultCLIBuilder{},
|
||||
}
|
||||
|
||||
rtnval, err := clh.generateSocket(defaultID, true)
|
||||
assert.NoError(err)
|
||||
assert.NotNil(rtnval)
|
||||
|
||||
if socket, ok := rtnval.(types.HybridVSock); ok {
|
||||
assert.Equal(socket.UdsPath, "/run/vc/vm/"+defaultID+"/"+clhSocket)
|
||||
assert.Equal(socket.Port, defaultPort)
|
||||
assert.NotEqual(socket.ContextID, 0, "ContextID 0 is reserved for the hypervisor communication")
|
||||
assert.NotEqual(socket.ContextID, 1, "ContextID 1 is reserved")
|
||||
assert.NotEqual(socket.ContextID, 2, "ContextID 2 is reserved for the host communication")
|
||||
assert.NotEqual(socket.ContextID, 0xffffffff, "ContextID 0xffffffff is reserved")
|
||||
} else {
|
||||
t.Fail()
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestClhReset(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
clh := &cloudHypervisor{
|
||||
ctx: context.Background(),
|
||||
cliBuilder: &DefaultCLIBuilder{},
|
||||
}
|
||||
|
||||
clh.state.PID = 10
|
||||
clh.state.VirtiofsdPID = 11
|
||||
clh.state.state = clhReady
|
||||
|
||||
clh.reset()
|
||||
|
||||
assert.Equal(clh.state.PID, 0)
|
||||
assert.Equal(clh.state.VirtiofsdPID, 0)
|
||||
assert.Equal(clh.state.state, clhNotReady)
|
||||
|
||||
}
|
||||
|
||||
func TestClhVirtiofsdArgs(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
|
||||
defaultSocketPath := "/a/b/c/doit.sock"
|
||||
|
||||
clhConfig := newClhConfig()
|
||||
clh := &cloudHypervisor{
|
||||
ctx: context.Background(),
|
||||
cliBuilder: &DefaultCLIBuilder{},
|
||||
config: clhConfig,
|
||||
}
|
||||
|
||||
args, err := clh.virtiofsdArgs(defaultSocketPath)
|
||||
assert.NoError(err)
|
||||
assert.Equal(strings.Join(args, " "), "-f -o vhost_user_socket="+defaultSocketPath+" -o source=/run/kata-containers/shared/sandboxes -o cache="+virtioFsCacheAlways)
|
||||
|
||||
}
|
||||
|
||||
func TestClhPath(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
|
||||
clhConfig := newClhConfig()
|
||||
clh := &cloudHypervisor{
|
||||
ctx: context.Background(),
|
||||
cliBuilder: &DefaultCLIBuilder{},
|
||||
config: clhConfig,
|
||||
}
|
||||
|
||||
defaultPath, _ := clh.config.HypervisorAssetPath()
|
||||
|
||||
path, err := clh.clhPath()
|
||||
assert.NoError(err)
|
||||
assert.Equal(path, defaultPath)
|
||||
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user