Base ARM64 Implementation

pinwhell · Jun 23, 2022 · ece4373 · ece4373
1 parent 00bc253
commit ece4373
Show file tree

Hide file tree

Showing 9 changed files with 75 additions and 47 deletions.
diff --git a/.gitignore b/.gitignore
@@ -395,4 +395,6 @@ FodyWeavers.xsd
 *.msp
 
 # JetBrains Rider
-*.sln.iml
+*.sln.iml
+
+DevEnv/
diff --git a/LittleXrefs/src/ILXTool.cpp b/LittleXrefs/src/ILXTool.cpp
@@ -1,6 +1,6 @@
 #include "ILXTool.h"
 #include "Function.h"
-#include "ReferenceEngine.h"
+#include "IReferenceEngine.h"
 #include "FunctionReferenceList.h"
 #include <iostream>
 #include <conio.h>
@@ -110,6 +110,12 @@ void ILXTool::Run()
 
         FindReferences(typeName, offset, funcRefsLists);
         funcRefsLists.Print();
-        system("pause");
+        _getch();
     }
 }
+
+LXARM64Tool::LXARM64Tool(LittleXrefs* pLXRefs)
+    : ILXTool(pLXRefs, CS_ARCH_ARM64, CS_MODE_ARM)
+{
+    RefsEngine = new Arm64ReferenceEngine(GetCapstoneHandle());
+}
diff --git a/LittleXrefs/src/ILXTool.h b/LittleXrefs/src/ILXTool.h
@@ -9,7 +9,7 @@ using namespace LX;
 
 struct FunctionReferenceList;
 struct Function;
-struct ReferenceEngine;
+struct IReferenceEngine;
 
 class ILXTool
 {
@@ -19,7 +19,7 @@ class ILXTool
 	std::vector<Function*> allFunctions;
 	void ParseAllFunction();
 protected:
-	ReferenceEngine* RefsEngine;
+	IReferenceEngine* RefsEngine;
 public:
 	ILXTool(LittleXrefs* _pLXRefs, cs_arch arch, cs_mode archMode);
 	~ILXTool();
@@ -38,3 +38,10 @@ class LXARMTool : public ILXTool {
 	~LXARMTool();
 };
 
+class LXARM64Tool : public ILXTool {
+public:
+	LXARM64Tool(LittleXrefs* pLXRefs);
+	~LXARM64Tool();
+};
+
+
diff --git a/LittleXrefs/src/ReferenceEngine.cpp → LittleXrefs/src/IReferenceEngine.cpp b/LittleXrefs/src/ReferenceEngine.cpp → LittleXrefs/src/IReferenceEngine.cpp
@@ -1,19 +1,15 @@
-#include "ReferenceEngine.h"
+#include "IReferenceEngine.h"
 #include "FunctionReferenceList.h"
 #include "Function.h"
 #include "CapstoneHelper.h"
 
-ReferenceEngine::ReferenceEngine(csh _CapstoneDisasm)
+IReferenceEngine::IReferenceEngine(csh _CapstoneDisasm, uint16_t _abiRegBase)
 	: CapstoneHandle(_CapstoneDisasm)
+	, abiRegBase(_abiRegBase)
 {
 }
 
-ArmReferenceEngine::ArmReferenceEngine(csh CapstoneDisasm)
-	:ReferenceEngine(CapstoneDisasm)
-{
-}
-
-void ArmReferenceEngine::FindRefereces(Function* pFunc, const std::string& typeName, uint64_t offset, FunctionReferenceList& outRefsLists)
+void IReferenceEngine::FindRefereces(Function* pFunc, const std::string& typeName, uint64_t offset, FunctionReferenceList& outRefsLists)
 {
 	std::vector<uint16_t> matchingTypenameParamsId;
 
@@ -27,13 +23,19 @@ void ArmReferenceEngine::FindRefereces(Function* pFunc, const std::string& typeN
 			cs_insn* pDisasmdInstEnd = pDisasmdInst + count;
 
 			for (uint16_t currParamId : matchingTypenameParamsId)
-				FindRefereces(pFunc, pDisasmdInst, pDisasmdInstEnd, (uint16_t)(arm_reg::ARM_REG_R0 + currParamId), offset, outRefsLists);
+				FindRefereces(pFunc, pDisasmdInst, pDisasmdInstEnd, (uint16_t)(abiRegBase + currParamId), offset, outRefsLists);
 
 			cs_free(pDisasmdInst, count);
 		}
 	}
 }
 
