def setHostScan():
defaultConfig = es.get_scan_config()
value = cleanPostData(request.json.get("scan_hosts"))
hostsList, ports = (value, defaultConfig["scanPorts"])
if api_hostScan(hostsList=hostsList, ports=ports):
return jsonify({"message": "ok"})
else:
return jsonify({"message": "error"})
def AllConfig():
defaultConfig = es.get_scan_config()
scanHosts = defaultConfig["scanHosts"].split(",")
if request.json is None:
return jsonify({
"scanHosts": scanHosts,
"scanPeriod": defaultConfig["scanPeriod"],
"scanThreads": defaultConfig["scanThreads"],
"scanCMS": defaultConfig["scanCMS"],
"scanPorts": defaultConfig["scanPorts"]
})
else:
scanHosts = cleanPostData(request.json.get('scanHosts'))
scanPeriod = cleanPostData(request.json.get('scanPeriod'))
scanPorts = cleanPostData(request.json.get('scanPorts'))
es.update_scan_config("scanHosts", scanHosts)
es.update_scan_config("scanPeriod", scanPeriod)
es.update_scan_config("scanPorts", scanPorts)
return jsonify({"message": "ok"})
def VulTaskSearch():
value = cleanPostData(request.json.get('searchValue'))
try:
searchGroup = []
for item in value.split(";"):
(n, v) = item.split(":")
searchGroup.append({"match": {n: v}})
rg = es.get_assets_doc(searchGroup)
searchResult = []
if rg:
count = 0
for v in rg:
searchResult.append(v["_source"])
count += 1
return jsonify({"searchResult": searchResult, "count": count})
except:
pass