1 files changed, 14 insertions, 49 deletions

diff --git a/scripts/lib/wic/engine.py b/scripts/lib/wic/engine.py
index 9d596be3a7..b9e60cbe4e 100644
--- a/scripts/lib/wic/engine.py
+++ b/scripts/lib/wic/engine.py
@@ -345,64 +345,29 @@ class Disk:
                                                    path))
 
     def copy(self, src, dest):
-        """Copy files or directories to/from the vfat or ext* partition."""
-        pnum = dest.part if isinstance(src, str) else src.part
-        partimg = self._get_part_image(pnum)
+        """Copy partition image into wic image."""
+        pnum =  dest.part if isinstance(src, str) else src.part
 
         if self.partitions[pnum].fstype.startswith('ext'):
-            if isinstance(src, str): # host to image case
-                if os.path.isdir(src):
-                    base = os.path.abspath(src)
-                    base_parent = os.path.dirname(base)
-                    cmds = []
-                    made = set()
-
-                    for root, dirs, files in os.walk(base):
-                        for fname in files:
-                            host_file = os.path.join(root, fname)
-                            rel = os.path.relpath(host_file, base_parent)
-                            dest_file = os.path.join(dest.path, rel)
-                            dest_dir = os.path.dirname(dest_file)
-
-                            # create dir structure (mkdir -p)
-                            parts = dest_dir.strip('/').split('/')
-                            cur = ''
-                            for p in parts:
-                                cur = cur + '/' + p
-                                if cur not in made:
-                                    cmds.append(f'mkdir "{cur}"')
-                                    made.add(cur)
-
-                            cmds.append(f'write "{host_file}" "{dest_file}"')
-
-                    # write script to a temp file
-                    with tempfile.NamedTemporaryFile(mode='w', delete=False,
-                                                     prefix='wic-debugfs-') as tf:
-                        for line in cmds:
-                            tf.write(line + '\n')
-                        scriptname = tf.name
-
-                    cmd = f"{self.debugfs} -w -f {scriptname} {partimg}"
-
-                else: # single file
-                    cmd = "printf 'cd {}\nwrite {} {}\n' | {} -w {}".\
-                          format(os.path.dirname(dest.path), src,
-                                 os.path.basename(src), self.debugfs, partimg)
-
-            else: # image to host case
+            if isinstance(src, str):
+                cmd = "printf 'cd {}\nwrite {} {}\n' | {} -w {}".\
+                      format(os.path.dirname(dest.path), src, os.path.basename(src),
+                             self.debugfs, self._get_part_image(pnum))
+            else: # copy from wic
+                # run both dump and rdump to support both files and directory
                 cmd = "printf 'cd {}\ndump /{} {}\nrdump /{} {}\n' | {} {}".\
                       format(os.path.dirname(src.path), src.path,
-                             dest, src.path, dest, self.debugfs, partimg)
-
+                             dest, src.path, dest, self.debugfs,
+                             self._get_part_image(pnum))
         else: # fat
             if isinstance(src, str):
                 cmd = "{} -i {} -snop {} ::{}".format(self.mcopy,
-                                                      partimg,
-                                                      src, dest.path)
+                                                  self._get_part_image(pnum),
+                                                  src, dest.path)
             else:
                 cmd = "{} -i {} -snop ::{} {}".format(self.mcopy,
-                                                      partimg,
-                                                      src.path, dest)
+                                                  self._get_part_image(pnum),
+                                                  src.path, dest)
 
         exec_cmd(cmd, as_shell=True)
         self._put_part_image(pnum)