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