1 files changed, 16 insertions, 0 deletions
diff --git a/plugins/workflow_objc/MessageHandler.h b/plugins/workflow_objc/MessageHandler.h
new file mode 100644
index 00000000..8826b094
--- /dev/null
+++ b/plugins/workflow_objc/MessageHandler.h
@@ -0,0 +1,16 @@
+#pragma once
+
+#include <binaryninjaapi.h>
+
+class MessageHandler {
+
+    std::set<uint64_t> m_msgSendFunctions;
+    static std::set<uint64_t> findMsgSendFunctions(BinaryNinja::Ref<BinaryNinja::BinaryView> data);
+
+public:
+    MessageHandler(BinaryNinja::Ref<BinaryNinja::BinaryView> data);
+
+    std::set<uint64_t> getMessageSendFunctions() const { return m_msgSendFunctions; }
+    bool hasMessageSendFunctions() const { return m_msgSendFunctions.size() != 0; }
+    bool isMessageSend(uint64_t);
+};