/*
 * Copyright (c) 2013 Grzegorz Kostka (kostka.grzegorz@gmail.com)
 *
 *
 * HelenOS:
 * Copyright (c) 2012 Martin Sucha
 * 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 lwext4
 * @{
 */
/**
 * @file  ext4_dir.h
 * @brief Directory handle procedures.
 */

#ifndef EXT4_DIR_H_
#define EXT4_DIR_H_

#ifdef __cplusplus
extern "C" {
#endif

#include <ext4_config.h>
#include <ext4_types.h>
#include <ext4_misc.h>
#include <ext4_blockdev.h>
#include <ext4_super.h>

#include <stdint.h>

struct ext4_dir_iter {
	struct ext4_inode_ref *inode_ref;
	struct ext4_block curr_blk;
	uint64_t curr_off;
	struct ext4_dir_en *curr;
};

struct ext4_dir_search_result {
	struct ext4_block block;
	struct ext4_dir_en *dentry;
};


/**@brief Get i-node number from directory entry.
 * @param de Directory entry
 * @return I-node number
 */
static inline uint32_t
ext4_dir_en_get_inode(struct ext4_dir_en *de)
{
	return to_le32(de->inode);
}

/**@brief Set i-node number to directory entry.
 * @param de Directory entry
 * @param inode I-node number
 */
static inline void
ext4_dir_en_set_inode(struct ext4_dir_en *de, uint32_t inode)
{
	de->inode = to_le32(inode);
}

/**@brief Set i-node number to directory entry. (For HTree root)
 * @param de Directory entry
 * @param inode I-node number
 */
static inline void
ext4_dx_dot_en_set_inode(struct ext4_dir_idx_dot_en *de, uint32_t inode)
{
	de->inode = to_le32(inode);
}

/**@brief Get directory entry length.
 * @param de Directory entry
 * @return Entry length
 */
static inline uint16_t ext4_dir_en_get_entry_len(struct ext4_dir_en *de)
{
	return to_le16(de->entry_len);
}

/**@brief Set directory entry length.
 * @param de     Directory entry
 * @param l Entry length
 */
static inline void ext4_dir_en_set_entry_len(struct ext4_dir_en *de, uint16_t l)
{
	de->entry_len = to_le16(l);
}

/**@brief Get directory entry name length.
 * @param sb Superblock
 * @param de Directory entry
 * @return Entry name length
 */
static inline uint16_t ext4_dir_en_get_name_len(struct ext4_sblock *sb,
						struct ext4_dir_en *de)
{
	if(de == NULL)
		return 0;

	uint16_t v = de->name_len;

	if ((ext4_get32(sb, rev_level) == 0) &&
	    (ext4_get32(sb, minor_rev_level) < 5))
		v |= ((uint16_t)de->in.name_length_high) << 8;

	return v;
}

/**@brief Set directory entry name length.
 * @param sb     Superblock
 * @param de     Directory entry
 * @param len Entry name length
 */
static inline void ext4_dir_en_set_name_len(struct ext4_sblock *sb,
					    struct ext4_dir_en *de,
					    uint16_t len)
{
	de->name_len = (len << 8) >> 8;

	if ((ext4_get32(sb, rev_level) == 0) &&
	    (ext4_get32(sb, minor_rev_level) < 5))
		de->in.name_length_high = len >> 8;
}

/**@brief Get i-node type of directory entry.
 * @param sb Superblock
 * @param de Directory entry
 * @return I-node type (file, dir, etc.)
 */
static inline uint8_t ext4_dir_en_get_inode_type(struct ext4_sblock *sb,
						 struct ext4_dir_en *de)
{
	if ((ext4_get32(sb, rev_level) > 0) ||
	    (ext4_get32(sb, minor_rev_level) >= 5))
		return de->in.inode_type;

	return EXT4_DE_UNKNOWN;
}
/**@brief Set i-node type of directory entry.
 * @param sb   Superblock
 * @param de   Directory entry
 * @param t I-node type (file, dir, etc.)
 */

static inline void ext4_dir_en_set_inode_type(struct ext4_sblock *sb,
					      struct ext4_dir_en *de, uint8_t t)
{
	if ((ext4_get32(sb, rev_level) > 0) ||
	    (ext4_get32(sb, minor_rev_level) >= 5))
		de->in.inode_type = t;
}

/**@brief Verify checksum of a linear directory leaf block
 * @param inode_ref Directory i-node
 * @param dirent    Linear directory leaf block
 * @return true means the block passed checksum verification
 */
bool ext4_dir_csum_verify(struct ext4_inode_ref *inode_ref,
			  struct ext4_dir_en *dirent);

/**@brief Initialize directory iterator.
 * Set position to the first valid entry from the required position.
 * @param it        Pointer to iterator to be initialized
 * @param inode_ref Directory i-node
 * @param pos       Position to start reading entries from
 * @return Error code
 */
int ext4_dir_iterator_init(struct ext4_dir_iter *it,
			   struct ext4_inode_ref *inode_ref, uint64_t pos);

/**@brief Jump to the next valid entry
 * @param it Initialized iterator
 * @return Error code
 */
int ext4_dir_iterator_next(struct ext4_dir_iter *it);

/**@brief Uninitialize directory iterator.
 *        Release all allocated structures.
 * @param it Iterator to be finished
 * @return Error code
 */
int ext4_dir_iterator_fini(struct ext4_dir_iter *it);

/**@brief Write directory entry to concrete data block.
 * @param sb        Superblock
 * @param en     Pointer to entry to be written
 * @param entry_len Length of new entry
 * @param child     Child i-node to be written to new entry
 * @param name      Name of the new entry
 * @param name_len  Length of entry name
 */
void ext4_dir_write_entry(struct ext4_sblock *sb, struct ext4_dir_en *en,
			  uint16_t entry_len, struct ext4_inode_ref *child,
			  const char *name, size_t name_len);

/**@brief Add new entry to the directory.
 * @param parent Directory i-node
 * @param name   Name of new entry
 * @param child  I-node to be referenced from new entry
 * @return Error code
 */
int ext4_dir_add_entry(struct ext4_inode_ref *parent, const char *name,
		       uint32_t name_len, struct ext4_inode_ref *child);

/**@brief Find directory entry with passed name.
 * @param result Result structure to be returned if entry found
 * @param parent Directory i-node
 * @param name   Name of entry to be found
 * @param name_len  Name length
 * @return Error code
 */
int ext4_dir_find_entry(struct ext4_dir_search_result *result,
			struct ext4_inode_ref *parent, const char *name,
			uint32_t name_len);

/**@brief Remove directory entry.
 * @param parent Directory i-node
 * @param name   Name of the entry to be removed
 * @param name_len  Name length
 * @return Error code
 */
int ext4_dir_remove_entry(struct ext4_inode_ref *parent, const char *name,
			  uint32_t name_len);

/**@brief Try to insert entry to concrete data block.
 * @param sb           Superblock
 * @param inode_ref    Directory i-node
 * @param dst_blk      Block to try to insert entry to
 * @param child        Child i-node to be inserted by new entry
 * @param name         Name of the new entry
 * @param name_len     Length of the new entry name
 * @return Error code
 */
int ext4_dir_try_insert_entry(struct ext4_sblock *sb,
			      struct ext4_inode_ref *inode_ref,
			      struct ext4_block *dst_blk,
			      struct ext4_inode_ref *child, const char *name,
			      uint32_t name_len);

/**@brief Try to find entry in block by name.
 * @param block     Block containing entries
 * @param sb        Superblock
 * @param name_len  Length of entry name
 * @param name      Name of entry to be found
 * @param res_entry Output pointer to found entry, NULL if not found
 * @return Error code
 */
int ext4_dir_find_in_block(struct ext4_block *block, struct ext4_sblock *sb,
			   size_t name_len, const char *name,
			   struct ext4_dir_en **res_entry);

/**@brief Simple function to release allocated data from result.
 * @param parent Parent inode
 * @param result Search result to destroy
 * @return Error code
 *
 */
int ext4_dir_destroy_result(struct ext4_inode_ref *parent,
			    struct ext4_dir_search_result *result);

void ext4_dir_set_csum(struct ext4_inode_ref *inode_ref,
		       struct ext4_dir_en *dirent);


void ext4_dir_init_entry_tail(struct ext4_dir_entry_tail *t);

#ifdef __cplusplus
}
#endif

#endif /* EXT4_DIR_H_ */

/**
 * @}
 */
