Project

General

Profile

Download (86.7 KB) Statistics
| Branch: | Tag: | Revision:
1
#!/bin/sh
2
#
3
# builder_common.sh
4
#
5
# part of pfSense (https://www.pfsense.org)
6
# Copyright (c) 2004-2016 Rubicon Communications, LLC (Netgate)
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 are met:
21
#
22
# 1. Redistributions of source code must retain the above copyright notice,
23
#    this list of conditions and the following disclaimer.
24
#
25
# 2. Redistributions in binary form must reproduce the above copyright
26
#    notice, this list of conditions and the following disclaimer in
27
#    the documentation and/or other materials provided with the
28
#    distribution.
29
#
30
# 3. All advertising materials mentioning features or use of this software
31
#    must display the following acknowledgment:
32
#    "This product includes software developed by the pfSense Project
33
#    for use in the pfSense® software distribution. (http://www.pfsense.org/).
34
#
35
# 4. The names "pfSense" and "pfSense Project" must not be used to
36
#    endorse or promote products derived from this software without
37
#    prior written permission. For written permission, please contact
38
#    coreteam@pfsense.org.
39
#
40
# 5. Products derived from this software may not be called "pfSense"
41
#    nor may "pfSense" appear in their names without prior written
42
#    permission of the Electric Sheep Fencing, LLC.
43
#
44
# 6. Redistributions of any form whatsoever must retain the following
45
#    acknowledgment:
46
#
47
# "This product includes software developed by the pfSense Project
48
# for use in the pfSense software distribution (http://www.pfsense.org/).
49
#
50
# THIS SOFTWARE IS PROVIDED BY THE pfSense PROJECT ``AS IS'' AND ANY
51
# EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
52
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
53
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE pfSense PROJECT OR
54
# ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
55
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
56
# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
57
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
58
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
59
# STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
60
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
61
# OF THE POSSIBILITY OF SUCH DAMAGE.
62

    
63
if [ -n "${IMAGES_FINAL_DIR}" -a "${IMAGES_FINAL_DIR}" != "/" ]; then
64
	mkdir -p ${IMAGES_FINAL_DIR}
65
else
66
	echo "IMAGES_FINAL_DIR is not defined"
67
	print_error_pfS
