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

#include "../sysinfo.h"

​

#include "zbxjson.h"

#include "zbxstr.h"

​

#include "../common/stats.h"

#include "../common/diskdevices.h"

​

#define ZBX_DEV_PFX     "/dev/"

#define ZBX_DEV_READ        0

#define ZBX_DEV_WRITE       1

#define ZBX_SYS_BLKDEV_PFX  "/sys/dev/block/"

​

#if defined(KERNEL_2_4)

#   define INFO_FILE_NAME   "/proc/partitions"

#   define PARSE(line)  if (sscanf(line, ZBX_FS_UI64 ZBX_FS_UI64 " %*d %s "     \

                    ZBX_FS_UI64 " %*d " ZBX_FS_UI64 " %*d "         \

                    ZBX_FS_UI64 " %*d " ZBX_FS_UI64 " %*d %*d %*d %*d", \

                &rdev_major,                            \

                &rdev_minor,                            \

                name,                               \

                &ds[ZBX_DSTAT_R_OPER],                      \

                &ds[ZBX_DSTAT_R_SECT],                      \

                &ds[ZBX_DSTAT_W_OPER],                      \

                &ds[ZBX_DSTAT_W_SECT]                       \

                ) != 7) continue

#else

#   define INFO_FILE_NAME   "/proc/diskstats"

#   define PARSE(line)  if (sscanf(line, ZBX_FS_UI64 ZBX_FS_UI64 " %s "         \

                    ZBX_FS_UI64 " %*d " ZBX_FS_UI64 " %*d "         \

                    ZBX_FS_UI64 " %*d " ZBX_FS_UI64 " %*d %*d %*d %*d", \

                &rdev_major,                            \

                &rdev_minor,                            \

                name,                               \

                &ds[ZBX_DSTAT_R_OPER],                      \

                &ds[ZBX_DSTAT_R_SECT],                      \

                &ds[ZBX_DSTAT_W_OPER],                      \

                &ds[ZBX_DSTAT_W_SECT]                       \

                ) != 7                              \

                &&                              \

                /* some disk partitions */                  \

                sscanf(line, ZBX_FS_UI64 ZBX_FS_UI64 " %s "         \

                    ZBX_FS_UI64 ZBX_FS_UI64                 \

                    ZBX_FS_UI64 ZBX_FS_UI64,                \

                &rdev_major,                            \

                &rdev_minor,                            \

                name,                               \

                &ds[ZBX_DSTAT_R_OPER],                      \

                &ds[ZBX_DSTAT_R_SECT],                      \

                &ds[ZBX_DSTAT_W_OPER],                      \

                &ds[ZBX_DSTAT_W_SECT]                       \

                ) != 7                              \

                ) continue

#endif

​

int zbx_get_diskstat(const char *devname, zbx_uint64_t *dstat)

{

    FILE        *f;

    char        tmp[MAX_STRING_LEN], name[MAX_STRING_LEN], dev_path[MAX_STRING_LEN];

    int     ret = FAIL, dev_exists = FAIL, found = 0;

    zbx_uint64_t    ds[ZBX_DSTAT_MAX], rdev_major, rdev_minor;

    zbx_stat_t  dev_st;

​

    for (int i = 0; i < ZBX_DSTAT_MAX; i++)

        dstat[i] = (zbx_uint64_t)__UINT64_C(0);

​

    if (NULL != devname && '\0' != *devname && 0 != strcmp(devname, "all"))

    {

        *dev_path = '\0';

        if (0 != strncmp(devname, ZBX_DEV_PFX, ZBX_CONST_STRLEN(ZBX_DEV_PFX)))

            zbx_strscpy(dev_path, ZBX_DEV_PFX);

        zbx_strscat(dev_path, devname);

​

        if (zbx_stat(dev_path, &dev_st) == 0)

            dev_exists = SUCCEED;

    }