1 files changed, 8 insertions, 1 deletions
diff --git a/src/processor/exploitability_linux.cc b/src/processor/exploitability_linux.cc
index c11aed4c..4f98dfeb 100644
--- a/src/processor/exploitability_linux.cc
+++ b/src/processor/exploitability_linux.cc
@@ -106,8 +106,15 @@ ExploitabilityRating ExploitabilityLinux::CheckPlatformExploitability() {
     case MD_CONTEXT_AMD64:
       instruction_ptr = context->GetContextAMD64()->rip;
       break;
+    case MD_CONTEXT_ARM:
+      instruction_ptr =
+          context->GetContextARM()->iregs[MD_CONTEXT_ARM_REG_PC];
+      break;
+    case MD_CONTEXT_ARM64:
+      instruction_ptr =
+          context->GetContextARM64()->iregs[MD_CONTEXT_ARM64_REG_PC];
+      break;
     default:
-      // TODO(liuandrew): Add support ARM and arm64 architectures.
       BPLOG(INFO) << "Unsupported architecture.";
       return EXPLOITABILITY_ERR_PROCESSING;
   }