core: 引入 img_type_warp

This commit is contained in:
bin456789 2024-09-06 23:00:32 +08:00
parent b9219cc0c7
commit 3364eb9e2d
No known key found for this signature in database
GPG Key ID: EE301B386DE6C11B
2 changed files with 89 additions and 76 deletions

View File

@ -234,7 +234,7 @@ test_url_grace() {
test_url_real() { test_url_real() {
grace=$1 grace=$1
url=$2 url=$2
expect_type=$3 expect_type_combo=$3
var_to_eval=$4 var_to_eval=$4
info test url info test url
@ -247,13 +247,14 @@ test_url_real() {
# TODO: 好像无法识别 nixos 官方源的跳转 # TODO: 好像无法识别 nixos 官方源的跳转
# 有的服务器不支持 rangecurl会下载整个文件 # 有的服务器不支持 rangecurl会下载整个文件
# 用 dd 限制下载 1M # 所以用 head 限制 1M
# 并过滤 curl 23 错误dd限制了空间 # 过滤 curl 23 错误head 限制了大小
# 也可用 ulimit -f 但好像 cygwin 不支持 # 也可用 ulimit -f 但好像 cygwin 不支持
# ${PIPESTATUS[n]} 表示第n个管道的返回值
echo $url echo $url
for i in $(seq 5 -1 0); do for i in $(seq 5 -1 0); do
if command curl --insecure --connect-timeout 10 -Lfr 0-1048575 "$url" \ if command curl --insecure --connect-timeout 10 -Lfr 0-1048575 "$url" \
1> >(exec dd bs=1M count=1 of=$tmp_file iflag=fullblock 2>/dev/null) \ 1> >(exec head -c 1048576 >$tmp_file) \
2> >(exec grep -v 'curl: (23)' >&2); then 2> >(exec grep -v 'curl: (23)' >&2); then
break break
else else
@ -268,19 +269,54 @@ test_url_real() {
fi fi
done done
if [ -n "$expect_type" ]; then if [ -n "$expect_type_combo" ]; then
# gzip的mime有很多种写法 fix_file_type() {
# centos7中显示为 x-gzip在其他系统中显示为 gzip可能还有其他 # gzip的mime有很多种写法
# 所以不用mime判断 # centos7中显示为 x-gzip在其他系统中显示为 gzip可能还有其他
# https://www.digipres.org/formats/sources/tika/formats/#application/gzip # 所以不用mime判断
# https://www.digipres.org/formats/sources/tika/formats/#application/gzip
# 有些 file 版本输出的是 # ISO 9660 CD-ROM filesystem data ,要去掉开头的井号
# 下面两种都是 raw
# DOS/MBR boot sector
# x86 boot sector; partition 1: ...
sed 's/^# //' | cut -d' ' -f1 | to_lower |
sed -e 's,dos/mbr,raw,' -e 's,x86,raw,'
}
file_enhanced() {
file_type=$(file -b $tmp_file | fix_file_type)
if [ "$file_type" = "xz" ] || [ "$file_type" = "gzip" ]; then
# 要安装 xz 或者 gzip不然会报错
# ERROR:[xz: Wait failed, No child process]
install_pkg "$file_type"
# 加 if 是为了避免以下情况外面是xz但是识别不到里面的东西即使装了xz,
# 即使 file 报错返回值也是 0
# [root@localhost ~]# file -bZ /reinstall-tmp/img-test
# ERROR:[xz: Unexpected end of input]
if file_type_in="$(file -bZ $tmp_file | fix_file_type)" &&
! grep -iq "^Error" <<<"$file_type_in"; then
file_type="$file_type_in.$file_type"
fi
fi
echo "$file_type"
}
# 有些 file 版本输出的是 # ISO 9660 CD-ROM filesystem data ,要去掉开头的井号
install_pkg file install_pkg file
real_type=$(file -b $tmp_file | sed 's/^# //' | cut -d' ' -f1 | to_lower) real_type_combo=$(file_enhanced $tmp_file)
[ -n "$var_to_eval" ] && eval $var_to_eval=$real_type echo "$real_type_combo"
if ! grep -wo "$real_type" <<<"$expect_type"; then if ! grep -Foq "|$real_type_combo|" <<<"|$expect_type_combo|"; then
failed "$url expected: $expect_type. actual: $real_type." failed "$url expected: $expect_type_combo. actual: $real_type_combo."
fi
IFS=. read -r real_type real_type_warp <<<"$real_type_combo"
if [ -n "$var_to_eval" ]; then
eval "$var_to_eval=$real_type"
eval "$var_to_eval""_warp=$real_type_warp"
fi fi
fi fi
} }
@ -934,7 +970,7 @@ setos() {
filename=$(curl -L $mirror | grep -oP "ubuntu-$releasever.*?-live-server-$basearch_alt.iso" | head -1) filename=$(curl -L $mirror | grep -oP "ubuntu-$releasever.*?-live-server-$basearch_alt.iso" | head -1)
iso=$mirror/$filename iso=$mirror/$filename
# 在 ubuntu 20.04 上file 命令检测 ubuntu 22.04 iso 结果是 DOS/MBR boot sector # 在 ubuntu 20.04 上file 命令检测 ubuntu 22.04 iso 结果是 DOS/MBR boot sector
test_url $iso 'iso|dos/mbr' test_url $iso 'iso|raw'
eval ${step}_iso=$iso eval ${step}_iso=$iso
# ks # ks
@ -985,7 +1021,7 @@ setos() {
# 传统安装 # 传统安装
# 该服务器文件缓存 miss 时会响应 206 + Location 头 # 该服务器文件缓存 miss 时会响应 206 + Location 头
# 但 curl 这种情况不会重定向,所以添加 ascii 类型让它不要报错 # 但 curl 这种情况不会重定向,所以添加 ascii 类型让它不要报错
test_url $mirror/nixos-$releasever/store-paths.xz xz/ascii test_url $mirror/nixos-$releasever/store-paths.xz 'xz|ascii'
eval ${step}_mirror=$mirror eval ${step}_mirror=$mirror
fi fi
} }
@ -1061,38 +1097,30 @@ setos() {
# 注意 windows server 2008 r2 serverdatacenter 不用改 # 注意 windows server 2008 r2 serverdatacenter 不用改
image_name=${image_name/windows server 2008 server/windows longhorn server} image_name=${image_name/windows server 2008 server/windows longhorn server}
test_url $iso 'iso|dos/mbr' test_url $iso 'iso|raw'
eval "${step}_iso='$iso'" eval "${step}_iso='$iso'"
eval "${step}_image_name='$image_name'" eval "${step}_image_name='$image_name'"
} }
# shellcheck disable=SC2154 # shellcheck disable=SC2154
setos_dd() { setos_dd() {
# 下面两种都是 raw # raw 包含 vhd
# DOS/MBR boot sector test_url $img 'raw|raw.gzip|raw.xz' img_type
# x86 boot sector; partition 1: ...
test_url $img 'xz|gzip|dos/mbr|x86' img_type
# 修正 raw 的 img_type
if [ "$img_type" = dos/mbr ] || [ "$img_type" = x86 ]; then
img_type=raw
fi
if is_efi; then if is_efi; then
install_pkg hexdump install_pkg hexdump
if ! [ "$img_type" = raw ]; then
install_pkg $img_type
fi
extract() { extract() {
if [ "$img_type" = raw ]; then case "$img_type_warp" in
cat "$1" '') cat "$1" ;;
else xz | gzip)
install_pkg $img_type_warp
# xz/gzip -d 文件必须有正确的扩展名,否则报扩展名错误 # xz/gzip -d 文件必须有正确的扩展名,否则报扩展名错误
# 因此用 stdin # 因此用 stdin
$img_type -dc <"$1" "$img_type_warp" -dc <"$1"
fi ;;
*) error_and_exit "warp type $img_type_warp not support." ;;
esac
} }
# openwrt 镜像 efi part type 不是 esp # openwrt 镜像 efi part type 不是 esp
@ -1126,6 +1154,7 @@ Continue with DD?
fi fi
eval "${step}_img='$img'" eval "${step}_img='$img'"
eval "${step}_img_type='$img_type'" eval "${step}_img_type='$img_type'"
eval "${step}_img_type_warp='$img_type_warp'"
} }
setos_centos_alma_rocky_fedora() { setos_centos_alma_rocky_fedora() {
@ -1293,13 +1322,7 @@ Continue with DD?
# 集中测试云镜像格式 # 集中测试云镜像格式
if is_use_cloud_image && [ "$step" = finalos ]; then if is_use_cloud_image && [ "$step" = finalos ]; then
# shellcheck disable=SC2154 # shellcheck disable=SC2154
test_url $finalos_img 'xz|gzip|qemu' finalos_img_type test_url $finalos_img 'qemu|qemu.gzip|qemu.xz' finalos_img_type
# openeuler 云镜像格式是 .qcow2.xz
if [ "$distro" = openeuler ]; then
# shellcheck disable=SC2034
finalos_img_type=qemu
fi
fi fi
} }

