mirror of https://github.com/pwndbg/pwndbg.git
Use gdb/lldb from pypi (#3119)
* add lldb/gdb pypi * add gdb * debug * debug * fix paths * fix lldb * add lldb-server * bump lock * fix nix * fix lint * rename gdbinit * fix pwndbg nix * revert missing files * fix docs * fix docs * remove .skip-venv, not needed anymore * cleanup version comment * fix lint script * lint * fix nix develop * fix docs * fix docs script * use 'pwndbg' binary for tests * lint * fix kernel tests * fix ubuntu22.04 * bump lldb * refactor gdbinit/lldbinit * test1 * fix logger * fix tests no-homepull/3137/head
patryk4815
6 months ago
committed by
GitHub
parent
22e26203a5
commit
df12edc0d5
@ -0,0 +1,90 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import cProfile
|
||||||
|
import logging
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
import time
|
||||||
|
import traceback
|
||||||
|
|
||||||
|
import gdb
|
||||||
|
|
||||||
|
from pwndbginit import gdbpatches # noqa: F401
|
||||||
|
from pwndbginit.common import verify_venv
|
||||||
|
|
||||||
|
|
||||||
|
def init_logger():
|
||||||
|
log_level_env = os.environ.get("PWNDBG_LOGLEVEL", "WARNING")
|
||||||
|
log_level = getattr(logging, log_level_env.upper())
|
||||||
|
|
||||||
|
root_logger = logging.getLogger()
|
||||||
|
root_logger.setLevel(log_level)
|
||||||
|
|
||||||
|
# Add a custom StreamHandler we will use to customize log message formatting. We
|
||||||
|
# configure the handler later, after pwndbg has been imported.
|
||||||
|
handler = logging.StreamHandler()
|
||||||
|
root_logger.addHandler(handler)
|
||||||
|
|
||||||
|
return handler
|
||||||
|
|
||||||
|
|
||||||
|
def check_doubleload():
|
||||||
|
if "pwndbg" in sys.modules:
|
||||||
|
print(
|
||||||
|
"Detected double-loading of Pwndbg (likely from both .gdbinit and the Pwndbg portable build)."
|
||||||
|
)
|
||||||
|
print(
|
||||||
|
"To fix this, please remove the line 'source your-path/gdbinit.py' from your .gdbinit file."
|
||||||
|
)
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
|
||||||
|
def main() -> None:
|
||||||
|
handler = init_logger()
|
||||||
|
profiler = cProfile.Profile()
|
||||||
|
|
||||||
|
start_time = None
|
||||||
|
if os.environ.get("PWNDBG_PROFILE") == "1":
|
||||||
|
start_time = time.time()
|
||||||
|
profiler.enable()
|
||||||
|
|
||||||
|
check_doubleload()
|
||||||
|
verify_venv()
|
||||||
|
|
||||||
|
# Force UTF-8 encoding (to_string=True to skip output appearing to the user)
|
||||||
|
try:
|
||||||
|
gdb.execute("set target-wide-charset UTF-8", to_string=True)
|
||||||
|
gdb.execute("set charset UTF-8", to_string=True)
|
||||||
|
except gdb.error as e:
|
||||||
|
print(f"Warning: Cannot set gdb charset: '{e}'")
|
||||||
|
|
||||||
|
import pwndbg # noqa: F811
|
||||||
|
import pwndbg.dbg.gdb
|
||||||
|
|
||||||
|
pwndbg.dbg = pwndbg.dbg_mod.gdb.GDB()
|
||||||
|
pwndbg.dbg.setup()
|
||||||
|
|
||||||
|
import pwndbg.log
|
||||||
|
import pwndbg.profiling
|
||||||
|
|
||||||
|
# ColorFormatter relies on pwndbg being loaded, so we can't set it up until now
|
||||||
|
handler.setFormatter(pwndbg.log.ColorFormatter())
|
||||||
|
|
||||||
|
pwndbg.profiling.init(profiler, start_time)
|
||||||
|
if os.environ.get("PWNDBG_PROFILE") == "1":
|
||||||
|
pwndbg.profiling.profiler.stop("pwndbg-load.pstats")
|
||||||
|
pwndbg.profiling.profiler.start()
|
||||||
|
|
||||||
|
# We need reimport it here so that it's available at the global scope
|
||||||
|
# when some starts a Python interpreter in GDB
|
||||||
|
gdb.execute("py import pwndbg")
|
||||||
|
|
||||||
|
|
||||||
|
def main_try():
|
||||||
|
# We wrap everything in try/except so that we can exit GDB with an error code
|
||||||
|
# This is used by tests to check if gdbinit.py failed
|
||||||
|
try:
|
||||||
|
main()
|
||||||
|
except Exception:
|
||||||
|
print(traceback.format_exc(), file=sys.stderr, flush=True)
|
||||||
|
os._exit(1)
|
||||||
@ -0,0 +1,52 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import importlib.abc
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
import traceback
|
||||||
|
|
||||||
|
import gdb
|
||||||
|
|
||||||
|
|
||||||
|
def fix_exit():
|
||||||
|
major_ver = int(gdb.VERSION.split(".")[0])
|
||||||
|
if major_ver <= 15:
|
||||||
|
# On certain verions of gdb (used on ubuntu 24.04) using sys.exit() can cause
|
||||||
|
# a segfault. See:
|
||||||
|
# https://github.com/pwndbg/pwndbg/pull/2900#issuecomment-2825456636
|
||||||
|
# https://sourceware.org/bugzilla/show_bug.cgi?id=31946
|
||||||
|
def _patched_exit(exit_code):
|
||||||
|
# argparse requires a SystemExit exception, otherwise our CLI commands will exit incorrectly on invalid arguments
|
||||||
|
stack_list = traceback.extract_stack(limit=2)
|
||||||
|
if len(stack_list) == 2:
|
||||||
|
p = stack_list[0]
|
||||||
|
if p.filename.endswith("/argparse.py"):
|
||||||
|
raise SystemExit()
|
||||||
|
|
||||||
|
sys.stdout.flush()
|
||||||
|
sys.stderr.flush()
|
||||||
|
os._exit(exit_code)
|
||||||
|
|
||||||
|
sys.exit = _patched_exit
|
||||||
|
|
||||||
|
|
||||||
|
def fix_stdout():
|
||||||
|
# Add the original stdout methods back to gdb._GdbOutputFile for pwnlib colors
|
||||||
|
sys.stdout.isatty = sys.__stdout__.isatty
|
||||||
|
sys.stdout.fileno = sys.__stdout__.fileno
|
||||||
|
|
||||||
|
|
||||||
|
def fix_readline():
|
||||||
|
# Fix gdb readline bug: https://github.com/pwndbg/pwndbg/issues/2232#issuecomment-2542564965
|
||||||
|
class GdbRemoveReadlineFinder(importlib.abc.MetaPathFinder):
|
||||||
|
def find_spec(self, fullname, path=None, target=None):
|
||||||
|
if fullname == "readline":
|
||||||
|
raise ImportError("readline module disabled under GDB")
|
||||||
|
return None
|
||||||
|
|
||||||
|
sys.meta_path.insert(0, GdbRemoveReadlineFinder())
|
||||||
|
|
||||||
|
|
||||||
|
fix_stdout()
|
||||||
|
fix_readline()
|
||||||
|
fix_exit()
|
||||||
@ -0,0 +1,40 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import cProfile
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
import time
|
||||||
|
|
||||||
|
import lldb
|
||||||
|
|
||||||
|
from pwndbginit.common import verify_venv
|
||||||
|
|
||||||
|
|
||||||
|
def main(debugger: lldb.SBDebugger, major: int, minor: int, debug: bool = False) -> None:
|
||||||
|
if "pwndbg" in sys.modules:
|
||||||
|
print("Detected double-loading of Pwndbg.")
|
||||||
|
print("This should not happen. Please report this issue if you're not sure how to fix it.")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
verify_venv()
|
||||||
|
profiler = cProfile.Profile()
|
||||||
|
|
||||||
|
start_time = None
|
||||||
|
if os.environ.get("PWNDBG_PROFILE") == "1":
|
||||||
|
start_time = time.time()
|
||||||
|
profiler.enable()
|
||||||
|
|
||||||
|
import pwndbg # noqa: F811
|
||||||
|
import pwndbg.dbg.lldb
|
||||||
|
|
||||||
|
pwndbg.dbg_mod.lldb.LLDB_VERSION = (major, minor)
|
||||||
|
|
||||||
|
pwndbg.dbg = pwndbg.dbg_mod.lldb.LLDB()
|
||||||
|
pwndbg.dbg.setup(debugger, "pwndbglldbhandler", debug=debug)
|
||||||
|
|
||||||
|
import pwndbg.profiling
|
||||||
|
|
||||||
|
pwndbg.profiling.init(profiler, start_time)
|
||||||
|
if os.environ.get("PWNDBG_PROFILE") == "1":
|
||||||
|
pwndbg.profiling.profiler.stop("pwndbg-load.pstats")
|
||||||
|
pwndbg.profiling.profiler.start()
|
||||||
@ -0,0 +1,86 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import os
|
||||||
|
import shutil
|
||||||
|
import subprocess
|
||||||
|
import sys
|
||||||
|
import sysconfig
|
||||||
|
from typing import Tuple
|
||||||
|
|
||||||
|
|
||||||
|
def get_gdb_version(path: str) -> Tuple[str, str]:
|
||||||
|
result = subprocess.run(
|
||||||
|
[
|
||||||
|
path,
|
||||||
|
"-nx",
|
||||||
|
"--batch",
|
||||||
|
"-iex",
|
||||||
|
"py import sysconfig; print(sysconfig.get_config_var('INSTSONAME'), sysconfig.get_config_var('VERSION'))",
|
||||||
|
],
|
||||||
|
capture_output=True,
|
||||||
|
text=True,
|
||||||
|
)
|
||||||
|
return tuple(result.stdout.strip().split(" ", 2))
|
||||||
|
|
||||||
|
|
||||||
|
def get_venv_bin_path():
|
||||||
|
bin_dir = "Scripts" if os.name == "nt" else "bin"
|
||||||
|
return os.path.join(sys.prefix, bin_dir)
|
||||||
|
|
||||||
|
|
||||||
|
def prepend_venv_bin_to_path():
|
||||||
|
# Set virtualenv's bin path (needed for utility tools like ropper, pwntools etc)
|
||||||
|
venv_bin = get_venv_bin_path()
|
||||||
|
path_elements = os.environ.get("PATH", "").split(os.pathsep)
|
||||||
|
if venv_bin in path_elements:
|
||||||
|
return
|
||||||
|
|
||||||
|
path_elements.insert(0, venv_bin)
|
||||||
|
os.environ["PATH"] = os.pathsep.join(path_elements)
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
prepend_venv_bin_to_path()
|
||||||
|
|
||||||
|
gdb_argv = [
|
||||||
|
sys.argv[0],
|
||||||
|
"-q",
|
||||||
|
"-nx",
|
||||||
|
"-iex",
|
||||||
|
"py import pwndbginit.gdbinit; pwndbginit.gdbinit.main_try()",
|
||||||
|
*sys.argv[1:],
|
||||||
|
]
|
||||||
|
sys.argv = gdb_argv
|
||||||
|
|
||||||
|
try:
|
||||||
|
from gdb_for_pwndbg.gdb import main
|
||||||
|
|
||||||
|
main()
|
||||||
|
return
|
||||||
|
except ImportError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
gdb_path = shutil.which("gdb")
|
||||||
|
if not gdb_path:
|
||||||
|
print("ERROR: Could not find 'gdb' binary")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
envs = os.environ.copy()
|
||||||
|
envs["PYTHONNOUSERSITE"] = "1"
|
||||||
|
envs["PYTHONPATH"] = ":".join(sys.path)
|
||||||
|
envs["PYTHONHOME"] = f"{sys.prefix}:{sys.exec_prefix}"
|
||||||
|
|
||||||
|
expected = (sysconfig.get_config_var("INSTSONAME"), sysconfig.get_config_var("VERSION"))
|
||||||
|
have = get_gdb_version(gdb_path)
|
||||||
|
if have != expected:
|
||||||
|
print(
|
||||||
|
f"ERROR: GDB is compiled for Python {have}, but your Python interpreter is {expected}"
|
||||||
|
)
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
os.execve(gdb_path, sys.argv, env=envs)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
@ -0,0 +1 @@
|
|||||||
|
# This file is used just to handle lldb-commands, it must be empty
|
||||||
Loading…
Reference in new issue