aboutsummaryrefslogtreecommitdiffstats
path: root/resources/libraries/python
diff options
context:
space:
mode:
Diffstat (limited to 'resources/libraries/python')
-rw-r--r--resources/libraries/python/QemuManager.py89
-rw-r--r--resources/libraries/python/QemuUtils.py250
2 files changed, 200 insertions, 139 deletions
diff --git a/resources/libraries/python/QemuManager.py b/resources/libraries/python/QemuManager.py
index ce55be0a51..66a21aa115 100644
--- a/resources/libraries/python/QemuManager.py
+++ b/resources/libraries/python/QemuManager.py
@@ -56,15 +56,12 @@ class QemuManager:
nf_dtcr = kwargs[u"nf_dtcr"] \
if isinstance(kwargs[u"nf_dtcr"], int) else 2
- img = Constants.QEMU_VM_KERNEL
-
for nf_chain in range(1, nf_chains + 1):
for nf_node in range(1, nf_nodes + 1):
qemu_id = (nf_chain - 1) * nf_nodes + nf_node
name = f"{node}_{qemu_id}"
- sock1 = f"/run/vpp/sock-{qemu_id}-1"
- sock2 = f"/run/vpp/sock-{qemu_id}-2"
idx1 = (nf_chain - 1) * nf_nodes * 2 + nf_node * 2 - 1
+
vif1_mac = Topology.get_interface_mac(
self.nodes[node], f"vhost{idx1}"
) if kwargs[u"vnf"] == u"testpmd_mac" \
@@ -83,27 +80,15 @@ class QemuManager:
vs_dtc=vs_dtc, nf_dtc=nf_dtc, nf_dtcr=nf_dtcr
)
- self.machines[name] = QemuUtils(
- node=self.nodes[node], qemu_id=qemu_id,
- smp=len(self.machines_affinity[name]), mem=4096,
- vnf=kwargs[u"vnf"], img=img
- )
- self.machines[name].configure_kernelvm_vnf(
- mac1=f"52:54:00:00:{qemu_id:02x}:01",
- mac2=f"52:54:00:00:{qemu_id:02x}:02",
- vif1_mac=vif1_mac, vif2_mac=vif2_mac, queues=queues,
- jumbo_frames=kwargs[u"jumbo"]
- )
- self.machines[name].qemu_add_vhost_user_if(
- sock1, jumbo_frames=kwargs[u"jumbo"], queues=queues,
- queue_size=kwargs[u"perf_qemu_qsz"],
- csum=kwargs[u"enable_csum"], gso=kwargs[u"enable_gso"]
- )
- self.machines[name].qemu_add_vhost_user_if(
- sock2, jumbo_frames=kwargs[u"jumbo"], queues=queues,
- queue_size=kwargs[u"perf_qemu_qsz"],
- csum=kwargs[u"enable_csum"], gso=kwargs[u"enable_gso"]
- )
+ try:
+ getattr(self, f'_c_{kwargs["vnf"]}')(
+ qemu_id=qemu_id, name=name, queues=queues, **kwargs
+ )
+ except AttributeError:
+ self._c_default(
+ qemu_id=qemu_id, name=name, queues=queues,
+ vif1_mac=vif1_mac, vif2_mac=vif2_mac, **kwargs
+ )
def construct_vms_on_all_nodes(self, **kwargs):
"""Construct 1..Mx1..N VMs(s) with specified name on all nodes.
@@ -124,7 +109,9 @@ class QemuManager:
"""
for machine, machine_affinity in \
zip(self.machines.values(), self.machines_affinity.values()):
- machine.qemu_start()
+ index = list(self.machines.values()).index(machine)
+ name = list(self.machines.keys())[index]
+ self.nodes[name] = machine.qemu_start()
if pinning:
machine.qemu_set_affinity(*machine_affinity)
@@ -134,8 +121,58 @@ class QemuManager:
:param force: Force kill all Qemu instances by pkill qemu if True.
:type force: bool
"""
+ for node in list(self.nodes.values()):
+ if node["type"] == NodeType.VM:
+ try:
+ self.nodes.popitem(node)
+ except TypeError:
+ pass
for machine in self.machines.values():
if force:
machine.qemu_kill_all()
else:
machine.qemu_kill()
+
+ def _c_default(self, **kwargs):
+ """Instantiate one VM with default configuration.
+
+ :param kwargs: Named parameters.
+ :type kwargs: dict
+ """
+ qemu_id = kwargs[u"qemu_id"]
+ name = kwargs[u"name"]
+
+ self.machines[name] = QemuUtils(
+ node=self.nodes[kwargs[u"node"]],
+ qemu_id=qemu_id,
+ smp=len(self.machines_affinity[name]),
+ mem=4096,
+ vnf=kwargs[u"vnf"],
+ img=Constants.QEMU_VM_KERNEL
+ )
+ self.machines[name].add_default_params()
+ self.machines[name].add_kernelvm_params()
+ self.machines[name].configure_kernelvm_vnf(
+ mac1=f"52:54:00:00:{qemu_id:02x}:01",
+ mac2=f"52:54:00:00:{qemu_id:02x}:02",
+ vif1_mac=kwargs[u"vif1_mac"],
+ vif2_mac=kwargs[u"vif2_mac"],
+ queues=kwargs[u"queues"],
+ jumbo_frames=kwargs[u"jumbo"]
+ )
+ self.machines[name].add_vhost_user_if(
+ f"/run/vpp/sock-{qemu_id}-1",
+ jumbo_frames=kwargs[u"jumbo"],
+ queues=kwargs[u"queues"],
+ queue_size=kwargs[u"perf_qemu_qsz"],
+ csum=kwargs[u"enable_csum"],
+ gso=kwargs[u"enable_gso"]
+ )
+ self.machines[name].add_vhost_user_if(
+ f"/run/vpp/sock-{qemu_id}-2",
+ jumbo_frames=kwargs[u"jumbo"],
+ queues=kwargs[u"queues"],
+ queue_size=kwargs[u"perf_qemu_qsz"],
+ csum=kwargs[u"enable_csum"],
+ gso=kwargs[u"enable_gso"]
+ )
diff --git a/resources/libraries/python/QemuUtils.py b/resources/libraries/python/QemuUtils.py
index 3ba64f0dbf..bb0c20c554 100644
--- a/resources/libraries/python/QemuUtils.py
+++ b/resources/libraries/python/QemuUtils.py
@@ -28,7 +28,6 @@ from resources.libraries.python.OptionString import OptionString
from resources.libraries.python.ssh import exec_cmd, exec_cmd_no_error
from resources.libraries.python.topology import NodeType, Topology
from resources.libraries.python.VppConfigGenerator import VppConfigGenerator
-from resources.libraries.python.VPPUtil import VPPUtil
__all__ = [u"QemuUtils"]
@@ -95,9 +94,9 @@ class QemuUtils:
self._opt[u"vnf"] = vnf
# Temporary files.
self._temp = dict()
+ self._temp[u"log"] = f"/tmp/serial_{qemu_id}.log"
self._temp[u"pidfile"] = f"/run/qemu_{qemu_id}.pid"
if img == Constants.QEMU_VM_IMAGE:
- self._opt[u"vm_type"] = u"nestedvm"
self._temp[u"qmp"] = f"/run/qmp_{qemu_id}.sock"
self._temp[u"qga"] = f"/run/qga_{qemu_id}.sock"
elif img == Constants.QEMU_VM_KERNEL:
@@ -105,8 +104,6 @@ class QemuUtils:
node, f"ls -1 {Constants.QEMU_VM_KERNEL}* | tail -1",
message=u"Qemu Kernel VM image not found!"
)
- self._opt[u"vm_type"] = u"kernelvm"
- self._temp[u"log"] = f"/tmp/serial_{qemu_id}.log"
self._temp[u"ini"] = f"/etc/vm_init_{qemu_id}.conf"
self._opt[u"initrd"], _ = exec_cmd_no_error(
node, f"ls -1 {Constants.QEMU_VM_KERNEL_INITRD}* | tail -1",
@@ -116,17 +113,6 @@ class QemuUtils:
raise RuntimeError(f"QEMU: Unknown VM image option: {img}")
# Computed parameters for QEMU command line.
self._params = OptionString(prefix=u"-")
- self.add_params()
-
- def add_params(self):
- """Set QEMU command line parameters."""
- self.add_default_params()
- if self._opt.get(u"vm_type", u"") == u"nestedvm":
- self.add_nestedvm_params()
- elif self._opt.get(u"vm_type", u"") == u"kernelvm":
- self.add_kernelvm_params()
- else:
- raise RuntimeError(u"QEMU: Unsupported VM type!")
def add_default_params(self):
"""Set default QEMU command line parameters."""
@@ -136,9 +122,7 @@ class QemuUtils:
u"name", f"vnf{self._opt.get(u'qemu_id')},debug-threads=on"
)
self._params.add(u"no-user-config")
- self._params.add_with_value(u"monitor", u"none")
- self._params.add_with_value(u"display", u"none")
- self._params.add_with_value(u"vga", u"none")
+ self._params.add(u"nographic")
self._params.add(u"enable-kvm")
self._params.add_with_value(u"pidfile", self._temp.get(u"pidfile"))
self._params.add_with_value(u"cpu", u"host")
@@ -156,32 +140,59 @@ class QemuUtils:
self._params.add_with_value(u"numa", u"node,memdev=mem")
self._params.add_with_value(u"balloon", u"none")
- def add_nestedvm_params(self):
- """Set NestedVM QEMU parameters."""
+ def add_net_user(self):
+ """Set managment port forwarding."""
self._params.add_with_value(
- u"net",
- f"nic,macaddr=52:54:00:00:{self._opt.get(u'qemu_id'):02x}:ff"
+ u"netdev", f"user,id=mgmt,net=192.168.76.0/24,"
+ f"hostfwd=tcp::{self._vm_info[u'port']}-:22"
)
self._params.add_with_value(
- u"net", f"user,hostfwd=tcp::{self._vm_info[u'port']}-:22"
+ u"device", f"virtio-net,netdev=mgmt"
)
- locking = u",file.locking=off"
+
+ def add_qmp_qga(self):
+ """Set QMP, QGA management."""
self._params.add_with_value(
- u"drive", f"file={self._opt.get(u'img')},"
- f"format=raw,cache=none,if=virtio{locking}"
+ u"chardev", f"socket,path={self._temp.get(u'qga')},"
+ f"server,nowait,id=qga0"
)
+ self._params.add_with_value(u"device", u"isa-serial,chardev=qga0")
self._params.add_with_value(
u"qmp", f"unix:{self._temp.get(u'qmp')},server,nowait"
)
+
+ def add_serial(self):
+ """Set serial to file redirect."""
self._params.add_with_value(
u"chardev", f"socket,host=127.0.0.1,"
f"port={self._vm_info[u'serial']},id=gnc0,server,nowait")
self._params.add_with_value(u"device", u"isa-serial,chardev=gnc0")
self._params.add_with_value(
- u"chardev", f"socket,path={self._temp.get(u'qga')},"
- f"server,nowait,id=qga0"
+ u"serial", f"file:{self._temp.get(u'log')}"
+ )
+
+ def add_drive_cdrom(self, drive_file):
+ """Set CD-ROM drive.
+
+ :param drive_file: Path to drive image.
+ :type drive_file: str
+ """
+ self._params.add_with_value(
+ u"drive", f"file={drive_file},media=cdrom"
+ )
+
+ def add_drive(self, drive_file, drive_format):
+ """Set drive with custom format.
+
+ :param drive_file: Path to drive image.
+ :param drive_format: Drive image format.
+ :type drive_file: str
+ :type drive_format: str
+ """
+ self._params.add_with_value(
+ u"drive", f"file={drive_file},format={drive_format},"
+ u"cache=none,if=virtio,file.locking=off"
)
- self._params.add_with_value(u"device", u"isa-serial,chardev=qga0")
def add_kernelvm_params(self):
"""Set KernelVM QEMU parameters."""
@@ -203,6 +214,62 @@ class QemuUtils:
f"init={self._temp.get(u'ini')} fastboot'"
)
+ def add_vhost_user_if(
+ self, socket, server=True, jumbo_frames=False, queue_size=None,
+ queues=1, csum=False, gso=False):
+ """Add Vhost-user interface.
+
+ :param socket: Path of the unix socket.
+ :param server: If True the socket shall be a listening socket.
+ :param jumbo_frames: Set True if jumbo frames are used in the test.
+ :param queue_size: Vring queue size.
+ :param queues: Number of queues.
+ :param csum: Checksum offloading.
+ :param gso: Generic segmentation offloading.
+ :type socket: str
+ :type server: bool
+ :type jumbo_frames: bool
+ :type queue_size: int
+ :type queues: int
+ :type csum: bool
+ :type gso: bool
+ """
+ self._vhost_id += 1
+ self._params.add_with_value(
+ u"chardev", f"socket,id=char{self._vhost_id},"
+ f"path={socket}{u',server' if server is True else u''}"
+ )
+ self._params.add_with_value(
+ u"netdev", f"vhost-user,id=vhost{self._vhost_id},"
+ f"chardev=char{self._vhost_id},queues={queues}"
+ )
+ mac = f"52:54:00:00:{self._opt.get(u'qemu_id'):02x}:" \
+ f"{self._vhost_id:02x}"
+ queue_size = f"rx_queue_size={queue_size},tx_queue_size={queue_size}" \
+ if queue_size else u""
+ self._params.add_with_value(
+ u"device", f"virtio-net-pci,netdev=vhost{self._vhost_id},mac={mac},"
+ f"addr={self._vhost_id+5}.0,mq=on,vectors={2 * queues + 2},"
+ f"csum={u'on' if csum else u'off'},gso={u'on' if gso else u'off'},"
+ f"guest_tso4=off,guest_tso6=off,guest_ecn=off,"
+ f"{queue_size}"
+ )
+
+ # Add interface MAC and socket to the node dict.
+ if_data = {u"mac_address": mac, u"socket": socket}
+ if_name = f"vhost{self._vhost_id}"
+ self._vm_info[u"interfaces"][if_name] = if_data
+ # Add socket to temporary file list.
+ self._temp[if_name] = socket
+
+ def add_vfio_pci_if(self, pci):
+ """Add VFIO PCI interface.
+
+ :param pci: PCI address of interface.
+ :type pci: str
+ """
+ self._params.add_with_value(u"device", f"vfio-pci,host={pci}")
+
def create_kernelvm_config_vpp(self, **kwargs):
"""Create QEMU VPP config files.
@@ -224,11 +291,19 @@ class QemuUtils:
vpp_config.add_unix_cli_listen()
vpp_config.add_unix_exec(running)
vpp_config.add_socksvr()
+ vpp_config.add_buffers_per_numa(107520)
vpp_config.add_cpu_main_core(u"0")
if self._opt.get(u"smp") > 1:
vpp_config.add_cpu_corelist_workers(f"1-{self._opt.get(u'smp')-1}")
vpp_config.add_plugin(u"disable", u"default")
- if "virtio" not in self._opt.get(u'vnf'):
+ vpp_config.add_plugin(u"enable", u"ping_plugin.so")
+ if "ipsec" in self._opt.get(u'vnf'):
+ vpp_config.add_plugin(u"enable", u"crypto_native_plugin.so")
+ vpp_config.add_plugin(u"enable", u"crypto_ipsecmb_plugin.so")
+ vpp_config.add_plugin(u"enable", u"crypto_openssl_plugin.so")
+ if "2vfpt" in self._opt.get(u'vnf'):
+ vpp_config.add_plugin(u"enable", u"avf_plugin.so")
+ if "vhost" in self._opt.get(u'vnf'):
vpp_config.add_plugin(u"enable", u"dpdk_plugin.so")
vpp_config.add_dpdk_dev(u"0000:00:06.0", u"0000:00:07.0")
vpp_config.add_dpdk_dev_default_rxq(kwargs[u"queues"])
@@ -401,54 +476,6 @@ class QemuUtils:
self.qemu_kill_all()
raise
- def qemu_add_vhost_user_if(
- self, socket, server=True, jumbo_frames=False, queue_size=None,
- queues=1, csum=False, gso=False):
- """Add Vhost-user interface.
-
- :param socket: Path of the unix socket.
- :param server: If True the socket shall be a listening socket.
- :param jumbo_frames: Set True if jumbo frames are used in the test.
- :param queue_size: Vring queue size.
- :param queues: Number of queues.
- :param csum: Checksum offloading.
- :param gso: Generic segmentation offloading.
- :type socket: str
- :type server: bool
- :type jumbo_frames: bool
- :type queue_size: int
- :type queues: int
- :type csum: bool
- :type gso: bool
- """
- self._vhost_id += 1
- self._params.add_with_value(
- u"chardev", f"socket,id=char{self._vhost_id},"
- f"path={socket}{u',server' if server is True else u''}"
- )
- self._params.add_with_value(
- u"netdev", f"vhost-user,id=vhost{self._vhost_id},"
- f"chardev=char{self._vhost_id},queues={queues}"
- )
- mac = f"52:54:00:00:{self._opt.get(u'qemu_id'):02x}:" \
- f"{self._vhost_id:02x}"
- queue_size = f"rx_queue_size={queue_size},tx_queue_size={queue_size}" \
- if queue_size else u""
- self._params.add_with_value(
- u"device", f"virtio-net-pci,netdev=vhost{self._vhost_id},mac={mac},"
- f"addr={self._vhost_id+5}.0,mq=on,vectors={2 * queues + 2},"
- f"csum={u'on' if csum else u'off'},gso={u'on' if gso else u'off'},"
- f"guest_tso4=off,guest_tso6=off,guest_ecn=off,"
- f"{queue_size}"
- )
-
- # Add interface MAC and socket to the node dict.
- if_data = {u"mac_address": mac, u"socket": socket}
- if_name = f"vhost{self._vhost_id}"
- self._vm_info[u"interfaces"][if_name] = if_data
- # Add socket to temporary file list.
- self._temp[if_name] = socket
-
def _qemu_qmp_exec(self, cmd):
"""Execute QMP command.
@@ -507,16 +534,40 @@ class QemuUtils:
return json.loads(stdout.split(u"\n", 1)[0]) if stdout else dict()
def _wait_until_vm_boot(self):
- """Wait until QEMU with NestedVM is booted."""
- if self._opt.get(u"vm_type") == u"nestedvm":
- self._wait_until_nestedvm_boot()
- self._update_vm_interfaces()
- elif self._opt.get(u"vm_type") == u"kernelvm":
- self._wait_until_kernelvm_boot()
+ """Wait until QEMU VM is booted."""
+ try:
+ getattr(self, f'_wait_{self._opt["vnf"]}')()
+ except AttributeError:
+ self._wait_default()
+
+ def _wait_default(self, retries=60):
+ """Wait until QEMU with VPP is booted.
+
+ :param retries: Number of retries.
+ :type retries: int
+ """
+ for _ in range(retries):
+ command = f"tail -1 {self._temp.get(u'log')}"
+ stdout = None
+ try:
+ stdout, _ = exec_cmd_no_error(self._node, command, sudo=True)
+ sleep(1)
+ except RuntimeError:
+ pass
+ if "vpp " in stdout and "built by" in stdout:
+ break
+ if u"Press enter to exit" in stdout:
+ break
+ if u"reboot: Power down" in stdout:
+ raise RuntimeError(
+ f"QEMU: NF failed to run on {self._node[u'host']}!"
+ )
else:
- raise RuntimeError(u"QEMU: Unsupported VM type!")
+ raise RuntimeError(
+ f"QEMU: Timeout, VM not booted on {self._node[u'host']}!"
+ )
- def _wait_until_nestedvm_boot(self, retries=12):
+ def _wait_nestedvm(self, retries=12):
"""Wait until QEMU with NestedVM is booted.
First try to flush qga until there is output.
@@ -564,33 +615,6 @@ class QemuUtils:
f"QEMU: Timeout, VM not booted on {self._node[u'host']}!"
)
- def _wait_until_kernelvm_boot(self, retries=60):
- """Wait until QEMU KernelVM is booted.
-
- :param retries: Number of retries.
- :type retries: int
- """
- vpp_ver = VPPUtil.vpp_show_version(self._node)
-
- for _ in range(retries):
- command = f"tail -1 {self._temp.get(u'log')}"
- stdout = None
- try:
- stdout, _ = exec_cmd_no_error(self._node, command, sudo=True)
- sleep(1)
- except RuntimeError:
- pass
- if vpp_ver in stdout or u"Press enter to exit" in stdout:
- break
- if u"reboot: Power down" in stdout:
- raise RuntimeError(
- f"QEMU: NF failed to run on {self._node[u'host']}!"
- )
- else:
- raise RuntimeError(
- f"QEMU: Timeout, VM not booted on {self._node[u'host']}!"
- )
-
def _update_vm_interfaces(self):
"""Update interface names in VM node dict."""
# Send guest-network-get-interfaces command via QGA, output example: