moved from Mate-Extra

1 files changed, 282 insertions, 0 deletions

diff --git a/src/gpm-load.c b/src/gpm-load.c
new file mode 100644
index 0000000..eab0129
--- /dev/null
+++ b/src/gpm-load.c
@@ -0,0 +1,282 @@
+/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*-
+ *
+ * Copyright (C) 2007 Richard Hughes <[email protected]>
+ *
+ * Licensed under the GNU General Public License Version 2
+ *
+ * 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+#include "config.h"
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <time.h>
+#include <errno.h>
+
+#include <string.h>
+#include <sys/time.h>
+#include <sys/types.h>
+#if defined(sun) && defined(__SVR4)
+#include <kstat.h>
+#include <sys/sysinfo.h>
+#endif
+#ifdef HAVE_UNISTD_H
+#include <unistd.h>
+#endif /* HAVE_UNISTD_H */
+
+#include <glib/gi18n.h>
+
+#include "gpm-common.h"
+#include "gpm-marshal.h"
+#include "egg-debug.h"
+
+#include "gpm-load.h"
+
+static void     gpm_load_finalize   (GObject	  *object);
+
+#define GPM_LOAD_GET_PRIVATE(o) (G_TYPE_INSTANCE_GET_PRIVATE ((o), GPM_TYPE_LOAD, GpmLoadPrivate))
+
+struct GpmLoadPrivate
+{
+	long unsigned	 old_idle;
+	long unsigned	 old_total;
+};
+
+static gpointer gpm_load_object = NULL;
+
+G_DEFINE_TYPE (GpmLoad, gpm_load, G_TYPE_OBJECT)
+
+/**
+ * gpm_load_class_init:
+ * @klass: This class instance
+ **/
+static void
+gpm_load_class_init (GpmLoadClass *klass)
+{
+	GObjectClass *object_class = G_OBJECT_CLASS (klass);
+	object_class->finalize = gpm_load_finalize;
+	g_type_class_add_private (klass, sizeof (GpmLoadPrivate));
+}
+
+#if defined(sun) && defined(__SVR4)
+
+/**
+ * gpm_load_get_cpu_values:
+ * @cpu_idle: The idle time reported by the CPU
+ * @cpu_total: The total time reported by the CPU
+ * Return value: Success of reading /proc/stat.
+ **/
+static gboolean
+gpm_load_get_cpu_values (long unsigned *cpu_idle, long unsigned *cpu_total)
+{	
+	long unsigned cpu_user = 0;
+	long unsigned cpu_kernel = 0;
+	long unsigned cpu_wait = 0;
+	kstat_ctl_t *kc = NULL;
+	kstat_named_t *kn = NULL;
+	kstat_t *ks = NULL;
+	cpu_stat_t  data;
+	int ncpus;
+	int count;
+	   	
+	kc = kstat_open();
+	if (!kc) {
+		egg_warning ("Cannot open kstat!\n");
+		return FALSE;
+	}
+
+	ks = kstat_lookup(kc, "unix", 0, "system_misc");
+  	if (kstat_read(kc, ks, NULL) == -1) {
+		egg_warning ("Cannot read kstat on module unix!\n");
+		goto out;
+	}
+	kn = kstat_data_lookup (ks, "ncpus");
+	if (!kn) {
+		egg_warning ("Cannot get number of cpus in current system!\n");
+		goto out;
+	}
+	ncpus = kn->value.ui32;
+
+	/* 
+ 	 * To aggresive ticks used of all cpus,
+	 * traverse kstat chain to access very cpu_stat instane.
+	 */
+	for(count = 0, *cpu_idle =0, *cpu_total = 0; count < ncpus; count++){
+		
+		ks = kstat_lookup(kc, "cpu_stat", count, NULL);
+		if (ks == NULL) {
+			egg_warning ("Null output for kstat on cpu%d\n", count);
+			goto out;
+		}
+     
+		if (kstat_read(kc, ks, &data) == -1) {
+			egg_warning ("Cannot read kstat entry on cpu%d\n", count);
+			goto out;
+		}
+
+		egg_debug ("cpu%d:\t%lu\t%lu\t%lu\t%lu\n", count,
+					data.cpu_sysinfo.cpu[CPU_IDLE],
+					data.cpu_sysinfo.cpu[CPU_USER],
+					data.cpu_sysinfo.cpu[CPU_KERNEL],
+					data.cpu_sysinfo.cpu[CPU_WAIT]);
+
+		*cpu_idle += data.cpu_sysinfo.cpu[CPU_IDLE];
+		cpu_user += data.cpu_sysinfo.cpu[CPU_USER];
+		cpu_kernel += data.cpu_sysinfo.cpu[CPU_KERNEL];
+		cpu_wait += data.cpu_sysinfo.cpu[CPU_WAIT];
+	}
+	kstat_close(kc);
+	/*
+         * Summing up all these times gives you the system uptime.
+         * This is what the uptime command does.
+         */
+        *cpu_total = cpu_user + cpu_kernel + cpu_wait + *cpu_idle;
+        return TRUE;
+
+out:
+	kstat_close(kc);
+	return FALSE;
+}
+
+#else
+
+/**
+ * gpm_load_get_cpu_values:
+ * @cpu_idle: The idle time reported by the CPU
+ * @cpu_total: The total time reported by the CPU
+ * Return value: Success of reading /proc/stat.
+ **/
+static gboolean
+gpm_load_get_cpu_values (long unsigned *cpu_idle, long unsigned *cpu_total)
+{
+	long unsigned cpu_user;
+	long unsigned cpu_nice;
+	long unsigned cpu_system;
+	int len;
+	char tmp[5];
+	char str[80];
+	FILE *fd;
+	char *suc;
+	gboolean ret = FALSE;
+
+	/* open file */
+	fd = fopen("/proc/stat", "r");
+	if (!fd)
+		goto out;
+
+	/* get data */
+	suc = fgets (str, 80, fd);
+	if (suc == NULL)
+		goto out;
+
+	/* parse */
+	len = sscanf (str, "%s %lu %lu %lu %lu", tmp,
+		      &cpu_user, &cpu_nice, &cpu_system, cpu_idle);
+	if (len != 5)
+		goto out;
+
+	/* summing up all these times gives you the system uptime in jiffies */
+	*cpu_total = cpu_user + cpu_nice + cpu_system + *cpu_idle;
+	ret = TRUE;
+out:
+	if (!fd)
+		fclose (fd);
+	return ret;
+}
+#endif /* sun & __SVR4 */
+
+/**
+ * gpm_load_get_current:
+ * @load: This class instance
+ * Return value: The CPU idle load
+ **/
+gdouble
+gpm_load_get_current (GpmLoad *load)
+{
+	double percentage_load;
+	long unsigned cpu_idle;
+	long unsigned cpu_total;
+	long unsigned diff_idle;
+	long unsigned diff_total;
+	gboolean ret;
+
+	/* work out the differences */
+	ret = gpm_load_get_cpu_values (&cpu_idle, &cpu_total);
+	if (!ret)
+		return 0.0;
+
+	diff_idle = cpu_idle - load->priv->old_idle;
+	diff_total = cpu_total - load->priv->old_total;
+
+	/* If we divide the total time by idle time we get the load. */
+	if (diff_idle > 0)
+		percentage_load = (double) diff_total / (double) diff_idle;
+	else
+		percentage_load = 100;
+
+	load->priv->old_idle = cpu_idle;
+	load->priv->old_total = cpu_total;
+
+	return percentage_load;
+}
+
+/**
+ * gpm_load_init:
+ */
+static void
+gpm_load_init (GpmLoad *load)
+{
+	load->priv = GPM_LOAD_GET_PRIVATE (load);
+
+	load->priv->old_idle = 0;
+	load->priv->old_total = 0;
+
+	/* we have to populate the values at startup */
+	gpm_load_get_cpu_values (&load->priv->old_idle, &load->priv->old_total);
+}
+
+/**
+ * gpm_load_coldplug:
+ *
+ * @object: This load instance
+ */
+static void
+gpm_load_finalize (GObject *object)
+{
+	GpmLoad *load;
+	g_return_if_fail (object != NULL);
+	g_return_if_fail (GPM_IS_LOAD (object));
+	load = GPM_LOAD (object);
+	g_return_if_fail (load->priv != NULL);
+	G_OBJECT_CLASS (gpm_load_parent_class)->finalize (object);
+}
+
+/**
+ * gpm_load_new:
+ * Return value: new GpmLoad instance.
+ **/
+GpmLoad *
+gpm_load_new (void)
+{
+	if (gpm_load_object != NULL) {
+		g_object_ref (gpm_load_object);
+	} else {
+		gpm_load_object = g_object_new (GPM_TYPE_LOAD, NULL);
+		g_object_add_weak_pointer (gpm_load_object, &gpm_load_object);
+	}
+	return GPM_LOAD (gpm_load_object);
+}
+