diff --git a/binja_script.py b/binja_script.py index 3f65e7483..71ac2cd52 100644 --- a/binja_script.py +++ b/binja_script.py @@ -1,6 +1,7 @@ from __future__ import annotations import ctypes +import os import sys import threading import xmlrpc.client @@ -20,8 +21,8 @@ xmlrpc.client.MAXINT = 10**100 xmlrpc.client.MININT = -(10**100) -host = "127.0.0.1" -port = 31337 +host = os.environ.get("PWNDBG_BINJA_SERVER_HOST", "127.0.0.1") +port = int(os.environ.get("PWNDBG_BINJA_SERVER_PORT", "31337")) logger = binaryninja.log.Logger(0, "pwndbg-integration") diff --git a/ida_script.py b/ida_script.py index 6fd940dfb..1c3263647 100644 --- a/ida_script.py +++ b/ida_script.py @@ -2,6 +2,7 @@ from __future__ import annotations import datetime +import os import threading import xmlrpc.client as xmlclient from xml.sax.saxutils import escape @@ -48,8 +49,8 @@ xmlclient.Marshaller.dispatch[type(1 << 63)] = create_marshaller("%d< xmlclient.Marshaller.dispatch[int] = create_marshaller("%d") xmlclient.Marshaller.dispatch[idaapi.cfuncptr_t] = create_marshaller(just_to_str=True) -host = "127.0.0.1" -port = 31337 +host = os.environ.get("PWNDBG_IDA_SERVER_HOST", "127.0.0.1") +port = int(os.environ.get("PWNDBG_IDA_SERVER_PORT", "31337")) mutex = threading.Condition()