From 20adc82b8f85a78fe2cd6ddc2b3a8c78558999a5 Mon Sep 17 00:00:00 2001
From: Mark Rowe <mark@vector35.com>
Date: Mon, 23 Mar 2026 19:49:56 -0700
Subject: [ObjC] Set return type on objc_msgSend calls that send an init
 message to a known class

---
 plugins/workflow_objc/src/activities/super_init.rs | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

(limited to 'plugins/workflow_objc/src/activities/super_init.rs')

diff --git a/plugins/workflow_objc/src/activities/super_init.rs b/plugins/workflow_objc/src/activities/super_init.rs
index 81b9a453..7b6bb3f7 100644
--- a/plugins/workflow_objc/src/activities/super_init.rs
+++ b/plugins/workflow_objc/src/activities/super_init.rs
@@ -39,8 +39,7 @@ fn return_type_for_super_init(call: &util::Call, view: &BinaryView) -> Option<Re
         util::match_constant_pointer_or_load_of_constant_pointer(&call.call.params[1])?;
     let selector = Selector::from_address(view, selector_addr).ok()?;
 
-    // TODO: This will match `initialize` and `initiate` which are not init methods.
-    if !selector.name.starts_with("init") {
+    if !selector.is_init_family() {
         return None;
     }
 
-- 
cgit v1.3.1