Zabbix

/*

** 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 "pg_service.h"

#include "zbxcommon.h"

#include "pg_cache.h"

#include "zbxalgo.h"

#include "zbxcacheconfig.h"

#include "zbxnix.h"

#include "zbxpgservice.h"

#include "zbxserialize.h"

#include "zbxthreads.h"

#include "zbxtime.h"

#include "zbxtypes.h"

#include "zbxversion.h"

​

/******************************************************************************

 *                                                                            *

 * Purpose: move hosts between proxy groups in cache                          *

 *                                                                            *

 * Parameter: pgs     - [IN] proxy group service                              *

 *            message - [IN] IPC message with host relocation data            *

 *                                                                            *

 ******************************************************************************/

static void pg_update_host_pgroup(zbx_pg_service_t *pgs, zbx_ipc_message_t *message)

{

    unsigned char   *ptr = message->data;

    zbx_uint64_t    hostid, srcid, dstid;

​

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

​

    pg_cache_lock(pgs->cache);

​

    (void)pg_cache_update_groups(pgs->cache);

​

    while (ptr - message->data < message->size)

    {

        zbx_pg_group_t  *group;

​

        ptr += zbx_deserialize_value(ptr, &hostid);

        ptr += zbx_deserialize_value(ptr, &srcid);

        ptr += zbx_deserialize_value(ptr, &dstid);

​

        if (srcid == dstid)

        {

            /* handle host name change by re-assigning host to the same proxy */

            /* causing its hostmap revision to change                         */

            zbx_pg_host_t   *host;

            zbx_pg_proxy_t  *proxy;

​

            if (NULL == (host = (zbx_pg_host_t *)zbx_hashset_search(&pgs->cache->hostmap, &hostid)))

                continue;

​

            pg_cache_set_host_proxy(pgs->cache, hostid, host->proxyid);

​

            if (NULL != (proxy = (zbx_pg_proxy_t *)zbx_hashset_search(&pgs->cache->proxies,

                    &host->proxyid)) && NULL != proxy->group)

            {

                pg_cache_queue_group_update(pgs->cache, proxy->group);

            }

        }

        else

        {

            if (0 != srcid)

            {

                if (NULL != (group = (zbx_pg_group_t *)zbx_hashset_search(&pgs->cache->groups, &srcid)))

                    pg_cache_group_remove_host(pgs->cache, group, hostid);

            }

​

            if (0 != dstid)

            {

                if (NULL != (group = (zbx_pg_group_t *)zbx_hashset_search(&pgs->cache->groups, &dstid)))

                    pg_cache_group_add_host(pgs->cache, group, hostid);

            }

        }

    }

​

    pg_cache_unlock(pgs->cache);

​

    zabbix_log(LOG_LEVEL_DEBUG, "End of %s()", __func__);