mkarchiso 15 KB
Newer Older
1
#!/bin/bash
2

3
4
set -e -u

5
6
export LANG=C

7
8
9
app_name=${0##*/}
arch=$(uname -m)
pkg_list=""
10
run_cmd=""
11
12
13
14
15
16
quiet="y"
pacman_conf="/etc/pacman.conf"
export iso_label="ARCH_$(date +%Y%m)"
iso_publisher="Arch Linux <http://www.archlinux.org>"
iso_application="Arch Linux Live/Rescue CD"
install_dir="arch"
17
18
work_dir="work"
out_dir="out"
19
sfs_comp="xz"
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

# Show an INFO message
# $1: message string
_msg_info() {
    local _msg="${1}"
    echo "[mkarchiso] INFO: ${_msg}"
}

# Show an ERROR message then exit with status
# $1: message string
# $2: exit code number (with 0 does not exit)
_msg_error() {
    local _msg="${1}"
    local _error=${2}
    echo
    echo "[mkarchiso] ERROR: ${_msg}"
    echo
    if [[ ${_error} -gt 0 ]]; then
        exit ${_error}
    fi
}

# Show space usage similar to df, but better formatted.
# $1: mount-point or mounted device.
_show_space_usage () {
    local _where="${1}"
    local _fs _total _used _avail _pct_u=0 _mnt
    read _fs _total _used _avail _pct_u _mnt < <(df -m "${_where}" | tail -1) &> /dev/null
    _msg_info "Total: ${_total} MiB (100%) | Used: ${_used} MiB (${_pct_u}) | Avail: ${_avail} MiB ($((100 - ${_pct_u%\%}))%)"
}

51
_chroot_init() {
52
53
54
    if [[ -f "${work_dir}/mkarchiso.init" ]]; then
        _msg_info "Initial enviroment already installed, skipping."
    else
55
        mkdir -p ${work_dir}/airootfs
56
        _pacman "base syslinux"
57
        : > "${work_dir}/mkarchiso.init"
58
59
60
61
    fi
}

_chroot_run() {
62
    eval arch-chroot ${work_dir}/airootfs "${run_cmd}"
63
64
}

65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
# Mount a filesystem (trap signals in case of error for unmounting it
# $1: source image
# $2: mount-point
_mount_fs() {
    local _src="${1}"
    local _dst="${2}"
    trap "_umount_fs ${_src}" EXIT HUP INT TERM
    mkdir -p "${_dst}"
    _msg_info "Mounting '${_src}' on '${_dst}'"
    mount "${_src}" "${_dst}"
    _show_space_usage "${_dst}"
}

# Unmount a filesystem (and untrap signals)
# $1: mount-point or device/image
_umount_fs() {
    local _dst="${1}"
    _show_space_usage "${_dst}"
    _msg_info "Unmounting '${_dst}'"
    umount "${_dst}"
    rmdir "${_dst}"
    trap - EXIT HUP INT TERM
}

# Compare if a file/directory (source) is newer than other file (target)
# $1: source file/directory
# $2: target file
# return: 0 if target does not exists or if target is older than source.
#         1 if target is newer than source
_is_directory_changed() {
    local _src="${1}"
    local _dst="${2}"

    if [ -e "${_dst}" ]; then
        if [[ $(find ${_src} -newer ${_dst} | wc -l) -gt 0 ]]; then
            _msg_info "Target '${_dst}' is older than '${_src}', updating."
            rm -f "${_dst}"
            return 0
        else
            _msg_info "Target '${_dst}' is up to date with '${_src}', skipping."
            return 1
        fi
    else
        _msg_info "Target '${_dst}' does not exist, making it from '${_src}'"
        return 0
    fi
}

# Show help usage, with an exit status.
# $1: exit status number.
_usage ()
116
{
117
    echo "usage ${app_name} [options] command <command options>"
118
    echo " general options:"
119
    echo "    -p PACKAGE(S)    Package(s) to install, can be used multiple times"
120
    echo "    -r <command>     Run <command> inside airootfs"
121
122
    echo "    -C <file>        Config file for pacman."
    echo "                     Default: '${pacman_conf}'"
123
    echo "    -L <label>       Set a label for the disk"
124
    echo "                     Default: '${iso_label}'"
125
    echo "    -P <publisher>   Set a publisher for the disk"
126
    echo "                     Default: '${iso_publisher}'"
127
    echo "    -A <application> Set an application name for the disk"
128
    echo "                     Default: '${iso_application}'"
129
    echo "    -D <install_dir> Set an install_dir. All files will by located here."
130
    echo "                     Default: '${install_dir}'"
131
    echo "                     NOTE: Max 8 characters, use only [a-z0-9]"
132
    echo "    -w <work_dir>    Set the working directory"
133
    echo "                     Default: '${work_dir}'"
134
    echo "    -o <out_dir>     Set the output directory"
135
    echo "                     Default: '${out_dir}'"
136
137
    echo "    -c <comp_type>   Set SquashFS compression type (gzip, lzma, lzo, xz)"
    echo "                     Default: '${sfs_comp}'"
138
139
    echo "    -v               Enable verbose output"
    echo "    -h               This message"
140
    echo " commands:"
141
142
143
144
145
146
    echo "   init"
    echo "      Make base layout and install base group"
    echo "   install"
    echo "      Install all specified packages (-p)"
    echo "   run"
    echo "      run command specified by -r"
147
    echo "   prepare"
148
    echo "      build all images"
149
    echo "   checksum"
150
    echo "      make a checksum.md5 for self-test"
151
    echo "   pkglist"
152
    echo "      make a pkglist.txt of packages installed on airootfs"
153
    echo "   iso <image name>"
154
    echo "      build an iso image from the working dir"
155
    exit ${1}
156
157
}

158
# Shows configuration according to command mode.
159
# $1: init | install | run | prepare | checksum | iso
160
161
162
163
164
165
166
167
168
_show_config () {
    local _mode="$1"
    echo
    _msg_info "Configuration settings"
    _msg_info "                  Command:   ${command_name}"
    _msg_info "             Architecture:   ${arch}"
    _msg_info "        Working directory:   ${work_dir}"
    _msg_info "   Installation directory:   ${install_dir}"
    case "${_mode}" in
169
170
171
172
        init)
            _msg_info "       Pacman config file:   ${pacman_conf}"
            ;;
        install)
