blob: 404a2ee0763a36f1da25324346a0925403341785 (
plain)
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
|
#!/usr/bin/python
# -*- coding: utf-8 -*-
import cmd
from trex_rpc_client import RpcClient
class TrexConsole(cmd.Cmd):
"""Trex Console"""
def __init__(self, rpc_client):
cmd.Cmd.__init__(self)
self.prompt = "TRex > "
self.intro = "\n-=TRex Console V1.0=-\n"
self.rpc_client = rpc_client
self.verbose = 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)]
def do_verbose (self, line):
'''shows or set verbose mode\nusage: verbose [on/off]\n'''
if line == "":
print "verbose is " + ("on" if self.verbose else "off")
elif line == "on":
self.verbose = True
self.rpc_client.set_verbose(True)
print "verbose set to on\n"
elif line == "off":
self.verbose = False
self.rpc_client.set_verbose(False)
print "verbose set to off\n"
else:
print "please specify 'on' or 'off'\n"
def do_query_server(self, line):
'''\nquery the RPC server for supported remote commands\n'''
rc, msg = self.rpc_client.query_rpc_server()
if not rc:
print "\n*** Failed to query RPC server: " + str(msg)
print "RPC server supports the following commands: \n\n" + msg
def do_ping (self, line):
'''pings the RPC server \n'''
print "Pinging RPC server"
rc, msg = self.rpc_client.ping_rpc_server()
if rc:
print "[SUCCESS]\n"
else:
print "[FAILED]\n"
def do_quit(self, line):
'''\nexit the client\n'''
return True
def default(self, line):
print "'{0}' is an unrecognized command\n".format(line)
# aliasing
do_EOF = do_q = do_quit
def main ():
# RPC client
try:
rpc_client = RpcClient(5050)
rpc_client.connect()
except Exception as e:
print "\n*** " + str(e) + "\n"
exit(-1)
# console
console = TrexConsole(rpc_client)
console.cmdloop()
if __name__ == '__main__':
main()
|
