1
#!/bin/bash
2
# mount.sh - Mounts/unmounts essential filesystems for chroot
3

4
function mnt() {
5
    echo "MOUNTING..."
6
    sudo mount -t proc /proc ${2}proc
7
    sudo mount -t sysfs /sys ${2}sys
8
    sudo mount -o bind /dev ${2}dev
9
    sudo mount -o bind /dev/pts ${2}dev/pts
10
    echo "Entering chroot environment..."
11
    sudo chroot ${2}
12
}
13

14
function umnt() {
15
    echo "UNMOUNTING..."
16
    sudo umount ${2}proc
17
    sudo umount ${2}sys
18
    sudo umount ${2}dev/pts
19
    sudo umount ${2}dev
20
}
21

22
if [ "$1" == "-m" ] && [ -n "$2" ]; then
23
    mnt $1 $2
24
elif [ "$1" == "-u" ] && [ -n "$2" ]; then
25
    umnt $1 $2
26
else
27
    echo "Usage: $0 [-m|-u] <rootfs_path>"
28
    echo "  -m: Mount and chroot into the rootfs"
29
    echo "  -u: Unmount the rootfs"
30
    echo "Example: $0 -m /path/to/rootfs/"
31
fi

1
#!/usr/bin/env python3
2
# -*- coding: utf-8 -*-
3

4
import os
5
import sys
6
import subprocess
7
import shutil
8

9
# --- 用户配置 ---
10
# 在这里修改为要挂载的根文件系统的绝对路径
11
ROOTFS_PATH = "/home/hao/k230_sdk/output/k230_canmv_lckfb_defconfig/images/debian13"
12

13
# 如果需要跨架构 chroot (例如在 x86 电脑上 chroot ARM 系统),
14
# 指定 QEMU 静态二进制文件的路径。如果不需要，请留空 ""。
15
# 通常在 /usr/bin/ 目录下, 例如 qemu-aarch64-static, qemu-arm-static 等。
16
QEMU_STATIC_BINARY = "/usr/bin/qemu-riscv64-static"
17
# --- 配置结束 ---
18

19

20
def run_command(command, check=True):
21
    """执行一个 shell 命令并处理可能发生的错误"""
22
    print(f"🚀 执行: {' '.join(command)}")
23
    try:
24
        subprocess.run(command, check=check)
25
    except FileNotFoundError:
26
        print(f"❌ 命令未找到: {command[0]}。请确保该程序已安装并在 PATH 环境变量中。", file=sys.stderr)
27
        sys.exit(1)
28
    except subprocess.CalledProcessError as e:
29
        print(f"❌ 命令执行失败: {' '.join(e.cmd)} (返回码: {e.returncode})", file=sys.stderr)
30
        if not check:
31
             print("   (警告: 此命令失败，但程序将继续执行)")
32
        else:
33
            sys.exit(1)
34

35

36
def unmount_filesystems():
37
    """
38
    按正确顺序卸载所有 chroot 文件系统。
39
    该函数设计得非常健壮，会检查每个挂载点是否存在且已挂载。
40
    """
41
    print("\n🧹 开始安全卸载程序...")
42
    mount_points = ['dev/pts', 'dev', 'sys', 'proc']
43

44
    for mp in mount_points:
45
        target_path = os.path.join(ROOTFS_PATH, mp)
46
        if os.path.exists(target_path) and os.path.ismount(target_path):
47
            run_command(['sudo', 'umount', '-l', target_path], check=False) # 使用 -l (lazy) 更安全
48

49
    if QEMU_STATIC_BINARY:
50
        qemu_dest_path = os.path.join(ROOTFS_PATH, 'usr', 'bin', os.path.basename(QEMU_STATIC_BINARY))
51
        if os.path.exists(qemu_dest_path):
52
            print(f"   清理 QEMU 模拟器: {qemu_dest_path}")
53
            run_command(['sudo', 'rm', qemu_dest_path])
54

55
    print("✅ 清理完成。")
56

57

58
def mount_and_chroot():
59
    """
60
    挂载所需的文件系统并进入 chroot 环境。
61
    使用 try...finally 确保无论 chroot 内部发生什么，都会执行卸载。
62
    """
63
    try:
64
        print("🛠️  开始挂载 chroot 所需的文件系统...")
65
        mounts = [
66
            ('proc', 'proc', 'proc', None),
67
            ('sysfs', 'sys', 'sysfs', None),
68
            ('/dev', 'dev', None, 'bind'),
69
            ('/dev/pts', 'dev/pts', None, 'bind'),
70
        ]
