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
|
#!/usr/bin/env python
import socket
from mprpc import RPCClient
import usrp_mpm as mpm
import argparse
import random
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument("-a", "--address", default="0.0.0.0", type=str, help="Destination address")
parser.add_argument("-p", "--port", default=0, type=int, help="Destination port")
sub_parsers = parser.add_subparsers(dest="command")
rpc_parser = sub_parsers.add_parser("rpc", help="Issue RPC")
rpc_parser.add_argument("-c", "--call", required=True, help="command to issue")
rpc_parser.add_argument("arguments", nargs="*")
disc_parser = sub_parsers.add_parser("disc", help="Issue discovery")
echo_parser = sub_parsers.add_parser("echo", help="Issue UDP echo")
return parser.parse_args()
def rpc(address, port, command, *args):
if not port:
port = mpm.types.MPM_RPC_PORT
client = RPCClient(address, port)
if args:
args = [eval(arg.lstrip("=")) if arg.startswith("=") else arg for arg in args]
result = client.call(command, *args)
else:
result = client.call(command)
return result
def discovery(address, port):
if not port:
port = mpm.types.MPM_DISCOVERY_PORT
sock = socket.socket(
socket.AF_INET,
socket.SOCK_DGRAM)
sock.sendto(mpm.types.MPM_DISCOVERY_MESSAGE, (address, port))
sock.settimeout(1.0) # wait max 1 second
while True:
try:
data, sender = sock.recvfrom(8000)
print("Received response from: {}".format(sender[0]))
print("Dicovery data: {}".format(data))
except:
break
def echo(address, port):
if not port:
port = mpm.types.MPM_DISCOVERY_PORT
sock = socket.socket(
socket.AF_INET,
socket.SOCK_DGRAM)
message = "MPM-ECHO" + bytearray(random.getrandbits(8) for _ in xrange(8000-8))
sock.sendto(message, (address, port))
sock.settimeout(0.05) # wait max 50 ms
while True:
try:
data, sender = sock.recvfrom(9000)
print("Received response from: {}".format(sender[0]))
print("Echo data size: {}".format(len(data)))
except:
break
def main():
args = parse_args()
if args.command == "rpc":
if args.arguments:
result = rpc(args.address, args.port, args.call, *args.arguments)
else:
result = rpc(args.address, args.port, args.call)
print(result)
elif args.command == "disc":
discovery(args.address, args.port)
result = True
elif args.command == "echo":
echo(args.address, args.port)
result = True
return result
if __name__ == "__main__":
exit(not(main()))
|
