mirror of
https://github.com/ntop/ntopng.git
synced 2026-05-06 03:45:26 +00:00
72 lines
2.6 KiB
Lua
72 lines
2.6 KiB
Lua
--
|
|
-- (C) 2019 - ntop.org
|
|
--
|
|
|
|
local dirs = ntop.getDirs()
|
|
package.path = dirs.installdir .. "/scripts/lua/modules/?.lua;" .. package.path
|
|
require "lua_utils"
|
|
require "alert_utils"
|
|
|
|
local alerts_api = require("alerts_api")
|
|
local alert_consts = require("alert_consts")
|
|
|
|
local do_trace = false
|
|
local config_alerts_local = nil
|
|
local config_alerts_remote = nil
|
|
local available_modules = nil
|
|
local ifid = nil
|
|
|
|
-- #################################################################
|
|
|
|
-- The function below ia called once (#pragma once)
|
|
function setup(str_granularity)
|
|
if(do_trace) then print("alert.lua:setup("..str_granularity..") called\n") end
|
|
ifid = interface.getId()
|
|
local ifname = interface.setActiveInterfaceId(ifid)
|
|
|
|
-- Load the threshold checking functions
|
|
available_modules = alerts_api.load_check_modules("host", str_granularity)
|
|
|
|
config_alerts_local = getLocalHostsConfiguredAlertThresholds(ifname, str_granularity, available_modules)
|
|
config_alerts_remote = getRemoteHostsConfiguredAlertThresholds(ifname, str_granularity, available_modules)
|
|
end
|
|
|
|
-- #################################################################
|
|
|
|
-- The function below is called once per host
|
|
function checkAlerts(granularity)
|
|
local info = host.getFullInfo()
|
|
local host_key = hostinfo2hostkey({ip = info.ip, vlan = info.vlan}, nil, true --[[ force @[vlan] even when vlan is 0 --]])
|
|
local config_alerts = ternary(info["localhost"], config_alerts_local, config_alerts_remote)
|
|
local host_config = config_alerts[host_key] or {}
|
|
local global_config = ternary(info["localhost"], config_alerts["local_hosts"], config_alerts["remote_hosts"]) or {}
|
|
local has_configured_alerts = (table.len(host_config) or table.len(global_config))
|
|
local entity_info = alerts_api.hostAlertEntity(info.ip, info.vlan)
|
|
|
|
if has_configured_alerts then
|
|
for _, check in pairs(available_modules) do
|
|
local config = host_config[check.key] or global_config[check.key]
|
|
|
|
if config or check.always_enabled then
|
|
check.check_function({
|
|
granularity = granularity,
|
|
alert_entity = entity_info,
|
|
entity_info = info,
|
|
alert_config = config,
|
|
check_module = check,
|
|
})
|
|
end
|
|
end
|
|
end
|
|
|
|
alerts_api.releaseEntityAlerts(entity_info, host.getExpiredAlerts(granularity2id(granularity)))
|
|
end
|
|
|
|
-- #################################################################
|
|
|
|
function releaseAlerts(granularity)
|
|
local info = host.getFullInfo()
|
|
local entity_info = alerts_api.hostAlertEntity(info.ip, info.vlan)
|
|
|
|
alerts_api.releaseEntityAlerts(entity_info, host.getAlerts(granularity))
|
|
end
|