cfg_worker.py

import xbin

"""
xbin Standalone CFG Plugin Example
==================================

This is a truly standalone worker that can live in its own repository.
"""

@xbin.plugin(
    name="external_cfg", 
    category="cfg_generation", 
    version="1.0"
)
class ExternalCFGWorker:
    def on_new_binary(self, path, analyses):
        # Category-aware: only run if relevant
        if "cfg_generation" not in analyses:
            return

        print(f"[*] Generating CFG for: {path}")
        
        # Mock CFG data
        cfg_data = {
            "nodes": [
                {"id": "0x401000", "label": "main"},
                {"id": "0x401010", "label": "init_session"}
            ],
            "edges": [
                {"source": "0x401000", "target": "0x401010"}
            ]
        }
        
        # Use the clean module-level API
        xbin.post_result(
            item_key="0x401000",
            data=cfg_data,
            confidence=0.85
        )
        print("[+] CFG posted.")

if __name__ == "__main__":
    xbin.start_worker()