initrd: Better check the created initrd before installing
[dragonfly.git] / initrd / mkinitrd.sh
1 #!/bin/sh
2 #
3 # Copyright (c) 2010, 2018
4 #       The DragonFly Project.  All rights reserved.
5 #
6 # Redistribution and use in source and binary forms, with or without
7 # modification, are permitted provided that the following conditions
8 # are met:
9 #
10 # 1. Redistributions of source code must retain the above copyright
11 #    notice, this list of conditions and the following disclaimer.
12 # 2. Redistributions in binary form must reproduce the above copyright
13 #    notice, this list of conditions and the following disclaimer in
14 #    the documentation and/or other materials provided with the
15 #    distribution.
16 # 3. Neither the name of The DragonFly Project nor the names of its
17 #    contributors may be used to endorse or promote products derived
18 #    from this software without specific, prior written permission.
19 #
20 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21 # ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
23 # FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
24 # COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
25 # INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
26 # BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
27 # LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
28 # AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
29 # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
30 # OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 # SUCH DAMAGE.
32 #
33
34 #
35 # Description
36 #
37 # This tool packs the (statically linked) rescue tools (at /rescue by
38 # default) and contents specified by "-c <content_dirs>", such as the
39 # necessary etc files, into an UFS-formatted VN image.  This image is
40 # installed at /boot/kernel/initrd.img.gz and used as the initial ramdisk
41 # to help mount the real root filesystem when it is encrypted or on LVM.
42 #
43
44
45 #
46 # Default configurations
47 #
48
49 # Directory hierarchy on the initrad
50 #
51 # * 'new_root' will always be created
52 # * 'sbin' will be symlinked to 'bin'
53 # * 'tmp' will be symlinked to 'var/tmp'
54 #
55 INITRD_DIRS="bin dev etc mnt var"
56
57 # Directory of the statically linked rescue tools which will be copied
58 # onto the initrd.
59 RESCUE_DIR="/rescue"
60
61 # Specify the location that the initrd will be installed to, i.e.,
62 # <BOOT_DIR>/kernel/initrd.img.gz
63 BOOT_DIR="/boot"
64
65 # Maximum size (number of MB) allowed for the initrd image
66 INITRD_SIZE_MAX="15"  # MB
67
68
69 #
70 # Helper functions
71 #
72
73 log() {
74         echo "$@" >&2
75 }
76
77 error() {
78         local rc=$1
79         shift
80         log "$@"
81         exit ${rc}
82 }
83
84 check_dirs() {
85         for _dir; do
86                 [ -d "${_dir}" ] ||
87                     error 1 "Directory '${_dir}' does not exist"
88         done
89         return 0
90 }
91
92 # Calculate the total size of the given directory, taking care of the
93 # hard links.
94 #
95 # NOTE: Do not use 'du' since it gives the disk usage of the files,
96 #       which varies between different filesystems.
97 #
98 calc_size() {
99         find "$1" -ls | \
100             awk '{ print $7,$1 }' | \
101             sort -n -k 2 | \
102             uniq -f 1 | \
103             awk '{ sum+=$1 } END { print sum }'  # byte
104 }
105
106
107 #
108 # Functions
109 #
110
111 calc_initrd_size() {
112         log "Calculating required initrd size ..."
113         isize=0
114         for _dir; do
115                 csize=$(calc_size ${_dir})
116                 log "* ${_dir}: ${csize} bytes"
117                 isize=$((${isize} + ${csize}))
118         done
119         # Round initrd size up by MB
120         isize_mb=$(echo ${isize} | awk '
121             function ceil(x) {
122                 y = int(x);
123                 return (x>y ? y+1 : y);
124             }
125             {
126                 mb = $1/1024/1024;
127                 print ceil(mb);
128             }')
129         # Add additional 1 MB
130         echo $((${isize_mb} + 1))
131 }
132
133 create_vn() {
134         kldload -n vn
135         VN_DEV=$(vnconfig -c -S ${INITRD_SIZE}m -Z -T vn ${INITRD_FILE}) &&
136             echo "Configured ${VN_DEV}" ||
137             error 1 "Failed to configure VN device"
138
139         newfs -i 131072 -m 0 /dev/${VN_DEV}s0 &&
140             echo "Formatted initrd image with UFS" ||
141             error 1 "Failed to format the initrd image"
142         mount_ufs /dev/${VN_DEV}s0 ${BUILD_DIR} &&
143             echo "Mounted initrd image on ${BUILD_DIR}" ||
144             error 1 "Failed to mount initrd image on ${BUILD_DIR}"
145 }
146
147 destroy_vn() {
148         umount /dev/${VN_DEV}s0 &&
149             echo "Unmounted initrd image" ||
150             error 1 "Failed to umount initrd image"
151         vnconfig -u ${VN_DEV} &&
152             echo "Unconfigured ${VN_DEV}" ||
153             error 1 "Failed to unconfigure ${VN_DEV}"
154 }
155
156 make_hier() {
157         mkdir -p ${BUILD_DIR}/new_root ||
158             error 1 "Failed to mkdir ${BUILD_DIR}/new_root"
159         # Symlink 'sbin' to 'bin'
160         ln -sf bin ${BUILD_DIR}/sbin
161         # Symlink 'tmp' to 'var/tmp', as '/var' will be mounted with
162         # tmpfs, saving a second tmpfs been mounted on '/tmp'.
163         ln -sf var/tmp ${BUILD_DIR}/tmp
164         for _dir in ${INITRD_DIRS}; do
165                 [ ! -d "${BUILD_DIR}/${_dir}" ] &&
166                     mkdir -p ${BUILD_DIR}/${_dir}
167         done
168         echo "Created directory structure"
169 }
170
171 copy_rescue() {
172         cpdup -o -u ${RESCUE_DIR}/ ${BUILD_DIR}/bin/ &&
173             echo "Copied ${RESCUE_DIR} to ${BUILD_DIR}/bin" ||
174             error 1 "Failed to copy ${RESCUE_DIR} to ${BUILD_DIR}/bin"
175 }
176
177 copy_content() {
178         for _dir in ${CONTENT_DIRS}; do
179                 cpdup -o -u ${_dir}/ ${BUILD_DIR}/ &&
180                     echo "Copied ${_dir} to ${BUILD_DIR}" ||
181                     error 1 "Failed to copy ${dir} to ${BUILD_DIR}"
182         done
183 }
184
185 print_info() {
186         lt ${BUILD_DIR}
187         df -h ${BUILD_DIR}
188 }
189
190 # Check the validity of the created initrd image before moving over.
191 # This prevents creating an empty and broken initrd image by running
192 # this tool but without ${CONTENT_DIRS} prepared.
193 #
194 # NOTE: Need more improvements.
195 #
196 check_initrd()
197 {
198         [ -x "${BUILD_DIR}/sbin/oinit" ] &&
199         [ -x "${BUILD_DIR}/bin/sh" ] &&
200         [ -x "${BUILD_DIR}/etc/rc" ] || {
201                 destroy_vn
202                 error 1 "Ivalid initrd image!"
203         }
204 }
205
206 usage() {
207         error 2 \
208             "usage: ${0##*/} [-b boot_dir] [-r rescue_dir]" \
209             "[-s size] [-S max_size] -c <content_dirs>"
210 }
211
212
213 #
214 # Main
215 #
216
217 while getopts :b:c:hr:s:S: opt; do
218         case ${opt} in
219         b)
220                 BOOT_DIR="${OPTARG}"
221                 ;;
222         c)
223                 CONTENT_DIRS="${OPTARG}"
224                 ;;
225         h)
226                 usage
227                 ;;
228         r)
229                 RESCUE_DIR="${OPTARG}"
230                 ;;
231         s)
232                 INITRD_SIZE="${OPTARG}"
233                 ;;
234         S)
235                 INITRD_SIZE_MAX="${OPTARG}"
236                 ;;
237         \?)
238                 log "Invalid option -${OPTARG}"
239                 usage
240                 ;;
241         :)
242                 log "Option -${OPTARG} requires an argument"
243                 usage
244                 ;;
245         esac
246 done
247
248 shift $((OPTIND - 1))
249 [ $# -ne 0 ] && usage
250 [ -z "${BOOT_DIR}" -o -z "${RESCUE_DIR}" -o -z "${CONTENT_DIRS}" ] && usage
251 check_dirs ${BOOT_DIR} ${RESCUE_DIR} ${CONTENT_DIRS}
252
253 VN_DEV=""
254 INITRD_SIZE=${INITRD_SIZE%[mM]}  # MB
255 INITRD_SIZE_MAX=${INITRD_SIZE_MAX%[mM]}  # MB
256
257 BUILD_DIR=$(mktemp -d -t initrd) || error $? "Cannot create build directory"
258 echo "Initrd build directory: ${BUILD_DIR}"
259 INITRD_FILE="${BUILD_DIR}.img"
260 INITRD_DEST="${BOOT_DIR}/kernel/initrd.img.gz"
261
262 CSIZE=$(calc_initrd_size ${RESCUE_DIR} ${CONTENT_DIRS})
263 echo "Required initrd image size: ${CSIZE} MB"
264 if [ -n "${INITRD_SIZE}" -a "${INITRD_SIZE}" != "0" ]; then
265         if [ ${CSIZE} -gt ${INITRD_SIZE} ]; then
266                 error 1 "Given initrd size (${INITRD_SIZE} MB) too small"
267         fi
268 else
269         INITRD_SIZE=${CSIZE}
270 fi
271 echo "Initrd size: ${INITRD_SIZE} MB"
272
273 if [ -n "${INITRD_SIZE_MAX}" -a "${INITRD_SIZE_MAX}" != "0" ] && \
274    [ ${INITRD_SIZE} -gt ${INITRD_SIZE_MAX} ]; then
275         error 1 "Exceeded the maximum size (${INITRD_SIZE_MAX} MB)"
276 fi
277
278 create_vn
279 make_hier
280 copy_rescue
281 copy_content
282 print_info
283 destroy_vn
284 rm -rf ${BUILD_DIR}
285
286 echo -n "Compressing ${INITRD_FILE} ..."
287 gzip -9 ${INITRD_FILE}
288 echo " OK"
289
290 if [ -f "${INITRD_DEST}" ]; then
291         echo -n "Backing up ${INITRD_DEST} ..."
292         mv ${INITRD_DEST} ${INITRD_DEST}.old
293         echo " OK (${INITRD_DEST}.old)"
294 fi
295
296 echo -n "Copying ${INITRD_FILE}.gz to ${INITRD_DEST} ..."
297 install -o root -g wheel -m 444 ${INITRD_FILE}.gz ${INITRD_DEST}
298 echo " OK"
299 rm -f ${INITRD_FILE}.gz