79 lines
2.4 KiB
Python
Executable File
79 lines
2.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import subprocess
|
|
import json
|
|
import re
|
|
|
|
class batman:
|
|
""" Bindings for B.A.T.M.A.N. advanced batctl tool
|
|
"""
|
|
def __init__(self, mesh_interface = "bat0"):
|
|
self.mesh_interface = mesh_interface
|
|
|
|
def vis_data(self):
|
|
return self.vis_data_batadv_vis() + self.vis_data_batctl_legacy()
|
|
|
|
def vis_data_helper(self,lines):
|
|
vd = []
|
|
for line in lines:
|
|
try:
|
|
utf8_line = line.decode("utf-8")
|
|
vd.append(json.loads(utf8_line))
|
|
except e:
|
|
pass
|
|
return vd
|
|
|
|
def vis_data_batctl_legacy(self):
|
|
""" Parse "batctl -m <mesh_interface> vd json -n" into an array of dictionaries.
|
|
"""
|
|
output = subprocess.check_output(["batctl","-m",self.mesh_interface,"vd","json","-n"])
|
|
lines = output.splitlines()
|
|
return self.vis_data_helper(lines)
|
|
|
|
def vis_data_batadv_vis(self):
|
|
""" Parse "batadv-vis -i <mesh_interface> -f json" into an array of dictionaries.
|
|
"""
|
|
output = subprocess.check_output(["batadv-vis","-i",self.mesh_interface,"-f","json"])
|
|
lines = output.splitlines()
|
|
return self.vis_data_helper(lines)
|
|
|
|
def gateway_list(self):
|
|
""" Parse "batctl -m <mesh_interface> gwl -n" into an array of dictionaries.
|
|
"""
|
|
output = subprocess.check_output(["batctl","-m",self.mesh_interface,"gwl","-n"])
|
|
output_utf8 = output.decode("utf-8")
|
|
# TODO Parse information
|
|
lines = output_utf8.splitlines()
|
|
own_mac = re.match(r"^.*MainIF/MAC: [^/]+/([0-9a-f:]+).*$",lines[0]).group(1)
|
|
# Remove header line
|
|
del lines[0]
|
|
# Fill gateway list
|
|
gw = []
|
|
gw_mode = self.gateway_mode()
|
|
if gw_mode['mode'] == 'server':
|
|
gw.append({'mac': own_mac, 'bandwidth': gw_mode['bandwidth']})
|
|
for line in lines:
|
|
gw_line = line.split()
|
|
# When in client gateway mode maybe gw_line[0] is not the right.
|
|
gw.append({'mac':gw_line[0], 'bandwidth': gw_line[-1]})
|
|
return gw
|
|
|
|
def gateway_mode(self):
|
|
""" Parse "batctl -m <mesh_interface> gw"
|
|
"""
|
|
output = subprocess.check_output(["batctl","-m",self.mesh_interface,"gw"])
|
|
elements = output.decode("utf-8").split()
|
|
mode = elements[0]
|
|
if mode == "server":
|
|
return {'mode': 'server', 'bandwidth': elements[3]}
|
|
else:
|
|
return {'mode': mode}
|
|
|
|
if __name__ == "__main__":
|
|
bc = batman()
|
|
vd = bc.vis_data()
|
|
gw = bc.gateway_list()
|
|
for x in vd:
|
|
print(x)
|
|
print(gw)
|
|
print(bc.gateway_mode())
|