From 23cd3e7048252777e6bb38288b66c14c2ec718db Mon Sep 17 00:00:00 2001
From: Papoteur <papoteur@mageia.org>
Date: Fri, 30 Jun 2023 11:28:07 +0200
Subject: Give user ownership to the backup image.

---
 backend/magiback     | 21 ++++++++++++++++++++-
 backend/raw_write.py |  5 ++++-
 2 files changed, 24 insertions(+), 2 deletions(-)

(limited to 'backend')

diff --git a/backend/magiback b/backend/magiback
index c65b3a3..526e98a 100755
--- a/backend/magiback
+++ b/backend/magiback
@@ -38,6 +38,12 @@ class Isodumper(raw_write.Dumper):
 					<arg type='s' name='label' direction='in'/>
 					<arg type='s' name='key' direction='in'/>
 				</method>
+				<method name='set_backup_mode'>
+					<arg type='x' name='uid' direction='in'/>
+					<arg type='x' name='gid' direction='in'/>
+				</method>
+				<method name='unset_backup_mode'>
+				</method>
 				<method name='end'>
 					<arg type='b' name='success' direction='out'/>
 					<arg type='s' name='message' direction='out'/>
@@ -69,11 +75,14 @@ class Isodumper(raw_write.Dumper):
         self.signature_checked = False
         self.writing_perm = False
         self.writing_target = ""
+        self.backup_mode = False
+        self.uid = 0
+        self.gid = 0
 
     def do_write(self, source, target, size, dbus_context):
         self.finished.clear()
         if check_permission('org.mageia.Magiback.Isodumper.write', dbus_context):
-            self.thread = threading.Thread(target=self._do_write, args=(source, target, size, ))
+            self.thread = threading.Thread(target=self._do_write, args=(source, target, size, self.backup_mode, self.uid, self.uid))
             self.thread.start()
             logging.debug("Writing thread started")
             self.writing_perm = True
@@ -124,6 +133,16 @@ class Isodumper(raw_write.Dumper):
         self.thread = threading.Thread(target=self._check_write, args=(target, source,))
         self.thread.start()
 
+    def set_backup_mode(self, uid, gid):
+        self.backup_mode = True
+        self.uid = uid
+        self.gid = gid
+
+    def unset_backup_mode(self):
+        self.backup_mode = False
+        self.uid = 0
+        self.gid = 0
+
     def run(self):
         self.loop.run()
 
diff --git a/backend/raw_write.py b/backend/raw_write.py
index a29b604..fe0fda9 100755
--- a/backend/raw_write.py
+++ b/backend/raw_write.py
@@ -39,7 +39,7 @@ from subprocess import Popen, PIPE
 
 class Dumper(object):
 
-    def _do_write(self,source,target, b):
+    def _do_write(self,source,target, b, backup_mode, uid, gid):
         # Writing step
         #Dump mode
         self.returncode=0
@@ -107,6 +107,9 @@ class Dumper(object):
                            self.return_message = message
                            self.finished.set()
                            return
+                if backup_mode:
+                    # Restore user as owner
+                    os.chown(target, uid, gid)
                 try:
                     ofc.close()
                 except:
-- 
cgit v1.2.1