View File

@ -1649,39 +1649,26 @@ get_http_file_size_to() {
var_name=$1 var_name=$1
url=$2 url=$2
size='' pipe_extract() {
if wget --spider -S $url -o /tmp/headers.log; then # alpine busybox 自带 gzip但官方版也许性能更好
# 网址重定向可能得到多个 Content-Length, 选最后一个 case "$img_type_warp" in
if size=$(grep 'Content-Length:' /tmp/headers.log | xz | gzip) apk add $img_type_warp ;;
tail -1 | awk '{print $2}' | grep .); then '') ;;
eval "$var_name='$size'" *) error_and_exit "Not supported img_type_warp: $img_type_warp" ;;
fi esac
if [ -n "$img_type_warp" ]; then
"$img_type_warp" -dc
else else
error_and_exit "Can't access $url" cat
fi fi
} }
dd_gzip_xz_raw() { dd_gzip_xz_raw() {
# 用官方 wget一来带进度条二来自带重试 # 用官方 wget一来带进度条二来自带重试功能
# alpine busybox 自带 gzip xz但官方版也许性能更好 apk add wget
# raw 包括 vhd
# shellcheck disable=SC2154
case "$img_type" in
xz | gzip) apk add wget $img_type ;;
raw) apk add wget ;;
*) error_and_exit 'Not supported' ;;
esac
pipe_extract() { if ! wget $img -O- | pipe_extract >/dev/$xda 2>/tmp/dd_stderr; then
if [ "$img_type" = raw ]; then
cat
else
$img_type -dc
fi
}
if ! command wget $img -O- --tries=5 --progress=bar:force |
pipe_extract >/dev/$xda 2>/tmp/dd_stderr; then
# vhd 文件结尾有 512 字节额外信息,可以忽略 # vhd 文件结尾有 512 字节额外信息,可以忽略
if grep -iq 'No space' /tmp/dd_stderr; then if grep -iq 'No space' /tmp/dd_stderr; then
apk add parted apk add parted
@ -2536,11 +2523,13 @@ download_qcow() {
mount /dev/disk/by-label/installer /installer mount /dev/disk/by-label/installer /installer
qcow_file=/installer/cloud_image.qcow2 qcow_file=/installer/cloud_image.qcow2
if [ "$distro" = openeuler ]; then if [ -n "$img_type_warp" ]; then
prog=xz # 边下载边解压,单线程下载
apk add wget $prog # 用官方 wget ,带进度条
command wget $img -O- --tries=5 --progress=bar:force | $prog -dc >$qcow_file apk add wget
wget $img -O- | pipe_extract >$qcow_file
else else
# 多线程下载
download "$img" "$qcow_file" download "$img" "$qcow_file"
fi fi
} }
@ -4077,6 +4066,7 @@ trans() {
fi fi
# dd qemu 切换成云镜像模式,暂时没用到 # dd qemu 切换成云镜像模式,暂时没用到
# shellcheck disable=SC2154
if [ "$distro" = "dd" ] && [ "$img_type" = "qemu" ]; then if [ "$distro" = "dd" ] && [ "$img_type" = "qemu" ]; then
# 移到 reinstall.sh ? # 移到 reinstall.sh ?
distro=any distro=any
@ -4105,8 +4095,8 @@ trans() {
;; ;;
esac esac
;; ;;
gzip | xz | raw) raw)
# 暂时没用到 gzip xz 格式的云镜像 # 暂时没用到 raw 格式的云镜像
dd_gzip_xz_raw dd_gzip_xz_raw
resize_after_install_cloud_image resize_after_install_cloud_image
modify_os_on_disk linux modify_os_on_disk linux
@ -4114,7 +4104,7 @@ trans() {
esac esac
elif [ "$distro" = "dd" ]; then elif [ "$distro" = "dd" ]; then
case "$img_type" in case "$img_type" in
gzip | xz | raw) raw)
dd_gzip_xz_raw dd_gzip_xz_raw
modify_os_on_disk windows modify_os_on_disk windows
;; ;;