68
fi
69

    
70
lc() {
71
	echo "${1}" | tr '[[:upper:]]' '[[:lower:]]'
72
}
73

    
74
git_last_commit() {
75
	export CURRENT_COMMIT=$(git -C ${BUILDER_ROOT} log -1 --format='%H')
76
	export CURRENT_AUTHOR=$(git -C ${BUILDER_ROOT} log -1 --format='%an')
77
	echo ">>> Last known commit $CURRENT_AUTHOR - $CURRENT_COMMIT"
78
	echo "$CURRENT_COMMIT" > $SCRATCHDIR/build_commit_info.txt
79
}
80

    
81
# Create core pkg repository
82
core_pkg_create_repo() {
83
	if [ ! -d "${CORE_PKG_REAL_PATH}/All" ]; then
84
		return
85
	fi
86

    
87
	############ ATTENTION ##############
88
	#
89
	# For some reason pkg-repo fail without / in the end of directory name
90
	# so removing it will break command
91
	#
92
	# https://github.com/freebsd/pkg/issues/1364
93
	#
94
	echo -n ">>> Creating core packages repository... "
95
	if pkg repo -q "${CORE_PKG_REAL_PATH}/"; then
96
		echo "Done!"
97
	else
98
		echo "Failed!"
99
		print_error_pfS
100
	fi
101

    
102
	# Use the same directory structure as poudriere does to avoid
103
	# breaking snapshot repositories during rsync
104
	ln -sf $(basename ${CORE_PKG_REAL_PATH}) ${CORE_PKG_PATH}/.latest
105
	ln -sf .latest/All ${CORE_PKG_ALL_PATH}
106
	ln -sf .latest/digests.txz ${CORE_PKG_PATH}/digests.txz
107
	ln -sf .latest/meta.txz ${CORE_PKG_PATH}/meta.txz
108
	ln -sf .latest/packagesite.txz ${CORE_PKG_PATH}/packagesite.txz
109
}
110

    
111
# Create core pkg (base, kernel)
112
core_pkg_create() {
113
	local _template="${1}"
114
	local _flavor="${2}"
115
	local _version="${3}"
116
	local _root="${4}"
117
	local _filter="${5}"
118

    
119
	[ -d "${CORE_PKG_TMP}" ] \
120
		&& rm -rf ${CORE_PKG_TMP}
121

    
122
	local _templates_path=${BUILDER_TOOLS}/templates/core_pkg/${_template}
123
	local _template_metadir=${_templates_path}/metadir
124
	local _metadir=${CORE_PKG_TMP}/${_template}_metadir
125

    
126
	if [ ! -d ${_template_metadir} ]; then
127
		echo "ERROR: Template dir not found for pkg ${_template}"
128
		exit
129
	fi
130

    
131
	mkdir -p ${CORE_PKG_TMP}
132

    
133
	cp -r ${_template_metadir} ${_metadir}
134

    
135
	local _manifest=${_metadir}/+MANIFEST
136
	local _plist=${CORE_PKG_TMP}/${_template}_plist
137
	local _exclude_plist=${CORE_PKG_TMP}/${_template}_exclude_plist
138

    
139
	if [ -f "${_templates_path}/pkg-plist" ]; then
140
		cp ${_templates_path}/pkg-plist ${_plist}
141
	else
142
		if [ -n "${_filter}" ]; then
143
			_filter="-name ${_filter}"
144
		fi
145
		(cd ${_root} && find . ${_filter} -type f -or -type l | sed 's,^.,,' | sort -u) > ${_plist}
146
	fi
147

    
148
	if [ -f "${_templates_path}/exclude_plist" ]; then
149
		cp ${_templates_path}/exclude_plist ${_exclude_plist}
150
	else
151
		touch ${_exclude_plist}
152
	fi
153

    
154
	sed \
155
		-i '' \
156
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
157
		-e "s,%%PRODUCT_URL%%,${PRODUCT_URL},g" \
158
		-e "s,%%FLAVOR%%,${_flavor:+-}${_flavor},g" \
159
		-e "s,%%FLAVOR_DESC%%,${_flavor:+ (${_flavor})},g" \
160
		-e "s,%%VERSION%%,${_version},g" \
161
		${_metadir}/* \
162
		${_plist} \
163
		${exclude_plist}
164

    
165
	if [ -f "${_exclude_plist}" ]; then
166
		sort -u ${_exclude_plist} > ${_plist}.exclude
167
		mv ${_plist} ${_plist}.tmp
168
		comm -23 ${_plist}.tmp ${_plist}.exclude > ${_plist}
169
		rm -f ${_plist}.tmp ${plist}.exclude
170
	fi
171

    
172
	# Add license information
173
	local _portname=$(sed '/^name: /!d; s,^[^"]*",,; s,",,' ${_metadir}/+MANIFEST)
174
	local _licenses_dir="/usr/local/share/licenses/${_portname}-${_version}"
175
	mkdir -p ${_root}${_licenses_dir}
176
	cp ${BUILDER_ROOT}/license.txt ${_root}${_licenses_dir}/ESF
177
	echo "This package has a single license: ESF (Electric Sheep Fencing License)." \
178
		> ${_root}${_licenses_dir}/LICENSE
179
	cat <<EOF >${_root}${_licenses_dir}/catalog.mk
180
_LICENSE=ESF
181
_LICENSE_NAME=Electric Sheep Fencing License
182
_LICENSE_PERMS=dist-mirror dist-sell pkg-mirror pkg-sell auto-accept
183
_LICENSE_GROUPS=
184
_LICENSE_DISTFILES=
185
EOF
186
	cat <<EOF >>${_plist}
187
${_licenses_dir}/catalog.mk
188
${_licenses_dir}/LICENSE
189
${_licenses_dir}/ESF
190
EOF
191

    
192
	mkdir -p ${CORE_PKG_REAL_PATH}/All
193
	if ! pkg create -o ${CORE_PKG_REAL_PATH}/All -p ${_plist} -r ${_root} -m ${_metadir}; then
194
		echo ">>> ERROR: Error building package ${_template} ${_flavor}"
195
		print_error_pfS
196
	fi
197

    
198
	# Cleanup _licenses_dir
199
	rm -rf ${_root}${_licenses_dir}
200
}
201

    
202
# This routine will output that something went wrong
203
print_error_pfS() {
204
	echo
205
	echo "####################################"
206
	echo "Something went wrong, check errors!" >&2
207
	echo "####################################"
208
	echo
209
	echo "NOTE: a lot of times you can run './build.sh --clean-builder' to resolve."
210
	echo
211
	if [ "$1" != "" ]; then
212
		echo $1
213
	fi
214
	[ -n "${LOGFILE}" -a -f "${LOGFILE}" ] && \
215
		echo "Log saved on ${LOGFILE}" && \
216
		tail -n20 ${LOGFILE} >&2
217
	echo
218
	kill $$
219
	exit 1
220
}
221

    
222
prestage_on_ram_setup() {
223
	# Do not use memory disks for release build
224
	if [ -n "${_IS_RELEASE}" ]; then
225
		return
226
	fi
227

    
228
	[ -d "${STAGE_CHROOT_DIR}" ] \
229
		|| mkdir -p ${STAGE_CHROOT_DIR}
230
	[ -d "${FINAL_CHROOT_DIR}" ] \
231
		|| mkdir -p ${FINAL_CHROOT_DIR}
232

    
233
	_AVAIL_MEM=$(($(sysctl -n hw.usermem) / 1024 / 1024))
234
	if [ $_AVAIL_MEM -lt 2000 ]; then
235
		echo ">>> Builder has less than 2GiB RAM skipping memory disks"
236
		return
237
	else
238
		echo "######################################################################################"
239
		echo
240
		echo ">>> Builder has more than 2GiB RAM enabling memory disks"
241
		echo ">>> WARNING: Remember to remove these memory disks by running $0 --disable-memorydisks"
242
		echo
243
		echo "######################################################################################"
244
	fi
245

    
246
	if df /dev/ufs/prestagebacking >/dev/null 2>&1; then
247
		echo ">>> Detected preexisting memory disk enabled for ${STAGE_CHROOT_DIR}."
248
	else
249
		mdconfig -a -t swap -u 10001 -s ${MEMORYDISK_SIZE}
250
		newfs -L prestagebacking -U /dev/md10001
251
		mount /dev/ufs/prestagebacking ${STAGE_CHROOT_DIR}
252
	fi
253

    
254
	if df /dev/ufs/finalstagebacking >/dev/null 2>&1; then
255
		echo ">>> Detected preexisting memory disk enabled for ${FINAL_CHROOT_DIR}."
256
	else
257
		mdconfig -a -t swap -u 10002 -s ${MEMORYDISK_SIZE}
258
		newfs -L finalstagebacking -U /dev/md10002
259
		mount /dev/ufs/finalstagebacking ${FINAL_CHROOT_DIR}
260
	fi
261
}
262

    
263
prestage_on_ram_cleanup() {
264
	if [ -c /dev/md10001 ]; then
265
		if [ -d ${STAGE_CHROOT_DIR} ]; then
266
			umount ${STAGE_CHROOT_DIR}
267
		fi
268
		mdconfig -d -u 10001
269
	fi
270
	if [ -c /dev/md10002 ]; then
271
		if [ -d ${FINAL_CHROOT_DIR} ]; then
272
			umount ${FINAL_CHROOT_DIR}
273
		fi
274
		mdconfig -d -u 10002
275
	fi
276
}
277

    
278
# This routine will verify that the kernel has been
279
# installed OK to the staging area.
280
ensure_kernel_exists() {
281
	if [ ! -f "$1/boot/kernel/kernel.gz" ]; then
282
		echo ">>> ERROR: Could not locate $1/boot/kernel.gz"
283
		print_error_pfS
284
	fi
285
	KERNEL_SIZE=$(stat -f "%z" $1/boot/kernel/kernel.gz)
286
	if [ "$KERNEL_SIZE" -lt 3500 ]; then
287
		echo ">>> ERROR: Kernel $1/boot/kernel.gz appears to be smaller than it should be: $KERNEL_SIZE"
288
		print_error_pfS
289
	fi
290
}
291

    
292
get_pkg_name() {
293
	echo "${PRODUCT_NAME}-${1}-${CORE_PKG_VERSION}"
294
}
295

    
296
# This routine builds all related kernels
297
build_all_kernels() {
298
	# Set KERNEL_BUILD_PATH if it has not been set
299
	if [ -z "${KERNEL_BUILD_PATH}" ]; then
300
		KERNEL_BUILD_PATH=$SCRATCHDIR/kernels
301
		echo ">>> KERNEL_BUILD_PATH has not been set. Setting to ${KERNEL_BUILD_PATH}!"
302
	fi
303

    
304
	[ -d "${KERNEL_BUILD_PATH}" ] \
305
		&& rm -rf ${KERNEL_BUILD_PATH}
306

    
307
	# Build embedded kernel
308
	for BUILD_KERNEL in $BUILD_KERNELS; do
309
		unset KERNCONF
310
		unset KERNEL_DESTDIR
311
		unset KERNELCONF
312
		unset KERNEL_NAME
313
		export KERNCONF=$BUILD_KERNEL
314
		export KERNEL_DESTDIR="$KERNEL_BUILD_PATH/$BUILD_KERNEL"
315
		export KERNELCONF="${FREEBSD_SRC_DIR}/sys/${TARGET}/conf/$BUILD_KERNEL"
316
		export KERNEL_NAME=${BUILD_KERNEL}
317

    
318
		LOGFILE="${BUILDER_LOGS}/kernel.${KERNCONF}.${TARGET}.log"
319
		echo ">>> Building $BUILD_KERNEL kernel."  | tee -a ${LOGFILE}
320

    
321
		if [ ! -e "${FREEBSD_SRC_DIR}/sys/${TARGET}/conf/${BUILD_KERNEL}" ]; then
322
			echo ">>> ERROR: Could not find $KERNELCONF"
323
			print_error_pfS
324
		fi
325

    
326
		if [ -n "${NO_BUILDKERNEL}" -a -f "${CORE_PKG_ALL_PATH}/$(get_pkg_name kernel-${KERNEL_NAME}).txz" ]; then
327
			echo ">>> NO_BUILDKERNEL set, skipping build" | tee -a ${LOGFILE}
328
			continue
329
		fi
330

    
331
		export SRC_CONF=${SRC_CONF}
332
		buildkernel
333

    
334
		echo ">>> Staging $BUILD_KERNEL kernel..." | tee -a ${LOGFILE}
335
		installkernel
336

    
337
		ensure_kernel_exists $KERNEL_DESTDIR
338

    
339
		echo -n ">>> Creating pkg of $KERNEL_NAME-debug kernel to staging area... "  | tee -a ${LOGFILE}
340
		core_pkg_create kernel-debug ${KERNEL_NAME} ${CORE_PKG_VERSION} ${KERNEL_DESTDIR} \*.symbols
341
		find ${KERNEL_DESTDIR} -name '*.symbols' -type f -delete
342
		echo " Done" | tee -a ${LOGFILE}
343

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

    
347
		rm -rf $KERNEL_DESTDIR 2>&1 1>/dev/null
348

    
349
		echo " Done" | tee -a ${LOGFILE}
350
	done
351
}
352

    
353
install_default_kernel() {
354
	if [ -z "${1}" ]; then
355
		echo ">>> ERROR: install_default_kernel called without a kernel config name"| tee -a ${LOGFILE}
356
		print_error_pfS
357
	fi
358

    
359
	export KERNEL_NAME="${1}"
360

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

    
363
	# Copy kernel package to chroot, otherwise pkg won't find it to install
364
	if ! pkg_chroot_add ${FINAL_CHROOT_DIR} kernel-${KERNEL_NAME}; then
365
		echo ">>> ERROR: Error installing kernel package $(get_pkg_name kernel-${KERNEL_NAME}).txz" | tee -a ${LOGFILE}
366
		print_error_pfS
367
	fi
368

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

    
372
	if [ ! -f $FINAL_CHROOT_DIR/boot/kernel/kernel.gz ]; then
373
		echo ">>> ERROR: No kernel installed on $FINAL_CHROOT_DIR and the resulting image will be unusable. STOPPING!" | tee -a ${LOGFILE}
374
		print_error_pfS
375
	fi
376
	mkdir -p $FINAL_CHROOT_DIR/pkgs
377
	if [ -z "${2}" -o -n "${INSTALL_EXTRA_KERNELS}" ]; then
378
		cp ${CORE_PKG_ALL_PATH}/$(get_pkg_name kernel-${KERNEL_NAME}).txz $FINAL_CHROOT_DIR/pkgs
379
		if [ -n "${INSTALL_EXTRA_KERNELS}" ]; then
380
			for _EXTRA_KERNEL in $INSTALL_EXTRA_KERNELS; do
381
				_EXTRA_KERNEL_PATH=${CORE_PKG_ALL_PATH}/$(get_pkg_name kernel-${_EXTRA_KERNEL}).txz
382
				if [ -f "${_EXTRA_KERNEL_PATH}" ]; then
383
					echo -n ". adding ${_EXTRA_KERNEL_PATH} on image /pkgs folder"
384
					cp ${_EXTRA_KERNEL_PATH} $FINAL_CHROOT_DIR/pkgs
385
				else
386
					echo ">>> ERROR: Requested kernel $(get_pkg_name kernel-${_EXTRA_KERNEL}).txz was not found to be put on image /pkgs folder!"
387
					print_error_pfS
388
				fi
389
			done
390
		fi
391
	fi
392
	echo "Done." | tee -a ${LOGFILE}
393

    
394
	unset KERNEL_NAME
395
}
396

    
397
# Creates a full update file
398
create_Full_update_tarball() {
399
	local _variant="$1"
400

    
401
	LOGFILE=${BUILDER_LOGS}/full_update.${TARGET}
402
	if [ "${UPDATES_TARBALL_FILENAME}" = "" ]; then
403
		echo ">>> UPDATES_TARBALL_FILENAME is empty skipping generation of full update image!" | tee -a ${LOGFILE}
404
		return
405
	fi
406

    
407
	local _image_path=${UPDATES_TARBALL_FILENAME}
408
	if [ -n "${_variant}" ]; then
409
		_image_path=$(echo "$_image_path" | \
410
			sed "s/-Full-Update-/-Full-Update-${_variant}-/")
411
		VARIANTUPDATES="${VARIANTUPDATES}${VARIANTUPDATES:+ }${_image_path}"
412
	fi
413

    
414
	mkdir -p $UPDATESDIR
415

    
416
	customize_stagearea_for_image "fullupdate" "" $_variant
417
	install_default_kernel ${DEFAULT_KERNEL}
418

    
419
	rm -rf ${FINAL_CHROOT_DIR}/cf
420
	rm -rf ${FINAL_CHROOT_DIR}/conf
421
	rm -f ${FINAL_CHROOT_DIR}/etc/dh-parameters.*
422
	rm -f ${FINAL_CHROOT_DIR}/etc/rc.conf
423
	rm -f ${FINAL_CHROOT_DIR}/etc/pwd.db 2>/dev/null
424
	rm -f ${FINAL_CHROOT_DIR}/etc/group 2>/dev/null
425
	rm -f ${FINAL_CHROOT_DIR}/etc/spwd.db 2>/dev/null
426
	rm -f ${FINAL_CHROOT_DIR}/etc/passwd 2>/dev/null
427
	rm -f ${FINAL_CHROOT_DIR}/etc/master.passwd 2>/dev/null
428
	rm -f ${FINAL_CHROOT_DIR}/etc/fstab 2>/dev/null
429
	rm -f ${FINAL_CHROOT_DIR}/etc/bogons 2>/dev/null
430
	# Remove loader.conf and friends.  Ticket #560
431
	rm ${FINAL_CHROOT_DIR}/boot/loader.conf 2>/dev/null
432
	rm ${FINAL_CHROOT_DIR}/boot/loader.conf.local 2>/dev/null
433

    
434
	# Old systems will run (pre|post)_upgrade_command from /tmp
435
	if [ -f ${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/pre_upgrade_command ]; then
436
		cp -p \
437
			${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/pre_upgrade_command \
438
			${FINAL_CHROOT_DIR}/tmp
439
	fi
440
	if [ -f ${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/post_upgrade_command ]; then
441
		cp -p \
442
			${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/post_upgrade_command \
443
			${FINAL_CHROOT_DIR}/tmp
444
	fi
445

    
446
	echo ">>> Creating ${_image_path} ..." | tee -a ${LOGFILE}
447
	tar --exclude=./dev -czPf ${_image_path} -C ${FINAL_CHROOT_DIR} .
448

    
449
	echo ">>> Full Update created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
450
}
451

    
452
# Outputs various set variables aka env
453
print_flags() {
454

    
455
	echo
456
	printf "             Product version: %s\n" $PRODUCT_VERSION
457
	printf "                   Stage DIR: %s\n" $STAGE_CHROOT_DIR
458
	printf "                 Updates dir: %s\n" $UPDATESDIR
459
	printf " Image Preparation Stage DIR: %s\n" $FINAL_CHROOT_DIR
460
	printf "                  Source DIR: %s\n" $FREEBSD_SRC_DIR
461
	printf "          FreeBSD repository: %s\n" $FREEBSD_REPO_BASE
462
	printf "          FreeBSD-src branch: %s\n" $FREEBSD_BRANCH
463
	printf "     FreeBSD original branch: %s\n" $FREEBSD_PARENT_BRANCH
464
	printf "               BUILD_KERNELS: %s\n" $BUILD_KERNELS
465
	printf "           Git Branch or Tag: %s\n" $GIT_REPO_BRANCH_OR_TAG
466
	printf "            MODULES_OVERRIDE: %s\n" $MODULES_OVERRIDE
467
	printf "    VMDK_DISK_CAPACITY_IN_GB: %s\n" $VMDK_DISK_CAPACITY_IN_GB
468
	printf "                 OVFTEMPLATE: %s\n" $OVFTEMPLATE
469
	printf "                     OVFVMDK: %s\n" $OVFVMDK
470
	printf "                    SRC_CONF: %s\n" $SRC_CONF
471
	printf "                     ISOPATH: %s\n" $ISOPATH
472
	printf "                MEMSTICKPATH: %s\n" $MEMSTICKPATH
473
	printf "    UPDATES_TARBALL_FILENAME: %s\n" $UPDATES_TARBALL_FILENAME
474
if [ -n "$SHOW_ENV" ]; then
475
	for LINE in $(env | egrep -v '(terminal|PASS|NAME|USER|SSH|GROUP|HOST)'); do
476
		echo "SHOW_ENV: $LINE"
477
	done
478
fi
479
	echo
480
}
481

    
482
# This builds FreeBSD (make buildworld)
483
# Imported from FreeSBIE
484
make_world() {
485

    
486
	LOGFILE=${BUILDER_LOGS}/buildworld.${TARGET}
487
	if [ -n "${NO_BUILDWORLD}" ]; then
488
		echo ">>> NO_BUILDWORLD set, skipping build" | tee -a ${LOGFILE}
489
		return
490
	fi
491

    
492
	# Set SRC_CONF variable if it's not already set.
493
	if [ -z "${SRC_CONF}" ]; then
494
		echo ">>> SRC_CONF is unset make sure this is what you want!" | tee -a ${LOGFILE}
495
	else
496
		echo ">>> Setting SRC_CONF to $SRC_CONF" | tee -a ${LOGFILE}
497
	fi
498

    
499
	# Set default parameters
500
	export MAKE_ARGS="${MAKEJ_WORLD} __MAKE_CONF=${MAKE_CONF} SRCCONF=${SRC_CONF} TARGET=${TARGET} TARGET_ARCH=${TARGET_ARCH}"
501

    
502
	echo ">>> LOGFILE set to $LOGFILE." | tee -a ${LOGFILE}
503
	makeargs="${MAKE_ARGS}"
504
	echo ">>> Building world for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
505
	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}
506
	(env LOCAL_ITOOLS="${EXTRA_TOOLS}" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} -DNO_CLEAN ${makeargs} buildworld || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
507
	echo ">>> Building world for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
508

    
509
	LOGFILE=${BUILDER_LOGS}/installworld.${TARGET}
510
	echo ">>> LOGFILE set to $LOGFILE." | tee -a ${LOGFILE}
511
	# Create if cleaned up
512
	makeargs="${MAKE_ARGS} DESTDIR=${STAGE_CHROOT_DIR} WITHOUT_TOOLCHAIN=1"
513
	echo ">>> Installing world for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
514
	echo ">>> Builder is running the command: env LOCAL_ITOOLS=\"${EXTRA_TOOLS}\" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} installworld" | tee -a ${LOGFILE}
515
	(env LOCAL_ITOOLS="${EXTRA_TOOLS}" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} installworld || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
516
	echo ">>> Installing world for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
517

    
518
	makeargs="${MAKE_ARGS} DESTDIR=${STAGE_CHROOT_DIR}"
519
	echo ">>> Distribution world for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
520
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} distribution " | tee -a ${LOGFILE}
521
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} distribution  || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
522
	echo ">>> Distribution world for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
523

    
524
	[ -d "${STAGE_CHROOT_DIR}/usr/local/bin" ] \
525
		|| mkdir -p ${STAGE_CHROOT_DIR}/usr/local/bin
526
	makeargs="${MAKE_ARGS} DESTDIR=${STAGE_CHROOT_DIR}"
527
	echo ">>> Building and installing crypto tools and athstats for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
528
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/crypto ${makeargs} clean all install " | tee -a ${LOGFILE}
529
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/crypto ${makeargs} clean all install || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
530
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} clean" | tee -a ${LOGFILE}
531
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} clean || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
532
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} all" | tee -a ${LOGFILE}
533
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} all || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
534
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} install" | tee -a ${LOGFILE}
535
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} install || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
536
	echo ">>> Building and installing crypto tools and athstats for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
537

    
538
	unset makeargs
539
}
540

    
541
nanobsd_image_filename() {
542
	local _size="$1"
543
	local _type="$2"
544
	local _upgrade="$3"
545

    
546
	if [ -z "$_upgrade" ]; then
547
		local _template=${NANOBSD_IMG_TEMPLATE}
548
	else
549
		local _template=${NANOBSD_UPGRADE_TEMPLATE}
550
	fi
551

    
552
	echo "$_template" | sed \
553
		-e "s,%%SIZE%%,${_size},g" \
554
		-e "s,%%TYPE%%,${_type},g"
555

    
556
	return 0
557
}
558

    
559
# This routine originated in nanobsd.sh
560
nanobsd_set_flash_details () {
561
	a1=$(echo $1 | tr '[:upper:]' '[:lower:]')
562

    
563
	# Source:
564
	#	SanDisk CompactFlash Memory Card
565
	#	Product Manual
566
	#	Version 10.9
567
	#	Document No. 20-10-00038
568
	#	April 2005
569
	# Table 2-7
570
	# NB: notice math error in SDCFJ-4096-388 line.
571
	#
572
	case "${a1}" in
573
		2048|2048m|2048mb|2g)
574
			NANO_MEDIASIZE=$((1989999616/512))
575
			;;
576
		4096|4096m|4096mb|4g)
577
			NANO_MEDIASIZE=$((3989999616/512))
578
			;;
579
		8192|8192m|8192mb|8g)
580
			NANO_MEDIASIZE=$((7989999616/512))
581
			;;
582
		16384|16384m|16384mb|16g)
583
			NANO_MEDIASIZE=$((15989999616/512))
584
			;;
585
		*)
586
			echo "Unknown Flash capacity"
587
			exit 2
588
			;;
589
	esac
590

    
591
	NANO_HEADS=16
592
	NANO_SECTS=63
593

    
594
	echo ">>> [nanoo] $1"
595
	echo ">>> [nanoo] NANO_MEDIASIZE: $NANO_MEDIASIZE"
596
	echo ">>> [nanoo] NANO_HEADS: $NANO_HEADS"
597
	echo ">>> [nanoo] NANO_SECTS: $NANO_SECTS"
598
	echo ">>> [nanoo] NANO_BOOT0CFG: $NANO_BOOT0CFG"
599
}
600

    
601
# This routine originated in nanobsd.sh
602
create_nanobsd_diskimage () {
603
	if [ -z "${1}" ]; then
604
		echo ">>> ERROR: Type of image has not been specified"
605
		print_error_pfS
606
	fi
607
	if [ -z "${2}" ]; then
608
		echo ">>> ERROR: Size of image has not been specified"
609
		print_error_pfS
610
	fi
611

    
612
	if [ "${1}" = "nanobsd" ]; then
613
		# It's serial
614
		export NANO_BOOTLOADER="boot/boot0sio"
615
	elif [ "${1}" = "nanobsd-vga" ]; then
616
		# It's vga
617
		export NANO_BOOTLOADER="boot/boot0"
618
	else
619
		echo ">>> ERROR: Type of image to create unknown"
620
		print_error_pfS
621
	fi
622

    
623
	if [ -z "${2}" ]; then
624
		echo ">>> ERROR: Media size(s) not specified."
625
		print_error_pfS
626
	fi
627

    
628
	if [ -z "${2}" ]; then
629
		echo ">>> ERROR: FLASH_SIZE is not set."
630
		print_error_pfS
631
	fi
632

    
633
	LOGFILE=${BUILDER_LOGS}/${1}.${TARGET}
634
	# Prepare folder to be put in image
635
	customize_stagearea_for_image "${1}"
636
	install_default_kernel ${DEFAULT_KERNEL} "no"
637

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

    
640
	echo "nanobsd" > $FINAL_CHROOT_DIR/etc/platform
641

    
642
	local BOOTCONF=${FINAL_CHROOT_DIR}/boot.config
643
	local LOADERCONF=${FINAL_CHROOT_DIR}/boot/loader.conf
644

    
645
	if [ "${1}" = "nanobsd" ]; then
646
		# Tell loader to use serial console early.
647
		echo "-S115200 -h" >> ${BOOTCONF}
648

    
649
		# Remove old console options if present.
650
		[ -f "${LOADERCONF}" ] \
651
			&& sed -i "" -Ee "/(console|boot_multicons|boot_serial|hint.uart)/d" ${LOADERCONF}
652
		# Activate serial console+video console in loader.conf
653
		echo 'loader_color="NO"' >> ${LOADERCONF}
654
		echo 'beastie_disable="YES"' >> ${LOADERCONF}
655
		echo 'boot_serial="YES"' >> ${LOADERCONF}
656
		echo 'console="comconsole"' >> ${LOADERCONF}
657
		echo 'comconsole_speed="115200"' >> ${LOADERCONF}
658
	fi
659
	echo 'autoboot_delay="5"' >> ${LOADERCONF}
660

    
661
	# Old systems will run (pre|post)_upgrade_command from /tmp
662
	if [ -f ${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/pre_upgrade_command ]; then
663
		cp -p \
664
			${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/pre_upgrade_command \
665
			${FINAL_CHROOT_DIR}/tmp
666
	fi
667
	if [ -f ${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/post_upgrade_command ]; then
668
		cp -p \
669
			${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/post_upgrade_command \
670
			${FINAL_CHROOT_DIR}/tmp
671
	fi
672

    
673
	for _NANO_MEDIASIZE in ${2}; do
674
		if [ -z "${_NANO_MEDIASIZE}" ]; then
675
			continue;
676
		fi
677

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

    
681
		IMG="${IMAGES_FINAL_DIR}/$(nanobsd_image_filename ${_NANO_MEDIASIZE} ${1})"
682
		IMGUPDATE="${IMAGES_FINAL_DIR}/$(nanobsd_image_filename ${_NANO_MEDIASIZE} ${1} 1)"
683

    
684
		nanobsd_set_flash_details ${_NANO_MEDIASIZE}
685

    
686
		# These are defined in FlashDevice and on builder_default.sh
687
		echo $NANO_MEDIASIZE \
688
			$NANO_IMAGES \
689
			$NANO_SECTS \
690
			$NANO_HEADS \
691
			$NANO_CODESIZE \
692
			$NANO_CONFSIZE \
693
			$NANO_DATASIZE |
694
awk '
695
{
696
	printf "# %s\n", $0
697

    
698
	# size of cylinder in sectors
699
	cs = $3 * $4
700

    
701
	# number of full cylinders on media
702
	cyl = int ($1 / cs)
703

    
704
	# output fdisk geometry spec, truncate cyls to 1023
705
	if (cyl <= 1023)
706
		print "g c" cyl " h" $4 " s" $3
707
	else
708
		print "g c" 1023 " h" $4 " s" $3
709

    
710
	if ($7 > 0) {
711
		# size of data partition in full cylinders
712
		dsl = int (($7 + cs - 1) / cs)
713
	} else {
714
		dsl = 0;
715
	}
716

    
717
	# size of config partition in full cylinders
718
	csl = int (($6 + cs - 1) / cs)
719

    
720
	if ($5 == 0) {
721
		# size of image partition(s) in full cylinders
722
		isl = int ((cyl - dsl - csl) / $2)
723
	} else {
724
		isl = int (($5 + cs - 1) / cs)
725
	}
726

    
727
	# First image partition start at second track
728
	print "p 1 165 " $3, isl * cs - $3
729
	c = isl * cs;
730

    
731
	# Second image partition (if any) also starts offset one
732
	# track to keep them identical.
733
	if ($2 > 1) {
734
		print "p 2 165 " $3 + c, isl * cs - $3
735
		c += isl * cs;
736
	}
737

    
738
	# Config partition starts at cylinder boundary.
739
	print "p 3 165 " c, csl * cs
740
	c += csl * cs
741

    
742
	# Data partition (if any) starts at cylinder boundary.
743
	if ($7 > 0) {
744
		print "p 4 165 " c, dsl * cs
745
	} else if ($7 < 0 && $1 > c) {
746
		print "p 4 165 " c, $1 - c
747
	} else if ($1 < c) {
748
		print "Disk space overcommitted by", \
749
		    c - $1, "sectors" > "/dev/stderr"
750
		exit 2
751
	}
752

    
753
	# Force slice 1 to be marked active. This is necessary
754
	# for booting the image from a USB device to work.
755
	print "a 1"
756
}
757
	' > ${IMAGES_FINAL_DIR}/_.fdisk
758

    
759
		MNT=${IMAGES_FINAL_DIR}/_.mnt
760
		mkdir -p ${MNT}
761

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

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

    
768
		fdisk -i -f ${IMAGES_FINAL_DIR}/_.fdisk ${MD} 2>&1 >> ${LOGFILE}
769
		fdisk ${MD} 2>&1 >> ${LOGFILE}
770

    
771
		boot0cfg -t 100 -B -b ${FINAL_CHROOT_DIR}/${NANO_BOOTLOADER} ${NANO_BOOT0CFG} ${MD} 2>&1 >> ${LOGFILE}
772

    
773
		# Create first image
774
		bsdlabel -m i386 -w -B -b ${FINAL_CHROOT_DIR}/boot/boot ${MD}s1 2>&1 >> ${LOGFILE}
775
		bsdlabel -m i386 ${MD}s1 2>&1 >> ${LOGFILE}
776
		local _label=$(lc ${PRODUCT_NAME})
777
		newfs -L ${_label}0 ${NANO_NEWFS} /dev/${MD}s1a 2>&1 >> ${LOGFILE}
778
		mount /dev/ufs/${_label}0 ${MNT}
779
		if [ $? -ne 0 ]; then
780
			echo ">>> ERROR: Something wrong happened during mount of first slice image creation. STOPPING!" | tee -a ${LOGFILE}
781
			print_error_pfS
782
		fi
783
		# Consider the unmounting as well
784
		trap "umount /dev/ufs/${_label}0; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
785

    
786
		clone_directory_contents ${FINAL_CHROOT_DIR} ${MNT}
787

    
788
		# Set NanoBSD image size
789
		echo "${_NANO_MEDIASIZE}" > ${MNT}/etc/nanosize.txt
790

    
791
		echo "/dev/ufs/${_label}0 / ufs ro,sync,noatime 1 1" > ${MNT}/etc/fstab
792
		if [ $NANO_CONFSIZE -gt 0 ] ; then
793
			echo "/dev/ufs/cf /cf ufs ro,sync,noatime 1 1" >> ${MNT}/etc/fstab
794
		fi
795

    
796
		umount ${MNT}
797
		# Restore the original trap
798
		trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
799

    
800
		# Setting NANO_IMAGES to 1 and NANO_INIT_IMG2 will tell
801
		# NanoBSD to only create one partition.  We default to 2
802
		# partitions in case anything happens to the first the
803
		# operator can boot from the 2nd and should be OK.
804

    
805
		# Before just going to use dd for duplicate think!
806
		# The images are created as sparse so lets take advantage
807
		# of that by just exec some commands.
808
		if [ $NANO_IMAGES -gt 1 -a $NANO_INIT_IMG2 -gt 0 ] ; then
809
			# Duplicate to second image (if present)
810
			echo ">>> Creating NanoBSD second slice by duplicating first slice." | tee -a ${LOGFILE}
811
			# Create second image
812
			dd if=/dev/${MD}s1 of=/dev/${MD}s2 conv=sparse bs=64k 2>&1 >> ${LOGFILE}
813
			tunefs -L ${_label}1 /dev/${MD}s2a 2>&1 >> ${LOGFILE}
814
			mount /dev/ufs/${_label}1 ${MNT}
815
			if [ $? -ne 0 ]; then
816
				echo ">>> ERROR: Something wrong happened during mount of second slice image creation. STOPPING!" | tee -a ${LOGFILE}
817
				print_error_pfS
818
			fi
819
			# Consider the unmounting as well
820
			trap "umount /dev/ufs/${_label}1; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
821

    
822
			echo "/dev/ufs/${_label}1 / ufs ro,sync,noatime 1 1" > ${MNT}/etc/fstab
823
			if [ $NANO_CONFSIZE -gt 0 ] ; then
824
				echo "/dev/ufs/cf /cf ufs ro,sync,noatime 1 1" >> ${MNT}/etc/fstab
825
			fi
826

    
827
			umount ${MNT}
828
			# Restore the trap back
829
			trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
830
		fi
831

    
832
		# Create Data slice, if any.
833
		# Note the changing of the variable to NANO_CONFSIZE
834
		# from NANO_DATASIZE.  We also added glabel support
835
		# and populate the Product configuration from the /cf
836
		# directory located in FINAL_CHROOT_DIR
837
		if [ $NANO_CONFSIZE -gt 0 ] ; then
838
			echo ">>> Creating /cf area to hold config.xml"
839
			newfs -L cf ${NANO_NEWFS} /dev/${MD}s3 2>&1 >> ${LOGFILE}
840
			# Mount data partition and copy contents of /cf
841
			# Can be used later to create custom default config.xml while building
842
			mount /dev/ufs/cf ${MNT}
843
			if [ $? -ne 0 ]; then
844
				echo ">>> ERROR: Something wrong happened during mount of cf slice image creation. STOPPING!" | tee -a ${LOGFILE}
845
				print_error_pfS
846
			fi
847
			# Consider the unmounting as well
848
			trap "umount /dev/ufs/cf; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
849

    
850
			clone_directory_contents ${FINAL_CHROOT_DIR}/cf ${MNT}
851

    
852
			umount ${MNT}
853
			# Restore the trap back
854
			trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
855
		else
856
			">>> [nanoo] NANO_CONFSIZE is not set. Not adding a /conf partition.. You sure about this??" | tee -a ${LOGFILE}
857
		fi
858

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

    
862
		mdconfig -d -u $MD
863
		# Restore default action
864
		trap "-" 1 2 15 EXIT
865

    
866
		# Check each image and ensure that they are over
867
		# 3 megabytes.  If either image is under 20 megabytes
868
		# in size then error out.
869
		IMGSIZE=$(stat -f "%z" ${IMG})
870
		IMGUPDATESIZE=$(stat -f "%z" ${IMGUPDATE})
871
		CHECKSIZE="20040710"
872
		if [ "$IMGSIZE" -lt "$CHECKSIZE" ]; then
873
			echo ">>> ERROR: Something went wrong when building NanoBSD.  The image size is under 20 megabytes!" | tee -a ${LOGFILE}
874
			print_error_pfS
875
		fi
876
		if [ "$IMGUPDATESIZE" -lt "$CHECKSIZE" ]; then
877
			echo ">>> ERROR: Something went wrong when building NanoBSD upgrade image.  The image size is under 20 megabytes!" | tee -a ${LOGFILE}
878
			print_error_pfS
879
		fi
880

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

    
884
		gzip -qf $IMG &
885
		_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
886
		gzip -qf $IMGUPDATE &
887
		_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
888
	done
889

    
890
	unset IMG
891
	unset IMGUPDATE
892
	unset IMGUPDATESIZE
893
	unset IMGSIZE
894

    
895
	ls -lah $IMAGES_FINAL_DIR
896
}
897

    
898
# This routine creates a ova image that contains
899
# a ovf and vmdk file. These files can be imported
900
# right into vmware or virtual box.
901
# (and many other emulation platforms)
902
# http://www.vmware.com/pdf/ovf_whitepaper_specification.pdf
903
create_ova_image() {
904
	# XXX create a .ovf php creator that you can pass:
905
	#     1. populatedSize
906
	#     2. license
907
	#     3. product name
908
	#     4. version
909
	#     5. number of network interface cards
910
	#     6. allocationUnits
911
	#     7. capacity
912
	#     8. capacityAllocationUnits
913

    
914
	LOGFILE=${BUILDER_LOGS}/ova.${TARGET}.log
915

    
916
	if [ -d "${OVA_TMP}" ]; then
917
		chflags -R noschg ${OVA_TMP}
918
		rm -rf ${OVA_TMP}
919
	fi
920

    
921
	local _mntdir=${OVA_TMP}/mnt
922
	mkdir -p ${_mntdir}
923

    
924
	if [ -z "${OVA_SWAP_PART_SIZE_IN_GB}" -o "${OVA_SWAP_PART_SIZE_IN_GB}" = "0" ]; then
925
		# first partition size (freebsd-ufs)
926
		local OVA_FIRST_PART_SIZE_IN_GB=${VMDK_DISK_CAPACITY_IN_GB}
927
		# Calculate real first partition size, removing 128 blocks (65536 bytes) beginning/loader
928
		local OVA_FIRST_PART_SIZE=$((${OVA_FIRST_PART_SIZE_IN_GB}*1024*1024*1024-65536))
929
		# Unset swap partition size variable
930
		unset OVA_SWAP_PART_SIZE
931
		# Parameter used by mkimg
932
		unset OVA_SWAP_PART_PARAM
933
	else
934
		# first partition size (freebsd-ufs)
935
		local OVA_FIRST_PART_SIZE_IN_GB=$((VMDK_DISK_CAPACITY_IN_GB-OVA_SWAP_PART_SIZE_IN_GB))
936
		# Use first partition size in g
937
		local OVA_FIRST_PART_SIZE="${OVA_FIRST_PART_SIZE_IN_GB}g"
938
		# Calculate real swap size, removing 128 blocks (65536 bytes) beginning/loader
939
		local OVA_SWAP_PART_SIZE=$((${OVA_SWAP_PART_SIZE_IN_GB}*1024*1024*1024-65536))
940
		# Parameter used by mkimg
941
		local OVA_SWAP_PART_PARAM="-p freebsd-swap/swap0::${OVA_SWAP_PART_SIZE}"
942
	fi
943

    
944
	# Prepare folder to be put in image
945
	customize_stagearea_for_image "ova"
946
	install_default_kernel ${DEFAULT_KERNEL} "no"
947

    
948
	# Fill fstab
949
	echo ">>> Installing platform specific items..." | tee -a ${LOGFILE}
950
	echo "/dev/gpt/${PRODUCT_NAME}	/	ufs		rw	1	1" > ${FINAL_CHROOT_DIR}/etc/fstab
951
	if [ -n "${OVA_SWAP_PART_SIZE}" ]; then
952
		echo "/dev/gpt/swap0	none	swap	sw	0	0" >> ${FINAL_CHROOT_DIR}/etc/fstab
953
	fi
954

    
955
	# Create / partition
956
	echo -n ">>> Creating / partition... " | tee -a ${LOGFILE}
957
	truncate -s ${OVA_FIRST_PART_SIZE} ${OVA_TMP}/${OVFUFS}
958
	local _md=$(mdconfig -a -f ${OVA_TMP}/${OVFUFS})
959
	trap "mdconfig -d -u ${_md}; return" 1 2 15 EXIT
960

    
961
	newfs -L ${PRODUCT_NAME} -j /dev/${_md} 2>&1 >>${LOGFILE}
962

    
963
	if ! mount /dev/${_md} ${_mntdir} 2>&1 >>${LOGFILE}; then
964
		echo "Failed!" | tee -a ${LOGFILE}
965
		echo ">>> ERROR: Error mounting temporary vmdk image. STOPPING!" | tee -a ${LOGFILE}
966
		print_error_pfS
967
	fi
968
	trap "umount ${_mntdir}; mdconfig -d -u ${_md}; return" 1 2 15 EXIT
969

    
970
	echo "Done!" | tee -a ${LOGFILE}
971

    
972
	clone_directory_contents ${FINAL_CHROOT_DIR} ${_mntdir}
973

    
974
	sync
975
	umount ${_mntdir} 2>&1 >>${LOGFILE}
976
	mdconfig -d -u ${_md}
977
	trap "-" 1 2 15 EXIT
978

    
979
	# Create raw disk
980
	echo -n ">>> Creating raw disk... " | tee -a ${LOGFILE}
981
	mkimg \
982
		-s gpt \
983
		-f raw \
984
		-b ${FINAL_CHROOT_DIR}/boot/pmbr \
985
		-p freebsd-boot:=${FINAL_CHROOT_DIR}/boot/gptboot \
986
		-p freebsd-ufs/${PRODUCT_NAME}:=${OVA_TMP}/${OVFUFS} \
987
		${OVA_SWAP_PART_PARAM} \
988
		-o ${OVA_TMP}/${OVFRAW} 2>&1 >> ${LOGFILE}
989

    
990
	if [ $? -ne 0 -o ! -f ${OVA_TMP}/${OVFRAW} ]; then
991
		if [ -f ${OVA_TMP}/${OVFUFS} ]; then
992
			rm -f ${OVA_TMP}/${OVFUFS}
993
		fi
994
		if [ -f ${OVA_TMP}/${OVFRAW} ]; then
995
			rm -f ${OVA_TMP}/${OVFRAW}
996
		fi
997
		echo "Failed!" | tee -a ${LOGFILE}
998
		echo ">>> ERROR: Error creating temporary vmdk image. STOPPING!" | tee -a ${LOGFILE}
999
		print_error_pfS
1000
	fi
1001
	echo "Done!" | tee -a ${LOGFILE}
1002

    
1003
	# We don't need it anymore
1004
	rm -f ${OVA_TMP}/${OVFUFS} >/dev/null 2>&1
1005

    
1006
	# Convert raw to vmdk
1007
	echo -n ">>> Creating vmdk disk... " | tee -a ${LOGFILE}
1008
	vmdktool -z9 -v ${OVA_TMP}/${OVFVMDK} ${OVA_TMP}/${OVFRAW}
1009

    
1010
	if [ $? -ne 0 -o ! -f ${OVA_TMP}/${OVFVMDK} ]; then
1011
		if [ -f ${OVA_TMP}/${OVFRAW} ]; then
1012
			rm -f ${OVA_TMP}/${OVFRAW}
1013
		fi
1014
		if [ -f ${OVA_TMP}/${OVFVMDK} ]; then
1015
			rm -f ${OVA_TMP}/${OVFVMDK}
1016
		fi
1017
		echo "Failed!" | tee -a ${LOGFILE}
1018
		echo ">>> ERROR: Error creating vmdk image. STOPPING!" | tee -a ${LOGFILE}
1019
		print_error_pfS
1020
	fi
1021
	echo "Done!" | tee -a ${LOGFILE}
1022

    
1023
	rm -f ${OVA_TMP}/${OVFRAW}
1024

    
1025
	ova_setup_ovf_template
1026

    
1027
	echo -n ">>> Writing final ova image... " | tee -a ${LOGFILE}
1028
	# Create OVA file for vmware
1029
	gtar -C ${OVA_TMP} -cpf ${OVAPATH} ${PRODUCT_NAME}.ovf ${OVFVMDK}
1030
	echo "Done!" | tee -a ${LOGFILE}
1031
	rm -f ${OVA_TMP}/${OVFVMDK} >/dev/null 2>&1
1032

    
1033
	echo ">>> OVA created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1034
}
1035

    
1036
# called from create_ova_image
1037
ova_setup_ovf_template() {
1038
	if [ ! -f ${OVFTEMPLATE} ]; then
1039
		echo ">>> ERROR: OVF template file (${OVFTEMPLATE}) not found."
1040
		print_error_pfS
1041
	fi
1042

    
1043
	#  OperatingSystemSection (${PRODUCT_NAME}.ovf)
1044
	#  42   FreeBSD 32-Bit
1045
	#  78   FreeBSD 64-Bit
1046
	if [ "${TARGET}" = "amd64" ]; then
1047
		local _os_id="78"
1048
		local _os_type="freebsd64Guest"
1049
		local _os_descr="FreeBSD 64-Bit"
1050
	elif [ "${TARGET}" = "i386" ]; then
1051
		local _os_id="42"
1052
		local _os_type="freebsdGuest"
1053
		local _os_descr="FreeBSD"
1054
	else
1055
		echo ">>> ERROR: Platform not supported for OVA (${TARGET})"
1056
		print_error_pfS
1057
	fi
1058

    
1059
	local POPULATED_SIZE=$(du -d0 -k $FINAL_CHROOT_DIR | cut -f1)
1060
	local POPULATED_SIZE_IN_BYTES=$((${POPULATED_SIZE}*1024))
1061
	local VMDK_FILE_SIZE=$(stat -f "%z" ${OVA_TMP}/${OVFVMDK})
1062

    
1063
	sed \
1064
		-e "s,%%VMDK_FILE_SIZE%%,${VMDK_FILE_SIZE},g" \
1065
		-e "s,%%VMDK_DISK_CAPACITY_IN_GB%%,${VMDK_DISK_CAPACITY_IN_GB},g" \
1066
		-e "s,%%POPULATED_SIZE_IN_BYTES%%,${POPULATED_SIZE_IN_BYTES},g" \
1067
		-e "s,%%OS_ID%%,${_os_id},g" \
1068
		-e "s,%%OS_TYPE%%,${_os_type},g" \
1069
		-e "s,%%OS_DESCR%%,${_os_descr},g" \
1070
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
1071
		-e "s,%%PRODUCT_NAME_SUFFIX%%,${PRODUCT_NAME_SUFFIX},g" \
1072
		-e "s,%%PRODUCT_VERSION%%,${PRODUCT_VERSION},g" \
1073
		-e "s,%%PRODUCT_URL%%,${PRODUCT_URL},g" \
1074
		-e "s#%%VENDOR_NAME%%#${VENDOR_NAME}#g" \
1075
		-e "s#%%OVF_INFO%%#${OVF_INFO}#g" \
1076
		-e "/^%%PRODUCT_LICENSE%%/r ${BUILDER_ROOT}/license.txt" \
1077
		-e "/^%%PRODUCT_LICENSE%%/d" \
1078
		${OVFTEMPLATE} > ${OVA_TMP}/${PRODUCT_NAME}.ovf
1079
}
1080

    
1081
# Cleans up previous builds
1082
clean_builder() {
1083
	# Clean out directories
1084
	echo ">>> Cleaning up previous build environment...Please wait!"
1085

    
1086
	staginareas_clean_each_run
1087

    
1088
	if [ -d "${STAGE_CHROOT_DIR}" ]; then
1089
		BASENAME=$(basename ${STAGE_CHROOT_DIR})
1090
		echo -n ">>> Cleaning ${STAGE_CHROOT_DIR} ..."
1091
		chflags -R noschg ${STAGE_CHROOT_DIR} 2>&1 >/dev/null
1092
		rm -rf ${STAGE_CHROOT_DIR}/* 2>/dev/null
1093
		echo "Done."
1094
	fi
1095

    
1096
	if [ -z "${NO_CLEAN_FREEBSD_OBJ}" -a -d "${FREEBSD_SRC_DIR}" ]; then
1097
		OBJTREE=$(env TARGET=${TARGET} TARGET_ARCH=${TARGET_ARCH} make -C ${FREEBSD_SRC_DIR} -V OBJTREE)
1098
		if [ -d "${OBJTREE}" ]; then
1099
			echo -n ">>> Cleaning FreeBSD objects dir staging..."
1100
			echo -n "."
1101
			chflags -R noschg ${OBJTREE} 2>&1 >/dev/null
1102
			echo -n "."
1103
			rm -rf ${OBJTREE}/*
1104
			echo "Done!"
1105
		fi
1106
		if [ -d "${KERNEL_BUILD_PATH}" ]; then
1107
			echo -n ">>> Cleaning previously built kernel stage area..."
1108
			rm -rf $KERNEL_BUILD_PATH/*
1109
			echo "Done!"
1110
		fi
1111
	fi
1112
	mkdir -p $KERNEL_BUILD_PATH
1113

    
1114
	echo -n ">>> Cleaning previously built images..."
1115
	rm -rf $IMAGES_FINAL_DIR/*
1116
	rm -rf $STAGINGAREA/*
1117
	echo "Done!"
1118

    
1119
	if [ -z "${NO_CLEAN_FREEBSD_SRC}" ]; then
1120
		if [ -d "$FREEBSD_SRC_DIR" ]; then
1121
			echo -n ">>> Ensuring $FREEBSD_SRC_DIR is clean..."
1122
			rm -rf ${FREEBSD_SRC_DIR}
1123
			echo "Done!"
1124
		fi
1125
	fi
1126

    
1127
	echo -n ">>> Cleaning previous builder logs..."
1128
	if [ -d "$BUILDER_LOGS" ]; then
1129
		rm -rf ${BUILDER_LOGS}
1130
	fi
1131
	mkdir -p ${BUILDER_LOGS}
1132

    
1133
	echo "Done!"
1134

    
1135
	echo ">>> Cleaning of builder environment has finished."
1136
}
1137

    
1138
clone_directory_contents() {
1139
	if [ ! -e "$2" ]; then
1140
		mkdir -p "$2"
1141
	fi
1142
	if [ ! -d "$1" -o ! -d "$2" ]; then
1143
		if [ -z "${LOGFILE}" ]; then
1144
			echo ">>> ERROR: Argument $1 supplied is not a directory!"
1145
		else
1146
			echo ">>> ERROR: Argument $1 supplied is not a directory!" | tee -a ${LOGFILE}
1147
		fi
1148
		print_error_pfS
1149
	fi
1150
	echo -n ">>> Using TAR to clone $1 to $2 ..."
1151
	tar -C ${1} -c -f - . | tar -C ${2} -x -p -f -
1152
	echo "Done!"
1153
}
1154

    
1155
clone_to_staging_area() {
1156
	# Clone everything to the final staging area
1157
	echo -n ">>> Cloning everything to ${STAGE_CHROOT_DIR} staging area..."
1158
	LOGFILE=${BUILDER_LOGS}/cloning.${TARGET}.log
1159

    
1160
	tar -C ${PRODUCT_SRC} -c -f - . | \
1161
		tar -C ${STAGE_CHROOT_DIR} -x -p -f -
1162

    
1163
	if [ "${PRODUCT_NAME}" != "pfSense" ]; then
1164
		mv ${STAGE_CHROOT_DIR}/usr/local/sbin/pfSense-upgrade \
1165
			${STAGE_CHROOT_DIR}/usr/local/sbin/${PRODUCT_NAME}-upgrade
1166
	fi
1167

    
1168
	if [ -f ${STAGE_CHROOT_DIR}/etc/master.passwd ]; then
1169
		chroot ${STAGE_CHROOT_DIR} pwd_mkdb /etc/master.passwd
1170
	fi
1171
	mkdir -p ${STAGE_CHROOT_DIR}/etc/mtree
1172
	mtree -Pcp ${STAGE_CHROOT_DIR}/var > ${STAGE_CHROOT_DIR}/etc/mtree/var.dist
1173
	mtree -Pcp ${STAGE_CHROOT_DIR}/etc > ${STAGE_CHROOT_DIR}/etc/mtree/etc.dist
1174
	if [ -d ${STAGE_CHROOT_DIR}/usr/local/etc ]; then
1175
		mtree -Pcp ${STAGE_CHROOT_DIR}/usr/local/etc > ${STAGE_CHROOT_DIR}/etc/mtree/localetc.dist
1176
	fi
1177

    
1178
	## Add buildtime and lastcommit information
1179
	# This is used for detecting updates.
1180
	echo "$BUILTDATESTRING" > $STAGE_CHROOT_DIR/etc/version.buildtime
1181
	# Record last commit info if it is available.
1182
	if [ -f $SCRATCHDIR/build_commit_info.txt ]; then
1183
		cp $SCRATCHDIR/build_commit_info.txt $STAGE_CHROOT_DIR/etc/version.lastcommit
1184
	fi
1185

    
1186
	local _exclude_files="${CORE_PKG_TMP}/base_exclude_files"
1187
	sed \
1188
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
1189
		-e "s,%%VERSION%%,${_version},g" \
1190
		${BUILDER_TOOLS}/templates/core_pkg/base/exclude_files \
1191
		> ${_exclude_files}
1192

    
1193
	mkdir -p ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR} >/dev/null 2>&1
1194

    
1195
	# Include a sample pkg stable conf to base
1196
	setup_pkg_repo \
1197
		${PKG_REPO_DEFAULT} \
1198
		${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/${PRODUCT_NAME}-repo.conf \
1199
		${TARGET} \
1200
		${TARGET_ARCH}
1201

    
1202
	mtree \
1203
		-c \
1204
		-k uid,gid,mode,size,flags,sha256digest \
1205
		-p ${STAGE_CHROOT_DIR} \
1206
		-X ${_exclude_files} \
1207
		> ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/base.mtree
1208
	tar \
1209
		-C ${STAGE_CHROOT_DIR} \
1210
		-cJf ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/base.txz \
1211
		-X ${_exclude_files} \
1212
		.
1213

    
1214
	local _share_repos_path="${SCRATCHDIR}/repo-tmp/${PRODUCT_SHARE_DIR}/pkg/repos"
1215
	rm -rf ${SCRATCHDIR}/repo-tmp >/dev/null 2>&1
1216
	mkdir -p ${_share_repos_path} >/dev/null 2>&1
1217

    
1218
	local _freebsd_major_version=$( \
1219
		sed -n '/^REVISION=/ {; s,\.[0-9]*"$,,; s,^.*",,; p; q; };' \
1220
		${FREEBSD_SRC_DIR}/sys/conf/newvers.sh \
1221
	)
1222
	local _default_abi="FreeBSD:${_freebsd_major_version}:${TARGET_ARCH}"
1223

    
1224
	local _default_altabi="freebsd:${_freebsd_major_version}"
1225
	if [ "${TARGET_ARCH}" = "armv6" ]; then
1226
		_default_altabi="${_default_altabi}:${TARGET_ARCH}:32:el:eabi:hardfp"
1227
	elif [ "${TARGET_ARCH}" = "i386" ]; then
1228
		_default_altabi="${_default_altabi}:x86:32"
1229
	else
1230
		_default_altabi="${_default_altabi}:x86:64"
1231
	fi
1232

    
1233
	# Add all repos
1234
	for _template in ${PKG_REPO_BASE}/${PRODUCT_NAME}-repo*.conf; do
1235
		_template_filename=$(basename ${_template})
1236
		setup_pkg_repo \
1237
			${_template} \
1238
			${_share_repos_path}/${_template_filename} \
1239
			${TARGET} \
1240
			${TARGET_ARCH}
1241

    
1242
		cp -f ${_template%%.conf}.descr ${_share_repos_path}
1243

    
1244
		if [ -f ${_template%%.conf}.abi ]; then
1245
			sed -e "s,%%ARCH%%,${TARGET_ARCH},g" ${_template%%.conf}.abi \
1246
				> ${_share_repos_path}/${_template_filename%%.conf}.abi
1247
		else
1248
			echo ${_default_abi} \
1249
				> ${_share_repos_path}/${_template_filename%%.conf}.abi
1250
		fi
1251

    
1252
		if [ -f ${_template%%.conf}.altabi ]; then
1253
			sed -e "s,%%ARCH%%,${TARGET_ARCH},g" ${_template%%.conf}.altabi \
1254
				> ${_share_repos_path}/${_template_filename%%.conf}.altabi
1255
		else
1256
			echo ${_default_altabi} \
1257
				> ${_share_repos_path}/${_template_filename%%.conf}.altabi
1258
		fi
1259
	done
1260

    
1261
	core_pkg_create repo "" ${CORE_PKG_VERSION} ${SCRATCHDIR}/repo-tmp
1262

    
1263
	core_pkg_create rc "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1264
	core_pkg_create base "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1265
	core_pkg_create base-nanobsd "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1266
	core_pkg_create default-config "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1267

    
1268
	local DEFAULTCONF=${STAGE_CHROOT_DIR}/conf.default/config.xml
1269

    
1270
	# Save current WAN and LAN if value
1271
	local _old_wan_if=$(xml sel -t -v "${XML_ROOTOBJ}/interfaces/wan/if" ${DEFAULTCONF})
1272
	local _old_lan_if=$(xml sel -t -v "${XML_ROOTOBJ}/interfaces/lan/if" ${DEFAULTCONF})
1273

    
1274
	# Change default interface names to match vmware driver
1275
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/wan/if" -v "vmx0" ${DEFAULTCONF}
1276
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/lan/if" -v "vmx1" ${DEFAULTCONF}
1277
	core_pkg_create default-config "vmware" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1278

    
1279
	# Restore default values to be used by serial package
1280
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/wan/if" -v "${_old_wan_if}" ${DEFAULTCONF}
1281
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/lan/if" -v "${_old_lan_if}" ${DEFAULTCONF}
1282

    
1283
	# Activate serial console in config.xml
1284
	xml ed -L -P -d "${XML_ROOTOBJ}/system/enableserial" ${DEFAULTCONF}
1285
	xml ed -P -s "${XML_ROOTOBJ}/system" -t elem -n "enableserial" \
1286
		${DEFAULTCONF} > ${DEFAULTCONF}.tmp
1287
	xml fo -t ${DEFAULTCONF}.tmp > ${DEFAULTCONF}
1288
	rm -f ${DEFAULTCONF}.tmp
1289

    
1290
	echo force > ${STAGE_CHROOT_DIR}/cf/conf/enableserial_force
1291

    
1292
	core_pkg_create default-config-serial "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1293

    
1294
	rm -f ${STAGE_CHROOT_DIR}/cf/conf/enableserial_force
1295
	rm -f ${STAGE_CHROOT_DIR}/cf/conf/config.xml
1296

    
1297
	# Make sure pkg is present
1298
	pkg_bootstrap ${STAGE_CHROOT_DIR}
1299

    
1300
	echo "Done!"
1301
}
1302

    
1303
create_final_staging_area() {
1304
	if [ -z "${FINAL_CHROOT_DIR}" ]; then
1305
		echo ">>> ERROR: FINAL_CHROOT_DIR is not set, cannot continue!" | tee -a ${LOGFILE}
1306
		print_error_pfS
1307
	fi
1308

    
1309
	if [ -d "${FINAL_CHROOT_DIR}" ]; then
1310
		echo -n ">>> Previous ${FINAL_CHROOT_DIR} detected cleaning up..." | tee -a ${LOGFILE}
1311
		chflags -R noschg ${FINAL_CHROOT_DIR} 2>&1 1>/dev/null
1312
		rm -rf ${FINAL_CHROOT_DIR}/* 2>&1 1>/dev/null
1313
		echo "Done." | tee -a ${LOGFILE}
1314
	fi
1315

    
1316
	echo ">>> Preparing Final image staging area: $(LC_ALL=C date)" 2>&1 | tee -a ${LOGFILE}
1317
	echo ">>> Cloning ${STAGE_CHROOT_DIR} to ${FINAL_CHROOT_DIR}" 2>&1 | tee -a ${LOGFILE}
1318
	clone_directory_contents ${STAGE_CHROOT_DIR} ${FINAL_CHROOT_DIR}
1319

    
1320
	if [ ! -f $FINAL_CHROOT_DIR/sbin/init ]; then
1321
		echo ">>> ERROR: Something went wrong during cloning -- Please verify!" 2>&1 | tee -a ${LOGFILE}
1322
		print_error_pfS
1323
	fi
1324
}
1325

    
1326
customize_stagearea_for_image() {
1327
	local _image_type="$1"
1328
	local _default_config="" # filled with $2 below
1329
	local _image_variant="$3"
1330

    
1331
	if [ -n "$2" ]; then
1332
		_default_config="$2"
1333
	elif [ "${_image_type}" = "nanobsd" -o \
1334
	     "${_image_type}" = "memstickserial" -o \
1335
	     "${_image_type}" = "memstickadi" ]; then
1336
		_default_config="default-config-serial"
1337
	elif [ "${_image_type}" = "ova" ]; then
1338
		_default_config="default-config-vmware"
1339
	else
1340
		_default_config="default-config"
1341
	fi
1342

    
1343
	# Prepare final stage area
1344
	create_final_staging_area
1345

    
1346
	pkg_chroot_add ${FINAL_CHROOT_DIR} rc
1347
	pkg_chroot_add ${FINAL_CHROOT_DIR} repo
1348

    
1349
	if [ "${_image_type}" = "nanobsd" -o \
1350
	     "${_image_type}" = "nanobsd-vga" ]; then
1351

    
1352
		mkdir -p ${FINAL_CHROOT_DIR}/root/var/db \
1353
			 ${FINAL_CHROOT_DIR}/root/var/cache \
1354
			 ${FINAL_CHROOT_DIR}/var/db/pkg \
1355
			 ${FINAL_CHROOT_DIR}/var/cache/pkg
1356
		mv -f ${FINAL_CHROOT_DIR}/var/db/pkg ${FINAL_CHROOT_DIR}/root/var/db
1357
		mv -f ${FINAL_CHROOT_DIR}/var/cache/pkg ${FINAL_CHROOT_DIR}/root/var/cache
1358
		ln -sf ../../root/var/db/pkg ${FINAL_CHROOT_DIR}/var/db/pkg
1359
		ln -sf ../../root/var/cache/pkg ${FINAL_CHROOT_DIR}/var/cache/pkg
1360

    
1361
		pkg_chroot_add ${FINAL_CHROOT_DIR} base-nanobsd
1362
	else
1363
		pkg_chroot_add ${FINAL_CHROOT_DIR} base
1364
	fi
1365

    
1366
	if [ "${_image_type}" = "iso" -o \
1367
	     "${_image_type}" = "memstick" -o \
1368
	     "${_image_type}" = "memstickserial" -o \
1369
	     "${_image_type}" = "memstickadi" ]; then
1370
		install_bsdinstaller
1371
		mkdir -p ${FINAL_CHROOT_DIR}/pkgs
1372
		cp ${CORE_PKG_ALL_PATH}/*default-config*.txz ${FINAL_CHROOT_DIR}/pkgs
1373
	fi
1374

    
1375
	pkg_chroot_add ${FINAL_CHROOT_DIR} ${_default_config}
1376

    
1377
	# XXX: Workaround to avoid pkg to complain regarding release
1378
	#      repo on first boot since packages are installed from
1379
	#      staging server during build phase
1380
	if [ -n "${USE_PKG_REPO_STAGING}" ]; then
1381
		_read_cmd="select value from repodata where key='packagesite'"
1382
		if [ -n "${_IS_RELEASE}" ]; then
1383
			local _tgt_server="${PKG_REPO_SERVER_RELEASE}"
1384
		else
1385
			local _tgt_server="${PKG_REPO_SERVER_DEVEL}"
1386
		fi
1387
		for _db in ${FINAL_CHROOT_DIR}/var/db/pkg/repo-*sqlite; do
1388
			_cur=$(/usr/local/bin/sqlite3 ${_db} "${_read_cmd}")
1389
			_new=$(echo "${_cur}" | sed -e "s,^${PKG_REPO_SERVER_STAGING},${_tgt_server},")
1390
			/usr/local/bin/sqlite3 ${_db} "update repodata set value='${_new}' where key='packagesite'"
1391
		done
1392
	fi
1393

    
1394
	if [ -n "$_image_variant" -a \
1395
	    -d ${BUILDER_TOOLS}/templates/custom_logos/${_image_variant} ]; then
1396
		mkdir -p ${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/custom_logos
1397
		cp -f \
1398
			${BUILDER_TOOLS}/templates/custom_logos/${_image_variant}/*.png \
1399
			${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/custom_logos
1400
	fi
1401
}
1402

    
1403
create_distribution_tarball() {
1404
	mkdir -p ${FINAL_CHROOT_DIR}/distrib
1405

    
1406
	tar -C ${FINAL_CHROOT_DIR} --exclude ./distrib --exclude ./pkgs -cJf ${FINAL_CHROOT_DIR}/distrib/${PRODUCT_NAME}.txz .
1407
}
1408

    
1409
create_iso_image() {
1410
	LOGFILE=${BUILDER_LOGS}/isoimage.${TARGET}
1411
	echo ">>> Building bootable ISO image for ${TARGET}" | tee -a ${LOGFILE}
1412
	if [ -z "${DEFAULT_KERNEL}" ]; then
1413
		echo ">>> ERROR: Could not identify DEFAULT_KERNEL to install on image!" | tee -a ${LOGFILE}
1414
		print_error_pfS
1415
	fi
1416

    
1417
	customize_stagearea_for_image "iso"
1418
	install_default_kernel ${DEFAULT_KERNEL}
1419

    
1420
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1421

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

    
1425
	# This check is for supporting create memstick/ova images
1426
	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}
1427
	echo "-o label=${FSLABEL} -o publisher=\"${PRODUCT_NAME} project.\" $ISOPATH ${FINAL_CHROOT_DIR}" | tee -a ${LOGFILE}
1428

    
1429
	create_distribution_tarball
1430

    
1431
	# Remove /rescue from iso since cd9660 cannot deal with hardlinks
1432
	rm -rf ${FINAL_CHROOT_DIR}/rescue
1433

    
1434
	makefs -t cd9660 -o bootimage="i386;${FINAL_CHROOT_DIR}/boot/cdboot" -o no-emul-boot -o rockridge \
1435
		-o label=${FSLABEL} -o publisher="${PRODUCT_NAME} project." $ISOPATH ${FINAL_CHROOT_DIR} 2>&1 >> ${LOGFILE}
1436
	if [ $? -ne 0 -o ! -f $ISOPATH ]; then
1437
		if [ -f ${ISOPATH} ]; then
1438
			rm -f $ISOPATH
1439
		fi
1440
		echo ">>> ERROR: Something wrong happened during ISO image creation. STOPPING!" | tee -a ${LOGFILE}
1441
		print_error_pfS
1442
	fi
1443
	gzip -qf $ISOPATH &
1444
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1445

    
1446
	echo ">>> ISO created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1447
}
1448

    
1449
create_memstick_image() {
1450
	local _variant="$1"
1451

    
1452
	LOGFILE=${BUILDER_LOGS}/memstick.${TARGET}
1453
	if [ "${MEMSTICKPATH}" = "" ]; then
1454
		echo ">>> MEMSTICKPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1455
		return
1456
	fi
1457

    
1458
	local _image_path=${MEMSTICKPATH}
1459
	if [ -n "${_variant}" ]; then
1460
		_image_path=$(echo "$_image_path" | \
1461
			sed "s/-memstick-/-memstick-${_variant}-/")
1462
		VARIANTIMAGES="${VARIANTIMAGES}${VARIANTIMAGES:+ }${_image_path}"
1463
	fi
1464

    
1465
	customize_stagearea_for_image "memstick" "" $_variant
1466
	install_default_kernel ${DEFAULT_KERNEL}
1467

    
1468
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1469

    
1470
	echo ">>> Creating memstick to ${_image_path}." 2>&1 | tee -a ${LOGFILE}
1471
	echo "/dev/ufs/${PRODUCT_NAME} / ufs ro 0 0" > ${FINAL_CHROOT_DIR}/etc/fstab
1472
	echo "kern.cam.boot_delay=10000" >> ${FINAL_CHROOT_DIR}/boot/loader.conf.local
1473

    
1474
	create_distribution_tarball
1475

    
1476
	makefs -B little -o label=${PRODUCT_NAME},version=2 ${_image_path} ${FINAL_CHROOT_DIR}
1477
	if [ $? -ne 0 ]; then
1478
		if [ -f ${_image_path} ]; then
1479
			rm -f $_image_path
1480
		fi
1481
		echo ">>> ERROR: Something wrong happened during MEMSTICK image creation. STOPPING!" | tee -a ${LOGFILE}
1482
		print_error_pfS
1483
	fi
1484
	MD=$(mdconfig -a -t vnode -f $_image_path)
1485
	# Just in case
1486
	trap "mdconfig -d -u ${MD}" 1 2 15 EXIT
1487
	gpart create -s BSD ${MD} 2>&1 >> ${LOGFILE}
1488
	gpart bootcode -b ${FINAL_CHROOT_DIR}/boot/boot ${MD} 2>&1 >> ${LOGFILE}
1489
	gpart add -t freebsd-ufs ${MD} 2>&1 >> ${LOGFILE}
1490
	trap "-" 1 2 15 EXIT
1491
	mdconfig -d -u ${MD} 2>&1 | tee -a ${LOGFILE}
1492
	gzip -qf $_image_path &
1493
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1494

    
1495
	echo ">>> MEMSTICK created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1496
}
1497

    
1498
create_memstick_serial_image() {
1499
	LOGFILE=${BUILDER_LOGS}/memstickserial.${TARGET}
1500
	if [ "${MEMSTICKSERIALPATH}" = "" ]; then
1501
		echo ">>> MEMSTICKSERIALPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1502
		return
1503
	fi
1504

    
1505
	customize_stagearea_for_image "memstickserial"
1506
	install_default_kernel ${DEFAULT_KERNEL}
1507

    
1508
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1509

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

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

    
1515
	BOOTCONF=${FINAL_CHROOT_DIR}/boot.config
1516
	LOADERCONF=${FINAL_CHROOT_DIR}/boot/loader.conf
1517

    
1518
	echo ">>> Activating serial console..." 2>&1 | tee -a ${LOGFILE}
1519
	# Activate serial console in boot.config
1520
	if [ -f ${BOOTCONF} ]; then
1521
		sed -i "" '/-D$/d' ${BOOTCONF}
1522
	fi
1523
	echo "-S115200 -D" >> ${BOOTCONF}
1524

    
1525
	# Remove old console options if present.
1526
	[ -f "${LOADERCONF}" ] \
1527
		&& sed -i "" -Ee "/(console|boot_multicons|boot_serial)/d" ${LOADERCONF}
1528
	# Activate serial console+video console in loader.conf
1529
	echo 'boot_multicons="YES"' >>  ${LOADERCONF}
1530
	echo 'boot_serial="YES"' >> ${LOADERCONF}
1531
	echo 'console="comconsole,vidconsole"' >> ${LOADERCONF}
1532
	echo 'comconsole_speed="115200"' >> ${LOADERCONF}
1533

    
1534
	create_distribution_tarball
1535

    
1536
	makefs -B little -o label=${PRODUCT_NAME},version=2 ${MEMSTICKSERIALPATH} ${FINAL_CHROOT_DIR}
1537
	if [ $? -ne 0 ]; then
1538
		if [ -f ${MEMSTICKSERIALPATH} ]; then
1539
			rm -f $MEMSTICKSERIALPATH
1540
		fi
1541
		echo ">>> ERROR: Something wrong happened during MEMSTICKSERIAL image creation. STOPPING!" | tee -a ${LOGFILE}
1542
		print_error_pfS
1543
	fi
1544
	MD=$(mdconfig -a -t vnode -f $MEMSTICKSERIALPATH)
1545
	# Just in case
1546
	trap "mdconfig -d -u ${MD}" 1 2 15 EXIT
1547
	gpart create -s BSD ${MD} 2>&1 >> ${LOGFILE}
1548
	gpart bootcode -b ${FINAL_CHROOT_DIR}/boot/boot ${MD} 2>&1 >> ${LOGFILE}
1549
	gpart add -t freebsd-ufs ${MD} 2>&1 >> ${LOGFILE}
1550
	trap "-" 1 2 15 EXIT
1551
	mdconfig -d -u ${MD} 2>&1 >> ${LOGFILE}
1552
	gzip -qf $MEMSTICKSERIALPATH &
1553
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1554

    
1555
	echo ">>> MEMSTICKSERIAL created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1556
}
1557

    
1558
create_memstick_adi_image() {
1559
	LOGFILE=${BUILDER_LOGS}/memstickadi.${TARGET}
1560
	if [ "${MEMSTICKADIPATH}" = "" ]; then
1561
		echo ">>> MEMSTICKADIPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1562
		return
1563
	fi
1564

    
1565
	customize_stagearea_for_image "memstickadi"
1566
	install_default_kernel ${DEFAULT_KERNEL}
1567

    
1568
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1569

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

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

    
1575
	BOOTCONF=${FINAL_CHROOT_DIR}/boot.config
1576
	LOADERCONF=${FINAL_CHROOT_DIR}/boot/loader.conf
1577

    
1578
	echo ">>> Activating serial console..." 2>&1 | tee -a ${LOGFILE}
1579
	# Activate serial console in boot.config
1580
	if [ -f ${BOOTCONF} ]; then
1581
		sed -i "" '/-[Dh]$/d' ${BOOTCONF}
1582
	fi
1583
	echo "-S115200 -h" >> ${BOOTCONF}
1584

    
1585
	# Remove old console options if present.
1586
	[ -f "${LOADERCONF}" ] \
1587
		&& sed -i "" -Ee "/(console|boot_multicons|boot_serial|hint.uart)/d" ${LOADERCONF}
1588
	# Activate serial console+video console in loader.conf
1589
	echo 'boot_serial="YES"' >> ${LOADERCONF}
1590
	echo 'console="comconsole"' >> ${LOADERCONF}
1591
	echo 'comconsole_speed="115200"' >> ${LOADERCONF}
1592
	echo 'comconsole_port="0x2F8"' >> ${LOADERCONF}
1593
	echo 'hint.uart.0.flags="0x00"' >> ${LOADERCONF}
1594
	echo 'hint.uart.1.flags="0x10"' >> ${LOADERCONF}
1595

    
1596
	create_distribution_tarball
1597

    
1598
	makefs -B little -o label=${PRODUCT_NAME},version=2 ${MEMSTICKADIPATH} ${FINAL_CHROOT_DIR}
1599
	if [ $? -ne 0 ]; then
1600
		if [ -f ${MEMSTICKADIPATH} ]; then
1601
			rm -f $MEMSTICKADIPATH
1602
		fi
1603
		echo ">>> ERROR: Something wrong happened during MEMSTICKADI image creation. STOPPING!" | tee -a ${LOGFILE}
1604
		print_error_pfS
1605
	fi
1606
	MD=$(mdconfig -a -t vnode -f $MEMSTICKADIPATH)
1607
	# Just in case
1608
	trap "mdconfig -d -u ${MD}" 1 2 15 EXIT
1609
	gpart create -s BSD ${MD} 2>&1 >> ${LOGFILE}
1610
	gpart bootcode -b ${FINAL_CHROOT_DIR}/boot/boot ${MD} 2>&1 >> ${LOGFILE}
1611
	gpart add -t freebsd-ufs ${MD} 2>&1 >> ${LOGFILE}
1612
	trap "-" 1 2 15 EXIT
1613
	mdconfig -d -u ${MD} 2>&1 >> ${LOGFILE}
1614
	gzip -qf $MEMSTICKADIPATH &
1615
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1616

    
1617
	echo ">>> MEMSTICKADI created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1618
}
1619

    
1620
# Create pkg conf on desired place with desired arch/branch
1621
setup_pkg_repo() {
1622
	if [ -z "${4}" ]; then
1623
		return
1624
	fi
1625

    
1626
	local _template="${1}"
1627
	local _target="${2}"
1628
	local _arch="${3}"
1629
	local _target_arch="${4}"
1630
	local _staging="${5}"
1631

    
1632
	if [ -z "${_template}" -o ! -f "${_template}" ]; then
1633
		echo ">>> ERROR: It was not possible to find pkg conf template ${_template}"
1634
		print_error_pfS
1635
	fi
1636

    
1637
	if [ -n "${_staging}" -a -n "${USE_PKG_REPO_STAGING}" ]; then
1638
		local _pkg_repo_server_devel=${PKG_REPO_SERVER_STAGING}
1639
		local _pkg_repo_branch_devel=${PKG_REPO_BRANCH_STAGING}
1640
		local _pkg_repo_server_release=${PKG_REPO_SERVER_STAGING}
1641
		local _pkg_repo_branch_release=${PKG_REPO_BRANCH_STAGING}
1642
	else
1643
		local _pkg_repo_server_devel=${PKG_REPO_SERVER_DEVEL}
1644
		local _pkg_repo_branch_devel=${PKG_REPO_BRANCH_DEVEL}
1645
		local _pkg_repo_server_release=${PKG_REPO_SERVER_RELEASE}
1646
		local _pkg_repo_branch_release=${PKG_REPO_BRANCH_RELEASE}
1647
	fi
1648

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

    
1651
	sed \
1652
		-e "s/%%ARCH%%/${_target_arch}/" \
1653
		-e "s/%%PKG_REPO_BRANCH_DEVEL%%/${_pkg_repo_branch_devel}/g" \
1654
		-e "s/%%PKG_REPO_BRANCH_RELEASE%%/${_pkg_repo_branch_release}/g" \
1655
		-e "s,%%PKG_REPO_SERVER_DEVEL%%,${_pkg_repo_server_devel},g" \
1656
		-e "s,%%PKG_REPO_SERVER_RELEASE%%,${_pkg_repo_server_release},g" \
1657
		-e "s/%%PRODUCT_NAME%%/${PRODUCT_NAME}/g" \
1658
		${_template} \
1659
		> ${_target}
1660
}
1661

    
1662
# This routine ensures any ports / binaries that the builder
1663
# system needs are on disk and ready for execution.
1664
builder_setup() {
1665
	# If Product-builder is already installed, just leave
1666
	if pkg info -e -q ${PRODUCT_NAME}-builder; then
1667
		return
1668
	fi
1669

    
1670
	if [ ! -f ${PKG_REPO_PATH} ]; then
1671
		[ -d $(dirname ${PKG_REPO_PATH}) ] \
1672
			|| mkdir -p $(dirname ${PKG_REPO_PATH})
1673

    
1674
		update_freebsd_sources
1675

    
1676
		local _arch=$(uname -m)
1677
		setup_pkg_repo \
1678
			${PKG_REPO_DEFAULT} \
1679
			${PKG_REPO_PATH} \
1680
			${_arch} \
1681
			${_arch} \
1682
			"staging"
1683

    
1684
		# Use fingerprint keys from repo
1685
		sed -i '' -e "/fingerprints:/ s,\"/,\"${BUILDER_ROOT}/src/," \
1686
			${PKG_REPO_PATH}
1687
	fi
1688

    
1689
	pkg install ${PRODUCT_NAME}-builder
1690
}
1691

    
1692
# Updates FreeBSD sources
1693
update_freebsd_sources() {
1694
	if [ "${1}" = "full" ]; then
1695
		local _full=1
1696
		local _clone_params=""
1697
	else
1698
		local _full=0
1699
		local _clone_params="--depth 1 --single-branch"
1700
	fi
1701

    
1702
	if [ ! -d "${FREEBSD_SRC_DIR}" ]; then
1703
		mkdir -p ${FREEBSD_SRC_DIR}
1704
	fi
1705

    
1706
	if [ -n "${NO_BUILDWORLD}" -a -n "${NO_BUILDKERNEL}" ]; then
1707
		echo ">>> NO_BUILDWORLD and NO_BUILDKERNEL set, skipping update of freebsd sources" | tee -a ${LOGFILE}
1708
		return
1709
	fi
1710

    
1711
	echo -n ">>> Obtaining FreeBSD sources ${FREEBSD_BRANCH}..."
1712
	local _FREEBSD_BRANCH=${FREEBSD_BRANCH:-"devel"}
1713
	local _CLONE=1
1714

    
1715
	if [ -d "${FREEBSD_SRC_DIR}/.git" ]; then
1716
		CUR_BRANCH=$(cd ${FREEBSD_SRC_DIR} && git branch | grep '^\*' | cut -d' ' -f2)
1717
		if [ ${_full} -eq 0 -a "${CUR_BRANCH}" = "${_FREEBSD_BRANCH}" ]; then
1718
			_CLONE=0
1719
			( cd ${FREEBSD_SRC_DIR} && git clean -fd; git fetch origin; git reset --hard origin/${_FREEBSD_BRANCH} ) 2>&1 | grep -C3 -i -E 'error|fatal'
1720
		else
1721
			rm -rf ${FREEBSD_SRC_DIR}
1722
		fi
1723
	fi
1724

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

    
1729
	if [ ! -d "${FREEBSD_SRC_DIR}/.git" ]; then
1730
		echo ">>> ERROR: It was not possible to clone FreeBSD src repo"
1731
		print_error_pfS
1732
	fi
1733

    
1734
	if [ -n "${GIT_FREEBSD_COSHA1}" ]; then
1735
		( cd ${FREEBSD_SRC_DIR} && git checkout ${GIT_FREEBSD_COSHA1} ) 2>&1 | grep -C3 -i -E 'error|fatal'
1736
	fi
1737
	echo "Done!"
1738
}
1739

    
1740
pkg_chroot() {
1741
	local _root="${1}"
1742
	shift
1743

    
1744
	if [ $# -eq 0 ]; then
1745
		return -1
1746
	fi
1747

    
1748
	if [ -z "${_root}" -o "${_root}" = "/" -o ! -d "${_root}" ]; then
1749
		return -1
1750
	fi
1751

    
1752
	mkdir -p \
1753
		${SCRATCHDIR}/pkg_cache \
1754
		${_root}/var/cache/pkg \
1755
		${_root}/dev
1756

    
1757
	/sbin/mount -t nullfs ${SCRATCHDIR}/pkg_cache ${_root}/var/cache/pkg
1758
	/sbin/mount -t devfs devfs ${_root}/dev
1759
	cp -f /etc/resolv.conf ${_root}/etc/resolv.conf
1760
	touch ${BUILDER_LOGS}/install_pkg_install_ports.txt
1761
	script -aq ${BUILDER_LOGS}/install_pkg_install_ports.txt pkg -c ${_root} $@ >/dev/null 2>&1
1762
	local result=$?
1763
	rm -f ${_root}/etc/resolv.conf
1764
	/sbin/umount -f ${_root}/dev
1765
	/sbin/umount -f ${_root}/var/cache/pkg
1766

    
1767
	return $result
1768
}
1769

    
1770

    
1771
pkg_chroot_add() {
1772
	if [ -z "${1}" -o -z "${2}" ]; then
1773
		return 1
1774
	fi
1775

    
1776
	local _target="${1}"
1777
	local _pkg="$(get_pkg_name ${2}).txz"
1778

    
1779
	if [ ! -d "${_target}" ]; then
1780
		echo ">>> ERROR: Target dir ${_target} not found"
1781
		print_error_pfS
1782
	fi
1783

    
1784
	if [ ! -f ${CORE_PKG_ALL_PATH}/${_pkg} ]; then
1785
		echo ">>> ERROR: Package ${_pkg} not found"
1786
		print_error_pfS
1787
	fi
1788

    
1789
	cp ${CORE_PKG_ALL_PATH}/${_pkg} ${_target}
1790
	pkg_chroot ${_target} add /${_pkg}
1791
	rm -f ${_target}/${_pkg}
1792
}
1793

    
1794
pkg_bootstrap() {
1795
	local _root=${1:-"${STAGE_CHROOT_DIR}"}
1796

    
1797
	setup_pkg_repo \
1798
		${PKG_REPO_DEFAULT} \
1799
		${_root}${PKG_REPO_PATH} \
1800
		${TARGET} \
1801
		${TARGET_ARCH} \
1802
		"staging"
1803

    
1804
	pkg_chroot ${_root} bootstrap -f
1805
}
1806

    
1807
# This routine assists with installing various
1808
# freebsd ports files into the pfsense-fs staging
1809
# area.
1810
install_pkg_install_ports() {
1811
	local MAIN_PKG="${1}"
1812

    
1813
	if [ -z "${MAIN_PKG}" ]; then
1814
		MAIN_PKG=${PRODUCT_NAME}
1815
	fi
1816

    
1817
	echo ">>> Installing pkg repository in chroot (${STAGE_CHROOT_DIR})..."
1818

    
1819
	[ -d ${STAGE_CHROOT_DIR}/var/cache/pkg ] || \
1820
		mkdir -p ${STAGE_CHROOT_DIR}/var/cache/pkg
1821

    
1822
	[ -d ${SCRATCHDIR}/pkg_cache ] || \
1823
		mkdir -p ${SCRATCHDIR}/pkg_cache
1824

    
1825
	echo -n ">>> Installing built ports (packages) in chroot (${STAGE_CHROOT_DIR})... "
1826
	# First mark all packages as automatically installed
1827
	pkg_chroot ${STAGE_CHROOT_DIR} set -A 1 -a
1828
	# Install all necessary packages
1829
	if ! pkg_chroot ${STAGE_CHROOT_DIR} install ${MAIN_PKG} ${custom_package_list}; then
1830
		echo "Failed!"
1831
		print_error_pfS
1832
	fi
1833
	# Make sure required packages are set as non-automatic
1834
	pkg_chroot ${STAGE_CHROOT_DIR} set -A 0 pkg ${MAIN_PKG} ${custom_package_list}
1835
	# Remove unnecessary packages
1836
	pkg_chroot ${STAGE_CHROOT_DIR} autoremove
1837
	echo "Done!"
1838
}
1839

    
1840
install_bsdinstaller() {
1841
	local _params=""
1842

    
1843
	# Use staging repo on RELEASE
1844
	if [ -n "${_IS_RELEASE}" ]; then
1845
		mkdir -p ${FINAL_CHROOT_DIR}/tmp/pkg-repo
1846
		cp -f ${STAGE_CHROOT_DIR}${PKG_REPO_PATH} \
1847
			${FINAL_CHROOT_DIR}/tmp/pkg-repo
1848
		_params="--repo-conf-dir /tmp/pkg-repo "
1849
	fi
1850

    
1851
	echo ">>> Installing BSDInstaller in chroot (${FINAL_CHROOT_DIR})... (starting)"
1852
	pkg_chroot ${FINAL_CHROOT_DIR} ${_params}install -f bsdinstaller
1853
	sed -i '' -e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME}," \
1854
		  -e "s,%%PRODUCT_VERSION%%,${PRODUCT_VERSION}," \
1855
		  -e "s,%%ARCH%%,${TARGET}," \
1856
		  ${FINAL_CHROOT_DIR}/usr/local/share/dfuibe_lua/conf/pfSense.lua \
1857
		  ${FINAL_CHROOT_DIR}/usr/local/share/dfuibe_lua/conf/pfSense_rescue.lua
1858
	if [ -n "${_IS_RELEASE}" ]; then
1859
		rm -rf ${FINAL_CHROOT_DIR}/tmp/pkg-repo
1860
	fi
1861
	echo ">>> Installing BSDInstaller in chroot (${FINAL_CHROOT_DIR})... (finished)"
1862
}
1863

    
1864
staginareas_clean_each_run() {
1865
	echo -n ">>> Cleaning build directories: "
1866
	if [ -d "${FINAL_CHROOT_DIR}" ]; then
1867
		BASENAME=$(basename ${FINAL_CHROOT_DIR})
1868
		echo -n "$BASENAME "
1869
		chflags -R noschg ${FINAL_CHROOT_DIR} 2>&1 >/dev/null
1870
		rm -rf ${FINAL_CHROOT_DIR}/* 2>/dev/null
1871
	fi
1872
	echo "Done!"
1873
}
1874

    
1875
# Imported from FreeSBIE
1876
buildkernel() {
1877
	if [ -n "${NO_BUILDKERNEL}" ]; then
1878
		echo ">>> NO_BUILDKERNEL set, skipping build" | tee -a ${LOGFILE}
1879
		return
1880
	fi
1881

    
1882
	if [ -z "${KERNCONF}" ]; then
1883
		echo ">>> ERROR: No kernel configuration defined probably this is not what you want! STOPPING!" | tee -a ${LOGFILE}
1884
		print_error_pfS
1885
	fi
1886

    
1887
	if [ -n "${KERNELCONF}" ]; then
1888
		export KERNCONFDIR=$(dirname ${KERNELCONF})
1889
		export KERNCONF=$(basename ${KERNELCONF})
1890
	fi
1891

    
1892
	SRCCONFBASENAME=$(basename ${SRC_CONF})
1893
	echo ">>> KERNCONFDIR: ${KERNCONFDIR}"
1894
	echo ">>> ARCH:        ${TARGET}"
1895
	echo ">>> SRC_CONF:    ${SRCCONFBASENAME}"
1896

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

    
1902
# Imported from FreeSBIE
1903
installkernel() {
1904
	if [ -z "${KERNCONF}" ]; then
1905
		echo ">>> ERROR: No kernel configuration defined probably this is not what you want! STOPPING!" | tee -a ${LOGFILE}
1906
		print_error_pfS
1907
	fi
1908

    
1909
	if [ -n "${KERNELCONF}" ]; then
1910
		export KERNCONFDIR=$(dirname ${KERNELCONF})
1911
		export KERNCONF=$(basename ${KERNELCONF})
1912
	fi
1913

    
1914
	mkdir -p ${STAGE_CHROOT_DIR}/boot
1915
	makeargs="${MAKEJ_KERNEL} SRCCONF=${SRC_CONF} __MAKE_CONF=${MAKE_CONF} TARGET_ARCH=${TARGET_ARCH} TARGET=${TARGET} DESTDIR=${KERNEL_DESTDIR}"
1916
	echo ">>> Builder is running the command: script -aq $LOGFILE make ${makeargs} installkernel KERNCONF=${KERNCONF}"  | tee -a $LOGFILE
1917
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} installkernel KERNCONF=${KERNCONF} || print_error_pfS;) | egrep '^>>>'
1918
	gzip -f9 $KERNEL_DESTDIR/boot/kernel/kernel
1919
}
1920

    
1921
# Launch is ran first to setup a few variables that we need
1922
# Imported from FreeSBIE
1923
launch() {
1924
	if [ "$(id -u)" != "0" ]; then
1925
		echo "Sorry, this must be done as root."
1926
	fi
1927

    
1928
	echo ">>> Operation $0 has started at $(date)"
1929
}
1930

    
1931
finish() {
1932
	echo ">>> Operation $0 has ended at $(date)"
1933
}
1934

    
1935
pkg_repo_rsync() {
1936
	local _repo_path_param="${1}"
1937
	local _ignore_final_rsync="${2}"
1938

    
1939
	if [ -z "${_repo_path_param}" -o ! -d "${_repo_path_param}" ]; then
1940
		return
1941
	fi
1942

    
1943
	if [ -n "${SKIP_FINAL_RSYNC}" ]; then
1944
		_ignore_final_rsync="1"
1945
	fi
1946

    
1947
	# Sanitize path
1948
	_repo_path=$(realpath ${_repo_path_param})
1949

    
1950
	local _repo_dir=$(dirname ${_repo_path})
1951
	local _repo_base=$(basename ${_repo_path})
1952

    
1953
	# Add ./ it's an rsync trick to make it chdir to directory before sending it
1954
	_repo_path="${_repo_dir}/./${_repo_base}"
1955

    
1956
	if [ -z "${LOGFILE}" ]; then
1957
		local _logfile="/dev/null"
1958
	else
1959
		local _logfile="${LOGFILE}"
1960
	fi
1961

    
1962
	if [ -n "${PKG_REPO_SIGNING_COMMAND}" -a -z "${DO_NOT_SIGN_PKG_REPO}" ]; then
1963
		# Detect poudriere directory structure
1964
		if [ -L "${_repo_path}/.latest" ]; then
1965
			local _real_repo_path=$(readlink -f ${_repo_path}/.latest)
1966
		else
1967
			local _real_repo_path=${_repo_path}
1968
		fi
1969

    
1970
		echo -n ">>> Signing repository... " | tee -a ${_logfile}
1971
		############ ATTENTION ##############
1972
		#
1973
		# For some reason pkg-repo fail without / in the end of directory name
1974
		# so removing it will break command
1975
		#
1976
		# https://github.com/freebsd/pkg/issues/1364
1977
		#
1978
		if script -aq ${_logfile} pkg repo ${_real_repo_path}/ \
1979
		    signing_command: ${PKG_REPO_SIGNING_COMMAND} >/dev/null 2>&1; then
1980
			echo "Done!" | tee -a ${_logfile}
1981
		else
1982
			echo "Failed!" | tee -a ${_logfile}
1983
			echo ">>> ERROR: An error occurred trying to sign repo"
1984
			print_error_pfS
1985
		fi
1986

    
1987
		local _pkgfile="${_repo_path}/Latest/pkg.txz"
1988
		if [ -e ${_pkgfile} ]; then
1989
			echo -n ">>> Signing Latest/pkg.txz for bootstraping... " | tee -a ${_logfile}
1990

    
1991
			if sha256 -q ${_pkgfile} | ${PKG_REPO_SIGNING_COMMAND} \
1992
			    > ${_pkgfile}.sig 2>/dev/null; then
1993
				echo "Done!" | tee -a ${_logfile}
1994
			else
1995
				echo "Failed!" | tee -a ${_logfile}
1996
				echo ">>> ERROR: An error occurred trying to sign Latest/pkg.txz"
1997
				print_error_pfS
1998
			fi
1999
		fi
2000
	fi
2001

    
2002
	if [ -n "${DO_NOT_UPLOAD}" ]; then
2003
		return
2004
	fi
2005

    
2006
	for _pkg_rsync_hostname in ${PKG_RSYNC_HOSTNAME}; do
2007
		# Make sure destination directory exist
2008
		ssh -p ${PKG_RSYNC_SSH_PORT} \
2009
			${PKG_RSYNC_USERNAME}@${_pkg_rsync_hostname} \
2010
			"mkdir -p ${PKG_RSYNC_DESTDIR}"
2011

    
2012
		echo -n ">>> Sending updated repository to ${_pkg_rsync_hostname}... " | tee -a ${_logfile}
2013
		if script -aq ${_logfile} rsync -Have "ssh -p ${PKG_RSYNC_SSH_PORT}" \
2014
			--timeout=60 --delete-delay ${_repo_path} \
2015
			${PKG_RSYNC_USERNAME}@${_pkg_rsync_hostname}:${PKG_RSYNC_DESTDIR} >/dev/null 2>&1
2016
		then
2017
			echo "Done!" | tee -a ${_logfile}
2018
		else
2019
			echo "Failed!" | tee -a ${_logfile}
2020
			echo ">>> ERROR: An error occurred sending repo to remote hostname"
2021
			print_error_pfS
2022
		fi
2023

    
2024
		if [ -z "${USE_PKG_REPO_STAGING}" -o -n "${_ignore_final_rsync}" ]; then
2025
			return
2026
		fi
2027

    
2028
		if [ -n "${_IS_RELEASE}" -o "${_repo_path_param}" = "${CORE_PKG_PATH}" ]; then
2029
			for _pkg_final_rsync_hostname in ${PKG_FINAL_RSYNC_HOSTNAME}; do
2030
				# Send .real* directories first to prevent having a broken repo while transfer happens
2031
				local _cmd="rsync -Have \"ssh -p ${PKG_FINAL_RSYNC_SSH_PORT}\" \
2032
					--timeout=60 ${PKG_RSYNC_DESTDIR}/./${_repo_base%%-core}* \
2033
					--include=\"/*\" --include=\"*/.real*\" --include=\"*/.real*/***\" \
2034
					--exclude=\"*\" \
2035
					${PKG_FINAL_RSYNC_USERNAME}@${_pkg_final_rsync_hostname}:${PKG_FINAL_RSYNC_DESTDIR}"
2036

    
2037
				echo -n ">>> Sending updated packages to ${_pkg_final_rsync_hostname}... " | tee -a ${_logfile}
2038
				if script -aq ${_logfile} ssh -p ${PKG_RSYNC_SSH_PORT} \
2039
					${PKG_RSYNC_USERNAME}@${_pkg_rsync_hostname} ${_cmd} >/dev/null 2>&1; then
2040
					echo "Done!" | tee -a ${_logfile}
2041
				else
2042
					echo "Failed!" | tee -a ${_logfile}
2043
					echo ">>> ERROR: An error occurred sending repo to final hostname"
2044
					print_error_pfS
2045
				fi
2046

    
2047
				_cmd="rsync -Have \"ssh -p ${PKG_FINAL_RSYNC_SSH_PORT}\" \
2048
					--timeout=60 --delete-delay ${PKG_RSYNC_DESTDIR}/./${_repo_base%%-core}* \
2049
					${PKG_FINAL_RSYNC_USERNAME}@${_pkg_final_rsync_hostname}:${PKG_FINAL_RSYNC_DESTDIR}"
2050

    
2051
				echo -n ">>> Sending updated repositories metadata to ${_pkg_final_rsync_hostname}... " | tee -a ${_logfile}
2052
				if script -aq ${_logfile} ssh -p ${PKG_RSYNC_SSH_PORT} \
2053
					${PKG_RSYNC_USERNAME}@${_pkg_rsync_hostname} ${_cmd} >/dev/null 2>&1; then
2054
					echo "Done!" | tee -a ${_logfile}
2055
				else
2056
					echo "Failed!" | tee -a ${_logfile}
2057
					echo ">>> ERROR: An error occurred sending repo to final hostname"
2058
					print_error_pfS
2059
				fi
2060
			done
2061
		fi
2062
	done
2063
}
2064

    
2065
poudriere_create_patch() {
2066
	local _jail_patch="${SCRATCHDIR}/poudriere_jail.${POUDRIERE_BRANCH}.patch"
2067

    
2068
	if [ -z "${FREEBSD_PARENT_BRANCH}" ]; then
2069
		echo ">>> ERROR: FREEBSD_PARENT_BRANCH is not set"
2070
	fi
2071

    
2072
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2073

    
2074
	# Get FreeBSD source and apply patches
2075
	update_freebsd_sources full
2076

    
2077
	[ -f "${_jail_patch}" ] && \
2078
		rm -f "${_jail_patch}"
2079

    
2080
	# Create a big patch with all our changes to use on jail
2081
	( \
2082
		cd ${FREEBSD_SRC_DIR} && \
2083
		git diff $(git merge-base origin/${FREEBSD_PARENT_BRANCH} ${FREEBSD_BRANCH}) > ${_jail_patch}
2084
	) >/dev/null 2>&1
2085

    
2086
	# Check if patch was created
2087
	if [ ! -s "${_jail_patch}" ]; then
2088
		echo ">>> ERROR: Patch does not exist or is empty, aborting..." | tee -a ${LOGFILE}
2089
		print_error_pfS
2090
	fi
2091
}
2092

    
2093
poudriere_possible_archs() {
2094
	local _arch=$(uname -m)
2095
	local _archs="i386.i386"
2096

    
2097
	# If host is amd64, we'll create both repos, and if possible armv6
2098
	if [ "${_arch}" = "amd64" ]; then
2099
		_archs="amd64.amd64 ${_archs}"
2100

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

    
2105
			if binmiscctl lookup armv6 >/dev/null 2>&1; then
2106
				_archs="${_archs} arm.armv6"
2107
			fi
2108
		fi
2109
	fi
2110

    
2111
	if [ -n "${ARCH_LIST}" ]; then
2112
		local _found=0
2113
		for _desired_arch in ${ARCH_LIST}; do
2114
			_found=0
2115
			for _possible_arch in ${_archs}; do
2116
				if [ "${_desired_arch}" = "${_possible_arch}" ]; then
2117
					_found=1
2118
					break
2119
				fi
2120
			done
2121
			if [ ${_found} -eq 0 ]; then
2122
				echo ">>> ERROR: Impossible to build for arch: ${_desired_arch}"
2123
				print_error_pfS
2124
			fi
2125
		done
2126
		_archs="${ARCH_LIST}"
2127
	fi
2128

    
2129
	echo ${_archs}
2130
}
2131

    
2132
poudriere_jail_name() {
2133
	local _jail_arch="${1}"
2134

    
2135
	if [ -z "${_jail_arch}" ]; then
2136
		return 1
2137
	fi
2138

    
2139
	# Remove arch
2140
	echo "${PRODUCT_NAME}_${POUDRIERE_BRANCH}_${_jail_arch##*.}"
2141
}
2142

    
2143
poudriere_rename_ports() {
2144
	if [ "${PRODUCT_NAME}" = "pfSense" ]; then
2145
		return;
2146
	fi
2147

    
2148
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2149

    
2150
	local _ports_dir="/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}"
2151

    
2152
	echo -n ">>> Renaming product ports on ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
2153
	for d in $(find ${_ports_dir} -depth 2 -type d -name '*pfSense*'); do
2154
		local _pdir=$(dirname ${d})
2155
		local _pname=$(echo $(basename ${d}) | sed "s,pfSense,${PRODUCT_NAME},")
2156
		local _plist=""
2157

    
2158
		if [ -e ${_pdir}/${_pname} ]; then
2159
			rm -rf ${_pdir}/${_pname}
2160
		fi
2161

    
2162
		cp -r ${d} ${_pdir}/${_pname}
2163

    
2164
		if [ -f ${_pdir}/${_pname}/pkg-plist ]; then
2165
			_plist=${_pdir}/${_pname}/pkg-plist
2166
		fi
2167

    
2168
		sed -i '' -e "s,pfSense,${PRODUCT_NAME},g" \
2169
			  -e "s,https://www.pfsense.org,${PRODUCT_URL},g" \
2170
			  -e "/^MAINTAINER=/ s,^.*$,MAINTAINER=	${PRODUCT_EMAIL}," \
2171
			${_pdir}/${_pname}/Makefile \
2172
			${_pdir}/${_pname}/pkg-descr ${_plist}
2173

    
2174
		# PHP module is special
2175
		if echo "${_pname}" | grep -q "^php[0-9]*-${PRODUCT_NAME}-module"; then
2176
			local _product_capital=$(echo ${PRODUCT_NAME} | tr '[a-z]' '[A-Z]')
2177
			sed -i '' -e "s,PHP_PFSENSE,PHP_${_product_capital},g" \
2178
				  -e "s,PFSENSE_SHARED_LIBADD,${_product_capital}_SHARED_LIBADD,g" \
2179
				  -e "s,pfSense,${PRODUCT_NAME},g" \
2180
				  -e "s,${PRODUCT_NAME}\.c,pfSense.c,g" \
2181
				${_pdir}/${_pname}/files/config.m4
2182

    
2183
			sed -i '' -e "s,COMPILE_DL_PFSENSE,COMPILE_DL_${_product_capital}," \
2184
				  -e "s,pfSense_module_entry,${PRODUCT_NAME}_module_entry,g" \
2185
				  -e "/ZEND_GET_MODULE/ s,pfSense,${PRODUCT_NAME}," \
2186
				  -e "/PHP_PFSENSE_WORLD_EXTNAME/ s,pfSense,${PRODUCT_NAME}," \
2187
				${_pdir}/${_pname}/files/pfSense.c \
2188
				${_pdir}/${_pname}/files/php_pfSense.h
2189
		fi
2190

    
2191
		if [ -d ${_pdir}/${_pname}/files ]; then
2192
			for fd in $(find ${_pdir}/${_pname}/files -type d -name '*pfSense*'); do
2193
				local _fddir=$(dirname ${fd})
2194
				local _fdname=$(echo $(basename ${fd}) | sed "s,pfSense,${PRODUCT_NAME},")
2195

    
2196
				mv ${fd} ${_fddir}/${_fdname}
2197
			done
2198
		fi
2199
	done
2200
	echo "Done!" | tee -a ${LOGFILE}
2201
}
2202

    
2203
poudriere_create_ports_tree() {
2204
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2205

    
2206
	if ! poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
2207
		local _branch=""
2208
		if [ -z "${POUDRIERE_PORTS_GIT_URL}" ]; then
2209
			echo ">>> ERROR: POUDRIERE_PORTS_GIT_URL is not defined"
2210
			print_error_pfS
2211
		fi
2212
		if [ -n "${POUDRIERE_PORTS_GIT_BRANCH}" ]; then
2213
			_branch="-B ${POUDRIERE_PORTS_GIT_BRANCH}"
2214
		fi
2215
		echo -n ">>> Creating poudriere ports tree, it may take some time... " | tee -a ${LOGFILE}
2216
		if ! script -aq ${LOGFILE} poudriere ports -c -p "${POUDRIERE_PORTS_NAME}" -m git ${_branch} >/dev/null 2>&1; then
2217
			echo "" | tee -a ${LOGFILE}
2218
			echo ">>> ERROR: Error creating poudriere ports tree, aborting..." | tee -a ${LOGFILE}
2219
			print_error_pfS
2220
		fi
2221
		echo "Done!" | tee -a ${LOGFILE}
2222
		poudriere_rename_ports
2223
	fi
2224
}
2225

    
2226
poudriere_init() {
2227
	local _error=0
2228
	local _archs=$(poudriere_possible_archs)
2229
	local _jail_patch="${SCRATCHDIR}/poudriere_jail.${POUDRIERE_BRANCH}.patch"
2230

    
2231
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2232

    
2233
	# Sanity checks
2234
	if [ -z "${ZFS_TANK}" ]; then
2235
		echo ">>> ERROR: \$ZFS_TANK is empty" | tee -a ${LOGFILE}
2236
		error=1
2237
	fi
2238

    
2239
	if [ -z "${ZFS_ROOT}" ]; then
2240
		echo ">>> ERROR: \$ZFS_ROOT is empty" | tee -a ${LOGFILE}
2241
		error=1
2242
	fi
2243

    
2244
	if [ -z "${POUDRIERE_PORTS_NAME}" ]; then
2245
		echo ">>> ERROR: \$POUDRIERE_PORTS_NAME is empty" | tee -a ${LOGFILE}
2246
		error=1
2247
	fi
2248

    
2249
	if [ ${_error} -eq 1 ]; then
2250
		print_error_pfS
2251
	fi
2252

    
2253
	# Check if zpool exists
2254
	if ! zpool list ${ZFS_TANK} >/dev/null 2>&1; then
2255
		echo ">>> ERROR: ZFS tank ${ZFS_TANK} not found, please create it and try again..." | tee -a ${LOGFILE}
2256
		print_error_pfS
2257
	fi
2258

    
2259
	# Check if zfs rootfs exists
2260
	if ! zfs list ${ZFS_TANK}${ZFS_ROOT} >/dev/null 2>&1; then
2261
		echo -n ">>> Creating ZFS filesystem ${ZFS_TANK}${ZFS_ROOT}... "
2262
		if zfs create -o atime=off -o mountpoint=/usr/local${ZFS_ROOT} \
2263
		    ${ZFS_TANK}${ZFS_ROOT} >/dev/null 2>&1; then
2264
			echo "Done!"
2265
		else
2266
			echo "Failed!"
2267
			print_error_pfS
2268
		fi
2269
	fi
2270

    
2271
	# Make sure poudriere is installed
2272
	if ! pkg info --quiet poudriere; then
2273
		echo ">>> Installing poudriere..." | tee -a ${LOGFILE}
2274
		if ! pkg install poudriere >/dev/null 2>&1; then
2275
			echo ">>> ERROR: poudriere was not installed, aborting..." | tee -a ${LOGFILE}
2276
			print_error_pfS
2277
		fi
2278
	fi
2279

    
2280
	# Create poudriere.conf
2281
	if [ -z "${POUDRIERE_PORTS_GIT_URL}" ]; then
2282
		echo ">>> ERROR: POUDRIERE_PORTS_GIT_URL is not defined"
2283
		print_error_pfS
2284
	fi
2285
	echo ">>> Creating poudriere.conf" | tee -a ${LOGFILE}
2286
	cat <<EOF >/usr/local/etc/poudriere.conf
2287
ZPOOL=${ZFS_TANK}
2288
ZROOTFS=${ZFS_ROOT}
2289
RESOLV_CONF=/etc/resolv.conf
2290
BASEFS=/usr/local/poudriere
2291
USE_PORTLINT=no
2292
USE_TMPFS=yes
2293
NOLINUX=yes
2294
DISTFILES_CACHE=/usr/ports/distfiles
2295
CHECK_CHANGED_OPTIONS=yes
2296
CHECK_CHANGED_DEPS=yes
2297
ATOMIC_PACKAGE_REPOSITORY=yes
2298
COMMIT_PACKAGES_ON_FAILURE=no
2299
KEEP_OLD_PACKAGES=yes
2300
KEEP_OLD_PACKAGES_COUNT=5
2301
EOF
2302

    
2303
	# Create specific items conf
2304
	[ ! -d /usr/local/etc/poudriere.d ] \
2305
		&& mkdir -p /usr/local/etc/poudriere.d
2306

    
2307
	cat <<EOF >/usr/local/etc/poudriere.d/${POUDRIERE_PORTS_NAME}-poudriere.conf
2308
GIT_URL="${POUDRIERE_PORTS_GIT_URL}"
2309
EOF
2310

    
2311
	# Create DISTFILES_CACHE if it doesn't exist
2312
	if [ ! -d /usr/ports/distfiles ]; then
2313
		mkdir -p /usr/ports/distfiles
2314
	fi
2315

    
2316
	# Remove old jails
2317
	for jail_arch in ${_archs}; do
2318
		jail_name=$(poudriere_jail_name ${jail_arch})
2319

    
2320
		if poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
2321
			echo ">>> Poudriere jail ${jail_name} already exists, deleting it..." | tee -a ${LOGFILE}
2322
			poudriere jail -d -j "${jail_name}" >/dev/null 2>&1
2323
		fi
2324
	done
2325

    
2326
	# Remove old ports tree
2327
	if poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
2328
		echo ">>> Poudriere ports tree ${POUDRIERE_PORTS_NAME} already exists, deleting it..." | tee -a ${LOGFILE}
2329
		poudriere ports -d -p "${POUDRIERE_PORTS_NAME}"
2330
	fi
2331

    
2332
	poudriere_create_patch
2333

    
2334
	local native_xtools=""
2335
	# Now we are ready to create jails
2336
	for jail_arch in ${_archs}; do
2337
		jail_name=$(poudriere_jail_name ${jail_arch})
2338

    
2339
		if [ "${jail_arch}" = "arm.armv6" ]; then
2340
			native_xtools="-x"
2341
		else
2342
			native_xtools=""
2343
		fi
2344

    
2345
		echo -n ">>> Creating jail ${jail_name}, it may take some time... " | tee -a ${LOGFILE}
2346
		# XXX: Change -m to git when it's available in poudriere
2347
		if ! script -aq ${LOGFILE} poudriere jail -c -j "${jail_name}" -v ${FREEBSD_PARENT_BRANCH} \
2348
				-a ${jail_arch} -m svn -P ${_jail_patch} ${native_xtools} >/dev/null 2>&1; then
2349
			echo "" | tee -a ${LOGFILE}
2350
			echo ">>> ERROR: Error creating jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
2351
			print_error_pfS
2352
		fi
2353
		echo "Done!" | tee -a ${LOGFILE}
2354
	done
2355

    
2356
	poudriere_create_ports_tree
2357

    
2358
	echo ">>> Poudriere is now configured!" | tee -a ${LOGFILE}
2359
}
2360

    
2361
poudriere_update_jails() {
2362
	local _archs=$(poudriere_possible_archs)
2363
	local _jail_patch="${SCRATCHDIR}/poudriere_jail.${POUDRIERE_BRANCH}.patch"
2364

    
2365
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2366

    
2367
	poudriere_create_patch
2368

    
2369
	local native_xtools=""
2370
	for jail_arch in ${_archs}; do
2371
		jail_name=$(poudriere_jail_name ${jail_arch})
2372

    
2373
		local _create_or_update="-u"
2374
		local _create_or_update_text="Updating"
2375
		if ! poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
2376
			echo ">>> Poudriere jail ${jail_name} not found, creating..." | tee -a ${LOGFILE}
2377
			_create_or_update="-c -v ${FREEBSD_PARENT_BRANCH} -a ${jail_arch} -m svn"
2378
			_create_or_update_text="Creating"
2379
		fi
2380

    
2381
		if [ "${jail_arch}" = "arm.armv6" ]; then
2382
			native_xtools="-x"
2383
		else
2384
			native_xtools=""
2385
		fi
2386

    
2387
		echo -n ">>> ${_create_or_update_text} jail ${jail_name}, it may take some time... " | tee -a ${LOGFILE}
2388
		if ! script -aq ${LOGFILE} poudriere jail ${_create_or_update} -j "${jail_name}" -P ${_jail_patch} ${native_xtools} >/dev/null 2>&1; then
2389
			echo "" | tee -a ${LOGFILE}
2390
			echo ">>> ERROR: Error ${_create_or_update_text} jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
2391
			print_error_pfS
2392
		fi
2393
		echo "Done!" | tee -a ${LOGFILE}
2394
	done
2395
}
2396

    
2397
poudriere_update_ports() {
2398
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2399

    
2400
	# Create ports tree if necessary
2401
	if ! poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
2402
		poudriere_create_ports_tree
2403
	else
2404
		echo -n ">>> Resetting local changes on ports tree ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
2405
		script -aq ${LOGFILE} git -C "/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}" reset --hard >/dev/null 2>&1
2406
		script -aq ${LOGFILE} git -C "/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}" clean -fd >/dev/null 2>&1
2407
		echo "Done!" | tee -a ${LOGFILE}
2408
		echo -n ">>> Updating ports tree ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
2409
		script -aq ${LOGFILE} poudriere ports -u -p "${POUDRIERE_PORTS_NAME}" >/dev/null 2>&1
2410
		echo "Done!" | tee -a ${LOGFILE}
2411
		poudriere_rename_ports
2412
	fi
2413
}
2414

    
2415
poudriere_bulk() {
2416
	local _archs=$(poudriere_possible_archs)
2417

    
2418
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2419

    
2420
	if [ -z "${DO_NOT_UPLOAD}" -a -z "${PKG_RSYNC_HOSTNAME}" ]; then
2421
		echo ">>> ERROR: PKG_RSYNC_HOSTNAME is not set"
2422
		print_error_pfS
2423
	fi
2424

    
2425
	rm -f ${LOGFILE}
2426

    
2427
	poudriere_create_ports_tree
2428

    
2429
	[ -d /usr/local/etc/poudriere.d ] || \
2430
		mkdir -p /usr/local/etc/poudriere.d
2431

    
2432
	if [ -f "${BUILDER_TOOLS}/conf/pfPorts/make.conf" ]; then
2433
		cp -f "${BUILDER_TOOLS}/conf/pfPorts/make.conf" /usr/local/etc/poudriere.d/${POUDRIERE_PORTS_NAME}-make.conf
2434
	fi
2435

    
2436
	# Change version of pfSense meta ports for snapshots
2437
	if [ -z "${_IS_RELEASE}" ]; then
2438
		local _meta_pkg_version="$(echo "${PRODUCT_VERSION}" | sed 's,DEVELOPMENT,ALPHA,')-${DATESTRING}"
2439
		sed -i '' \
2440
			-e "/^DISTVERSION/ s,^.*,DISTVERSION=	${_meta_pkg_version}," \
2441
			-e "/^PORTREVISION=/d" \
2442
			/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}/security/${PRODUCT_NAME}/Makefile
