/* Copyright (C) 2007 Benoît Dejean
   This file is part of LibGTop 2.

   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/procwd.h>
#include <glibtop/error.h>

#include <glibtop_private.h>

#include <unistd.h>
#include <dirent.h>
#include <sys/types.h>


void
_glibtop_init_proc_wd_s(glibtop *server)
{
	server->sysdeps.proc_wd =
	(1 << GLIBTOP_PROC_WD_EXE) +
	(1 << GLIBTOP_PROC_WD_ROOT) +
	(1 << GLIBTOP_PROC_WD_NUMBER);

}

static gboolean is_in(GPtrArray *array, const char *str)
{
	guint i;

	for (i = 0; i != array->len; ++i) {
		if (strcmp(g_ptr_array_index(array, i), str) == 0)
			return TRUE;
	}

	return FALSE;
}


char**
glibtop_get_proc_wd_s(glibtop *server, glibtop_proc_wd *buf, pid_t pid)
{
	GPtrArray *dirs;
	char path[80];
	char dir[256];
	DIR *task;

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

	g_snprintf(path, sizeof path, "/proc/%u/root", pid);
	if (safe_readlink(path, buf->root, sizeof buf->root))
		buf->flags |= (1 << GLIBTOP_PROC_WD_ROOT);

	g_snprintf(path, sizeof path, "/proc/%u/exe", pid);
	if (safe_readlink(path, buf->exe, sizeof buf->exe))
		buf->flags |= (1 << GLIBTOP_PROC_WD_EXE);

	dirs = g_ptr_array_sized_new(2);

	g_snprintf(path, sizeof path, "/proc/%u/cwd", pid);
	if (safe_readlink(path, dir, sizeof dir))
		g_ptr_array_add(dirs, g_strdup(dir));

	g_snprintf(path, sizeof path, "/proc/%u/task", pid);
	if ((task = opendir(path)) != NULL) {
		struct dirent *sub;
		while ((sub = readdir(task)) != NULL) {
			/* task dirs have numeric name */
			if (!isdigit(sub->d_name[0]))
				continue;
			g_snprintf(path, sizeof path, "/proc/%u/task/%s/cwd", pid, sub->d_name);
			if (safe_readlink(path, dir, sizeof dir) && !is_in(dirs, dir))
				g_ptr_array_add(dirs, g_strdup(dir));
		}
		closedir(task);
	}

	buf->number = dirs->len;
	buf->flags |= (1 << GLIBTOP_PROC_WD_NUMBER);

	g_ptr_array_add(dirs, NULL);

	return (char**) g_ptr_array_free(dirs, FALSE);
}