Skip to content

magisk模块添加abl分区备份与恢复功能 #35

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
liujixings wants to merge 1 commit into from
+403 −16
Closed

magisk模块添加abl分区备份与恢复功能 #35

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
281 changes: 279 additions & 2 deletions magisk_module/bin/bl_flasher.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ if [ -z "$MODDIR" ]; then
fi
RUNTIME_DIR="$MODDIR/tmp"
BY_NAME_DIR="/dev/block/by-name"
BACKUP_DIR="/data/adb/abl_backup"
IMAGE_NAMES="abl"
LOG_FILE="$RUNTIME_DIR/flash.log"
STATE_FILE="$RUNTIME_DIR/state"
Expand Down Expand Up @@ -43,6 +44,158 @@ write_log() {
printf '[%s] %s\n' "$(timestamp)" "$*" >> "$LOG_FILE"
}

ensure_backup_dir() {
if mkdir -p "$BACKUP_DIR" 2>/dev/null; then
return 0
fi
write_log "无法创建备份目录: $BACKUP_DIR"
return 1
}

file_hash() {
_file="$1"
if command -v sha256sum >/dev/null 2>&1; then
_out=$(sha256sum "$_file" 2>/dev/null) || return 1
printf '%s\n' "${_out%% *}"
return 0
fi
if command -v md5sum >/dev/null 2>&1; then
_out=$(md5sum "$_file" 2>/dev/null) || return 1
printf '%s\n' "${_out%% *}"
return 0
fi
if command -v cksum >/dev/null 2>&1; then
_out=$(cksum "$_file" 2>/dev/null) || return 1
printf '%s\n' "${_out%% *}"
return 0
fi
return 1
}

verify_partition_backup_hash() {
_source="$1"
_backup="$2"

_source_hash=$(file_hash "$_source" | tr -d '[:space:]') || {
write_log "备份哈希校验失败：无法计算源分区哈希 $_source"
return 1
}
_backup_hash=$(file_hash "$_backup" | tr -d '[:space:]') || {
write_log "备份哈希校验失败：无法计算备份文件哈希 $_backup"
return 1
}

if [ -z "$_source_hash" ] || [ -z "$_backup_hash" ]; then
write_log "备份哈希校验失败：哈希为空 source=$_source backup=$_backup"
return 1
fi

if [ "$_source_hash" != "$_backup_hash" ]; then
write_log "备份哈希校验失败：源分区与备份文件不一致 source=$_source backup=$_backup"
return 1
fi

write_log "备份哈希校验通过：$_backup"
printf '%s\n' "$_backup_hash"
return 0
}

backup_partition_image() {
part_path="$1"
part_name="$2"
tag="$3"

ensure_backup_dir || return 1

ts=$(date '+%Y%m%d-%H%M%S')
tmp_file="$BACKUP_DIR/${part_name}_${ts}_tmp_${tag}.img"

write_log "备份 $part_name 到 $tmp_file"
if ! dd if="$part_path" of="$tmp_file" bs=4M conv=fsync >> "$LOG_FILE" 2>&1; then
write_log "备份失败: $part_name"
rm -f "$tmp_file"
return 1
fi
sync

hash=$(verify_partition_backup_hash "$part_path" "$tmp_file" | tr -d '[:space:]' | cut -c1-16)
if [ -z "$hash" ]; then
write_log "备份失败：源分区与备份文件哈希校验未通过"
rm -f "$tmp_file"
return 1
fi
final_file="$BACKUP_DIR/${part_name}_${ts}_${hash}_${tag}.img"

if ! mv "$tmp_file" "$final_file"; then
write_log "备份重命名失败: $tmp_file"
rm -f "$tmp_file"
return 1
fi

write_log "备份完成: $final_file"
printf '%s\n' "$final_file"
return 0
}

latest_stored_abl() {
ls -t "$BACKUP_DIR"/abl_*_backup.img "$BACKUP_DIR"/abl_*_saved.img 2>/dev/null | head -n 1
}

latest_backup_abl() {
ls -t "$BACKUP_DIR"/abl_*_backup.img 2>/dev/null | head -n 1
}

latest_saved_abl() {
ls -t "$BACKUP_DIR"/abl_*_saved.img 2>/dev/null | head -n 1
}

prune_old_saved_backups() {
keep_file="$1"
for old_file in "$BACKUP_DIR"/abl_*_saved.img; do
[ -e "$old_file" ] || continue
[ "$old_file" = "$keep_file" ] && continue
rm -f "$old_file"
write_log "已删除旧 saved 备份: $old_file"
done
}

