/*

   *  linux/fs/ext4/file.c

   *
   * Copyright (C) 1992, 1993, 1994, 1995
   * Remy Card (card@masi.ibp.fr)
   * Laboratoire MASI - Institut Blaise Pascal
   * Universite Pierre et Marie Curie (Paris VI)
   *
   *  from
   *
   *  linux/fs/minix/file.c
   *
   *  Copyright (C) 1991, 1992  Linus Torvalds
   *

   *  ext4 fs regular file handling primitives

   *
   *  64-bit file support on 64-bit platforms by Jakub Jelinek
   *	(jj@sunsite.ms.mff.cuni.cz)
   */
  
  #include <linux/time.h>
  #include <linux/fs.h>

  #include <linux/mount.h>
  #include <linux/path.h>

  #include <linux/dax.h>

  #include <linux/quotaops.h>

  #include <linux/pagevec.h>

  #include <linux/uio.h>

  #include "ext4.h"
  #include "ext4_jbd2.h"

  #include "xattr.h"
  #include "acl.h"
  
  /*
   * Called when an inode is released. Note that this is different

   * from ext4_file_open: open gets called at every open, but release

   * gets called only when /all/ the files are closed.
   */

  static int ext4_release_file(struct inode *inode, struct file *filp)

  {

  	if (ext4_test_inode_state(inode, EXT4_STATE_DA_ALLOC_CLOSE)) {

  		ext4_alloc_da_blocks(inode);

  		ext4_clear_inode_state(inode, EXT4_STATE_DA_ALLOC_CLOSE);

  	}

  	/* if we are the last writer on the inode, drop the block reservation */
  	if ((filp->f_mode & FMODE_WRITE) &&

  			(atomic_read(&inode->i_writecount) == 1) &&
  		        !EXT4_I(inode)->i_reserved_data_blocks)

  	{

  		down_write(&EXT4_I(inode)->i_data_sem);

  		ext4_discard_preallocations(inode);

  		up_write(&EXT4_I(inode)->i_data_sem);

  	}
  	if (is_dx(inode) && filp->private_data)

  		ext4_htree_free_dir_info(filp->private_data);

  
  	return 0;
  }

  static void ext4_unwritten_wait(struct inode *inode)

  {
  	wait_queue_head_t *wq = ext4_ioend_wq(inode);

  	wait_event(*wq, (atomic_read(&EXT4_I(inode)->i_unwritten) == 0));

  }
  
  /*
   * This tests whether the IO in question is block-aligned or not.
   * Ext4 utilizes unwritten extents when hole-filling during direct IO, and they
   * are converted to written only after the IO is complete.  Until they are
   * mapped, these blocks appear as holes, so dio_zero_block() will assume that
   * it needs to zero out portions of the start and/or end block.  If 2 AIO
   * threads are at work on the same unwritten block, they must be synchronized
   * or one thread will zero the other's data, causing corruption.
   */
  static int

  ext4_unaligned_aio(struct inode *inode, struct iov_iter *from, loff_t pos)

  {
  	struct super_block *sb = inode->i_sb;
  	int blockmask = sb->s_blocksize - 1;

  	if (pos >= i_size_read(inode))

  		return 0;

  	if ((pos | iov_iter_alignment(from)) & blockmask)

  		return 1;
  
  	return 0;
  }

  static ssize_t

  ext4_file_write_iter(struct kiocb *iocb, struct iov_iter *from)

  {

  	struct inode *inode = file_inode(iocb->ki_filp);

  	int o_direct = iocb->ki_flags & IOCB_DIRECT;

  	int unaligned_aio = 0;

  	int overwrite = 0;

  	ssize_t ret;

  	inode_lock(inode);
  	ret = generic_write_checks(iocb, from);
  	if (ret <= 0)
  		goto out;

  	/*

  	 * Unaligned direct AIO must be serialized among each other as zeroing
  	 * of partial blocks of two competing unaligned AIOs can result in data
  	 * corruption.

  	 */

  	if (o_direct && ext4_test_inode_flag(inode, EXT4_INODE_EXTENTS) &&

  	    !is_sync_kiocb(iocb) &&

  	    ext4_unaligned_aio(inode, from, iocb->ki_pos)) {
  		unaligned_aio = 1;

  		ext4_unwritten_wait(inode);
  	}

  	/*

  	 * If we have encountered a bitmap-format file, the size limit
  	 * is smaller than s_maxbytes, which is for extent-mapped files.
  	 */

  	if (!(ext4_test_inode_flag(inode, EXT4_INODE_EXTENTS))) {

  		struct ext4_sb_info *sbi = EXT4_SB(inode->i_sb);

  		if (iocb->ki_pos >= sbi->s_bitmap_maxbytes) {

  			ret = -EFBIG;

  			goto out;

  		}

  		iov_iter_truncate(from, sbi->s_bitmap_maxbytes - iocb->ki_pos);

  	}

  	iocb->private = &overwrite;

  	if (o_direct) {

  		size_t length = iov_iter_count(from);
  		loff_t pos = iocb->ki_pos;

  		/* check whether we do a DIO overwrite or not */

  		if (ext4_should_dioread_nolock(inode) && !unaligned_aio &&

  		    pos + length <= i_size_read(inode)) {

  			struct ext4_map_blocks map;
  			unsigned int blkbits = inode->i_blkbits;
  			int err, len;
  
  			map.m_lblk = pos >> blkbits;

  			map.m_len = EXT4_MAX_BLOCKS(length, pos, blkbits);

  			len = map.m_len;
  
  			err = ext4_map_blocks(NULL, inode, &map, 0);
  			/*
  			 * 'err==len' means that all of blocks has
  			 * been preallocated no matter they are
  			 * initialized or not.  For excluding
  			 * unwritten extents, we need to check
  			 * m_flags.  There are two conditions that
  			 * indicate for initialized extents.  1) If we
  			 * hit extent cache, EXT4_MAP_MAPPED flag is
  			 * returned; 2) If we do a real lookup,
  			 * non-flags are returned.  So we should check
  			 * these two conditions.
  			 */
  			if (err == len && (map.m_flags & EXT4_MAP_MAPPED))
  				overwrite = 1;
  		}

  	}

  	ret = __generic_file_write_iter(iocb, from);

  	inode_unlock(inode);

  	if (ret > 0)
  		ret = generic_write_sync(iocb, ret);

  	return ret;
  
  out:

  	inode_unlock(inode);

  	return ret;

  }

  #ifdef CONFIG_FS_DAX
  static int ext4_dax_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
  {

  	int result;
  	handle_t *handle = NULL;

  	struct inode *inode = file_inode(vma->vm_file);
  	struct super_block *sb = inode->i_sb;

  	bool write = vmf->flags & FAULT_FLAG_WRITE;
  
  	if (write) {
  		sb_start_pagefault(sb);
  		file_update_time(vma->vm_file);

  		down_read(&EXT4_I(inode)->i_mmap_sem);

  		handle = ext4_journal_start_sb(sb, EXT4_HT_WRITE_PAGE,
  						EXT4_DATA_TRANS_BLOCKS(sb));

  	} else
  		down_read(&EXT4_I(inode)->i_mmap_sem);

  
  	if (IS_ERR(handle))
  		result = VM_FAULT_SIGBUS;
  	else

  		result = dax_fault(vma, vmf, ext4_dax_get_block);

  
  	if (write) {
  		if (!IS_ERR(handle))
  			ext4_journal_stop(handle);

  		up_read(&EXT4_I(inode)->i_mmap_sem);

  		sb_end_pagefault(sb);

  	} else
  		up_read(&EXT4_I(inode)->i_mmap_sem);

  
  	return result;

  }

  static int ext4_dax_pmd_fault(struct vm_area_struct *vma, unsigned long addr,
  						pmd_t *pmd, unsigned int flags)
  {

  	int result;
  	handle_t *handle = NULL;
  	struct inode *inode = file_inode(vma->vm_file);
  	struct super_block *sb = inode->i_sb;
  	bool write = flags & FAULT_FLAG_WRITE;
  
  	if (write) {
  		sb_start_pagefault(sb);
  		file_update_time(vma->vm_file);

  		down_read(&EXT4_I(inode)->i_mmap_sem);

  		handle = ext4_journal_start_sb(sb, EXT4_HT_WRITE_PAGE,
  				ext4_chunk_trans_blocks(inode,
  							PMD_SIZE / PAGE_SIZE));

  	} else
  		down_read(&EXT4_I(inode)->i_mmap_sem);

  
  	if (IS_ERR(handle))
  		result = VM_FAULT_SIGBUS;
  	else

  		result = dax_pmd_fault(vma, addr, pmd, flags,

  					 ext4_dax_get_block);

  
  	if (write) {
  		if (!IS_ERR(handle))
  			ext4_journal_stop(handle);

  		up_read(&EXT4_I(inode)->i_mmap_sem);

  		sb_end_pagefault(sb);

  	} else
  		up_read(&EXT4_I(inode)->i_mmap_sem);

  
  	return result;

  }

  /*

   * Handle write fault for VM_MIXEDMAP mappings. Similarly to ext4_dax_fault()

   * handler we check for races agaist truncate. Note that since we cycle through
   * i_mmap_sem, we are sure that also any hole punching that began before we
   * were called is finished by now and so if it included part of the file we
   * are working on, our pte will get unmapped and the check for pte_same() in
   * wp_pfn_shared() fails. Thus fault gets retried and things work out as
   * desired.
   */
  static int ext4_dax_pfn_mkwrite(struct vm_area_struct *vma,
  				struct vm_fault *vmf)
  {
  	struct inode *inode = file_inode(vma->vm_file);
  	struct super_block *sb = inode->i_sb;

  	loff_t size;

  	int ret;

  
  	sb_start_pagefault(sb);
  	file_update_time(vma->vm_file);
  	down_read(&EXT4_I(inode)->i_mmap_sem);
  	size = (i_size_read(inode) + PAGE_SIZE - 1) >> PAGE_SHIFT;
  	if (vmf->pgoff >= size)
  		ret = VM_FAULT_SIGBUS;

  	else
  		ret = dax_pfn_mkwrite(vma, vmf);

  	up_read(&EXT4_I(inode)->i_mmap_sem);
  	sb_end_pagefault(sb);
  
  	return ret;

  }
  
  static const struct vm_operations_struct ext4_dax_vm_ops = {
  	.fault		= ext4_dax_fault,

  	.pmd_fault	= ext4_dax_pmd_fault,

  	.page_mkwrite	= ext4_dax_fault,

  	.pfn_mkwrite	= ext4_dax_pfn_mkwrite,

  };
  #else
  #define ext4_dax_vm_ops	ext4_file_vm_ops
  #endif

  static const struct vm_operations_struct ext4_file_vm_ops = {

  	.fault		= ext4_filemap_fault,

  	.map_pages	= filemap_map_pages,

  	.page_mkwrite   = ext4_page_mkwrite,
  };
  
  static int ext4_file_mmap(struct file *file, struct vm_area_struct *vma)
  {

  	struct inode *inode = file->f_mapping->host;
  
  	if (ext4_encrypted_inode(inode)) {

  		int err = fscrypt_get_encryption_info(inode);

  		if (err)
  			return 0;

  		if (!fscrypt_has_encryption_key(inode))

  			return -ENOKEY;

  	}

  	file_accessed(file);

  	if (IS_DAX(file_inode(file))) {
  		vma->vm_ops = &ext4_dax_vm_ops;

  		vma->vm_flags |= VM_MIXEDMAP | VM_HUGEPAGE;

  	} else {
  		vma->vm_ops = &ext4_file_vm_ops;
  	}

  	return 0;
  }

  static int ext4_file_open(struct inode * inode, struct file * filp)
  {
  	struct super_block *sb = inode->i_sb;
  	struct ext4_sb_info *sbi = EXT4_SB(inode->i_sb);
  	struct vfsmount *mnt = filp->f_path.mnt;

  	struct dentry *dir;

  	struct path path;
  	char buf[64], *cp;

  	int ret;

  
  	if (unlikely(!(sbi->s_mount_flags & EXT4_MF_MNTDIR_SAMPLED) &&
  		     !(sb->s_flags & MS_RDONLY))) {
  		sbi->s_mount_flags |= EXT4_MF_MNTDIR_SAMPLED;
  		/*
  		 * Sample where the filesystem has been mounted and
  		 * store it in the superblock for sysadmin convenience
  		 * when trying to sort through large numbers of block
  		 * devices or filesystem images.
  		 */
  		memset(buf, 0, sizeof(buf));

  		path.mnt = mnt;
  		path.dentry = mnt->mnt_root;

  		cp = d_path(&path, buf, sizeof(buf));

  		if (!IS_ERR(cp)) {

  			handle_t *handle;
  			int err;

  			handle = ext4_journal_start_sb(sb, EXT4_HT_MISC, 1);

  			if (IS_ERR(handle))
  				return PTR_ERR(handle);

  			BUFFER_TRACE(sbi->s_sbh, "get_write_access");

  			err = ext4_journal_get_write_access(handle, sbi->s_sbh);
  			if (err) {
  				ext4_journal_stop(handle);
  				return err;
  			}

  			strlcpy(sbi->s_es->s_last_mounted, cp,
  				sizeof(sbi->s_es->s_last_mounted));

  			ext4_handle_dirty_super(handle, sb);
  			ext4_journal_stop(handle);

  		}
  	}

  	if (ext4_encrypted_inode(inode)) {

  		ret = fscrypt_get_encryption_info(inode);

  		if (ret)
  			return -EACCES;

  		if (!fscrypt_has_encryption_key(inode))

  			return -ENOKEY;
  	}

  	dir = dget_parent(file_dentry(filp));

  	if (ext4_encrypted_inode(d_inode(dir)) &&

  			!fscrypt_has_permitted_context(d_inode(dir), inode)) {

  		ext4_warning(inode->i_sb,

  			     "Inconsistent encryption contexts: %lu/%lu",

  			     (unsigned long) d_inode(dir)->i_ino,

  			     (unsigned long) inode->i_ino);

  		dput(dir);

  		return -EPERM;
  	}

  	dput(dir);

  	/*
  	 * Set up the jbd2_inode if we are opening the inode for
  	 * writing and the journal is present
  	 */

  	if (filp->f_mode & FMODE_WRITE) {

  		ret = ext4_inode_attach_jinode(inode);

  		if (ret < 0)
  			return ret;

  	}

  	return dquot_file_open(inode, filp);

  }

  /*

   * Here we use ext4_map_blocks() to get a block mapping for a extent-based
   * file rather than ext4_ext_walk_space() because we can introduce
   * SEEK_DATA/SEEK_HOLE for block-mapped and extent-mapped file at the same
   * function.  When extent status tree has been fully implemented, it will
   * track all extent status for a file and we can directly use it to
   * retrieve the offset for SEEK_DATA/SEEK_HOLE.
   */
  
  /*
   * When we retrieve the offset for SEEK_DATA/SEEK_HOLE, we would need to
   * lookup page cache to check whether or not there has some data between
   * [startoff, endoff] because, if this range contains an unwritten extent,
   * we determine this extent as a data or a hole according to whether the
   * page cache has data or not.
   */

  static int ext4_find_unwritten_pgoff(struct inode *inode,
  				     int whence,

  				     ext4_lblk_t end_blk,

  				     loff_t *offset)

  {
  	struct pagevec pvec;

  	unsigned int blkbits;

  	pgoff_t index;
  	pgoff_t end;

  	loff_t endoff;

  	loff_t startoff;
  	loff_t lastoff;
  	int found = 0;

  	blkbits = inode->i_sb->s_blocksize_bits;

  	startoff = *offset;
  	lastoff = startoff;

  	endoff = (loff_t)end_blk << blkbits;

  	index = startoff >> PAGE_SHIFT;
  	end = endoff >> PAGE_SHIFT;

  
  	pagevec_init(&pvec, 0);
  	do {
  		int i, num;
  		unsigned long nr_pages;
  
  		num = min_t(pgoff_t, end - index, PAGEVEC_SIZE);
  		nr_pages = pagevec_lookup(&pvec, inode->i_mapping, index,
  					  (pgoff_t)num);
  		if (nr_pages == 0) {

  			if (whence == SEEK_DATA)

  				break;

  			BUG_ON(whence != SEEK_HOLE);

  			/*
  			 * If this is the first time to go into the loop and
  			 * offset is not beyond the end offset, it will be a
  			 * hole at this offset
  			 */
  			if (lastoff == startoff || lastoff < endoff)
  				found = 1;
  			break;
  		}
  
  		/*
  		 * If this is the first time to go into the loop and
  		 * offset is smaller than the first page offset, it will be a
  		 * hole at this offset.
  		 */

  		if (lastoff == startoff && whence == SEEK_HOLE &&

  		    lastoff < page_offset(pvec.pages[0])) {
  			found = 1;
  			break;
  		}
  
  		for (i = 0; i < nr_pages; i++) {
  			struct page *page = pvec.pages[i];
  			struct buffer_head *bh, *head;
  
  			/*
  			 * If the current offset is not beyond the end of given
  			 * range, it will be a hole.
  			 */

  			if (lastoff < endoff && whence == SEEK_HOLE &&

  			    page->index > end) {
  				found = 1;
  				*offset = lastoff;
  				goto out;
  			}
  
  			lock_page(page);
  
  			if (unlikely(page->mapping != inode->i_mapping)) {
  				unlock_page(page);
  				continue;
  			}
  
  			if (!page_has_buffers(page)) {
  				unlock_page(page);
  				continue;
  			}
  
  			if (page_has_buffers(page)) {
  				lastoff = page_offset(page);
  				bh = head = page_buffers(page);
  				do {
  					if (buffer_uptodate(bh) ||
  					    buffer_unwritten(bh)) {

  						if (whence == SEEK_DATA)

  							found = 1;
  					} else {

  						if (whence == SEEK_HOLE)

  							found = 1;
  					}
  					if (found) {
  						*offset = max_t(loff_t,
  							startoff, lastoff);
  						unlock_page(page);
  						goto out;
  					}
  					lastoff += bh->b_size;
  					bh = bh->b_this_page;
  				} while (bh != head);
  			}
  
  			lastoff = page_offset(page) + PAGE_SIZE;
  			unlock_page(page);
  		}
  
  		/*
  		 * The no. of pages is less than our desired, that would be a
  		 * hole in there.
  		 */

  		if (nr_pages < num && whence == SEEK_HOLE) {

  			found = 1;
  			*offset = lastoff;
  			break;
  		}
  
  		index = pvec.pages[i - 1]->index + 1;
  		pagevec_release(&pvec);
  	} while (index <= end);
  
  out:
  	pagevec_release(&pvec);
  	return found;
  }
  
  /*
   * ext4_seek_data() retrieves the offset for SEEK_DATA.
   */
  static loff_t ext4_seek_data(struct file *file, loff_t offset, loff_t maxsize)
  {
  	struct inode *inode = file->f_mapping->host;

  	struct extent_status es;
  	ext4_lblk_t start, last, end;
  	loff_t dataoff, isize;
  	int blkbits;

  	int ret;

  	inode_lock(inode);

  
  	isize = i_size_read(inode);
  	if (offset >= isize) {

  		inode_unlock(inode);

  		return -ENXIO;
  	}

  
  	blkbits = inode->i_sb->s_blocksize_bits;
  	start = offset >> blkbits;
  	last = start;
  	end = isize >> blkbits;
  	dataoff = offset;
  
  	do {

  		ret = ext4_get_next_extent(inode, last, end - last + 1, &es);
  		if (ret <= 0) {
  			/* No extent found -> no data */
  			if (ret == 0)
  				ret = -ENXIO;
  			inode_unlock(inode);
  			return ret;

  		}

  		last = es.es_lblk;
  		if (last != start)
  			dataoff = (loff_t)last << blkbits;
  		if (!ext4_es_is_unwritten(&es))

  			break;

  		/*
  		 * If there is a unwritten extent at this offset,
  		 * it will be as a data or a hole according to page
  		 * cache that has data or not.
  		 */

  		if (ext4_find_unwritten_pgoff(inode, SEEK_DATA,
  					      es.es_lblk + es.es_len, &dataoff))
  			break;
  		last += es.es_len;

  		dataoff = (loff_t)last << blkbits;

  		cond_resched();

  	} while (last <= end);

  	inode_unlock(inode);

  	if (dataoff > isize)
  		return -ENXIO;
  
  	return vfs_setpos(file, dataoff, maxsize);

  }
  
  /*

   * ext4_seek_hole() retrieves the offset for SEEK_HOLE.

   */
  static loff_t ext4_seek_hole(struct file *file, loff_t offset, loff_t maxsize)
  {
  	struct inode *inode = file->f_mapping->host;

  	struct extent_status es;
  	ext4_lblk_t start, last, end;
  	loff_t holeoff, isize;
  	int blkbits;

  	int ret;

  	inode_lock(inode);

  
  	isize = i_size_read(inode);
  	if (offset >= isize) {

  		inode_unlock(inode);

  		return -ENXIO;
  	}

  	blkbits = inode->i_sb->s_blocksize_bits;
  	start = offset >> blkbits;
  	last = start;
  	end = isize >> blkbits;
  	holeoff = offset;

  	do {

  		ret = ext4_get_next_extent(inode, last, end - last + 1, &es);
  		if (ret < 0) {
  			inode_unlock(inode);
  			return ret;

  		}

  		/* Found a hole? */
  		if (ret == 0 || es.es_lblk > last) {
  			if (last != start)
  				holeoff = (loff_t)last << blkbits;
  			break;

  		}

  		/*
  		 * If there is a unwritten extent at this offset,
  		 * it will be as a data or a hole according to page
  		 * cache that has data or not.
  		 */

  		if (ext4_es_is_unwritten(&es) &&
  		    ext4_find_unwritten_pgoff(inode, SEEK_HOLE,
  					      last + es.es_len, &holeoff))
  			break;

  		last += es.es_len;
  		holeoff = (loff_t)last << blkbits;
  		cond_resched();

  	} while (last <= end);

  	inode_unlock(inode);

  	if (holeoff > isize)
  		holeoff = isize;
  
  	return vfs_setpos(file, holeoff, maxsize);

  }
  
  /*

   * ext4_llseek() handles both block-mapped and extent-mapped maxbytes values
   * by calling generic_file_llseek_size() with the appropriate maxbytes
   * value for each.

   */

  loff_t ext4_llseek(struct file *file, loff_t offset, int whence)

  {
  	struct inode *inode = file->f_mapping->host;
  	loff_t maxbytes;
  
  	if (!(ext4_test_inode_flag(inode, EXT4_INODE_EXTENTS)))
  		maxbytes = EXT4_SB(inode->i_sb)->s_bitmap_maxbytes;
  	else
  		maxbytes = inode->i_sb->s_maxbytes;

  	switch (whence) {

  	case SEEK_SET:
  	case SEEK_CUR:
  	case SEEK_END:

  		return generic_file_llseek_size(file, offset, whence,

  						maxbytes, i_size_read(inode));
  	case SEEK_DATA:
  		return ext4_seek_data(file, offset, maxbytes);
  	case SEEK_HOLE:
  		return ext4_seek_hole(file, offset, maxbytes);
  	}
  
  	return -EINVAL;

  }

  const struct file_operations ext4_file_operations = {

  	.llseek		= ext4_llseek,

  	.read_iter	= generic_file_read_iter,

  	.write_iter	= ext4_file_write_iter,

  	.unlocked_ioctl = ext4_ioctl,

  #ifdef CONFIG_COMPAT

  	.compat_ioctl	= ext4_compat_ioctl,

  #endif

  	.mmap		= ext4_file_mmap,

  	.open		= ext4_file_open,

  	.release	= ext4_release_file,
  	.fsync		= ext4_sync_file,

  	.get_unmapped_area = thp_get_unmapped_area,

  	.splice_read	= generic_file_splice_read,

  	.splice_write	= iter_file_splice_write,

  	.fallocate	= ext4_fallocate,

  };

  const struct inode_operations ext4_file_inode_operations = {

  	.setattr	= ext4_setattr,

  	.getattr	= ext4_getattr,

  	.listxattr	= ext4_listxattr,

  	.get_acl	= ext4_get_acl,

  	.set_acl	= ext4_set_acl,

  	.fiemap		= ext4_fiemap,

  };