/*
 * Copyright (c) 2013-2019 Huawei Technologies Co., Ltd. All rights reserved.
 * Copyright (c) 2020-2022 Huawei Device Co., Ltd. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice, this list of
 *    conditions and the following disclaimer.
 *
 * 2. 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.
 *
 * 3. Neither the name of the copyright holder nor the names of its contributors may be used
 *    to endorse or promote products derived from this software without specific prior written
 *    permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "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 COPYRIGHT HOLDER OR
 * CONTRIBUTORS 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.
 */

#include "los_config.h"
#include "sys/mount.h"

#ifdef LOSCFG_SHELL

#include "los_typedef.h"
#include "shell.h"
#include "sys/stat.h"
#include "stdlib.h"
#include "unistd.h"
#include "fcntl.h"
#include "sys/statfs.h"
#include "stdio.h"
#include "pthread.h"

#include "shcmd.h"
#include "securec.h"
#include "show.h"
#include "los_syscall.h"

#include "los_process_pri.h"
#include <ctype.h>
#include "fs/fs_operation.h"

typedef enum {
    RM_RECURSIVER,
    RM_FILE,
    RM_DIR,
    CP_FILE,
    CP_COUNT
} wildcard_type;

#define ERROR_OUT_IF(condition, message_function, handler) \
    do { \
        if (condition) { \
            message_function; \
            handler; \
        } \
    } while (0)

static inline void set_err(int errcode, const char *err_message)
{
    set_errno(errcode);
    perror(err_message);
}

int osShellCmdDoChdir(const char *path)
{
    char *fullpath = NULL;
    char *fullpath_bak = NULL;
    int ret;
    char *shell_working_directory = OsShellGetWorkingDirectory();
    if (shell_working_directory == NULL) {
        return -1;
    }

    if (path == NULL) {
        LOS_TaskLock();
        PRINTK("%s\n", shell_working_directory);
        LOS_TaskUnlock();

        return 0;
    }

    ERROR_OUT_IF(strlen(path) > PATH_MAX, set_err(ENOTDIR, "cd error"), return -1);

    ret = vfs_normalize_path(shell_working_directory, path, &fullpath);
    ERROR_OUT_IF(ret < 0, set_err(-ret, "cd error"), return -1);

    fullpath_bak = fullpath;
    ret = chdir(fullpath);
    if (ret < 0) {
        free(fullpath_bak);
        perror("cd");
        return -1;
    }

    /* copy full path to working directory */

    LOS_TaskLock();
    ret = strncpy_s(shell_working_directory, PATH_MAX, fullpath, strlen(fullpath));
    if (ret != EOK) {
        free(fullpath_bak);
        LOS_TaskUnlock();
        return -1;
    }
    LOS_TaskUnlock();
    /* release normalize directory path name */

    free(fullpath_bak);

    return 0;
}

int osShellCmdLs(int argc, const char **argv)
{
    char *fullpath = NULL;
    const char *filename = NULL;
    int ret;
    char *shell_working_directory = OsShellGetWorkingDirectory();
    if (shell_working_directory == NULL) {
        return -1;
    }

    ERROR_OUT_IF(argc > 1, PRINTK("ls or ls [DIRECTORY]\n"), return -1);

    if (argc == 0) {
        ls(shell_working_directory);
        return 0;
    }

    filename = argv[0];
    ret = vfs_normalize_path(shell_working_directory, filename, &fullpath);
    ERROR_OUT_IF(ret < 0, set_err(-ret, "ls error"), return -1);

    ls(fullpath);
    free(fullpath);

    return 0;
}

int osShellCmdCd(int argc, const char **argv)
{
    if (argc == 0) {
        (void)osShellCmdDoChdir("/");
        return 0;
    }

    (void)osShellCmdDoChdir(argv[0]);

    return 0;
}

#define CAT_BUF_SIZE  512
#define CAT_TASK_PRIORITY  10
#define CAT_TASK_STACK_SIZE  0x3000
pthread_mutex_t g_mutex_cat = PTHREAD_MUTEX_INITIALIZER;

int osShellCmdDoCatShow(UINTPTR arg)
{
    int ret = 0;
    char buf[CAT_BUF_SIZE];
    size_t size, written, toWrite;
    ssize_t cnt;
    char *fullpath = (char *)arg;
    FILE *ini = NULL;

    (void)pthread_mutex_lock(&g_mutex_cat);
    ini = fopen(fullpath, "r");
    if (ini == NULL) {
        ret = -1;
        perror("cat error");
        goto out;
    }

    do {
        (void)memset_s(buf, sizeof(buf), 0, CAT_BUF_SIZE);
        size = fread(buf, 1, CAT_BUF_SIZE, ini);
        if ((int)size < 0) {
            ret = -1;
            perror("cat error");
            goto out_with_fclose;
        }

        for (toWrite = size, written = 0; toWrite > 0;) {
            cnt = write(1, buf + written, toWrite);
            if (cnt == 0) {
                /* avoid task-starvation */
                (void)LOS_TaskDelay(1);
                continue;
            } else if (cnt < 0) {
                perror("cat write error");
                break;
            }

            written += cnt;
            toWrite -= cnt;
        }
    }
    while (size > 0);

out_with_fclose:
    (void)fclose(ini);
out:
    free(fullpath);
    (void)pthread_mutex_unlock(&g_mutex_cat);
    return ret;
}

