/*
 * Copyright (c) 2022 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifdef HKS_CONFIG_FILE
#include HKS_CONFIG_FILE
#else
#include "hks_config.h"
#endif

#include "hks_file_operator.h"

#ifndef _STORAGE_LITE_
#include <dirent.h>
#include <errno.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <unistd.h>

/* use product definitions temporarily */
#define DEFAULT_FILE_PERMISSION 0700
#else

#include <utils_file.h>

#endif /* _STORAGE_LITE_ */

#include "securec.h"

#include "hks_log.h"
#include "hks_mem.h"

#ifndef _CUT_AUTHENTICATE_
static int32_t GetFileName(const char *path, const char *fileName, char *fullFileName, uint32_t fullFileNameLen)
{
    if (path != NULL) {
        if (strncpy_s(fullFileName, fullFileNameLen, path, strlen(path)) != EOK) {
            return HKS_ERROR_INTERNAL_ERROR;
        }

        if (path[strlen(path) - 1] != '/') {
            if (strncat_s(fullFileName, fullFileNameLen, "/", strlen("/")) != EOK) {
                return HKS_ERROR_INTERNAL_ERROR;
            }
        }

        if (strncat_s(fullFileName, fullFileNameLen, fileName, strlen(fileName)) != EOK) {
            return HKS_ERROR_INTERNAL_ERROR;
        }
    } else {
        if (strncpy_s(fullFileName, fullFileNameLen, fileName, strlen(fileName)) != EOK) {
            return HKS_ERROR_INTERNAL_ERROR;
        }
    }

    return HKS_SUCCESS;
}

static int32_t GetFullFileName(const char *path, const char *fileName, char **fullFileName)
{
    uint32_t nameLen = HKS_MAX_FILE_NAME_LEN;
    char *tmpFileName = (char *)HksMalloc(nameLen);
    if (tmpFileName == NULL) {
        return HKS_ERROR_MALLOC_FAIL;
    }
    (void)memset_s(tmpFileName, nameLen, 0, nameLen);

    int32_t ret = GetFileName(path, fileName, tmpFileName, nameLen);
    if (ret != HKS_SUCCESS) {
        HKS_LOG_E("get full fileName failed");
        HKS_FREE_PTR(tmpFileName);
        return ret;
    }

    *fullFileName = tmpFileName;
    return HKS_SUCCESS;
}

#ifndef _STORAGE_LITE_
static int32_t IsFileExist(const char *fileName)
{
    struct stat fileStat;
    int32_t ret = stat(fileName, &fileStat);
    if (ret == -1) {
        if (errno == ENOENT) {
            return HKS_ERROR_NOT_EXIST;
        } else {
            HKS_LOG_E("file stat failed, errno = 0x%x", errno);
            return HKS_ERROR_OPEN_FILE_FAIL;
        }
    }

    return HKS_SUCCESS;
}

static uint32_t FileRead(const char *fileName, uint32_t offset, uint8_t *buf, uint32_t len)
{
    (void)offset;
    int32_t fd = open(fileName, O_RDONLY);
    if (fd < 0) {
        HKS_LOG_E("failed to open file, errno = 0x%x", errno);
        return 0;
    }

    int32_t size = read(fd, buf, len);
    close(fd);
    if (size < 0) {
        HKS_LOG_E("failed to read file, errno = 0x%x", errno);
        return 0;
    }

    return (uint32_t)size;
}

static uint32_t FileSize(const char *fileName)
{
    struct stat fileStat;
    (void)memset_s(&fileStat, sizeof(fileStat), 0, sizeof(fileStat));
    if (stat(fileName, &fileStat) != 0) {
        HKS_LOG_E("file stat failed, errno = 0x%x", errno);
        return 0;
    }

    return (uint32_t)fileStat.st_size;
}

static int32_t FileWrite(const char *fileName, uint32_t offset, const uint8_t *buf, uint32_t len)
{
    (void)offset;
    int32_t fd = open(fileName, O_WRONLY | O_CREAT, S_IRUSR | S_IWUSR);
    if (fd < 0) {
        HKS_LOG_E("open file failed, errno = 0x%x", errno);
        return HKS_ERROR_OPEN_FILE_FAIL;
    }

    int32_t size = write(fd, buf, len);
    if (size < 0) {
        HKS_LOG_E("write file size failed, errno = 0x%x", errno);
        close(fd);
        return HKS_ERROR_WRITE_FILE_FAIL;
    }
    fsync(fd);
    close(fd);
    return HKS_SUCCESS;
}

static int32_t FileRemove(const char *fileName)
{
    struct stat fileStat;
    int32_t ret = stat(fileName, &fileStat);
    if (ret == -1) {
        if (errno == ENOENT) {
            return HKS_SUCCESS; /* if file not exist, return ok */
        } else {
            HKS_LOG_E("file stat failed, errno = 0x%x", errno);
            return HKS_ERROR_OPEN_FILE_FAIL;
        }
    }

    if (S_ISDIR(fileStat.st_mode)) {
        return HKS_ERROR_INVALID_ARGUMENT; /* FileRemove func only care about files. */
    }

    if (unlink(fileName) != 0) {
        HKS_LOG_E("remove file failed, errno = 0x%x", errno);
        return HKS_ERROR_REMOVE_FILE_FAIL;
    }

    return HKS_SUCCESS;
}

