mirror of
https://github.com/WerWolv/ImHex
synced 2024-10-23 03:01:38 +00:00
817 lines
32 KiB
C++
817 lines
32 KiB
C++
#pragma once
|
|
|
|
#include <hex.hpp>
|
|
|
|
#include "imgui.h"
|
|
#include "imgui_memory_editor.h"
|
|
|
|
#include "providers/provider.hpp"
|
|
#include "helpers/utils.hpp"
|
|
#include "lang/token.hpp"
|
|
|
|
#include <cstring>
|
|
#include <random>
|
|
#include <string>
|
|
|
|
namespace hex::lang {
|
|
|
|
using namespace ::std::literals::string_literals;
|
|
|
|
namespace {
|
|
|
|
std::string makeDisplayable(u8 *data, size_t size) {
|
|
std::string result;
|
|
for (u8* c = data; c < (data + size); c++) {
|
|
if (iscntrl(*c) || *c > 0x7F)
|
|
result += " ";
|
|
else
|
|
result += *c;
|
|
}
|
|
|
|
if (*(data + size - 1) == '\x00')
|
|
result.pop_back();
|
|
|
|
return result;
|
|
}
|
|
|
|
}
|
|
|
|
class PatternData {
|
|
public:
|
|
PatternData(u64 offset, size_t size, u32 color = 0)
|
|
: m_offset(offset), m_size(size), m_color(color) {
|
|
constexpr u32 Palette[] = { 0x70b4771f, 0x700e7fff, 0x702ca02c, 0x702827d6, 0x70bd6794, 0x704b568c, 0x70c277e3, 0x707f7f7f, 0x7022bdbc, 0x70cfbe17 };
|
|
|
|
if (color != 0)
|
|
return;
|
|
|
|
this->m_color = Palette[PatternData::s_paletteOffset++];
|
|
|
|
if (PatternData::s_paletteOffset >= (sizeof(Palette) / sizeof(u32)))
|
|
PatternData::s_paletteOffset = 0;
|
|
}
|
|
virtual ~PatternData() = default;
|
|
|
|
virtual PatternData* clone() = 0;
|
|
|
|
[[nodiscard]] u64 getOffset() const { return this->m_offset; }
|
|
[[nodiscard]] size_t getSize() const { return this->m_size; }
|
|
|
|
[[nodiscard]] const std::string& getVariableName() const { return this->m_variableName; }
|
|
void setVariableName(std::string name) { this->m_variableName = std::move(name); }
|
|
|
|
[[nodiscard]] const std::string& getTypeName() const { return this->m_typeName; }
|
|
void setTypeName(std::string name) { this->m_typeName = std::move(name); }
|
|
|
|
[[nodiscard]] u32 getColor() const { return this->m_color; }
|
|
void setColor(u32 color) { this->m_color = color; }
|
|
|
|
[[nodiscard]] std::endian getEndian() const { return this->m_endian; }
|
|
void setEndian(std::endian endian) { this->m_endian = endian; }
|
|
|
|
virtual void createEntry(prv::Provider* &provider) = 0;
|
|
[[nodiscard]] virtual std::string getFormattedName() const = 0;
|
|
|
|
virtual std::optional<u32> highlightBytes(size_t offset) {
|
|
auto currOffset = this->getOffset();
|
|
if (offset >= currOffset && offset < (currOffset + this->getSize()))
|
|
return this->getColor();
|
|
else
|
|
return { };
|
|
}
|
|
|
|
virtual std::map<u64, u32> getHighlightedAddresses() {
|
|
if (this->m_highlightedAddresses.empty()) {
|
|
for (u64 i = 0; i < this->getSize(); i++)
|
|
this->m_highlightedAddresses.insert({ this->getOffset() + i, this->getColor() });
|
|
}
|
|
|
|
return this->m_highlightedAddresses;
|
|
}
|
|
|
|
virtual void sort(ImGuiTableSortSpecs *sortSpecs, prv::Provider *provider) { }
|
|
|
|
static bool sortPatternDataTable(ImGuiTableSortSpecs *sortSpecs, prv::Provider *provider, lang::PatternData* left, lang::PatternData* right) {
|
|
if (sortSpecs->Specs->ColumnUserID == ImGui::GetID("name")) {
|
|
if (sortSpecs->Specs->SortDirection == ImGuiSortDirection_Ascending)
|
|
return left->getVariableName() > right->getVariableName();
|
|
else
|
|
return left->getVariableName() < right->getVariableName();
|
|
}
|
|
else if (sortSpecs->Specs->ColumnUserID == ImGui::GetID("offset")) {
|
|
if (sortSpecs->Specs->SortDirection == ImGuiSortDirection_Ascending)
|
|
return left->getOffset() > right->getOffset();
|
|
else
|
|
return left->getOffset() < right->getOffset();
|
|
}
|
|
else if (sortSpecs->Specs->ColumnUserID == ImGui::GetID("size")) {
|
|
if (sortSpecs->Specs->SortDirection == ImGuiSortDirection_Ascending)
|
|
return left->getSize() > right->getSize();
|
|
else
|
|
return left->getSize() < right->getSize();
|
|
}
|
|
else if (sortSpecs->Specs->ColumnUserID == ImGui::GetID("value")) {
|
|
size_t biggerSize = std::max(left->getSize(), right->getSize());
|
|
std::vector<u8> leftBuffer(biggerSize, 0x00), rightBuffer(biggerSize, 0x00);
|
|
|
|
provider->read(left->getOffset(), leftBuffer.data(), left->getSize());
|
|
provider->read(right->getOffset(), rightBuffer.data(), right->getSize());
|
|
|
|
if (left->m_endian != std::endian::native)
|
|
std::reverse(leftBuffer.begin(), leftBuffer.end());
|
|
if (right->m_endian != std::endian::native)
|
|
std::reverse(rightBuffer.begin(), rightBuffer.end());
|
|
|
|
if (sortSpecs->Specs->SortDirection == ImGuiSortDirection_Ascending)
|
|
return leftBuffer > rightBuffer;
|
|
else
|
|
return leftBuffer < rightBuffer;
|
|
}
|
|
else if (sortSpecs->Specs->ColumnUserID == ImGui::GetID("type")) {
|
|
if (sortSpecs->Specs->SortDirection == ImGuiSortDirection_Ascending)
|
|
return left->getTypeName() > right->getTypeName();
|
|
else
|
|
return left->getTypeName() < right->getTypeName();
|
|
}
|
|
else if (sortSpecs->Specs->ColumnUserID == ImGui::GetID("color")) {
|
|
if (sortSpecs->Specs->SortDirection == ImGuiSortDirection_Ascending)
|
|
return left->getColor() > right->getColor();
|
|
else
|
|
return left->getColor() < right->getColor();
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
static void resetPalette() { PatternData::s_paletteOffset = 0; }
|
|
|
|
protected:
|
|
void createDefaultEntry(std::string_view value) const {
|
|
ImGui::TableNextRow();
|
|
ImGui::TreeNodeEx(this->getVariableName().c_str(), ImGuiTreeNodeFlags_Leaf | ImGuiTreeNodeFlags_NoTreePushOnOpen | ImGuiTreeNodeFlags_SpanFullWidth | ImGuiTreeNodeFlags_AllowItemOverlap);
|
|
ImGui::TableNextColumn();
|
|
if (ImGui::Selectable(("##PatternDataLine"s + std::to_string(this->getOffset())).c_str(), false, ImGuiSelectableFlags_SpanAllColumns | ImGuiSelectableFlags_AllowItemOverlap)) {
|
|
Region selectRegion = { this->getOffset(), this->getSize() };
|
|
View::postEvent(Events::SelectionChangeRequest, &selectRegion);
|
|
}
|
|
ImGui::SameLine();
|
|
ImGui::Text("%s", this->getVariableName().c_str());
|
|
ImGui::TableNextColumn();
|
|
ImGui::ColorButton("color", ImColor(this->getColor()), ImGuiColorEditFlags_NoTooltip, ImVec2(ImGui::GetColumnWidth(), ImGui::GetTextLineHeight()));
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("0x%08llx : 0x%08llx", this->getOffset(), this->getOffset() + this->getSize() - 1);
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("0x%04llx", this->getSize());
|
|
ImGui::TableNextColumn();
|
|
ImGui::TextColored(ImColor(0xFF9BC64D), "%s", this->getFormattedName().c_str());
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("%s", value.data());
|
|
}
|
|
|
|
protected:
|
|
std::endian m_endian = std::endian::native;
|
|
std::map<u64, u32> m_highlightedAddresses;
|
|
|
|
private:
|
|
u64 m_offset;
|
|
size_t m_size;
|
|
|
|
u32 m_color;
|
|
std::string m_variableName;
|
|
std::string m_typeName;
|
|
|
|
static inline u8 s_paletteOffset = 0;
|
|
|
|
};
|
|
|
|
class PatternDataPadding : public PatternData {
|
|
public:
|
|
PatternDataPadding(u64 offset, size_t size) : PatternData(offset, size, 0x00FFFFFF) { }
|
|
|
|
PatternData* clone() override {
|
|
return new PatternDataPadding(*this);
|
|
}
|
|
|
|
void createEntry(prv::Provider* &provider) override {
|
|
}
|
|
|
|
[[nodiscard]] std::string getFormattedName() const override {
|
|
return "";
|
|
}
|
|
};
|
|
|
|
class PatternDataPointer : public PatternData {
|
|
public:
|
|
PatternDataPointer(u64 offset, size_t size, PatternData *pointedAt, u32 color = 0)
|
|
: PatternData(offset, size, color), m_pointedAt(pointedAt) {
|
|
this->m_pointedAt->setVariableName("*" + this->m_pointedAt->getVariableName());
|
|
}
|
|
|
|
PatternData* clone() override {
|
|
return new PatternDataPointer(*this);
|
|
}
|
|
|
|
void createEntry(prv::Provider* &provider) override {
|
|
u64 data = 0;
|
|
provider->read(this->getOffset(), &data, this->getSize());
|
|
data = hex::changeEndianess(data, this->getSize(), this->getEndian());
|
|
|
|
ImGui::TableNextRow();
|
|
ImGui::TableNextColumn();
|
|
bool open = ImGui::TreeNodeEx(this->getVariableName().c_str(), ImGuiTreeNodeFlags_SpanFullWidth);
|
|
ImGui::TableNextColumn();
|
|
ImGui::ColorButton("color", ImColor(this->getColor()), ImGuiColorEditFlags_NoTooltip, ImVec2(ImGui::GetColumnWidth(), ImGui::GetTextLineHeight()));
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("0x%08llx : 0x%08llx", this->getOffset(), this->getOffset() + this->getSize() - 1);
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("0x%04llx", this->getSize());
|
|
ImGui::TableNextColumn();
|
|
ImGui::TextColored(ImColor(0xFF9BC64D), "%s*", this->m_pointedAt->getFormattedName().c_str());
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("*(0x%llx)", data);
|
|
|
|
if (open) {
|
|
this->m_pointedAt->createEntry(provider);
|
|
|
|
ImGui::TreePop();
|
|
}
|
|
}
|
|
|
|
std::optional<u32> highlightBytes(size_t offset) override {
|
|
if (offset >= this->getOffset() && offset < (this->getOffset() + this->getSize()))
|
|
return this->getColor();
|
|
else if (auto color = this->m_pointedAt->highlightBytes(offset); color.has_value())
|
|
return color.value();
|
|
else
|
|
return { };
|
|
}
|
|
|
|
std::map<u64, u32> getHighlightedAddresses() override {
|
|
if (this->m_highlightedAddresses.empty()) {
|
|
auto ownAddresses = PatternData::getHighlightedAddresses();
|
|
auto pointedToAddresses = this->m_pointedAt->getHighlightedAddresses();
|
|
|
|
ownAddresses.merge(pointedToAddresses);
|
|
this->m_highlightedAddresses = ownAddresses;
|
|
}
|
|
|
|
return this->m_highlightedAddresses;
|
|
}
|
|
[[nodiscard]] std::string getFormattedName() const override {
|
|
return "Pointer";
|
|
}
|
|
|
|
[[nodiscard]] PatternData* getPointedAtPattern() {
|
|
return this->m_pointedAt;
|
|
}
|
|
|
|
private:
|
|
PatternData *m_pointedAt;
|
|
};
|
|
|
|
class PatternDataUnsigned : public PatternData {
|
|
public:
|
|
PatternDataUnsigned(u64 offset, size_t size, u32 color = 0)
|
|
: PatternData(offset, size, color) { }
|
|
|
|
PatternData* clone() override {
|
|
return new PatternDataUnsigned(*this);
|
|
}
|
|
|
|
void createEntry(prv::Provider* &provider) override {
|
|
u64 data = 0;
|
|
provider->read(this->getOffset(), &data, this->getSize());
|
|
data = hex::changeEndianess(data, this->getSize(), this->getEndian());
|
|
|
|
this->createDefaultEntry(hex::format("%lu (0x%0*lx)", data, this->getSize() * 2, data));
|
|
}
|
|
|
|
[[nodiscard]] std::string getFormattedName() const override {
|
|
switch (this->getSize()) {
|
|
case 1: return "u8";
|
|
case 2: return "u16";
|
|
case 4: return "u32";
|
|
case 8: return "u64";
|
|
case 16: return "u128";
|
|
default: return "Unsigned data";
|
|
}
|
|
}
|
|
};
|
|
|
|
class PatternDataSigned : public PatternData {
|
|
public:
|
|
PatternDataSigned(u64 offset, size_t size, u32 color = 0)
|
|
: PatternData(offset, size, color) { }
|
|
|
|
PatternData* clone() override {
|
|
return new PatternDataSigned(*this);
|
|
}
|
|
|
|
void createEntry(prv::Provider* &provider) override {
|
|
u64 data = 0;
|
|
provider->read(this->getOffset(), &data, this->getSize());
|
|
data = hex::changeEndianess(data, this->getSize(), this->getEndian());
|
|
|
|
s64 signedData = hex::signExtend(data, this->getSize(), 64);
|
|
|
|
this->createDefaultEntry(hex::format("%ld (0x%0*lx)", signedData, this->getSize() * 2, data));
|
|
}
|
|
|
|
[[nodiscard]] std::string getFormattedName() const override {
|
|
switch (this->getSize()) {
|
|
case 1: return "s8";
|
|
case 2: return "s16";
|
|
case 4: return "s32";
|
|
case 8: return "s64";
|
|
case 16: return "s128";
|
|
default: return "Signed data";
|
|
}
|
|
}
|
|
};
|
|
|
|
class PatternDataFloat : public PatternData {
|
|
public:
|
|
PatternDataFloat(u64 offset, size_t size, u32 color = 0)
|
|
: PatternData(offset, size, color) { }
|
|
|
|
PatternData* clone() override {
|
|
return new PatternDataFloat(*this);
|
|
}
|
|
|
|
void createEntry(prv::Provider* &provider) override {
|
|
if (this->getSize() == 4) {
|
|
u32 data = 0;
|
|
provider->read(this->getOffset(), &data, 4);
|
|
data = hex::changeEndianess(data, 4, this->getEndian());
|
|
|
|
this->createDefaultEntry(hex::format("%e (0x%0*lX)", *reinterpret_cast<float*>(&data), this->getSize() * 2, data));
|
|
} else if (this->getSize() == 8) {
|
|
u64 data = 0;
|
|
provider->read(this->getOffset(), &data, 8);
|
|
data = hex::changeEndianess(data, 8, this->getEndian());
|
|
|
|
this->createDefaultEntry(hex::format("%e (0x%0*lX)", *reinterpret_cast<double*>(&data), this->getSize() * 2, data));
|
|
}
|
|
}
|
|
|
|
[[nodiscard]] std::string getFormattedName() const override {
|
|
switch (this->getSize()) {
|
|
case 4: return "float";
|
|
case 8: return "double";
|
|
default: return "Floating point data";
|
|
}
|
|
}
|
|
};
|
|
|
|
class PatternDataBoolean : public PatternData {
|
|
public:
|
|
explicit PatternDataBoolean(u64 offset, u32 color = 0)
|
|
: PatternData(offset, 1, color) { }
|
|
|
|
PatternData* clone() override {
|
|
return new PatternDataBoolean(*this);
|
|
}
|
|
|
|
void createEntry(prv::Provider* &provider) override {
|
|
u8 boolean;
|
|
provider->read(this->getOffset(), &boolean, 1);
|
|
|
|
if (boolean == 0)
|
|
this->createDefaultEntry("false");
|
|
else if (boolean == 1)
|
|
this->createDefaultEntry("true");
|
|
else
|
|
this->createDefaultEntry("true*");
|
|
}
|
|
|
|
[[nodiscard]] std::string getFormattedName() const override {
|
|
return "bool";
|
|
}
|
|
};
|
|
|
|
class PatternDataCharacter : public PatternData {
|
|
public:
|
|
explicit PatternDataCharacter(u64 offset, u32 color = 0)
|
|
: PatternData(offset, 1, color) { }
|
|
|
|
PatternData* clone() override {
|
|
return new PatternDataCharacter(*this);
|
|
}
|
|
|
|
void createEntry(prv::Provider* &provider) override {
|
|
char character;
|
|
provider->read(this->getOffset(), &character, 1);
|
|
|
|
this->createDefaultEntry(hex::format("'%c'", character));
|
|
}
|
|
|
|
[[nodiscard]] std::string getFormattedName() const override {
|
|
return "char";
|
|
}
|
|
};
|
|
|
|
class PatternDataString : public PatternData {
|
|
public:
|
|
PatternDataString(u64 offset, size_t size, u32 color = 0)
|
|
: PatternData(offset, size, color) { }
|
|
|
|
PatternData* clone() override {
|
|
return new PatternDataString(*this);
|
|
}
|
|
|
|
void createEntry(prv::Provider* &provider) override {
|
|
std::vector<u8> buffer(this->getSize() + 1, 0x00);
|
|
provider->read(this->getOffset(), buffer.data(), this->getSize());
|
|
buffer[this->getSize()] = '\0';
|
|
|
|
this->createDefaultEntry(hex::format("\"%s\"", makeDisplayable(buffer.data(), this->getSize()).c_str()));
|
|
}
|
|
|
|
[[nodiscard]] std::string getFormattedName() const override {
|
|
return "String";
|
|
}
|
|
};
|
|
|
|
class PatternDataArray : public PatternData {
|
|
public:
|
|
PatternDataArray(u64 offset, size_t size, std::vector<PatternData*> entries, u32 color = 0)
|
|
: PatternData(offset, size, color), m_entries(std::move(entries)) { }
|
|
|
|
PatternDataArray(const PatternDataArray &other) : PatternData(other.getOffset(), other.getSize(), other.getColor()) {
|
|
for (const auto &entry : other.m_entries)
|
|
this->m_entries.push_back(entry->clone());
|
|
}
|
|
|
|
PatternData* clone() override {
|
|
return new PatternDataArray(*this);
|
|
}
|
|
|
|
void createEntry(prv::Provider* &provider) override {
|
|
if (this->m_entries.empty())
|
|
return;
|
|
|
|
ImGui::TableNextRow();
|
|
ImGui::TableNextColumn();
|
|
bool open = ImGui::TreeNodeEx(this->getVariableName().c_str(), ImGuiTreeNodeFlags_SpanFullWidth);
|
|
ImGui::TableNextColumn();
|
|
ImGui::ColorButton("color", ImColor(this->getColor()), ImGuiColorEditFlags_NoTooltip, ImVec2(ImGui::GetColumnWidth(), ImGui::GetTextLineHeight()));
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("0x%08llx : 0x%08llx", this->getOffset(), this->getOffset() + this->getSize() - 1);
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("0x%04llx", this->getSize());
|
|
ImGui::TableNextColumn();
|
|
ImGui::TextColored(ImColor(0xFF9BC64D), "%s", this->m_entries[0]->getTypeName().c_str());
|
|
ImGui::SameLine(0, 0);
|
|
|
|
ImGui::TextUnformatted("[");
|
|
ImGui::SameLine(0, 0);
|
|
ImGui::TextColored(ImColor(0xFF00FF00), "%llu", this->m_entries.size());
|
|
ImGui::SameLine(0, 0);
|
|
ImGui::TextUnformatted("]");
|
|
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("%s", "{ ... }");
|
|
|
|
if (open) {
|
|
for (auto &member : this->m_entries)
|
|
member->createEntry(provider);
|
|
|
|
ImGui::TreePop();
|
|
}
|
|
}
|
|
|
|
std::optional<u32> highlightBytes(size_t offset) override{
|
|
for (auto &entry : this->m_entries) {
|
|
if (auto color = entry->highlightBytes(offset); color.has_value())
|
|
return color.value();
|
|
}
|
|
|
|
return { };
|
|
}
|
|
|
|
std::map<u64, u32> getHighlightedAddresses() override {
|
|
if (this->m_highlightedAddresses.empty()) {
|
|
for (auto &entry : this->m_entries) {
|
|
this->m_highlightedAddresses.merge(entry->getHighlightedAddresses());
|
|
}
|
|
}
|
|
|
|
return this->m_highlightedAddresses;
|
|
}
|
|
[[nodiscard]] std::string getFormattedName() const override {
|
|
return this->m_entries[0]->getTypeName() + "[" + std::to_string(this->m_entries.size()) + "]";
|
|
}
|
|
|
|
private:
|
|
std::vector<PatternData*> m_entries;
|
|
};
|
|
|
|
class PatternDataStruct : public PatternData {
|
|
public:
|
|
PatternDataStruct(u64 offset, size_t size, const std::vector<PatternData*> & members, u32 color = 0)
|
|
: PatternData(offset, size, color), m_members(members), m_sortedMembers(members) { }
|
|
|
|
PatternDataStruct(const PatternDataStruct &other) : PatternData(other.getOffset(), other.getSize(), other.getColor()) {
|
|
for (const auto &member : other.m_members)
|
|
this->m_members.push_back(member->clone());
|
|
}
|
|
|
|
PatternData* clone() override {
|
|
return new PatternDataStruct(*this);
|
|
}
|
|
|
|
void createEntry(prv::Provider* &provider) override {
|
|
ImGui::TableNextRow();
|
|
ImGui::TableNextColumn();
|
|
bool open = ImGui::TreeNodeEx(this->getVariableName().c_str(), ImGuiTreeNodeFlags_SpanFullWidth);
|
|
ImGui::TableNextColumn();
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("0x%08llx : 0x%08llx", this->getOffset(), this->getOffset() + this->getSize() - (this->getSize() == 0 ? 0 : 1));
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("0x%04llx", this->getSize());
|
|
ImGui::TableNextColumn();
|
|
ImGui::TextColored(ImColor(0xFFD69C56), "struct"); ImGui::SameLine(); ImGui::Text("%s", this->getTypeName().c_str());
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("%s", "{ ... }");
|
|
|
|
if (open) {
|
|
for (auto &member : this->m_sortedMembers)
|
|
member->createEntry(provider);
|
|
|
|
ImGui::TreePop();
|
|
}
|
|
|
|
}
|
|
|
|
std::optional<u32> highlightBytes(size_t offset) override{
|
|
for (auto &member : this->m_members) {
|
|
if (auto color = member->highlightBytes(offset); color.has_value())
|
|
return color.value();
|
|
}
|
|
|
|
return { };
|
|
}
|
|
|
|
std::map<u64, u32> getHighlightedAddresses() override {
|
|
if (this->m_highlightedAddresses.empty()) {
|
|
for (auto &member : this->m_members) {
|
|
this->m_highlightedAddresses.merge(member->getHighlightedAddresses());
|
|
}
|
|
}
|
|
|
|
return this->m_highlightedAddresses;
|
|
}
|
|
|
|
void sort(ImGuiTableSortSpecs *sortSpecs, prv::Provider *provider) override {
|
|
this->m_sortedMembers = this->m_members;
|
|
|
|
std::sort(this->m_sortedMembers.begin(), this->m_sortedMembers.end(), [&sortSpecs, &provider](PatternData *left, PatternData *right) {
|
|
return PatternData::sortPatternDataTable(sortSpecs, provider, left, right);
|
|
});
|
|
|
|
for (auto &member : this->m_members)
|
|
member->sort(sortSpecs, provider);
|
|
}
|
|
|
|
[[nodiscard]] std::string getFormattedName() const override {
|
|
return "struct " + PatternData::getTypeName();
|
|
}
|
|
|
|
const auto& getMembers() const {
|
|
return this->m_members;
|
|
}
|
|
|
|
private:
|
|
std::vector<PatternData*> m_members;
|
|
std::vector<PatternData*> m_sortedMembers;
|
|
};
|
|
|
|
class PatternDataUnion : public PatternData {
|
|
public:
|
|
PatternDataUnion(u64 offset, size_t size, const std::vector<PatternData*> & members, u32 color = 0)
|
|
: PatternData(offset, size, color), m_members(members), m_sortedMembers(members) { }
|
|
|
|
PatternDataUnion(const PatternDataUnion &other) : PatternData(other.getOffset(), other.getSize(), other.getColor()) {
|
|
for (const auto &member : other.m_members)
|
|
this->m_members.push_back(member->clone());
|
|
}
|
|
|
|
PatternData* clone() override {
|
|
return new PatternDataUnion(*this);
|
|
}
|
|
|
|
void createEntry(prv::Provider* &provider) override {
|
|
ImGui::TableNextRow();
|
|
ImGui::TableNextColumn();
|
|
bool open = ImGui::TreeNodeEx(this->getVariableName().c_str(), ImGuiTreeNodeFlags_SpanFullWidth);
|
|
ImGui::TableNextColumn();
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("0x%08llx : 0x%08llx", this->getOffset(), std::max(this->getOffset() + this->getSize() - (this->getSize() == 0 ? 0 : 1), u64(0)));
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("0x%04llx", this->getSize());
|
|
ImGui::TableNextColumn();
|
|
ImGui::TextColored(ImColor(0xFFD69C56), "union"); ImGui::SameLine(); ImGui::Text("%s", PatternData::getTypeName().c_str());
|
|
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("%s", "{ ... }");
|
|
|
|
if (open) {
|
|
for (auto &member : this->m_sortedMembers)
|
|
member->createEntry(provider);
|
|
|
|
ImGui::TreePop();
|
|
}
|
|
|
|
}
|
|
|
|
std::optional<u32> highlightBytes(size_t offset) override{
|
|
for (auto &member : this->m_members) {
|
|
if (auto color = member->highlightBytes(offset); color.has_value())
|
|
return color.value();
|
|
}
|
|
|
|
return { };
|
|
}
|
|
|
|
std::map<u64, u32> getHighlightedAddresses() override {
|
|
if (this->m_highlightedAddresses.empty()) {
|
|
for (auto &member : this->m_members) {
|
|
this->m_highlightedAddresses.merge(member->getHighlightedAddresses());
|
|
}
|
|
}
|
|
|
|
return this->m_highlightedAddresses;
|
|
}
|
|
|
|
void sort(ImGuiTableSortSpecs *sortSpecs, prv::Provider *provider) override {
|
|
this->m_sortedMembers = this->m_members;
|
|
|
|
std::sort(this->m_sortedMembers.begin(), this->m_sortedMembers.end(), [&sortSpecs, &provider](PatternData *left, PatternData *right) {
|
|
return PatternData::sortPatternDataTable(sortSpecs, provider, left, right);
|
|
});
|
|
|
|
for (auto &member : this->m_members)
|
|
member->sort(sortSpecs, provider);
|
|
}
|
|
|
|
[[nodiscard]] std::string getFormattedName() const override {
|
|
return "union " + PatternData::getTypeName();;
|
|
}
|
|
|
|
const auto& getMembers() const {
|
|
return this->m_members;
|
|
}
|
|
|
|
private:
|
|
std::vector<PatternData*> m_members;
|
|
std::vector<PatternData*> m_sortedMembers;
|
|
};
|
|
|
|
class PatternDataEnum : public PatternData {
|
|
public:
|
|
PatternDataEnum(u64 offset, size_t size, std::vector<std::pair<Token::IntegerLiteral, std::string>> enumValues, u32 color = 0)
|
|
: PatternData(offset, size, color), m_enumValues(std::move(enumValues)) { }
|
|
|
|
PatternData* clone() override {
|
|
return new PatternDataEnum(*this);
|
|
}
|
|
|
|
void createEntry(prv::Provider* &provider) override {
|
|
u64 value = 0;
|
|
provider->read(this->getOffset(), &value, this->getSize());
|
|
value = hex::changeEndianess(value, this->getSize(), this->getEndian());
|
|
|
|
std::string valueString = PatternData::getTypeName() + "::";
|
|
|
|
bool foundValue = false;
|
|
for (auto &[entryValueLiteral, entryName] : this->m_enumValues) {
|
|
bool matches = std::visit([&, name = entryName](auto &&entryValue) {
|
|
if (value == entryValue) {
|
|
valueString += name;
|
|
foundValue = true;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}, entryValueLiteral.second);
|
|
if (matches)
|
|
break;
|
|
}
|
|
|
|
if (!foundValue)
|
|
valueString += "???";
|
|
|
|
ImGui::TableNextRow();
|
|
ImGui::TreeNodeEx(this->getVariableName().c_str(), ImGuiTreeNodeFlags_Leaf | ImGuiTreeNodeFlags_NoTreePushOnOpen | ImGuiTreeNodeFlags_SpanFullWidth);
|
|
ImGui::TableNextColumn();
|
|
if (ImGui::Selectable(("##PatternDataLine"s + std::to_string(this->getOffset())).c_str(), false, ImGuiSelectableFlags_SpanAllColumns)) {
|
|
Region selectRegion = { this->getOffset(), this->getSize() };
|
|
View::postEvent(Events::SelectionChangeRequest, &selectRegion);
|
|
}
|
|
ImGui::SameLine();
|
|
ImGui::Text("%s", this->getVariableName().c_str());
|
|
ImGui::TableNextColumn();
|
|
ImGui::ColorButton("color", ImColor(this->getColor()), ImGuiColorEditFlags_NoTooltip, ImVec2(ImGui::GetColumnWidth(), ImGui::GetTextLineHeight()));
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("0x%08llx : 0x%08llx", this->getOffset(), this->getOffset() + this->getSize() - 1);
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("0x%04llx", this->getSize());
|
|
ImGui::TableNextColumn();
|
|
ImGui::TextColored(ImColor(0xFFD69C56), "enum"); ImGui::SameLine(); ImGui::Text("%s", PatternData::getTypeName().c_str());
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("%s", hex::format("%s (0x%0*lx)", valueString.c_str(), this->getSize() * 2, value).c_str());
|
|
}
|
|
|
|
[[nodiscard]] std::string getFormattedName() const override {
|
|
return "enum " + PatternData::getTypeName();
|
|
}
|
|
|
|
const auto& getEnumValues() const {
|
|
return this->m_enumValues;
|
|
}
|
|
|
|
private:
|
|
std::vector<std::pair<Token::IntegerLiteral, std::string>> m_enumValues;
|
|
};
|
|
|
|
class PatternDataBitfield : public PatternData {
|
|
public:
|
|
PatternDataBitfield(u64 offset, size_t size, std::vector<std::pair<std::string, size_t>> fields, u32 color = 0)
|
|
: PatternData(offset, size, color), m_fields(std::move(fields)) { }
|
|
|
|
PatternData* clone() override {
|
|
return new PatternDataBitfield(*this);
|
|
}
|
|
|
|
void createEntry(prv::Provider* &provider) override {
|
|
std::vector<u8> value(this->getSize(), 0);
|
|
provider->read(this->getOffset(), &value[0], value.size());
|
|
|
|
if (this->m_endian == std::endian::big)
|
|
std::reverse(value.begin(), value.end());
|
|
|
|
ImGui::TableNextRow();
|
|
ImGui::TableNextColumn();
|
|
bool open = ImGui::TreeNodeEx(this->getVariableName().c_str(), ImGuiTreeNodeFlags_SpanFullWidth);
|
|
ImGui::TableNextColumn();
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("0x%08llx : 0x%08llx", this->getOffset(), this->getOffset() + this->getSize() - 1);
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("0x%04llx", this->getSize());
|
|
ImGui::TableNextColumn();
|
|
ImGui::TextColored(ImColor(0xFFD69C56), "bitfield"); ImGui::SameLine(); ImGui::Text("%s", PatternData::getTypeName().c_str());
|
|
ImGui::TableNextColumn();
|
|
|
|
std::string valueString = "{ ";
|
|
for (u64 i = 0; i < value.size(); i++)
|
|
valueString += hex::format("%02x ", value[i]);
|
|
valueString += "}";
|
|
|
|
ImGui::TextUnformatted(valueString.c_str());
|
|
|
|
if (open) {
|
|
u16 bitOffset = 0;
|
|
for (auto &[entryName, entrySize] : this->m_fields) {
|
|
ImGui::TableNextRow();
|
|
ImGui::TreeNodeEx(this->getVariableName().c_str(), ImGuiTreeNodeFlags_Leaf | ImGuiTreeNodeFlags_NoTreePushOnOpen | ImGuiTreeNodeFlags_SpanFullWidth);
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("%s", entryName.c_str());
|
|
ImGui::TableNextColumn();
|
|
ImGui::ColorButton("color", ImColor(this->getColor()), ImGuiColorEditFlags_NoTooltip, ImVec2(ImGui::GetColumnWidth(), ImGui::GetTextLineHeight()));
|
|
ImGui::TableNextColumn();
|
|
ImGui::Text("0x%08llx : 0x%08llx", this->getOffset() + (bitOffset >> 3), this->getOffset() + ((bitOffset + entrySize) >> 3));
|
|
ImGui::TableNextColumn();
|
|
if (entrySize == 1)
|
|
ImGui::Text("%llu bit", entrySize);
|
|
else
|
|
ImGui::Text("%llu bits", entrySize);
|
|
ImGui::TableNextColumn();
|
|
ImGui::TextColored(ImColor(0xFF9BC64D), "bits");
|
|
ImGui::TableNextColumn();
|
|
{
|
|
u128 fieldValue = 0;
|
|
std::memcpy(&fieldValue, value.data() + (bitOffset / 8), (entrySize / 8) + 1);
|
|
ImGui::Text("%llx", hex::extract((bitOffset + entrySize) - 1 - ((bitOffset / 8) * 8), bitOffset - ((bitOffset / 8) * 8), fieldValue));
|
|
}
|
|
bitOffset += entrySize;
|
|
}
|
|
|
|
ImGui::TreePop();
|
|
}
|
|
|
|
}
|
|
|
|
[[nodiscard]] std::string getFormattedName() const override {
|
|
return "bitfield " + PatternData::getTypeName();
|
|
}
|
|
|
|
const auto& getFields() const {
|
|
return this->m_fields;
|
|
}
|
|
|
|
private:
|
|
std::vector<std::pair<std::string, size_t>> m_fields;
|
|
};
|
|
|
|
|
|
} |