mkarchiso 8.68 KB
Newer Older
1
#! /bin/bash
2
3
4
5

PKGLIST=""
QUIET="y"
FORCE="n"
6
PACCONFIG="/etc/pacman.conf"
7
export LABEL="ARCH_$(date +%Y%m)"
8
9
PUBLISHER="Arch Linux <http://www.archlinux.org>"
APPLICATION="Arch Linux Live/Rescue CD"
10
COMPRESSION="gzip"
11
CREATE_DEFAULT="n"
12
13
14
15
16
17
18
19
20

APPNAME=$(basename "${0}")

# usage: usage <exitvalue>
usage ()
{
    echo "usage ${APPNAME} [options] command <command options>"
    echo " general options:"
    echo "    -f               Force overwrite of working files/squashfs image/bootable image"
21
    echo "    -p PACKAGE(S)    Additional package(s) to install, can be used multiple times"
22
    echo "    -C <file>        Config file for pacman. Default $PACCONFIG"
23
24
25
    echo "    -L <label>       Set a label for the disk"
    echo "    -P <publisher>   Set a publisher for the disk"
    echo "    -A <application> Set an application name for the disk"
26
27
28
29
    echo "    -c <compressor>  Set SquashFS compression type: gzip, lzma or lzo. Default $COMPRESSION"
    echo "                     NOTES:"
    echo "                            lzma: not officially supported yet by Linux (2.6.36)"
    echo "                            lzo:  needs Linux >= 2.6.36"
30
31
32
    echo "    -d               Create default user directory /home/arch"
    echo "    -v               Enable verbose output"
    echo "    -h               This message"
33
    echo " commands:"
34
35
36
    echo "   create <dir>"
    echo "      create a base directory layout to work with"
    echo "      includes all specified packages"
37
    echo "   iso <dir> <image name>"
38
    echo "      build an iso image from the working dir"
39
40
41
    exit $1
}

42
while getopts 'p:C:L:P:A:c:dfvh' arg; do
43
44
    case "${arg}" in
        p) PKGLIST="${PKGLIST} ${OPTARG}" ;;
45
        C) PACCONFIG="${OPTARG}" ;;
46
47
48
    	L) LABEL="${OPTARG}" ;;
    	P) PUBLISHER="${OPTARG}" ;;
    	A) APPLICATION="${OPTARG}" ;;
49
        c) COMPRESSION="${OPTARG}" ;;
50
        d) CREATE_DEFAULT="y" ;;
51
52
53
54
55
56
57
        f) FORCE="y" ;;
        v) QUIET="n" ;;
        h|?) usage 0 ;;
        *) echo "invalid argument '${arg}'"; usage 1 ;;
    esac
done

58
59
60
#trim spaces
PKGLIST="$(echo $PKGLIST)"

61
62
63
64
65
66
67
68
shift $(($OPTIND - 1))

# do UID checking here so someone can at least get usage instructions
if [ "$EUID" != "0" ]; then
    echo "error: This script must be run as root."
    exit 1
fi

69
70
71
72
73
if [ ! -f "$PACCONFIG" ]; then
    echo "error: pacman config file '$PACCONFIG' does not exist"
    exit 1
fi

74
command_name="${1}"
75
76
77
work_dir=""
imgname=""

78
case "${command_name}" in
79
80
    create) work_dir="${2}"; imgname="none" ;;
    iso) work_dir="${2}"; imgname="${3}" ;;
81
82
83
    *) echo "invalid command name '${command_name}'"; usage 1 ;;
esac

84
85
[ "x${imgname}" = "x" ] && echo "Image name must be specified" && usage 1
[ "x${work_dir}" = "x" ] && echo "Please specify a working directory" && usage 1
86

87
88
89
90
echo "${APPNAME} : Configuration Settings"
echo "        working directory:   ${work_dir}"
echo "               image name:   ${imgname}"

