Home | History | Annotate | Line # | Download | only in embedded
mkimage revision 1.27 
      1 #!/bin/sh
      2 # $NetBSD: mkimage,v 1.27 2013/02/10 15:51:59 christos Exp $
      3 #
      4 # Copyright (c) 2013 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 DIR="$(cd "$(dirname "$0")" && pwd)"
     36 PROG="$(basename "$0")"
     37 
     38 DISKLABEL=${TOOL_DISKLABEL:-disklabel}
     39 FDISK=${TOOL_FDISK:-fdisk}
     40 MAKEFS=${TOOL_MAKEFS:-makefs}
     41 MTREE=${TOOL_MTREE:-mtree}
     42 
     43 src="/usr/src"
     44 release="/usr/obj/evbarm/release"
     45 sets="base comp etc games man misc modules text"
     46 xsets="xbase xcomp xetc xfont xserver" 
     47 minfree="10%"
     48 bar="==="
     49 
     50 tmp="$(mktemp -d "/tmp/$PROG.XXXXXX")"
     51 mnt="${tmp}/mnt"
     52 mkdir -p "${mnt}/etc" "${mnt}/dev" "${mnt}/boot"
     53 
     54 trap "cleanup" 0 1 2 3 15
     55 
     56 cleanup() {
     57 	case "$tmp" in
     58 	/tmp/$PROG.*)	rm -fr "$tmp";;
     59 	esac
     60 }
     61 
     62 getsize() {
     63 	set -- $(ls -l $1)
     64 	echo $5
     65 }
     66 
     67 usage() {
     68 	cat << EOF 1>&2
     69 Usage: $PROG -h <host-arch> [-K <kerneldir>] [-S <srcdir>] [-D <destdir>] [-c <custom-files-dir>] [-s <Mb size>] [<image>]
     70 EOF
     71 	exit 1
     72 }
     73 
     74 # First pass for options to get the host
     75 OPTS="K:D:S:c:h:s:x"
     76 while getopts "$OPTS" f
     77 do
     78 	case $f in
     79 	h)	h="$OPTARG";;
     80 	*)	;;
     81 	esac
     82 done
     83 
     84 if [ -z "$h" ]
     85 then
     86 	usage
     87 fi
     88 
     89 if [ ! -f "${DIR}/conf/${h}.conf" ]
     90 then
     91 	echo $PROG: ${DIR}/conf/${h}.conf is not present 1>&2
     92 	exit 1
     93 fi
     94 
     95 . "${DIR}/conf/${h}.conf"
     96 
     97 selected_sets="$sets"
     98 
     99 OPTIND=1
    100 while getopts "$OPTS" f
    101 do
    102 	case $f in
    103 	D)	release="$OPTARG";;
    104 	K)	kernel="$OPTARG";;
    105 	S)	src="$OPTARG";;
    106 	c)	custom="$OPTARG";;
    107 	h)	;;
    108 	s)	size="$OPTARG";;
    109 	x)	selected_sets="$sets $xsets";;
    110 	*)	usage;;
    111 	esac
    112 done
    113 
    114 shift $(( $OPTIND - 1 ))
    115 if [ -n "$1" ]; then
    116 	# take the next argument as being the image name
    117 	image="$1"
    118 	shift
    119 fi
    120 
    121 case "$image" in
    122 *.gz)	compress=true; image="${image%.gz}";;
    123 *)	compress=false;;
    124 esac
    125 
    126 echo ${bar} configuring sets ${bar}
    127 (echo '/set type=dir uname=root gname=wheel mode=0755'
    128 for i in $selected_sets; do
    129 	s="${release}/etc/mtree/set.$i"
    130 	[ -f "$s" ] && cat "$s"
    131 done) > "$tmp/selected_sets"
    132 
    133 make_fstab
    134 customize
    135 populate
    136 
    137 (cd ${mnt}; ${MTREE} -N ${release}/etc -c -k all | 
    138     ${MTREE} -N ${release}/etc -C -k all) >> "$tmp/selected_sets"
    139 if [ -n ${msdosid} ]; then
    140 	echo ${bar} Populating msdos filesystem ${bar}
    141 	${MAKEFS} -N ${release}/etc -t msdos \
    142 	    -O $((${init} / 2))m -s $((${boot} / 2))m ${image} ${mnt}/boot
    143 fi
    144 
    145 echo ${bar} Populating ffs filesystem ${bar}
    146 ${MAKEFS} -N ${release}/etc -t ffs -rx \
    147     -O $(((${init} + ${boot} + ${swap}) / 2))m \
    148     -F "$tmp/selected_sets" ${image} "${release}" "${mnt}"
    149 
    150 if [ "${size}" = 0 ]; then
    151 	size="$(getsize "${image}")"
    152 fi
    153 newsize=$((${size} / 2 / 1024))
    154 
    155 echo ${bar} Adding label ${bar}
    156 make_label > ${tmp}/label
    157 ${DISKLABEL} -R -F ${image} ${tmp}/label
    158 if [ -n ${msdosid} ]; then
    159 	echo ${bar} Running fdisk ${bar}
    160 	${FDISK} -f -u -0 -s ${msdosid}/${init}/${boot} -F ${image}
    161 fi
    162 
    163 if $compress; then
    164 	echo ${bar} Compressing image ${bar}
    165 	rm -f "${image}.gz"
    166 	gzip -9 ${image}
    167 	image="${image}.gz"
    168 fi
    169 
    170 echo ${bar} Image is ${image} ${bar}
    171