// SPDX-License-Identifier: GPL-2.0
/*
 *  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 "hfsplus_fs.h"
#include "hfsplus_raw.h"

int hfsplus_cat_case_cmp_key(const hfsplus_btree_key *k1,
                             const hfsplus_btree_key *k2)
{
        __be32 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_strcasecmp(&k1->cat.name, &k2->cat.name);
}

int hfsplus_cat_bin_cmp_key(const hfsplus_btree_key *k1,
                            const hfsplus_btree_key *k2)
{
        __be32 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_strcmp(&k1->cat.name, &k2->cat.name);
}

/* Generates key for catalog file/folders record. */
int hfsplus_cat_build_key(struct super_block *sb,
                hfsplus_btree_key *key, u32 parent, const struct qstr *str)
{
        int len, err;

        key->cat.parent = cpu_to_be32(parent);
        err = hfsplus_asc2uni(sb, &key->cat.name, HFSPLUS_MAX_STRLEN,
                        str->name, str->len);
        if (unlikely(err < 0))
                return err;

        len = be16_to_cpu(key->cat.name.length);
        key->key_len = cpu_to_be16(6 + 2 * len);
        return 0;
}

/* Generates key for catalog thread record. */
void hfsplus_cat_build_key_with_cnid(struct super_block *sb,
                        hfsplus_btree_key *key, u32 parent)
{
        key->cat.parent = cpu_to_be32(parent);
        key->cat.name.length = 0;
        key->key_len = cpu_to_be16(6);
}

static void hfsplus_cat_build_key_uni(hfsplus_btree_key *key, u32 parent,
                                      struct 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);
}

void hfsplus_cat_set_perms(struct inode *inode, struct hfsplus_perm *perms)
{
        if (inode->i_flags & S_IMMUTABLE)
                perms->rootflags |= HFSPLUS_FLG_IMMUTABLE;
        else
                perms->rootflags &= ~HFSPLUS_FLG_IMMUTABLE;
        if (inode->i_flags & S_APPEND)
                perms->rootflags |= HFSPLUS_FLG_APPEND;
        else
                perms->rootflags &= ~HFSPLUS_FLG_APPEND;

        perms->userflags = HFSPLUS_I(inode)->userflags;
        perms->mode = cpu_to_be16(inode->i_mode);
        perms->owner = cpu_to_be32(i_uid_read(inode));
        perms->group = cpu_to_be32(i_gid_read(inode));

        if (S_ISREG(inode->i_mode))
                perms->dev = cpu_to_be32(inode->i_nlink);
        else if (S_ISBLK(inode->i_mode) || S_ISCHR(inode->i_mode))
                perms->dev = cpu_to_be32(inode->i_rdev);
        else
                perms->dev = 0;
}

static int hfsplus_cat_build_record(hfsplus_cat_entry *entry,
                u32 cnid, struct inode *inode)
{
        struct hfsplus_sb_info *sbi = HFSPLUS_SB(inode->i_sb);

        if (S_ISDIR(inode->i_mode)) {
                struct hfsplus_cat_folder *folder;

                folder = &entry->folder;
                memset(folder, 0, sizeof(*folder));
                folder->type = cpu_to_be16(HFSPLUS_FOLDER);
                if (test_bit(HFSPLUS_SB_HFSX, &sbi->flags))
                        folder->flags |= cpu_to_be16(HFSPLUS_HAS_FOLDER_COUNT);
                folder->id = cpu_to_be32(inode->i_ino);
                HFSPLUS_I(inode)->create_date =
                        folder->create_date =
                        folder->content_mod_date =
                        folder->attribute_mod_date =
                        folder->access_date = hfsp_now2mt();
                hfsplus_cat_set_perms(inode, &folder->permissions);
                if (inode == sbi->hidden_dir)
                        /* invisible and namelocked */
                        folder->user_info.frFlags = cpu_to_be16(0x5000);
                return sizeof(*folder);
        } else {
                struct hfsplus_cat_file *file;

                file = &entry->file;
                memset(file, 0, sizeof(*file));
                file->type = cpu_to_be16(HFSPLUS_FILE);
                file->flags = cpu_to_be16(HFSPLUS_FILE_THREAD_EXISTS);
                file->id = cpu_to_be32(cnid);
                HFSPLUS_I(inode)->create_date =
                        file->create_date =
                        file->content_mod_date =
                        file->attribute_mod_date =
                        file->access_date = hfsp_now2mt();
                if (cnid == inode->i_ino) {
                        hfsplus_cat_set_perms(inode, &file->permissions);
                        if (S_ISLNK(inode->i_mode)) {
                                file->user_info.fdType =
                                        cpu_to_be32(HFSP_SYMLINK_TYPE);
                                file->user_info.fdCreator =
                                        cpu_to_be32(HFSP_SYMLINK_CREATOR);
                        } else {
                                file->user_info.fdType =
                                        cpu_to_be32(sbi->type);
                                file->user_info.fdCreator =
                                        cpu_to_be32(sbi->creator);
                        }
                        if (HFSPLUS_FLG_IMMUTABLE &
                                        (file->permissions.rootflags |
                                        file->permissions.userflags))
                                file->flags |=
                                        cpu_to_be16(HFSPLUS_FILE_LOCKED);
                } 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->create_date =
                                HFSPLUS_I(sbi->hidden_dir)->create_date;
                        file->permissions.dev =
                                cpu_to_be32(HFSPLUS_I(inode)->linkid);
                }
                return sizeof(*file);
        }
}

