/* Hurd unionfs
   Copyright (C) 2001, 2002 Free Software Foundation, Inc.
   Written by Moritz Schulte <moritz@duesseldorf.ccc.de>.

   This program 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.
 
   This program 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307
   USA.  */

/* node management.  */

#define _GNU_SOURCE

#include <hurd/netfs.h>
#include <stdlib.h>
#include <error.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <stdio.h>

#include "unionfs.h"
#include "node.h"
#include "ulfs.h"
#include "lib.h"

/* Declarations for functions only used in this file.  */

/* Deallocate all ports contained in NODE and free per-ulfs data
   structures.  */
void node_ulfs_free (node_t *node);

/* Create a new node, derived from a light node, add a reference to
   the light node.  */
error_t
node_create (lnode_t *lnode, node_t **node)
{
  netnode_t *netnode_new = malloc (sizeof (netnode_t));
  error_t err = 0;
  node_t *node_new;

  debug_msg ("node_create for lnode: %s", lnode->name);

  if (! netnode_new)
    err = ENOMEM;
  else
    {
      node_new = netfs_make_node (netnode_new);
      if (! node_new)
	{
	  err = ENOMEM;
	  free (netnode_new);
	}
      else
	{
	  node_new->nn->ulfs = NULL;
	  err = node_ulfs_init (node_new);
	  if (err)
	    node_destroy (node_new);
	  else
	    {
	      lnode->node = node_new;
	      lnode_ref_add (lnode);
	      node_new->nn->lnode = lnode;
	      node_new->nn->flags = 0;
	      node_new->nn->ncache_next = NULL;
	      node_new->nn->ncache_prev = NULL;
	      *node = node_new;
	    }
	}
    }
  return err;
}

/* Destroy a node, remove one reference from the associated light
   node.  */
void
node_destroy (node_t *node)
{
  debug_msg ("node destroy: %s", node->nn->lnode->name);
  assert (! (node->nn->ncache_next || node->nn->ncache_prev));
  node_ulfs_free (node);
  mutex_lock (&node->nn->lnode->lock);
  node->nn->lnode->node = NULL;
  lnode_ref_remove (node->nn->lnode);
  free (node->nn);
  free (node);
}

/* Make sure that all ports to the underlying filesystems of NODE,
   which must be locked, are uptodate.  */
error_t
node_update (node_t *node)
{
  error_t err = 0;
  char *path;

  debug_msg ("node_update for lnode: %s", node->nn->lnode->name);

  if (! node_is_root (node))
    {
      mutex_lock (&netfs_root_node->lock);
      err = lnode_path_construct (node->nn->lnode, &path);
      if (! err)
	{
	  node_ulfs_t *root_ulfs;
	  struct stat stat;
	  file_t port;
	  int i = 0;
	  
	  root_ulfs = netfs_root_node->nn->ulfs;
	  node_ulfs_iterate_unlocked (node)
	    {
	      if (! (node_ulfs->flags & FLAG_NODE_ULFS_FIXED))
		{
		  /* We really have to update the port.  */
		  if (port_valid (node_ulfs->port))
		    port_dealloc (node_ulfs->port);
		  err = file_lookup ((root_ulfs + i)->port, path,
				     O_READ | O_NOTRANS, O_NOTRANS,
				     0, &port, &stat);
		  if (! err)
		    {
		      if (stat.st_ino == underlying_node_stat.st_ino
			  && stat.st_fsid == underlying_node_stat.st_fsid)
			/* It's OUR root node.  */
			err = ELOOP;
		      else
			{
			  port_dealloc (port);
			  err = file_lookup ((root_ulfs + i)->port, path,
					     O_READ, 0, 0, &port, &stat);
			}
		    }

		  if (err)
		    {
		      port = MACH_PORT_NULL;
		      err = 0;
		    }
		  node_ulfs->port = port;
		}
	      i++;
	    }
	  free (path);
	  node->nn->flags |= FLAG_NODE_ULFS_UPTODATE;
	}
      mutex_unlock (&netfs_root_node->lock);
    }
  return err;
}

/* Lookup a file named NAME beneath DIR on the underlying filesystems
   with FLAGS as openflags.  Return the first port successfully looked
   up in *PORT and according stat information in *STAT.  */
error_t
node_lookup_file (node_t *dir, char *name, int flags,
		  file_t *port, struct stat *s)
{
  error_t err = ENOENT;
  struct stat stat;
  file_t p;

  node_ulfs_iterate_unlocked (dir)
    {
      if (err == ENOENT && port_valid (node_ulfs->port))
	{
	  err = file_lookup (node_ulfs->port, name,
			     flags | O_NOTRANS, O_NOTRANS,
			     0, &p, &stat);
	  if (! err)
	    {
	      if (stat.st_ino == underlying_node_stat.st_ino
		  && stat.st_fsid == underlying_node_stat.st_fsid)
		/* It's OUR root node.  */
		err = ELOOP;
	      else if (1) //stat.st_mode & S_ITRANS)
		{
		  port_dealloc (p);
		  err = file_lookup (node_ulfs->port, name,
				     flags, 0, 0, &p, &stat);
		}
	    }
	}
    }
  if (! err)
    {
      *s = stat;
      *port = p;
    }
  return err;
}

/* Deallocate all ports contained in NODE and free per-ulfs data
   structures.  */
