Zabbix

/*

** Copyright (C) 2001-2024 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 "zbxcommshigh.h"

​

#include "zbxjson.h"

#include "zbxlog.h"

#include "zbxip.h"

#include "zbxcomms.h"

#include "zbxnum.h"

​

#if !defined(_WINDOWS) && !defined(__MINGW32)

#include "zbxnix.h"

#endif

​

#include "zbxcfg.h"

​

void    zbx_addrs_failover(zbx_vector_addr_ptr_t *addrs)

{

    if (1 < addrs->values_num)

    {

        zbx_addr_t  *addr = addrs->values[0];

​

        zbx_vector_addr_ptr_remove(addrs, 0);

        zbx_vector_addr_ptr_append(addrs, addr);

    }

}

​

static int  zbx_tcp_connect_failover(zbx_socket_t *s, const char *source_ip, zbx_vector_addr_ptr_t *addrs,

        int timeout, int connect_timeout, unsigned int tls_connect, const char *tls_arg1, const char *tls_arg2,

        int loglevel)

{

    int i, ret = FAIL;

​

    for (i = 0; i < addrs->values_num; i++)

    {

        zbx_addr_t  *addr;

​

        addr = (zbx_addr_t *)addrs->values[0];

​

        if (FAIL != (ret = zbx_tcp_connect(s, source_ip, addr->ip, addr->port, connect_timeout, tls_connect,

                tls_arg1, tls_arg2)))

        {

            zbx_socket_set_deadline(s, timeout);

            break;

        }

​

        zabbix_log(loglevel, "Unable to connect to [%s]:%d [%s]",

                ((zbx_addr_t *)addrs->values[0])->ip, ((zbx_addr_t *)addrs->values[0])->port,

                zbx_socket_strerror());

​

        zbx_addrs_failover(addrs);

    }

​

    return ret;

}

​

int zbx_connect_to_server(zbx_socket_t *sock, const char *source_ip, zbx_vector_addr_ptr_t *addrs, int timeout,

        int connect_timeout, int retry_interval, int level, const zbx_config_tls_t *config_tls)

{

    int     res = FAIL;

    const char  *tls_arg1, *tls_arg2;

​

    zabbix_log(LOG_LEVEL_DEBUG, "In %s() [%s]:%d [timeout:%d, connection timeout:%d]", __func__,

            addrs->values[0]->ip, addrs->values[0]->port, timeout, connect_timeout);

​

    switch (config_tls->connect_mode)

    {

        case ZBX_TCP_SEC_UNENCRYPTED:

            tls_arg1 = NULL;

            tls_arg2 = NULL;

            break;

#if defined(HAVE_GNUTLS) || defined(HAVE_OPENSSL)

        case ZBX_TCP_SEC_TLS_CERT:

            tls_arg1 = config_tls->server_cert_issuer;

            tls_arg2 = config_tls->server_cert_subject;

            break;

        case ZBX_TCP_SEC_TLS_PSK:

            tls_arg1 = config_tls->psk_identity;

            tls_arg2 = NULL;    /* zbx_tls_connect() will find PSK */