+ArmReferenceEngine::ArmReferenceEngine(csh CapstoneDisasm)
+	: IReferenceEngine(CapstoneDisasm, ARM_REG_R0)
+{
+}
+//arm_reg::ARM_REG_R0
+
 void ArmReferenceEngine::FindRefereces(Function* pFunc, cs_insn* pStart, cs_insn* pEnd, uint16_t trackReg, uint64_t offset, FunctionReferenceList& outRefsLists)
 {
 	for (cs_insn* pCurrInst = pStart; pCurrInst < pEnd; pCurrInst++)
@@ -100,9 +102,13 @@ void ArmReferenceEngine::FindRefereces(Function* pFunc, cs_insn* pStart, cs_insn
 	return;
 }
 
-void ArmReferenceEngine::FindRefereces(unsigned char* pEntry, uint16_t trackReg, uint64_t offset, FunctionReferenceList& outRefsLists)
+Arm64ReferenceEngine::Arm64ReferenceEngine(csh CapstoneDisasm)
+	: IReferenceEngine(CapstoneDisasm, ARM64_REG_X0)
+{
+
+}
+
+void Arm64ReferenceEngine::FindRefereces(Function* pFunc, cs_insn* pStart, cs_insn* pEnd, uint16_t trackReg, uint64_t offset, FunctionReferenceList& outRefsLists)
 {
-
-
-
+
 }
diff --git a/LittleXrefs/src/IReferenceEngine.h b/LittleXrefs/src/IReferenceEngine.h
@@ -0,0 +1,31 @@
+#pragma once
+
+#include <capstone/capstone.h>
+#include <string>
+
+struct Function;
+struct FunctionReferenceList;
+
+struct IReferenceEngine
+{
+	csh CapstoneHandle;
+	uint16_t abiRegBase;
+
+	IReferenceEngine(csh CapstoneDisasm, uint16_t abiRegBase);
+
+	virtual void FindRefereces(Function* pFunc, cs_insn* pStart, cs_insn* pEnd, uint16_t trackReg, uint64_t offset, FunctionReferenceList& outRefsLists) = 0;
+	void FindRefereces(Function* pFunc, const std::string& typeName, uint64_t offset, FunctionReferenceList& outRefsLists);
+};
+
+struct ArmReferenceEngine : IReferenceEngine {
+	ArmReferenceEngine(csh CapstoneDisasm);
+
+	void FindRefereces(Function* pFunc, cs_insn* pStart, cs_insn* pEnd, uint16_t trackReg, uint64_t offset, FunctionReferenceList& outRefsLists) override;
+};
+
+struct Arm64ReferenceEngine : IReferenceEngine {
+	Arm64ReferenceEngine(csh CapstoneDisasm);
+
+	void FindRefereces(Function* pFunc, cs_insn* pStart, cs_insn* pEnd, uint16_t trackReg, uint64_t offset, FunctionReferenceList& outRefsLists) override;
+};
+
diff --git a/LittleXrefs/src/LittleXrefs.vcxproj b/LittleXrefs/src/LittleXrefs.vcxproj
@@ -27,7 +27,7 @@
     <ClInclude Include="LittleXrefs.h" />
     <ClInclude Include="Parameter.h" />
     <ClInclude Include="Reference.h" />
-    <ClInclude Include="ReferenceEngine.h" />
+    <ClInclude Include="IReferenceEngine.h" />
   </ItemGroup>
   <ItemGroup>
     <ClCompile Include="CapstoneHelper.cpp" />
@@ -39,7 +39,7 @@
     <ClCompile Include="main.cpp" />
     <ClCompile Include="Parameter.cpp" />
     <ClCompile Include="Reference.cpp" />
-    <ClCompile Include="ReferenceEngine.cpp" />
+    <ClCompile Include="IReferenceEngine.cpp" />
   </ItemGroup>
   <PropertyGroup Label="Globals">
     <VCProjectVersion>15.0</VCProjectVersion>

diff --git a/LittleXrefs/src/LittleXrefs.vcxproj.filters b/LittleXrefs/src/LittleXrefs.vcxproj.filters
@@ -36,7 +36,7 @@
     <ClInclude Include="Parameter.h">
       <Filter>Header Files</Filter>
     </ClInclude>
-    <ClInclude Include="ReferenceEngine.h">
+    <ClInclude Include="IReferenceEngine.h">
       <Filter>Header Files</Filter>
     </ClInclude>
     <ClInclude Include="CapstoneHelper.h">
@@ -68,7 +68,7 @@
     <ClCompile Include="Parameter.cpp">
       <Filter>Source Files</Filter>
     </ClCompile>
-    <ClCompile Include="ReferenceEngine.cpp">
+    <ClCompile Include="IReferenceEngine.cpp">
       <Filter>Source Files</Filter>
     </ClCompile>
     <ClCompile Include="CapstoneHelper.cpp">

diff --git a/LittleXrefs/src/ReferenceEngine.h b/LittleXrefs/src/ReferenceEngine.h
diff --git a/LittleXrefs/src/main.cpp b/LittleXrefs/src/main.cpp
@@ -9,7 +9,7 @@ int main()
 		exit(1);
 	}
 
-	ILXTool* pIlxTool = new LXARMTool(LX::g_pLXrefs);
+	ILXTool* pIlxTool = new LXARM64Tool(LX::g_pLXrefs);
 
 	pIlxTool->Run();
 }