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
|
#!/usr/bin/env python
import socket
from mprpc import RPCClient
import usrp_mpm as mpm
import argparse
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")
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:
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(4096)
print("Received respons from: {}".format(sender[0]))
print("Dicovery data: {}".format(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)
if __name__ == "__main__":
exit(not(main()))
|
