Merge pull request #67 from ActiveState/BE-4449-cve-2024-6232

Add tests to show that CVE-2024-6232 is okay

Author: icanhasmath

Lib/test/test_tarfile.py

@@ -785,6 +785,43 @@ def test_pax_number_fields(self):
         finally:
             tar.close()
 
+    def test_pax_header_bad_formats(self):
+        # The fields from the pax header have priority over the
+        # TarInfo.
+        pax_header_replacements = (
+            b" foo=bar\n",
+            b"0 \n",
+            b"1 \n",
+            b"2 \n",
+            b"3 =\n",
+            b"4 =a\n",
+            b"1000000 foo=bar\n",
+            b"0 foo=bar\n",
+            b"-12 foo=bar\n",
+            b"000000000000000000000000036 foo=bar\n",
+        )
+        pax_headers = {"foo": "bar"}
+        for replacement in pax_header_replacements:
+            tar = tarfile.open(tmpname, "w", format=tarfile.PAX_FORMAT,
+                                encoding="iso8859-1")
+            try:
+                t = tarfile.TarInfo()
+                t.name = "pax"  # non-ASCII
+                t.uid = 1
+                t.pax_headers = pax_headers
+                tar.addfile(t)
+            finally:
+                tar.close()
+            with open(tmpname, "rb") as f:
+                data = f.read()
+                self.assertIn(b"11 foo=bar\n", data)
+                data = data.replace(b"11 foo=bar\n", replacement)
+            with open(tmpname, "wb") as f:
+                f.truncate()
+                f.write(data)
+            with self.assertRaisesRegexp(tarfile.ReadError, r"file could not be opened successfully"):
+                tarfile.open(tmpname, encoding="iso8859-1")
+
 
 class WriteTestBase(unittest.TestCase):
     # Put all write tests in here that are supposed to be tested