/*
 *  linux/fs/hfsplus/catalog.c
 *
 * Copyright (C) 2001
 * Brad Boyer (flar@allandria.com)
 * (C) 2003 Ardis Technologies <roman@ardistech.com>
 *
 * Handling of catalog records
 */

#include <linux/sched.h>

#include "hfsplus_fs.h"
#include "hfsplus_raw.h"

int hfsplus_cmp_cat_key(hfsplus_btree_key *k1, hfsplus_btree_key *k2)
{
	u32 k1p, k2p;

	k1p = k1->cat.parent;
	k2p = k2->cat.parent;
	if (k1p != k2p)
		return be32_to_cpu(k1p) < be32_to_cpu(k2p) ? -1 : 1;

	return hfsplus_unistrcmp(&k1->cat.name, &k2->cat.name);
}

void hfsplus_fill_cat_key(hfsplus_btree_key *key, u32 parent,
			  struct qstr *str)
{
	int len;

	key->cat.parent = cpu_to_be32(parent);
	if (str) {
		hfsplus_asc2uni(&key->cat.name, str->name, str->len);
		len = be16_to_cpu(key->cat.name.length);
	} else
		len = key->cat.name.length = 0;
	key->key_len = cpu_to_be16(6 + 2 * len);
}

static void hfsplus_fill_cat_key_uni(hfsplus_btree_key *key, u32 parent,
				     hfsplus_unistr *name)
{
	int ustrlen;

	ustrlen = be16_to_cpu(name->length);
	key->cat.parent = cpu_to_be32(parent);
	key->cat.name.length = cpu_to_be16(ustrlen);
	ustrlen *= 2;
	memcpy(key->cat.name.unicode, name->unicode, ustrlen);
	key->key_len = cpu_to_be16(6 + ustrlen);
}

static void hfsplus_set_perms(struct inode *inode, hfsplus_perm *perms)
{
	perms->mode = cpu_to_be32(inode->i_mode);
	perms->owner = cpu_to_be32(inode->i_uid);
	perms->group = cpu_to_be32(inode->i_gid);
}

static int hfsplus_fill_cat_entry(hfsplus_cat_entry *entry, u32 cnid, struct inode *inode)
{
	if (S_ISDIR(inode->i_mode)) {
		hfsplus_cat_folder *folder;

		folder = &entry->folder;
		memset(folder, 0, sizeof(*folder));
		folder->type = cpu_to_be16(HFSPLUS_FOLDER);
		folder->id = cpu_to_be32(inode->i_ino);
		folder->create_date = folder->content_mod_date = 
			folder->attribute_mod_date = folder->access_date = 
			hfsp_now2mt();
		hfsplus_set_perms(inode, &folder->permissions);
		if (inode == HFSPLUS_SB(inode->i_sb).hidden_dir)
			/* invisible and namelocked */
			folder->user_info.frFlags = cpu_to_be16(0x5000);
		return sizeof(*folder);
	} else {
		hfsplus_cat_file *file;

		file = &entry->file;
		memset(file, 0, sizeof(*file));
		file->type = cpu_to_be16(HFSPLUS_FILE);
		file->id = cpu_to_be32(cnid);
		file->create_date = file->content_mod_date =
			file->attribute_mod_date = file->access_date =
			hfsp_now2mt();
		if (cnid == inode->i_ino) {
			hfsplus_set_perms(inode, &file->permissions);
			file->user_info.fdType = cpu_to_be32(HFSPLUS_SB(inode->i_sb).type);
			file->user_info.fdCreator = cpu_to_be32(HFSPLUS_SB(inode->i_sb).creator);
		} else {
			file->user_info.fdType = cpu_to_be32(HFSP_HARDLINK_TYPE);
			file->user_info.fdCreator = cpu_to_be32(HFSP_HFSPLUS_CREATOR);
			file->user_info.fdFlags = cpu_to_be16(0x100);
			file->permissions.dev = cpu_to_be32(HFSPLUS_I(inode).dev);
		}
		return sizeof(*file);
	}
}

static int hfsplus_fill_cat_thread(hfsplus_cat_entry *entry, int type,
				   u32 parentid, struct qstr *str)
{
	entry->type = cpu_to_be16(type);
	entry->thread.reserved = 0;
	entry->thread.parentID = cpu_to_be32(parentid);
	hfsplus_asc2uni(&entry->thread.nodeName, str->name, str->len);
	return 10 + be16_to_cpu(entry->thread.nodeName.length) * 2;
}

