archiso 7.82 KB
Newer Older
1
# args: source, newroot, mountpoint
2
3
_mnt_fs() {
    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
16
17
18
19
    if [[ "${cowfile_size}" == "100" ]]; then
        rw_dev_size=${ro_dev_size}
    else
        # size calculation done in this way to avoid integer overflow when ro_dev_size is > 10.2G
        rw_dev_size=$((ro_dev_size/100*cowfile_size))
    fi
20
21

    if [[ "${cow_persistent}" == "P" ]]; then
22
23
        if [[ -f "/run/archiso/cowspace/${cow_directory}/${img_name}.cow" ]]; then
            msg ":: Found '/run/archiso/cowspace/${cow_directory}/${img_name}.cow', using as persistent."
24
        else
25
            msg ":: Creating '/run/archiso/cowspace/${cow_directory}/${img_name}.cow' as persistent."
26
            dd of="/run/archiso/cowspace/${cow_directory}/${img_name}.cow" count=0 seek=${rw_dev_size} &> /dev/null
27
28
        fi
    else
29
30
31
        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"
32
        fi
33
        msg ":: Creating '/run/archiso/cowspace/${cow_directory}/${img_name}.cow' as non-persistent."
34
        dd of="/run/archiso/cowspace/${cow_directory}/${img_name}.cow" count=0 seek=${rw_dev_size} &> /dev/null
35
36
    fi

37
    rw_dev=$(losetup --find --show "/run/archiso/cowspace/${cow_directory}/${img_name}.cow")
38
    echo ${rw_dev} >> /run/archiso/used_block_devices
39

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

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

46
# args: /path/to/image_file, mountpoint
47
48
49
_mnt_sfs() {
    local img="${1}"
    local mnt="${2}"
50
    local img_fullname="${img##*/}"
51
    local sfs_dev
52
53

    if [[ "${copytoram}" == "y" ]]; then
54
        msg -n ":: Copying squashfs image to RAM..."
55
56
        if ! cp "${img}" "/run/archiso/copytoram/${img_fullname}" ; then
            echo "ERROR: while copy '${img}' to '/run/archiso/copytoram/${img_fullname}'"
57
58
            launch_interactive_shell
        fi
59
        img="/run/archiso/copytoram/${img_fullname}"
60
        msg "done."
61
    fi
62
    sfs_dev=$(losetup --find --show --read-only "${img}")
63
    echo ${sfs_dev} >> /run/archiso/used_block_devices
64
    _mnt_dev "${sfs_dev}" "${mnt}" "-r"
Simo Leone's avatar
Simo Leone committed
65
66
}

67
68
69
70
71
72
# args: device, mountpoint, flags
_mnt_dev() {
    local dev="${1}"
    local mnt="${2}"
    local flg="${3}"

73
74
    mkdir -p "${mnt}"

75
76
77
78
79
80
81
82
83
    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

84
85
    if mount "${flg}" "${dev}" "${mnt}"; then
        msg ":: Device '${dev}' mounted successfully."
86
    else
87
        echo "ERROR; Failed to mount '${dev}'"
88
89
90
91
92
93
        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
}

94
95
_verify_checksum() {
    local _status
96
    cd "/run/archiso/bootmnt/${archisobasedir}"
97
    md5sum -c checksum.${arch}.md5 > /tmp/checksum.${arch}.log 2>&1
98
99
100
101
102
    _status=$?
    cd "${OLDPWD}"
    return ${_status}
}

103
run_hook() {
104
105
106
107
    [[ -z "${arch}" ]] && arch="$(uname -m)"
    [[ -z "${cowspace_size}" ]] && cowspace_size="75%"
    [[ -z "${copytoram_size}" ]] && copytoram_size="75%"
    [[ -z "${archisobasedir}" ]] && archisobasedir="arch"
108
    [[ -z "${dm_snap_prefix}" ]] && dm_snap_prefix="arch"
109
    [[ -z "${archisodevice}" ]] && archisodevice="/dev/disk/by-label/${archisolabel}"
110
111
112
113
114
    if [[ -z "${cowfile_size}" ]]; then
        cowfile_size="100"
    else
        cowfile_size=${cowfile_size/%}
    fi
115

116
    if [[ -z "${aitab}" ]]; then
117
        aitab="/run/archiso/bootmnt/${archisobasedir}/aitab"
118
    else
119
        aitab="/run/archiso/bootmnt/${archisobasedir}/${aitab}"
120
    fi
121
122
123
124
125
126
127
128
129
130
131
132

    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}"

133
134
135
136
    # set mount handler for archiso
    mount_handler="archiso_mount_handler"
}

137
138
139
# This function is called normally from init script, but it can be called
# as chain from other mount handlers.
# args: /path/to/newroot
140
archiso_mount_handler() {
141
142
    local newroot="${1}"

143
144
145
146
147
148
149
    if ! mountpoint -q "/run/archiso/bootmnt"; then
        _mnt_dev "${archisodevice}" "/run/archiso/bootmnt" "-r"
        if [[ "${copytoram}" != "y" ]]; then
            echo $(readlink -f ${archisodevice}) >> /run/archiso/used_block_devices
        fi
    fi

150

151
152
    if [[ ! -f "${aitab}" ]]; then
        echo "ERROR: '${aitab}' file does not exist."
153
154
155
        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
156
157
    fi

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

174
    if [[ "${copytoram}" == "y" ]]; then
175
176
177
        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
178
179
    fi

180
    if [[ -n "${cow_device}" ]]; then
181
        _mnt_dev "${cow_device}" "/run/archiso/cowspace" "-r"
182
        echo $(readlink -f ${cow_device}) >> /run/archiso/used_block_devices
183
        mount -o remount,rw "/run/archiso/cowspace"
184
    else
185
186
187
        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
188
    fi
189
    mkdir -p "/run/archiso/cowspace/${cow_directory}"
190

191
192
193
194
195
    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
196
            _mnt_sfs "/run/archiso/bootmnt/${archisobasedir}/${aitab_arch}/${aitab_img}.fs.sfs" "/run/archiso/sfs/${aitab_img}"
197
            _mnt_fs "/run/archiso/sfs/${aitab_img}/${aitab_img}.fs" "${newroot}" "${aitab_mnt}"
198
        else
199
            _mnt_sfs "/run/archiso/bootmnt/${archisobasedir}/${aitab_arch}/${aitab_img}.sfs" "${newroot}${aitab_mnt}"
Aaron Griffin's avatar
Aaron Griffin committed
200
        fi
201
    done < "${aitab}"
202

203
    if [[ "${copytoram}" == "y" ]]; then
204
        umount /run/archiso/bootmnt
Aaron Griffin's avatar
Aaron Griffin committed
205
206
    fi
}
Dan McGee's avatar
Dan McGee committed
207
208

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