add support for setting the uid and gid of entries

closes #115

Author: Changaco

libarchive/entry.py

@@ -42,10 +42,18 @@ def filetype(self):
     def uid(self):
         return ffi.entry_uid(self._entry_p)
 
+    @uid.setter
+    def uid(self, uid):
+        ffi.entry_set_uid(self._entry_p, uid)
+
     @property
     def gid(self):
         return ffi.entry_gid(self._entry_p)
 
+    @gid.setter
+    def gid(self, gid):
+        ffi.entry_set_gid(self._entry_p, gid)
+
     def get_blocks(self, block_size=ffi.page_size):
         archive_p = self._archive_p
         buf = create_string_buffer(block_size)

libarchive/ffi.py

@@ -195,6 +195,8 @@ def get_write_filter_function(filter_name):
 
 ffi('entry_set_size', [c_archive_entry_p, c_longlong], None)
 ffi('entry_set_filetype', [c_archive_entry_p, c_uint], None)
+ffi('entry_set_uid', [c_archive_entry_p, c_longlong], None)
+ffi('entry_set_gid', [c_archive_entry_p, c_longlong], None)
 ffi('entry_set_perm', [c_archive_entry_p, c_int], None)
 ffi('entry_set_atime', [c_archive_entry_p, c_time_t, c_long], None)
 ffi('entry_set_mtime', [c_archive_entry_p, c_time_t, c_long], None)

libarchive/write.py

@@ -78,6 +78,7 @@ def add_file_from_memory(
         self, entry_path, entry_size, entry_data,
         filetype=REGULAR_FILE, permission=DEFAULT_UNIX_PERMISSION,
         atime=None, mtime=None, ctime=None, birthtime=None,
+        uid=None, gid=None,
     ):
         """"Add file from memory to archive.
 
@@ -99,6 +100,8 @@ def add_file_from_memory(
             birthtime (int | Tuple[int]):
                 the file's birth time (for archive formats that support it),
                 in seconds or as a tuple (seconds, nanoseconds)
+            uid (int): the file owner's identifier
+            gid (int): the file group's identifier
         """
         archive_pointer = self._pointer
 
@@ -117,6 +120,11 @@ def add_file_from_memory(
         entry_set_filetype(archive_entry_pointer, filetype)
         entry_set_perm(archive_entry_pointer, permission)
 
+        if uid is not None:
+            archive_entry.uid = uid
+        if gid is not None:
+            archive_entry.gid = gid
+
         if atime is not None:
             if not isinstance(atime, tuple):
                 atime = (atime, 0)

tests/test_rwx.py

@@ -161,7 +161,8 @@ def write_callback(data):
     with libarchive.custom_writer(write_callback, archfmt) as archive:
         archive.add_file_from_memory(
             entry_path, entry_size, entry_data,
-            atime=atime, mtime=mtime, ctime=ctime, birthtime=btime
+            atime=atime, mtime=mtime, ctime=ctime, birthtime=btime,
+            uid=1000, gid=1000,
         )
 
     buf = b''.join(blocks)
@@ -178,3 +179,5 @@ def write_callback(data):
                 assert archive_entry.birthtime in (
                     btime[0], format_time(*btime)
                 )
+            assert archive_entry.uid == 1000
+            assert archive_entry.gid == 1000