core: 重构

reinstall.sh

@@ -1,10 +1,20 @@
 #!/bin/bash
+# shellcheck disable=SC2086
+
 set -eE
 confhome=https://raw.githubusercontent.com/bin456789/reinstall/main
 localtest_confhome=http://192.168.253.1
 
-trap 'error line $LINENO return $?' ERR
+this_script=$(realpath "$0")
-this_script=$(realpath $0)
+trap 'trap_err $LINENO $?' ERR
+
+trap_err() {
+    line_no=$1
+    ret_no=$2
+
+    error "Line $line_no return $ret_no"
+    sed -n "$line_no"p "$this_script"
+}
 
 usage_and_exit() {
     cat <<EOF
@@ -24,20 +34,23 @@ EOF
 }
 
 info() {
-    color='\e[32m'
+    upper=$(echo "$*" | tr '[:lower:]' '[:upper:]')
-    plain='\e[0m'
+    echo_color_text '\e[32m' "***** $upper *****"
-    upper=$(echo "$@" | tr '[:lower:]' '[:upper:]')
+}
-    echo -e "$color***** $upper *****$plain"
+
+warn() {
+    echo_color_text '\e[33m' "Warn: $*"
 }
 
 error() {
-    color='\e[31m'
+    echo_color_text '\e[31m' "Error: $*"
-    plain='\e[0m'
+}
-    echo -e "${color}Error: $*${plain}"
+
-    # 如果从trap调用，显示错误行
+echo_color_text() {
-    if [ "$1" = line ]; then
+    color="$1"
-        sed -n "$2"p $this_script
+    shift
-    fi
+    plain="\e[0m"
+    echo -e "$color$*$plain"
 }
 
 error_and_exit() {
@@ -68,10 +81,6 @@ is_in_alpine() {
     [ -f /etc/alpine-release ]
 }
 
-is_in_arch() {
-    [ -f /etc/arch-release ]
-}
-
 is_use_cloud_image() {
     [ -n "$cloud_image" ] && [ "$cloud_image" = 1 ]
 }
@@ -116,6 +125,21 @@ get_host_by_url() {
     cut -d/ -f3 <<<$1
 }
 
+insert_into_file() {
+    file=$1
+    location=$2
+    regex_to_find=$3
+
+    line_num=$(grep -E -n "$regex_to_find" "$file" | cut -d: -f1)
+    if [ "$location" = before ]; then
+        line_num=$((line_num - 1))
+    elif ! [ "$location" = after ]; then
+        return 1
+    fi
+
+    sed -i "${line_num}r /dev/stdin" "$file"
+}
+
 test_url() {
     url=$1
     expect_type=$2
@@ -991,8 +1015,7 @@ mod_alpine_initrd() {
     fi
 
     # hack 1 添加 ipv6 模块
-    line_num=$(grep -E -n 'configure_ip\(\)' init | cut -d: -f1)
+    insert_into_file init after 'configure_ip\(\)' <<EOF
-    cat <<EOF | sed -i "${line_num}r /dev/stdin" init
         depmod
         modprobe ipv6
 EOF
@@ -1013,8 +1036,7 @@ EOF
     # udhcpc:  bound
     # udhcpc6: deconfig
     # udhcpc6: bound
-    line_num=$(grep -E -n 'deconfig\|renew\|bound' usr/share/udhcpc/default.script | cut -d: -f1)
+    insert_into_file usr/share/udhcpc/default.script after 'deconfig\|renew\|bound' <<EOF
-    cat <<EOF | sed -i "${line_num}r /dev/stdin" usr/share/udhcpc/default.script
         if [ "\$1" = deconfig ]; then
             return
         fi
@@ -1028,8 +1050,7 @@ EOF
     # hack 4 网络配置
     collect_netconf
     is_in_china && is_in_china=true || is_in_china=false
-    line_num=$(grep -E -n 'MAC_ADDRESS=' init | cut -d: -f1)
+    insert_into_file init after 'MAC_ADDRESS=' <<EOF
-    cat <<EOF | sed -i "${line_num}r /dev/stdin" init
         source /alpine-network.sh \
         "$mac_addr" "$ipv4_addr" "$ipv4_gateway" "$ipv6_addr" "$ipv6_gateway" "$is_in_china"
 EOF
@@ -1037,15 +1058,13 @@ EOF
     # hack 5 运行 trans.start
     # exec /bin/busybox switch_root $switch_root_opts $sysroot $chart_init "$KOPT_init" $KOPT_init_args # 3.17
     # exec              switch_root $switch_root_opts $sysroot $chart_init "$KOPT_init" $KOPT_init_args # 3.18
-    line_num=$(grep -E -n '^exec (/bin/busybox )?switch_root' init | cut -d: -f1)
-    line_num=$((line_num - 1))
     # 1. alpine arm initramfs 时间问题 要添加 --no-check-certificate
     # 2. aws t4g arm 如果没设置console=ttyx，在initramfs里面wget https会出现bad header错误，chroot后正常
     # Connecting to raw.githubusercontent.com (185.199.108.133:443)
     # 60C0BB2FFAFF0000:error:0A00009C:SSL routines:ssl3_get_record:http request:ssl/record/ssl3_record.c:345:
     # ssl_client: SSL_connect
     # wget: bad header line: <20>
-    cat <<EOF | sed -i "${line_num}r /dev/stdin" init
+    insert_into_file init before '^exec (/bin/busybox )?switch_root' <<EOF
         # echo "wget --no-check-certificate -O- $confhome/trans.sh | /bin/ash" >\$sysroot/etc/local.d/trans.start
         # wget --no-check-certificate -O \$sysroot/etc/local.d/trans.start $confhome/trans.sh
         cp /trans.start \$sysroot/etc/local.d/trans.start

trans.sh

@@ -1,16 +1,30 @@
 #!/bin/ash
 # shellcheck shell=dash
-# shellcheck disable=SC3047,SC3036,SC3010,SC3001
+# shellcheck disable=SC2086,SC3047,SC3036,SC3010,SC3001
 # alpine 默认使用 busybox ash
 
 # 命令出错终止运行，将进入到登录界面，防止失联
 set -eE
-trap 'error line $LINENO return $?' ERR
+trap 'trap_err $LINENO $?' ERR
-this_script=$(realpath $0)
 
-catch() {
+# 复制本脚本到 /tmp/trans.sh，用于打印错误
-    if [ "$1" != "0" ]; then
+# 也有可能从管道运行，这时删除 /tmp/trans.sh
-        error "Error $1 occurred on $2"
+case "$0" in
+*trans.*) cp -f "$0" /tmp/trans.sh ;;
+*) rm -f /tmp/trans.sh ;;
+esac
+
+# 还原改动，不然本脚本会被复制到新系统
+rm -f /etc/local.d/trans.start
+rm -f /etc/runlevels/default/local
+
+trap_err() {
+    line_no=$1
+    ret_no=$2
+
+    error "Line $line_no return $ret_no"
+    if [ -f "/tmp/trans.sh" ]; then
+        sed -n "$line_no"p /tmp/trans.sh
     fi
 }
 
@@ -18,10 +32,6 @@ error() {
     color='\e[31m'
     plain='\e[0m'
     echo -e "${color}Error: $*${plain}"
-    # 如果从trap调用，显示错误行
-    if [ "$1" = line ]; then
-        sed -n "$2"p $this_script
-    fi
 }
 
 error_and_exit() {
@@ -42,11 +52,6 @@ add_community_repo() {
     fi
 }
 
-cp() {
-    # 防止 alias cp='cp -i'
-    command cp "$@"
-}
-
 download() {
     url=$1
     file=$2
@@ -146,6 +151,7 @@ setup_lighttpd() {
 # 所以如果有显示器且有鼠标，tty0 放最后面，否则 tty0 放前面
 get_ttys() {
     prefix=$1
+    # shellcheck disable=SC2154
     wget $confhome/ttys.sh -O- | sh -s $prefix
 }
 
@@ -270,6 +276,21 @@ is_dhcpv6() {
     [ "$dhcpv6" = 1 ]
 }
 
+insert_into_file() {
+    file=$1
+    location=$2
+    regex_to_find=$3
+
+    line_num=$(grep -E -n "$regex_to_find" "$file" | cut -d: -f1)
+    if [ "$location" = before ]; then
+        line_num=$((line_num - 1))
+    elif ! [ "$location" = after ]; then
+        return 1
+    fi
+
+    sed -i "${line_num}r /dev/stdin" "$file"
+}
+
 install_alpine() {
     # 网络
     # 坑1 udhcpc下，ip -4 addr 无法知道是否是 dhcp
@@ -392,10 +413,6 @@ EOF
     true >/etc/apk/repositories
     setup-apkrepos -1
 
-    # 还原改动，不然本脚本会被复制到新系统
-    rm -f /etc/local.d/trans.start
-    rm -f /etc/runlevels/default/local
-
     # 安装到硬盘
     # alpine默认使用 syslinux (efi 环境除外)，这里强制使用 grub，方便用脚本再次重装
     KERNELOPTS="$(get_ttys console=)"
@@ -448,6 +465,7 @@ create_part() {
     wipefs -a /dev/$xda
 
     # xda*1 星号用于 nvme0n1p1 的字母 p
+    # shellcheck disable=SC2154
     if [ "$distro" = windows ]; then
         apk add ntfs-3g-progs virt-what wimlib rsync dos2unix
         # 虽然ntfs3不需要fuse，但wimmount需要，所以还是要保留
@@ -465,6 +483,7 @@ create_part() {
                 set 2 msftres on \
                 set 3 msftdata on
             update_part /dev/$xda
+
             mkfs.fat -n efi /dev/$xda*1              #1 efi
             echo                                     #2 msr
             mkfs.ext4 -F -L os /dev/$xda*3           #3 os
@@ -477,10 +496,13 @@ create_part() {
                 mkpart primary ntfs -6GiB 100% \
                 set 1 boot on
             update_part /dev/$xda
+
             mkfs.ext4 -F -L os /dev/$xda*1           #1 os
             mkfs.ntfs -f -F -L installer /dev/$xda*2 #2 installer
         fi
-    elif is_use_cloud_image && { [ "$distro" = centos ] || [ "$distro" = alma ] || [ "$distro" = rocky ]; }; then
+    elif is_use_cloud_image; then
+        # 这几个系统不使用dd，而是复制文件，因为dd这几个系统的qcow2需要10g硬盘
+        if { [ "$distro" = centos ] || [ "$distro" = alma ] || [ "$distro" = rocky ]; }; then
             apk add dosfstools e2fsprogs
             if is_efi; then
                 parted /dev/$xda -s -- \
@@ -490,6 +512,7 @@ create_part() {
                     mkpart '" "' ext4 -2GiB 100% \
                     set 1 esp on
                 update_part /dev/$xda
+
                 mkfs.fat -n efi /dev/$xda*1           #1 efi
                 echo                                  #2 os 用目标系统的格式化工具
                 mkfs.ext4 -F -L installer /dev/$xda*3 #3 installer
@@ -501,18 +524,12 @@ create_part() {
                     mkpart '" "' ext4 -2GiB 100% \
                     set 1 bios_grub on
                 update_part /dev/$xda
+
                 echo                                  #1 bios_boot
                 echo                                  #2 os 用目标系统的格式化工具
                 mkfs.ext4 -F -L installer /dev/$xda*3 #3 installer
             fi
-    elif is_use_cloud_image; then
+        fi
-        parted /dev/$xda -s -- \
-            mklabel gpt \
-            mkpart '" "' ext4 1MiB -2GiB \
-            mkpart '" "' ext4 -2GiB 100%
-        update_part /dev/$xda
-        mkfs.ext4 -F -L os /dev/$xda*1        #1 os
-        mkfs.ext4 -F -L installer /dev/$xda*2 #2 installer
     else
         # 安装红帽系或ubuntu
         # 对于红帽系是临时分区表，安装时除了 installer 分区，其他分区会重建为默认的大小
@@ -528,6 +545,7 @@ create_part() {
                 mkpart '" "' ext4 -2GiB 100% \
                 set 1 boot on
             update_part /dev/$xda
+
             mkfs.fat -n efi /dev/$xda*1       #1 efi
             mkfs.ext4 -F -L os /dev/$xda*2    #2 os
             mkfs.fat -n installer /dev/$xda*3 #3 installer
@@ -540,6 +558,7 @@ create_part() {
                 mkpart '" "' ext4 -2GiB 100% \
                 set 1 bios_grub on
             update_part /dev/$xda
+
             echo                              #1 bios_boot
             mkfs.ext4 -F -L os /dev/$xda*2    #2 os
             mkfs.fat -n installer /dev/$xda*3 #3 installer
@@ -551,6 +570,7 @@ create_part() {
                 mkpart primary ext4 -2GiB 100% \
                 set 1 boot on
             update_part /dev/$xda
+
             mkfs.ext4 -F -L os /dev/$xda*1    #1 os
             mkfs.fat -n installer /dev/$xda*2 #2 installer
         fi
@@ -571,17 +591,13 @@ create_part() {
 mount_pseudo_fs() {
     os_dir=$1
 
-    if [[ "$os_dir" != "*/" ]]; then
-        os_dir=$os_dir/
-    fi
-
     # https://wiki.archlinux.org/title/Chroot#Using_chroot
-    mount -t proc /proc ${os_dir}proc/
+    mount -t proc /proc $os_dir/proc/
-    mount -t sysfs /sys ${os_dir}sys/
+    mount -t sysfs /sys $os_dir/sys/
-    mount --rbind /dev ${os_dir}dev/
+    mount --rbind /dev $os_dir/dev/
-    mount --rbind /run ${os_dir}run/
+    mount --rbind /run $os_dir/run/
     if is_efi; then
-        mount --rbind /sys/firmware/efi/efivars ${os_dir}sys/firmware/efi/efivars/
+        mount --rbind /sys/firmware/efi/efivars $os_dir/sys/firmware/efi/efivars/
     fi
 }
 
@@ -668,8 +684,7 @@ download_cloud_init_config() {
     if ! grep -w swap $os_dir/etc/fstab; then
         # btrfs
         if mount | grep 'on /os type btrfs'; then
-            line_num=$(grep -E -n '^runcmd:' $ci_file | cut -d: -f1)
+            insert_into_file $ci_file after '^runcmd:' <<EOF
-            cat <<EOF | sed -i "${line_num}r /dev/stdin" $ci_file
   - btrfs filesystem mkswapfile --size 1G /swapfile
   - swapon /swapfile
   - echo "/swapfile none swap defaults 0 0" >> /etc/fstab
@@ -719,11 +734,15 @@ modify_dd_os() {
 
     download_cloud_init_config $os_dir
 
+    # 为红帽系禁用 selinux kdump
     if [ -f $os_dir/etc/redhat-release ]; then
         find_and_mount /boot
         find_and_mount /boot/efi
         disable_selinux_kdump $os_dir
-    elif grep opensuse-tumbleweed $os_dir/etc/os-release; then
+    fi
+
+    # opensuse tumbleweed 需安装 wicked
+    if grep opensuse-tumbleweed $os_dir/etc/os-release; then
         cp -f /etc/resolv.conf $os_dir/etc/resolv.conf
         mount_pseudo_fs $os_dir
         chroot $os_dir zypper install -y wicked
@@ -1007,7 +1026,7 @@ EOF
             fi
 
             # 显示分区
-            lsblk /dev/nbd0
+            lsblk -o NAME,SIZE,FSTYPE,LABEL /dev/nbd0
 
             # 将前1M dd到内存
             dd if=/dev/nbd0 of=/first-1M bs=1M count=1
@@ -1244,10 +1263,9 @@ install_windows() {
     sed -i "s|%arch%|$arch|; s|%image_name%|$image_name|; s|%locale%|$locale|" /tmp/Autounattend.xml
 
     # 修改应答文件，分区配置
-    line_num=$(grep -E -n '<ModifyPartitions>' /tmp/Autounattend.xml | cut -d: -f1)
     if is_efi; then
         sed -i "s|%installto_partitionid%|3|" /tmp/Autounattend.xml
-        cat <<EOF | sed -i "${line_num}r /dev/stdin" /tmp/Autounattend.xml
+        insert_into_file /tmp/Autounattend.xml after '<ModifyPartitions>' <<EOF
             <ModifyPartition wcm:action="add">
                 <Order>1</Order>
                 <PartitionID>1</PartitionID>
@@ -1265,7 +1283,7 @@ install_windows() {
 EOF
     else
         sed -i "s|%installto_partitionid%|1|" /tmp/Autounattend.xml
-        cat <<EOF | sed -i "${line_num}r /dev/stdin" /tmp/Autounattend.xml
+        insert_into_file /tmp/Autounattend.xml after '<ModifyPartitions>' <<EOF
             <ModifyPartition wcm:action="add">
                 <Order>1</Order>
                 <PartitionID>1</PartitionID>