prune_old_backup_backups() {
keep_file="$1"
for old_file in "$BACKUP_DIR"/abl_*_backup.img; do
[ -e "$old_file" ] || continue
[ "$old_file" = "$keep_file" ] && continue
rm -f "$old_file"
write_log "已删除旧 backup 备份: $old_file"
done
}

short_hash() {
_file="$1"
file_hash "$_file" | tr -d '[:space:]' | cut -c1-16
}

hash_from_filename() {
_name=$(basename "$1")
_name=${_name%.img}
printf '%s\n' "$_name" | awk -F'_' '{print $(NF-1)}'
}

verify_image_hash() {
_file="$1"
_expected=$(hash_from_filename "$_file")
_actual=$(short_hash "$_file")
if [ -z "$_expected" ] || [ -z "$_actual" ]; then
write_log "哈希校验失败（空值）: $_file"
return 1
fi
if [ "$_expected" != "$_actual" ]; then
write_log "哈希校验失败: $_file expected=$_expected actual=$_actual"
return 1
fi
write_log "哈希校验通过: $_file ($_actual)"
return 0
}

detect_current_slot() {
case "$(getprop ro.boot.slot_suffix 2>/dev/null)" in
_a) printf '%s\n' '_a' ;;
Expand Down Expand Up @@ -129,6 +282,7 @@ cleanup_lock() { rm -rf "$LOCK_DIR"; rm -f "$PID_FILE"; }