173
174
175
            _msg_info "       Pacman config file:   ${pacman_conf}"
            _msg_info "                 Packages:   ${pkg_list}"
            ;;
176
177
178
        run)
            _msg_info "              Run command:   ${run_cmd}"
            ;;
179
180
        prepare)
            ;;
181
182
        checksum)
            ;;
183
184
        pkglist)
            ;;
185
186
187
188
189
190
        iso)
            _msg_info "               Image name:   ${img_name}"
            _msg_info "               Disk label:   ${iso_label}"
            _msg_info "           Disk publisher:   ${iso_publisher}"
            _msg_info "         Disk application:   ${iso_application}"
            ;;
191
    esac
192
193
    echo
}
194

195
# Install desired packages to airootfs
196
197
_pacman ()
{
198
    _msg_info "Installing packages to '${work_dir}/airootfs/'..."
199
200

    if [[ "${quiet}" = "y" ]]; then
201
        pacstrap -C "${pacman_conf}" -c -d -G -M "${work_dir}/airootfs" $* &> /dev/null
202
    else
203
        pacstrap -C "${pacman_conf}" -c -d -G -M "${work_dir}/airootfs" $*
204
    fi
205

206
    _msg_info "Packages installed successfully!"
207
208
}

209
# Cleanup airootfs
210
_cleanup () {
211
    _msg_info "Cleaning up what we can on airootfs..."
212

213
    # Delete initcpio image(s)
214
215
    if [[ -d "${work_dir}/airootfs/boot" ]]; then
        find "${work_dir}/airootfs/boot" -type f -name '*.img' -delete
216
217
    fi
    # Delete kernel(s)
218
219
    if [[ -d "${work_dir}/airootfs/boot" ]]; then
        find "${work_dir}/airootfs/boot" -type f -name 'vmlinuz*' -delete
220
221
    fi
    # Delete pacman database sync cache files (*.tar.gz)
222
223
    if [[ -d "${work_dir}/airootfs/var/lib/pacman" ]]; then
        find "${work_dir}/airootfs/var/lib/pacman" -maxdepth 1 -type f -delete
224
225
    fi
    # Delete pacman database sync cache
226
227
    if [[ -d "${work_dir}/airootfs/var/lib/pacman/sync" ]]; then
        find "${work_dir}/airootfs/var/lib/pacman/sync" -delete
228
229
    fi
    # Delete pacman package cache
230
231
    if [[ -d "${work_dir}/airootfs/var/cache/pacman/pkg" ]]; then
        find "${work_dir}/airootfs/var/cache/pacman/pkg" -type f -delete
232
233
    fi
    # Delete all log files, keeps empty dirs.
234
235
    if [[ -d "${work_dir}/airootfs/var/log" ]]; then
        find "${work_dir}/airootfs/var/log" -type f -delete
236
237
    fi
    # Delete all temporary files and dirs
238
239
    if [[ -d "${work_dir}/airootfs/var/tmp" ]]; then
        find "${work_dir}/airootfs/var/tmp" -mindepth 1 -delete
240
    fi
241
    # Delete package pacman related files.
242
    find "${work_dir}" \( -name "*.pacnew" -o -name "*.pacsave" -o -name "*.pacorig" \) -delete
243
    _msg_info "Done!"
244
}
245

