gh-75707: tarfile: Add optional open() argument "reprodicuble"

This makes it possible to create reproducible .tar.gz files without
overriding time.time(), by setting the gzip header field mtime to 0.

Author: ampleyfly

Lib/tarfile.py

@@ -339,7 +339,7 @@ class _Stream:
     """
 
     def __init__(self, name, mode, comptype, fileobj, bufsize,
-                 compresslevel, preset):
+                 compresslevel, preset, reproducible):
         """Construct a _Stream object.
         """
         self._extfileobj = True
@@ -374,7 +374,7 @@ def __init__(self, name, mode, comptype, fileobj, bufsize,
                     self.exception = zlib.error
                     self._init_read_gz()
                 else:
-                    self._init_write_gz(compresslevel)
+                    self._init_write_gz(compresslevel, reproducible)
 
             elif comptype == "bz2":
                 try:
@@ -423,15 +423,19 @@ def __del__(self):
         if hasattr(self, "closed") and not self.closed:
             self.close()
 
-    def _init_write_gz(self, compresslevel):
+    def _init_write_gz(self, compresslevel, reproducible):
         """Initialize for writing with gzip compression.
         """
         self.cmp = self.zlib.compressobj(compresslevel,
                                          self.zlib.DEFLATED,
                                          -self.zlib.MAX_WBITS,
                                          self.zlib.DEF_MEM_LEVEL,
                                          0)
-        timestamp = struct.pack("<L", int(time.time()))
+        if reproducible:
+            timestamp = 0
+        else:
+            timestamp = int(time.time())
+        timestamp = struct.pack("<L", timestamp)
         self.__write(b"\037\213\010\010" + timestamp + b"\002\377")
         if self.name.endswith(".gz"):
             self.name = self.name[:-3]
@@ -1726,7 +1730,7 @@ class TarFile(object):
     def __init__(self, name=None, mode="r", fileobj=None, format=None,
             tarinfo=None, dereference=None, ignore_zeros=None, encoding=None,
             errors="surrogateescape", pax_headers=None, debug=None,
-            errorlevel=None, copybufsize=None, stream=False):
+            errorlevel=None, copybufsize=None, stream=False, reproducible=False):
         """Open an (uncompressed) tar archive 'name'. 'mode' is either 'r' to
            read from an existing archive, 'a' to append data to an existing
            file or 'w' to create a new file overwriting an existing one. 'mode'
@@ -1932,8 +1936,9 @@ def not_compressed(comptype):
 
             compresslevel = kwargs.pop("compresslevel", 6)
             preset = kwargs.pop("preset", None)
+            reproducible = kwargs.pop("reproducible", False)
             stream = _Stream(name, filemode, comptype, fileobj, bufsize,
-                             compresslevel, preset)
+                             compresslevel, preset, reproducible)
             try:
                 t = cls(name, filemode, stream, **kwargs)
             except:

Lib/test/test_tarfile.py

@@ -2115,6 +2115,11 @@ def test_create_with_compresslevel(self):
         with tarfile.open(tmpname, 'r:gz', compresslevel=1) as tobj:
             pass
 
+    def test_create_reproducible(self):
+        with tarfile.open(tmpname, self.mode, compresslevel=1, reproducible=True) as tobj:
+            tobj.add(self.file_path)
+        with self.open(tmpname, 'r') as fobj:
+            self.assertEqual(fobj.mtime, None)
 
 class Bz2CreateTest(Bz2Test, CreateTest):