int osShellCmdCat(int argc, const char **argv)
{
    char *fullpath = NULL;
    int ret;
    unsigned int ca_task;
    struct Vnode *vnode = NULL;
    TSK_INIT_PARAM_S init_param;
    char *shell_working_directory = OsShellGetWorkingDirectory();
    if (shell_working_directory == NULL) {
        return -1;
    }

    ERROR_OUT_IF(argc != 1, PRINTK("cat [FILE]\n"), return -1);

    ret = vfs_normalize_path(shell_working_directory, argv[0], &fullpath);
    ERROR_OUT_IF(ret < 0, set_err(-ret, "cat error"), return -1);

    VnodeHold();
    ret = VnodeLookup(fullpath, &vnode, O_RDONLY);
    if (ret != LOS_OK) {
        set_errno(-ret);
        perror("cat error");
        VnodeDrop();
        free(fullpath);
        return -1;
    }
    if (vnode->type != VNODE_TYPE_REG) {
        set_errno(EINVAL);
        perror("cat error");
        VnodeDrop();
        free(fullpath);
        return -1;
    }
    VnodeDrop();
    (void)memset_s(&init_param, sizeof(init_param), 0, sizeof(TSK_INIT_PARAM_S));
    init_param.pfnTaskEntry = (TSK_ENTRY_FUNC)osShellCmdDoCatShow;
    init_param.usTaskPrio   = CAT_TASK_PRIORITY;
    init_param.auwArgs[0]   = (UINTPTR)fullpath;
    init_param.uwStackSize  = CAT_TASK_STACK_SIZE;
    init_param.pcName       = "shellcmd_cat";
    init_param.uwResved     = LOS_TASK_STATUS_DETACHED | OS_TASK_FLAG_SPECIFIES_PROCESS;
    init_param.processID    = 2; /* 2: kProcess */

    ret = (int)LOS_TaskCreate(&ca_task, &init_param);
    if (ret != LOS_OK) {
        free(fullpath);
    }

    return ret;
}

static int nfs_mount_ref(const char *server_ip_and_path, const char *mount_path,
                         unsigned int uid, unsigned int gid) __attribute__((weakref("nfs_mount")));

static unsigned long get_mountflags(const char *options)
{
    unsigned long mountfalgs = 0;
    char *p;
    while ((options != NULL) && (p = strsep((char**)&options, ",")) != NULL) {
        if (strncmp(p, "ro", strlen("ro")) == 0) {
            mountfalgs |= MS_RDONLY;
        } else if (strncmp(p, "rw", strlen("rw")) == 0) {
            mountfalgs &= ~MS_RDONLY;
        } else if (strncmp(p, "nosuid", strlen("nosuid")) == 0) {
            mountfalgs |= MS_NOSUID;
        } else if (strncmp(p, "suid", strlen("suid")) == 0) {
            mountfalgs &= ~MS_NOSUID;
        } else {
            continue;
        }
    }

    return mountfalgs;
}
static inline void print_mount_usage(void)
{
    PRINTK("mount [DEVICE] [PATH] [NAME]\n");
}

int osShellCmdMount(int argc, const char **argv)
{
    int ret;
    char *fullpath = NULL;
    const char *filename = NULL;
    unsigned int gid, uid;
    char *data = NULL;
    char *filessystemtype = NULL;
    unsigned long mountfalgs;
    char *shell_working_directory = OsShellGetWorkingDirectory();
    if (shell_working_directory == NULL) {
        return -1;
    }

    ERROR_OUT_IF(argc < 3, print_mount_usage(), return OS_FAIL);

    if (strncmp(argv[0], "-t", 2) == 0 || strncmp(argv[0], "-o", 2) == 0) // 2: length of "-t"
    {
        if (argc < 4) { // 4: required number of parameters
            PRINTK("mount -t/-o [DEVICE] [PATH] [NAME]\n");
            return -1;
        }

        filename = argv[2]; // 2: index of file path
        ret = vfs_normalize_path(shell_working_directory, filename, &fullpath);
        ERROR_OUT_IF(ret < 0, set_err(-ret, "mount error"), return -1);

        if (strncmp(argv[3], "nfs", 3) == 0) { // 3: index of fs type
            if (argc <= 6) { // 6: arguments include uid or gid
                uid = ((argc >= 5) && (argv[4] != NULL)) ? (unsigned int)strtoul(argv[4], (char **)NULL, 0) : 0;
                gid = ((argc == 6) && (argv[5] != NULL)) ? (unsigned int)strtoul(argv[5], (char **)NULL, 0) : 0;
   
                if (nfs_mount_ref != NULL) {
                    ret = nfs_mount_ref(argv[1], fullpath, uid, gid);
                    if (ret != LOS_OK) {
                        PRINTK("mount -t [DEVICE] [PATH] [NAME]\n");
                    }
                } else {
                    PRINTK("can't find nfs_mount\n");
                }
                free(fullpath);
                return 0;
            }
        }

        filessystemtype = (argc >= 4) ? (char *)argv[3] : NULL; /* 4: specify fs type, 3: fs type */
        mountfalgs = (argc >= 5) ? get_mountflags((const char *)argv[4]) : 0; /* 4: usr option */
        data = (argc >= 6) ? (char *)argv[5] : NULL; /* 5: usr option data, 6: six args needed for data */

        if (strcmp(argv[1], "0") == 0) {
            ret = mount((const char *)NULL, fullpath, filessystemtype, mountfalgs, data);
        } else {
            ret = mount(argv[1], fullpath, filessystemtype, mountfalgs, data); /* 3: fs type */
        }
        if (ret != LOS_OK) {
            perror("mount error");
        } else {
            PRINTK("mount ok\n");
        }
    } else {
        filename = argv[1];
        ret = vfs_normalize_path(shell_working_directory, filename, &fullpath);
        ERROR_OUT_IF(ret < 0, set_err(-ret, "mount error"), return -1);

        if (strncmp(argv[2], "nfs", 3) == 0) { // 2: index of fs type, 3: length of "nfs"
            if (argc <= 5) { // 5: arguments include gid and uid
                uid = ((argc >= 4) && (argv[3] != NULL)) ? (unsigned int)strtoul(argv[3], (char **)NULL, 0) : 0;
                gid = ((argc == 5) && (argv[4] != NULL)) ? (unsigned int)strtoul(argv[4], (char **)NULL, 0) : 0;

                if (nfs_mount_ref != NULL) {
                    ret = nfs_mount_ref(argv[0], fullpath, uid, gid);
                    if (ret != LOS_OK) {
                        PRINTK("mount [DEVICE] [PATH] [NAME]\n");
                    }
                } else {
                    PRINTK("can't find nfs_mount\n");
                }
                free(fullpath);
                return 0;
            }

            print_mount_usage();
            free(fullpath);
            return 0;
        }

        mountfalgs = (argc >= 4) ? get_mountflags((const char *)argv[3]) : 0;  /* 3: usr option */
        data = (argc >= 5) ? (char *)argv[4] : NULL; /* 4: usr option data, 5: number of args needed for data */

        if (strcmp(argv[0], "0") == 0) {
            ret = mount((const char *)NULL, fullpath, argv[2], mountfalgs, data);
        } else {
            ret = mount(argv[0], fullpath, argv[2], mountfalgs, data);  /* 2: fs type */
        }
        if (ret != LOS_OK) {
            perror("mount error");
        } else {
            PRINTK("mount ok\n");
        }
    }

    free(fullpath);
    return 0;
}

