Revert "Revert "Add root uid judgment before running"" (#183)
This reverts commit 0941d67557
.
pull/184/head
parent
0941d67557
commit
cbae6a10b1
|
@ -26,6 +26,14 @@ curl() {
|
|||
$(type -P curl) -L -q --retry 5 --retry-delay 10 --retry-max-time 60 "$@"
|
||||
}
|
||||
|
||||
check_if_running_as_root() {
|
||||
# If you want to run as another user, please modify $UID to be owned by this user
|
||||
if [[ "$UID" -ne '0' ]]; then
|
||||
echo "error: You must run this script as root!"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
download_files() {
|
||||
if ! curl -R -H 'Cache-Control: no-cache' -o "${dir_tmp}/${2}" "${1}"; then
|
||||
echo 'error: Download failed! Please check your network or try again.'
|
||||
|
@ -56,6 +64,7 @@ install_file() {
|
|||
}
|
||||
|
||||
main() {
|
||||
check_if_running_as_root
|
||||
download_files $DOWNLOAD_LINK_GEOIP $file_ip
|
||||
download_files $DOWNLOAD_LINK_GEOSITE $file_dlc
|
||||
check_sum
|
||||
|
|
|
@ -44,6 +44,14 @@ systemd_cat_config() {
|
|||
fi
|
||||
}
|
||||
|
||||
check_if_running_as_root() {
|
||||
# If you want to run as another user, please modify $UID to be owned by this user
|
||||
if [[ "$UID" -ne '0' ]]; then
|
||||
echo "error: You must run this script as root!"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
identify_the_operating_system_and_architecture() {
|
||||
if [[ "$(uname)" == 'Linux' ]]; then
|
||||
case "$(uname -m)" in
|
||||
|
@ -491,6 +499,7 @@ show_help() {
|
|||
}
|
||||
|
||||
main() {
|
||||
check_if_running_as_root
|
||||
identify_the_operating_system_and_architecture
|
||||
judgment_parameters "$@"
|
||||
|
||||
|
|
Loading…
Reference in New Issue