Cool_Tools/ImHex
1
0
Fork 0
mirror of synced 2025-01-29 19:17:28 +01:00
Code Releases Wiki Activity

feat: Added option to load files into memory

Browse Source
This commit is contained in:
WerWolv 2023-08-09 20:04:12 +02:00
parent ef556d07ed
commit 65c56a887c
4 changed files with 54 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
plugins/builtin/include/content/providers/file_provider.hpp
View File

@ -58,6 +58,9 @@ namespace hex::plugin::builtin {
[[nodiscard]] std::pair<Region, bool> getRegionValidity(u64 address) const override;
private:
void convertToMemoryFile();
protected:
std::fs::path m_path;
wolv::io::File m_file;

1
plugins/builtin/romfs/lang/en_US.json
View File

@ -485,6 +485,7 @@
"hex.builtin.provider.file.error.open": "Failed to open file {}: {}",
"hex.builtin.provider.file.access": "Last access time",
"hex.builtin.provider.file.creation": "Creation time",
"hex.builtin.provider.file.menu.into_memory": "Load into memory",
"hex.builtin.provider.file.modification": "Last modification time",
"hex.builtin.provider.file.path": "File path",
"hex.builtin.provider.file.size": "Size",

46
plugins/builtin/source/content/providers/file_provider.cpp
View File

@ -1,10 +1,13 @@
#include "content/providers/file_provider.hpp"
#include "content/providers/memory_file_provider.hpp"
#include <cstring>
#include <hex/api/localization.hpp>
#include <hex/api/project_file_manager.hpp>
#include <hex/api/achievement_manager.hpp>
#include <hex/api/task.hpp>
#include <hex/helpers/utils.hpp>
#include <hex/helpers/fmt.hpp>
@ -204,14 +207,9 @@ namespace hex::plugin::builtin {
std::vector<FileProvider::MenuEntry> FileProvider::getMenuEntries(){
return {
{"hex.builtin.provider.file.menu.open_folder"_lang, [path = this->m_path] {
fs::openFolderWithSelectionExternal(path);
}},
{"hex.builtin.provider.file.menu.open_file"_lang, [path = this->m_path] {
fs::openFileExternal(path);
}},
{ "hex.builtin.provider.file.menu.open_folder"_lang, [this] { fs::openFolderWithSelectionExternal(this->m_path); } },
{ "hex.builtin.provider.file.menu.open_file"_lang, [this] { fs::openFileExternal(this->m_path); } },
{ "hex.builtin.provider.file.menu.into_memory", [this] { this->convertToMemoryFile(); } }
};
}
@ -299,4 +297,36 @@ namespace hex::plugin::builtin {
return { Region::Invalid(), false };
}
void FileProvider::convertToMemoryFile() {
auto newProvider = hex::ImHexApi::Provider::createProvider("hex.builtin.provider.mem_file", true);
if (auto memoryProvider = dynamic_cast<MemoryFileProvider*>(newProvider); memoryProvider != nullptr) {
if (!memoryProvider->open())
ImHexApi::Provider::remove(newProvider);
else {
const auto size = this->getActualSize();
TaskManager::createTask("Loading into memory", size, [this, size, memoryProvider](Task &task) {
task.setInterruptCallback([memoryProvider]{
ImHexApi::Provider::remove(memoryProvider);
});
constexpr static size_t BufferSize = 0x10000;
std::vector<u8> buffer(BufferSize);
memoryProvider->resize(size);
for (u64 i = 0; i < size; i += BufferSize) {
auto copySize = std::min<size_t>(BufferSize, size - i);
this->read(i, buffer.data(), copySize, true);
memoryProvider->writeRaw(i, buffer.data(), copySize);
task.update(i);
}
memoryProvider->markDirty(true);
ImHexApi::Provider::remove(this, false);
});
}
}
}
}

18
plugins/builtin/source/content/providers/memory_file_provider.cpp
View File

@ -47,12 +47,16 @@ namespace hex::plugin::builtin {
auto newProvider = hex::ImHexApi::Provider::createProvider("hex.builtin.provider.file", true);
if (auto fileProvider = dynamic_cast<FileProvider*>(newProvider); fileProvider != nullptr && (fileProvider->setPath(path), !fileProvider->open()))
ImHexApi::Provider::remove(newProvider);
else {
fileProvider->markDirty(false);
EventManager::post<EventProviderOpened>(newProvider);
ImHexApi::Provider::remove(this, true);
if (auto fileProvider = dynamic_cast<FileProvider*>(newProvider); fileProvider != nullptr) {
fileProvider->setPath(path);
if (!fileProvider->open())
ImHexApi::Provider::remove(newProvider);
else {
fileProvider->markDirty(false);
EventManager::post<EventProviderOpened>(newProvider);
ImHexApi::Provider::remove(this, true);
}
}
});
}
@ -134,11 +138,13 @@ namespace hex::plugin::builtin {
this->m_data = settings["data"].get<std::vector<u8>>();
this->m_name = settings["name"].get<std::string>();
this->m_readOnly = settings["readOnly"].get<bool>();
}
[[nodiscard]] nlohmann::json MemoryFileProvider::storeSettings(nlohmann::json settings) const {
settings["data"] = this->m_data;
settings["name"] = this->m_name;
settings["readOnly"] = this->m_readOnly;
return Provider::storeSettings(settings);
}