2443
	fi
2444

    
2445
	for jail_arch in ${_archs}; do
2446
		jail_name=$(poudriere_jail_name ${jail_arch})
2447

    
2448
		if ! poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
2449
			echo ">>> Poudriere jail ${jail_name} not found, skipping..." | tee -a ${LOGFILE}
2450
			continue
2451
		fi
2452

    
2453
		if [ -f "${POUDRIERE_BULK}.${jail_arch}" ]; then
2454
			_ref_bulk="${POUDRIERE_BULK}.${jail_arch}"
2455
		else
2456
			_ref_bulk="${POUDRIERE_BULK}"
2457
		fi
2458

    
2459
		_bulk=${SCRATCHDIR}/poudriere_bulk.${POUDRIERE_BRANCH}
2460
		sed -e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" ${_ref_bulk} > ${_bulk}
2461

    
2462
		if ! poudriere bulk -f ${_bulk} -j ${jail_name} -p ${POUDRIERE_PORTS_NAME}; then
2463
			echo ">>> ERROR: Something went wrong..."
2464
			print_error_pfS
2465
		fi
2466

    
2467
		echo ">>> Cleaning up old packages from repo..."
2468
		if ! poudriere pkgclean -f ${_bulk} -j ${jail_name} -p ${POUDRIERE_PORTS_NAME} -y; then
