102 lines
2.9 KiB
Python
102 lines
2.9 KiB
Python
import subprocess
|
|
import json
|
|
import re
|
|
|
|
|
|
class Batman(object):
|
|
"""
|
|
Bindings for B.A.T.M.A.N. Advanced
|
|
commandline interface "batctl"
|
|
"""
|
|
def __init__(self, mesh_interface='bat0', alfred_sockpath=None):
|
|
self.mesh_interface = mesh_interface
|
|
self.alfred_sock = alfred_sockpath
|
|
|
|
def vis_data(self, batadv_vis=False):
|
|
vds = self.vis_data_batctl_legacy()
|
|
if batadv_vis:
|
|
vds += self.vis_data_batadv_vis()
|
|
return vds
|
|
|
|
@staticmethod
|
|
def vis_data_helper(lines):
|
|
vd_tmp = []
|
|
for line in lines:
|
|
try:
|
|
utf8_line = line.decode('utf-8')
|
|
vd_tmp.append(json.loads(utf8_line))
|
|
except UnicodeDecodeError:
|
|
pass
|
|
return vd_tmp
|
|
|
|
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()
|
|
vds = self.vis_data_helper(lines)
|
|
return vds
|
|
|
|
def vis_data_batadv_vis(self):
|
|
"""
|
|
Parse "batadv-vis -i <mesh_interface> -f json"
|
|
into an array of dictionaries.
|
|
"""
|
|
cmd = ['batadv-vis', '-i', self.mesh_interface, '-f', 'json']
|
|
if self.alfred_sock:
|
|
cmd.extend(['-u', self.alfred_sock])
|
|
output = subprocess.check_output(cmd)
|
|
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')
|
|
lines = output_utf8.splitlines()
|
|
|
|
own_mac = re.match(r"^.*MainIF/MAC: [^/]+/([0-9a-f:]+).*$",
|
|
lines[0]).group(1)
|
|
|
|
gateways = []
|
|
gw_mode = self.gateway_mode()
|
|
if gw_mode['mode'] == 'server':
|
|
gateways.append(own_mac)
|
|
|
|
for line in lines:
|
|
gw_line = re.match(r"^(?:=>)? +([0-9a-f:]+) ", line)
|
|
if gw_line:
|
|
gateways.append(gw_line.group(1))
|
|
|
|
return gateways
|
|
|
|
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())
|