int osShellCmdUmount(int argc, const char **argv)
{
    int ret;
    const char *filename = NULL;
    char *fullpath = NULL;
    char *target_path = NULL;
    int cmp_num;
    char *work_path = NULL;
    char *shell_working_directory = OsShellGetWorkingDirectory();
    if (shell_working_directory == NULL) {
        return -1;
    }
    work_path = shell_working_directory;

    ERROR_OUT_IF(argc == 0, PRINTK("umount [PATH]\n"), return 0);

    filename = argv[0];
    ret = vfs_normalize_path(shell_working_directory, filename, &fullpath);
    ERROR_OUT_IF(ret < 0, set_err(-ret, "umount error"), return -1);

    target_path = fullpath;
    cmp_num = strlen(fullpath);
    ret = strncmp(work_path, target_path, cmp_num);
    if (ret == 0) {
        work_path += cmp_num;
        if (*work_path == '/' || *work_path == '\0') {
            set_errno(EBUSY);
            perror("umount error");
            free(fullpath);
            return -1;
        }
    }

    ret = umount(fullpath);
    free(fullpath);
    if (ret != LOS_OK) {
        perror("umount error");
        return 0;
    }

    PRINTK("umount ok\n");
    return 0;
}

int osShellCmdMkdir(int argc, const char **argv)
{
    int ret;
    char *fullpath = NULL;
    const char *filename = NULL;
    char *shell_working_directory = OsShellGetWorkingDirectory();
    if (shell_working_directory == NULL) {
        return -1;
    }

    ERROR_OUT_IF(argc != 1, PRINTK("mkdir [DIRECTORY]\n"), return 0);

    filename = argv[0];
    ret = vfs_normalize_path(shell_working_directory, filename, &fullpath);
    ERROR_OUT_IF(ret < 0, set_err(-ret, "mkdir error"), return -1);

    ret = mkdir(fullpath, S_IRWXU | S_IRWXG | S_IRWXO);
    if (ret == -1) {
        perror("mkdir error");
    }
    free(fullpath);
    return 0;
}

int osShellCmdPwd(int argc, const char **argv)
{
    char buf[SHOW_MAX_LEN] = {0};
    DIR *dir = NULL;
    char *shell_working_directory = OsShellGetWorkingDirectory();
    if (shell_working_directory == NULL) {
        return -1;
    }

    ERROR_OUT_IF(argc > 0, PRINTK("\nUsage: pwd\n"), return -1);

    dir = opendir(shell_working_directory);
    if (dir == NULL) {
        perror("pwd error");
        return -1;
    }

    LOS_TaskLock();
    if (strncpy_s(buf, SHOW_MAX_LEN, shell_working_directory, SHOW_MAX_LEN - 1) != EOK) {
        LOS_TaskUnlock();
        PRINTK("pwd error: strncpy_s error!\n");
        (void)closedir(dir);
        return -1;
    }
    LOS_TaskUnlock();

    PRINTK("%s\n", buf);
    (void)closedir(dir);
    return 0;
}

static inline void print_statfs_usage(void)
{
    PRINTK("Usage  :\n");
    PRINTK("    statfs <path>\n");
    PRINTK("    path  : Mounted file system path that requires query information\n");
    PRINTK("Example:\n");
    PRINTK("    statfs /ramfs\n");
}

int osShellCmdStatfs(int argc, const char **argv)
{
    struct statfs sfs;
    int result;
    unsigned long long total_size, free_size;
    char *fullpath = NULL;
    const char *filename = NULL;
    char *shell_working_directory = OsShellGetWorkingDirectory();
    if (shell_working_directory == NULL) {
        return -1;
    }

    ERROR_OUT_IF(argc != 1, PRINTK("statfs failed! Invalid argument!\n"), return -1);

    (void)memset_s(&sfs, sizeof(sfs), 0, sizeof(sfs));

    filename = argv[0];
    result = vfs_normalize_path(shell_working_directory, filename, &fullpath);
    ERROR_OUT_IF(result < 0, set_err(-result, "statfs error"), return -1);

    result = statfs(fullpath, &sfs);
    free(fullpath);

    if (result != 0 || sfs.f_type == 0) {
        PRINTK("statfs failed! Invalid argument!\n");
        print_statfs_usage();
        return -1;
    }

    total_size  = (unsigned long long)sfs.f_bsize * sfs.f_blocks;
    free_size   = (unsigned long long)sfs.f_bsize * sfs.f_bfree;

    PRINTK("statfs got:\n f_type     = %d\n cluster_size   = %d\n", sfs.f_type, sfs.f_bsize);
    PRINTK(" total_clusters = %llu\n free_clusters  = %llu\n", sfs.f_blocks, sfs.f_bfree);
    PRINTK(" avail_clusters = %llu\n f_namelen    = %d\n", sfs.f_bavail, sfs.f_namelen);
    PRINTK("\n%s\n total size: %4llu Bytes\n free  size: %4llu Bytes\n", argv[0], total_size, free_size);

    return 0;
}

