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

PKGLIST=""
QUIET="y"
FORCE="n"
6
MOUNTFILE="$(pwd)/mounts"
Aaron Griffin's avatar
Aaron Griffin committed
7
IMG_TYPE="iso"
8
9
10
11
12
13
14
15
16

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"
17
    echo "    -p PACKAGE(S)    Additional package(s) to install, can be used multiple times"
18
19
20
21
    echo "    -t <iso,disk>    Type of image to create. Defaults to iso."
    echo "    -v               Enable verbose output."
    echo "    -h               This message."
    echo " commands:"
22
23
24
    echo "   install <dir> : install packages to the working dir"
    echo "   squash <dir> <sqfs name> : generate a squashfs image of the working dir"
    echo "   image -p <bootloader> <dir> <image name> : build an image from the working dir"
25
26
27
28
29
30
31
32
33
34
35
36
37
38
    exit $1
}

while getopts 'i:P:p:a:t:fvh' arg; do
    case "${arg}" in
        p) PKGLIST="${PKGLIST} ${OPTARG}" ;;
        t) IMG_TYPE="${OPTARG}" ;;
        f) FORCE="y" ;;
        v) QUIET="n" ;;
        h|?) usage 0 ;;
        *) echo "invalid argument '${arg}'"; usage 1 ;;
    esac
done

39
40
41
#trim spaces
PKGLIST="$(echo $PKGLIST)"

42
43
44
45
46
47
48
49
50
51
shift $(($OPTIND - 1))
echo "ARGS: $@"

# 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

command_name="${1}"
52
53
54
work_dir=""
imgname=""

55
case "${command_name}" in
56
    install) work_dir="${2}"; imgname="none" ;;
57
    squash) work_dir="${2}"; imgname="${3}" ;;
58
    image) work_dir="${2}"; imgname="${3}" ;;
59
60
61
    *) echo "invalid command name '${command_name}'"; usage 1 ;;
esac

62
63
[ "x${imgname}" = "x" ] && (echo "Image name must be specified" && usage 1)
[ "x${work_dir}" = "x" ] && (echo "Please specify a working directory" && usage 1)
64

65
66
67
68
69
70

echo "${APPNAME} : Configuration Settings"
echo "        working directory:   ${work_dir}"
echo "               image name:   ${imgname}"
echo "               image type:   ${IMG_TYPE}"

71
72
73
74
75
# usage: _pacman <packages>...
_pacman ()
{
    local ret
    if [ "${QUIET}" = "y" ]; then
76
        mkarchroot -f ${work_dir} $* 2>&1 >/dev/null
77
78
        ret=$?
    else
79
        mkarchroot -f ${work_dir} $*
80
81
        ret=$?
    fi
82
83
84
85

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

86
87
88
89
90
    if [ $ret -ne 0 ]; then
        exit 1
    fi
}

91
command_install () {
92
    echo "====> Installing packages to '${work_dir}'"
93
94
95
96
97
    if [ -e "${work_dir}" -a "${FORCE}" = "n" ]; then
        echo "error: Working dir '${work_dir}' already exists, aborting."
        exit 1
    fi

98
    mkdir -p "${work_dir}"
99

100
101
102
103
    if [ "${PKGLIST}" = "" ]; then
        echo "error: no packages to install"
        exit 1
    fi
104

105
106
    echo "Installing packages..."
    _pacman "${PKGLIST}"
107

108
109
110
111
112
    if [ -d "${work_dir}/lib/modules/" ]; then
        echo "Updating kernel module dependencies"
        kernelver=$(_kversion)
        depmod -a -b "${work_dir}" "${kernelver}"
    fi
113

114
115
116
117
118
    echo "Cleaning up what we can"
    if [ -d "${work_dir}/boot/" ]; then
        # remove the initcpio images that were generated for the host system
        find "${work_dir}/boot" -name *.img -delete
    fi
119

120
121
122
123
124
    #TODO is this needed? do it at the Makefile level?
    if [ -d "${work_dir}/home/" ]; then
        echo "Creating default home directory"
        install -d -o1000 -g100 -m0755 "${work_dir}/home/arch"
    fi
125
126
127
128

    # delete a lot of unnecessary cache/log files
    kill_dirs="var/abs var/cache/man var/cache/pacman var/log/* var/mail tmp/* initrd"
    for x in ${kill_dirs}; do
129
130
        if [ -e "${work_dir}/${x}" ]; then
            rm -rf "${work_dir}/${x}"
131
132
133
134
        fi
    done

    # pacman DBs are big, delete all sync dbs
135
    rm -rf "${work_dir}/var/lib/pacman/sync"
136
}
137

