diff options
Diffstat (limited to 'plugin_proc_stat.c')
-rw-r--r-- | plugin_proc_stat.c | 381 |
1 files changed, 381 insertions, 0 deletions
diff --git a/plugin_proc_stat.c b/plugin_proc_stat.c new file mode 100644 index 0000000..f90b02d --- /dev/null +++ b/plugin_proc_stat.c @@ -0,0 +1,381 @@ +/* $Id: plugin_proc_stat.c 1079 2010-01-15 21:44:04Z volker $ + * $URL: https://ssl.bulix.org/svn/lcd4linux/trunk/plugin_proc_stat.c $ + * + * plugin for /proc/stat parsing + * + * Copyright (C) 2003 Michael Reinelt <michael@reinelt.co.at> + * Copyright (C) 2004 The LCD4Linux Team <lcd4linux-devel@users.sourceforge.net> + * + * This file is part of LCD4Linux. + * + * LCD4Linux 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, or (at your option) + * any later version. + * + * LCD4Linux is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + * + */ + +/* + * exported functions: + * + * int plugin_init_proc_stat (void) + * adds functions to access /proc/stat + * + */ + + +#include "config.h" + +#include <stdlib.h> +#include <stdio.h> +#include <string.h> +#include <ctype.h> +#include <errno.h> + +#ifdef __MAC_OS_X_VERSION_10_3 +#include <mach/mach_host.h> +#include <mach/host_info.h> +#endif + +#include "debug.h" +#include "plugin.h" +#include "qprintf.h" +#include "hash.h" + + +static HASH Stat; +static FILE *stream = NULL; + + +static void hash_put1(const char *key1, const char *val) +{ + hash_put_delta(&Stat, key1, val); +} + + +static void hash_put2(const char *key1, const char *key2, const char *val) +{ + char key[32]; + + qprintf(key, sizeof(key), "%s.%s", key1, key2); + hash_put1(key, val); +} + + +static void hash_put3(const char *key1, const char *key2, const char *key3, const char *val) +{ + char key[32]; + + qprintf(key, sizeof(key), "%s.%s.%s", key1, key2, key3); + hash_put1(key, val); +} + + +static int parse_proc_stat(void) +{ + int age; + + /* reread every 10 msec only */ + age = hash_age(&Stat, NULL); + if (age > 0 && age <= 10) + return 0; + +#ifndef __MAC_OS_X_VERSION_10_3 + + /* Linux Kernel, /proc-filesystem */ + + if (stream == NULL) + stream = fopen("/proc/stat", "r"); + if (stream == NULL) { + error("fopen(/proc/stat) failed: %s", strerror(errno)); + return -1; + } + + rewind(stream); + + while (!feof(stream)) { + char buffer[1024]; + if (fgets(buffer, sizeof(buffer), stream) == NULL) + break; + + if (strncmp(buffer, "cpu", 3) == 0) { + char *key[] = { "user", "nice", "system", "idle", "iow", "irq", "sirq" }; + char delim[] = " \t\n"; + char *cpu, *beg, *end; + int i; + + cpu = buffer; + + /* skip "cpu" or "cpu0" block */ + if ((end = strpbrk(buffer, delim)) != NULL) + *end = '\0'; + beg = end ? end + 1 : NULL; + + for (i = 0; i < 7 && beg != NULL; i++) { + while (strchr(delim, *beg)) + beg++; + if ((end = strpbrk(beg, delim))) + *end = '\0'; + hash_put2(cpu, key[i], beg); + beg = end ? end + 1 : NULL; + } + } + + else if (strncmp(buffer, "page ", 5) == 0) { + char *key[] = { "in", "out" }; + char delim[] = " \t\n"; + char *beg, *end; + int i; + + for (i = 0, beg = buffer + 5; i < 2 && beg != NULL; i++) { + while (strchr(delim, *beg)) + beg++; + if ((end = strpbrk(beg, delim))) + *end = '\0'; + hash_put2("page", key[i], beg); + beg = end ? end + 1 : NULL; + } + } + + else if (strncmp(buffer, "swap ", 5) == 0) { + char *key[] = { "in", "out" }; + char delim[] = " \t\n"; + char *beg, *end; + int i; + + for (i = 0, beg = buffer + 5; i < 2 && beg != NULL; i++) { + while (strchr(delim, *beg)) + beg++; + if ((end = strpbrk(beg, delim))) + *end = '\0'; + hash_put2("swap", key[i], beg); + beg = end ? end + 1 : NULL; + } + } + + else if (strncmp(buffer, "intr ", 5) == 0) { + char delim[] = " \t\n"; + char *beg, *end, num[4]; + int i; + + for (i = 0, beg = buffer + 5; i < 17 && beg != NULL; i++) { + while (strchr(delim, *beg)) + beg++; + if ((end = strpbrk(beg, delim))) + *end = '\0'; + if (i == 0) + strcpy(num, "sum"); + else + qprintf(num, sizeof(num), "%d", i - 1); + hash_put2("intr", num, beg); + beg = end ? end + 1 : NULL; + } + } + + else if (strncmp(buffer, "disk_io:", 8) == 0) { + char *key[] = { "io", "rio", "rblk", "wio", "wblk" }; + char delim[] = " ():,\t\n"; + char *dev, *beg, *end, *p; + int i; + + dev = buffer + 8; + while (dev != NULL) { + while (strchr(delim, *dev)) + dev++; + if ((end = strchr(dev, ')'))) + *end = '\0'; + while ((p = strchr(dev, ',')) != NULL) + *p = ':'; + beg = end ? end + 1 : NULL; + for (i = 0; i < 5 && beg != NULL; i++) { + while (strchr(delim, *beg)) + beg++; + if ((end = strpbrk(beg, delim))) + *end = '\0'; + hash_put3("disk_io", dev, key[i], beg); + beg = end ? end + 1 : NULL; + } + dev = beg; + } + } + + else { + char delim[] = " \t\n"; + char *beg, *end; + + beg = buffer; + if ((end = strpbrk(beg, delim))) + *end = '\0'; + beg = end ? end + 1 : NULL; + if ((end = strpbrk(beg, delim))) + *end = '\0'; + while (strchr(delim, *beg)) + beg++; + hash_put1(buffer, beg); + } + } + +#else + + /* MACH Kernel, MacOS X */ + + kern_return_t err; + mach_msg_type_number_t count; + host_info_t r_load; + host_cpu_load_info_data_t cpu_load; + char s_val[8]; + + r_load = &cpu_load; + count = HOST_CPU_LOAD_INFO_COUNT; + err = host_statistics(mach_host_self(), HOST_CPU_LOAD_INFO, r_load, &count); + if (KERN_SUCCESS != err) { + error("Error getting cpu load"); + return -1; + } + snprintf(s_val, sizeof(s_val), "%d", cpu_load.cpu_ticks[CPU_STATE_USER]); + hash_put2("cpu", "user", s_val); + snprintf(s_val, sizeof(s_val), "%d", cpu_load.cpu_ticks[CPU_STATE_NICE]); + hash_put2("cpu", "nice", s_val); + snprintf(s_val, sizeof(s_val), "%d", cpu_load.cpu_ticks[CPU_STATE_SYSTEM]); + hash_put2("cpu", "system", s_val); + snprintf(s_val, sizeof(s_val), "%d", cpu_load.cpu_ticks[CPU_STATE_IDLE]); + hash_put2("cpu", "idle", s_val); + +#endif + + return 0; +} + + +static void my_proc_stat(RESULT * result, const int argc, RESULT * argv[]) +{ + char *string; + double number; + + if (parse_proc_stat() < 0) { + SetResult(&result, R_STRING, ""); + return; + } + + switch (argc) { + case 1: + string = hash_get(&Stat, R2S(argv[0]), NULL); + if (string == NULL) + string = ""; + SetResult(&result, R_STRING, string); + break; + case 2: + number = hash_get_delta(&Stat, R2S(argv[0]), NULL, R2N(argv[1])); + SetResult(&result, R_NUMBER, &number); + break; + default: + error("proc_stat(): wrong number of parameters"); + SetResult(&result, R_STRING, ""); + } +} + + +static void my_cpu(RESULT * result, RESULT * arg1, RESULT * arg2) +{ + char *key; + int delay; + double value; + double cpu_user, cpu_nice, cpu_system, cpu_idle, cpu_total; + double cpu_iow, cpu_irq, cpu_sirq; + + if (parse_proc_stat() < 0) { + SetResult(&result, R_STRING, ""); + return; + } + + key = R2S(arg1); + delay = R2N(arg2); + + cpu_user = hash_get_delta(&Stat, "cpu.user", NULL, delay); + cpu_nice = hash_get_delta(&Stat, "cpu.nice", NULL, delay); + cpu_system = hash_get_delta(&Stat, "cpu.system", NULL, delay); + cpu_idle = hash_get_delta(&Stat, "cpu.idle", NULL, delay); + + /* new fields for kernel 2.6 */ + /* even if we dont have this param (ie kernel 2.4) */ + /* the return is 0.0 and not change the results */ + cpu_iow = hash_get_delta(&Stat, "cpu.iow", NULL, delay); + cpu_irq = hash_get_delta(&Stat, "cpu.irq", NULL, delay); + cpu_sirq = hash_get_delta(&Stat, "cpu.sirq", NULL, delay); + + cpu_total = cpu_user + cpu_nice + cpu_system + cpu_idle + cpu_iow + cpu_irq + cpu_sirq; + + if (strcasecmp(key, "user") == 0) + value = cpu_user; + else if (strcasecmp(key, "nice") == 0) + value = cpu_nice; + else if (strcasecmp(key, "system") == 0) + value = cpu_system; + else if (strcasecmp(key, "idle") == 0) + value = cpu_idle; + else if (strcasecmp(key, "iowait") == 0) + value = cpu_iow; + else if (strcasecmp(key, "irq") == 0) + value = cpu_irq; + else if (strcasecmp(key, "softirq") == 0) + value = cpu_sirq; + else if (strcasecmp(key, "busy") == 0) + value = cpu_total - cpu_idle; + + if (cpu_total > 0.0) + value = 100 * value / cpu_total; + else + value = 0.0; + + SetResult(&result, R_NUMBER, &value); +} + + +static void my_disk(RESULT * result, RESULT * arg1, RESULT * arg2, RESULT * arg3) +{ + char *dev, *key, buffer[32]; + int delay; + double value; + + if (parse_proc_stat() < 0) { + SetResult(&result, R_STRING, ""); + return; + } + + dev = R2S(arg1); + key = R2S(arg2); + delay = R2N(arg3); + + qprintf(buffer, sizeof(buffer), "disk_io\\.%s\\.%s", dev, key); + value = hash_get_regex(&Stat, buffer, NULL, delay); + + SetResult(&result, R_NUMBER, &value); +} + + +int plugin_init_proc_stat(void) +{ + hash_create(&Stat); + AddFunction("proc_stat", -1, my_proc_stat); + AddFunction("proc_stat::cpu", 2, my_cpu); + AddFunction("proc_stat::disk", 3, my_disk); + return 0; +} + +void plugin_exit_proc_stat(void) +{ + if (stream != NULL) { + fclose(stream); + stream = NULL; + } + hash_destroy(&Stat); +} |