Update components/passbeaker.py

components/passbeaker.py

@@ -1,3 +1,7 @@
+import hashlib
+import itertools
+from multiprocessing import Pool
+
 class PasswordCracker:
     def __init__(self, password_hash, wordlist_file, algorithm, salt=None, parallel=False, complexity_check=False):
         self.password_hash = password_hash
@@ -6,15 +10,84 @@ class PasswordCracker:
         self.salt = salt
         self.parallel = parallel
         self.complexity_check = complexity_check
+        self.total_passwords = 0
+        self.matched_password = None
+
+    def crack_hash(self, word):
+        if self.salt:
+            word_with_salt = f'{self.salt}{word}'
+        else:
+            word_with_salt = word
+        hashed_word = hashlib.new(self.algorithm, word_with_salt.encode()).hexdigest()
+        if hashed_word == self.password_hash:
+            self.matched_password = word
+            return True
+        return False
+
+    def generate_passwords(self, min_length, max_length, character_set):
+        passwords = []
+        for length in range(min_length, max_length + 1):
+            for combination in itertools.product(character_set, repeat=length):
+                password = ''.join(combination)
+                passwords.append(password)
+        return passwords
+
+    def evaluate_complexity(self, password):
+        has_lowercase = False
+        has_uppercase = False
+        has_digit = False
+        has_special = False
+
+        for char in password:
+            if char.islower():
+                has_lowercase = True
+            elif char.isupper():
+                has_uppercase = True
+            elif char.isdigit():
+                has_digit = True
+            else:
+                has_special = True
+
+        if len(password) >= 8 and has_lowercase and has_uppercase and has_digit and has_special:
+            return True
+        return False
+
+    def crack_password(self, password):
+        if self.complexity_check and not self.evaluate_complexity(password):
+            return
+        if self.matched_password is None:
+            if self.crack_hash(password):
+                return
+
+    def crack_passwords(self, passwords):
+        for password in passwords:
+            self.total_passwords += 1
+            if self.crack_hash(password):
+                break
+
+    def crack_passwords_parallel(self, passwords):
+        pool = Pool()
+        pool.map(self.crack_password, passwords)
+        pool.close()
 
     def crack_passwords_with_wordlist(self):
-        # ?????? ??? ??????? ?????? ?? ???????
-        pass
+        passwords = self.wordlist_file.splitlines()
+        if self.parallel:
+            self.crack_passwords_parallel(passwords)
+        else:
+            self.crack_passwords(passwords)
 
     def crack_passwords_with_brute_force(self, min_length, max_length, character_set):
-        # ?????? ??? ??????? ?????? ??????? ????????
-        pass
+        passwords = self.generate_passwords(min_length, max_length, character_set)
+        if self.parallel:
+            self.crack_passwords_parallel(passwords)
+        else:
+            self.crack_passwords(passwords)
 
     def get_statistics(self):
-        # ????? ??? ????????? ??????????
-        return "Statistics data"
+        result = f"Total Number of Passwords Tried: {self.total_passwords}\n"
+        if self.matched_password:
+            result += f"Password Cracked! Password: {self.matched_password}"
+        else:
+            result += "Password Failed."
+        return result