static int hfsplus_fill_cat_thread(struct super_block *sb,
                                   hfsplus_cat_entry *entry, int type,
                                   u32 parentid, const struct qstr *str)
{
        int err;

        entry->type = cpu_to_be16(type);
        entry->thread.reserved = 0;
        entry->thread.parentID = cpu_to_be32(parentid);
        err = hfsplus_asc2uni(sb, &entry->thread.nodeName, HFSPLUS_MAX_STRLEN,
                                str->name, str->len);
        if (unlikely(err < 0))
                return err;

        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, u32 cnid,
                     struct hfs_find_data *fd)
{
        hfsplus_cat_entry tmp;
        int err;
        u16 type;

        hfsplus_cat_build_key_with_cnid(sb, fd->search_key, cnid);
        err = hfs_brec_read(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) {
                pr_err("found bad thread record in catalog\n");
                return -EIO;
        }

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

        hfsplus_cat_build_key_uni(fd->search_key,
                be32_to_cpu(tmp.thread.parentID),
                &tmp.thread.nodeName);
        return hfs_brec_find(fd, hfs_find_rec_by_key);
}

static void hfsplus_subfolders_inc(struct inode *dir)
{
        struct hfsplus_sb_info *sbi = HFSPLUS_SB(dir->i_sb);

        if (test_bit(HFSPLUS_SB_HFSX, &sbi->flags)) {
                /*
                 * Increment subfolder count. Note, the value is only meaningful
                 * for folders with HFSPLUS_HAS_FOLDER_COUNT flag set.
                 */
                HFSPLUS_I(dir)->subfolders++;
        }
}

static void hfsplus_subfolders_dec(struct inode *dir)
{
        struct hfsplus_sb_info *sbi = HFSPLUS_SB(dir->i_sb);

        if (test_bit(HFSPLUS_SB_HFSX, &sbi->flags)) {
                /*
                 * Decrement subfolder count. Note, the value is only meaningful
                 * for folders with HFSPLUS_HAS_FOLDER_COUNT flag set.
                 *
                 * Check for zero. Some subfolders may have been created
                 * by an implementation ignorant of this counter.
                 */
                if (HFSPLUS_I(dir)->subfolders)
                        HFSPLUS_I(dir)->subfolders--;
        }
}

