From 181cec4348da40331b3e8ab365732c025ec149b2 Mon Sep 17 00:00:00 2001 From: Reinhard Tartler Date: Wed, 27 Apr 2011 19:24:15 +0200 Subject: Import upstream version 0.11.0~svn1143 --- plugin_diskstats.c | 155 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 155 insertions(+) create mode 100644 plugin_diskstats.c (limited to 'plugin_diskstats.c') diff --git a/plugin_diskstats.c b/plugin_diskstats.c new file mode 100644 index 0000000..bfaaa0e --- /dev/null +++ b/plugin_diskstats.c @@ -0,0 +1,155 @@ +/* $Id: plugin_diskstats.c 840 2007-09-09 12:17:42Z michael $ + * $URL: https://ssl.bulix.org/svn/lcd4linux/trunk/plugin_diskstats.c $ + * + * plugin for /proc/diskstats parsing + * + * Copyright (C) 2003 Michael Reinelt + * Copyright (C) 2004 The LCD4Linux Team + * + * 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_diskstats (void) + * adds functions to access /proc/stat + * + */ + + +#include "config.h" + +#include +#include +#include +#include +#include + +#include "debug.h" +#include "plugin.h" +#include "hash.h" + + +static HASH DISKSTATS; +static FILE *stream = NULL; + + +static int parse_diskstats(void) +{ + int age; + + /* reread every 10 msec only */ + age = hash_age(&DISKSTATS, NULL); + if (age > 0 && age <= 10) + return 0; + + if (stream == NULL) + stream = fopen("/proc/diskstats", "r"); + if (stream == NULL) { + error("fopen(/proc/diskstats) failed: %s", strerror(errno)); + return -1; + } + + rewind(stream); + + while (!feof(stream)) { + char buffer[1024]; + char dev[64]; + char *beg, *end; + unsigned int num, len; + + if (fgets(buffer, sizeof(buffer), stream) == NULL) + break; + + /* fetch device name (3rd column) as key */ + num = 0; + beg = buffer; + end = beg; + while (*beg) { + while (*beg == ' ') + beg++; + end = beg + 1; + while (*end && *end != ' ') + end++; + if (num++ == 2) + break; + beg = end ? end + 1 : NULL; + } + len = end ? (unsigned) (end - beg) : strlen(beg); + + if (len >= sizeof(dev)) + len = sizeof(dev) - 1; + strncpy(dev, beg, len); + dev[len] = '\0'; + + hash_put_delta(&DISKSTATS, dev, buffer); + + } + return 0; +} + + +static void my_diskstats(RESULT * result, RESULT * arg1, RESULT * arg2, RESULT * arg3) +{ + char *dev, *key; + int delay; + double value; + + if (parse_diskstats() < 0) { + SetResult(&result, R_STRING, ""); + return; + } + + dev = R2S(arg1); + key = R2S(arg2); + delay = R2N(arg3); + + value = hash_get_regex(&DISKSTATS, dev, key, delay); + + SetResult(&result, R_NUMBER, &value); +} + + +int plugin_init_diskstats(void) +{ + int i; + char *header[] = { "major", "minor", "name", + "reads", "read_merges", "read_sectors", "read_ticks", + "writes", "write_merges", "write_sectors", "write_ticks", + "in_flight", "io_ticks", "time_in_queue", "" + }; + + hash_create(&DISKSTATS); + hash_set_delimiter(&DISKSTATS, " \n"); + for (i = 0; *header[i] != '\0'; i++) { + hash_set_column(&DISKSTATS, i, header[i]); + } + + AddFunction("diskstats", 3, my_diskstats); + return 0; +} + +void plugin_exit_diskstats(void) +{ + if (stream != NULL) { + fclose(stream); + stream = NULL; + } + hash_destroy(&DISKSTATS); +} -- cgit v1.2.3