/*

   This file is part of LibGTop 2.0.

   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 <unistd.h>

#include <glibtop.h>

#include <glibtop/error.h>

#include <glibtop/procargs.h>

#include <glibtop_suid.h>

#include <sys/param.h>

#include <sys/sysctl.h>

#include <limits.h>

static const unsigned long _glibtop_sysdeps_proc_args =

(1L << GLIBTOP_PROC_ARGS_SIZE);

/* Init function. */

void

_glibtop_init_proc_args_p (glibtop *server)

{

	server->sysdeps.proc_args = _glibtop_sysdeps_proc_args;

}

/* Provides detailed information about a process. */

char *

glibtop_get_proc_args_p (glibtop *server, glibtop_proc_args *buf,

			 pid_t pid, unsigned max_len)

{

	char argbuf[4096];

	char *start, *end, *str;

	size_t bufsize = 4096;

	int mib [3];

	unsigned length;

	char *args;

	int *ip;

	glibtop_init_p (server, (1 << GLIBTOP_SYSDEPS_PROC_ARGS), 0);

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

	mib [0] = CTL_KERN;

	mib [1] = KERN_PROCARGS;

	mib [2] = pid;

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

	glibtop_suid_enter (server);

	if (sysctl (mib, 3, argbuf, &bufsize, NULL, 0) < 0) {

		glibtop_suid_leave (server);

		return NULL;

	}

	glibtop_suid_leave (server);

	/* !!! END OF SUID ROOT PART !!! */

    	end = &argbuf[bufsize];

	ip = (int *)end - 2;

	while (*--ip) {

		if (ip == (int *)argbuf) {

			return NULL;

		}

	}

	start = (char *)(ip + 1);

	while (*--ip) {

		if (ip == (int *)argbuf) {

			return NULL;

		}

	}

	ip++;

        if ((ip[0] & 0xbfff0000) == 0xbfff0000) {

		ip += 2;

		start = (char*)(ip + (strlen ((char*)ip) + 3) / 4);

		while (!*start)

			start++;

	}

	for (str = start + strlen (start); str < end - 1; str++) {

		if (*str == '=') {

			while (*str != '\0' && str > start)

				str--;

			break;

		}

	}

	end = str;

	if (start[0] == '-' || start[0] == '?' || start[0] <= ' ') {

		return NULL;

	}

	length = end - start;

	if (max_len != 0 && length > max_len)

		length = max_len;

	args = g_malloc(length + 1);

	if (args == NULL) {

		glibtop_warn_io_r (server, "malloc(procargs)");

		return NULL;

	}

	memcpy (args, start, length);

	args [length] = '\0';

	buf->size = length;

	buf->flags = _glibtop_sysdeps_proc_args;

	return args;

}