pentbox-python/main.py at main · butterjack/pentbox-python · 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
import pyinputplus as pyip
import pyfiglet

from SymmetricEncryption import SymmetricEncryption
from AsymmetricEncryption import AsymmetricEncryption
from Encoding import Encoding
from Hashing import Hashing

import click

def menu():
    while(True):
        choice = pyip.inputMenu(['encoding','hashing','pwdCracker','symmetric-encrypt','asymmetric-encrypt','quit'])
        if(choice=='encoding'):
            Encoding.menu()
        elif(choice=='hashing'):
            Hashing.hash_menu()
        elif(choice=='pwdCracker'):
            Hashing.crack_menu()
        elif(choice=='symmetric-encrypt'):
            SymmetricEncryption.menu()
        elif(choice=='asymmetric-encrypt'):
            AsymmetricEncryption.menu()

        elif(choice=='quit'):
            return


@click.command()
@click.option('-m','--mode','mode', default='nothing' , help='choose the function to try')
def main(mode):
    """
        -----------------------------------------------------------------------
        --------------------------Security Project-----------------------------
        -----------------------------------------------------------------------
        The first menu does the following : encode and decode a given text to
          various types of encoding utf8, ascii, base16, base32, base64
        -----------------------------------------------------------------------
        The second menu does the following : hash a given text according to
        various types of hashing MD5, SHA1, SHA256, SHA384, SHA512, SHA3_256,
        SHA3_384, SHAKE_128, SHA224, BLAKE2B, BLAKE2S, SHAKE_256, SHA3_512,
        SHA3_224
        -----------------------------------------------------------------------
        The third menu does the following: crack a given hashed password using
        all the hashing functions mentioned above using a dictionary of 63175
        words under the name of pentbox-wlist.txt
        -----------------------------------------------------------------------
        The fourth menu does the following: encrypt a given message with an AES
        (Advanced Encryption Standard) or Salsa20 encryption which are symmetric
        encryption
        -----------------------------------------------------------------------
        The last menu does the following: encrypt a given message with an RSA or
        Elgamal encryption which are asymmetric encryption
        -----------------------------------------------------------------------

    """
    if(mode=='symmetric'):
        SymmetricEncryption.menu()
    if(mode=='encoding'):
        Encoding.menu()
    if(mode=='hashing'):
        Hashing.hash_menu()
    if(mode=='crack'):
        Hashing.crack_menu()
    if(mode=='asymmetric'):
        AsymmetricEncryption.menu()


    if(mode=='nothing'):
        menu()


if __name__ == '__main__':
    main()