ImHex/source/providers/file_provider.cpp

#include "providers/file_provider.hpp"

#undef __STRICT_ANSI__
#include <cstdio>

#include <sys/stat.h>
#include <time.h>

#include "helpers/utils.hpp"
#include "helpers/project_file_handler.hpp"


#ifdef OS_MACOS
    #define off64_t off_t
    #define fopen64 fopen
    #define fseeko64 fseek
    #define ftello64 ftell
#endif

namespace hex::prv {

    FileProvider::FileProvider(std::string_view path) : Provider(), m_path(path) {
        this->m_fileStatsValid = stat(path.data(), &this->m_fileStats) == 0;

        this->m_file = fopen64(path.data(), "r+b");

        this->m_readable = true;
        this->m_writable = true;

        if (this->m_file == nullptr) {
            this->m_file = fopen64(path.data(), "rb");
            this->m_writable = false;
        }

        if (this->m_file != nullptr)
            ProjectFile::setFilePath(path);
    }

    FileProvider::~FileProvider() {
        if (this->m_file != nullptr)
            fclose(this->m_file);
    }


    bool FileProvider::isAvailable() {
        return this->m_file != nullptr;
    }

    bool FileProvider::isReadable() {
        return isAvailable() && this->m_readable;
    }

    bool FileProvider::isWritable() {
        return isAvailable() && this->m_writable;
    }


    void FileProvider::read(u64 offset, void *buffer, size_t size) {
        if ((offset + size) > this->getSize() || buffer == nullptr || size == 0)
            return;

        fseeko64(this->m_file, this->getCurrentPage() * PageSize + offset, SEEK_SET);
        size_t readSize = fread(buffer, 1, size, this->m_file);



        for (u64 i = 0; i < readSize; i++)
            if (this->m_patches.back().contains(offset + i))
                reinterpret_cast<u8*>(buffer)[i] = this->m_patches.back()[offset + i];
    }

    void FileProvider::write(u64 offset, const void *buffer, size_t size) {
        if (buffer == nullptr || size == 0)
            return;

        this->m_patches.push_back(this->m_patches.back());

        for (u64 i = 0; i < size; i++)
            this->m_patches.back()[offset + i] = reinterpret_cast<const u8*>(buffer)[i];
    }

    void FileProvider::readRaw(u64 offset, void *buffer, size_t size) {
        if ((offset + size) > this->getSize() || buffer == nullptr || size == 0)
            return;

        fseeko64(this->m_file, this->getCurrentPage() * PageSize + offset, SEEK_SET);
        fread(buffer, 1, size, this->m_file);
    }

    void FileProvider::writeRaw(u64 offset, const void *buffer, size_t size) {
        if (buffer == nullptr || size == 0)
            return;

        fseeko64(this->m_file, offset, SEEK_SET);
        fwrite(buffer, 1, size, this->m_file);
    }
    size_t FileProvider::getActualSize() {
        fseeko64(this->m_file, 0, SEEK_END);
        return ftello64(this->m_file);
    }

    std::vector<std::pair<std::string, std::string>> FileProvider::getDataInformation() {
        std::vector<std::pair<std::string, std::string>> result;

        result.emplace_back("File path", this->m_path);
        result.emplace_back("Size", hex::toByteString(this->getActualSize()));

        if (this->m_fileStatsValid) {
            result.emplace_back("Creation time", ctime(&this->m_fileStats.st_ctime));
            result.emplace_back("Last access time", ctime(&this->m_fileStats.st_atime));
            result.emplace_back("Last modification time", ctime(&this->m_fileStats.st_mtime));
        }

        return result;
    }

}