fix: deadstore FP where use happens after first return statement

amarna/rules/local_rules/DeadStoreRule.py

@@ -37,6 +37,7 @@ def code_element_function(self, tree: Tree) -> None:
                 implicits_and_arguments.append(str(args.children[0]))
 
         defines: Set[Token] = set()
+        in_return = []
         for main_child in tree.children:
             if main_child.data != "code_block":
                 continue
@@ -62,18 +63,18 @@ def code_element_function(self, tree: Tree) -> None:
 
                     # in a return statement, check which variables were not used
                     for subcode in child.children[0].find_data("code_element_return"):
-                        tokens = [
-                            str(tok) for tok in subcode.scan_values(lambda v: isinstance(v, Token))
-                        ]
-                        for dead_store in sorted(defines):
-                            if dead_store in tokens or dead_store in implicits_and_arguments:
-                                continue
-                            sarif = create_result_token(
-                                self.fname,
-                                self.RULE_NAME,
-                                self.RULE_TEXT,
-                                dead_store,
-                            )
-                            if sarif in self.results:
-                                continue
-                            self.results.append(sarif)
+                        for tok in subcode.scan_values(lambda v: isinstance(v, Token)):
+                            in_return.append(str(tok))
+
+        for dead_store in sorted(defines):
+            if dead_store in in_return or dead_store in implicits_and_arguments:
+                continue
+            sarif = create_result_token(
+                self.fname,
+                self.RULE_NAME,
+                self.RULE_TEXT,
+                dead_store,
+            )
+            if sarif in self.results:
+                continue
+            self.results.append(sarif)

tests/dead_store3.cairo

@@ -0,0 +1,9 @@
+func are_equal(x, y) -> (eq):
+    let (a) = 44
+    if x == y:
+        return (FALSE)
+    end
+    if x == a:
+        return (TRUE)
+    end
+end

tests/expected/dead_store3.expected

@@ -0,0 +1 @@
+[unused-function] in dead_store3.cairo:1:1