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