OsMutationKvm.sh

#!/bin/bash
# Reinstall Any OpenVZ/LXC VPS to Debian/CentOS/Alpine
# Author: Lloyd@nodeseek.com
# WARNING: A fresh system will be installed and all old data will be wiped.
# License: GPLv3; Partly based on https://gist.github.com/trimsj/c1fefd650b5f49ceb8f3efc1b6a1404d

function print_help(){
    echo -ne "\e[1;32m"
    cat <<- EOF
                                                                                     
		 ██████╗ ███████╗███╗   ███╗██╗   ██╗████████╗ █████╗ ████████╗██╗ ██████╗ ███╗   ██╗
		██╔═══██╗██╔════╝████╗ ████║██║   ██║╚══██╔══╝██╔══██╗╚══██╔══╝██║██╔═══██╗████╗  ██║
		██║   ██║███████╗██╔████╔██║██║   ██║   ██║   ███████║   ██║   ██║██║   ██║██╔██╗ ██║
		██║   ██║╚════██║██║╚██╔╝██║██║   ██║   ██║   ██╔══██║   ██║   ██║██║   ██║██║╚██╗██║
		╚██████╔╝███████║██║ ╚═╝ ██║╚██████╔╝   ██║   ██║  ██║   ██║   ██║╚██████╔╝██║ ╚████║
		 ╚═════╝ ╚══════╝╚═╝     ╚═╝ ╚═════╝    ╚═╝   ╚═╝  ╚═╝   ╚═╝   ╚═╝ ╚═════╝ ╚═╝  ╚═══╝
                                                                                     
		Reinstall Any OpenVZ/LXC VPS to Debian/CentOS/Alpine;
		[warning] A fresh system will be installed and all old data will be wiped!
		Author: Lloyd@nodeseek.com
	EOF
    echo -ne "\e[m"
}

function read_virt_tech(){
    cttype=$(virt-what | sed -n 1p)
    if [[ $cttype == "lxc" || $cttype == "openvz" ]]; then
        [[ $cttype == "lxc" ]] && echo -e '\e[1;33mYour container type: lxc\e[m' || echo -e '\e[1;33mYour container type: openvz\e[m'
    else
        while [ "$cttype" != 'lxc' -a "$cttype" != 'openvz' ] ; do
            echo -ne "\e[1;33mplease input container type (lxc/openvz):\e[m"
            read cttype
        done
    fi
}

function install(){
    if [ -n "$(command -v apt)" ] ; then
        cmd1="apt-get"
        cmd2="apt-get install -y"
    elif [ -n "$(command -v yum)" ] ; then
        cmd1="yum"
        cmd2="yum install -y"
    elif [ -n "$(command -v dnf)" ] ; then
        cmd1="dnf"
        cmd2="dnf install -y"
    elif [ -n "$(command -v apk)" ] ; then
        cmd1="apk"
        cmd2="apk add"
    else
        echo "Error: Not Supported Os"
        exit 1
    fi
    $cmd1 update
    $cmd2 "$@"
}

function read_lxc_template(){
    last_lxc_version=$(curl -Ls "https://api.github.com/repos/LloydAsp/OsMutation/releases/latest" | grep "LXC" | grep '"tag_name":' | sed -E 's/.*"([^"]+)".*/\1/')
    if [[ -n $last_lxc_version ]]; then
        image_list=$(curl -Ls "https://api.github.com/repos/LloydAsp/OsMutation/releases/latest" | grep "LXC" | grep '"browser_download_url":' | sed -E 's/.*"([^"]+)".*/\1/')

        os_list=$(curl -Ls "https://api.github.com/repos/LloydAsp/OsMutation/releases/latest" | grep "LXC" | grep '"browser_download_url":' | sed -E 's/.*"([^"]+)".*/\1/' | sed "s/https\:\/\/github.com\/LloydAsp\/OsMutation\/releases\/download\/${last_lxc_version}\///g" | sed "s/\.tar\.gz//g")
        echo "$os_list" | nl

        while [ -z "${os_index##*[!0-9]*}" ]; do
            echo -ne "\e[1;33mplease select os (input number):\e[m"
            read os_index
        done

        download_link=$(echo "$image_list" | head -n $os_index | tail -n 1)
    else
        server=http://images.linuxcontainers.org
        path=$(wget -qO- ${server}/meta/1.0/index-system | \
            grep -v edge | grep default | \
            awk '-F;' '(( $1=="debian" || $1=="centos" || $1=="alpine") && ( $3=="amd64" || $3=="i386")) {print $NF}')

        os_list=$( echo "$path" | sed -E 's%/images/(.*)/default/.*/%\1%g' | sed 's%/%-%g' )
        echo "$os_list" | nl

        while [ -z "${os_index##*[!0-9]*}" ]; do
            echo -ne "\e[1;33mplease select os (input number):\e[m"
            read os_index 
        done

        path=$( echo "$path" | head -n $os_index | tail -n 1)
        os_selected=$(echo "$os_list" | head -n $os_index | tail -n 1 )
        download_link=${server}/${path}/rootfs.tar.xz
    fi
}