int hfsplus_create_cat(u32 cnid, struct inode *dir,
                const struct qstr *str, struct inode *inode)
{
        struct super_block *sb = dir->i_sb;
        struct hfs_find_data fd;
        hfsplus_cat_entry entry;
        int entry_size;
        int err;

        hfs_dbg("name %s, cnid %u, i_nlink %d\n",
                str->name, cnid, inode->i_nlink);
        err = hfs_find_init(HFSPLUS_SB(sb)->cat_tree, &fd);
        if (err)
                return err;

        /*
         * Fail early and avoid ENOSPC during the btree operations. We may
         * have to split the root node at most once.
         */
        err = hfs_bmap_reserve(fd.tree, 2 * fd.tree->depth);
        if (err)
                goto err2;

        hfsplus_cat_build_key_with_cnid(sb, fd.search_key, cnid);
        entry_size = hfsplus_fill_cat_thread(sb, &entry,
                S_ISDIR(inode->i_mode) ?
                        HFSPLUS_FOLDER_THREAD : HFSPLUS_FILE_THREAD,
                dir->i_ino, str);
        if (unlikely(entry_size < 0)) {
                err = entry_size;
                goto err2;
        }

        err = hfs_brec_find(&fd, hfs_find_rec_by_key);
        if (err != -ENOENT) {
                if (!err)
                        err = -EEXIST;
                goto err2;
        }
        err = hfs_brec_insert(&fd, &entry, entry_size);
        if (err)
                goto err2;

        err = hfsplus_cat_build_key(sb, fd.search_key, dir->i_ino, str);
        if (unlikely(err))
                goto err1;

        entry_size = hfsplus_cat_build_record(&entry, cnid, inode);
        err = hfs_brec_find(&fd, hfs_find_rec_by_key);
        if (err != -ENOENT) {
                /* panic? */
                if (!err)
                        err = -EEXIST;
                goto err1;
        }
        err = hfs_brec_insert(&fd, &entry, entry_size);
        if (err)
                goto err1;

        dir->i_size++;
        if (S_ISDIR(inode->i_mode))
                hfsplus_subfolders_inc(dir);
        inode_set_mtime_to_ts(dir, inode_set_ctime_current(dir));
        hfsplus_mark_inode_dirty(dir, HFSPLUS_I_CAT_DIRTY);

        hfs_find_exit(&fd);
        return 0;

err1:
        hfsplus_cat_build_key_with_cnid(sb, fd.search_key, cnid);
        if (!hfs_brec_find(&fd, hfs_find_rec_by_key))
                hfs_brec_remove(&fd);
err2:
        hfs_find_exit(&fd);
        return err;
}

int hfsplus_delete_cat(u32 cnid, struct inode *dir, const struct qstr *str)
{
        struct super_block *sb = dir->i_sb;
        struct hfs_find_data fd;
        struct hfsplus_fork_raw fork;
        struct list_head *pos;
        int err, off;
        u16 type;

        hfs_dbg("name %s, cnid %u\n", str ? str->name : NULL, cnid);
        err = hfs_find_init(HFSPLUS_SB(sb)->cat_tree, &fd);
        if (err)
                return err;

        /*
         * Fail early and avoid ENOSPC during the btree operations. We may
         * have to split the root node at most once.
         */
        err = hfs_bmap_reserve(fd.tree, 2 * (int)fd.tree->depth - 2);
        if (err)
                goto out;

        if (!str) {
                int len;

                hfsplus_cat_build_key_with_cnid(sb, fd.search_key, cnid);
                err = hfs_brec_find(&fd, hfs_find_rec_by_key);
                if (err)
                        goto out;

                off = fd.entryoffset +
                        offsetof(struct hfsplus_cat_thread, nodeName);
                fd.search_key->cat.parent = cpu_to_be32(dir->i_ino);
                hfs_bnode_read(fd.bnode,
                        &fd.search_key->cat.name.length, off, 2);
                len = be16_to_cpu(fd.search_key->cat.name.length) * 2;
                hfs_bnode_read(fd.bnode,
                        &fd.search_key->cat.name.unicode,
                        off + 2, len);
                fd.search_key->key_len = cpu_to_be16(6 + len);
        } else {
                err = hfsplus_cat_build_key(sb, fd.search_key, dir->i_ino, str);
                if (unlikely(err))
                        goto out;
        }

        err = hfs_brec_find(&fd, hfs_find_rec_by_key);
        if (err)
                goto out;

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

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

        /* we only need to take spinlock for exclusion with ->release() */
        spin_lock(&HFSPLUS_I(dir)->open_dir_lock);
        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--;
        }
        spin_unlock(&HFSPLUS_I(dir)->open_dir_lock);

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

        hfsplus_cat_build_key_with_cnid(sb, fd.search_key, cnid);
        err = hfs_brec_find(&fd, hfs_find_rec_by_key);
        if (err)
                goto out;

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

        dir->i_size--;
        if (type == HFSPLUS_FOLDER)
                hfsplus_subfolders_dec(dir);
        inode_set_mtime_to_ts(dir, inode_set_ctime_current(dir));
        hfsplus_mark_inode_dirty(dir, HFSPLUS_I_CAT_DIRTY);

        if (type == HFSPLUS_FILE || type == HFSPLUS_FOLDER) {
                if (HFSPLUS_SB(sb)->attr_tree)
                        hfsplus_delete_all_attrs(dir, cnid);
        }