246
247
248
249
# Makes a SquashFS filesystem image of file/directory passes as argument with desired compression.
# $1: Source file/directory
_mksfs () {
    local _src="${1}"
250

251
252
    if [[ ! -e "${work_dir}/${_src}" ]]; then
        _msg_error "The path '${work_dir}/${_src}' does not exist" 1
253
254
    fi

255
256
257
258
259
    local _sfs_img="${work_dir}/${_src}.sfs"

    _msg_info "Creating SquashFS image for '${work_dir}/${_src}', This may take some time..."
    local _seconds=${SECONDS}
    if [[ "${quiet}" = "y" ]]; then
260
        mksquashfs "${work_dir}/${_src}" "${_sfs_img}" -noappend -comp "${sfs_comp}" -no-progress &> /dev/null
261
    else
262
        mksquashfs "${work_dir}/${_src}" "${_sfs_img}" -noappend -comp "${sfs_comp}" -no-progress
263
    fi
264
265
    _seconds=$((SECONDS - _seconds))
    printf "[mkarchiso] INFO: Image creation done in %02d:%02d minutes\n" $((_seconds / 60)) $((_seconds % 60))
266
}
267

268
269
270
271
# Makes a filesystem from a source directory.
# $1: Source directory
_mkfs () {
    local _src="${1}"
272

273
    local _fs_src="${work_dir}/${_src}"
274
    local _fs_img="${work_dir}/${_src}.img"
275

276
277
    if [[ ! -e "${_fs_src}" ]]; then
        _msg_error "The path '${_fs_src}' does not exist" 1
278
279
    fi

280
    _msg_info "Creating ext4 image of 32GiB..."
281
    rm -f "${_fs_img}"
282
    truncate -s 32G "${_fs_img}"
283
284
285
    local _qflag=""
    if [[ ${quiet} == "y" ]]; then
        _qflag="-q"
286
    fi
287
288
    mkfs.ext4 ${_qflag} -O ^has_journal -E lazy_itable_init=0 -m 0 -F "${_fs_img}"
    tune2fs -c 0 -i 0 "${_fs_img}" &> /dev/null
289
    _msg_info "Done!"
290
    _mount_fs "${_fs_img}" "${work_dir}/mnt/${_src}"
291
    _msg_info "Copying '${_fs_src}/' to '${work_dir}/mnt/${_src}/'..."
292
    cp -aT "${_fs_src}/" "${work_dir}/mnt/${_src}/"
293
    _msg_info "Done!"
294
    _umount_fs "${work_dir}/mnt/${_src}"
295
296
}

