#pragma once #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include namespace hex::pl { using namespace ::std::literals::string_literals; namespace { template T> std::string makeDisplayable(T *data, size_t size) { std::string result; for (T* c = data; c < (data + size); c++) { if (iscntrl(*c) || *c > 0x7F) result += " "; else result += static_cast(*c); } if (size != 0 && *(data + size - 1) == '\x00') result.pop_back(); return result; } } class Inlinable { public: [[nodiscard]] bool isInlined() const { return this->m_inlined; } void setInlined(bool inlined) { this->m_inlined = inlined; } private: bool m_inlined = false; }; class PatternCreationLimiter { public: explicit PatternCreationLimiter(Evaluator *evaluator) : m_evaluator(evaluator) { if (this->m_evaluator == nullptr) return; this->m_evaluator->patternCreated(); } PatternCreationLimiter(const PatternCreationLimiter &other) { this->m_evaluator = other.m_evaluator; if (this->m_evaluator == nullptr) return; this->m_evaluator->patternCreated(); } virtual ~PatternCreationLimiter() { if (this->m_evaluator == nullptr) return; this->m_evaluator->patternDestroyed(); } protected: Evaluator *m_evaluator = nullptr; }; class PatternData : public PatternCreationLimiter { public: PatternData(u64 offset, size_t size, Evaluator *evaluator, u32 color = 0) : PatternCreationLimiter(evaluator), 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[SharedData::patternPaletteOffset++]; this->m_manualColor = false; if (SharedData::patternPaletteOffset >= (sizeof(Palette) / sizeof(u32))) SharedData::patternPaletteOffset = 0; } PatternData(const PatternData &other) = default; virtual ~PatternData() = default; virtual PatternData* clone() = 0; [[nodiscard]] u64 getOffset() const { return this->m_offset; } virtual void setOffset(u64 offset) { this->m_offset = offset; } [[nodiscard]] size_t getSize() const { return this->m_size; } void setSize(size_t size) { this->m_size = 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::optional& getComment() const { return this->m_comment; } void setComment(std::string comment) { this->m_comment = std::move(comment); } [[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; } virtual void setColor(u32 color) { this->m_color = color; this->m_manualColor = true; } [[nodiscard]] bool hasOverriddenColor() const { return this->m_manualColor; } [[nodiscard]] std::endian getEndian() const { return this->m_endian; } void setEndian(std::endian endian) { this->m_endian = endian; } [[nodiscard]] std::string getDisplayName() const { return this->m_displayName.value_or(this->m_variableName); } void setDisplayName(const std::string &name) { this->m_displayName = name; } [[nodiscard]] Evaluator* getEvaluator() const { return this->m_evaluator; } [[nodiscard]] const auto& getTransformFunction() const { return this->m_transformFunction; } void setTransformFunction(const ContentRegistry::PatternLanguage::Function &function) { this->m_transformFunction = function; } [[nodiscard]] const auto& getFormatterFunction() const { return this->m_formatterFunction; } void setFormatterFunction(const ContentRegistry::PatternLanguage::Function &function) { this->m_formatterFunction = function; } virtual void createEntry(prv::Provider* &provider) = 0; [[nodiscard]] virtual std::string getFormattedName() const = 0; [[nodiscard]] virtual const PatternData* getPattern(u64 offset) const { if (offset >= this->getOffset() && offset < (this->getOffset() + this->getSize())) return this; else return nullptr; } virtual void getHighlightedAddresses(std::map &highlight) const { if (this->isHidden()) return; for (u64 i = 0; i < this->getSize(); i++) highlight.insert({ this->getOffset() + i, this->getColor() }); this->m_evaluator->handleAbort(); } virtual void sort(ImGuiTableSortSpecs *sortSpecs, prv::Provider *provider) { } [[nodiscard]] virtual std::string toString(prv::Provider *provider) const { return hex::format("{} {} @ 0x{:X}", this->getTypeName(), this->getVariableName(), this->getOffset()); } static bool sortPatternDataTable(ImGuiTableSortSpecs *sortSpecs, prv::Provider *provider, pl::PatternData* left, pl::PatternData* right) { if (sortSpecs->Specs->ColumnUserID == ImGui::GetID("name")) { if (sortSpecs->Specs->SortDirection == ImGuiSortDirection_Ascending) return left->getDisplayName() > right->getDisplayName(); else return left->getDisplayName() < right->getDisplayName(); } 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 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; } void draw(prv::Provider *provider) { if (isHidden()) return; this->createEntry(provider); } static void resetPalette() { SharedData::patternPaletteOffset = 0; } void setHidden(bool hidden) { this->m_hidden = hidden; } [[nodiscard]] bool isHidden() const { return this->m_hidden; } void setLocal(bool local) { this->m_local = local; } [[nodiscard]] bool isLocal() const { return this->m_local; } [[nodiscard]] virtual bool operator!=(const PatternData &other) const final { return !operator==(other); } [[nodiscard]] virtual bool operator==(const PatternData &other) const = 0; template [[nodiscard]] bool areCommonPropertiesEqual(const PatternData &other) const { return typeid(other) == typeid(std::remove_cvref_t) && this->m_offset == other.m_offset && this->m_size == other.m_size && this->m_hidden == other.m_hidden && this->m_endian == other.m_endian && this->m_variableName == other.m_variableName && this->m_typeName == other.m_typeName && this->m_comment == other.m_comment && this->m_local == other.m_local; } [[nodiscard]] std::string formatDisplayValue(const std::string &value, const Token::Literal &literal) const { if (!this->m_formatterFunction.has_value()) return value; else { try { auto result = this->m_formatterFunction->func(this->getEvaluator(), { literal }); if (result.has_value()) { if (auto displayValue = std::get_if(&result.value()); displayValue != nullptr) return *displayValue; else return "???"; } else { return "???"; } } catch (LogConsole::EvaluateError &error) { return "Error: " + error.second; } } } protected: void createDefaultEntry(const std::string &value, const Token::Literal &literal) const { ImGui::TableNextRow(); ImGui::TreeNodeEx(this->getDisplayName().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)) { EventManager::post(Region { this->getOffset(), this->getSize() }); } this->drawCommentTooltip(); ImGui::SameLine(); ImGui::TextUnformatted(this->getDisplayName().c_str()); ImGui::TableNextColumn(); ImGui::ColorButton("color", ImColor(this->getColor()), ImGuiColorEditFlags_NoTooltip, ImVec2(ImGui::GetColumnWidth(), ImGui::GetTextLineHeight())); ImGui::TableNextColumn(); ImGui::TextFormatted("0x{0:08X} : 0x{1:08X}", this->getOffset(), this->getOffset() + this->getSize() - 1); ImGui::TableNextColumn(); ImGui::TextFormatted("0x{0:04X}", this->getSize()); ImGui::TableNextColumn(); ImGui::TextFormattedColored(ImColor(0xFF9BC64D), "{}", this->getFormattedName()); ImGui::TableNextColumn(); ImGui::TextFormatted("{}", formatDisplayValue(value, literal)); } void drawCommentTooltip() const { if (ImGui::IsItemHovered(ImGuiHoveredFlags_AllowWhenBlockedByActiveItem) && this->getComment().has_value()) { ImGui::BeginTooltip(); ImGui::TextUnformatted(this->getComment()->c_str()); ImGui::EndTooltip(); } } protected: std::endian m_endian = std::endian::native; bool m_hidden = false; private: u64 m_offset; size_t m_size; u32 m_color; std::optional m_displayName; std::string m_variableName; std::optional m_comment; std::string m_typeName; std::optional m_formatterFunction; std::optional m_transformFunction; bool m_local = false; bool m_manualColor = false; }; class PatternDataPadding : public PatternData { public: PatternDataPadding(u64 offset, size_t size, Evaluator *evaluator) : PatternData(offset, size, evaluator, 0xFF000000) { } PatternData* clone() override { return new PatternDataPadding(*this); } void createEntry(prv::Provider* &provider) override { } [[nodiscard]] std::string getFormattedName() const override { return ""; } [[nodiscard]] bool operator==(const PatternData &other) const override { return areCommonPropertiesEqual(other); } }; class PatternDataPointer : public PatternData { public: PatternDataPointer(u64 offset, size_t size, Evaluator *evaluator, u32 color = 0) : PatternData(offset, size, evaluator, color), m_pointedAt(nullptr) { } PatternDataPointer(const PatternDataPointer &other) : PatternData(other) { this->m_pointedAt = other.m_pointedAt->clone(); } ~PatternDataPointer() override { delete this->m_pointedAt; } 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->getDisplayName().c_str(), ImGuiTreeNodeFlags_SpanFullWidth | ImGuiTreeNodeFlags_AllowItemOverlap); this->drawCommentTooltip(); ImGui::TableNextColumn(); ImGui::ColorButton("color", ImColor(this->getColor()), ImGuiColorEditFlags_NoTooltip, ImVec2(ImGui::GetColumnWidth(), ImGui::GetTextLineHeight())); ImGui::TableNextColumn(); ImGui::TextFormatted("0x{0:08X} : 0x{1:08X}", this->getOffset(), this->getOffset() + this->getSize() - 1); ImGui::TableNextColumn(); ImGui::TextFormatted("0x{0:04X}", this->getSize()); ImGui::TableNextColumn(); ImGui::TextFormattedColored(ImColor(0xFF9BC64D), "{}", this->getFormattedName()); ImGui::TableNextColumn(); ImGui::TextFormatted("{}", formatDisplayValue(hex::format("*(0x{0:X})", data), u128(data))); if (open) { this->m_pointedAt->createEntry(provider); ImGui::TreePop(); } } void getHighlightedAddresses(std::map &highlight) const override { PatternData::getHighlightedAddresses(highlight); this->m_pointedAt->getHighlightedAddresses(highlight); } [[nodiscard]] std::string getFormattedName() const override { std::string result = this->m_pointedAt->getFormattedName() + "* : "; switch (this->getSize()) { case 1: result += "u8"; break; case 2: result += "u16"; break; case 4: result += "u32"; break; case 8: result += "u64"; break; case 16: result += "u128"; break; } return result; } void setPointedAtPattern(PatternData *pattern) { this->m_pointedAt = pattern; this->m_pointedAt->setVariableName(hex::format("*({})", this->getVariableName())); this->m_pointedAt->setOffset(this->m_pointedAtAddress); } void setPointedAtAddress(u64 address) { this->m_pointedAtAddress = address; } [[nodiscard]] u64 getPointedAtAddress() const { return this->m_pointedAtAddress; } [[nodiscard]] PatternData* getPointedAtPattern() { return this->m_pointedAt; } void setColor(u32 color) override { PatternData::setColor(color); this->m_pointedAt->setColor(color); } [[nodiscard]] bool operator==(const PatternData &other) const override { return areCommonPropertiesEqual(other) && *static_cast(&other)->m_pointedAt == *this->m_pointedAt; } void rebase(u64 base) { if (this->m_pointedAt != nullptr) { this->m_pointedAtAddress = (this->m_pointedAt->getOffset() - this->m_pointerBase) + base; this->m_pointedAt->setOffset(this->m_pointedAtAddress); } this->m_pointerBase = base; } [[nodiscard]] const PatternData* getPattern(u64 offset) const override { if (offset >= this->getOffset() && offset < (this->getOffset() + this->getSize())) return this; else return this->m_pointedAt->getPattern(offset); } private: PatternData *m_pointedAt = nullptr; u64 m_pointedAtAddress = 0; u64 m_pointerBase = 0; }; class PatternDataUnsigned : public PatternData { public: PatternDataUnsigned(u64 offset, size_t size, Evaluator *evaluator, u32 color = 0) : PatternData(offset, size, evaluator, color) { } PatternData* clone() override { return new PatternDataUnsigned(*this); } void createEntry(prv::Provider* &provider) override { u128 data = 0; provider->read(this->getOffset(), &data, this->getSize()); data = hex::changeEndianess(data, this->getSize(), this->getEndian()); this->createDefaultEntry(hex::format("{:d} (0x{:0{}X})", data, 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"; } } [[nodiscard]] bool operator==(const PatternData &other) const override { return areCommonPropertiesEqual(other); } }; class PatternDataSigned : public PatternData { public: PatternDataSigned(u64 offset, size_t size, Evaluator *evaluator, u32 color = 0) : PatternData(offset, size, evaluator, color) { } PatternData* clone() override { return new PatternDataSigned(*this); } void createEntry(prv::Provider* &provider) override { s128 data = 0; provider->read(this->getOffset(), &data, this->getSize()); data = hex::changeEndianess(data, this->getSize(), this->getEndian()); data = hex::signExtend(this->getSize() * 8, data); this->createDefaultEntry(hex::format("{:d} (0x{:0{}X})", data, data, 1 * 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"; } } [[nodiscard]] bool operator==(const PatternData &other) const override { return areCommonPropertiesEqual(other); } }; class PatternDataFloat : public PatternData { public: PatternDataFloat(u64 offset, size_t size, Evaluator *evaluator, u32 color = 0) : PatternData(offset, size, evaluator, 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{}X})", *reinterpret_cast(&data), data, this->getSize() * 2), *reinterpret_cast(&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{}X})", *reinterpret_cast(&data), data, this->getSize() * 2), *reinterpret_cast(&data)); } } [[nodiscard]] std::string getFormattedName() const override { switch (this->getSize()) { case 4: return "float"; case 8: return "double"; default: return "Floating point data"; } } [[nodiscard]] bool operator==(const PatternData &other) const override { return areCommonPropertiesEqual(other); } }; class PatternDataBoolean : public PatternData { public: explicit PatternDataBoolean(u64 offset, Evaluator *evaluator, u32 color = 0) : PatternData(offset, 1, evaluator, 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", false); else if (boolean == 1) this->createDefaultEntry("true", true); else this->createDefaultEntry("true*", true); } [[nodiscard]] std::string getFormattedName() const override { return "bool"; } [[nodiscard]] bool operator==(const PatternData &other) const override { return areCommonPropertiesEqual(other); } }; class PatternDataCharacter : public PatternData { public: explicit PatternDataCharacter(u64 offset, Evaluator *evaluator, u32 color = 0) : PatternData(offset, 1, evaluator, 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("'{0}'", character), character); } [[nodiscard]] std::string getFormattedName() const override { return "char"; } [[nodiscard]] bool operator==(const PatternData &other) const override { return areCommonPropertiesEqual(other); } }; class PatternDataCharacter16 : public PatternData { public: explicit PatternDataCharacter16(u64 offset, Evaluator *evaluator, u32 color = 0) : PatternData(offset, 2, evaluator, color) { } PatternData* clone() override { return new PatternDataCharacter16(*this); } void createEntry(prv::Provider* &provider) override { char16_t character; provider->read(this->getOffset(), &character, 2); character = hex::changeEndianess(character, this->getEndian()); u128 literal = character; this->createDefaultEntry(hex::format("'{0}'", std::wstring_convert, char16_t>{}.to_bytes(character)), literal); } [[nodiscard]] std::string getFormattedName() const override { return "char16"; } [[nodiscard]] std::string toString(prv::Provider *provider) const override { char16_t character; provider->read(this->getOffset(), &character, 2); character = hex::changeEndianess(character, this->getEndian()); return std::wstring_convert, char16_t>{}.to_bytes(character); } [[nodiscard]] bool operator==(const PatternData &other) const override { return areCommonPropertiesEqual(other); } }; class PatternDataString : public PatternData { public: PatternDataString(u64 offset, size_t size, Evaluator *evaluator, u32 color = 0) : PatternData(offset, size, evaluator, color) { } PatternData* clone() override { return new PatternDataString(*this); } void createEntry(prv::Provider* &provider) override { auto size = std::min(this->getSize(), 0x7F); if (size == 0) return; std::string buffer(size, 0x00); provider->read(this->getOffset(), buffer.data(), size); std::erase_if(buffer, [](auto c){ return c == 0x00; }); this->createDefaultEntry(hex::format("\"{0}\" {1}", makeDisplayable(buffer.data(), this->getSize()), size > this->getSize() ? "(truncated)" : ""), buffer); } [[nodiscard]] std::string getFormattedName() const override { return "String"; } [[nodiscard]] std::string toString(prv::Provider *provider) const override { std::string buffer(this->getSize(), 0x00); provider->read(this->getOffset(), buffer.data(), buffer.size()); std::erase_if(buffer, [](auto c){ return c == 0x00; }); return buffer; } [[nodiscard]] bool operator==(const PatternData &other) const override { return areCommonPropertiesEqual(other); } }; class PatternDataString16 : public PatternData { public: PatternDataString16(u64 offset, size_t size, Evaluator *evaluator, u32 color = 0) : PatternData(offset, size, evaluator, color) { } PatternData* clone() override { return new PatternDataString16(*this); } void createEntry(prv::Provider* &provider) override { auto size = std::min(this->getSize(), 0x100); if (size == 0) return; std::u16string buffer(this->getSize()/sizeof(char16_t), 0x00); provider->read(this->getOffset(), buffer.data(), size); for (auto &c : buffer) c = hex::changeEndianess(c, 2, this->getEndian()); std::erase_if(buffer, [](auto c){ return c == 0x00; }); auto utf8String = std::wstring_convert, char16_t>{}.to_bytes(buffer); this->createDefaultEntry(hex::format("\"{0}\" {1}", utf8String, size > this->getSize() ? "(truncated)" : ""), utf8String); } [[nodiscard]] std::string getFormattedName() const override { return "String16"; } [[nodiscard]] std::string toString(prv::Provider *provider) const override { std::u16string buffer(this->getSize()/sizeof(char16_t), 0x00); provider->read(this->getOffset(), buffer.data(), this->getSize()); for (auto &c : buffer) c = hex::changeEndianess(c, 2, this->getEndian()); std::erase_if(buffer, [](auto c){ return c == 0x00; }); return std::wstring_convert, char16_t>{}.to_bytes(buffer); } [[nodiscard]] bool operator==(const PatternData &other) const override { return areCommonPropertiesEqual(other); } }; class PatternDataDynamicArray : public PatternData, public Inlinable { public: PatternDataDynamicArray(u64 offset, size_t size, Evaluator *evaluator, u32 color = 0) : PatternData(offset, size, evaluator, color) { } PatternDataDynamicArray(const PatternDataDynamicArray &other) : PatternData(other) { std::vector entries; for (const auto &entry : other.m_entries) entries.push_back(entry->clone()); this->setEntries(entries); } ~PatternDataDynamicArray() override { for (const auto &entry : this->m_entries) delete entry; } PatternData* clone() override { return new PatternDataDynamicArray(*this); } void setColor(u32 color) override { PatternData::setColor(color); for (auto &entry : this->m_entries) entry->setColor(color); } void createEntry(prv::Provider* &provider) override { if (this->m_entries.empty()) return; bool open = true; if (!this->isInlined()) { ImGui::TableNextRow(); ImGui::TableNextColumn(); open = ImGui::TreeNodeEx(this->getDisplayName().c_str(), ImGuiTreeNodeFlags_SpanFullWidth); this->drawCommentTooltip(); ImGui::TableNextColumn(); ImGui::ColorButton("color", ImColor(this->getColor()), ImGuiColorEditFlags_NoTooltip, ImVec2(ImGui::GetColumnWidth(), ImGui::GetTextLineHeight())); ImGui::TableNextColumn(); ImGui::TextFormatted("0x{0:08X} : 0x{1:08X}", this->getOffset(), this->getOffset() + this->getSize() - 1); ImGui::TableNextColumn(); ImGui::TextFormatted("0x{0:04X}", this->getSize()); ImGui::TableNextColumn(); ImGui::TextFormattedColored(ImColor(0xFF9BC64D), "{0}", this->m_entries[0]->getTypeName()); ImGui::SameLine(0, 0); ImGui::TextUnformatted("["); ImGui::SameLine(0, 0); ImGui::TextFormattedColored(ImColor(0xFF00FF00), "{0}", this->m_entries.size()); ImGui::SameLine(0, 0); ImGui::TextUnformatted("]"); ImGui::TableNextColumn(); ImGui::TextFormatted("{}", this->formatDisplayValue("{ ... }", this)); } if (open) { for (u64 i = 0; i < this->m_entries.size(); i++) { this->m_entries[i]->draw(provider); if (i >= (this->m_displayEnd - 1)) { ImGui::TableNextRow(); ImGui::TableNextColumn(); ImGui::Selectable("... (Double-click to see more items)", false, ImGuiSelectableFlags_SpanAllColumns); if (ImGui::IsItemHovered() && ImGui::IsMouseDoubleClicked(ImGuiMouseButton_Left)) this->m_displayEnd += 50; break; } } if (!this->isInlined()) ImGui::TreePop(); } else { this->m_displayEnd = 50; } } void getHighlightedAddresses(std::map &highlight) const override { for (auto &entry : this->m_entries) { entry->getHighlightedAddresses(highlight); } } [[nodiscard]] std::string getFormattedName() const override { return this->m_entries[0]->getTypeName() + "[" + std::to_string(this->m_entries.size()) + "]"; } void setOffset(u64 offset) override { for (auto &entry : this->m_entries) entry->setOffset(entry->getOffset() - this->getOffset() + offset); PatternData::setOffset(offset); } [[nodiscard]] const std::vector& getEntries() { return this->m_entries; } void setEntries(const std::vector &entries) { this->m_entries = entries; if (this->hasOverriddenColor()) { for (auto &entry : this->m_entries) { entry->setColor(this->getColor()); } } } [[nodiscard]] bool operator==(const PatternData &other) const override { if (!areCommonPropertiesEqual(other)) return false; auto &otherArray = *static_cast(&other); if (this->m_entries.size() != otherArray.m_entries.size()) return false; for (u64 i = 0; i < this->m_entries.size(); i++) { if (*this->m_entries[i] != *otherArray.m_entries[i]) return false; } return true; } [[nodiscard]] const PatternData* getPattern(u64 offset) const override { auto iter = std::find_if(this->m_entries.begin(), this->m_entries.end(), [this, offset](PatternData *pattern){ return offset >= pattern->getOffset() && offset < (pattern->getOffset() + pattern->getSize()); }); if (iter == this->m_entries.end()) return nullptr; else return *iter; } private: std::vector m_entries; u64 m_displayEnd = 50; }; class PatternDataStaticArray : public PatternData, public Inlinable { public: PatternDataStaticArray(u64 offset, size_t size, Evaluator *evaluator, u32 color = 0) : PatternData(offset, size, evaluator, color) { } PatternDataStaticArray(const PatternDataStaticArray &other) : PatternData(other) { this->setEntries(other.getTemplate()->clone(), other.getEntryCount()); } ~PatternDataStaticArray() override { delete this->m_template; delete this->m_highlightTemplate; } PatternData* clone() override { return new PatternDataStaticArray(*this); } void createEntry(prv::Provider* &provider) override { if (this->getEntryCount() == 0) return; bool open = true; if (!this->isInlined()) { ImGui::TableNextRow(); ImGui::TableNextColumn(); open = ImGui::TreeNodeEx(this->getDisplayName().c_str(), ImGuiTreeNodeFlags_SpanFullWidth); this->drawCommentTooltip(); ImGui::TableNextColumn(); ImGui::ColorButton("color", ImColor(this->getColor()), ImGuiColorEditFlags_NoTooltip, ImVec2(ImGui::GetColumnWidth(), ImGui::GetTextLineHeight())); ImGui::TableNextColumn(); ImGui::TextFormatted("0x{0:08X} : 0x{1:08X}", this->getOffset(), this->getOffset() + this->getSize() - 1); ImGui::TableNextColumn(); ImGui::TextFormatted("0x{0:04X}", this->getSize()); ImGui::TableNextColumn(); ImGui::TextFormattedColored(ImColor(0xFF9BC64D), "{0}", this->m_template->getTypeName().c_str()); ImGui::SameLine(0, 0); ImGui::TextUnformatted("["); ImGui::SameLine(0, 0); ImGui::TextFormattedColored(ImColor(0xFF00FF00), "{0}", this->m_entryCount); ImGui::SameLine(0, 0); ImGui::TextUnformatted("]"); ImGui::TableNextColumn(); ImGui::TextFormatted("{}", this->formatDisplayValue("{ ... }", this)); } if (open) { auto entry = this->m_template->clone(); for (u64 index = 0; index < this->m_entryCount; index++) { entry->setVariableName(hex::format("[{0}]", index)); entry->setOffset(this->getOffset() + index * this->m_template->getSize()); entry->draw(provider); if (index >= (this->m_displayEnd - 1)) { ImGui::TableNextRow(); ImGui::TableNextColumn(); ImGui::Selectable("... (Double-click to see more items)", false, ImGuiSelectableFlags_SpanAllColumns); if (ImGui::IsItemHovered() && ImGui::IsMouseDoubleClicked(ImGuiMouseButton_Left)) this->m_displayEnd += 50; break; } } delete entry; if (!this->isInlined()) ImGui::TreePop(); } else { this->m_displayEnd = 50; } } void getHighlightedAddresses(std::map &highlight) const override { auto entry = this->m_template->clone(); for (u64 address = this->getOffset(); address < this->getOffset() + this->getSize(); address += entry->getSize()) { entry->setOffset(address); entry->getHighlightedAddresses(highlight); } delete entry; } void setOffset(u64 offset) override { this->m_template->setOffset(this->m_template->getOffset() - this->getOffset() + offset); PatternData::setOffset(offset); } void setColor(u32 color) override { PatternData::setColor(color); this->m_template->setColor(color); } [[nodiscard]] std::string getFormattedName() const override { return this->m_template->getTypeName() + "[" + std::to_string(this->m_entryCount) + "]"; } [[nodiscard]] PatternData* getTemplate() const { return this->m_template; } [[nodiscard]] size_t getEntryCount() const { return this->m_entryCount; } void setEntryCount(size_t count) { this->m_entryCount = count; } void setEntries(PatternData* templ, size_t count) { this->m_template = templ; this->m_highlightTemplate = this->m_template->clone(); this->m_entryCount = count; if (this->hasOverriddenColor()) this->setColor(this->m_template->getColor()); this->m_template->setEndian(templ->getEndian()); } [[nodiscard]] bool operator==(const PatternData &other) const override { if (!areCommonPropertiesEqual(other)) return false; auto &otherArray = *static_cast(&other); return *this->m_template == *otherArray.m_template && this->m_entryCount == otherArray.m_entryCount; } [[nodiscard]] const PatternData* getPattern(u64 offset) const override { if (offset >= this->getOffset() && offset < (this->getOffset() + this->getSize())) { this->m_highlightTemplate->setOffset((offset / this->m_highlightTemplate->getSize()) * this->m_highlightTemplate->getSize()); return this->m_highlightTemplate; } else { return nullptr; } } private: PatternData *m_template; mutable PatternData *m_highlightTemplate; size_t m_entryCount; u64 m_displayEnd = 50; }; class PatternDataStruct : public PatternData, public Inlinable { public: PatternDataStruct(u64 offset, size_t size, Evaluator *evaluator, u32 color = 0) : PatternData(offset, size, evaluator, color){ } PatternDataStruct(const PatternDataStruct &other) : PatternData(other) { for (const auto &member : other.m_members) this->m_members.push_back(member->clone()); this->m_sortedMembers = this->m_members; } ~PatternDataStruct() override { for (const auto &member : this->m_members) delete member; } PatternData* clone() override { return new PatternDataStruct(*this); } void createEntry(prv::Provider* &provider) override { bool open = true; if (!this->isInlined()) { ImGui::TableNextRow(); ImGui::TableNextColumn(); open = ImGui::TreeNodeEx(this->getDisplayName().c_str(), ImGuiTreeNodeFlags_SpanFullWidth); this->drawCommentTooltip(); ImGui::TableNextColumn(); ImGui::TableNextColumn(); ImGui::TextFormatted("0x{0:08X} : 0x{1:08X}", this->getOffset(), this->getOffset() + this->getSize() - (this->getSize() == 0 ? 0 : 1)); ImGui::TableNextColumn(); ImGui::TextFormatted("0x{0:04X}", this->getSize()); ImGui::TableNextColumn(); ImGui::TextFormattedColored(ImColor(0xFFD69C56), "struct"); ImGui::SameLine(); ImGui::TextUnformatted(this->getTypeName().c_str()); ImGui::TableNextColumn(); ImGui::TextFormatted("{}", this->formatDisplayValue("{ ... }", this)); } if (open) { for (auto &member : this->m_sortedMembers) member->draw(provider); if (!this->isInlined()) ImGui::TreePop(); } } void getHighlightedAddresses(std::map &highlight) const override { for (auto &member : this->m_members) { member->getHighlightedAddresses(highlight); } } void setOffset(u64 offset) override { for (auto &member : this->m_members) member->setOffset(member->getOffset() - this->getOffset() + offset); PatternData::setOffset(offset); } void setColor(u32 color) override { PatternData::setColor(color); for (auto &member : this->m_members) member->setColor(color); } 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(); } [[nodiscard]] const auto& getMembers() const { return this->m_members; } void setMembers(const std::vector & members) { this->m_members.clear(); for (auto &member : members) { if (member == nullptr) continue; this->m_members.push_back(member); } this->m_sortedMembers = this->m_members; } [[nodiscard]] bool operator==(const PatternData &other) const override { if (!areCommonPropertiesEqual(other)) return false; auto &otherStruct = *static_cast(&other); if (this->m_members.size() != otherStruct.m_members.size()) return false; for (u64 i = 0; i < this->m_members.size(); i++) { if (*this->m_members[i] != *otherStruct.m_members[i]) return false; } return true; } [[nodiscard]] const PatternData* getPattern(u64 offset) const override { auto iter = std::find_if(this->m_members.begin(), this->m_members.end(), [offset](PatternData *pattern){ return offset >= pattern->getOffset() && offset < (pattern->getOffset() + pattern->getSize()); }); if (iter == this->m_members.end()) return nullptr; else return *iter; } private: std::vector m_members; std::vector m_sortedMembers; }; class PatternDataUnion : public PatternData, public Inlinable { public: PatternDataUnion(u64 offset, size_t size, Evaluator *evaluator, u32 color = 0) : PatternData(offset, size, evaluator, color) { } PatternDataUnion(const PatternDataUnion &other) : PatternData(other) { for (const auto &member : other.m_members) this->m_members.push_back(member->clone()); this->m_sortedMembers = this->m_members; } ~PatternDataUnion() override { for (const auto &member : this->m_members) delete member; } PatternData* clone() override { return new PatternDataUnion(*this); } void createEntry(prv::Provider* &provider) override { bool open = true; if (!this->isInlined()) { ImGui::TableNextRow(); ImGui::TableNextColumn(); open = ImGui::TreeNodeEx(this->getDisplayName().c_str(), ImGuiTreeNodeFlags_SpanFullWidth); this->drawCommentTooltip(); ImGui::TableNextColumn(); ImGui::TableNextColumn(); ImGui::TextFormatted("0x{0:08X} : 0x{1:08X}", this->getOffset(), std::max(this->getOffset() + this->getSize() - (this->getSize() == 0 ? 0 : 1), u64(0))); ImGui::TableNextColumn(); ImGui::TextFormatted("0x{0:04X}", this->getSize()); ImGui::TableNextColumn(); ImGui::TextFormattedColored(ImColor(0xFFD69C56), "union"); ImGui::SameLine(); ImGui::TextUnformatted(PatternData::getTypeName().c_str()); ImGui::TableNextColumn(); ImGui::TextFormatted("{}", this->formatDisplayValue("{ ... }", this)); } if (open) { for (auto &member : this->m_sortedMembers) member->draw(provider); if (!this->isInlined()) ImGui::TreePop(); } } void getHighlightedAddresses(std::map &highlight) const override { for (auto &member : this->m_members) { member->getHighlightedAddresses(highlight); } } void setOffset(u64 offset) override { for (auto &member : this->m_members) member->setOffset(member->getOffset() - this->getOffset() + offset); PatternData::setOffset(offset); } void setColor(u32 color) override { PatternData::setColor(color); for (auto &member : this->m_members) member->setColor(color); } 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();; } [[nodiscard]] const auto& getMembers() const { return this->m_members; } void setMembers(const std::vector & members) { this->m_members.clear(); for (auto &member : members) { if (member == nullptr) continue; this->m_members.push_back(member); } this->m_sortedMembers = this->m_members; } [[nodiscard]] bool operator==(const PatternData &other) const override { if (!areCommonPropertiesEqual(other)) return false; auto &otherUnion = *static_cast(&other); if (this->m_members.size() != otherUnion.m_members.size()) return false; for (u64 i = 0; i < this->m_members.size(); i++) { if (*this->m_members[i] != *otherUnion.m_members[i]) return false; } return true; } [[nodiscard]] const PatternData* getPattern(u64 offset) const override { auto largestMember = std::find_if(this->m_members.begin(), this->m_members.end(), [this](PatternData *pattern) { return pattern->getSize() == this->getSize(); }); if (largestMember == this->m_members.end()) return nullptr; else return *largestMember; } private: std::vector m_members; std::vector m_sortedMembers; }; class PatternDataEnum : public PatternData { public: PatternDataEnum(u64 offset, size_t size, Evaluator *evaluator, u32 color = 0) : PatternData(offset, size, evaluator, color) { } 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(overloaded { [&, name = entryName](auto &&entryValue) { if (value == entryValue) { valueString += name; foundValue = true; return true; } return false; }, [](std::string) { return false; }, [](PatternData*) { return false; } }, entryValueLiteral); if (matches) break; } if (!foundValue) valueString += "???"; ImGui::TableNextRow(); ImGui::TreeNodeEx(this->getDisplayName().c_str(), ImGuiTreeNodeFlags_Leaf | ImGuiTreeNodeFlags_NoTreePushOnOpen | ImGuiTreeNodeFlags_SpanFullWidth | ImGuiTreeNodeFlags_AllowItemOverlap); this->drawCommentTooltip(); ImGui::TableNextColumn(); if (ImGui::Selectable(("##PatternDataLine"s + std::to_string(this->getOffset())).c_str(), false, ImGuiSelectableFlags_SpanAllColumns)) { EventManager::post(Region { this->getOffset(), this->getSize() }); } ImGui::SameLine(); ImGui::TextUnformatted(this->getDisplayName().c_str()); ImGui::TableNextColumn(); ImGui::ColorButton("color", ImColor(this->getColor()), ImGuiColorEditFlags_NoTooltip, ImVec2(ImGui::GetColumnWidth(), ImGui::GetTextLineHeight())); ImGui::TableNextColumn(); ImGui::TextFormatted("0x{0:08X} : 0x{1:08X}", this->getOffset(), this->getOffset() + this->getSize() - 1); ImGui::TableNextColumn(); ImGui::TextFormatted("0x{0:04X}", this->getSize()); ImGui::TableNextColumn(); ImGui::TextFormattedColored(ImColor(0xFFD69C56), "enum"); ImGui::SameLine(); ImGui::TextUnformatted(PatternData::getTypeName().c_str()); ImGui::TableNextColumn(); ImGui::TextFormatted("{}", this->formatDisplayValue(hex::format("{} (0x{:0{}X})", valueString.c_str(), value, this->getSize() * 2), this)); } [[nodiscard]] std::string getFormattedName() const override { return "enum " + PatternData::getTypeName(); } [[nodiscard]] const auto& getEnumValues() const { return this->m_enumValues; } void setEnumValues(const std::vector> &enumValues) { this->m_enumValues = enumValues; } [[nodiscard]] bool operator==(const PatternData &other) const override { if (!areCommonPropertiesEqual(other)) return false; auto &otherEnum = *static_cast(&other); if (this->m_enumValues.size() != otherEnum.m_enumValues.size()) return false; for (u64 i = 0; i < this->m_enumValues.size(); i++) { if (this->m_enumValues[i] != otherEnum.m_enumValues[i]) return false; } return true; } private: std::vector> m_enumValues; }; class PatternDataBitfieldField : public PatternData { public: PatternDataBitfieldField(u64 offset, u8 bitOffset, u8 bitSize, PatternData *bitField, Evaluator *evaluator, u32 color = 0) : m_bitOffset(bitOffset), m_bitSize(bitSize), m_bitField(bitField), PatternData(offset, 0, evaluator, color) { } PatternData* clone() override { return new PatternDataBitfieldField(*this); } void createEntry(prv::Provider* &provider) override { std::vector value(this->m_bitField->getSize(), 0); provider->read(this->m_bitField->getOffset(), &value[0], value.size()); if (this->m_bitField->getEndian() == std::endian::little) std::reverse(value.begin(), value.end()); ImGui::TableNextRow(); ImGui::TableNextColumn(); ImGui::TextUnformatted(this->getDisplayName().c_str()); ImGui::TableNextColumn(); ImGui::ColorButton("color", ImColor(this->getColor()), ImGuiColorEditFlags_NoTooltip, ImVec2(ImGui::GetColumnWidth(), ImGui::GetTextLineHeight())); ImGui::TableNextColumn(); if (this->m_bitSize == 1) ImGui::TextFormatted("0x{0:08X} bit {1}", this->getOffset() + this->m_bitOffset / 8, (this->m_bitOffset + (this->m_bitSize - 1)) % 8); else ImGui::TextFormatted("0x{0:08X} bits {1} - {2}", this->getOffset() + this->m_bitOffset / 8, this->m_bitOffset % 8, this->m_bitOffset % 8 + (this->m_bitSize - 1) % 8); ImGui::TableNextColumn(); if (this->m_bitSize == 1) ImGui::TextFormatted("{0} bit", this->m_bitSize); else ImGui::TextFormatted("{0} bits", this->m_bitSize); ImGui::TableNextColumn(); ImGui::TextFormattedColored(ImColor(0xFF9BC64D), "bits"); ImGui::TableNextColumn(); { u8 numBytes = (this->m_bitSize / 8) + 1; u64 extractedValue = hex::extract(this->m_bitOffset + (this->m_bitSize - 1), this->m_bitOffset, value); ImGui::TextFormatted("{}", this->formatDisplayValue(hex::format("{0} (0x{1:X})", extractedValue, extractedValue), this)); } } [[nodiscard]] std::string getFormattedName() const override { return "bits"; } [[nodiscard]] u8 getBitOffset() const { return this->m_bitOffset; } [[nodiscard]] u8 getBitSize() const { return this->m_bitSize; } [[nodiscard]] bool operator==(const PatternData &other) const override { if (!areCommonPropertiesEqual(other)) return false; auto &otherBitfieldField = *static_cast(&other); return this->m_bitOffset == otherBitfieldField.m_bitOffset && this->m_bitSize == otherBitfieldField.m_bitSize; } private: u8 m_bitOffset, m_bitSize; PatternData *m_bitField; }; class PatternDataBitfield : public PatternData, public Inlinable { public: PatternDataBitfield(u64 offset, size_t size, Evaluator *evaluator, u32 color = 0) : PatternData(offset, size, evaluator, color) { } PatternDataBitfield(const PatternDataBitfield &other) : PatternData(other) { for (auto &field : other.m_fields) this->m_fields.push_back(field->clone()); } ~PatternDataBitfield() override { for (auto field : this->m_fields) delete field; } PatternData* clone() override { return new PatternDataBitfield(*this); } void createEntry(prv::Provider* &provider) override { std::vector value(this->getSize(), 0); provider->read(this->getOffset(), &value[0], value.size()); if (this->m_endian == std::endian::little) std::reverse(value.begin(), value.end()); bool open = true; if (!this->isInlined()) { ImGui::TableNextRow(); ImGui::TableNextColumn(); open = ImGui::TreeNodeEx(this->getDisplayName().c_str(), ImGuiTreeNodeFlags_SpanFullWidth); this->drawCommentTooltip(); ImGui::TableNextColumn(); ImGui::TableNextColumn(); ImGui::TextFormatted("0x{0:08X} : 0x{1:08X}", this->getOffset(), this->getOffset() + this->getSize() - 1); ImGui::TableNextColumn(); ImGui::TextFormatted("0x{0:04X}", this->getSize()); ImGui::TableNextColumn(); ImGui::TextFormattedColored(ImColor(0xFFD69C56), "bitfield"); ImGui::SameLine(); ImGui::TextUnformatted(PatternData::getTypeName().c_str()); ImGui::TableNextColumn(); std::string valueString = "{ "; for (auto i : value) valueString += hex::format("{0:02X} ", i); valueString += "}"; ImGui::TextFormatted("{}", this->formatDisplayValue(valueString, this)); } if (open) { for (auto &field : this->m_fields) field->draw(provider); if (!this->isInlined()) ImGui::TreePop(); } } void setOffset(u64 offset) override { for (auto &field : this->m_fields) field->setOffset(field->getOffset() - this->getOffset() + offset); PatternData::setOffset(offset); } [[nodiscard]] std::string getFormattedName() const override { return "bitfield " + PatternData::getTypeName(); } [[nodiscard]] const auto& getFields() const { return this->m_fields; } void setFields(const std::vector &fields) { this->m_fields = fields; for (auto &field : this->m_fields) { field->setSize(this->getSize()); field->setColor(this->getColor()); } } void setColor(u32 color) override { PatternData::setColor(color); for (auto &field : this->m_fields) field->setColor(color); } [[nodiscard]] bool operator==(const PatternData &other) const override { if (!areCommonPropertiesEqual(other)) return false; auto &otherBitfield = *static_cast(&other); if (this->m_fields.size() != otherBitfield.m_fields.size()) return false; for (u64 i = 0; i < this->m_fields.size(); i++) { if (*this->m_fields[i] != *otherBitfield.m_fields[i]) return false; } return true; } private: std::vector m_fields; }; }