コード例 #1
0
 def flush_buf(self):
     msg = self.messenger
     stdios = [msg.o_stdout, msg.o_stderr]
     r, _, _ = select.select(stdios, [], [], 0)  # Not blocking.
     if msg.o_stdout in r:  # Send message from stdout
         data = os.read(msg.o_stdout, 16384)
         msg.write_msg(remote_interface.StdOut(data))
     if msg.o_stderr in r:  # Send message from stdout
         data = os.read(msg.o_stderr, 16384)
         msg.write_msg(remote_interface.StdErr(data))
コード例 #2
0
ファイル: remote_runner.py プロジェクト: alanoe/avocado-vt
 def query_master(self, *args, **kargs):
     """Read CmdRespond from master."""
     # First of all, dump stdio to master. Otherwise it can block on
     # logging.info() function.
     msg = self.messenger
     stdios = [msg.o_stdout, msg.o_stderr]
     r, _, _ = select.select(stdios, [], [], 0)  # Not blocking.
     if msg.o_stdout in r:  # Send message from stdout
         data = os.read(msg.o_stdout, 16384)
         msg.write_msg(remote_interface.StdOut(data))
     if msg.o_stderr in r:  # Send message from stdout
         data = os.read(msg.o_stderr, 16384)
         msg.write_msg(remote_interface.StdErr(data))
     # Sent actual request.
     cmd = remote_interface.CmdQuery(args, kargs)
     msg.write_msg(cmd)
     succ, data = msg.read_msg()
     assert succ
     assert isinstance(data, remote_interface.CmdRespond)
     return data.respond
コード例 #3
0
ファイル: remote_runner.py プロジェクト: alanoe/avocado-vt
    def cmd_loop(self):
        """
        Wait for commands from master and receive results and outputs from
        commands.
        """
        try:
            while (not self._exit):
                stdios = [self.stdin, self.o_stdout, self.o_stderr]
                r_pipes = [cmd.r_pipe for cmd in self.cmds.values()
                           if cmd.r_pipe is not None]
                stdouts = [cmd.stdout_pipe for cmd in self.cmds.values()
                           if cmd.stdout_pipe is not None]
                stderrs = [cmd.stderr_pipe for cmd in self.cmds.values()
                           if cmd.stderr_pipe is not None]

                r, _, _ = select.select(
                    stdios + r_pipes + stdouts + stderrs, [], [])

                if self.stdin in r:  # command from controller
                    cmd = CmdSlave(self.read_msg()[1])
                    self.cmds[cmd.cmd_id] = cmd
                    try:
                        # There is hidden bug. We can bump into condition when
                        # we running some function-command. That function dump
                        # some data to stdio. But, stdio is out of
                        # consideration at this point. It can stuck.
                        cmd(self)
                        self.write_msg(cmd.basecmd)
                    except Exception:
                        err_msg = traceback.format_exc()
                        self.write_msg(
                            remote_interface.CommanderError(err_msg))

                if self.o_stdout in r:  # Send message from stdout
                    msg = os.read(self.o_stdout, 16384)
                    self.write_msg(remote_interface.StdOut(msg))
                if self.o_stderr in r:  # Send message from stdout
                    msg = os.read(self.o_stderr, 16384)
                    self.write_msg(remote_interface.StdErr(msg))

                # test all commands for io
                for cmd in self.cmds.values():
                    if cmd.stdout_pipe in r:  # command stdout
                        data = os.read(cmd.stdout_pipe, 16384)
                        if data != "":  # pipe is not closed on another side.
                            self.write_msg(remote_interface.StdOut(data,
                                                                   cmd.cmd_id))
                        else:
                            os.close(cmd.stdout_pipe)
                            cmd.stdout_pipe = None
                    if cmd.stderr_pipe in r:  # command stderr
                        data = os.read(cmd.stderr_pipe, 16384)
                        if data != "":  # pipe is not closed on another side.
                            self.write_msg(remote_interface.StdErr(data,
                                                                   cmd.cmd_id))
                        else:
                            os.close(cmd.stderr_pipe)
                            cmd.stderr_pipe = None
                    if cmd.r_pipe in r:  # command results
                        if cmd.work():
                            cmd.finish(self)
                        self.write_msg(cmd.basecmd)
        except Exception:
            err_msg = traceback.format_exc()
            self.write_msg(remote_interface.CommanderError(err_msg))