out:
        hfs_find_exit(&fd);

        return err;
}

int hfsplus_rename_cat(u32 cnid,
                       struct inode *src_dir, const struct qstr *src_name,
                       struct inode *dst_dir, const struct qstr *dst_name)
{
        struct super_block *sb = src_dir->i_sb;
        struct hfs_find_data src_fd, dst_fd;
        hfsplus_cat_entry entry;
        int entry_size, type;
        int err;

        hfs_dbg("cnid %u - ino %lu, name %s - ino %lu, name %s\n",
                cnid, src_dir->i_ino, src_name->name,
                dst_dir->i_ino, dst_name->name);
        err = hfs_find_init(HFSPLUS_SB(sb)->cat_tree, &src_fd);
        if (err)
                return err;
        dst_fd = src_fd;

        /*
         * Fail early and avoid ENOSPC during the btree operations. We may
         * have to split the root node at most twice.
         */
        err = hfs_bmap_reserve(src_fd.tree, 4 * (int)src_fd.tree->depth - 1);
        if (err)
                goto out;

        /* find the old dir entry and read the data */
        err = hfsplus_cat_build_key(sb, src_fd.search_key,
                        src_dir->i_ino, src_name);
        if (unlikely(err))
                goto out;

        err = hfs_brec_find(&src_fd, hfs_find_rec_by_key);
        if (err)
                goto out;
        if (src_fd.entrylength > sizeof(entry) || src_fd.entrylength < 0) {
                err = -EIO;
                goto out;
        }

        hfs_bnode_read(src_fd.bnode, &entry, src_fd.entryoffset,
                                src_fd.entrylength);
        type = be16_to_cpu(entry.type);

        /* create new dir entry with the data from the old entry */
        err = hfsplus_cat_build_key(sb, dst_fd.search_key,
                        dst_dir->i_ino, dst_name);
        if (unlikely(err))
                goto out;

        err = hfs_brec_find(&dst_fd, hfs_find_rec_by_key);
        if (err != -ENOENT) {
                if (!err)
                        err = -EEXIST;
                goto out;
        }

        err = hfs_brec_insert(&dst_fd, &entry, src_fd.entrylength);
        if (err)
                goto out;
        dst_dir->i_size++;
        if (type == HFSPLUS_FOLDER)
                hfsplus_subfolders_inc(dst_dir);
        inode_set_mtime_to_ts(dst_dir, inode_set_ctime_current(dst_dir));

        /* finally remove the old entry */
        err = hfsplus_cat_build_key(sb, src_fd.search_key,
                        src_dir->i_ino, src_name);
        if (unlikely(err))
                goto out;

        err = hfs_brec_find(&src_fd, hfs_find_rec_by_key);
        if (err)
                goto out;
        err = hfs_brec_remove(&src_fd);
        if (err)
                goto out;
        src_dir->i_size--;
        if (type == HFSPLUS_FOLDER)
                hfsplus_subfolders_dec(src_dir);
        inode_set_mtime_to_ts(src_dir, inode_set_ctime_current(src_dir));

        /* remove old thread entry */
        hfsplus_cat_build_key_with_cnid(sb, src_fd.search_key, cnid);
        err = hfs_brec_find(&src_fd, hfs_find_rec_by_key);
        if (err)
                goto out;
        type = hfs_bnode_read_u16(src_fd.bnode, src_fd.entryoffset);
        err = hfs_brec_remove(&src_fd);
        if (err)
                goto out;

        /* create new thread entry */
        hfsplus_cat_build_key_with_cnid(sb, dst_fd.search_key, cnid);
        entry_size = hfsplus_fill_cat_thread(sb, &entry, type,
                dst_dir->i_ino, dst_name);
        if (unlikely(entry_size < 0)) {
                err = entry_size;
                goto out;
        }

        err = hfs_brec_find(&dst_fd, hfs_find_rec_by_key);
        if (err != -ENOENT) {
                if (!err)
                        err = -EEXIST;
                goto out;
        }
        err = hfs_brec_insert(&dst_fd, &entry, entry_size);

        hfsplus_mark_inode_dirty(dst_dir, HFSPLUS_I_CAT_DIRTY);
        hfsplus_mark_inode_dirty(src_dir, HFSPLUS_I_CAT_DIRTY);
out:
        hfs_bnode_put(dst_fd.bnode);
        hfs_find_exit(&src_fd);
        return err;
}