int osShellCmdTouch(int argc, const char **argv)
{
    int ret;
    int fd = -1;
    char *fullpath = NULL;
    const char *filename = NULL;
    char *shell_working_directory = OsShellGetWorkingDirectory();
    if (shell_working_directory == NULL) {
        return -1;
    }

    ERROR_OUT_IF(argc != 1, PRINTK("touch [FILE]\n"), return -1);

    filename = argv[0];
    ret = vfs_normalize_path(shell_working_directory, filename, &fullpath);
    ERROR_OUT_IF(ret < 0, set_err(-ret, "touch error"), return -1);

    fd = open(fullpath, O_RDWR | O_CREAT, S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH);
    free(fullpath);
    if (fd == -1) {
        perror("touch error");
        return -1;
    }

    (void)close(fd);
    return 0;
}

#define CP_BUF_SIZE 4096
pthread_mutex_t g_mutex_cp = PTHREAD_MUTEX_INITIALIZER;

static int os_shell_cmd_do_cp(const char *src_filepath, const char *dst_filename)
{
    int  ret;
    char *src_fullpath = NULL;
    char *dst_fullpath = NULL;
    const char *src_filename = NULL;
    char *dst_filepath = NULL;
    char *buf = NULL;
    const char *filename = NULL;
    ssize_t r_size, w_size;
    int src_fd = -1;
    int dst_fd = -1;
    struct stat stat_buf;
    mode_t src_mode;
    char *shell_working_directory = OsShellGetWorkingDirectory();
    if (shell_working_directory == NULL) {
        return -1;
    }

    buf = (char *)malloc(CP_BUF_SIZE);
    if (buf == NULL) {
        PRINTK("cp error: Out of memory!\n");
        return -1;
    }

    /* Get source fullpath. */

    ret = vfs_normalize_path(shell_working_directory, src_filepath, &src_fullpath);
    if (ret < 0) {
        set_errno(-ret);
        PRINTK("cp error: %s\n", strerror(errno));
        free(buf);
        return -1;
    }

    /* Is source path exist? */

    ret = stat(src_fullpath, &stat_buf);
    if (ret == -1) {
        PRINTK("cp %s error: %s\n", src_fullpath, strerror(errno));
        goto errout_with_srcpath;
    }
    src_mode = stat_buf.st_mode;
    /* Is source path a directory? */

    if (S_ISDIR(stat_buf.st_mode)) {
        PRINTK("cp %s error: Source file can't be a directory.\n", src_fullpath);
        goto errout_with_srcpath;
    }

    /* Get dest fullpath. */

    dst_fullpath = strdup(dst_filename);
    if (dst_fullpath == NULL) {
        PRINTK("cp error: Out of memory.\n");
        goto errout_with_srcpath;
    }

    /* Is dest path exist? */

    ret = stat(dst_fullpath, &stat_buf);
    if (ret == 0) {
        /* Is dest path a directory? */

        if (S_ISDIR(stat_buf.st_mode)) {
            /* Get source file name without '/'. */

            src_filename = src_filepath;
            while (1) {
                filename = strchr(src_filename, '/');
                if (filename == NULL) {
                    break;
                }
                src_filename = filename + 1;
            }

            /* Add the source file after dest path. */

            ret = vfs_normalize_path(dst_fullpath, src_filename, &dst_filepath);
            if (ret < 0) {
                set_errno(-ret);
                PRINTK("cp error. %s.\n", strerror(errno));
                goto errout_with_path;
            }
            free(dst_fullpath);
            dst_fullpath = dst_filepath;
        }
    }

    /* Is dest file same as source file? */

    if (strcmp(src_fullpath, dst_fullpath) == 0) {
        PRINTK("cp error: '%s' and '%s' are the same file\n", src_fullpath, dst_fullpath);
        goto errout_with_path;
    }

    /* Copy begins. */

    (void)pthread_mutex_lock(&g_mutex_cp);
    src_fd = open(src_fullpath, O_RDONLY);
    if (src_fd < 0) {
        PRINTK("cp error: can't open %s. %s.\n", src_fullpath, strerror(errno));
        goto errout_with_mutex;
    }

    dst_fd = open(dst_fullpath, O_CREAT | O_WRONLY | O_TRUNC, src_mode);
    if (dst_fd < 0) {
        PRINTK("cp error: can't create %s. %s.\n", dst_fullpath, strerror(errno));
        goto errout_with_srcfd;
    }

    do {
        (void)memset_s(buf, CP_BUF_SIZE, 0, CP_BUF_SIZE);
        r_size = read(src_fd, buf, CP_BUF_SIZE);
        if (r_size < 0) {
            PRINTK("cp %s %s failed. %s.\n", src_fullpath, dst_fullpath, strerror(errno));
            goto errout_with_fd;
        }
        w_size = write(dst_fd, buf, r_size);
        if (w_size != r_size) {
            PRINTK("cp %s %s failed. %s.\n", src_fullpath, dst_fullpath, strerror(errno));
            goto errout_with_fd;
        }
    } while (r_size == CP_BUF_SIZE);

    /* Release resource. */

    free(buf);
    free(src_fullpath);
    free(dst_fullpath);
    (void)close(src_fd);
    (void)close(dst_fd);
    (void)pthread_mutex_unlock(&g_mutex_cp);
    return LOS_OK;

errout_with_fd:
    (void)close(dst_fd);
errout_with_srcfd:
    (void)close(src_fd);
errout_with_mutex:
    (void)pthread_mutex_unlock(&g_mutex_cp);
errout_with_path:
    free(dst_fullpath);
errout_with_srcpath:
    free(src_fullpath);
    free(buf);
    return -1;
}

