voltron/install.sh at master · attackgithub/voltron · GitHub

History

executable file

255 lines (220 loc) · 7.58 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

220

221

222

223

224

225

226

227

228

229

230

231

232

233

234

235

236

237

238

239

240

241

242

243

244

245

246

247

248

249

250

251

252

253

254

255

#!/bin/bash

# Install Voltron for whichever debuggers are detected (only GDB and LLDB so

# far).

function usage {

cat <<END

Voltron installer script.

This script will attempt to find GDB/LLDB, infer the correct Python to use, and

install Voltron. By default it will attempt to detect a single version of each

GDB and LLDB, and will install into the user's site-packages directory. The

options below can be used to change this behaviour.

Usage: ./install.sh [ -s -d -S ] [ -v virtualenv ] [ -b BACKEND ]

-s Install to system's site-packages directory

-d Install in developer mode (-e flag passed to pip)

-v venv Install into a virtualenv (only for LLDB)

-b debugger Select debugger backend ("", "gdb", "lldb", or "gdb,lldb") for

which to install

-S Skip package manager (apt/yum) update

END

exit 1

}

GDB=$(command -v gdb)

LLDB=$(command -v lldb)

APT_GET=$(command -v apt-get)

YUM_YUM=$(command -v yum)

YUM_DNF=$(command -v dnf)

# Default to --user install without sudo

USER_MODE='--user'

SUDO=''

[[ -z "${GDB}" ]]

BACKEND_GDB=$?

[[ -z "${LLDB}" ]]

BACKEND_LLDB=$?

if [ -z "${LLDB}" ]; then

for i in `seq 6 8`; do

LLDB=$(command -v lldb-3.$i)

if [ -n "${LLDB}" ]; then

break

done

while getopts ":dsSb:v:" opt; do

case $opt in

USER_MODE=''

SUDO=$(command -v sudo)

;;

DEV_MODE="-e"

;;

VENV="${OPTARG}"

USER_MODE=''

SUDO=''

;;

SKIP_UPDATE='-s'

;;

[[ ! "${OPTARG}" =~ "gdb" ]]

BACKEND_GDB=$?

[[ ! "${OPTARG}" =~ "lldb" ]]

BACKEND_LLDB=$?

;;

\?)

usage

;;

esac

done

if [ "${BACKEND_GDB}" -eq 1 ] && [ -z "${GDB}" ]; then

echo "Requested to install voltron for gdb, but gdb not present on the system"

exit 1

if [ "${BACKEND_LLDB}" -eq 1 ] && [ -z "${LLDB}" ]; then

echo "Requested to install voltron for lldb, but lldb not present on the system"

exit 1

set -ex

function install_apt {

if [ -n "${APT_GET}" ]; then

if [ -z "${SKIP_UPDATE}" ]; then

sudo apt-get update

if echo $PYVER|grep "3\."; then

sudo apt-get -y install libreadline6-dev python3-dev python3-setuptools python3-yaml python3-pip

else

sudo apt-get -y install libreadline6-dev python-dev python-setuptools python-yaml python-pip

}

function install_yum {

local CMD=""

if [ -n "${YUM_DNF}" ]; then

CMD=$YUM_DNF

else

if [ -n "${YUM_YUM}" ]; then

CMD=$YUM_YUM

if [ "${CMD}" != "" ]; then

local PARAMS="--assumeyes"

if [ -z "${SKIP_UPDATE}" ]; then

PARAMS="$PARAMS --refresh"

if echo $PYVER|grep "3\."; then

sudo $CMD $PARAMS install readline-devel python3-devel python3-setuptools python3-yaml python3-pip

else

sudo $CMD $PARAMS install readline-devel python-devel python-setuptools python-yaml python-pip

}

function install_packages {

install_apt

install_yum

}

if [ "${BACKEND_GDB}" -eq 1 ]; then

# Find the Python version used by GDB

GDB_PYVER=$(${GDB} -batch -q --nx -ex 'pi import platform; print(".".join(platform.python_version_tuple()[:2]))')

GDB_PYTHON=$(${GDB} -batch -q --nx -ex 'pi import sys; print(sys.executable)')

GDB_PYTHON="${GDB_PYTHON/%$GDB_PYVER/}${GDB_PYVER}"

install_packages

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())')

