Project

General

Profile

Download (71.1 KB) Statistics
| Branch: | Tag: | Revision:
1
#!/bin/sh
2
#
3
# builder_common.sh
4
#
5
# Copyright (c) 2004-2015 Electric Sheep Fencing, LLC
6
# Copyright (C) 2014 Ermal Luçi
7
# All rights reserved.
8
#
9
# NanoBSD portions of the code
10
# Copyright (c) 2005 Poul-Henning Kamp.
11
# and copied from nanobsd.sh
12
# All rights reserved.
13
#
14
# FreeSBIE portions of the code
15
# Copyright (c) 2005 Dario Freni
16
# and copied from FreeSBIE project
17
# All rights reserved.
18
#
19
# Redistribution and use in source and binary forms, with or without
20
# modification, are permitted provided that the following conditions
21
# are met:
22
#
23
# 1. Redistributions of source code must retain the above copyright
24
#    notice, this list of conditions and the following disclaimer.
25
#
26
# 2. Redistributions in binary form must reproduce the above copyright
27
#    notice, this list of conditions and the following disclaimer in the
28
#    documentation and/or other materials provided with the distribution.
29
#
30
# THIS SOFTWARE IS PROVIDED BY THE pfSense PROJECT ``AS IS'' AND ANY
31
# EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
32
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
33
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE pfSense PROJECT OR
34
# ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
35
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
36
# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
37
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
38
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
39
# STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
40
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
41
# OF THE POSSIBILITY OF SUCH DAMAGE.
42
# Redistribution and use in source and binary forms, with or without
43
# modification, are permitted provided that the following conditions are met:
44
#
45

    
46
if [ -n "${IMAGES_FINAL_DIR}" -a "${IMAGES_FINAL_DIR}" != "/" ]; then
47
	mkdir -p ${IMAGES_FINAL_DIR}
48
else
49
	echo "IMAGES_FINAL_DIR is not defined"
50
	print_error_pfS
