Find private keys within gzip-compresssed files

Author: jwreschnig-utiq

pre_commit_hooks/detect_private_key.py

@@ -1,6 +1,8 @@
 from __future__ import annotations
 
 import argparse
+import gzip
+
 from collections.abc import Sequence
 
 BLACKLIST = [
@@ -29,6 +31,17 @@ def main(argv: Sequence[str] | None = None) -> int:
             content = f.read()
             if any(line in content for line in BLACKLIST):
                 private_key_files.append(filename)
+                continue
+        try:
+            if filename.endswith(".gz"):
+                with gzip.open(filename, 'rb') as f:
+                    content = f.read()
+                    if any(line in content for line in BLACKLIST):
+                        private_key_files.append(filename)
+                        continue
+        except gzip.BadGzipFile:
+            pass
+
 
     if private_key_files:
         for private_key_file in private_key_files:

tests/detect_private_key_test.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import gzip
 import pytest
 
 from pre_commit_hooks.detect_private_key import main
@@ -26,3 +27,9 @@ def test_main(input_s, expected_retval, tmpdir):
     path = tmpdir.join('file.txt')
     path.write_binary(input_s)
     assert main([str(path)]) == expected_retval
+
+@pytest.mark.parametrize(('input_s', 'expected_retval'), TESTS)
+def test_main_gzip(input_s, expected_retval, tmpdir):
+    path = tmpdir.join('file.txt.gz')
+    path.write_binary(gzip.compress(input_s))
+    assert main([str(path)]) == expected_retval