Add bs_func to Decompilation class and edit hooks

libbs/artifacts/decompilation.py

@@ -9,6 +9,7 @@ class Decompilation(Artifact):
         "text",
         "line_map",
         "decompiler",
+        "bs_func",
     )
 
     def __init__(
@@ -17,13 +18,15 @@ def __init__(
         text: str = None,
         line_map: dict = None,
         decompiler: str = None,
+        bs_func = None,
         **kwargs
     ):
         super().__init__(**kwargs)
         self.addr = addr
         self.text = text
         self.line_map = line_map or {}
         self.decompiler = decompiler
+        self.bs_func = bs_func
 
     def __str__(self):
         return f"//ADDR: {hex(self.addr)}\n// SOURCE: {self.decompiler}\n{self.text}"

libbs/decompilers/ida/hooks.py

@@ -552,7 +552,8 @@ def _send_decompilation_event(self, cfunc):
         dec = Decompilation(
             addr=cfunc.entry_ea,
             text=str(cfunc),
-            decompiler="ida"
+            decompiler="ida",
+            bs_func=function
         )
         self.interface.decompilation_changed(dec, function=function, func_addr=lifted_addr)