From d5a9f2bc82d4fb4f57aa25200541e2d0e0788877 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?L=C3=A9o=20Lam?= <leo@leolam.fr>
Date: Fri, 1 Jan 2021 18:58:24 +0100
Subject: [PATCH] identify_matching_functions_by_call: Add support for matching
 one fn

---
 tools/identify_matching_functions_by_call.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/tools/identify_matching_functions_by_call.py b/tools/identify_matching_functions_by_call.py
index 337dee16..8496cc06 100755
--- a/tools/identify_matching_functions_by_call.py
+++ b/tools/identify_matching_functions_by_call.py
@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 from typing import Dict, List
+import argparse
 
 import cxxfilt
 from colorama import Fore
@@ -34,9 +35,18 @@ class Checker(checker.FunctionChecker):
 
 
 def main() -> None:
+    parser = argparse.ArgumentParser("Identifies matching functions by looking at function calls in matching functions")
+    parser.add_argument("-f", "--fn", help="Functions to analyze", nargs="*")
+    args = parser.parse_args()
+
+    functions_to_analyze = set(args.fn) if args.fn else set()
+
     functions_by_addr: Dict[int, utils.FunctionInfo] = {fn.addr: fn for fn in utils.get_functions()}
     fn_checker = Checker()
     for fn in functions_by_addr.values():
+        if functions_to_analyze and fn.decomp_name not in functions_to_analyze:
+            continue
+
         if fn.status != utils.FunctionStatus.Matching:
             continue