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

#include "../sysinfo.h"

​

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

​

static int  get_cpu_num(int online)

{

    int mib[2], cpu_num;

    size_t  len = sizeof(cpu_num);

​

    mib[0] = CTL_HW;

    mib[1] = (1 == online ? HW_AVAILCPU : HW_NCPU);

​

    if (0 != sysctl(mib, 2, &cpu_num, &len, NULL, 0))

        return -1;

​

    return cpu_num;

}

​

int system_cpu_num(AGENT_REQUEST *request, AGENT_RESULT *result)

{

    char    *tmp;

    int cpu_num, online = 0;

​

    if (1 < request->nparam)

    {

        SET_MSG_RESULT(result, zbx_strdup(NULL, "Too many parameters."));

        return SYSINFO_RET_FAIL;

    }

​

    tmp = get_rparam(request, 0);

​

    if (NULL == tmp || '\0' == *tmp || 0 == strcmp(tmp, "online"))

        online = 1;

    else if (0 != strcmp(tmp, "max"))

    {

        SET_MSG_RESULT(result, zbx_strdup(NULL, "Invalid first parameter."));

        return SYSINFO_RET_FAIL;

    }

​

    if (-1 == (cpu_num = get_cpu_num(online)))

    {

        SET_MSG_RESULT(result, zbx_strdup(NULL, "Cannot obtain number of CPUs."));

        return SYSINFO_RET_FAIL;

    }

​

    SET_UI64_RESULT(result, cpu_num);

​

    return SYSINFO_RET_OK;