archiso 5.87 KB
Newer Older
1
2
3
# Initialize loopback device logic (we using on-demand mode)
# args: none
_init_loop_dev() {
4
5
6
7
    modprobe loop
    loop_dev_count=99
    loop_dev_range=$(cat /sys/block/loop0/range)
    loop_dev_minor=$((loop_dev_count*loop_dev_range))
8
9
10
11
12
}

# Call this function before _make_loop_dev() each time.
# args: none
_next_loop_dev() {
13
14
    loop_dev_count=$((loop_dev_count+1))
    loop_dev_minor=$((loop_dev_count*loop_dev_range))
15
16
}

17
# Setup a loopback device for image passed as argument and echo the path to loopback device used.
18
19
20
# args: /path/to/image_file
_make_loop_dev() {
    local img="${1}"
21
22
23
    mknod /dev/loop${loop_dev_count} b 7 ${loop_dev_minor} &> /dev/null
    losetup /dev/loop${loop_dev_count} "${img}" &> /dev/null
    echo /dev/loop${loop_dev_count}
24
25
}

Simo Leone's avatar
Simo Leone committed
26
# args: source, mountpoint
27
28
29
30
31
32
33
_mnt_fs() {
    local img="${1}"
    local mnt="${2}"
    local img_fullname="${img##*/}";
    local img_name="${img_fullname%%.*}"
    local ro_dev ro_dev_size ro_dev_fs_type rw_dev

34
    mkdir -p "${mnt}"
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

    _next_loop_dev
    ro_dev=$(_make_loop_dev "${img}")
    ro_dev_size=$(blockdev --getsz ${ro_dev})
    ro_dev_fs_type=$(blkid -o value -s TYPE -p ${ro_dev} 2> /dev/null)

    dd of="/cowspace/${img_name}.cow" count=0 seek=${ro_dev_size} &> /dev/null
    _next_loop_dev
    rw_dev=$(_make_loop_dev "/cowspace/${img_name}.cow")

    echo "0 ${ro_dev_size} snapshot ${ro_dev} ${rw_dev} N 8" | dmsetup create ${img_name}

    msg ":: Mounting '/dev/mapper/${img_name}' (${ro_dev_fs_type}) to '${mnt}'"
    if ! mount -t "${ro_dev_fs_type}" "/dev/mapper/${img_name}" "${mnt}" ; then
        echo "ERROR: while mounting '/dev/mapper/${img_name}' to '${mnt}'"
        launch_interactive_shell
    fi
Simo Leone's avatar
Simo Leone committed
52
53
}

54
# args: /path/to/image_file, mountpoint
55
56
57
58
59
60
61
62
_mnt_sfs() {
    local img="${1}"
    local mnt="${2}"
    local img_fullname="${img##*/}";

    mkdir -p "${mnt}"

    if [[ "${copytoram}" == "y" ]]; then
63
        msg -n ":: Copying squashfs image to RAM..."
64
65
        if ! cp "${img}" "/copytoram/${img_fullname}" ; then
            echo "ERROR: while copy '${img}' to '/copytoram/${img_fullname}'"
66
67
            launch_interactive_shell
        fi
68
        img="/copytoram/${img_fullname}"
69
        msg "done."
70
    fi
71
72
73
74
    _next_loop_dev
    msg ":: Mounting '${img}' (SquashFS) to '${mnt}'"
    if ! mount -r -t squashfs $(_make_loop_dev "${img}") "${mnt}" &> /dev/null ; then
        echo "ERROR: while mounting '${img}' to '${mnt}'"
75
76
        launch_interactive_shell
    fi
Simo Leone's avatar
Simo Leone committed
77
78
}

79
run_hook() {
80
81
82
83
84
85
86
    [[ -z "${arch}" ]] && arch="$(uname -m)"
    [[ -z "${cowspace_size}" ]] && cowspace_size="75%"
    [[ -z "${copytoram_size}" ]] && copytoram_size="75%"
    [[ -z "${archisobasedir}" ]] && archisobasedir="arch"
    [[ -z "${archisodevice}" ]] && archisodevice="/dev/disk/by-label/${archisolabel}"
    if [[ -z "${aitab}" ]]; then
        aitab="/bootmnt/${archisobasedir}/aitab"
87
    else
88
        aitab="/bootmnt/${aitab}"
89
    fi
90
91
92
93
    # set mount handler for archiso
    mount_handler="archiso_mount_handler"
}

