def packet_test_cb(self, buf, *args):
"""
Used to trigger packet sender when receiver is ready,
"""
ret = SshCbReturn()
ret.buf = buf
# ret.removecb = True
ret.nextargs = args
verbose = args[0]
if verbose:
self.debug(buf)
return ret
def packet_test_cb(self, buf, *args):
"""
Used to trigger packet sender when receiver is ready,
"""
ret = SshCbReturn()
ret.buf = buf
# ret.removecb = True
ret.nextargs = args
verbose = args[0]
if verbose:
self.debug(buf)
return ret
def packet_test_cb(self, buf, *args):
"""
Used to trigger packet sender when receiver is ready,
"""
ret = SshCbReturn()
ret.buf = buf
ret.nextargs = args
sender = args[0]
if re.search(START_MESSAGE, buf):
sender.start()
if sender.done_time:
if verbose:
self.ssh.debug('Sender is Done. Setting command timer to 5 seconds')
ret.settimer = 5
if verbose:
self.ssh.debug(buf)
return ret
def packet_test_cb(self, buf, *args):
"""
Used to trigger packet sender when receiver is ready,
"""
ret = SshCbReturn()
ret.buf = buf
ret.nextargs = args
sender = args[0]
if re.search(START_MESSAGE, buf):
sender.start()
if sender.done_time:
if verbose:
self.ssh.debug('Sender is Done. Setting command timer to 5 seconds')
ret.settimer = 5
if verbose:
self.ssh.debug(buf)
return ret
def _bundle_status_cb(self, buf, cmdtimeout, parttimeout, starttime,
lasttime, check_image_stage):
ret = SshCbReturn(stop=False)
#if the over timeout or the callback interval has expired,
# then return stop=true
#interval timeout should not be hit due to the setting of the
# timer value, but check here anyways
if (cmdtimeout != 0) and (int(time.time() - starttime) > cmdtimeout):
self.log.debug('bundle_status_cb command timed out after ' +
str(cmdtimeout) + ' seconds')
ret.statuscode = -100
ret.stop = True
return ret
if not check_image_stage:
ret.settimer = parttimeout
if (parttimeout != 0 and lasttime != 0) and \
(int(time.time()-lasttime) > parttimeout):
self.log.debug('bundle_status_cb inter-part time out after ' +
str(parttimeout) + ' seconds')
ret.statuscode = -100
ret.stop = True
return ret
if re.search('[P|p]art:', buf):
sys.stdout.write("\r\x1b[K" + str(buf).strip())
sys.stdout.flush()
check_image_stage = False
else:
#Print command output and write to ssh.cmd['output'] buffer
ret.buf = buf
self.log.debug(str(buf))
#Command is still going, reset timer thread to intervaltimeout,
# provide arguments for next time this is called from ssh cmd.
ret.stop = False
ret.nextargs = [
cmdtimeout, parttimeout, starttime,
time.time(), check_image_stage
]
return ret
def remote_tail_monitor_cb(self,
buf,
instance_id,
max_lines,
lines_read,
start_time,
timeout,
print_method,
prefix,
idle_timeout):
ret = SshCbReturn(stop=False, settimer=idle_timeout)
return_buf = ""
now = time.time()
if (timeout and (now - start_time) >= timeout) or (max_lines and lines_read >= max_lines):
ret.statuscode = 0
ret.stop = True
try:
for line in str(buf).splitlines():
lines_read += 1
print_method(str(prefix) + str(line))
except Exception, e:
return_buf = "Error in remote_tail_monitor:" + str(e)
ret.statuscode = 69
ret.stop = True
def expect_prompt_cb(sshconnection,
buf,
command=None,
prompt_match="^\w+(>|#|\$)",
verbose=None,
debug_method=None):
prompt = prompt_match + "\s*$"
start_match = None
if command is not None:
start_match = prompt + "\s*" + command + "\s*$"
if verbose is None:
verbose = sshconnection.verbose
ret = SshCbReturn(stop=False)
ret.buf = buf
def debug(msg, ssh=sshconnection):
if verbose:
if debug_method:
debug_method(msg)
else:
print(msg)
debug('Starting expect_prompt_cb, prompt:' + str(prompt))
debug('Starting buf:"' + str(buf) + '"')
# Create a callback return obj for the cmd() loop to consume...
lines = buf.splitlines()
# See if we have a prompt for password, assume we only have one match and were
# blocking waiting on password input
for line in lines:
sshconnection.debug('line:' + str(line))
if re.search(prompt, line):
debug('Got prompt match in buffer. start_match:{0}, Line:"{1}"'.
format(start_match, line))
if start_match:
if re.search(start_match, line):
sshconnection.debug(
'Found match for start_match:{0}, line:{1}'.format(
start_match, line))
command = None
start_match = None
else:
ret.removecb = True
ret.stop = True
debug('Ending buf:"' + str(ret.buf) + '"')
return ret
else:
debug('\nPrompt not found, continuing...')
ret.removecb = False
ret.nextargs = [command, prompt_match, verbose, debug_method]
# debug('Ending buf:"' + str(ret.buf) + '"')
return ret
def _bundle_status_cb(self, buf, cmdtimeout, parttimeout, starttime,
lasttime, check_image_stage):
ret = SshCbReturn(stop=False)
#if the over timeout or the callback interval has expired,
# then return stop=true
#interval timeout should not be hit due to the setting of the
# timer value, but check here anyways
if (cmdtimeout != 0) and ( int(time.time()-starttime) > cmdtimeout):
self.log.debug('bundle_status_cb command timed out after ' +
str(cmdtimeout)+' seconds')
ret.statuscode=-100
ret.stop = True
return ret
if not check_image_stage:
ret.settimer = parttimeout
if (parttimeout != 0 and lasttime != 0) and \
(int(time.time()-lasttime) > parttimeout):
self.log.debug('bundle_status_cb inter-part time out after ' +
str(parttimeout) + ' seconds')
ret.statuscode=-100
ret.stop = True
return ret
if re.search('[P|p]art:',buf):
sys.stdout.write("\r\x1b[K"+str(buf).strip())
sys.stdout.flush()
check_image_stage=False
else:
#Print command output and write to ssh.cmd['output'] buffer
ret.buf = buf
self.log.debug(str(buf))
#Command is still going, reset timer thread to intervaltimeout,
# provide arguments for next time this is called from ssh cmd.
ret.stop = False
ret.nextargs =[cmdtimeout, parttimeout, starttime,
time.time(), check_image_stage]
return ret
def expect_prompt_cb(sshconnection,
buf,
command=None,
prompt_match="^\w+(>|#|\$)",
verbose=None,
debug_method=None):
prompt = prompt_match + "\s*$"
start_match = None
if command is not None:
start_match = prompt + "\s*" + command + "\s*$"
if verbose is None:
verbose = sshconnection.verbose
ret = SshCbReturn(stop=False)
ret.buf = buf
def debug(msg, ssh=sshconnection):
if verbose:
if debug_method:
debug_method(msg)
else:
print(msg)
debug('Starting expect_prompt_cb, prompt:' + str(prompt))
debug('Starting buf:"' + str(buf) + '"')
# Create a callback return obj for the cmd() loop to consume...
lines = buf.splitlines()
# See if we have a prompt for password, assume we only have one match and were
# blocking waiting on password input
for line in lines:
sshconnection.debug('line:' + str(line))
if re.search(prompt, line):
debug('Got prompt match in buffer. start_match:{0}, Line:"{1}"'
.format(start_match, line))
if start_match:
if re.search(start_match, line):
sshconnection.debug('Found match for start_match:{0}, line:{1}'
.format(start_match, line))
command = None
start_match = None
else:
ret.removecb = True
ret.stop = True
debug('Ending buf:"' + str(ret.buf) + '"')
return ret
else:
debug('\nPrompt not found, continuing...')
ret.removecb = False
ret.nextargs = [command, prompt_match, verbose, debug_method]
# debug('Ending buf:"' + str(ret.buf) + '"')
return ret
def remote_tail_monitor_cb(self, buf, instance_id, max_lines, lines_read,
start_time, timeout, print_method, prefix,
idle_timeout):
ret = SshCbReturn(stop=False, settimer=idle_timeout)
return_buf = ""
now = time.time()
if (timeout and
(now - start_time) >= timeout) or (max_lines
and lines_read >= max_lines):
ret.statuscode = 0
ret.stop = True
try:
for line in str(buf).splitlines():
lines_read += 1
print_method(str(prefix) + str(line))
except Exception, e:
return_buf = "Error in remote_tail_monitor:" + str(e)
ret.statuscode = 69
ret.stop = True
def update_output(buf):
ret = SshCbReturn()
logger.debug(buf)
ret.settimer = args.timeout # Add seconds back to timeout
ret.buf = buf
return ret
def expect_password_cb(buf,
password,
prompt="^Password",
nextcb=None,
cbargs=[],
retry=0,
password_attempts=0,
verbose=False,
debug_method=None):
'''
Sample callback to handle password prompts to be provided to ssh.cmd()
:param buf: output from cmd()
:param password: string password to be supplied to a detected 'password' prompt
:param nextcb: function/method callback to be returned, this cmd() will use to
handle it's future output.
:param prompt: regex string used to match prompt. case insensitive match used
:
'''
ret = SshCbReturn(stop=False)
def debug(msg):
if verbose:
if debug_method:
debug_method(msg)
else:
print(msg)
def add_to_buffer(lines_to_add, newbuf):
for line in lines_to_add:
debug('Adding line to buf:"' + str(line) + '"')
if newbuf is None:
newbuf = line + '\n'
else:
newbuf += line + '\n'
return newbuf
def bufadd(line):
return add_to_buffer(line, ret.buf)
debug('STARTING expect_password_cb: password:'******'Starting buf:"' + str(buf) + '"')
# Create a callback return obj for the cmd() loop to consume...
lines = buf.splitlines()
# See if we've already gone through password dialog, if so there may be a left
# over newline. Remove it.
if password_attempts and lines[0] == "":
debug('Removing first blank line(s) after sending password')
lines.pop(0)
if not lines:
# self.debug('Skipping blanklines...')
ret.buf = None
ret.nextargs = [
password, prompt, nextcb, cbargs, retry, password_attempts,
verbose
]
return ret
# See if we have a prompt for password, assume we only have one match and were blocking
# waiting on password input
prompt_indices = [
i for i, s in enumerate(lines) if re.match(prompt, s, re.IGNORECASE)
]
if prompt_indices:
debug('Got password prompt, sending password...')
# Check to see if we've already tried a password, and if we should retry or fail
if password_attempts > retry:
raise CommandExpectPasswordException("Password dialog attempts:" +
str(password_attempts) +
" exceeded retry limit:" +
str(retry))
prompt_index = prompt_indices[0]
# Add any lines other than password prompt back to return buffer
# Remove line with password prompt first
lines.pop(prompt_index)
ret.buf = bufadd(lines)
# Add password to CbReturn sendstring value to be sent to channel in cmd() loop...
ret.sendstring = str(password).rstrip() + "\n"
# Increment our password attempts value, and args to return obj for next time
# we get called. Expecting to get called at least 1 more time to handle the blank
# line after password dialog, may also get called for a login, and prompted
# for a password again...
password_attempts += 1
ret.removecb = False
ret.nextcb = None
ret.nextargs = [
password, prompt, nextcb, cbargs, retry, password_attempts,
verbose, debug_method
]
debug('Ending buf:"' + str(ret.buf) + '"')
return ret
else:
debug('\nPassword prompt not found, continuing. password_attempts:' +
str(password_attempts) + ', prompt:' + str(prompt) +
', len lines: ' + str(len(lines)))
# Assume the password dialog is complete, pass buffer to next handler,
# attempt to make password portion look transparent
ret.buf = bufadd(lines)
if nextcb is not None:
debug('Got nextcb, calling it on our buffer now...')
ret = nextcb(ret.buf, *cbargs)
if ret.nextcb and not ret.removecb:
nextcb = ret.nextcb
else:
nextcb = None
# tweak the return values, store the real ones in our nextargs and handle
# things in this method instead
ret.nextcb = None
ret.removecb = False
ret.nextargs = [
password, prompt, nextcb, cbargs, retry, password_attempts, verbose
]
debug('Ending buf:"' + str(ret.buf) + '"')
return ret
def expect_password_cb(buf,
password,
prompt="^Password",
nextcb=None,
cbargs=[],
retry=0,
password_attempts=0,
verbose=False,
debug_method=None):
'''
Sample callback to handle password prompts to be provided to ssh.cmd()
:param buf: output from cmd()
:param password: string password to be supplied to a detected 'password' prompt
:param nextcb: function/method callback to be returned, this cmd() will use to
handle it's future output.
:param prompt: regex string used to match prompt. case insensitive match used
:
'''
ret = SshCbReturn(stop=False)
def debug(msg):
if verbose:
if debug_method:
debug_method(msg)
else:
print(msg)
def add_to_buffer(lines_to_add, newbuf):
for line in lines_to_add:
debug('Adding line to buf:"' + str(line) + '"')
if newbuf is None:
newbuf = line + '\n'
else:
newbuf += line + '\n'
return newbuf
def bufadd(line):
return add_to_buffer(line, ret.buf)
debug('STARTING expect_password_cb: password:'******'Starting buf:"' + str(buf) + '"')
# Create a callback return obj for the cmd() loop to consume...
lines = buf.splitlines()
# See if we've already gone through password dialog, if so there may be a left
# over newline. Remove it.
if password_attempts and lines[0] == "":
debug('Removing first blank line(s) after sending password')
lines.pop(0)
if not lines:
# self.debug('Skipping blanklines...')
ret.buf = None
ret.nextargs = [password, prompt, nextcb, cbargs, retry, password_attempts,
verbose]
return ret
# See if we have a prompt for password, assume we only have one match and were blocking
# waiting on password input
prompt_indices = [i for i, s in enumerate(lines) if re.match(prompt, s, re.IGNORECASE)]
if prompt_indices:
debug('Got password prompt, sending password...')
# Check to see if we've already tried a password, and if we should retry or fail
if password_attempts > retry:
raise CommandExpectPasswordException(
"Password dialog attempts:" + str(password_attempts) +
" exceeded retry limit:" + str(retry))
prompt_index = prompt_indices[0]
# Add any lines other than password prompt back to return buffer
# Remove line with password prompt first
lines.pop(prompt_index)
ret.buf = bufadd(lines)
# Add password to CbReturn sendstring value to be sent to channel in cmd() loop...
ret.sendstring = str(password).rstrip() + "\n"
# Increment our password attempts value, and args to return obj for next time
# we get called. Expecting to get called at least 1 more time to handle the blank
# line after password dialog, may also get called for a login, and prompted
# for a password again...
password_attempts += 1
ret.removecb = False
ret.nextcb = None
ret.nextargs = [password, prompt, nextcb, cbargs, retry, password_attempts, verbose,
debug_method]
debug('Ending buf:"' + str(ret.buf) + '"')
return ret
else:
debug('\nPassword prompt not found, continuing. password_attempts:' +
str(password_attempts) + ', prompt:' + str(prompt) +
', len lines: ' + str(len(lines)))
# Assume the password dialog is complete, pass buffer to next handler,
# attempt to make password portion look transparent
ret.buf = bufadd(lines)
if nextcb is not None:
debug('Got nextcb, calling it on our buffer now...')
ret = nextcb(ret.buf, *cbargs)
if ret.nextcb and not ret.removecb:
nextcb = ret.nextcb
else:
nextcb = None
# tweak the return values, store the real ones in our nextargs and handle
# things in this method instead
ret.nextcb = None
ret.removecb = False
ret.nextargs = [password, prompt, nextcb, cbargs, retry, password_attempts, verbose]
debug('Ending buf:"' + str(ret.buf) + '"')
return ret
def update_output(buf):
ret = SshCbReturn()
logger.debug(buf)
ret.settimer = args.timeout # Add seconds back to timeout
ret.buf = buf
return ret
