anduin / Server Init

#!/usr/bin/env bash #=============================================================================== # Concise server preparation script with error confirmation (idempotent) #=============================================================================== 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 \ -o ConnectTimeout=5 \ "$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 (only if changed) CURRENT_HOST=$(run_remote "hostname") if [[ "$CURRENT_HOST" != "$HOSTNAME" ]]; then 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 else print_ok "Hostname already '$HOSTNAME', skipping" fi # 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 & persist random password (once) if run_remote "[ -f /etc/$NEWUSER.pass ]" &>/dev/null; then # In this case, the password is already set print_ok "Don't have to change password. Reusing existing password for $NEWUSER" PASS_NEW=$(run_remote "sudo cat /etc/$NEWUSER.pass") else PASS_NEW=$(uuidgen) print_ok "Setting password for $NEWUSER" run_remote "echo '$NEWUSER:$PASS_NEW' | sudo chpasswd" run_remote "echo '$PASS_NEW' | sudo tee /etc/$NEWUSER.pass > /dev/null" run_remote "sudo chmod 600 /etc/$NEWUSER.pass" run_remote "sudo chown root:root /etc/$NEWUSER.pass" print_ok "New password generated for $NEWUSER and persisted at /etc/$NEWUSER.pass. Please back it up! It can still be used to log in via serial console or rescue mode!" fi local_pass_file="./password_${NEWUSER}_at_${SERVER}.txt" rm -f "$local_pass_file" 2>/dev/null || true sshpass -p "$REMOTE_PASS" ssh -q -o StrictHostKeyChecking=no \ "$REMOTE_USER@$SERVER" "sudo cat /etc/$NEWUSER.pass" \ > "$local_pass_file" sudo chown "$USER:$USER" "$local_pass_file" sudo chmod 600 "$local_pass_file" print_ok "Password for $NEWUSER saved locally at $local_pass_file [DO NOT SHARE THIS FILE! IT CAN BE USED TO LOG IN VIA SERIAL CONSOLE OR RESCUE MODE!]" # 7) Copy SSH key (only if absent) [ ! -f ~/.ssh/id_rsa.pub ] && run_local ssh-keygen -t rsa -N "" -f ~/.ssh/id_rsa PUBKEY=$(<~/.ssh/id_rsa.pub) print_ok "Ensuring SSH key in authorized_keys" run_remote "mkdir -p /home/$NEWUSER/.ssh && \ sudo bash -c 'grep -qxF \"$PUBKEY\" /home/$NEWUSER/.ssh/authorized_keys 2>/dev/null || \ echo \"$PUBKEY\" >> /home/$NEWUSER/.ssh/authorized_keys' && \ sudo chown -R $NEWUSER:$NEWUSER /home/$NEWUSER/.ssh && \ sudo chmod 700 /home/$NEWUSER/.ssh && \ sudo chmod 600 /home/$NEWUSER/.ssh/authorized_keys" # Switch to new user for subsequent operations print_ok "Switching to new user $NEWUSER" 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 && \ sudo systemctl restart sshd || sudo systemctl restart ssh" # 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 && \ 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 && sudo ufw allow OpenSSH && echo y | sudo ufw enable" # 12) Install & configure Fail2Ban print_ok "Installing Fail2Ban" run_remote "sudo apt-get update && sudo apt-get install -y fail2ban" print_ok "Configuring Fail2Ban" run_remote <<'EOF' sudo tee /etc/fail2ban/jail.local > /dev/null <<EOJ [sshd] enabled = true port = ssh filter = sshd backend = systemd logpath = journal maxretry = 3 findtime = 600 bantime = 3600 EOJ sudo systemctl restart fail2ban EOF print_ok "Fail2Ban setup complete" # 13) Enable BBR (only once) print_ok "Enabling BBR congestion control" run_remote <<'EOF' grep -q 'net.ipv4.tcp_congestion_control = bbr' /etc/sysctl.d/99-bbr.conf 2>/dev/null || { 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 # 14) 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" # 15) Install or upgrade latest HWE kernel if needed print_ok "Checking HWE kernel package on remote" run_remote <<'EOF' set -euo pipefail # Try to find the HWE package HWE_PKG=$(apt search linux-generic-hwe- 2>/dev/null | grep -o 'linux-generic-hwe-[^/ ]*' | head -1) if [ -z "$HWE_PKG" ]; then echo "[ OK ] No HWE kernel package found for this release, skipping" else inst=$(apt-cache policy "$HWE_PKG" | awk '/Installed:/ {print $2}') cand=$(apt-cache policy "$HWE_PKG" | awk '/Candidate:/ {print $2}') if dpkg -s "$HWE_PKG" &>/dev/null; then if [ "$inst" != "$cand" ] && [ "$cand" != "(none)" ]; then echo "[ OK ] Upgrading $HWE_PKG from $inst to $cand" sudo apt-get update sudo apt-get install -y "$HWE_PKG" echo reboot_required > /tmp/.reboot_flag else echo "[ OK ] $HWE_PKG is already at latest version ($inst), skipping" fi else if [ "$cand" != "(none)" ]; then echo "[ OK ] Installing $HWE_PKG ($cand)" sudo apt-get update sudo apt-get install -y "$HWE_PKG" echo reboot_required > /tmp/.reboot_flag else echo "[ OK ] $HWE_PKG has no installation candidate, skipping" fi fi fi EOF # 16) Conditionally reboot & wait if run_remote 'test -f /tmp/.reboot_flag'; then print_ok "Rebooting server to apply new kernel" run_remote "rm -f /tmp/.reboot_flag" run_remote "sudo reboot" || true sleep 5 wait_ssh else print_ok "No new kernel installed, skipping reboot" fi # 17) Final updates & cleanup print_ok "Installing upgrades & cleanup" run_remote "sudo apt-get update && sudo apt-get upgrade -y && sudo apt-get autoremove -y" # 18) Performance tuning print_ok "Tuning CPU performance & timezone" run_remote "sudo apt-get install -y linux-tools-$(uname -r) cpupower && \ sudo cpupower frequency-set -g performance || true && \ sudo timedatectl set-timezone GMT" # 19) Remove snap print_ok "Removing snapd" run_remote <<'EOF' # 1) 如果 snapd.service 存在，就 disable 一下；否则跳过 if systemctl list-unit-files | grep -q '^snapd\.service'; then sudo systemctl disable --now snapd || true else echo "[ OK ] snapd.service not found, skipping disable" fi # 2) 如果 dpkg 里检测到 snapd 包，就 purge 并清理数据目录 if dpkg -l snapd &>/dev/null; then sudo apt-get purge -y snapd sudo rm -rf /snap /var/lib/snapd /var/cache/snapd else echo "[ OK ] snapd package not installed, skipping purge" fi # 3) 在所有机器都写上 no-snap 的 pin sudo tee /etc/apt/preferences.d/no-snap.pref > /dev/null <<EOP Package: snapd Pin: release a=* Pin-Priority: -10 EOP EOF # 20) Final cleanup & benchmark print_ok "Final autoremove & benchmark" run_remote "sudo apt-get autoremove -y --purge && \ sudo apt-get install -y sysbench && sysbench cpu --threads=$(nproc) run && \ sudo apt-get autoremove -y sysbench --purge" print_ok "Setup complete. Connect via: ssh $NEWUSER@$SERVER" # After this script, server will: # * Only allow SSH key login # * Root login disabled, password authentication disabled # * Have a new hostname set # * Have a new user with sudo privileges and can log in via SSH # * Have a random password stored securely at /etc/<new_user>.pass # * Have SSH key copied to authorized_keys so you can log in without a password # * Be hardened with UFW, Fail2Ban and allowed SSH connections(only) # * Have BBR enabled for better network performance # * Have the latest HWE kernel installed # * Have the best mirror selected for package updates # * Have snap removed # * Have CPU performance tuned to 'performance' mode # * Have timezone set to GMT # * Have all unnecessary users removed (Check /etc/passwd for remaining users) # * Have all unnecessary packages removed # * Have the latest updates installed # * Have sysbench installed for performance testing # * Have a final benchmark run to verify CPU performance # * Have a final cleanup of unnecessary packages