51
fi
52

    
53
lc() {
54
	echo "${1}" | tr '[[:upper:]]' '[[:lower:]]'
55
}
56

    
57
git_last_commit() {
58
	export CURRENT_COMMIT=$(git -C ${BUILDER_ROOT} log -1 --format='%H')
59
	export CURRENT_AUTHOR=$(git -C ${BUILDER_ROOT} log -1 --format='%an')
60
	echo ">>> Last known commit $CURRENT_AUTHOR - $CURRENT_COMMIT"
61
	echo "$CURRENT_COMMIT" > $SCRATCHDIR/build_commit_info.txt
62
}
63

    
64
# Create core pkg repository
65
core_pkg_create_repo() {
66
	if [ ! -d "${CORE_PKG_PATH}/All" ]; then
67
		return
68
	fi
69

    
70
	echo -n ">>> Creating core packages repository... "
71
	if pkg repo -q "${CORE_PKG_PATH}"; then
72
		echo "Done!"
73
	else
74
		echo "Failed!"
75
		print_error_pfS
76
	fi
77
}
78

    
79
# Create core pkg (base, kernel)
80
core_pkg_create() {
81
	local _template="${1}"
82
	local _flavor="${2}"
83
	local _version="${3}"
84
	local _root="${4}"
85

    
86
	[ -d "${CORE_PKG_TMP}" ] \
87
		&& rm -rf ${CORE_PKG_TMP}
88

    
89
	local _templates_path=${BUILDER_TOOLS}/templates/core_pkg/${_template}
90
	local _template_metadir=${_templates_path}/metadir
91
	local _metadir=${CORE_PKG_TMP}/${_template}_metadir
92

    
93
	if [ ! -d ${_template_metadir} ]; then
94
		echo "ERROR: Template dir not found for pkg ${_template}"
95
		exit
96
	fi
97

    
98
	mkdir -p ${CORE_PKG_TMP}
99

    
100
	cp -r ${_template_metadir} ${_metadir}
101

    
102
	local _manifest=${_metadir}/+MANIFEST
103
	local _plist=${CORE_PKG_TMP}/${_template}_plist
104
	local _exclude_plist=${CORE_PKG_TMP}/${_template}_exclude_plist
105

    
106
	if [ -f "${_templates_path}/pkg-plist" ]; then
107
		cp ${_templates_path}/pkg-plist ${_plist}
108
	else
109
		(cd ${_root} && find . -type f -or -type l | sed 's,^.,,' | sort -u) > ${_plist}
110
	fi
111

    
112
	if [ -f "${_templates_path}/exclude_plist" ]; then
113
		cp ${_templates_path}/exclude_plist ${_exclude_plist}
114
	else
115
		touch ${_exclude_plist}
116
	fi
117

    
118
	sed \
119
		-i '' \
120
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
121
		-e "s,%%PRODUCT_URL%%,${PRODUCT_URL},g" \
122
		-e "s,%%FLAVOR%%,${_flavor},g" \
123
		-e "s,%%VERSION%%,${_version},g" \
124
		${_metadir}/* \
125
		${_plist} \
126
		${exclude_plist}
127

    
128
	if [ -f "${_exclude_plist}" ]; then
129
		sort -u ${_exclude_plist} > ${_plist}.exclude
130
		mv ${_plist} ${_plist}.tmp
131
		comm -23 ${_plist}.tmp ${_plist}.exclude > ${_plist}
132
		rm -f ${_plist}.tmp ${plist}.exclude
133
	fi
134

    
135
	mkdir -p ${CORE_PKG_PATH}/All
136
	if ! pkg create -o ${CORE_PKG_PATH}/All -p ${_plist} -r ${_root} -m ${_metadir}; then
137
		echo ">>> ERROR: Error building package ${_template} ${_flavor}"
138
		print_error_pfS
139
	fi
140
}
141

    
142
# This routine will output that something went wrong
143
print_error_pfS() {
144
	echo
145
	echo "####################################"
146
	echo "Something went wrong, check errors!" >&2
147
	echo "####################################"
148
	echo
149
	echo "NOTE: a lot of times you can run './build.sh --clean-builder' to resolve."
150
	echo
151
	if [ "$1" != "" ]; then
152
		echo $1
153
	fi
154
	[ -n "${LOGFILE}" -a -f "${LOGFILE}" ] && \
155
		echo "Log saved on ${LOGFILE}" && \
156
		tail -n20 ${LOGFILE} >&2
157
	echo
158
	kill $$
159
	exit 1
160
}
161

    
162
prestage_on_ram_setup() {
163
	[ -d "${STAGE_CHROOT_DIR}" ] \
164
		|| mkdir -p ${STAGE_CHROOT_DIR}
165
	[ -d "${FINAL_CHROOT_DIR}" ] \
166
		|| mkdir -p ${FINAL_CHROOT_DIR}
167

    
168
	_AVAIL_MEM=$(($(sysctl -n hw.usermem) / 1024 / 1024))
169
	if [ $_AVAIL_MEM -lt 2000 ]; then
170
		echo ">>> Builder has less than 2GiB RAM skipping memory disks"
171
		return
172
	else
173
		echo "######################################################################################"
174
		echo
175
		echo ">>> Builder has more than 2GiB RAM enabling memory disks"
176
		echo ">>> WARNING: Remember to remove these memory disks by running $0 --disable-memorydisks"
177
		echo
178
		echo "######################################################################################"
179
	fi
180

    
181
	if df /dev/ufs/prestagebacking >/dev/null 2>&1; then
182
		echo ">>> Detected preexisting memory disk enabled for ${STAGE_CHROOT_DIR}."
183
	else
184
		mdconfig -a -t swap -u 10001 -s ${MEMORYDISK_SIZE}
185
		newfs -L prestagebacking -U /dev/md10001
186
		mount /dev/ufs/prestagebacking ${STAGE_CHROOT_DIR}
187
	fi
188

    
189
	if df /dev/ufs/finalstagebacking >/dev/null 2>&1; then
190
		echo ">>> Detected preexisting memory disk enabled for ${FINAL_CHROOT_DIR}."
191
	else
192
		mdconfig -a -t swap -u 10002 -s ${MEMORYDISK_SIZE}
193
		newfs -L finalstagebacking -U /dev/md10002
194
		mount /dev/ufs/finalstagebacking ${FINAL_CHROOT_DIR}
195
	fi
196
}
197

    
198
prestage_on_ram_cleanup() {
199
	if [ -c /dev/md10001 ]; then
200
		if [ -d ${STAGE_CHROOT_DIR} ]; then
201
			umount ${STAGE_CHROOT_DIR}
202
		fi
203
		mdconfig -d -u 10001
204
	fi
205
	if [ -c /dev/md10002 ]; then
206
		if [ -d ${FINAL_CHROOT_DIR} ]; then
207
			umount ${FINAL_CHROOT_DIR}
208
		fi
209
		mdconfig -d -u 10002
210
	fi
211
}
212

    
213
# This routine will verify that the kernel has been
214
# installed OK to the staging area.
215
ensure_kernel_exists() {
216
	if [ ! -f "$1/boot/kernel/kernel.gz" ]; then
217
		echo ">>> ERROR: Could not locate $1/boot/kernel.gz"
218
		print_error_pfS
219
	fi
220
	KERNEL_SIZE=$(stat -f "%z" $1/boot/kernel/kernel.gz)
221
	if [ "$KERNEL_SIZE" -lt 3500 ]; then
222
		echo ">>> ERROR: Kernel $1/boot/kernel.gz appears to be smaller than it should be: $KERNEL_SIZE"
223
		print_error_pfS
224
	fi
225
}
226

    
227
get_pkg_name() {
228
	echo "${PRODUCT_NAME}-${1}-${CORE_PKG_VERSION}"
229
}
230

    
231
# This routine builds all related kernels
232
build_all_kernels() {
233
	# Set KERNEL_BUILD_PATH if it has not been set
234
	if [ -z "${KERNEL_BUILD_PATH}" ]; then
235
		KERNEL_BUILD_PATH=$SCRATCHDIR/kernels
236
		echo ">>> KERNEL_BUILD_PATH has not been set. Setting to ${KERNEL_BUILD_PATH}!"
237
	fi
238

    
239
	[ -d "${KERNEL_BUILD_PATH}" ] \
240
		&& rm -rf ${KERNEL_BUILD_PATH}
241

    
242
	# Build embedded kernel
243
	for BUILD_KERNEL in $BUILD_KERNELS; do
244
		unset KERNCONF
245
		unset KERNEL_DESTDIR
246
		unset KERNELCONF
247
		unset KERNEL_NAME
248
		export KERNCONF=$BUILD_KERNEL
249
		export KERNEL_DESTDIR="$KERNEL_BUILD_PATH/$BUILD_KERNEL"
250
		export KERNELCONF="${FREEBSD_SRC_DIR}/sys/${TARGET}/conf/$BUILD_KERNEL"
251
		export KERNEL_NAME=${BUILD_KERNEL}
252

    
253
		LOGFILE="${BUILDER_LOGS}/kernel.${KERNCONF}.${TARGET}.log"
254
		echo ">>> Building $BUILD_KERNEL kernel."  | tee -a ${LOGFILE}
255

    
256
		if [ ! -e "${FREEBSD_SRC_DIR}/sys/${TARGET}/conf/${BUILD_KERNEL}" ]; then
257
			echo ">>> ERROR: Could not find $KERNELCONF"
258
			print_error_pfS
259
		fi
260

    
261
		if [ -n "${NO_BUILDKERNEL}" -a -f "${CORE_PKG_PATH}/All/$(get_pkg_name kernel-${KERNEL_NAME}).txz" ]; then
262
			echo ">>> NO_BUILDKERNEL set, skipping build" | tee -a ${LOGFILE}
263
			continue
264
		fi
265

    
266
		export SRC_CONF=${SRC_CONF}
267
		buildkernel
268

    
269
		echo ">>> Staging $BUILD_KERNEL kernel..." | tee -a ${LOGFILE}
270
		installkernel
271

    
272
		ensure_kernel_exists $KERNEL_DESTDIR
273

    
274
		echo -n ">>> Creating pkg of $KERNEL_NAME kernel to staging area..."  | tee -a ${LOGFILE}
275
		core_pkg_create kernel ${KERNEL_NAME} ${CORE_PKG_VERSION} ${KERNEL_DESTDIR}
276

    
277
		rm -rf $KERNEL_DESTDIR 2>&1 1>/dev/null
278

    
279
		echo ".Done" | tee -a ${LOGFILE}
280
	done
281
}
282

    
283
install_default_kernel() {
284
	if [ -z "${1}" ]; then
285
		echo ">>> ERROR: install_default_kernel called without a kernel config name"| tee -a ${LOGFILE}
286
		print_error_pfS
287
	fi
288

    
289
	export KERNEL_NAME="${1}"
290

    
291
	echo -n ">>> Installing kernel to be used by image ${KERNEL_NAME}..." | tee -a ${LOGFILE}
292

    
293
	# Copy kernel package to chroot, otherwise pkg won't find it to install
294
	if ! pkg_chroot_add ${FINAL_CHROOT_DIR} kernel-${KERNEL_NAME}; then
295
		echo ">>> ERROR: Error installing kernel package $(get_pkg_name kernel-${KERNEL_NAME}).txz" | tee -a ${LOGFILE}
296
		print_error_pfS
297
	fi
298

    
299
	# Lock kernel to avoid user end up removing it for any reason
300
	pkg_chroot ${FINAL_CHROOT_DIR} lock -q -y $(get_pkg_name kernel-${KERNEL_NAME})
301

    
302
	if [ ! -f $FINAL_CHROOT_DIR/boot/kernel/kernel.gz ]; then
303
		echo ">>> ERROR: No kernel installed on $FINAL_CHROOT_DIR and the resulting image will be unusable. STOPPING!" | tee -a ${LOGFILE}
304
		print_error_pfS
305
	fi
306
	mkdir -p $FINAL_CHROOT_DIR/pkgs
307
	if [ -z "${2}" -o -n "${INSTALL_EXTRA_KERNELS}" ]; then
308
		cp ${CORE_PKG_PATH}/All/$(get_pkg_name kernel-${KERNEL_NAME}).txz $FINAL_CHROOT_DIR/pkgs
309
		if [ -n "${INSTALL_EXTRA_KERNELS}" ]; then
310
			for _EXTRA_KERNEL in $INSTALL_EXTRA_KERNELS; do
311
				_EXTRA_KERNEL_PATH=${CORE_PKG_PATH}/All/$(get_pkg_name kernel-${_EXTRA_KERNEL}).txz
312
				if [ -f "${_EXTRA_KERNEL_PATH}" ]; then
313
					echo -n ". adding ${_EXTRA_KERNEL_PATH} on image /pkgs folder"
314
					cp ${_EXTRA_KERNEL_PATH} $FINAL_CHROOT_DIR/pkgs
315
				else
316
					echo ">>> ERROR: Requested kernel $(get_pkg_name kernel-${_EXTRA_KERNEL}).txz was not found to be put on image /pkgs folder!"
317
					print_error_pfS
318
				fi
319
			done
320
		fi
321
	fi
322
	echo "Done." | tee -a ${LOGFILE}
323

    
324
	unset KERNEL_NAME
325
}
326

    
327
# Creates a full update file
328
create_Full_update_tarball() {
329
	mkdir -p $UPDATESDIR
330

    
331
	customize_stagearea_for_image "fullupdate"
332
	install_default_kernel ${DEFAULT_KERNEL}
333

    
334
	rm -rf ${FINAL_CHROOT_DIR}/cf
335
	rm -rf ${FINAL_CHROOT_DIR}/conf
336
	rm -f ${FINAL_CHROOT_DIR}/etc/rc.conf
337
	rm -f ${FINAL_CHROOT_DIR}/etc/pwd.db 2>/dev/null
338
	rm -f ${FINAL_CHROOT_DIR}/etc/group 2>/dev/null
339
	rm -f ${FINAL_CHROOT_DIR}/etc/spwd.db 2>/dev/null
340
	rm -f ${FINAL_CHROOT_DIR}/etc/passwd 2>/dev/null
341
	rm -f ${FINAL_CHROOT_DIR}/etc/master.passwd 2>/dev/null
342
	rm -f ${FINAL_CHROOT_DIR}/etc/fstab 2>/dev/null
343
	rm -f ${FINAL_CHROOT_DIR}/etc/bogons 2>/dev/null
344
	# Remove loader.conf and friends.  Ticket #560
345
	rm ${FINAL_CHROOT_DIR}/boot/loader.conf 2>/dev/null
346
	rm ${FINAL_CHROOT_DIR}/boot/loader.conf.local 2>/dev/null
347

    
348
	# Old systems will run (pre|post)_upgrade_command from /tmp
349
	if [ -f ${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/pre_upgrade_command ]; then
350
		cp -p \
351
			${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/pre_upgrade_command \
352
			${FINAL_CHROOT_DIR}/tmp
353
	fi
354
	if [ -f ${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/post_upgrade_command ]; then
355
		cp -p \
356
			${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/post_upgrade_command \
357
			${FINAL_CHROOT_DIR}/tmp
358
	fi
359

    
360
	echo ">>> Creating ${UPDATES_TARBALL_FILENAME} ..." | tee -a ${LOGFILE}
361
	tar --exclude=./dev -czPf ${UPDATES_TARBALL_FILENAME} -C ${FINAL_CHROOT_DIR} .
362
}
363

    
364
# Outputs various set variables aka env
365
print_flags() {
366

    
367
	echo
368
	printf "             Product version: %s\n" $PRODUCT_VERSION
369
	printf "                   Stage DIR: %s\n" $STAGE_CHROOT_DIR
370
	printf "                 Updates dir: %s\n" $UPDATESDIR
371
	printf " Image Preparation Stage DIR: %s\n" $FINAL_CHROOT_DIR
372
	printf "                  Source DIR: %s\n" $FREEBSD_SRC_DIR
373
	printf "          FreeBSD repository: %s\n" $FREEBSD_REPO_BASE
374
	printf "          FreeBSD-src branch: %s\n" $FREEBSD_BRANCH
375
	printf "     FreeBSD original branch: %s\n" $FREEBSD_PARENT_BRANCH
376
	printf "               BUILD_KERNELS: %s\n" $BUILD_KERNELS
377
	printf "           Git Branch or Tag: %s\n" $GIT_REPO_BRANCH_OR_TAG
378
	printf "            MODULES_OVERRIDE: %s\n" $MODULES_OVERRIDE
379
	printf "    VMDK_DISK_CAPACITY_IN_GB: %s\n" $VMDK_DISK_CAPACITY_IN_GB
380
	printf "   OVA_FIRST_PART_SIZE_IN_GB: %s\n" $OVA_FIRST_PART_SIZE_IN_GB
381
	printf "    OVA_SWAP_PART_SIZE_IN_GB: %s\n" $OVA_SWAP_PART_SIZE_IN_GB
382
	printf "                 OVFTEMPLATE: %s\n" $OVFTEMPLATE
383
	printf "                     OVFVMDK: %s\n" $OVFVMDK
384
	printf "                    SRC_CONF: %s\n" $SRC_CONF
385
	printf "                     ISOPATH: %s\n" $ISOPATH
386
	printf "                MEMSTICKPATH: %s\n" $MEMSTICKPATH
387
	printf "    UPDATES_TARBALL_FILENAME: %s\n" $UPDATES_TARBALL_FILENAME
388
if [ -n "$SHOW_ENV" ]; then
389
	for LINE in $(env | egrep -v '(terminal|PASS|NAME|USER|SSH|GROUP|HOST)'); do
390
		echo "SHOW_ENV: $LINE"
391
	done
392
fi
393
	echo
394
}
395

    
396
# This builds FreeBSD (make buildworld)
397
# Imported from FreeSBIE
398
make_world() {
399

    
400
	LOGFILE=${BUILDER_LOGS}/buildworld.${TARGET}
401
	if [ -n "${NO_BUILDWORLD}" ]; then
402
		echo ">>> NO_BUILDWORLD set, skipping build" | tee -a ${LOGFILE}
403
		return
404
	fi
405

    
406
	# Set SRC_CONF variable if it's not already set.
407
	if [ -z "${SRC_CONF}" ]; then
408
		echo ">>> SRC_CONF is unset make sure this is what you want!" | tee -a ${LOGFILE}
409
	else
410
		echo ">>> Setting SRC_CONF to $SRC_CONF" | tee -a ${LOGFILE}
411
	fi
412

    
413
	# Set default parameters
414
	export MAKE_ARGS="${MAKEJ_WORLD} __MAKE_CONF=${MAKE_CONF} SRCCONF=${SRC_CONF} TARGET=${TARGET} TARGET_ARCH=${TARGET_ARCH}"
415

    
416
	echo ">>> LOGFILE set to $LOGFILE." | tee -a ${LOGFILE}
417
	makeargs="${MAKE_ARGS}"
418
	echo ">>> Building world for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
419
	echo ">>> Builder is running the command: env LOCAL_ITOOLS=\"${EXTRA_TOOLS}\" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} -DNO_CLEAN ${makeargs} buildworld" | tee -a ${LOGFILE}
420
	(env LOCAL_ITOOLS="${EXTRA_TOOLS}" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} -DNO_CLEAN ${makeargs} buildworld || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
421
	echo ">>> Building world for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
422

    
423
	LOGFILE=${BUILDER_LOGS}/installworld.${TARGET}
424
	echo ">>> LOGFILE set to $LOGFILE." | tee -a ${LOGFILE}
425
	# Create if cleaned up
426
	makeargs="${MAKE_ARGS} DESTDIR=${STAGE_CHROOT_DIR} WITHOUT_TOOLCHAIN=1"
427
	echo ">>> Installing world for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
428
	echo ">>> Builder is running the command: env LOCAL_ITOOLS=\"${EXTRA_TOOLS}\" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} installworld" | tee -a ${LOGFILE}
429
	(env LOCAL_ITOOLS="${EXTRA_TOOLS}" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} installworld || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
430
	echo ">>> Installing world for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
431

    
432
	makeargs="${MAKE_ARGS} DESTDIR=${STAGE_CHROOT_DIR}"
433
	echo ">>> Distribution world for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
434
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} distribution " | tee -a ${LOGFILE}
435
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} distribution  || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
436
	echo ">>> Distribution world for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
437

    
438
	[ -d "${STAGE_CHROOT_DIR}/usr/local/bin" ] \
439
		|| mkdir -p ${STAGE_CHROOT_DIR}/usr/local/bin
440
	makeargs="${MAKE_ARGS} DESTDIR=${STAGE_CHROOT_DIR}"
441
	echo ">>> Building and installing crypto tools and athstats for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
442
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/crypto ${makeargs} clean all install " | tee -a ${LOGFILE}
443
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/crypto ${makeargs} clean all install || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
444
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} clean" | tee -a ${LOGFILE}
445
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} clean || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
446
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} all" | tee -a ${LOGFILE}
447
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} all || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
448
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} install" | tee -a ${LOGFILE}
449
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} install || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
450
	echo ">>> Building and installing crypto tools and athstats for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
451

    
452
	unset makeargs
453
}
454

    
455
# This routine originated in nanobsd.sh
456
nanobsd_set_flash_details () {
457
	a1=$(echo $1 | tr '[:upper:]' '[:lower:]')
458

    
459
	# Source:
460
	#	SanDisk CompactFlash Memory Card
461
	#	Product Manual
462
	#	Version 10.9
463
	#	Document No. 20-10-00038
464
	#	April 2005
465
	# Table 2-7
466
	# NB: notice math error in SDCFJ-4096-388 line.
467
	#
468
	case "${a1}" in
469
		1024|1024m|1024mb|1g)
470
			NANO_MEDIASIZE=$((997129216/512))
471
			;;
472
		2048|2048m|2048mb|2g)
473
			NANO_MEDIASIZE=$((1989999616/512))
474
			;;
475
		4096|4096m|4096mb|4g)
476
			NANO_MEDIASIZE=$((3989999616/512))
477
			;;
478
		8192|8192m|8192mb|8g)
479
			NANO_MEDIASIZE=$((7989999616/512))
480
			;;
481
		16384|16384m|16384mb|16g)
482
			NANO_MEDIASIZE=$((15989999616/512))
483
			;;
484
		*)
485
			echo "Unknown Flash capacity"
486
			exit 2
487
			;;
488
	esac
489

    
490
	NANO_HEADS=16
491
	NANO_SECTS=63
492

    
493
	echo ">>> [nanoo] $1"
494
	echo ">>> [nanoo] NANO_MEDIASIZE: $NANO_MEDIASIZE"
495
	echo ">>> [nanoo] NANO_HEADS: $NANO_HEADS"
496
	echo ">>> [nanoo] NANO_SECTS: $NANO_SECTS"
497
	echo ">>> [nanoo] NANO_BOOT0CFG: $NANO_BOOT0CFG"
498
}
499

    
500
# This routine originated in nanobsd.sh
501
create_nanobsd_diskimage () {
502
	if [ -z "${1}" ]; then
503
		echo ">>> ERROR: Type of image has not been specified"
504
		print_error_pfS
505
	fi
506
	if [ -z "${2}" ]; then
507
		echo ">>> ERROR: Size of image has not been specified"
508
		print_error_pfS
509
	fi
510

    
511
	if [ "${1}" = "nanobsd" ]; then
512
		# It's serial
513
		export NANO_BOOTLOADER="boot/boot0sio"
514
	elif [ "${1}" = "nanobsd-vga" ]; then
515
		# It's vga
516
		export NANO_BOOTLOADER="boot/boot0"
517
	else
518
		echo ">>> ERROR: Type of image to create unknown"
519
		print_error_pfS
520
	fi
521

    
522
	if [ -z "${2}" ]; then
523
		echo ">>> ERROR: Media size(s) not specified."
524
		print_error_pfS
525
	fi
526

    
527
	if [ -z "${2}" ]; then
528
		echo ">>> ERROR: FLASH_SIZE is not set."
529
		print_error_pfS
530
	fi
531

    
532
	LOGFILE=${BUILDER_LOGS}/${1}.${TARGET}
533
	# Prepare folder to be put in image
534
	customize_stagearea_for_image "${1}"
535
	install_default_kernel ${DEFAULT_KERNEL} "no"
536

    
537
	echo ">>> Fixing up NanoBSD Specific items..." | tee -a ${LOGFILE}
538

    
539
	echo "nanobsd" > $FINAL_CHROOT_DIR/etc/platform
540

    
541
	local BOOTCONF=${FINAL_CHROOT_DIR}/boot.config
542
	local LOADERCONF=${FINAL_CHROOT_DIR}/boot/loader.conf
543

    
544
	if [ "${1}" = "nanobsd" ]; then
545
		# Tell loader to use serial console early.
546
		echo "-S115200 -h" >> ${BOOTCONF}
547
	fi
548

    
549
	# Remove old console options if present.
550
	[ -f "${LOADERCONF}" ] \
551
		&& sed -i "" -Ee "/(console|boot_multicons|boot_serial|hint.uart)/d" ${LOADERCONF}
552
	# Activate serial console+video console in loader.conf
553
	echo 'autoboot_delay="5"' >> ${LOADERCONF}
554
	echo 'loader_color="NO"' >> ${LOADERCONF}
555
	echo 'beastie_disable="YES"' >> ${LOADERCONF}
556
	echo 'boot_serial="YES"' >> ${LOADERCONF}
557
	echo 'console="comconsole"' >> ${LOADERCONF}
558
	echo 'comconsole_speed="115200"' >> ${LOADERCONF}
559

    
560
	for _NANO_MEDIASIZE in ${2}; do
561
		if [ -z "${_NANO_MEDIASIZE}" ]; then
562
			continue;
563
		fi
564

    
565
		echo ">>> building NanoBSD(${1}) disk image with size ${_NANO_MEDIASIZE} for platform (${TARGET})..." | tee -a ${LOGFILE}
566
		echo "" > $BUILDER_LOGS/nanobsd_cmds.sh
567

    
568
		IMG="${IMAGES_FINAL_DIR}/${PRODUCT_NAME}-${PRODUCT_VERSION}-${_NANO_MEDIASIZE}-${TARGET}-${1}${TIMESTAMP_SUFFIX}.img"
569
		IMGUPDATE="${IMAGES_FINAL_DIR}/${PRODUCT_NAME}-${PRODUCT_VERSION}-${_NANO_MEDIASIZE}-${TARGET}-${1}-upgrade${TIMESTAMP_SUFFIX}.img"
570

    
571
		nanobsd_set_flash_details ${_NANO_MEDIASIZE}
572

    
573
		# These are defined in FlashDevice and on builder_default.sh
574
		echo $NANO_MEDIASIZE \
575
			$NANO_IMAGES \
576
			$NANO_SECTS \
577
			$NANO_HEADS \
578
			$NANO_CODESIZE \
579
			$NANO_CONFSIZE \
580
			$NANO_DATASIZE |
581
awk '
582
{
583
	printf "# %s\n", $0
584

    
585
	# size of cylinder in sectors
586
	cs = $3 * $4
587

    
588
	# number of full cylinders on media
589
	cyl = int ($1 / cs)
590

    
591
	# output fdisk geometry spec, truncate cyls to 1023
592
	if (cyl <= 1023)
593
		print "g c" cyl " h" $4 " s" $3
594
	else
595
		print "g c" 1023 " h" $4 " s" $3
596

    
597
	if ($7 > 0) {
598
		# size of data partition in full cylinders
599
		dsl = int (($7 + cs - 1) / cs)
600
	} else {
601
		dsl = 0;
602
	}
603

    
604
	# size of config partition in full cylinders
605
	csl = int (($6 + cs - 1) / cs)
606

    
607
	if ($5 == 0) {
608
		# size of image partition(s) in full cylinders
609
		isl = int ((cyl - dsl - csl) / $2)
610
	} else {
611
		isl = int (($5 + cs - 1) / cs)
612
	}
613

    
614
	# First image partition start at second track
615
	print "p 1 165 " $3, isl * cs - $3
616
	c = isl * cs;
617

    
618
	# Second image partition (if any) also starts offset one
619
	# track to keep them identical.
620
	if ($2 > 1) {
621
		print "p 2 165 " $3 + c, isl * cs - $3
622
		c += isl * cs;
623
	}
624

    
625
	# Config partition starts at cylinder boundary.
626
	print "p 3 165 " c, csl * cs
627
	c += csl * cs
628

    
629
	# Data partition (if any) starts at cylinder boundary.
630
	if ($7 > 0) {
631
		print "p 4 165 " c, dsl * cs
632
	} else if ($7 < 0 && $1 > c) {
633
		print "p 4 165 " c, $1 - c
634
	} else if ($1 < c) {
635
		print "Disk space overcommitted by", \
636
		    c - $1, "sectors" > "/dev/stderr"
637
		exit 2
638
	}
639

    
640
	# Force slice 1 to be marked active. This is necessary
641
	# for booting the image from a USB device to work.
642
	print "a 1"
643
}
644
	' > ${IMAGES_FINAL_DIR}/_.fdisk
645

    
646
		MNT=${IMAGES_FINAL_DIR}/_.mnt
647
		mkdir -p ${MNT}
648

    
649
		dd if=/dev/zero of=${IMG} bs=${NANO_SECTS}b \
650
			count=0 seek=$((${NANO_MEDIASIZE}/${NANO_SECTS})) 2>&1 >> ${LOGFILE}
651

    
652
		MD=$(mdconfig -a -t vnode -f ${IMG} -x ${NANO_SECTS} -y ${NANO_HEADS})
653
		trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
654

    
655
		fdisk -i -f ${IMAGES_FINAL_DIR}/_.fdisk ${MD} 2>&1 >> ${LOGFILE}
656
		fdisk ${MD} 2>&1 >> ${LOGFILE}
657

    
658
		boot0cfg -B -b ${FINAL_CHROOT_DIR}/${NANO_BOOTLOADER} ${NANO_BOOT0CFG} ${MD} 2>&1 >> ${LOGFILE}
659

    
660
		# Create first image
661
		bsdlabel -m i386 -w -B -b ${FINAL_CHROOT_DIR}/boot/boot ${MD}s1 2>&1 >> ${LOGFILE}
662
		bsdlabel -m i386 ${MD}s1 2>&1 >> ${LOGFILE}
663
		local _label=$(lc ${PRODUCT_NAME})
664
		newfs -L ${_label}0 ${NANO_NEWFS} /dev/${MD}s1a 2>&1 >> ${LOGFILE}
665
		mount /dev/ufs/${_label}0 ${MNT}
666
		if [ $? -ne 0 ]; then
667
			echo ">>> ERROR: Something wrong happened during mount of first slice image creation. STOPPING!" | tee -a ${LOGFILE}
668
			print_error_pfS
669
		fi
670
		# Consider the unmounting as well
671
		trap "umount /dev/ufs/${_label}0; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
672

    
673
		clone_directory_contents ${FINAL_CHROOT_DIR} ${MNT}
674

    
675
		# Set NanoBSD image size
676
		echo "${_NANO_MEDIASIZE}" > ${MNT}/etc/nanosize.txt
677
		rm -f $MNT/cf/conf/* 2>/dev/null
678

    
679
		echo "/dev/ufs/${_label}0 / ufs ro,sync,noatime 1 1" > ${MNT}/etc/fstab
680
		if [ $NANO_CONFSIZE -gt 0 ] ; then
681
			echo "/dev/ufs/cf /cf ufs ro,sync,noatime 1 1" >> ${MNT}/etc/fstab
682
		fi
683

    
684
		umount ${MNT}
685
		# Restore the original trap
686
		trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
687

    
688
		# Setting NANO_IMAGES to 1 and NANO_INIT_IMG2 will tell
689
		# NanoBSD to only create one partition.  We default to 2
690
		# partitions in case anything happens to the first the
691
		# operator can boot from the 2nd and should be OK.
692

    
693
		# Before just going to use dd for duplicate think!
694
		# The images are created as sparse so lets take advantage
695
		# of that by just exec some commands.
696
		if [ $NANO_IMAGES -gt 1 -a $NANO_INIT_IMG2 -gt 0 ] ; then
697
			# Duplicate to second image (if present)
698
			echo ">>> Creating NanoBSD second slice by duplicating first slice." | tee -a ${LOGFILE}
699
			# Create second image
700
			dd if=/dev/${MD}s1 of=/dev/${MD}s2 conv=sparse bs=64k 2>&1 >> ${LOGFILE}
701
			tunefs -L ${_label}1 /dev/${MD}s2a 2>&1 >> ${LOGFILE}
702
			mount /dev/ufs/${_label}1 ${MNT}
703
			if [ $? -ne 0 ]; then
704
				echo ">>> ERROR: Something wrong happened during mount of second slice image creation. STOPPING!" | tee -a ${LOGFILE}
705
				print_error_pfS
706
			fi
707
			# Consider the unmounting as well
708
			trap "umount /dev/ufs/${_label}1; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
709

    
710
			echo "/dev/ufs/${_label}1 / ufs ro,sync,noatime 1 1" > ${MNT}/etc/fstab
711
			if [ $NANO_CONFSIZE -gt 0 ] ; then
712
				echo "/dev/ufs/cf /cf ufs ro,sync,noatime 1 1" >> ${MNT}/etc/fstab
713
			fi
714

    
715
			umount ${MNT}
716
			# Restore the trap back
717
			trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
718
		fi
719

    
720
		# Create Data slice, if any.
721
		# Note the changing of the variable to NANO_CONFSIZE
722
		# from NANO_DATASIZE.  We also added glabel support
723
		# and populate the Product configuration from the /cf
724
		# directory located in FINAL_CHROOT_DIR
725
		if [ $NANO_CONFSIZE -gt 0 ] ; then
726
			echo ">>> Creating /cf area to hold config.xml"
727
			newfs -L cf ${NANO_NEWFS} /dev/${MD}s3 2>&1 >> ${LOGFILE}
728
			# Mount data partition and copy contents of /cf
729
			# Can be used later to create custom default config.xml while building
730
			mount /dev/ufs/cf ${MNT}
731
			if [ $? -ne 0 ]; then
732
				echo ">>> ERROR: Something wrong happened during mount of cf slice image creation. STOPPING!" | tee -a ${LOGFILE}
733
				print_error_pfS
734
			fi
735
			# Consider the unmounting as well
736
			trap "umount /dev/ufs/cf; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
737

    
738
			clone_directory_contents ${FINAL_CHROOT_DIR}/cf ${MNT}
739

    
740
			umount ${MNT}
741
			# Restore the trap back
742
			trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
743
		else
744
			">>> [nanoo] NANO_CONFSIZE is not set. Not adding a /conf partition.. You sure about this??" | tee -a ${LOGFILE}
745
		fi
746

    
747
		echo ">>> [nanoo] Creating NanoBSD upgrade file from first slice..." | tee -a ${LOGFILE}
748
		dd if=/dev/${MD}s1 of=$IMGUPDATE conv=sparse bs=64k 2>&1 >> ${LOGFILE}
749

    
750
		mdconfig -d -u $MD
751
		# Restore default action
752
		trap "-" 1 2 15 EXIT
753

    
754
		# Check each image and ensure that they are over
755
		# 3 megabytes.  If either image is under 20 megabytes
756
		# in size then error out.
757
		IMGSIZE=$(stat -f "%z" ${IMG})
758
		IMGUPDATESIZE=$(stat -f "%z" ${IMGUPDATE})
759
		CHECKSIZE="20040710"
760
		if [ "$IMGSIZE" -lt "$CHECKSIZE" ]; then
761
			echo ">>> ERROR: Something went wrong when building NanoBSD.  The image size is under 20 megabytes!" | tee -a ${LOGFILE}
762
			print_error_pfS
763
		fi
764
		if [ "$IMGUPDATESIZE" -lt "$CHECKSIZE" ]; then
765
			echo ">>> ERROR: Something went wrong when building NanoBSD upgrade image.  The image size is under 20 megabytes!" | tee -a ${LOGFILE}
766
			print_error_pfS
767
		fi
768

    
769
		# Wrap up the show, Johnny
770
		echo ">>> NanoBSD Image completed for size: $_NANO_MEDIASIZE." | tee -a ${LOGFILE}
771

    
772
		gzip -qf $IMG &
773
		_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
774
		gzip -qf $IMGUPDATE &
775
		_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
776
	done
777

    
778
	unset IMG
779
	unset IMGUPDATE
780
	unset IMGUPDATESIZE
781
	unset IMGSIZE
782

    
783
	ls -lah $IMAGES_FINAL_DIR
784
}
785

    
786
# This routine creates a ova image that contains
787
# a ovf and vmdk file. These files can be imported
788
# right into vmware or virtual box.
789
# (and many other emulation platforms)
790
# http://www.vmware.com/pdf/ovf_whitepaper_specification.pdf
791
create_ova_image() {
792
	# XXX create a .ovf php creator that you can pass:
793
	#     1. populatedSize
794
	#     2. license
795
	#     3. product name
796
	#     4. version
797
	#     5. number of network interface cards
798
	#     6. allocationUnits
799
	#     7. capacity
800
	#     8. capacityAllocationUnits
801

    
802
	LOGFILE=${BUILDER_LOGS}/ova.${TARGET}.log
803

    
804
	[ -d "${OVA_TMP}" ] \
805
		&& rm -rf ${OVA_TMP}
806

    
807
	mkdir -p ${OVA_TMP}
808

    
809
	# Prepare folder to be put in image
810
	customize_stagearea_for_image "ova"
811
	install_default_kernel ${DEFAULT_KERNEL} "no"
812

    
813
	# Fill fstab
814
	echo ">>> Installing platform specific items..." | tee -a ${LOGFILE}
815
	echo "/dev/gpt/${PRODUCT_NAME}	/	ufs		rw	0	0" > ${FINAL_CHROOT_DIR}/etc/fstab
816
	echo "/dev/gpt/swap0	none	swap	sw	0	0" >> ${FINAL_CHROOT_DIR}/etc/fstab
817

    
818
	# Create / partition
819
	echo -n ">>> Creating / partition... " | tee -a ${LOGFILE}
820
	makefs \
821
		-B little \
822
		-o label=${PRODUCT_NAME} \
823
		-s ${OVA_FIRST_PART_SIZE_IN_GB}g \
824
		${OVA_TMP}/${OVFUFS} \
825
		${FINAL_CHROOT_DIR} 2>&1 >> ${LOGFILE}
826

    
827
	if [ $? -ne 0 -o ! -f ${OVA_TMP}/${OVFUFS} ]; then
828
		if [ -f ${OVA_TMP}/${OVFUFS} ]; then
829
			rm -f ${OVA_TMP}/${OVFUFS}
830
		fi
831
		echo "Failed!" | tee -a ${LOGFILE}
832
		echo ">>> ERROR: Error creating vmdk / partition. STOPPING!" | tee -a ${LOGFILE}
833
		print_error_pfS
834
	fi
835
	echo "Done!" | tee -a ${LOGFILE}
836

    
837
	# Create raw disk
838
	echo -n ">>> Creating raw disk... " | tee -a ${LOGFILE}
839
	mkimg \
840
		-s gpt \
841
		-f raw \
842
		-b /boot/pmbr \
843
		-p freebsd-boot:=/boot/gptboot \
844
		-p freebsd-ufs/${PRODUCT_NAME}:=${OVA_TMP}/${OVFUFS} \
845
		-p freebsd-swap/swap0::${OVA_SWAP_PART_SIZE} \
846
		-o ${OVA_TMP}/${OVFRAW} 2>&1 >> ${LOGFILE}
847

    
848
	if [ $? -ne 0 -o ! -f ${OVA_TMP}/${OVFRAW} ]; then
849
		if [ -f ${OVA_TMP}/${OVFUFS} ]; then
850
			rm -f ${OVA_TMP}/${OVFUFS}
851
		fi
852
		if [ -f ${OVA_TMP}/${OVFRAW} ]; then
853
			rm -f ${OVA_TMP}/${OVFRAW}
854
		fi
855
		echo "Failed!" | tee -a ${LOGFILE}
856
		echo ">>> ERROR: Error creating temporary vmdk image. STOPPING!" | tee -a ${LOGFILE}
857
		print_error_pfS
858
	fi
859
	echo "Done!" | tee -a ${LOGFILE}
860

    
861
	# We don't need it anymore
862
	rm -f ${OVA_TMP}/${OVFUFS} >/dev/null 2>&1
863

    
864
	# Convert raw to vmdk
865
	echo -n ">>> Creating vmdk disk... " | tee -a ${LOGFILE}
866
	vmdktool -z9 -v ${OVA_TMP}/${OVFVMDK} ${OVA_TMP}/${OVFRAW}
867

    
868
	if [ $? -ne 0 -o ! -f ${OVA_TMP}/${OVFVMDK} ]; then
869
		if [ -f ${OVA_TMP}/${OVFRAW} ]; then
870
			rm -f ${OVA_TMP}/${OVFRAW}
871
		fi
872
		if [ -f ${OVA_TMP}/${OVFVMDK} ]; then
873
			rm -f ${OVA_TMP}/${OVFVMDK}
874
		fi
875
		echo "Failed!" | tee -a ${LOGFILE}
876
		echo ">>> ERROR: Error creating vmdk image. STOPPING!" | tee -a ${LOGFILE}
877
		print_error_pfS
878
	fi
879
	echo "Done!" | tee -a ${LOGFILE}
880

    
881
	rm -f ${OVA_TMP}/i${OVFRAW}
882

    
883
	ova_setup_ovf_template
884

    
885
	echo -n ">>> Writing final ova image... " | tee -a ${LOGFILE}
886
	# Create OVA file for vmware
887
	gtar -C ${OVA_TMP} -cpf ${OVAPATH} ${PRODUCT_NAME}.ovf ${OVFVMDK}
888
	echo "Done!" | tee -a ${LOGFILE}
889
	rm -f ${OVA_TMP}/${OVFVMDK} >/dev/null 2>&1
890

    
891
	echo ">>> OVA created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
892
}
893

    
894
# called from create_ova_image
895
ova_setup_ovf_template() {
896
	if [ ! -f ${OVFTEMPLATE} ]; then
897
		echo ">>> ERROR: OVF template file (${OVFTEMPLATE}) not found."
898
		print_error_pfS
899
	fi
900

    
901
	#  OperatingSystemSection (${PRODUCT_NAME}.ovf)
902
	#  42   FreeBSD 32-Bit
903
	#  78   FreeBSD 64-Bit
904
	if [ "${TARGET}" = "amd64" ]; then
905
		local _os_id="78"
906
		local _os_type="freebsd64Guest"
907
		local _os_descr="FreeBSD 64-Bit"
908
	elif [ "${TARGET}" = "i386" ]; then
909
		local _os_id="42"
910
		local _os_type="freebsdGuest"
911
		local _os_descr="FreeBSD"
912
	else
913
		echo ">>> ERROR: Platform not supported for OVA (${TARGET})"
914
		print_error_pfS
915
	fi
916

    
917
	local POPULATED_SIZE=$(du -d0 -k $FINAL_CHROOT_DIR | cut -f1)
918
	local POPULATED_SIZE_IN_BYTES=$((${POPULATED_SIZE}*1024))
919
	local VMDK_FILE_SIZE=$(stat -f "%z" ${OVA_TMP}/${OVFVMDK})
920

    
921
	sed \
922
		-e "s,%%VMDK_FILE_SIZE%%,${VMDK_FILE_SIZE},g" \
923
		-e "s,%%VMDK_DISK_CAPACITY_IN_GB%%,${VMDK_DISK_CAPACITY_IN_GB},g" \
924
		-e "s,%%POPULATED_SIZE_IN_BYTES%%,${POPULATED_SIZE_IN_BYTES},g" \
925
		-e "s,%%OS_ID%%,${_os_id},g" \
926
		-e "s,%%OS_TYPE%%,${_os_type},g" \
927
		-e "s,%%OS_DESCR%%,${_os_descr},g" \
928
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
929
		-e "s,%%PRODUCT_VERSION%%,${PRODUCT_VERSION},g" \
930
		-e "s,%%PRODUCT_URL%%,${PRODUCT_URL},g" \
931
		-e "/^%%PRODUCT_LICENSE%%/r ${BUILDER_ROOT}/license.txt" \
932
		-e "/^%%PRODUCT_LICENSE%%/d" \
933
		${OVFTEMPLATE} > ${OVA_TMP}/${PRODUCT_NAME}.ovf
934
}
935

    
936
# Cleans up previous builds
937
clean_builder() {
938
	# Clean out directories
939
	echo ">>> Cleaning up previous build environment...Please wait!"
940

    
941
	staginareas_clean_each_run
942

    
943
	if [ -d "${STAGE_CHROOT_DIR}" ]; then
944
		BASENAME=$(basename ${STAGE_CHROOT_DIR})
945
		echo -n ">>> Cleaning ${STAGE_CHROOT_DIR} ..."
946
		chflags -R noschg ${STAGE_CHROOT_DIR} 2>&1 >/dev/null
947
		rm -rf ${STAGE_CHROOT_DIR}/* 2>/dev/null
948
		echo "Done."
949
	fi
950

    
951
	if [ -z "${NO_CLEAN_FREEBSD_OBJ}" -a -d "${FREEBSD_SRC_DIR}" ]; then
952
		OBJTREE=$(env TARGET=${TARGET} TARGET_ARCH=${TARGET_ARCH} make -C ${FREEBSD_SRC_DIR} -V OBJTREE)
953
		if [ -d "${OBJTREE}" ]; then
954
			echo -n ">>> Cleaning FreeBSD objects dir staging..."
955
			echo -n "."
956
			chflags -R noschg ${OBJTREE} 2>&1 >/dev/null
957
			echo -n "."
958
			rm -rf ${OBJTREE}/*
959
			echo "Done!"
960
		fi
961
		if [ -d "${KERNEL_BUILD_PATH}" ]; then
962
			echo -n ">>> Cleaning previously built kernel stage area..."
963
			rm -rf $KERNEL_BUILD_PATH/*
964
			echo "Done!"
965
		fi
966
	fi
967
	mkdir -p $KERNEL_BUILD_PATH
968

    
969
	echo -n ">>> Cleaning previously built images..."
970
	rm -rf $IMAGES_FINAL_DIR/*
971
	rm -rf $STAGINGAREA/*
972
	echo "Done!"
973

    
974
	if [ -z "${NO_CLEAN_FREEBSD_SRC}" ]; then
975
		if [ -d "$FREEBSD_SRC_DIR" ]; then
976
			echo -n ">>> Ensuring $FREEBSD_SRC_DIR is clean..."
977
			rm -rf ${FREEBSD_SRC_DIR}
978
			echo "Done!"
979
		fi
980
	fi
981

    
982
	echo -n ">>> Cleaning previous builder logs..."
983
	if [ -d "$BUILDER_LOGS" ]; then
984
		rm -rf ${BUILDER_LOGS}
985
	fi
986
	mkdir -p ${BUILDER_LOGS}
987

    
988
	echo "Done!"
989

    
990
	echo ">>> Cleaning of builder environment has finished."
991
}
992

    
993
clone_directory_contents() {
994
	if [ ! -d "$1" -o ! -d "$2" ]; then
995
		if [ -z "${LOGFILE}" ]; then
996
			echo ">>> ERROR: Argument $1 supplied is not a directory!"
997
		else
998
			echo ">>> ERROR: Argument $1 supplied is not a directory!" | tee -a ${LOGFILE}
999
		fi
1000
		print_error_pfS
1001
	fi
1002
	echo -n ">>> Using TAR to clone $1 to $2 ..."
1003
	tar -C ${1} -c -f - . | tar -C ${2} -x -p -f -
1004
	echo "Done!"
1005
}
1006

    
1007
clone_to_staging_area() {
1008
	# Clone everything to the final staging area
1009
	echo -n ">>> Cloning everything to ${STAGE_CHROOT_DIR} staging area..."
1010
	LOGFILE=${BUILDER_LOGS}/cloning.${TARGET}.log
1011

    
1012
	tar -C ${PRODUCT_SRC} -c -f - . | \
1013
		tar -C ${STAGE_CHROOT_DIR} -x -p -f -
1014

    
1015
	if [ -f ${STAGE_CHROOT_DIR}/etc/master.passwd ]; then
1016
		chroot ${STAGE_CHROOT_DIR} cap_mkdb /etc/master.passwd
1017
		chroot ${STAGE_CHROOT_DIR} pwd_mkdb /etc/master.passwd
1018
	fi
1019
	mkdir -p ${STAGE_CHROOT_DIR}/etc/mtree
1020
	mtree -Pcp ${STAGE_CHROOT_DIR}/var > ${STAGE_CHROOT_DIR}/etc/mtree/var.dist
1021
	mtree -Pcp ${STAGE_CHROOT_DIR}/etc > ${STAGE_CHROOT_DIR}/etc/mtree/etc.dist
1022
	if [ -d ${STAGE_CHROOT_DIR}/usr/local/etc ]; then
1023
		mtree -Pcp ${STAGE_CHROOT_DIR}/usr/local/etc > ${STAGE_CHROOT_DIR}/etc/mtree/localetc.dist
1024
	fi
1025

    
1026
	## Add buildtime and lastcommit information
1027
	# This is used for detecting updates.
1028
	echo "$BUILTDATESTRING" > $STAGE_CHROOT_DIR/etc/version.buildtime
1029
	# Record last commit info if it is available.
1030
	if [ -f $SCRATCHDIR/build_commit_info.txt ]; then
1031
		cp $SCRATCHDIR/build_commit_info.txt $STAGE_CHROOT_DIR/etc/version.lastcommit
1032
	fi
1033

    
1034
	local _exclude_files="${CORE_PKG_TMP}/base_exclude_files"
1035
	sed \
1036
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
1037
		-e "s,%%FLAVOR%%,${_flavor},g" \
1038
		-e "s,%%VERSION%%,${_version},g" \
1039
		${BUILDER_TOOLS}/templates/core_pkg/base/exclude_files \
1040
		> ${_exclude_files}
1041

    
1042
	mkdir -p ${STAGE_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME} >/dev/null 2>&1
1043
	mtree \
1044
		-c \
1045
		-k uid,gid,mode,size,flags,sha256digest \
1046
		-p ${STAGE_CHROOT_DIR} \
1047
		-X ${_exclude_files} \
1048
		> ${STAGE_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/base.mtree
1049
	tar \
1050
		-C ${STAGE_CHROOT_DIR} \
1051
		-cJf ${STAGE_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/base.txz \
1052
		-X ${_exclude_files} \
1053
		.
1054

    
1055
	core_pkg_create base "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1056
	core_pkg_create base-nanobsd "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1057
	core_pkg_create default-config "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1058

    
1059
	local DEFAULTCONF=${STAGE_CHROOT_DIR}/conf.default/config.xml
1060

    
1061
	# Change default interface names to match vmware driver
1062
	sed -i '' -e 's,em0,vmx0,' -e 's,em1,vmx1,' ${DEFAULTCONF}
1063
	core_pkg_create default-config-vmware "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1064

    
1065
	# Restore default values to be used by serial package
1066
	sed -i '' -e 's,vmx0,em0,' -e 's,vmx1,em1,' ${DEFAULTCONF}
1067

    
1068
	# Activate serial console in config.xml
1069
	# If it was there before, clear the setting to be sure we don't add it twice.
1070
	sed -i "" -e "/		<enableserial\/>/d" ${DEFAULTCONF}
1071
	# Enable serial in the config
1072
	sed -i "" -e "s/	<\/system>/		<enableserial\/>\\$(echo -e \\\n)	<\/system>/" ${DEFAULTCONF}
1073

    
1074
	echo force > ${STAGE_CHROOT_DIR}/cf/conf/enableserial_force
1075

    
1076
	core_pkg_create default-config-serial "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1077

    
1078
	rm -f ${STAGE_CHROOT_DIR}/cf/conf/enableserial_force
1079
	rm -f ${STAGE_CHROOT_DIR}/cf/conf/config.xml
1080

    
1081
	# Make sure pkg is present
1082
	pkg_bootstrap ${STAGE_CHROOT_DIR}
1083

    
1084
	echo "Done!"
1085
}
1086

    
1087
create_final_staging_area() {
1088
	if [ -z "${FINAL_CHROOT_DIR}" ]; then
1089
		echo ">>> ERROR: FINAL_CHROOT_DIR is not set, cannot continue!" | tee -a ${LOGFILE}
1090
		print_error_pfS
1091
	fi
1092

    
1093
	if [ -d "${FINAL_CHROOT_DIR}" ]; then
1094
		echo -n ">>> Previous ${FINAL_CHROOT_DIR} detected cleaning up..." | tee -a ${LOGFILE}
1095
		chflags -R noschg ${FINAL_CHROOT_DIR} 2>&1 1>/dev/null
1096
		rm -rf ${FINAL_CHROOT_DIR}/* 2>&1 1>/dev/null
1097
		echo "Done." | tee -a ${LOGFILE}
1098
	fi
1099

    
1100
	echo ">>> Preparing Final image staging area: $(LC_ALL=C date)" 2>&1 | tee -a ${LOGFILE}
1101
	echo ">>> Cloning ${STAGE_CHROOT_DIR} to ${FINAL_CHROOT_DIR}" 2>&1 | tee -a ${LOGFILE}
1102
	clone_directory_contents ${STAGE_CHROOT_DIR} ${FINAL_CHROOT_DIR}
1103

    
1104
	if [ ! -f $FINAL_CHROOT_DIR/sbin/init ]; then
1105
		echo ">>> ERROR: Something went wrong during cloning -- Please verify!" 2>&1 | tee -a ${LOGFILE}
1106
		print_error_pfS
1107
	fi
1108
}
1109

    
1110
customize_stagearea_for_image() {
1111
	# Prepare final stage area
1112
	create_final_staging_area
1113

    
1114
	if [ "${1}" = "nanobsd" -o \
1115
	     "${1}" = "nanobsd-vga" ]; then
1116

    
1117
		mkdir -p ${FINAL_CHROOT_DIR}/root/var/db \
1118
			 ${FINAL_CHROOT_DIR}/root/var/cache \
1119
			 ${FINAL_CHROOT_DIR}/var/db/pkg \
1120
			 ${FINAL_CHROOT_DIR}/var/cache/pkg
1121
		mv -f ${FINAL_CHROOT_DIR}/var/db/pkg ${FINAL_CHROOT_DIR}/root/var/db
1122
		mv -f ${FINAL_CHROOT_DIR}/var/cache/pkg ${FINAL_CHROOT_DIR}/root/var/cache
1123
		ln -sf ../../root/var/db/pkg ${FINAL_CHROOT_DIR}/var/db/pkg
1124
		ln -sf ../../root/var/cache/pkg ${FINAL_CHROOT_DIR}/var/cache/pkg
1125

    
1126
		pkg_chroot_add ${FINAL_CHROOT_DIR} base-nanobsd
1127
	else
1128
		pkg_chroot_add ${FINAL_CHROOT_DIR} base
1129
	fi
1130

    
1131
	if [ "${1}" = "iso" -o \
1132
	     "${1}" = "memstick" -o \
1133
	     "${1}" = "memstickserial" -o \
1134
	     "${1}" = "memstickadi" ]; then
1135
		install_bsdinstaller
1136
		mkdir -p ${FINAL_CHROOT_DIR}/pkgs
1137
		cp ${CORE_PKG_PATH}/All/*default-config*.txz ${FINAL_CHROOT_DIR}/pkgs
1138
	fi
1139

    
1140
	if [ "${1}" = "nanobsd" -o \
1141
	     "${1}" = "memstickserial" -o \
1142
	     "${1}" = "memstickadi" ]; then
1143
		pkg_chroot_add ${FINAL_CHROOT_DIR} default-config-serial
1144
	elif [ "${1}" = "ova" ]; then
1145
		pkg_chroot_add ${FINAL_CHROOT_DIR} default-config-vmware
1146
	else
1147
		pkg_chroot_add ${FINAL_CHROOT_DIR} default-config
1148
	fi
1149
}
1150

    
1151
create_distribution_tarball() {
1152
	mkdir -p ${FINAL_CHROOT_DIR}/install
1153

    
1154
	tar -C ${FINAL_CHROOT_DIR} --exclude ./install --exclude ./pkgs -cJf ${FINAL_CHROOT_DIR}/install/${PRODUCT_NAME}.txz .
1155
}
1156

    
1157
create_iso_image() {
1158
	LOGFILE=${BUILDER_LOGS}/isoimage.${TARGET}
1159
	echo ">>> Building bootable ISO image for ${TARGET}" | tee -a ${LOGFILE}
1160
	if [ -z "${DEFAULT_KERNEL}" ]; then
1161
		echo ">>> ERROR: Could not identify DEFAULT_KERNEL to install on image!" | tee -a ${LOGFILE}
1162
		print_error_pfS
1163
	fi
1164

    
1165
	customize_stagearea_for_image "iso"
1166
	install_default_kernel ${DEFAULT_KERNEL}
1167

    
1168
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1169

    
1170
	FSLABEL=$(echo ${PRODUCT_NAME} | tr '[:lower:]' '[:upper:]')
1171
	echo "/dev/iso9660/${FSLABEL} / cd9660 ro 0 0" > ${FINAL_CHROOT_DIR}/etc/fstab
1172

    
1173
	# This check is for supporting create memstick/ova images
1174
	echo -n ">>> Running command: script -aq $LOGFILE makefs -t cd9660 -o bootimage=\"i386;${FINAL_CHROOT_DIR}/boot/cdboot \"-o no-emul-boot -o rockridge " | tee -a ${LOGFILE}
1175
	echo "-o label=${FSLABEL} -o publisher=\"${PRODUCT_NAME} project.\" $ISOPATH ${FINAL_CHROOT_DIR}" | tee -a ${LOGFILE}
1176

    
1177
	create_distribution_tarball
1178

    
1179
	# Remove /rescue from iso since cd9660 cannot deal with hardlinks
1180
	rm -rf ${FINAL_CHROOT_DIR}/rescue
1181

    
1182
	makefs -t cd9660 -o bootimage="i386;${FINAL_CHROOT_DIR}/boot/cdboot" -o no-emul-boot -o rockridge \
1183
		-o label=${FSLABEL} -o publisher="${PRODUCT_NAME} project." $ISOPATH ${FINAL_CHROOT_DIR} 2>&1 >> ${LOGFILE}
1184
	if [ $? -ne 0 -o ! -f $ISOPATH ]; then
1185
		if [ -f ${ISOPATH} ]; then
1186
			rm -f $ISOPATH
1187
		fi
1188
		echo ">>> ERROR: Something wrong happened during ISO image creation. STOPPING!" | tee -a ${LOGFILE}
1189
		print_error_pfS
1190
	fi
1191
	gzip -qf $ISOPATH &
1192
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1193

    
1194
	echo ">>> ISO created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1195
}
1196

    
1197
create_memstick_image() {
1198

    
1199
	LOGFILE=${BUILDER_LOGS}/memstick.${TARGET}
1200
	if [ "${MEMSTICKPATH}" = "" ]; then
1201
		echo ">>> MEMSTICKPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1202
		return
1203
	fi
1204

    
1205
	if [ ! -d ${FINAL_CHROOT_DIR}/boot ]; then
1206
		customize_stagearea_for_image "memstick"
1207
		install_default_kernel ${DEFAULT_KERNEL}
1208
	fi
1209

    
1210
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1211

    
1212
	echo ">>> Creating memstick to ${MEMSTICKPATH}." 2>&1 | tee -a ${LOGFILE}
1213
	echo "/dev/ufs/${PRODUCT_NAME} / ufs ro 0 0" > ${FINAL_CHROOT_DIR}/etc/fstab
1214
	echo "kern.cam.boot_delay=10000" >> ${FINAL_CHROOT_DIR}/boot/loader.conf.local
1215

    
1216
	create_distribution_tarball
1217

    
1218
	makefs -B little -o label=${PRODUCT_NAME} ${MEMSTICKPATH} ${FINAL_CHROOT_DIR}
1219
	if [ $? -ne 0 ]; then
1220
		if [ -f ${MEMSTICKPATH} ]; then
1221
			rm -f $MEMSTICKPATH
1222
		fi
1223
		echo ">>> ERROR: Something wrong happened during MEMSTICK image creation. STOPPING!" | tee -a ${LOGFILE}
1224
		print_error_pfS
1225
	fi
1226
	MD=$(mdconfig -a -t vnode -f $MEMSTICKPATH)
1227
	# Just in case
1228
	trap "mdconfig -d -u ${MD}" 1 2 15 EXIT
1229
	gpart create -s BSD ${MD} 2>&1 >> ${LOGFILE}|
1230
	gpart bootcode -b ${FINAL_CHROOT_DIR}/boot/boot ${MD} 2>&1 >> ${LOGFILE}
1231
	gpart add -t freebsd-ufs ${MD} 2>&1 >> ${LOGFILE}
1232
	trap "-" 1 2 15 EXIT
1233
	mdconfig -d -u ${MD} 2>&1 | tee -a ${LOGFILE}
1234
	gzip -qf $MEMSTICKPATH &
1235
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1236

    
1237
	echo ">>> MEMSTICK created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1238
}
1239

    
1240
create_memstick_serial_image() {
1241
	LOGFILE=${BUILDER_LOGS}/memstickserial.${TARGET}
1242
	if [ "${MEMSTICKSERIALPATH}" = "" ]; then
1243
		echo ">>> MEMSTICKSERIALPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1244
		return
1245
	fi
1246

    
1247
	if [ ! -d ${FINAL_CHROOT_DIR}/boot ]; then
1248
		customize_stagearea_for_image "memstickserial"
1249
		install_default_kernel ${DEFAULT_KERNEL}
1250
	fi
1251

    
1252
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1253

    
1254
	echo "/dev/ufs/${PRODUCT_NAME} / ufs ro 0 0" > ${FINAL_CHROOT_DIR}/etc/fstab
1255
	echo "kern.cam.boot_delay=10000" >> ${FINAL_CHROOT_DIR}/boot/loader.conf.local
1256

    
1257
	echo ">>> Creating serial memstick to ${MEMSTICKSERIALPATH}." 2>&1 | tee -a ${LOGFILE}
1258

    
1259
	BOOTCONF=${FINAL_CHROOT_DIR}/boot.config
1260
	LOADERCONF=${FINAL_CHROOT_DIR}/boot/loader.conf
1261

    
1262
	echo ">>> Activating serial console..." 2>&1 | tee -a ${LOGFILE}
1263
	# Activate serial console in boot.config
1264
	if [ -f ${BOOTCONF} ]; then
1265
		sed -i "" '/-D$/d' ${BOOTCONF}
1266
	fi
1267
	echo "-S115200 -D" >> ${BOOTCONF}
1268

    
1269
	# Remove old console options if present.
1270
	[ -f "${LOADERCONF}" ] \
1271
		&& sed -i "" -Ee "/(console|boot_multicons|boot_serial)/d" ${LOADERCONF}
1272
	# Activate serial console+video console in loader.conf
1273
	echo 'boot_multicons="YES"' >>  ${LOADERCONF}
1274
	echo 'boot_serial="YES"' >> ${LOADERCONF}
1275
	echo 'console="comconsole,vidconsole"' >> ${LOADERCONF}
1276
	echo 'comconsole_speed="115200"' >> ${LOADERCONF}
1277

    
1278
	create_distribution_tarball
1279

    
1280
	makefs -B little -o label=${PRODUCT_NAME} ${MEMSTICKSERIALPATH} ${FINAL_CHROOT_DIR}
1281
	if [ $? -ne 0 ]; then
1282
		if [ -f ${MEMSTICKSERIALPATH} ]; then
1283
			rm -f $MEMSTICKSERIALPATH
1284
		fi
1285
		echo ">>> ERROR: Something wrong happened during MEMSTICKSERIAL image creation. STOPPING!" | tee -a ${LOGFILE}
1286
		print_error_pfS
1287
	fi
1288
	MD=$(mdconfig -a -t vnode -f $MEMSTICKSERIALPATH)
1289
	# Just in case
1290
	trap "mdconfig -d -u ${MD}" 1 2 15 EXIT
1291
	gpart create -s BSD ${MD} 2>&1 >> ${LOGFILE}
1292
	gpart bootcode -b ${FINAL_CHROOT_DIR}/boot/boot ${MD} 2>&1 >> ${LOGFILE}
1293
	gpart add -t freebsd-ufs ${MD} 2>&1 >> ${LOGFILE}
1294
	trap "-" 1 2 15 EXIT
1295
	mdconfig -d -u ${MD} 2>&1 >> ${LOGFILE}
1296
	gzip -qf $MEMSTICKSERIALPATH &
1297
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1298

    
1299
	echo ">>> MEMSTICKSERIAL created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1300
}
1301

    
1302
create_memstick_adi_image() {
1303
	LOGFILE=${BUILDER_LOGS}/memstickadi${TARGET}
1304
	if [ "${MEMSTICKADIPATH}" = "" ]; then
1305
		echo ">>> MEMSTICKADIPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1306
		return
1307
	fi
1308

    
1309
	if [ ! -d ${FINAL_CHROOT_DIR}/boot ]; then
1310
		customize_stagearea_for_image "memstickadi"
1311
		install_default_kernel ${DEFAULT_KERNEL}
1312
	fi
1313

    
1314
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1315

    
1316
	echo "/dev/ufs/${PRODUCT_NAME} / ufs ro 0 0" > ${FINAL_CHROOT_DIR}/etc/fstab
1317
	echo "kern.cam.boot_delay=10000" >> ${FINAL_CHROOT_DIR}/boot/loader.conf.local
1318

    
1319
	echo ">>> Creating serial memstick to ${MEMSTICKADIPATH}." 2>&1 | tee -a ${LOGFILE}
1320

    
1321
	BOOTCONF=${FINAL_CHROOT_DIR}/boot.config
1322
	LOADERCONF=${FINAL_CHROOT_DIR}/boot/loader.conf
1323

    
1324
	echo ">>> Activating serial console..." 2>&1 | tee -a ${LOGFILE}
1325
	# Activate serial console in boot.config
1326
	if [ -f ${BOOTCONF} ]; then
1327
		sed -i "" '/-[Dh]$/d' ${BOOTCONF}
1328
	fi
1329
	echo "-S115200 -h" >> ${BOOTCONF}
1330

    
1331
	# Remove old console options if present.
1332
	[ -f "${LOADERCONF}" ] \
1333
		&& sed -i "" -Ee "/(console|boot_multicons|boot_serial|hint.uart)/d" ${LOADERCONF}
1334
	# Activate serial console+video console in loader.conf
1335
	echo 'boot_serial="YES"' >> ${LOADERCONF}
1336
	echo 'console="comconsole"' >> ${LOADERCONF}
1337
	echo 'comconsole_speed="115200"' >> ${LOADERCONF}
1338
	echo 'comconsole_port="0x2F8"' >> ${LOADERCONF}
1339
	echo 'hint.uart.0.flags="0x00"' >> ${LOADERCONF}
1340
	echo 'hint.uart.1.flags="0x10"' >> ${LOADERCONF}
1341

    
1342
	create_distribution_tarball
1343

    
1344
	makefs -B little -o label=${PRODUCT_NAME} ${MEMSTICKADIPATH} ${FINAL_CHROOT_DIR}
1345
	if [ $? -ne 0 ]; then
1346
		if [ -f ${MEMSTICKADIPATH} ]; then
1347
			rm -f $MEMSTICKADIPATH
1348
		fi
1349
		echo ">>> ERROR: Something wrong happened during MEMSTICKADI image creation. STOPPING!" | tee -a ${LOGFILE}
1350
		print_error_pfS
1351
	fi
1352
	MD=$(mdconfig -a -t vnode -f $MEMSTICKADIPATH)
1353
	# Just in case
1354
	trap "mdconfig -d -u ${MD}" 1 2 15 EXIT
1355
	gpart create -s BSD ${MD} 2>&1 >> ${LOGFILE}
1356
	gpart bootcode -b ${FINAL_CHROOT_DIR}/boot/boot ${MD} 2>&1 >> ${LOGFILE}
1357
	gpart add -t freebsd-ufs ${MD} 2>&1 >> ${LOGFILE}
1358
	trap "-" 1 2 15 EXIT
1359
	mdconfig -d -u ${MD} 2>&1 >> ${LOGFILE}
1360
	gzip -qf $MEMSTICKADIPATH &
1361
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1362

    
1363
	echo ">>> MEMSTICKADI created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1364
}
1365

    
1366
# Create pkg conf on desired place with desired arch/branch
1367
setup_pkg_repo() {
1368
	if [ -z "${4}" ]; then
1369
		return
1370
	fi
1371

    
1372
	local _target="${1}"
1373
	local _arch="${2}"
1374
	local _target_arch="${3}"
1375
	local _branch="${4}"
1376

    
1377
	mkdir -p $(dirname ${_target}) >/dev/null 2>&1
1378

    
1379
	sed \
1380
		-e "s/%%ARCH%%/${_arch}_${_target_arch}/" \
1381
		-e "s/%%GIT_REPO_BRANCH_OR_TAG%%/${_branch}/g" \
1382
		-e "s,%%PKG_REPO_SERVER%%,${PKG_REPO_SERVER},g" \
1383
		-e "s/%%PRODUCT_NAME%%/${PRODUCT_NAME}/g" \
1384
		${FREEBSD_SRC_DIR}/release/pkg_repos/${PRODUCT_NAME}.conf.template \
1385
		> ${_target}
1386
}
1387

    
1388
# This routine ensures any ports / binaries that the builder
1389
# system needs are on disk and ready for execution.
1390
builder_setup() {
1391
	# If Product-builder is already installed, just leave
1392
	if pkg info -e -q ${PRODUCT_NAME}-builder; then
1393
		return
1394
	fi
1395

    
1396
	if [ ! -f /usr/local/etc/pkg/repos/${PRODUCT_NAME}.conf ]; then
1397
		[ -d /usr/local/etc/pkg/repos ] \
1398
			|| mkdir -p /usr/local/etc/pkg/repos
1399

    
1400
		update_freebsd_sources
1401

    
1402
		local _arch=$(uname -m)
1403
		setup_pkg_repo /usr/local/etc/pkg/repos/${PRODUCT_NAME}.conf ${_arch} ${_arch} ${PKG_REPO_CONF_BRANCH}
1404
	fi
1405

    
1406
	pkg install ${PRODUCT_NAME}-builder
1407
}
1408

    
1409
# Updates FreeBSD sources
1410
update_freebsd_sources() {
1411
	if [ "${1}" = "full" ]; then
1412
		local _full=1
1413
		local _clone_params=""
1414
	else
1415
		local _full=0
1416
		local _clone_params="--depth 1 --single-branch"
1417
	fi
1418

    
1419
	if [ ! -d "${FREEBSD_SRC_DIR}" ]; then
1420
		mkdir -p ${FREEBSD_SRC_DIR}
1421
	fi
1422

    
1423
	if [ -n "${NO_BUILDWORLD}" -a -n "${NO_BUILDKERNEL}" ]; then
1424
		echo ">>> NO_BUILDWORLD and NO_BUILDKERNEL set, skipping update of freebsd sources" | tee -a ${LOGFILE}
1425
		return
1426
	fi
1427

    
1428
	echo -n ">>> Obtaining FreeBSD sources ${FREEBSD_BRANCH}..."
1429
	local _FREEBSD_BRANCH=${FREEBSD_BRANCH:-"devel"}
1430
	local _CLONE=1
1431

    
1432
	if [ -d "${FREEBSD_SRC_DIR}/.git" ]; then
1433
		CUR_BRANCH=$(cd ${FREEBSD_SRC_DIR} && git branch | grep '^\*' | cut -d' ' -f2)
1434
		if [ ${_full} -eq 0 -a "${CUR_BRANCH}" = "${_FREEBSD_BRANCH}" ]; then
1435
			_CLONE=0
1436
			( cd ${FREEBSD_SRC_DIR} && git clean -fd; git fetch origin; git reset --hard origin/${_FREEBSD_BRANCH} ) 2>&1 | grep -C3 -i -E 'error|fatal'
1437
		else
1438
			rm -rf ${FREEBSD_SRC_DIR}
1439
		fi
1440
	fi
1441

    
1442
	if [ ${_CLONE} -eq 1 ]; then
1443
		( git clone --branch ${_FREEBSD_BRANCH} ${_clone_params} ${FREEBSD_REPO_BASE} ${FREEBSD_SRC_DIR} ) 2>&1 | grep -C3 -i -E 'error|fatal'
1444
	fi
1445

    
1446
	if [ ! -d "${FREEBSD_SRC_DIR}/.git" ]; then
1447
		echo ">>> ERROR: It was not possible to clone FreeBSD src repo"
1448
		print_error_pfS
1449
	fi
1450

    
1451
	if [ -n "${GIT_FREEBSD_COSHA1}" ]; then
1452
		( cd ${FREEBSD_SRC_DIR} && git checkout ${GIT_FREEBSD_COSHA1} ) 2>&1 | grep -C3 -i -E 'error|fatal'
1453
	fi
1454
	echo "Done!"
1455
}
1456

    
1457
pkg_chroot() {
1458
	local _root="${1}"
1459
	shift
1460

    
1461
	if [ $# -eq 0 ]; then
1462
		return -1
1463
	fi
1464

    
1465
	if [ -z "${_root}" -o "${_root}" = "/" -o ! -d "${_root}" ]; then
1466
		return -1
1467
	fi
1468

    
1469
	mkdir -p \
1470
		${SCRATCHDIR}/pkg_cache \
1471
		${_root}/var/cache/pkg \
1472
		${_root}/dev
1473

    
1474
	/sbin/mount -t nullfs ${SCRATCHDIR}/pkg_cache ${_root}/var/cache/pkg
1475
	/sbin/mount -t devfs devfs ${_root}/dev
1476
	cp -f /etc/resolv.conf ${_root}/etc/resolv.conf
1477
	touch ${BUILDER_LOGS}/install_pkg_install_ports.txt
1478
	script -aq ${BUILDER_LOGS}/install_pkg_install_ports.txt pkg -c ${_root} $@ >/dev/null 2>&1
1479
	rm -f ${_root}/etc/resolv.conf
1480
	/sbin/umount -f ${_root}/dev
1481
	/sbin/umount -f ${_root}/var/cache/pkg
1482
}
1483

    
1484

    
1485
pkg_chroot_add() {
1486
	if [ -z "${1}" -o -z "${2}" ]; then
1487
		return 1
1488
	fi
1489

    
1490
	local _target="${1}"
1491
	local _pkg="$(get_pkg_name ${2}).txz"
1492

    
1493
	if [ ! -d "${_target}" ]; then
1494
		echo ">>> ERROR: Target dir ${_target} not found"
1495
		print_error_pfS
1496
	fi
1497

    
1498
	if [ ! -f ${CORE_PKG_PATH}/All/${_pkg} ]; then
1499
		echo ">>> ERROR: Package ${_pkg} not found"
1500
		print_error_pfS
1501
	fi
1502

    
1503
	cp ${CORE_PKG_PATH}/All/${_pkg} ${_target}
1504
	pkg_chroot ${_target} add /${_pkg}
1505
	rm -f ${_target}/${_pkg}
1506
}
1507

    
1508
pkg_bootstrap() {
1509
	local _root=${1:-"${STAGE_CHROOT_DIR}"}
1510

    
1511
	setup_pkg_repo ${_root}/usr/local/etc/pkg/repos/${PRODUCT_NAME}.conf ${TARGET} ${TARGET_ARCH} ${PKG_REPO_CONF_BRANCH}
1512

    
1513
	pkg_chroot ${_root} bootstrap -f
1514
}
1515

    
1516
# This routine assists with installing various
1517
# freebsd ports files into the pfsenese-fs staging
1518
# area.
1519
install_pkg_install_ports() {
1520
	local MAIN_PKG="${1}"
1521

    
1522
	if [ -z "${MAIN_PKG}" ]; then
1523
		MAIN_PKG=${PRODUCT_NAME}
1524
	fi
1525

    
1526
	echo ">>> Installing pkg repository in chroot (${STAGE_CHROOT_DIR})..."
1527

    
1528
	[ -d ${STAGE_CHROOT_DIR}/var/cache/pkg ] || \
1529
		mkdir -p ${STAGE_CHROOT_DIR}/var/cache/pkg
1530

    
1531
	[ -d ${SCRATCHDIR}/pkg_cache ] || \
1532
		mkdir -p ${SCRATCHDIR}/pkg_cache
1533

    
1534
	echo ">>> Installing built ports (packages) in chroot (${STAGE_CHROOT_DIR})... (starting)"
1535
	pkg_chroot ${STAGE_CHROOT_DIR} install ${MAIN_PKG} ${custom_package_list}
1536
	pkg_chroot ${STAGE_CHROOT_DIR} autoremove
1537
	echo ">>> Installing built ports (packages) in chroot (${STAGE_CHROOT_DIR})... (finshied)"
1538
}
1539

    
1540
install_bsdinstaller() {
1541
	echo ">>> Installing BSDInstaller in chroot (${FINAL_CHROOT_DIR})... (starting)"
1542
	pkg_chroot ${FINAL_CHROOT_DIR} install -f bsdinstaller
1543
	sed -i '' -e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME}," \
1544
		  -e "s,%%PRODUCT_VERSION%%,${PRODUCT_VERSION}," \
1545
		  -e "s,%%ARCH%%,${TARGET}," \
1546
		  ${FINAL_CHROOT_DIR}/usr/local/share/dfuibe_lua/conf/pfSense.lua \
1547
		  ${FINAL_CHROOT_DIR}/usr/local/share/dfuibe_lua/conf/pfSense_rescue.lua
1548
	echo ">>> Installing BSDInstaller in chroot (${FINAL_CHROOT_DIR})... (finished)"
1549
}
1550

    
1551
staginareas_clean_each_run() {
1552
	echo -n ">>> Cleaning build directories: "
1553
	if [ -d "${FINAL_CHROOT_DIR}" ]; then
1554
		BASENAME=$(basename ${FINAL_CHROOT_DIR})
1555
		echo -n "$BASENAME "
1556
		chflags -R noschg ${FINAL_CHROOT_DIR} 2>&1 >/dev/null
1557
		rm -rf ${FINAL_CHROOT_DIR}/* 2>/dev/null
1558
	fi
1559
	echo "Done!"
1560
}
1561

    
1562
# Imported from FreeSBIE
1563
buildkernel() {
1564
	if [ -n "${NO_BUILDKERNEL}" ]; then
1565
		echo ">>> NO_BUILDKERNEL set, skipping build" | tee -a ${LOGFILE}
1566
		return
1567
	fi
1568

    
1569
	if [ -z "${KERNCONF}" ]; then
1570
		echo ">>> ERROR: No kernel configuration defined probably this is not what you want! STOPPING!" | tee -a ${LOGFILE}
1571
		print_error_pfS
1572
	fi
1573

    
1574
	if [ -n "${KERNELCONF}" ]; then
1575
		export KERNCONFDIR=$(dirname ${KERNELCONF})
1576
		export KERNCONF=$(basename ${KERNELCONF})
1577
	fi
1578

    
1579
	SRCCONFBASENAME=$(basename ${SRC_CONF})
1580
	echo ">>> KERNCONFDIR: ${KERNCONFDIR}"
1581
	echo ">>> ARCH:        ${TARGET}"
1582
	echo ">>> SRC_CONF:    ${SRCCONFBASENAME}"
1583

    
1584
	makeargs="${MAKEJ_KERNEL} SRCCONF=${SRC_CONF} __MAKE_CONF=${MAKE_CONF} TARGET_ARCH=${TARGET_ARCH} TARGET=${TARGET}"
1585
	echo ">>> Builder is running the command: script -aq $LOGFILE make -DNO_KERNELCLEAN $makeargs buildkernel KERNCONF=${KERNCONF}" | tee -a $LOGFILE
1586
	(script -q $LOGFILE make -C ${FREEBSD_SRC_DIR} -DNO_KERNELCLEAN $makeargs buildkernel KERNCONF=${KERNCONF} || print_error_pfS;) | egrep '^>>>'
1587
}
1588

    
1589
# Imported from FreeSBIE
1590
installkernel() {
1591
	if [ -z "${KERNCONF}" ]; then
1592
		echo ">>> ERROR: No kernel configuration defined probably this is not what you want! STOPPING!" | tee -a ${LOGFILE}
1593
		print_error_pfS
1594
	fi
1595

    
1596
	if [ -n "${KERNELCONF}" ]; then
1597
		export KERNCONFDIR=$(dirname ${KERNELCONF})
1598
		export KERNCONF=$(basename ${KERNELCONF})
1599
	fi
1600

    
1601
	mkdir -p ${STAGE_CHROOT_DIR}/boot
1602
	makeargs="${MAKEJ_KERNEL} SRCCONF=${SRC_CONF} __MAKE_CONF=${MAKE_CONF} TARGET_ARCH=${TARGET_ARCH} TARGET=${TARGET} DESTDIR=${KERNEL_DESTDIR}"
1603
	echo ">>> Builder is running the command: script -aq $LOGFILE make ${makeargs} installkernel KERNCONF=${KERNCONF}"  | tee -a $LOGFILE
1604
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} installkernel KERNCONF=${KERNCONF} || print_error_pfS;) | egrep '^>>>'
1605
	gzip -f9 $KERNEL_DESTDIR/boot/kernel/kernel
1606
}
1607

    
1608
# Launch is ran first to setup a few variables that we need
1609
# Imported from FreeSBIE
1610
launch() {
1611
	if [ "$(id -u)" != "0" ]; then
1612
		echo "Sorry, this must be done as root."
1613
	fi
1614

    
1615
	echo ">>> Operation $0 has started at $(date)"
1616
}
1617

    
1618
finish() {
1619
	echo ">>> Operation $0 has ended at $(date)"
1620
}
1621

    
1622
pkg_repo_rsync() {
1623
	local _repo_path="${1}"
1624

    
1625
	if [ -n "${DO_NOT_UPLOAD}" -o -z "${_repo_path}" -o ! -d "${_repo_path}" ]; then
1626
		return
1627
	fi
1628

    
1629
	if [ -z "${LOGFILE}" ]; then
1630
		local _logfile="/dev/null"
1631
	else
1632
		local _logfile="${LOGFILE}"
1633
	fi
1634

    
1635
	echo -n ">>> Sending updated repository to ${PKG_RSYNC_HOSTNAME}... " | tee -a ${_logfile}
1636
	if script -aq ${_logfile} rsync -ave "ssh -p ${PKG_RSYNC_SSH_PORT}" \
1637
		--timeout=60 --delete-delay ${_repo_path} \
1638
		${PKG_RSYNC_USERNAME}@${PKG_RSYNC_HOSTNAME}:${PKG_RSYNC_DESTDIR} >/dev/null 2>&1
1639
	then
1640
		echo "Done!" | tee -a ${_logfile}
1641
	else
1642
		echo "Failed!" | tee -a ${_logfile}
1643
		echo ">>> ERROR: An error occurred sending repo to remote hostname"
1644
		print_error_pfS
1645
	fi
1646
}
1647

    
1648
poudriere_create_patch() {
1649
	local _jail_patch="${SCRATCHDIR}/poudriere_jail.${GIT_REPO_BRANCH_OR_TAG}.patch"
1650

    
1651
	if [ -z "${FREEBSD_PARENT_BRANCH}" ]; then
1652
		echo ">>> ERROR: FREEBSD_PARENT_BRANCH is not set"
1653
	fi
1654

    
1655
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1656

    
1657
	# Get FreeBSD source and apply patches
1658
	update_freebsd_sources full
1659

    
1660
	[ -f "${_jail_patch}" ] && \
1661
		rm -f "${_jail_patch}"
1662

    
1663
	# Create a big patch with all our changes to use on jail
1664
	( \
1665
		cd ${FREEBSD_SRC_DIR} && \
1666
		git diff $(git merge-base origin/${FREEBSD_PARENT_BRANCH} ${FREEBSD_BRANCH}) > ${_jail_patch}
1667
	) >/dev/null 2>&1
1668

    
1669
	# Check if patch was created
1670
	if [ ! -s "${_jail_patch}" ]; then
1671
		echo ">>> ERROR: Patch does not exist or is empty, aborting..." | tee -a ${LOGFILE}
1672
		print_error_pfS
1673
	fi
1674
}
1675

    
1676
poudriere_possible_archs() {
1677
	local _arch=$(uname -m)
1678
	local _archs="i386.i386"
1679

    
1680
	# IF host is amd64, we'll create both repos, and if possible armv6
1681
	if [ "${_arch}" = "amd64" ]; then
1682
		_archs="amd64.amd64 ${_archs}"
1683

    
1684
		if [ -f /usr/local/bin/qemu-arm-static ]; then
1685
			# Make sure binmiscctl is ok
1686
			/usr/local/etc/rc.d/qemu_user_static forcestart >/dev/null 2>&1
1687

    
1688
			if binmiscctl lookup armv6 >/dev/null 2>&1; then
1689
				_archs="${_archs} arm.armv6"
1690
			fi
1691
		fi
1692
	fi
1693

    
1694
	if [ -n "${ARCH_LIST}" ]; then
1695
		local _found=0
1696
		for _desired_arch in ${ARCH_LIST}; do
1697
			_found=0
1698
			for _possible_arch in ${_archs}; do
1699
				if [ "${_desired_arch}" = "${_possible_arch}" ]; then
1700
					_found=1
1701
					break
1702
				fi
1703
			done
1704
			if [ ${_found} -eq 0 ]; then
1705
				echo ">>> ERROR: Impossible to build for arch: ${_desired_arch}"
1706
				print_error_pfS
1707
			fi
1708
		done
1709
		_archs="${ARCH_LIST}"
1710
	fi
1711

    
1712
	echo ${_archs}
1713
}
1714

    
1715
poudriere_jail_name() {
1716
	local _jail_arch="${1}"
1717

    
1718
	if [ -z "${_jail_arch}" ]; then
1719
		return 1
1720
	fi
1721

    
1722
	# Poudriere doesn't like periods in jail names
1723
	_jail_arch=$(echo "${_jail_arch}" | tr '.' '_')
1724

    
1725
	echo "${PRODUCT_NAME}_${GIT_REPO_BRANCH_OR_TAG}_${_jail_arch}"
1726
}
1727

    
1728
poudriere_create_ports_tree() {
1729
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1730

    
1731
	if ! poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
1732
		local _branch=""
1733
		if [ -z "${POUDRIERE_PORTS_GIT_URL}" ]; then
1734
			echo ">>> ERROR: POUDRIERE_PORTS_GIT_URL is not defined"
1735
			print_error_pfS
1736
		fi
1737
		if [ -n "${POUDRIERE_PORTS_GIT_BRANCH}" ]; then
1738
			_branch="-B ${POUDRIERE_PORTS_GIT_BRANCH}"
1739
		fi
1740
		echo -n ">>> Creating poudriere ports tree, it may take some time... " | tee -a ${LOGFILE}
1741
		if ! script -aq ${LOGFILE} poudriere ports -c -p "${POUDRIERE_PORTS_NAME}" -m git ${_branch} >/dev/null 2>&1; then
1742
			echo "" | tee -a ${LOGFILE}
1743
			echo ">>> ERROR: Error creating poudriere ports tree, aborting..." | tee -a ${LOGFILE}
1744
			print_error_pfS
1745
		fi
1746
		echo "Done!" | tee -a ${LOGFILE}
1747
	fi
1748
}
1749

    
1750
poudriere_init() {
1751
	local _error=0
1752
	local _archs=$(poudriere_possible_archs)
1753
	local _jail_patch="${SCRATCHDIR}/poudriere_jail.${GIT_REPO_BRANCH_OR_TAG}.patch"
1754

    
1755
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1756

    
1757
	# Sanity checks
1758
	if [ -z "${ZFS_TANK}" ]; then
1759
		echo ">>> ERROR: \$ZFS_TANK is empty" | tee -a ${LOGFILE}
1760
		error=1
1761
	fi
1762

    
1763
	if [ -z "${ZFS_ROOT}" ]; then
1764
		echo ">>> ERROR: \$ZFS_ROOT is empty" | tee -a ${LOGFILE}
1765
		error=1
1766
	fi
1767

    
1768
	if [ -z "${POUDRIERE_PORTS_NAME}" ]; then
1769
		echo ">>> ERROR: \$POUDRIERE_PORTS_NAME is empty" | tee -a ${LOGFILE}
1770
		error=1
1771
	fi
1772

    
1773
	if [ ${_error} -eq 1 ]; then
1774
		print_error_pfS
1775
	fi
1776

    
1777
	# Check if zpool exists
1778
	if ! zpool list ${ZFS_TANK} >/dev/null 2>&1; then
1779
		echo ">>> ERROR: ZFS tank ${ZFS_TANK} not found, please create it and try again..." | tee -a ${LOGFILE}
1780
		print_error_pfS
1781
	fi
1782

    
1783
	# Check if zfs rootfs exists
1784
	if ! zfs list ${ZFS_TANK}${ZFS_ROOT} >/dev/null 2>&1; then
1785
		echo ">>> ERROR: ZFS filesystem ${ZFS_TANK}${ZFS_ROOT} not found, please create it and try again..." | tee -a ${LOGFILE}
1786
		print_error_pfS
1787
	fi
1788

    
1789
	# Make sure poudriere is installed
1790
	if ! pkg info --quiet poudriere; then
1791
		echo ">>> Installing poudriere..." | tee -a ${LOGFILE}
1792
		if ! pkg install poudriere >/dev/null 2>&1; then
1793
			echo ">>> ERROR: poudriere was not installed, aborting..." | tee -a ${LOGFILE}
1794
			print_error_pfS
1795
		fi
1796
	fi
1797

    
1798
	# Create poudriere.conf
1799
	if [ -z "${POUDRIERE_PORTS_GIT_URL}" ]; then
1800
		echo ">>> ERROR: POUDRIERE_PORTS_GIT_URL is not defined"
1801
		print_error_pfS
1802
	fi
1803
	echo ">>> Creating poudriere.conf" | tee -a ${LOGFILE}
1804
	cat <<EOF >/usr/local/etc/poudriere.conf
1805
ZPOOL=${ZFS_TANK}
1806
ZROOTFS=${ZFS_ROOT}
1807
RESOLV_CONF=/etc/resolv.conf
1808
BASEFS=/usr/local/poudriere
1809
USE_PORTLINT=no
1810
USE_TMPFS=yes
1811
NOLINUX=yes
1812
DISTFILES_CACHE=/usr/ports/distfiles
1813
CHECK_CHANGED_OPTIONS=yes
1814
CHECK_CHANGED_DEPS=yes
1815
ATOMIC_PACKAGE_REPOSITORY=yes
1816
COMMIT_PACKAGES_ON_FAILURE=no
1817
GIT_URL="${POUDRIERE_PORTS_GIT_URL}"
1818
EOF
1819

    
1820
	# Remove old jails
1821
	for jail_arch in ${_archs}; do
1822
		jail_name=$(poudriere_jail_name ${jail_arch})
1823

    
1824
		if poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
1825
			echo ">>> Poudriere jail ${jail_name} already exists, deleting it..." | tee -a ${LOGFILE}
1826
			poudriere jail -d -j "${jail_name}" >/dev/null 2>&1
1827
		fi
1828
	done
1829

    
1830
	# Remove old ports tree
1831
	if poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
1832
		echo ">>> Poudriere ports tree ${POUDRIERE_PORTS_NAME} already exists, deleting it..." | tee -a ${LOGFILE}
1833
		poudriere ports -d -p "${POUDRIERE_PORTS_NAME}"
1834
	fi
1835

    
1836
	poudriere_create_patch
1837

    
1838
	local native_xtools=""
1839
	# Now we are ready to create jails
1840
	for jail_arch in ${_archs}; do
1841
		jail_name=$(poudriere_jail_name ${jail_arch})
1842

    
1843
		if [ "${jail_arch}" = "arm.armv6" ]; then
1844
			native_xtools="-x"
1845
		else
1846
			native_xtools=""
1847
		fi
1848

    
1849
		echo -n ">>> Creating jail ${jail_name}, it may take some time... " | tee -a ${LOGFILE}
1850
		# XXX: Change -m to git when it's available in poudriere
1851
		if ! script -aq ${LOGFILE} poudriere jail -c -j "${jail_name}" -v ${FREEBSD_PARENT_BRANCH} \
1852
				-a ${jail_arch} -m svn -P ${_jail_patch} ${native_xtools} >/dev/null 2>&1; then
1853
			echo "" | tee -a ${LOGFILE}
1854
			echo ">>> ERROR: Error creating jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
1855
			print_error_pfS
1856
		fi
1857
		echo "Done!" | tee -a ${LOGFILE}
1858
	done
1859

    
1860
	poudriere_create_ports_tree
1861

    
1862
	echo ">>> Poudriere is now configured!" | tee -a ${LOGFILE}
1863
}
1864

    
1865
poudriere_update_jails() {
1866
	local _archs=$(poudriere_possible_archs)
1867
	local _jail_patch="${SCRATCHDIR}/poudriere_jail.${GIT_REPO_BRANCH_OR_TAG}.patch"
1868

    
1869
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1870

    
1871
	poudriere_create_patch
1872

    
1873
	local native_xtools=""
1874
	for jail_arch in ${_archs}; do
1875
		jail_name=$(poudriere_jail_name ${jail_arch})
1876

    
1877
		if ! poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
1878
			echo ">>> Poudriere jail ${jail_name} not found, skipping..." | tee -a ${LOGFILE}
1879
			continue
1880
		fi
1881

    
1882
		if [ "${jail_arch}" = "arm.armv6" ]; then
1883
			native_xtools="-x"
1884
		else
1885
			native_xtools=""
1886
		fi
1887

    
1888
		echo -n ">>> Updating jail ${jail_name}, it may take some time... " | tee -a ${LOGFILE}
1889
		if ! script -aq ${LOGFILE} poudriere jail -u -j "${jail_name}" -P ${_jail_patch} ${native_xtools} >/dev/null 2>&1; then
1890
			echo "" | tee -a ${LOGFILE}
1891
			echo ">>> ERROR: Error updating jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
1892
			print_error_pfS
1893
		fi
1894
		echo "Done!" | tee -a ${LOGFILE}
1895
	done
1896
}
1897

    
1898
poudriere_update_ports() {
1899
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1900

    
1901
	# Create ports tree if necessary
1902
	if ! poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
1903
		poudriere_create_ports_tree
1904
	else
1905
		echo -n ">>> Reseting local changes on ports tree ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
1906
		script -aq ${LOGFILE} git -C "/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}" reset --hard >/dev/null 2>&1
1907
		script -aq ${LOGFILE} git -C "/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}" clean -fd >/dev/null 2>&1
1908
		echo "Done!" | tee -a ${LOGFILE}
1909
		echo -n ">>> Updating ports tree ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
1910
		script -aq ${LOGFILE} poudriere ports -u -p "${POUDRIERE_PORTS_NAME}" >/dev/null 2>&1
1911
		echo "Done!" | tee -a ${LOGFILE}
1912
	fi
1913
}
1914

    
1915
poudriere_bulk() {
1916
	local _archs=$(poudriere_possible_archs)
1917

    
1918
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1919

    
1920
	if [ -z "${DO_NOT_UPLOAD}" -a -z "${PKG_RSYNC_HOSTNAME}" ]; then
1921
		echo ">>> ERROR: PKG_RSYNC_HOSTNAME is not set"
1922
		print_error_pfS
1923
	fi
1924

    
1925
	poudriere_create_ports_tree
1926

    
1927
	[ -d /usr/local/etc/poudriere.d ] || \
1928
		mkdir -p /usr/local/etc/poudriere.d
1929

    
1930
	if [ -f "${BUILDER_TOOLS}/conf/pfPorts/make.conf" ]; then
1931
		cp -f "${BUILDER_TOOLS}/conf/pfPorts/make.conf" /usr/local/etc/poudriere.d/${POUDRIERE_PORTS_NAME}-make.conf
1932
	fi
1933

    
1934
	# Change version of pfSense meta ports for snapshots
1935
	if [ -z "${_IS_RELEASE}" ]; then
1936
		for meta_pkg in ${PRODUCT_NAME} ${PRODUCT_NAME}-vmware; do
1937
			local _meta_pkg_version="$(echo "${PRODUCT_VERSION}" | sed 's,DEVELOPMENT,ALPHA,')-${DATESTRING}"
1938
			sed -i '' -e "/^DISTVERSION/ s,^.*,DISTVERSION=	${_meta_pkg_version}," \
1939
				/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}/security/${meta_pkg}/Makefile
1940
		done
1941
	fi
1942

    
1943
	for jail_arch in ${_archs}; do
1944
		jail_name=$(poudriere_jail_name ${jail_arch})
1945

    
1946
		if ! poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
1947
			echo ">>> Poudriere jail ${jail_name} not found, skipping..." | tee -a ${LOGFILE}
1948
			continue
1949
		fi
1950

    
1951
		if [ -f "${POUDRIERE_BULK}.${jail_arch}" ]; then
1952
			_ref_bulk="${POUDRIERE_BULK}.${jail_arch}"
1953
		else
1954
			_ref_bulk="${POUDRIERE_BULK}"
1955
		fi
1956

    
1957
		_bulk=${SCRATCHDIR}/poudriere_bulk.${GIT_REPO_BRANCH_OR_TAG}
1958
		sed -e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" ${_ref_bulk} > ${_bulk}
1959

    
1960
		if ! poudriere bulk -f ${_bulk} -j ${jail_name} -p ${POUDRIERE_PORTS_NAME}; then
1961
			echo ">>> ERROR: Something went wrong..."
1962
			print_error_pfS
1963
		fi
1964

    
1965
		echo ">>> Cleaning up old packages from repo..."
1966
		if ! poudriere pkgclean -f ${_bulk} -j ${jail_name} -p ${POUDRIERE_PORTS_NAME} -y; then
1967
			echo ">>> ERROR: Something went wrong..."
1968
			print_error_pfS
1969
		fi
1970

    
1971
		# ./ is intentional, it's a rsync trick to make it chdir to directory before send it
1972
		pkg_repo_rsync "/usr/local/poudriere/data/packages/./${jail_name}-${POUDRIERE_PORTS_NAME}"
1973
	done
1974
}
1975

    
1976
# This routine is called to write out to stdout
1977
# a string. The string is appended to $SNAPSHOTSLOGFILE
1978
# and we scp the log file to the builder host if
1979
# needed for the real time logging functions.
1980
snapshots_update_status() {
1981
	if [ -z "${SNAPSHOTS}" -o -z "$1" ]; then
1982
		return
1983
	fi
1984
	echo $1
1985
	echo "`date` -|- $1" >> $SNAPSHOTSLOGFILE
1986
	if [ -z "${DO_NOT_UPLOAD}" -a -n "${RSYNCIP}" ]; then
1987
		LU=`cat $SNAPSHOTSLASTUPDATE`
1988
		CT=`date "+%H%M%S"`
1989
		# Only update every minute
1990
		if [ "$LU" != "$CT" ]; then
1991
			ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCLOGS}"
1992
			scp -q $SNAPSHOTSLOGFILE ${RSYNCUSER}@${RSYNCIP}:${RSYNCLOGS}/build.log
1993
			date "+%H%M%S" > $SNAPSHOTSLASTUPDATE
1994
		fi
1995
	fi
1996
}
1997

    
1998
# Copy the current log file to $filename.old on
1999
# the snapshot www server (real time logs)
2000
snapshots_rotate_logfile() {
2001
	if [ -z "${DO_NOT_UPLOAD}" -a -n "${RSYNCIP}" ]; then
2002
		scp -q $SNAPSHOTSLOGFILE ${RSYNCUSER}@${RSYNCIP}:${RSYNCLOGS}/build.log.old
2003
	fi
2004

    
2005
	# Cleanup log file
2006
	rm -f $SNAPSHOTSLOGFILE;    touch $SNAPSHOTSLOGFILE
2007
	rm -f $SNAPSHOTSLASTUPDATE; touch $SNAPSHOTSLASTUPDATE
2008

    
2009
}
2010

    
2011
snapshots_copy_to_staging_nanobsd() {
2012
	for NANOTYPE in nanobsd nanobsd-vga; do
2013
		for FILESIZE in ${1}; do
2014
			FILENAMEFULL="${PRODUCT_NAME}-${PRODUCT_VERSION}-${FILESIZE}-${TARGET}-${NANOTYPE}${TIMESTAMP_SUFFIX}.img.gz"
2015
			FILENAMEUPGRADE="${PRODUCT_NAME}-${PRODUCT_VERSION}-${FILESIZE}-${TARGET}-${NANOTYPE}-upgrade${TIMESTAMP_SUFFIX}.img.gz"
2016
			mkdir -p $STAGINGAREA/nanobsd
2017
			mkdir -p $STAGINGAREA/nanobsdupdates
2018

    
2019
			cp $IMAGES_FINAL_DIR/$FILENAMEFULL $STAGINGAREA/nanobsd/ 2>/dev/null
2020
			cp $IMAGES_FINAL_DIR/$FILENAMEUPGRADE $STAGINGAREA/nanobsdupdates 2>/dev/null
2021

    
2022
			if [ -f $STAGINGAREA/nanobsd/$FILENAMEFULL ]; then
2023
				md5 $STAGINGAREA/nanobsd/$FILENAMEFULL > $STAGINGAREA/nanobsd/$FILENAMEFULL.md5 2>/dev/null
2024
				sha256 $STAGINGAREA/nanobsd/$FILENAMEFULL > $STAGINGAREA/nanobsd/$FILENAMEFULL.sha256 2>/dev/null
2025
			fi
2026
			if [ -f $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE ]; then
2027
				md5 $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE > $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE.md5 2>/dev/null
2028
				sha256 $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE > $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE.sha256 2>/dev/null
2029
			fi
2030

    
2031
			# Copy NanoBSD auto update:
2032
			if [ -f $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE ]; then
2033
				cp $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE $STAGINGAREA/latest-${NANOTYPE}-$FILESIZE.img.gz 2>/dev/null
2034
				sha256 $STAGINGAREA/latest-${NANOTYPE}-$FILESIZE.img.gz > $STAGINGAREA/latest-${NANOTYPE}-$FILESIZE.img.gz.sha256 2>/dev/null
2035
				# NOTE: Updates need a file with output similar to date output
2036
				# Use the file generated at start of snapshots_dobuilds() to be consistent on times
2037
				cp $BUILTDATESTRINGFILE $STAGINGAREA/version-${NANOTYPE}-$FILESIZE
2038
			fi
2039
		done
2040
	done
2041
}
2042

    
2043
snapshots_copy_to_staging_iso_updates() {
2044
	# Copy ISOs
2045
	md5 ${ISOPATH}.gz > ${ISOPATH}.md5
2046
	sha256 ${ISOPATH}.gz > ${ISOPATH}.sha256
2047
	cp ${ISOPATH}* $STAGINGAREA/ 2>/dev/null
2048

    
2049
	# Copy memstick items
2050
	md5 ${MEMSTICKPATH}.gz > ${MEMSTICKPATH}.md5
2051
	sha256 ${MEMSTICKPATH}.gz > ${MEMSTICKPATH}.sha256
2052
	cp ${MEMSTICKPATH}* $STAGINGAREA/ 2>/dev/null
2053

    
2054
	md5 ${MEMSTICKSERIALPATH}.gz > ${MEMSTICKSERIALPATH}.md5
2055
	sha256 ${MEMSTICKSERIALPATH}.gz > ${MEMSTICKSERIALPATH}.sha256
2056
	cp ${MEMSTICKSERIALPATH}* $STAGINGAREA/ 2>/dev/null
2057

    
2058
	if [ "${TARGET}" = "amd64" ]; then
2059
		md5 ${MEMSTICKADIPATH}.gz > ${MEMSTICKADIPATH}.md5
2060
		sha256 ${MEMSTICKADIPATH}.gz > ${MEMSTICKADIPATH}.sha256
2061
		cp ${MEMSTICKADIPATH}* $STAGINGAREA/ 2>/dev/null
2062
	fi
2063

    
2064
	md5 ${UPDATES_TARBALL_FILENAME} > ${UPDATES_TARBALL_FILENAME}.md5
2065
	sha256 ${UPDATES_TARBALL_FILENAME} > ${UPDATES_TARBALL_FILENAME}.sha256
2066
	cp ${UPDATES_TARBALL_FILENAME}* $STAGINGAREA/ 2>/dev/null
2067
	# NOTE: Updates need a file with output similar to date output
2068
	# Use the file generated at start of snapshots_dobuilds() to be consistent on times
2069
	if [ -z "${_IS_RELEASE}" ]; then
2070
		cp $BUILTDATESTRINGFILE $STAGINGAREA/version 2>/dev/null
2071
	fi
2072
}
2073

    
2074
snapshots_scp_files() {
2075
	if [ -z "${RSYNC_COPY_ARGUMENTS}" ]; then
2076
		RSYNC_COPY_ARGUMENTS="-ave ssh --timeout=60 --bwlimit=${RSYNCKBYTELIMIT}" #--bwlimit=50
2077
	fi
2078

    
2079
	snapshots_update_status ">>> Copying core pkg repo to ${PKG_RSYNC_HOSTNAME}"
2080
	# Add ./ before last directory, it's rsync trick to make it chdir to parent directory before send
2081
	pkg_repo_rsync $(echo "${CORE_PKG_PATH}" | sed -E 's,/$,,; s,/([^/]*)$,/./\1,')
2082
	snapshots_update_status ">>> Finished copying core pkg repo"
2083

    
2084
	snapshots_update_status ">>> Copying files to ${RSYNCIP}"
2085

    
2086
	# Ensure directory(s) are available
2087
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/livecd_installer"
2088
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/updates"
2089
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/nanobsd"
2090
	if [ -d $STAGINGAREA/virtualization ]; then
2091
		ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/virtualization"
2092
	fi
2093
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/.updaters"
2094
	# ensure permissions are correct for r+w
2095
	ssh ${RSYNCUSER}@${RSYNCIP} "chmod -R ug+rw ${RSYNCPATH}/."
2096
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/${PRODUCT_NAME}-*iso* \
2097
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/livecd_installer/
2098
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/${PRODUCT_NAME}-memstick* \
2099
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/livecd_installer/
2100
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/${PRODUCT_NAME}-*Update* \
2101
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/updates/
2102
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/nanobsd/* \
2103
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/nanobsd/
2104
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/nanobsdupdates/* \
2105
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/updates/
2106
	if [ -d $STAGINGAREA/virtualization ]; then
2107
		rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/virtualization/* \
2108
			${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/virtualization/
2109
	fi
2110

    
2111
	# Rather than copy these twice, use ln to link to the latest one.
2112

    
2113
	ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest.tgz"
2114
	ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest.tgz.sha256"
2115

    
2116
	LATESTFILENAME="`ls $UPDATESDIR/*.tgz | grep Full | grep -v md5 | grep -v sha256 | tail -n1`"
2117
	LATESTFILENAME=`basename ${LATESTFILENAME}`
2118
	ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${LATESTFILENAME} \
2119
		${RSYNCPATH}/.updaters/latest.tgz"
2120
	ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${LATESTFILENAME}.sha256 \
2121
		${RSYNCPATH}/.updaters/latest.tgz.sha256"
2122

    
2123
	for i in ${FLASH_SIZE}
2124
	do
2125
		ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz"
2126
		ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz.sha256"
2127
		ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz"
2128
		ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz.sha256"
2129

    
2130
		FILENAMEUPGRADE="${PRODUCT_NAME}-${PRODUCT_VERSION}-${i}-${TARGET}-nanobsd-upgrade${TIMESTAMP_SUFFIX}.img.gz"
2131
		ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE} \
2132
			${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz"
2133
		ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE}.sha256 \
2134
			${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz.sha256"
2135

    
2136
		FILENAMEUPGRADE="${PRODUCT_NAME}-${PRODUCT_VERSION}-${i}-${TARGET}-nanobsd-vga-upgrade${TIMESTAMP_SUFFIX}.img.gz"
2137
		ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE} \
2138
			${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz"
2139
		ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE}.sha256 \
2140
			${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz.sha256"
2141
	done
2142

    
2143
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/version* \
2144
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/.updaters
2145
	snapshots_update_status ">>> Finished copying files."
2146
}
(2-2/3)