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