#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;
}
int system_cpu_load(AGENT_REQUEST *request, AGENT_RESULT *result)
{
char *tmp;
int mode, per_cpu = 1, cpu_num;
double load[ZBX_AVG_COUNT], value;
if (2 < 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, "all"))
per_cpu = 0;
else if (0 != strcmp(tmp, "percpu"))
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Invalid first parameter."));
return SYSINFO_RET_FAIL;
}
tmp = get_rparam(request, 1);
if (NULL == tmp || '\0' == *tmp || 0 == strcmp(tmp, "avg1"))
mode = ZBX_AVG1;