/*
 * Copyright (c) 2012 Frantisek Princ
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/** @addtogroup libext4
 * @{
 */ 

/**
 * @file	libext4_inode.c
 * @brief	Ext4 inode structure operations.
 */

#include <byteorder.h>
#include <errno.h>
#include <libblock.h>
#include "libext4.h"

/** Compute number of bits for block count.
 *
 *  @param block_size	filesystem block_size
 *  @return		number of bits
 */
static uint32_t ext4_inode_block_bits_count(uint32_t block_size)
{
	uint32_t bits = 8;
	uint32_t size = block_size;

	do {
		bits++;
		size = size >> 1;
	} while (size > 256);

	return bits;
}

/** Get mode of the i-node.
 *
 * @param sb		superblock
 * @param inode		i-node to load mode from
 * @return 		mode of the i-node
 */
uint32_t ext4_inode_get_mode(ext4_superblock_t *sb, ext4_inode_t *inode)
{
	if (ext4_superblock_get_creator_os(sb) == EXT4_SUPERBLOCK_OS_HURD) {
		return ((uint32_t)uint16_t_le2host(inode->osd2.hurd2.mode_high)) << 16 |
		    ((uint32_t)uint16_t_le2host(inode->mode));
	}
	return uint16_t_le2host(inode->mode);
}

/** Set mode of the i-node.
 *
 * @param sb		superblock
 * @param inode		i-node to set mode to
 * @param mode		mode to set to i-node
 */
void ext4_inode_set_mode(ext4_superblock_t *sb, ext4_inode_t *inode, uint32_t mode)
{
	inode->mode = host2uint16_t_le((mode << 16) >> 16);

	if (ext4_superblock_get_creator_os(sb) == EXT4_SUPERBLOCK_OS_HURD) {
		inode->osd2.hurd2.mode_high = host2uint16_t_le(mode >> 16);
	}
}

/** Get ID of the i-node owner (user id).
 *
 * @param inode		i-node to load uid from
 * @return		user ID of the i-node owner
 */
uint32_t ext4_inode_get_uid(ext4_inode_t *inode)
{
	return uint32_t_le2host(inode->uid);
}

/** Set ID of the i-node owner.
 *
 * @param inode		i-node to set uid to
 * @param uid		ID of the i-node owner
 */
void ext4_inode_set_uid(ext4_inode_t *inode, uint32_t uid)
{
	inode->uid = host2uint32_t_le(uid);
}

/** Get real i-node size.
 *
 * @param sb		superblock
 * @param inode		i-node to load size from
 * @return		real size of i-node
 */
uint64_t ext4_inode_get_size(ext4_superblock_t *sb, ext4_inode_t *inode)
{
	uint32_t major_rev = ext4_superblock_get_rev_level(sb);

	if ((major_rev > 0) && ext4_inode_is_type(sb, inode, EXT4_INODE_MODE_FILE)) {
		return ((uint64_t)uint32_t_le2host(inode->size_hi)) << 32 |
			    ((uint64_t)uint32_t_le2host(inode->size_lo));
	}
	return uint32_t_le2host(inode->size_lo);
}

/** Set real i-node size.
 *
 *  @param inode	inode to set size to
 *  @param size		size of the i-node
 */
void ext4_inode_set_size(ext4_inode_t *inode, uint64_t size) {
	inode->size_lo = host2uint32_t_le((size << 32) >> 32);
	inode->size_hi = host2uint32_t_le(size >> 32);
}

uint32_t ext4_inode_get_access_time(ext4_inode_t *inode)
{
	return uint32_t_le2host(inode->access_time);
}

/** TODO comment
 *
 */
void ext4_inode_set_access_time(ext4_inode_t *inode, uint32_t time)
{
	inode->access_time = host2uint32_t_le(time);
}

/** TODO comment
 *
 */
uint32_t ext4_inode_get_change_inode_time(ext4_inode_t *inode)
{
	return uint32_t_le2host(inode->change_inode_time);
}

/** TODO comment
 *
 */
void ext4_inode_set_change_inode_time(ext4_inode_t *inode, uint32_t time)
{
	inode->change_inode_time = host2uint32_t_le(time);
}

/** TODO comment
 *
 */
uint32_t ext4_inode_get_modification_time(ext4_inode_t *inode)
{
	return uint32_t_le2host(inode->modification_time);
}

/** TODO comment
 *
 */
void ext4_inode_set_modification_time(ext4_inode_t *inode, uint32_t time)
{
	inode->modification_time = host2uint32_t_le(time);
}

/** TODO comment
 *
 */
uint32_t ext4_inode_get_deletion_time(ext4_inode_t *inode)
{
	return uint32_t_le2host(inode->deletion_time);
}

/** TODO comment
 *
 */
void ext4_inode_set_deletion_time(ext4_inode_t *inode, uint32_t time)
{
	inode->deletion_time = host2uint32_t_le(time);
}

