#!/usr/bin/env php
<?php
/*
 +-------------------------------------------------------------------------+
 | Copyright (C) 2004-2026 The Cacti Group                                 |
 |                                                                         |
 | This program is free software; you can redistribute it and/or           |
 | modify it under the terms of the GNU General Public License             |
 | as published by the Free Software Foundation; either version 2          |
 | of the License, or (at your option) any later version.                  |
 |                                                                         |
 | This program is distributed in the hope that it will be useful,         |
 | but WITHOUT ANY WARRANTY; without even the implied warranty of          |
 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the           |
 | GNU General Public License for more details.                            |
 +-------------------------------------------------------------------------+
 | Cacti: The Complete RRDtool-based Graphing Solution                     |
 +-------------------------------------------------------------------------+
 | This code is designed, written, and maintained by the Cacti Group. See  |
 | about.php and/or the AUTHORS file for specific developer information.   |
 +-------------------------------------------------------------------------+
 | http://www.cacti.net/                                                   |
 +-------------------------------------------------------------------------+
*/

global $config;

if (!isset($called_by_script_server)) {
	include_once(__DIR__ . '/../include/cli_check.php');
	include_once(__DIR__ . '/../lib/snmp.php');

	array_shift($_SERVER['argv']);
	print call_user_func_array('ss_fortigate_virus', $_SERVER['argv']);
} else {
	include_once(__DIR__ . '/../lib/snmp.php');
}

function ss_fortigate_virus(int $host_id = 0) : mixed {
	global $environ, $poller_id, $config;

	if ($host_id <= 0) {
		return 'vir_detected:0 vir_blocked:0 vir_det_web:0 vir_det_smb:0 vir_det_mail:0' . PHP_EOL;
	}

	$host = db_fetch_row_prepared('SELECT *
		FROM host
		WHERE id = ?',
		[$host_id]);

	$oids = [
		'vir_detected' => '.1.3.6.1.4.1.12356.101.8.2.1.1.1.1',
		'vir_blocked'  => '.1.3.6.1.4.1.12356.101.8.2.1.1.2.1',
		'vir_det_web'  => '.1.3.6.1.4.1.12356.101.8.2.1.1.3.1',
		'vir_det_smb'  => '.1.3.6.1.4.1.12356.101.8.2.1.1.21.1'
	];

	$result = '';

	foreach ($oids as $name => $oid) {
		$x = cacti_snmp_get($host['hostname'],
			$host['snmp_community'],
			$oid,
			$host['snmp_version'],
			$host['snmp_username'],
			$host['snmp_password'],
			$host['snmp_auth_protocol'],
			$host['snmp_priv_passphrase'],
			$host['snmp_priv_protocol'],
			$host['snmp_context'],
			$host['snmp_port'],
			$host['snmp_timeout'],
			$host['snmp_retries'],
			SNMP_POLLER,
			$host['snmp_engine_id']);

		if (is_numeric($x)) {
			$result .= $name . ':' . $x . ' ';
		} else {
			$result .= $name . ':0 ';
		}
	}

	$oids = [
		'vir_det_smtp' => '.1.3.6.1.4.1.12356.101.8.2.1.1.5.1',
		'vir_det_pop3' => '.1.3.6.1.4.1.12356.101.8.2.1.1.7.1',
		'vir_det_imap' => '.1.3.6.1.4.1.12356.101.8.2.1.1.9.1'
	];

	$email = 0;

	foreach ($oids as $name => $oid) {
		$x = cacti_snmp_get($host['hostname'],
			$host['snmp_community'],
			$oid,
			$host['snmp_version'],
			$host['snmp_username'],
			$host['snmp_password'],
			$host['snmp_auth_protocol'],
			$host['snmp_priv_passphrase'],
			$host['snmp_priv_protocol'],
			$host['snmp_context'],
			$host['snmp_port'],
			$host['snmp_timeout'],
			$host['snmp_retries'],
			SNMP_POLLER,
			$host['snmp_engine_id']);

		if (is_numeric($x)) {
			$email += $x;
		}
	}

	$result .= 'vir_det_mail:' . $email . PHP_EOL;

	return $result;
}