summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rwxr-xr-xsrc/console/trex_console.py38
-rw-r--r--src/console/trex_rpc_client.py12
-rw-r--r--src/rpc-server/src/commands/trex_rpc_cmd_general.cpp47
-rw-r--r--src/rpc-server/src/commands/trex_rpc_cmds.h29
-rw-r--r--src/rpc-server/src/trex_rpc_cmds_table.cpp9
5 files changed, 127 insertions, 8 deletions
diff --git a/src/console/trex_console.py b/src/console/trex_console.py
index 404a2ee0..3e452bf5 100755
--- a/src/console/trex_console.py
+++ b/src/console/trex_console.py
@@ -1,6 +1,8 @@
#!/usr/bin/python
# -*- coding: utf-8 -*-
import cmd
+import json
+
from trex_rpc_client import RpcClient
class TrexConsole(cmd.Cmd):
@@ -12,12 +14,18 @@ class TrexConsole(cmd.Cmd):
self.intro = "\n-=TRex Console V1.0=-\n"
self.rpc_client = rpc_client
self.verbose = False
+ rc, msg = self.rpc_client.query_rpc_server()
+
+ if rc:
+ lst = msg.split('\n')
+ self.supported_rpc = [str(x) for x in lst if x]
# a cool hack - i stole this function and added space
def completenames(self, text, *ignored):
dotext = 'do_'+text
return [a[3:]+' ' for a in self.get_names() if a.startswith(dotext)]
+ # set verbose on / off
def do_verbose (self, line):
'''shows or set verbose mode\nusage: verbose [on/off]\n'''
if line == "":
@@ -34,7 +42,7 @@ class TrexConsole(cmd.Cmd):
else:
print "please specify 'on' or 'off'\n"
-
+ # query the server for registered commands
def do_query_server(self, line):
'''\nquery the RPC server for supported remote commands\n'''
rc, msg = self.rpc_client.query_rpc_server()
@@ -44,7 +52,7 @@ class TrexConsole(cmd.Cmd):
print "RPC server supports the following commands: \n\n" + msg
def do_ping (self, line):
- '''pings the RPC server \n'''
+ '''pings the RPC server\n'''
print "Pinging RPC server"
rc, msg = self.rpc_client.ping_rpc_server()
@@ -53,6 +61,23 @@ class TrexConsole(cmd.Cmd):
else:
print "[FAILED]\n"
+ def do_rpc (self, line):
+ '''Launches a RPC on the server\n'''
+ if line == "":
+ print "Please provide method name\n"
+ return
+
+ rc, msg = self.rpc_client.invoke_rpc_method(line)
+ if rc:
+ print "[SUCCESS]\n"
+ else:
+ print "[FAILED]\n"
+
+ print "Server Response:\n\n{0}\n".format(json.dumps(msg))
+
+ def complete_rpc (self, text, line, begidx, endidx):
+ return [x for x in self.supported_rpc if x.startswith(text)]
+
def do_quit(self, line):
'''\nexit the client\n'''
return True
@@ -73,8 +98,13 @@ def main ():
exit(-1)
# console
- console = TrexConsole(rpc_client)
- console.cmdloop()
+ try:
+ console = TrexConsole(rpc_client)
+ console.cmdloop()
+ except KeyboardInterrupt as e:
+ print "\n\n*** Caught Ctrl + C... Exiting...\n\n"
+ return
if __name__ == '__main__':
main()
+
diff --git a/src/console/trex_rpc_client.py b/src/console/trex_rpc_client.py
index e6ff5fee..ac4aa298 100644
--- a/src/console/trex_rpc_client.py
+++ b/src/console/trex_rpc_client.py
@@ -50,13 +50,15 @@ class RpcClient():
response_json = json.loads(response)
if (response_json.get("jsonrpc") != "2.0"):
- return False, "Bad Server Response: " + str(response)
+ return False, "Malfromed Response ({0})".format(str(response))
+ # error reported by server
if ("error" in response_json):
- return False, "Server Has Reported An Error: " + str(response)
+ return False, response_json["error"]["message"]
+ # if no error there should be a result
if ("result" not in response_json):
- return False, "Bad Server Response: " + str(response)
+ return False, "Malfromed Response ({0})".format(str(response))
return True, response_json["result"]
@@ -97,5 +99,9 @@ class RpcClient():
print "[SUCCESS]\n"
+ def __del__ (self):
+ print "Shutting down RPC client\n"
+ self.context.destroy(linger = 0)
+
diff --git a/src/rpc-server/src/commands/trex_rpc_cmd_general.cpp b/src/rpc-server/src/commands/trex_rpc_cmd_general.cpp
new file mode 100644
index 00000000..2be4a76f
--- /dev/null
+++ b/src/rpc-server/src/commands/trex_rpc_cmd_general.cpp
@@ -0,0 +1,47 @@
+/*
+ Itay Marom
+ Cisco Systems, Inc.
+*/
+
+/*
+Copyright (c) 2015-2015 Cisco Systems, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+ http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+#include "trex_rpc_cmds.h"
+#include <../linux_dpdk/version.h>
+
+using namespace std;
+
+/**
+ * get status
+ *
+ */
+TrexRpcCommand::rpc_cmd_rc_e
+TrexRpcCmdGetStatus::_run(const Json::Value &params, Json::Value &result) {
+
+ /* validate count */
+ if (params.size() != 0) {
+ return (TrexRpcCommand::RPC_CMD_PARAM_COUNT_ERR);
+ }
+
+ Json::Value &section = result["result"];
+
+ section["general"]["version"] = VERSION_BUILD_NUM;
+ section["general"]["build_date"] = get_build_date();
+ section["general"]["build_time"] = get_build_time();
+ section["general"]["version_user"] = VERSION_USER;
+
+ return (RPC_CMD_OK);
+}
+
diff --git a/src/rpc-server/src/commands/trex_rpc_cmds.h b/src/rpc-server/src/commands/trex_rpc_cmds.h
index 44b72c4c..0778b75d 100644
--- a/src/rpc-server/src/commands/trex_rpc_cmds.h
+++ b/src/rpc-server/src/commands/trex_rpc_cmds.h
@@ -28,6 +28,11 @@ limitations under the License.
/* all the RPC commands decl. goes here */
/******************* test section ************/
+
+/**
+ * add
+ *
+ */
class TrexRpcCmdTestAdd : public TrexRpcCommand {
public:
TrexRpcCmdTestAdd() : TrexRpcCommand("rpc_test_add") {}
@@ -35,6 +40,10 @@ protected:
virtual rpc_cmd_rc_e _run(const Json::Value &params, Json::Value &result);
};
+/**
+ * sub
+ *
+ */
class TrexRpcCmdTestSub : public TrexRpcCommand {
public:
TrexRpcCmdTestSub() : TrexRpcCommand("rpc_test_sub") {} ;
@@ -42,6 +51,10 @@ protected:
virtual rpc_cmd_rc_e _run(const Json::Value &params, Json::Value &result);
};
+/**
+ * ping
+ *
+ */
class TrexRpcCmdPing : public TrexRpcCommand {
public:
TrexRpcCmdPing() : TrexRpcCommand("rpc_ping") {};
@@ -49,6 +62,10 @@ protected:
virtual rpc_cmd_rc_e _run(const Json::Value &params, Json::Value &result);
};
+/**
+ * get all registered commands
+ *
+ */
class TrexRpcCmdGetReg : public TrexRpcCommand {
public:
TrexRpcCmdGetReg() : TrexRpcCommand("rpc_get_reg_cmds") {};
@@ -56,5 +73,17 @@ protected:
virtual rpc_cmd_rc_e _run(const Json::Value &params, Json::Value &result);
};
+/**
+ * get status
+ *
+ */
+class TrexRpcCmdGetStatus : public TrexRpcCommand {
+public:
+ TrexRpcCmdGetStatus() : TrexRpcCommand("rpc_get_status") {};
+protected:
+ virtual rpc_cmd_rc_e _run(const Json::Value &params, Json::Value &result);
+};
+
+
/**************** test section end *************/
#endif /* __TREX_RPC_CMD_H__ */
diff --git a/src/rpc-server/src/trex_rpc_cmds_table.cpp b/src/rpc-server/src/trex_rpc_cmds_table.cpp
index 7635f14f..04a56389 100644
--- a/src/rpc-server/src/trex_rpc_cmds_table.cpp
+++ b/src/rpc-server/src/trex_rpc_cmds_table.cpp
@@ -32,6 +32,7 @@ TrexRpcCommandsTable::TrexRpcCommandsTable() {
register_command(new TrexRpcCmdTestSub());
register_command(new TrexRpcCmdPing());
register_command(new TrexRpcCmdGetReg());
+ register_command(new TrexRpcCmdGetStatus());
}
TrexRpcCommandsTable::~TrexRpcCommandsTable() {
@@ -41,7 +42,13 @@ TrexRpcCommandsTable::~TrexRpcCommandsTable() {
}
TrexRpcCommand * TrexRpcCommandsTable::lookup(const string &method_name) {
- return m_rpc_cmd_table[method_name];
+ auto search = m_rpc_cmd_table.find(method_name);
+
+ if (search != m_rpc_cmd_table.end()) {
+ return search->second;
+ } else {
+ return NULL;
+ }
}