Exemplo n.º 1
0
 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
Exemplo n.º 2
0
 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
Exemplo n.º 3
0
    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)
Exemplo n.º 4
0
 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
Exemplo n.º 5
0
 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
Exemplo n.º 6
0
 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
Exemplo n.º 7
0
 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
Exemplo n.º 8
0
 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
Exemplo n.º 9
0
 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
Exemplo n.º 10
0
 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
Exemplo n.º 11
0
 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
Exemplo n.º 12
0
 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
Exemplo n.º 13
0
    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
Exemplo n.º 14
0
 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
Exemplo n.º 15
0
 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
Exemplo n.º 16
0
    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
Exemplo n.º 17
0
    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)
Exemplo n.º 18
0
    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