archiso 8.2 KB
Newer Older
1
# args: source, newroot, mountpoint
2
_mnt_dmsnapshot() {
3
    local img="${1}"
4
5
    local newroot="${2}"
    local mnt="${3}"
6
7
    local img_fullname="${img##*/}";
    local img_name="${img_fullname%%.*}"
8
    local dm_snap_name="${dm_snap_prefix}_${img_name}"
9
    local ro_dev ro_dev_size rw_dev
10

11
    ro_dev=$(losetup --find --show --read-only "${img}")
12
    echo ${ro_dev} >> /run/archiso/used_block_devices
13
    ro_dev_size=$(blockdev --getsz ${ro_dev})
14
15

    if [[ "${cow_persistent}" == "P" ]]; then
16
17
        if [[ -f "/run/archiso/cowspace/${cow_directory}/${img_name}.cow" ]]; then
            msg ":: Found '/run/archiso/cowspace/${cow_directory}/${img_name}.cow', using as persistent."
18
        else
19
            msg ":: Creating '/run/archiso/cowspace/${cow_directory}/${img_name}.cow' as persistent."
20
            truncate -s "${cow_spacesize}" "/run/archiso/cowspace/${cow_directory}/${img_name}.cow"
21
22
        fi
    else
23
24
25
        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"
26
        fi
27
        msg ":: Creating '/run/archiso/cowspace/${cow_directory}/${img_name}.cow' as non-persistent."
28
        truncate -s "${cow_spacesize}" "/run/archiso/cowspace/${cow_directory}/${img_name}.cow"
29
30
    fi

31
    rw_dev=$(losetup --find --show "/run/archiso/cowspace/${cow_directory}/${img_name}.cow")
32
    echo ${rw_dev} >> /run/archiso/used_block_devices
33

34
    dmsetup create ${dm_snap_name} --table "0 ${ro_dev_size} snapshot ${ro_dev} ${rw_dev} ${cow_persistent} ${cow_chunksize}"
35

36
37
38
39
    if [[ "${cow_persistent}" != "P" ]]; then
        rm -f "/run/archiso/cowspace/${cow_directory}/${img_name}.cow"
    fi

40
    _mnt_dev "/dev/mapper/${dm_snap_name}" "${newroot}${mnt}" "-w" "defaults"
41
    echo $(readlink -f /dev/mapper/${dm_snap_name}) >> /run/archiso/used_block_devices
Simo Leone's avatar
Simo Leone committed
42
43
}

44
45
46
47
48
49
50
51
52
53
# args: source, newroot, mountpoint
_mnt_overlayfs() {
    local src="${1}"
    local newroot="${2}"
    local mnt="${3}"
    mkdir -p /run/archiso/cowspace/${cow_directory}/upperdir /run/archiso/cowspace/${cow_directory}/workdir
    mount -t overlay -o lowerdir=${src},upperdir=/run/archiso/cowspace/${cow_directory}/upperdir,workdir=/run/archiso/cowspace/${cow_directory}/workdir airootfs "${newroot}${mnt}"
}


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

    if [[ "${copytoram}" == "y" ]]; then
62
        msg -n ":: Copying squashfs image to RAM..."
63
64
        if ! cp "${img}" "/run/archiso/copytoram/${img_fullname}" ; then
            echo "ERROR: while copy '${img}' to '/run/archiso/copytoram/${img_fullname}'"
65
66
            launch_interactive_shell
        fi
67
        img="/run/archiso/copytoram/${img_fullname}"
68
        msg "done."
69
    fi
70
    sfs_dev=$(losetup --find --show --read-only "${img}")
71
    echo ${sfs_dev} >> /run/archiso/used_block_devices
72
    _mnt_dev "${sfs_dev}" "${mnt}" "-r" "defaults"
Simo Leone's avatar
Simo Leone committed
73
74
}

75
# args: device, mountpoint, flags, opts
76
77
78
79
_mnt_dev() {
    local dev="${1}"
    local mnt="${2}"
    local flg="${3}"
80
    local opts="${4}"
81

82
83
    mkdir -p "${mnt}"

84
85
86
87
88
89
90
91
92
    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

93
    if mount -o "${opts}" "${flg}" "${dev}" "${mnt}"; then
94
        msg ":: Device '${dev}' mounted successfully."
95
    else
96
        echo "ERROR; Failed to mount '${dev}'"
97
98
99
100
101
102
        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
}

103
104
_verify_checksum() {
    local _status
105
    cd "/run/archiso/bootmnt/${archisobasedir}/${arch}"
106
    sha512sum -c airootfs.sha512 > /tmp/checksum.log 2>&1
107
108
109
110
111
    _status=$?
    cd "${OLDPWD}"
    return ${_status}
}

