/*
 * pfs_dev.c
 *
 * Loosely based on fs_dev.c from libnx, et al.
 *
 * Copyright (c) 2020-2023, DarkMatterCore <pabloacurielz@gmail.com>.
 *
 * This file is part of nxdumptool (https://github.com/DarkMatterCore/nxdumptool).
 *
 * nxdumptool is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * nxdumptool is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

#include "nxdt_utils.h"
#include "nxdt_devoptab.h"
#include "ro_dev.h"

/* Helper macros. */

#define PFS_DEV_INIT_VARS       DEVOPTAB_INIT_VARS(PartitionFileSystemContext)
#define PFS_DEV_INIT_FILE_VARS  DEVOPTAB_INIT_FILE_VARS(PartitionFileSystemContext, PartitionFileSystemFileState)
#define PFS_DEV_INIT_DIR_VARS   DEVOPTAB_INIT_DIR_VARS(PartitionFileSystemContext, PartitionFileSystemDirectoryState)
#define PFS_DEV_INIT_FS_ACCESS  DEVOPTAB_DECL_FS_CTX(PartitionFileSystemContext)

/* Type definitions. */

typedef struct {
    u32 index;                              ///< Partition FS entry index.
    PartitionFileSystemEntry *pfs_entry;    ///< Partition FS entry metadata.
    const char *name;                       ///< Entry name.
    u64 offset;                             ///< Current offset within Partition FS entry data.
} PartitionFileSystemFileState;

typedef struct {
    u8 state;   ///< 0: "." entry; 1: ".." entry; 2: actual Partition FS entry.
    u32 index;  ///< Current Partition FS entry index.
} PartitionFileSystemDirectoryState;

/* Function prototypes. */

static int       pfsdev_open(struct _reent *r, void *fd, const char *path, int flags, int mode);
static int       pfsdev_close(struct _reent *r, void *fd);
static ssize_t   pfsdev_read(struct _reent *r, void *fd, char *ptr, size_t len);
static off_t     pfsdev_seek(struct _reent *r, void *fd, off_t pos, int dir);
static int       pfsdev_fstat(struct _reent *r, void *fd, struct stat *st);
static int       pfsdev_stat(struct _reent *r, const char *file, struct stat *st);
static DIR_ITER* pfsdev_diropen(struct _reent *r, DIR_ITER *dirState, const char *path);
static int       pfsdev_dirreset(struct _reent *r, DIR_ITER *dirState);
static int       pfsdev_dirnext(struct _reent *r, DIR_ITER *dirState, char *filename, struct stat *filestat);
static int       pfsdev_dirclose(struct _reent *r, DIR_ITER *dirState);
static int       pfsdev_statvfs(struct _reent *r, const char *path, struct statvfs *buf);

static const char *pfsdev_get_truncated_path(struct _reent *r, const char *path);

static void pfsdev_fill_stat(struct stat *st, u32 index, const PartitionFileSystemEntry *pfs_entry, time_t mount_time);

/* Global variables. */