/* The separator and EOF for a directory fullpath: '/'and '\0' */

#define SEPARATOR_EOF_LEN 2

static int os_shell_cmd_do_rmdir(const char *pathname)
{
    struct dirent *dirent = NULL;
    struct stat stat_info;
    DIR *d = NULL;
    char *fullpath = NULL;
    int ret;

    (void)memset_s(&stat_info, sizeof(stat_info), 0, sizeof(struct stat));
    if (stat(pathname, &stat_info) != 0) {
        return -1;
    }

    if (S_ISREG(stat_info.st_mode) || S_ISLNK(stat_info.st_mode)) {
        return remove(pathname);
    }
    d = opendir(pathname);
    if (d == NULL) {
        return -1;
    }
    while (1) {
        dirent = readdir(d);
        if (dirent == NULL) {
            break;
        }
        if (strcmp(dirent->d_name, "..") && strcmp(dirent->d_name, ".")) {
            size_t fullpath_buf_size = strlen(pathname) + strlen(dirent->d_name) + SEPARATOR_EOF_LEN;
            fullpath = (char *)malloc(fullpath_buf_size);
            if (fullpath == NULL) {
                PRINTK("malloc failure!\n");
                (void)closedir(d);
                return -1;
            }
            ret = snprintf_s(fullpath, fullpath_buf_size, fullpath_buf_size - 1, "%s/%s", pathname, dirent->d_name);
            if (ret < 0) {
                PRINTK("name is too long!\n");
                free(fullpath);
                (void)closedir(d);
                return -1;
            }
            (void)os_shell_cmd_do_rmdir(fullpath);
            free(fullpath);
        }
    }
    (void)closedir(d);
    return rmdir(pathname);
}

/*  Wildcard matching operations  */

static int os_wildcard_match(const char *src, const char *filename)
{
    int ret;

    if (*src != '\0') {
        if (*filename == '*') {
            while ((*filename == '*') || (*filename == '?')) {
                filename++;
            }

            if (*filename == '\0') {
                return 0;
            }

            while (*src != '\0' && !(*src == *filename)) {
                src++;
            }

            if (*src == '\0') {
                return -1;
            }

            ret = os_wildcard_match(src, filename);

            while ((ret != 0) && (*(++src) != '\0')) {
                if (*src == *filename) {
                    ret = os_wildcard_match(src, filename);
                }
            }
            return ret;
        } else {
            if ((*src == *filename) || (*filename == '?')) {
                return os_wildcard_match(++src, ++filename);
            }
            return -1;
        }
    }

    while (*filename != '\0') {
        if (*filename != '*') {
            return -1;
        }
        filename++;
    }
    return 0;
}

/*   To determine whether a wildcard character exists in a path   */

static int os_is_containers_wildcard(const char *filename)
{
    while (*filename != '\0') {
        if ((*filename == '*') || (*filename == '?')) {
            return 1;
        }
        filename++;
    }
    return 0;
}

/*  Delete a matching file or directory  */

static int os_wildcard_delete_file_or_dir(const char *fullpath, wildcard_type mark)
{
    int ret;

    switch (mark) {
        case RM_RECURSIVER:
            ret = os_shell_cmd_do_rmdir(fullpath);
            break;
        case RM_FILE:
            ret = unlink(fullpath);
            break;
        case RM_DIR:
            ret = rmdir(fullpath);
            break;
        default:
            return VFS_ERROR;
    }
    if (ret == -1) {
        PRINTK("%s  ", fullpath);
        perror("rm/rmdir error!");
        return ret;
    }

    PRINTK("%s match successful!delete!\n", fullpath);
    return 0;
}

/*  Split the path with wildcard characters  */

static char* os_wildcard_split_path(char *fullpath, char **handle, char **wait)
{
    int n = 0;
    int a = 0;
    int b = 0;
    int len  = strlen(fullpath);

    for (n = 0; n < len; n++) {
        if (fullpath[n] == '/') {
            if (b != 0) {
                fullpath[n] = '\0';
                *wait = fullpath + n + 1;
                break;
            }
            a = n;
        } else if (fullpath[n] == '*' || fullpath[n] == '?') {
            b = n;
            fullpath[a] = '\0';
            if (a == 0) {
                *handle = fullpath + a + 1;
                continue;
            }
            *handle = fullpath + a + 1;
        }
    }
    return fullpath;
}

/*  Handling entry of the path with wildcard characters  */

