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

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

16
# Setup a loopback device for image passed as argument and echo the path to loopback device used.
17
18
19
# args: /path/to/image_file
_make_loop_dev() {
    local img="${1}"
20
21
22
    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}
23
24
}

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

34
    mkdir -p "${mnt}"
35
36
37
38
39
40

    _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)

41
42

    if [[ "${cow_persistent}" == "P" ]]; then
43
44
        if [[ -f "/run/archiso/cowspace/${cow_directory}/${img_name}.cow" ]]; then
            msg ":: Found '/run/archiso/cowspace/${cow_directory}/${img_name}.cow', using as persistent."
45
        else
46
47
            msg ":: Creating '/run/archiso/cowspace/${cow_directory}/${img_name}.cow' as persistent."
            dd of="/run/archiso/cowspace/${cow_directory}/${img_name}.cow" count=0 seek=${ro_dev_size} &> /dev/null
48
49
        fi
    else
50
51
52
        if [[ -f "/run/archiso/cowspace/${cow_directory}/${img_name}.cow" ]]; then
            msg ":: Found '/run/archiso/cowspace/${cow_directory}/${img_name}.cow' but non-persistent requested, removing."
            rm -f "/run/archiso/cowspace/${cow_directory}/${img_name}.cow"
53
        fi
54
55
        msg ":: Creating '/run/archiso/cowspace/${cow_directory}/${img_name}.cow' as non-persistent."
        dd of="/run/archiso/cowspace/${cow_directory}/${img_name}.cow" count=0 seek=${ro_dev_size} &> /dev/null
56
57
    fi

58
    _next_loop_dev
59
    rw_dev=$(_make_loop_dev "/run/archiso/cowspace/${cow_directory}/${img_name}.cow")
60

61
    echo "0 ${ro_dev_size} snapshot ${ro_dev} ${rw_dev} ${cow_persistent} 8" | dmsetup create ${dm_snap_name}
62

63
64
65
    msg ":: Mounting '/dev/mapper/${dm_snap_name}' (${ro_dev_fs_type}) to '${mnt}'"
    if ! mount -t "${ro_dev_fs_type}" "/dev/mapper/${dm_snap_name}" "${mnt}" ; then
        echo "ERROR: while mounting '/dev/mapper/${dm_snap_name}' to '${mnt}'"
66
67
        launch_interactive_shell
    fi
Simo Leone's avatar
Simo Leone committed
68
69
}

70
# args: /path/to/image_file, mountpoint
71
72
73
74
75
76
77
78
_mnt_sfs() {
    local img="${1}"
    local mnt="${2}"
    local img_fullname="${img##*/}";

    mkdir -p "${mnt}"

    if [[ "${copytoram}" == "y" ]]; then
79
        msg -n ":: Copying squashfs image to RAM..."
80
81
        if ! cp "${img}" "/run/archiso/copytoram/${img_fullname}" ; then
            echo "ERROR: while copy '${img}' to '/run/archiso/copytoram/${img_fullname}'"
82
83
            launch_interactive_shell
        fi
84
        img="/run/archiso/copytoram/${img_fullname}"
85
        msg "done."
86
    fi
87
88
89
90
    _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}'"
91
92
        launch_interactive_shell
    fi
Simo Leone's avatar
Simo Leone committed
93
94
}

95
96
97
98
99
100
# args: device, mountpoint, flags
_mnt_dev() {
    local dev="${1}"
    local mnt="${2}"
    local flg="${3}"

101
102
    mkdir -p "${mnt}"

103
104
105
106
107
108
109
110
111
    msg ":: Mounting '${dev}' to '${mnt}'"

    while ! poll_device "${dev}" 30; do
        echo "ERROR: '${dev}' device did not 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
    done

112
113
    if mount "${flg}" "${dev}" "${mnt}"; then
        msg ":: Device '${dev}' mounted successfully."
114
    else
115
        echo "ERROR; Failed to mount '${dev}'"
116
117
118
119
120
121
        echo "   Falling back to interactive prompt"
        echo "   You can try to fix the problem manually, log out when you are finished"
        launch_interactive_shell
    fi
}

122
123
_verify_checksum() {
    local _status
124
    cd "/run/archiso/bootmnt/${archisobasedir}"
125
126
127
128
129
130
    md5sum -c checksum.md5 > /checksum.log 2>&1
    _status=$?
    cd "${OLDPWD}"
    return ${_status}
}

