You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

106 lines
2.8 KiB

/*
** Zabbix
** Copyright (C) 2001-2023 Zabbix SIA
**
** 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.
**
** You should have received a copy of the GNU General Public License
** along with this program; if not, write to the Free Software
** Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
**/
#include "test_get_value_ssh.h"
#include "zbxmocktest.h"
#include "zbxsysinfo.h"
#include "../../../src/zabbix_server/poller/checks_ssh.h"
#if defined(HAVE_SSH)
# include "../../../src/zabbix_server/poller/ssh_run.c"
#elif defined(HAVE_SSH2)
# include "../../../src/zabbix_server/poller/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;
zbx_init_agent_result(&result);
ret = get_value_ssh(item, SEC_PER_MIN, get_zbx_config_source_ip(), &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;
}