mirror of
https://github.com/rtbrick/bngblaster.git
synced 2024-05-06 15:54:57 +00:00
64 lines
1.6 KiB
Python
64 lines
1.6 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
import sys
|
||
|
|
import socket
|
||
|
|
import os
|
||
|
|
import json
|
||
|
|
|
||
|
|
def usage():
|
||
|
|
print("""
|
||
|
|
BNG Blaster Control Socket Client
|
||
|
|
|
||
|
|
{c} <socket> <command> [arguments]
|
||
|
|
|
||
|
|
Examples:
|
||
|
|
{c} run.sock session-info outer-vlan 1 inner-vlan 1
|
||
|
|
{c} run.sock igmp-join outer-vlan 1 inner-vlan 1 group 239.0.0.1 source1 1.1.1.1 source2 2.2.2.2 source3 3.3.3.3
|
||
|
|
{c} run.sock igmp-info outer-vlan 1 inner-vlan 1
|
||
|
|
""".format(c=sys.argv[0]))
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
def error(*args, **kwargs):
|
||
|
|
print(*args, file=sys.stderr, **kwargs)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
request = {}
|
||
|
|
if(len(sys.argv)) < 3:
|
||
|
|
usage()
|
||
|
|
|
||
|
|
socket_path = sys.argv[1]
|
||
|
|
|
||
|
|
request["command"] = sys.argv[2]
|
||
|
|
if(len(sys.argv)) > 4:
|
||
|
|
request["arguments"] = {}
|
||
|
|
for i in range(3, len(sys.argv), 2):
|
||
|
|
try:
|
||
|
|
request["arguments"][sys.argv[i]] = int(sys.argv[i+1])
|
||
|
|
except:
|
||
|
|
request["arguments"][sys.argv[i]] = sys.argv[i+1]
|
||
|
|
|
||
|
|
if os.path.exists(socket_path):
|
||
|
|
client = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
||
|
|
try:
|
||
|
|
client.connect(socket_path)
|
||
|
|
client.send(json.dumps(request).encode('utf-8'))
|
||
|
|
|
||
|
|
data = ""
|
||
|
|
while True:
|
||
|
|
junk = client.recv(1024)
|
||
|
|
if junk:
|
||
|
|
data += junk.decode('utf-8')
|
||
|
|
else:
|
||
|
|
break
|
||
|
|
print(json.dumps(json.loads(data), indent=4))
|
||
|
|
except Exception as e:
|
||
|
|
error(e)
|
||
|
|
finally:
|
||
|
|
client.close()
|
||
|
|
else:
|
||
|
|
error("socket %s not found" % socket_path)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|