def set_config(key, value):
""" Set a configuration value """
config = get_config()
config[key] = value
config_json = json.dumps(config)
config_path = get_config_path()
system.set_file_contents(config_path, config_json)
def set_udev_interface_mapping(self, interface_name, mac_addr):
""" Deploy a udev rule to force a predictable interface name to mac address mapping """
if not is_mounted(self.ROOT_MOUNT):
error("ERROR: Root volume not mounted", exit=True)
udev_path = f"{self.ROOT_MOUNT}/etc/udev/rules.d/70-persistent-net.rules"
udev_contents = get_file_contents(udev_path)
for line in udev_contents.split("\n"):
if interface_name in line:
error(
f"ERROR: '{interface_name}' in {udev_path} - remove to continue",
exit=True)
if mac_addr in line:
error(
f"ERROR: '{mac_addr}' in {udev_path} - remove to continue",
exit=True)
# {address} is meant to look like that, it is not an f-string missing its f
udev_parts = [
'SUBSYSTEM=="net"',
'ACTION=="add"',
'DRIVERS=="?*"',
('ATTR{address}=="' + mac_addr + '"'),
f'NAME="{interface_name}"',
]
# Join the entries to looks like 'SUBSYSTEM=="net", ACTION=="add",...' with a \n at the end
udev_line = ", ".join(udev_parts) + "\n"
print("")
print(f"Appending udev rule to file: {udev_path}")
print(udev_line)
set_file_contents(udev_path, udev_line, append=True)
print("udev file contents:")
print(get_file_contents(udev_path))
def set_interface(
self,
interface_name,
is_dhcp,
mac_addr,
ip_addr=None,
prefix=None,
gateway=None,
dns=(),
domain=None,
):
""" Deploy a RHEL styled interface file """
# create the /etc/sysconfig/network-scripts file
bootproto = "dhcp" if is_dhcp else "static"
iface_lines = [
f"DEVICE={interface_name}",
f"BOOTPROTO={bootproto}",
"ONBOOT=yes",
"USERCTL=no",
f"HWARDDR={mac_addr}",
]
if not is_dhcp:
iface_lines.append(f"IPADDR={ip_addr}")
iface_lines.append(f"PREFIX={prefix}")
if gateway is not None:
iface_lines.append("DEFROUTE=yes")
iface_lines.append(f"GATEWAY={gateway}")
for index, domain_server in enumerate(dns):
num = index + 1
iface_lines.append(f"DNS{num}={domain_server}")
if domain is not None:
iface_lines.append(f"DOMAIN={domain}")
iface_contents = "\n".join(iface_lines)
iface_path = f"{self.ROOT_MOUNT}/etc/sysconfig/network-scripts/ifcfg-{interface_name}"
print(f"Creating interface file at {iface_path}:")
print(iface_contents)
set_file_contents(iface_path, iface_contents)
def add_virtio_drivers(self, force=False):
""" Install VirtIO drivers to mounted system """
if not self.was_root_mounted:
error("ERROR: You must mount the volumes before you can add virtio drivers", exit=True)
self.debug_action(action="ADD VIRTIO DRIVERS")
ls_boot_lines = run(f"ls {self.ROOT_MOUNT}/boot")
initram_lines = [
line
for line in ls_boot_lines
if line.startswith("initramfs-") and line.endswith(".img") and "dump" not in line
]
for filename in initram_lines:
if "rescue" in filename or "kdump" in filename:
debug(f"Skipping rescue/kdump file: {filename}")
continue
kernel_version = filename.replace("initramfs-", "").replace(".img", "")
debug("Running lsinitrd to check for virtio drivers")
lsinitrd = self.chroot_run(f"lsinitrd /boot/{filename}")
virtio_line = next((line for line in lsinitrd if "virtio" in line.lower()), None)
if virtio_line is not None:
print(f"{filename} already has virtio drivers")
if force:
print("force=true, reinstalling")
else:
continue
print(f"Adding virtio drivers to {filename}")
drivers = "virtio_blk virtio_net virtio_scsi virtio_balloon"
cmd = f'dracut --add-drivers "{drivers}" -f /boot/{filename} {kernel_version}'
# Python+chroot causes dracut space delimiter to break - use a script file
script_file = f"{self.ROOT_MOUNT}/virtio.sh"
debug(f"writing script file: {script_file}")
debug(f"script file contents: {cmd}")
set_file_contents(script_file, cmd)
self.chroot_run("bash /virtio.sh")
debug(f"deleting script file: {script_file}")
os.remove(script_file)
self.debug_action(end=True)