2025-06-26 00:30:28 +08:00
|
|
|
#!/bin/bash
|
2025-07-03 22:59:24 +08:00
|
|
|
# Wrapper script to run pcileech-generate with sudo while preserving Python path
|
|
|
|
|
#
|
|
|
|
|
# generate.py is the correct entry point that orchestrates:
|
|
|
|
|
# • Device enumeration and selection
|
|
|
|
|
# • Driver rebinding to vfio-pci
|
|
|
|
|
# • Container execution with build.py
|
|
|
|
|
# • Optional firmware flashing
|
2025-06-26 00:30:28 +08:00
|
|
|
|
2025-07-03 22:59:24 +08:00
|
|
|
# Get the path to the installed pcileech-generate script
|
|
|
|
|
PCILEECH_GENERATE_PATH=$(which pcileech-generate)
|
2025-06-26 00:30:28 +08:00
|
|
|
|
2025-07-03 22:59:24 +08:00
|
|
|
if [ -z "$PCILEECH_GENERATE_PATH" ]; then
|
|
|
|
|
echo "Error: pcileech-generate not found in PATH"
|
|
|
|
|
echo "Make sure pcileechfwgenerator is properly installed: pip install pcileechfwgenerator"
|
2025-06-26 00:30:28 +08:00
|
|
|
exit 1
|
|
|
|
|
fi
|
|
|
|
|
|
|
|
|
|
# Get Python site-packages directories
|
|
|
|
|
PYTHON_USER_SITE=$(python3 -m site --user-site)
|
|
|
|
|
PYTHON_SITE_PACKAGES=$(python3 -c "import site; print(':'.join(site.getsitepackages()))")
|
|
|
|
|
|
|
|
|
|
# Run with sudo, preserving the PYTHONPATH
|
2025-07-03 22:59:24 +08:00
|
|
|
echo "Running pcileech-generate with sudo and preserved Python paths..."
|
|
|
|
|
sudo PYTHONPATH="$PYTHONPATH:$PYTHON_USER_SITE:$PYTHON_SITE_PACKAGES" "$PCILEECH_GENERATE_PATH" "$@"
|
