[WARP] Update python example to fix adding a source to a "read-only" container

2 files changed, 14 insertions, 3 deletions

diff --git a/plugins/warp/api/python/warp.py b/plugins/warp/api/python/warp.py
index 866d179b..fc487a4d 100644
--- a/plugins/warp/api/python/warp.py
+++ b/plugins/warp/api/python/warp.py
@@ -350,6 +350,13 @@ class WarpContainer(metaclass=_WarpContainerMetaclass):
         warpcore.BNWARPFreeContainerList(containers, count.value)
         return result
 
+    @staticmethod
+    def by_name(name: str) -> Optional['WarpContainer']:
+        for container in WarpContainer:
+            if container.name == name:
+                return container
+        return None
+
     @property
     def name(self) -> str:
         return warpcore.BNWARPContainerGetName(self.handle)
diff --git a/plugins/warp/examples/create_signatures.py b/plugins/warp/examples/create_signatures.py
index 6fbcae54..82ed724c 100755
--- a/plugins/warp/examples/create_signatures.py
+++ b/plugins/warp/examples/create_signatures.py
@@ -14,12 +14,15 @@ def process_binary(input_file: str, output_dir: str) -> None:
     if not bv:
         return
 
-    # Sources exist only in containers, so we will just pull off the first available container.
-    # In the future we might make container construction available to the API.
-    container = WarpContainer.all()[0]
+    # For the sake of this example we are going to assume the container "User"
+    # is available, in the future we might want to make containers constructable
+    container = WarpContainer.by_name("User")
     output_file = output_dir / f"{input_path.stem}_analysis.warp"
     # Add the source so we can add functions to it and then commit it (write to disk)
     source = container.add_source(str(output_file))
+    if source is None:
+        print(f"Failed to create source {str(output_file)}")
+        return
 
     # NOTE: You probably want to pull the platform from the function, but for this example it's fine.
     target = WarpTarget(bv.platform)
@@ -30,6 +33,7 @@ def process_binary(input_file: str, output_dir: str) -> None:
     # Actually write the warp file to disk.
     container.commit_source(source)
     bv.file.close()
+    print(f"committed {len(functions_to_warp)} functions to {output_file}")
 
 if __name__ == "__main__":
     if len(sys.argv) != 3: