Add new mbmon plugin

Adapted for mate-sensors-applet by [email protected]

3 files changed, 306 insertions, 0 deletions

diff --git a/plugins/mbmon/Makefile.am b/plugins/mbmon/Makefile.am
new file mode 100644
index 0000000..f391e67
--- /dev/null
+++ b/plugins/mbmon/Makefile.am
@@ -0,0 +1,21 @@
+# MBMON plugin
+plugindir = $(libdir)/mate-sensors-applet/plugins
+
+AM_CPPFLAGS = -DMATELOCALEDIR=\""$(datadir)/locale/"\" \
+	-DG_LOG_DOMAIN=\""Sensors Applet"\" \
+	-DPIXMAPS_DIR=\""$(datadir)/pixmaps/$(PACKAGE)/"\" \
+	-DDATADIR=\""$(datadir)"\" \
+	-DLIBDIR=\""$(libdir)"\" \
+        -DSYSCONFDIR=\""$(sysconfdir)"\" \
+        -DPREFIX=\""$(prefix)"\" \
+	-I$(top_srcdir) \
+	$(GLIB_CFLAGS)
+
+plugin_LTLIBRARIES = libmbmon.la
+
+libmbmon_la_SOURCES = \
+	mbmon-plugin.h	\
+	mbmon-plugin.c
+
+libmbmon_la_LDFLAGS = $(PLUGIN_LIBTOOL_FLAGS) $(GLIB_LIBS)
+libmbmon_la_LIBADD = $(top_builddir)/lib/libmate-sensors-applet-plugin.la
diff --git a/plugins/mbmon/mbmon-plugin.c b/plugins/mbmon/mbmon-plugin.c
new file mode 100644
index 0000000..c9d5fd3
--- /dev/null
+++ b/plugins/mbmon/mbmon-plugin.c
@@ -0,0 +1,261 @@
+/*
+ * Copyright (C) 2010 Daniele Napolitano <[email protected]>
+ *
+ * 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.
+ *
+ * This program 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif /* HAVE_CONFIG_H */
+
+#ifdef HAVE_SYS_TYPES_H
+#include <sys/types.h>
+#endif
+
+#ifdef HAVE_SYS_SOCKET_H
+#include <sys/socket.h>
+#endif
+
+#ifdef HAVE_NETINET_IN_H
+#include <netinet/in.h>
+#endif
+
+#ifdef HAVE_ARPA_INET_H
+#include <arpa/inet.h>
+#endif
+
+#include "mbmon-plugin.h"
+
+const gchar *plugin_name = "mbmon";
+
+#define MBMON_SERVER_IP_ADDRESS "127.0.0.1"
+#define MBMON_PORT_NUMBER 411
+#define MBMON_OUTPUT_BUFFER_LENGTH 1024
+
+enum {
+	MBMON_SOCKET_OPEN_ERROR,
+	MBMON_SOCKET_CONNECT_ERROR,
+	MBMON_GIOCHANNEL_ERROR,
+	MBMON_GIOCHANNEL_READ_ERROR
+
+};
+
+static const gchar *mbmon_plugin_query_mbmon_daemon(GError **error) {
+	int sockfd;
+	ssize_t n = 1;
+	gboolean first_run = FALSE;
+	gint output_length = 0;
+	gchar *pc;
+
+	struct sockaddr_in address;
+	static char* buffer = NULL;
+	static GTimeVal previous_query_time;
+	GTimeVal current_query_time;
+
+	if (NULL == buffer) {
+		// initialise buffer and current time
+		buffer = g_new0(char, MBMON_OUTPUT_BUFFER_LENGTH);
+		g_get_current_time(&previous_query_time);
+		first_run = TRUE;
+	}
+	g_get_current_time(&current_query_time);
+
+	/* only query if more than 2 seconds has elapsed,
+	mbmon daemon will send a new value every 2 seconds */
+	if (first_run || current_query_time.tv_sec - previous_query_time.tv_sec > 2) {
+		previous_query_time = current_query_time;
+
+		if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
+			// couldn't open socket
+			g_set_error(error, SENSORS_APPLET_PLUGIN_ERROR, MBMON_SOCKET_OPEN_ERROR, "Error opening socket for mbmon");
+			return NULL;
+		}
+
+		address.sin_family = AF_INET;
+		address.sin_addr.s_addr = inet_addr(MBMON_SERVER_IP_ADDRESS);
+		address.sin_port = htons(MBMON_PORT_NUMBER);
+
+		if (connect(sockfd, (struct sockaddr *)&address, (socklen_t)sizeof(address)) == -1) {
+			g_set_error(error, SENSORS_APPLET_PLUGIN_ERROR, MBMON_SOCKET_CONNECT_ERROR, "Error connecting to mbmon daemon on port %i on %s", htons(MBMON_PORT_NUMBER), MBMON_SERVER_IP_ADDRESS);
+			return NULL;
+		}
+
+		pc = buffer;
+		while ((n = read(sockfd, pc, MBMON_OUTPUT_BUFFER_LENGTH - output_length)) > 0) {
+			output_length += n;
+			pc = &pc[n];
+		}
+		/* always null terminate the end of the buffer
+		 * regardless of how much stuff is in it already */
+		buffer[output_length] = '\0';
+		close(sockfd);
+	}
+
+	return buffer;
+}
+
+static SensorType get_sensor_type (const char *name) {
+        SensorType type = CURRENT_SENSOR;
+
+	if (g_strrstr(name, "FAN")) {
+        type = FAN_SENSOR;
+	}
+    else if (g_strrstr(name, "TEMP")) {
+        type = TEMP_SENSOR;
+	} else {
+        type = VOLTAGE_SENSOR;
+    }
+        return type;
+}
+
+static IconType get_sensor_icon (SensorType type) {
+	switch (type) {
+	case TEMP_SENSOR:
+		return CPU_ICON;
+	case FAN_SENSOR:
+		return FAN_ICON;
+	default:
+		return GENERIC_ICON;
+	}
+}
+
+static void mbmon_plugin_get_sensors(GList **sensors) {
+
+	GError *error = NULL;
+	const gchar *mbmon_output;
+	gchar **output_vector = NULL, **pv, **pv2;
+
+	mbmon_output = mbmon_plugin_query_mbmon_daemon(&error);
+
+	if (error) {
+		g_error_free(error);
+		return;
+	}
+
+	pv = output_vector = g_strsplit(mbmon_output, "\n", -1);
+
+	while(pv[0] != NULL) {
+		pv2 = g_strsplit(pv[0], ":", -1);
+		gchar *name, *label;
+		SensorType type;
+        gboolean visible;
+        IconType icon;
+        gdouble low_value, high_value;
+
+		type = get_sensor_type(pv2[0]);
+		icon = get_sensor_icon(type);
+		name = g_strstrip(pv2[0]);
+		label = NULL;
+		visible = (type == TEMP_SENSOR ? TRUE : FALSE);
+
+		if(type == VOLTAGE_SENSOR) {
+		    if(g_strrstr(name, "VC0")) {
+		        label = g_strdup("Core Voltage 1");
+		    } else if(g_strrstr(name, "VC1")) {
+		        label = g_strdup("Core Voltage 2");
+		    } else if(g_strrstr(name, "V33")) {
+		        label = g_strdup("+3.3v Voltage");
+		    } else if(g_strrstr(name, "V50P")) {
+		        label = g_strdup("+5v Voltage");
+		    } else if(g_strrstr(name, "V12P")) {
+		        label = g_strdup("+12v Voltage");
+		    } else if(g_strrstr(name, "V12N")) {
+		        label = g_strdup("-12v Voltage");
+		    } else if(g_strrstr(name, "V50N")) {
+		        label = g_strdup("-5v Voltage");
+		    }
+		}
+
+		label = (label == NULL ? name : label);
+
+		sensors_applet_plugin_default_sensor_limits(type, &low_value, &high_value);
+
+		sensors_applet_plugin_add_sensor_with_limits(sensors,
+		                                             name,
+		                                             name,
+                                                     label,
+                                                     type,
+                                                     visible,
+                                                     low_value,
+                                                     high_value,
+                                                     icon,
+                                                     DEFAULT_GRAPH_COLOR);
+
+	    g_strfreev(pv2);
+		pv++;
+	}
+	g_strfreev(output_vector);
+
+}
+
+/* to be called to setup for mbmon sensors */
+static GList *mbmon_plugin_init(void) {
+	GList *sensors = NULL;
+	mbmon_plugin_get_sensors(&sensors);
+        return sensors;
+}
+
+
+/* returns the value of the sensor_list at the given iter, or if an
+   error occurs, instatiates error with an error message */
+static gdouble mbmon_plugin_get_sensor_value(const gchar *path,
+						  const gchar *id,
+						  SensorType type,
+						  GError **error) {
+
+	const gchar *mbmon_output;
+	gchar **output_vector = NULL, **pv, **pv2;
+
+	gfloat sensor_value = -1.0f;
+
+	mbmon_output = mbmon_plugin_query_mbmon_daemon(error);
+
+	if (*error) {
+		return sensor_value;
+	}
+
+	pv = output_vector = g_strsplit(mbmon_output, "\n", -1);
+
+	while(pv[0] != NULL) {
+		if (g_strrstr(pv[0], path) != NULL) {
+
+		    pv2 = g_strsplit(pv[0], ":", -1);
+			sensor_value = (gfloat)(g_ascii_strtod(pv2[1], NULL));
+	        g_strfreev(pv2);
+			break;
+		}
+		pv++;
+	}
+	g_strfreev(output_vector);
+
+	return (gdouble)sensor_value;
+}
+
+const gchar *sensors_applet_plugin_name(void)
+{
+        return plugin_name;
+}
+
+GList *sensors_applet_plugin_init(void)
+{
+        return mbmon_plugin_init();
+}
+
+gdouble sensors_applet_plugin_get_sensor_value(const gchar *path,
+                                                const gchar *id,
+                                                SensorType type,
+                                                GError **error) {
+        return mbmon_plugin_get_sensor_value(path, id, type, error);
+}
diff --git a/plugins/mbmon/mbmon-plugin.h b/plugins/mbmon/mbmon-plugin.h
new file mode 100644
index 0000000..18471b6
--- /dev/null
+++ b/plugins/mbmon/mbmon-plugin.h
@@ -0,0 +1,24 @@
+/*
+ * Copyright (C) 2010 Daniele Napolitano <[email protected]>
+ *
+ * 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.
+ *
+ * This program 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
+#ifndef MBMON_PLUGIN_H
+#define MBMON_PLUGIN_H
+
+#include <sensors-applet/sensors-applet-plugin.h>
+
+#endif /* MBMON_PLUGIN_H */