/*
   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/procmap.h>

#include <glibtop_suid.h>

#include <mach/mach_init.h>
#include <mach/mach_host.h>
#include <mach/vm_map.h>

static const unsigned long _glibtop_sysdeps_proc_map =
(1L << GLIBTOP_PROC_MAP_TOTAL) + (1L << GLIBTOP_PROC_MAP_NUMBER) +
(1L << GLIBTOP_PROC_MAP_SIZE);

static const unsigned long _glibtop_sysdeps_proc_map_entry =
(1L << GLIBTOP_MAP_ENTRY_START) + (1L << GLIBTOP_MAP_ENTRY_END) +
(1L << GLIBTOP_MAP_ENTRY_OFFSET) + (1L << GLIBTOP_MAP_ENTRY_PERM);

/* Init function. */

void
_glibtop_init_proc_map_p (glibtop *server)
{
	server->sysdeps.proc_map = _glibtop_sysdeps_proc_map;
}

/* Provides detailed information about a process. */

glibtop_map_entry *
glibtop_get_proc_map_p (glibtop *server, glibtop_proc_map *buf,	pid_t pid)
{
	glibtop_map_entry *maps;
	mach_port_t task;
	vm_address_t address;
	natural_t nesting_depth;
	int n;

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

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

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

	glibtop_suid_enter (server);

	if (task_for_pid (mach_task_self (), pid, &task) != KERN_SUCCESS) {
		glibtop_suid_leave (server);
		glibtop_warn_io_r (server, "task_for_pid (procmap)");
		return NULL;
	}

	maps = NULL;
	n = 0;
	address = 0;
	nesting_depth = 0;
	while (1) {
		vm_region_submap_info_data_64_t info;
		mach_msg_type_number_t info_count;
		vm_size_t size;

		info_count = VM_REGION_SUBMAP_INFO_COUNT_64;
		if (vm_region_recurse_64 (task, &address, &size, &nesting_depth,
			(vm_region_recurse_info_64_t)&info, &info_count))
			break;

		if (info.is_submap) {
			nesting_depth++;
			continue;
		}

		maps = g_realloc (maps, (n + 1) * sizeof (glibtop_map_entry));
		if (!maps) {
			glibtop_suid_leave (server);
			return NULL;
		}

		memset (maps + n, 0, sizeof (glibtop_map_entry));

		maps[n].start	= address;
		maps[n].end	= address + size - 1;
		maps[n].offset	= info.offset;
		if (info.protection & VM_PROT_READ)
			maps[n].perm |= GLIBTOP_MAP_PERM_READ;
		if (info.protection & VM_PROT_WRITE) 
			maps[n].perm |= GLIBTOP_MAP_PERM_WRITE;
		if (info.protection & VM_PROT_EXECUTE)      
			maps[n].perm |= GLIBTOP_MAP_PERM_EXECUTE;
		if (info.share_mode == SM_PRIVATE ||
		    info.share_mode == SM_PRIVATE_ALIASED)
			maps[n].perm |= GLIBTOP_MAP_PERM_PRIVATE;
		if (info.share_mode == SM_SHARED ||
		    info.share_mode == SM_TRUESHARED ||
		    info.share_mode == SM_SHARED_ALIASED)
			maps[n].perm |= GLIBTOP_MAP_PERM_SHARED;

		maps[n].flags	= _glibtop_sysdeps_proc_map_entry;

		address += size;
		n++;
	}

	glibtop_suid_leave (server);

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

	buf->number	= n;
	buf->size	= sizeof (glibtop_map_entry);
	buf->total	= n * sizeof (glibtop_map_entry);
	buf->flags	= _glibtop_sysdeps_proc_map;

	return maps;
}