#!/usr/bin/env bash set -euo pipefail echo "[+] Updating package index and installing fail2ban..." sudo apt update sudo apt install -y fail2ban echo "[+] Writing /etc/fail2ban/jail.local..." sudo tee /etc/fail2ban/jail.local > /dev/null <<'EOF' [sshd] enabled = true port = ssh filter = sshd logpath = /var/log/auth.log maxretry = 3 findtime = 600 bantime = 3600 EOF sleep 1 echo "[+] Restarting fail2ban service..." sudo systemctl restart fail2ban echo "=== Fail2Ban global status ===" # Allow script to continue even if fail2ban-client status fails (e.g., socket not yet ready) sudo fail2ban-client status || true echo "=== SSHD jail status ===" sudo fail2ban-client status sshd || true echo "Tip: To view the currently banned IP list again, run:" echo "sudo fail2ban-client status sshd" echo "Tip: To unban an IP address, run:" echo "sudo fail2ban-client set sshd unbanip <IP_ADDRESS>" echo "Tip: To ban an IP address manually, run:" echo "sudo fail2ban-client set sshd banip <IP_ADDRESS>" echo "Tip: To view the fail2ban logs, run:" echo "sudo journalctl -u fail2ban"

#!/usr/bin/env bash set -euo pipefail # Check current APT source format status check_apt_format() { local old_format=false local new_format=false # Check old format (.list) if [ -f "/etc/apt/sources.list" ]; then if grep -v '^#' /etc/apt/sources.list | grep -q '[^[:space:]]'; then old_format=true fi fi # Check for ubuntu.sources file in new format if [ -f "/etc/apt/sources.list.d/ubuntu.sources" ]; then if grep -v '^#' /etc/apt/sources.list.d/ubuntu.sources | grep -q '[^[:space:]]'; then new_format=true fi fi # Return status if $old_format && $new_format; then echo "both" elif $old_format; then echo "old" elif $new_format; then echo "new" else echo "none" fi } # Find the fastest mirror find_fastest_mirror() { # Redirect all output to stderr echo "Testing mirror speeds..." >&2 # Get current Ubuntu codename codename=$(lsb_release -cs) # Define 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/" # USTC "http://ftp.sjtu.edu.cn/ubuntu/" # SJTU "http://mirrors.tuna.tsinghua.edu.cn/ubuntu/" # Tsinghua "http://mirrors.aliyun.com/ubuntu/" # Aliyun "http://mirrors.163.com/ubuntu/" # NetEase "http://mirrors.cloud.tencent.com/ubuntu/" # Tencent Cloud "http://mirror.aiursoft.cn/ubuntu/" # Aiursoft "http://mirrors.huaweicloud.com/ubuntu/" # Huawei Cloud "http://mirrors.zju.edu.cn/ubuntu/" # Zhejiang University "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 # Test speed of each mirror for mirror in "${mirrors[@]}"; do echo "Testing $mirror ..." >&2 response="$(curl -o /dev/null -s -w "%{http_code} %{time_total}\n" \ --connect-timeout 1 --max-time 3 "${mirror}dists/${codename}/Release")" http_code=$(echo "$response" | awk '{print $1}') time_total=$(echo "$response" | awk '{print $2}') if [ "$http_code" -eq 200 ]; then results["$mirror"]="$time_total" echo " Success: $time_total seconds" >&2 else echo " Failed: HTTP code $http_code" >&2 results["$mirror"]="9999" fi done # Sort mirrors by response time sorted_mirrors="$( for url in "${!results[@]}"; do echo "$url ${results[$url]}" done | sort -k2 -n )" echo >&2 echo "=== Mirrors sorted by response time (ascending) ===" >&2 echo "$sorted_mirrors" >&2 echo >&2 # Choose the fastest mirror fastest_mirror="$(echo "$sorted_mirrors" | head -n 1 | awk '{print $1}')" if [[ "$fastest_mirror" == "" || "${results[$fastest_mirror]}" == "9999" ]]; then echo "No usable mirror found, using default mirror" >&2 fastest_mirror="http://archive.ubuntu.com/ubuntu/" fi echo "Fastest mirror found: $fastest_mirror" >&2 echo >&2 # Only this line will be returned to caller, without >&2 echo "$fastest_mirror" } # Generate old format source list generate_old_format() { local mirror="$1" local codename="$2" echo "Generating old format source list /etc/apt/sources.list" sudo tee /etc/apt/sources.list >/dev/null <<EOF deb $mirror $codename main restricted universe multiverse deb $mirror $codename-updates main restricted universe multiverse deb $mirror $codename-backports main restricted universe multiverse deb $mirror $codename-security main restricted universe multiverse EOF echo "Old format source list updated" } # Generate new format source list generate_new_format() { local mirror="$1" local codename="$2" echo "Generating new format source list /etc/apt/sources.list.d/ubuntu.sources" sudo tee /etc/apt/sources.list.d/ubuntu.sources >/dev/null <<EOF Types: deb URIs: $mirror Suites: $codename Components: main restricted universe multiverse Signed-By: /usr/share/keyrings/ubuntu-archive-keyring.gpg Types: deb URIs: $mirror Suites: $codename-updates Components: main restricted universe multiverse Signed-By: /usr/share/keyrings/ubuntu-archive-keyring.gpg Types: deb URIs: $mirror Suites: $codename-backports Components: main restricted universe multiverse Signed-By: /usr/share/keyrings/ubuntu-archive-keyring.gpg Types: deb URIs: $mirror Suites: $codename-security Components: main restricted universe multiverse Signed-By: /usr/share/keyrings/ubuntu-archive-keyring.gpg EOF echo "New format source list updated" } # Main function main() { # Ensure required packages are installed sudo apt update sudo apt install -y curl lsb-release bc # Get current source format format=$(check_apt_format) echo "Current APT source format status: $format" # Get Ubuntu codename codename=$(lsb_release -cs) echo "Ubuntu codename: $codename" # Find the fastest mirror echo "Searching for the fastest mirror..." fastest_mirror=$(find_fastest_mirror) # Decide update strategy based on format case "$format" in "none") echo "No valid APT source found, generating old format source list" generate_old_format "$fastest_mirror" "$codename" ;; "old") echo "System uses traditional format, updating old format source list" generate_old_format "$fastest_mirror" "$codename" ;; "new") echo "System uses modern format, updating new format source list" generate_new_format "$fastest_mirror" "$codename" ;; "both") echo "System uses both formats, old format will be removed, only new format will be retained" sudo mv /etc/apt/sources.list /etc/apt/sources.list.bak echo "Old format source list backed up to /etc/apt/sources.list.bak" generate_new_format "$fastest_mirror" "$codename" ;; esac # Update package list echo "Updating package list..." sudo apt update echo "APT source optimization completed!" aptVersion=$(apt --version | head -n 1 | awk '{print $2}') echo "Current APT version: $aptVersion" # If current APT version is 3.0 or higher, and using old format or none, convert to new format # sudo apt modernize-sources if [[ $(echo "$aptVersion >= 3.0" | bc) -eq 1 && ( "$format" == "old" || "$format" == "none" ) ]]; then echo "APT version is 3.0 or higher, converting to new format" sudo apt modernize-sources echo "APT sources converted to new format" fi } # Execute main function main