X-Git-Url: https://git.tokkee.org/?a=blobdiff_plain;f=src%2Fcpu.c;h=d9453aa31f95d403870f343d642636b487917ad6;hb=86090b3d409634d016da7bbb0cbe43ed5a851d30;hp=6832b5239a1b9bad95d931b5061636e1fa01118c;hpb=73681054cbff428d352d8ed707d24fab1bf67b24;p=collectd.git diff --git a/src/cpu.c b/src/cpu.c index 6832b523..d9453aa3 100644 --- a/src/cpu.c +++ b/src/cpu.c @@ -1,11 +1,10 @@ /** * collectd - src/cpu.c - * Copyright (C) 2005 Florian octo Forster + * Copyright (C) 2005-2007 Florian octo Forster * * This program is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License as published by the - * Free Software Foundation; either version 2 of the License, or (at your - * option) any later version. + * Free Software Foundation; only version 2 of the License is applicable. * * This program is distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty of @@ -23,13 +22,28 @@ #include "collectd.h" #include "common.h" #include "plugin.h" +#include "utils_debug.h" -#define MODULE_NAME "cpu" - -#if defined(KERNEL_LINUX) || defined(HAVE_LIBKSTAT) || defined(HAVE_SYSCTLBYNAME) -# define CPU_HAVE_READ 1 -#else -# define CPU_HAVE_READ 0 +#ifdef HAVE_MACH_KERN_RETURN_H +# include +#endif +#ifdef HAVE_MACH_MACH_INIT_H +# include +#endif +#ifdef HAVE_MACH_HOST_PRIV_H +# include +#endif +#if HAVE_MACH_MACH_ERROR_H +# include +#endif +#ifdef HAVE_MACH_PROCESSOR_INFO_H +# include +#endif +#ifdef HAVE_MACH_PROCESSOR_H +# include +#endif +#ifdef HAVE_MACH_VM_MAP_H +# include #endif #ifdef HAVE_LIBKSTAT @@ -55,40 +69,87 @@ # endif #endif /* HAVE_SYSCTLBYNAME */ -#ifdef HAVE_LIBKSTAT +#if defined(PROCESSOR_CPU_LOAD_INFO) || defined(KERNEL_LINUX) || defined(HAVE_LIBKSTAT) || defined(HAVE_SYSCTLBYNAME) +# define CPU_HAVE_READ 1 +#else +# define CPU_HAVE_READ 0 +#endif + +static data_source_t dsrc[5] = +{ + {"user", DS_TYPE_COUNTER, 0, 4294967295.0}, + {"nice", DS_TYPE_COUNTER, 0, 4294967295.0}, + {"syst", DS_TYPE_COUNTER, 0, 4294967295.0}, + {"idle", DS_TYPE_COUNTER, 0, 4294967295.0}, + {"wait", DS_TYPE_COUNTER, 0, 4294967295.0} +}; + +static data_set_t ds = +{ + "cpu", 5, dsrc +}; + +#if CPU_HAVE_READ +#ifdef PROCESSOR_CPU_LOAD_INFO +static mach_port_t port_host; +static processor_port_array_t cpu_list; +static mach_msg_type_number_t cpu_list_len; + +#if PROCESSOR_TEMPERATURE +static int cpu_temp_retry_counter = 0; +static int cpu_temp_retry_step = 1; +static int cpu_temp_retry_max = 1; +#endif /* PROCESSOR_TEMPERATURE */ +/* #endif PROCESSOR_CPU_LOAD_INFO */ + +#elif defined(KERNEL_LINUX) +/* no variables needed */ +/* #endif KERNEL_LINUX */ + +#elif defined(HAVE_LIBKSTAT) /* colleague tells me that Sun doesn't sell systems with more than 100 or so CPUs.. */ # define MAX_NUMCPU 256 extern kstat_ctl_t *kc; static kstat_t *ksp[MAX_NUMCPU]; static int numcpu; -#endif /* HAVE_LIBKSTAT */ +/* #endif HAVE_LIBKSTAT */ -#ifdef HAVE_SYSCTLBYNAME +#elif defined(HAVE_SYSCTLBYNAME) static int numcpu; #endif /* HAVE_SYSCTLBYNAME */ -static char *cpu_filename = "cpu-%s.rrd"; - -static char *ds_def[] = +static int init (void) { - "DS:user:COUNTER:25:0:100", - "DS:nice:COUNTER:25:0:100", - "DS:syst:COUNTER:25:0:100", - "DS:idle:COUNTER:25:0:100", - "DS:wait:COUNTER:25:0:100", - NULL -}; -static int ds_num = 5; +#if PROCESSOR_CPU_LOAD_INFO || PROCESSOR_TEMPERATURE + kern_return_t status; + int collectd_step; -void cpu_init (void) -{ -#ifdef HAVE_LIBKSTAT + port_host = mach_host_self (); + + /* FIXME: Free `cpu_list' if it's not NULL */ + if ((status = host_processors (port_host, &cpu_list, &cpu_list_len)) != KERN_SUCCESS) + { + syslog (LOG_ERR, "cpu plugin: host_processors returned %i\n", (int) status); + cpu_list_len = 0; + return (-1); + } + + DBG ("host_processors returned %i %s", (int) cpu_list_len, cpu_list_len == 1 ? "processor" : "processors"); + syslog (LOG_INFO, "cpu plugin: Found %i processor%s.", (int) cpu_list_len, cpu_list_len == 1 ? "" : "s"); + + collectd_step = atoi (COLLECTD_STEP); + if ((collectd_step > 0) && (collectd_step <= 86400)) + cpu_temp_retry_max = 86400 / collectd_step; + +/* #endif PROCESSOR_CPU_LOAD_INFO */ + +#elif defined(HAVE_LIBKSTAT) kstat_t *ksp_chain; numcpu = 0; if (kc == NULL) - return; + return (-1); /* Solaris doesn't count linear.. *sigh* */ for (numcpu = 0, ksp_chain = kc->kc_chain; @@ -106,68 +167,157 @@ void cpu_init (void) if (sysctlbyname ("hw.ncpu", &numcpu, &numcpu_size, NULL, 0) < 0) { syslog (LOG_WARNING, "cpu: sysctlbyname: %s", strerror (errno)); - return; + return (-1); } if (numcpu != 1) syslog (LOG_NOTICE, "cpu: Only one processor supported when using `sysctlbyname' (found %i)", numcpu); #endif - return; + return (0); +} /* int init */ + +static void submit (int cpu_num, unsigned long long user, + unsigned long long nice, unsigned long long syst, + unsigned long long idle, unsigned long long wait) +{ + value_t values[5]; + value_list_t vl = VALUE_LIST_INIT; + + values[0].counter = user; + values[1].counter = nice; + values[2].counter = syst; + values[3].counter = idle; + values[4].counter = wait; + + vl.values = values; + vl.values_len = 2; + vl.time = time (NULL); + strcpy (vl.host, hostname); + strcpy (vl.plugin, "cpu"); + strcpy (vl.plugin_instance, ""); + snprintf (vl.type_instance, sizeof (vl.type_instance), + "%i", cpu_num); + vl.type_instance[DATA_MAX_NAME_LEN - 1] = '\0'; + + plugin_dispatch_values ("cpu", &vl); } -void cpu_write (char *host, char *inst, char *val) +static int cpu_read (void) { - char file[512]; - int status; +#if PROCESSOR_CPU_LOAD_INFO || PROCESSOR_TEMPERATURE + int cpu; - status = snprintf (file, 512, cpu_filename, inst); - if (status < 1) - return; - else if (status >= 512) - return; + kern_return_t status; + +#if PROCESSOR_CPU_LOAD_INFO + processor_cpu_load_info_data_t cpu_info; + mach_msg_type_number_t cpu_info_len; +#endif +#if PROCESSOR_TEMPERATURE + processor_info_data_t cpu_temp; + mach_msg_type_number_t cpu_temp_len; +#endif - rrd_update_file (host, file, val, ds_def, ds_num); -} + host_t cpu_host; -#define BUFSIZE 512 -static void cpu_submit (int cpu_num, unsigned long long user, - unsigned long long nice, unsigned long long syst, - unsigned long long idle, unsigned long long wait) -{ - char buf[BUFSIZE]; - char cpu[16]; + for (cpu = 0; cpu < cpu_list_len; cpu++) + { +#if PROCESSOR_CPU_LOAD_INFO + cpu_host = 0; + cpu_info_len = PROCESSOR_BASIC_INFO_COUNT; - if (snprintf (buf, BUFSIZE, "%u:%llu:%llu:%llu:%llu:%llu", (unsigned int) curtime, - user, nice, syst, idle, wait) >= BUFSIZE) - return; - snprintf (cpu, 16, "%i", cpu_num); + if ((status = processor_info (cpu_list[cpu], + PROCESSOR_CPU_LOAD_INFO, &cpu_host, + (processor_info_t) &cpu_info, &cpu_info_len)) != KERN_SUCCESS) + { + syslog (LOG_ERR, "cpu plugin: processor_info failed with status %i", (int) status); + continue; + } - plugin_submit (MODULE_NAME, cpu, buf); -} -#undef BUFSIZE + if (cpu_info_len < CPU_STATE_MAX) + { + syslog (LOG_ERR, "cpu plugin: processor_info returned only %i elements..", cpu_info_len); + continue; + } -#if CPU_HAVE_READ -static void cpu_read (void) -{ -#ifdef KERNEL_LINUX -#define BUFSIZE 1024 + submit (cpu, cpu_info.cpu_ticks[CPU_STATE_USER], + cpu_info.cpu_ticks[CPU_STATE_NICE], + cpu_info.cpu_ticks[CPU_STATE_SYSTEM], + cpu_info.cpu_ticks[CPU_STATE_IDLE], + 0ULL); +#endif /* PROCESSOR_CPU_LOAD_INFO */ +#if PROCESSOR_TEMPERATURE + /* + * Not all Apple computers do have this ability. To minimize + * the messages sent to the syslog we do an exponential + * stepback if `processor_info' fails. We still try ~once a day + * though.. + */ + if (cpu_temp_retry_counter > 0) + { + cpu_temp_retry_counter--; + continue; + } + + cpu_temp_len = PROCESSOR_INFO_MAX; + + status = processor_info (cpu_list[cpu], + PROCESSOR_TEMPERATURE, + &cpu_host, + cpu_temp, &cpu_temp_len); + if (status != KERN_SUCCESS) + { + syslog (LOG_ERR, "cpu plugin: processor_info failed: %s", + mach_error_string (status)); + + cpu_temp_retry_counter = cpu_temp_retry_step; + cpu_temp_retry_step *= 2; + if (cpu_temp_retry_step > cpu_temp_retry_max) + cpu_temp_retry_step = cpu_temp_retry_max; + + continue; + } + + if (cpu_temp_len != 1) + { + DBG ("processor_info (PROCESSOR_TEMPERATURE) returned %i elements..?", + (int) cpu_temp_len); + continue; + } + + cpu_temp_retry_counter = 0; + cpu_temp_retry_step = 1; + + DBG ("cpu_temp = %i", (int) cpu_temp); +#endif /* PROCESSOR_TEMPERATURE */ + } +/* #endif PROCESSOR_CPU_LOAD_INFO */ + +#elif defined(KERNEL_LINUX) int cpu; unsigned long long user, nice, syst, idle; unsigned long long wait, intr, sitr; /* sitr == soft interrupt */ FILE *fh; - char buf[BUFSIZE]; + char buf[1024]; char *fields[9]; int numfields; + static complain_t complain_obj; + if ((fh = fopen ("/proc/stat", "r")) == NULL) { - syslog (LOG_WARNING, "cpu: fopen: %s", strerror (errno)); - return; + plugin_complain (LOG_ERR, &complain_obj, "cpu plugin: " + "fopen (/proc/stat) failed: %s", + strerror (errno)); + return (-1); } - while (fgets (buf, BUFSIZE, fh) != NULL) + plugin_relief (LOG_NOTICE, &complain_obj, "cpu plugin: " + "fopen (/proc/stat) succeeded."); + + while (fgets (buf, 1024, fh) != NULL) { if (strncmp (buf, "cpu", 3)) continue; @@ -199,11 +349,10 @@ static void cpu_read (void) wait = 0LL; } - cpu_submit (cpu, user, nice, syst, idle, wait); + submit (cpu, user, nice, syst, idle, wait); } fclose (fh); -#undef BUFSIZE /* #endif defined(KERNEL_LINUX) */ #elif defined(HAVE_LIBKSTAT) @@ -224,7 +373,7 @@ static void cpu_read (void) syst = (unsigned long long) cs.cpu_sysinfo.cpu[CPU_KERNEL]; wait = (unsigned long long) cs.cpu_sysinfo.cpu[CPU_WAIT]; - cpu_submit (ksp[cpu]->ks_instance, + submit (ksp[cpu]->ks_instance, user, 0LL, syst, idle, wait); } /* #endif defined(HAVE_LIBKSTAT) */ @@ -233,33 +382,37 @@ static void cpu_read (void) long cpuinfo[CPUSTATES]; size_t cpuinfo_size; + static complain_t complain_obj; + cpuinfo_size = sizeof (cpuinfo); if (sysctlbyname("kern.cp_time", &cpuinfo, &cpuinfo_size, NULL, 0) < 0) { - syslog (LOG_WARNING, "cpu: sysctlbyname: %s", strerror (errno)); + plugin_complain (LOG_ERR, &complain_obj, "cpu plugin: " + "sysctlbyname failed: %s.", + strerror (errno)); return; } + plugin_relief (LOG_NOTICE, &complain_obj, "cpu plugin: " + "sysctlbyname succeeded."); + cpuinfo[CP_SYS] += cpuinfo[CP_INTR]; /* FIXME: Instance is always `0' */ - cpu_submit (0, cpuinfo[CP_USER], cpuinfo[CP_NICE], cpuinfo[CP_SYS], cpuinfo[CP_IDLE], 0LL); + submit (0, cpuinfo[CP_USER], cpuinfo[CP_NICE], cpuinfo[CP_SYS], cpuinfo[CP_IDLE], 0LL); #endif - return; + return (0); } #endif /* CPU_HAVE_READ */ void module_register (void) { - plugin_register (MODULE_NAME, cpu_init, + plugin_register_data_set (&ds); + #if CPU_HAVE_READ - cpu_read, -#else - NULL, -#endif - cpu_write); + plugin_register_init ("cpu", init); + plugin_register_read ("cpu", cpu_read); +#endif /* CPU_HAVE_READ */ } - -#undef MODULE_NAME