def modify_account(self, account, new_account, new_passwd, os_type,
basevm_type):
sub_desc = "new name: {0} new password: {1}".format(
new_account, new_passwd)
if new_account == "null":
sub_desc = "new password: {0}".format(new_passwd)
elif new_passwd == "null":
sub_desc = "new name: {0}".format(new_account)
desc = "Modify user account '{0}': {1}".format(account, sub_desc)
if basevm_type == 'kvm':
if os_type == "windows.7":
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{0} 'net user {1} {2} ' ".format(
self.addr, account, new_passwd)
else:
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{0} 'bash -s' < {1}{5}/users_managing/modify_user.sh {2} {3} {4}".format(
self.addr, self.getAbsPath(), account, new_account,
new_passwd, INSTANTIATION_DIR)
elif basevm_type == 'aws':
if os_type == "windows":
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{0} 'net user {1} {2} ' ".format(
self.addr, account, new_passwd)
elif os_type in ['amazon_linux', 'amazon_linux2', 'red_hat']:
command_string = "ssh -i TESTKEY.pem -o StrictHostKeyChecking=no ec2-user@{0} 'sudo -s' 'bash -s' < {1}{5}/users_managing/modify_user.sh {2} {3} {4}".format(
self.addr, self.getAbsPath(), account, new_account,
new_passwd, INSTANTIATION_DIR)
elif os_type in ['ubuntu_16', 'ubuntu_18', 'ubuntu_20']:
command_string = "ssh -i TESTKEY.pem -o StrictHostKeyChecking=no ubuntu@{0} 'sudo -s' 'bash -s' < {1}{5}/users_managing/modify_user.sh {2} {3} {4}".format(
self.addr, self.getAbsPath(), account, new_account,
new_passwd, INSTANTIATION_DIR)
command = Command(command_string, desc)
return command
def command(self):
desc = "Deploy dummy malware"
command_string = "{0}{5}/malware_creation/malware_launch.sh {1} {2} {3} {4} {0}".format(
self.getAbsPath(), self.addr, self.malware_name, self.mode,
self.crspd_option, INSTANTIATION_DIR)
command = Command(command_string, desc)
return command
def package_install_command(self, package_manager, tool_name, version):
if self.addr != "host":
if version == "":
desc = "Install package '{0}'".format(tool_name)
else:
desc = "Install package '{0}' version {1}".format(
tool_name, version)
if package_manager == "chocolatey":
if version == "":
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{1} {2} install -y {3}".format(
self.account, self.addr, package_manager, tool_name)
else:
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{1} {2} install -y {3} --version {4}".format(
self.account, self.addr, package_manager, tool_name,
version)
else:
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no {0}@{1} {2} install -y {3} {4}".format(
self.account, self.addr, package_manager, tool_name,
version)
command = Command(command_string, desc)
return command
else:
return "sudo {0} install -y {1} {2}".format(
package_manager, tool_name, version)
def command_post_clone(self, image_addr):
desc = "Execute program post-cloning '{0}'".format(self.program)
command_string = "python {0}{7}/content_copy_program_run/run_program.py {1} {2} {3} {4} {5} {6}".format(
self.getAbsPath(), self.program, self.interpreter, self.args,
image_addr, self.image_passwd, self.log_file, INSTANTIATION_DIR)
command = Command(command_string, desc)
return command
def command(self):
desc = "Generate ssh keys and do hostname setup"
command_string = "{0}{5}/sshkey_hostname_setup/sshkey_setup.sh {1} {2} {3}; {0}{5}/sshkey_hostname_setup/hostname_setup.sh {1} {2} {4};".format(
self.getAbsPath(), self.vm_addr, self.root_passwd,
self.mstnode_account, self.hostname, INSTANTIATION_DIR)
command = Command(command_string, desc)
return command
def command(self):
desc = "Modify firewall ruleset"
command_string = "{0}{4}/ruleset_modification/ruleset_modify.sh {0} {1} {2} {3}".format(
self.getAbsPath(), self.image_addr, self.image_passwd,
self.ruleset_file, INSTANTIATION_DIR)
command = Command(command_string, desc)
return command
def command(self):
desc = "Copy file '{0}'".format(self.src)
command_string = "{0}{5}/content_copy_program_run/copy_content.sh {1} {2} {3} {4}".format(
self.getAbsPath(), self.src, self.dst, self.image_addr,
self.image_passwd, INSTANTIATION_DIR)
command = Command(command_string, desc)
return command
def ddos_attack(self):
desc = "Generate traffic capture file containing DDoS attack trace"
command_string = "{0}{8}/logs_preparation/pcap_ddosattack_generator.sh {0} {1} {2} {3} {4} {5} {6} {7}".format(
self.getAbsPath(), self.virbr_addr, self.image_addr,
self.image_passwd, self.noise_level, self.file_path,
self.file_name, self.cr_dir, INSTANTIATION_DIR)
command = Command(command_string, desc)
return command
def command(self):
desc = "Execute program '{0}'".format(self.program)
command_string = "python {0}{7}/content_copy_program_run/run_program.py \"{1}\" {2} {3} {4} {5} {6} {8}".format(
self.getAbsPath(), self.program, self.interpreter, self.args,
self.image_addr, self.image_passwd, self.log_file,
INSTANTIATION_DIR, self.os_type)
command = Command(command_string, desc)
return command
def ssh_attack(self, target_account, attack_source, number):
desc = "Generate traffic capture file containing ssh attack trace"
command_string = "{0}{11}/logs_preparation/pcap_sshattack_generator.sh {0} {1} {2} {3} {4} {5} {6} {7} {8} {9} {10}".format(
self.getAbsPath(), self.virbr_addr, target_account,
self.image_addr, self.image_passwd, attack_source, number,
self.noise_level, self.file_path, self.file_name, self.cr_dir,
INSTANTIATION_DIR)
command = Command(command_string, desc)
return command
def command(self):
if self.attack_type == "ssh_attack":
desc = "Perform ssh attack on account '{0}' (repeat {1} times)".format(
self.target_account, self.number)
command_string = "{0}{5}/attacks_emulation/install_paramiko.sh; python {0}{5}/attacks_emulation/attack_paramiko_ssh.py {1} {2} {3} {4}".format(
self.getAbsPath(), self.target_addr, self.target_account,
self.number, self.attack_time, INSTANTIATION_DIR)
command = Command(command_string, desc)
return command
def command(self):
desc = "Generate ssh keys and do hostname setup"
if self.os_type=="windows.7":
command_string ="{0}{1}/sshkey_hostname_setup/sshkey_setup_win_cmd.sh {0} {1} {2} {3} {4};".format(self.getAbsPath(), INSTANTIATION_DIR, self.vm_addr, self.root_passwd, self.mstnode_account)
elif self.os_type in ["windows.8.1","windows.10"] :
command_string ="{0}{1}/sshkey_hostname_setup/sshkey_setup_win_unix.sh {0} {1} {2} {3} {4};".format(self.getAbsPath(), INSTANTIATION_DIR, self.vm_addr, self.root_passwd, self.mstnode_account)
else:
command_string = "{0}{5}/sshkey_hostname_setup/sshkey_setup.sh {1} {2} {3}; {0}{5}/sshkey_hostname_setup/hostname_setup.sh {1} {2} {4};".format(self.getAbsPath(), self.vm_addr, self.root_passwd, self.mstnode_account, self.hostname, INSTANTIATION_DIR)
command = Command(command_string, desc)
return command
def add_account(self, new_account, new_passwd, full_name):
desc = "Add user account '{0}'".format(new_account)
if full_name:
full_name_arg = full_name
else:
full_name_arg = ""
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{0} 'bash -s' < {1}{5}/users_managing/add_user.sh {2} {3} yes {4}".format(
self.addr, self.getAbsPath(), new_account, new_passwd,
full_name_arg, INSTANTIATION_DIR)
command = Command(command_string, desc)
return command
def command(self):
desc = "Copy file '{0}'".format(self.src)
if (self.os_type == "windows.7"):
command_string = "{0}{5}/content_copy_program_run/copy_content_win.sh {1} \" {2} \" {3} {4}".format(
self.getAbsPath(), self.src, self.dst, self.image_addr,
self.image_passwd, INSTANTIATION_DIR)
else:
command_string = "{0}{4}/content_copy_program_run/copy_content.sh {1} {2} {3} {5} {6}".format(
self.getAbsPath(), self.src, self.dst, self.image_addr,
INSTANTIATION_DIR, self.basevm_type, self.os_type)
command = Command(command_string, desc)
return command
def modify_account(self, account, new_account, new_passwd, os_type):
sub_desc = "new name: {0} new password: {1}".format(new_account, new_passwd)
if new_account == "null":
sub_desc = "new password: {0}".format(new_passwd)
elif new_passwd == "null":
sub_desc = "new name: {0}".format(new_account)
desc = "Modify user account '{0}': {1}".format(account, sub_desc)
if os_type =="windows.7":
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{0} 'net user {1} {2} ' ".format(self.addr, account, new_passwd)
else:
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{0} 'bash -s' < {1}{5}/users_managing/modify_user.sh {2} {3} {4}".format(self.addr, self.getAbsPath(), account, new_account, new_passwd, INSTANTIATION_DIR)
command = Command(command_string, desc)
return command
def add_account(self, new_account, new_passwd, full_name, os_type):
desc = "Add user account '{0}'".format(new_account)
if full_name:
full_name_arg=full_name
else:
full_name_arg=""
if os_type=="windows.7" :
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{0} 'net user {2} {3} /ADD' ;".format(self.addr, self.getAbsPath(), new_account, new_passwd)
command_string += "sshpass -p {0} ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no {1}@{2} 'dir' ;".format(new_passwd, new_account, self.addr)
command_string += "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{0} 'net localgroup \"Remote Desktop Users\" {2} /ADD'".format(self.addr, self.getAbsPath(), new_account)
else:
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{0} 'bash -s' < {1}{5}/users_managing/add_user.sh {2} {3} yes {4}".format(self.addr, self.getAbsPath(), new_account, new_passwd, full_name_arg, INSTANTIATION_DIR)
command = Command(command_string, desc)
return command
def package_install_command(self, package_manager, tool_name, version,
os_type, basevm_type):
if self.addr != "host":
if version == "":
desc = "Install package '{0}'".format(tool_name)
else:
desc = "Install package '{0}' version {1}".format(
tool_name, version)
if basevm_type == 'kvm':
# Handle Windows package manager
if package_manager == "chocolatey":
if version == "":
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{1} {2} install -y {3}".format(
self.account, self.addr, package_manager,
tool_name)
else:
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{1} {2} install -y {3} --version {4}".format(
self.account, self.addr, package_manager,
tool_name, version)
# Handle other OS package managers
else:
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no {0}@{1} {2} install -y {3} {4}".format(
self.account, self.addr, package_manager, tool_name,
version)
elif basevm_type == 'aws':
# Handle RedHat-like package manager
if os_type in ['amazon_linux', 'amazon_linux2', 'red_hat']:
command_string = "ssh -i TESTKEY.pem -o StrictHostKeyChecking=no ec2-user@{1} 'sudo -s' '{2} install -y {3} {4}'".format(
self.account, self.addr, package_manager, tool_name,
version)
# Handle Ubuntu package manager
elif os_type in ['ubuntu_16', 'ubuntu_18', 'ubuntu_20']:
command_string = "ssh -i TESTKEY.pem -o StrictHostKeyChecking=no ubuntu@{1} 'sudo apt-get update; sudo {2} install -y {3} {4}'".format(
self.account, self.addr, package_manager, tool_name,
version)
command = Command(command_string, desc)
return command
else:
return "sudo {0} install -y {1} {2}".format(
package_manager, tool_name, version)
def add_account(self, new_account, new_passwd, full_name, os_type,
basevm_type):
desc = "Add user account '{0}'".format(new_account)
if full_name:
full_name_arg = full_name
else:
full_name_arg = ""
if basevm_type == 'kvm':
if os_type == "windows.7":
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{0} 'net user {2} {3} /ADD' ;".format(
self.addr, self.getAbsPath(), new_account, new_passwd)
command_string += "sshpass -p {0} ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no {1}@{2} 'dir' ;".format(
new_passwd, new_account, self.addr)
command_string += "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{0} 'net localgroup \"Remote Desktop Users\" {2} /ADD'".format(
self.addr, self.getAbsPath(), new_account)
else:
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{0} 'bash -s' < {1}{5}/users_managing/add_user.sh {2} {3} yes {4}".format(
self.addr, self.getAbsPath(), new_account, new_passwd,
full_name_arg, INSTANTIATION_DIR)
elif basevm_type == 'aws':
if os_type == "windows":
command_string = "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{0} 'net user {2} {3} /ADD' ;".format(
self.addr, self.getAbsPath(), new_account, new_passwd)
command_string += "sshpass -p {0} ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no {1}@{2} 'dir' ;".format(
new_passwd, new_account, self.addr)
command_string += "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@{0} 'net localgroup \"Remote Desktop Users\" {2} /ADD'".format(
self.addr, self.getAbsPath(), new_account)
elif os_type in ['amazon_linux', 'amazon_linux2', 'red_hat']:
command_string = "ssh -i TESTKEY.pem -o StrictHostKeyChecking=no ec2-user@{0} 'sudo -s' 'bash -s' < {1}{5}/users_managing/add_user.sh {2} {3} yes {4}".format(
self.addr, self.getAbsPath(), new_account, new_passwd,
full_name_arg, INSTANTIATION_DIR)
elif os_type in ['ubuntu_16', 'ubuntu_18', 'ubuntu_20']:
command_string = "ssh -i TESTKEY.pem -o StrictHostKeyChecking=no ubuntu@{0} 'sudo -s' 'bash -s' < {1}{5}/users_managing/add_user.sh {2} {3} yes {4}".format(
self.addr, self.getAbsPath(), new_account, new_passwd,
full_name_arg, INSTANTIATION_DIR)
command = Command(command_string, desc)
return command