# Install Voltron and dependencies

${SUDO} ${GDB_PYTHON} -m pip install -U $USER_MODE $DEV_MODE .

# Add Voltron to gdbinit

GDB_INIT_FILE="${HOME}/.gdbinit"

if [ -e ${GDB_INIT_FILE} ]; then

sed -i.bak '/voltron/d' ${GDB_INIT_FILE}

if [ -z $DEV_MODE ]; then

GDB_ENTRY_FILE="$GDB_SITE_PACKAGES/voltron/entry.py"

else

GDB_ENTRY_FILE="$(pwd)/voltron/entry.py"

echo "source $GDB_ENTRY_FILE" >> ${GDB_INIT_FILE}

if [ "${BACKEND_LLDB}" -eq 1 ]; then

# Find the Python version used by LLDB

LLDB_PYVER=$(${LLDB} -Qxb --one-line 'script import platform; print(".".join(platform.python_version_tuple()[:2]))'|tail -1)

LLDB_PYTHON=$(${LLDB} -Qxb --one-line 'script import sys; print(sys.executable)'|tail -1)

LLDB_PYTHON="${LLDB_PYTHON/%$LLDB_PYVER/}${LLDB_PYVER}"

${LLDB_PYTHON} -m pip install --user --upgrade six

if [ -n "${VENV}" ]; then

echo "Creating virtualenv..."

${LLDB_PYTHON} -m pip install --user virtualenv

${LLDB_PYTHON} -m virtualenv "${VENV}"

LLDB_PYTHON="${VENV}/bin/python"

LLDB_SITE_PACKAGES=$(find "${VENV}" -name site-packages)

elif [ -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)

install_packages

if [ "$LLDB_SITE_PACKAGES" == "$GDB_SITE_PACKAGES" ]; then

echo "Skipping installation for LLDB - same site-packages directory"

else

# Install Voltron and dependencies

${SUDO} ${LLDB_PYTHON} -m pip install -U $USER_MODE $DEV_MODE .

# Add Voltron to lldbinit

LLDB_INIT_FILE="${HOME}/.lldbinit"

if [ -e ${LLDB_INIT_FILE} ]; then

sed -i.bak '/voltron/d' ${LLDB_INIT_FILE}

if [ -z "${DEV_MODE}" ]; then

LLDB_ENTRY_FILE="$LLDB_SITE_PACKAGES/voltron/entry.py"

else

LLDB_ENTRY_FILE="$(pwd)/voltron/entry.py"

if [ -n "${VENV}" ]; then

echo "script import sys;sys.path.append('${LLDB_SITE_PACKAGES}')" >> ${LLDB_INIT_FILE}

echo "command script import $LLDB_ENTRY_FILE" >> ${LLDB_INIT_FILE}

if [ "${BACKEND_GDB}" -ne 1 ] && [ "${BACKEND_LLDB}" -ne 1 ]; 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())')

install_packages

# Install Voltron and dependencies

${SUDO} ${PYTHON} -m pip install -U $USER_MODE $DEV_MODE .

set +x

echo "=============================================================="

if [ "${BACKEND_GDB}" -eq 1 ]; then

echo "Installed for GDB (${GDB}):"

echo " Python: $GDB_PYTHON"

echo " Packages directory: $GDB_SITE_PACKAGES"

echo " Added voltron to: $GDB_INIT_FILE"

echo " Entry point: $GDB_ENTRY_FILE"

if [ "${BACKEND_LLDB}" -eq 1 ]; then

echo "Installed for LLDB (${LLDB}):"

echo " Python: $LLDB_PYTHON"

echo " Packages directory: $LLDB_SITE_PACKAGES"

echo " Added voltron to: $LLDB_INIT_FILE"

echo " Entry point: $LLDB_ENTRY_FILE"

if [ "${BACKEND_GDB}" -ne 1 ] && [ "${BACKEND_LLDB}" -ne 1 ]; then

if [ -z "${GDB}" ] && [ -z "${LLDB}" ]; then

echo -n "Couldn't find any debuggers. "

else

echo -n "No debuggers selected. "

echo "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."

voltron/install.sh at master · attackgithub/voltron · GitHub

Expand file tree

Breadcrumbs

install.sh

Latest commit

History

Breadcrumbs

install.sh

File metadata and controls