2469
			echo ">>> ERROR: Something went wrong..."
2470
			print_error_pfS
2471
		fi
2472

    
2473
		pkg_repo_rsync "/usr/local/poudriere/data/packages/${jail_name}-${POUDRIERE_PORTS_NAME}"
2474
	done
2475
}
2476

    
2477
# This routine is called to write out to stdout
2478
# a string. The string is appended to $SNAPSHOTSLOGFILE
2479
snapshots_update_status() {
2480
	if [ -z "$1" ]; then
2481
		return
2482
	fi
2483
	if [ -z "${SNAPSHOTS}" -a -z "${POUDRIERE_SNAPSHOTS}" ]; then
2484
		return
2485
	fi
2486
	echo "$*"
2487
	echo "`date` -|- $*" >> $SNAPSHOTSLOGFILE
2488
}
2489

    
2490
create_sha256() {
2491
	local _file="${1}"
2492

    
2493
	if [ ! -f "${_file}" ]; then
2494
		return 1
2495
	fi
2496

    
2497
	( \
2498
		cd $(dirname ${_file}) && \
2499
		sha256 $(basename ${_file}) > $(basename ${_file}).sha256 \
2500
	)
2501
}
2502

    
2503
snapshots_create_latest_symlink() {
2504
	local _image="${1}"
2505

    
2506
	if [ -z "${_image}" ]; then
2507
		return
2508
	fi
2509

    
2510
	if [ -z "${TIMESTAMP_SUFFIX}" ]; then
2511
		return
2512
	fi
2513

    
2514
	if [ ! -f "${_image}" ]; then
2515
		return
2516
	fi
2517

    
2518
	local _symlink=$(echo ${_image} | sed "s,${TIMESTAMP_SUFFIX},-latest,")
2519
	ln -sf $(basename ${_image}) ${_symlink}
2520
	ln -sf $(basename ${_image}).sha256 ${_symlink}.sha256
2521
}
2522

    
2523
snapshots_copy_to_staging_nanobsd() {
2524
	for NANOTYPE in nanobsd nanobsd-vga; do
2525
		for FILESIZE in ${1}; do
2526
			FILENAMEFULL="$(nanobsd_image_filename ${FILESIZE} ${NANOTYPE}).gz"
2527
			FILENAMEUPGRADE="$(nanobsd_image_filename ${FILESIZE} ${NANOTYPE} 1).gz"
2528
			mkdir -p $STAGINGAREA/nanobsd
2529
			mkdir -p $STAGINGAREA/nanobsdupdates
2530

    
2531
			cp -l $IMAGES_FINAL_DIR/$FILENAMEFULL $STAGINGAREA/nanobsd/ 2>/dev/null
2532
			cp -l $IMAGES_FINAL_DIR/$FILENAMEUPGRADE $STAGINGAREA/nanobsdupdates 2>/dev/null
2533

    
2534
			if [ -f $STAGINGAREA/nanobsd/$FILENAMEFULL ]; then
2535
				create_sha256 $STAGINGAREA/nanobsd/$FILENAMEFULL
2536
			fi
2537
			if [ -f $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE ]; then
2538
				create_sha256 $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE
2539
			fi
2540

    
2541
			# Copy NanoBSD auto update:
2542
			if [ -f $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE ]; then
2543
				cp -l $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE $STAGINGAREA/latest-${NANOTYPE}-$FILESIZE.img.gz 2>/dev/null
2544
				create_sha256 $STAGINGAREAA/latest-${NANOTYPE}-$FILESIZE.img.gz
2545
				# NOTE: Updates need a file with output similar to date output
2546
				# Use the file generated at start of snapshots_dobuilds() to be consistent on times
2547
				cp $BUILTDATESTRINGFILE $STAGINGAREA/version-${NANOTYPE}-$FILESIZE
2548
			fi
2549
		done
2550
	done
2551
}
2552

    
2553
snapshots_copy_to_staging_iso_updates() {
2554
	local _img=""
2555

    
2556
	for _img in ${ISOPATH} ${MEMSTICKPATH} ${MEMSTICKSERIALPATH} ${MEMSTICKADIPATH} ${VARIANTIMAGES}; do
2557
		if [ ! -f "${_img}.gz" ]; then
2558
			continue
2559
		fi
2560
		_img="${_img}.gz"
2561
		create_sha256 ${_img}
2562
		cp -l ${_img}* $STAGINGAREA/ 2>/dev/null
2563
		snapshots_create_latest_symlink ${STAGINGAREA}/$(basename ${_img})
2564
	done
2565

    
2566
	for _img in ${UPDATES_TARBALL_FILENAME} ${VARIANTUPDATES}; do
2567
		if [ -f "${_img}" ]; then
2568
			create_sha256 ${_img}
2569
			cp -l ${_img}* $STAGINGAREA/ 2>/dev/null
2570
			snapshots_create_latest_symlink ${STAGINGAREA}/$(basename ${_img})
2571
		fi
2572
	done
2573

    
2574
	if [ -f "${OVAPATH}" ]; then
2575
		mkdir -p ${STAGINGAREA}/virtualization
2576
		create_sha256 ${OVAPATH}
2577
		cp -l ${OVAPATH}* $STAGINGAREA/virtualization 2>/dev/null
2578
		snapshots_create_latest_symlink ${STAGINGAREA}/virtualization/$(basename ${OVAPATH})
2579
	fi
2580

    
2581
	# NOTE: Updates need a file with output similar to date output
2582
	# Use the file generated at start of snapshots_dobuilds() to be consistent on times
2583
	if [ -z "${_IS_RELEASE}" ]; then
2584
		cp $BUILTDATESTRINGFILE $STAGINGAREA/version 2>/dev/null
2585
	fi
2586
}
2587

    
2588
snapshots_scp_files() {
2589
	if [ -z "${RSYNC_COPY_ARGUMENTS}" ]; then
2590
		RSYNC_COPY_ARGUMENTS="-ave ssh --timeout=60"
2591
	fi
2592

    
2593
	snapshots_update_status ">>> Copying core pkg repo to ${PKG_RSYNC_HOSTNAME}"
2594
	pkg_repo_rsync "${CORE_PKG_PATH}"
2595
	snapshots_update_status ">>> Finished copying core pkg repo"
2596

    
2597
	for _rsyncip in ${RSYNCIP}; do
2598
		snapshots_update_status ">>> Copying files to ${_rsyncip}"
2599

    
2600
		# Ensure directory(s) are available
2601
		ssh ${RSYNCUSER}@${_rsyncip} "mkdir -p ${RSYNCPATH}/installer"
2602
		ssh ${RSYNCUSER}@${_rsyncip} "mkdir -p ${RSYNCPATH}/updates"
2603
		ssh ${RSYNCUSER}@${_rsyncip} "mkdir -p ${RSYNCPATH}/nanobsd"
2604
		if [ -d $STAGINGAREA/virtualization ]; then
2605
			ssh ${RSYNCUSER}@${_rsyncip} "mkdir -p ${RSYNCPATH}/virtualization"
2606
		fi
2607
		ssh ${RSYNCUSER}@${_rsyncip} "mkdir -p ${RSYNCPATH}/.updaters"
2608
		# ensure permissions are correct for r+w
2609
		ssh ${RSYNCUSER}@${_rsyncip} "chmod -R ug+rw ${RSYNCPATH}/."
2610
		rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/${PRODUCT_NAME}${PRODUCT_NAME_SUFFIX}-*iso* \
2611
			${RSYNCUSER}@${_rsyncip}:${RSYNCPATH}/installer/
2612
		rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/${PRODUCT_NAME}${PRODUCT_NAME_SUFFIX}-memstick* \
2613
			${RSYNCUSER}@${_rsyncip}:${RSYNCPATH}/installer/
2614
		rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/${PRODUCT_NAME}${PRODUCT_NAME_SUFFIX}-*Update* \
2615
			${RSYNCUSER}@${_rsyncip}:${RSYNCPATH}/updates/
2616
		rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/nanobsd/* \
2617
			${RSYNCUSER}@${_rsyncip}:${RSYNCPATH}/nanobsd/
2618
		rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/nanobsdupdates/* \
2619
			${RSYNCUSER}@${_rsyncip}:${RSYNCPATH}/updates/
2620
		if [ -d $STAGINGAREA/virtualization ]; then
2621
			rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/virtualization/* \
2622
				${RSYNCUSER}@${_rsyncip}:${RSYNCPATH}/virtualization/
2623
		fi
2624

    
2625
		# Rather than copy these twice, use ln to link to the latest one.
2626

    
2627
		ssh ${RSYNCUSER}@${_rsyncip} "rm -f ${RSYNCPATH}/.updaters/latest.tgz"
2628
		ssh ${RSYNCUSER}@${_rsyncip} "rm -f ${RSYNCPATH}/.updaters/latest.tgz.sha256"
2629

    
2630
		LATESTFILENAME=$(basename ${UPDATES_TARBALL_FILENAME})
2631
		ssh ${RSYNCUSER}@${_rsyncip} "ln -s ${RSYNCPATH}/updates/${LATESTFILENAME} \
2632
			${RSYNCPATH}/.updaters/latest.tgz"
2633
		ssh ${RSYNCUSER}@${_rsyncip} "ln -s ${RSYNCPATH}/updates/${LATESTFILENAME}.sha256 \
2634
			${RSYNCPATH}/.updaters/latest.tgz.sha256"
2635

    
2636
		for i in ${FLASH_SIZE}
2637
		do
2638
			ssh ${RSYNCUSER}@${_rsyncip} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz"
2639
			ssh ${RSYNCUSER}@${_rsyncip} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz.sha256"
2640
			ssh ${RSYNCUSER}@${_rsyncip} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz"
2641
			ssh ${RSYNCUSER}@${_rsyncip} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz.sha256"
2642

    
2643
			FILENAMEUPGRADE="$(nanobsd_image_filename ${i} nanobsd 1).gz"
2644
			ssh ${RSYNCUSER}@${_rsyncip} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE} \
2645
				${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz"
2646
			ssh ${RSYNCUSER}@${_rsyncip} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE}.sha256 \
2647
				${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz.sha256"
2648

    
2649
			FILENAMEUPGRADE="$(nanobsd_image_filename ${i} nanobsd-vga 1).gz"
2650
			ssh ${RSYNCUSER}@${_rsyncip} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE} \
2651
				${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz"
2652
			ssh ${RSYNCUSER}@${_rsyncip} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE}.sha256 \
2653
				${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz.sha256"
2654
		done
2655

    
2656
		rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/version* \
2657
			${RSYNCUSER}@${_rsyncip}:${RSYNCPATH}/.updaters
2658
		snapshots_update_status ">>> Finished copying files."
2659
	done
2660
}
(2-2/3)