hddtemp plugin from Scott Bronson

git-svn-id: https://ssl.bulix.org/svn/lcd4linux/trunk@907 3ae390bd-cb1e-0410-b409-cd5a39f66f1f

1 files changed, 312 insertions, 0 deletions

diff --git a/plugin_hddtemp.c b/plugin_hddtemp.c
new file mode 100644
index 0000000..c8d742b
--- /dev/null
+++ b/plugin_hddtemp.c
@@ -0,0 +1,312 @@
+/* $Id$
+ * $URL$
+ *
+ * plugin hddtemp
+ *
+ * Copyright (C) 2007 Scott Bronson <brons_lcd4linux@rinspin.com>
+ * Copyright (C) 2004, 2005, 2006, 2007, 2008 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.
+ *
+ */
+
+/* Example:
+ * (I tried to put this info on the wiki but Akismet claimed it was spam)
+
+	Widget SmallHDDTemp {
+		class 'Text'
+		expression hddtemp('/dev/hda')
+		width 4
+		precision 1
+		align 'R'
+		update tick
+	}
+
+	hddtemp home page: http://www.guzu.net/linux/hddtemp.php
+
+	Quick Examples:
+	 * hddtemp() -- return temperature of first drive (order is defined by the hddtemp daemon).
+	 * hddtemp('/dev/hda') -- return temperature of /dev/hda on localhost.
+	 * hddtemp('ST3400832A') -- return temperature of the drive with the given ID.
+	 * hddtemp('burly', '') -- return temperature of the first drive on host burly.
+	 * hddtemp('burly', 17634, '/dev/hda') -- return temperature of /dev/hda on burly connecting to the hddtemp daemon listening on port 17634.
+
+You can find out what drives are being monitored by running
+		telnet HOST PORT
+
+i.e.
+
+	$ telnet localhost 7634
+	Trying 127.0.0.1...
+	Connected to localhost.
+	Escape character is '^]'.
+	|/dev/hda|ST3400832A|53|C|Connection closed by foreign host.
+
+This tells me that /dev/hda is a Seagate ST3400832A running at an awfully
+toasty 53 degC.
+
+ */
+
+
+/* 
+ * exported functions:
+ *
+ * int plugin_init_hddtemp (void)
+ *  adds various functions
+ * void plugin_exit_hddtemp (void)
+ *
+ */
+
+#include "config.h"
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
+#include <errno.h>
+
+/* network specific includes */
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <netdb.h>
+#include <arpa/inet.h>
+
+/* these should always be included */
+#include "debug.h"
+#include "plugin.h"
+#include "cfg.h"
+#include "thread.h"
+
+
+static int socket_open(const char *name, int port)
+{
+    struct sockaddr_in server;
+    struct hostent *host_info;
+    unsigned long addr;
+    int sock;
+
+    sock = socket(PF_INET, SOCK_STREAM, 0);
+    if (sock < 0) {
+	error("[hddtemp] failed to create socket: %s", strerror(errno));
+	return -1;
+    }
+
+    memset(&server, 0, sizeof(server));
+    if ((addr = inet_addr(name)) != INADDR_NONE) {
+	memcpy((char *) &server.sin_addr, &addr, sizeof(addr));
+    } else {
+	host_info = gethostbyname(name);
+	if (NULL == host_info) {
+	    error("[hddtemp] Unknown server: %s", name);
+	    return -1;
+	}
+	memcpy((char *) &server.sin_addr, host_info->h_addr, host_info->h_length);
+    }
+
+    server.sin_family = AF_INET;
+    server.sin_port = htons(port);
+
+    if (connect(sock, (struct sockaddr *) &server, sizeof(server)) < 0) {
+	error("[hddtemp] can't connect to server %s: %s", name, strerror(errno));
+	return -1;
+    }
+
+    return sock;
+}
+
+
+// replaces all null chars in a buffer with a given character.
+static void replace_nulls(char *buf, size_t len, char repl)
+{
+    char *end = buf + len;
+
+    while (buf < end) {
+	if (*buf == '\0') {
+	    *buf = repl;
+	}
+
+	buf += 1;
+    }
+}
+
+
+static size_t hddtemp_read(int socket, char *buf, size_t bufsiz)
+{
+    size_t count = 0;
+    ssize_t len;
+
+    while (count < bufsiz) {
+	len = read(socket, buf + count, bufsiz - count);
+	if (len > 0) {
+	    replace_nulls(buf + count, len, '@');
+	    count += len;
+	} else if (len < 0) {
+	    // an error!
+	    error("[hddtemp] Couldn't read: %s", strerror(errno));
+	    return -1;
+	} else {
+	    // remote socket has closed
+	    break;
+	}
+    }
+
+    // null-terminate the string
+    if (count < bufsiz) {
+	buf[count] = '\0';
+    } else {
+	buf[bufsiz - 1] = '\0';
+    }
+
+    return count;
+}
+
+
+static int hddtemp_connect(const char *host, int port, char *buf, size_t bufsiz)
+{
+    int sock, ret;
+
+    sock = socket_open(host, port);
+    if (sock < 0) {
+	error("[hddtemp] Error accessing %s:%d: %s", host, port, strerror(errno));
+	return -1;
+    }
+
+    ret = hddtemp_read(sock, buf, bufsiz);
+    close(sock);
+
+    return ret;
+}
+
+
+static char *hddtemp_fetch(char *buf, size_t bufsiz, const char *host, int port, const char *arg)
+{
+    char *line, *end;
+
+    // fetch a buffer of all hddtemps
+    if (!hddtemp_connect(host, port, buf, bufsiz)) {
+	return "err";
+    }
+    // find the line containing the user's specification
+    line = strstr(buf, arg);
+    if (!line) {
+	return "--";
+    }
+    // back up to the beginning of the line
+    while (line > buf && line[-1] != '\n') {
+	line--;
+    }
+
+    // quick sanity check -- each line begins with the separator
+    if (*line != '|') {
+	return "fmt1";
+    }
+    // |device|identifier|temp|units|
+    // skip 3 bars to find the start of the temperature
+    line = strchr(line + 1, '|');
+    if (line && line[0])
+	line = strchr(line + 1, '|');
+    if (!line || !line[0]) {
+	return "fmt2";
+    }
+    // set the next bar to '\0' to null-terminate temp
+    line += 1;
+    end = strchr(line, '|');
+    if (!end) {
+	return "fmt3";
+    }
+    *end = '\0';
+
+    return line;
+}
+
+
+#ifndef TEST
+
+static void hddtemp_call(RESULT * result, int argc, RESULT * argv[])
+{
+    char buf[4096];
+    const char *search = "";
+    const char *host = "localhost";
+    int port = 7634;
+    char *out;
+
+    switch (argc) {
+    case 0:
+	break;
+    case 1:
+	search = R2S(argv[0]);
+	break;
+    case 2:
+	host = R2S(argv[0]);
+	search = R2S(argv[1]);
+    default:
+	if (argv > 3) {
+	    error("hddtemp(): too many parameters");
+	}
+	host = R2S(argv[0]);
+	port = (int) R2N(argv[1]);
+	search = R2S(argv[2]);
+    }
+
+    out = hddtemp_fetch(buf, sizeof(buf), host, port, search);
+    SetResult(&result, R_STRING, out);
+}
+
+
+int plugin_init_hddtemp(void)
+{
+    info("Adding hddtemp functions!");
+
+    AddFunction("hddtemp", -1, hddtemp_call);
+
+    return 0;
+}
+
+
+void plugin_exit_hddtemp(void)
+{
+}
+#endif
+
+
+
+#ifdef TEST
+// Add this to your makefile to make the test_hddtemp app:
+// test_hddtemp: plugin_hddtemp.c
+//      gcc -Wall -Werror -g -DTEST plugin_hddtemp.c -o $@
+
+#include <stdarg.h>
+
+void message(const int level, const char *format, ...)
+{
+    va_list ap;
+
+    va_start(ap, format);
+    vprintf(format, ap);
+    putchar('\n');
+    va_end(ap);
+}
+
+int main(int argc, char **argv)
+{
+    char buf[4096];
+    char *out = hddtemp_fetch(buf, sizeof(buf), argv[1] ? argv[1] : "/dev/hda");
+    printf("OUTPUT: %s\n", out);
+    return 0;
+}
+#endif