HellFire/HellFire.py at master · StevenD33/HellFire · 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
# Import the necessary packages
# from consolemenu import *
# from consolemenu.items import *
from Audit import audit
from Basic_Hardening import basic_hardening
# Create the menu
# menu = ConsoleMenu("Hellfire")

# # Create some items
# # A CommandItem runs a console command

# audit_item = CommandItem("Run Audit on your system",  "python Audit.py",)
# basic_hardening_item = CommandItem("Run Basic Hardening on your system",  "python Basic_Hardening.py")
# advanced_hardening_item = CommandItem("Run Advanced Hardening on your system",  "python Advanced_Hardening.py")
# personalised_hardening_item = CommandItem("Run Personalised Hardening on your system",  "python Personalised_Hardening.py")

# function_item = FunctionItem("Call a Python function",audit)

# # A SelectionMenu constructs a menu from a list of strings
# selection_menu = SelectionMenu(["item1", "item2", "item3"])

# # A SubmenuItem lets you add a menu (the selection_menu above, for example)
# # as a submenu of another menu
# submenu_item = SubmenuItem("Submenu item", selection_menu, menu)

# # Once we're done creating them, we just add the items to the menu
# menu.append_item(audit_item)
# menu.append_item(basic_hardening_item)
# menu.append_item(advanced_hardening_item)
# menu.append_item(personalised_hardening_item)
# menu.append_item(submenu_item)
# menu.append_item(function_item)

# menu.show()


import sys, os

# Main definition - constants
menu_actions  = {}

# Main menu
def main_menu():
    os.system('clear')
    print(" |\     /|(  ____ \( \      ( \      (  ____ \\__   __/(  ____ )(  ____ \ ")
    print(" | )   ( || (    \/| (      | (      | (    \/   ) (   | (    )|| (    \/ ")
    print(" | (___) || (__    | |      | |      | (__       | |   | (____)|| (__     ")
    print(" |  ___  ||  __)   | |      | |      |  __)      | |   |     __)|  __)    ")
    print(" | (   ) || (      | |      | |      | (         | |   | (\ (   | (       ")
    print(" | )   ( || (____/\| (____/\| (____/\| )      ___) (___| ) \ \__| (____/\ ")
    print(" |/     \|(_______/(_______/(_______/|/       \_______/|/   \__/(_______/ ")

    print ("1. Run Audit on your system")
    print ("2. Run Basic Hardening on your system")
    print ("3. Run Advanced Hardening on your system")
    print ("4. Run Personalised Hardening on your system")
    print ("\n0. Quit")
    choice = input(" >>  ")
    if choice == "1":
        audit()

    if choice == "2":
        basic_hardening()
    else:
        print("Invalid selection, please try again.\n")


# Main Program
if __name__ == "__main__":
    # Launch main menu
    main_menu()