ntopng/scripts/lua/if_stats_ndpi.lua
2018-09-19 17:54:28 +02:00

120 lines
3.2 KiB
Lua

--
-- (C) 2013-18 - ntop.org
--
dirs = ntop.getDirs()
package.path = dirs.installdir .. "/scripts/lua/modules/?.lua;" .. package.path
require "lua_utils"
require "graph_utils"
local ts_utils = require("ts_utils")
local ifid = _GET["ifid"]
if ifid ~= nil and ifid ~= "" then
if_name = getInterfaceName(ifid)
else
if_name = ifname
ifid = interface.name2id(ifname)
end
interface.select(if_name)
local ifstats = interface.getStats()
local format = _GET["format"]
if(format == "json") then
sendHTTPHeader('application/json')
json_format = true
else
sendHTTPContentTypeHeader('text/html')
json_format = false
end
-- Add ARP to stats
if(ifstats.stats ~= nil) then
local arp = { }
arp["bytes.sent"] = 0
arp["bytes.rcvd"] = ifstats.eth["ARP_bytes"]
arp["packets.sent"] = 0
arp["packets.rcvd"] = ifstats.eth["ARP_packets"]
arp.breed = "Unrated"
ifstats["ndpi"]["ARP"] = arp
if(ifstats["ndpi"]["Unknown"] ~= nil) then
ifstats["ndpi"]["Unknown"]["bytes.rcvd"] = ifstats["ndpi"]["Unknown"]["bytes.rcvd"] - ifstats.eth["ARP_bytes"]
ifstats["ndpi"]["Unknown"]["packets.rcvd"] = ifstats["ndpi"]["Unknown"]["packets.rcvd"] - ifstats.eth["ARP_packets"]
end
end
local total = ifstats.stats.bytes
local vals = {}
for k, v in pairs(ifstats["ndpi"]) do
-- io.write("->"..k.."\n")
if v["bytes.rcvd"] > 0 or v["bytes.sent"] > 0 then
vals[k] = k
end
end
table.sort(vals)
if(json_format) then print('[\n') end
local num = 0
for _k in pairsByKeys(vals, asc) do
k = vals[_k]
if(not(json_format)) then
print('<tr id="t_protocol_'..k..'">')
print('<th style="width: 33%;">')
else
if(num > 0) then
print(',\n')
end
end
local proto_cache_key = "ntopng.cache.has_ndpi_" .. ifid.."_".. k
local has_ndpi_proto = ntop.getCache(proto_cache_key)
if has_ndpi_proto ~= "1" then
if ts_utils.exists("iface:ndpi", {ifid=ifid, protocol=k}) then
has_ndpi_proto = "1"
ntop.setCache(proto_cache_key, "1", 300)
end
end
if(has_ndpi_proto == "1") then
if(not(json_format)) then
print("<A HREF=\""..ntop.getHttpPrefix().."/lua/if_stats.lua?ifid=" .. ifid .. "&page=historical&ts_schema=iface:ndpi&protocol=".. k .."\">".. k .." "..formatBreed(ifstats["ndpi"][k]["breed"]).."</A>")
else
print('{ "proto": "'..k..'", "breed": "'..ifstats["ndpi"][k]["breed"]..'", ')
end
else
if(not(json_format)) then
print(k.." "..formatBreed(ifstats["ndpi"][k]["breed"]))
else
print('{ "proto": "'..k..'", ')
end
end
t = ifstats["ndpi"][k]["bytes.sent"]+ifstats["ndpi"][k]["bytes.rcvd"]
if(not(json_format)) then
if(k ~= "ARP") then print(" <A HREF=\""..ntop.getHttpPrefix().."/lua/flows_stats.lua?application="..k.."\"><i class=\"fa fa-search-plus\"></i></A>") end
print("</th><td class=\"text-right\" style=\"width: 20%;\">" ..bytesToSize(t).. "</td>")
print("<td ><span style=\"width: 60%; float: left;\">")
percentageBar(total, t, "") -- k
-- print("</td>\n")
print("</span><span style=\"width: 40%; margin-left: 15px;\" >" ..round((t * 100)/total, 1).. " %</span></td></tr>\n")
else
print('"bytes": '..tostring(t)..' }')
end
num = num + 1
end
if(json_format) then print('\n]\n') end