32ab1c1a06
* refresh store on open * show the user if the request failed * do not refresh if a refresh is in progress This makes ImHex crash * InProgress * fix formatting * use a localized message * set localized message in other langs
257 lines
11 KiB
C++
257 lines
11 KiB
C++
#include "content/views/view_store.hpp"
|
|
|
|
#include <hex/api/content_registry.hpp>
|
|
|
|
#include <imgui.h>
|
|
#define IMGUI_DEFINE_MATH_OPERATORS
|
|
#include <imgui_internal.h>
|
|
|
|
#include <hex/helpers/utils.hpp>
|
|
#include <hex/helpers/crypto.hpp>
|
|
#include <hex/helpers/logger.hpp>
|
|
#include <hex/helpers/magic.hpp>
|
|
#include <hex/helpers/file.hpp>
|
|
#include <hex/helpers/fs.hpp>
|
|
#include <hex/helpers/tar.hpp>
|
|
|
|
#include <fstream>
|
|
#include <filesystem>
|
|
#include <functional>
|
|
#include <nlohmann/json.hpp>
|
|
|
|
namespace hex::plugin::builtin {
|
|
|
|
using namespace std::literals::string_literals;
|
|
using namespace std::literals::chrono_literals;
|
|
|
|
ViewStore::ViewStore() : View("hex.builtin.view.store.name") {
|
|
ContentRegistry::Interface::addMenuItem("hex.builtin.menu.help", 3000, [&, this] {
|
|
if (ImGui::MenuItem("hex.builtin.view.store.name"_lang)) {
|
|
if (this->m_requestStatus == RequestStatus::NotAttempted)
|
|
this->refresh();
|
|
TaskManager::doLater([] { ImGui::OpenPopup(View::toWindowName("hex.builtin.view.store.name").c_str()); });
|
|
this->getWindowOpenState() = true;
|
|
}
|
|
});
|
|
}
|
|
|
|
void ViewStore::drawStore() {
|
|
ImGui::Header("hex.builtin.view.store.desc"_lang, true);
|
|
|
|
if (ImGui::Button("hex.builtin.view.store.reload"_lang)) {
|
|
this->refresh();
|
|
}
|
|
|
|
auto drawTab = [this](auto title, fs::ImHexPath pathType, auto &content, const std::function<void(const StoreEntry &)> &downloadDoneCallback) {
|
|
if (ImGui::BeginTabItem(title)) {
|
|
if (ImGui::BeginTable("##pattern_language", 3, ImGuiTableFlags_ScrollY | ImGuiTableFlags_Borders | ImGuiTableFlags_SizingStretchSame | ImGuiTableFlags_RowBg)) {
|
|
ImGui::TableSetupScrollFreeze(0, 1);
|
|
ImGui::TableSetupColumn("hex.builtin.view.store.row.name"_lang, ImGuiTableColumnFlags_WidthFixed);
|
|
ImGui::TableSetupColumn("hex.builtin.view.store.row.description"_lang, ImGuiTableColumnFlags_None);
|
|
ImGui::TableSetupColumn("", ImGuiTableColumnFlags_WidthFixed);
|
|
|
|
ImGui::TableHeadersRow();
|
|
|
|
u32 id = 1;
|
|
for (auto &entry : content) {
|
|
ImGui::TableNextRow(ImGuiTableRowFlags_None, ImGui::GetTextLineHeight() + 4.0F * ImGui::GetStyle().FramePadding.y);
|
|
ImGui::TableNextColumn();
|
|
ImGui::TextUnformatted(entry.name.c_str());
|
|
ImGui::TableNextColumn();
|
|
ImGui::TextUnformatted(entry.description.c_str());
|
|
ImGui::TableNextColumn();
|
|
|
|
ImGui::PushID(id);
|
|
ImGui::BeginDisabled(this->m_download.valid() && this->m_download.wait_for(0s) != std::future_status::ready);
|
|
{
|
|
if (entry.downloading) {
|
|
ImGui::TextSpinner("");
|
|
|
|
if (this->m_download.valid() && this->m_download.wait_for(0s) == std::future_status::ready) {
|
|
entry.downloading = false;
|
|
|
|
auto response = this->m_download.get();
|
|
if (response.code == 200) {
|
|
entry.installed = true;
|
|
entry.hasUpdate = false;
|
|
|
|
if (entry.isFolder) {
|
|
Tar tar(this->m_downloadPath, Tar::Mode::Read);
|
|
tar.extractAll(this->m_downloadPath.parent_path() / this->m_downloadPath.stem());
|
|
}
|
|
|
|
downloadDoneCallback(entry);
|
|
} else
|
|
log::error("Download failed! HTTP Code {}", response.code);
|
|
|
|
|
|
this->m_download = {};
|
|
}
|
|
|
|
} else if (entry.hasUpdate) {
|
|
if (ImGui::Button("hex.builtin.view.store.update"_lang)) {
|
|
entry.downloading = this->download(pathType, entry.fileName, entry.link, true);
|
|
}
|
|
} else if (!entry.installed) {
|
|
if (ImGui::Button("hex.builtin.view.store.download"_lang)) {
|
|
entry.downloading = this->download(pathType, entry.fileName, entry.link, false);
|
|
}
|
|
} else {
|
|
if (ImGui::Button("hex.builtin.view.store.remove"_lang)) {
|
|
entry.installed = !this->remove(pathType, entry.fileName);
|
|
}
|
|
}
|
|
}
|
|
ImGui::EndDisabled();
|
|
ImGui::PopID();
|
|
id++;
|
|
}
|
|
|
|
ImGui::EndTable();
|
|
}
|
|
ImGui::EndTabItem();
|
|
}
|
|
};
|
|
|
|
if (ImGui::BeginTabBar("storeTabs")) {
|
|
drawTab("hex.builtin.view.store.tab.patterns"_lang, fs::ImHexPath::Patterns, this->m_patterns, [](auto) {});
|
|
drawTab("hex.builtin.view.store.tab.libraries"_lang, fs::ImHexPath::PatternsInclude, this->m_includes, [](auto) {});
|
|
drawTab("hex.builtin.view.store.tab.magics"_lang, fs::ImHexPath::Magic, this->m_magics, [](auto) { magic::compile(); });
|
|
drawTab("hex.builtin.view.store.tab.constants"_lang, fs::ImHexPath::Constants, this->m_constants, [](auto) {});
|
|
drawTab("hex.builtin.view.store.tab.encodings"_lang, fs::ImHexPath::Encodings, this->m_encodings, [](auto) {});
|
|
drawTab("hex.builtin.view.store.tab.yara"_lang, fs::ImHexPath::Yara, this->m_yara, [](auto) {});
|
|
|
|
ImGui::EndTabBar();
|
|
}
|
|
}
|
|
|
|
void ViewStore::refresh() {
|
|
// do not refresh if a refresh is already in progress
|
|
if (this->m_requestStatus == RequestStatus::InProgress)
|
|
return;
|
|
this->m_requestStatus = RequestStatus::InProgress;
|
|
|
|
this->m_patterns.clear();
|
|
this->m_includes.clear();
|
|
this->m_magics.clear();
|
|
this->m_constants.clear();
|
|
this->m_yara.clear();
|
|
this->m_encodings.clear();
|
|
|
|
this->m_apiRequest = this->m_net.getString(ImHexApiURL + "/store"s, 30'0000);
|
|
}
|
|
|
|
void ViewStore::parseResponse() {
|
|
auto response = this->m_apiRequest.get();
|
|
this->m_requestStatus = response.code == 200 ? RequestStatus::Succeded : RequestStatus::Failed;
|
|
if (this->m_requestStatus == RequestStatus::Succeded) {
|
|
auto json = nlohmann::json::parse(response.body);
|
|
|
|
auto parseStoreEntries = [](auto storeJson, const std::string &name, fs::ImHexPath pathType, std::vector<StoreEntry> &results) {
|
|
// Check if the response handles the type of files
|
|
if (storeJson.contains(name)) {
|
|
|
|
for (auto &entry : storeJson[name]) {
|
|
|
|
// Check if entry is valid
|
|
if (entry.contains("name") && entry.contains("desc") && entry.contains("file") && entry.contains("url") && entry.contains("hash") && entry.contains("folder")) {
|
|
|
|
// Parse entry
|
|
StoreEntry storeEntry = { entry["name"], entry["desc"], entry["file"], entry["url"], entry["hash"], entry["folder"], false, false, false };
|
|
|
|
// Check if file is installed already or has an update available
|
|
for (const auto &folder : fs::getDefaultPaths(pathType)) {
|
|
|
|
auto path = folder / std::fs::path(storeEntry.fileName);
|
|
|
|
if (fs::exists(path) && fs::isPathWritable(folder)) {
|
|
storeEntry.installed = true;
|
|
|
|
std::ifstream file(path, std::ios::in | std::ios::binary);
|
|
std::vector<u8> data(fs::getFileSize(path), 0x00);
|
|
file.read(reinterpret_cast<char *>(data.data()), data.size());
|
|
|
|
auto fileHash = crypt::sha256(data);
|
|
|
|
// Compare installed file hash with hash of repo file
|
|
if (std::vector(fileHash.begin(), fileHash.end()) != crypt::decode16(storeEntry.hash))
|
|
storeEntry.hasUpdate = true;
|
|
}
|
|
}
|
|
|
|
results.push_back(storeEntry);
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
parseStoreEntries(json, "patterns", fs::ImHexPath::Patterns, this->m_patterns);
|
|
parseStoreEntries(json, "includes", fs::ImHexPath::PatternsInclude, this->m_includes);
|
|
parseStoreEntries(json, "magic", fs::ImHexPath::Magic, this->m_magics);
|
|
parseStoreEntries(json, "constants", fs::ImHexPath::Constants, this->m_constants);
|
|
parseStoreEntries(json, "yara", fs::ImHexPath::Yara, this->m_yara);
|
|
parseStoreEntries(json, "encodings", fs::ImHexPath::Encodings, this->m_encodings);
|
|
}
|
|
this->m_apiRequest = {};
|
|
}
|
|
|
|
void ViewStore::drawContent() {
|
|
if (ImGui::BeginPopupModal(View::toWindowName("hex.builtin.view.store.name").c_str(), &this->getWindowOpenState())) {
|
|
if (this->m_apiRequest.valid()) {
|
|
if (this->m_apiRequest.wait_for(0s) != std::future_status::ready)
|
|
ImGui::TextSpinner("hex.builtin.view.store.loading"_lang);
|
|
else
|
|
this->parseResponse();
|
|
}
|
|
|
|
if (this->m_requestStatus == RequestStatus::Failed)
|
|
ImGui::TextFormattedColored(ImGui::GetCustomColorVec4(ImGuiCustomCol_ToolbarRed), "hex.builtin.view.store.netfailed"_lang);
|
|
|
|
this->drawStore();
|
|
|
|
ImGui::EndPopup();
|
|
} else {
|
|
this->getWindowOpenState() = false;
|
|
}
|
|
}
|
|
|
|
bool ViewStore::download(fs::ImHexPath pathType, const std::string &fileName, const std::string &url, bool update) {
|
|
bool downloading = false;
|
|
for (const auto &path : fs::getDefaultPaths(pathType)) {
|
|
if (!fs::isPathWritable(path))
|
|
continue;
|
|
|
|
auto fullPath = path / std::fs::path(fileName);
|
|
|
|
if (!update || fs::exists(fullPath)) {
|
|
downloading = true;
|
|
this->m_downloadPath = fullPath;
|
|
this->m_download = this->m_net.downloadFile(url, fullPath, 30'0000);
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!downloading) {
|
|
View::showErrorPopup("hex.builtin.view.store.download_error"_lang);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ViewStore::remove(fs::ImHexPath pathType, const std::string &fileName) {
|
|
bool removed = true;
|
|
for (const auto &path : fs::getDefaultPaths(pathType)) {
|
|
const auto filePath = path / fileName;
|
|
const auto folderPath = (path / std::fs::path(fileName).stem());
|
|
|
|
fs::remove(filePath);
|
|
fs::removeAll(folderPath);
|
|
|
|
removed = removed && !fs::exists(filePath) && !fs::exists(folderPath);
|
|
}
|
|
|
|
return removed;
|
|
}
|
|
|
|
} |