297
298
299
command_checksum () {
    _show_config checksum

300
301
302
303
    local _chk_arch

    for _chk_arch in i686 x86_64; do
        if _is_directory_changed "${work_dir}/iso/${install_dir}" "${work_dir}/iso/${install_dir}/checksum.${_chk_arch}.md5"; then
304
            _msg_info "Creating checksum file for self-test (${_chk_arch})..."
305
306
            cd "${work_dir}/iso/${install_dir}"
            if [[ -d "${_chk_arch}" ]]; then
307
                find ${_chk_arch} -type f -print0 | xargs -0 md5sum > checksum.${_chk_arch}.md5
308
309
310
311
312
            fi
            cd ${OLDPWD}
            _msg_info "Done!"
        fi
    done
313
314
}

315
316
317
command_pkglist () {
    _show_config pkglist

318
    if _is_directory_changed "${work_dir}/airootfs/var/lib/pacman/local" "${work_dir}/iso/${install_dir}/pkglist.${arch}.txt"; then
319
        _msg_info "Creating a list of installed packages on live-enviroment..."
320
        pacman -Sl -r "${work_dir}/airootfs" --config "${pacman_conf}" | \
321
322
323
324
325
326
327
            awk '/\[installed\]$/ {print $1 "/" $2 "-" $3}' > \
            "${work_dir}/iso/${install_dir}/pkglist.${arch}.txt"
        _msg_info "Done!"
    fi

}

328
# Create an ISO9660 filesystem from "iso" directory.
329
command_iso () {
330
331
    local _iso_efi_boot_args=""

332
333
334
    if [[ ! -f "${work_dir}/iso/isolinux/isolinux.bin" ]]; then
         _msg_error "The file '${work_dir}/iso/isolinux/isolinux.bin' does not exist." 1
    fi
335
336
337
    if [[ ! -f "${work_dir}/iso/isolinux/isohdpfx.bin" ]]; then
         _msg_error "The file '${work_dir}/iso/isolinux/isohdpfx.bin' does not exist." 1
    fi
338

339
340
    # If exists, add an EFI "El Torito" boot image (FAT filesystem) to ISO-9660 image.
    if [[ -f "${work_dir}/iso/EFI/archiso/efiboot.img" ]]; then
341
342
343
344
        _iso_efi_boot_args="-eltorito-alt-boot
                            -e EFI/archiso/efiboot.img
                            -no-emul-boot
                            -isohybrid-gpt-basdat"
345
346
    fi

347
348
    _show_config iso

349
350
    if _is_directory_changed "${work_dir}/iso" "${out_dir}/${img_name}"; then
        mkdir -p ${out_dir}
351
352
353
354
355
        _msg_info "Creating ISO image..."
        local _qflag=""
        if [[ ${quiet} == "y" ]]; then
            _qflag="-quiet"
        fi
356
        xorriso -as mkisofs ${_qflag} \
357
            -iso-level 3 \
358
359
360
361
362
363
364
            -full-iso9660-filenames \
            -volid "${iso_label}" \
            -appid "${iso_application}" \
            -publisher "${iso_publisher}" \
            -preparer "prepared by mkarchiso" \
            -eltorito-boot isolinux/isolinux.bin \
            -eltorito-catalog isolinux/boot.cat \
365
            -no-emul-boot -boot-load-size 4 -boot-info-table \
366
            -isohybrid-mbr ${work_dir}/iso/isolinux/isohdpfx.bin \
367
            ${_iso_efi_boot_args} \
368
369
            -output "${out_dir}/${img_name}" \
            "${work_dir}/iso/"
370
        _msg_info "Done! | $(ls -sh ${out_dir}/${img_name})"
371
372
373
    fi
}