138
# command_squash path image
139
command_squash () {
140
141
    echo "====> Generating SquashFS image ${imgname}"
    if [ -e "${imgname}" ]; then
142
143
        if [ "${FORCE}" = "y" ]; then
            echo -n "Removing old SquashFS image..."
144
            rm "${imgname}"
145
146
            echo "done."
        else
147
            echo "error: SquashFS image '${imgname}' already exists, aborting."
148
149
150
151
            exit 1
        fi
    fi

152
    echo "Creating SquashFS image. This may take some time..."
153
154
    start=$(date +%s)
    if [ "${QUIET}" = "y" ]; then
155
        mksquashfs "${work_dir}" "${imgname}" -noappend >/dev/null
156
    else
157
        mksquashfs "${work_dir}" "${imgname}" -noappend
158
159
160
    fi
    minutes=$(echo $start $(date +%s) | awk '{ printf "%0.2f",($2-$1)/60 }')
    echo "Image creation done in $minutes minutes."
161
}
162

163
command_image () {
164
165
166
167
168
169
170
171
172
173
174
    echo "====> Making bootable image"
    if [ -e "${imgname}" ]; then
        if [ "${FORCE}" = "y" ]; then
            echo "Removing existing bootable image..."
            rm -rf "${imgname}"
        else
            echo "error: Image '${imgname}' already exists, aborting."
            exit 1
        fi
    fi

Aaron Griffin's avatar
Aaron Griffin committed
175
176
177
178
179
180
    if [ ! -e "${MOUNTFILE}" ]; then
        echo "error: mount file '${MOUNTFILE}' does not exist, aborting."
        exit 1
    fi
    cp "${MOUNTFILE}" "${work_dir}/mounts"

181
    bootflags=""
Aaron Griffin's avatar
Aaron Griffin committed
182
183
184
    if [ "$PKGLIST" = "grub" -o "$PKGLIST" = "grub-gfx" ]; then
        #test for either (eww, gross)
        tst=0
185
        pacman -Qi grub >/dev/null 2>&1
Aaron Griffin's avatar
Aaron Griffin committed
186
        tst=$(($tst + $?))
187
        pacman -Qi grub-gfx >/dev/null 2>&1
Aaron Griffin's avatar
Aaron Griffin committed
188
189
190
191
192
193
        tst=$(($tst + $?))
        if [ $tst -ge 2 ]; then
            echo "grub or grub-gfx not found on host system. Cannot install!"
            exit 1
        fi

194
195
196
197
        if [ ! -e "${work_dir}/boot/grub/stage2_eltorito" ]; then
            echo "error: grub stage files not found in '${work_dir}/boot/grub"
            exit 1
        fi
Aaron Griffin's avatar
Aaron Griffin committed
198
199
200

        bootflags="-b boot/grub/stage2_eltorito"
    elif [ "$PKGLIST" = "isolinux" ]; then
201
        if ! pacman -Qi $PKGLIST >/dev/null 2>&1; then
Aaron Griffin's avatar
Aaron Griffin committed
202
203
            echo "$PKGLIST not found on host system. Cannot install!"
            exit 1
204
        fi
Aaron Griffin's avatar
Aaron Griffin committed
205

206
207
208
209
        if [ ! -e "${work_dir}/boot/isolinux/isolinux.bin" ]; then
            echo "error: isolinux bin file not found in '${work_dir}/boot/isolinux"
            exit 1
        fi
Aaron Griffin's avatar
Aaron Griffin committed
210
211

        bootflags="-b boot/isolinux/isolinux.bin -c boot/isolinux/boot.cat"
212
    else
Aaron Griffin's avatar
Aaron Griffin committed
213
214
215
216
217
218
        echo "No bootloader specified. Use the -p flag to specify"
        echo "  Supported Bootloaders:"
        echo "     grub"
        echo "     grub-gfx"
        echo "     isolinux"
        exit 1
219
    fi
220

Aaron Griffin's avatar
Aaron Griffin committed
221
    if [ "$IMG_TYPE" = "disk" ]; then
222
        echo "Creating DISK image..."
223
        mkusbimg "${work_dir}" "${imgname}"
Aaron Griffin's avatar
Aaron Griffin committed
224
    elif [ "$IMG_TYPE" = "iso" ]; then
225
        echo "Creating ISO image..."
226
        qflag=""
Aaron Griffin's avatar
Aaron Griffin committed
227
        #[ "${QUIET}" = "y" ] && qflag="-q"
228
        mkisofs ${qflag} -r -l $bootflags -uid 0 -gid 0 \
Aaron Griffin's avatar
Aaron Griffin committed
229
            -input-charset utf-8 -p "prepared by mkarchiso" \
230
            -no-emul-boot -boot-load-size 4 -boot-info-table \
Aaron Griffin's avatar
Aaron Griffin committed
231
232
            -publisher "ArchLinux <archlinux.org>" \
            -A "ArchLinux Live/Rescue CD" \
233
            -o "${imgname}" "${work_dir}"
Aaron Griffin's avatar
Aaron Griffin committed
234
235
236
    else
        echo "Invalid image type '$IMG_TYPE' specified"
        exit 1
237
    fi
238
239
240
241
}

# Go through the main commands in order. If 'all' was specified, then we want
# to do everything. Start with 'install'.
242
if [ "${command_name}" = "install"  ]; then
243
244
    command_install
fi
245
if [ "${command_name}" = "squash" ]; then
246
247
    command_squash
fi
248
249
if [ "${command_name}" = "image" ]; then
    command_image
250
251
252
fi

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