From cc70274079a0de78497011e2c665838522b6e1f5 Mon Sep 17 00:00:00 2001 From: Alejandro Sirgo Rica Date: Wed, 2 Oct 2024 12:13:25 +0200 Subject: src: check if the system is hibernated before /image/create Mount the system partition in readonly mode and check for the hiberfil.sys file if the target system is a Windows. Fail the image creation process if the target system is hibernated. --- src/utils/fs.py | 11 +++++++---- src/utils/probe.py | 7 +++++++ 2 files changed, 14 insertions(+), 4 deletions(-) (limited to 'src/utils') diff --git a/src/utils/fs.py b/src/utils/fs.py index 381f65b..d6cf36a 100644 --- a/src/utils/fs.py +++ b/src/utils/fs.py @@ -29,7 +29,7 @@ def find_mountpoint(path): return path -def mount_mkdir(source, target): +def mount_mkdir(source, target, readonly=False): """ Mounts and creates the mountpoint directory if it's not present. @@ -43,18 +43,21 @@ def mount_mkdir(source, target): return False if not os.path.ismount(target): - return mount(source, target) + return mount(source, target, readonly) return True -def mount(source, target): +def mount(source, target, readonly): """ Mounts source into target directoru using mount(8). Return true if exit code is 0. False otherwise. """ - cmd = f'mount {source} {target}' + if readonly: + cmd = f'mount -o ro {source} {target}' + else: + cmd = f'mount {source} {target}' proc = subprocess.run(cmd.split(), stderr=DEVNULL) return not proc.returncode diff --git a/src/utils/probe.py b/src/utils/probe.py index 2ed029b..e53f08f 100644 --- a/src/utils/probe.py +++ b/src/utils/probe.py @@ -178,3 +178,10 @@ def os_probe(mountpoint): return getwindowsversion(winreghives) else: return 'unknown' + +def is_hibernation_enabled(mountpoint): + if get_os_family(mountpoint) != OSFamily.WINDOWS: + return False + + hiberfile_path = f'{mountpoint}/hiberfil.sys' + return os.path.exists(hiberfile_path) -- cgit v1.2.3-18-g5258