77 lines
2.8 KiB
Python
77 lines
2.8 KiB
Python
|
import hashlib
|
||
|
|
import tempfile
|
||
|
|
import unittest
|
||
|
|
from glob import glob
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from collectionparser.crack import crack
|
||
|
|
from collectionparser.parser import Processor
|
||
|
|
|
||
|
|
|
||
|
|
class CrackTest(unittest.TestCase):
|
||
|
|
def test_hash_saved_by_processor(self):
|
||
|
|
with tempfile.TemporaryDirectory() as tmp:
|
||
|
|
src = Path(tmp) / "in.txt"
|
||
|
|
src.write_text(
|
||
|
|
"[email protected]:5f4dcc3b5aa765d61d8327deb882cf99\n[email protected]:plainpw\n",
|
||
|
|
encoding="utf8",
|
||
|
|
)
|
||
|
|
processor = Processor(
|
||
|
|
str(Path(tmp) / "s.txt"),
|
||
|
|
str(Path(tmp) / "f.txt"),
|
||
|
|
str(Path(tmp) / "h.txt"),
|
||
|
|
)
|
||
|
|
processor.process(src)
|
||
|
|
hashes = "".join(
|
||
|
|
Path(p).read_text(encoding="utf8")
|
||
|
|
for p in glob(str(Path(tmp) / "h.txt.part.*"))
|
||
|
|
)
|
||
|
|
success = "".join(
|
||
|
|
Path(p).read_text(encoding="utf8")
|
||
|
|
for p in glob(str(Path(tmp) / "s.txt.part.*"))
|
||
|
|
)
|
||
|
|
self.assertEqual("[email protected]:5f4dcc3b5aa765d61d8327deb882cf99\n", hashes)
|
||
|
|
self.assertEqual("[email protected]:plainpw\n", success)
|
||
|
|
|
||
|
|
def test_crack_md5_and_sha1(self):
|
||
|
|
with tempfile.TemporaryDirectory() as tmp:
|
||
|
|
hashes = Path(tmp) / "hashes.txt"
|
||
|
|
hashes.write_bytes(
|
||
|
|
b"[email protected]:" + hashlib.md5(b"hunter2").hexdigest().encode() + b"\n"
|
||
|
|
b"[email protected]:" + hashlib.sha1(b"secret").hexdigest().encode() + b"\n"
|
||
|
|
b"[email protected]:not-a-hash\n"
|
||
|
|
)
|
||
|
|
wordlist = Path(tmp) / "words.txt"
|
||
|
|
wordlist.write_bytes(b"foo\nhunter2\nbar\nsecret\n")
|
||
|
|
out = Path(tmp) / "cracked.txt"
|
||
|
|
total = crack(str(hashes), str(out), wordlists=[str(wordlist)], workers=2)
|
||
|
|
self.assertEqual(2, total)
|
||
|
|
self.assertEqual(
|
||
|
|
{b"[email protected]:hunter2", b"[email protected]:secret"},
|
||
|
|
set(out.read_bytes().splitlines()),
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_crack_from_plaintext_passwords(self):
|
||
|
|
with tempfile.TemporaryDirectory() as tmp:
|
||
|
|
hashes = Path(tmp) / "hashes.txt"
|
||
|
|
hashes.write_bytes(
|
||
|
|
b"[email protected]:" + hashlib.md5(b"reused").hexdigest().encode() + b"\n"
|
||
|
|
)
|
||
|
|
empty = Path(tmp) / "empty.txt"
|
||
|
|
empty.write_bytes(b"")
|
||
|
|
plaintext = Path(tmp) / "plain.txt"
|
||
|
|
plaintext.write_bytes(b"[email protected]:reused\n")
|
||
|
|
out = Path(tmp) / "cracked.txt"
|
||
|
|
total = crack(
|
||
|
|
str(hashes),
|
||
|
|
str(out),
|
||
|
|
wordlists=[str(empty)],
|
||
|
|
passwords_from=[str(plaintext)],
|
||
|
|
workers=1,
|
||
|
|
)
|
||
|
|
self.assertEqual(1, total)
|
||
|
|
self.assertEqual([b"[email protected]:reused"], out.read_bytes().splitlines())
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|