mirror of https://github.com/avast/PurpleDome
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
3.2 KiB
Python
96 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
""" Demo program to set up and control the machines """
|
|
|
|
import argparse
|
|
|
|
import yaml
|
|
|
|
from app.calderacontrol import CalderaControl
|
|
from app.machinecontrol import Machine
|
|
from app.attack_log import AttackLog
|
|
|
|
|
|
def create_machines(arguments):
|
|
"""
|
|
|
|
@param arguments: The arguments from argparse
|
|
"""
|
|
# TODO: Add argparse and make it flexible
|
|
|
|
with open(arguments.configfile) as fh:
|
|
config = yaml.safe_load(fh)
|
|
|
|
attack_logger = AttackLog(arguments.verbose)
|
|
target_ = Machine(config["targets"]["target1"], attack_logger)
|
|
attacker_1 = Machine(config["attackers"]["attacker"])
|
|
|
|
print("Got them")
|
|
|
|
# TODO Automatically create all machines defined in config file
|
|
|
|
# attacker_1.destroy()
|
|
print("destroyed")
|
|
attacker_1.create(reboot=False)
|
|
print("Attacker up")
|
|
attacker_1.up()
|
|
print(attacker_1.install_caldera_server())
|
|
attacker_1.start_caldera_server()
|
|
print("Attacker done")
|
|
|
|
target_.destroy()
|
|
target_.set_caldera_server(attacker_1.get_ip())
|
|
target_.install_caldera_service()
|
|
target_.create()
|
|
print("Target up")
|
|
target_.up()
|
|
target_.start_caldera_client()
|
|
print("Target done")
|
|
|
|
print("Caldera server running at: http://{}:8888/".format(attacker_1.get_ip()))
|
|
# target_.install_caldera_client(attacker_1.getip(), "target1elf")
|
|
|
|
|
|
def download_caldera_client(arguments):
|
|
""" Downloads the caldera client
|
|
|
|
@param arguments: The arguments from argparse
|
|
"""
|
|
|
|
attack_logger = AttackLog(arguments.verbose)
|
|
caldera_control = CalderaControl(args.ip, attack_logger, None)
|
|
caldera_control.fetch_client(platform=arguments.platform,
|
|
file=arguments.file,
|
|
target_dir=arguments.target_dir,
|
|
extension=".go")
|
|
|
|
|
|
def create_parser():
|
|
""" Creates the parser for the command line arguments"""
|
|
|
|
main_parser = argparse.ArgumentParser("Controls a Caldera server to attack other systems")
|
|
main_parser.add_argument('--verbose', '-v', action='count', default=0)
|
|
subparsers = main_parser.add_subparsers(help="sub-commands")
|
|
|
|
# Sub parser for machine creation
|
|
parser_create = subparsers.add_parser("create", help="create systems")
|
|
parser_create.set_defaults(func=create_machines)
|
|
parser_create.add_argument("--configfile", default="experiment.yaml", help="Config file to create from")
|
|
|
|
parser_download_caldera_client = subparsers.add_parser("fetch_client", help="download the caldera client")
|
|
parser_download_caldera_client.set_defaults(func=download_caldera_client)
|
|
parser_download_caldera_client.add_argument("--ip", default="192.168.178.189", help="Ip of Caldera to connect to")
|
|
parser_download_caldera_client.add_argument("--platform", default="windows", help="platform to download the client for")
|
|
parser_download_caldera_client.add_argument("--file", default="sandcat.go", help="The agent to download")
|
|
parser_download_caldera_client.add_argument("--target_dir", default=".", help="The target dir to download the file to")
|
|
|
|
return main_parser
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
parser = create_parser()
|
|
|
|
args = parser.parse_args()
|
|
|
|
args.func(args)
|