/* -*- 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_private.h>
#include <glibtop/procuid.h>

#include <errno.h>

#include <glibtop_private.h>

/*
 * The differences between old and new procfs API are:
 * - old has '/proc/<pid>' file and ioctl() is used to obtain the data.
 * - new has '/proc/<pid>/*' files and read() & friends are used to
 *   obtain the data.
 * - names of structures and members are different. It's mostly the
 *   prefix. Old uses `pr' and new uses `ps'.
 *
 * Since almost every line would be in #ifdef, I think it would
 * be more readable to just copy everything twice. This is not a big
 * file, after all.
 */

#ifdef HAVE_PROCFS_H

/* Read /proc/<pid>/psinfo. */

int
glibtop_get_proc_data_psinfo_s (glibtop *server, struct psinfo *psinfo, pid_t pid)
{
    int fd;
    char buffer [BUFSIZ];

    sprintf (buffer, "/proc/%d/psinfo", (int) pid);
    fd = s_open (buffer, O_RDONLY);
    if (fd < 0) {
	glibtop_warn_io_r (server, "open (%s)", buffer);
	return -1;
    }

    if (s_pread (fd, psinfo, sizeof (struct psinfo), 0) !=
	sizeof (struct psinfo))
	{
	    s_close (fd);
	    glibtop_warn_io_r (server, "pread (%s)", buffer);
	    return -1;
	}

    s_close (fd);
    return 0;
}

int
glibtop_get_proc_data_usage_s (glibtop *server, struct prusage *prusage, pid_t pid)
{
    int fd;
    char buffer [BUFSIZ];

    sprintf (buffer, "/proc/%d/usage", (int) pid);
    fd = s_open (buffer, O_RDONLY);
    if (fd < 0) {
	glibtop_warn_io_r (server, "open (%s)", buffer);
	return -1;
    }

    if (s_pread (fd, prusage, sizeof (struct prusage), 0) !=
	sizeof (struct prusage))
	{
	    s_close (fd);
	    glibtop_warn_io_r (server, "pread (%s)", buffer);
	    return -1;
	}

    s_close (fd);
    return 0;
}

int
glibtop_get_proc_credentials_s(glibtop *server,
      			       struct prcred *prcred,
			       gid_t *groups,
			       pid_t pid)
{
    int fd;
    size_t toread;
    char buffer[BUFSIZ];

    sprintf(buffer, "/proc/%d/cred", (int)pid);
    if((fd = s_open(buffer, O_RDONLY)) < 0)
	{
	    if(errno != EPERM && errno != EACCES)
		glibtop_warn_io_r(server, "open (%s)", buffer);
	    return -1;
	}
    if(s_pread(fd, prcred, sizeof(struct prcred), 0) !=
       sizeof(struct prcred))
	{
	    s_close(fd);
	    glibtop_warn_io_r(server, "pread (%s)", buffer);
	    return -1;
	}
    if(prcred->pr_ngroups >= 0)
	{
	    if(prcred->pr_ngroups <= GLIBTOP_MAX_GROUPS)
	        toread = prcred->pr_ngroups * sizeof(gid_t);
	    else
	        toread = GLIBTOP_MAX_GROUPS * sizeof(gid_t);
	    if(s_pread(fd, groups, toread,
	               (off_t)&(((struct prcred *)0)->pr_groups[0])) != toread)
		prcred->pr_ngroups = 0;
	}
    s_close(fd);
    return 0;
}

int
glibtop_get_proc_status_s(glibtop *server, struct pstatus *pstatus, pid_t pid)
{
    int fd;
    char buffer[BUFSIZ];

    sprintf(buffer, "/proc/%d/status", (int)pid);
    if((fd = s_open(buffer, O_RDONLY)) < 0)
	{
	    if(errno != EPERM && errno != EACCES)
		glibtop_warn_io_r(server, "open (%s)", buffer);
	    return -1;
	}
    if(s_pread(fd, pstatus, sizeof(struct pstatus), 0) !=
       sizeof(struct pstatus))
	{
	    s_close(fd);
	    glibtop_warn_io_r(server, "pread (%s)", buffer);
	    return -1;
	}
    s_close(fd);
    return 0;
}

#else /* old API */

int
glibtop_get_proc_data_psinfo_s (glibtop *server,
      				struct prpsinfo *psinfo,
				pid_t pid)
{
    int fd;
    char buffer [BUFSIZ];

    sprintf (buffer, "/proc/%d", (int) pid);
    fd = s_open (buffer, O_RDONLY);
    if (fd < 0) {
	if(errno != EPERM && errno != EACCES)
	    glibtop_warn_io_r (server, "open (%s)", buffer);
	return -1;
    }

    if(ioctl(fd, PIOCPSINFO, psinfo) < 0)
	{
	    s_close (fd);
	    glibtop_warn_io_r (server, "ioctl(%s, PIOCPSINFO)", buffer);
	    return -1;
	}

    s_close (fd);
    return 0;
}

int
glibtop_get_proc_data_usage_s (glibtop *server,
      			       struct prusage *prusage,
			       pid_t pid)
{
    int fd;
    char buffer [BUFSIZ];

    sprintf (buffer, "/proc/%d", (int) pid);
    fd = s_open (buffer, O_RDONLY);
    if (fd < 0) {
	if(errno != EPERM && errno != EACCES)
	    glibtop_warn_io_r (server, "open (%s)", buffer);
	return -1;
    }

    if(ioctl(fd, PIOCUSAGE, prusage) < 0)
	{
	    s_close (fd);
	    glibtop_warn_io_r (server, "ioctl(%s, PIOCUSAGE)", buffer);
	    return -1;
	}

    s_close (fd);
    return 0;
}

int
glibtop_get_proc_credentials_s(glibtop *server,
      			       struct prcred *prcred,
			       gid_t *groups,
			       pid_t pid)
{
    int fd;
    size_t toread;
    char buffer[BUFSIZ];

    sprintf(buffer, "/proc/%d", (int)pid);
    if((fd = s_open(buffer, O_RDONLY)) < 0)
	{
	    if(errno != EPERM && errno != EACCES)
		glibtop_warn_io_r(server, "open (%s)", buffer);
	    return -1;
	}
    if(ioctl(fd, PIOCCRED, prcred) < 0)
	{
	    s_close(fd);
	    glibtop_warn_io_r(server, "ioctl(%s, PIOCCRED)", buffer);
	    return -1;
	}
    s_close(fd);
    return 0;
}

int
glibtop_get_proc_status_s(glibtop *server, struct prstatus *pstatus, pid_t pid)
{
    int fd;
    char buffer[BUFSIZ];

    sprintf(buffer, "/proc/%d", (int)pid);
    if((fd = s_open(buffer, O_RDONLY)) < 0)
	{
	    if(errno != EPERM && errno != EACCES)
		glibtop_warn_io_r(server, "open (%s)", buffer);
	    return -1;
	}
    if(ioctl(fd, PIOCSTATUS, pstatus) < 0)
	{
	    s_close(fd);
	    glibtop_warn_io_r(server, "ioctl(%s, PIOCSTATUS)", buffer);
	    return -1;
	}
    s_close(fd);
    return 0;
}
#endif