91
92
93
94
95
# usage: _pacman <packages>...
_pacman ()
{
    local ret
    if [ "${QUIET}" = "y" ]; then
96
        mkarchroot -n -C "$PACCONFIG" -f "${work_dir}/root-image" $* 2>&1 >/dev/null
97
98
        ret=$?
    else
99
        mkarchroot -n -C "$PACCONFIG" -f "${work_dir}/root-image" $*
100
101
        ret=$?
    fi
102
103
104
105

    # Cleanup
    find "${work_dir}" -name *.pacnew -name *.pacsave -name *.pacorig -delete

106
107
108
109
110
    if [ $ret -ne 0 ]; then
        exit 1
    fi
}

111
112
113
114
115
116
117
118
119
command_create () {
    echo "====> Creating working directory: ${work_dir}"
    mkdir -p "${work_dir}/iso/"
    mkdir -p "${work_dir}/root-image/"
    echo "# archiso isomounts file
# img - location of image/directory to mount relative to addons directory
# arch - architecture of this image
# mount point - absolute location on the post-initrd root
# type - either 'bind' or 'squashfs' for now
120

121
# syntax: <img> <arch> <mount point> <type>
122

123
124
125
# NOTE: Order matters. If the same file exists in multiple
#  images, the FIRST one mounted, top-down, will take precedence

126
127
root-image.sqfs i686 / squashfs
#root-image-x86_64.sqfs x86_64 / squashfs" > "${work_dir}/isomounts"
128

129
    echo "README for this archiso created directory
130

131
132
133
All directories in this dir, except for 'iso' will be squashed
with squashfs and put into the iso dir as iso/<dirname>.sqfs
This should be reflected in the isomounts file
134

135
The iso dir is later used to build the actual bootable iso.
136
137
Please ensure the proper bootloader is installed or copied
to the iso/ directory.
138
139

...TODO: Write more..." > "${work_dir}/README"
140

141
142
143
    if [ "${PKGLIST}" != "" ]; then
        echo "====> Installing packages to '${work_dir}/root-image/'"
        _pacman "${PKGLIST}"
144

145
146
147
        echo "Cleaning up what we can"
        if [ -d "${work_dir}/root-image/boot/" ]; then
            # remove the initcpio images that were generated for the host system
148
            find "${work_dir}/root-image/boot" -name '*.img' -delete
149
150
        fi

151
152
153
154
155
        if [ ${CREATE_DEFAULT} == "y" ]; then
            if [ -d "${work_dir}/root-image/home/" ]; then
                echo "Creating default home directory"
                install -d -o1000 -g100 -m0755 "${work_dir}/root-image/home/arch"
            fi
156
157
        fi

158
159
160
161
162
163
164
165
166
167
168
169
        # Delete pacman database sync cache files (*.tar.gz)
        find "${work_dir}/root-image/var/lib/pacman" -maxdepth 1 -type f -delete
        # Delete pacman database sync cache
        find "${work_dir}/root-image/var/lib/pacman/sync" -delete
        # Delete pacman package cache
        find "${work_dir}/root-image/var/cache/pacman/pkg" -type f -delete
        # Delete all log files, keeps empty dirs.
        find "${work_dir}/root-image/var/log" -type f -delete
        # Delete all temporary files and dirs
        find "${work_dir}/root-image/var/tmp" -mindepth 1 -delete
        # Delete all temporary files and dirs
        find "${work_dir}/root-image/tmp" -mindepth 1 -delete
170
    fi
171
}
172

