Home | History | Annotate | Line # | Download | only in embedded
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