/** Get ID of the i-node owner's group.
 *
 * @param inode         i-node to load gid from
 * @return              group ID of the i-node owner
 */
uint32_t ext4_inode_get_gid(ext4_inode_t *inode)
{
	return uint32_t_le2host(inode->gid);
}

/** Set ID ot the i-node owner's group.
 *
 * @param inode		i-node to set gid to
 * @param gid		group ID of the i-node owner
 */
void ext4_inode_set_gid(ext4_inode_t *inode, uint32_t gid)
{
	inode->gid = host2uint32_t_le(gid);
}

/** Get number of links to i-node.
 *
 * @param inode 	i-node to load number of links from
 * @return		number of links to i-node
 */
uint16_t ext4_inode_get_links_count(ext4_inode_t *inode)
{
	return uint16_t_le2host(inode->links_count);
}

/** Set number of links to i-node.
 *
 * @param inode 	i-node to set number of links to
 * @param count		number of links to i-node
 */
void ext4_inode_set_links_count(ext4_inode_t *inode, uint16_t count)
{
	inode->links_count = host2uint16_t_le(count);
}

/** TODO comment
 *
 */
uint64_t ext4_inode_get_blocks_count(ext4_superblock_t *sb, ext4_inode_t *inode)
{
	if (ext4_superblock_has_feature_read_only(sb, EXT4_FEATURE_RO_COMPAT_HUGE_FILE)) {

		// 48-bit field
		uint64_t count = ((uint64_t)uint16_t_le2host(inode->osd2.linux2.blocks_high)) << 32 |
				uint32_t_le2host(inode->blocks_count_lo);

		if (ext4_inode_has_flag(inode, EXT4_INODE_FLAG_HUGE_FILE)) {
	    	uint32_t block_size = ext4_superblock_get_block_size(sb);
	    	uint32_t block_bits = ext4_inode_block_bits_count(block_size);
			return count  << (block_bits - 9);
		} else {
			return count;
		}
	} else {
		return uint32_t_le2host(inode->blocks_count_lo);
    }

}


/** TODO comment
 *
 */
int ext4_inode_set_blocks_count(ext4_superblock_t *sb, ext4_inode_t *inode,
		uint64_t count)
{
    // 32-bit maximum
    uint64_t max = 0;
    max = ~max >> 32;

    if (count <= max) {
    	inode->blocks_count_lo = host2uint32_t_le(count);
    	inode->osd2.linux2.blocks_high = 0;
    	ext4_inode_clear_flag(inode, EXT4_INODE_FLAG_HUGE_FILE);
    	return EOK;
    }

    if (!ext4_superblock_has_feature_read_only(sb, EXT4_FEATURE_RO_COMPAT_HUGE_FILE)) {
    	return EINVAL;
    }

    // 48-bit maximum
    max = 0;
    max = ~max >> 16;

    if (count <= max) {
    	inode->blocks_count_lo = host2uint32_t_le(count);
    	inode->osd2.linux2.blocks_high = host2uint16_t_le(count >> 32);
    	ext4_inode_clear_flag(inode, EXT4_INODE_FLAG_HUGE_FILE);
    } else {
    	uint32_t block_size = ext4_superblock_get_block_size(sb);
    	uint32_t block_bits = ext4_inode_block_bits_count(block_size);
    	ext4_inode_set_flag(inode, EXT4_INODE_FLAG_HUGE_FILE);
    	count = count >> (block_bits - 9);
    	inode->blocks_count_lo = host2uint32_t_le(count);
    	inode->osd2.linux2.blocks_high = host2uint16_t_le(count >> 32);
    }
    return EOK;
}

/** Get flags (features) of i-node.
 * 
 * @param inode		i-node to get flags from
 * @return		flags (bitmap)
 */
uint32_t ext4_inode_get_flags(ext4_inode_t *inode) {
	return uint32_t_le2host(inode->flags);
}

/** Set flags (features) of i-node.
 *
 * @param inode		i-node to set flags to
 * @param flags		flags to set to i-node
 */
void ext4_inode_set_flags(ext4_inode_t *inode, uint32_t flags) {
	inode->flags = host2uint32_t_le(flags);
}

/** TODO comment
 *
 */
uint32_t ext4_inode_get_generation(ext4_inode_t *inode)
{
	return uint32_t_le2host(inode->generation);
}

/** TODO comment
 *
 */
void ext4_inode_set_generation(ext4_inode_t *inode, uint32_t generation)
{
	inode->generation = host2uint32_t_le(generation);
}

/** TODO comment
 *
 */
uint64_t ext4_inode_get_file_acl(ext4_inode_t *inode, ext4_superblock_t *sb)
{
	if (ext4_superblock_get_creator_os(sb) == EXT4_SUPERBLOCK_OS_LINUX) {
		return ((uint32_t)uint16_t_le2host(inode->osd2.linux2.file_acl_high)) << 16 |
		    (uint32_t_le2host(inode->file_acl_lo));
	}

	return uint32_t_le2host(inode->file_acl_lo);
}

