mkimage revision 1.2
1 #! /bin/sh 2 3 # $NetBSD: mkimage,v 1.2 2012/01/20 02:19:47 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 bar="===" 87 custom=custom 88 h=usermode1.$(uname -n) 89 image=usermode.img 90 overhead=8 # in MB 91 sets="base etc modules" 92 setsdir=/usr/build/release/$(uname -m)/binary/sets 93 size=0 # in MB 94 specialdirs="/kern /proc" 95 usermodedirs="/var.run /var.log /etc.cow /root.cow /pkgs" 96 sudo="sudo" 97 98 while [ $# -gt 0 ]; do 99 case "$1" in 100 -S) setsdir=$2; shift ;; 101 -c) custom=$2; shift ;; 102 -h) h=$2; shift ;; 103 -s) size=$2; shift ;; 104 -x) set -x ;; 105 *) break ;; 106 esac 107 shift 108 done 109 110 if [ $# -gt 0 ]; then 111 # take the next argument as being the image name 112 image="$1" 113 shift 114 fi 115 116 total=0 117 for s in ${sets}; do 118 total=$(expr ${total} + $(sizeone ${setsdir}/${s}.tgz)) 119 done 120 # calculate size of custom files 121 custsize=0 122 if [ -d "${custom}" ]; then 123 custsize=$(ls -lR "${custom}" | awk 'NF == 9 { tot += $5 } END { print tot }') 124 fi 125 total=$(expr \( \( ${total} + ${custsize} \) / 1000000 \) + ${overhead}) 126 if [ $size -eq 0 ]; then 127 # auto-size the pkgs fs 128 size=${total} 129 else 130 # check that we've been given enough space 131 if [ ${total} -gt ${size} ]; then 132 echo "File system size given as ${size} MB, but it needs ${total} MB" >&2 133 exit 1 134 fi 135 fi 136 137 echo "${bar} making a new ${size} MB image in ${image} ${bar}" 138 dd if=/dev/zero of=${image} bs=1m count=${size} 139 140 vnddev=$(next_avail vnd) 141 echo "${bar} mounting image via vnd ${vnddev} ${bar}" 142 ${sudo} vnconfig ${vnddev} ${image} 143 ${sudo} newfs /dev/r${vnddev}a 144 ${sudo} mount /dev/${vnddev}a /mnt 145 146 echo "${bar} installing sets ${bar}" 147 (cd /mnt && 148 for s in ${sets}; do 149 echo ${s} 150 ${sudo} tar xpzf ${setsdir}/${s}.tgz 151 done 152 ) 153 154 echo "${bar} performing customisations ${bar}" 155 ${sudo} rm -f /mnt/etc/motd 156 157 tmp=/tmp/usermode.$$ 158 cat > ${tmp} << EOF 159 # NetBSD/usermode /etc/fstab 160 /dev/ld0a / ffs ro 1 1 161 /dev/ld1a /pkgs ffs ro 1 2 162 kernfs /kern kernfs rw 163 ptyfs /dev/pts ptyfs rw 164 procfs /proc procfs rw 165 # mount /root as tmpfs on top of existing dir 166 tmpfs /root.cow tmpfs rw,-s2M 0 0 167 /root.cow /root union rw,hidden 0 0 168 # mount /etc as tmpfs on top of existing dir 169 tmpfs /etc.cow tmpfs rw,-s12M 0 0 170 /etc.cow /etc union rw,hidden 0 0 171 # mount /var/run as tmpfs on top of existing dir 172 tmpfs /var.run tmpfs rw,-s1M 0 0 173 /var.run /var/run union rw,hidden - - 174 # mount /var/log as tmpfs on top of existing dir 175 tmpfs /var.log tmpfs rw,-s10M 0 0 176 /var.log /var/log union rw,hidden - - 177 tmpfs /var/db tmpfs rw,-s8M 0 0 178 tmpfs /tmp tmpfs rw,-s32M 0 0 179 /dev/cd0a /cdrom cd9660 ro,noauto 180 EOF 181 ${sudo} mv ${tmp} /mnt/etc/fstab 182 183 cat > ${tmp} << EOF 184 # $NetBSD: mkimage,v 1.2 2012/01/20 02:19:47 agc Exp $ 185 # 186 # see rc.conf(5) for more information. 187 # 188 # Use program=YES to enable program, NO to disable it. program_flags are 189 # passed to the program on the command line. 190 # 191 192 # Load the defaults in from /etc/defaults/rc.conf (if it's readable). 193 # These can be overridden below. 194 # 195 if [ -r /etc/defaults/rc.conf ]; then 196 . /etc/defaults/rc.conf 197 fi 198 199 # If this is not set to YES, the system will drop into single-user mode. 200 # 201 rc_configured=YES 202 203 # make sure we have the right rw filesystem at boot 204 critical_filesystems_local="/var/db /var.run /var/run /var.log /var/log /etc.cow /etc /root.cow /root" 205 206 # Add local overrides below 207 # 208 dhcpcd=YES 209 sshd=YES 210 211 hostname=${h} 212 EOF 213 ${sudo} mv ${tmp} /mnt/etc/rc.conf 214 215 echo "${bar} making extra directories ${bar}" 216 for d in ${usermodedirs}; do 217 ${sudo} mkdir -p /mnt/${d} 218 done 219 for d in ${specialdirs}; do 220 ${sudo} mkdir -p /mnt/${d} 221 done 222 223 echo "${bar} customising /var/tmp ${bar}" 224 ${sudo} rm -rf /mnt/var/tmp 225 (cd /mnt/var && ${sudo} ln -s /tmp tmp) 226 227 # package-related stuff 228 (cat /mnt/etc/csh.cshrc;echo "setenv PKG_DBDIR /usr/pkg/.dbdir") > ${tmp} 229 ${sudo} mv ${tmp} /mnt/etc/csh.cshrc 230 (cat /mnt/etc/profile;echo "export PKG_DBDIR=/usr/pkg/.dbdir") > ${tmp} 231 ${sudo} mv ${tmp} /mnt/etc/profile 232 (echo "PKG_DBDIR=/usr/pkg/.dbdir") > ${tmp} 233 ${sudo} mv ${tmp} /mnt/etc/mk.conf 234 (cd /mnt/usr && ${sudo} ln -s /pkgs/usr/pkg pkg) 235 236 # last, customisation stage 237 if [ -d ${custom} ]; then 238 echo "${bar} user customisations from files in ${custom} ${bar}" 239 (cd ${custom} && ${sudo} pax -rwpe . /mnt) 240 fi 241 242 df /mnt 243 244 ${sudo} umount /mnt 245 ${sudo} vnconfig -u ${vnddev} 246 247 exit 0 248
Indexes created Tue Sep 30 20:09:53 GMT 2025