pre-commit/pre-commit-hooks
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit-hooks.git synced 2026-04-09 21:04:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Allow arbitrary binary to pass the private key hook check. Resolves #64

Browse source
This commit is contained in:
Anthony Sottile 2015-06-11 13:28:10 -07:00
parent 6077f2fefb
commit 07c938e1b1
2 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

12
pre_commit_hooks/detect_private_key.py
View file

@ -1,7 +1,6 @@
from __future__ import print_function from __future__ import print_function
import argparse import argparse
import io
import sys import sys
@ -13,12 +12,11 @@ def detect_private_key(argv=None):
private_key_files = [] private_key_files = []
for filename in args.filenames: for filename in args.filenames:
with io.open(filename, 'r') as f: content = open(filename, 'rb').read()
content = f.read() if b'BEGIN RSA PRIVATE KEY' in content:
if 'BEGIN RSA PRIVATE KEY' in content: private_key_files.append(content)
private_key_files.append(content) if b'BEGIN DSA PRIVATE KEY' in content:
if 'BEGIN DSA PRIVATE KEY' in content: private_key_files.append(content)
private_key_files.append(content)
if private_key_files: if private_key_files:
for private_key_file in private_key_files: for private_key_file in private_key_files:

2
tests/detect_private_key_test.py
View file

@ -10,6 +10,8 @@ TESTS = (
(b'-----BEGIN DSA PRIVATE KEY-----', 1), (b'-----BEGIN DSA PRIVATE KEY-----', 1),
(b'ssh-rsa DATA', 0), (b'ssh-rsa DATA', 0),
(b'ssh-dsa DATA', 0), (b'ssh-dsa DATA', 0),
# Some arbitrary binary data
(b'\xa2\xf1\x93\x12', 0),
) )