/** TODO comment
 *
 */
void ext4_inode_set_file_acl(ext4_inode_t *inode, ext4_superblock_t *sb,
		uint64_t file_acl)
{
	inode->file_acl_lo = host2uint32_t_le((file_acl << 32) >> 32);

	if (ext4_superblock_get_creator_os(sb) == EXT4_SUPERBLOCK_OS_LINUX) {
		inode->osd2.linux2.file_acl_high = host2uint16_t_le(file_acl >> 32);
	}
}

/***********************************************************************/

/** Get block address of specified direct block.
 *
 * @param inode		i-node to load block from
 * @param idx		index of logical block
 * @return		physical block address
 */
uint32_t ext4_inode_get_direct_block(ext4_inode_t *inode, uint32_t idx)
{
	assert(idx < EXT4_INODE_DIRECT_BLOCK_COUNT);
	return uint32_t_le2host(inode->blocks[idx]);
}

/** Set block address of specified direct block.
 *
 * @param inode		i-node to set block address to
 * @param idx		index of logical block
 * @param fblock	physical block address
 */
void ext4_inode_set_direct_block(ext4_inode_t *inode, uint32_t idx, uint32_t fblock)
{
	assert(idx < EXT4_INODE_DIRECT_BLOCK_COUNT);
	inode->blocks[idx] = host2uint32_t_le(fblock);
}

/** Get block address of specified indirect block.
 *
 * @param inode		i-node to get block address from
 * @param idx		index of indirect block 
 * @return		physical block address
 */
uint32_t ext4_inode_get_indirect_block(ext4_inode_t *inode, uint32_t idx)
{
	return uint32_t_le2host(inode->blocks[idx + EXT4_INODE_INDIRECT_BLOCK]);
}

/** Set block address of specified indirect block.
 *
 * @param inode		i-node to set block address to
 * @param idx		index of indirect block
 * @param fblock	physical block address
 */
void ext4_inode_set_indirect_block(ext4_inode_t *inode, uint32_t idx, uint32_t fblock)
{
	inode->blocks[idx + EXT4_INODE_INDIRECT_BLOCK] = host2uint32_t_le(fblock);
}

/** Check if i-node has specified type.
 *
 * @param sb		superblock
 * @param inode		i-node to check type of
 * @param type		type to check
 * @return 		result of check operation
 */
bool ext4_inode_is_type(ext4_superblock_t *sb, ext4_inode_t *inode, uint32_t type)
{
	uint32_t mode = ext4_inode_get_mode(sb, inode);
	return (mode & EXT4_INODE_MODE_TYPE_MASK) == type;
}

/** Get extent header from the root of the extent tree.
 *
 * @param inode		i-node to get extent header from
 * @return		pointer to extent header of the root node
 */
ext4_extent_header_t * ext4_inode_get_extent_header(ext4_inode_t *inode)
{
	return (ext4_extent_header_t *)inode->blocks;
}

/** Check if i-node has specified flag.
 *
 * @param inode		i-node to check flags of
 * @param flag		flag to check
 * @return		result of check operation
 */
bool ext4_inode_has_flag(ext4_inode_t *inode, uint32_t flag)
{
	if (ext4_inode_get_flags(inode) & flag) {
		return true;
	}
	return false;
}

/** Remove specified flag from i-node.
 *
 * @param inode		i-node to clear flag on
 * @param clear_flag	flag to be cleared
 */
void ext4_inode_clear_flag(ext4_inode_t *inode, uint32_t clear_flag)
{
	uint32_t flags = ext4_inode_get_flags(inode);
	flags = flags & (~clear_flag);
	ext4_inode_set_flags(inode, flags);
}

/** Set specified flag to i-node.
 *
 * @param inode		i-node to set flag on
 * @param set_flag	falt to be set
 */
void ext4_inode_set_flag(ext4_inode_t *inode, uint32_t set_flag)
{
	uint32_t flags = ext4_inode_get_flags(inode);
	flags = flags | set_flag;
	ext4_inode_set_flags(inode, flags);
}

/** Check if i-node can be truncated.
 *
 * @param sb		superblock
 * @param inode		i-node to check
 * @return 		result of the check operation
 */
bool ext4_inode_can_truncate(ext4_superblock_t *sb, ext4_inode_t *inode)
{
	 if (ext4_inode_has_flag(inode, EXT4_INODE_FLAG_APPEND)
			 || ext4_inode_has_flag(inode, EXT4_INODE_FLAG_IMMUTABLE)) {
		 return false;
	 }

	 if (ext4_inode_is_type(sb, inode, EXT4_INODE_MODE_FILE)
			 || ext4_inode_is_type(sb, inode, EXT4_INODE_MODE_DIRECTORY)) {
		 return true;
	 }

	 return false;
}

/**
 * @}
 */