mirror of https://github.com/snare/voltron.git
139 lines
4.3 KiB
Bash
Executable File
139 lines
4.3 KiB
Bash
Executable File
#!/bin/bash
|
|
#
|
|
# Install Voltron for whichever debuggers are detected (only GDB and LLDB so
|
|
# far).
|
|
#
|
|
# Adapted from pwndbg's install script.
|
|
#
|
|
# Usage: ./install.sh [ -u -d ]
|
|
# -u Install to user's site-packages directory
|
|
# -d Install in developer mode (-e flag passed to pip)
|
|
#
|
|
SUDO='sudo'
|
|
GDB=$(command -v gdb)
|
|
LLDB=$(command -v lldb)
|
|
|
|
set -ex
|
|
|
|
while getopts ":ud" opt; do
|
|
case $opt in
|
|
u)
|
|
USER_MODE='--user'
|
|
SUDO=''
|
|
;;
|
|
d)
|
|
DEV_MODE="-e"
|
|
;;
|
|
\?)
|
|
echo "Invalid option: -$OPTARG" >&2
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
function install_apt {
|
|
if uname | grep -i Linux &>/dev/null; then
|
|
sudo apt-get update
|
|
if echo $PYVER|grep "3\."; then
|
|
sudo apt-get -y install libreadline6-dev python3-dev python3-setuptools python3-yaml
|
|
else
|
|
sudo apt-get -y install libreadline6-dev python-dev python-setuptools python-yaml
|
|
fi
|
|
fi
|
|
}
|
|
|
|
if [ -n "${GDB}" ]; then
|
|
# Find the Python version used by GDB
|
|
PYVER=$(gdb -batch -q --nx -ex 'pi import platform; print(".".join(platform.python_version_tuple()[:2]))')
|
|
PYTHON=$(gdb -batch -q --nx -ex 'pi import sys; print(sys.executable)')
|
|
PYTHON="${PYTHON}${PYVER}"
|
|
|
|
install_apt
|
|
|
|
if [ -z $USER_MODE ]; then
|
|
GDB_SITE_PACKAGES=$(gdb -batch -q --nx -ex 'pi import site; print(site.getsitepackages()[0])')
|
|
else
|
|
GDB_SITE_PACKAGES=$(gdb -batch -q --nx -ex 'pi import site; print(site.getusersitepackages())')
|
|
fi
|
|
|
|
# Install Voltron and dependencies
|
|
${SUDO} ${PYTHON} -m pip install $USER_MODE $DEV_MODE -U .
|
|
|
|
# Add Voltron to gdbinit
|
|
if ! grep voltron ~/.gdbinit &>/dev/null; then
|
|
echo "source $GDB_SITE_PACKAGES/voltron/entry.py" >> ~/.gdbinit
|
|
fi
|
|
fi
|
|
|
|
if [ -n "${LLDB}" ]; then
|
|
# Find the Python version used by LLDB
|
|
PYVER=$(lldb -Qxb --one-line 'script import platform; print(".".join(platform.python_version_tuple()[:2]))'|tail -1)
|
|
PYTHON=$(lldb -Qxb --one-line 'script import sys; print(sys.executable)'|tail -1)
|
|
PYTHON="${PYTHON}${PYVER}"
|
|
if [ -z $USER_MODE ]; then
|
|
LLDB_SITE_PACKAGES=$(lldb -Qxb --one-line 'script import site; print(site.getsitepackages()[0])'|tail -1)
|
|
else
|
|
LLDB_SITE_PACKAGES=$(lldb -Qxb --one-line 'script import site; print(site.getusersitepackages())'|tail -1)
|
|
fi
|
|
|
|
install_apt
|
|
|
|
if [ "$LLDB_SITE_PACKAGES" == "$GDB_SITE_PACKAGES" ]; then
|
|
echo "Skipping installation for LLDB - same site-packages directory"
|
|
else
|
|
# Install Voltron and dependencies
|
|
${SUDO} ${PYTHON} -m pip install $USER_MODE $DEV_MODE -U .
|
|
fi
|
|
|
|
# Add Voltron to lldbinit
|
|
if ! grep voltron ~/.lldbinit &>/dev/null; then
|
|
INIT_FILE='~/.lldbinit'
|
|
echo "command script import $LLDB_SITE_PACKAGES/voltron/entry.py" >> ~/.lldbinit
|
|
fi
|
|
fi
|
|
|
|
if [ -z "${GDB}" ] && [ -z "${LLDB}" ]; then
|
|
# Find system Python
|
|
PYTHON=$(command -v python)
|
|
PYVER=$(${PYTHON} -c 'import platform; print(".".join(platform.python_version_tuple()[:2]))')
|
|
if [ -z $USER_MODE ]; then
|
|
PYTHON_SITE_PACKAGES=$(${PYTHON} -c 'import site; print(site.getsitepackages()[0])')
|
|
else
|
|
PYTHON_SITE_PACKAGES=$(${PYTHON} -c 'import site; print(site.getusersitepackages())')
|
|
fi
|
|
|
|
install_apt
|
|
|
|
# Install Voltron and dependencies
|
|
${SUDO} ${PYTHON} -m pip install $USER_MODE $DEV_MODE -U .
|
|
fi
|
|
|
|
set +x
|
|
echo "=============================================================="
|
|
if [ -n "${GDB}" ]; then
|
|
echo "Installed for GDB (${GDB}):"
|
|
echo " Python: $PYTHON"
|
|
echo " Packages directory: $GDB_SITE_PACKAGES"
|
|
if [ -n "${INIT_FILE}" ]; then
|
|
echo " Added voltron to: ~/.gdbinit"
|
|
else
|
|
echo " Already loaded in: ~/.gdbinit"
|
|
fi
|
|
fi
|
|
if [ -n "${LLDB}" ]; then
|
|
echo "Installed for LLDB (${LLDB}):"
|
|
echo " Python: $PYTHON"
|
|
echo " Packages directory: $LLDB_SITE_PACKAGES"
|
|
if [ -n "${INIT_FILE}" ]; then
|
|
echo " Added voltron to: ~/.lldbinit"
|
|
else
|
|
echo " Already loaded in: ~/.lldbinit"
|
|
fi
|
|
fi
|
|
if [ -z "${GDB}" ] && [ -z "${LLDB}" ]; then
|
|
echo "Couldn't find any debuggers. Installed using the Python in your path:"
|
|
echo " Python: $PYTHON"
|
|
echo " Packages directory: $PYTHON_SITE_PACKAGES"
|
|
echo " Did not add Voltron to any debugger init files."
|
|
fi
|