class SSHMonitor(BaseMonitor):
'''
SSHMonitor monitors target ip
and runs a command over SSH in case it is not responding.
'''
def __init__(self, name, username, password, hostname, port,
status_command, restart_command=None, logger=None):
'''
:param name: name of the object
:param username: ssh login username
:param password: ssh login password
:param hostname: ssh server ip
:param port: ssh server port
:param status_command: command to make sure target is alive
:param restart_command: command to restart the target in case it is deadore
:param logger: logger for this object (default: None)
'''
super(SSHMonitor, self).__init__(name, logger)
self._status_command = status_command
self._restart_command = restart_command
self._ssh = ReconnectingSSHConnection(hostname, port, username, password)
def teardown(self):
self._ssh.close()
super(SSHMonitor, self).teardown()
def _ssh_command(self, cmd):
return_code = stdout = stderr = None
try:
return_code, stdout, stderr = self._ssh.exec_command(cmd)
except KeyboardInterrupt:
raise
except Exception as e:
self.logger.debug('SSHMonitor: ssh command exec error: %s' % str(e))
return return_code, stdout, stderr
def post_test(self):
return_code, stdout, stderr = self._ssh_command(self._status_command)
if return_code != 0:
self.report.add('status_command', self._status_command)
self.report.add('status_command return code', return_code)
self.report.add('status_command stdout', stdout)
self.report.add('status_command stderr', stderr)
self.report.failed("got non-zero return code")
if self._restart_command:
self.logger.info('target not responding - restarting target !!!')
return_code, stdout, stderr = self._ssh_command(self._restart_command)
self.report.add('restart_command', self._restart_command)
self.report.add('restart_command return code', return_code)
self.report.add('restart_command stdout', stdout)
self.report.add('restart_command stderr', stderr)
super(SSHMonitor, self).post_test()
def pre_test(self, test_number):
super(SSHMonitor, self).pre_test(test_number)
while self._ssh_command(self._status_command)[0] != 0:
self.logger.debug('waiting for target to be up')
time.sleep(1)
def _monitor_func(self):
'''
Nothing is done here, so we use a sleep for now.
'''
time.sleep(0.1)
class SshFileMonitor(BaseMonitor):
'''
SshFileMonitor monitors for files using a file_mask.
If found - moves files to local folder, renaming with test number.
'''
def __init__(self,
name,
username,
password,
hostname,
port,
file_mask,
local_dir,
use_scp=False,
fail_if_exists=True,
setup_commands=[],
on_fail_command=None,
on_fail_delay=0,
logger=None):
'''
:param name: name of the object
:param username: ssh login username
:param password: ssh login password
:param hostname: ssh server ip
:param port: ssh server port
:param file_mask: file_mask to fetch
:param local_dir: local_path to store fetched files
:param use_scp: use the SCP protocol for transferring files instead of SFTP
:param fail_if_exists: fail test if file exists (default: True)
:param on_fail_command: command to run on failure (default: None)
:param on_fail_delay: time to sleep after running on_fail_command (default: 0)
:param logger: logger for this object (default: None)
'''
super(SshFileMonitor, self).__init__(name, logger)
self._file_mask = file_mask
self._local_dir = local_dir
self._fail_if_exists = fail_if_exists
self._setup_commands = setup_commands
self._on_fail_command = on_fail_command
self._on_fail_delay = on_fail_delay
self._ssh = ReconnectingSSHConnection(hostname, port, username,
password, use_scp)
def _ssh_command(self, cmd):
return_code = None
try:
return_code, self._stdout, self._stderr = self._ssh.exec_command(
cmd)
except KeyboardInterrupt:
raise
except Exception as e:
self.logger.debug('SSHFileMonitor: ssh command exec error: %s' %
str(e))
return return_code
def setup(self):
'''
Called at the begining of the fuzzing session
'''
super(SshFileMonitor, self).setup()
timestamp = datetime.datetime.now().strftime("%Y_%m_%d-%H_%M_%S")
self._local_dir = os.path.join(self._local_dir, timestamp)
os.makedirs(self._local_dir)
for command in self._setup_commands:
self.logger.debug('running remote setup command: %s' % command)
res = self._ssh_command(command)
if res != 0:
self.logger.debug('Error running command: %s. got %s' %
(command, res))
self.logger.debug('stdout: %s' % self._stdout)
self.logger.debug('stderr: %s' % self._stderr)
def teardown(self):
if self._ssh:
self._ssh.close()
self._ssh = None
super(SshFileMonitor, self).teardown()
def post_test(self):
cmd = 'ls %s' % self._file_mask
return_code = self._ssh_command(cmd)
self.report.add('file_mask', self._file_mask)
if return_code == 0:
ls_stdout = self._stdout
ls_stderr = self._stderr
del ls_stderr
self.report.failed('found core file')
self.report.add('remote file', ls_stdout.strip())
remote_path = ls_stdout.strip()
local_path = os.path.join(self._local_dir,
'test_%05d' % self.test_number)
self._ssh.get(str(remote_path), str(local_path))
self._ssh.remove(str(remote_path))
self.report.add('local file', local_path)
if self._on_fail_command:
self.logger.info('running remote on fail command: %s',
self._on_fail_command)
self.report.add('on_fail_command', self._on_fail_command)
self.report.add('on_fail_delay', self._on_fail_delay)
retval = self._ssh_command(self._on_fail_command)
self.logger.info('on fail command returned %s' % retval)
if retval != 0:
self.logger.info('on fail command stdout %s' %
self._stdout)
self.logger.info('on fail command stderr %s' %
self._stderr)
else:
self.logger.info('sleeping for %s seconds' %
self._on_fail_delay)
time.sleep(self._on_fail_delay)
super(SshFileMonitor, self).post_test()
def X_pre_test(self, test_number):
super(SshFileMonitor, self).pre_test(test_number)
def _monitor_func(self):
'''
Nothing is done here, so we use a sleep for now.
'''
time.sleep(0.1)
class SSHMonitor(BaseMonitor):
'''
SSHMonitor monitors target ip
and runs a command over SSH in case it is not responding.
'''
def __init__(self,
name,
username,
password,
hostname,
port,
status_command,
restart_command=None,
logger=None):
'''
:param name: name of the object
:param username: ssh login username
:param password: ssh login password
:param hostname: ssh server ip
:param port: ssh server port
:param status_command: command to make sure target is alive
:param restart_command: command to restart the target in case it is deadore
:param logger: logger for this object (default: None)
'''
super(SSHMonitor, self).__init__(name, logger)
self._status_command = status_command
self._restart_command = restart_command
self._ssh = ReconnectingSSHConnection(hostname, port, username,
password)
def teardown(self):
self._ssh.close()
super(SSHMonitor, self).teardown()
def _ssh_command(self, cmd):
return_code = stdout = stderr = None
try:
return_code, stdout, stderr = self._ssh.exec_command(cmd)
except KeyboardInterrupt:
raise
except Exception as e:
self.logger.debug('SSHMonitor: ssh command exec error: %s' %
str(e))
return return_code, stdout, stderr
def post_test(self):
return_code, stdout, stderr = self._ssh_command(self._status_command)
if return_code != 0:
self.report.add('status_command', self._status_command)
self.report.add('status_command return code', return_code)
self.report.add('status_command stdout', stdout)
self.report.add('status_command stderr', stderr)
self.report.failed("got non-zero return code")
if self._restart_command:
self.logger.info(
'target not responding - restarting target !!!')
return_code, stdout, stderr = self._ssh_command(
self._restart_command)
self.report.add('restart_command', self._restart_command)
self.report.add('restart_command return code', return_code)
self.report.add('restart_command stdout', stdout)
self.report.add('restart_command stderr', stderr)
super(SSHMonitor, self).post_test()
def pre_test(self, test_number):
super(SSHMonitor, self).pre_test(test_number)
while self._ssh_command(self._status_command)[0] != 0:
self.logger.debug('waiting for target to be up')
time.sleep(1)
def _monitor_func(self):
'''
Nothing is done here, so we use a sleep for now.
'''
time.sleep(0.1)
class SshFileMonitor(BaseMonitor):
'''
SshFileMonitor monitors for files using a file_mask.
If found - moves files to local folder, renaming with test number.
'''
def __init__(self, name, username, password, hostname, port,
file_mask,
local_dir,
use_scp=False,
fail_if_exists=True,
setup_commands=[],
on_fail_command=None,
on_fail_delay=0,
logger=None):
'''
:param name: name of the object
:param username: ssh login username
:param password: ssh login password
:param hostname: ssh server ip
:param port: ssh server port
:param file_mask: file_mask to fetch
:param local_dir: local_path to store fetched files
:param use_scp: use the SCP protocol for transferring files instead of SFTP
:param fail_if_exists: fail test if file exists (default: True)
:param on_fail_command: command to run on failure (default: None)
:param on_fail_delay: time to sleep after running on_fail_command (default: 0)
:param logger: logger for this object (default: None)
'''
super(SshFileMonitor, self).__init__(name, logger)
self._file_mask = file_mask
self._local_dir = local_dir
self._fail_if_exists = fail_if_exists
self._setup_commands = setup_commands
self._on_fail_command = on_fail_command
self._on_fail_delay = on_fail_delay
self._ssh = ReconnectingSSHConnection(hostname, port, username, password, use_scp)
def _ssh_command(self, cmd):
return_code = None
try:
return_code, self._stdout, self._stderr = self._ssh.exec_command(cmd)
except KeyboardInterrupt:
raise
except Exception as e:
self.logger.debug('SSHFileMonitor: ssh command exec error: %s' % str(e))
return return_code
def setup(self):
'''
Called at the begining of the fuzzing session
'''
super(SshFileMonitor, self).setup()
timestamp = datetime.datetime.now().strftime("%Y_%m_%d-%H_%M_%S")
self._local_dir = os.path.join(self._local_dir, timestamp)
os.makedirs(self._local_dir)
for command in self._setup_commands:
self.logger.debug('running remote setup command: %s' % command)
res = self._ssh_command(command)
if res != 0:
self.logger.debug('Error running command: %s. got %s' % (command, res))
self.logger.debug('stdout: %s' % self._stdout)
self.logger.debug('stderr: %s' % self._stderr)
def teardown(self):
if self._ssh:
self._ssh.close()
self._ssh = None
super(SshFileMonitor, self).teardown()
def post_test(self):
cmd = 'ls %s' % self._file_mask
return_code = self._ssh_command(cmd)
self.report.add('file_mask', self._file_mask)
if return_code == 0:
ls_stdout = self._stdout
ls_stderr = self._stderr
del ls_stderr
self.report.failed('found core file')
self.report.add('remote file', ls_stdout.strip())
remote_path = ls_stdout.strip()
local_path = os.path.join(self._local_dir, 'test_%05d' % self.test_number)
self._ssh.get(str(remote_path), str(local_path))
self._ssh.remove(str(remote_path))
self.report.add('local file', local_path)
if self._on_fail_command:
self.logger.info('running remote on fail command: %s', self._on_fail_command)
self.report.add('on_fail_command', self._on_fail_command)
self.report.add('on_fail_delay', self._on_fail_delay)
retval = self._ssh_command(self._on_fail_command)
self.logger.info('on fail command returned %s' % retval)
if retval != 0:
self.logger.info('on fail command stdout %s' % self._stdout)
self.logger.info('on fail command stderr %s' % self._stderr)
else:
self.logger.info('sleeping for %s seconds' % self._on_fail_delay)
time.sleep(self._on_fail_delay)
super(SshFileMonitor, self).post_test()
def X_pre_test(self, test_number):
super(SshFileMonitor, self).pre_test(test_number)
def _monitor_func(self):
'''
Nothing is done here, so we use a sleep for now.
'''
time.sleep(0.1)