static const devoptab_t pfsdev_devoptab = {
    .name         = NULL,
    .structSize   = sizeof(PartitionFileSystemFileState),
    .open_r       = pfsdev_open,
    .close_r      = pfsdev_close,
    .write_r      = rodev_write,                                ///< Not supported by Partition FS sections.
    .read_r       = pfsdev_read,
    .seek_r       = pfsdev_seek,
    .fstat_r      = pfsdev_fstat,
    .stat_r       = pfsdev_stat,
    .link_r       = rodev_link,                                 ///< Not supported by Partition FS sections.
    .unlink_r     = rodev_unlink,                               ///< Not supported by Partition FS sections.
    .chdir_r      = rodev_chdir,                                ///< No need to deal with cwd shenanigans, so we won't support it.
    .rename_r     = rodev_rename,                               ///< Not supported by Partition FS sections.
    .mkdir_r      = rodev_mkdir,                                ///< Not supported by Partition FS sections.
    .dirStateSize = sizeof(PartitionFileSystemDirectoryState),
    .diropen_r    = pfsdev_diropen,                             ///< Partition FS sections don't support directories, but we'll allow operations on the FS root.
    .dirreset_r   = pfsdev_dirreset,                            ///< Partition FS sections don't support directories, but we'll allow operations on the FS root.
    .dirnext_r    = pfsdev_dirnext,                             ///< Partition FS sections don't support directories, but we'll allow operations on the FS root.
    .dirclose_r   = pfsdev_dirclose,                            ///< Partition FS sections don't support directories, but we'll allow operations on the FS root.
    .statvfs_r    = pfsdev_statvfs,
    .ftruncate_r  = rodev_ftruncate,                            ///< Not supported by Partition FS sections.
    .fsync_r      = rodev_fsync,                                ///< Not supported by Partition FS sections.
    .deviceData   = NULL,
    .chmod_r      = rodev_chmod,                                ///< Not supported by Partition FS sections.
    .fchmod_r     = rodev_fchmod,                               ///< Not supported by Partition FS sections.
    .rmdir_r      = rodev_rmdir,                                ///< Not supported by Partition FS sections.
    .lstat_r      = pfsdev_stat,                                ///< Symlinks aren't supported, so we'll just alias lstat() to stat().
    .utimes_r     = rodev_utimes,                               ///< Not supported by Partition FS sections.
    .fpathconf_r  = rodev_fpathconf,                            ///< Not supported by Partition FS sections.
    .pathconf_r   = rodev_pathconf,                             ///< Not supported by Partition FS sections.
    .symlink_r    = rodev_symlink,                              ///< Not supported by Partition FS sections.
    .readlink_r   = rodev_readlink                              ///< Not supported by Partition FS sections.
};

const devoptab_t *pfsdev_get_devoptab()
{
    return &pfsdev_devoptab;
}

static int pfsdev_open(struct _reent *r, void *fd, const char *path, int flags, int mode)
{
    NX_IGNORE_ARG(mode);

    PFS_DEV_INIT_FILE_VARS;
    PFS_DEV_INIT_FS_ACCESS;

    /* Validate input. */
    if (!file || (flags & (O_WRONLY | O_RDWR | O_APPEND | O_CREAT | O_TRUNC | O_EXCL))) DEVOPTAB_SET_ERROR_AND_EXIT(EROFS);

    /* Get truncated path. */
    if (!(path = pfsdev_get_truncated_path(r, path))) DEVOPTAB_EXIT;

    //LOG_MSG_DEBUG("Opening \"%s:/%s\" with flags 0x%X.", dev_ctx->name, path, flags);

    /* Reset file descriptor. */
    memset(file, 0, sizeof(PartitionFileSystemFileState));

    /* Get information about the requested Partition FS entry. */
    if (!pfsGetEntryIndexByName(fs_ctx, path, &(file->index)) || !(file->pfs_entry = pfsGetEntryByIndex(fs_ctx, file->index)) || \
        !(file->name = pfsGetEntryNameByIndex(fs_ctx, file->index))) DEVOPTAB_SET_ERROR(ENOENT);

end:
    DEVOPTAB_DEINIT_VARS;
    DEVOPTAB_RETURN_INT(0);
}

static int pfsdev_close(struct _reent *r, void *fd)
{
    PFS_DEV_INIT_FILE_VARS;

    /* Sanity check. */
    if (!file) DEVOPTAB_SET_ERROR_AND_EXIT(EINVAL);

    //LOG_MSG_DEBUG("Closing \"%s:/%s\".", dev_ctx->name, file->name);

    /* Reset file descriptor. */
    memset(file, 0, sizeof(PartitionFileSystemFileState));

end:
    DEVOPTAB_DEINIT_VARS;
    DEVOPTAB_RETURN_INT(0);
}

