libgtop/sysdeps/osf1/procstate.c

/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 4 -*- */

/* $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.h>
#include <glibtop/error.h>
#include <glibtop/procstate.h>

#include <glibtop_suid.h>

#include <mach.h>
#include <mach/mach_types.h>
#include <mach/task_info.h>

static const unsigned long _glibtop_sysdeps_proc_state =
(1L << GLIBTOP_PROC_STATE_CMD) +
(1L << GLIBTOP_PROC_STATE_UID) + (1L << GLIBTOP_PROC_STATE_GID) +
(1L << GLIBTOP_PROC_STATE_RUID) + (1L << GLIBTOP_PROC_STATE_RGID);

/* Init function. */

int
glibtop_init_proc_state_p (glibtop *server)
{
    server->sysdeps.proc_state = _glibtop_sysdeps_proc_state;
}

/* Provides detailed information about a process. */

int
glibtop_get_proc_state_p (glibtop *server, glibtop_proc_state *buf,
			  pid_t pid)
{
    struct tbl_procinfo procinfo;
    int minim_state = 99, ret;
    task_t thistask;

    glibtop_init_p (server, GLIBTOP_SYSDEPS_PROC_STATE, 0);
	
    memset (buf, 0, sizeof (glibtop_proc_state));

    /* !!! THE FOLLOWING CODE RUNS SUID ROOT - CHANGE WITH CAUTION !!! */

    glibtop_suid_enter (server);
	
    ret = table (TBL_PROCINFO, pid, (char *) &procinfo, 1,
		 sizeof (struct tbl_procinfo)); 

    glibtop_suid_leave (server);
		     
    /* !!! END OF SUID ROOT PART !!! */
	
    if (ret != 1) return;

    /* Check whether the process actually exists. */
    if (procinfo.pi_status == PI_EMPTY) return;

    /* Check whether it is not a zombie. */
    if (procinfo.pi_status == PI_ZOMBIE) {
	buf->state = GLIBTOP_PROCESS_ZOMBIE;
	buf->flags = (1L << GLIBTOP_PROC_STATE_STATE);
	return;
    }

    strncpy (buf->cmd, procinfo.pi_comm, sizeof (buf->cmd)-1);

    buf->cmd [sizeof (buf->cmd)-1] = 0;

    buf->uid = procinfo.pi_svuid;
    buf->gid = procinfo.pi_svgid;
    buf->ruid = procinfo.pi_ruid;
    buf->rgid = procinfo.pi_rgid;

    buf->flags = _glibtop_sysdeps_proc_state;

    /* !!! THE FOLLOWING CODE RUNS SUID ROOT - CHANGE WITH CAUTION !!! */

    glibtop_suid_enter (server);
                
    /* Get task structure. */
    ret = task_by_unix_pid (task_self(), procinfo.pi_pid, &thistask);
        
    if (ret == KERN_SUCCESS) {
	thread_array_t			threadarr;
	unsigned int			threadarr_l;
	thread_basic_info_t		threadinfo;
	thread_basic_info_data_t	threadinfodata;
	int				j;

	/* Get thread array. */
	(void) task_threads (thistask, &threadarr, &threadarr_l);

	threadinfo = &threadinfodata;
	for (j = 0; j < threadarr_l; j++) {
	    unsigned int		threadinfo_l = THREAD_BASIC_INFO_COUNT;
	    int			tret;

	    tret = thread_info (threadarr [j], THREAD_BASIC_INFO,
				(thread_info_t) threadinfo, &threadinfo_l);

	    if (tret == KERN_SUCCESS) {
		if (minim_state > threadinfo->run_state) 
		    minim_state=threadinfo->run_state;
	    }
	}
    }

    glibtop_suid_leave (server);

    /* !!! END OF SUID ROOT PART !!! */
        
    if (ret != KERN_SUCCESS) return;

    switch (minim_state) {
    case TH_STATE_RUNNING:
	buf->state = GLIBTOP_PROCESS_RUNNING;
	break;
    case TH_STATE_UNINTERRUPTIBLE: 
	buf->state = GLIBTOP_PROCESS_UNINTERRUPTIBLE;
	break;
    case TH_STATE_WAITING:
	buf->state = GLIBTOP_PROCESS_INTERRUPTIBLE;
	break;
    case TH_STATE_STOPPED:      
    case TH_STATE_HALTED:
	buf->state = GLIBTOP_PROCESS_STOPPED;
	break;
    default:
	if (ret != KERN_SUCCESS)
	    buf->state = GLIBTOP_PROCESS_ZOMBIE;
	break;
    }

    if (buf->state)
	buf->flags |= (1L << GLIBTOP_PROC_STATE_STATE);
}