[VS] Fix nil check cases.

This commit is contained in:
Nicolo Maio 2023-10-18 12:38:47 +02:00
parent 22101abf00
commit 3aeab7041c

View file

@ -40,6 +40,19 @@ end
-- #######################################################
-- Function to normalize values (check nil cases)
local function normalize_values(primary_key, secondary_key)
if (primary_key == nil) then
return nil
elseif (primary_key[secondary_key] == nil) then
return nil
else
return primary_key[secondary_key]
end
end
-- #######################################################
-- @brief Format an alert into a human-readable string
-- @param ifid The integer interface id of the generated alert
-- @param alert The alert description table, including alert data such as the generating entity, timestamp, granularity, type
@ -57,10 +70,10 @@ function alert_vulnerability_scan.format(ifid, alert, alert_type_params)
if (not isEmptyString(alert_type_params.tcp_ports_case)) then
msg = msg .. i18n('vulnerability_scan_alert_ports_changed_cases.'..alert_type_params.tcp_ports_case, {
open_ports_num = alert_type_params.tcp_open_ports.num,
open_ports = alert_type_params.tcp_open_ports.ports,
closed_ports_num = alert_type_params.tcp_closed_ports.num,
closed_ports = alert_type_params.tcp_closed_ports.ports,
open_ports_num = normalize_values(alert_type_params.tcp_open_ports,"num"),
open_ports = normalize_values(alert_type_params.tcp_open_ports,"ports"),
closed_ports_num = normalize_values(alert_type_params.tcp_closed_ports,"num"),
closed_ports = normalize_values(alert_type_params.tcp_closed_ports,"ports"),
protocol = i18n("tcp")
})
@ -70,10 +83,10 @@ function alert_vulnerability_scan.format(ifid, alert, alert_type_params)
if (not isEmptyString(alert_type_params.udp_ports_case)) then
msg = msg .. i18n('vulnerability_scan_alert_ports_changed_cases.'..alert_type_params.udp_ports_case, {
open_ports_num = alert_type_params.udp_open_ports.num,
open_ports = alert_type_params.udp_open_ports.ports,
closed_ports_num = alert_type_params.udp_closed_ports.num,
closed_ports = alert_type_params.udp_closed_ports.ports,
open_ports_num = normalize_values(alert_type_params.udp_open_ports,"num"),
open_ports = normalize_values(alert_type_params.udp_open_ports,"ports"),
closed_ports_num = normalize_values(alert_type_params.udp_closed_ports,"num"),
closed_ports = normalize_values(alert_type_params.udp_closed_ports,"ports"),
protocol = i18n("udp")
})
msg = msg:gsub("%,", ", ")