static ssize_t pfsdev_read(struct _reent *r, void *fd, char *ptr, size_t len)
{
    PFS_DEV_INIT_FILE_VARS;
    PFS_DEV_INIT_FS_ACCESS;

    /* Sanity check. */
    if (!file || !ptr || !len) DEVOPTAB_SET_ERROR_AND_EXIT(EINVAL);

    //LOG_MSG_DEBUG("Reading 0x%lX byte(s) at offset 0x%lX from \"%s:/%s\".", len, file->offset, dev_ctx->name, file->name);

    /* Read file data. */
    if (!pfsReadEntryData(fs_ctx, file->pfs_entry, ptr, len, file->offset)) DEVOPTAB_SET_ERROR_AND_EXIT(EIO);

    /* Adjust offset. */
    file->offset += len;

end:
    DEVOPTAB_DEINIT_VARS;
    DEVOPTAB_RETURN_INT((ssize_t)len);
}

static off_t pfsdev_seek(struct _reent *r, void *fd, off_t pos, int dir)
{
    off_t offset = 0;

    PFS_DEV_INIT_FILE_VARS;

    /* Sanity check. */
    if (!file) DEVOPTAB_SET_ERROR_AND_EXIT(EINVAL);

    /* Find the offset to seek from. */
    switch(dir)
    {
        case SEEK_SET:  /* Set absolute position relative to zero (start offset). */
            break;
        case SEEK_CUR:  /* Set position relative to the current position. */
            offset = (off_t)file->offset;
            break;
        case SEEK_END:  /* Set position relative to EOF. */
            offset = (off_t)file->pfs_entry->size;
            break;
        default:        /* Invalid option. */
            DEVOPTAB_SET_ERROR_AND_EXIT(EINVAL);
    }

    /* Don't allow negative seeks beyond the beginning of file. */
    if (pos < 0 && offset < -pos) DEVOPTAB_SET_ERROR_AND_EXIT(EINVAL);

    /* Calculate actual offset. */
    offset += pos;

    /* Don't allow positive seeks beyond the end of file. */
    if (offset > (off_t)file->pfs_entry->size) DEVOPTAB_SET_ERROR_AND_EXIT(EOVERFLOW);

    //LOG_MSG_DEBUG("Seeking to offset 0x%lX from \"%s:/%s\".", offset, dev_ctx->name, file->name);

    /* Adjust offset. */
    file->offset = (u64)offset;

end:
    DEVOPTAB_DEINIT_VARS;
    DEVOPTAB_RETURN_INT(offset);
}

static int pfsdev_fstat(struct _reent *r, void *fd, struct stat *st)
{
    PFS_DEV_INIT_FILE_VARS;

    /* Sanity check. */
    if (!file || !st) DEVOPTAB_SET_ERROR_AND_EXIT(EINVAL);

    //LOG_MSG_DEBUG("Getting file stats for \"%s:/%s\".", dev_ctx->name, file->name);

    /* Fill stat info. */
    pfsdev_fill_stat(st, file->index, file->pfs_entry, dev_ctx->mount_time);

end:
    DEVOPTAB_DEINIT_VARS;
    DEVOPTAB_RETURN_INT(0);
}

static int pfsdev_stat(struct _reent *r, const char *file, struct stat *st)
{
    u32 index = 0;
    PartitionFileSystemEntry *pfs_entry = NULL;

    PFS_DEV_INIT_VARS;
    PFS_DEV_INIT_FS_ACCESS;

    /* Sanity check. */
    if (!st) DEVOPTAB_SET_ERROR_AND_EXIT(EINVAL);

    /* Get truncated path. */
    if (!(file = pfsdev_get_truncated_path(r, file))) DEVOPTAB_EXIT;

    //LOG_MSG_DEBUG("Getting file stats for \"%s:/%s\".", dev_ctx->name, file);

    /* Get information about the requested Partition FS entry. */
    if (!pfsGetEntryIndexByName(fs_ctx, file, &index) || !(pfs_entry = pfsGetEntryByIndex(fs_ctx, index))) DEVOPTAB_SET_ERROR_AND_EXIT(ENOENT);

    /* Fill stat info. */
    pfsdev_fill_stat(st, index, pfs_entry, dev_ctx->mount_time);

end:
    DEVOPTAB_DEINIT_VARS;
    DEVOPTAB_RETURN_INT(0);
}

