Add support for MultiRegister (PR1733)

dyninst · Jul 15, 2024 · 70fd6bc · 70fd6bc
1 parent 0cd46f5
commit 70fd6bc
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/instructionAPI/statefull_visitor.cpp b/instructionAPI/statefull_visitor.cpp
@@ -6,6 +6,7 @@
 #include "InstructionDecoder.h"
 #include "Operand.h"
 #include "Register.h"
+#include "MultiRegister.h"
 #include "Visitor.h"
 
 #include <array>
@@ -25,6 +26,7 @@ struct stateful_visitor : di::Visitor {
   bool foundImm{};
   bool foundBin{};
   bool foundDer{};
+  bool foundMultiReg{};
 
   void visit(di::BinaryFunction*) override { foundBin = true; }
 
@@ -34,13 +36,17 @@ struct stateful_visitor : di::Visitor {
 
   void visit(di::Dereference*) override { foundDer = true; }
 
+  void visit(di::MultiRegisterAST*) override { foundMultiReg = true; }
+
+
   // clang-format off
   friend std::ostream& operator<<(std::ostream &os, stateful_visitor const& v) {
     std::cout << std::boolalpha
               << "  foundReg: " << v.foundReg << '\n'
               << "  foundImm: " << v.foundImm << '\n'
               << "  foundBin: " << v.foundBin << '\n'
-              << "  foundDer: " << v.foundDer << '\n';
+              << "  foundDer: " << v.foundDer << '\n'
+              << "  foundMultiReg: " << v.foundMultiReg << '\n';
     return os;
   }
 

diff --git a/instructionAPI/stateless_visitor.cpp b/instructionAPI/stateless_visitor.cpp
@@ -6,6 +6,7 @@
 #include "InstructionDecoder.h"
 #include "Operand.h"
 #include "Register.h"
+#include "MultiRegister.h"
 #include "Visitor.h"
 
 #include <array>
@@ -38,6 +39,11 @@ class printer : public di::Visitor {
   void visit(di::RegisterAST* r) override {
     std::cout << "  Register '" << r->format(di::defaultStyle) << "'\n";
   }
+
+  void visit(di::MultiRegisterAST* r) override {
+    std::cout << "  MultiRegister '" << r->format(di::defaultStyle) << "'\n";
+  }
+
 };
 
 void print(di::Instruction const& insn) {