anduin / Server Init

#!/usr/bin/env bash #=============================================================================== # Concise server preparation script with error confirmation #=============================================================================== set -euo pipefail export LC_ALL=C export LANG=en_US.UTF-8 export DEBIAN_FRONTEND=noninteractive #----------------------------------- # Colors & Prompts #----------------------------------- Green="\033[32m"; Red="\033[31m"; Yellow="\033[33m"; Blue="\033[36m"; Font="\033[0m" OK="${Green}[ OK ]${Font}"; ERROR="${Red}[FAILED]${Font}"; WARN="${Yellow}[ WARN ]${Font}" print_ok(){ echo -e "${OK} $1"; } print_error(){ echo -e "${ERROR} $1"; } print_warn(){ echo -e "${WARN} $1"; } #----------------------------------- # Error handling & confirmation #----------------------------------- on_error(){ print_error "Error at line $1."; areYouSure; } trap 'on_error $LINENO' ERR areYouSure(){ print_warn "Continue despite errors? [y/N]" read -r ans case $ans in [yY]*) print_ok "Continuing...";; *) print_error "Aborted."; exit 1;; esac } #----------------------------------- # Helpers #----------------------------------- run_local(){ print_ok "Local: $*"; "$@"; } run_remote(){ sshpass -p "$REMOTE_PASS" ssh -o StrictHostKeyChecking=no "$REMOTE_USER@$SERVER" "$*"; } wait_ssh(){ print_ok "Waiting for SSH on $SERVER...(Running ssh $REMOTE_USER@$SERVER)" until sshpass -p "$REMOTE_PASS" ssh -q -o StrictHostKeyChecking=no "$REMOTE_USER@$SERVER" exit; do print_warn "SSH not ready, retrying in 5s..." sleep 5 done print_ok "SSH available." } usage(){ echo "Usage: $0 <orig_user> <orig_pass> <server> <new_hostname> <new_user>"; exit 1; } #----------------------------------- # Main #----------------------------------- [ $# -ne 5 ] && usage USER="$1"; PASS="$2"; SERVER="$3"; HOSTNAME="$4"; NEWUSER="$5" REMOTE_USER="$USER"; REMOTE_PASS="$PASS" # 1) Install sshpass locally run_local sudo apt-get update -y run_local sudo apt-get install -y sshpass # 2) Clear known_hosts, wait for SSH run_local ssh-keygen -R "$SERVER" -f ~/.ssh/known_hosts wait_ssh # 3) Hostname & reboot print_ok "Setting hostname to $HOSTNAME" run_remote "sudo hostnamectl set-hostname $HOSTNAME" run_remote "sudo reboot" || true print_ok "Server rebooting..." sleep 5 wait_ssh # 4) Create or verify new user if run_remote "id -u $NEWUSER" &>/dev/null; then print_ok "User $NEWUSER exists" else print_ok "Creating user $NEWUSER" run_remote "sudo adduser --disabled-password --gecos '' $NEWUSER" fi # 5) Grant sudo & set up passwordless print_ok "Granting sudo to $NEWUSER" run_remote "sudo usermod -aG sudo $NEWUSER" print_ok "Setting passwordless sudo for $NEWUSER" run_remote "echo '$NEWUSER ALL=(ALL) NOPASSWD:ALL' | sudo tee /etc/sudoers.d/$NEWUSER" # 6) Generate & set random password PASS_NEW=$(uuidgen) print_ok "Setting password for $NEWUSER" run_remote "echo '$NEWUSER:$PASS_NEW' | sudo chpasswd" print_ok "New password for $NEWUSER: $PASS_NEW" # 7) Copy SSH key [ ! -f ~/.ssh/id_rsa.pub ] && run_local ssh-keygen -t rsa -N "" -f ~/.ssh/id_rsa print_ok "Copying SSH key" sshpass -p "$PASS_NEW" ssh-copy-id -i ~/.ssh/id_rsa.pub "$NEWUSER@$SERVER" # Switch to new user for subsequent operations print_ok "Switching to new user $NEWUSER instead of $REMOTE_USER" REMOTE_USER="$NEWUSER"; REMOTE_PASS="$PASS_NEW" wait_ssh # 8) Harden SSH print_ok "Hardening SSH settings" run_remote "sudo sed -i 's/PermitRootLogin yes/PermitRootLogin no/; s/PasswordAuthentication yes/PasswordAuthentication no/; s/#PubkeyAuthentication yes/PubkeyAuthentication yes/' /etc/ssh/sshd_config" run_remote "sudo systemctl restart sshd" # 9) Remove other non-system users print_ok "Removing other users" others=$(run_remote \ "awk -F: -v skip='$NEWUSER' '\$3>=1000 && \$1!=skip {print \$1}' /etc/passwd") for u in $others; do print_warn "Deleting user $u" run_remote "sudo pkill -u $u || true; sudo deluser --remove-home $u" done # 10) Reset machine-id print_ok "Resetting machine-id" run_remote "sudo rm -f /etc/machine-id /var/lib/dbus/machine-id" run_remote "sudo systemd-machine-id-setup; sudo cp /etc/machine-id /var/lib/dbus/machine-id" # 11) Enable UFW & OpenSSH print_ok "Enabling UFW firewall" run_remote "sudo apt-get install -y ufw" run_remote "sudo ufw allow OpenSSH && echo y | sudo ufw enable" # 12) Enable BBR print_ok "Enabling BBR congestion control" #run_remote "sudo bash -c 'grep -q bbr /etc/sysctl.conf || { echo net.core.default_qdisc=fq >>/etc/sysctl.conf; echo net.ipv4.tcp_congestion_control=bbr >>/etc/sysctl.conf; sysctl -p; }'" run_remote <<'EOF' sudo tee /etc/sysctl.d/99-bbr.conf > /dev/null <<SYSCTL net.core.default_qdisc = fq net.ipv4.tcp_congestion_control = bbr SYSCTL sudo sysctl --system EOF # 13) Select best mirror & update print_ok "Selecting best mirror & updating" run_remote "curl -s https://gist.aiursoft.cn/anduin/879917820a6c4b268fc12c21f1b3fe7a/raw/HEAD/mirror.sh | bash" run_remote "sudo apt-get update" # 14) Install latest HWE kernel print_ok "Installing latest HWE kernel" run_remote "sudo apt-get install -y \$(apt search linux-generic-hwe- | awk -F/ '/linux-generic-hwe-/{print \$1}' | head -1)" # 15) Reboot & wait print_ok "Rebooting server" run_remote "sudo reboot" || true sleep 5 wait_ssh # 16) Final updates & cleanup print_ok "Installing upgrades & cleanup" run_remote "sudo apt-get update && sudo apt-get upgrade -y && sudo apt-get autoremove -y" # 17) Performance tuning print_ok "Tuning CPU performance & timezone" run_remote "sudo apt-get install -y linux-tools-$(uname -r)" run_remote "sudo apt-get install -y cpupower" run_remote "sudo cpupower frequency-set -g performance" || true run_remote "sudo timedatectl set-timezone GMT" # 18) Remove snap print_ok "Removing snapd" run_remote "sudo systemctl disable --now snapd && sudo apt-get purge -y snapd && sudo rm -rf /snap /var/lib/snapd /var/cache/snapd" run_remote "sudo tee /etc/apt/preferences.d/no-snap.pref <<EOF Package: snapd Pin: release a=* Pin-Priority: -10 EOF" # 19) Final cleanup & benchmark print_ok "Final autoremove & benchmark" run_remote "sudo apt-get autoremove -y --purge" run_remote "sudo apt-get install -y sysbench && sysbench cpu --threads=\$(nproc) run" run remote "sudo apt-get autoremove -y sysbench --purge" print_ok "Setup complete. Connect via: ssh $NEWUSER@$SERVER"

