88 lines
2.5 KiB
C++
88 lines
2.5 KiB
C++
#pragma once
|
|
#if !defined(OS_WEB)
|
|
|
|
#include <hex/providers/provider.hpp>
|
|
|
|
#include <set>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#if defined(OS_WINDOWS)
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#include <windows.h>
|
|
#endif
|
|
|
|
namespace hex::plugin::builtin {
|
|
|
|
class DiskProvider : public hex::prv::Provider {
|
|
public:
|
|
DiskProvider();
|
|
~DiskProvider() override = default;
|
|
|
|
[[nodiscard]] bool isAvailable() const override;
|
|
[[nodiscard]] bool isReadable() const override;
|
|
[[nodiscard]] bool isWritable() const override;
|
|
[[nodiscard]] bool isResizable() const override;
|
|
[[nodiscard]] bool isSavable() const override;
|
|
|
|
void readRaw(u64 offset, void *buffer, size_t size) override;
|
|
void writeRaw(u64 offset, const void *buffer, size_t size) override;
|
|
[[nodiscard]] size_t getActualSize() const override;
|
|
|
|
void setPath(const std::fs::path &path);
|
|
|
|
[[nodiscard]] bool open() override;
|
|
void close() override;
|
|
|
|
[[nodiscard]] std::string getName() const override;
|
|
[[nodiscard]] std::vector<Description> getDataDescription() const override;
|
|
|
|
[[nodiscard]] bool hasLoadInterface() const override { return true; }
|
|
bool drawLoadInterface() override;
|
|
|
|
void loadSettings(const nlohmann::json &settings) override;
|
|
[[nodiscard]] nlohmann::json storeSettings(nlohmann::json settings = { }) const override;
|
|
|
|
[[nodiscard]] std::string getTypeName() const override {
|
|
return "hex.builtin.provider.disk";
|
|
}
|
|
|
|
[[nodiscard]] std::pair<Region, bool> getRegionValidity(u64 address) const override;
|
|
std::variant<std::string, i128> queryInformation(const std::string &category, const std::string &argument) override;
|
|
|
|
protected:
|
|
void reloadDrives();
|
|
|
|
struct DriveInfo {
|
|
std::string path;
|
|
std::string friendlyName;
|
|
|
|
auto operator<=>(const DriveInfo &other) const {
|
|
return this->path <=> other.path;
|
|
}
|
|
};
|
|
|
|
std::set<DriveInfo> m_availableDrives;
|
|
std::fs::path m_path;
|
|
std::string m_friendlyName;
|
|
|
|
#if defined(OS_WINDOWS)
|
|
HANDLE m_diskHandle = INVALID_HANDLE_VALUE;
|
|
#else
|
|
std::string m_pathBuffer;
|
|
int m_diskHandle = -1;
|
|
#endif
|
|
|
|
size_t m_diskSize = 0;
|
|
size_t m_sectorSize = 0;
|
|
|
|
u64 m_sectorBufferAddress = 0;
|
|
std::vector<u8> m_sectorBuffer;
|
|
|
|
bool m_readable = false;
|
|
bool m_writable = false;
|
|
};
|
|
|
|
}
|
|
#endif
|