mirror of
https://github.com/snovvcrash/usbrip.git
synced 2024-05-21 14:36:03 +02:00
254 lines
6.8 KiB
Bash
Executable File
254 lines
6.8 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
: '
|
|
%file install.sh
|
|
%author Sam Freeside (@snovvcrash) <snovvcrash@protonmail[.]ch>
|
|
%date 2018-05-28
|
|
|
|
%brief usbrip installer.
|
|
|
|
%license
|
|
Copyright (C) 2020 Sam Freeside
|
|
|
|
This file is part of usbrip.
|
|
|
|
usbrip is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
usbrip is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with usbrip. If not, see <http://www.gnu.org/licenses/>.
|
|
%endlicense
|
|
'
|
|
|
|
# Usage: sudo -H ./install.sh [-l/--local] [-s/--storages]
|
|
|
|
shopt -s expand_aliases
|
|
|
|
# ----------------------- Constants ------------------------
|
|
|
|
OPT="/opt/usbrip"
|
|
LOG="/var/opt/usbrip/log"
|
|
STORAGE="/var/opt/usbrip/storage"
|
|
SYMLINK="/usr/local/bin/usbrip"
|
|
|
|
G="\033[1;32m" # GREEN
|
|
Y="\033[1;33m" # YELLOW
|
|
R="\033[1;31m" # RED
|
|
W="\033[1;37m" # WHITE
|
|
NC="\033[0m" # NO COLOR
|
|
|
|
# --------------------- usbrip aliases ---------------------
|
|
|
|
alias createHistoryStorage="${OPT}/venv/bin/usbrip storage create history -e"
|
|
|
|
alias generateAuthorizedDeviceList="${OPT}/venv/bin/usbrip events gen_auth /var/opt/usbrip/trusted/auth.json -e -a vid pid"
|
|
|
|
alias createViolationsStorage="${OPT}/venv/bin/usbrip storage create violations -i /var/opt/usbrip/trusted/auth.json -e -a vid pid"
|
|
|
|
# --------------- Check for root privileges ----------------
|
|
|
|
if [[ $EUID -ne 0 ]]; then
|
|
/usr/bin/printf "${R}>>>>${NC} Please run as root:\nsudo -H %s\n" "${0}"
|
|
exit 1
|
|
fi
|
|
|
|
# -------------------- Handle switches ---------------------
|
|
|
|
LOCAL=false
|
|
STORAGES=false
|
|
|
|
if [[ "$1" == "-l" ]] || [[ "$1" == "--local" ]]; then
|
|
LOCAL=true
|
|
elif [[ "$1" == "-s" ]] || [[ "$1" == "--storages" ]]; then
|
|
STORAGES=true
|
|
fi
|
|
|
|
if [[ "$2" == "-l" ]] || [[ "$2" == "--local" ]]; then
|
|
LOCAL=true
|
|
elif [[ "$2" == "-s" ]] || [[ "$2" == "--storages" ]]; then
|
|
STORAGES=true
|
|
fi
|
|
|
|
# -------------- Check for required packages ---------------
|
|
|
|
# python3-venv
|
|
|
|
if ! /usr/bin/dpkg-query -W -f='${Status}' python3-venv 2>&1 | /bin/grep "ok installed" > /dev/null; then
|
|
/usr/bin/printf "${Y}>>>>${NC} Unresolved dependency: python3-venv. Do you want to install this package as:\n%s\n" \
|
|
"\"sudo apt install python3-venv -y\"?"
|
|
select YN in "Yes" "No"; do
|
|
case ${YN} in
|
|
"Yes" )
|
|
$(which apt) install "python3-venv" -y
|
|
break
|
|
;;
|
|
"No" )
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
fi
|
|
|
|
# p7zip-full
|
|
|
|
if ! /usr/bin/dpkg-query -W -f='${Status}' p7zip-full 2>&1 | /bin/grep "ok installed" > /dev/null && ${STORAGES}; then
|
|
/usr/bin/printf "${Y}>>>>${NC} Unresolved dependency: p7zip-full. Do you want to install this package as:\n%s\n" \
|
|
"\"sudo apt install p7zip-full -y\"?"
|
|
select YN in "Yes" "No"; do
|
|
case ${YN} in
|
|
"Yes" )
|
|
$(which apt) install "p7zip-full" -y
|
|
break
|
|
;;
|
|
"No" )
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
fi
|
|
|
|
# ------------------- Create directories -------------------
|
|
|
|
# OPT
|
|
|
|
/usr/bin/printf "${W}>>>>${NC} Creating directory: '${OPT}'\n"
|
|
|
|
if [[ -d "${OPT}" ]]; then
|
|
/usr/bin/printf "${R}>>>>${NC} ${OPT} already exists. First run:\n%s\n" \
|
|
"sudo uninstall.sh --all"
|
|
exit 1
|
|
fi
|
|
|
|
if /bin/mkdir "${OPT}"; then
|
|
/usr/bin/printf "${G}>>>>${NC} Successfully created directory: '${OPT}'\n\n"
|
|
else
|
|
/usr/bin/printf "${R}>>>>${NC} Failed to create directory: '${OPT}'\n"
|
|
exit 1
|
|
fi
|
|
|
|
# LOG
|
|
|
|
/usr/bin/printf "${W}>>>>${NC} Creating directory: '${LOG}'\n"
|
|
|
|
if [[ -d "${LOG}" ]]; then
|
|
/usr/bin/printf "${R}>>>>${NC} ${LOG} already exists. First run:\n%s\n" \
|
|
"sudo uninstall.sh --all"
|
|
exit 1
|
|
fi
|
|
|
|
if /bin/mkdir -p "${LOG}"; then
|
|
/usr/bin/printf "${G}>>>>${NC} Successfully created directory: '${LOG}'\n\n"
|
|
else
|
|
/usr/bin/printf "${R}>>>>${NC} Failed to create directory: '${LOG}'\n"
|
|
exit 1
|
|
fi
|
|
|
|
# STORAGE
|
|
|
|
/usr/bin/printf "${W}>>>>${NC} Creating directory: '${STORAGE}'\n"
|
|
|
|
if [[ -d "${STORAGE}" ]]; then
|
|
/usr/bin/printf "${R}>>>>${NC} ${STORAGE} already exists. First run:\n%s\n" \
|
|
"sudo uninstall.sh --all"
|
|
exit 1
|
|
fi
|
|
|
|
if /bin/mkdir -p "${STORAGE}"; then
|
|
/usr/bin/printf "${G}>>>>${NC} Successfully created directory: '${STORAGE}'\n\n"
|
|
else
|
|
/usr/bin/printf "${R}>>>>${NC} Failed to create directory: '${STORAGE}'\n"
|
|
exit 1
|
|
fi
|
|
|
|
# ------------ Build Python virtual environment ------------
|
|
|
|
/usr/bin/printf "${W}>>>>${NC} Building Python virtual environment\n"
|
|
|
|
if /usr/bin/python3 -m venv "${OPT}/venv"; then
|
|
/usr/bin/printf "${G}>>>>${NC} Successfully builded Python virtual environment\n\n"
|
|
else
|
|
/usr/bin/printf "${R}>>>>${NC} Failed to build Python virtual environment\n"
|
|
exit 1
|
|
fi
|
|
|
|
# ------------------------ Install -------------------------
|
|
|
|
/usr/bin/printf "${W}>>>>${NC} Installing usbrip\n"
|
|
|
|
if $LOCAL; then
|
|
if ${OPT}/venv/bin/python "${PWD}/setup.py" install; then
|
|
/usr/bin/printf "${G}>>>>${NC} Successfully installed usbrip using local dependencies\n\n"
|
|
else
|
|
/usr/bin/printf "${R}>>>>${NC} Failed to install usbrip using local dependencies\n"
|
|
exit 1
|
|
fi
|
|
else
|
|
if ${OPT}/venv/bin/pip install "${PWD}"; then
|
|
/usr/bin/printf "${G}>>>>${NC} Successfully installed usbrip using PyPI dependencies\n\n"
|
|
else
|
|
/usr/bin/printf "${R}>>>>${NC} Failed to install usbrip using PyPI dependencies\n"
|
|
exit 1
|
|
fi
|
|
fi
|
|
|
|
${OPT}/venv/bin/python "${PWD}/setup.py" clean
|
|
echo
|
|
|
|
# --------------------- Create symlink ---------------------
|
|
|
|
if [[ -e "${SYMLINK}" ]]; then
|
|
/bin/rm "${SYMLINK}"
|
|
fi
|
|
|
|
if /bin/ln -s "${OPT}/venv/bin/usbrip" "${SYMLINK}"; then
|
|
/usr/bin/printf "${G}>>>>${NC} Created symlink: '${SYMLINK}'\n"
|
|
fi
|
|
|
|
# ----------------- Create usbrip storages -----------------
|
|
|
|
if ${STORAGES}; then
|
|
# History
|
|
|
|
/usr/bin/printf "${W}>>>>${NC} Creating usbrip history storage\n"
|
|
|
|
if createHistoryStorage; then
|
|
/usr/bin/printf "${G}>>>>${NC} Successfully created usbrip history storage\n\n"
|
|
else
|
|
/usr/bin/printf "${R}>>>>${NC} Failed to create usbrip history storage\n"
|
|
exit 1
|
|
fi
|
|
|
|
# Gen Auth
|
|
|
|
/usr/bin/printf "${W}>>>>${NC} Generating authorized device list\n"
|
|
|
|
if generateAuthorizedDeviceList; then
|
|
/usr/bin/printf "${G}>>>>${NC} Successfully generated authorized device list\n\n"
|
|
else
|
|
/usr/bin/printf "${R}>>>>${NC} Failed to generate authorized device list\n"
|
|
exit 1
|
|
fi
|
|
|
|
# Violations
|
|
|
|
/usr/bin/printf "${W}>>>>${NC} Creating usbrip violations storage\n"
|
|
|
|
if createViolationsStorage; then
|
|
/usr/bin/printf "${G}>>>>${NC} Successfully created usbrip violations storage\n\n"
|
|
else
|
|
/usr/bin/printf "${R}>>>>${NC} Failed to create usbrip violations storage\n"
|
|
exit 1
|
|
fi
|
|
fi
|
|
|
|
# -------------------------- Done --------------------------
|
|
|
|
/usr/bin/printf "${G}>>>>${NC} Done.\n"
|