ImHex/tests/algorithms/include/test_provider.hpp

#include <hex/providers/provider.hpp>

#include <hex/helpers/file.hpp>
#include <hex/helpers/logger.hpp>
#include <stdexcept>

namespace hex::test {
    using namespace hex::prv;

    class TestProvider : public prv::Provider {
    public:
        TestProvider(std::vector<u8> data) : Provider(){
            this->setData(data);
        }
        ~TestProvider() override = default;

        [[nodiscard]] bool isAvailable() const override { return true; }
        [[nodiscard]] bool isReadable() const override { return true; }
        [[nodiscard]] bool isWritable() const override { return false; }
        [[nodiscard]] bool isResizable() const override { return false; }
        [[nodiscard]] bool isSavable() const override { return false; }

        void setData(const std::vector<u8> &data) {
            if (data.empty()) {
                hex::log::fatal("No data provided");
                throw std::runtime_error("");
            }

            this->m_data = data;
        }

        [[nodiscard]] std::string getName() const override {
            return "";
        }

        [[nodiscard]] std::vector<std::pair<std::string, std::string>> getDataInformation() const override {
            return { };
        }

        void readRaw(u64 offset, void *buffer, size_t size) override {
            if (offset + size >= this->m_data.size()) return;

            std::memcpy(buffer, &this->m_data[offset], size);
        }

        void writeRaw(u64 offset, const void *buffer, size_t size) override {
            if (offset + size >= this->m_data.size()) return;

            std::memcpy(&this->m_data[offset], buffer, size);
        }

        size_t getActualSize() const override {
            return this->m_data.size();
        }

    private:
        std::vector<u8> m_data;
    };

}