/* $Id$ */

/* Copyright (C) 1998-99 Martin Baulig
   This file is part of LibGTop 1.0.

   Contributed by Martin Baulig <martin@home-of-linux.org>, April 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., 59 Temple Place - Suite 330,
   Boston, MA 02111-1307, USA.
*/

#include <glibtop/open.h>

#include <unistd.h>
#include <sys/types.h>
#include <sys/processor.h>

#include <glibtop_private.h>

/* We need to call this when kstat_chain_update() returns new KID.
 * In that case all kstat pointers and data are invalid, so we
 * need to reread everything. The condition shouldn't happen very
 * often.
 */

void
glibtop_get_kstats(glibtop *server)
{
    kstat_ctl_t *kc = server->machine.kc;
    kstat_t *ksp;
    int nproc_same, new_ncpu;

    new_ncpu = sysconf(_SC_NPROCESSORS_CONF);

    if(!kc)
    {
        server->ncpu = new_ncpu;
        server->machine.vminfo_kstat = NULL;
        server->machine.system = NULL;
        server->machine.syspages = NULL;
        server->machine.bunyip = NULL;
        return;
    }

    do {
	ksp = kstat_lookup(kc, "unix", -1, "vminfo");
	server->machine.vminfo_kstat = ksp;
	if(ksp)
	{
	    kstat_read(kc, ksp, &server->machine.vminfo);
	    server->machine.vminfo_snaptime = ksp->ks_snaptime;
	}

	/* We don't know why was kstat chain invalidated. It could have
	   been because the number of processors changed. The sysconf()
	   man page says that values returned won't change during the
	   life time of a process, but let's hope that's just an error in
	   the documentation. */

	if(nproc_same = new_ncpu == server->ncpu)
	{
	    int checked, i;
	    char cpu[20];

	    for(i = 0, checked = 0; i < GLIBTOP_NCPU || checked == new_ncpu; ++i)
		if(server->machine.cpu_stat_kstat[i])
		{
		    sprintf(cpu, "cpu_stat%d", i);
		    if(!(server->machine.cpu_stat_kstat[i] =
			     kstat_lookup(kc, "cpu_stat", -1, cpu)))
		    {
			nproc_same = 0;
			break;
		    }
		    ++checked;
		}
	}
	if(!nproc_same)
	{
	    processorid_t p;
	    int found;
	    char cpu[20];

	    if(new_ncpu > GLIBTOP_NCPU)
		new_ncpu = GLIBTOP_NCPU;
	    server->ncpu = new_ncpu;
	    for(p = 0, found = 0; p < GLIBTOP_NCPU && found != new_ncpu; ++p)
	    {
		if(p_online(p, P_STATUS) < 0)
		{
		    server->machine.cpu_stat_kstat[p] = NULL;
		    continue;
		}
		sprintf(cpu, "cpu_stat%d", (int)p);
		server->machine.cpu_stat_kstat[p] =
			kstat_lookup(kc, "cpu_stat", -1, cpu);
		++found;
	    }
	}

	server->machine.system   = kstat_lookup(kc, "unix", -1, "system_misc");
	server->machine.syspages = kstat_lookup(kc, "unix", -1, "system_pages");
	server->machine.bunyip   = kstat_lookup(kc, "bunyip", -1, "mempages");

    } while(kstat_chain_update(kc) > 0 &&
	    (new_ncpu = sysconf(_SC_NPROCESSORS_CONF)));

    /* We'll ignore -1 from kstat_chain_update here, since it really
       shouldn't happen */
}

void
glibtop_open_s (glibtop *server, const char *program_name,
		const unsigned long features, const unsigned flags)
{
    kstat_ctl_t *kc;
    kstat_t *ksp;
    kstat_named_t *kn;

    server->name = program_name;

    server->machine.pagesize = sysconf(_SC_PAGESIZE) >> 10;
    server->machine.ticks = sysconf(_SC_CLK_TCK);
    server->machine.kc = kc = kstat_open ();

#if 0
    for (ksp = server->machine.kc->kc_chain; ksp != NULL; ksp = ksp->ks_next) {
	if (!strcmp (ksp->ks_class, "vm") && !strcmp (ksp->ks_name, "vminfo")) {
	    server->machine.vminfo_kstat = ksp;
	    kstat_read (server->machine.kc, ksp, &server->machine.vminfo);
	    server->machine.vminfo_snaptime = ksp->ks_snaptime;
	    continue;
	}

	if (!strcmp (ksp->ks_class, "misc") && !strncmp (ksp->ks_name, "cpu_stat", 8)) {
	    int cpu;

	    if ((sscanf (ksp->ks_name+8, "%d", &cpu) != 1) || (cpu > 63))
		continue;

	    if (cpu >= server->ncpu)
		server->ncpu = cpu+1;

	    server->machine.cpu_stat_kstat [cpu] = ksp;
	    continue;
	}
    }

#endif

    if (!kc)
	glibtop_warn_io_r (server, "kstat_open ()");

    server->ncpu = -1;  /* Force processor detection */
    glibtop_get_kstats(server);

    server->machine.boot = 0;
    if((ksp = server->machine.system) && kstat_read(kc, ksp, NULL) >= 0)
    {
	kn = (kstat_named_t *)kstat_data_lookup(ksp, "boot_time");
	if(kn)
	    server->machine.boot = kn->value.ui32;
    }

    server->machine.kd = kvm_open(NULL, NULL, NULL, O_RDONLY, NULL);
    if(!server->machine.kd)
        glibtop_warn_io_r(server, "kvm_open()");

    fprintf (stderr, "Sleeping 2 seconds, please wait ...\n");
    sleep (2);
}