/* Try to get a catalog entry for given catalog id */
int hfsplus_find_cat(struct super_block *sb, unsigned long cnid,
		     struct hfsplus_find_data *fd)
{
	hfsplus_cat_entry tmp;
	int err;
	u16 type;

	hfsplus_fill_cat_key(fd->search_key, cnid, NULL);
	err = hfsplus_btree_find_entry(fd, &tmp, sizeof(hfsplus_cat_entry));
	if (err)
		return err;

	type = be16_to_cpu(tmp.type);
	if (type != HFSPLUS_FOLDER_THREAD && type != HFSPLUS_FILE_THREAD) {
		printk("HFS+-fs: Found bad thread record in catalog\n");
		return -EIO;
	}

	if (be16_to_cpu(tmp.thread.nodeName.length) > 255) {
		printk(KERN_ERR "hfs: catalog name length corrupted\n");
		return -EIO;
	}

	hfsplus_fill_cat_key_uni(fd->search_key, be32_to_cpu(tmp.thread.parentID),
				 &tmp.thread.nodeName);
	return hfsplus_btree_find(fd);
}

int hfsplus_create_cat(u32 cnid, struct inode *dir, struct qstr *str, struct inode *inode)
{
	struct hfsplus_find_data fd;
	struct super_block *sb;
	hfsplus_cat_entry entry;
	int entry_size;
	int err;

	dprint(DBG_CAT_MOD, "create_cat: %s,%u(%d)\n", str->name, cnid, inode->i_nlink);
	sb = dir->i_sb;
	hfsplus_find_init(HFSPLUS_SB(sb).cat_tree, &fd);

	hfsplus_fill_cat_key(fd.search_key, cnid, NULL);
	entry_size = hfsplus_fill_cat_thread(&entry, S_ISDIR(inode->i_mode) ?
			HFSPLUS_FOLDER_THREAD : HFSPLUS_FILE_THREAD,
			dir->i_ino, str);
	err = hfsplus_btree_find(&fd);
	if (err != -ENOENT) {
		if (!err)
			err = -EEXIST;
		goto out;
	}
	err = hfsplus_bnode_insert_rec(&fd, &entry, entry_size);
	if (err)
		goto out;

	hfsplus_fill_cat_key(fd.search_key, dir->i_ino, str);
	entry_size = hfsplus_fill_cat_entry(&entry, cnid, inode);
	err = hfsplus_btree_find(&fd);
	if (err != -ENOENT) {
		/* panic? */
		if (!err)
			err = -EEXIST;
		goto out;
	}
	err = hfsplus_bnode_insert_rec(&fd, &entry, entry_size);
	if (!err) {
		dir->i_size++;
		mark_inode_dirty(dir);
	}
out:
	hfsplus_find_exit(&fd);

	return err;
}

int hfsplus_delete_cat(u32 cnid, struct inode *dir, struct qstr *str)
{
	struct super_block *sb;
	struct hfsplus_find_data fd;
	hfsplus_fork_raw fork;
	struct list_head *pos;
	int err, off;
	u16 type;

	dprint(DBG_CAT_MOD, "delete_cat: %s,%u\n", str ? str->name : NULL, cnid);
	sb = dir->i_sb;
	hfsplus_find_init(HFSPLUS_SB(sb).cat_tree, &fd);

	if (!str) {
		int len;

		hfsplus_fill_cat_key(fd.search_key, cnid, NULL);
		err = hfsplus_btree_find(&fd);
		if (err)
			goto out;

		off = fd.entryoffset + offsetof(hfsplus_cat_thread, nodeName);
		fd.search_key->cat.parent = cpu_to_be32(dir->i_ino);
		hfsplus_bnode_readbytes(fd.bnode, &fd.search_key->cat.name.length, off, 2);
		len = be16_to_cpu(fd.search_key->cat.name.length) * 2;
		hfsplus_bnode_readbytes(fd.bnode, &fd.search_key->cat.name.unicode, off + 2, len);
		fd.search_key->key_len = cpu_to_be16(6 + len);
	} else
		hfsplus_fill_cat_key(fd.search_key, dir->i_ino, str);

	err = hfsplus_btree_find(&fd);
	if (err)
		goto out;

	type = hfsplus_bnode_read_u16(fd.bnode, fd.entryoffset);
	if (type == HFSPLUS_FILE) {
#if 0
		off = fd.entryoffset + offsetof(hfsplus_cat_file, data_fork);
		hfsplus_bnode_readbytes(fd.bnode, &fork, off, sizeof(fork));
		hfsplus_free_fork(sb, cnid, &fork, HFSPLUS_TYPE_DATA);
#endif

		off = fd.entryoffset + offsetof(hfsplus_cat_file, rsrc_fork);
		hfsplus_bnode_readbytes(fd.bnode, &fork, off, sizeof(fork));
		hfsplus_free_fork(sb, cnid, &fork, HFSPLUS_TYPE_RSRC);
	}

	list_for_each(pos, &HFSPLUS_I(dir).open_dir_list) {
		struct hfsplus_readdir_data *rd =
			list_entry(pos, struct hfsplus_readdir_data, list);
		if (fd.tree->keycmp(fd.search_key, (void *)&rd->key) < 0)
			rd->file->f_pos--;
	}

	err = hfsplus_bnode_remove_rec(&fd);
	if (err)
		goto out;

	hfsplus_fill_cat_key(fd.search_key, cnid, NULL);
	err = hfsplus_btree_find(&fd);
	if (err)
		goto out;

	err = hfsplus_bnode_remove_rec(&fd);
	if (err) 
		goto out;

	dir->i_size--;
	mark_inode_dirty(dir);
out:
	hfsplus_find_exit(&fd);

	return err;
}

