mirror of
https://github.com/ultravioletrs/cocos.git
synced 2026-08-07 07:14:50 +00:00
Jovan Djukic
1f0eccfae7
CI / lint (push) Has been cancelled
CI / test (agent) (push) Has been cancelled
CI / test (cli) (push) Has been cancelled
CI / test (cmd) (push) Has been cancelled
CI / test (internal) (push) Has been cancelled
CI / test (manager, true) (push) Has been cancelled
CI / test (pkg) (push) Has been cancelled
CI / upload-coverage (push) Has been cancelled
421 lines
11 KiB
Go
421 lines
11 KiB
Go
// Copyright (c) Ultraviolet
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
package qemu
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestConstructQemuArgs(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
config Config
|
|
expected []string
|
|
}{
|
|
{
|
|
name: "Default configuration",
|
|
config: Config{
|
|
QemuBinPath: "qemu-system-x86_64",
|
|
EnableKVM: true,
|
|
Machine: "q35",
|
|
CPU: "EPYC",
|
|
SMPCount: 4,
|
|
MaxCPUs: 64,
|
|
MemID: "ram1",
|
|
MemoryConfig: MemoryConfig{
|
|
Size: "2048M",
|
|
Slots: 5,
|
|
Max: "30G",
|
|
},
|
|
OVMFCodeConfig: OVMFCodeConfig{
|
|
If: "pflash",
|
|
Format: "raw",
|
|
Unit: 0,
|
|
File: "/usr/share/OVMF/OVMF_CODE.fd",
|
|
ReadOnly: "on",
|
|
},
|
|
OVMFVarsConfig: OVMFVarsConfig{
|
|
If: "pflash",
|
|
Format: "raw",
|
|
Unit: 1,
|
|
File: "/usr/share/OVMF/OVMF_VARS.fd",
|
|
},
|
|
NetDevConfig: NetDevConfig{
|
|
ID: "vmnic",
|
|
HostFwdAgent: 7020,
|
|
GuestFwdAgent: 7002,
|
|
},
|
|
VirtioNetPciConfig: VirtioNetPciConfig{
|
|
DisableLegacy: "on",
|
|
IOMMUPlatform: true,
|
|
Addr: "0x2",
|
|
},
|
|
KernelConfig: KernelConfig{
|
|
KernelFile: "img/bzImage",
|
|
RootFsFile: "img/rootfs.cpio.gz",
|
|
},
|
|
KernelCommandLine: "quiet console=null",
|
|
NoGraphic: true,
|
|
Monitor: "pty",
|
|
},
|
|
expected: []string{
|
|
"-enable-kvm",
|
|
"-machine", "q35",
|
|
"-cpu", "EPYC",
|
|
"-smp", "4,maxcpus=64",
|
|
"-m", "2048M,slots=5,maxmem=30G",
|
|
"-drive", "if=pflash,format=raw,unit=0,file=/usr/share/OVMF/OVMF_CODE.fd,readonly=on",
|
|
"-drive", "if=pflash,format=raw,unit=1,file=/usr/share/OVMF/OVMF_VARS.fd",
|
|
"-netdev", "user,id=vmnic,hostfwd=tcp::7020-:7002",
|
|
"-device", "virtio-net-pci,disable-legacy=on,iommu_platform=true,netdev=vmnic,addr=0x2,romfile=",
|
|
"-kernel", "img/bzImage",
|
|
"-append", "quiet console=null",
|
|
"-initrd", "img/rootfs.cpio.gz",
|
|
"-nographic",
|
|
"-monitor", "pty",
|
|
},
|
|
},
|
|
{
|
|
name: "SEV-SNP enabled configuration",
|
|
config: Config{
|
|
QemuBinPath: "qemu-system-x86_64",
|
|
EnableKVM: true,
|
|
EnableSEVSNP: true,
|
|
Machine: "q35",
|
|
CPU: "EPYC",
|
|
SMPCount: 4,
|
|
MaxCPUs: 64,
|
|
MemID: "ram1",
|
|
MemoryConfig: MemoryConfig{
|
|
Size: "2048M",
|
|
Slots: 5,
|
|
Max: "30G",
|
|
},
|
|
OVMFCodeConfig: OVMFCodeConfig{
|
|
If: "pflash",
|
|
Format: "raw",
|
|
Unit: 0,
|
|
File: "/usr/share/OVMF/OVMF_CODE.fd",
|
|
ReadOnly: "on",
|
|
},
|
|
OVMFVarsConfig: OVMFVarsConfig{
|
|
If: "pflash",
|
|
Format: "raw",
|
|
Unit: 1,
|
|
File: "/usr/share/OVMF/OVMF_VARS.fd",
|
|
},
|
|
NetDevConfig: NetDevConfig{
|
|
ID: "vmnic",
|
|
HostFwdAgent: 7020,
|
|
GuestFwdAgent: 7002,
|
|
},
|
|
VirtioNetPciConfig: VirtioNetPciConfig{
|
|
DisableLegacy: "on",
|
|
IOMMUPlatform: true,
|
|
Addr: "0x2",
|
|
},
|
|
KernelConfig: KernelConfig{
|
|
KernelFile: "img/bzImage",
|
|
RootFsFile: "img/rootfs.cpio.gz",
|
|
},
|
|
SEVSNPConfig: SEVSNPConfig{
|
|
ID: "sev0",
|
|
CBitPos: 51,
|
|
ReducedPhysBits: 1,
|
|
},
|
|
IGVMConfig: IGVMConfig{
|
|
ID: "igvm0",
|
|
File: "/test/path/cocos-igvm.igvm",
|
|
},
|
|
KernelCommandLine: "quiet console=null",
|
|
NoGraphic: true,
|
|
Monitor: "pty",
|
|
},
|
|
expected: []string{
|
|
"-enable-kvm",
|
|
"-machine", "q35",
|
|
"-cpu", "EPYC",
|
|
"-smp", "4,maxcpus=64",
|
|
"-m", "2048M,slots=5,maxmem=30G",
|
|
"-netdev", "user,id=vmnic,hostfwd=tcp::7020-:7002",
|
|
"-device", "virtio-net-pci,disable-legacy=on,iommu_platform=true,netdev=vmnic,addr=0x2,romfile=",
|
|
"-machine", "confidential-guest-support=sev0,memory-backend=ram1,igvm-cfg=igvm0",
|
|
"-object", "memory-backend-memfd,id=ram1,size=2048M,share=true,prealloc=false",
|
|
"-object", "sev-snp-guest,id=sev0,cbitpos=51,reduced-phys-bits=1",
|
|
"-object", "igvm-cfg,id=igvm0,file=/test/path/cocos-igvm.igvm",
|
|
"-kernel", "img/bzImage",
|
|
"-append", "quiet console=null",
|
|
"-initrd", "img/rootfs.cpio.gz",
|
|
"-nographic",
|
|
"-monitor", "pty",
|
|
},
|
|
},
|
|
{
|
|
name: "GPU passthrough configuration",
|
|
config: Config{
|
|
QemuBinPath: "qemu-system-x86_64",
|
|
EnableKVM: true,
|
|
Machine: "q35",
|
|
CPU: "EPYC",
|
|
SMPCount: 4,
|
|
MaxCPUs: 64,
|
|
MemID: "ram1",
|
|
MemoryConfig: MemoryConfig{
|
|
Size: "2048M",
|
|
Slots: 5,
|
|
Max: "30G",
|
|
},
|
|
OVMFCodeConfig: OVMFCodeConfig{
|
|
If: "pflash",
|
|
Format: "raw",
|
|
Unit: 0,
|
|
File: "/usr/share/OVMF/OVMF_CODE.fd",
|
|
ReadOnly: "on",
|
|
},
|
|
OVMFVarsConfig: OVMFVarsConfig{
|
|
If: "pflash",
|
|
Format: "raw",
|
|
Unit: 1,
|
|
File: "/usr/share/OVMF/OVMF_VARS.fd",
|
|
},
|
|
NetDevConfig: NetDevConfig{
|
|
ID: "vmnic",
|
|
HostFwdAgent: 7020,
|
|
GuestFwdAgent: 7002,
|
|
},
|
|
VirtioNetPciConfig: VirtioNetPciConfig{
|
|
DisableLegacy: "on",
|
|
IOMMUPlatform: true,
|
|
Addr: "0x2",
|
|
},
|
|
KernelConfig: KernelConfig{
|
|
KernelFile: "img/bzImage",
|
|
RootFsFile: "img/rootfs.cpio.gz",
|
|
},
|
|
GPUConfig: GPUConfig{
|
|
EnableGPU: true,
|
|
GPUBDF: "0000:02:00.0",
|
|
PCIeRootPort: "pci.1",
|
|
PCIeBus: "pcie.0",
|
|
FWCfgPciMmio: "262144",
|
|
},
|
|
KernelCommandLine: "quiet console=null",
|
|
NoGraphic: true,
|
|
Monitor: "pty",
|
|
},
|
|
expected: []string{
|
|
"-enable-kvm",
|
|
"-machine", "q35",
|
|
"-cpu", "EPYC",
|
|
"-smp", "4,maxcpus=64",
|
|
"-m", "2048M,slots=5,maxmem=30G",
|
|
"-drive", "if=pflash,format=raw,unit=0,file=/usr/share/OVMF/OVMF_CODE.fd,readonly=on",
|
|
"-drive", "if=pflash,format=raw,unit=1,file=/usr/share/OVMF/OVMF_VARS.fd",
|
|
"-netdev", "user,id=vmnic,hostfwd=tcp::7020-:7002",
|
|
"-device", "virtio-net-pci,disable-legacy=on,iommu_platform=true,netdev=vmnic,addr=0x2,romfile=",
|
|
"-device", "pcie-root-port,id=pci.1,bus=pcie.0",
|
|
"-device", "vfio-pci,host=0000:02:00.0,bus=pci.1",
|
|
"-fw_cfg", "name=opt/ovmf/X-PciMmio64Mb,string=262144",
|
|
"-kernel", "img/bzImage",
|
|
"-append", "quiet console=null",
|
|
"-initrd", "img/rootfs.cpio.gz",
|
|
"-nographic",
|
|
"-monitor", "pty",
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
result := tt.config.ConstructQemuArgs()
|
|
if !reflect.DeepEqual(result, tt.expected) {
|
|
t.Errorf("ConstructQemuArgs() = %v, want %v", result, tt.expected)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestConstructQemuArgs_HostData(t *testing.T) {
|
|
config := Config{
|
|
EnableSEVSNP: true,
|
|
SEVSNPConfig: SEVSNPConfig{
|
|
ID: "sev0",
|
|
CBitPos: 51,
|
|
ReducedPhysBits: 1,
|
|
EnableHostData: true,
|
|
HostData: "test-host-data",
|
|
},
|
|
}
|
|
|
|
result := config.ConstructQemuArgs()
|
|
|
|
expected := "-object"
|
|
expectedValue := "sev-snp-guest,id=sev0,cbitpos=51,reduced-phys-bits=1,host-data=test-host-data"
|
|
|
|
found := false
|
|
for i, arg := range result {
|
|
if arg == expected && i+1 < len(result) {
|
|
if result[i+1] == expectedValue {
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
if !found {
|
|
t.Errorf("ConstructQemuArgs() did not contain expected SEV-SNP configuration with host data")
|
|
}
|
|
}
|
|
|
|
func TestConstructQemuArgs_TDX(t *testing.T) {
|
|
config := Config{
|
|
EnableKVM: true,
|
|
EnableTDX: true,
|
|
Machine: "q35",
|
|
CPU: "EPYC",
|
|
SMPCount: 4,
|
|
MaxCPUs: 64,
|
|
MemID: "ram1",
|
|
MemoryConfig: MemoryConfig{
|
|
Size: "4096M",
|
|
Slots: 8,
|
|
Max: "64G",
|
|
},
|
|
NetDevConfig: NetDevConfig{
|
|
ID: "vmnic",
|
|
HostFwdAgent: 7020,
|
|
GuestFwdAgent: 7002,
|
|
},
|
|
VirtioNetPciConfig: VirtioNetPciConfig{
|
|
DisableLegacy: "on",
|
|
IOMMUPlatform: true,
|
|
Addr: "0x2",
|
|
},
|
|
TDXConfig: TDXConfig{
|
|
ID: "tdx0",
|
|
QuoteGenerationPort: 4050,
|
|
OVMF: "/usr/share/ovmf/OVMF.fd",
|
|
},
|
|
KernelConfig: KernelConfig{
|
|
KernelFile: "img/bzImage",
|
|
RootFsFile: "img/rootfs.cpio.gz",
|
|
},
|
|
KernelCommandLine: "quiet console=null",
|
|
NoGraphic: true,
|
|
Monitor: "pty",
|
|
}
|
|
|
|
expected := []string{
|
|
"-enable-kvm",
|
|
"-machine", "q35",
|
|
"-cpu", "EPYC",
|
|
"-smp", "4,maxcpus=64",
|
|
"-m", "4096M,slots=8,maxmem=64G",
|
|
"-netdev", "user,id=vmnic,hostfwd=tcp::7020-:7002",
|
|
"-device", "virtio-net-pci,disable-legacy=on,iommu_platform=true,netdev=vmnic,addr=0x2,romfile=",
|
|
"-object", "{\"qom-type\":\"tdx-guest\",\"id\":\"tdx0\",\"quote-generation-socket\":{\"type\": \"vsock\", \"cid\":\"2\",\"port\":\"4050\"}}",
|
|
"-machine", "confidential-guest-support=tdx0,memory-backend=ram1,hpet=off",
|
|
"-object", "memory-backend-memfd,id=ram1,size=4096M,share=true,prealloc=false",
|
|
"-bios", "/usr/share/ovmf/OVMF.fd",
|
|
"-nodefaults",
|
|
"-kernel", "img/bzImage",
|
|
"-append", "quiet console=null",
|
|
"-initrd", "img/rootfs.cpio.gz",
|
|
"-nographic",
|
|
"-monitor", "pty",
|
|
}
|
|
|
|
result := config.ConstructQemuArgs()
|
|
if !reflect.DeepEqual(result, expected) {
|
|
t.Errorf("ConstructQemuArgs() = %v, want %v", result, expected)
|
|
}
|
|
}
|
|
|
|
func TestConstructQemuArgs_DiskBootSkipsKernelAndInitrd(t *testing.T) {
|
|
config := Config{
|
|
EnableKVM: true,
|
|
EnableDisk: true,
|
|
Machine: "q35",
|
|
CPU: "EPYC",
|
|
SMPCount: 4,
|
|
MaxCPUs: 64,
|
|
MemID: "ram1",
|
|
MemoryConfig: MemoryConfig{
|
|
Size: "2048M",
|
|
Slots: 5,
|
|
Max: "30G",
|
|
},
|
|
NetDevConfig: NetDevConfig{
|
|
ID: "vmnic",
|
|
HostFwdAgent: 7020,
|
|
GuestFwdAgent: 7002,
|
|
},
|
|
VirtioNetPciConfig: VirtioNetPciConfig{
|
|
DisableLegacy: "on",
|
|
IOMMUPlatform: true,
|
|
Addr: "0x2",
|
|
},
|
|
DiskConfig: DiskConfig{
|
|
DstFile: "img/disk.img",
|
|
ID: "disk0",
|
|
Format: "qcow2",
|
|
SCSIID: "scsi0",
|
|
},
|
|
KernelConfig: KernelConfig{
|
|
KernelFile: "img/bzImage",
|
|
RootFsFile: "img/rootfs.cpio.gz",
|
|
},
|
|
NoGraphic: true,
|
|
Monitor: "pty",
|
|
}
|
|
|
|
result := config.ConstructQemuArgs()
|
|
|
|
for _, forbidden := range []string{"-kernel", "-append", "-initrd"} {
|
|
for _, arg := range result {
|
|
if arg == forbidden {
|
|
t.Fatalf("ConstructQemuArgs() unexpectedly contained %s during disk boot: %v", forbidden, result)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestConstructQemuArgs_EnableDisk(t *testing.T) {
|
|
config := Config{
|
|
EnableDisk: true,
|
|
DiskConfig: DiskConfig{
|
|
SrcFile: "img/enc_os.qcow2",
|
|
DstFile: "img/enc_os_dst.qcow2",
|
|
ID: "disk0",
|
|
Format: "qcow2",
|
|
SCSIID: "scsi0",
|
|
},
|
|
}
|
|
|
|
result := config.ConstructQemuArgs()
|
|
|
|
expected := []string{
|
|
"-drive", "file=img/enc_os_dst.qcow2,if=none,id=disk0,format=qcow2",
|
|
"-device", "virtio-scsi-pci,id=scsi0,disable-legacy=on,iommu_platform=true",
|
|
"-device", "scsi-hd,drive=disk0,bus=scsi0.0",
|
|
}
|
|
|
|
var found []bool = make([]bool, len(expected))
|
|
for i, arg := range result {
|
|
for j := 0; j < len(expected); j += 2 {
|
|
if arg == expected[j] && i+1 < len(result) && result[i+1] == expected[j+1] {
|
|
found[j] = true
|
|
found[j+1] = true
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
for j, f := range found {
|
|
if !f {
|
|
t.Errorf("ConstructQemuArgs() did not contain expected disk configuration: %s", expected[j])
|
|
}
|
|
}
|
|
}
|