print_status() {
ensure_runtime
ensure_backup_dir >/dev/null 2>&1 || true
current_slot=$(getprop ro.boot.slot_suffix 2>/dev/null)
case "$current_slot" in _a|_b) ;; *) current_slot='' ;; esac
target_slot=''
Expand All @@ -148,14 +302,24 @@ print_status() {
_state=''; [ -f "$STATE_FILE" ] && read -r _state < "$STATE_FILE"
_msg=''; [ -f "$MESSAGE_FILE" ] && read -r _msg < "$MESSAGE_FILE"
_upd=''; [ -f "$UPDATED_FILE" ] && read -r _upd < "$UPDATED_FILE"
latest_backup=''; latest_backup=$(latest_backup_abl)
latest_saved=''; latest_saved=$(latest_saved_abl)
has_backup='0'
has_saved='0'
[ -n "$latest_backup" ] && has_backup='1'
[ -n "$latest_saved" ] && has_saved='1'

_out="CURRENT_SLOT=${current_slot}
TARGET_SLOT=${target_slot}
RUNNING=${running}
PID=${pid}
STATE=${_state}
MESSAGE=${_msg}
UPDATED_AT=${_upd}"
UPDATED_AT=${_upd}
HAS_BACKUP=${has_backup}
LATEST_BACKUP=${latest_backup}
HAS_SAVED=${has_saved}
LATEST_SAVED=${latest_saved}"

emit "$_out"
}
Expand Down Expand Up @@ -184,6 +348,15 @@ run_flash() {
write_log "当前槽位: $current_slot 目标槽位: $target_slot"
efisp_failed='0'
abl_part=$(partition_path abl "$target_slot")
source_abl_part=$(partition_path abl "$current_slot")

new_backup_file=$(backup_partition_image "$source_abl_part" "abl" 'backup')
if [ -z "$new_backup_file" ]; then
write_state 'error' '自动备份当前槽位 abl 失败'
write_log '自动备份当前槽位 abl 失败，已中止刷写'
exit 1
fi
prune_old_backup_backups "$new_backup_file"

if [ "$update_efisp" = 'update-efisp' ] || [ "$update_efisp" = '1' ] || [ "$update_efisp" = 'true' ]; then
#patch abl and flash efisp first, since it's the only one that can brick the device if something goes wrong
Expand Down Expand Up @@ -220,6 +393,21 @@ run_flash() {
for name in $IMAGE_NAMES; do
part=$(partition_path "$name" "$target_slot")
srcpart=$(partition_path "$name" "$current_slot")
used_saved=''
if [ "$name" = 'abl' ]; then
saved_img=$(latest_saved_abl)
if [ -n "$saved_img" ] && [ -f "$saved_img" ]; then
if ! verify_image_hash "$saved_img"; then
write_state 'error' 'saved 备份哈希校验失败，已中止刷写'
exit 1
fi
srcpart="$saved_img"
used_saved="$saved_img"
write_log "刷写使用 saved 镜像: $srcpart"
else
write_log "未找到 saved 备份，回退使用当前槽位分区: $srcpart"
fi
fi

write_log "blockdev --setrw $part"
if ! blockdev --setrw "$part" >> "$LOG_FILE" 2>&1; then
Expand All @@ -231,6 +419,10 @@ run_flash() {
fi
sync
write_log "$name 完成"
if [ "$name" = 'abl' ] && [ -n "$used_saved" ]; then
rm -f "$used_saved"
write_log "saved 刷写成功，已删除: $used_saved"
fi
done

if [ "$efisp_failed" = '1' ]; then
Expand All @@ -245,6 +437,66 @@ run_flash() {
fi
}

run_restore() {
ensure_runtime
if ! mkdir "$LOCK_DIR" 2>/dev/null; then
write_log '已有任务在执行，拒绝重复启动恢复'
exit 1
fi
printf '%s\n' "$$" > "$PID_FILE"
trap cleanup_lock EXIT INT TERM HUP
: > "$LOG_FILE"

ensure_backup_dir || { write_state 'error' '无法创建备份目录'; exit 1; }

current_slot=$(detect_current_slot 2>/dev/null || true)
if [ -z "$current_slot" ]; then
write_state 'error' '无法识别当前槽位'; write_log '无法识别当前槽位'; exit 1
fi

current_part=$(partition_path abl "$current_slot")
write_state 'running' "正在恢复当前槽位 $current_slot 的 abl"
write_log "恢复目标分区: $current_part"

restore_img=$(latest_backup_abl)
if [ -z "$restore_img" ] || [ ! -f "$restore_img" ]; then
write_state 'error' '未找到可恢复的 abl 备份'
write_log '未找到可恢复的 abl 备份'
exit 1
fi

if ! verify_image_hash "$restore_img"; then
write_state 'error' 'backup 备份哈希校验失败，已中止恢复'
write_log 'backup 备份哈希校验失败，已中止恢复'
exit 1
fi

saved_file=$(backup_partition_image "$current_part" "abl" 'saved')
if [ -z "$saved_file" ]; then
write_state 'error' '恢复前保存当前 abl 失败'
write_log '恢复前保存当前 abl 失败'
exit 1
fi
prune_old_saved_backups "$saved_file"

if ! blockdev --setrw "$current_part" >> "$LOG_FILE" 2>&1; then
write_state 'error' '当前槽位 abl 设置可写失败'
write_log '当前槽位 abl 设置可写失败'
exit 1
fi

write_log "开始恢复: $restore_img -> $current_part"
if ! dd if="$restore_img" of="$current_part" bs=4M conv=fsync >> "$LOG_FILE" 2>&1; then
write_state 'error' '恢复 abl 失败'
write_log '恢复 abl 失败'
exit 1
fi

sync
write_state 'success' 'abl 恢复完成'
write_log "恢复完成，当前槽位已写入: $restore_img"
}

start_flash() {
update_efisp="${1:-skip-efisp}"
ensure_runtime
Expand All @@ -269,6 +521,29 @@ PID=${pid}"
fi
}

start_restore() {
ensure_runtime
if pid=$(current_pid 2>/dev/null); then
emit "ALREADY_RUNNING=${pid}"; return 0
fi
if command -v nohup >/dev/null 2>&1; then
nohup sh "$0" restore >/dev/null 2>&1 &
else
sh "$0" restore </dev/null >/dev/null 2>&1 &
fi
sleep 1
if pid=$(current_pid 2>/dev/null); then
emit "STARTED=1
PID=${pid}"
else
_st=''; [ -f "$STATE_FILE" ] && read -r _st < "$STATE_FILE"
case "$_st" in
success|error|warning) emit "FINISHED=${_st}" ;;
*) emit 'STARTED=0' ;;
esac
fi
}

print_log() { ensure_runtime; tr '\n' '\t' < "$LOG_FILE"; }
tail_log() { ensure_runtime; tail -n "${1:-200}" "$LOG_FILE" | tr '\n' '\t'; }

Expand All @@ -284,8 +559,10 @@ case "$1" in
status) print_status ;;
flash) run_flash "$2" ;;
start) start_flash "$2" ;;
restore) run_restore ;;
start-restore) start_restore ;;
log) print_log ;;
tail) tail_log "$2" ;;
clear-log) clear_log ;;
*) printf 'Usage: %s {status|flash|start|log|tail [lines]|clear-log}\n' "$0" >&2; exit 1 ;;
*) printf 'Usage: %s {status|flash|start|restore|start-restore|log|tail [lines]|clear-log}\n' "$0" >&2; exit 1 ;;
esac
Loading
Loading