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 "dbconn.h"

#include "zbxcommon.h"

#include "zbxdb.h"

#include "zbxcfg.h"

#include "zbxmutexs.h"

#include "zbxshmem.h"

#include "zbxalgo.h"

#include "zbxdbschema.h"

#include "zbxnum.h"

#include "zbxstr.h"

#include "zbxtypes.h"

#if defined(HAVE_POSTGRESQL)

#   include "zbx_dbversion_constants.h"

#endif

​

#define ZBX_MAX_SQL_SIZE    262144  /* 256KB */

#ifndef ZBX_MAX_OVERFLOW_SQL_SIZE

#   define ZBX_MAX_OVERFLOW_SQL_SIZE    ZBX_MAX_SQL_SIZE

#elif 0 != ZBX_MAX_OVERFLOW_SQL_SIZE && \

    (1024 > ZBX_MAX_OVERFLOW_SQL_SIZE || ZBX_MAX_OVERFLOW_SQL_SIZE > ZBX_MAX_SQL_SIZE)

#error ZBX_MAX_OVERFLOW_SQL_SIZE is out of range

#endif

​

ZBX_CONST_PTR_VECTOR_IMPL(const_db_field_ptr, const zbx_db_field_t *)

ZBX_PTR_VECTOR_IMPL(db_value_ptr, zbx_db_value_t *)

​

const char  *idcache_tables[] = {"events", "event_tag", "problem_tag", "dservices", "dhosts", "alerts",

                    "escalations", "autoreg_host", "event_suppress", "trigger_queue",

                    "proxy_history", "proxy_dhistory", "proxy_autoreg_host", "host_proxy"};

​

#define ZBX_IDS_SIZE    ARRSIZE(idcache_tables)

​

static int  compare_table_names(const void *d1, const void *d2)

{

    const char *n1 = *(const char * const *)d1;

    const char *n2 = *(const char * const *)d2;

​

    return strcmp(n1, n2);

}

​

typedef struct

{

    zbx_uint64_t    lastids[ZBX_IDS_SIZE];

}

zbx_db_idcache_t;

​

/* nextid cache for tables updated only by server/proxy */

static zbx_mutex_t  idcache_mutex = ZBX_MUTEX_NULL;

zbx_shmem_info_t    *idcache_mem;

static zbx_db_idcache_t *idcache = NULL;

​

int zbx_db_init(char **error)

{

    qsort(idcache_tables, ZBX_IDS_SIZE, sizeof(idcache_tables[0]), compare_table_names);

​

    if (SUCCEED != dbconn_init(error))

        return FAIL;

​

    if (SUCCEED != zbx_mutex_create(&idcache_mutex, ZBX_MUTEX_CACHE_IDS, error))

        return FAIL;

​

    if (SUCCEED != zbx_shmem_create_min(&idcache_mem, sizeof(zbx_db_idcache_t), "table ids cache", NULL, 0, error))

        return FAIL;

​

    idcache = idcache_mem->base;

    memset(idcache, 0, sizeof(zbx_db_idcache_t));

​

    return SUCCEED;

}

​

void    zbx_db_deinit(void)

{

    dbconn_deinit();

​

    zbx_shmem_destroy(idcache_mem);

    idcache_mem = NULL;

​