1.4.5 added OpenAttackDefenseTools/ataka template

Author: gfelber

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "vagd"
-version = "1.4.4"
+version = "1.4.5"
 authors = [{ name = "0x6fe1be2" }]
 description = "VirtuAlization GDb integrations in pwntools"
 readme = "README.md"

src/vagd/cli.py

@@ -119,6 +119,9 @@ def template(
   local: Optional[bool] = typer.Option(
     False, "--local", help="create local template"
   ),
+  ataka: Optional[bool] = typer.Option(
+    False, "--ataka", help="create an ataka compatible template"
+  ),
 ):
   """
   creates a template
@@ -189,16 +192,20 @@ def template(
 
     template = "".join(templateChunks).format(
       "{}",
+      dependencies=", ".join(dependencies),
       aliases=aliases,
       binary=binary,
       ip=ip,
       port=str(port),
       env=env,
+      ataka_env="# ataka envs\nIP = os.getenv('TARGET_IP', IP)\nEXTRA = json.loads(os.getenv('TARGET_EXTRA', '[]'))"
+      if ataka
+      else "",
+      vms=("\n" + " " * 4).join(vms),
       libc=libc,
       aslr=aslr,
       is_local=True if local else "args.LOCAL",
-      dependencies=", ".join(dependencies),
-      vms=("\n" + " " * 4).join(vms),
+      is_ataka=" or os.getenv('TARGET_IP')" if ataka else "",
     )
 
     if output_exploit:

src/vagd/res/template.txt

@@ -15,11 +15,13 @@ c"""
 context.binary = exe = ELF(BINARY, checksec=False)
 context.aslr = {aslr}
 
+{ataka_env}
+
 {aliases}
 
 def setup():
   global vm
-  if args.REMOTE or {is_local}:
+  if args.REMOTE or {is_local}{is_ataka}:
     return
 
   try:
@@ -34,7 +36,7 @@ def setup():
 
 vm = None
 def get_target(**kw) -> tubes.tube:
-  if args.REMOTE:
+  if args.REMOTE{is_ataka}:
     # context.log_level = 'debug'
     return remote(IP, PORT)