diff --git a/plugins/builtin/include/content/views/view_information.hpp b/plugins/builtin/include/content/views/view_information.hpp index b8a7096e3..779e9c754 100644 --- a/plugins/builtin/include/content/views/view_information.hpp +++ b/plugins/builtin/include/content/views/view_information.hpp @@ -29,10 +29,10 @@ namespace hex::plugin::builtin { std::array m_valueCounts = { 0 }; bool m_analyzing = false; - std::pair m_analyzedRegion = { 0, 0 }; + Region m_analyzedRegion = { 0, 0 }; - std::string m_fileDescription; - std::string m_mimeType; + std::string m_dataDescription; + std::string m_dataMimeType; void analyze(); }; diff --git a/plugins/builtin/source/content/views/view_information.cpp b/plugins/builtin/source/content/views/view_information.cpp index 2654a6a3b..c6fc4b475 100644 --- a/plugins/builtin/source/content/views/view_information.cpp +++ b/plugins/builtin/source/content/views/view_information.cpp @@ -3,6 +3,7 @@ #include #include +#include #include #include #include @@ -26,20 +27,20 @@ namespace hex::plugin::builtin { ViewInformation::ViewInformation() : View("hex.builtin.view.information.name") { EventManager::subscribe(this, [this]() { - this->m_dataValid = false; + this->m_dataValid = false; this->m_highestBlockEntropy = 0; this->m_blockEntropy.clear(); this->m_averageEntropy = 0; - this->m_blockSize = 0; + this->m_blockSize = 0; this->m_valueCounts.fill(0x00); - this->m_mimeType = ""; - this->m_fileDescription = ""; + this->m_dataMimeType.clear(); + this->m_dataDescription.clear(); this->m_analyzedRegion = { 0, 0 }; }); EventManager::subscribe(this, [this](Region region) { if (this->m_blockSize != 0) - this->m_entropyHandlePosition = region.address / this->m_blockSize; + this->m_entropyHandlePosition = region.getStartAddress() / this->m_blockSize; }); EventManager::subscribe(this, [this](const auto*) { @@ -84,41 +85,49 @@ namespace hex::plugin::builtin { std::thread([this] { auto provider = ImHexApi::Provider::get(); - auto task = ImHexApi::Tasks::createTask("hex.builtin.view.information.analyzing", provider->getSize()); + auto task = ImHexApi::Tasks::createTask("hex.builtin.view.information.analyzing", provider->getActualSize()); this->m_analyzedRegion = { provider->getBaseAddress(), provider->getBaseAddress() + provider->getSize() }; { magic::compile(); - this->m_fileDescription = magic::getDescription(provider); - this->m_mimeType = magic::getMIMEType(provider); + this->m_dataDescription = magic::getDescription(provider); + this->m_dataMimeType = magic::getMIMEType(provider); } this->m_dataValid = true; { - this->m_blockSize = std::max(std::ceil(provider->getSize() / 2048.0F), 256); - std::vector buffer(this->m_blockSize, 0x00); - std::memset(this->m_valueCounts.data(), 0x00, this->m_valueCounts.size() * sizeof(u32)); + this->m_blockSize = std::max(std::ceil(provider->getActualSize() / 2048.0F), 256); + + std::array valueCounts = { 0 }, blockValueCounts = { 0 }; + this->m_blockEntropy.clear(); this->m_valueCounts.fill(0); - for (u64 i = 0; i < provider->getSize(); i += this->m_blockSize) { - std::array blockValueCounts = { 0 }; - provider->read(i + provider->getBaseAddress(), buffer.data(), std::min(u64(this->m_blockSize), provider->getSize() - i)); + auto reader = prv::BufferedReader(provider); - for (size_t j = 0; j < this->m_blockSize; j++) { - blockValueCounts[buffer[j]]++; - this->m_valueCounts[buffer[j]]++; + u64 count = 0; + for (u8 byte : reader) { + valueCounts[byte]++; + blockValueCounts[byte]++; + + count++; + if ((count % this->m_blockSize) == 0) [[unlikely]] { + this->m_blockEntropy.push_back(calculateEntropy(blockValueCounts, this->m_blockSize)); + blockValueCounts = { 0 }; + task.update(count); } - - this->m_blockEntropy.push_back(calculateEntropy(blockValueCounts, this->m_blockSize)); - task.update(i); } - this->m_averageEntropy = calculateEntropy(this->m_valueCounts, provider->getSize()); - this->m_highestBlockEntropy = *std::max_element(this->m_blockEntropy.begin(), this->m_blockEntropy.end()); + this->m_valueCounts = valueCounts; + + this->m_averageEntropy = calculateEntropy(valueCounts, provider->getSize()); + if (!this->m_blockEntropy.empty()) + this->m_highestBlockEntropy = *std::max_element(this->m_blockEntropy.begin(), this->m_blockEntropy.end()); + else + this->m_highestBlockEntropy = 0; } this->m_analyzing = false; @@ -129,15 +138,11 @@ namespace hex::plugin::builtin { if (ImGui::Begin(View::toWindowName("hex.builtin.view.information.name").c_str(), &this->getWindowOpenState(), ImGuiWindowFlags_NoCollapse)) { if (ImGui::BeginChild("##scrolling", ImVec2(0, 0), false, ImGuiWindowFlags_NoMove | ImGuiWindowFlags_NoNav)) { - auto provider = ImHexApi::Provider::get(); if (ImHexApi::Provider::isValid() && provider->isReadable()) { - ImGui::TextUnformatted("hex.builtin.view.information.control"_lang); - ImGui::Separator(); - ImGui::BeginDisabled(this->m_analyzing); { - if (ImGui::Button("hex.builtin.view.information.analyze"_lang)) + if (ImGui::Button("hex.builtin.view.information.analyze"_lang, ImVec2(ImGui::GetContentRegionAvailWidth(), 0))) this->analyze(); } ImGui::EndDisabled(); @@ -148,83 +153,131 @@ namespace hex::plugin::builtin { ImGui::NewLine(); } - ImGui::NewLine(); - ImGui::TextUnformatted("hex.builtin.view.information.region"_lang); - ImGui::Separator(); - - for (auto &[name, value] : provider->getDataInformation()) { - ImGui::LabelText(name.c_str(), "%s", value.c_str()); - } - if (this->m_dataValid) { - ImGui::LabelText("hex.builtin.view.information.region"_lang, "%s", hex::format("0x{:X} - 0x{:X}", this->m_analyzedRegion.first, this->m_analyzedRegion.second).c_str()); + // Analyzed region + ImGui::Header("hex.builtin.view.information.region"_lang, true); - ImGui::NewLine(); + if (ImGui::BeginTable("information", 2, ImGuiTableFlags_SizingFixedFit | ImGuiTableFlags_RowBg)) { + ImGui::TableSetupColumn("type"); + ImGui::TableSetupColumn("value", ImGuiTableColumnFlags_WidthStretch); - if (!this->m_fileDescription.empty() || !this->m_mimeType.empty()) { - ImGui::TextUnformatted("hex.builtin.view.information.magic"_lang); - ImGui::Separator(); - } + ImGui::TableNextRow(); - if (!this->m_fileDescription.empty()) { - ImGui::TextUnformatted("hex.builtin.view.information.description"_lang); - ImGui::TextFormattedWrapped("{}", this->m_fileDescription.c_str()); - ImGui::NewLine(); - } - - if (!this->m_mimeType.empty()) { - ImGui::TextUnformatted("hex.builtin.view.information.mime"_lang); - ImGui::TextFormattedWrapped("{}", this->m_mimeType.c_str()); - ImGui::NewLine(); - } - - ImGui::TextUnformatted("hex.builtin.view.information.info_analysis"_lang); - ImGui::Separator(); - - ImGui::PushStyleColor(ImGuiCol_FrameBg, ImGui::GetColorU32(ImGuiCol_WindowBg)); - ImPlot::PushStyleColor(ImPlotCol_FrameBg, ImGui::GetColorU32(ImGuiCol_WindowBg)); - - ImGui::TextUnformatted("hex.builtin.view.information.distribution"_lang); - ImPlot::SetNextPlotLimits(0, 256, 0.5, float(*std::max_element(this->m_valueCounts.begin(), this->m_valueCounts.end())) * 1.1F, ImGuiCond_Always); - if (ImPlot::BeginPlot("##distribution", "Address", "Count", ImVec2(-1, 0), ImPlotFlags_NoChild | ImPlotFlags_NoLegend | ImPlotFlags_NoMenus | ImPlotFlags_NoBoxSelect, ImPlotAxisFlags_Lock, ImPlotAxisFlags_Lock | ImPlotAxisFlags_LogScale)) { - static auto x = [] { - std::array result { 0 }; - std::iota(result.begin(), result.end(), 0); - return result; - }(); - - - ImPlot::PlotBars("##bytes", x.data(), this->m_valueCounts.data(), x.size(), 0.67); - - ImPlot::EndPlot(); - } - - ImGui::NewLine(); - - ImGui::TextUnformatted("hex.builtin.view.information.entropy"_lang); - - ImPlot::SetNextPlotLimits(0, this->m_blockEntropy.size(), -0.1, 1.1, ImGuiCond_Always); - if (ImPlot::BeginPlot("##entropy", "Address", "Entropy", ImVec2(-1, 0), ImPlotFlags_NoChild | ImPlotFlags_CanvasOnly | ImPlotFlags_AntiAliased, ImPlotAxisFlags_Lock | ImPlotAxisFlags_NoTickLabels, ImPlotAxisFlags_Lock)) { - ImPlot::PlotLine("##entropy_line", this->m_blockEntropy.data(), this->m_blockEntropy.size()); - - if (ImPlot::DragLineX("Position", &this->m_entropyHandlePosition, false)) { - u64 address = u64(this->m_entropyHandlePosition * this->m_blockSize) + provider->getBaseAddress(); - address = std::min(address, provider->getBaseAddress() + provider->getSize() - 1); - ImHexApi::HexEditor::setSelection(address, 1); + for (auto &[name, value] : provider->getDataInformation()) { + ImGui::TableNextColumn(); + ImGui::TextFormatted("{}", name); + ImGui::TableNextColumn(); + ImGui::TextFormattedWrapped("{}", value); } - ImPlot::EndPlot(); - } + ImGui::TableNextColumn(); + ImGui::TextFormatted("{}", "hex.builtin.view.information.region"_lang); + ImGui::TableNextColumn(); + ImGui::TextFormatted("0x{:X} - 0x{:X}", this->m_analyzedRegion.getStartAddress(), this->m_analyzedRegion.getEndAddress()); - ImPlot::PopStyleColor(); - ImGui::PopStyleColor(); + ImGui::EndTable(); + } ImGui::NewLine(); - ImGui::LabelText("hex.builtin.view.information.block_size"_lang, "%s", hex::format("hex.builtin.view.information.block_size.desc"_lang, this->m_blockEntropy.size(), this->m_blockSize).c_str()); - ImGui::LabelText("hex.builtin.view.information.file_entropy"_lang, "%.8f", this->m_averageEntropy); - ImGui::LabelText("hex.builtin.view.information.highest_entropy"_lang, "%.8f", this->m_highestBlockEntropy); + // Magic information + if (!(this->m_dataDescription.empty() && this->m_dataMimeType.empty())) { + ImGui::Header("hex.builtin.view.information.magic"_lang); + + if (ImGui::BeginTable("magic", 2, ImGuiTableFlags_SizingFixedFit | ImGuiTableFlags_RowBg)) { + ImGui::TableSetupColumn("type"); + ImGui::TableSetupColumn("value", ImGuiTableColumnFlags_WidthStretch); + + ImGui::TableNextRow(); + + if (!this->m_dataDescription.empty()) { + ImGui::TableNextColumn(); + ImGui::TextUnformatted("hex.builtin.view.information.description"_lang); + ImGui::TableNextColumn(); + ImGui::TextFormattedWrapped("{}", this->m_dataDescription.c_str()); + } + + if (!this->m_dataMimeType.empty()) { + ImGui::TableNextColumn(); + ImGui::TextUnformatted("hex.builtin.view.information.mime"_lang); + ImGui::TableNextColumn(); + ImGui::TextFormattedWrapped("{}", this->m_dataMimeType.c_str()); + } + + ImGui::EndTable(); + } + } + + // Information analysis + { + + ImGui::Header("hex.builtin.view.information.info_analysis"_lang); + + ImGui::PushStyleColor(ImGuiCol_FrameBg, ImGui::GetColorU32(ImGuiCol_WindowBg)); + ImPlot::PushStyleColor(ImPlotCol_FrameBg, ImGui::GetColorU32(ImGuiCol_WindowBg)); + + ImGui::TextUnformatted("hex.builtin.view.information.distribution"_lang); + ImPlot::SetNextPlotLimits(0, 256, 0.5, float(*std::max_element(this->m_valueCounts.begin(), this->m_valueCounts.end())) * 1.1F, ImGuiCond_Always); + if (ImPlot::BeginPlot("##distribution", "Address", "Count", ImVec2(-1, 0), ImPlotFlags_NoChild | ImPlotFlags_NoLegend | ImPlotFlags_NoMenus | ImPlotFlags_NoBoxSelect, ImPlotAxisFlags_Lock, ImPlotAxisFlags_Lock | ImPlotAxisFlags_LogScale)) { + static auto x = [] { + std::array result { 0 }; + std::iota(result.begin(), result.end(), 0); + return result; + }(); + + ImPlot::PlotBars("##bytes", x.data(), this->m_valueCounts.data(), x.size(), 1.0); + + ImPlot::EndPlot(); + } + + ImGui::NewLine(); + + ImGui::TextUnformatted("hex.builtin.view.information.entropy"_lang); + + ImPlot::SetNextPlotLimits(0, this->m_blockEntropy.size(), -0.1, 1.1, ImGuiCond_Always); + if (ImPlot::BeginPlot("##entropy", "Address", "Entropy", ImVec2(-1, 0), ImPlotFlags_NoChild | ImPlotFlags_CanvasOnly | ImPlotFlags_AntiAliased, ImPlotAxisFlags_Lock | ImPlotAxisFlags_NoTickLabels, ImPlotAxisFlags_Lock)) { + ImPlot::PlotLine("##entropy_line", this->m_blockEntropy.data(), this->m_blockEntropy.size()); + + if (ImPlot::DragLineX("Position", &this->m_entropyHandlePosition, false)) { + u64 address = u64(this->m_entropyHandlePosition * this->m_blockSize) + provider->getBaseAddress(); + address = std::min(address, provider->getBaseAddress() + provider->getSize() - 1); + ImHexApi::HexEditor::setSelection(address, 1); + } + + ImPlot::EndPlot(); + } + + ImPlot::PopStyleColor(); + ImGui::PopStyleColor(); + + ImGui::NewLine(); + } + + // Entropy information + if (ImGui::BeginTable("entropy", 2, ImGuiTableFlags_SizingFixedFit | ImGuiTableFlags_RowBg)) { + ImGui::TableSetupColumn("type"); + ImGui::TableSetupColumn("value", ImGuiTableColumnFlags_WidthStretch); + + ImGui::TableNextRow(); + + ImGui::TableNextColumn(); + ImGui::TextFormatted("{}", "hex.builtin.view.information.block_size"_lang); + ImGui::TableNextColumn(); + ImGui::TextFormatted("hex.builtin.view.information.block_size.desc"_lang, this->m_blockEntropy.size(), this->m_blockSize); + + ImGui::TableNextColumn(); + ImGui::TextFormatted("{}", "hex.builtin.view.information.file_entropy"_lang); + ImGui::TableNextColumn(); + ImGui::TextFormatted("{:.8f}", this->m_averageEntropy); + + ImGui::TableNextColumn(); + ImGui::TextFormatted("{}", "hex.builtin.view.information.highest_entropy"_lang); + ImGui::TableNextColumn(); + ImGui::TextFormatted("{:.8f}", this->m_highestBlockEntropy); + + ImGui::EndTable(); + } if (this->m_averageEntropy > 0.83 && this->m_highestBlockEntropy > 0.9) { ImGui::NewLine();