int hfsplus_rename_cat(u32 cnid,
		       struct inode *src_dir, struct qstr *src_name,
		       struct inode *dst_dir, struct qstr *dst_name)
{
	struct super_block *sb;
	struct hfsplus_find_data src_fd, dst_fd;
	hfsplus_cat_entry entry;
	int entry_size, type;
	int err = 0;

	dprint(DBG_CAT_MOD, "rename_cat: %u - %lu,%s - %lu,%s\n", cnid, src_dir->i_ino, src_name->name,
		dst_dir->i_ino, dst_name->name);
	sb = src_dir->i_sb;
	hfsplus_find_init(HFSPLUS_SB(sb).cat_tree, &src_fd);
	dst_fd = src_fd;

	/* find the old dir entry and read the data */
	hfsplus_fill_cat_key(src_fd.search_key, src_dir->i_ino, src_name);
	err = hfsplus_btree_find(&src_fd);
	if (err)
		goto out;
		
	hfsplus_bnode_readbytes(src_fd.bnode, &entry, src_fd.entryoffset,
				src_fd.entrylength);

	/* create new dir entry with the data from the old entry */
	hfsplus_fill_cat_key(dst_fd.search_key, dst_dir->i_ino, dst_name);
	err = hfsplus_btree_find(&dst_fd);
	if (err != -ENOENT) {
		if (!err)
			err = -EEXIST;
		goto out;
	}

	err = hfsplus_bnode_insert_rec(&dst_fd, &entry, src_fd.entrylength);
	if (err)
		goto out;
	dst_dir->i_size++;
	mark_inode_dirty(dst_dir);

	/* finally remove the old entry */
	hfsplus_fill_cat_key(src_fd.search_key, src_dir->i_ino, src_name);
	err = hfsplus_btree_find(&src_fd);
	if (err)
		goto out;
	err = hfsplus_bnode_remove_rec(&src_fd);
	if (err)
		goto out;
	src_dir->i_size--;
	mark_inode_dirty(src_dir);

	/* remove old thread entry */
	hfsplus_fill_cat_key(src_fd.search_key, cnid, NULL);
	err = hfsplus_btree_find(&src_fd);
	if (err)
		goto out;
	type = hfsplus_bnode_read_u16(src_fd.bnode, src_fd.entryoffset);
	err = hfsplus_bnode_remove_rec(&src_fd);
	if (err)
		goto out;

	/* create new thread entry */
	hfsplus_fill_cat_key(dst_fd.search_key, cnid, NULL);
	entry_size = hfsplus_fill_cat_thread(&entry, type, dst_dir->i_ino, dst_name);
	err = hfsplus_btree_find(&dst_fd);
	if (err != -ENOENT) {
		if (!err)
			err = -EEXIST;
		goto out;
	}
	err = hfsplus_bnode_insert_rec(&dst_fd, &entry, entry_size);
out:
	hfsplus_put_bnode(dst_fd.bnode);
	hfsplus_find_exit(&src_fd);
	return err;
}