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
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import cmd
import json
import ast
import argparse
import random
import string
import sys
import trex_root_path
from client_utils.jsonrpc_client import TrexStatelessClient
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_force_acquire (self, line):
'''Acquires ports by force\n'''
self.do_acquire(line, True)
def parse_ports_from_line (self, line):
port_list = set()
if line:
for port_id in line.split(' '):
if (not port_id.isdigit()) or (int(port_id) < 0) or (int(port_id) >= self.rpc_client.get_port_count()):
print "Please provide a list of ports seperated by spaces between 0 and {0}".format(self.rpc_client.get_port_count() - 1)
return None
port_list.add(int(port_id))
port_list = list(port_list)
else:
port_list = [i for i in xrange(0, self.rpc_client.get_port_count())]
return port_list
def do_acquire (self, line, force = False):
'''Acquire ports\n'''
port_list = self.parse_ports_from_line(line)
if not port_list:
return
print "\nTrying to acquire ports: " + (" ".join(str(x) for x in port_list)) + "\n"
rc, resp_list = self.rpc_client.take_ownership(port_list, force)
if not rc:
print "\n*** " + resp_list + "\n"
return
for i, rc in enumerate(resp_list):
if rc[0]:
print "Port {0} - Acquired".format(port_list[i])
else:
print "Port {0} - ".format(port_list[i]) + rc[1]
print "\n"
def do_release (self, line):
'''Release ports\n'''
if line:
port_list = self.parse_ports_from_line(line)
else:
port_list = self.rpc_client.get_owned_ports()
if not port_list:
return
rc, resp_list = self.rpc_client.release_ports(port_list)
print "\n"
for i, rc in enumerate(resp_list):
if rc[0]:
print "Port {0} - Released".format(port_list[i])
else:
print "Port {0} - Failed to release port, probably not owned by you or port is under traffic"
print "\n"
def do_get_port_stats (self, line):
'''Get ports stats\n'''
port_list = self.parse_ports_from_line(line)
if not port_list:
return
rc, resp_list = self.rpc_client.get_port_stats(port_list)
if not rc:
print "\n*** " + resp_list + "\n"
return
for i, rc in enumerate(resp_list):
if rc[0]:
print "\nPort {0} stats:\n{1}\n".format(port_list[i], self.rpc_client.pretty_json(json.dumps(rc[1])))
else:
print "\nPort {0} - ".format(i) + rc[1] + "\n"
print "\n"
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
self.supported_rpc = self.rpc_client.get_supported_cmds()
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" + self.rpc_client.pretty_json(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 do_whoami (self, line):
'''Prints console user name\n'''
print "\n" + self.rpc_client.whoami() + "\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)
# do
#def do_snapshot (self, line):
#for key, value in self.rpc_client.snapshot()[1]['streams'].iteritems():
#print str(key) + " " + str(value)
# 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)
parser.add_argument("-u", "--user", help = "User Name [default is random generated]\n",
default = 'user_' + ''.join(random.choice(string.digits) for _ in range(5)),
type = str)
return parser
def main ():
parser = setParserOptions()
options = parser.parse_args(sys.argv[1:])
# RPC client
rpc_client = TrexStatelessClient(options.server, options.port, options.user)
# 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()
|