DictionaryAttackExample/fun_dictionary_attack.py at main · DataBach-maker/DictionaryAttackExample · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
## libraries
import time
import hashlib
import pandas as pd

import random

class dictionary_attack:
    def __init__(self, path=str):

        self.path = path
        self.df = self.readCSV(self.path)

        self.df = self.identify_weak_passwords(self.df)
        self.df = self.generate_additional_passwords(self.df)

        self.df = self.create_hash_table(self.df)

        self.success = "The password was found: "
        self.fail = "The password could not be cracked "

    def readCSV(self, path):

        df = pd.read_csv(
            path,
            delimiter="\n",
            header=None,
            names=["Passwords"],
            encoding="ISO-8859-1",
        )

        return df

    def identify_weak_passwords(self, df):

        strong_password_regex = '(?=.*\d)(?=.*[!@#$%^&*]+)(?![.\n])(?=.*[A-Z])(?=.*[a-z]).*$'
        df["is_Weak_Password"] = ~df.Passwords.str.contains(strong_password_regex, na=False)

        return df

    def generate_additional_passwords(self, df):

        expression_one = df.loc[df['is_Weak_Password'] == True].Passwords + "!"
        expression_two = df.loc[df['is_Weak_Password'] == True].Passwords + str(random.randint(1940, 2022))

        expressions = [expression_one, expression_two]

        for expression in expressions:
            result_df = pd.concat((df["Passwords"], expression), axis=0)

        return result_df


    def create_hash_table(self, path):

        def sha_encode(password):
            return hashlib.sha256(password.encode()).hexdigest()

        df["Hash"] = df["Passwords"].astype(str).apply(sha_encode)
        return df


    def crack_password(self, password):

        password_hash = hashlib.sha256(password.encode()).hexdigest()
        solution = self.df.loc[self.df["Hash"] == password_hash]

        if solution.empty:
            return self.fail
        else:
            return self.success + password


if __name__ == "__main__":
    difficult_password = "difficultpassword##!!?*?235465367453"
    easy_password = "123456"

    file_path = r"C:\Users\Nutzer\Desktop\Projects\DictionaryAttack\rockyou.txt"

    attack = DictionaryAttack(file_path)

    start_time_difficult = time.time()
    attempt_result_difficult = attack.crack_password(difficult_password)
    end_time_difficult = time.time()
    total_time_elapsed_difficult = end_time_difficult - start_time_difficult

    print(
        attempt_result_difficult,
        ", The code ran: ",
        "{0:.10f}".format(total_time_elapsed_difficult),
    )

    start_time_easy = time.time()
    attempt_result_easy = attack.crack_password(easy_password)
    end_time_easy = time.time()
    total_time_elapsed_easy = end_time_easy - start_time_easy

    print(
        attempt_result_easy,
        ", The code ran: ",
        "{0:.10f}".format(total_time_elapsed_easy),
    )