function download_rootfs(){
    cd /oldroot;
    rm -rf $(ls /oldroot | grep -vE "(^dev|^proc|^sys|^run|^x)") ;

    #rootfs.tar.xz
    wget -qO- $download_link | tar -C /oldroot -xJv --delay-directory-restore --exclude="dev" --exclude="proc" --exclude="run" --exclude="sys"
}


function migrate_configuration(){
    dest_os_dir="$1"
    # save root password and ssh directory
    sed -i '/^root:/d' "/$dest_os_dir/etc/shadow"
    grep '^root:' /etc/shadow >> "/$dest_os_dir/etc/shadow"
    [ -d /root/.ssh ] && cp -a "/root/.ssh" "/$dest_os_dir/root/"
    mkdir -p "/$dest_os_dir/etc/udev/rules.d/"
    [ -f /etc/udev/rules.d/70-persistent-net.rules ] && cp -a "/etc/udev/rules.d/70-persistent-net.rules" "/$dest_os_dir/etc/udev/rules.d/70-persistent-net.rules"

    # save network configuration
    dev=$(ip route get 8.8.8.8 | sed -n 's/.*dev \([^\ ]*\).*/\1/p' | head -n 1)
    [ -d "/$dest_os_dir/etc/network/" ] || mkdir -p "/$dest_os_dir/etc/network/"
    ipaddr_with_mask=$(ip addr show dev $dev | sed -nE '/global/s/.*inet (.+) brd.*$/\1/p' | head -n 1)
    hostname=$(hostname)
    route_part="$(ip route show default 0.0.0.0/0 | sed -E 's/^(.*dev [^ ]+).*$/\1/')"
    gateway_line="up ip route add $route_part"

    # manual save network
    if [ -f /etc/network/interfaces ] && grep static /etc/network/interfaces > /dev/null ; then
        cp -rf /etc/network/interfaces "/$dest_os_dir/etc/network/interfaces"
    else
        cat > "/$dest_os_dir/etc/network/interfaces" <<- EOF
			auto lo
			iface lo inet loopback

			auto $dev
			iface $dev inet static
			address $ipaddr_with_mask
			$gateway_line

			hostname $hostname
		EOF
    fi

    rm "/$dest_os_dir/etc/resolv.conf"
	cat > "/$dest_os_dir/etc/resolv.conf" <<- EOF
		nameserver 8.8.8.8
		nameserver 2001:4860:4860::8888
	EOF
}

function chroot_run(){
    if grep -qi alpine /x/etc/issue; then
        chroot "/x/" sh -c "[ -f /bin/bash ] || apk add bash"
    fi
    chroot "/x/" /bin/bash -c "$*"
}

change_hostname() {
    local new_hostname="$1"
    local os_release

    # Detect the Linux distribution
    if [ -f "/etc/alpine-release" ]; then
        os_release="alpine"
    elif [ -f "/etc/debian_version" ]; then
        os_release="debian"
    elif [ -f "/etc/centos-release" ]; then
        os_release="centos"
    else
        echo "Unsupported Linux distribution."
        return 1
    fi

    # Change hostname based on distribution
    case "$os_release" in
        "alpine")
            echo "$new_hostname" > /etc/hostname
            hostname -F /etc/hostname
            ;;
        "debian")
            echo "$new_hostname" > /etc/hostname
            hostnamectl set-hostname "$new_hostname"
            ;;
        "centos")
            echo "NETWORKING=yes" > /etc/sysconfig/network
            echo "HOSTNAME=$new_hostname" >> /etc/sysconfig/network
            hostnamectl set-hostname "$new_hostname"
            ;;
    esac

    echo "Hostname changed to $new_hostname. Please restart your system to apply changes."
}