374
# create airootfs.sfs filesystem, and push it in "iso" directory.
375
376
377
378
command_prepare () {
    _show_config prepare

    _cleanup
379
380
381
382
383
384

    local _src="${work_dir}/airootfs"
    local _dst="${work_dir}/iso/${install_dir}/${arch}"

    mkdir -p "${_dst}"

385
    if _is_directory_changed "${_src}" "${_dst}/airootfs.sfs"; then
386
        _mkfs airootfs
387
388
389
        _mksfs airootfs.img
        mv "${_src}.sfs" "${_dst}"
        rm "${_src}.img"
390
    fi
391
392
}

393
# Install packages on airootfs.
394
# A basic check to avoid double execution/reinstallation is done via hashing package names.
395
command_install () {
396
397
398
399
400
401
402
403
404
405
406
407
    if [[ ! -f "${pacman_conf}" ]]; then
        _msg_error "Pacman config file '${pacman_conf}' does not exist" 1
    fi

    #trim spaces
    pkg_list="$(echo ${pkg_list})"

    if [[ -z ${pkg_list} ]]; then
        _msg_error "Packages must be specified" 0
        _usage 1
    fi

408
    _show_config install
409
410
411

    local _pkg_list_hash
    _pkg_list_hash=$(echo ${pkg_list} | sort -u | md5sum | cut -c1-32)
412
    if [[ -f "${work_dir}/install.${_pkg_list_hash}" ]]; then
413
414
415
        _msg_info "These packages are already installed, skipping."
    else
        _pacman "${pkg_list}"
416
        : > "${work_dir}/install.${_pkg_list_hash}"
417
    fi
418
419
}

420
421
422
423
424
425
426
427
428
command_init() {
    _show_config init
    _chroot_init
}

command_run() {
    _show_config run
    _chroot_run
}
429
430
431
432
433

if [[ ${EUID} -ne 0 ]]; then
    _msg_error "This script must be run as root." 1
fi

434
while getopts 'p:r:C:L:P:A:D:w:o:c:vh' arg; do
435
436
    case "${arg}" in
        p) pkg_list="${pkg_list} ${OPTARG}" ;;
437
        r) run_cmd="${OPTARG}" ;;
438
439
440
441
442
        C) pacman_conf="${OPTARG}" ;;
        L) iso_label="${OPTARG}" ;;
        P) iso_publisher="${OPTARG}" ;;
        A) iso_application="${OPTARG}" ;;
        D) install_dir="${OPTARG}" ;;
443
444
        w) work_dir="${OPTARG}" ;;
        o) out_dir="${OPTARG}" ;;
445
        c) sfs_comp="${OPTARG}" ;;
446
447
448
449
450
451
452
453
454
455
456
457
458
459
        v) quiet="n" ;;
        h|?) _usage 0 ;;
        *)
            _msg_error "Invalid argument '${arg}'" 0
            _usage 1
            ;;
    esac
done

shift $((OPTIND - 1))

if [[ $# -lt 1 ]]; then
    _msg_error "No command specified" 0
    _usage 1
460
fi
461
462
463
command_name="${1}"

case "${command_name}" in
464
465
466
467
468
469
470
471
    init)
        command_init
        ;;
    install)
        command_install
        ;;
    run)
        command_run
472
473
474
475
        ;;
    prepare)
        command_prepare
        ;;
476
477
478
    checksum)
        command_checksum
        ;;
479
480
481
    pkglist)
        command_pkglist
        ;;
482
    iso)
483
        if [[ $# -lt 2 ]]; then
484
485
486
            _msg_error "No image specified" 0
            _usage 1
        fi
487
        img_name="${2}"
488
489
490
491
492
493
494
        command_iso
        ;;
    *)
        _msg_error "Invalid command name '${command_name}'" 0
        _usage 1
        ;;
esac
495
496

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