mkimage revision 1.49 1 #!/bin/sh
2 # $NetBSD: mkimage,v 1.49 2014/12/14 00:28:46 christos Exp $
3 #
4 # Copyright (c) 2013, 2014 The NetBSD Foundation, Inc.
5 # All rights reserved.
6 #
7 # This code is derived from software contributed to The NetBSD Foundation
8 # by Christos Zoulas.
9 #
10 # Redistribution and use in source and binary forms, with or without
11 # modification, are permitted provided that the following conditions
12 # are met:
13 # 1. Redistributions of source code must retain the above copyright
14 # notice, this list of conditions and the following disclaimer.
15 # 2. Redistributions in binary form must reproduce the above copyright
16 # notice, this list of conditions and the following disclaimer in the
17 # documentation and/or other materials provided with the distribution.
18 # 3. Neither the name of The NetBSD Foundation nor the names of its
19 # contributors may be used to endorse or promote products derived
20 # from this software without specific prior written permission.
21 #
22 # THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
23 # ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
24 # TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
25 # PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
26 # BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
27 # CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
28 # SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
29 # INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
30 # CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
31 # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
32 # POSSIBILITY OF SUCH DAMAGE.
33 #
34
35 set -e
36
37 DIR="$(cd "$(dirname "$0")" && pwd)"
38 PROG="$(basename "$0")"
39
40 DISKLABEL=${TOOL_DISKLABEL:-disklabel}
41 FDISK=${TOOL_FDISK:-fdisk}
42 MAKEFS=${TOOL_MAKEFS:-makefs}
43 MTREE=${TOOL_MTREE:-mtree}
44 INSTALLBOOT=${TOOL_INSTALLBOOT:-installboot}
45 GZIP_CMD=${TOOL_GZIP:-gzip} # ${GZIP} is special to gzip(1)
46
47 src="/usr/src"
48 sets="base comp etc games man misc modules text"
49 xsets="xbase xcomp xetc xfont xserver"
50 minfree="10%"
51 bar="==="
52
53 tmp="$(mktemp -d "/tmp/$PROG.XXXXXX")"
54 mnt="${tmp}/mnt"
55 mkdir -p "${mnt}/etc" "${mnt}/dev"
56
57 trap "cleanup" 0 1 2 3 15
58
59 cleanup() {
60 case "$tmp" in
61 /tmp/$PROG.*) rm -fr "$tmp";;
62 esac
63 }
64
65 getsize() {
66 set -- $(ls -l $1)
67 echo $5
68 }
69
70 usage() {
71 cat << EOF 1>&2
72 Usage: $PROG -h <host-arch> [-bdmx] [-K <kerneldir>] [-S <srcdir>] [-D <destdir>] [-c <custom-files-dir>] [-s <Mb size>] [<image>]
73
74 -b Boot only, no sets loaded
75 -r root device kind (sd, wd, ld)
76 -d Add the debug sets
77 -m Optimize the OS installation to mimimize disk writes for SSDs
78 -x Load the x sets too, not just the base ones
79 EOF
80 exit 1
81 }
82
83 # First pass for options to get the host and src directories
84 OPTS="K:D:S:bc:dh:mr:s:x"
85 while getopts "$OPTS" f
86 do
87 case $f in
88 h) h="$OPTARG";;
89 S) src="$OPTARG";;
90 *) ;;
91 esac
92 done
93
94 if [ -z "$h" ]
95 then
96 usage
97 fi
98
99 if [ ! -f "${DIR}/conf/${h}.conf" ]
100 then
101 echo $PROG: ${DIR}/conf/${h}.conf is not present 1>&2
102 exit 1
103 fi
104
105 . "${DIR}/conf/${h}.conf"
106 release="/usr/obj/${MACHINE}/release"
107
108 selected_sets="$sets"
109 dsets=false
110 xsets=false
111 minwrites=false
112 rootdev=ld
113
114 OPTIND=1
115 while getopts "$OPTS" f
116 do
117 case $f in
118 D) release="$OPTARG";;
119 K) kernel="$OPTARG";;
120 S) ;;
121 b) bootonly=true;;
122 d) dsets=true
123 selected_sets="$selected_sets debug"
124 if $xsets; then
125 selected_sets="$selected_sets xdebug"
126 fi
127 ;;
128 c) custom="$OPTARG";;
129 h) ;;
130 m) minwrites=true;;
131 r) rootdev="$OPTARG";;
132 s) size="$OPTARG";;
133 x) xsets=true
134 selected_sets="$selected_sets $xsets"
135 if $dsets; then
136 selected_sets="$selected_sets xdebug"
137 fi
138 ;;
139 *) usage;;
140 esac
141 done
142
143 shift $(( $OPTIND - 1 ))
144 if [ -n "$1" ]; then
145 # take the next argument as being the image name
146 image="$1"
147 shift
148 fi
149
150 case "$image" in
151 *.gz) compress=true; image="${image%.gz}";;
152 *) compress=false;;
153 esac
154
155 if [ -z "${bootonly}" ]; then
156 echo ${bar} configuring sets ${bar}
157 (echo '/set type=dir uname=root gname=wheel mode=0755'
158 for i in $selected_sets; do
159 s="${release}/etc/mtree/set.$i"
160 if [ -f "$s" ]; then
161 cat "$s"
162 fi
163 done) > "$tmp/selected_sets"
164 fi
165
166 make_fstab
167 customize
168 populate
169
170 if [ -z "${bootonly}" ]; then
171 (cd ${mnt}; ${MTREE} -N ${release}/etc -c -k all |
172 ${MTREE} -N ${release}/etc -C -k all) >> "$tmp/selected_sets"
173 fi
174 if [ -n "${msdosid}" ]; then
175 echo ${bar} Populating msdos filesystem ${bar}
176 ${MAKEFS} -N ${release}/etc -t msdos \
177 -O $((${init} / 2))m -s $((${boot} / 2 + ${init} / 2))m \
178 ${image} ${mnt}/boot
179 fi
180
181 if [ -z "${bootonly}" ]; then
182 echo ${bar} Populating ffs filesystem ${bar}
183 ${MAKEFS} -N ${release}/etc -t ffs -rx \
184 -O ${ffsoffset} \
185 -o d=4096 -b $((${extra}))m \
186 -F "$tmp/selected_sets" ${image} "${release}" "${mnt}"
187 fi
188
189 if [ "${size}" = 0 ]; then
190 size="$(getsize "${image}")"
191 fi
192 newsize=$((${size} / 2 / 1024))
193 compare=$((${newsize} * 2 * 1024))
194 while [ "${compare}" != "${size}" ]
195 do
196 size="$((size + size - compare))"
197 newsize="$((${size} / 2 / 1024))"
198 compare="$((${newsize} * 2 * 1024))"
199 done
200
201 echo ${bar} Adding label ${bar}
202 make_label > ${tmp}/label
203 ${DISKLABEL} -R -F ${image} ${tmp}/label
204 if [ -n "${msdosid}" ]; then
205 echo ${bar} Running fdisk ${bar}
206 initsecs=$((${init} * 1024))
207 bootsecs=$((${boot} * 1024))
208 ${FDISK} -f -u -0 -s ${msdosid}/${initsecs}/${bootsecs} -F ${image}
209 elif [ -n "${netbsdid}" ]; then
210 echo ${bar} Running fdisk ${bar}
211 ${FDISK} -f -i ${image}
212 ${FDISK} -f -a -u -0 -s 169/${init} ${image}
213 ${INSTALLBOOT} -f -v ${image} ${release}/usr/mdec/bootxx_ffsv1
214 fi
215
216 if $compress; then
217 echo ${bar} Compressing image ${bar}
218 rm -f "${image}.gz"
219 ${GZIP_CMD} -9 ${image}
220 image="${image}.gz"
221 fi
222
223 echo ${bar} Image is ${image} ${bar}
224