ntopng/scripts/lua/iface_ports_list.lua
Arianna Avanzini 86ab3f0585 src/Lua.cpp: finally add minimal SQL interface for Lua
This commit adds a Lua interface that allows Lua scripts to directly
query flows with a simple SQL-like statement. This also enforces the
checks and data conversions in the iface_ports_list.lua script.
2015-06-19 16:48:09 +02:00

99 lines
2.2 KiB
Lua

--
-- (C) 2013-15 - ntop.org
--
dirs = ntop.getDirs()
package.path = dirs.installdir .. "/scripts/lua/modules/?.lua;" .. package.path
require "lua_utils"
sendHTTPHeader('text/html; charset=iso-8859-1')
interface.select(ifname)
host = _GET["host"]
client_ports = nil
server_ports = nil
function fill_ports_array(field_key, flows_stats, host)
local ports_array = {}
for key, value in pairs(flows_stats) do
if ((host == nil) or (flows_stats[key][field_key..".ip"] == host)) then
p = flows_stats[key][field_key..".port"]
if(ports_array[p] == nil) then ports_array[p] = 0 end
ports_array[p] = ports_array[p] + flows_stats[key]["bytes"]
end
end
return ports_array
end
if (host == nil) then
flows_stats = interface.queryFlowsInfo("SELECT PORTS FROM FLOWS")
else
host_table = hostkey2hostinfo(host)
host = host_table["host"]
vlan = host_table["vlan"] or 0
flows_stats = interface.queryFlowsInfo("SELECT PORTS FROM FLOWS WHERE host = "..host.." AND vlan = "..tostring(vlan))
end
client_ports = fill_ports_array("cli", flows_stats, host)
server_ports = fill_ports_array("srv", flows_stats, host)
if(_GET["mode"] == "server") then
ports = server_ports
else
ports = client_ports
end
_ports = { }
tot = 0
for k,v in pairs(ports) do
_ports[v] = k
tot = tot + v
end
threshold = (tot * 5) / 100
print "[ "
num = 0
accumulate = 0
for key, value in pairsByKeys(_ports, rev) do
if(key < threshold) then
break
end
if(num > 0) then
print ",\n"
end
print("\t { \"label\": \"" .. value .."\", \"value\": ".. key ..", \"url\": \""..ntop.getHttpPrefix().."/lua/port_details.lua?port="..value)
if(host ~= nil) then
print("&host="..host)
end
print("\" }")
accumulate = accumulate + key
num = num + 1
if(num == max_num_entries) then
break
end
end
-- In case there is some leftover do print it as "Other"
if(accumulate < tot) then
if(num > 0) then
print (",\n")
end
print("\t { \"label\": \"Other\", \"value\": ".. (tot-accumulate) .." }")
end
if(tot == 0) then
print("\t { \"label\": \"Other\", \"value\": ".. 0 .." }")
end
print "\n]"