From: Ondrej Lichtner <olichtne(a)redhat.com&gt; This commit adds the {get, set}_result_sent methods, and a new attribute (_result_sent) to the NetTestCommand class. This attribute is used to indicate that the result of a command was already sent to the Controller. The set method takes an optional parameter 'value' that can be used to set the attribute to False if needed (not currently used). Signed-off-by: Ondrej Lichtner <olichtne(a)redhat.com&gt; --- lnst/Common/NetTestCommand.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/lnst/Common/NetTestCommand.py b/lnst/Common/NetTestCommand.py index 0ce8998..9ffd876 100644 --- a/lnst/Common/NetTestCommand.py +++ b/lnst/Common/NetTestCommand.py @@ -99,6 +99,7 @@ class NetTestCommand: self._result = None self._log_ctl = log_ctl self._start_time = None + self._result_sent = False if "bg_id" not in self._command: self._id = None @@ -111,6 +112,12 @@ class NetTestCommand: def forked(self): return self._process != None + def set_result_sent(self, value=True): + self._result_sent = value + + def get_result_sent(self): + return self._result_sent + def finished(self): return self._finished -- 2.5.0

From: Ondrej Lichtner <olichtne(a)redhat.com&gt; This commit makes shure that when the rpc method kill_command is called, it first checks if the command didn't finish first. If it did, it will only be removed from the command context. This situation can occur when the Controller detects a command timeout and sends a kill_command request, but on the Slave the command is already finished and the result was sent in the mean time. This commit should fix this issue and the command will be seen as finished on time. If the command was killed we now also add a killed flag to the result message to indicate this action. Related to #120 Signed-off-by: Ondrej Lichtner <olichtne(a)redhat.com&gt; --- lnst/Common/NetTestCommand.py | 1 + lnst/Controller/Machine.py | 5 +++-- lnst/Slave/NetTestSlave.py | 29 ++++++++++++++++++++++++----- 3 files changed, 28 insertions(+), 7 deletions(-) diff --git a/lnst/Common/NetTestCommand.py b/lnst/Common/NetTestCommand.py index f80968e..7385e47 100644 --- a/lnst/Common/NetTestCommand.py +++ b/lnst/Common/NetTestCommand.py @@ -212,6 +212,7 @@ class NetTestCommand: result = self._cmd_cls.get_result() result["passed"] = True result["msg"] = "Command killed." + result["killed"] = True self._result = result return self._result diff --git a/lnst/Controller/Machine.py b/lnst/Controller/Machine.py index 8a40ceb..e60e1ea 100644 --- a/lnst/Controller/Machine.py +++ b/lnst/Controller/Machine.py @@ -316,8 +316,9 @@ class Machine(object): else: cmd_res = self._rpc_call("kill_command", None) - cmd_res["passed"] = False - cmd_res["msg"] = str(exc) + if "killed" in cmd_res and cmd_res["killed"]: + cmd_res["passed"] = False + cmd_res["msg"] = str(exc) signal.alarm(0) signal.signal(signal.SIGALRM, prev_handler) diff --git a/lnst/Slave/NetTestSlave.py b/lnst/Slave/NetTestSlave.py index a725a62..775d4a5 100644 --- a/lnst/Slave/NetTestSlave.py +++ b/lnst/Slave/NetTestSlave.py @@ -366,6 +366,18 @@ class SlaveMethods: def run_command(self, command): cmd = NetTestCommand(self._command_context, command, self._resource_table, self._log_ctl) + + if self._command_context.get_cmd(cmd.get_id()) != None: + prev_cmd = self._command_context.get_cmd(cmd.get_id()) + if not prev_cmd.get_result_sent(): + if cmd.get_id() is None: + raise Exception("Previous foreground command still "\ + "running!") + else: + raise Exception("Different command with id '%s' "\ + "still running!" % cmd.get_id()) + else: + self._command_context.del_cmd(cmd) self._command_context.add_cmd(cmd) res = cmd.run() @@ -385,9 +397,16 @@ class SlaveMethods: def kill_command(self, id): cmd = self._command_context.get_cmd(id) - cmd.kill(None) - self._command_context.del_cmd(cmd) - return cmd.get_result() + if cmd is not None: + if not cmd.get_result_sent(): + cmd.kill(None) + result = cmd.get_result() + cmd.set_result_sent() + return result + else: + pass + else: + raise Exception("No command with id '%s'." % id) def machine_cleanup(self): logging.info("Performing machine cleanup.") @@ -852,7 +871,7 @@ class NetTestSlave: self._server_handler.send_data_to_ctl(msg) cmd = self._cmd_context.get_cmd(None) cmd.join() - self._cmd_context.del_cmd(cmd) + cmd.set_result_sent() else: cmd = self._cmd_context.get_cmd(msg["cmd_id"]) cmd.join() @@ -862,7 +881,7 @@ class NetTestSlave: if cmd.finished(): msg["result"] = cmd.get_result() self._server_handler.send_data_to_ctl(msg) - self._cmd_context.del_cmd(cmd) + cmd.set_result_sent() elif msg["type"] == "netlink": self._if_manager.handle_netlink_msgs(msg["data"]) elif msg["type"] == "from_netns": -- 2.5.0