/**
   * eCryptfs: Linux filesystem encryption layer
   *
   * Copyright (C) 1997-2004 Erez Zadok
   * Copyright (C) 2001-2004 Stony Brook University

   * Copyright (C) 2004-2007 International Business Machines Corp.

   *   Author(s): Michael A. Halcrow <mahalcro@us.ibm.com>
   *              Michael C. Thompsion <mcthomps@us.ibm.com>
   *
   * 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.
   */
  
  #include <linux/file.h>
  #include <linux/vmalloc.h>
  #include <linux/pagemap.h>
  #include <linux/dcache.h>
  #include <linux/namei.h>
  #include <linux/mount.h>

  #include <linux/fs_stack.h>

  #include <linux/slab.h>

  #include <linux/xattr.h>

  #include <asm/unaligned.h>

  #include "ecryptfs_kernel.h"
  
  static struct dentry *lock_parent(struct dentry *dentry)
  {
  	struct dentry *dir;

  	dir = dget_parent(dentry);

  	inode_lock_nested(d_inode(dir), I_MUTEX_PARENT);

  	return dir;
  }

  static void unlock_dir(struct dentry *dir)
  {

  	inode_unlock(d_inode(dir));

  	dput(dir);
  }

  static int ecryptfs_inode_test(struct inode *inode, void *lower_inode)
  {

  	return ecryptfs_inode_to_lower(inode) == lower_inode;

  }

  static int ecryptfs_inode_set(struct inode *inode, void *opaque)

  {

  	struct inode *lower_inode = opaque;
  
  	ecryptfs_set_inode_lower(inode, lower_inode);
  	fsstack_copy_attr_all(inode, lower_inode);
  	/* i_size will be overwritten for encrypted regular files */
  	fsstack_copy_inode_size(inode, lower_inode);
  	inode->i_ino = lower_inode->i_ino;

  	inode->i_version++;

  	inode->i_mapping->a_ops = &ecryptfs_aops;

  
  	if (S_ISLNK(inode->i_mode))
  		inode->i_op = &ecryptfs_symlink_iops;
  	else if (S_ISDIR(inode->i_mode))
  		inode->i_op = &ecryptfs_dir_iops;
  	else
  		inode->i_op = &ecryptfs_main_iops;
  
  	if (S_ISDIR(inode->i_mode))
  		inode->i_fop = &ecryptfs_dir_fops;
  	else if (special_file(inode->i_mode))
  		init_special_inode(inode, inode->i_mode, inode->i_rdev);
  	else
  		inode->i_fop = &ecryptfs_main_fops;

  	return 0;
  }

  static struct inode *__ecryptfs_get_inode(struct inode *lower_inode,
  					  struct super_block *sb)

  {
  	struct inode *inode;

  	if (lower_inode->i_sb != ecryptfs_superblock_to_lower(sb))
  		return ERR_PTR(-EXDEV);
  	if (!igrab(lower_inode))
  		return ERR_PTR(-ESTALE);

  	inode = iget5_locked(sb, (unsigned long)lower_inode,
  			     ecryptfs_inode_test, ecryptfs_inode_set,
  			     lower_inode);
  	if (!inode) {

  		iput(lower_inode);

  		return ERR_PTR(-EACCES);

  	}

  	if (!(inode->i_state & I_NEW))

  		iput(lower_inode);

  
  	return inode;
  }
  
  struct inode *ecryptfs_get_inode(struct inode *lower_inode,
  				 struct super_block *sb)
  {
  	struct inode *inode = __ecryptfs_get_inode(lower_inode, sb);
  
  	if (!IS_ERR(inode) && (inode->i_state & I_NEW))
  		unlock_new_inode(inode);

  	return inode;

  }

  /**
   * ecryptfs_interpose
   * @lower_dentry: Existing dentry in the lower filesystem
   * @dentry: ecryptfs' dentry
   * @sb: ecryptfs's super_block

   *
   * Interposes upper and lower dentries.
   *
   * Returns zero on success; non-zero otherwise
   */
  static int ecryptfs_interpose(struct dentry *lower_dentry,

  			      struct dentry *dentry, struct super_block *sb)

  {

  	struct inode *inode = ecryptfs_get_inode(d_inode(lower_dentry), sb);

  	if (IS_ERR(inode))
  		return PTR_ERR(inode);

  	d_instantiate(dentry, inode);

  	return 0;
  }

  static int ecryptfs_do_unlink(struct inode *dir, struct dentry *dentry,
  			      struct inode *inode)
  {
  	struct dentry *lower_dentry = ecryptfs_dentry_to_lower(dentry);
  	struct inode *lower_dir_inode = ecryptfs_inode_to_lower(dir);
  	struct dentry *lower_dir_dentry;
  	int rc;
  
  	dget(lower_dentry);
  	lower_dir_dentry = lock_parent(lower_dentry);

  	rc = vfs_unlink(lower_dir_inode, lower_dentry, NULL);

  	if (rc) {
  		printk(KERN_ERR "Error in vfs_unlink; rc = [%d]
  ", rc);
  		goto out_unlock;
  	}
  	fsstack_copy_attr_times(dir, lower_dir_inode);
  	set_nlink(inode, ecryptfs_inode_to_lower(inode)->i_nlink);
  	inode->i_ctime = dir->i_ctime;
  	d_drop(dentry);
  out_unlock:
  	unlock_dir(lower_dir_dentry);
  	dput(lower_dentry);
  	return rc;
  }

  /**

   * ecryptfs_do_create
   * @directory_inode: inode of the new file's dentry's parent in ecryptfs
   * @ecryptfs_dentry: New file's dentry in ecryptfs
   * @mode: The mode of the new file

   *
   * Creates the underlying file and the eCryptfs inode which will link to
   * it. It will also update the eCryptfs directory inode to mimic the
   * stat of the lower directory inode.
   *

   * Returns the new eCryptfs inode on success; an ERR_PTR on error condition

   */

  static struct inode *

  ecryptfs_do_create(struct inode *directory_inode,

  		   struct dentry *ecryptfs_dentry, umode_t mode)

  {
  	int rc;
  	struct dentry *lower_dentry;
  	struct dentry *lower_dir_dentry;

  	struct inode *inode;

  
  	lower_dentry = ecryptfs_dentry_to_lower(ecryptfs_dentry);
  	lower_dir_dentry = lock_parent(lower_dentry);

  	rc = vfs_create(d_inode(lower_dir_dentry), lower_dentry, mode, true);

  	if (rc) {

  		printk(KERN_ERR "%s: Failure to create dentry in lower fs; "

  		       "rc = [%d]
  ", __func__, rc);

  		inode = ERR_PTR(rc);

  		goto out_lock;

  	}

  	inode = __ecryptfs_get_inode(d_inode(lower_dentry),

  				     directory_inode->i_sb);

  	if (IS_ERR(inode)) {

  		vfs_unlink(d_inode(lower_dir_dentry), lower_dentry, NULL);

  		goto out_lock;

  	}

  	fsstack_copy_attr_times(directory_inode, d_inode(lower_dir_dentry));
  	fsstack_copy_inode_size(directory_inode, d_inode(lower_dir_dentry));

  out_lock:
  	unlock_dir(lower_dir_dentry);

  	return inode;

  }
  
  /**

   * ecryptfs_initialize_file
   *
   * Cause the file to be changed from a basic empty file to an ecryptfs
   * file with a header and first data page.
   *
   * Returns zero on success
   */

  int ecryptfs_initialize_file(struct dentry *ecryptfs_dentry,
  			     struct inode *ecryptfs_inode)

  {

  	struct ecryptfs_crypt_stat *crypt_stat =

  		&ecryptfs_inode_to_private(ecryptfs_inode)->crypt_stat;

  	int rc = 0;

  	if (S_ISDIR(ecryptfs_inode->i_mode)) {

  		ecryptfs_printk(KERN_DEBUG, "This is a directory
  ");

  		crypt_stat->flags &= ~(ECRYPTFS_ENCRYPTED);

  		goto out;

  	}

  	ecryptfs_printk(KERN_DEBUG, "Initializing crypto context
  ");

  	rc = ecryptfs_new_file_context(ecryptfs_inode);

  	if (rc) {

  		ecryptfs_printk(KERN_ERR, "Error creating new file "
  				"context; rc = [%d]
  ", rc);
  		goto out;

  	}

  	rc = ecryptfs_get_lower_file(ecryptfs_dentry, ecryptfs_inode);

  	if (rc) {
  		printk(KERN_ERR "%s: Error attempting to initialize "

  			"the lower file for the dentry with name "

  			"[%pd]; rc = [%d]
  ", __func__,
  			ecryptfs_dentry, rc);

  		goto out;

  	}

  	rc = ecryptfs_write_metadata(ecryptfs_dentry, ecryptfs_inode);

  	if (rc)

  		printk(KERN_ERR "Error writing headers; rc = [%d]
  ", rc);

  	ecryptfs_put_lower_file(ecryptfs_inode);

  out:
  	return rc;
  }
  
  /**
   * ecryptfs_create
   * @dir: The inode of the directory in which to create the file.
   * @dentry: The eCryptfs dentry
   * @mode: The mode of the new file.

   *
   * Creates a new file.
   *
   * Returns zero on success; non-zero on error condition
   */
  static int
  ecryptfs_create(struct inode *directory_inode, struct dentry *ecryptfs_dentry,

  		umode_t mode, bool excl)

  {

  	struct inode *ecryptfs_inode;

  	int rc;

  	ecryptfs_inode = ecryptfs_do_create(directory_inode, ecryptfs_dentry,
  					    mode);

  	if (IS_ERR(ecryptfs_inode)) {

  		ecryptfs_printk(KERN_WARNING, "Failed to create file in"
  				"lower filesystem
  ");

  		rc = PTR_ERR(ecryptfs_inode);

  		goto out;
  	}
  	/* At this point, a file exists on "disk"; we need to make sure
  	 * that this on disk file is prepared to be an ecryptfs file */

  	rc = ecryptfs_initialize_file(ecryptfs_dentry, ecryptfs_inode);
  	if (rc) {

  		ecryptfs_do_unlink(directory_inode, ecryptfs_dentry,
  				   ecryptfs_inode);

  		iget_failed(ecryptfs_inode);

  		goto out;
  	}

  	d_instantiate_new(ecryptfs_dentry, ecryptfs_inode);

  out:
  	return rc;
  }

  static int ecryptfs_i_size_read(struct dentry *dentry, struct inode *inode)
  {
  	struct ecryptfs_crypt_stat *crypt_stat;
  	int rc;
  
  	rc = ecryptfs_get_lower_file(dentry, inode);
  	if (rc) {
  		printk(KERN_ERR "%s: Error attempting to initialize "
  			"the lower file for the dentry with name "

  			"[%pd]; rc = [%d]
  ", __func__,
  			dentry, rc);

  		return rc;
  	}
  
  	crypt_stat = &ecryptfs_inode_to_private(inode)->crypt_stat;
  	/* TODO: lock for crypt_stat comparison */
  	if (!(crypt_stat->flags & ECRYPTFS_POLICY_APPLIED))
  		ecryptfs_set_default_sizes(crypt_stat);
  
  	rc = ecryptfs_read_and_validate_header_region(inode);
  	ecryptfs_put_lower_file(inode);
  	if (rc) {
  		rc = ecryptfs_read_and_validate_xattr_region(dentry, inode);
  		if (!rc)
  			crypt_stat->flags |= ECRYPTFS_METADATA_IN_XATTR;
  	}
  
  	/* Must return 0 to allow non-eCryptfs files to be looked up, too */
  	return 0;
  }

  /**

   * ecryptfs_lookup_interpose - Dentry interposition for a lookup

   */

  static struct dentry *ecryptfs_lookup_interpose(struct dentry *dentry,
  				     struct dentry *lower_dentry)

  {

  	struct inode *inode, *lower_inode = d_inode(lower_dentry);

  	struct ecryptfs_dentry_info *dentry_info;

  	struct vfsmount *lower_mnt;

  	int rc = 0;

  	dentry_info = kmem_cache_alloc(ecryptfs_dentry_info_cache, GFP_KERNEL);

  	if (!dentry_info) {

  		printk(KERN_ERR "%s: Out of memory whilst attempting "
  		       "to allocate ecryptfs_dentry_info struct
  ",
  			__func__);

  		dput(lower_dentry);

  		return ERR_PTR(-ENOMEM);

  	}

  
  	lower_mnt = mntget(ecryptfs_dentry_to_lower_mnt(dentry->d_parent));

  	fsstack_copy_attr_atime(d_inode(dentry->d_parent),
  				d_inode(lower_dentry->d_parent));

  	BUG_ON(!d_count(lower_dentry));

  
  	ecryptfs_set_dentry_private(dentry, dentry_info);

  	dentry_info->lower_path.mnt = lower_mnt;
  	dentry_info->lower_path.dentry = lower_dentry;

  	if (d_really_is_negative(lower_dentry)) {

  		/* We want to add because we couldn't find in lower */

  		d_add(dentry, NULL);

  		return NULL;

  	}

  	inode = __ecryptfs_get_inode(lower_inode, dentry->d_sb);

  	if (IS_ERR(inode)) {

  		printk(KERN_ERR "%s: Error interposing; rc = [%ld]
  ",
  		       __func__, PTR_ERR(inode));

  		return ERR_CAST(inode);

  	}

  	if (S_ISREG(inode->i_mode)) {
  		rc = ecryptfs_i_size_read(dentry, inode);

  		if (rc) {

  			make_bad_inode(inode);

  			return ERR_PTR(rc);

  		}

  	}

  	if (inode->i_state & I_NEW)
  		unlock_new_inode(inode);

  	return d_splice_alias(inode, dentry);

  }
  
  /**
   * ecryptfs_lookup
   * @ecryptfs_dir_inode: The eCryptfs directory inode
   * @ecryptfs_dentry: The eCryptfs dentry that we are looking up

   * @flags: lookup flags

   *
   * Find a file on disk. If the file does not exist, then we'll add it to the
   * dentry cache and continue on to read it from the disk.
   */
  static struct dentry *ecryptfs_lookup(struct inode *ecryptfs_dir_inode,
  				      struct dentry *ecryptfs_dentry,

  				      unsigned int flags)

  {
  	char *encrypted_and_encoded_name = NULL;

  	struct ecryptfs_mount_crypt_stat *mount_crypt_stat;

  	struct dentry *lower_dir_dentry, *lower_dentry;

  	const char *name = ecryptfs_dentry->d_name.name;
  	size_t len = ecryptfs_dentry->d_name.len;

  	struct dentry *res;

  	int rc = 0;

  	lower_dir_dentry = ecryptfs_dentry_to_lower(ecryptfs_dentry->d_parent);

  	mount_crypt_stat = &ecryptfs_superblock_to_private(
  				ecryptfs_dentry->d_sb)->mount_crypt_stat;

  	if (mount_crypt_stat
  	    && (mount_crypt_stat->flags & ECRYPTFS_GLOBAL_ENCRYPT_FILENAMES)) {
  		rc = ecryptfs_encrypt_and_encode_filename(
  			&encrypted_and_encoded_name, &len,
  			mount_crypt_stat, name, len);
  		if (rc) {
  			printk(KERN_ERR "%s: Error attempting to encrypt and encode "
  			       "filename; rc = [%d]
  ", __func__, rc);
  			return ERR_PTR(rc);
  		}
  		name = encrypted_and_encoded_name;

  	}

  
  	lower_dentry = lookup_one_len_unlocked(name, lower_dir_dentry, len);

  	if (IS_ERR(lower_dentry)) {

  		ecryptfs_printk(KERN_DEBUG, "%s: lookup_one_len() returned "

  				"[%ld] on lower_dentry = [%s]
  ", __func__,
  				PTR_ERR(lower_dentry),

  				name);

  		res = ERR_CAST(lower_dentry);

  	} else {
  		res = ecryptfs_lookup_interpose(ecryptfs_dentry, lower_dentry);

  	}

  	kfree(encrypted_and_encoded_name);

  	return res;

  }
  
  static int ecryptfs_link(struct dentry *old_dentry, struct inode *dir,
  			 struct dentry *new_dentry)
  {
  	struct dentry *lower_old_dentry;
  	struct dentry *lower_new_dentry;
  	struct dentry *lower_dir_dentry;
  	u64 file_size_save;
  	int rc;

  	file_size_save = i_size_read(d_inode(old_dentry));

  	lower_old_dentry = ecryptfs_dentry_to_lower(old_dentry);
  	lower_new_dentry = ecryptfs_dentry_to_lower(new_dentry);
  	dget(lower_old_dentry);
  	dget(lower_new_dentry);
  	lower_dir_dentry = lock_parent(lower_new_dentry);

  	rc = vfs_link(lower_old_dentry, d_inode(lower_dir_dentry),

  		      lower_new_dentry, NULL);

  	if (rc || d_really_is_negative(lower_new_dentry))

  		goto out_lock;

  	rc = ecryptfs_interpose(lower_new_dentry, new_dentry, dir->i_sb);

  	if (rc)
  		goto out_lock;

  	fsstack_copy_attr_times(dir, d_inode(lower_dir_dentry));
  	fsstack_copy_inode_size(dir, d_inode(lower_dir_dentry));
  	set_nlink(d_inode(old_dentry),
  		  ecryptfs_inode_to_lower(d_inode(old_dentry))->i_nlink);
  	i_size_write(d_inode(new_dentry), file_size_save);

  out_lock:
  	unlock_dir(lower_dir_dentry);
  	dput(lower_new_dentry);
  	dput(lower_old_dentry);

  	return rc;
  }
  
  static int ecryptfs_unlink(struct inode *dir, struct dentry *dentry)
  {

  	return ecryptfs_do_unlink(dir, dentry, d_inode(dentry));

  }
  
  static int ecryptfs_symlink(struct inode *dir, struct dentry *dentry,
  			    const char *symname)
  {
  	int rc;
  	struct dentry *lower_dentry;
  	struct dentry *lower_dir_dentry;

  	char *encoded_symname;

  	size_t encoded_symlen;
  	struct ecryptfs_mount_crypt_stat *mount_crypt_stat = NULL;

  
  	lower_dentry = ecryptfs_dentry_to_lower(dentry);
  	dget(lower_dentry);
  	lower_dir_dentry = lock_parent(lower_dentry);

  	mount_crypt_stat = &ecryptfs_superblock_to_private(
  		dir->i_sb)->mount_crypt_stat;
  	rc = ecryptfs_encrypt_and_encode_filename(&encoded_symname,
  						  &encoded_symlen,

  						  mount_crypt_stat, symname,
  						  strlen(symname));
  	if (rc)

  		goto out_lock;

  	rc = vfs_symlink(d_inode(lower_dir_dentry), lower_dentry,

  			 encoded_symname);

  	kfree(encoded_symname);

  	if (rc || d_really_is_negative(lower_dentry))

  		goto out_lock;

  	rc = ecryptfs_interpose(lower_dentry, dentry, dir->i_sb);

  	if (rc)
  		goto out_lock;

  	fsstack_copy_attr_times(dir, d_inode(lower_dir_dentry));
  	fsstack_copy_inode_size(dir, d_inode(lower_dir_dentry));

  out_lock:
  	unlock_dir(lower_dir_dentry);
  	dput(lower_dentry);

  	if (d_really_is_negative(dentry))

  		d_drop(dentry);
  	return rc;
  }

  static int ecryptfs_mkdir(struct inode *dir, struct dentry *dentry, umode_t mode)

  {
  	int rc;
  	struct dentry *lower_dentry;
  	struct dentry *lower_dir_dentry;
  
  	lower_dentry = ecryptfs_dentry_to_lower(dentry);
  	lower_dir_dentry = lock_parent(lower_dentry);

  	rc = vfs_mkdir(d_inode(lower_dir_dentry), lower_dentry, mode);
  	if (rc || d_really_is_negative(lower_dentry))

  		goto out;

  	rc = ecryptfs_interpose(lower_dentry, dentry, dir->i_sb);

  	if (rc)
  		goto out;

  	fsstack_copy_attr_times(dir, d_inode(lower_dir_dentry));
  	fsstack_copy_inode_size(dir, d_inode(lower_dir_dentry));
  	set_nlink(dir, d_inode(lower_dir_dentry)->i_nlink);

  out:
  	unlock_dir(lower_dir_dentry);

  	if (d_really_is_negative(dentry))

  		d_drop(dentry);
  	return rc;
  }
  
  static int ecryptfs_rmdir(struct inode *dir, struct dentry *dentry)
  {

  	struct dentry *lower_dentry;

  	struct dentry *lower_dir_dentry;

  	int rc;

  
  	lower_dentry = ecryptfs_dentry_to_lower(dentry);

  	dget(dentry);

  	lower_dir_dentry = lock_parent(lower_dentry);

  	dget(lower_dentry);

  	rc = vfs_rmdir(d_inode(lower_dir_dentry), lower_dentry);

  	dput(lower_dentry);

  	if (!rc && d_really_is_positive(dentry))
  		clear_nlink(d_inode(dentry));
  	fsstack_copy_attr_times(dir, d_inode(lower_dir_dentry));
  	set_nlink(dir, d_inode(lower_dir_dentry)->i_nlink);

  	unlock_dir(lower_dir_dentry);
  	if (!rc)
  		d_drop(dentry);

  	dput(dentry);

  	return rc;
  }
  
  static int

  ecryptfs_mknod(struct inode *dir, struct dentry *dentry, umode_t mode, dev_t dev)

  {
  	int rc;
  	struct dentry *lower_dentry;
  	struct dentry *lower_dir_dentry;
  
  	lower_dentry = ecryptfs_dentry_to_lower(dentry);
  	lower_dir_dentry = lock_parent(lower_dentry);

  	rc = vfs_mknod(d_inode(lower_dir_dentry), lower_dentry, mode, dev);
  	if (rc || d_really_is_negative(lower_dentry))

  		goto out;

  	rc = ecryptfs_interpose(lower_dentry, dentry, dir->i_sb);

  	if (rc)
  		goto out;

  	fsstack_copy_attr_times(dir, d_inode(lower_dir_dentry));
  	fsstack_copy_inode_size(dir, d_inode(lower_dir_dentry));

  out:
  	unlock_dir(lower_dir_dentry);

  	if (d_really_is_negative(dentry))

  		d_drop(dentry);
  	return rc;
  }
  
  static int
  ecryptfs_rename(struct inode *old_dir, struct dentry *old_dentry,

  		struct inode *new_dir, struct dentry *new_dentry,
  		unsigned int flags)

  {
  	int rc;
  	struct dentry *lower_old_dentry;
  	struct dentry *lower_new_dentry;
  	struct dentry *lower_old_dir_dentry;
  	struct dentry *lower_new_dir_dentry;

  	struct dentry *trap = NULL;

  	struct inode *target_inode;

  	if (flags)
  		return -EINVAL;

  	lower_old_dentry = ecryptfs_dentry_to_lower(old_dentry);
  	lower_new_dentry = ecryptfs_dentry_to_lower(new_dentry);
  	dget(lower_old_dentry);
  	dget(lower_new_dentry);
  	lower_old_dir_dentry = dget_parent(lower_old_dentry);
  	lower_new_dir_dentry = dget_parent(lower_new_dentry);

  	target_inode = d_inode(new_dentry);

  	trap = lock_rename(lower_old_dir_dentry, lower_new_dir_dentry);
  	/* source should not be ancestor of target */
  	if (trap == lower_old_dentry) {
  		rc = -EINVAL;
  		goto out_lock;
  	}
  	/* target should not be ancestor of source */
  	if (trap == lower_new_dentry) {
  		rc = -ENOTEMPTY;
  		goto out_lock;
  	}

  	rc = vfs_rename(d_inode(lower_old_dir_dentry), lower_old_dentry,
  			d_inode(lower_new_dir_dentry), lower_new_dentry,

  			NULL, 0);

  	if (rc)
  		goto out_lock;

  	if (target_inode)
  		fsstack_copy_attr_all(target_inode,
  				      ecryptfs_inode_to_lower(target_inode));

  	fsstack_copy_attr_all(new_dir, d_inode(lower_new_dir_dentry));

  	if (new_dir != old_dir)

  		fsstack_copy_attr_all(old_dir, d_inode(lower_old_dir_dentry));

  out_lock:
  	unlock_rename(lower_old_dir_dentry, lower_new_dir_dentry);

  	dput(lower_new_dir_dentry);
  	dput(lower_old_dir_dentry);

  	dput(lower_new_dentry);
  	dput(lower_old_dentry);
  	return rc;
  }

  static char *ecryptfs_readlink_lower(struct dentry *dentry, size_t *bufsiz)

  {

  	DEFINE_DELAYED_CALL(done);

  	struct dentry *lower_dentry = ecryptfs_dentry_to_lower(dentry);

  	const char *link;

  	char *buf;

  	int rc;

  	link = vfs_get_link(lower_dentry, &done);
  	if (IS_ERR(link))
  		return ERR_CAST(link);

  	rc = ecryptfs_decode_and_decrypt_filename(&buf, bufsiz, dentry->d_sb,

  						  link, strlen(link));
  	do_delayed_call(&done);
  	if (rc)
  		return ERR_PTR(rc);
  
  	return buf;

  }

  static const char *ecryptfs_get_link(struct dentry *dentry,

  				     struct inode *inode,
  				     struct delayed_call *done)

  {

  	size_t len;

  	char *buf;
  
  	if (!dentry)
  		return ERR_PTR(-ECHILD);
  
  	buf = ecryptfs_readlink_lower(dentry, &len);

  	if (IS_ERR(buf))

  		return buf;

  	fsstack_copy_attr_atime(d_inode(dentry),
  				d_inode(ecryptfs_dentry_to_lower(dentry)));

  	buf[len] = '\0';

  	set_delayed_call(done, kfree_link, buf);
  	return buf;

  }

  /**
   * upper_size_to_lower_size
   * @crypt_stat: Crypt_stat associated with file
   * @upper_size: Size of the upper file
   *

   * Calculate the required size of the lower file based on the

   * specified size of the upper file. This calculation is based on the
   * number of headers in the underlying file and the extent size.
   *
   * Returns Calculated size of the lower file.
   */
  static loff_t
  upper_size_to_lower_size(struct ecryptfs_crypt_stat *crypt_stat,
  			 loff_t upper_size)
  {
  	loff_t lower_size;

  	lower_size = ecryptfs_lower_header_size(crypt_stat);

  	if (upper_size != 0) {
  		loff_t num_extents;
  
  		num_extents = upper_size >> crypt_stat->extent_shift;
  		if (upper_size & ~crypt_stat->extent_mask)
  			num_extents++;
  		lower_size += (num_extents * crypt_stat->extent_size);
  	}
  	return lower_size;
  }
  
  /**

   * truncate_upper

   * @dentry: The ecryptfs layer dentry

   * @ia: Address of the ecryptfs inode's attributes
   * @lower_ia: Address of the lower inode's attributes

   *
   * Function to handle truncations modifying the size of the file. Note
   * that the file sizes are interpolated. When expanding, we are simply

   * writing strings of 0's out. When truncating, we truncate the upper
   * inode and update the lower_ia according to the page index
   * interpolations. If ATTR_SIZE is set in lower_ia->ia_valid upon return,
   * the caller must use lower_ia in a call to notify_change() to perform
   * the truncation of the lower inode.

   *
   * Returns zero on success; non-zero otherwise
   */

  static int truncate_upper(struct dentry *dentry, struct iattr *ia,
  			  struct iattr *lower_ia)

  {
  	int rc = 0;

  	struct inode *inode = d_inode(dentry);

  	struct ecryptfs_crypt_stat *crypt_stat;
  	loff_t i_size = i_size_read(inode);
  	loff_t lower_size_before_truncate;
  	loff_t lower_size_after_truncate;

  	if (unlikely((ia->ia_size == i_size))) {
  		lower_ia->ia_valid &= ~ATTR_SIZE;

  		return 0;

  	}

  	rc = ecryptfs_get_lower_file(dentry, inode);

  	if (rc)
  		return rc;

  	crypt_stat = &ecryptfs_inode_to_private(d_inode(dentry))->crypt_stat;

  	/* Switch on growing or shrinking file */

  	if (ia->ia_size > i_size) {

  		char zero[] = { 0x00 };

  		lower_ia->ia_valid &= ~ATTR_SIZE;

  		/* Write a single 0 at the last position of the file;
  		 * this triggers code that will fill in 0's throughout
  		 * the intermediate portion of the previous end of the
  		 * file and the new and of the file */

  		rc = ecryptfs_write(inode, zero,

  				    (ia->ia_size - 1), 1);
  	} else { /* ia->ia_size < i_size_read(inode) */
  		/* We're chopping off all the pages down to the page
  		 * in which ia->ia_size is located. Fill in the end of

  		 * that page from (ia->ia_size & ~PAGE_MASK) to
  		 * PAGE_SIZE with zeros. */

  		size_t num_zeros = (PAGE_SIZE
  				    - (ia->ia_size & ~PAGE_MASK));

  		if (!(crypt_stat->flags & ECRYPTFS_ENCRYPTED)) {

  			truncate_setsize(inode, ia->ia_size);

  			lower_ia->ia_size = ia->ia_size;
  			lower_ia->ia_valid |= ATTR_SIZE;

  			goto out;

  		}

  		if (num_zeros) {
  			char *zeros_virt;
  
  			zeros_virt = kzalloc(num_zeros, GFP_KERNEL);
  			if (!zeros_virt) {
  				rc = -ENOMEM;

  				goto out;

  			}

  			rc = ecryptfs_write(inode, zeros_virt,

  					    ia->ia_size, num_zeros);

  			kfree(zeros_virt);

  			if (rc) {

  				printk(KERN_ERR "Error attempting to zero out "
  				       "the remainder of the end page on "
  				       "reducing truncate; rc = [%d]
  ", rc);

  				goto out;

  			}
  		}

  		truncate_setsize(inode, ia->ia_size);

  		rc = ecryptfs_write_inode_size_to_metadata(inode);

  		if (rc) {
  			printk(KERN_ERR	"Problem with "
  			       "ecryptfs_write_inode_size_to_metadata; "
  			       "rc = [%d]
  ", rc);

  			goto out;

  		}

  		/* We are reducing the size of the ecryptfs file, and need to
  		 * know if we need to reduce the size of the lower file. */
  		lower_size_before_truncate =
  		    upper_size_to_lower_size(crypt_stat, i_size);
  		lower_size_after_truncate =

  		    upper_size_to_lower_size(crypt_stat, ia->ia_size);
  		if (lower_size_after_truncate < lower_size_before_truncate) {
  			lower_ia->ia_size = lower_size_after_truncate;
  			lower_ia->ia_valid |= ATTR_SIZE;
  		} else
  			lower_ia->ia_valid &= ~ATTR_SIZE;

  	}

  out:

  	ecryptfs_put_lower_file(inode);

  	return rc;
  }

  static int ecryptfs_inode_newsize_ok(struct inode *inode, loff_t offset)
  {
  	struct ecryptfs_crypt_stat *crypt_stat;
  	loff_t lower_oldsize, lower_newsize;
  
  	crypt_stat = &ecryptfs_inode_to_private(inode)->crypt_stat;
  	lower_oldsize = upper_size_to_lower_size(crypt_stat,
  						 i_size_read(inode));
  	lower_newsize = upper_size_to_lower_size(crypt_stat, offset);
  	if (lower_newsize > lower_oldsize) {
  		/*
  		 * The eCryptfs inode and the new *lower* size are mixed here
  		 * because we may not have the lower i_mutex held and/or it may
  		 * not be appropriate to call inode_newsize_ok() with inodes
  		 * from other filesystems.
  		 */
  		return inode_newsize_ok(inode, lower_newsize);
  	}
  
  	return 0;
  }

  /**
   * ecryptfs_truncate
   * @dentry: The ecryptfs layer dentry
   * @new_length: The length to expand the file to
   *
   * Simple function that handles the truncation of an eCryptfs inode and
   * its corresponding lower inode.
   *
   * Returns zero on success; non-zero otherwise
   */
  int ecryptfs_truncate(struct dentry *dentry, loff_t new_length)
  {
  	struct iattr ia = { .ia_valid = ATTR_SIZE, .ia_size = new_length };
  	struct iattr lower_ia = { .ia_valid = 0 };
  	int rc;

  	rc = ecryptfs_inode_newsize_ok(d_inode(dentry), new_length);

  	if (rc)
  		return rc;

  	rc = truncate_upper(dentry, &ia, &lower_ia);
  	if (!rc && lower_ia.ia_valid & ATTR_SIZE) {
  		struct dentry *lower_dentry = ecryptfs_dentry_to_lower(dentry);

  		inode_lock(d_inode(lower_dentry));

  		rc = notify_change(lower_dentry, &lower_ia, NULL);

  		inode_unlock(d_inode(lower_dentry));

  	}
  	return rc;
  }

  static int

  ecryptfs_permission(struct inode *inode, int mask)

  {

  	return inode_permission(ecryptfs_inode_to_lower(inode), mask);

  }
  
  /**
   * ecryptfs_setattr
   * @dentry: dentry handle to the inode to modify
   * @ia: Structure with flags of what to change and values
   *
   * Updates the metadata of an inode. If the update is to the size
   * i.e. truncation, then ecryptfs_truncate will handle the size modification
   * of both the ecryptfs inode and the lower inode.
   *
   * All other metadata changes will be passed right to the lower filesystem,
   * and we will just update our inode to look like the lower.
   */
  static int ecryptfs_setattr(struct dentry *dentry, struct iattr *ia)
  {
  	int rc = 0;
  	struct dentry *lower_dentry;

  	struct iattr lower_ia;

  	struct inode *inode;
  	struct inode *lower_inode;
  	struct ecryptfs_crypt_stat *crypt_stat;

  	crypt_stat = &ecryptfs_inode_to_private(d_inode(dentry))->crypt_stat;

  	if (!(crypt_stat->flags & ECRYPTFS_STRUCT_INITIALIZED)) {
  		rc = ecryptfs_init_crypt_stat(crypt_stat);
  		if (rc)
  			return rc;
  	}

  	inode = d_inode(dentry);

  	lower_inode = ecryptfs_inode_to_lower(inode);

  	lower_dentry = ecryptfs_dentry_to_lower(dentry);
  	mutex_lock(&crypt_stat->cs_mutex);

  	if (d_is_dir(dentry))

  		crypt_stat->flags &= ~(ECRYPTFS_ENCRYPTED);

  	else if (d_is_reg(dentry)

  		 && (!(crypt_stat->flags & ECRYPTFS_POLICY_APPLIED)
  		     || !(crypt_stat->flags & ECRYPTFS_KEY_VALID))) {

  		struct ecryptfs_mount_crypt_stat *mount_crypt_stat;

  		mount_crypt_stat = &ecryptfs_superblock_to_private(
  			dentry->d_sb)->mount_crypt_stat;

  		rc = ecryptfs_get_lower_file(dentry, inode);

  		if (rc) {
  			mutex_unlock(&crypt_stat->cs_mutex);
  			goto out;
  		}

  		rc = ecryptfs_read_metadata(dentry);

  		ecryptfs_put_lower_file(inode);

  		if (rc) {

  			if (!(mount_crypt_stat->flags
  			      & ECRYPTFS_PLAINTEXT_PASSTHROUGH_ENABLED)) {
  				rc = -EIO;

  				printk(KERN_WARNING "Either the lower file "

  				       "is not in a valid eCryptfs format, "

  				       "or the key could not be retrieved. "
  				       "Plaintext passthrough mode is not "

  				       "enabled; returning -EIO
  ");

  				mutex_unlock(&crypt_stat->cs_mutex);

  				goto out;
  			}
  			rc = 0;

  			crypt_stat->flags &= ~(ECRYPTFS_I_SIZE_INITIALIZED
  					       | ECRYPTFS_ENCRYPTED);

  		}

  	}
  	mutex_unlock(&crypt_stat->cs_mutex);

  	rc = setattr_prepare(dentry, ia);

  	if (rc)
  		goto out;
  	if (ia->ia_valid & ATTR_SIZE) {
  		rc = ecryptfs_inode_newsize_ok(inode, ia->ia_size);
  		if (rc)
  			goto out;
  	}

  	memcpy(&lower_ia, ia, sizeof(lower_ia));
  	if (ia->ia_valid & ATTR_FILE)
  		lower_ia.ia_file = ecryptfs_file_to_lower(ia->ia_file);

  	if (ia->ia_valid & ATTR_SIZE) {

  		rc = truncate_upper(dentry, ia, &lower_ia);

  		if (rc < 0)
  			goto out;
  	}

  
  	/*
  	 * mode change is for clearing setuid/setgid bits. Allow lower fs
  	 * to interpret this in its own way.
  	 */

  	if (lower_ia.ia_valid & (ATTR_KILL_SUID | ATTR_KILL_SGID))
  		lower_ia.ia_valid &= ~ATTR_MODE;

  	inode_lock(d_inode(lower_dentry));

  	rc = notify_change(lower_dentry, &lower_ia, NULL);

  	inode_unlock(d_inode(lower_dentry));

  out:

  	fsstack_copy_attr_all(inode, lower_inode);

  	return rc;
  }

  static int ecryptfs_getattr_link(const struct path *path, struct kstat *stat,
  				 u32 request_mask, unsigned int flags)

  {

  	struct dentry *dentry = path->dentry;

  	struct ecryptfs_mount_crypt_stat *mount_crypt_stat;
  	int rc = 0;
  
  	mount_crypt_stat = &ecryptfs_superblock_to_private(
  						dentry->d_sb)->mount_crypt_stat;

  	generic_fillattr(d_inode(dentry), stat);

  	if (mount_crypt_stat->flags & ECRYPTFS_GLOBAL_ENCRYPT_FILENAMES) {
  		char *target;
  		size_t targetsiz;

  		target = ecryptfs_readlink_lower(dentry, &targetsiz);
  		if (!IS_ERR(target)) {

  			kfree(target);
  			stat->size = targetsiz;

  		} else {
  			rc = PTR_ERR(target);

  		}
  	}
  	return rc;
  }

  static int ecryptfs_getattr(const struct path *path, struct kstat *stat,
  			    u32 request_mask, unsigned int flags)

  {

  	struct dentry *dentry = path->dentry;

  	struct kstat lower_stat;
  	int rc;

  	rc = vfs_getattr(ecryptfs_dentry_to_lower_path(dentry), &lower_stat,
  			 request_mask, flags);

  	if (!rc) {

  		fsstack_copy_attr_all(d_inode(dentry),
  				      ecryptfs_inode_to_lower(d_inode(dentry)));
  		generic_fillattr(d_inode(dentry), stat);

  		stat->blocks = lower_stat.blocks;
  	}
  	return rc;
  }

  int

  ecryptfs_setxattr(struct dentry *dentry, struct inode *inode,
  		  const char *name, const void *value,

  		  size_t size, int flags)
  {

  	int rc;

  	struct dentry *lower_dentry;
  
  	lower_dentry = ecryptfs_dentry_to_lower(dentry);

  	if (!(d_inode(lower_dentry)->i_opflags & IOP_XATTR)) {

  		rc = -EOPNOTSUPP;

  		goto out;
  	}

  	rc = vfs_setxattr(lower_dentry, name, value, size, flags);

  	if (!rc && inode)
  		fsstack_copy_attr_all(inode, d_inode(lower_dentry));

  out:
  	return rc;
  }

  ssize_t

  ecryptfs_getxattr_lower(struct dentry *lower_dentry, struct inode *lower_inode,
  			const char *name, void *value, size_t size)

  {

  	int rc;

  	if (!(lower_inode->i_opflags & IOP_XATTR)) {

  		rc = -EOPNOTSUPP;

  		goto out;
  	}

  	inode_lock(lower_inode);

  	rc = __vfs_getxattr(lower_dentry, lower_inode, name, value, size);

  	inode_unlock(lower_inode);

  out:
  	return rc;
  }

  static ssize_t

  ecryptfs_getxattr(struct dentry *dentry, struct inode *inode,
  		  const char *name, void *value, size_t size)

  {

  	return ecryptfs_getxattr_lower(ecryptfs_dentry_to_lower(dentry),
  				       ecryptfs_inode_to_lower(inode),
  				       name, value, size);

  }
  
  static ssize_t
  ecryptfs_listxattr(struct dentry *dentry, char *list, size_t size)
  {
  	int rc = 0;
  	struct dentry *lower_dentry;
  
  	lower_dentry = ecryptfs_dentry_to_lower(dentry);

  	if (!d_inode(lower_dentry)->i_op->listxattr) {

  		rc = -EOPNOTSUPP;

  		goto out;
  	}

  	inode_lock(d_inode(lower_dentry));

  	rc = d_inode(lower_dentry)->i_op->listxattr(lower_dentry, list, size);

  	inode_unlock(d_inode(lower_dentry));

  out:
  	return rc;
  }

  static int ecryptfs_removexattr(struct dentry *dentry, struct inode *inode,
  				const char *name)

  {

  	int rc;

  	struct dentry *lower_dentry;

  	struct inode *lower_inode;

  
  	lower_dentry = ecryptfs_dentry_to_lower(dentry);

  	lower_inode = ecryptfs_inode_to_lower(inode);

  	if (!(lower_inode->i_opflags & IOP_XATTR)) {

  		rc = -EOPNOTSUPP;

  		goto out;
  	}

  	inode_lock(lower_inode);

  	rc = __vfs_removexattr(lower_dentry, name);

  	inode_unlock(lower_inode);

  out:
  	return rc;
  }

  const struct inode_operations ecryptfs_symlink_iops = {

  	.get_link = ecryptfs_get_link,

  	.permission = ecryptfs_permission,
  	.setattr = ecryptfs_setattr,

  	.getattr = ecryptfs_getattr_link,

  	.listxattr = ecryptfs_listxattr,

  };

  const struct inode_operations ecryptfs_dir_iops = {

  	.create = ecryptfs_create,
  	.lookup = ecryptfs_lookup,
  	.link = ecryptfs_link,
  	.unlink = ecryptfs_unlink,
  	.symlink = ecryptfs_symlink,
  	.mkdir = ecryptfs_mkdir,
  	.rmdir = ecryptfs_rmdir,
  	.mknod = ecryptfs_mknod,
  	.rename = ecryptfs_rename,
  	.permission = ecryptfs_permission,
  	.setattr = ecryptfs_setattr,

  	.listxattr = ecryptfs_listxattr,

  };

  const struct inode_operations ecryptfs_main_iops = {

  	.permission = ecryptfs_permission,
  	.setattr = ecryptfs_setattr,

  	.getattr = ecryptfs_getattr,

  	.listxattr = ecryptfs_listxattr,

  };
  
  static int ecryptfs_xattr_get(const struct xattr_handler *handler,
  			      struct dentry *dentry, struct inode *inode,
  			      const char *name, void *buffer, size_t size)
  {
  	return ecryptfs_getxattr(dentry, inode, name, buffer, size);
  }
  
  static int ecryptfs_xattr_set(const struct xattr_handler *handler,
  			      struct dentry *dentry, struct inode *inode,
  			      const char *name, const void *value, size_t size,
  			      int flags)
  {
  	if (value)
  		return ecryptfs_setxattr(dentry, inode, name, value, size, flags);
  	else {
  		BUG_ON(flags != XATTR_REPLACE);
  		return ecryptfs_removexattr(dentry, inode, name);
  	}
  }
  
  const struct xattr_handler ecryptfs_xattr_handler = {
  	.prefix = "",  /* match anything */
  	.get = ecryptfs_xattr_get,
  	.set = ecryptfs_xattr_set,
  };
  
  const struct xattr_handler *ecryptfs_xattr_handlers[] = {
  	&ecryptfs_xattr_handler,
  	NULL

  };