static int os_wildcard_extract_directory(char *fullpath, void *dst, wildcard_type mark)
{
    char separator[] = "/";
    char src[PATH_MAX] = {0};
    struct dirent *dirent = NULL;
    char *f = NULL;
    char *s = NULL;
    char *t = NULL;
    int ret = 0;
    DIR *d = NULL;
    struct stat stat_buf;
    int deleteFlag = 0;

    f = os_wildcard_split_path(fullpath, &s, &t);

    if (s == NULL) {
        if (mark == CP_FILE) {
            ret = os_shell_cmd_do_cp(fullpath, dst);
        } else if (mark == CP_COUNT) {
            ret = stat(fullpath, &stat_buf);
            if (ret == 0 && (S_ISREG(stat_buf.st_mode) || S_ISLNK(stat_buf.st_mode))) {
                (*(int *)dst)++;
            }
        } else {
            ret = os_wildcard_delete_file_or_dir(fullpath, mark);
        }
        return ret;
    }

    d = (*f == '\0') ? opendir("/") : opendir(f);
    if (d == NULL) {
        perror("opendir error");
        return VFS_ERROR;
    }

    while (1) {
        dirent = readdir(d);
        if (dirent == NULL) {
            break;
        }

        ret = strcpy_s(src, PATH_MAX, f);
        if (ret != EOK) {
            goto closedir_out;
        }

        ret = os_wildcard_match(dirent->d_name, s);
        if (ret == 0) {
            ret = strcat_s(src, sizeof(src), separator);
            if (ret != EOK) {
                goto closedir_out;
            }
            ret = strcat_s(src, sizeof(src), dirent->d_name);
            if (ret != EOK) {
                goto closedir_out;
            }
            if (t == NULL) {
                if (mark == CP_FILE) {
                    ret = os_shell_cmd_do_cp(src, dst);
                } else if (mark == CP_COUNT) {
                    ret = stat(src, &stat_buf);
                    if (ret == 0 && (S_ISREG(stat_buf.st_mode) || S_ISLNK(stat_buf.st_mode))) {
                        (*(int *)dst)++;
                        if ((*(int *)dst) > 1) {
                            break;
                        }
                    }
                } else {
                    ret = os_wildcard_delete_file_or_dir(src, mark);
                    if (ret == 0) {
                        deleteFlag = 1;
                    }
                }
            } else {
                ret = strcat_s(src, sizeof(src), separator);
                if (ret != EOK) {
                    goto closedir_out;
                }
                ret = strcat_s(src, sizeof(src), t);
                if (ret != EOK) {
                    goto closedir_out;
                }
                ret = os_wildcard_extract_directory(src, dst, mark);
                if (mark == CP_COUNT && (*(int *)dst) > 1) {
                    break;
                }
            }
        }
    }
    (void)closedir(d);
    if (deleteFlag == 1) {
        ret = 0;
    }
    return ret;
closedir_out:
    (void)closedir(d);
    return VFS_ERROR;
}

int osShellCmdCp(int argc, const char **argv)
{
    int  ret;
    const char *src = NULL;
    const char *dst = NULL;
    char *src_fullpath = NULL;
    char *dst_fullpath = NULL;
    struct stat stat_buf;
    int count = 0;
    char *shell_working_directory = OsShellGetWorkingDirectory();
    if (shell_working_directory == NULL) {
        return -1;
    }

    ERROR_OUT_IF(argc < 2, PRINTK("cp [SOURCEFILE] [DESTFILE]\n"), return -1);

    src = argv[0];
    dst = argv[1];

    /* Get source fullpath. */

    ret = vfs_normalize_path(shell_working_directory, src, &src_fullpath);
    if (ret < 0) {
        set_errno(-ret);
        PRINTK("cp error:%s\n", strerror(errno));
        return -1;
    }

    if (src[strlen(src) - 1] == '/') {
        PRINTK("cp %s error: Source file can't be a directory.\n", src);
        goto errout_with_srcpath;
    }

    /* Get dest fullpath. */

    ret = vfs_normalize_path(shell_working_directory, dst, &dst_fullpath);
    if (ret < 0) {
        set_errno(-ret);
        PRINTK("cp error: can't open %s. %s\n", dst, strerror(errno));
        goto errout_with_srcpath;
    }

    /* Is dest path exist? */

    ret = stat(dst_fullpath, &stat_buf);
    if (ret < 0) {
        /* Is dest path a directory? */

        if (dst[strlen(dst) - 1] == '/') {
            PRINTK("cp error: %s, %s.\n", dst_fullpath, strerror(errno));
            goto errout_with_path;
        }
    } else {
        if ((S_ISREG(stat_buf.st_mode) || S_ISLNK(stat_buf.st_mode)) && dst[strlen(dst) - 1] == '/') {
            PRINTK("cp error: %s is not a directory.\n", dst_fullpath);
            goto errout_with_path;
        }
    }

    if (os_is_containers_wildcard(src_fullpath)) {
        if (ret < 0 || S_ISREG(stat_buf.st_mode) || S_ISLNK(stat_buf.st_mode)) {
            char *src_copy = strdup(src_fullpath);
            if (src_copy == NULL) {
                PRINTK("cp error : Out of memory.\n");
                goto errout_with_path;
            }
            (void)os_wildcard_extract_directory(src_copy, &count, CP_COUNT);
            free(src_copy);
            if (count > 1) {
                PRINTK("cp error : %s is not a directory.\n", dst_fullpath);
                goto errout_with_path;
            }
        }
        ret = os_wildcard_extract_directory(src_fullpath, dst_fullpath, CP_FILE);
    } else {
        ret = os_shell_cmd_do_cp(src_fullpath, dst_fullpath);
    }
    free(dst_fullpath);
    free(src_fullpath);
    return ret;

errout_with_path:
    free(dst_fullpath);
errout_with_srcpath:
    free(src_fullpath);
    return VFS_ERROR;
}

static inline void print_rm_usage(void)
{
    PRINTK("rm [FILE] or rm [-r/-R] [FILE]\n");
}

