mkimage revision 1.1 1 #! /bin/sh
2
3 # $NetBSD: mkimage,v 1.1 2012/01/15 02:01:02 agc Exp $
4
5 # Copyright (c) 2012 Alistair Crooks <agc (at] NetBSD.org>
6 # All rights reserved.
7 #
8 # Redistribution and use in source and binary forms, with or without
9 # modification, are permitted provided that the following conditions
10 # are met:
11 # 1. Redistributions of source code must retain the above copyright
12 # notice, this list of conditions and the following disclaimer.
13 # 2. Redistributions in binary form must reproduce the above copyright
14 # notice, this list of conditions and the following disclaimer in the
15 # documentation and/or other materials provided with the distribution.
16 #
17 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18 # IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19 # OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 # IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21 # INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22 # NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26 # THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 #
28
29 # find next available vnd, from kre
30 next_avail ()
31 {
32 local dev="$1"
33 local N=$(( ${#dev} + 1 ))
34 local unit units
35
36 units=$(
37 sysctl -n hw.disknames |
38 tr ' ' '\012' |
39 grep '^'"${dev}"'[0-9]' |
40 sort -n -k 1.$N )
41
42 test -z "${units}" && {
43 test -e "/dev/${dev}0a" || {
44 echo >&2 "No ${dev}s available!"
45 return 1
46 }
47 echo "${dev}0"
48 return
49 }
50
51 N=0
52 for unit in ${units}
53 do
54 if [ "${unit}" = "${dev}${N}" ]
55 then
56 N=$(( N + 1 ))
57 else
58 echo "${dev}${N}"
59 return
60 fi
61 done
62
63 test -e /dev/"${dev}${N}a" || {
64 echo >&2 "All ${dev}s in use"
65 return 1
66 }
67
68 echo "${dev}${N}"
69 }
70
71 # find the size of the gzipped files in a .tgz archive
72 sizeone() {
73 case "$1" in
74 *.tgz|*.tar.gz)
75 tar tvzf "$1" | awk '{ tot += $5 } END { print tot }'
76 ;;
77 *.tbz|*.tar.bz2)
78 tar tvjf "$1" | awk '{ tot += $5 } END { print tot }'
79 ;;
80 *)
81 echo 0
82 ;;
83 esac
84 }
85
86 setsdir=/usr/build/release/i386/binary/sets
87 size=0 # in MB
88 overhead=8 # in MB
89 image=usermode.img
90 h=usermode1.$(uname -n)
91 sets="base etc modules"
92 usermodedirs="/var.run /var.log /etc.cow /root.cow /pkgs"
93 specialdirs="/kern /proc"
94 bar="==="
95
96 while [ $# -gt 0 ]; do
97 case "$1" in
98 -S) setsdir=$2; shift ;;
99 -h) h=$2; shift ;;
100 -s) size=$2; shift ;;
101 -x) set -x ;;
102 *) break ;;
103 esac
104 shift
105 done
106
107 if [ $# -gt 0 ]; then
108 # take the next argument as being the image name
109 image="$1"
110 shift
111 fi
112
113 total=0
114 for s in ${sets}; do
115 total=$(expr ${total} + $(sizeone ${setsdir}/${s}.tgz))
116 done
117 total=$(expr \( ${total} / 1000000 \) + ${overhead})
118 if [ $size -eq 0 ]; then
119 # auto-size the pkgs fs
120 size=${total}
121 else
122 # check that we've been given enough space
123 if [ ${total} -gt ${size} ]; then
124 echo "File system size given as ${size} MB, but it needs ${total} MB" >&2
125 exit 1
126 fi
127 fi
128
129 echo "${bar} making a new ${size} MB image in ${image} ${bar}"
130 dd if=/dev/zero of=${image} bs=1m count=${size}
131
132 vnddev=$(next_avail vnd)
133 echo "${bar} mounting image via vnd ${vnddev} ${bar}"
134 sudo vnconfig ${vnddev} ${image}
135 sudo newfs /dev/r${vnddev}a
136 sudo mount /dev/${vnddev}a /mnt
137
138 echo "${bar} installing sets ${bar}"
139 (cd /mnt &&
140 for s in ${sets}; do
141 echo ${s}
142 sudo tar xpzf ${setsdir}/${s}.tgz
143 done
144 )
145
146 echo "${bar} performing customisations ${bar}"
147 sudo rm -f /mnt/etc/motd
148
149 tmp=/tmp/usermode.$$
150 cat > ${tmp} << EOF
151 # NetBSD/usermode /etc/fstab
152 /dev/ld0a / ffs ro 1 1
153 /dev/ld1a /pkgs ffs ro 1 2
154 kernfs /kern kernfs rw
155 ptyfs /dev/pts ptyfs rw
156 procfs /proc procfs rw
157 # mount /root as tmpfs on top of existing dir
158 tmpfs /root.cow tmpfs rw,-s2M 0 0
159 /root.cow /root union rw,hidden 0 0
160 # mount /etc as tmpfs on top of existing dir
161 tmpfs /etc.cow tmpfs rw,-s12M 0 0
162 /etc.cow /etc union rw,hidden 0 0
163 # mount /var/run as tmpfs on top of existing dir
164 tmpfs /var.run tmpfs rw,-s1M 0 0
165 /var.run /var/run union rw,hidden - -
166 # mount /var/log as tmpfs on top of existing dir
167 tmpfs /var.log tmpfs rw,-s10M 0 0
168 /var.log /var/log union rw,hidden - -
169 tmpfs /var/db tmpfs rw,-s8M 0 0
170 tmpfs /tmp tmpfs rw,-s32M 0 0
171 /dev/cd0a /cdrom cd9660 ro,noauto
172 EOF
173 sudo mv ${tmp} /mnt/etc/fstab
174
175 cat > ${tmp} << EOF
176 # $NetBSD: mkimage,v 1.1 2012/01/15 02:01:02 agc Exp $
177 #
178 # see rc.conf(5) for more information.
179 #
180 # Use program=YES to enable program, NO to disable it. program_flags are
181 # passed to the program on the command line.
182 #
183
184 # Load the defaults in from /etc/defaults/rc.conf (if it's readable).
185 # These can be overridden below.
186 #
187 if [ -r /etc/defaults/rc.conf ]; then
188 . /etc/defaults/rc.conf
189 fi
190
191 # If this is not set to YES, the system will drop into single-user mode.
192 #
193 rc_configured=YES
194
195 # make sure we have the right rw filesystem at boot
196 critical_filesystems_local="/var/db /var.run /var/run /var.log /var/log /etc.cow /etc /root.cow /root"
197
198 # Add local overrides below
199 #
200 dhcpcd=YES
201 sshd=YES
202
203 hostname=${h}
204 EOF
205 sudo mv ${tmp} /mnt/etc/rc.conf
206
207 echo "${bar} making extra directories ${bar}"
208 for d in ${usermodedirs}; do
209 sudo mkdir -p /mnt/${d}
210 done
211 for d in ${specialdirs}; do
212 sudo mkdir -p /mnt/${d}
213 done
214
215 echo "${bar} customising /var/tmp ${bar}"
216 sudo rm -rf /mnt/var/tmp
217 (cd /mnt/var && sudo ln -s /tmp tmp)
218
219 # package-related stuff
220 (cat /mnt/etc/csh.cshrc;echo "setenv PKG_DBDIR /usr/pkg/.dbdir") > ${tmp}
221 sudo mv ${tmp} /mnt/etc/csh.cshrc
222 (cat /mnt/etc/profile;echo "export PKG_DBDIR=/usr/pkg/.dbdir") > ${tmp}
223 sudo mv ${tmp} /mnt/etc/profile
224 (echo "PKG_DBDIR=/usr/pkg/.dbdir") > ${tmp}
225 sudo mv ${tmp} /mnt/etc/mk.conf
226 (cd /mnt/usr && sudo ln -s /pkgs/usr/pkg pkg)
227
228 df /mnt
229
230 sudo umount /mnt
231 sudo vnconfig -u ${vnddev}
232
233 exit 0
234