/*
** Copyright (C) 2001-2025 Zabbix SIA
**
** This program is free software: you can redistribute it and/or modify it under the terms of
** the GNU Affero General Public License as published by the Free Software Foundation, version 3.
**
** 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 Affero General Public License for more details.
**
** You should have received a copy of the GNU Affero General Public License along with this program.
** If not, see <https://www.gnu.org/licenses/>.
**/

#include "zbxpoller.h"

#include "test_get_value_ssh.h"
#include "zbxmocktest.h"

#include "zbxsysinfo.h"

#if defined(HAVE_SSH)
#	include "../../../src/libs/zbxpoller/ssh_run.c"
#elif defined(HAVE_SSH2)
#	include "../../../src/libs/zbxpoller/ssh2_run.c"
#endif

int	__wrap_ssh_run(zbx_dc_item_t *item, AGENT_RESULT *result, const char *encoding, const char *options,
		int timeout, const char *config_source_ip);

#if defined(HAVE_SSH2) || defined(HAVE_SSH)
int	zbx_get_value_ssh_test_run(zbx_dc_item_t *item, char **error)
{
	AGENT_RESULT	result;
	int		ret;
	const char	*config_ssh_key_location = NULL;

	zbx_init_agent_result(&result);
	ret = zbx_ssh_get_value(item, get_zbx_config_source_ip(), config_ssh_key_location, &result);

	if (NULL != result.msg && '\0' != *(result.msg))
	{
		*error = zbx_malloc(NULL, sizeof(char) * strlen(result.msg));
		zbx_strlcpy(*error, result.msg, strlen(result.msg) * sizeof(char));
	}

	zbx_free_agent_result(&result);

	return ret;
}
#endif

int	__wrap_ssh_run(zbx_dc_item_t *item, AGENT_RESULT *result, const char *encoding, const char *options,
		int timeout, const char *config_source_ip)
{
	int	ret = SYSINFO_RET_OK;

	zabbix_log(LOG_LEVEL_DEBUG, "In %s()", __func__);

	ZBX_UNUSED(item);
	ZBX_UNUSED(encoding);
	ZBX_UNUSED(timeout);
	ZBX_UNUSED(config_source_ip);

#if defined(HAVE_SSH) || defined(HAVE_SSH2)
	char	*err_msg = NULL;
#if defined(HAVE_SSH)
	ssh_session	session;

	if (NULL == (session = ssh_new()))
#elif defined(HAVE_SSH2)
	LIBSSH2_SESSION	*session;

	if (NULL == (session = libssh2_session_init()))
#endif
	{
		SET_MSG_RESULT(result, zbx_strdup(NULL, "Cannot initialize SSH session"));
		zabbix_log(LOG_LEVEL_DEBUG, "Cannot initialize SSH session");
		ret = NOTSUPPORTED;
		goto ret;
	}

	if (0 != ssh_parse_options(session, options, &err_msg))
	{
		SET_MSG_RESULT(result, err_msg);
		ret = NOTSUPPORTED;
	}

#if defined(HAVE_SSH)
	ssh_free(session);
#elif defined(HAVE_SSH2)
	libssh2_session_free(session);
#endif
ret:
#else
	ZBX_UNUSED(result);
	ZBX_UNUSED(options);
#endif
	zabbix_log(LOG_LEVEL_DEBUG, "End of %s():%s", __func__, zbx_result_string(ret));

	return ret;
}