int osShellCmdRm(int argc, const char **argv)
{
    int  ret = 0;
    char *fullpath = NULL;
    const char *filename = NULL;
    char *shell_working_directory = OsShellGetWorkingDirectory();
    if (shell_working_directory == NULL) {
        return -1;
    }

    ERROR_OUT_IF(argc != 1 && argc != 2, print_rm_usage(), return -1);

    if (argc == 2) { // 2: arguments include "-r" or "-R"
        ERROR_OUT_IF(strcmp(argv[0], "-r") != 0 && strcmp(argv[0], "-R") != 0, print_rm_usage(), return -1);

        filename = argv[1];
        ret = vfs_normalize_path(shell_working_directory, filename, &fullpath);
        ERROR_OUT_IF(ret < 0, set_err(-ret, "rm error"), return -1);

        if (os_is_containers_wildcard(fullpath)) {
            ret = os_wildcard_extract_directory(fullpath, NULL, RM_RECURSIVER);
        } else {
            ret = os_shell_cmd_do_rmdir(fullpath);
        }
    } else {
        filename = argv[0];
        ret = vfs_normalize_path(shell_working_directory, filename, &fullpath);
        ERROR_OUT_IF(ret < 0, set_err(-ret, "rm error"), return -1);

        if (os_is_containers_wildcard(fullpath)) {
            ret = os_wildcard_extract_directory(fullpath, NULL, RM_FILE);
        } else {
            ret = unlink(fullpath);
        }
    }
    if (ret == -1) {
        perror("rm error");
    }
    free(fullpath);
    return 0;
}

int osShellCmdRmdir(int argc, const char **argv)
{
    int  ret;
    char *fullpath = NULL;
    const char *filename = NULL;
    char *shell_working_directory = OsShellGetWorkingDirectory();
    if (shell_working_directory == NULL) {
        return -1;
    }

    ERROR_OUT_IF(argc == 0, PRINTK("rmdir [DIRECTORY]\n"), return -1);

    filename = argv[0];
    ret = vfs_normalize_path(shell_working_directory, filename, &fullpath);
    ERROR_OUT_IF(ret < 0, set_err(-ret, "rmdir error"), return -1);

    if (os_is_containers_wildcard(fullpath)) {
        ret = os_wildcard_extract_directory(fullpath, NULL, RM_DIR);
    } else {
        ret = rmdir(fullpath);
    }
    if (ret == -1) {
        PRINTK("rmdir %s failed. Error: %s.\n", fullpath, strerror(errno));
    }
    free(fullpath);

    return 0;
}

int osShellCmdSync(int argc, const char **argv)
{
    ERROR_OUT_IF(argc > 0, PRINTK("\nUsage: sync\n"), return -1);

    sync();
    return 0;
}

int osShellCmdLsfd(int argc, const char **argv)
{
    ERROR_OUT_IF(argc > 0, PRINTK("\nUsage: lsfd\n"), return -1);

    lsfd();

    return 0;
}

int checkNum(const char *arg)
{
    int i = 0;
    if (arg == NULL) {
        return -1;
    }
    if (arg[0] == '-') {
        /* exclude the '-' */

        i = 1;
    }
    for (; arg[i] != 0; i++) {
        if (!isdigit(arg[i])) {
            return -1;
        }
    }
    return 0;
}

#ifdef LOSCFG_KERNEL_SYSCALL
int osShellCmdSu(int argc, const char **argv)
{
    int su_uid;
    int su_gid;

    if (argc == 0) {
        /* for su root */

        su_uid = 0;
        su_gid = 0;
    } else {
        ERROR_OUT_IF((argc != 2), PRINTK("su [uid_num] [gid_num]\n"), return -1);
        ERROR_OUT_IF((checkNum(argv[0]) != 0) || (checkNum(argv[1]) != 0), /* check argv is digit */
        PRINTK("check uid_num and gid_num is digit\n"), return -1);

        su_uid = atoi(argv[0]);
        su_gid = atoi(argv[1]);

        ERROR_OUT_IF((su_uid < 0) || (su_uid > 60000) || (su_gid < 0) ||
            (su_gid > 60000), PRINTK("uid_num or gid_num out of range!they should be [0~60000]\n"), return -1);
    }

    SysSetUserID(su_uid);
    SysSetGroupID(su_gid);
    return 0;
}
#endif

int osShellCmdChmod(int argc, const char **argv)
{
    int i = 0;
    int mode = 0;
    int ret;
    char *fullpath = NULL;
    const char *filename = NULL;
    struct IATTR attr = {0};
    char *shell_working_directory = NULL;
    const char *p = NULL;
#define MODE_BIT 3 /* 3 bits express 1 mode */

    ERROR_OUT_IF((argc != 2), PRINTK("Usage: chmod <MODE> [FILE]\n"), return -1);

    p = argv[0];
    while (p[i]) {
        if ((p[i] <= '7') && (p[i] >= '0')) {
            mode = ((uint)mode << MODE_BIT) | (uint)(p[i] - '0');
        } else {
            PRINTK("check the input <MODE>\n");
            return -1;
        }
        i++;
    }
    filename = argv[1];

    shell_working_directory = OsShellGetWorkingDirectory();
    if (shell_working_directory == NULL) {
        return -1;
    }
    ret = vfs_normalize_path(shell_working_directory, filename, &fullpath);
    ERROR_OUT_IF(ret < 0, set_err(-ret, "chmod error\n"), return -1);

    attr.attr_chg_mode = mode;
    attr.attr_chg_valid = CHG_MODE; /* change mode */
    ret = chattr(fullpath, &attr);
    if (ret < 0) {
        free(fullpath);
        PRINTK("chmod error! %s\n", strerror(errno));
        return ret;
    }

    free(fullpath);
    return 0;
}