void
node_ulfs_free (node_t *node)
{
  node_ulfs_iterate_unlocked (node)
    if (port_valid (node_ulfs->port)
	&& node_ulfs->port != underlying_node)
      port_dealloc (node_ulfs->port);
  free (node->nn->ulfs);
}

/* Initialize per-ulfs data structures for NODE.  The ulfs_lock must
   be held by the caller.  */
error_t
node_ulfs_init (node_t *node)
{
  node_ulfs_t *ulfs_new;
  error_t err = 0;
  
  ulfs_new = malloc (ulfs_num * sizeof (node_ulfs_t));
  if (! ulfs_new)
    err = ENOMEM;
  else
    {
      if (node->nn->ulfs)
	node_ulfs_free (node);

      node->nn->ulfs = ulfs_new;
      node->nn->ulfs_num = ulfs_num;
      
      node_ulfs_iterate_unlocked (node)
	{
	  node_ulfs->flags = 0;
	  node_ulfs->port = port_null;
	}
    }

  return err;
}

/* Read the merged directory entries from NODE, which must be
   locked, into *DIRENTS.  */
error_t
node_entries_get (node_t *node, node_dirent_t **dirents)
{
  struct dirent **dirent_list, **dirent;
  node_dirent_t *node_dirent_list = NULL;
  char *dirent_data;
  error_t err = 0;

  /* Add a dirent to the list.  If an entry with the specified name
     already exists, reuse that entry.  Otherwise create a new
     one.  */
  error_t node_dirent_add (char *name, ino_t fileno, int type)
    {
      node_dirent_t *node_dirent;
      error_t e = 0;

      for (node_dirent = node_dirent_list;
	   node_dirent && strcmp (node_dirent->dirent->d_name, name);
	   node_dirent = node_dirent->next);

      if (node_dirent)
	{
	  /* Reuse existing entry.  */

	  node_dirent->dirent->d_fileno = fileno;
	  node_dirent->dirent->d_type = type;
	}
      else
	{
	  /* Create new entry.  */

	  node_dirent_t *node_dirent_new;

	  node_dirent_new = malloc (sizeof (node_dirent_t));
	  if (node_dirent_new)
	    {
	      int name_len = strlen (name);
	      int size = DIRENT_LEN (name_len);
	      struct dirent *dirent_new;

	      dirent_new = malloc (size);
	      if (dirent_new)
		{
		  /* Fill dirent.  */
		  dirent_new->d_fileno = fileno;
		  dirent_new->d_type = type;
		  dirent_new->d_reclen = size;
		  strcpy ((char *) dirent_new + DIRENT_NAME_OFFS, name);

		  /* Add dirent to the list.  */
		  node_dirent_new->dirent = dirent_new;
		  node_dirent_new->next = node_dirent_list;
		  node_dirent_list = node_dirent_new;
		}
	      else
		{
		  free (node_dirent_new);
		  e = ENOMEM;
		}
	    }
	  else
	    e = ENOMEM;
	}
      return e;
    }

  node_ulfs_iterate_unlocked(node)
    {
      if (port_valid (node_ulfs->port))
	{
	  err = dir_entries_get (node_ulfs->port,
				 &dirent_data, &dirent_list);
	  if (! err)
	    {
	      for (dirent = dirent_list; (! err) && *dirent; dirent++)
		if (strcmp ((*dirent)->d_name, ".")
		    && strcmp ((*dirent)->d_name, ".."))
		  err = node_dirent_add ((*dirent)->d_name,
					 (*dirent)->d_fileno,
					 (*dirent)->d_type);
	      free (dirent_list);
	    }
	}
    }

  if (err)
    node_entries_free (node_dirent_list);
  else
    *dirents = node_dirent_list;

  return err;
}

/* Free DIRENTS.  */
void
node_entries_free (node_dirent_t *dirents)
{
  node_dirent_t *dirent, *dirent_next;

  for (dirent = dirents; dirent; dirent = dirent_next)
    {
      dirent_next = dirent->next;
      free (dirent->dirent);
      free (dirent);
    }
}

/* Create the root node (and it's according lnode) and store it in
   *ROOT_NODE.  */
error_t
node_create_root (node_t **root_node)
{
  lnode_t *lnode;
  node_t *node;
  error_t err;

  err = lnode_create (NULL, &lnode);
  if (! err)
    {
      err = node_create (lnode, &node);
      mutex_unlock (&lnode->lock);
    }
  else
    lnode_destroy (lnode);

  if (! err)
    *root_node = node;
  return err;
}

/* Initialize the ports to the underlying filesystems for the root
   node.  */
error_t
node_init_root (node_t *node)
{
  error_t err;

  mutex_lock (&ulfs_lock);
  
  err = node_ulfs_init (node);
  if (! err)
    {
      ulfs_t *ulfs;
      int i = 0;

      node_ulfs_iterate_unlocked (node)
	{
	  if (! err)
	    err = ulfs_get_num (i, &ulfs);
	  if (! err)
	    {
	      if (ulfs->path)
		node_ulfs->port = file_name_lookup (ulfs->path,
						    O_READ | O_DIRECTORY, 0);
	      else
		node_ulfs->port = underlying_node;

	      if (! port_valid (node_ulfs->port))
		err = errno;
	      else
		{
		  node_ulfs->flags |= FLAG_NODE_ULFS_FIXED;
		  i++;
		}
	    }
	}
    }

  mutex_unlock (&ulfs_lock);
  return err;
}