Files
Hunt-AI/dco.py
2024-11-24 10:53:02 -05:00

20 lines
759 B
Python

from Modules.Imports.all_imports import *
from Protocols.menu import MENU_OPTIONS as BASE_MENU_OPTIONS
if __name__ == "__main__":
box_name = get_box_name()
target_ip = get_target_ip() # Ask for the IP before anything else
if target_ip: # Ensure the IP is valid before proceeding
OPEN_PORTS = scan_ports(target_ip, box_name) # Pass the valid IP and capture open ports
# Update MENU_OPTIONS to pass necessary arguments
MENU_OPTIONS = {
key: {
**value,
"submenu": value["submenu"], # Pass the actual function
}
for key, value in BASE_MENU_OPTIONS.items()
}
display_menu(MENU_OPTIONS, OPEN_PORTS, target_ip) # Display the dynamic menu