mirror of
https://github.com/zardus/ctf-tools
synced 2024-12-14 06:52:36 +00:00
268 lines
5.4 KiB
Bash
Executable file
268 lines
5.4 KiB
Bash
Executable file
#!/bin/bash -e
|
|
# set -evx
|
|
|
|
function usage()
|
|
{
|
|
cat <<END
|
|
Usage: $(basename $0) [-s] (list|setup|install|uninstall|bin|search) tool
|
|
|
|
Where:
|
|
-s allow running things with sudo (i.e., to install debs)
|
|
tool the name of the tool. if "all", does the action on all tools
|
|
|
|
Actions:
|
|
setup set up the environment (adds ctf-tools/bin to your \$PATH in .bashrc)
|
|
list list all tools (-i: only installed, -u: only uninstalled)
|
|
install installs a tool
|
|
uninstall uninstalls a tool
|
|
reinstall reinstalls a tool
|
|
upgrade upgrades a tool
|
|
bin re-links tool binaries into ctf-tools/bin
|
|
search search description and name of tools
|
|
|
|
END
|
|
}
|
|
|
|
function tool_log()
|
|
{
|
|
echo "$(tput setaf 4 2>/dev/null)TOOLS | $TOOL |$(tput sgr0 2>/dev/null) $@"
|
|
}
|
|
|
|
|
|
function detect_distribution()
|
|
{
|
|
if which pacman 2>&1 >/dev/null; then
|
|
echo "archlinux"
|
|
elif which apt-get 2>&1 >/dev/null; then
|
|
echo "debian"
|
|
else
|
|
echo ""
|
|
fi
|
|
}
|
|
|
|
|
|
function base_build_setup_ubuntu()
|
|
{
|
|
PACKAGE_REQS="build-essential libtool g++ gcc linuxtexinfo curl wget automake autoconf python-dev git subversion unzip"
|
|
PACKAGE_COUNT=$(echo $PACKAGE_REQS | tr ' ' '\n' | wc -l)
|
|
if [ $(dpkg -l $PACKAGE_REQS | grep "^ii" | wc -l) -ne $PACKAGE_COUNT ]
|
|
then
|
|
if [ "$ALLOW_SUDO" -eq 1 ]
|
|
then
|
|
sudo apt-get -y install $PACKAGE_REQS
|
|
else
|
|
TOOL=SETUP tool_log "Please install the following packages: $PACKAGE_REQS"
|
|
fi
|
|
fi
|
|
|
|
if ! dpkg --print-foreign-architectures | grep -q i386
|
|
then
|
|
if [ "$ALLOW_SUDO" -eq 1 ]
|
|
then
|
|
sudo dpkg --add-architecture i386
|
|
sudo apt-get update
|
|
else
|
|
TOOL=SETUP tool_log "Certain tools need i386 libraries (enable with 'dpkg --add-architecture i386; apt-get update')."
|
|
fi
|
|
fi
|
|
}
|
|
|
|
|
|
function base_build_setup_arch()
|
|
{
|
|
PACKAGE_REQS="curl wget python2 python3 git subversion unzip"
|
|
if [ "$ALLOW_SUDO" -eq 1 ]; then
|
|
sudo pacman -Syu --noconfirm $PACKAGE_REQS
|
|
sudo pacman -Syu --noconfirm base-devel || true
|
|
else
|
|
TOOL=SETUP tool_log "Please install the following packages: $PACKAGE_REQS"
|
|
fi
|
|
|
|
|
|
if ! grep "^\[multilib\]$" /etc/pacman.conf >/dev/null; then
|
|
if [ "$ALLOW_SUDO" -eq 1 ]
|
|
then
|
|
sudo sh -c 'cat >> /etc/pacman.conf' <<EOF
|
|
|
|
[multilib]
|
|
Include = /etc/pacman.d/mirrorlist
|
|
EOF
|
|
else
|
|
TOOL=SETUP tool_log "Certain tools need i386 libraries (enable multilib in /etc/pacman.conf')."
|
|
return
|
|
fi
|
|
fi
|
|
|
|
if [ "$ALLOW_SUDO" -eq 1 ] \
|
|
&& grep "^\[multilib\]$" /etc/pacman.conf >/dev/null \
|
|
&& ! sudo pacman -Qk gcc-multilib >/dev/null
|
|
then
|
|
sudo pacman -Syy --noconfirm
|
|
#sudo pacman -Syu --noconfirm multilib-devel
|
|
# unfortunately we cannot do --noconfirm if we might choose to replace
|
|
# a package such as gcc with gcc-multilib, therefore this workaround
|
|
printf "\ny\ny\ny\n" | sudo pacman -Syu multilib-devel
|
|
fi
|
|
}
|
|
|
|
|
|
function base_build_setup()
|
|
{
|
|
case "$1" in
|
|
"ubuntu")
|
|
base_build_setup_ubuntu
|
|
;;
|
|
"archlinux")
|
|
base_build_setup_arch
|
|
;;
|
|
*)
|
|
TOOL=SETUP tool_log "Cannot detect or unsupported distribution"
|
|
esac
|
|
}
|
|
|
|
|
|
|
|
DISTRI=$(detect_distribution)
|
|
|
|
while [[ $1 == -* ]]
|
|
do
|
|
case $1 in
|
|
-s)
|
|
export ALLOW_SUDO=1
|
|
;;
|
|
*)
|
|
usage
|
|
exit
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
[ -z "$ALLOW_SUDO" ] && export ALLOW_SUDO=0
|
|
|
|
ACTION=$1
|
|
TOOL=$2
|
|
|
|
if [ "$TOOL" == "all" ]
|
|
then
|
|
for t in $($0 list)
|
|
do
|
|
$0 $ACTION $t
|
|
done
|
|
elif [ -z "$TOOL" -a "$ACTION" != "list" -a "$ACTION" != "setup" ]
|
|
then
|
|
usage
|
|
exit
|
|
fi
|
|
|
|
cd $(dirname "${BASH_SOURCE[0]}")/..
|
|
|
|
case $ACTION in
|
|
setup)
|
|
|
|
base_build_setup "$DISTRI"
|
|
|
|
<<<<<<< HEAD
|
|
echo "PATH=\"$PWD/bin:\$PATH\"" >> ~/.bashrc
|
|
if [ -e ~/.zshrc ]
|
|
then
|
|
=======
|
|
if [ -e ~/.zshrc ]
|
|
then
|
|
>>>>>>> 82571b6917b6dbb212e722d35c6f7d2d99b7e0cb
|
|
echo "PATH=\"$PWD/bin:\$PATH\"" >> ~/.zshrc
|
|
fi
|
|
|
|
if [ -e ~/.config/fish/config.fish ]; then
|
|
echo "set -x PATH $PATH $PWD/bin" >> ~/.config/fish/config.fish
|
|
fi
|
|
|
|
;;
|
|
list)
|
|
for t in *
|
|
do
|
|
[ ! -e "$t/install" ] && continue
|
|
echo $t
|
|
done
|
|
;;
|
|
bin)
|
|
cd bin
|
|
ln -sf ../$TOOL/bin/* .
|
|
tool_log "bin symlinks updated"
|
|
cd ..
|
|
;;
|
|
install)
|
|
cd $TOOL
|
|
if git status --ignored . | egrep -q 'Untracked|Ignored'
|
|
then
|
|
tool_log "appears to already be installed. Uninstall first?"
|
|
exit 0
|
|
fi
|
|
|
|
tool_log "starting install, logging to $PWD/install.log"
|
|
rm -f install.log
|
|
if [ -x ./install-root -a "$ALLOW_SUDO" -eq 1 ]; then
|
|
sudo env DISTRI=$DISTRI ./install-root >> install.log 2>&1
|
|
fi
|
|
if env DISTRI=$DISTRI ./install >>install.log 2>&1
|
|
then
|
|
tool_log "install finished"
|
|
else
|
|
tool_log "INSTALL FAILED"
|
|
cat install.log >&2
|
|
exit 1
|
|
fi
|
|
|
|
cd ..
|
|
$0 bin $TOOL
|
|
;;
|
|
uninstall)
|
|
cd $TOOL
|
|
|
|
[ -x ./uninstall ] && ./uninstall
|
|
git clean -dffx .
|
|
tool_log "uninstall finished"
|
|
|
|
cd ..
|
|
;;
|
|
upgrade)
|
|
cd $TOOL
|
|
if [ -x ./upgrade ]
|
|
then
|
|
./upgrade
|
|
tool_log "upgrade complete!"
|
|
else
|
|
tool_log "no upgrade script -- reinstalling"
|
|
$0 uninstall $TOOL
|
|
$0 install $TOOL
|
|
fi
|
|
;;
|
|
reinstall)
|
|
$0 uninstall $TOOL
|
|
$0 install $TOOL
|
|
;;
|
|
search)
|
|
cat README.md | grep "<\!--tool-->" | sed "s/<\!--[^-]*-->//g" | grep -i "$TOOL"
|
|
;;
|
|
test)
|
|
if ! cat README.md | grep "<\!--tool-->" | grep "| \[$TOOL\](" | grep -q -- "--test--"
|
|
then
|
|
tool_log "Tests not enabled."
|
|
else
|
|
$0 install $TOOL
|
|
|
|
cd $TOOL
|
|
if [ -f ./test ]
|
|
then
|
|
tool_log "Running test script."
|
|
./test
|
|
tool_log "test script succeeded!"
|
|
else
|
|
tool_log "Install succeeded. No test script!"
|
|
fi
|
|
fi
|
|
;;
|
|
*)
|
|
echo "TOOLS | ERROR | unknown action $ACTION"
|
|
;;
|
|
esac
|