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