diff options
Diffstat (limited to 'plugins/mbmon/mbmon-plugin.c')
| -rw-r--r-- | plugins/mbmon/mbmon-plugin.c | 261 |
1 files changed, 261 insertions, 0 deletions
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(¤t_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); +} |