173
174
175
176
177
178
179
180
181
182
# _mksquash dirname
_mksquash () {
    if [ ! -d "$1" ]; then
        echo "Error: '$1' is not a directory"
        return 1
    fi

    sqimg="${work_dir}/iso/$(basename ${1}).sqfs"
    echo "====> Generating SquashFS image for '${1}'"
    if [ -e "${sqimg}" ]; then
183
184
185
        dirhaschanged=$(find ${1} -newer ${sqimg})
        if [ "${dirhaschanged}" != "" ]; then
            echo "SquashFS image '${sqimg}' is not up to date, rebuilding..."
186
            rm "${sqimg}"
187
        else
188
189
            echo "SquashFS image '${sqimg}' is up to date, skipping."
            return
190
191
192
        fi
    fi

193
    echo "Creating SquashFS image. This may take some time..."
194
195
    start=$(date +%s)
    if [ "${QUIET}" = "y" ]; then
196
        mksquashfs "${1}" "${sqimg}" -noappend -comp "${COMPRESSION}" >/dev/null
197
    else
198
        mksquashfs "${1}" "${sqimg}" -noappend -comp "${COMPRESSION}"
199
200
201
    fi
    minutes=$(echo $start $(date +%s) | awk '{ printf "%0.2f",($2-$1)/60 }')
    echo "Image creation done in $minutes minutes."
202
}
203

204
205
206
207
208
209
210
211
212
_imgcommon () {
    for d in $(find "${work_dir}" -maxdepth 1 -type d -name '[^.]*'); do
        if [ "$d" != "${work_dir}/iso" -a \
             "$(basename "$d")" != "iso" -a \
             "$d" != "${work_dir}" ]; then
            _mksquash "$d"
        fi
    done

213
    echo "====> Making bootable image"
214
215
216
217
218
219
220
221
222
223
224
225
226

    # Sanity checks
    if [ ! -d "${work_dir}/iso" ]; then
        echo "Error: '${work_dir}/iso' doesn't exist. What did you do?!"
        exit 1
    fi

    if [ ! -f "${work_dir}/isomounts" ]; then
        echo "Error: the isomounts file doesn't exist. This image won't do anything"
        echo "  Protecting you from yourself and erroring out here..."
        exit 1
    fi

227
228
229
230
231
    if [ -e "${imgname}" ]; then
        if [ "${FORCE}" = "y" ]; then
            echo "Removing existing bootable image..."
            rm -rf "${imgname}"
        else
232
            echo "Error: Image '${imgname}' already exists, aborting."
233
234
235
236
            exit 1
        fi
    fi

237
    cp "${work_dir}/isomounts" "${work_dir}/iso/"
238

239
240
    if ! sed "s|archisolabel=[^ ]*|archisolabel=${LABEL}|" -i ${work_dir}/iso/boot/syslinux/syslinux.cfg; then
        echo "Error: ${work_dir}/iso/boot/syslinux/syslinux.cfg, doesn't exist, aborting."
241
242
        exit 1
    fi
243
244
245
246
}

command_iso () {
    _imgcommon
Aaron Griffin's avatar
Aaron Griffin committed
247

248
249
    echo "Creating ISO image..."
    qflag=""
Gerhard Brauer's avatar
Gerhard Brauer committed
250
    [ "${QUIET}" = "y" ] && qflag="-quiet"
251
    mkisofs ${qflag} -r -l \
252
        -b boot/syslinux/isolinux.bin -c boot/syslinux/boot.cat \
253
        -uid 0 -gid 0 \
254
        -udf -allow-limited-size -iso-level 3 \
255
256
        -input-charset utf-8 -p "prepared by mkarchiso" \
        -no-emul-boot -boot-load-size 4 -boot-info-table \
257
258
        -publisher "${PUBLISHER}" \
        -A "${APPLICATION}" \
259
        -V "${LABEL}" \
260
        -o "${imgname}" "${work_dir}/iso/"
261
    isohybrid "${imgname}"
262
263
}

264
265
# Go through the main commands in order. If 'all' was specified, then we want
# to do everything. Start with 'install'.
266
267
if [ "${command_name}" = "create"  ]; then
    command_create
268
fi
269
270
if [ "${command_name}" = "iso" ]; then
    command_iso
271
fi
272
273

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