mirror of
https://github.com/ntop/ntopng.git
synced 2026-04-28 23:19:33 +00:00
123 lines
3.6 KiB
Lua
123 lines
3.6 KiB
Lua
--
|
|
-- (C) 2020 - ntop.org
|
|
--
|
|
|
|
local dirs = ntop.getDirs()
|
|
package.path = dirs.installdir .. "/scripts/lua/modules/?.lua;" .. package.path
|
|
package.path = dirs.installdir .. "/scripts/lua/modules/import_export/?.lua;" .. package.path
|
|
package.path = dirs.installdir .. "/scripts/lua/modules/vulnerability_scan/?.lua;" .. package.path
|
|
|
|
require "lua_utils"
|
|
local import_export = require "import_export"
|
|
local host_pools = require "host_pools"
|
|
local am_utils = require "am_utils"
|
|
local vs_utils = require "vs_utils"
|
|
|
|
-- ##############################################
|
|
|
|
local am_import_export = {}
|
|
|
|
-- ##############################################
|
|
|
|
function am_import_export:create(args)
|
|
-- Instance of the base class
|
|
local _am_import_export = import_export:create()
|
|
|
|
-- Subclass using the base class instance
|
|
self.key = "am"
|
|
-- self is passed as argument so it will be set as base class metatable
|
|
-- and this will actually make it possible to override functions
|
|
local _am_import_export_instance = _am_import_export:create(self)
|
|
|
|
-- Compute
|
|
|
|
-- Return the instance
|
|
return _am_import_export_instance
|
|
end
|
|
|
|
-- ##############################################
|
|
|
|
-- @brief Import configuration
|
|
-- @param conf The configuration to be imported
|
|
-- @return A table with a key "success" set to true is returned on success. A key "err" is set in case of failure, with one of the errors defined in rest_utils.consts.err.
|
|
function am_import_export:import(config)
|
|
local res = {}
|
|
|
|
local conf = config.res
|
|
local vs_conf = config.vs_config
|
|
|
|
-- TODO Validate the configuration and set
|
|
-- res.err = rest_utils.consts.err.bad_content
|
|
|
|
local old_hosts = am_utils.getHosts(true --[[ config only ]])
|
|
|
|
for host_key, host_conf in pairs(conf or {}) do
|
|
local host = am_utils.key2host(host_key)
|
|
|
|
if not host.is_infrastructure then -- Infrastructure is handled separately on another import_export
|
|
if old_hosts[host_key] then
|
|
am_utils.editHost(host.host, host_conf.ifname, host.measurement, host_conf.threshold, host_conf.granularity, host_conf.token, host_conf.save_result, host_conf.readonly)
|
|
else
|
|
am_utils.addHost(host.host, host_conf.ifname, host.measurement, host_conf.threshold, host_conf.granularity, host_conf.token, host_conf.save_result, host_conf.readonly)
|
|
end
|
|
end
|
|
end
|
|
|
|
|
|
-- vs_utils.restore_config_backup(vs_conf)
|
|
|
|
if not res.err then
|
|
res.success = true
|
|
end
|
|
|
|
return res
|
|
end
|
|
|
|
-- ##############################################
|
|
|
|
-- @brief Export configuration
|
|
-- @return The current configuration
|
|
function am_import_export:export()
|
|
local conf = am_utils.getHosts(true --[[ only retrieve the configuration ]])
|
|
local res = {}
|
|
|
|
for host_key, host_conf in pairs(conf) do
|
|
local host = am_utils.key2host(host_key)
|
|
|
|
-- Infrastructure has its own import_export utilities so
|
|
-- hosts that belong to infrastructure monitoring are skipped.
|
|
if not host.is_infrastructure then
|
|
res[host_key] = host_conf
|
|
end
|
|
end
|
|
|
|
|
|
local hosts_to_scan = vs_utils.retrieve_hosts_backup()
|
|
|
|
local am_configs = {
|
|
res = res,
|
|
vs_config = hosts_to_scan
|
|
}
|
|
|
|
return am_configs
|
|
end
|
|
|
|
-- ##############################################
|
|
|
|
-- @brief Reset configuration
|
|
function am_import_export:reset()
|
|
local conf = am_utils.getHosts(true --[[ only retrieve the configuration ]])
|
|
|
|
for host_key, _ in pairs(conf) do
|
|
local host = am_utils.key2host(host_key)
|
|
|
|
-- Don't delete infrastructure hosts
|
|
if not host.is_infrastructure then
|
|
am_utils.deleteHost(host.host, host.measurement)
|
|
end
|
|
end
|
|
end
|
|
|
|
-- ##############################################
|
|
|
|
return am_import_export
|