1 files changed, 47 insertions, 0 deletions
diff --git a/src/utils/postinstall.py b/src/utils/postinstall.py
index f6332b0..7fef437 100644
--- a/src/utils/postinstall.py
+++ b/src/utils/postinstall.py
@@ -22,6 +22,50 @@ from socket import gethostname
 CONFIGUREOS_LEGACY_ENABLED = False
 
 
+def set_windows_hostname(disk, partition, name):
+    logging.info(f'Configuring hostname')
+
+    if len(name) > 15:
+        logging.warning(f'Windows does nor permit hostnames that exceed 15 characters. Truncating {name}')
+        name = name[0:15]
+
+    byte_name = name.encode(WINDOWS_HIVE_ENCODING)
+
+    device = get_partition_device(disk, partition)
+    mountpoint = device.replace('dev', 'mnt')
+
+    if not mount_mkdir(device, mountpoint):
+        raise OgError(f'Unable to mount {device} into {mountpoint}')
+
+    try:
+        hive_path = mountpoint + WINDOWS_HIVE_SYSTEM
+        hive = hive_handler_open(hive_path, write=True)
+        root = hive.root()
+
+        select_node = get_node_child_from_path(hive, root, 'Select')
+        current_control_set_number = get_value_from_node(hive, select_node, 'Current')
+
+        control_set = f'ControlSet{current_control_set_number:03}'
+
+        computer_name_node = get_node_child_from_path(hive, root, f'{control_set}/Control/ComputerName/ComputerName')
+        name_value = {'key': 'ComputerName', 't': RegistryType.SZ.value, 'value': byte_name}
+        hive.node_set_value(computer_name_node, name_value)
+
+        parameters_node = get_node_child_from_path(hive, root, f'{control_set}/Services/Tcpip/Parameters')
+
+        hostname_value = {'key': 'HostName', 't': RegistryType.SZ.value, 'value': byte_name}
+        hive.node_set_value(parameters_node, hostname_value)
+
+        nvhostname_value = {'key': 'NV Hostname', 't': RegistryType.SZ.value, 'value': byte_name}
+        hive.node_set_value(parameters_node, nvhostname_value)
+
+        hive.commit(hive_path)
+    except Exception as e:
+        raise OgError(f'Unable to configure Windows hostname: {e}') from e
+    finally:
+        umount(mountpoint)
+
+
 def configure_os_custom(disk, partition):
     if not shutil.which('configureOsCustom'):
         raise OgError('configureOsCustom not found')
@@ -115,6 +159,9 @@ def configure_os_linux(disk, partition):
 
 
 def configure_os_windows(disk, partition):
+    hostname = gethostname()
+    set_windows_hostname(disk, partition, hostname)
+
     if is_uefi_supported():
         restore_windows_efi_bootloader(disk, partition)