treewide: cleanup malloc calls
[collectd.git] / src / processes.c
1 /**
2  * collectd - src/processes.c
3  * Copyright (C) 2005       Lyonel Vincent
4  * Copyright (C) 2006-2010  Florian octo Forster
5  * Copyright (C) 2008       Oleg King
6  * Copyright (C) 2009       Sebastian Harl
7  * Copyright (C) 2009       Andrés J. Díaz
8  * Copyright (C) 2009       Manuel Sanmartin
9  * Copyright (C) 2010       Clément Stenac
10  * Copyright (C) 2012       Cosmin Ioiart
11  *
12  * This program is free software; you can redistribute it and/or modify it
13  * under the terms of the GNU General Public License as published by the
14  * Free Software Foundation; either version 2 of the License, or (at your
15  * option) any later version.
16  *
17  * This program is distributed in the hope that it will be useful, but
18  * WITHOUT ANY WARRANTY; without even the implied warranty of
19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
20  * General Public License for more details.
21  *
22  * You should have received a copy of the GNU General Public License along
23  * with this program; if not, write to the Free Software Foundation, Inc.,
24  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
25  *
26  * Authors:
27  *   Lyonel Vincent <lyonel at ezix.org>
28  *   Florian octo Forster <octo at collectd.org>
29  *   Oleg King <king2 at kaluga.ru>
30  *   Sebastian Harl <sh at tokkee.org>
31  *   Andrés J. Díaz <ajdiaz at connectical.com>
32  *   Manuel Sanmartin
33  *   Clément Stenac <clement.stenac at diwi.org>
34  *   Cosmin Ioiart <cioiart at gmail.com>
35  **/
36
37 #include "collectd.h"
38 #include "common.h"
39 #include "plugin.h"
40 #include "configfile.h"
41
42 /* Include header files for the mach system, if they exist.. */
43 #if HAVE_THREAD_INFO
44 #  if HAVE_MACH_MACH_INIT_H
45 #    include <mach/mach_init.h>
46 #  endif
47 #  if HAVE_MACH_HOST_PRIV_H
48 #    include <mach/host_priv.h>
49 #  endif
50 #  if HAVE_MACH_MACH_ERROR_H
51 #    include <mach/mach_error.h>
52 #  endif
53 #  if HAVE_MACH_MACH_HOST_H
54 #    include <mach/mach_host.h>
55 #  endif
56 #  if HAVE_MACH_MACH_PORT_H
57 #    include <mach/mach_port.h>
58 #  endif
59 #  if HAVE_MACH_MACH_TYPES_H
60 #    include <mach/mach_types.h>
61 #  endif
62 #  if HAVE_MACH_MESSAGE_H
63 #    include <mach/message.h>
64 #  endif
65 #  if HAVE_MACH_PROCESSOR_SET_H
66 #    include <mach/processor_set.h>
67 #  endif
68 #  if HAVE_MACH_TASK_H
69 #    include <mach/task.h>
70 #  endif
71 #  if HAVE_MACH_THREAD_ACT_H
72 #    include <mach/thread_act.h>
73 #  endif
74 #  if HAVE_MACH_VM_REGION_H
75 #    include <mach/vm_region.h>
76 #  endif
77 #  if HAVE_MACH_VM_MAP_H
78 #    include <mach/vm_map.h>
79 #  endif
80 #  if HAVE_MACH_VM_PROT_H
81 #    include <mach/vm_prot.h>
82 #  endif
83 #  if HAVE_SYS_SYSCTL_H
84 #    include <sys/sysctl.h>
85 #  endif
86 /* #endif HAVE_THREAD_INFO */
87
88 #elif KERNEL_LINUX
89 #  if HAVE_LINUX_CONFIG_H
90 #    include <linux/config.h>
91 #  endif
92 #  ifndef CONFIG_HZ
93 #    define CONFIG_HZ 100
94 #  endif
95 /* #endif KERNEL_LINUX */
96
97 #elif HAVE_LIBKVM_GETPROCS && (HAVE_STRUCT_KINFO_PROC_FREEBSD || HAVE_STRUCT_KINFO_PROC_OPENBSD)
98 #  include <kvm.h>
99 #  include <sys/param.h>
100 #  include <sys/sysctl.h>
101 #  include <sys/user.h>
102 #  include <sys/proc.h>
103 /* #endif HAVE_LIBKVM_GETPROCS && (HAVE_STRUCT_KINFO_PROC_FREEBSD || HAVE_STRUCT_KINFO_PROC_OPENBSD) */
104
105 #elif HAVE_PROCINFO_H
106 #  include <procinfo.h>
107 #  include <sys/types.h>
108
109 #define MAXPROCENTRY 32
110 #define MAXTHRDENTRY 16
111 #define MAXARGLN 1024
112 /* #endif HAVE_PROCINFO_H */
113
114 #elif KERNEL_SOLARIS
115 /* Hack: Avoid #error when building a 32-bit binary with
116  * _FILE_OFFSET_BITS=64. There is a reason for this #error, as one
117  * of the structures in <sys/procfs.h> uses an off_t, but that
118  * isn't relevant to our usage of procfs. */
119 #if !defined(_LP64) && _FILE_OFFSET_BITS == 64
120 #  define SAVE_FOB_64
121 #  undef _FILE_OFFSET_BITS
122 #endif
123
124 # include <procfs.h>
125
126 #ifdef SAVE_FOB_64
127 #  define _FILE_OFFSET_BITS 64
128 #  undef SAVE_FOB_64
129 #endif
130
131 # include <dirent.h>
132 /* #endif KERNEL_SOLARIS */
133
134 #else
135 # error "No applicable input method."
136 #endif
137
138 #if HAVE_REGEX_H
139 # include <regex.h>
140 #endif
141
142 #if HAVE_KSTAT_H
143 # include <kstat.h>
144 #endif
145
146 #ifndef CMDLINE_BUFFER_SIZE
147 # if defined(ARG_MAX) && (ARG_MAX < 4096)
148 #  define CMDLINE_BUFFER_SIZE ARG_MAX
149 # else
150 #  define CMDLINE_BUFFER_SIZE 4096
151 # endif
152 #endif
153
154 typedef struct procstat_entry_s
155 {
156         unsigned long id;
157         unsigned long age;
158
159         unsigned long num_proc;
160         unsigned long num_lwp;
161         unsigned long vmem_size;
162         unsigned long vmem_rss;
163         unsigned long vmem_data;
164         unsigned long vmem_code;
165         unsigned long stack_size;
166
167         unsigned long vmem_minflt;
168         unsigned long vmem_majflt;
169         derive_t      vmem_minflt_counter;
170         derive_t      vmem_majflt_counter;
171
172         unsigned long cpu_user;
173         unsigned long cpu_system;
174         derive_t      cpu_user_counter;
175         derive_t      cpu_system_counter;
176
177         /* io data */
178         derive_t io_rchar;
179         derive_t io_wchar;
180         derive_t io_syscr;
181         derive_t io_syscw;
182
183         derive_t cswitch_vol;
184         derive_t cswitch_invol;
185
186         struct procstat_entry_s *next;
187 } procstat_entry_t;
188
189 #define PROCSTAT_NAME_LEN 256
190 typedef struct procstat
191 {
192         char          name[PROCSTAT_NAME_LEN];
193 #if HAVE_REGEX_H
194         regex_t *re;
195 #endif
196
197         unsigned long num_proc;
198         unsigned long num_lwp;
199         unsigned long vmem_size;
200         unsigned long vmem_rss;
201         unsigned long vmem_data;
202         unsigned long vmem_code;
203         unsigned long stack_size;
204
205         derive_t vmem_minflt_counter;
206         derive_t vmem_majflt_counter;
207
208         derive_t cpu_user_counter;
209         derive_t cpu_system_counter;
210
211         /* io data */
212         derive_t io_rchar;
213         derive_t io_wchar;
214         derive_t io_syscr;
215         derive_t io_syscw;
216
217         derive_t cswitch_vol;
218         derive_t cswitch_invol;
219
220         struct procstat   *next;
221         struct procstat_entry_s *instances;
222 } procstat_t;
223
224 static procstat_t *list_head_g = NULL;
225
226 static _Bool report_ctx_switch = 0;
227
228 #if HAVE_THREAD_INFO
229 static mach_port_t port_host_self;
230 static mach_port_t port_task_self;
231
232 static processor_set_name_array_t pset_list;
233 static mach_msg_type_number_t     pset_list_len;
234 /* #endif HAVE_THREAD_INFO */
235
236 #elif KERNEL_LINUX
237 static long pagesize_g;
238 /* #endif KERNEL_LINUX */
239
240 #elif HAVE_LIBKVM_GETPROCS && (HAVE_STRUCT_KINFO_PROC_FREEBSD || HAVE_STRUCT_KINFO_PROC_OPENBSD)
241 static int pagesize;
242 /* #endif HAVE_LIBKVM_GETPROCS && (HAVE_STRUCT_KINFO_PROC_FREEBSD || HAVE_STRUCT_KINFO_PROC_OPENBSD) */
243
244 #elif HAVE_PROCINFO_H
245 static  struct procentry64 procentry[MAXPROCENTRY];
246 static  struct thrdentry64 thrdentry[MAXTHRDENTRY];
247 static int pagesize;
248
249 #ifndef _AIXVERSION_610
250 int     getprocs64 (void *procsinfo, int sizproc, void *fdsinfo, int sizfd, pid_t *index, int count);
251 int     getthrds64( pid_t, void *, int, tid64_t *, int );
252 #endif
253 int getargs (void *processBuffer, int bufferLen, char *argsBuffer, int argsLen);
254 #endif /* HAVE_PROCINFO_H */
255
256 /* put name of process from config to list_head_g tree
257  * list_head_g is a list of 'procstat_t' structs with
258  * processes names we want to watch */
259 static void ps_list_register (const char *name, const char *regexp)
260 {
261         procstat_t *new;
262         procstat_t *ptr;
263         int status;
264
265         new = malloc (sizeof (*new));
266         if (new == NULL)
267         {
268                 ERROR ("processes plugin: ps_list_register: malloc failed.");
269                 return;
270         }
271         memset (new, 0, sizeof (procstat_t));
272         sstrncpy (new->name, name, sizeof (new->name));
273
274 #if HAVE_REGEX_H
275         if (regexp != NULL)
276         {
277                 DEBUG ("ProcessMatch: adding \"%s\" as criteria to process %s.", regexp, name);
278                 new->re = malloc (sizeof (*new->re));
279                 if (new->re == NULL)
280                 {
281                         ERROR ("processes plugin: ps_list_register: malloc failed.");
282                         sfree (new);
283                         return;
284                 }
285
286                 status = regcomp (new->re, regexp, REG_EXTENDED | REG_NOSUB);
287                 if (status != 0)
288                 {
289                         DEBUG ("ProcessMatch: compiling the regular expression \"%s\" failed.", regexp);
290                         sfree (new->re);
291                         sfree (new);
292                         return;
293                 }
294         }
295 #else
296         if (regexp != NULL)
297         {
298                 ERROR ("processes plugin: ps_list_register: "
299                                 "Regular expression \"%s\" found in config "
300                                 "file, but support for regular expressions "
301                                 "has been disabled at compile time.",
302                                 regexp);
303                 sfree (new);
304                 return;
305         }
306 #endif
307
308         for (ptr = list_head_g; ptr != NULL; ptr = ptr->next)
309         {
310                 if (strcmp (ptr->name, name) == 0)
311                 {
312                         WARNING ("processes plugin: You have configured more "
313                                         "than one `Process' or "
314                                         "`ProcessMatch' with the same name. "
315                                         "All but the first setting will be "
316                                         "ignored.");
317 #if HAVE_REGEX_H
318                         sfree (new->re);
319 #endif
320                         sfree (new);
321                         return;
322                 }
323
324                 if (ptr->next == NULL)
325                         break;
326         }
327
328         if (ptr == NULL)
329                 list_head_g = new;
330         else
331                 ptr->next = new;
332 } /* void ps_list_register */
333
334 /* try to match name against entry, returns 1 if success */
335 static int ps_list_match (const char *name, const char *cmdline, procstat_t *ps)
336 {
337 #if HAVE_REGEX_H
338         if (ps->re != NULL)
339         {
340                 int status;
341                 const char *str;
342
343                 str = cmdline;
344                 if ((str == NULL) || (str[0] == 0))
345                         str = name;
346
347                 assert (str != NULL);
348
349                 status = regexec (ps->re, str,
350                                 /* nmatch = */ 0,
351                                 /* pmatch = */ NULL,
352                                 /* eflags = */ 0);
353                 if (status == 0)
354                         return (1);
355         }
356         else
357 #endif
358         if (strcmp (ps->name, name) == 0)
359                 return (1);
360
361         return (0);
362 } /* int ps_list_match */
363
364 static void ps_update_counter (
365         _Bool init,
366         derive_t *group_counter,
367         derive_t *curr_counter, unsigned long *curr_value,
368         derive_t new_counter, unsigned long new_value) {
369     if (init)
370     {
371         *curr_value = new_value;
372         *curr_counter += new_value;
373         *group_counter += new_value;
374         return;
375     }
376
377     if (new_counter < *curr_counter)
378     {
379         *curr_value = new_counter + (ULONG_MAX - *curr_counter);
380     }
381     else
382     {
383         *curr_value = new_counter - *curr_counter;
384     }
385     *curr_counter = new_counter;
386     *group_counter += *curr_value;
387 }
388
389 /* add process entry to 'instances' of process 'name' (or refresh it) */
390 static void ps_list_add (const char *name, const char *cmdline, procstat_entry_t *entry)
391 {
392         procstat_t *ps;
393         procstat_entry_t *pse;
394
395         if (entry->id == 0)
396                 return;
397
398         for (ps = list_head_g; ps != NULL; ps = ps->next)
399         {
400         _Bool want_init;
401
402                 if ((ps_list_match (name, cmdline, ps)) == 0)
403                         continue;
404
405                 for (pse = ps->instances; pse != NULL; pse = pse->next)
406                         if ((pse->id == entry->id) || (pse->next == NULL))
407                                 break;
408
409                 if ((pse == NULL) || (pse->id != entry->id))
410                 {
411                         procstat_entry_t *new;
412
413                         new = malloc (sizeof (*new));
414                         if (new == NULL)
415                                 return;
416                         memset (new, 0, sizeof (procstat_entry_t));
417                         new->id = entry->id;
418
419                         if (pse == NULL)
420                                 ps->instances = new;
421                         else
422                                 pse->next = new;
423
424                         pse = new;
425                 }
426
427                 pse->age = 0;
428                 pse->num_proc   = entry->num_proc;
429                 pse->num_lwp    = entry->num_lwp;
430                 pse->vmem_size  = entry->vmem_size;
431                 pse->vmem_rss   = entry->vmem_rss;
432                 pse->vmem_data  = entry->vmem_data;
433                 pse->vmem_code  = entry->vmem_code;
434                 pse->stack_size = entry->stack_size;
435                 pse->io_rchar   = entry->io_rchar;
436                 pse->io_wchar   = entry->io_wchar;
437                 pse->io_syscr   = entry->io_syscr;
438                 pse->io_syscw   = entry->io_syscw;
439                 pse->cswitch_vol   = entry->cswitch_vol;
440                 pse->cswitch_invol = entry->cswitch_invol;
441
442                 ps->num_proc   += pse->num_proc;
443                 ps->num_lwp    += pse->num_lwp;
444                 ps->vmem_size  += pse->vmem_size;
445                 ps->vmem_rss   += pse->vmem_rss;
446                 ps->vmem_data  += pse->vmem_data;
447                 ps->vmem_code  += pse->vmem_code;
448                 ps->stack_size += pse->stack_size;
449
450                 ps->io_rchar   += ((pse->io_rchar == -1)?0:pse->io_rchar);
451                 ps->io_wchar   += ((pse->io_wchar == -1)?0:pse->io_wchar);
452                 ps->io_syscr   += ((pse->io_syscr == -1)?0:pse->io_syscr);
453                 ps->io_syscw   += ((pse->io_syscw == -1)?0:pse->io_syscw);
454
455                 ps->cswitch_vol   += ((pse->cswitch_vol == -1)?0:pse->cswitch_vol);
456                 ps->cswitch_invol += ((pse->cswitch_invol == -1)?0:pse->cswitch_invol);
457
458                 want_init = (entry->vmem_minflt_counter == 0)
459                                 && (entry->vmem_majflt_counter == 0);
460                 ps_update_counter (want_init,
461                                 &ps->vmem_minflt_counter,
462                                 &pse->vmem_minflt_counter, &pse->vmem_minflt,
463                                 entry->vmem_minflt_counter, entry->vmem_minflt);
464                 ps_update_counter (want_init,
465                                 &ps->vmem_majflt_counter,
466                                 &pse->vmem_majflt_counter, &pse->vmem_majflt,
467                                 entry->vmem_majflt_counter, entry->vmem_majflt);
468
469                 want_init = (entry->cpu_user_counter == 0)
470                                 && (entry->cpu_system_counter == 0);
471                 ps_update_counter (want_init,
472                                 &ps->cpu_user_counter,
473                                 &pse->cpu_user_counter, &pse->cpu_user,
474                                 entry->cpu_user_counter, entry->cpu_user);
475                 ps_update_counter (want_init,
476                                 &ps->cpu_system_counter,
477                                 &pse->cpu_system_counter, &pse->cpu_system,
478                                 entry->cpu_system_counter, entry->cpu_system);
479         }
480 }
481
482 /* remove old entries from instances of processes in list_head_g */
483 static void ps_list_reset (void)
484 {
485         procstat_t *ps;
486         procstat_entry_t *pse;
487         procstat_entry_t *pse_prev;
488
489         for (ps = list_head_g; ps != NULL; ps = ps->next)
490         {
491                 ps->num_proc    = 0;
492                 ps->num_lwp     = 0;
493                 ps->vmem_size   = 0;
494                 ps->vmem_rss    = 0;
495                 ps->vmem_data   = 0;
496                 ps->vmem_code   = 0;
497                 ps->stack_size  = 0;
498                 ps->io_rchar = -1;
499                 ps->io_wchar = -1;
500                 ps->io_syscr = -1;
501                 ps->io_syscw = -1;
502                 ps->cswitch_vol   = -1;
503                 ps->cswitch_invol = -1;
504
505                 pse_prev = NULL;
506                 pse = ps->instances;
507                 while (pse != NULL)
508                 {
509                         if (pse->age > 10)
510                         {
511                                 DEBUG ("Removing this procstat entry cause it's too old: "
512                                                 "id = %lu; name = %s;",
513                                                 pse->id, ps->name);
514
515                                 if (pse_prev == NULL)
516                                 {
517                                         ps->instances = pse->next;
518                                         free (pse);
519                                         pse = ps->instances;
520                                 }
521                                 else
522                                 {
523                                         pse_prev->next = pse->next;
524                                         free (pse);
525                                         pse = pse_prev->next;
526                                 }
527                         }
528                         else
529                         {
530                                 pse->age++;
531                                 pse_prev = pse;
532                                 pse = pse->next;
533                         }
534                 } /* while (pse != NULL) */
535         } /* for (ps = list_head_g; ps != NULL; ps = ps->next) */
536 }
537
538 /* put all pre-defined 'Process' names from config to list_head_g tree */
539 static int ps_config (oconfig_item_t *ci)
540 {
541         int i;
542
543         for (i = 0; i < ci->children_num; ++i) {
544                 oconfig_item_t *c = ci->children + i;
545
546                 if (strcasecmp (c->key, "Process") == 0)
547                 {
548                         if ((c->values_num != 1)
549                                         || (OCONFIG_TYPE_STRING != c->values[0].type)) {
550                                 ERROR ("processes plugin: `Process' expects exactly "
551                                                 "one string argument (got %i).",
552                                                 c->values_num);
553                                 continue;
554                         }
555
556                         if (c->children_num != 0) {
557                                 WARNING ("processes plugin: the `Process' config option "
558                                                 "does not expect any child elements -- ignoring "
559                                                 "content (%i elements) of the <Process '%s'> block.",
560                                                 c->children_num, c->values[0].value.string);
561                         }
562
563                         ps_list_register (c->values[0].value.string, NULL);
564                 }
565                 else if (strcasecmp (c->key, "ProcessMatch") == 0)
566                 {
567                         if ((c->values_num != 2)
568                                         || (OCONFIG_TYPE_STRING != c->values[0].type)
569                                         || (OCONFIG_TYPE_STRING != c->values[1].type))
570                         {
571                                 ERROR ("processes plugin: `ProcessMatch' needs exactly "
572                                                 "two string arguments (got %i).",
573                                                 c->values_num);
574                                 continue;
575                         }
576
577                         if (c->children_num != 0) {
578                                 WARNING ("processes plugin: the `ProcessMatch' config option "
579                                                 "does not expect any child elements -- ignoring "
580                                                 "content (%i elements) of the <ProcessMatch '%s' '%s'> "
581                                                 "block.", c->children_num, c->values[0].value.string,
582                                                 c->values[1].value.string);
583                         }
584
585                         ps_list_register (c->values[0].value.string,
586                                         c->values[1].value.string);
587                 }
588                 else if (strcasecmp (c->key, "CollectContextSwitch") == 0)
589                 {
590                         cf_util_get_boolean (c, &report_ctx_switch);
591                 }
592                 else
593                 {
594                         ERROR ("processes plugin: The `%s' configuration option is not "
595                                         "understood and will be ignored.", c->key);
596                         continue;
597                 }
598         }
599
600         return (0);
601 }
602
603 static int ps_init (void)
604 {
605 #if HAVE_THREAD_INFO
606         kern_return_t status;
607
608         port_host_self = mach_host_self ();
609         port_task_self = mach_task_self ();
610
611         if (pset_list != NULL)
612         {
613                 vm_deallocate (port_task_self,
614                                 (vm_address_t) pset_list,
615                                 pset_list_len * sizeof (processor_set_t));
616                 pset_list = NULL;
617                 pset_list_len = 0;
618         }
619
620         if ((status = host_processor_sets (port_host_self,
621                                         &pset_list,
622                                         &pset_list_len)) != KERN_SUCCESS)
623         {
624                 ERROR ("host_processor_sets failed: %s\n",
625                                 mach_error_string (status));
626                 pset_list = NULL;
627                 pset_list_len = 0;
628                 return (-1);
629         }
630 /* #endif HAVE_THREAD_INFO */
631
632 #elif KERNEL_LINUX
633         pagesize_g = sysconf(_SC_PAGESIZE);
634         DEBUG ("pagesize_g = %li; CONFIG_HZ = %i;",
635                         pagesize_g, CONFIG_HZ);
636 /* #endif KERNEL_LINUX */
637
638 #elif HAVE_LIBKVM_GETPROCS && (HAVE_STRUCT_KINFO_PROC_FREEBSD || HAVE_STRUCT_KINFO_PROC_OPENBSD)
639         pagesize = getpagesize();
640 /* #endif HAVE_LIBKVM_GETPROCS && (HAVE_STRUCT_KINFO_PROC_FREEBSD || HAVE_STRUCT_KINFO_PROC_OPENBSD) */
641
642 #elif HAVE_PROCINFO_H
643         pagesize = getpagesize();
644 #endif /* HAVE_PROCINFO_H */
645
646         return (0);
647 } /* int ps_init */
648
649 /* submit global state (e.g.: qty of zombies, running, etc..) */
650 static void ps_submit_state (const char *state, double value)
651 {
652         value_t values[1];
653         value_list_t vl = VALUE_LIST_INIT;
654
655         values[0].gauge = value;
656
657         vl.values = values;
658         vl.values_len = 1;
659         sstrncpy (vl.host, hostname_g, sizeof (vl.host));
660         sstrncpy (vl.plugin, "processes", sizeof (vl.plugin));
661         sstrncpy (vl.plugin_instance, "", sizeof (vl.plugin_instance));
662         sstrncpy (vl.type, "ps_state", sizeof (vl.type));
663         sstrncpy (vl.type_instance, state, sizeof (vl.type_instance));
664
665         plugin_dispatch_values (&vl);
666 }
667
668 /* submit info about specific process (e.g.: memory taken, cpu usage, etc..) */
669 static void ps_submit_proc_list (procstat_t *ps)
670 {
671         value_t values[2];
672         value_list_t vl = VALUE_LIST_INIT;
673
674         vl.values = values;
675         vl.values_len = 2;
676         sstrncpy (vl.host, hostname_g, sizeof (vl.host));
677         sstrncpy (vl.plugin, "processes", sizeof (vl.plugin));
678         sstrncpy (vl.plugin_instance, ps->name, sizeof (vl.plugin_instance));
679
680         sstrncpy (vl.type, "ps_vm", sizeof (vl.type));
681         vl.values[0].gauge = ps->vmem_size;
682         vl.values_len = 1;
683         plugin_dispatch_values (&vl);
684
685         sstrncpy (vl.type, "ps_rss", sizeof (vl.type));
686         vl.values[0].gauge = ps->vmem_rss;
687         vl.values_len = 1;
688         plugin_dispatch_values (&vl);
689
690         sstrncpy (vl.type, "ps_data", sizeof (vl.type));
691         vl.values[0].gauge = ps->vmem_data;
692         vl.values_len = 1;
693         plugin_dispatch_values (&vl);
694
695         sstrncpy (vl.type, "ps_code", sizeof (vl.type));
696         vl.values[0].gauge = ps->vmem_code;
697         vl.values_len = 1;
698         plugin_dispatch_values (&vl);
699
700         sstrncpy (vl.type, "ps_stacksize", sizeof (vl.type));
701         vl.values[0].gauge = ps->stack_size;
702         vl.values_len = 1;
703         plugin_dispatch_values (&vl);
704
705         sstrncpy (vl.type, "ps_cputime", sizeof (vl.type));
706         vl.values[0].derive = ps->cpu_user_counter;
707         vl.values[1].derive = ps->cpu_system_counter;
708         vl.values_len = 2;
709         plugin_dispatch_values (&vl);
710
711         sstrncpy (vl.type, "ps_count", sizeof (vl.type));
712         vl.values[0].gauge = ps->num_proc;
713         vl.values[1].gauge = ps->num_lwp;
714         vl.values_len = 2;
715         plugin_dispatch_values (&vl);
716
717         sstrncpy (vl.type, "ps_pagefaults", sizeof (vl.type));
718         vl.values[0].derive = ps->vmem_minflt_counter;
719         vl.values[1].derive = ps->vmem_majflt_counter;
720         vl.values_len = 2;
721         plugin_dispatch_values (&vl);
722
723         if ( (ps->io_rchar != -1) && (ps->io_wchar != -1) )
724         {
725                 sstrncpy (vl.type, "ps_disk_octets", sizeof (vl.type));
726                 vl.values[0].derive = ps->io_rchar;
727                 vl.values[1].derive = ps->io_wchar;
728                 vl.values_len = 2;
729                 plugin_dispatch_values (&vl);
730         }
731
732         if ( (ps->io_syscr != -1) && (ps->io_syscw != -1) )
733         {
734                 sstrncpy (vl.type, "ps_disk_ops", sizeof (vl.type));
735                 vl.values[0].derive = ps->io_syscr;
736                 vl.values[1].derive = ps->io_syscw;
737                 vl.values_len = 2;
738                 plugin_dispatch_values (&vl);
739         }
740
741         if ( report_ctx_switch )
742         {
743                 sstrncpy (vl.type, "contextswitch", sizeof (vl.type));
744                 sstrncpy (vl.type_instance, "voluntary", sizeof (vl.type_instance));
745                 vl.values[0].derive = ps->cswitch_vol;
746                 vl.values_len = 1;
747                 plugin_dispatch_values (&vl);
748
749                 sstrncpy (vl.type, "contextswitch", sizeof (vl.type));
750                 sstrncpy (vl.type_instance, "involuntary", sizeof (vl.type_instance));
751                 vl.values[0].derive = ps->cswitch_invol;
752                 vl.values_len = 1;
753                 plugin_dispatch_values (&vl);
754         }
755
756         DEBUG ("name = %s; num_proc = %lu; num_lwp = %lu; "
757                         "vmem_size = %lu; vmem_rss = %lu; vmem_data = %lu; "
758                         "vmem_code = %lu; "
759                         "vmem_minflt_counter = %"PRIi64"; vmem_majflt_counter = %"PRIi64"; "
760                         "cpu_user_counter = %"PRIi64"; cpu_system_counter = %"PRIi64"; "
761                         "io_rchar = %"PRIi64"; io_wchar = %"PRIi64"; "
762                         "io_syscr = %"PRIi64"; io_syscw = %"PRIi64"; "
763                         "cswitch_vol = %"PRIi64"; cswitch_invol = %"PRIi64";",
764                         ps->name, ps->num_proc, ps->num_lwp,
765                         ps->vmem_size, ps->vmem_rss,
766                         ps->vmem_data, ps->vmem_code,
767                         ps->vmem_minflt_counter, ps->vmem_majflt_counter,
768                         ps->cpu_user_counter, ps->cpu_system_counter,
769                         ps->io_rchar, ps->io_wchar, ps->io_syscr, ps->io_syscw,
770                         ps->cswitch_vol, ps->cswitch_invol);
771 } /* void ps_submit_proc_list */
772
773 #if KERNEL_LINUX || KERNEL_SOLARIS
774 static void ps_submit_fork_rate (derive_t value)
775 {
776         value_t values[1];
777         value_list_t vl = VALUE_LIST_INIT;
778
779         values[0].derive = value;
780
781         vl.values = values;
782         vl.values_len = 1;
783         sstrncpy(vl.host, hostname_g, sizeof (vl.host));
784         sstrncpy(vl.plugin, "processes", sizeof (vl.plugin));
785         sstrncpy(vl.plugin_instance, "", sizeof (vl.plugin_instance));
786         sstrncpy(vl.type, "fork_rate", sizeof (vl.type));
787         sstrncpy(vl.type_instance, "", sizeof (vl.type_instance));
788
789         plugin_dispatch_values(&vl);
790 }
791 #endif /* KERNEL_LINUX || KERNEL_SOLARIS*/
792
793 /* ------- additional functions for KERNEL_LINUX/HAVE_THREAD_INFO ------- */
794 #if KERNEL_LINUX
795 static procstat_t *ps_read_tasks_status (long pid, procstat_t *ps)
796 {
797         char           dirname[64];
798         DIR           *dh;
799         char           filename[64];
800         FILE          *fh;
801         struct dirent *ent;
802         derive_t cswitch_vol = 0;
803         derive_t cswitch_invol = 0;
804         char buffer[1024];
805         char *fields[8];
806         int numfields;
807
808         ssnprintf (dirname, sizeof (dirname), "/proc/%li/task", pid);
809
810         if ((dh = opendir (dirname)) == NULL)
811         {
812                 DEBUG ("Failed to open directory `%s'", dirname);
813                 return (NULL);
814         }
815
816         while ((ent = readdir (dh)) != NULL)
817         {
818                 char *tpid;
819
820                 if (!isdigit ((int) ent->d_name[0]))
821                         continue;
822
823                 tpid = ent->d_name;
824
825                 ssnprintf (filename, sizeof (filename), "/proc/%li/task/%s/status", pid, tpid);
826                 if ((fh = fopen (filename, "r")) == NULL)
827                 {
828                         DEBUG ("Failed to open file `%s'", filename);
829                         continue;
830                 }
831
832                 while (fgets (buffer, sizeof(buffer), fh) != NULL)
833                 {
834                         derive_t tmp;
835                         char *endptr;
836
837                         if (strncmp (buffer, "voluntary_ctxt_switches", 23) != 0
838                                 && strncmp (buffer, "nonvoluntary_ctxt_switches", 26) != 0)
839                                 continue;
840
841                         numfields = strsplit (buffer, fields,
842                                 STATIC_ARRAY_SIZE (fields));
843
844                         if (numfields < 2)
845                                 continue;
846
847                         errno = 0;
848                         endptr = NULL;
849                         tmp = (derive_t) strtoll (fields[1], &endptr, /* base = */ 10);
850                         if ((errno == 0) && (endptr != fields[1]))
851                         {
852                                 if (strncmp (buffer, "voluntary_ctxt_switches", 23) == 0)
853                                 {
854                                         cswitch_vol += tmp;
855                                 }
856                                 else if (strncmp (buffer, "nonvoluntary_ctxt_switches", 26) == 0)
857                                 {
858                                         cswitch_invol += tmp;
859                                 }
860                         }
861                 } /* while (fgets) */
862
863                 if (fclose (fh))
864                 {
865                         char errbuf[1024];
866                                 WARNING ("processes: fclose: %s",
867                                         sstrerror (errno, errbuf, sizeof (errbuf)));
868                 }
869         }
870         closedir (dh);
871
872         ps->cswitch_vol = cswitch_vol;
873         ps->cswitch_invol = cswitch_invol;
874
875         return (ps);
876 } /* int *ps_read_tasks_status */
877
878 /* Read data from /proc/pid/status */
879 static procstat_t *ps_read_status (long pid, procstat_t *ps)
880 {
881         FILE *fh;
882         char buffer[1024];
883         char filename[64];
884         unsigned long lib = 0;
885         unsigned long exe = 0;
886         unsigned long data = 0;
887         unsigned long threads = 0;
888         char *fields[8];
889         int numfields;
890
891         ssnprintf (filename, sizeof (filename), "/proc/%li/status", pid);
892         if ((fh = fopen (filename, "r")) == NULL)
893                 return (NULL);
894
895         while (fgets (buffer, sizeof(buffer), fh) != NULL)
896         {
897                 unsigned long tmp;
898                 char *endptr;
899
900                 if (strncmp (buffer, "Vm", 2) != 0
901                                 && strncmp (buffer, "Threads", 7) != 0)
902                         continue;
903
904                 numfields = strsplit (buffer, fields,
905                                 STATIC_ARRAY_SIZE (fields));
906
907                 if (numfields < 2)
908                         continue;
909
910                 errno = 0;
911                 endptr = NULL;
912                 tmp = strtoul (fields[1], &endptr, /* base = */ 10);
913                 if ((errno == 0) && (endptr != fields[1]))
914                 {
915                         if (strncmp (buffer, "VmData", 6) == 0)
916                         {
917                                 data = tmp;
918                         }
919                         else if (strncmp (buffer, "VmLib", 5) == 0)
920                         {
921                                 lib = tmp;
922                         }
923                         else if  (strncmp(buffer, "VmExe", 5) == 0)
924                         {
925                                 exe = tmp;
926                         }
927                         else if  (strncmp(buffer, "Threads", 7) == 0)
928                         {
929                                 threads = tmp;
930                         }
931                 }
932         } /* while (fgets) */
933
934         if (fclose (fh))
935         {
936                 char errbuf[1024];
937                 WARNING ("processes: fclose: %s",
938                                 sstrerror (errno, errbuf, sizeof (errbuf)));
939         }
940
941         ps->vmem_data = data * 1024;
942         ps->vmem_code = (exe + lib) * 1024;
943         if (threads != 0)
944                 ps->num_lwp = threads;
945
946         return (ps);
947 } /* procstat_t *ps_read_vmem */
948
949 static procstat_t *ps_read_io (long pid, procstat_t *ps)
950 {
951         FILE *fh;
952         char buffer[1024];
953         char filename[64];
954
955         char *fields[8];
956         int numfields;
957
958         ssnprintf (filename, sizeof (filename), "/proc/%li/io", pid);
959         if ((fh = fopen (filename, "r")) == NULL)
960                 return (NULL);
961
962         while (fgets (buffer, sizeof (buffer), fh) != NULL)
963         {
964                 derive_t *val = NULL;
965                 long long tmp;
966                 char *endptr;
967
968                 if (strncasecmp (buffer, "rchar:", 6) == 0)
969                         val = &(ps->io_rchar);
970                 else if (strncasecmp (buffer, "wchar:", 6) == 0)
971                         val = &(ps->io_wchar);
972                 else if (strncasecmp (buffer, "syscr:", 6) == 0)
973                         val = &(ps->io_syscr);
974                 else if (strncasecmp (buffer, "syscw:", 6) == 0)
975                         val = &(ps->io_syscw);
976                 else
977                         continue;
978
979                 numfields = strsplit (buffer, fields,
980                                 STATIC_ARRAY_SIZE (fields));
981
982                 if (numfields < 2)
983                         continue;
984
985                 errno = 0;
986                 endptr = NULL;
987                 tmp = strtoll (fields[1], &endptr, /* base = */ 10);
988                 if ((errno != 0) || (endptr == fields[1]))
989                         *val = -1;
990                 else
991                         *val = (derive_t) tmp;
992         } /* while (fgets) */
993
994         if (fclose (fh))
995         {
996                 char errbuf[1024];
997                 WARNING ("processes: fclose: %s",
998                                 sstrerror (errno, errbuf, sizeof (errbuf)));
999         }
1000
1001         return (ps);
1002 } /* procstat_t *ps_read_io */
1003
1004 static int ps_read_process (long pid, procstat_t *ps, char *state)
1005 {
1006         char  filename[64];
1007         char  buffer[1024];
1008
1009         char *fields[64];
1010         char  fields_len;
1011
1012         size_t buffer_len;
1013
1014         char  *buffer_ptr;
1015         size_t name_start_pos;
1016         size_t name_end_pos;
1017         size_t name_len;
1018
1019         derive_t cpu_user_counter;
1020         derive_t cpu_system_counter;
1021         long long unsigned vmem_size;
1022         long long unsigned vmem_rss;
1023         long long unsigned stack_size;
1024
1025         ssize_t status;
1026
1027         memset (ps, 0, sizeof (procstat_t));
1028
1029         ssnprintf (filename, sizeof (filename), "/proc/%li/stat", pid);
1030
1031         status = read_file_contents (filename, buffer, sizeof(buffer) - 1);
1032         if (status <= 0)
1033                 return (-1);
1034         buffer_len = (size_t) status;
1035         buffer[buffer_len] = 0;
1036
1037         /* The name of the process is enclosed in parens. Since the name can
1038          * contain parens itself, spaces, numbers and pretty much everything
1039          * else, use these to determine the process name. We don't use
1040          * strchr(3) and strrchr(3) to avoid pointer arithmetic which would
1041          * otherwise be required to determine name_len. */
1042         name_start_pos = 0;
1043         while ((buffer[name_start_pos] != '(')
1044                         && (name_start_pos < buffer_len))
1045                 name_start_pos++;
1046
1047         name_end_pos = buffer_len;
1048         while ((buffer[name_end_pos] != ')')
1049                         && (name_end_pos > 0))
1050                 name_end_pos--;
1051
1052         /* Either '(' or ')' is not found or they are in the wrong order.
1053          * Anyway, something weird that shouldn't happen ever. */
1054         if (name_start_pos >= name_end_pos)
1055         {
1056                 ERROR ("processes plugin: name_start_pos = %zu >= name_end_pos = %zu",
1057                                 name_start_pos, name_end_pos);
1058                 return (-1);
1059         }
1060
1061         name_len = (name_end_pos - name_start_pos) - 1;
1062         if (name_len >= sizeof (ps->name))
1063                 name_len = sizeof (ps->name) - 1;
1064
1065         sstrncpy (ps->name, &buffer[name_start_pos + 1], name_len + 1);
1066
1067         if ((buffer_len - name_end_pos) < 2)
1068                 return (-1);
1069         buffer_ptr = &buffer[name_end_pos + 2];
1070
1071         fields_len = strsplit (buffer_ptr, fields, STATIC_ARRAY_SIZE (fields));
1072         if (fields_len < 22)
1073         {
1074                 DEBUG ("processes plugin: ps_read_process (pid = %li):"
1075                                 " `%s' has only %i fields..",
1076                                 pid, filename, fields_len);
1077                 return (-1);
1078         }
1079
1080         *state = fields[0][0];
1081
1082         if (*state == 'Z')
1083         {
1084                 ps->num_lwp  = 0;
1085                 ps->num_proc = 0;
1086         }
1087         else
1088         {
1089                 ps->num_lwp = strtoul (fields[17], /* endptr = */ NULL, /* base = */ 10);
1090                 if ((ps_read_status(pid, ps)) == NULL)
1091                 {
1092                         /* No VMem data */
1093                         ps->vmem_data = -1;
1094                         ps->vmem_code = -1;
1095                         DEBUG("ps_read_process: did not get vmem data for pid %li", pid);
1096                 }
1097                 if (ps->num_lwp <= 0)
1098                         ps->num_lwp = 1;
1099                 ps->num_proc = 1;
1100         }
1101
1102         /* Leave the rest at zero if this is only a zombi */
1103         if (ps->num_proc == 0)
1104         {
1105                 DEBUG ("processes plugin: This is only a zombie: pid = %li; "
1106                                 "name = %s;", pid, ps->name);
1107                 return (0);
1108         }
1109
1110         cpu_user_counter   = atoll (fields[11]);
1111         cpu_system_counter = atoll (fields[12]);
1112         vmem_size          = atoll (fields[20]);
1113         vmem_rss           = atoll (fields[21]);
1114         ps->vmem_minflt_counter = atol (fields[7]);
1115         ps->vmem_majflt_counter = atol (fields[9]);
1116
1117         {
1118                 unsigned long long stack_start = atoll (fields[25]);
1119                 unsigned long long stack_ptr   = atoll (fields[26]);
1120
1121                 stack_size = (stack_start > stack_ptr)
1122                         ? stack_start - stack_ptr
1123                         : stack_ptr - stack_start;
1124         }
1125
1126         /* Convert jiffies to useconds */
1127         cpu_user_counter   = cpu_user_counter   * 1000000 / CONFIG_HZ;
1128         cpu_system_counter = cpu_system_counter * 1000000 / CONFIG_HZ;
1129         vmem_rss = vmem_rss * pagesize_g;
1130
1131         ps->cpu_user_counter = cpu_user_counter;
1132         ps->cpu_system_counter = cpu_system_counter;
1133         ps->vmem_size = (unsigned long) vmem_size;
1134         ps->vmem_rss = (unsigned long) vmem_rss;
1135         ps->stack_size = (unsigned long) stack_size;
1136
1137         if ( (ps_read_io (pid, ps)) == NULL)
1138         {
1139                 /* no io data */
1140                 ps->io_rchar = -1;
1141                 ps->io_wchar = -1;
1142                 ps->io_syscr = -1;
1143                 ps->io_syscw = -1;
1144
1145                 DEBUG("ps_read_process: not get io data for pid %li", pid);
1146         }
1147
1148         if ( report_ctx_switch )
1149         {
1150                 if ( (ps_read_tasks_status(pid, ps)) == NULL)
1151                 {
1152                         ps->cswitch_vol = -1;
1153                         ps->cswitch_invol = -1;
1154
1155                         DEBUG("ps_read_tasks_status: not get context "
1156                                         "switch data for pid %li", pid);
1157                 }
1158         }
1159
1160         /* success */
1161         return (0);
1162 } /* int ps_read_process (...) */
1163
1164 static char *ps_get_cmdline (long pid, char *name, char *buf, size_t buf_len)
1165 {
1166         char  *buf_ptr;
1167         size_t len;
1168
1169         char file[PATH_MAX];
1170         int  fd;
1171
1172         size_t n;
1173
1174         if ((pid < 1) || (NULL == buf) || (buf_len < 2))
1175                 return NULL;
1176
1177         ssnprintf (file, sizeof (file), "/proc/%li/cmdline", pid);
1178
1179         errno = 0;
1180         fd = open (file, O_RDONLY);
1181         if (fd < 0) {
1182                 char errbuf[4096];
1183                 /* ENOENT means the process exited while we were handling it.
1184                  * Don't complain about this, it only fills the logs. */
1185                 if (errno != ENOENT)
1186                         WARNING ("processes plugin: Failed to open `%s': %s.", file,
1187                                         sstrerror (errno, errbuf, sizeof (errbuf)));
1188                 return NULL;
1189         }
1190
1191         buf_ptr = buf;
1192         len     = buf_len;
1193
1194         n = 0;
1195
1196         while (42) {
1197                 ssize_t status;
1198
1199                 status = read (fd, (void *)buf_ptr, len);
1200
1201                 if (status < 0) {
1202                         char errbuf[1024];
1203
1204                         if ((EAGAIN == errno) || (EINTR == errno))
1205                                 continue;
1206
1207                         WARNING ("processes plugin: Failed to read from `%s': %s.", file,
1208                                         sstrerror (errno, errbuf, sizeof (errbuf)));
1209                         close (fd);
1210                         return NULL;
1211                 }
1212
1213                 n += status;
1214
1215                 if (status == 0)
1216                         break;
1217
1218                 buf_ptr += status;
1219                 len     -= status;
1220
1221                 if (len <= 0)
1222                         break;
1223         }
1224
1225         close (fd);
1226
1227         if (0 == n) {
1228                 /* cmdline not available; e.g. kernel thread, zombie */
1229                 if (NULL == name)
1230                         return NULL;
1231
1232                 ssnprintf (buf, buf_len, "[%s]", name);
1233                 return buf;
1234         }
1235
1236         assert (n <= buf_len);
1237
1238         if (n == buf_len)
1239                 --n;
1240         buf[n] = '\0';
1241
1242         --n;
1243         /* remove trailing whitespace */
1244         while ((n > 0) && (isspace (buf[n]) || ('\0' == buf[n]))) {
1245                 buf[n] = '\0';
1246                 --n;
1247         }
1248
1249         /* arguments are separated by '\0' in /proc/<pid>/cmdline */
1250         while (n > 0) {
1251                 if ('\0' == buf[n])
1252                         buf[n] = ' ';
1253                 --n;
1254         }
1255         return buf;
1256 } /* char *ps_get_cmdline (...) */
1257
1258 static int read_fork_rate (void)
1259 {
1260         FILE *proc_stat;
1261         char buffer[1024];
1262         value_t value;
1263         _Bool value_valid = 0;
1264
1265         proc_stat = fopen ("/proc/stat", "r");
1266         if (proc_stat == NULL)
1267         {
1268                 char errbuf[1024];
1269                 ERROR ("processes plugin: fopen (/proc/stat) failed: %s",
1270                                 sstrerror (errno, errbuf, sizeof (errbuf)));
1271                 return (-1);
1272         }
1273
1274         while (fgets (buffer, sizeof (buffer), proc_stat) != NULL)
1275         {
1276                 int status;
1277                 char *fields[3];
1278                 int fields_num;
1279
1280                 fields_num = strsplit (buffer, fields,
1281                                 STATIC_ARRAY_SIZE (fields));
1282                 if (fields_num != 2)
1283                         continue;
1284
1285                 if (strcmp ("processes", fields[0]) != 0)
1286                         continue;
1287
1288                 status = parse_value (fields[1], &value, DS_TYPE_DERIVE);
1289                 if (status == 0)
1290                         value_valid = 1;
1291
1292                 break;
1293         }
1294         fclose(proc_stat);
1295
1296         if (!value_valid)
1297                 return (-1);
1298
1299         ps_submit_fork_rate (value.derive);
1300         return (0);
1301 }
1302 #endif /*KERNEL_LINUX */
1303
1304 #if KERNEL_SOLARIS
1305 static char *ps_get_cmdline (long pid, char *name __attribute__((unused)), /* {{{ */
1306     char *buffer, size_t buffer_size)
1307 {
1308         char path[PATH_MAX];
1309         psinfo_t info;
1310         ssize_t status;
1311
1312         snprintf(path, sizeof (path), "/proc/%li/psinfo", pid);
1313
1314         status = read_file_contents (path, (void *) &info, sizeof (info));
1315         if ((status < 0) || (((size_t) status) != sizeof (info)))
1316         {
1317                 ERROR ("processes plugin: Unexpected return value "
1318                                 "while reading \"%s\": "
1319                                 "Returned %zd but expected %zu.",
1320                                 path, status, buffer_size);
1321                 return (NULL);
1322         }
1323
1324         info.pr_psargs[sizeof (info.pr_psargs) - 1] = 0;
1325         sstrncpy (buffer, info.pr_psargs, buffer_size);
1326
1327         return (buffer);
1328 } /* }}} int ps_get_cmdline */
1329
1330 /*
1331  * Reads process information on the Solaris OS. The information comes mainly from
1332  * /proc/PID/status, /proc/PID/psinfo and /proc/PID/usage
1333  * The values for input and ouput chars are calculated "by hand"
1334  * Added a few "solaris" specific process states as well
1335  */
1336 static int ps_read_process(long pid, procstat_t *ps, char *state)
1337 {
1338         char filename[64];
1339         char f_psinfo[64], f_usage[64];
1340         char *buffer;
1341
1342         pstatus_t *myStatus;
1343         psinfo_t *myInfo;
1344         prusage_t *myUsage;
1345
1346         snprintf(filename, sizeof (filename), "/proc/%li/status", pid);
1347         snprintf(f_psinfo, sizeof (f_psinfo), "/proc/%li/psinfo", pid);
1348         snprintf(f_usage, sizeof (f_usage), "/proc/%li/usage", pid);
1349
1350
1351         buffer = malloc(sizeof (pstatus_t));
1352         memset(buffer, 0, sizeof (pstatus_t));
1353         read_file_contents(filename, buffer, sizeof (pstatus_t));
1354         myStatus = (pstatus_t *) buffer;
1355
1356         buffer = malloc(sizeof (psinfo_t));
1357         memset(buffer, 0, sizeof(psinfo_t));
1358         read_file_contents(f_psinfo, buffer, sizeof (psinfo_t));
1359         myInfo = (psinfo_t *) buffer;
1360
1361         buffer = malloc(sizeof (prusage_t));
1362         memset(buffer, 0, sizeof(prusage_t));
1363         read_file_contents(f_usage, buffer, sizeof (prusage_t));
1364         myUsage = (prusage_t *) buffer;
1365
1366         sstrncpy(ps->name, myInfo->pr_fname, sizeof (myInfo->pr_fname));
1367         ps->num_lwp = myStatus->pr_nlwp;
1368         if (myInfo->pr_wstat != 0) {
1369                 ps->num_proc = 0;
1370                 ps->num_lwp = 0;
1371                 *state = (char) 'Z';
1372                 return (0);
1373         } else {
1374                 ps->num_proc = 1;
1375                 ps->num_lwp = myInfo->pr_nlwp;
1376         }
1377
1378         /*
1379          * Convert system time and user time from nanoseconds to microseconds
1380          * for compatibility with the linux module
1381          */
1382         ps->cpu_system_counter = myStatus -> pr_stime.tv_nsec / 1000;
1383         ps->cpu_user_counter = myStatus -> pr_utime.tv_nsec / 1000;
1384
1385         /*
1386          * Convert rssize from KB to bytes to be consistent w/ the linux module
1387          */
1388         ps->vmem_rss = myInfo->pr_rssize * 1024;
1389         ps->vmem_size = myInfo->pr_size * 1024;
1390         ps->vmem_minflt_counter = myUsage->pr_minf;
1391         ps->vmem_majflt_counter = myUsage->pr_majf;
1392
1393         /*
1394          * TODO: Data and code segment calculations for Solaris
1395          */
1396
1397         ps->vmem_data = -1;
1398         ps->vmem_code = -1;
1399         ps->stack_size = myStatus->pr_stksize;
1400
1401         /*
1402          * Calculating input/ouput chars
1403          * Formula used is total chars / total blocks => chars/block
1404          * then convert input/output blocks to chars
1405          */
1406         ulong_t tot_chars = myUsage->pr_ioch;
1407         ulong_t tot_blocks = myUsage->pr_inblk + myUsage->pr_oublk;
1408         ulong_t chars_per_block = 1;
1409         if (tot_blocks != 0)
1410                 chars_per_block = tot_chars / tot_blocks;
1411         ps->io_rchar = myUsage->pr_inblk * chars_per_block;
1412         ps->io_wchar = myUsage->pr_oublk * chars_per_block;
1413         ps->io_syscr = myUsage->pr_sysc;
1414         ps->io_syscw = myUsage->pr_sysc;
1415
1416         /*
1417          * TODO: context switch counters for Solaris
1418    */
1419         ps->cswitch_vol   = -1;
1420         ps->cswitch_invol = -1;
1421
1422
1423         /*
1424          * TODO: Find way of setting BLOCKED and PAGING status
1425          */
1426
1427         *state = (char) 'R';
1428         if (myStatus->pr_flags & PR_ASLEEP)
1429                 *state = (char) 'S';
1430         else if (myStatus->pr_flags & PR_STOPPED)
1431                 *state = (char) 'T';
1432         else if (myStatus->pr_flags & PR_DETACH)
1433                 *state = (char) 'E';
1434         else if (myStatus->pr_flags & PR_DAEMON)
1435                 *state = (char) 'A';
1436         else if (myStatus->pr_flags & PR_ISSYS)
1437                 *state = (char) 'Y';
1438         else if (myStatus->pr_flags & PR_ORPHAN)
1439                 *state = (char) 'O';
1440
1441         sfree(myStatus);
1442         sfree(myInfo);
1443         sfree(myUsage);
1444
1445         return (0);
1446 }
1447
1448 /*
1449  * Reads the number of threads created since the last reboot. On Solaris these
1450  * are retrieved from kstat (module cpu, name sys, class misc, stat nthreads).
1451  * The result is the sum for all the threads created on each cpu
1452  */
1453 static int read_fork_rate (void)
1454 {
1455         extern kstat_ctl_t *kc;
1456         kstat_t *ksp_chain = NULL;
1457         derive_t result = 0;
1458
1459         if (kc == NULL)
1460                 return (-1);
1461
1462         for (ksp_chain = kc->kc_chain;
1463                         ksp_chain != NULL;
1464                         ksp_chain = ksp_chain->ks_next)
1465         {
1466                 if ((strcmp (ksp_chain->ks_module, "cpu") == 0)
1467                                 && (strcmp (ksp_chain->ks_name, "sys") == 0)
1468                                 && (strcmp (ksp_chain->ks_class, "misc") == 0))
1469                 {
1470                         long long tmp;
1471
1472                         kstat_read (kc, ksp_chain, NULL);
1473
1474                         tmp = get_kstat_value(ksp_chain, "nthreads");
1475                         if (tmp != -1LL)
1476                                 result += tmp;
1477                 }
1478         }
1479
1480         ps_submit_fork_rate (result);
1481         return (0);
1482 }
1483 #endif /* KERNEL_SOLARIS */
1484
1485 #if HAVE_THREAD_INFO
1486 static int mach_get_task_name (task_t t, int *pid, char *name, size_t name_max_len)
1487 {
1488         int mib[4];
1489
1490         struct kinfo_proc kp;
1491         size_t            kp_size;
1492
1493         mib[0] = CTL_KERN;
1494         mib[1] = KERN_PROC;
1495         mib[2] = KERN_PROC_PID;
1496
1497         if (pid_for_task (t, pid) != KERN_SUCCESS)
1498                 return (-1);
1499         mib[3] = *pid;
1500
1501         kp_size = sizeof (kp);
1502         if (sysctl (mib, 4, &kp, &kp_size, NULL, 0) != 0)
1503                 return (-1);
1504
1505         if (name_max_len > (MAXCOMLEN + 1))
1506                 name_max_len = MAXCOMLEN + 1;
1507
1508         strncpy (name, kp.kp_proc.p_comm, name_max_len - 1);
1509         name[name_max_len - 1] = '\0';
1510
1511         DEBUG ("pid = %i; name = %s;", *pid, name);
1512
1513         /* We don't do the special handling for `p_comm == "LaunchCFMApp"' as
1514          * `top' does it, because it is a lot of work and only used when
1515          * debugging. -octo */
1516
1517         return (0);
1518 }
1519 #endif /* HAVE_THREAD_INFO */
1520 /* ------- end of additional functions for KERNEL_LINUX/HAVE_THREAD_INFO ------- */
1521
1522 /* do actual readings from kernel */
1523 static int ps_read (void)
1524 {
1525 #if HAVE_THREAD_INFO
1526         kern_return_t            status;
1527
1528         int                      pset;
1529         processor_set_t          port_pset_priv;
1530
1531         int                      task;
1532         task_array_t             task_list;
1533         mach_msg_type_number_t   task_list_len;
1534
1535         int                      task_pid;
1536         char                     task_name[MAXCOMLEN + 1];
1537
1538         int                      thread;
1539         thread_act_array_t       thread_list;
1540         mach_msg_type_number_t   thread_list_len;
1541         thread_basic_info_data_t thread_data;
1542         mach_msg_type_number_t   thread_data_len;
1543
1544         int running  = 0;
1545         int sleeping = 0;
1546         int zombies  = 0;
1547         int stopped  = 0;
1548         int blocked  = 0;
1549
1550         procstat_t *ps;
1551         procstat_entry_t pse;
1552
1553         ps_list_reset ();
1554
1555         /*
1556          * The Mach-concept is a little different from the traditional UNIX
1557          * concept: All the work is done in threads. Threads are contained in
1558          * `tasks'. Therefore, `task status' doesn't make much sense, since
1559          * it's actually a `thread status'.
1560          * Tasks are assigned to sets of processors, so that's where you go to
1561          * get a list.
1562          */
1563         for (pset = 0; pset < pset_list_len; pset++)
1564         {
1565                 if ((status = host_processor_set_priv (port_host_self,
1566                                                 pset_list[pset],
1567                                                 &port_pset_priv)) != KERN_SUCCESS)
1568                 {
1569                         ERROR ("host_processor_set_priv failed: %s\n",
1570                                         mach_error_string (status));
1571                         continue;
1572                 }
1573
1574                 if ((status = processor_set_tasks (port_pset_priv,
1575                                                 &task_list,
1576                                                 &task_list_len)) != KERN_SUCCESS)
1577                 {
1578                         ERROR ("processor_set_tasks failed: %s\n",
1579                                         mach_error_string (status));
1580                         mach_port_deallocate (port_task_self, port_pset_priv);
1581                         continue;
1582                 }
1583
1584                 for (task = 0; task < task_list_len; task++)
1585                 {
1586                         ps = NULL;
1587                         if (mach_get_task_name (task_list[task],
1588                                                 &task_pid,
1589                                                 task_name, PROCSTAT_NAME_LEN) == 0)
1590                         {
1591                                 /* search for at least one match */
1592                                 for (ps = list_head_g; ps != NULL; ps = ps->next)
1593                                         /* FIXME: cmdline should be here instead of NULL */
1594                                         if (ps_list_match (task_name, NULL, ps) == 1)
1595                                                 break;
1596                         }
1597
1598                         /* Collect more detailed statistics for this process */
1599                         if (ps != NULL)
1600                         {
1601                                 task_basic_info_data_t        task_basic_info;
1602                                 mach_msg_type_number_t        task_basic_info_len;
1603                                 task_events_info_data_t       task_events_info;
1604                                 mach_msg_type_number_t        task_events_info_len;
1605                                 task_absolutetime_info_data_t task_absolutetime_info;
1606                                 mach_msg_type_number_t        task_absolutetime_info_len;
1607
1608                                 memset (&pse, '\0', sizeof (pse));
1609                                 pse.id = task_pid;
1610
1611                                 task_basic_info_len = TASK_BASIC_INFO_COUNT;
1612                                 status = task_info (task_list[task],
1613                                                 TASK_BASIC_INFO,
1614                                                 (task_info_t) &task_basic_info,
1615                                                 &task_basic_info_len);
1616                                 if (status != KERN_SUCCESS)
1617                                 {
1618                                         ERROR ("task_info failed: %s",
1619                                                         mach_error_string (status));
1620                                         continue; /* with next thread_list */
1621                                 }
1622
1623                                 task_events_info_len = TASK_EVENTS_INFO_COUNT;
1624                                 status = task_info (task_list[task],
1625                                                 TASK_EVENTS_INFO,
1626                                                 (task_info_t) &task_events_info,
1627                                                 &task_events_info_len);
1628                                 if (status != KERN_SUCCESS)
1629                                 {
1630                                         ERROR ("task_info failed: %s",
1631                                                         mach_error_string (status));
1632                                         continue; /* with next thread_list */
1633                                 }
1634
1635                                 task_absolutetime_info_len = TASK_ABSOLUTETIME_INFO_COUNT;
1636                                 status = task_info (task_list[task],
1637                                                 TASK_ABSOLUTETIME_INFO,
1638                                                 (task_info_t) &task_absolutetime_info,
1639                                                 &task_absolutetime_info_len);
1640                                 if (status != KERN_SUCCESS)
1641                                 {
1642                                         ERROR ("task_info failed: %s",
1643                                                         mach_error_string (status));
1644                                         continue; /* with next thread_list */
1645                                 }
1646
1647                                 pse.num_proc++;
1648                                 pse.vmem_size = task_basic_info.virtual_size;
1649                                 pse.vmem_rss = task_basic_info.resident_size;
1650                                 /* Does not seem to be easily exposed */
1651                                 pse.vmem_data = 0;
1652                                 pse.vmem_code = 0;
1653
1654                                 pse.vmem_minflt_counter = task_events_info.cow_faults;
1655                                 pse.vmem_majflt_counter = task_events_info.faults;
1656
1657                                 pse.cpu_user_counter = task_absolutetime_info.total_user;
1658                                 pse.cpu_system_counter = task_absolutetime_info.total_system;
1659
1660                                 /* context switch counters not implemented */
1661                                 pse.cswitch_vol   = -1;
1662                                 pse.cswitch_invol = -1;
1663                         }
1664
1665                         status = task_threads (task_list[task], &thread_list,
1666                                         &thread_list_len);
1667                         if (status != KERN_SUCCESS)
1668                         {
1669                                 /* Apple's `top' treats this case a zombie. It
1670                                  * makes sense to some extend: A `zombie'
1671                                  * thread is nonsense, since the task/process
1672                                  * is dead. */
1673                                 zombies++;
1674                                 DEBUG ("task_threads failed: %s",
1675                                                 mach_error_string (status));
1676                                 if (task_list[task] != port_task_self)
1677                                         mach_port_deallocate (port_task_self,
1678                                                         task_list[task]);
1679                                 continue; /* with next task_list */
1680                         }
1681
1682                         for (thread = 0; thread < thread_list_len; thread++)
1683                         {
1684                                 thread_data_len = THREAD_BASIC_INFO_COUNT;
1685                                 status = thread_info (thread_list[thread],
1686                                                 THREAD_BASIC_INFO,
1687                                                 (thread_info_t) &thread_data,
1688                                                 &thread_data_len);
1689                                 if (status != KERN_SUCCESS)
1690                                 {
1691                                         ERROR ("thread_info failed: %s",
1692                                                         mach_error_string (status));
1693                                         if (task_list[task] != port_task_self)
1694                                                 mach_port_deallocate (port_task_self,
1695                                                                 thread_list[thread]);
1696                                         continue; /* with next thread_list */
1697                                 }
1698
1699                                 if (ps != NULL)
1700                                         pse.num_lwp++;
1701
1702                                 switch (thread_data.run_state)
1703                                 {
1704                                         case TH_STATE_RUNNING:
1705                                                 running++;
1706                                                 break;
1707                                         case TH_STATE_STOPPED:
1708                                         /* What exactly is `halted'? */
1709                                         case TH_STATE_HALTED:
1710                                                 stopped++;
1711                                                 break;
1712                                         case TH_STATE_WAITING:
1713                                                 sleeping++;
1714                                                 break;
1715                                         case TH_STATE_UNINTERRUPTIBLE:
1716                                                 blocked++;
1717                                                 break;
1718                                         /* There is no `zombie' case here,
1719                                          * since there are no zombie-threads.
1720                                          * There's only zombie tasks, which are
1721                                          * handled above. */
1722                                         default:
1723                                                 WARNING ("Unknown thread status: %i",
1724                                                                 thread_data.run_state);
1725                                                 break;
1726                                 } /* switch (thread_data.run_state) */
1727
1728                                 if (task_list[task] != port_task_self)
1729                                 {
1730                                         status = mach_port_deallocate (port_task_self,
1731                                                         thread_list[thread]);
1732                                         if (status != KERN_SUCCESS)
1733                                                 ERROR ("mach_port_deallocate failed: %s",
1734                                                                 mach_error_string (status));
1735                                 }
1736                         } /* for (thread_list) */
1737
1738                         if ((status = vm_deallocate (port_task_self,
1739                                                         (vm_address_t) thread_list,
1740                                                         thread_list_len * sizeof (thread_act_t)))
1741                                         != KERN_SUCCESS)
1742                         {
1743                                 ERROR ("vm_deallocate failed: %s",
1744                                                 mach_error_string (status));
1745                         }
1746                         thread_list = NULL;
1747                         thread_list_len = 0;
1748
1749                         /* Only deallocate the task port, if it isn't our own.
1750                          * Don't know what would happen in that case, but this
1751                          * is what Apple's top does.. ;) */
1752                         if (task_list[task] != port_task_self)
1753                         {
1754                                 status = mach_port_deallocate (port_task_self,
1755                                                 task_list[task]);
1756                                 if (status != KERN_SUCCESS)
1757                                         ERROR ("mach_port_deallocate failed: %s",
1758                                                         mach_error_string (status));
1759                         }
1760
1761                         if (ps != NULL)
1762                                 /* FIXME: cmdline should be here instead of NULL */
1763                                 ps_list_add (task_name, NULL, &pse);
1764                 } /* for (task_list) */
1765
1766                 if ((status = vm_deallocate (port_task_self,
1767                                 (vm_address_t) task_list,
1768                                 task_list_len * sizeof (task_t))) != KERN_SUCCESS)
1769                 {
1770                         ERROR ("vm_deallocate failed: %s",
1771                                         mach_error_string (status));
1772                 }
1773                 task_list = NULL;
1774                 task_list_len = 0;
1775
1776                 if ((status = mach_port_deallocate (port_task_self, port_pset_priv))
1777                                 != KERN_SUCCESS)
1778                 {
1779                         ERROR ("mach_port_deallocate failed: %s",
1780                                         mach_error_string (status));
1781                 }
1782         } /* for (pset_list) */
1783
1784         ps_submit_state ("running", running);
1785         ps_submit_state ("sleeping", sleeping);
1786         ps_submit_state ("zombies", zombies);
1787         ps_submit_state ("stopped", stopped);
1788         ps_submit_state ("blocked", blocked);
1789
1790         for (ps = list_head_g; ps != NULL; ps = ps->next)
1791                 ps_submit_proc_list (ps);
1792 /* #endif HAVE_THREAD_INFO */
1793
1794 #elif KERNEL_LINUX
1795         int running  = 0;
1796         int sleeping = 0;
1797         int zombies  = 0;
1798         int stopped  = 0;
1799         int paging   = 0;
1800         int blocked  = 0;
1801
1802         struct dirent *ent;
1803         DIR           *proc;
1804         long           pid;
1805
1806         char cmdline[CMDLINE_BUFFER_SIZE];
1807
1808         int        status;
1809         procstat_t ps;
1810         procstat_entry_t pse;
1811         char       state;
1812
1813         procstat_t *ps_ptr;
1814
1815         running = sleeping = zombies = stopped = paging = blocked = 0;
1816         ps_list_reset ();
1817
1818         if ((proc = opendir ("/proc")) == NULL)
1819         {
1820                 char errbuf[1024];
1821                 ERROR ("Cannot open `/proc': %s",
1822                                 sstrerror (errno, errbuf, sizeof (errbuf)));
1823                 return (-1);
1824         }
1825
1826         while ((ent = readdir (proc)) != NULL)
1827         {
1828                 if (!isdigit (ent->d_name[0]))
1829                         continue;
1830
1831                 if ((pid = atol (ent->d_name)) < 1)
1832                         continue;
1833
1834                 status = ps_read_process (pid, &ps, &state);
1835                 if (status != 0)
1836                 {
1837                         DEBUG ("ps_read_process failed: %i", status);
1838                         continue;
1839                 }
1840
1841                 memset (&pse, 0, sizeof (pse));
1842                 pse.id       = pid;
1843                 pse.age      = 0;
1844
1845                 pse.num_proc   = ps.num_proc;
1846                 pse.num_lwp    = ps.num_lwp;
1847                 pse.vmem_size  = ps.vmem_size;
1848                 pse.vmem_rss   = ps.vmem_rss;
1849                 pse.vmem_data  = ps.vmem_data;
1850                 pse.vmem_code  = ps.vmem_code;
1851                 pse.stack_size = ps.stack_size;
1852
1853                 pse.vmem_minflt = 0;
1854                 pse.vmem_minflt_counter = ps.vmem_minflt_counter;
1855                 pse.vmem_majflt = 0;
1856                 pse.vmem_majflt_counter = ps.vmem_majflt_counter;
1857
1858                 pse.cpu_user = 0;
1859                 pse.cpu_user_counter = ps.cpu_user_counter;
1860                 pse.cpu_system = 0;
1861                 pse.cpu_system_counter = ps.cpu_system_counter;
1862
1863                 pse.io_rchar = ps.io_rchar;
1864                 pse.io_wchar = ps.io_wchar;
1865                 pse.io_syscr = ps.io_syscr;
1866                 pse.io_syscw = ps.io_syscw;
1867
1868                 pse.cswitch_vol = ps.cswitch_vol;
1869                 pse.cswitch_invol = ps.cswitch_invol;
1870
1871                 switch (state)
1872                 {
1873                         case 'R': running++;  break;
1874                         case 'S': sleeping++; break;
1875                         case 'D': blocked++;  break;
1876                         case 'Z': zombies++;  break;
1877                         case 'T': stopped++;  break;
1878                         case 'W': paging++;   break;
1879                 }
1880
1881                 ps_list_add (ps.name,
1882                                 ps_get_cmdline (pid, ps.name, cmdline, sizeof (cmdline)),
1883                                 &pse);
1884         }
1885
1886         closedir (proc);
1887
1888         ps_submit_state ("running",  running);
1889         ps_submit_state ("sleeping", sleeping);
1890         ps_submit_state ("zombies",  zombies);
1891         ps_submit_state ("stopped",  stopped);
1892         ps_submit_state ("paging",   paging);
1893         ps_submit_state ("blocked",  blocked);
1894
1895         for (ps_ptr = list_head_g; ps_ptr != NULL; ps_ptr = ps_ptr->next)
1896                 ps_submit_proc_list (ps_ptr);
1897
1898         read_fork_rate();
1899 /* #endif KERNEL_LINUX */
1900
1901 #elif HAVE_LIBKVM_GETPROCS && HAVE_STRUCT_KINFO_PROC_FREEBSD
1902         int running  = 0;
1903         int sleeping = 0;
1904         int zombies  = 0;
1905         int stopped  = 0;
1906         int blocked  = 0;
1907         int idle     = 0;
1908         int wait     = 0;
1909
1910         kvm_t *kd;
1911         char errbuf[_POSIX2_LINE_MAX];
1912         struct kinfo_proc *procs;          /* array of processes */
1913         struct kinfo_proc *proc_ptr = NULL;
1914         int count;                         /* returns number of processes */
1915         int i;
1916
1917         procstat_t *ps_ptr;
1918         procstat_entry_t pse;
1919
1920         ps_list_reset ();
1921
1922         /* Open the kvm interface, get a descriptor */
1923         kd = kvm_openfiles (NULL, "/dev/null", NULL, 0, errbuf);
1924         if (kd == NULL)
1925         {
1926                 ERROR ("processes plugin: Cannot open kvm interface: %s",
1927                                 errbuf);
1928                 return (0);
1929         }
1930
1931         /* Get the list of processes. */
1932         procs = kvm_getprocs(kd, KERN_PROC_ALL, 0, &count);
1933         if (procs == NULL)
1934         {
1935                 ERROR ("processes plugin: Cannot get kvm processes list: %s",
1936                                 kvm_geterr(kd));
1937                 kvm_close (kd);
1938                 return (0);
1939         }
1940
1941         /* Iterate through the processes in kinfo_proc */
1942         for (i = 0; i < count; i++)
1943         {
1944                 /* Create only one process list entry per _process_, i.e.
1945                  * filter out threads (duplicate PID entries). */
1946                 if ((proc_ptr == NULL) || (proc_ptr->ki_pid != procs[i].ki_pid))
1947                 {
1948                         char cmdline[CMDLINE_BUFFER_SIZE] = "";
1949                         _Bool have_cmdline = 0;
1950
1951                         proc_ptr = &(procs[i]);
1952                         /* Don't probe system processes and processes without arguments */
1953                         if (((procs[i].ki_flag & P_SYSTEM) == 0)
1954                                         && (procs[i].ki_args != NULL))
1955                         {
1956                                 char **argv;
1957                                 int argc;
1958                                 int status;
1959
1960                                 /* retrieve the arguments */
1961                                 argv = kvm_getargv (kd, proc_ptr, /* nchr = */ 0);
1962                                 argc = 0;
1963                                 if ((argv != NULL) && (argv[0] != NULL))
1964                                 {
1965                                         while (argv[argc] != NULL)
1966                                                 argc++;
1967
1968                                         status = strjoin (cmdline, sizeof (cmdline), argv, argc, " ");
1969                                         if (status < 0)
1970                                                 WARNING ("processes plugin: Command line did not fit into buffer.");
1971                                         else
1972                                                 have_cmdline = 1;
1973                                 }
1974                         } /* if (process has argument list) */
1975
1976                         pse.id       = procs[i].ki_pid;
1977                         pse.age      = 0;
1978
1979                         pse.num_proc = 1;
1980                         pse.num_lwp  = procs[i].ki_numthreads;
1981
1982                         pse.vmem_size = procs[i].ki_size;
1983                         pse.vmem_rss = procs[i].ki_rssize * pagesize;
1984                         pse.vmem_data = procs[i].ki_dsize * pagesize;
1985                         pse.vmem_code = procs[i].ki_tsize * pagesize;
1986                         pse.stack_size = procs[i].ki_ssize * pagesize;
1987                         pse.vmem_minflt = 0;
1988                         pse.vmem_minflt_counter = procs[i].ki_rusage.ru_minflt;
1989                         pse.vmem_majflt = 0;
1990                         pse.vmem_majflt_counter = procs[i].ki_rusage.ru_majflt;
1991
1992                         pse.cpu_user = 0;
1993                         pse.cpu_system = 0;
1994                         pse.cpu_user_counter = 0;
1995                         pse.cpu_system_counter = 0;
1996                         /*
1997                          * The u-area might be swapped out, and we can't get
1998                          * at it because we have a crashdump and no swap.
1999                          * If it's here fill in these fields, otherwise, just
2000                          * leave them 0.
2001                          */
2002                         if (procs[i].ki_flag & P_INMEM)
2003                         {
2004                                 pse.cpu_user_counter = procs[i].ki_rusage.ru_utime.tv_usec
2005                                         + (1000000lu * procs[i].ki_rusage.ru_utime.tv_sec);
2006                                 pse.cpu_system_counter = procs[i].ki_rusage.ru_stime.tv_usec
2007                                         + (1000000lu * procs[i].ki_rusage.ru_stime.tv_sec);
2008                         }
2009
2010                         /* no I/O data */
2011                         pse.io_rchar = -1;
2012                         pse.io_wchar = -1;
2013                         pse.io_syscr = -1;
2014                         pse.io_syscw = -1;
2015
2016                         /* context switch counters not implemented */
2017                         pse.cswitch_vol   = -1;
2018                         pse.cswitch_invol = -1;
2019
2020                         ps_list_add (procs[i].ki_comm, have_cmdline ? cmdline : NULL, &pse);
2021
2022                         switch (procs[i].ki_stat)
2023                         {
2024                                 case SSTOP:     stopped++;      break;
2025                                 case SSLEEP:    sleeping++;     break;
2026                                 case SRUN:      running++;      break;
2027                                 case SIDL:      idle++;         break;
2028                                 case SWAIT:     wait++;         break;
2029                                 case SLOCK:     blocked++;      break;
2030                                 case SZOMB:     zombies++;      break;
2031                         }
2032                 } /* if ((proc_ptr == NULL) || (proc_ptr->ki_pid != procs[i].ki_pid)) */
2033         }
2034
2035         kvm_close(kd);
2036
2037         ps_submit_state ("running",  running);
2038         ps_submit_state ("sleeping", sleeping);
2039         ps_submit_state ("zombies",  zombies);
2040         ps_submit_state ("stopped",  stopped);
2041         ps_submit_state ("blocked",  blocked);
2042         ps_submit_state ("idle",     idle);
2043         ps_submit_state ("wait",     wait);
2044
2045         for (ps_ptr = list_head_g; ps_ptr != NULL; ps_ptr = ps_ptr->next)
2046                 ps_submit_proc_list (ps_ptr);
2047 /* #endif HAVE_LIBKVM_GETPROCS && HAVE_STRUCT_KINFO_PROC_FREEBSD */
2048
2049 #elif HAVE_LIBKVM_GETPROCS && HAVE_STRUCT_KINFO_PROC_OPENBSD
2050         int running  = 0;
2051         int sleeping = 0;
2052         int zombies  = 0;
2053         int stopped  = 0;
2054         int onproc   = 0;
2055         int idle     = 0;
2056         int dead     = 0;
2057
2058         kvm_t *kd;
2059         char errbuf[1024];
2060         struct kinfo_proc *procs;          /* array of processes */
2061         struct kinfo_proc *proc_ptr = NULL;
2062         int count;                         /* returns number of processes */
2063         int i;
2064
2065         procstat_t *ps_ptr;
2066         procstat_entry_t pse;
2067
2068         ps_list_reset ();
2069
2070         /* Open the kvm interface, get a descriptor */
2071         kd = kvm_open (NULL, NULL, NULL, 0, errbuf);
2072         if (kd == NULL)
2073         {
2074                 ERROR ("processes plugin: Cannot open kvm interface: %s",
2075                                 errbuf);
2076                 return (0);
2077         }
2078
2079         /* Get the list of processes. */
2080         procs = kvm_getprocs(kd, KERN_PROC_ALL, 0, sizeof(struct kinfo_proc), &count);
2081         if (procs == NULL)
2082         {
2083                 ERROR ("processes plugin: Cannot get kvm processes list: %s",
2084                                 kvm_geterr(kd));
2085                 kvm_close (kd);
2086                 return (0);
2087         }
2088
2089         /* Iterate through the processes in kinfo_proc */
2090         for (i = 0; i < count; i++)
2091         {
2092                 /* Create only one process list entry per _process_, i.e.
2093                  * filter out threads (duplicate PID entries). */
2094                 if ((proc_ptr == NULL) || (proc_ptr->p_pid != procs[i].p_pid))
2095                 {
2096                         char cmdline[CMDLINE_BUFFER_SIZE] = "";
2097                         _Bool have_cmdline = 0;
2098
2099                         proc_ptr = &(procs[i]);
2100                         /* Don't probe zombie processes  */
2101                         if (!P_ZOMBIE(proc_ptr))
2102                         {
2103                                 char **argv;
2104                                 int argc;
2105                                 int status;
2106
2107                                 /* retrieve the arguments */
2108                                 argv = kvm_getargv (kd, proc_ptr, /* nchr = */ 0);
2109                                 argc = 0;
2110                                 if ((argv != NULL) && (argv[0] != NULL))
2111                                 {
2112                                         while (argv[argc] != NULL)
2113                                                 argc++;
2114
2115                                         status = strjoin (cmdline, sizeof (cmdline), argv, argc, " ");
2116                                         if (status < 0)
2117                                                 WARNING ("processes plugin: Command line did not fit into buffer.");
2118                                         else
2119                                                 have_cmdline = 1;
2120                                 }
2121                         } /* if (process has argument list) */
2122
2123                         memset (&pse, 0, sizeof (pse));
2124                         pse.id       = procs[i].p_pid;
2125                         pse.age      = 0;
2126
2127                         pse.num_proc = 1;
2128                         pse.num_lwp  = 1; /* XXX: accumulate p_tid values for a single p_pid ? */
2129
2130                         pse.vmem_rss = procs[i].p_vm_rssize * pagesize;
2131                         pse.vmem_data = procs[i].p_vm_dsize * pagesize;
2132                         pse.vmem_code = procs[i].p_vm_tsize * pagesize;
2133                         pse.stack_size = procs[i].p_vm_ssize * pagesize;
2134                         pse.vmem_size = pse.stack_size + pse.vmem_code + pse.vmem_data;
2135                         pse.vmem_minflt = 0;
2136                         pse.vmem_minflt_counter = procs[i].p_uru_minflt;
2137                         pse.vmem_majflt = 0;
2138                         pse.vmem_majflt_counter = procs[i].p_uru_majflt;
2139
2140                         pse.cpu_user = 0;
2141                         pse.cpu_system = 0;
2142                         pse.cpu_user_counter = procs[i].p_uutime_usec +
2143                                                 (1000000lu * procs[i].p_uutime_sec);
2144                         pse.cpu_system_counter = procs[i].p_ustime_usec +
2145                                                 (1000000lu * procs[i].p_ustime_sec);
2146
2147                         /* no I/O data */
2148                         pse.io_rchar = -1;
2149                         pse.io_wchar = -1;
2150                         pse.io_syscr = -1;
2151                         pse.io_syscw = -1;
2152
2153                         /* context switch counters not implemented */
2154                         pse.cswitch_vol   = -1;
2155                         pse.cswitch_invol = -1;
2156
2157                         ps_list_add (procs[i].p_comm, have_cmdline ? cmdline : NULL, &pse);
2158
2159                         switch (procs[i].p_stat)
2160                         {
2161                                 case SSTOP:     stopped++;      break;
2162                                 case SSLEEP:    sleeping++;     break;
2163                                 case SRUN:      running++;      break;
2164                                 case SIDL:      idle++;         break;
2165                                 case SONPROC:   onproc++;       break;
2166                                 case SDEAD:     dead++;         break;
2167                                 case SZOMB:     zombies++;      break;
2168                         }
2169                 } /* if ((proc_ptr == NULL) || (proc_ptr->p_pid != procs[i].p_pid)) */
2170         }
2171
2172         kvm_close(kd);
2173
2174         ps_submit_state ("running",  running);
2175         ps_submit_state ("sleeping", sleeping);
2176         ps_submit_state ("zombies",  zombies);
2177         ps_submit_state ("stopped",  stopped);
2178         ps_submit_state ("onproc",   onproc);
2179         ps_submit_state ("idle",     idle);
2180         ps_submit_state ("dead",     dead);
2181
2182         for (ps_ptr = list_head_g; ps_ptr != NULL; ps_ptr = ps_ptr->next)
2183                 ps_submit_proc_list (ps_ptr);
2184 /* #endif HAVE_LIBKVM_GETPROCS && HAVE_STRUCT_KINFO_PROC_OPENBSD */
2185
2186 #elif HAVE_PROCINFO_H
2187         /* AIX */
2188         int running  = 0;
2189         int sleeping = 0;
2190         int zombies  = 0;
2191         int stopped  = 0;
2192         int paging   = 0;
2193         int blocked  = 0;
2194
2195         pid_t pindex = 0;
2196         int nprocs;
2197
2198         procstat_t *ps;
2199         procstat_entry_t pse;
2200
2201         ps_list_reset ();
2202         while ((nprocs = getprocs64 (procentry, sizeof(struct procentry64),
2203                                         /* fdsinfo = */ NULL, sizeof(struct fdsinfo64),
2204                                         &pindex, MAXPROCENTRY)) > 0)
2205         {
2206                 int i;
2207
2208                 for (i = 0; i < nprocs; i++)
2209                 {
2210                         tid64_t thindex;
2211                         int nthreads;
2212                         char arglist[MAXARGLN+1];
2213                         char *cargs;
2214                         char *cmdline;
2215
2216                         if (procentry[i].pi_state == SNONE) continue;
2217                         /* if (procentry[i].pi_state == SZOMB)  FIXME */
2218
2219                         cmdline = procentry[i].pi_comm;
2220                         cargs = procentry[i].pi_comm;
2221                         if ( procentry[i].pi_flags & SKPROC )
2222                         {
2223                                 if (procentry[i].pi_pid == 0)
2224                                         cmdline = "swapper";
2225                                 cargs = cmdline;
2226                         }
2227                         else
2228                         {
2229                                 if (getargs(&procentry[i], sizeof(struct procentry64), arglist, MAXARGLN) >= 0)
2230                                 {
2231                                         int n;
2232
2233                                         n = -1;
2234                                         while (++n < MAXARGLN)
2235                                         {
2236                                                 if (arglist[n] == '\0')
2237                                                 {
2238                                                         if (arglist[n+1] == '\0')
2239                                                                 break;
2240                                                         arglist[n] = ' ';
2241                                                 }
2242                                         }
2243                                         cargs = arglist;
2244                                 }
2245                         }
2246
2247                         pse.id       = procentry[i].pi_pid;
2248                         pse.age      = 0;
2249                         pse.num_lwp  = procentry[i].pi_thcount;
2250                         pse.num_proc = 1;
2251
2252                         thindex=0;
2253                         while ((nthreads = getthrds64(procentry[i].pi_pid,
2254                                                         thrdentry, sizeof(struct thrdentry64),
2255                                                         &thindex, MAXTHRDENTRY)) > 0)
2256                         {
2257                                 int j;
2258
2259                                 for (j=0; j< nthreads; j++)
2260                                 {
2261                                         switch (thrdentry[j].ti_state)
2262                                         {
2263                                                 /* case TSNONE: break; */
2264                                                 case TSIDL:     blocked++;      break; /* FIXME is really blocked */
2265                                                 case TSRUN:     running++;      break;
2266                                                 case TSSLEEP:   sleeping++;     break;
2267                                                 case TSSWAP:    paging++;       break;
2268                                                 case TSSTOP:    stopped++;      break;
2269                                                 case TSZOMB:    zombies++;      break;
2270                                         }
2271                                 }
2272                                 if (nthreads < MAXTHRDENTRY)
2273                                         break;
2274                         }
2275
2276                         pse.cpu_user = 0;
2277                         /* tv_usec is nanosec ??? */
2278                         pse.cpu_user_counter = procentry[i].pi_ru.ru_utime.tv_sec * 1000000 +
2279                                 procentry[i].pi_ru.ru_utime.tv_usec / 1000;
2280
2281                         pse.cpu_system = 0;
2282                         /* tv_usec is nanosec ??? */
2283                         pse.cpu_system_counter = procentry[i].pi_ru.ru_stime.tv_sec * 1000000 +
2284                                 procentry[i].pi_ru.ru_stime.tv_usec / 1000;
2285
2286                         pse.vmem_minflt = 0;
2287                         pse.vmem_minflt_counter = procentry[i].pi_minflt;
2288                         pse.vmem_majflt = 0;
2289                         pse.vmem_majflt_counter = procentry[i].pi_majflt;
2290
2291                         pse.vmem_size = procentry[i].pi_tsize + procentry[i].pi_dvm * pagesize;
2292                         pse.vmem_rss = (procentry[i].pi_drss + procentry[i].pi_trss) * pagesize;
2293                         /* Not supported */
2294                         pse.vmem_data = 0;
2295                         pse.vmem_code = 0;
2296                         pse.stack_size =  0;
2297
2298                         pse.io_rchar = -1;
2299                         pse.io_wchar = -1;
2300                         pse.io_syscr = -1;
2301                         pse.io_syscw = -1;
2302
2303                         pse.cswitch_vol   = -1;
2304                         pse.cswitch_invol = -1;
2305
2306                         ps_list_add (cmdline, cargs, &pse);
2307                 } /* for (i = 0 .. nprocs) */
2308
2309                 if (nprocs < MAXPROCENTRY)
2310                         break;
2311         } /* while (getprocs64() > 0) */
2312         ps_submit_state ("running",  running);
2313         ps_submit_state ("sleeping", sleeping);
2314         ps_submit_state ("zombies",  zombies);
2315         ps_submit_state ("stopped",  stopped);
2316         ps_submit_state ("paging",   paging);
2317         ps_submit_state ("blocked",  blocked);
2318
2319         for (ps = list_head_g; ps != NULL; ps = ps->next)
2320                 ps_submit_proc_list (ps);
2321 /* #endif HAVE_PROCINFO_H */
2322
2323 #elif KERNEL_SOLARIS
2324         /*
2325          * The Solaris section adds a few more process states and removes some
2326          * process states compared to linux. Most notably there is no "PAGING"
2327          * and "BLOCKED" state for a process.  The rest is similar to the linux
2328          * code.
2329          */
2330         int running = 0;
2331         int sleeping = 0;
2332         int zombies = 0;
2333         int stopped = 0;
2334         int detached = 0;
2335         int daemon = 0;
2336         int system = 0;
2337         int orphan = 0;
2338
2339         struct dirent *ent;
2340         DIR *proc;
2341
2342         int status;
2343         procstat_t *ps_ptr;
2344         char state;
2345
2346         char cmdline[PRARGSZ];
2347
2348         ps_list_reset ();
2349
2350         proc = opendir ("/proc");
2351         if (proc == NULL)
2352                 return (-1);
2353
2354         while ((ent = readdir(proc)) != NULL)
2355         {
2356                 long pid;
2357                 struct procstat ps;
2358                 procstat_entry_t pse;
2359                 char *endptr;
2360
2361                 if (!isdigit ((int) ent->d_name[0]))
2362                         continue;
2363
2364                 pid = strtol (ent->d_name, &endptr, 10);
2365                 if (*endptr != 0) /* value didn't completely parse as a number */
2366                         continue;
2367
2368                 status = ps_read_process (pid, &ps, &state);
2369                 if (status != 0)
2370                 {
2371                         DEBUG("ps_read_process failed: %i", status);
2372                         continue;
2373                 }
2374
2375                 memset (&pse, 0, sizeof (pse));
2376                 pse.id = pid;
2377                 pse.age = 0;
2378
2379                 pse.num_proc   = ps.num_proc;
2380                 pse.num_lwp    = ps.num_lwp;
2381                 pse.vmem_size  = ps.vmem_size;
2382                 pse.vmem_rss   = ps.vmem_rss;
2383                 pse.vmem_data  = ps.vmem_data;
2384                 pse.vmem_code  = ps.vmem_code;
2385                 pse.stack_size = ps.stack_size;
2386
2387                 pse.vmem_minflt = 0;
2388                 pse.vmem_minflt_counter = ps.vmem_minflt_counter;
2389                 pse.vmem_majflt = 0;
2390                 pse.vmem_majflt_counter = ps.vmem_majflt_counter;
2391
2392                 pse.cpu_user = 0;
2393                 pse.cpu_user_counter = ps.cpu_user_counter;
2394                 pse.cpu_system = 0;
2395                 pse.cpu_system_counter = ps.cpu_system_counter;
2396
2397                 pse.io_rchar = ps.io_rchar;
2398                 pse.io_wchar = ps.io_wchar;
2399                 pse.io_syscr = ps.io_syscr;
2400                 pse.io_syscw = ps.io_syscw;
2401
2402                 pse.cswitch_vol = -1;
2403                 pse.cswitch_invol = -1;
2404
2405                 switch (state)
2406                 {
2407                         case 'R': running++;  break;
2408                         case 'S': sleeping++; break;
2409                         case 'E': detached++; break;
2410                         case 'Z': zombies++;  break;
2411                         case 'T': stopped++;  break;
2412                         case 'A': daemon++;   break;
2413                         case 'Y': system++;   break;
2414                         case 'O': orphan++;   break;
2415                 }
2416
2417
2418                 ps_list_add (ps.name,
2419                                 ps_get_cmdline (pid, ps.name, cmdline, sizeof (cmdline)),
2420                                 &pse);
2421         } /* while(readdir) */
2422         closedir (proc);
2423
2424         ps_submit_state ("running",  running);
2425         ps_submit_state ("sleeping", sleeping);
2426         ps_submit_state ("zombies",  zombies);
2427         ps_submit_state ("stopped",  stopped);
2428         ps_submit_state ("detached", detached);
2429         ps_submit_state ("daemon",   daemon);
2430         ps_submit_state ("system",   system);
2431         ps_submit_state ("orphan",   orphan);
2432
2433         for (ps_ptr = list_head_g; ps_ptr != NULL; ps_ptr = ps_ptr->next)
2434                 ps_submit_proc_list (ps_ptr);
2435
2436         read_fork_rate();
2437 #endif /* KERNEL_SOLARIS */
2438
2439         return (0);
2440 } /* int ps_read */
2441
2442 void module_register (void)
2443 {
2444         plugin_register_complex_config ("processes", ps_config);
2445         plugin_register_init ("processes", ps_init);
2446         plugin_register_read ("processes", ps_read);
2447 } /* void module_register */