Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 5 additions & 2 deletions src/bricoler/bricoler.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@
from .mtree import MtreeFile
from .task import Task, TaskParameter, TaskMeta, TaskSchedule
from .util import chdir, host_machine, info, run_cmd, warn
from .vm import FreeBSDVM, VMImage, VMHypervisor, BhyveRun, QEMURun, SSHCommandRunner
from .vm import FreeBSDVM, VMImage, VMHypervisor, BhyveRun, QEMURun, RVVMRun, SSHCommandRunner


class FreeBSDSrcRepository(GitRepository):
Expand Down Expand Up @@ -658,7 +658,10 @@ class FreeBSDVMBootTask(Task):
}

def run(self, ctx):
cls = QEMURun if self.hypervisor == VMHypervisor.QEMU else BhyveRun
match self.hypervisor:
case VMHypervisor.BHYVE: cls = BhyveRun
case VMHypervisor.QEMU: cls = QEMURun
case VMHypervisor.RVVM: cls = RVVMRun
if self.p9_shares:
p9_shares = [tuple(desc.split(':')) for desc in self.p9_shares.split(',')]
else:
Expand Down
42 changes: 34 additions & 8 deletions src/bricoler/vm.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,6 +65,7 @@ def select(self, d: Dict[str, str], default=None) -> str:
class VMHypervisor(Enum):
BHYVE = 'bhyve'
QEMU = 'qemu'
RVVM = 'rvvm'


class VMRun:
Expand All @@ -76,6 +77,7 @@ class BlockDriver(Enum):
class NetworkDriver(Enum):
VIRTIO = 1,
E1000 = 2,
NONE = 3,

def __init__(
self,
Expand Down Expand Up @@ -163,11 +165,10 @@ def bootrom_path(self) -> Path:
})

def network_driver_name(self) -> str:
driver = self.nic_driver
if driver == VMRun.NetworkDriver.VIRTIO:
return "virtio-net"
elif driver == VMRun.NetworkDriver.E1000:
return "e1000"
match self.nic_driver:
case VMRun.NetworkDriver.VIRTIO: return "virtio-net"
case VMRun.NetworkDriver.E1000: return "e1000"
raise ValueError(f"Unsupported network driver {driver} is not supported by bhyve")

def setup(self) -> List[Any]:
if BhyveRun.has_monitor_mode():
Expand Down Expand Up @@ -210,7 +211,8 @@ def add_device(desc):
add_device(f"{self.block_driver_name()},{self.image.path}")
for disk in self.extra_disks:
add_device(f"{self.block_driver_name()},{disk}")
add_device(f"{self.network_driver_name()},slirp,open,hostfwd=tcp:{self.ssh_addr[0]}:{self.ssh_addr[1]}-:22")
if self.nic_driver != VMRun.NetworkDriver.NONE:
add_device(f"{self.network_driver_name()},slirp,open,hostfwd=tcp:{self.ssh_addr[0]}:{self.ssh_addr[1]}-:22")
for share in self.p9_shares:
add_device(f"virtio-9p,{share[0]}={share[1]}")

Expand All @@ -219,6 +221,27 @@ def add_device(desc):
return [str(a) for a in bhyve_cmd]


class RVVMRun(VMRun):
def setup(self) -> List[str]:
rvvm_cmd = [
"rvvm",
"/usr/local/share/RVVM/fw_payload.bin",
"-image", f"{self.image.path}",
"-mem", f"{self.memory}M",
"-smp", f"{self.ncpus}",
"-nogui",
]

match self.nic_driver:
case VMRun.NetworkDriver.VIRTIO:
rvvm_cmd.extend(["-portfwd", f"tcp/{self.ssh_addr[0]}:{self.ssh_addr[1]}=22"])
case VMRun.NetworkDriver.NONE:
rvvm_cmd.extend(["-nonet"])
case _:
raise ValueError(f"Unsupported network driver {self.nic_driver} is not supported by RVVM")

return rvvm_cmd

class QEMURun(VMRun):
def bios_path(self) -> Optional[Path]:
return self.image.select({
Expand Down Expand Up @@ -290,8 +313,6 @@ def setup(self) -> List[Any]:
"-device", "virtio-rng-pci",
"-device", f"{self.block_driver_name()},drive=image",
"-drive", f"file={self.image.path},if=none,id=image,format=raw",
"-device", f"{self.nic_driver_name()},netdev=net0",
"-netdev", f"user,id=net0,hostfwd=tcp:{self.ssh_addr[0]}:{self.ssh_addr[1]}-:22",
"-gdb", f"tcp:{self.gdb_addr[0]}:{self.gdb_addr[1]}",
]
for disk in self.extra_disks:
Expand All @@ -312,6 +333,11 @@ def setup(self) -> List[Any]:
kernel_path = self.kernel_path()
if kernel_path is not None:
qemu_cmd.extend(["-kernel", kernel_path])
if self.nic_driver != VMRun.NetworkDriver.NONE:
qemu_cmd.extend([
"-device", f"{self.nic_driver_name()},netdev=net0",
"-netdev", f"user,id=net0,hostfwd=tcp:{self.ssh_addr[0]}:{self.ssh_addr[1]}-:22",
])

return [str(a) for a in qemu_cmd]

Expand Down