1 /**
2 * collectd - src/processes.c
3 * Copyright (C) 2005 Lyonel Vincent
4 * Copyright (C) 2006 Florian Forster (Mach code)
5 *
6 * This program is free software; you can redistribute it and/or modify it
7 * under the terms of the GNU General Public License as published by the
8 * Free Software Foundation; either version 2 of the License, or (at your
9 * option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful, but
12 * WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License along
17 * with this program; if not, write to the Free Software Foundation, Inc.,
18 * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
19 *
20 * Authors:
21 * Lyonel Vincent <lyonel at ezix.org>
22 * Florian octo Forster <octo at verplant.org>
23 **/
25 #include "collectd.h"
26 #include "common.h"
27 #include "plugin.h"
28 #include "utils_debug.h"
29 #include "configfile.h"
31 /* Include header files for the mach system, if they exist.. */
32 #if HAVE_THREAD_INFO
33 # if HAVE_MACH_MACH_INIT_H
34 # include <mach/mach_init.h>
35 # endif
36 # if HAVE_MACH_HOST_PRIV_H
37 # include <mach/host_priv.h>
38 # endif
39 # if HAVE_MACH_MACH_ERROR_H
40 # include <mach/mach_error.h>
41 # endif
42 # if HAVE_MACH_MACH_HOST_H
43 # include <mach/mach_host.h>
44 # endif
45 # if HAVE_MACH_MACH_PORT_H
46 # include <mach/mach_port.h>
47 # endif
48 # if HAVE_MACH_MACH_TYPES_H
49 # include <mach/mach_types.h>
50 # endif
51 # if HAVE_MACH_MESSAGE_H
52 # include <mach/message.h>
53 # endif
54 # if HAVE_MACH_PROCESSOR_SET_H
55 # include <mach/processor_set.h>
56 # endif
57 # if HAVE_MACH_TASK_H
58 # include <mach/task.h>
59 # endif
60 # if HAVE_MACH_THREAD_ACT_H
61 # include <mach/thread_act.h>
62 # endif
63 # if HAVE_MACH_VM_REGION_H
64 # include <mach/vm_region.h>
65 # endif
66 # if HAVE_MACH_VM_MAP_H
67 # include <mach/vm_map.h>
68 # endif
69 # if HAVE_MACH_VM_PROT_H
70 # include <mach/vm_prot.h>
71 # endif
72 # if HAVE_SYS_SYSCTL_H
73 # include <sys/sysctl.h>
74 # endif
75 /* #endif HAVE_THREAD_INFO */
77 #elif KERNEL_LINUX
78 # if HAVE_LINUX_CONFIG_H
79 # include <linux/config.h>
80 # endif
81 # ifndef CONFIG_HZ
82 # define CONFIG_HZ 100
83 # endif
84 #endif /* KERNEL_LINUX */
86 #define MODULE_NAME "processes"
88 #if HAVE_THREAD_INFO || KERNEL_LINUX
89 # define PROCESSES_HAVE_READ 1
90 #else
91 # define PROCESSES_HAVE_READ 0
92 #endif
94 #define BUFSIZE 256
96 static char *processes_file = "processes.rrd";
97 static char *processes_ds_def[] =
98 {
99 "DS:running:GAUGE:"COLLECTD_HEARTBEAT":0:65535",
100 "DS:sleeping:GAUGE:"COLLECTD_HEARTBEAT":0:65535",
101 "DS:zombies:GAUGE:"COLLECTD_HEARTBEAT":0:65535",
102 "DS:stopped:GAUGE:"COLLECTD_HEARTBEAT":0:65535",
103 "DS:paging:GAUGE:"COLLECTD_HEARTBEAT":0:65535",
104 "DS:blocked:GAUGE:"COLLECTD_HEARTBEAT":0:65535",
105 NULL
106 };
107 static int processes_ds_num = 6;
109 static char *ps_rss_file = "processes/ps_rss-%s.rrd";
110 static char *ps_rss_ds_def[] =
111 {
112 /* max = 2^63 - 1 */
113 "DS:byte:GAUGE:"COLLECTD_HEARTBEAT":0:9223372036854775807",
114 NULL
115 };
116 static int ps_rss_ds_num = 1;
118 static char *ps_cputime_file = "processes/ps_cputime-%s.rrd";
119 static char *ps_cputime_ds_def[] =
120 {
121 /* 1 second in user-mode per second ought to be enough.. */
122 "DS:user:COUNTER:"COLLECTD_HEARTBEAT":0:1000000",
123 "DS:syst:COUNTER:"COLLECTD_HEARTBEAT":0:1000000",
124 NULL
125 };
126 static int ps_cputime_ds_num = 2;
128 static char *ps_count_file = "processes/ps_count-%s.rrd";
129 static char *ps_count_ds_def[] =
130 {
131 "DS:processes:GAUGE:"COLLECTD_HEARTBEAT":0:65535",
132 "DS:threads:GAUGE:"COLLECTD_HEARTBEAT":0:65535",
133 NULL
134 };
135 static int ps_count_ds_num = 2;
137 static char *ps_pagefaults_file = "processes/ps_pagefaults-%s.rrd";
138 static char *ps_pagefaults_ds_def[] =
139 {
140 /* max = 2^63 - 1 */
141 "DS:minflt:COUNTER:"COLLECTD_HEARTBEAT":0:9223372036854775807",
142 "DS:majflt:COUNTER:"COLLECTD_HEARTBEAT":0:9223372036854775807",
143 NULL
144 };
145 static int ps_pagefaults_ds_num = 2;
147 static char *config_keys[] =
148 {
149 "Process",
150 NULL
151 };
152 static int config_keys_num = 1;
154 typedef struct procstat_entry_s
155 {
156 unsigned long id;
157 unsigned long age;
159 unsigned long num_proc;
160 unsigned long num_lwp;
161 unsigned long vmem_rss;
163 unsigned long vmem_minflt;
164 unsigned long vmem_majflt;
165 unsigned long vmem_minflt_counter;
166 unsigned long vmem_majflt_counter;
168 unsigned long cpu_user;
169 unsigned long cpu_system;
170 unsigned long cpu_user_counter;
171 unsigned long cpu_system_counter;
173 struct procstat_entry_s *next;
174 } procstat_entry_t;
176 #define PROCSTAT_NAME_LEN 256
177 typedef struct procstat
178 {
179 char name[PROCSTAT_NAME_LEN];
181 unsigned long num_proc;
182 unsigned long num_lwp;
183 unsigned long vmem_rss;
185 unsigned long vmem_minflt_counter;
186 unsigned long vmem_majflt_counter;
188 unsigned long cpu_user_counter;
189 unsigned long cpu_system_counter;
191 struct procstat *next;
192 struct procstat_entry_s *instances;
193 } procstat_t;
195 static procstat_t *list_head_g = NULL;
197 #if HAVE_THREAD_INFO
198 static mach_port_t port_host_self;
199 static mach_port_t port_task_self;
201 static processor_set_name_array_t pset_list;
202 static mach_msg_type_number_t pset_list_len;
203 /* #endif HAVE_THREAD_INFO */
205 #elif KERNEL_LINUX
206 static long pagesize_g;
207 #endif /* KERNEL_LINUX */
209 #if HAVE_THREAD_INFO | KERNEL_LINUX
210 static void ps_list_register (const char *name)
211 {
212 procstat_t *new;
213 procstat_t *ptr;
215 if ((new = (procstat_t *) malloc (sizeof (procstat_t))) == NULL)
216 return;
217 memset (new, 0, sizeof (procstat_t));
218 strncpy (new->name, name, PROCSTAT_NAME_LEN);
220 for (ptr = list_head_g; ptr != NULL; ptr = ptr->next)
221 {
222 if (strcmp (ptr->name, name) == 0)
223 return;
224 if (ptr->next == NULL)
225 break;
226 }
228 if (ptr == NULL)
229 list_head_g = new;
230 else
231 ptr->next = new;
232 }
234 static procstat_t *ps_list_search (const char *name)
235 {
236 procstat_t *ptr;
238 for (ptr = list_head_g; ptr != NULL; ptr = ptr->next)
239 if (strcmp (ptr->name, name) == 0)
240 break;
242 return (ptr);
243 }
245 static void ps_list_add (const char *name, procstat_entry_t *entry)
246 {
247 procstat_t *ps;
248 procstat_entry_t *pse;
250 if (entry->id == 0)
251 return;
253 if ((ps = ps_list_search (name)) == NULL)
254 return;
256 for (pse = ps->instances; pse != NULL; pse = pse->next)
257 if ((pse->id == entry->id) || (pse->next == NULL))
258 break;
260 if ((pse == NULL) || (pse->id != entry->id))
261 {
262 procstat_entry_t *new;
264 new = (procstat_entry_t *) malloc (sizeof (procstat_entry_t));
265 if (new == NULL)
266 return;
267 memset (new, 0, sizeof (procstat_entry_t));
268 new->id = entry->id;
270 if (pse == NULL)
271 ps->instances = new;
272 else
273 pse->next = new;
275 pse = new;
276 }
278 pse->age = 0;
279 pse->num_proc = entry->num_proc;
280 pse->num_lwp = entry->num_lwp;
281 pse->vmem_rss = entry->vmem_rss;
283 ps->num_proc += pse->num_proc;
284 ps->num_lwp += pse->num_lwp;
285 ps->vmem_rss += pse->vmem_rss;
287 if ((entry->vmem_minflt_counter == 0)
288 && (entry->vmem_majflt_counter == 0))
289 {
290 pse->vmem_minflt_counter += entry->vmem_minflt;
291 pse->vmem_minflt = entry->vmem_minflt;
293 pse->vmem_majflt_counter += entry->vmem_majflt;
294 pse->vmem_majflt = entry->vmem_majflt;
295 }
296 else
297 {
298 if (entry->vmem_minflt_counter < pse->vmem_minflt_counter)
299 {
300 pse->vmem_minflt = entry->vmem_minflt_counter
301 + (ULONG_MAX - pse->vmem_minflt_counter);
302 }
303 else
304 {
305 pse->vmem_minflt = entry->vmem_minflt_counter - pse->vmem_minflt_counter;
306 }
307 pse->vmem_minflt_counter = entry->vmem_minflt_counter;
309 if (entry->vmem_majflt_counter < pse->vmem_majflt_counter)
310 {
311 pse->vmem_majflt = entry->vmem_majflt_counter
312 + (ULONG_MAX - pse->vmem_majflt_counter);
313 }
314 else
315 {
316 pse->vmem_majflt = entry->vmem_majflt_counter - pse->vmem_majflt_counter;
317 }
318 pse->vmem_majflt_counter = entry->vmem_majflt_counter;
319 }
321 ps->vmem_minflt_counter += pse->vmem_minflt;
322 ps->vmem_majflt_counter += pse->vmem_majflt;
324 if ((entry->cpu_user_counter == 0)
325 && (entry->cpu_system_counter == 0))
326 {
327 pse->cpu_user_counter += entry->cpu_user;
328 pse->cpu_user = entry->cpu_user;
330 pse->cpu_system_counter += entry->cpu_system;
331 pse->cpu_system = entry->cpu_system;
332 }
333 else
334 {
335 if (entry->cpu_user_counter < pse->cpu_user_counter)
336 {
337 pse->cpu_user = entry->cpu_user_counter
338 + (ULONG_MAX - pse->cpu_user_counter);
339 }
340 else
341 {
342 pse->cpu_user = entry->cpu_user_counter - pse->cpu_user_counter;
343 }
344 pse->cpu_user_counter = entry->cpu_user_counter;
346 if (entry->cpu_system_counter < pse->cpu_system_counter)
347 {
348 pse->cpu_system = entry->cpu_system_counter
349 + (ULONG_MAX - pse->cpu_system_counter);
350 }
351 else
352 {
353 pse->cpu_system = entry->cpu_system_counter - pse->cpu_system_counter;
354 }
355 pse->cpu_system_counter = entry->cpu_system_counter;
356 }
358 ps->cpu_user_counter += pse->cpu_user;
359 ps->cpu_system_counter += pse->cpu_system;
360 }
362 static void ps_list_reset (void)
363 {
364 procstat_t *ps;
365 procstat_entry_t *pse;
366 procstat_entry_t *pse_prev;
368 for (ps = list_head_g; ps != NULL; ps = ps->next)
369 {
370 ps->num_proc = 0;
371 ps->num_lwp = 0;
372 ps->vmem_rss = 0;
374 pse_prev = NULL;
375 pse = ps->instances;
376 while (pse != NULL)
377 {
378 if (pse->age > 10)
379 {
380 DBG ("Removing this procstat entry cause it's too old: "
381 "id = %lu; name = %s;",
382 pse->id, ps->name);
384 if (pse_prev == NULL)
385 {
386 ps->instances = pse->next;
387 free (pse);
388 pse = ps->instances;
389 }
390 else
391 {
392 pse_prev->next = pse->next;
393 free (pse);
394 pse = pse_prev->next;
395 }
396 }
397 else
398 {
399 pse->age++;
400 pse_prev = pse;
401 pse = pse->next;
402 }
403 } /* while (pse != NULL) */
404 } /* for (ps = list_head_g; ps != NULL; ps = ps->next) */
405 }
406 #endif /* HAVE_THREAD_INFO | KERNEL_LINUX */
408 static int ps_config (char *key, char *value)
409 {
410 if (strcasecmp (key, "Process") == 0)
411 {
412 ps_list_register (value);
413 }
414 else
415 {
416 return (-1);
417 }
419 return (0);
420 }
422 static void ps_init (void)
423 {
424 #if HAVE_THREAD_INFO
425 kern_return_t status;
427 port_host_self = mach_host_self ();
428 port_task_self = mach_task_self ();
430 if (pset_list != NULL)
431 {
432 vm_deallocate (port_task_self,
433 (vm_address_t) pset_list,
434 pset_list_len * sizeof (processor_set_t));
435 pset_list = NULL;
436 pset_list_len = 0;
437 }
439 if ((status = host_processor_sets (port_host_self,
440 &pset_list,
441 &pset_list_len)) != KERN_SUCCESS)
442 {
443 syslog (LOG_ERR, "host_processor_sets failed: %s\n",
444 mach_error_string (status));
445 pset_list = NULL;
446 pset_list_len = 0;
447 return;
448 }
449 /* #endif HAVE_THREAD_INFO */
451 #elif KERNEL_LINUX
452 pagesize_g = sysconf(_SC_PAGESIZE);
453 DBG ("pagesize_g = %li; CONFIG_HZ = %i;",
454 pagesize_g, CONFIG_HZ);
455 #endif /* KERNEL_LINUX */
457 return;
458 }
460 static void ps_write (char *host, char *inst, char *val)
461 {
462 rrd_update_file (host, processes_file, val,
463 processes_ds_def, processes_ds_num);
464 }
466 static void ps_rss_write (char *host, char *inst, char *val)
467 {
468 char filename[256];
469 int status;
471 status = snprintf (filename, 256, ps_rss_file, inst);
472 if ((status < 1) || (status >= 256))
473 return;
475 rrd_update_file (host, filename, val, ps_rss_ds_def, ps_rss_ds_num);
476 }
478 static void ps_cputime_write (char *host, char *inst, char *val)
479 {
480 char filename[256];
481 int status;
483 status = snprintf (filename, 256, ps_cputime_file, inst);
484 if ((status < 1) || (status >= 256))
485 return;
487 DBG ("host = %s; filename = %s; val = %s;",
488 host, filename, val);
489 rrd_update_file (host, filename, val,
490 ps_cputime_ds_def, ps_cputime_ds_num);
491 }
493 static void ps_count_write (char *host, char *inst, char *val)
494 {
495 char filename[256];
496 int status;
498 status = snprintf (filename, 256, ps_count_file, inst);
499 if ((status < 1) || (status >= 256))
500 return;
502 DBG ("host = %s; filename = %s; val = %s;",
503 host, filename, val);
504 rrd_update_file (host, filename, val,
505 ps_count_ds_def, ps_count_ds_num);
506 }
508 static void ps_pagefaults_write (char *host, char *inst, char *val)
509 {
510 char filename[256];
511 int status;
513 status = snprintf (filename, 256, ps_pagefaults_file, inst);
514 if ((status < 1) || (status >= 256))
515 return;
517 DBG ("host = %s; filename = %s; val = %s;",
518 host, filename, val);
519 rrd_update_file (host, filename, val,
520 ps_pagefaults_ds_def, ps_pagefaults_ds_num);
521 }
523 #if PROCESSES_HAVE_READ
524 static void ps_submit (int running,
525 int sleeping,
526 int zombies,
527 int stopped,
528 int paging,
529 int blocked)
530 {
531 char buf[BUFSIZE];
533 if (snprintf (buf, BUFSIZE, "%u:%i:%i:%i:%i:%i:%i",
534 (unsigned int) curtime,
535 running, sleeping, zombies, stopped, paging,
536 blocked) >= BUFSIZE)
537 return;
539 DBG ("running = %i; sleeping = %i; zombies = %i; stopped = %i; paging = %i; blocked = %i;",
540 running, sleeping, zombies, stopped, paging, blocked);
542 plugin_submit (MODULE_NAME, "-", buf);
543 }
545 static void ps_submit_proc_list (procstat_t *ps)
546 {
547 char buffer[64];
549 if (ps == NULL)
550 return;
552 snprintf (buffer, 64, "%u:%lu",
553 (unsigned int) curtime,
554 ps->vmem_rss);
555 buffer[63] = '\0';
556 plugin_submit ("ps_rss", ps->name, buffer);
558 snprintf (buffer, 64, "%u:%u:%u",
559 (unsigned int) curtime,
560 /* Make the counter overflow */
561 (unsigned int) (ps->cpu_user_counter & 0xFFFFFFFF),
562 (unsigned int) (ps->cpu_system_counter & 0xFFFFFFFF));
563 buffer[63] = '\0';
564 plugin_submit ("ps_cputime", ps->name, buffer);
566 snprintf (buffer, 64, "%u:%lu:%lu",
567 (unsigned int) curtime,
568 ps->num_proc, ps->num_lwp);
569 buffer[63] = '\0';
570 plugin_submit ("ps_count", ps->name, buffer);
572 snprintf (buffer, 64, "%u:%lu:%lu",
573 (unsigned int) curtime,
574 ps->vmem_minflt_counter, ps->vmem_majflt_counter);
575 buffer[63] = '\0';
576 plugin_submit ("ps_pagefaults", ps->name, buffer);
578 DBG ("name = %s; num_proc = %lu; num_lwp = %lu; vmem_rss = %lu; "
579 "vmem_minflt_counter = %lu; vmem_majflt_counter = %lu; "
580 "cpu_user_counter = %lu; cpu_system_counter = %lu;",
581 ps->name, ps->num_proc, ps->num_lwp, ps->vmem_rss,
582 ps->vmem_minflt_counter, ps->vmem_majflt_counter, ps->cpu_user_counter,
583 ps->cpu_system_counter);
585 }
587 #if KERNEL_LINUX
588 static int *ps_read_tasks (int pid)
589 {
590 int *list = NULL;
591 int list_size = 1; /* size of allocated space, in elements */
592 int list_len = 0; /* number of currently used elements */
594 char dirname[64];
595 DIR *dh;
596 struct dirent *ent;
598 snprintf (dirname, 64, "/proc/%i/task", pid);
599 dirname[63] = '\0';
601 if ((dh = opendir (dirname)) == NULL)
602 {
603 syslog (LOG_NOTICE, "processes plugin: Failed to open directory `%s'",
604 dirname);
605 return (NULL);
606 }
608 while ((ent = readdir (dh)) != NULL)
609 {
610 if (!isdigit (ent->d_name[0]))
611 continue;
613 if ((list_len + 1) >= list_size)
614 {
615 int *new_ptr;
616 int new_size = 2 * list_size;
617 /* Comes in sizes: 2, 4, 8, 16, ... */
619 new_ptr = (int *) realloc (list, (size_t) (sizeof (int) * new_size));
620 if (new_ptr == NULL)
621 {
622 if (list != NULL)
623 free (list);
624 syslog (LOG_ERR, "processes plugin: "
625 "Failed to allocate more memory.");
626 return (NULL);
627 }
629 list = new_ptr;
630 list_size = new_size;
632 memset (list + list_len, 0, sizeof (int) * (list_size - list_len));
633 }
635 list[list_len] = atoi (ent->d_name);
636 if (list[list_len] != 0)
637 list_len++;
638 }
640 closedir (dh);
642 assert (list_len < list_size);
643 assert (list[list_len] == 0);
645 return (list);
646 }
648 int ps_read_process (int pid, procstat_t *ps, char *state)
649 {
650 char filename[64];
651 char buffer[1024];
652 FILE *fh;
654 char *fields[64];
655 char fields_len;
657 int *tasks;
658 int i;
660 int ppid;
661 int name_len;
663 long long unsigned cpu_user_counter;
664 long long unsigned cpu_system_counter;
665 long long unsigned vmem_rss;
667 memset (ps, 0, sizeof (procstat_t));
669 snprintf (filename, 64, "/proc/%i/stat", pid);
670 filename[63] = '\0';
672 if ((fh = fopen (filename, "r")) == NULL)
673 return (-1);
675 if (fgets (buffer, 1024, fh) == NULL)
676 {
677 fclose (fh);
678 return (-1);
679 }
681 fclose (fh);
683 fields_len = strsplit (buffer, fields, 64);
684 if (fields_len < 24)
685 {
686 DBG ("`%s' has only %i fields..",
687 filename, fields_len);
688 return (-1);
689 }
690 else if (fields_len != 41)
691 {
692 DBG ("WARNING: (fields_len = %i) != 41", fields_len);
693 }
695 /* copy the name, strip brackets in the process */
696 name_len = strlen (fields[1]) - 2;
697 if ((fields[1][0] != '(') || (fields[1][name_len + 1] != ')'))
698 {
699 DBG ("No brackets found in process name: `%s'", fields[1]);
700 return (-1);
701 }
702 fields[1] = fields[1] + 1;
703 fields[1][name_len] = '\0';
704 strncpy (ps->name, fields[1], PROCSTAT_NAME_LEN);
706 ppid = atoi (fields[3]);
708 if ((tasks = ps_read_tasks (pid)) == NULL)
709 {
710 /* This happends for zombied, e.g. */
711 DBG ("ps_read_tasks (%i) failed.", pid);
712 *state = 'Z';
713 ps->num_lwp = 0;
714 ps->num_proc = 0;
715 }
716 else
717 {
718 *state = '\0';
719 ps->num_lwp = 0;
720 ps->num_proc = 1;
721 for (i = 0; tasks[i] != 0; i++)
722 ps->num_lwp++;
724 free (tasks);
725 tasks = NULL;
726 }
728 /* Leave the rest at zero if this is only an LWP */
729 if (ps->num_proc == 0)
730 {
731 DBG ("This is only an LWP: pid = %i; name = %s;",
732 pid, ps->name);
733 return (0);
734 }
736 cpu_user_counter = atoll (fields[13]);
737 cpu_system_counter = atoll (fields[14]);
738 vmem_rss = atoll (fields[23]);
739 ps->vmem_minflt_counter = atol (fields[9]);
740 ps->vmem_majflt_counter = atol (fields[11]);
742 /* Convert jiffies to useconds */
743 cpu_user_counter = cpu_user_counter * 1000000 / CONFIG_HZ;
744 cpu_system_counter = cpu_system_counter * 1000000 / CONFIG_HZ;
745 vmem_rss = vmem_rss * pagesize_g;
747 ps->cpu_user_counter = (unsigned long) cpu_user_counter;
748 ps->cpu_system_counter = (unsigned long) cpu_system_counter;
749 ps->vmem_rss = (unsigned long) vmem_rss;
751 *state = fields[2][0];
753 /* success */
754 return (0);
755 } /* int ps_read_process (...) */
756 #endif /* KERNEL_LINUX */
758 #if HAVE_THREAD_INFO
759 static int mach_get_task_name (task_t t, int *pid, char *name, size_t name_max_len)
760 {
761 int mib[4];
763 struct kinfo_proc kp;
764 size_t kp_size;
766 mib[0] = CTL_KERN;
767 mib[1] = KERN_PROC;
768 mib[2] = KERN_PROC_PID;
770 if (pid_for_task (t, pid) != KERN_SUCCESS)
771 return (-1);
772 mib[3] = *pid;
774 kp_size = sizeof (kp);
775 if (sysctl (mib, 4, &kp, &kp_size, NULL, 0) != 0)
776 return (-1);
778 if (name_max_len > (MAXCOMLEN + 1))
779 name_max_len = MAXCOMLEN + 1;
781 strncpy (name, kp.kp_proc.p_comm, name_max_len - 1);
782 name[name_max_len - 1] = '\0';
784 DBG ("pid = %i; name = %s;", *pid, name);
786 /* We don't do the special handling for `p_comm == "LaunchCFMApp"' as
787 * `top' does it, because it is a lot of work and only used when
788 * debugging. -octo */
790 return (0);
791 }
792 #endif /* HAVE_THREAD_INFO */
794 static void ps_read (void)
795 {
796 #if HAVE_THREAD_INFO
797 kern_return_t status;
799 int pset;
800 processor_set_t port_pset_priv;
802 int task;
803 task_array_t task_list;
804 mach_msg_type_number_t task_list_len;
806 int task_pid;
807 char task_name[MAXCOMLEN + 1];
809 int thread;
810 thread_act_array_t thread_list;
811 mach_msg_type_number_t thread_list_len;
812 thread_basic_info_data_t thread_data;
813 mach_msg_type_number_t thread_data_len;
815 int running = 0;
816 int sleeping = 0;
817 int zombies = 0;
818 int stopped = 0;
819 int blocked = 0;
821 procstat_t *ps;
822 procstat_entry_t pse;
824 ps_list_reset ();
826 /*
827 * The Mach-concept is a little different from the traditional UNIX
828 * concept: All the work is done in threads. Threads are contained in
829 * `tasks'. Therefore, `task status' doesn't make much sense, since
830 * it's actually a `thread status'.
831 * Tasks are assigned to sets of processors, so that's where you go to
832 * get a list.
833 */
834 for (pset = 0; pset < pset_list_len; pset++)
835 {
836 if ((status = host_processor_set_priv (port_host_self,
837 pset_list[pset],
838 &port_pset_priv)) != KERN_SUCCESS)
839 {
840 syslog (LOG_ERR, "host_processor_set_priv failed: %s\n",
841 mach_error_string (status));
842 continue;
843 }
845 if ((status = processor_set_tasks (port_pset_priv,
846 &task_list,
847 &task_list_len)) != KERN_SUCCESS)
848 {
849 syslog (LOG_ERR, "processor_set_tasks failed: %s\n",
850 mach_error_string (status));
851 mach_port_deallocate (port_task_self, port_pset_priv);
852 continue;
853 }
855 for (task = 0; task < task_list_len; task++)
856 {
857 ps = NULL;
858 if (mach_get_task_name (task_list[task],
859 &task_pid,
860 task_name, PROCSTAT_NAME_LEN) == 0)
861 ps = ps_list_search (task_name);
863 /* Collect more detailed statistics for this process */
864 if (ps != NULL)
865 {
866 task_basic_info_data_t task_basic_info;
867 mach_msg_type_number_t task_basic_info_len;
868 task_events_info_data_t task_events_info;
869 mach_msg_type_number_t task_events_info_len;
870 task_absolutetime_info_data_t task_absolutetime_info;
871 mach_msg_type_number_t task_absolutetime_info_len;
873 memset (&pse, '\0', sizeof (pse));
874 pse.id = task_pid;
876 task_basic_info_len = TASK_BASIC_INFO_COUNT;
877 status = task_info (task_list[task],
878 TASK_BASIC_INFO,
879 (task_info_t) &task_basic_info,
880 &task_basic_info_len);
881 if (status != KERN_SUCCESS)
882 {
883 syslog (LOG_ERR, "task_info failed: %s",
884 mach_error_string (status));
885 continue; /* with next thread_list */
886 }
888 task_events_info_len = TASK_EVENTS_INFO_COUNT;
889 status = task_info (task_list[task],
890 TASK_EVENTS_INFO,
891 (task_info_t) &task_events_info,
892 &task_events_info_len);
893 if (status != KERN_SUCCESS)
894 {
895 syslog (LOG_ERR, "task_info failed: %s",
896 mach_error_string (status));
897 continue; /* with next thread_list */
898 }
900 task_absolutetime_info_len = TASK_ABSOLUTETIME_INFO_COUNT;
901 status = task_info (task_list[task],
902 TASK_ABSOLUTETIME_INFO,
903 (task_info_t) &task_absolutetime_info,
904 &task_absolutetime_info_len);
905 if (status != KERN_SUCCESS)
906 {
907 syslog (LOG_ERR, "task_info failed: %s",
908 mach_error_string (status));
909 continue; /* with next thread_list */
910 }
912 pse.num_proc++;
913 pse.vmem_rss = task_basic_info.resident_size;
915 pse.vmem_minflt_counter = task_events_info.cow_faults;
916 pse.vmem_majflt_counter = task_events_info.faults;
918 pse.cpu_user_counter = task_absolutetime_info.total_user;
919 pse.cpu_system_counter = task_absolutetime_info.total_system;
920 }
922 status = task_threads (task_list[task], &thread_list,
923 &thread_list_len);
924 if (status != KERN_SUCCESS)
925 {
926 /* Apple's `top' treats this case a zombie. It
927 * makes sense to some extend: A `zombie'
928 * thread is nonsense, since the task/process
929 * is dead. */
930 zombies++;
931 DBG ("task_threads failed: %s",
932 mach_error_string (status));
933 if (task_list[task] != port_task_self)
934 mach_port_deallocate (port_task_self,
935 task_list[task]);
936 continue; /* with next task_list */
937 }
939 for (thread = 0; thread < thread_list_len; thread++)
940 {
941 thread_data_len = THREAD_BASIC_INFO_COUNT;
942 status = thread_info (thread_list[thread],
943 THREAD_BASIC_INFO,
944 (thread_info_t) &thread_data,
945 &thread_data_len);
946 if (status != KERN_SUCCESS)
947 {
948 syslog (LOG_ERR, "thread_info failed: %s",
949 mach_error_string (status));
950 if (task_list[task] != port_task_self)
951 mach_port_deallocate (port_task_self,
952 thread_list[thread]);
953 continue; /* with next thread_list */
954 }
956 if (ps != NULL)
957 pse.num_lwp++;
959 switch (thread_data.run_state)
960 {
961 case TH_STATE_RUNNING:
962 running++;
963 break;
964 case TH_STATE_STOPPED:
965 /* What exactly is `halted'? */
966 case TH_STATE_HALTED:
967 stopped++;
968 break;
969 case TH_STATE_WAITING:
970 sleeping++;
971 break;
972 case TH_STATE_UNINTERRUPTIBLE:
973 blocked++;
974 break;
975 /* There is no `zombie' case here,
976 * since there are no zombie-threads.
977 * There's only zombie tasks, which are
978 * handled above. */
979 default:
980 syslog (LOG_WARNING,
981 "Unknown thread status: %s",
982 thread_data.run_state);
983 break;
984 } /* switch (thread_data.run_state) */
986 if (task_list[task] != port_task_self)
987 {
988 status = mach_port_deallocate (port_task_self,
989 thread_list[thread]);
990 if (status != KERN_SUCCESS)
991 syslog (LOG_ERR, "mach_port_deallocate failed: %s",
992 mach_error_string (status));
993 }
994 } /* for (thread_list) */
996 if ((status = vm_deallocate (port_task_self,
997 (vm_address_t) thread_list,
998 thread_list_len * sizeof (thread_act_t)))
999 != KERN_SUCCESS)
1000 {
1001 syslog (LOG_ERR, "vm_deallocate failed: %s",
1002 mach_error_string (status));
1003 }
1004 thread_list = NULL;
1005 thread_list_len = 0;
1007 /* Only deallocate the task port, if it isn't our own.
1008 * Don't know what would happen in that case, but this
1009 * is what Apple's top does.. ;) */
1010 if (task_list[task] != port_task_self)
1011 {
1012 status = mach_port_deallocate (port_task_self,
1013 task_list[task]);
1014 if (status != KERN_SUCCESS)
1015 syslog (LOG_ERR, "mach_port_deallocate failed: %s",
1016 mach_error_string (status));
1017 }
1019 if (ps != NULL)
1020 ps_list_add (task_name, &pse);
1021 } /* for (task_list) */
1023 if ((status = vm_deallocate (port_task_self,
1024 (vm_address_t) task_list,
1025 task_list_len * sizeof (task_t))) != KERN_SUCCESS)
1026 {
1027 syslog (LOG_ERR, "vm_deallocate failed: %s",
1028 mach_error_string (status));
1029 }
1030 task_list = NULL;
1031 task_list_len = 0;
1033 if ((status = mach_port_deallocate (port_task_self, port_pset_priv))
1034 != KERN_SUCCESS)
1035 {
1036 syslog (LOG_ERR, "mach_port_deallocate failed: %s",
1037 mach_error_string (status));
1038 }
1039 } /* for (pset_list) */
1041 ps_submit (running, sleeping, zombies, stopped, -1, blocked);
1043 for (ps = list_head_g; ps != NULL; ps = ps->next)
1044 ps_submit_proc_list (ps);
1045 /* #endif HAVE_THREAD_INFO */
1047 #elif KERNEL_LINUX
1048 int running = 0;
1049 int sleeping = 0;
1050 int zombies = 0;
1051 int stopped = 0;
1052 int paging = 0;
1053 int blocked = 0;
1055 struct dirent *ent;
1056 DIR *proc;
1057 int pid;
1059 int status;
1060 procstat_t ps;
1061 procstat_entry_t pse;
1062 char state;
1064 procstat_t *ps_ptr;
1066 running = sleeping = zombies = stopped = paging = blocked = 0;
1067 ps_list_reset ();
1069 if ((proc = opendir ("/proc")) == NULL)
1070 {
1071 syslog (LOG_ERR, "Cannot open `/proc': %s", strerror (errno));
1072 return;
1073 }
1075 while ((ent = readdir (proc)) != NULL)
1076 {
1077 if (!isdigit (ent->d_name[0]))
1078 continue;
1080 if ((pid = atoi (ent->d_name)) < 1)
1081 continue;
1083 status = ps_read_process (pid, &ps, &state);
1084 if (status != 0)
1085 {
1086 DBG ("ps_read_process failed: %i", status);
1087 continue;
1088 }
1090 pse.id = pid;
1091 pse.age = 0;
1093 pse.num_proc = ps.num_proc;
1094 pse.num_lwp = ps.num_lwp;
1095 pse.vmem_rss = ps.vmem_rss;
1097 pse.vmem_minflt = 0;
1098 pse.vmem_minflt_counter = ps.vmem_minflt_counter;
1099 pse.vmem_majflt = 0;
1100 pse.vmem_majflt_counter = ps.vmem_majflt_counter;
1102 pse.cpu_user = 0;
1103 pse.cpu_user_counter = ps.cpu_user_counter;
1104 pse.cpu_system = 0;
1105 pse.cpu_system_counter = ps.cpu_system_counter;
1107 switch (state)
1108 {
1109 case 'R': running++; break;
1110 case 'S': sleeping++; break;
1111 case 'D': blocked++; break;
1112 case 'Z': zombies++; break;
1113 case 'T': stopped++; break;
1114 case 'W': paging++; break;
1115 }
1117 ps_list_add (ps.name, &pse);
1118 }
1120 closedir (proc);
1122 ps_submit (running, sleeping, zombies, stopped, paging, blocked);
1124 for (ps_ptr = list_head_g; ps_ptr != NULL; ps_ptr = ps_ptr->next)
1125 ps_submit_proc_list (ps_ptr);
1126 #endif /* KERNEL_LINUX */
1127 }
1128 #else
1129 # define ps_read NULL
1130 #endif /* PROCESSES_HAVE_READ */
1132 void module_register (void)
1133 {
1134 plugin_register (MODULE_NAME, ps_init, ps_read, ps_write);
1135 plugin_register ("ps_rss", NULL, NULL, ps_rss_write);
1136 plugin_register ("ps_cputime", NULL, NULL, ps_cputime_write);
1137 plugin_register ("ps_count", NULL, NULL, ps_count_write);
1138 plugin_register ("ps_pagefaults", NULL, NULL, ps_pagefaults_write);
1139 cf_register (MODULE_NAME, ps_config, config_keys, config_keys_num);
1140 }
1142 #undef BUFSIZE
1143 #undef MODULE_NAME