94
95
96
# This function is called normally from init script, but it can be called
# as chain from other mount handlers.
# args: /path/to/newroot
97
archiso_mount_handler() {
98
99
100
101
    local newroot="${1}"
    local fstype fserror

    _init_loop_dev
102

103
    msg ":: Waiting for boot device..."
104
    while ! poll_device "${archisodevice}" 30; do
105
106
107
108
        echo "ERROR: boot device didn't show up after 30 seconds..."
        echo "   Falling back to interactive prompt"
        echo "   You can try to fix the problem manually, log out when you are finished"
        launch_interactive_shell
109
    done
110

111
112
113
114
115
116
    fstype=$(blkid -o value -s TYPE -p "${archisodevice}" 2> /dev/null)
    if [[ -n "${fstype}" ]]; then
        if mount -r -t "${fstype}" "${archisodevice}" /bootmnt; then
            if [[ -f "${aitab}" ]]; then
                msg ":: Mounted archiso volume successfully."
                fserror=0
117
            else
118
119
                echo "ERROR: Mounting was successful, but the '${aitab}' file does not exist."
                fserror=1
120
121
            fi
        else
122
123
            echo "ERROR; Failed to mount '${archisodevice}' (FS is ${fstype})"
            fserror=1
124
        fi
125
    else
126
127
        echo "ERROR: '${archisodevice}' found, but the filesystem type is unknown."
        fserror=1
128
129
    fi

130
    if [[ ${fserror} -eq 1 ]]; then
131
132
133
        echo "   Falling back to interactive prompt"
        echo "   You can try to fix the problem manually, log out when you are finished"
        launch_interactive_shell
Aaron Griffin's avatar
Aaron Griffin committed
134
135
    fi

136
    if [[ "${copytoram}" == "y" ]]; then
137
138
        msg -n ":: Mounting /copytoram (tmpfs) filesystem, size=${copytoram_size}..."
        mount -t tmpfs -o "size=${copytoram_size}",mode=0755 copytoram /copytoram
139
140
141
        msg "done."
    fi

142
143
    msg -n ":: Mounting /cowspace (tmpfs) filesystem, size=${cowspace_size}..."
    mount -t tmpfs -o "size=${cowspace_size}",mode=0755 cowspace /cowspace
144
145
    msg "done."

146
147
148
149
150
151
152
153
154
155
156
157
158
    local aitab_img aitab_mnt aitab_arch aitab_sfs_comp aitab_fs_type aitab_fs_size
    while read aitab_img aitab_mnt aitab_arch aitab_sfs_comp aitab_fs_type aitab_fs_size; do
        [[ "${aitab_img#\#}" != "${aitab_img}" ]] && continue
        [[ "${aitab_arch}" != "any" && "${aitab_arch}" != "${arch}" ]] && continue
        if [[ "${aitab_fs_type}" != "none" ]]; then
            if [[ "${aitab_sfs_comp}" != "none" ]]; then
                _mnt_sfs "/bootmnt/${archisobasedir}/${aitab_arch}/${aitab_img}.fs.sfs" "/sfs/${aitab_img}"
                _mnt_fs "/sfs/${aitab_img}/${aitab_img}.fs" "${newroot}${aitab_mnt}"
            else
                _mnt_fs "/bootmnt/${archisobasedir}/${aitab_arch}/${aitab_img}.fs" "${newroot}${aitab_mnt}"
            fi
        else
            _mnt_sfs "/bootmnt/${archisobasedir}/${aitab_arch}/${aitab_img}.sfs" "${newroot}${aitab_mnt}"
Aaron Griffin's avatar
Aaron Griffin committed
159
        fi
160
    done < "${aitab}"
161

162
163
    if [[ "${copytoram}" == "y" ]]; then
        umount /bootmnt
164
    else
165
166
        mkdir "${newroot}/bootmnt"
        mount --bind /bootmnt "${newroot}/bootmnt"
Aaron Griffin's avatar
Aaron Griffin committed
167
168
    fi
}
Dan McGee's avatar
Dan McGee committed
169
170

# vim:ft=sh:ts=4:sw=4:et: