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.

2 files changed, 14 insertions, 4 deletions

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)