71

72
        for source, dest_subdir, fstype, options in mounts:
73
            target_path = os.path.join(ROOTFS_PATH, dest_subdir)
74
            if not os.path.exists(target_path):
75
                print(f"   创建缺失的目录: {target_path}")
76
                run_command(['sudo', 'mkdir', '-p', target_path])
77

78
            command = ['sudo', 'mount']
79
            if fstype: command.extend(['-t', fstype])
80
            if options == 'bind': command.extend(['-o', 'bind'])
81
            command.extend([source, target_path])
82
            run_command(command)
83

84
        if QEMU_STATIC_BINARY:
85
            if not os.path.exists(QEMU_STATIC_BINARY):
86
                print(f"❌ 错误: QEMU 模拟器 '{QEMU_STATIC_BINARY}' 未在您的主机上找到。", file=sys.stderr)
87
                print("   如果您正在进行跨架构 chroot，请安装它 (例如: 'sudo apt install qemu-user-static')。", file=sys.stderr)
88
                return
89

90
            ### --- 新增功能：检查 binfmt_misc 是否启用 --- ###
91
            # 从 QEMU 二进制文件名推断出 binfmt_misc 的处理器名
92
            # 例如 "qemu-aarch64-static" -> "qemu-aarch64"
93
            handler_name = os.path.basename(QEMU_STATIC_BINARY).removesuffix('-static')
94
            binfmt_path = f"/proc/sys/fs/binfmt_misc/{handler_name}"
95

96
            print(f"   检查 QEMU binfmt_misc 处理器: {binfmt_path}")
97
            if not os.path.exists(binfmt_path):
98
                print(f"❌ 错误: QEMU binfmt_misc 处理器未启用 ({binfmt_path} 不存在)。", file=sys.stderr)
99
                print("   这意味着您的主机内核可能无法直接运行目标架构的二进制文件。", file=sys.stderr)
100
                print("   请尝试通过重启服务来注册处理器，例如运行:", file=sys.stderr)
101
                print("     sudo systemctl restart systemd-binfmt.service", file=sys.stderr)
102
                print("   或者:", file=sys.stderr)
103
                print("     sudo update-binfmts --enable", file=sys.stderr)
104
                return # 提前中止，finally 块会负责清理
105

106
            print("   ✅ QEMU binfmt_misc 处理器已启用。")
107
            ### --- 新增功能结束 --- ###
108

109
            qemu_dest_path = os.path.join(ROOTFS_PATH, 'usr', 'bin', os.path.basename(QEMU_STATIC_BINARY))
110
            print(f"   复制 QEMU 模拟器到 chroot 环境: {qemu_dest_path}")
111
            run_command(['sudo', 'cp', QEMU_STATIC_BINARY, qemu_dest_path])
112

113
        print("\n✅ 挂载完成。即将进入 chroot 环境...")
114
        print("   在 chroot 环境中，您可以执行所需命令。")
115
        print("   完成后，请键入 'exit' 以退出 chroot 并自动卸载所有文件系统。")
116

117
        run_command(['sudo', 'chroot', ROOTFS_PATH])
118

119
    except Exception as e:
120
        print(f"\n❌ 在挂载或 chroot 过程中发生意外错误: {e}", file=sys.stderr)
121
    finally:
122
        print("\n🚪 已退出 chroot 环境或发生错误。")
123
        unmount_filesystems()
124

125

126
def main():
127
    """脚本主入口"""
128
    if os.geteuid() != 0:
129
        print("❌ 错误: 此脚本需要 root 权限。请使用 'sudo' 运行。", file=sys.stderr)
130
        sys.exit(1)
131

132
    if not os.path.isdir(ROOTFS_PATH):
133
        print(f"❌ 错误: 配置的根文件系统路径 '{ROOTFS_PATH}' 不是一个有效的目录。", file=sys.stderr)
134
        sys.exit(1)
135

136
    if len(sys.argv) != 2 or sys.argv[1] not in ['-m', '-u']:
137
        print(f"用法: sudo {sys.argv[0]} [-m|-u]")
138
        print("  -m: 挂载文件系统并进入 chroot 环境")
139
        print("  -u: 仅卸载文件系统 (用于手动清理)")
140
        sys.exit(1)
141

142
    action = sys.argv[1]
143
    if action == '-m':
144
        mount_and_chroot()
145
    elif action == '-u':
146
        unmount_filesystems()
147

148
if __name__ == '__main__':
149
    main()