github-starred/burp-ui
2
0
Fork 0
mirror of https://github.com/ziirish/burp-ui.git synced 2026-05-21 06:45:24 -06:00
Code Issues 1 Projects Packages Wiki Activity Actions

fix: forward errors and empty results

Browse source
This commit is contained in:
ziirish 2015-12-06 19:03:48 +01:00
parent 097e592dc4
commit 187f036b14
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
burpui/api/servers.py
View file

@ -72,14 +72,17 @@ class ServersStats(Resource):
'clients': len(api.bui.cli.servers[serv].get_all_clients(serv)),
'alive': api.bui.cli.servers[serv].ping()
})
return
else:
output.put({
'name': serv,
'clients': len(api.bui.cli.servers[serv].get_all_clients(serv)),
'alive': api.bui.cli.servers[serv].ping()
})
return
output.put(None)
except BUIserverException as e:
api.abort(500, str(e))
output.put(str(e))
output = multiprocessing.Queue()
pools = [multiprocessing.Process(target=get_server_infos, args=(s, output)) for s in api.bui.cli.servers]
@ -89,6 +92,11 @@ class ServersStats(Resource):
for p in pools:
p.join()
r = [output.get() for p in pools]
for p in pools:
tmp = output.get()
if tmp and isinstance(tmp, dict):
r.append(tmp)
elif tmp:
api.abort(500, tmp)
return r