int PROC_MEM(AGENT_REQUEST *request, AGENT_RESULT *result)
{
char *procname, *proccomm, *param;
int do_task, pagesize, count, i,
proc_ok, comm_ok;
double value = 0.0,
memsize = 0;
int proccount = 0;
size_t sz;
struct passwd *usrinfo;
#ifdef KERN_PROC2
int mib[6];
struct kinfo_proc2 *proc = NULL;
#else
int mib[4];
struct kinfo_proc *proc = NULL;
#endif
char **argv = NULL, *args = NULL;
size_t argv_alloc = 0, args_alloc = 0;
int argc;
if (4 < request->nparam)
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Too many parameters."));
return SYSINFO_RET_FAIL;
}
procname = get_rparam(request, 0);
param = get_rparam(request, 1);
if (NULL != param && '\0' != *param)
{
errno = 0;
if (NULL == (usrinfo = getpwnam(param)))
{
if (0 == errno)
SET_MSG_RESULT(result, zbx_strdup(NULL, "Specified user does not exist."));
else
SET_MSG_RESULT(result, zbx_dsprintf(NULL, "Cannot obtain user information: %s",
zbx_strerror(errno)));
return SYSINFO_RET_FAIL;
}
}
else
usrinfo = NULL;
param = get_rparam(request, 2);
if (NULL == param || '\0' == *param || 0 == strcmp(param, "sum"))
do_task = ZBX_DO_SUM;
else if (0 == strcmp(param, "avg"))
do_task = ZBX_DO_AVG;
else if (0 == strcmp(param, "max"))
do_task = ZBX_DO_MAX;
else if (0 == strcmp(param, "min"))
do_task = ZBX_DO_MIN;
else
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Invalid third parameter."));
return SYSINFO_RET_FAIL;
}
proccomm = get_rparam(request, 3);
pagesize = getpagesize();
mib[0] = CTL_KERN;
if (NULL != usrinfo)
{
mib[2] = KERN_PROC_UID;
mib[3] = usrinfo->pw_uid;
}
else
{
mib[2] = KERN_PROC_ALL;
mib[3] = 0;
}
#ifdef KERN_PROC2
mib[1] = KERN_PROC2;
mib[4] = sizeof(struct kinfo_proc2);
mib[5] = 0;
sz = 0;
if (0 != sysctl(mib, 6, NULL, &sz, NULL, 0))
{
SET_MSG_RESULT(result, zbx_dsprintf(NULL, "Cannot obtain necessary buffer size from system: %s",
zbx_strerror(errno)));
return SYSINFO_RET_FAIL;
}
proc = (struct kinfo_proc2 *)zbx_malloc(proc, sz);
mib[5] = (int)(sz / sizeof(struct kinfo_proc2));
if (0 != sysctl(mib, 6, proc, &sz, NULL, 0))
{
zbx_free(proc);
SET_MSG_RESULT(result, zbx_dsprintf(NULL, "Cannot obtain process information: %s",
zbx_strerror(errno)));
return SYSINFO_RET_FAIL;
}
count = sz / sizeof(struct kinfo_proc2);
#else
mib[1] = KERN_PROC;
sz = 0;
if (0 != sysctl(mib, 4, NULL, &sz, NULL, 0))
{
SET_MSG_RESULT(result, zbx_dsprintf(NULL, "Cannot obtain necessary buffer size from system: %s",
zbx_strerror(errno)));
return SYSINFO_RET_FAIL;
}
proc = (struct kinfo_proc *)zbx_malloc(proc, sz);
if (0 != sysctl(mib, 4, proc, &sz, NULL, 0))
{
zbx_free(proc);
SET_MSG_RESULT(result, zbx_dsprintf(NULL, "Cannot obtain process information: %s",
zbx_strerror(errno)));
return SYSINFO_RET_FAIL;
}
count = sz / sizeof(struct kinfo_proc);
#endif
for (i = 0; i < count; i++)
{
proc_ok = 0;
comm_ok = 0;
if (NULL == procname || '\0' == *procname || 0 == strcmp(procname, proc[i].ZBX_P_COMM))
proc_ok = 1;
if (NULL != proccomm && '\0' != *proccomm)
{
if (SUCCEED == proc_argv(proc[i].ZBX_P_PID, &argv, &argv_alloc, &argc))
{
collect_args(argv, argc, &args, &args_alloc);
if (NULL != zbx_regexp_match(args, proccomm, NULL))
comm_ok = 1;
}
}
else
comm_ok = 1;
if (proc_ok && comm_ok)
{
value = proc[i].ZBX_P_VM_TSIZE
+ proc[i].ZBX_P_VM_DSIZE
+ proc[i].ZBX_P_VM_SSIZE;
value *= pagesize;
if (0 == proccount++)
memsize = value;
else
{
if (ZBX_DO_MAX == do_task)
memsize = MAX(memsize, value);
else if (ZBX_DO_MIN == do_task)
memsize = MIN(memsize, value);
else
memsize += value;
}
}
}
zbx_free(proc);
zbx_free(argv);
zbx_free(args);
if (ZBX_DO_AVG == do_task)
SET_DBL_RESULT(result, proccount == 0 ? 0 : memsize/proccount);
else
SET_UI64_RESULT(result, memsize);
return SYSINFO_RET_OK;
}
int PROC_NUM(AGENT_REQUEST *request, AGENT_RESULT *result)
{
char *procname, *proccomm, *param;
int zbx_proc_stat, count, i,
proc_ok, stat_ok, comm_ok;
int proccount = 0;
size_t sz;
struct passwd *usrinfo;
#ifdef KERN_PROC2
int mib[6];
struct kinfo_proc2 *proc = NULL;
#else
int mib[4];
struct kinfo_proc *proc = NULL;
#endif
char **argv = NULL, *args = NULL;
size_t argv_alloc = 0, args_alloc = 0;
int argc;
if (4 < request->nparam)
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Too many parameters."));
return SYSINFO_RET_FAIL;
}
procname = get_rparam(request, 0);
param = get_rparam(request, 1);
if (NULL != param && '\0' != *param)
{
errno = 0;
if (NULL == (usrinfo = getpwnam(param)))
{
if (0 == errno)
SET_MSG_RESULT(result, zbx_strdup(NULL, "Specified user does not exist."));
else
SET_MSG_RESULT(result, zbx_dsprintf(NULL, "Cannot obtain user information: %s",
zbx_strerror(errno)));
return SYSINFO_RET_FAIL;
}
}
else
usrinfo = NULL;
param = get_rparam(request, 2);
if (NULL == param || '\0' == *param || 0 == strcmp(param, "all"))
zbx_proc_stat = ZBX_PROC_STAT_ALL;
else if (0 == strcmp(param, "run"))
zbx_proc_stat = ZBX_PROC_STAT_RUN;
else if (0 == strcmp(param, "sleep"))
zbx_proc_stat = ZBX_PROC_STAT_SLEEP;
else if (0 == strcmp(param, "zomb"))
zbx_proc_stat = ZBX_PROC_STAT_ZOMB;
else
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Invalid third parameter."));
return SYSINFO_RET_FAIL;
}
proccomm = get_rparam(request, 3);
mib[0] = CTL_KERN;
if (NULL != usrinfo)
{
mib[2] = KERN_PROC_UID;
mib[3] = usrinfo->pw_uid;
}
else
{
mib[2] = KERN_PROC_ALL;
mib[3] = 0;
}
#ifdef KERN_PROC2
mib[1] = KERN_PROC2;
mib[4] = sizeof(struct kinfo_proc2);
mib[5] = 0;
sz = 0;
if (0 != sysctl(mib, 6, NULL, &sz, NULL, 0))
{
SET_MSG_RESULT(result, zbx_dsprintf(NULL, "Cannot obtain necessary buffer size from system: %s",
zbx_strerror(errno)));
return SYSINFO_RET_FAIL;
}
proc = (struct kinfo_proc2 *)zbx_malloc(proc, sz);
mib[5] = (int)(sz / sizeof(struct kinfo_proc2));
if (0 != sysctl(mib, 6, proc, &sz, NULL, 0))
{
zbx_free(proc);
SET_MSG_RESULT(result, zbx_dsprintf(NULL, "Cannot obtain process information: %s",
zbx_strerror(errno)));
return SYSINFO_RET_FAIL;
}
count = sz / sizeof(struct kinfo_proc2);
#else
mib[1] = KERN_PROC;
sz = 0;
if (0 != sysctl(mib, 4, NULL, &sz, NULL, 0))
{
SET_MSG_RESULT(result, zbx_dsprintf(NULL, "Cannot obtain necessary buffer size from system: %s",
zbx_strerror(errno)));
return SYSINFO_RET_FAIL;
}
proc = (struct kinfo_proc *)zbx_malloc(proc, sz);
if (0 != sysctl(mib, 4, proc, &sz, NULL, 0))
{
zbx_free(proc);
SET_MSG_RESULT(result, zbx_dsprintf(NULL, "Cannot obtain process information: %s",
zbx_strerror(errno)));
return SYSINFO_RET_FAIL;
}
count = sz / sizeof(struct kinfo_proc);
#endif
for (i = 0; i < count; i++)
{
proc_ok = 0;
stat_ok = 0;
comm_ok = 0;
if (NULL == procname || '\0' == *procname || 0 == strcmp(procname, proc[i].ZBX_P_COMM))
proc_ok = 1;
stat_ok = (zbx_proc_stat == ZBX_PROC_STAT_ALL ||
(zbx_proc_stat == ZBX_PROC_STAT_RUN && (proc[i].ZBX_P_STAT == SRUN || proc[i].ZBX_P_STAT == SONPROC)) ||
(zbx_proc_stat == ZBX_PROC_STAT_SLEEP && proc[i].ZBX_P_STAT == SSLEEP) ||
(zbx_proc_stat == ZBX_PROC_STAT_ZOMB && (proc[i].ZBX_P_STAT == SZOMB || proc[i].ZBX_P_STAT == SDEAD)));
if (NULL != proccomm && '\0' != *proccomm)
{
if (SUCCEED == proc_argv(proc[i].ZBX_P_PID, &argv, &argv_alloc, &argc))
{
collect_args(argv, argc, &args, &args_alloc);
if (NULL != zbx_regexp_match(args, proccomm, NULL))
comm_ok = 1;
}
}
else
comm_ok = 1;
if (proc_ok && stat_ok && comm_ok)
proccount++;
}
zbx_free(proc);
zbx_free(argv);
zbx_free(args);
SET_UI64_RESULT(result, proccount);
return SYSINFO_RET_OK;
}
int PROC_MEM(const char *cmd, const char *param, unsigned flags, AGENT_RESULT *result)
{
char procname[MAX_STRING_LEN],
buffer[MAX_STRING_LEN],
proccomm[MAX_STRING_LEN],
*args;
int do_task, pagesize, count, i,
proc_ok, comm_ok,
op, arg;
double value = 0.0,
memsize = 0;
int proccount = 0;
size_t sz;
struct kinfo_proc2 *proc, *pproc;
struct passwd *usrinfo;
if (num_param(param) > 4)
return SYSINFO_RET_FAIL;
if (0 != get_param(param, 1, procname, sizeof(procname)))
*procname = '\0';
else if (strlen(procname) > MAXCOMLEN)
procname[MAXCOMLEN] = '\0';
if (0 != get_param(param, 2, buffer, sizeof(buffer)))
*buffer = '\0';
if (*buffer != '\0')
{
usrinfo = getpwnam(buffer);
if (usrinfo == NULL) /* incorrect user name */
return SYSINFO_RET_FAIL;
}
else
usrinfo = NULL;
if (0 != get_param(param, 3, buffer, sizeof(buffer)))
*buffer = '\0';
if (*buffer != '\0')
{
if (0 == strcmp(buffer, "avg"))
do_task = DO_AVG;
else if (0 == strcmp(buffer, "max"))
do_task = DO_MAX;
else if (0 == strcmp(buffer, "min"))
do_task = DO_MIN;
else if (0 == strcmp(buffer, "sum"))
do_task = DO_SUM;
else
return SYSINFO_RET_FAIL;
}
else
do_task = DO_SUM;
if (0 != get_param(param, 4, proccomm, sizeof(proccomm)))
*proccomm = '\0';
pagesize = getpagesize();
if (NULL == kd && NULL == (kd = kvm_open(NULL, NULL, NULL, KVM_NO_FILES, NULL)))
return SYSINFO_RET_FAIL;
if (NULL != usrinfo)
{
op = KERN_PROC_UID;
arg = (int)usrinfo->pw_uid;
}
else
{
op = KERN_PROC_ALL;
arg = 0;
}
if (NULL == (proc = kvm_getproc2(kd, op, arg, sizeof(struct kinfo_proc2), &count)))
return SYSINFO_RET_FAIL;
for (pproc = proc, i = 0; i < count; pproc++, i++)
{
proc_ok = 0;
comm_ok = 0;
if (*procname == '\0' || 0 == strcmp(procname, pproc->p_comm))
proc_ok = 1;
if (*proccomm != '\0')
{
if (NULL != (args = proc_argv(pproc->p_pid)))
{
if (NULL != zbx_regexp_match(args, proccomm, NULL))
comm_ok = 1;
}
}
else
comm_ok = 1;
if (proc_ok && comm_ok)
{
value = pproc->p_vm_tsize
+ pproc->p_vm_dsize
+ pproc->p_vm_ssize;
value *= pagesize;
if (0 == proccount++)
memsize = value;
else
{
if (do_task == DO_MAX)
memsize = MAX(memsize, value);
else if (do_task == DO_MIN)
memsize = MIN(memsize, value);
else
memsize += value;
}
}
}
if (do_task == DO_AVG)
SET_DBL_RESULT(result, proccount == 0 ? 0 : memsize / proccount);
else
SET_UI64_RESULT(result, memsize);
return SYSINFO_RET_OK;
}
int PROC_NUM(const char *cmd, const char *param, unsigned flags, AGENT_RESULT *result)
{
char procname[MAX_STRING_LEN],
buffer[MAX_STRING_LEN],
proccomm[MAX_STRING_LEN],
*args;
int zbx_proc_stat, count, i,
proc_ok, stat_ok, comm_ok,
op, arg;
int proccount = 0;
size_t sz;
struct kinfo_proc2 *proc, *pproc;
struct passwd *usrinfo;
if (num_param(param) > 4)
return SYSINFO_RET_FAIL;
if (0 != get_param(param, 1, procname, sizeof(procname)))
*procname = '\0';
else if (strlen(procname) > MAXCOMLEN)
procname[MAXCOMLEN] = '\0';
if (0 != get_param(param, 2, buffer, sizeof(buffer)))
*buffer = '\0';
if (*buffer != '\0')
{
usrinfo = getpwnam(buffer);
if (usrinfo == NULL) /* incorrect user name */
return SYSINFO_RET_FAIL;
}
else
usrinfo = NULL;
if (0 != get_param(param, 3, buffer, sizeof(buffer)))
*buffer = '\0';
if (*buffer != '\0')
{
if (0 == strcmp(buffer, "run"))
zbx_proc_stat = ZBX_PROC_STAT_RUN;
else if (0 == strcmp(buffer, "sleep"))
zbx_proc_stat = ZBX_PROC_STAT_SLEEP;
else if (0 == strcmp(buffer, "zomb"))
zbx_proc_stat = ZBX_PROC_STAT_ZOMB;
else if (0 == strcmp(buffer, "all"))
zbx_proc_stat = ZBX_PROC_STAT_ALL;
else
return SYSINFO_RET_FAIL;
}
else
zbx_proc_stat = ZBX_PROC_STAT_ALL;
if (0 != get_param(param, 4, proccomm, sizeof(proccomm)))
*proccomm = '\0';
if (NULL == kd && NULL == (kd = kvm_open(NULL, NULL, NULL, KVM_NO_FILES, NULL)))
return SYSINFO_RET_FAIL;
if (NULL != usrinfo)
{
op = KERN_PROC_UID;
arg = (int)usrinfo->pw_uid;
}
else
{
op = KERN_PROC_ALL;
arg = 0;
}
if (NULL == (proc = kvm_getproc2(kd, op, arg, sizeof(struct kinfo_proc2), &count)))
return SYSINFO_RET_FAIL;
for (pproc = proc, i = 0; i < count; pproc++, i++)
{
proc_ok = 0;
stat_ok = 0;
comm_ok = 0;
if (*procname == '\0' || 0 == strcmp(procname, pproc->p_comm))
proc_ok = 1;
if (zbx_proc_stat != ZBX_PROC_STAT_ALL)
{
switch (zbx_proc_stat) {
case ZBX_PROC_STAT_RUN:
if (pproc->p_stat == LSRUN || pproc->p_stat == LSONPROC)
stat_ok = 1;
break;
case ZBX_PROC_STAT_SLEEP:
if (pproc->p_stat == LSSLEEP)
stat_ok = 1;
break;
case ZBX_PROC_STAT_ZOMB:
if (pproc->p_stat == SZOMB || pproc->p_stat == LSDEAD)
stat_ok = 1;
break;
}
}
else
stat_ok = 1;
if (*proccomm != '\0')
{
if (NULL != (args = proc_argv(pproc->p_pid)))
{
if (zbx_regexp_match(args, proccomm, NULL) != NULL)
comm_ok = 1;
}
}
else
comm_ok = 1;
if (proc_ok && stat_ok && comm_ok)
proccount++;
}
SET_UI64_RESULT(result, proccount);
return SYSINFO_RET_OK;
}
int PROC_MEM(AGENT_REQUEST *request, AGENT_RESULT *result)
{
char *procname, *proccomm, *param;
int do_task, pagesize, count, i,
proc_ok, comm_ok;
double value = 0.0,
memsize = 0;
int proccount = 0;
size_t sz;
struct passwd *usrinfo;
#ifdef KERN_PROC2
int mib[6];
struct kinfo_proc2 *proc = NULL;
#else
int mib[4];
struct kinfo_proc *proc = NULL;
#endif
char **argv = NULL, *args = NULL;
size_t argv_alloc = 0, args_alloc = 0;
int argc;
if (4 < request->nparam)
return SYSINFO_RET_FAIL;
procname = get_rparam(request, 0);
param = get_rparam(request, 1);
if (NULL != param && '\0' != *param)
{
if (NULL == (usrinfo = getpwnam(param))) /* incorrect user name */
return SYSINFO_RET_FAIL;
}
else
usrinfo = NULL;
param = get_rparam(request, 2);
if (NULL == param || '\0' == *param || 0 == strcmp(param, "sum"))
do_task = DO_SUM;
else if (0 == strcmp(param, "avg"))
do_task = DO_AVG;
else if (0 == strcmp(param, "max"))
do_task = DO_MAX;
else if (0 == strcmp(param, "min"))
do_task = DO_MIN;
else
return SYSINFO_RET_FAIL;
proccomm = get_rparam(request, 3);
pagesize = getpagesize();
mib[0] = CTL_KERN;
if (NULL != usrinfo)
{
mib[2] = KERN_PROC_UID;
mib[3] = usrinfo->pw_uid;
}
else
{
mib[2] = KERN_PROC_ALL;
mib[3] = 0;
}
#ifdef KERN_PROC2
mib[1] = KERN_PROC2;
mib[4] = sizeof(struct kinfo_proc2);
mib[5] = 0;
sz = 0;
if (0 != sysctl(mib, 6, NULL, &sz, NULL, 0))
return SYSINFO_RET_FAIL;
proc = (struct kinfo_proc2 *)zbx_malloc(proc, sz);
mib[5] = (int)(sz / sizeof(struct kinfo_proc2));
if (0 != sysctl(mib, 6, proc, &sz, NULL, 0))
{
zbx_free(proc);
return SYSINFO_RET_FAIL;
}
count = sz / sizeof(struct kinfo_proc2);
#else
mib[1] = KERN_PROC;
sz = 0;
if (0 != sysctl(mib, 4, NULL, &sz, NULL, 0))
return SYSINFO_RET_FAIL;
proc = (struct kinfo_proc *)zbx_malloc(proc, sz);
if (0 != sysctl(mib, 4, proc, &sz, NULL, 0))
{
zbx_free(proc);
return SYSINFO_RET_FAIL;
}
count = sz / sizeof(struct kinfo_proc);
#endif
for (i = 0; i < count; i++)
{
proc_ok = 0;
comm_ok = 0;
if (NULL == procname || '\0' == *procname || 0 == strcmp(procname, proc[i].ZBX_P_COMM))
proc_ok = 1;
if (NULL != proccomm && '\0' != *proccomm)
{
if (SUCCEED == proc_argv(proc[i].ZBX_P_PID, &argv, &argv_alloc, &argc))
{
collect_args(argv, argc, &args, &args_alloc);
if (NULL != zbx_regexp_match(args, proccomm, NULL))
comm_ok = 1;
}
}
else
comm_ok = 1;
if (proc_ok && comm_ok)
{
value = proc[i].ZBX_P_VM_TSIZE
+ proc[i].ZBX_P_VM_DSIZE
+ proc[i].ZBX_P_VM_SSIZE;
value *= pagesize;
if (0 == proccount++)
memsize = value;
else
{
if (do_task == DO_MAX)
memsize = MAX(memsize, value);
else if (do_task == DO_MIN)
memsize = MIN(memsize, value);
else
memsize += value;
}
}
}
zbx_free(proc);
zbx_free(argv);
zbx_free(args);
if (do_task == DO_AVG)
SET_DBL_RESULT(result, proccount == 0 ? 0 : memsize/proccount);
else
SET_UI64_RESULT(result, memsize);
return SYSINFO_RET_OK;
}
int PROC_NUM(const char *cmd, const char *param, unsigned flags, AGENT_RESULT *result)
{
char procname[MAX_STRING_LEN],
buffer[MAX_STRING_LEN],
proccomm[MAX_STRING_LEN];
int zbx_proc_stat, count, i,
proc_ok, stat_ok, comm_ok;
int proccount = 0;
size_t sz;
struct passwd *usrinfo;
#ifdef KERN_PROC2
int mib[6];
struct kinfo_proc2 *proc = NULL;
#else
int mib[4];
struct kinfo_proc *proc = NULL;
#endif
char **argv = NULL, *args = NULL;
size_t argv_alloc = 0, args_alloc = 0;
int argc;
if (num_param(param) > 4)
return SYSINFO_RET_FAIL;
if (0 != get_param(param, 1, procname, sizeof(procname)))
*procname = '\0';
else if (strlen(procname) > MAXCOMLEN)
procname[MAXCOMLEN] = '\0';
if (0 != get_param(param, 2, buffer, sizeof(buffer)))
*buffer = '\0';
if (*buffer != '\0')
{
usrinfo = getpwnam(buffer);
if (usrinfo == NULL) /* incorrect user name */
return SYSINFO_RET_FAIL;
}
else
usrinfo = NULL;
if (0 != get_param(param, 3, buffer, sizeof(buffer)))
*buffer = '\0';
if (*buffer != '\0')
{
if (0 == strcmp(buffer, "run"))
zbx_proc_stat = ZBX_PROC_STAT_RUN;
else if (0 == strcmp(buffer, "sleep"))
zbx_proc_stat = ZBX_PROC_STAT_SLEEP;
else if (0 == strcmp(buffer, "zomb"))
zbx_proc_stat = ZBX_PROC_STAT_ZOMB;
else if (0 == strcmp(buffer, "all"))
zbx_proc_stat = ZBX_PROC_STAT_ALL;
else
return SYSINFO_RET_FAIL;
}
else
zbx_proc_stat = ZBX_PROC_STAT_ALL;
if (0 != get_param(param, 4, proccomm, sizeof(proccomm)))
*proccomm = '\0';
mib[0] = CTL_KERN;
if (NULL != usrinfo)
{
mib[2] = KERN_PROC_UID;
mib[3] = usrinfo->pw_uid;
}
else
{
mib[2] = KERN_PROC_ALL;
mib[3] = 0;
}
#ifdef KERN_PROC2
mib[1] = KERN_PROC2;
mib[4] = sizeof(struct kinfo_proc2);
mib[5] = 0;
sz = 0;
if (0 != sysctl(mib, 6, NULL, &sz, NULL, 0))
return SYSINFO_RET_FAIL;
proc = (struct kinfo_proc2 *)zbx_malloc(proc, sz);
mib[5] = (int)(sz / sizeof(struct kinfo_proc2));
if (0 != sysctl(mib, 6, proc, &sz, NULL, 0))
{
zbx_free(proc);
return SYSINFO_RET_FAIL;
}
count = sz / sizeof(struct kinfo_proc2);
#else
mib[1] = KERN_PROC;
sz = 0;
if (0 != sysctl(mib, 4, NULL, &sz, NULL, 0))
return SYSINFO_RET_FAIL;
proc = (struct kinfo_proc *)zbx_malloc(proc, sz);
if (0 != sysctl(mib, 4, proc, &sz, NULL, 0))
{
zbx_free(proc);
return SYSINFO_RET_FAIL;
}
count = sz / sizeof(struct kinfo_proc);
#endif
for (i = 0; i < count; i++)
{
proc_ok = 0;
stat_ok = 0;
comm_ok = 0;
if (*procname == '\0' || 0 == strcmp(procname, proc[i].ZBX_P_COMM))
proc_ok = 1;
stat_ok = (zbx_proc_stat == ZBX_PROC_STAT_ALL ||
(zbx_proc_stat == ZBX_PROC_STAT_RUN && (proc[i].ZBX_P_STAT == SRUN || proc[i].ZBX_P_STAT == SONPROC)) ||
(zbx_proc_stat == ZBX_PROC_STAT_SLEEP && proc[i].ZBX_P_STAT == SSLEEP) ||
(zbx_proc_stat == ZBX_PROC_STAT_ZOMB && (proc[i].ZBX_P_STAT == SZOMB || proc[i].ZBX_P_STAT == SDEAD)));
if (*proccomm != '\0')
{
if (SUCCEED == proc_argv(proc[i].ZBX_P_PID, &argv, &argv_alloc, &argc))
{
collect_args(argv, argc, &args, &args_alloc);
if (zbx_regexp_match(args, proccomm, NULL) != NULL)
comm_ok = 1;
}
}
else
comm_ok = 1;
if (proc_ok && stat_ok && comm_ok)
proccount++;
}
zbx_free(proc);
zbx_free(argv);
zbx_free(args);
SET_UI64_RESULT(result, proccount);
return SYSINFO_RET_OK;
}
int PROC_MEM(const char *cmd, const char *param, unsigned flags, AGENT_RESULT *result)
{
char procname[MAX_STRING_LEN],
buffer[MAX_STRING_LEN],
proccomm[MAX_STRING_LEN];
int do_task, pagesize, count, i,
proc_ok, comm_ok;
double value = 0.0,
memsize = 0;
int proccount = 0;
size_t sz;
struct passwd *usrinfo;
#ifdef KERN_PROC2
int mib[6];
struct kinfo_proc2 *proc = NULL;
#else
int mib[4];
struct kinfo_proc *proc = NULL;
#endif
char **argv = NULL, *args = NULL;
size_t argv_alloc = 0, args_alloc = 0;
int argc;
if (num_param(param) > 4)
return SYSINFO_RET_FAIL;
if (0 != get_param(param, 1, procname, sizeof(procname)))
*procname = '\0';
else if (strlen(procname) > MAXCOMLEN)
procname[MAXCOMLEN] = '\0';
if (0 != get_param(param, 2, buffer, sizeof(buffer)))
*buffer = '\0';
if (*buffer != '\0')
{
usrinfo = getpwnam(buffer);
if (usrinfo == NULL) /* incorrect user name */
return SYSINFO_RET_FAIL;
}
else
usrinfo = NULL;
if (0 != get_param(param, 3, buffer, sizeof(buffer)))
*buffer = '\0';
if (*buffer != '\0')
{
if (0 == strcmp(buffer, "avg"))
do_task = DO_AVG;
else if (0 == strcmp(buffer, "max"))
do_task = DO_MAX;
else if (0 == strcmp(buffer, "min"))
do_task = DO_MIN;
else if (0 == strcmp(buffer, "sum"))
do_task = DO_SUM;
else
return SYSINFO_RET_FAIL;
}
else
do_task = DO_SUM;
if (0 != get_param(param, 4, proccomm, sizeof(proccomm)))
*proccomm = '\0';
pagesize = getpagesize();
mib[0] = CTL_KERN;
if (NULL != usrinfo)
{
mib[2] = KERN_PROC_UID;
mib[3] = usrinfo->pw_uid;
}
else
{
mib[2] = KERN_PROC_ALL;
mib[3] = 0;
}
#ifdef KERN_PROC2
mib[1] = KERN_PROC2;
mib[4] = sizeof(struct kinfo_proc2);
mib[5] = 0;
sz = 0;
if (0 != sysctl(mib, 6, NULL, &sz, NULL, 0))
return SYSINFO_RET_FAIL;
proc = (struct kinfo_proc2 *)zbx_malloc(proc, sz);
mib[5] = (int)(sz / sizeof(struct kinfo_proc2));
if (0 != sysctl(mib, 6, proc, &sz, NULL, 0))
{
zbx_free(proc);
return SYSINFO_RET_FAIL;
}
count = sz / sizeof(struct kinfo_proc2);
#else
mib[1] = KERN_PROC;
sz = 0;
if (0 != sysctl(mib, 4, NULL, &sz, NULL, 0))
return SYSINFO_RET_FAIL;
proc = (struct kinfo_proc *)zbx_malloc(proc, sz);
if (0 != sysctl(mib, 4, proc, &sz, NULL, 0))
{
zbx_free(proc);
return SYSINFO_RET_FAIL;
}
count = sz / sizeof(struct kinfo_proc);
#endif
for (i = 0; i < count; i++)
{
proc_ok = 0;
comm_ok = 0;
if (*procname == '\0' || 0 == strcmp(procname, proc[i].ZBX_P_COMM))
proc_ok = 1;
if (*proccomm != '\0')
{
if (SUCCEED == proc_argv(proc[i].ZBX_P_PID, &argv, &argv_alloc, &argc))
{
collect_args(argv, argc, &args, &args_alloc);
if (NULL != zbx_regexp_match(args, proccomm, NULL))
comm_ok = 1;
}
}
else
comm_ok = 1;
if (proc_ok && comm_ok)
{
value = proc[i].ZBX_P_VM_TSIZE
+ proc[i].ZBX_P_VM_DSIZE
+ proc[i].ZBX_P_VM_SSIZE;
value *= pagesize;
if (0 == proccount++)
memsize = value;
else
{
if (do_task == DO_MAX)
memsize = MAX(memsize, value);
else if (do_task == DO_MIN)
memsize = MIN(memsize, value);
else
memsize += value;
}
}
}
zbx_free(proc);
zbx_free(argv);
zbx_free(args);
if (do_task == DO_AVG)
SET_DBL_RESULT(result, proccount == 0 ? 0 : memsize/proccount);
else
SET_UI64_RESULT(result, memsize);
return SYSINFO_RET_OK;
}