static DIR_ITER *pfsdev_diropen(struct _reent *r, DIR_ITER *dirState, const char *path)
{
    DIR_ITER *ret = NULL;

    PFS_DEV_INIT_DIR_VARS;

    /* Get truncated path. */
    /* We can only work with the FS root here, so we won't accept anything else. */
    if (!(path = pfsdev_get_truncated_path(r, path))) DEVOPTAB_EXIT;
    if (*path) DEVOPTAB_SET_ERROR_AND_EXIT(ENOENT);

    //LOG_MSG_DEBUG("Opening directory \"%s:/\".", dev_ctx->name);

    /* Reset directory state. */
    memset(dir, 0, sizeof(PartitionFileSystemDirectoryState));

    /* Update return value. */
    ret = dirState;

end:
    DEVOPTAB_DEINIT_VARS;
    DEVOPTAB_RETURN_PTR(ret);
}

static int pfsdev_dirreset(struct _reent *r, DIR_ITER *dirState)
{
    PFS_DEV_INIT_DIR_VARS;

    //LOG_MSG_DEBUG("Resetting directory state for \"%s:/\".", dev_ctx->name);

    /* Reset directory state. */
    dir->state = 0;
    dir->index = 0;

end:
    DEVOPTAB_DEINIT_VARS;
    DEVOPTAB_RETURN_INT(0);
}

static int pfsdev_dirnext(struct _reent *r, DIR_ITER *dirState, char *filename, struct stat *filestat)
{
    PartitionFileSystemEntry *pfs_entry = NULL;
    const char *fname = NULL;

    PFS_DEV_INIT_DIR_VARS;
    PFS_DEV_INIT_FS_ACCESS;

    /* Sanity check. */
    if (!filename || !filestat) DEVOPTAB_SET_ERROR_AND_EXIT(EINVAL);

    //LOG_MSG_DEBUG("Getting info for next directory entry in \"%s:/\" (state %u, index %u).", dev_ctx->name, dir->state, dir->index);

    if (dir->state < 2)
    {
        /* Fill bogus directory entry. */
        memset(filestat, 0, sizeof(struct stat));

        filestat->st_nlink = (2 + pfsGetEntryCount(fs_ctx)); // One for self, one for parent.
        filestat->st_mode = (S_IFDIR | S_IRUSR | S_IRGRP | S_IROTH);
        filestat->st_atime = filestat->st_mtime = filestat->st_ctime = dev_ctx->mount_time;

        strcpy(filename, dir->state == 0 ? "." : "..");

        /* Update state. */
        dir->state++;

        DEVOPTAB_EXIT;
    }

    /* Check if we haven't reached EOD. */
    if (dir->index >= pfsGetEntryCount(fs_ctx)) DEVOPTAB_SET_ERROR_AND_EXIT(ENOENT);

    /* Get Partition FS entry. */
    if (!(pfs_entry = pfsGetEntryByIndex(fs_ctx, dir->index)) || !(fname = pfsGetEntryName(fs_ctx, pfs_entry))) DEVOPTAB_SET_ERROR_AND_EXIT(EIO);
    if (strlen(fname) > NAME_MAX) DEVOPTAB_SET_ERROR_AND_EXIT(ENAMETOOLONG);

    /* Copy filename. */
    strcpy(filename, fname);

    /* Fill stat info. */
    pfsdev_fill_stat(filestat, dir->index, pfs_entry, dev_ctx->mount_time);

    /* Adjust index. */
    dir->index++;

end:
    DEVOPTAB_DEINIT_VARS;
    DEVOPTAB_RETURN_INT(0);
}

static int pfsdev_dirclose(struct _reent *r, DIR_ITER *dirState)
{
    PFS_DEV_INIT_DIR_VARS;

    //LOG_MSG_DEBUG("Closing directory \"%s:/\".", dev_ctx->name);

    /* Reset directory state. */
    memset(dir, 0, sizeof(PartitionFileSystemDirectoryState));

end:
    DEVOPTAB_DEINIT_VARS;
    DEVOPTAB_RETURN_INT(0);
}

