示例#1
0
 def execute_command(
     self, command, bufsize=-1, timeout=None, stdin_str="", stdin_file=None,
         raise_exceptions=False):
     timeout = timeout or DEFAULT_TIMEOUT
     chan = self._transport.open_session()
     chan.settimeout(POLLING_RATE)
     chan.exec_command(command)
     stdin_str = stdin_str if stdin_file is None else stdin_file.read()
     stdin = chan.makefile("wb", bufsize)
     stdout = chan.makefile("rb", bufsize)
     stderr = chan.makefile_stderr("rb", bufsize)
     stdin.write(stdin_str)
     stdin.close()
     stdout_str = stderr_str = ""
     exit_status = None
     max_time = time.time() + timeout
     while not chan.exit_status_ready():
         stderr_str += self._read_channel(stderr)
         stdout_str += self._read_channel(stdout)
         if max_time < time.time():
             raise socket.timeout(
                 "Command timed out\nSTDOUT:{0}\nSTDERR:{1}\n".format(
                     stdout_str, stderr_str))
     exit_status = chan.recv_exit_status()
     stdout_str += self._read_channel(stdout)
     stderr_str += self._read_channel(stderr)
     chan.close()
     return stdin_str, stdout_str, stderr_str, exit_status
示例#2
0
 def execute_command(self,
                     command,
                     bufsize=-1,
                     timeout=None,
                     stdin_str="",
                     stdin_file=None,
                     raise_exceptions=False):
     timeout = timeout or DEFAULT_TIMEOUT
     chan = self._transport.open_session()
     chan.settimeout(POLLING_RATE)
     chan.exec_command(command)
     stdin_str = stdin_str if stdin_file is None else stdin_file.read()
     stdin = chan.makefile("wb", bufsize)
     stdout = chan.makefile("rb", bufsize)
     stderr = chan.makefile_stderr("rb", bufsize)
     stdin.write(stdin_str)
     stdin.close()
     stdout_str = stderr_str = ""
     exit_status = None
     max_time = time.time() + timeout
     while not chan.exit_status_ready():
         stderr_str += self._read_channel(stderr)
         stdout_str += self._read_channel(stdout)
         if max_time < time.time():
             raise socket.timeout(
                 "Command timed out\nSTDOUT:{0}\nSTDERR:{1}\n".format(
                     stdout_str, stderr_str))
     exit_status = chan.recv_exit_status()
     stdout_str += self._read_channel(stdout)
     stderr_str += self._read_channel(stderr)
     chan.close()
     return stdin_str, stdout_str, stderr_str, exit_status
示例#3
0
 def _read_shell_response(self, uuid, max_time):
     stdout = stderr = ""
     exit_status = None
     while max_time > time.time():
         stdout += self._read_channel(self.channel.recv)
         stderr += self._read_channel(self.channel.recv_stderr)
         if stdout.count(uuid) == 2:
             list_ = stdout.split(uuid)
             stdout = list_[1]
             try:
                 exit_status = int(list_[2])
             except (ValueError, TypeError):
                 exit_status = None
             break
     else:
         raise socket.timeout(
             "Command timed out\nSTDOUT:{0}\nSTDERR:{1}\n".format(
                 stdout, stderr))
     response = ExecResponse(
         stdin=None, stdout=stdout.strip(), stderr=stderr,
         exit_status=exit_status)
     return response
示例#4
0
 def _read_shell_response(self, uuid, max_time):
     stdout = stderr = ""
     exit_status = None
     while max_time > time.time():
         stdout += self._read_channel(self.channel.recv)
         stderr += self._read_channel(self.channel.recv_stderr)
         if stdout.count(uuid) == 2:
             list_ = stdout.split(uuid)
             stdout = list_[1]
             try:
                 exit_status = int(list_[2])
             except (ValueError, TypeError):
                 exit_status = None
             break
     else:
         raise socket.timeout(
             "Command timed out\nSTDOUT:{0}\nSTDERR:{1}\n".format(
                 stdout, stderr))
     response = ExecResponse(
         stdin=None, stdout=stdout.strip(), stderr=stderr,
         exit_status=exit_status)
     return response
示例#5
0
 def _wait_for_active_shell(self, max_time):
     while not self.channel.send_ready():
         time.sleep(POLLING_RATE)
         if max_time < time.time():
             raise socket.timeout("Timed out waiting for active shell")
示例#6
0
 def _wait_for_active_shell(self, max_time):
     while not self.channel.send_ready():
         time.sleep(common.POLLING_RATE)
         if max_time < time.time():
             raise socket.timeout("Timed out waiting for active shell")