X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Fcpu.c;h=64ce5df9e27b8158942c70203ceb763a54f1bead;hb=0ebe3efc7b797546bcae87a946cff1c4cf472919;hp=b087d6654d2edfa84480fd5e6d5dffec80aa5392;hpb=dfe12aadb65933d7dd15d5e42d3c6218900bed15;p=collectd.git diff --git a/src/cpu.c b/src/cpu.c index b087d665..64ce5df9 100644 --- a/src/cpu.c +++ b/src/cpu.c @@ -1,11 +1,12 @@ /** * collectd - src/cpu.c - * Copyright (C) 2005,2006 Florian octo Forster + * Copyright (C) 2005-2009 Florian octo Forster + * Copyright (C) 2008 Oleg King + * Copyright (C) 2009 Simon Kuhnle * * 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 @@ -18,14 +19,13 @@ * * Authors: * Florian octo Forster + * Oleg King + * Simon Kuhnle **/ #include "collectd.h" #include "common.h" #include "plugin.h" -#include "utils_debug.h" - -#define MODULE_NAME "cpu" #ifdef HAVE_MACH_KERN_RETURN_H # include @@ -53,7 +53,8 @@ # include #endif /* HAVE_LIBKSTAT */ -#ifdef HAVE_SYSCTLBYNAME +#if (defined(HAVE_SYSCTL) && HAVE_SYSCTL) \ + || (defined(HAVE_SYSCTLBYNAME) && HAVE_SYSCTLBYNAME) # ifdef HAVE_SYS_SYSCTL_H # include # endif @@ -70,12 +71,26 @@ # define CP_IDLE 4 # define CPUSTATES 5 # endif -#endif /* HAVE_SYSCTLBYNAME */ - -#if defined(PROCESSOR_CPU_LOAD_INFO) || defined(KERNEL_LINUX) || defined(HAVE_LIBKSTAT) || defined(HAVE_SYSCTLBYNAME) -# define CPU_HAVE_READ 1 +#endif /* HAVE_SYSCTL || HAVE_SYSCTLBYNAME */ + +#if HAVE_SYSCTL +# if defined(CTL_HW) && defined(HW_NCPU) \ + && defined(CTL_KERN) && defined(KERN_CPTIME) && defined(CPUSTATES) +# define CAN_USE_SYSCTL 1 +# else +# define CAN_USE_SYSCTL 0 +# endif #else -# define CPU_HAVE_READ 0 +# define CAN_USE_SYSCTL 0 +#endif + +#if HAVE_STATGRAB_H +# include +#endif + +#if !PROCESSOR_CPU_LOAD_INFO && !KERNEL_LINUX && !HAVE_LIBKSTAT \ + && !CAN_USE_SYSCTL && !HAVE_SYSCTLBYNAME && !HAVE_LIBSTATGRAB +# error "No applicable input method." #endif #ifdef PROCESSOR_CPU_LOAD_INFO @@ -102,46 +117,37 @@ static kstat_t *ksp[MAX_NUMCPU]; static int numcpu; /* #endif HAVE_LIBKSTAT */ +#elif CAN_USE_SYSCTL +static int numcpu; +/* #endif CAN_USE_SYSCTL */ + #elif defined(HAVE_SYSCTLBYNAME) static int numcpu; -#endif /* HAVE_SYSCTLBYNAME */ +/* #endif HAVE_SYSCTLBYNAME */ -static char *cpu_filename = "cpu-%s.rrd"; +#elif defined(HAVE_LIBSTATGRAB) +/* no variables needed */ +#endif /* HAVE_LIBSTATGRAB */ -static char *ds_def[] = -{ - "DS:user:COUNTER:"COLLECTD_HEARTBEAT":0:U", - "DS:nice:COUNTER:"COLLECTD_HEARTBEAT":0:U", - "DS:syst:COUNTER:"COLLECTD_HEARTBEAT":0:U", - "DS:idle:COUNTER:"COLLECTD_HEARTBEAT":0:U", - "DS:wait:COUNTER:"COLLECTD_HEARTBEAT":0:U", - NULL -}; -static int ds_num = 5; - -static void cpu_init (void) +static int init (void) { #if PROCESSOR_CPU_LOAD_INFO || PROCESSOR_TEMPERATURE kern_return_t status; - int collectd_step; 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); + ERROR ("cpu plugin: host_processors returned %i", (int) status); cpu_list_len = 0; - return; + 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"); + DEBUG ("host_processors returned %i %s", (int) cpu_list_len, cpu_list_len == 1 ? "processor" : "processors"); + 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; - + cpu_temp_retry_max = 86400 / interval_g; /* #endif PROCESSOR_CPU_LOAD_INFO */ #elif defined(HAVE_LIBKSTAT) @@ -150,7 +156,7 @@ static void cpu_init (void) numcpu = 0; if (kc == NULL) - return; + return (-1); /* Solaris doesn't count linear.. *sigh* */ for (numcpu = 0, ksp_chain = kc->kc_chain; @@ -160,6 +166,25 @@ static void cpu_init (void) ksp[numcpu++] = ksp_chain; /* #endif HAVE_LIBKSTAT */ +#elif CAN_USE_SYSCTL + size_t numcpu_size; + int mib[2] = {CTL_HW, HW_NCPU}; + int status; + + numcpu = 0; + numcpu_size = sizeof (numcpu); + + status = sysctl (mib, STATIC_ARRAY_SIZE (mib), + &numcpu, &numcpu_size, NULL, 0); + if (status == -1) + { + char errbuf[1024]; + WARNING ("cpu plugin: sysctl: %s", + sstrerror (errno, errbuf, sizeof (errbuf))); + return (-1); + } +/* #endif CAN_USE_SYSCTL */ + #elif defined (HAVE_SYSCTLBYNAME) size_t numcpu_size; @@ -167,50 +192,43 @@ static void cpu_init (void) if (sysctlbyname ("hw.ncpu", &numcpu, &numcpu_size, NULL, 0) < 0) { - syslog (LOG_WARNING, "cpu: sysctlbyname: %s", strerror (errno)); - return; + char errbuf[1024]; + WARNING ("cpu plugin: sysctlbyname: %s", + sstrerror (errno, errbuf, sizeof (errbuf))); + return (-1); } if (numcpu != 1) - syslog (LOG_NOTICE, "cpu: Only one processor supported when using `sysctlbyname' (found %i)", numcpu); -#endif - - return; -} - -static void cpu_write (char *host, char *inst, char *val) -{ - char file[512]; - int status; + NOTICE ("cpu: Only one processor supported when using `sysctlbyname' (found %i)", numcpu); +/* #endif HAVE_SYSCTLBYNAME */ - status = snprintf (file, 512, cpu_filename, inst); - if (status < 1) - return; - else if (status >= 512) - return; +#elif defined(HAVE_LIBSTATGRAB) + /* nothing to initialize */ +#endif /* HAVE_LIBSTATGRAB */ - rrd_update_file (host, file, val, ds_def, ds_num); -} + return (0); +} /* int init */ -#if CPU_HAVE_READ -#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) +static void submit (int cpu_num, const char *type_instance, counter_t value) { - char buf[BUFSIZE]; - char cpu[16]; + value_t values[1]; + value_list_t vl = VALUE_LIST_INIT; - 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); + values[0].counter = value; - plugin_submit (MODULE_NAME, cpu, buf); + vl.values = values; + vl.values_len = 1; + sstrncpy (vl.host, hostname_g, sizeof (vl.host)); + sstrncpy (vl.plugin, "cpu", sizeof (vl.plugin)); + ssnprintf (vl.plugin_instance, sizeof (vl.type_instance), + "%i", cpu_num); + sstrncpy (vl.type, "cpu", sizeof (vl.type)); + sstrncpy (vl.type_instance, type_instance, sizeof (vl.type_instance)); + + plugin_dispatch_values (&vl); } -#undef BUFSIZE -static void cpu_read (void) +static int cpu_read (void) { #if PROCESSOR_CPU_LOAD_INFO || PROCESSOR_TEMPERATURE int cpu; @@ -238,21 +256,20 @@ static void cpu_read (void) 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\n", (int) status); + ERROR ("cpu plugin: processor_info failed with status %i", (int) status); continue; } if (cpu_info_len < CPU_STATE_MAX) { - syslog (LOG_ERR, "cpu-plugin: processor_info returned only %i elements..\n", cpu_info_len); + ERROR ("cpu plugin: processor_info returned only %i elements..", cpu_info_len); continue; } - cpu_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); + submit (cpu, "user", (counter_t) cpu_info.cpu_ticks[CPU_STATE_USER]); + submit (cpu, "nice", (counter_t) cpu_info.cpu_ticks[CPU_STATE_NICE]); + submit (cpu, "system", (counter_t) cpu_info.cpu_ticks[CPU_STATE_SYSTEM]); + submit (cpu, "idle", (counter_t) cpu_info.cpu_ticks[CPU_STATE_IDLE]); #endif /* PROCESSOR_CPU_LOAD_INFO */ #if PROCESSOR_TEMPERATURE /* @@ -275,7 +292,7 @@ static void cpu_read (void) cpu_temp, &cpu_temp_len); if (status != KERN_SUCCESS) { - syslog (LOG_ERR, "cpu-plugin: processor_info failed: %s", + ERROR ("cpu plugin: processor_info failed: %s", mach_error_string (status)); cpu_temp_retry_counter = cpu_temp_retry_step; @@ -288,7 +305,7 @@ static void cpu_read (void) if (cpu_temp_len != 1) { - DBG ("processor_info (PROCESSOR_TEMPERATURE) returned %i elements..?", + DEBUG ("processor_info (PROCESSOR_TEMPERATURE) returned %i elements..?", (int) cpu_temp_len); continue; } @@ -296,36 +313,30 @@ static void cpu_read (void) cpu_temp_retry_counter = 0; cpu_temp_retry_step = 1; - DBG ("cpu_temp = %i", (int) cpu_temp); + DEBUG ("cpu_temp = %i", (int) cpu_temp); #endif /* PROCESSOR_TEMPERATURE */ } /* #endif PROCESSOR_CPU_LOAD_INFO */ #elif defined(KERNEL_LINUX) -# define BUFSIZE 1024 int cpu; - unsigned long long user, nice, syst, idle; - unsigned long long wait, intr, sitr; /* sitr == soft interrupt */ + counter_t user, nice, syst, idle; + counter_t 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) { - plugin_complain (LOG_ERR, &complain_obj, "cpu plugin: " - "fopen (/proc/stat) failed: %s", - strerror (errno)); - return; + char errbuf[1024]; + ERROR ("cpu plugin: fopen (/proc/stat) failed: %s", + sstrerror (errno, errbuf, sizeof (errbuf))); + return (-1); } - plugin_relief (LOG_NOTICE, &complain_obj, "cpu plugin: " - "fopen (/proc/stat) succeeded."); - - while (fgets (buf, BUFSIZE, fh) != NULL) + while (fgets (buf, 1024, fh) != NULL) { if (strncmp (buf, "cpu", 3)) continue; @@ -342,85 +353,162 @@ static void cpu_read (void) syst = atoll (fields[3]); idle = atoll (fields[4]); + submit (cpu, "user", user); + submit (cpu, "nice", nice); + submit (cpu, "system", syst); + submit (cpu, "idle", idle); + if (numfields >= 8) { wait = atoll (fields[5]); intr = atoll (fields[6]); sitr = atoll (fields[7]); - /* I doubt anyone cares about the time spent in - * interrupt handlers.. */ - syst += intr + sitr; - } - else - { - wait = 0LL; - } + submit (cpu, "wait", wait); + submit (cpu, "interrupt", intr); + submit (cpu, "softirq", sitr); - cpu_submit (cpu, user, nice, syst, idle, wait); + if (numfields >= 9) + submit (cpu, "steal", atoll (fields[8])); + } } fclose (fh); -#undef BUFSIZE /* #endif defined(KERNEL_LINUX) */ #elif defined(HAVE_LIBKSTAT) int cpu; - unsigned long long user, syst, idle, wait; + counter_t user, syst, idle, wait; static cpu_stat_t cs; if (kc == NULL) - return; + return (-1); for (cpu = 0; cpu < numcpu; cpu++) { if (kstat_read (kc, ksp[cpu], &cs) == -1) continue; /* error message? */ - idle = (unsigned long long) cs.cpu_sysinfo.cpu[CPU_IDLE]; - user = (unsigned long long) cs.cpu_sysinfo.cpu[CPU_USER]; - syst = (unsigned long long) cs.cpu_sysinfo.cpu[CPU_KERNEL]; - wait = (unsigned long long) cs.cpu_sysinfo.cpu[CPU_WAIT]; + idle = (counter_t) cs.cpu_sysinfo.cpu[CPU_IDLE]; + user = (counter_t) cs.cpu_sysinfo.cpu[CPU_USER]; + syst = (counter_t) cs.cpu_sysinfo.cpu[CPU_KERNEL]; + wait = (counter_t) cs.cpu_sysinfo.cpu[CPU_WAIT]; - cpu_submit (ksp[cpu]->ks_instance, - user, 0LL, syst, idle, wait); + submit (ksp[cpu]->ks_instance, "user", user); + submit (ksp[cpu]->ks_instance, "system", syst); + submit (ksp[cpu]->ks_instance, "idle", idle); + submit (ksp[cpu]->ks_instance, "wait", wait); } /* #endif defined(HAVE_LIBKSTAT) */ +#elif CAN_USE_SYSCTL + uint64_t cpuinfo[numcpu][CPUSTATES]; + size_t cpuinfo_size; + int status; + int i; + + if (numcpu < 1) + { + ERROR ("cpu plugin: Could not determine number of " + "installed CPUs using sysctl(3)."); + return (-1); + } + + memset (cpuinfo, 0, sizeof (cpuinfo)); + +#if defined(KERN_CPTIME2) + if (numcpu > 1) { + for (i = 0; i < numcpu; i++) { + int mib[] = {CTL_KERN, KERN_CPTIME2, i}; + + cpuinfo_size = sizeof (cpuinfo[0]); + + status = sysctl (mib, STATIC_ARRAY_SIZE (mib), + cpuinfo[i], &cpuinfo_size, NULL, 0); + if (status == -1) { + char errbuf[1024]; + ERROR ("cpu plugin: sysctl failed: %s.", + sstrerror (errno, errbuf, sizeof (errbuf))); + return (-1); + } + } + } + else +#endif /* defined(KERN_CPTIME2) */ + { + int mib[] = {CTL_KERN, KERN_CPTIME}; + long cpuinfo_tmp[CPUSTATES]; + + cpuinfo_size = sizeof(cpuinfo_tmp); + + status = sysctl (mib, STATIC_ARRAY_SIZE (mib), + &cpuinfo_tmp, &cpuinfo_size, NULL, 0); + if (status == -1) + { + char errbuf[1024]; + ERROR ("cpu plugin: sysctl failed: %s.", + sstrerror (errno, errbuf, sizeof (errbuf))); + return (-1); + } + + for(i = 0; i < CPUSTATES; i++) { + cpuinfo[0][i] = cpuinfo_tmp[i]; + } + } + + for (i = 0; i < numcpu; i++) { + submit (i, "user", cpuinfo[i][CP_USER]); + submit (i, "nice", cpuinfo[i][CP_NICE]); + submit (i, "system", cpuinfo[i][CP_SYS]); + submit (i, "idle", cpuinfo[i][CP_IDLE]); + submit (i, "interrupt", cpuinfo[i][CP_INTR]); + } +/* #endif CAN_USE_SYSCTL */ + #elif defined(HAVE_SYSCTLBYNAME) 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) { - plugin_complain (LOG_ERR, &complain_obj, "cpu plugin: " - "sysctlbyname failed: %s.", - strerror (errno)); - return; + char errbuf[1024]; + ERROR ("cpu plugin: sysctlbyname failed: %s.", + sstrerror (errno, errbuf, sizeof (errbuf))); + return (-1); } - plugin_relief (LOG_NOTICE, &complain_obj, "cpu plugin: " - "sysctlbyname succeeded."); + submit (0, "user", cpuinfo[CP_USER]); + submit (0, "nice", cpuinfo[CP_NICE]); + submit (0, "system", cpuinfo[CP_SYS]); + submit (0, "idle", cpuinfo[CP_IDLE]); + submit (0, "interrupt", cpuinfo[CP_INTR]); +/* #endif HAVE_SYSCTLBYNAME */ - cpuinfo[CP_SYS] += cpuinfo[CP_INTR]; +#elif defined(HAVE_LIBSTATGRAB) + sg_cpu_stats *cs; + cs = sg_get_cpu_stats (); - /* FIXME: Instance is always `0' */ - cpu_submit (0, cpuinfo[CP_USER], cpuinfo[CP_NICE], cpuinfo[CP_SYS], cpuinfo[CP_IDLE], 0LL); -#endif + if (cs == NULL) + { + ERROR ("cpu plugin: sg_get_cpu_stats failed."); + return (-1); + } - return; + submit (0, "idle", (counter_t) cs->idle); + submit (0, "nice", (counter_t) cs->nice); + submit (0, "swap", (counter_t) cs->swap); + submit (0, "system", (counter_t) cs->kernel); + submit (0, "user", (counter_t) cs->user); + submit (0, "wait", (counter_t) cs->iowait); +#endif /* HAVE_LIBSTATGRAB */ + + return (0); } -#else -# define cpu_read NULL -#endif /* CPU_HAVE_READ */ void module_register (void) { - plugin_register (MODULE_NAME, cpu_init, cpu_read, cpu_write); -} - -#undef MODULE_NAME + plugin_register_init ("cpu", init); + plugin_register_read ("cpu", cpu_read); +} /* void module_register */