static int pfsdev_statvfs(struct _reent *r, const char *path, struct statvfs *buf)
{
    NX_IGNORE_ARG(path);

    u64 ext_fs_size = 0;

    PFS_DEV_INIT_VARS;
    PFS_DEV_INIT_FS_ACCESS;

    /* Sanity check. */
    if (!buf) DEVOPTAB_SET_ERROR_AND_EXIT(EINVAL);

    //LOG_MSG_DEBUG("Getting filesystem stats for \"%s:\"", dev_ctx->name);

    /* Get Partition FS total data size. */
    if (!pfsGetTotalDataSize(fs_ctx, &ext_fs_size)) DEVOPTAB_SET_ERROR_AND_EXIT(EIO);

    /* Fill filesystem stats. */
    memset(buf, 0, sizeof(struct statvfs));

    buf->f_bsize = 1;
    buf->f_frsize = 1;
    buf->f_blocks = ext_fs_size;
    buf->f_bfree = 0;
    buf->f_bavail = 0;
    buf->f_files = 0;
    buf->f_ffree = 0;
    buf->f_favail = 0;
    buf->f_fsid = 0;
    buf->f_flag = ST_NOSUID;
    buf->f_namemax = FS_MAX_PATH;

end:
    DEVOPTAB_DEINIT_VARS;
    DEVOPTAB_RETURN_INT(0);
}

static const char *pfsdev_get_truncated_path(struct _reent *r, const char *path)
{
    const u8 *p = (const u8*)path;
    ssize_t units = 0;
    u32 code = 0;
    size_t len = 0;
    bool path_sep_skipped = false;

    DEVOPTAB_DECL_ERROR_STATE;

    if (!r || !path || !*path) DEVOPTAB_SET_ERROR_AND_EXIT(EINVAL);

    //LOG_MSG_DEBUG("Input path: \"%s\".", path);

    /* Move the path pointer to the start of the actual path. */
    do {
        units = decode_utf8(&code, p);
        if (units < 0) DEVOPTAB_SET_ERROR_AND_EXIT(EILSEQ);
        p += units;
    } while(code >= ' ' && code != ':');

    /* We found a colon; p points to the actual path. */
    if (code == ':') path = (const char*)p;

    /* Skip the leading slash, if available. */
    if (path[0] == '/')
    {
        path++;
        path_sep_skipped = true;
    }

    /* Make sure there are no more colons or slashes and that the remainder of the string is valid UTF-8. */
    p = (const u8*)path;

    do {
        units = decode_utf8(&code, p);
        if (units < 0) DEVOPTAB_SET_ERROR_AND_EXIT(EILSEQ);
        if (code == ':' || code == '/') DEVOPTAB_SET_ERROR_AND_EXIT(EINVAL);
        p += units;
    } while(code >= ' ');

    /* Verify fixed path length. */
    len = strlen(path);
    if (!len && !path_sep_skipped) DEVOPTAB_SET_ERROR_AND_EXIT(EINVAL);
    if (len >= FS_MAX_PATH) DEVOPTAB_SET_ERROR_AND_EXIT(ENAMETOOLONG);

    //LOG_MSG_DEBUG("Truncated path: \"%s\".", path);

end:
    DEVOPTAB_RETURN_PTR(path);
}

static void pfsdev_fill_stat(struct stat *st, u32 index, const PartitionFileSystemEntry *pfs_entry, time_t mount_time)
{
    /* Clear stat struct. */
    memset(st, 0, sizeof(struct stat));

    /* Fill stat struct. */
    /* We're always dealing with a file entry. */
    st->st_ino = index;
    st->st_mode = (S_IFREG | S_IRUSR | S_IRGRP | S_IROTH);
    st->st_nlink = 1;
    st->st_size = (off_t)pfs_entry->size;
    st->st_atime = st->st_mtime = st->st_ctime = mount_time;
}