/* Copyright (C) 1998 Joshua Sled

   This file is part of LibGTop 1.0.

   Contributed by Joshua Sled <jsled@xcf.berkeley.edu>, July 1998.

   LibGTop 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.

   LibGTop 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 LibGTop; see the file COPYING. If not, write to the

   Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,

   Boston, MA 02110-1301, USA.

*/

#include <config.h>

#include <glibtop.h>

#include <glibtop/error.h>

#include <glibtop/cpu.h>

#include <glibtop_suid.h>

static const unsigned long _glibtop_sysdeps_cpu =

(1L << GLIBTOP_CPU_TOTAL) + (1L << GLIBTOP_CPU_USER) +

(1L << GLIBTOP_CPU_NICE) + (1L << GLIBTOP_CPU_SYS) +

(1L << GLIBTOP_CPU_IDLE) + (1L << GLIBTOP_CPU_FREQUENCY) +

(1L << GLIBTOP_CPU_IRQ);

static const unsigned long _glibtop_sysdeps_cpu_smp =

(1L << GLIBTOP_XCPU_TOTAL) + (1L << GLIBTOP_XCPU_USER) +

(1L << GLIBTOP_XCPU_NICE) + (1L << GLIBTOP_XCPU_SYS) +

(1L << GLIBTOP_XCPU_IDLE) + (1L << GLIBTOP_XCPU_IRQ);

/* MIB array for sysctl */

static int mib_length=2;

static int mib_cr [] = { CTL_KERN, KERN_CLOCKRATE };

static int mib_cptime [] = { CTL_KERN, KERN_CPTIME };

static int mib_cptime_s [] = { CTL_KERN, KERN_CPTIME2, 0 };

/* Init function. */

void

_glibtop_init_cpu_s (glibtop *server)

{

	server->sysdeps.cpu = _glibtop_sysdeps_cpu;

	if (server->ncpu)

		server->sysdeps.cpu |= _glibtop_sysdeps_cpu_smp;

}

/* Provides information about cpu usage. */

void

glibtop_get_cpu_s (glibtop *server, glibtop_cpu *buf)

{

	long cpts [CPUSTATES];

	int64_t *cp_times = NULL;

	struct clockinfo ci;

	size_t length;

	int ncpu, i;

	glibtop_init_s (&server, GLIBTOP_SYSDEPS_CPU, 0);

	memset (buf, 0, sizeof (glibtop_cpu));

	/* If this fails, the nlist may not be valid. */

	if (server->sysdeps.cpu == 0)

		return;

	length = sizeof (cpts);

	if (sysctl (mib_cptime, mib_length, cpts, &length, NULL, 0)) {

		glibtop_warn_io_r (server, "sysctl (kern.cptime)");

		return;

	}

	/* Get the clockrate data */

	length = sizeof (struct clockinfo);

	if (sysctl (mib_cr, mib_length, &ci, &length, NULL, 0)) {

		glibtop_warn_io_r (server, "sysctl (kern.clockrate)");

		return;

	}

	/* set user time */

	buf->user = cpts [CP_USER];

	/* set nice time */

	buf->nice = cpts [CP_NICE];

	/* set sys time */

	buf->sys = cpts [CP_SYS];

	/* set idle time */

	buf->idle = cpts [CP_IDLE];

	/* set irq */

	buf->irq = cpts [CP_INTR];

	/* set frequency */

	buf->frequency = (ci.stathz ? ci.stathz : ci.hz);

	/* set total */

	buf->total = cpts [CP_USER] + cpts [CP_NICE] \

		+ cpts [CP_SYS] + cpts [CP_IDLE] + cpts [CP_INTR];

	ncpu = server->ncpu + 1;

	/*

	if (!cp_times) {

		if ((cp_times = calloc(sizeof(int64_t *), ncpu)) == NULL)

			return;

	}

	*/

	length = CPUSTATES * sizeof(int64_t);

	for (i = 0; i < ncpu; i++) {

		mib_cptime_s[2] = i;

		cp_times = g_malloc (length);

		if (sysctl (mib_cptime_s, 3, cp_times, &length, NULL, 0) < 0)

			free(cp_times);

		if (cp_times) {

			buf->xcpu_user[i] = cp_times[CP_USER];

			buf->xcpu_nice[i] = cp_times[CP_NICE];

			buf->xcpu_sys[i] = cp_times[CP_SYS];

			buf->xcpu_idle[i] = cp_times[CP_IDLE];

			buf->xcpu_irq[i] = cp_times[CP_INTR];

		} else {

			buf->xcpu_user[i] = cpts [CP_USER] / ncpu;

			buf->xcpu_nice[i] = cpts [CP_NICE] / ncpu;

			buf->xcpu_sys[i] = cpts [CP_SYS] / ncpu;

			buf->xcpu_idle[i] = cpts [CP_IDLE] / ncpu;

			buf->xcpu_irq[i] = cpts [CP_INTR] / ncpu;

		}

		buf->xcpu_total[i] = buf->xcpu_user[i] + buf->xcpu_nice[i] \

				     + buf->xcpu_sys[i] + buf->xcpu_idle[i] \

				     + buf->xcpu_irq[i];

	}

	g_free (cp_times);

	/* Set the flags last. */

	buf->flags = _glibtop_sysdeps_cpu;

	if (ncpu > 1) {

		buf->flags |= _glibtop_sysdeps_cpu_smp;

	}

}