function post_install(){
    export PATH="/usr/sbin:/usr/bin:/sbin:/bin"

    root_partition="$(df / | awk 'NR==2 {print $1}')"
    root_partition_type="$(df -T / | awk 'NR==2 {print $2}')"
    target_disk=$(echo "$root_partition" | sed 's/[0-9]*$//')
    echo "$root_partition /               $root_partition_type   defaults    0       1" > /etc/fstab
    mkdir -p /boot/grub

    if grep -qi alpine /etc/issue; then
        install openssh bash
        rc-update add sshd default
        rc-update add mdev sysinit
        rc-update add devfs sysinit
        apk add ifupdown-ng
        rc-update add networking default
        sed -i 's/--auto/-a/' /etc/init.d/networking # fix bug in networking script of lxc
        change_hostname alpine
        install grub grub-bios linux-lts
        echo 'GRUB_CMDLINE_LINUX_DEFAULT="quiet rootfstype=ext4 modules=sd-mod,usb-storage,ext4"' >> /etc/default/grub
        grub-mkconfig -o /boot/grub/grub.cfg
    elif grep -qi debian /etc/issue; then
        install ssh ifupdown
        systemctl disable systemd-networkd.service
        change_hostname debian
        install fdisk linux-image-generic grub-pc
        grub-install "$target_disk"
        update-grub
        #upgrade-from-grub-legacy
    elif grep -qi centos /etc/issue; then
        install openssh ifupdown
        systemctl disable systemd-networkd.service
        change_hostname centos
        install fdisk linux-image-generic grub-pc
        grub-install "$target_disk"
        update-grub
        #upgrade-from-grub-legacy
    fi
    echo PermitRootLogin yes >> /etc/ssh/sshd_config
    rm -rf /x
    sync
}

function install_requirement(){
    if [ -n "$(command -v apk)" ] ; then
        install curl sed gawk wget gzip xz tar virt-what
    else
        install curl sed gawk wget gzip xz-utils virt-what
    fi
}

function clean_old_system(){
    pkill -9 -f systemd
    # and all remaining process using kill -9 PID PID PID PID PID PID ... EXCEPT the SSH you're actually using!
    # 这里好像还需要清理其他进程才行
    while true; do
        (mount | tac | grep oldroot | cut -d' ' -f3 | xargs umount) || break
    done
}

function main(){
    print_help
    echo -e '\e[1;32minstall requirement...\e[m'
    install_requirement

    read_lxc_template

    echo -e '\e[1;32mClean old system\e[m'
    clean_old_system

    echo -e '\e[1;32mdownloading template...\e[m'
    download_rootfs

    echo -e '\e[1;32mmigrating configuration\e[m'
    migrate_configuration oldroot
}

function make_temp_os(){
    # prevent no access on ipv6 only vps
    ping -c 3 api.github.com || echo "nameserver 2a00:1098:2c::1"  >  /etc/resolv.conf 
    
    temp_os_url='https://github.com/LloydAsp/templates/releases/download/v1.0.0/alpine-takeover.tar.gz'
    mkdir -p /x
    mount -t tmpfs tmpfs /x -o size=150M
    if [ -n "$(command -v wget)" ] ; then
        wget -qO- $temp_os_url | tar -C /x -xz
    elif [ -n "$(command -v curl)" ] ; then
        curl -sL $temp_os_url | tar -C /x -xz
    else
        install wget
        wget -qO- $temp_os_url | tar -C /x -xz
    fi
    
    mv /x/takeover/* /x

    # backup configuration
    migrate_configuration x

    # don't run follow
    cp OsMutationKvm.sh /x/
    cp OsMutationKvm.sh /

    cd /x
    curl -qo takeover.sh https://raw.githubusercontent.com/LloydAsp/OsMutation/main/takeover.sh
    chmod u+x takeover.sh
    ./takeover.sh
    # Note: disk entry "/" has changed now.

    /busybox chroot . apk add bash
    /busybox chroot . env main=1 /OsMutationKvm.sh

    echo -e '\e[1;32mpost processing...\e[m'
    /busybox cp /OsMutationKvm.sh /oldroot
    /busybox chroot /oldroot apk add bash &> /dev/null # for alpine
    /busybox chroot /oldroot env postinstall=1 /OsMutationKvm.sh

    while [ "$reboot_ans" != 'yes' -a "$reboot_ans" != 'no' ] ; do
        echo -ne "\e[1;33mreboot now? (yes/no):\e[m"
        read reboot_ans
    done

    if [ "$reboot_ans" == 'yes' ] ; then
        reboot -f
    fi
}


if [ -n "$main" ]; then
    main
    exit 0
fi

if [ -n "$postinstall" ]; then
    post_install 2>&1 | tee reinstall.log
    exit 0
fi


make_temp_os 2>&1 | tee reinstall.log