/*
 * Copyright (c) 2003-2007 Andrea Luzzardi <scox@sig11.org>
 *
 * This file is part of the pam_usb project. pam_usb is free software;
 * you can redistribute it and/or modify it under the terms of the GNU General
 * Public License version 2, as published by the Free Software Foundation.
 *
 * pam_usb 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
 * this program; if not, write to the Free Software Foundation, Inc., 51 Franklin
 * Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/types.h>
#ifndef __GNU__
#include <sys/mount.h>
#endif

#include "mem.h"
#include "conf.h"
#include "log.h"
#include "volume.h"

static int pusb_volume_mount(t_pusb_volume *volume)
{
	GError			*error = NULL;
	GVariant		*options = NULL;
	GVariantBuilder		builder;
	int			retval = 0;
	const gchar *const	*mount_points = NULL;

	g_variant_builder_init(&builder, G_VARIANT_TYPE_VARDICT);
	options = g_variant_builder_end(&builder);

	log_debug("Attempting to mount device %s.\n", volume->device);

	udisks_filesystem_call_mount_sync(volume->filesystem,
			options,
			&volume->mount_point,
			NULL,
			&error);

	if (!error)
	{
		volume->unmount = 1;
		retval = 1;
		log_debug("Mounted device %s to %s.\n",
				volume->device, volume->mount_point);
	}
	else if (error->code == UDISKS_ERROR_ALREADY_MOUNTED)
	{
		g_main_context_iteration(NULL, FALSE);
		mount_points = udisks_filesystem_get_mount_points(volume->filesystem);
		volume->mount_point = xstrdup(*mount_points);
		retval = 1;
		log_debug("Device %s mounted in between our probe and mount.\n",
				volume->device);
	}
	else
	{
		log_error("Failed to mount device %s.\n", volume->device);
	}

	if (error)
		g_error_free(error);

	return (retval);
}

static t_pusb_volume *pusb_volume_probe(t_pusb_options *opts,
		UDisksClient *udisks)
{
	t_pusb_volume		*volume = NULL;
	int			maxtries = (opts->probe_timeout * 1000000) / 100000;
	int			i;
	int			j;
	GList			*blocks = NULL;
	UDisksBlock		*block = NULL;
	UDisksObject		*object = NULL;
	const gchar *const	*mount_points = NULL;

	if (!*(opts->device.volume_uuid))
	{
		log_debug("No UUID configured for device.\n");
		return (NULL);
	}

	log_debug("Searching for volume with uuid %s.\n",
			opts->device.volume_uuid);

	for (i = 0; i < maxtries; ++i)
	{
		blocks = udisks_client_get_block_for_uuid(udisks, opts->device.volume_uuid);

		if (i == 1)
			log_info("Probing volume (this could take a while)...\n");

		for (j = 0; j < g_list_length(blocks); ++j)
		{
			block = UDISKS_BLOCK(g_list_nth(blocks, j)->data);
			object = UDISKS_OBJECT(g_dbus_interface_get_object(G_DBUS_INTERFACE(block)));

			if (udisks_object_peek_filesystem(object))
			{
				volume = xmalloc(sizeof(t_pusb_volume));
				volume->filesystem = udisks_object_get_filesystem(object);
				volume->unmount = 0;
				volume->device = xstrdup(udisks_block_get_device(block));
				volume->mount_point = NULL;

				mount_points = udisks_filesystem_get_mount_points(volume->filesystem);
				log_debug("Found mount points: %s\n", *mount_points);
				if (mount_points && *mount_points)
					volume->mount_point = xstrdup(*mount_points);

				break;
			}
		}

		g_list_foreach (blocks, (GFunc) g_object_unref, NULL);
		g_list_free (blocks);

		if (volume)
		{
			log_debug("Found volume %s.\n", opts->device.volume_uuid);
			break;
		}

		usleep(100000);
		g_main_context_iteration(NULL, FALSE);
	}

	if (!volume)
		log_debug("Could not find volume %s.\n",
				opts->device.volume_uuid);

	return (volume);
}

t_pusb_volume *pusb_volume_get(t_pusb_options *opts, UDisksClient *udisks)
{
	t_pusb_volume *volume = pusb_volume_probe(opts, udisks);

	if (!volume)
		return (NULL);

	if (volume->mount_point)
	{
		log_debug("Volume %s is already mounted.\n",
				opts->device.volume_uuid);
		return (volume);
	}

	if(!pusb_volume_mount(volume))
	{
		pusb_volume_destroy(volume);
		return (NULL);
	}

	return (volume);
}

void pusb_volume_destroy(t_pusb_volume *volume)
{
	GVariantBuilder	builder;
	GVariant	*options;
	int		ret;

	if (volume->unmount)
	{
		g_variant_builder_init(&builder, G_VARIANT_TYPE_VARDICT);
		options = g_variant_builder_end(&builder);

		log_debug("Attempting to unmount %s from %s.\n",
				volume->device, volume->mount_point);

		ret = udisks_filesystem_call_unmount_sync(volume->filesystem,
				options,
				NULL,
				NULL);
		if (!ret)
			log_error("Unable to unmount %s from %s\n",
				volume->device, volume->mount_point);

		log_debug("Unmount succeeded.\n");
	}

	g_object_unref(volume->filesystem);
	xfree(volume->device);
	xfree(volume->mount_point);
	xfree(volume);
}