1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import cmd
import json
import ast
import argparse
import sys
from trex_rpc_client import RpcClient
import trex_status
class TrexConsole(cmd.Cmd):
"""Trex Console"""
def __init__(self, rpc_client):
cmd.Cmd.__init__(self)
self.rpc_client = rpc_client
self.do_connect("")
self.intro = "\n-=TRex Console V1.0=-\n"
self.intro += "\nType 'help' or '?' for supported actions\n"
self.verbose = False
self.postcmd(False, "")
# 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\n'''
if line == "":
print "\nverbose is " + ("on\n" if self.verbose else "off\n")
elif line == "on":
self.verbose = True
self.rpc_client.set_verbose(True)
print "\nverbose set to on\n"
elif line == "off":
self.verbose = False
self.rpc_client.set_verbose(False)
print "\nverbose set to off\n"
else:
print "\nplease specify 'on' or 'off'\n"
# query the server for registered commands
def do_query_server(self, line):
'''query the RPC server for supported remote commands\n'''
rc, msg = self.rpc_client.query_rpc_server()
if not rc:
print "\n*** " + msg + "\n"
return
print "\nRPC server supports the following commands: \n\n"
for func in msg:
if func:
print func
print "\n"
def do_ping (self, line):
'''Pings the RPC server\n'''
print "\n-> Pinging RPC server"
rc, msg = self.rpc_client.ping_rpc_server()
if rc:
print "[SUCCESS]\n"
else:
print "\n*** " + msg + "\n"
return
def do_connect (self, line):
'''Connects to the server\n'''
if line == "":
rc, msg = self.rpc_client.connect()
else:
sp = line.split()
if (len(sp) != 2):
print "\n[usage] connect [server] [port] or without parameters\n"
return
rc, msg = self.rpc_client.connect(sp[0], sp[1])
if rc:
print "[SUCCESS]\n"
else:
print "\n*** " + msg + "\n"
return
rc, msg = self.rpc_client.query_rpc_server()
if rc:
self.supported_rpc = [str(x) for x in msg if x]
def do_rpc (self, line):
'''Launches a RPC on the server\n'''
if line == "":
print "\nUsage: [method name] [param dict as string]\n"
print "Example: rpc test_add {'x': 12, 'y': 17}\n"
return
sp = line.split(' ', 1)
method = sp[0]
params = None
bad_parse = False
if len(sp) > 1:
try:
params = ast.literal_eval(sp[1])
if not isinstance(params, dict):
bad_parse = True
except ValueError as e1:
bad_parse = True
except SyntaxError as e2:
bad_parse = True
if bad_parse:
print "\nValue should be a valid dict: '{0}'".format(sp[1])
print "\nUsage: [method name] [param dict as string]\n"
print "Example: rpc test_add {'x': 12, 'y': 17}\n"
return
rc, msg = self.rpc_client.invoke_rpc_method(method, params)
if rc:
print "\nServer Response:\n\n" + json.dumps(msg) + "\n"
else:
print "\n*** " + msg + "\n"
#print "Please try 'reconnect' to reconnect to server"
def complete_rpc (self, text, line, begidx, endidx):
return [x for x in self.supported_rpc if x.startswith(text)]
def do_status (self, line):
'''Shows a graphical console\n'''
self.do_verbose('off')
trex_status.show_trex_status(self.rpc_client)
def do_quit(self, line):
'''exit the client\n'''
return True
def do_disconnect (self, line):
'''Disconnect from the server\n'''
if not self.rpc_client.is_connected():
print "Not connected to server\n"
return
rc, msg = self.rpc_client.disconnect()
if rc:
print "[SUCCESS]\n"
else:
print msg + "\n"
def postcmd(self, stop, line):
if self.rpc_client.is_connected():
self.prompt = "TRex > "
else:
self.supported_rpc = None
self.prompt = "TRex (offline) > "
return stop
def default(self, line):
print "'{0}' is an unrecognized command. type 'help' or '?' for a list\n".format(line)
def do_help (self, line):
'''Shows This Help Screen\n'''
if line:
try:
func = getattr(self, 'help_' + line)
except AttributeError:
try:
doc = getattr(self, 'do_' + line).__doc__
if doc:
self.stdout.write("%s\n"%str(doc))
return
except AttributeError:
pass
self.stdout.write("%s\n"%str(self.nohelp % (line,)))
return
func()
return
print "\nSupported Console Commands:"
print "----------------------------\n"
cmds = [x[3:] for x in self.get_names() if x.startswith("do_")]
for cmd in cmds:
if cmd == "EOF":
continue
try:
doc = getattr(self, 'do_' + cmd).__doc__
if doc:
help = str(doc)
else:
help = "*** Undocumented Function ***\n"
except AttributeError:
help = "*** Undocumented Function ***\n"
print "{:<30} {:<30}".format(cmd + " - ", help)
# aliasing
do_exit = do_EOF = do_q = do_quit
def setParserOptions ():
parser = argparse.ArgumentParser(prog="trex_console.py")
parser.add_argument("-s", "--server", help = "T-Rex Server [default is localhost]",
default = "localhost",
type = str)
parser.add_argument("-p", "--port", help = "T-Rex Server Port [default is 5050]\n",
default = 5050,
type = int)
return parser
def main ():
parser = setParserOptions()
options = parser.parse_args(sys.argv[1:])
# RPC client
rpc_client = RpcClient(options.server, options.port)
# console
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()
|