131
run_hook() {
132
133
    modprobe loop

134
135
136
137
    [[ -z "${arch}" ]] && arch="$(uname -m)"
    [[ -z "${cowspace_size}" ]] && cowspace_size="75%"
    [[ -z "${copytoram_size}" ]] && copytoram_size="75%"
    [[ -z "${archisobasedir}" ]] && archisobasedir="arch"
138
    [[ -z "${dm_snap_prefix}" ]] && dm_snap_prefix="arch"
139
    [[ -z "${archisodevice}" ]] && archisodevice="/dev/disk/by-label/${archisolabel}"
140

141
    if [[ -z "${aitab}" ]]; then
142
        aitab="/run/archiso/bootmnt/${archisobasedir}/aitab"
143
    else
144
        aitab="/run/archiso/bootmnt/${archisobasedir}/${aitab}"
145
    fi
146
147
148
149
150
151
152
153
154
155
156
157

    if [[ -n "${cow_label}" ]]; then
        cow_device="/dev/disk/by-label/${cow_label}"
        [[ -z "${cow_persistent}" ]] && cow_persistent="P"
    elif [[ -n "${cow_device}" ]]; then
        [[ -z "${cow_persistent}" ]] && cow_persistent="P"
    else
        cow_persistent="N"
    fi

    [[ -z "${cow_directory}" ]] && cow_directory="persistent_${archisolabel}/${arch}"

158
159
160
161
    # set mount handler for archiso
    mount_handler="archiso_mount_handler"
}

162
163
164
# This function is called normally from init script, but it can be called
# as chain from other mount handlers.
# args: /path/to/newroot
165
archiso_mount_handler() {
166
167
168
    local newroot="${1}"

    _init_loop_dev
169

170
    mountpoint -q "/run/archiso/bootmnt" || _mnt_dev "${archisodevice}" "/run/archiso/bootmnt" "-r"
171

172
173
    if [[ ! -f "${aitab}" ]]; then
        echo "ERROR: '${aitab}' file does not exist."
174
175
176
        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
177
178
    fi

179
    if [[ "${checksum}" == "y" ]]; then
180
        if [[ -f "/run/archiso/bootmnt/${archisobasedir}/checksum.md5" ]]; then
181
182
183
184
185
186
187
188
189
190
191
192
193
194
            msg -n ":: Self-test requested, please wait..."
            if _verify_checksum; then
                msg "done. Checksum is OK, continue booting."
            else
                echo "ERROR: one or more files are corrupted"
                echo "see /checksum.log for details"
                launch_interactive_shell
            fi
        else
            echo "ERROR: checksum=y option specified but checksum.md5 not found"
            launch_interactive_shell
        fi
    fi

195
    if [[ "${copytoram}" == "y" ]]; then
196
197
198
        msg ":: Mounting /run/archiso/copytoram (tmpfs) filesystem, size=${copytoram_size}"
        mkdir -p /run/archiso/copytoram
        mount -t tmpfs -o "size=${copytoram_size}",mode=0755 copytoram /run/archiso/copytoram
199
200
    fi

201
    if [[ -n "${cow_device}" ]]; then
202
203
        _mnt_dev "${cow_device}" "/run/archiso/cowspace" "-r"
        mount -o remount,rw "/run/archiso/cowspace"
204
    else
205
206
207
        msg ":: Mounting /run/archiso/cowspace (tmpfs) filesystem, size=${cowspace_size}..."
        mkdir -p /run/archiso/cowspace
        mount -t tmpfs -o "size=${cowspace_size}",mode=0755 cowspace /run/archiso/cowspace
208
    fi
209
    mkdir -p "/run/archiso/cowspace/${cow_directory}"
210

211
212
213
214
215
216
    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
217
218
                _mnt_sfs "/run/archiso/bootmnt/${archisobasedir}/${aitab_arch}/${aitab_img}.fs.sfs" "/run/archiso/sfs/${aitab_img}"
                _mnt_fs "/run/archiso/sfs/${aitab_img}/${aitab_img}.fs" "${newroot}${aitab_mnt}"
219
            else
220
                _mnt_fs "/run/archiso/bootmnt/${archisobasedir}/${aitab_arch}/${aitab_img}.fs" "${newroot}${aitab_mnt}"
221
222
            fi
        else
223
            _mnt_sfs "/run/archiso/bootmnt/${archisobasedir}/${aitab_arch}/${aitab_img}.sfs" "${newroot}${aitab_mnt}"
Aaron Griffin's avatar
Aaron Griffin committed
224
        fi
225
    done < "${aitab}"
226

227
    if [[ "${copytoram}" == "y" ]]; then
228
        umount /run/archiso/bootmnt
Aaron Griffin's avatar
Aaron Griffin committed
229
230
    fi
}
Dan McGee's avatar
Dan McGee committed
231
232

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