int32_t HksIsDirExist(const char *path)
{
    if (path == NULL) {
        return HKS_ERROR_NULL_POINTER;
    }
    return IsFileExist(path);
}

int32_t HksMakeDir(const char *path)
{
    return mkdir(path, DEFAULT_FILE_PERMISSION);
}

void *HksOpenDir(const char *path)
{
    return (void *)opendir(path);
}

int32_t HksCloseDir(void *dirp)
{
    return closedir((DIR *)dirp);
}

int32_t HksGetDirFile(void *dirp, struct HksFileDirentInfo *direntInfo)
{
    DIR *dir = (DIR *)dirp;
    struct dirent *dire = readdir(dir);

    while (dire != NULL) {
        if (dire->d_type != DT_REG) { /* only care about files. */
            dire = readdir(dir);
            continue;
        }

        uint32_t len = strlen(dire->d_name);
        if (memcpy_s(direntInfo->fileName, sizeof(direntInfo->fileName) - 1, dire->d_name, len) != EOK) {
            return HKS_ERROR_BAD_STATE;
        }
        direntInfo->fileName[len] = '\0';
        return HKS_SUCCESS;
    }

    return HKS_ERROR_NOT_EXIST;
}

int32_t HksGetStoragePath(enum HksStoragePathType pathType, char *path, uint32_t *len)
{
    if ((path == NULL) || (len == NULL) || (*len <= 1)) {
        return HKS_ERROR_INVALID_ARGUMENT;
    }
    errno_t ret;
    uint32_t pathLen;
    if (pathType == HKS_STORAGE_MAIN_PATH) {
        pathLen = strlen(HKS_KEY_STORE_PATH);
        ret = memcpy_s(path, *len - 1, HKS_KEY_STORE_PATH, pathLen);
    } else if (pathType == HKS_STORAGE_BACKUP_PATH) {
        pathLen = strlen(HKS_KEY_STORE_BAK_PATH);
        ret = memcpy_s(path, *len - 1, HKS_KEY_STORE_BAK_PATH, pathLen);
    } else {
        return HKS_ERROR_INVALID_ARGUMENT;
    }
    if (ret != EOK) {
        HKS_LOG_E("memcpy failed");
        return HKS_ERROR_BAD_STATE;
    }
    path[pathLen] = '\0';
    *len = pathLen + 1;
    return HKS_SUCCESS;
}

int32_t HksRemoveDir(const char *dirPath)
{
    struct stat fileStat;
    int32_t ret = stat(dirPath, &fileStat);
    if (ret != 0) {
        HKS_LOG_E("file stat failed");
        return HKS_FAILURE;
    }

    if (S_ISDIR(fileStat.st_mode) == false) {
        HKS_LOG_E("path is not dir");
        return HKS_FAILURE;
    }

    DIR *dir = opendir(dirPath);
    if (dir  == NULL) {
        HKS_LOG_E("open dir failed");
        return HKS_ERROR_OPEN_FILE_FAIL;
    }

    struct dirent *dire = readdir(dir);
    while (dire != NULL) {
        if (dire->d_type == DT_REG) { /* only care about files. */
            ret = HksFileRemove(dirPath, dire->d_name);
            if (ret != HKS_SUCCESS) { /* Continue to delete remaining files */
                HKS_LOG_E("remove file failed when remove dir files, ret = %d.", ret);
            }
        }
        dire = readdir(dir);
    }

    closedir(dir);
    return HKS_SUCCESS;
}
#else
static int32_t IsFileExist(const char *fileName)
{
    unsigned int fileSize;
    int32_t ret = UtilsFileStat(fileName, &fileSize);
    if (ret < 0) {
        return HKS_ERROR_NOT_EXIST;
    }
    return HKS_SUCCESS;
}

static uint32_t FileRead(const char *fileName, uint32_t offset, uint8_t *buf, uint32_t len)
{
    /* now offset is 0, but we maybe extend hi1131 file interfaces in the future */
    if (offset != 0) {
        return 0;
    }

    unsigned int fileSize;
    int32_t ret = UtilsFileStat(fileName, &fileSize);
    if (ret < 0) {
        HKS_LOG_E("stat file failed, errno = 0x%x", ret);
        return 0;
    }

    if (len > fileSize) {
        HKS_LOG_E("read data over file size!\n, file size = %d\n, buf len = %d\n", fileSize, len);
        return 0;
    }

    int fd = UtilsFileOpen(fileName, O_RDONLY_FS, 0);
    if (fd < 0) {
        HKS_LOG_E("failed to open file, errno = 0x%x", fd);
        return 0;
    }

    ret = UtilsFileRead(fd, (char *)buf, len);
    UtilsFileClose(fd);
    if (ret < 0) {
        HKS_LOG_E("failed to read file, errno = 0x%x", ret);
        return 0;
    }

    return len;
}

static int32_t FileWrite(const char *fileName, uint32_t offset, const uint8_t *buf, uint32_t len)
{
    /* now offset is 0, but we may extend hi1131 file interfaces */
    if (offset != 0) {
        return HKS_ERROR_INVALID_ARGUMENT;
    }

    int fd = UtilsFileOpen(fileName, O_CREAT_FS | O_TRUNC_FS | O_RDWR_FS, 0);
    if (fd < 0) {
        HKS_LOG_E("failed to open key file, errno = 0x%x\n", fd);
        return HKS_ERROR_OPEN_FILE_FAIL;
    }

    int32_t ret = UtilsFileWrite(fd, (const char*)buf, len);
    (void)UtilsFileClose(fd);
    if (ret < 0) {
        HKS_LOG_E("failed to write key file, errno = 0x%x\n", ret);
        return HKS_ERROR_WRITE_FILE_FAIL;
    }

    return HKS_SUCCESS;
}

static uint32_t FileSize(const char *fileName)
{
    unsigned int fileSize;
    int32_t ret = UtilsFileStat(fileName, &fileSize);
    if (ret < 0) {
        HKS_LOG_E("stat file failed, errno = 0x%x", ret);
        return 0;
    }
    return fileSize;
}

static int32_t FileRemove(const char *fileName)
{
    int32_t ret = UtilsFileDelete(fileName);
    if (ret < 0) {
        HKS_LOG_E("remove file failed");
        return HKS_ERROR_REMOVE_FILE_FAIL;
    }
    return HKS_SUCCESS;
}

#endif

uint32_t HksFileRead(const char *path, const char *fileName, uint32_t offset, uint8_t *buf, uint32_t len)
{
    if ((fileName == NULL) || (buf == NULL) || (len == 0)) {
        return 0;
    }

    char *fullFileName = NULL;
    int32_t ret = GetFullFileName(path, fileName, &fullFileName);
    if (ret != HKS_SUCCESS) {
        return 0;
    }

    uint32_t size = FileRead(fullFileName, offset, buf, len);
    HKS_FREE_PTR(fullFileName);
    return size;
}

int32_t HksFileWrite(const char *path, const char *fileName, uint32_t offset, const uint8_t *buf, uint32_t len)
{
    if ((fileName == NULL) || (buf == NULL) || (len == 0)) {
        return HKS_ERROR_INVALID_ARGUMENT;
    }

    char *fullFileName = NULL;
    int32_t ret = GetFullFileName(path, fileName, &fullFileName);
    if (ret != HKS_SUCCESS) {
        return ret;
    }

    ret = FileWrite(fullFileName, offset, buf, len);
    HKS_FREE_PTR(fullFileName);
    return ret;
}

uint32_t HksFileSize(const char *path, const char *fileName)
{
    if (fileName == NULL) {
        return 0;
    }

    char *fullFileName = NULL;
    int32_t ret = GetFullFileName(path, fileName, &fullFileName);
    if (ret != HKS_SUCCESS) {
        return 0;
    }

    uint32_t size = FileSize(fullFileName);
    HKS_FREE_PTR(fullFileName);
    return size;
}

int32_t HksIsFileExist(const char *path, const char *fileName)
{
    if (fileName == NULL) {
        return HKS_ERROR_NULL_POINTER;
    }

    char *fullFileName = NULL;
    int32_t ret = GetFullFileName(path, fileName, &fullFileName);
    if (ret != HKS_SUCCESS) {
        return ret;
    }

    ret = IsFileExist(fullFileName);
    HKS_FREE_PTR(fullFileName);
    return ret;
}

int32_t HksFileRemove(const char *path, const char *fileName)
{
    if (fileName == NULL) {
        return HKS_ERROR_INVALID_ARGUMENT;
    }

    char *fullFileName = NULL;
    int32_t ret = GetFullFileName(path, fileName, &fullFileName);
    if (ret != HKS_SUCCESS) {
        return ret;
    }

    ret = FileRemove(fullFileName);
    HKS_FREE_PTR(fullFileName);
    return ret;
}

#ifdef HKS_SUPPORT_UPGRADE_STORAGE_DATA
/* return < 0 error; > 0 realFileSize; == 0 no data */
int32_t HksOldVersionFileRead(const char *fileName, uint32_t offset, uint8_t *buf, uint32_t len)
{
    HKS_LOG_D("piling");
    return HKS_ERROR_NOT_SUPPORTED;
}

int32_t HksOldVersionFileRemove(const char *fileName)
{
    HKS_LOG_D("piling");
    return HKS_ERROR_NOT_SUPPORTED;
}

/* return < 0 error; >= 0 realFileSize */
int32_t HksOldVersionFileSize(const char *fileName)
{
    HKS_LOG_D("piling");
    return HKS_ERROR_NOT_SUPPORTED;
}

/* return true, exist; false not exist */
bool HksOldVersionIsFileExist(const char *fileName)
{
    HKS_LOG_D("piling");
    return false;
}
#endif

int32_t HksGetFileName(const char *path, const char *fileName, char *fullFileName, uint32_t fullFileNameLen)
{
    return GetFileName(path, fileName, fullFileName, fullFileNameLen);
}

#endif /* _CUT_AUTHENTICATE_ */