mirror of https://github.com/pwndbg/pwndbg.git
This reverts commit a97aff4f1c.
pull/3482/merge
k4lizen
1 day ago
committed by
GitHub
parent
98cd6d9add
commit
dddb82b075
@ -1,24 +0,0 @@
|
||||
<!-- THIS PART OF THIS FILE IS AUTOGENERATED. DO NOT MODIFY IT. See scripts/generate-docs.sh -->
|
||||
# vmlinux
|
||||
|
||||
```text
|
||||
usage: vmlinux [-h] [-t TOOL] filepath
|
||||
|
||||
```
|
||||
|
||||
Load bzImage or vmlinux file to add kernel debug symbols.
|
||||
### Positional arguments
|
||||
|
||||
|Positional Argument|Help|
|
||||
| :--- | :--- |
|
||||
|filepath|Path to the bzImage or vmlinux file to load|
|
||||
|
||||
### Optional arguments
|
||||
|
||||
|Short|Long|Help|
|
||||
| :--- | :--- | :--- |
|
||||
|-h|--help|show this help message and exit|
|
||||
|-t|--tool|Path to the vmlinux-to-elf tool (if not in PATH)|
|
||||
|
||||
<!-- END OF AUTOGENERATED PART. Do not modify this line or the line below, they mark the end of the auto-generated part of the file. If you want to extend the documentation in a way which cannot easily be done by adding to the command help description, write below the following line. -->
|
||||
<!-- ------------\>8---- ----\>8---- ----\>8------------ -->
|
||||
@ -1,124 +0,0 @@
|
||||
"""
|
||||
Loads a bzImage or vmlinux file to add kernel debug symbols. This command uses vmlinux-to-elf
|
||||
to extract the ELF file from a bzImage and then loads it with the kernel base address.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import os
|
||||
import shutil
|
||||
import subprocess
|
||||
import tempfile
|
||||
|
||||
import pwndbg.aglib.kernel
|
||||
import pwndbg.color.message as M
|
||||
import pwndbg.commands
|
||||
import pwndbg.dbg
|
||||
from pwndbg.commands import CommandCategory
|
||||
|
||||
parser = argparse.ArgumentParser(
|
||||
description="Load bzImage or vmlinux file to add kernel debug symbols."
|
||||
)
|
||||
parser.add_argument(
|
||||
"filepath",
|
||||
type=str,
|
||||
help="Path to the bzImage or vmlinux file to load",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-t",
|
||||
"--tool",
|
||||
type=str,
|
||||
help="Path to the vmlinux-to-elf tool (if not in PATH)",
|
||||
)
|
||||
|
||||
|
||||
@pwndbg.commands.Command(parser, category=CommandCategory.KERNEL)
|
||||
@pwndbg.commands.OnlyWhenQemuKernel
|
||||
@pwndbg.commands.OnlyWhenPagingEnabled
|
||||
def vmlinux(filepath: str, tool: str = None) -> None:
|
||||
# Verify the input file exists
|
||||
if not os.path.isfile(filepath):
|
||||
print(M.error(f"File not found: {filepath}"))
|
||||
return
|
||||
|
||||
# Get the kernel base address
|
||||
base = pwndbg.aglib.kernel.arch_paginginfo().kbase
|
||||
if base is None:
|
||||
print(M.error("Unable to locate the kernel base address"))
|
||||
return
|
||||
|
||||
print(M.success(f"Found kernel base address: {hex(base)}"))
|
||||
|
||||
# Find vmlinux-to-elf tool
|
||||
if tool:
|
||||
# User specified the tool path
|
||||
vmlinux_tool = tool
|
||||
if not os.path.isfile(vmlinux_tool):
|
||||
print(M.error(f"Specified tool not found: {vmlinux_tool}"))
|
||||
return
|
||||
if not os.access(vmlinux_tool, os.X_OK):
|
||||
print(M.error(f"Specified tool is not executable: {vmlinux_tool}"))
|
||||
return
|
||||
else:
|
||||
# GDB may not inherit the full PATH, so check common locations
|
||||
vmlinux_tool = shutil.which("vmlinux-to-elf")
|
||||
if not vmlinux_tool:
|
||||
# Check common user install locations
|
||||
common_paths = [
|
||||
os.path.expanduser("~/.local/bin/vmlinux-to-elf"),
|
||||
"/usr/local/bin/vmlinux-to-elf",
|
||||
"/usr/bin/vmlinux-to-elf",
|
||||
]
|
||||
for path in common_paths:
|
||||
if os.path.isfile(path) and os.access(path, os.X_OK):
|
||||
vmlinux_tool = path
|
||||
break
|
||||
|
||||
if not vmlinux_tool:
|
||||
print(
|
||||
M.error(
|
||||
"vmlinux-to-elf tool not found in PATH or common locations.\n"
|
||||
"Please install it or ensure it's in: ~/.local/bin, /usr/local/bin, or /usr/bin\n"
|
||||
"You can install it with: pip install --user vmlinux-to-elf\n"
|
||||
"Or specify the tool path with: vmlinux <filepath> --tool <path-to-vmlinux-to-elf>"
|
||||
)
|
||||
)
|
||||
return
|
||||
|
||||
# Create a temporary file for the extracted ELF
|
||||
with tempfile.NamedTemporaryFile(delete=False, suffix=".elf") as tmpfile:
|
||||
tmpfile_path = tmpfile.name
|
||||
|
||||
try:
|
||||
# Run vmlinux-to-elf to extract the ELF file
|
||||
print(f"Extracting ELF from {filepath} using vmlinux-to-elf...")
|
||||
result = subprocess.run(
|
||||
[vmlinux_tool, filepath, tmpfile_path],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
)
|
||||
|
||||
if result.returncode != 0:
|
||||
print(M.error(f"vmlinux-to-elf failed with error:\n{result.stderr}"))
|
||||
return
|
||||
|
||||
print(M.success(f"Successfully extracted ELF to {tmpfile_path}"))
|
||||
|
||||
# Add the symbol file with the kernel base address
|
||||
print(f"Loading symbols at address {hex(base)}...")
|
||||
pwndbg.dbg.selected_inferior().add_symbol_file(tmpfile_path, base)
|
||||
print(M.success(f"Loaded kernel symbols from {filepath} successfully"))
|
||||
|
||||
except FileNotFoundError:
|
||||
print(
|
||||
M.error(
|
||||
"vmlinux-to-elf tool not found. Please ensure it is installed and in your PATH."
|
||||
)
|
||||
)
|
||||
except Exception as e:
|
||||
print(M.error(f"Error loading symbols: {str(e)}"))
|
||||
finally:
|
||||
# Note: We don't delete the temp file here because GDB needs it to remain accessible
|
||||
# for as long as the debugging session is active
|
||||
pass
|
||||
Loading…
Reference in new issue