pre-commit/pre-commit-hooks
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit-hooks.git synced 2026-03-29 18:16:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

detect_private_key: print *filenames*, not the key

Browse source
This commit is contained in:
Chris Kuehl 2015-08-01 16:59:21 -07:00
parent 516cc9fa72
commit ca2be345a1
2 changed files with 11 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

15
pre_commit_hooks/detect_private_key.py
View file

@ -3,6 +3,12 @@ from __future__ import print_function
import argparse
import sys
BLACKLIST = [
b'BEGIN RSA PRIVATE KEY',
b'BEGIN DSA PRIVATE KEY',
b'BEGIN EC PRIVATE KEY',
]
def detect_private_key(argv=None):
parser = argparse.ArgumentParser()
@ -12,11 +18,10 @@ def detect_private_key(argv=None):
private_key_files = []
for filename in args.filenames:
content = open(filename, 'rb').read()
if b'BEGIN RSA PRIVATE KEY' in content:
private_key_files.append(content)
if b'BEGIN DSA PRIVATE KEY' in content:
private_key_files.append(content)
with open(filename, 'rb') as f:
content = f.read()
if any(line in content for line in BLACKLIST):
private_key_files.append(filename)
if private_key_files:
for private_key_file in private_key_files:

1
tests/detect_private_key_test.py
View file

@ -8,6 +8,7 @@ from pre_commit_hooks.detect_private_key import detect_private_key
TESTS = (
(b'-----BEGIN RSA PRIVATE KEY-----', 1),
(b'-----BEGIN DSA PRIVATE KEY-----', 1),
(b'-----BEGIN EC PRIVATE KEY-----', 1),
(b'ssh-rsa DATA', 0),
(b'ssh-dsa DATA', 0),
# Some arbitrary binary data