112
113
114
115
116
117
118
119
120
_verify_signature() {
    local _status
    cd "/run/archiso/bootmnt/${archisobasedir}/${arch}"
    gpg --homedir /gpg --status-fd 1 --verify airootfs.sfs.sig 2>/dev/null | grep -qE '^\[GNUPG:\] GOODSIG'
    _status=$?
    cd "${OLDPWD}"
    return ${_status}
}

121
run_hook() {
122
    [[ -z "${arch}" ]] && arch="$(uname -m)"
123
124
    [[ -z "${copytoram_size}" ]] && copytoram_size="75%"
    [[ -z "${archisobasedir}" ]] && archisobasedir="arch"
125
    [[ -z "${dm_snap_prefix}" ]] && dm_snap_prefix="arch"
126
    [[ -z "${archisodevice}" ]] && archisodevice="/dev/disk/by-label/${archisolabel}"
127
    [[ -z "${cow_spacesize}" ]] && cow_spacesize="256M"
128
129
130
131
132
133
134
135
136
137

    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

138
    [[ -z "${cow_flags}" ]] && cow_flags="defaults"
139
    [[ -z "${cow_directory}" ]] && cow_directory="persistent_${archisolabel}/${arch}"
140
    [[ -z "${cow_chunksize}" ]] && cow_chunksize="8"
141

142
143
144
145
    # set mount handler for archiso
    mount_handler="archiso_mount_handler"
}

146
147
148
# This function is called normally from init script, but it can be called
# as chain from other mount handlers.
# args: /path/to/newroot
149
archiso_mount_handler() {
150
151
    local newroot="${1}"

152
    if ! mountpoint -q "/run/archiso/bootmnt"; then
153
        _mnt_dev "${archisodevice}" "/run/archiso/bootmnt" "-r" "defaults"
154
155
156
157
158
        if [[ "${copytoram}" != "y" ]]; then
            echo $(readlink -f ${archisodevice}) >> /run/archiso/used_block_devices
        fi
    fi

159
    if [[ "${checksum}" == "y" ]]; then
160
        if [[ -f "/run/archiso/bootmnt/${archisobasedir}/${arch}/airootfs.sha512" ]]; then
161
162
163
164
165
            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"
166
                echo "see /tmp/checksum.log for details"
167
168
169
                launch_interactive_shell
            fi
        else
170
            echo "ERROR: checksum=y option specified but ${archisobasedir}/${arch}/airootfs.sha512 not found"
171
172
173
174
            launch_interactive_shell
        fi
    fi

175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
    if [[ "${verify}" == "y" ]]; then
        if [[ -f "/run/archiso/bootmnt/${archisobasedir}/${arch}/airootfs.sfs.sig" ]]; then
            msg -n ":: Signature verification requested, please wait..."
            if _verify_signature; then
                msg "done. Signature is OK, continue booting."
            else
                echo "ERROR: one or more files are corrupted"
                launch_interactive_shell
            fi
        else
            echo "ERROR: verify=y option specified but ${archisobasedir}/${arch}/airootfs.sfs.sig not found"
            launch_interactive_shell
        fi
    fi

190
    if [[ "${copytoram}" == "y" ]]; then
191
192
193
        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
194
195
    fi

196
    if [[ -n "${cow_device}" ]]; then
197
        _mnt_dev "${cow_device}" "/run/archiso/cowspace" "-r" "${cow_flags}"
198
        echo $(readlink -f ${cow_device}) >> /run/archiso/used_block_devices
199
        mount -o remount,rw "/run/archiso/cowspace"
200
    else
201
        msg ":: Mounting /run/archiso/cowspace (tmpfs) filesystem, size=${cow_spacesize}..."
202
        mkdir -p /run/archiso/cowspace
203
        mount -t tmpfs -o "size=${cow_spacesize}",mode=0755 cowspace /run/archiso/cowspace
204
    fi
205
    mkdir -p -m 0700 "/run/archiso/cowspace/${cow_directory}"
206

207
    _mnt_sfs "/run/archiso/bootmnt/${archisobasedir}/${arch}/airootfs.sfs" "/run/archiso/sfs/airootfs"
208
209
210
211
212
    if [[ -f "/run/archiso/sfs/airootfs/airootfs.img" ]]; then
        _mnt_dmsnapshot "/run/archiso/sfs/airootfs/airootfs.img" "${newroot}" "/"
    else
        _mnt_overlayfs "/run/archiso/sfs/airootfs" "${newroot}" "/"
    fi
213

214
    if [[ "${copytoram}" == "y" ]]; then
215
        umount -d /run/archiso/bootmnt
Aaron Griffin's avatar
Aaron Griffin committed
216
217
    fi
}
Dan McGee's avatar
Dan McGee committed
218
219

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