/* 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., 51 Franklin Street, Fifth Floor,
   Boston, MA 02110-1301, USA.
*/

#include <config.h>
#include <glibtop/open.h>
#include <glibtop/cpu.h>
#include <glibtop/error.h>

#include <unistd.h>
#include <dlfcn.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;

    server->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);
	    /* Don't change snaptime if we only need to reinitialize kstats */
	    if(!(server->machine->vminfo_snaptime))
		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];

	    server->ncpu = new_ncpu = MIN(new_ncpu, GLIBTOP_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;
    int i, page;
    void *dl;

    server->name = program_name;

    page = sysconf(_SC_PAGESIZE) >> 10;
    for(i = 0; page; ++i, page >>= 1);
    server->machine->pagesize = i - 1;
    server->machine->ticks = sysconf(_SC_CLK_TCK);
    if(server->machine->kc)
    	kstat_close(server->machine->kc);
    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 */
    server->machine->vminfo_snaptime = 0;  /* Force snaptime read */
    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)
	    switch(kn->data_type)
	    {
#ifdef KSTAT_DATA_INT32
		case KSTAT_DATA_INT32:  server->machine->boot = kn->value.i32;
					break;
		case KSTAT_DATA_UINT32: server->machine->boot = kn->value.ui32;
					break;
		case KSTAT_DATA_INT64:  server->machine->boot = kn->value.i64;
					break;
		case KSTAT_DATA_UINT64: server->machine->boot = kn->value.ui64;
					break;
#else
		case KSTAT_DATA_LONG:      server->machine->boot = kn->value.l;
					   break;
		case KSTAT_DATA_ULONG:     server->machine->boot = kn->value.ul;
					   break;
		case KSTAT_DATA_LONGLONG:  server->machine->boot = kn->value.ll;
					   break;
		case KSTAT_DATA_ULONGLONG: server->machine->boot = kn->value.ull;
					   break;
#endif
	    }
    }

    /* Now let's have a bit of magic dust... */

#if GLIBTOP_SOLARIS_RELEASE >= 50600

    dl = dlopen("/usr/lib/libproc.so", RTLD_LAZY);
    if(server->machine->libproc)
    	dlclose(server->machine->libproc);
    server->machine->libproc = dl;
    if(dl)
    {
       void *func;

       func = dlsym(dl, "Pobjname");		/* Solaris 8 */
       if(!func)
	  func = dlsym(dl, "proc_objname");	/* Solaris 7 */
       server->machine->objname = (void (*)
				 (void *, uintptr_t, const char *, size_t))func;
       server->machine->pgrab = (struct ps_prochandle *(*)(pid_t, int, int *))
	  		       dlsym(dl, "Pgrab");
       server->machine->pfree = (void (*)(void *))dlsym(dl, "Pfree");
       
    }
    else
    {
       server->machine->objname = NULL;
       server->machine->pgrab = NULL;
       server->machine->pfree = NULL;
    }
#endif
    server->machine->me = getpid();
}