#!/usr/bin/env bash # Step 1: Ensure required packages are installed sudo apt update sudo apt install -y curl apt-transport-https lsb-release function switchSource() { # Get current Ubuntu codename (e.g., jammy, focal, bionic) codename=$(lsb_release -cs) # Define a list of potential mirrors mirrors=( "https://archive.ubuntu.com/ubuntu/" "https://mirror.aarnet.edu.au/pub/ubuntu/archive/" # Australia "https://mirror.fsmg.org.nz/ubuntu/" # New Zealand "https://mirrors.neterra.net/ubuntu/archive/" # Bulgaria "https://mirror.csclub.uwaterloo.ca/ubuntu/" # Canada "https://mirrors.dotsrc.org/ubuntu/" # Denmark "https://mirrors.nic.funet.fi/ubuntu/" # Finland "https://mirror.ubuntu.ikoula.com/" # France "https://mirror.xtom.com.hk/ubuntu/" # Hong Kong "https://mirrors.piconets.webwerks.in/ubuntu-mirror/ubuntu/" # India "https://ftp.udx.icscoe.jp/Linux/ubuntu/" # Japan "https://ftp.kaist.ac.kr/ubuntu/" # Korea "https://ubuntu.mirror.garr.it/ubuntu/" # Italy "https://ftp.uni-stuttgart.de/ubuntu/" # Germany "https://mirror.i3d.net/pub/ubuntu/" # Netherlands "https://mirroronet.pl/pub/mirrors/ubuntu/" # Poland "https://ubuntu.mobinhost.com/ubuntu/" # Iran "http://sg.archive.ubuntu.com/ubuntu/" # Singapore "http://ossmirror.mycloud.services/os/linux/ubuntu/" # Singapore "https://mirror.enzu.com/ubuntu/" # United States "http://jp.archive.ubuntu.com/ubuntu/" # Japan "http://kr.archive.ubuntu.com/ubuntu/" # Korea "http://us.archive.ubuntu.com/ubuntu/" # United States "http://tw.archive.ubuntu.com/ubuntu/" # Taiwan "https://mirror.twds.com.tw/ubuntu/" # Taiwan "https://ubuntu.mirrors.uk2.net/ubuntu/" # United Kingdom "http://mirrors.ustc.edu.cn/ubuntu/" # 中国科学技术大学 "http://ftp.sjtu.edu.cn/ubuntu/" # 上海交通大学 "http://mirrors.tuna.tsinghua.edu.cn/ubuntu/" # 清华大学 "http://mirrors.aliyun.com/ubuntu/" # 阿里云 "http://mirrors.163.com/ubuntu/" # 网易 "http://mirrors.cloud.tencent.com/ubuntu/" # 腾讯云 "http://mirror.aiursoft.cn/ubuntu/" # Aiursoft "http://mirrors.huaweicloud.com/ubuntu/" # 华为云 "http://mirrors.zju.edu.cn/ubuntu/" # 浙江大学 "http://azure.archive.ubuntu.com/ubuntu/" # Azure "https://mirrors.isu.net.sa/apt-mirror/" # Saudi Arabia "https://mirror.team-host.ru/ubuntu/" # Russia "https://labs.eif.urjc.es/mirror/ubuntu/" # Spain "https://mirror.alastyr.com/ubuntu/ubuntu-archive/" # Turkey "https://ftp.acc.umu.se/ubuntu/" # Sweden "https://mirror.kku.ac.th/ubuntu/" # Thailand "https://mirror.bizflycloud.vn/ubuntu/" # Vietnam ) declare -A results # Function to test speed of a single mirror test_speed() { url="$1" # Attempt to do a quick GET and measure total time response="$(curl -o /dev/null -s -w "%{http_code} %{time_total}\n" \ --connect-timeout 1 --max-time 2 "$url")" http_code=$(echo "$response" | awk '{print $1}') time_total=$(echo "$response" | awk '{print $2}') # If HTTP code == 200, mark the measured time; otherwise use a large value if [ "$http_code" -eq 200 ]; then results["$url"]="$time_total" else echo "Failed to access $url (HTTP code: $http_code)" results["$url"]="9999" fi } echo "Testing all mirrors for Ubuntu '$codename'..." for mirror in "${mirrors[@]}"; do test_speed "$mirror" done # Sort mirrors by time_total # Example of sorted_mirrors entry: "https://archive.ubuntu.com/ubuntu/ 0.034" sorted_mirrors="$( for url in "${!results[@]}"; do echo "$url ${results[$url]}" done | sort -k2 -n )" echo echo "=== Sorted mirrors by response time (ascending) ===" echo "$sorted_mirrors" echo # Pick the top (fastest) mirror from the sorted list fastest_mirror="$(echo "$sorted_mirrors" | head -n 1 | awk '{print $1}')" echo "Fastest mirror found: $fastest_mirror" echo "Updating /etc/apt/sources.list..." # Update /etc/apt/sources.list with the fastest mirror sudo tee /etc/apt/sources.list >/dev/null <<EOF deb $fastest_mirror $codename main restricted universe multiverse deb $fastest_mirror $codename-updates main restricted universe multiverse deb $fastest_mirror $codename-backports main restricted universe multiverse deb $fastest_mirror $codename-security main restricted universe multiverse EOF # Final check sudo apt update echo "All done!" } # Call the main function switchSource