int osShellCmdChown(int argc, const char **argv)
{
    int ret;
    char *fullpath = NULL;
    const char *filename = NULL;
    struct IATTR attr;
    uid_t owner = -1;
    gid_t group = -1;
    attr.attr_chg_valid = 0;

    ERROR_OUT_IF(((argc != 2) && (argc != 3)), PRINTK("Usage: chown [OWNER] [GROUP] FILE\n"), return -1);
    if (argc == 2) { // 2: chown owner of file
        ERROR_OUT_IF((checkNum(argv[0]) != 0), PRINTK("check OWNER is digit\n"), return -1);
        owner = atoi(argv[0]);
        filename = argv[1];
    }
    if (argc == 3) { // 3: chown both owner and group
        ERROR_OUT_IF((checkNum(argv[0]) != 0), PRINTK("check OWNER is digit\n"), return -1);
        ERROR_OUT_IF((checkNum(argv[1]) != 0), PRINTK("check GROUP is digit\n"), return -1);
        owner = atoi(argv[0]);
        group = atoi(argv[1]);
        filename = argv[2];
    }

    if (group != -1) {
        attr.attr_chg_gid = group;
        attr.attr_chg_valid |= CHG_GID;
    }
    if (owner != -1) {
        attr.attr_chg_uid = owner;
        attr.attr_chg_valid |= CHG_UID;
    }

    char *shell_working_directory = OsShellGetWorkingDirectory();
    if (shell_working_directory == NULL) {
        return -1;
    }
    ret = vfs_normalize_path(shell_working_directory, filename, &fullpath);
    ERROR_OUT_IF(ret < 0, set_err(-ret, "chown error\n"), return -1);

    ret = chattr(fullpath, &attr);
    if (ret < 0) {
        free(fullpath);
        PRINTK("chown error! %s\n", strerror(errno));
        return ret;
    }

    free(fullpath);
    return 0;
}

int osShellCmdChgrp(int argc, const char **argv)
{
    int ret;
    char *fullpath = NULL;
    const char *filename = NULL;
    struct IATTR attr;
    gid_t group;
    attr.attr_chg_valid = 0;
    ERROR_OUT_IF((argc != 2), PRINTK("Usage: chgrp GROUP FILE\n"), return -1);
    ERROR_OUT_IF((checkNum(argv[0]) != 0), PRINTK("check GROUP is digit\n"), return -1);
    group = atoi(argv[0]);
    filename = argv[1];

    if (group != -1) {
        attr.attr_chg_gid = group;
        attr.attr_chg_valid |= CHG_GID;
    }

    char *shell_working_directory = OsShellGetWorkingDirectory();
    if (shell_working_directory == NULL) {
        return -1;
    }
    ret = vfs_normalize_path(shell_working_directory, filename, &fullpath);
    ERROR_OUT_IF(ret < 0, set_err(-ret, "chmod error"), return -1);

    ret = chattr(fullpath, &attr);
    if (ret < 0) {
        free(fullpath);
        PRINTK("chgrp error! %s\n", strerror(errno));
        return ret;
    }

    free(fullpath);
    return 0;
}

#ifdef LOSCFG_SHELL_CMD_DEBUG
SHELLCMD_ENTRY(lsfd_shellcmd, CMD_TYPE_EX, "lsfd", XARGS, (CmdCallBackFunc)osShellCmdLsfd);
SHELLCMD_ENTRY(statfs_shellcmd, CMD_TYPE_EX, "statfs", XARGS, (CmdCallBackFunc)osShellCmdStatfs);
SHELLCMD_ENTRY(touch_shellcmd, CMD_TYPE_EX, "touch", XARGS, (CmdCallBackFunc)osShellCmdTouch);
#ifdef LOSCFG_KERNEL_SYSCALL
SHELLCMD_ENTRY(su_shellcmd, CMD_TYPE_EX, "su", XARGS, (CmdCallBackFunc)osShellCmdSu);
#endif
#endif
SHELLCMD_ENTRY(sync_shellcmd, CMD_TYPE_EX, "sync", XARGS, (CmdCallBackFunc)osShellCmdSync);
SHELLCMD_ENTRY(ls_shellcmd, CMD_TYPE_EX, "ls", XARGS, (CmdCallBackFunc)osShellCmdLs);
SHELLCMD_ENTRY(pwd_shellcmd, CMD_TYPE_EX, "pwd", XARGS, (CmdCallBackFunc)osShellCmdPwd);
SHELLCMD_ENTRY(cd_shellcmd, CMD_TYPE_EX, "cd", XARGS, (CmdCallBackFunc)osShellCmdCd);
SHELLCMD_ENTRY(cat_shellcmd, CMD_TYPE_EX, "cat", XARGS, (CmdCallBackFunc)osShellCmdCat);
SHELLCMD_ENTRY(rm_shellcmd, CMD_TYPE_EX, "rm", XARGS, (CmdCallBackFunc)osShellCmdRm);
SHELLCMD_ENTRY(rmdir_shellcmd, CMD_TYPE_EX, "rmdir", XARGS, (CmdCallBackFunc)osShellCmdRmdir);
SHELLCMD_ENTRY(mkdir_shellcmd, CMD_TYPE_EX, "mkdir", XARGS, (CmdCallBackFunc)osShellCmdMkdir);
SHELLCMD_ENTRY(chmod_shellcmd, CMD_TYPE_EX, "chmod", XARGS, (CmdCallBackFunc)osShellCmdChmod);
SHELLCMD_ENTRY(chown_shellcmd, CMD_TYPE_EX, "chown", XARGS, (CmdCallBackFunc)osShellCmdChown);
SHELLCMD_ENTRY(chgrp_shellcmd, CMD_TYPE_EX, "chgrp", XARGS, (CmdCallBackFunc)osShellCmdChgrp);
SHELLCMD_ENTRY(mount_shellcmd, CMD_TYPE_EX, "mount", XARGS, (CmdCallBackFunc)osShellCmdMount);
SHELLCMD_ENTRY(umount_shellcmd, CMD_TYPE_EX, "umount", XARGS, (CmdCallBackFunc)osShellCmdUmount);
SHELLCMD_ENTRY(cp_shellcmd, CMD_TYPE_EX, "cp", XARGS, (CmdCallBackFunc)osShellCmdCp);
#endif
