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