Project

General

Profile

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

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

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

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

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

    
70
	############ ATTENTION ##############
71
	#
72
	# For some reason pkg-repo fail without / in the end of directory name
73
	# so removing it will break command
74
	#
75
	# https://github.com/freebsd/pkg/issues/1364
76
	#
77
	echo -n ">>> Creating core packages repository... "
78
	if pkg repo -q "${CORE_PKG_REAL_PATH}/"; then
79
		echo "Done!"
80
	else
81
		echo "Failed!"
82
		print_error_pfS
83
	fi
84

    
85
	# Use the same directory structure as poudriere does to avoid
86
	# breaking snapshot repositories during rsync
87
	ln -sf $(basename ${CORE_PKG_REAL_PATH}) ${CORE_PKG_PATH}/.latest
88
	ln -sf .latest/All ${CORE_PKG_ALL_PATH}
89
	ln -sf .latest/digests.txz ${CORE_PKG_PATH}/digests.txz
90
	ln -sf .latest/meta.txz ${CORE_PKG_PATH}/meta.txz
91
	ln -sf .latest/packagesite.txz ${CORE_PKG_PATH}/packagesite.txz
92
}
93

    
94
# Create core pkg (base, kernel)
95
core_pkg_create() {
96
	local _template="${1}"
97
	local _flavor="${2}"
98
	local _version="${3}"
99
	local _root="${4}"
100
	local _filter="${5}"
101

    
102
	[ -d "${CORE_PKG_TMP}" ] \
103
		&& rm -rf ${CORE_PKG_TMP}
104

    
105
	local _templates_path=${BUILDER_TOOLS}/templates/core_pkg/${_template}
106
	local _template_metadir=${_templates_path}/metadir
107
	local _metadir=${CORE_PKG_TMP}/${_template}_metadir
108

    
109
	if [ ! -d ${_template_metadir} ]; then
110
		echo "ERROR: Template dir not found for pkg ${_template}"
111
		exit
112
	fi
113

    
114
	mkdir -p ${CORE_PKG_TMP}
115

    
116
	cp -r ${_template_metadir} ${_metadir}
117

    
118
	local _manifest=${_metadir}/+MANIFEST
119
	local _plist=${CORE_PKG_TMP}/${_template}_plist
120
	local _exclude_plist=${CORE_PKG_TMP}/${_template}_exclude_plist
121

    
122
	if [ -f "${_templates_path}/pkg-plist" ]; then
123
		cp ${_templates_path}/pkg-plist ${_plist}
124
	else
125
		if [ -n "${_filter}" ]; then
126
			_filter="-name ${_filter}"
127
		fi
128
		(cd ${_root} && find . ${_filter} -type f -or -type l | sed 's,^.,,' | sort -u) > ${_plist}
129
	fi
130

    
131
	if [ -f "${_templates_path}/exclude_plist" ]; then
132
		cp ${_templates_path}/exclude_plist ${_exclude_plist}
133
	else
134
		touch ${_exclude_plist}
135
	fi
136

    
137
	sed \
138
		-i '' \
139
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
140
		-e "s,%%PRODUCT_URL%%,${PRODUCT_URL},g" \
141
		-e "s,%%FLAVOR%%,${_flavor:+-}${_flavor},g" \
142
		-e "s,%%FLAVOR_DESC%%,${_flavor:+ (${_flavor})},g" \
143
		-e "s,%%VERSION%%,${_version},g" \
144
		${_metadir}/* \
145
		${_plist} \
146
		${exclude_plist}
147

    
148
	if [ -f "${_exclude_plist}" ]; then
149
		sort -u ${_exclude_plist} > ${_plist}.exclude
150
		mv ${_plist} ${_plist}.tmp
151
		comm -23 ${_plist}.tmp ${_plist}.exclude > ${_plist}
152
		rm -f ${_plist}.tmp ${plist}.exclude
153
	fi
154

    
155
	# Add license information
156
	local _portname=$(sed '/^name: /!d; s,^[^"]*",,; s,",,' ${_metadir}/+MANIFEST)
157
	local _licenses_dir="/usr/local/share/licenses/${_portname}-${_version}"
158
	mkdir -p ${_root}${_licenses_dir}
159
	cp ${BUILDER_ROOT}/license.txt ${_root}${_licenses_dir}/ESF
160
	echo "This package has a single license: ESF (Electric Sheep Fencing License)." \
161
		> ${_root}${_licenses_dir}/LICENSE
162
	cat <<EOF >${_root}${_licenses_dir}/catalog.mk
163
_LICENSE=ESF
164
_LICENSE_NAME=Electric Sheep Fencing License
165
_LICENSE_PERMS=dist-mirror dist-sell pkg-mirror pkg-sell auto-accept
166
_LICENSE_GROUPS=
167
_LICENSE_DISTFILES=
168
EOF
169
	cat <<EOF >>${_plist}
170
${_licenses_dir}/catalog.mk
171
${_licenses_dir}/LICENSE
172
${_licenses_dir}/ESF
173
EOF
174

    
175
	mkdir -p ${CORE_PKG_REAL_PATH}/All
176
	if ! pkg create -o ${CORE_PKG_REAL_PATH}/All -p ${_plist} -r ${_root} -m ${_metadir}; then
177
		echo ">>> ERROR: Error building package ${_template} ${_flavor}"
178
		print_error_pfS
179
	fi
180
}
181

    
182
# This routine will output that something went wrong
183
print_error_pfS() {
184
	echo
185
	echo "####################################"
186
	echo "Something went wrong, check errors!" >&2
187
	echo "####################################"
188
	echo
189
	echo "NOTE: a lot of times you can run './build.sh --clean-builder' to resolve."
190
	echo
191
	if [ "$1" != "" ]; then
192
		echo $1
193
	fi
194
	[ -n "${LOGFILE}" -a -f "${LOGFILE}" ] && \
195
		echo "Log saved on ${LOGFILE}" && \
196
		tail -n20 ${LOGFILE} >&2
197
	echo
198
	kill $$
199
	exit 1
200
}
201

    
202
prestage_on_ram_setup() {
203
	# Do not use memory disks for release build
204
	if [ -n "${_IS_RELEASE}" ]; then
205
		return
206
	fi
207

    
208
	[ -d "${STAGE_CHROOT_DIR}" ] \
209
		|| mkdir -p ${STAGE_CHROOT_DIR}
210
	[ -d "${FINAL_CHROOT_DIR}" ] \
211
		|| mkdir -p ${FINAL_CHROOT_DIR}
212

    
213
	_AVAIL_MEM=$(($(sysctl -n hw.usermem) / 1024 / 1024))
214
	if [ $_AVAIL_MEM -lt 2000 ]; then
215
		echo ">>> Builder has less than 2GiB RAM skipping memory disks"
216
		return
217
	else
218
		echo "######################################################################################"
219
		echo
220
		echo ">>> Builder has more than 2GiB RAM enabling memory disks"
221
		echo ">>> WARNING: Remember to remove these memory disks by running $0 --disable-memorydisks"
222
		echo
223
		echo "######################################################################################"
224
	fi
225

    
226
	if df /dev/ufs/prestagebacking >/dev/null 2>&1; then
227
		echo ">>> Detected preexisting memory disk enabled for ${STAGE_CHROOT_DIR}."
228
	else
229
		mdconfig -a -t swap -u 10001 -s ${MEMORYDISK_SIZE}
230
		newfs -L prestagebacking -U /dev/md10001
231
		mount /dev/ufs/prestagebacking ${STAGE_CHROOT_DIR}
232
	fi
233

    
234
	if df /dev/ufs/finalstagebacking >/dev/null 2>&1; then
235
		echo ">>> Detected preexisting memory disk enabled for ${FINAL_CHROOT_DIR}."
236
	else
237
		mdconfig -a -t swap -u 10002 -s ${MEMORYDISK_SIZE}
238
		newfs -L finalstagebacking -U /dev/md10002
239
		mount /dev/ufs/finalstagebacking ${FINAL_CHROOT_DIR}
240
	fi
241
}
242

    
243
prestage_on_ram_cleanup() {
244
	if [ -c /dev/md10001 ]; then
245
		if [ -d ${STAGE_CHROOT_DIR} ]; then
246
			umount ${STAGE_CHROOT_DIR}
247
		fi
248
		mdconfig -d -u 10001
249
	fi
250
	if [ -c /dev/md10002 ]; then
251
		if [ -d ${FINAL_CHROOT_DIR} ]; then
252
			umount ${FINAL_CHROOT_DIR}
253
		fi
254
		mdconfig -d -u 10002
255
	fi
256
}
257

    
258
# This routine will verify that the kernel has been
259
# installed OK to the staging area.
260
ensure_kernel_exists() {
261
	if [ ! -f "$1/boot/kernel/kernel.gz" ]; then
262
		echo ">>> ERROR: Could not locate $1/boot/kernel.gz"
263
		print_error_pfS
264
	fi
265
	KERNEL_SIZE=$(stat -f "%z" $1/boot/kernel/kernel.gz)
266
	if [ "$KERNEL_SIZE" -lt 3500 ]; then
267
		echo ">>> ERROR: Kernel $1/boot/kernel.gz appears to be smaller than it should be: $KERNEL_SIZE"
268
		print_error_pfS
269
	fi
270
}
271

    
272
get_pkg_name() {
273
	echo "${PRODUCT_NAME}-${1}-${CORE_PKG_VERSION}"
274
}
275

    
276
# This routine builds all related kernels
277
build_all_kernels() {
278
	# Set KERNEL_BUILD_PATH if it has not been set
279
	if [ -z "${KERNEL_BUILD_PATH}" ]; then
280
		KERNEL_BUILD_PATH=$SCRATCHDIR/kernels
281
		echo ">>> KERNEL_BUILD_PATH has not been set. Setting to ${KERNEL_BUILD_PATH}!"
282
	fi
283

    
284
	[ -d "${KERNEL_BUILD_PATH}" ] \
285
		&& rm -rf ${KERNEL_BUILD_PATH}
286

    
287
	# Build embedded kernel
288
	for BUILD_KERNEL in $BUILD_KERNELS; do
289
		unset KERNCONF
290
		unset KERNEL_DESTDIR
291
		unset KERNELCONF
292
		unset KERNEL_NAME
293
		export KERNCONF=$BUILD_KERNEL
294
		export KERNEL_DESTDIR="$KERNEL_BUILD_PATH/$BUILD_KERNEL"
295
		export KERNELCONF="${FREEBSD_SRC_DIR}/sys/${TARGET}/conf/$BUILD_KERNEL"
296
		export KERNEL_NAME=${BUILD_KERNEL}
297

    
298
		LOGFILE="${BUILDER_LOGS}/kernel.${KERNCONF}.${TARGET}.log"
299
		echo ">>> Building $BUILD_KERNEL kernel."  | tee -a ${LOGFILE}
300

    
301
		if [ ! -e "${FREEBSD_SRC_DIR}/sys/${TARGET}/conf/${BUILD_KERNEL}" ]; then
302
			echo ">>> ERROR: Could not find $KERNELCONF"
303
			print_error_pfS
304
		fi
305

    
306
		if [ -n "${NO_BUILDKERNEL}" -a -f "${CORE_PKG_ALL_PATH}/$(get_pkg_name kernel-${KERNEL_NAME}).txz" ]; then
307
			echo ">>> NO_BUILDKERNEL set, skipping build" | tee -a ${LOGFILE}
308
			continue
309
		fi
310

    
311
		export SRC_CONF=${SRC_CONF}
312
		buildkernel
313

    
314
		echo ">>> Staging $BUILD_KERNEL kernel..." | tee -a ${LOGFILE}
315
		installkernel
316

    
317
		ensure_kernel_exists $KERNEL_DESTDIR
318

    
319
		echo -n ">>> Creating pkg of $KERNEL_NAME-debug kernel to staging area... "  | tee -a ${LOGFILE}
320
		core_pkg_create kernel-debug ${KERNEL_NAME} ${CORE_PKG_VERSION} ${KERNEL_DESTDIR} \*.symbols
321
		find ${KERNEL_DESTDIR} -name '*.symbols' -type f -delete
322
		echo " Done" | tee -a ${LOGFILE}
323

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

    
327
		rm -rf $KERNEL_DESTDIR 2>&1 1>/dev/null
328

    
329
		echo " Done" | tee -a ${LOGFILE}
330
	done
331
}
332

    
333
install_default_kernel() {
334
	if [ -z "${1}" ]; then
335
		echo ">>> ERROR: install_default_kernel called without a kernel config name"| tee -a ${LOGFILE}
336
		print_error_pfS
337
	fi
338

    
339
	export KERNEL_NAME="${1}"
340

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

    
343
	# Copy kernel package to chroot, otherwise pkg won't find it to install
344
	if ! pkg_chroot_add ${FINAL_CHROOT_DIR} kernel-${KERNEL_NAME}; then
345
		echo ">>> ERROR: Error installing kernel package $(get_pkg_name kernel-${KERNEL_NAME}).txz" | tee -a ${LOGFILE}
346
		print_error_pfS
347
	fi
348

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

    
352
	if [ ! -f $FINAL_CHROOT_DIR/boot/kernel/kernel.gz ]; then
353
		echo ">>> ERROR: No kernel installed on $FINAL_CHROOT_DIR and the resulting image will be unusable. STOPPING!" | tee -a ${LOGFILE}
354
		print_error_pfS
355
	fi
356
	mkdir -p $FINAL_CHROOT_DIR/pkgs
357
	if [ -z "${2}" -o -n "${INSTALL_EXTRA_KERNELS}" ]; then
358
		cp ${CORE_PKG_ALL_PATH}/$(get_pkg_name kernel-${KERNEL_NAME}).txz $FINAL_CHROOT_DIR/pkgs
359
		if [ -n "${INSTALL_EXTRA_KERNELS}" ]; then
360
			for _EXTRA_KERNEL in $INSTALL_EXTRA_KERNELS; do
361
				_EXTRA_KERNEL_PATH=${CORE_PKG_ALL_PATH}/$(get_pkg_name kernel-${_EXTRA_KERNEL}).txz
362
				if [ -f "${_EXTRA_KERNEL_PATH}" ]; then
363
					echo -n ". adding ${_EXTRA_KERNEL_PATH} on image /pkgs folder"
364
					cp ${_EXTRA_KERNEL_PATH} $FINAL_CHROOT_DIR/pkgs
365
				else
366
					echo ">>> ERROR: Requested kernel $(get_pkg_name kernel-${_EXTRA_KERNEL}).txz was not found to be put on image /pkgs folder!"
367
					print_error_pfS
368
				fi
369
			done
370
		fi
371
	fi
372
	echo "Done." | tee -a ${LOGFILE}
373

    
374
	unset KERNEL_NAME
375
}
376

    
377
# Creates a full update file
378
create_Full_update_tarball() {
379
	local _variant="$1"
380

    
381
	LOGFILE=${BUILDER_LOGS}/full_update.${TARGET}
382
	if [ "${UPDATES_TARBALL_FILENAME}" = "" ]; then
383
		echo ">>> UPDATES_TARBALL_FILENAME is empty skipping generation of full update image!" | tee -a ${LOGFILE}
384
		return
385
	fi
386

    
387
	local _image_path=${UPDATES_TARBALL_FILENAME}
388
	if [ -n "${_variant}" ]; then
389
		_image_path=$(echo "$_image_path" | \
390
			sed "s/-Full-Update-/-Full-Update-${_variant}-/")
391
		VARIANTUPDATES="${VARIANTUPDATES}${VARIANTUPDATES:+ }${_image_path}"
392
	fi
393

    
394
	mkdir -p $UPDATESDIR
395

    
396
	customize_stagearea_for_image "fullupdate" "" $_variant
397
	install_default_kernel ${DEFAULT_KERNEL}
398

    
399
	rm -rf ${FINAL_CHROOT_DIR}/cf
400
	rm -rf ${FINAL_CHROOT_DIR}/conf
401
	rm -f ${FINAL_CHROOT_DIR}/etc/dh-parameters.*
402
	rm -f ${FINAL_CHROOT_DIR}/etc/rc.conf
403
	rm -f ${FINAL_CHROOT_DIR}/etc/pwd.db 2>/dev/null
404
	rm -f ${FINAL_CHROOT_DIR}/etc/group 2>/dev/null
405
	rm -f ${FINAL_CHROOT_DIR}/etc/spwd.db 2>/dev/null
406
	rm -f ${FINAL_CHROOT_DIR}/etc/passwd 2>/dev/null
407
	rm -f ${FINAL_CHROOT_DIR}/etc/master.passwd 2>/dev/null
408
	rm -f ${FINAL_CHROOT_DIR}/etc/fstab 2>/dev/null
409
	rm -f ${FINAL_CHROOT_DIR}/etc/bogons 2>/dev/null
410
	# Remove loader.conf and friends.  Ticket #560
411
	rm ${FINAL_CHROOT_DIR}/boot/loader.conf 2>/dev/null
412
	rm ${FINAL_CHROOT_DIR}/boot/loader.conf.local 2>/dev/null
413

    
414
	# Old systems will run (pre|post)_upgrade_command from /tmp
415
	if [ -f ${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/pre_upgrade_command ]; then
416
		cp -p \
417
			${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/pre_upgrade_command \
418
			${FINAL_CHROOT_DIR}/tmp
419
	fi
420
	if [ -f ${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/post_upgrade_command ]; then
421
		cp -p \
422
			${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/post_upgrade_command \
423
			${FINAL_CHROOT_DIR}/tmp
424
	fi
425

    
426
	echo ">>> Creating ${_image_path} ..." | tee -a ${LOGFILE}
427
	tar --exclude=./dev -czPf ${_image_path} -C ${FINAL_CHROOT_DIR} .
428

    
429
	echo ">>> Full Update created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
430
}
431

    
432
# Outputs various set variables aka env
433
print_flags() {
434

    
435
	echo
436
	printf "             Product version: %s\n" $PRODUCT_VERSION
437
	printf "                   Stage DIR: %s\n" $STAGE_CHROOT_DIR
438
	printf "                 Updates dir: %s\n" $UPDATESDIR
439
	printf " Image Preparation Stage DIR: %s\n" $FINAL_CHROOT_DIR
440
	printf "                  Source DIR: %s\n" $FREEBSD_SRC_DIR
441
	printf "          FreeBSD repository: %s\n" $FREEBSD_REPO_BASE
442
	printf "          FreeBSD-src branch: %s\n" $FREEBSD_BRANCH
443
	printf "     FreeBSD original branch: %s\n" $FREEBSD_PARENT_BRANCH
444
	printf "               BUILD_KERNELS: %s\n" $BUILD_KERNELS
445
	printf "           Git Branch or Tag: %s\n" $GIT_REPO_BRANCH_OR_TAG
446
	printf "            MODULES_OVERRIDE: %s\n" $MODULES_OVERRIDE
447
	printf "    VMDK_DISK_CAPACITY_IN_GB: %s\n" $VMDK_DISK_CAPACITY_IN_GB
448
	printf "                 OVFTEMPLATE: %s\n" $OVFTEMPLATE
449
	printf "                     OVFVMDK: %s\n" $OVFVMDK
450
	printf "                    SRC_CONF: %s\n" $SRC_CONF
451
	printf "                     ISOPATH: %s\n" $ISOPATH
452
	printf "                MEMSTICKPATH: %s\n" $MEMSTICKPATH
453
	printf "    UPDATES_TARBALL_FILENAME: %s\n" $UPDATES_TARBALL_FILENAME
454
if [ -n "$SHOW_ENV" ]; then
455
	for LINE in $(env | egrep -v '(terminal|PASS|NAME|USER|SSH|GROUP|HOST)'); do
456
		echo "SHOW_ENV: $LINE"
457
	done
458
fi
459
	echo
460
}
461

    
462
# This builds FreeBSD (make buildworld)
463
# Imported from FreeSBIE
464
make_world() {
465

    
466
	LOGFILE=${BUILDER_LOGS}/buildworld.${TARGET}
467
	if [ -n "${NO_BUILDWORLD}" ]; then
468
		echo ">>> NO_BUILDWORLD set, skipping build" | tee -a ${LOGFILE}
469
		return
470
	fi
471

    
472
	# Set SRC_CONF variable if it's not already set.
473
	if [ -z "${SRC_CONF}" ]; then
474
		echo ">>> SRC_CONF is unset make sure this is what you want!" | tee -a ${LOGFILE}
475
	else
476
		echo ">>> Setting SRC_CONF to $SRC_CONF" | tee -a ${LOGFILE}
477
	fi
478

    
479
	# Set default parameters
480
	export MAKE_ARGS="${MAKEJ_WORLD} __MAKE_CONF=${MAKE_CONF} SRCCONF=${SRC_CONF} TARGET=${TARGET} TARGET_ARCH=${TARGET_ARCH}"
481

    
482
	echo ">>> LOGFILE set to $LOGFILE." | tee -a ${LOGFILE}
483
	makeargs="${MAKE_ARGS}"
484
	echo ">>> Building world for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
485
	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}
486
	(env LOCAL_ITOOLS="${EXTRA_TOOLS}" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} -DNO_CLEAN ${makeargs} buildworld || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
487
	echo ">>> Building world for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
488

    
489
	LOGFILE=${BUILDER_LOGS}/installworld.${TARGET}
490
	echo ">>> LOGFILE set to $LOGFILE." | tee -a ${LOGFILE}
491
	# Create if cleaned up
492
	makeargs="${MAKE_ARGS} DESTDIR=${STAGE_CHROOT_DIR} WITHOUT_TOOLCHAIN=1"
493
	echo ">>> Installing world for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
494
	echo ">>> Builder is running the command: env LOCAL_ITOOLS=\"${EXTRA_TOOLS}\" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} installworld" | tee -a ${LOGFILE}
495
	(env LOCAL_ITOOLS="${EXTRA_TOOLS}" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} installworld || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
496
	echo ">>> Installing world for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
497

    
498
	makeargs="${MAKE_ARGS} DESTDIR=${STAGE_CHROOT_DIR}"
499
	echo ">>> Distribution world for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
500
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} distribution " | tee -a ${LOGFILE}
501
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} distribution  || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
502
	echo ">>> Distribution world for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
503

    
504
	[ -d "${STAGE_CHROOT_DIR}/usr/local/bin" ] \
505
		|| mkdir -p ${STAGE_CHROOT_DIR}/usr/local/bin
506
	makeargs="${MAKE_ARGS} DESTDIR=${STAGE_CHROOT_DIR}"
507
	echo ">>> Building and installing crypto tools and athstats for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
508
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/crypto ${makeargs} clean all install " | tee -a ${LOGFILE}
509
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/crypto ${makeargs} clean all install || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
510
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} clean" | tee -a ${LOGFILE}
511
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} clean || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
512
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} all" | tee -a ${LOGFILE}
513
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} all || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
514
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} install" | tee -a ${LOGFILE}
515
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} install || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
516
	echo ">>> Building and installing crypto tools and athstats for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
517

    
518
	unset makeargs
519
}
520

    
521
nanobsd_image_filename() {
522
	local _size="$1"
523
	local _type="$2"
524
	local _upgrade="$3"
525

    
526
	if [ -z "$_upgrade" ]; then
527
		local _template=${NANOBSD_IMG_TEMPLATE}
528
	else
529
		local _template=${NANOBSD_UPGRADE_TEMPLATE}
530
	fi
531

    
532
	echo "$_template" | sed \
533
		-e "s,%%SIZE%%,${_size},g" \
534
		-e "s,%%TYPE%%,${_type},g"
535

    
536
	return 0
537
}
538

    
539
# This routine originated in nanobsd.sh
540
nanobsd_set_flash_details () {
541
	a1=$(echo $1 | tr '[:upper:]' '[:lower:]')
542

    
543
	# Source:
544
	#	SanDisk CompactFlash Memory Card
545
	#	Product Manual
546
	#	Version 10.9
547
	#	Document No. 20-10-00038
548
	#	April 2005
549
	# Table 2-7
550
	# NB: notice math error in SDCFJ-4096-388 line.
551
	#
552
	case "${a1}" in
553
		2048|2048m|2048mb|2g)
554
			NANO_MEDIASIZE=$((1989999616/512))
555
			;;
556
		4096|4096m|4096mb|4g)
557
			NANO_MEDIASIZE=$((3989999616/512))
558
			;;
559
		8192|8192m|8192mb|8g)
560
			NANO_MEDIASIZE=$((7989999616/512))
561
			;;
562
		16384|16384m|16384mb|16g)
563
			NANO_MEDIASIZE=$((15989999616/512))
564
			;;
565
		*)
566
			echo "Unknown Flash capacity"
567
			exit 2
568
			;;
569
	esac
570

    
571
	NANO_HEADS=16
572
	NANO_SECTS=63
573

    
574
	echo ">>> [nanoo] $1"
575
	echo ">>> [nanoo] NANO_MEDIASIZE: $NANO_MEDIASIZE"
576
	echo ">>> [nanoo] NANO_HEADS: $NANO_HEADS"
577
	echo ">>> [nanoo] NANO_SECTS: $NANO_SECTS"
578
	echo ">>> [nanoo] NANO_BOOT0CFG: $NANO_BOOT0CFG"
579
}
580

    
581
# This routine originated in nanobsd.sh
582
create_nanobsd_diskimage () {
583
	if [ -z "${1}" ]; then
584
		echo ">>> ERROR: Type of image has not been specified"
585
		print_error_pfS
586
	fi
587
	if [ -z "${2}" ]; then
588
		echo ">>> ERROR: Size of image has not been specified"
589
		print_error_pfS
590
	fi
591

    
592
	if [ "${1}" = "nanobsd" ]; then
593
		# It's serial
594
		export NANO_BOOTLOADER="boot/boot0sio"
595
	elif [ "${1}" = "nanobsd-vga" ]; then
596
		# It's vga
597
		export NANO_BOOTLOADER="boot/boot0"
598
	else
599
		echo ">>> ERROR: Type of image to create unknown"
600
		print_error_pfS
601
	fi
602

    
603
	if [ -z "${2}" ]; then
604
		echo ">>> ERROR: Media size(s) not specified."
605
		print_error_pfS
606
	fi
607

    
608
	if [ -z "${2}" ]; then
609
		echo ">>> ERROR: FLASH_SIZE is not set."
610
		print_error_pfS
611
	fi
612

    
613
	LOGFILE=${BUILDER_LOGS}/${1}.${TARGET}
614
	# Prepare folder to be put in image
615
	customize_stagearea_for_image "${1}"
616
	install_default_kernel ${DEFAULT_KERNEL} "no"
617

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

    
620
	echo "nanobsd" > $FINAL_CHROOT_DIR/etc/platform
621

    
622
	local BOOTCONF=${FINAL_CHROOT_DIR}/boot.config
623
	local LOADERCONF=${FINAL_CHROOT_DIR}/boot/loader.conf
624

    
625
	if [ "${1}" = "nanobsd" ]; then
626
		# Tell loader to use serial console early.
627
		echo "-S115200 -h" >> ${BOOTCONF}
628

    
629
		# Remove old console options if present.
630
		[ -f "${LOADERCONF}" ] \
631
			&& sed -i "" -Ee "/(console|boot_multicons|boot_serial|hint.uart)/d" ${LOADERCONF}
632
		# Activate serial console+video console in loader.conf
633
		echo 'loader_color="NO"' >> ${LOADERCONF}
634
		echo 'beastie_disable="YES"' >> ${LOADERCONF}
635
		echo 'boot_serial="YES"' >> ${LOADERCONF}
636
		echo 'console="comconsole"' >> ${LOADERCONF}
637
		echo 'comconsole_speed="115200"' >> ${LOADERCONF}
638
	fi
639
	echo 'autoboot_delay="5"' >> ${LOADERCONF}
640

    
641
	# Old systems will run (pre|post)_upgrade_command from /tmp
642
	if [ -f ${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/pre_upgrade_command ]; then
643
		cp -p \
644
			${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/pre_upgrade_command \
645
			${FINAL_CHROOT_DIR}/tmp
646
	fi
647
	if [ -f ${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/post_upgrade_command ]; then
648
		cp -p \
649
			${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/post_upgrade_command \
650
			${FINAL_CHROOT_DIR}/tmp
651
	fi
652

    
653
	for _NANO_MEDIASIZE in ${2}; do
654
		if [ -z "${_NANO_MEDIASIZE}" ]; then
655
			continue;
656
		fi
657

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

    
661
		IMG="${IMAGES_FINAL_DIR}/$(nanobsd_image_filename ${_NANO_MEDIASIZE} ${1})"
662
		IMGUPDATE="${IMAGES_FINAL_DIR}/$(nanobsd_image_filename ${_NANO_MEDIASIZE} ${1} 1)"
663

    
664
		nanobsd_set_flash_details ${_NANO_MEDIASIZE}
665

    
666
		# These are defined in FlashDevice and on builder_default.sh
667
		echo $NANO_MEDIASIZE \
668
			$NANO_IMAGES \
669
			$NANO_SECTS \
670
			$NANO_HEADS \
671
			$NANO_CODESIZE \
672
			$NANO_CONFSIZE \
673
			$NANO_DATASIZE |
674
awk '
675
{
676
	printf "# %s\n", $0
677

    
678
	# size of cylinder in sectors
679
	cs = $3 * $4
680

    
681
	# number of full cylinders on media
682
	cyl = int ($1 / cs)
683

    
684
	# output fdisk geometry spec, truncate cyls to 1023
685
	if (cyl <= 1023)
686
		print "g c" cyl " h" $4 " s" $3
687
	else
688
		print "g c" 1023 " h" $4 " s" $3
689

    
690
	if ($7 > 0) {
691
		# size of data partition in full cylinders
692
		dsl = int (($7 + cs - 1) / cs)
693
	} else {
694
		dsl = 0;
695
	}
696

    
697
	# size of config partition in full cylinders
698
	csl = int (($6 + cs - 1) / cs)
699

    
700
	if ($5 == 0) {
701
		# size of image partition(s) in full cylinders
702
		isl = int ((cyl - dsl - csl) / $2)
703
	} else {
704
		isl = int (($5 + cs - 1) / cs)
705
	}
706

    
707
	# First image partition start at second track
708
	print "p 1 165 " $3, isl * cs - $3
709
	c = isl * cs;
710

    
711
	# Second image partition (if any) also starts offset one
712
	# track to keep them identical.
713
	if ($2 > 1) {
714
		print "p 2 165 " $3 + c, isl * cs - $3
715
		c += isl * cs;
716
	}
717

    
718
	# Config partition starts at cylinder boundary.
719
	print "p 3 165 " c, csl * cs
720
	c += csl * cs
721

    
722
	# Data partition (if any) starts at cylinder boundary.
723
	if ($7 > 0) {
724
		print "p 4 165 " c, dsl * cs
725
	} else if ($7 < 0 && $1 > c) {
726
		print "p 4 165 " c, $1 - c
727
	} else if ($1 < c) {
728
		print "Disk space overcommitted by", \
729
		    c - $1, "sectors" > "/dev/stderr"
730
		exit 2
731
	}
732

    
733
	# Force slice 1 to be marked active. This is necessary
734
	# for booting the image from a USB device to work.
735
	print "a 1"
736
}
737
	' > ${IMAGES_FINAL_DIR}/_.fdisk
738

    
739
		MNT=${IMAGES_FINAL_DIR}/_.mnt
740
		mkdir -p ${MNT}
741

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

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

    
748
		fdisk -i -f ${IMAGES_FINAL_DIR}/_.fdisk ${MD} 2>&1 >> ${LOGFILE}
749
		fdisk ${MD} 2>&1 >> ${LOGFILE}
750

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

    
753
		# Create first image
754
		bsdlabel -m i386 -w -B -b ${FINAL_CHROOT_DIR}/boot/boot ${MD}s1 2>&1 >> ${LOGFILE}
755
		bsdlabel -m i386 ${MD}s1 2>&1 >> ${LOGFILE}
756
		local _label=$(lc ${PRODUCT_NAME})
757
		newfs -L ${_label}0 ${NANO_NEWFS} /dev/${MD}s1a 2>&1 >> ${LOGFILE}
758
		mount /dev/ufs/${_label}0 ${MNT}
759
		if [ $? -ne 0 ]; then
760
			echo ">>> ERROR: Something wrong happened during mount of first slice image creation. STOPPING!" | tee -a ${LOGFILE}
761
			print_error_pfS
762
		fi
763
		# Consider the unmounting as well
764
		trap "umount /dev/ufs/${_label}0; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
765

    
766
		clone_directory_contents ${FINAL_CHROOT_DIR} ${MNT}
767

    
768
		# Set NanoBSD image size
769
		echo "${_NANO_MEDIASIZE}" > ${MNT}/etc/nanosize.txt
770

    
771
		echo "/dev/ufs/${_label}0 / ufs ro,sync,noatime 1 1" > ${MNT}/etc/fstab
772
		if [ $NANO_CONFSIZE -gt 0 ] ; then
773
			echo "/dev/ufs/cf /cf ufs ro,sync,noatime 1 1" >> ${MNT}/etc/fstab
774
		fi
775

    
776
		umount ${MNT}
777
		# Restore the original trap
778
		trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
779

    
780
		# Setting NANO_IMAGES to 1 and NANO_INIT_IMG2 will tell
781
		# NanoBSD to only create one partition.  We default to 2
782
		# partitions in case anything happens to the first the
783
		# operator can boot from the 2nd and should be OK.
784

    
785
		# Before just going to use dd for duplicate think!
786
		# The images are created as sparse so lets take advantage
787
		# of that by just exec some commands.
788
		if [ $NANO_IMAGES -gt 1 -a $NANO_INIT_IMG2 -gt 0 ] ; then
789
			# Duplicate to second image (if present)
790
			echo ">>> Creating NanoBSD second slice by duplicating first slice." | tee -a ${LOGFILE}
791
			# Create second image
792
			dd if=/dev/${MD}s1 of=/dev/${MD}s2 conv=sparse bs=64k 2>&1 >> ${LOGFILE}
793
			tunefs -L ${_label}1 /dev/${MD}s2a 2>&1 >> ${LOGFILE}
794
			mount /dev/ufs/${_label}1 ${MNT}
795
			if [ $? -ne 0 ]; then
796
				echo ">>> ERROR: Something wrong happened during mount of second slice image creation. STOPPING!" | tee -a ${LOGFILE}
797
				print_error_pfS
798
			fi
799
			# Consider the unmounting as well
800
			trap "umount /dev/ufs/${_label}1; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
801

    
802
			echo "/dev/ufs/${_label}1 / ufs ro,sync,noatime 1 1" > ${MNT}/etc/fstab
803
			if [ $NANO_CONFSIZE -gt 0 ] ; then
804
				echo "/dev/ufs/cf /cf ufs ro,sync,noatime 1 1" >> ${MNT}/etc/fstab
805
			fi
806

    
807
			umount ${MNT}
808
			# Restore the trap back
809
			trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
810
		fi
811

    
812
		# Create Data slice, if any.
813
		# Note the changing of the variable to NANO_CONFSIZE
814
		# from NANO_DATASIZE.  We also added glabel support
815
		# and populate the Product configuration from the /cf
816
		# directory located in FINAL_CHROOT_DIR
817
		if [ $NANO_CONFSIZE -gt 0 ] ; then
818
			echo ">>> Creating /cf area to hold config.xml"
819
			newfs -L cf ${NANO_NEWFS} /dev/${MD}s3 2>&1 >> ${LOGFILE}
820
			# Mount data partition and copy contents of /cf
821
			# Can be used later to create custom default config.xml while building
822
			mount /dev/ufs/cf ${MNT}
823
			if [ $? -ne 0 ]; then
824
				echo ">>> ERROR: Something wrong happened during mount of cf slice image creation. STOPPING!" | tee -a ${LOGFILE}
825
				print_error_pfS
826
			fi
827
			# Consider the unmounting as well
828
			trap "umount /dev/ufs/cf; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
829

    
830
			clone_directory_contents ${FINAL_CHROOT_DIR}/cf ${MNT}
831

    
832
			umount ${MNT}
833
			# Restore the trap back
834
			trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
835
		else
836
			">>> [nanoo] NANO_CONFSIZE is not set. Not adding a /conf partition.. You sure about this??" | tee -a ${LOGFILE}
837
		fi
838

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

    
842
		mdconfig -d -u $MD
843
		# Restore default action
844
		trap "-" 1 2 15 EXIT
845

    
846
		# Check each image and ensure that they are over
847
		# 3 megabytes.  If either image is under 20 megabytes
848
		# in size then error out.
849
		IMGSIZE=$(stat -f "%z" ${IMG})
850
		IMGUPDATESIZE=$(stat -f "%z" ${IMGUPDATE})
851
		CHECKSIZE="20040710"
852
		if [ "$IMGSIZE" -lt "$CHECKSIZE" ]; then
853
			echo ">>> ERROR: Something went wrong when building NanoBSD.  The image size is under 20 megabytes!" | tee -a ${LOGFILE}
854
			print_error_pfS
855
		fi
856
		if [ "$IMGUPDATESIZE" -lt "$CHECKSIZE" ]; then
857
			echo ">>> ERROR: Something went wrong when building NanoBSD upgrade image.  The image size is under 20 megabytes!" | tee -a ${LOGFILE}
858
			print_error_pfS
859
		fi
860

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

    
864
		gzip -qf $IMG &
865
		_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
866
		gzip -qf $IMGUPDATE &
867
		_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
868
	done
869

    
870
	unset IMG
871
	unset IMGUPDATE
872
	unset IMGUPDATESIZE
873
	unset IMGSIZE
874

    
875
	ls -lah $IMAGES_FINAL_DIR
876
}
877

    
878
# This routine creates a ova image that contains
879
# a ovf and vmdk file. These files can be imported
880
# right into vmware or virtual box.
881
# (and many other emulation platforms)
882
# http://www.vmware.com/pdf/ovf_whitepaper_specification.pdf
883
create_ova_image() {
884
	# XXX create a .ovf php creator that you can pass:
885
	#     1. populatedSize
886
	#     2. license
887
	#     3. product name
888
	#     4. version
889
	#     5. number of network interface cards
890
	#     6. allocationUnits
891
	#     7. capacity
892
	#     8. capacityAllocationUnits
893

    
894
	LOGFILE=${BUILDER_LOGS}/ova.${TARGET}.log
895

    
896
	[ -d "${OVA_TMP}" ] \
897
		&& rm -rf ${OVA_TMP}
898

    
899
	local _mntdir=${OVA_TMP}/mnt
900
	mkdir -p ${_mntdir}
901

    
902
	if [ -z "${OVA_SWAP_PART_SIZE_IN_GB}" -o "${OVA_SWAP_PART_SIZE_IN_GB}" = "0" ]; then
903
		# first partition size (freebsd-ufs)
904
		local OVA_FIRST_PART_SIZE_IN_GB=${VMDK_DISK_CAPACITY_IN_GB}
905
		# Calculate real first partition size, removing 128 blocks (65536 bytes) beginning/loader
906
		local OVA_FIRST_PART_SIZE=$((${OVA_FIRST_PART_SIZE_IN_GB}*1024*1024*1024-65536))
907
		# Unset swap partition size variable
908
		unset OVA_SWAP_PART_SIZE
909
		# Parameter used by mkimg
910
		unset OVA_SWAP_PART_PARAM
911
	else
912
		# first partition size (freebsd-ufs)
913
		local OVA_FIRST_PART_SIZE_IN_GB=$((VMDK_DISK_CAPACITY_IN_GB-OVA_SWAP_PART_SIZE_IN_GB))
914
		# Use first partition size in g
915
		local OVA_FIRST_PART_SIZE="${OVA_FIRST_PART_SIZE_IN_GB}g"
916
		# Calculate real swap size, removing 128 blocks (65536 bytes) beginning/loader
917
		local OVA_SWAP_PART_SIZE=$((${OVA_SWAP_PART_SIZE_IN_GB}*1024*1024*1024-65536))
918
		# Parameter used by mkimg
919
		local OVA_SWAP_PART_PARAM="-p freebsd-swap/swap0::${OVA_SWAP_PART_SIZE}"
920
	fi
921

    
922
	# Prepare folder to be put in image
923
	customize_stagearea_for_image "ova"
924
	install_default_kernel ${DEFAULT_KERNEL} "no"
925

    
926
	# Fill fstab
927
	echo ">>> Installing platform specific items..." | tee -a ${LOGFILE}
928
	echo "/dev/gpt/${PRODUCT_NAME}	/	ufs		rw	1	1" > ${FINAL_CHROOT_DIR}/etc/fstab
929
	if [ -n "${OVA_SWAP_PART_SIZE}" ]; then
930
		echo "/dev/gpt/swap0	none	swap	sw	0	0" >> ${FINAL_CHROOT_DIR}/etc/fstab
931
	fi
932

    
933
	# Create / partition
934
	echo -n ">>> Creating / partition... " | tee -a ${LOGFILE}
935
	truncate -s ${OVA_FIRST_PART_SIZE} ${OVA_TMP}/${OVFUFS}
936
	local _md=$(mdconfig -a -f ${OVA_TMP}/${OVAUFS})
937
	trap "mdconfig -d -u ${_md}; return" 1 2 15 EXIT
938

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

    
941
	if ! mount /dev/${_md} ${_mntdir} 2>&1 >>${LOGFILE}; then
942
		echo "Failed!" | tee -a ${LOGFILE}
943
		echo ">>> ERROR: Error mounting temporary vmdk image. STOPPING!" | tee -a ${LOGFILE}
944
		print_error_pfS
945
	fi
946
	trap "umount ${_mntdir}; mdconfig -d -u ${_md}; return" 1 2 15 EXIT
947

    
948
	echo "Done!" | tee -a ${LOGFILE}
949

    
950
	clone_directory_contents ${FINAL_CHROOT_DIR} ${_mntdir}
951

    
952
	sync
953
	umount ${_mntdir} 2>&1 >>${LOGFILE}
954
	mdconfig -d -u ${_md}
955
	trap "-" 1 2 15 EXIT
956

    
957
	# Create raw disk
958
	echo -n ">>> Creating raw disk... " | tee -a ${LOGFILE}
959
	mkimg \
960
		-s gpt \
961
		-f raw \
962
		-b ${FINAL_CHROOT_DIR}/boot/pmbr \
963
		-p freebsd-boot:=${FINAL_CHROOT_DIR}/boot/gptboot \
964
		-p freebsd-ufs/${PRODUCT_NAME}:=${OVA_TMP}/${OVFUFS} \
965
		${OVA_SWAP_PART_PARAM} \
966
		-o ${OVA_TMP}/${OVFRAW} 2>&1 >> ${LOGFILE}
967

    
968
	if [ $? -ne 0 -o ! -f ${OVA_TMP}/${OVFRAW} ]; then
969
		if [ -f ${OVA_TMP}/${OVFUFS} ]; then
970
			rm -f ${OVA_TMP}/${OVFUFS}
971
		fi
972
		if [ -f ${OVA_TMP}/${OVFRAW} ]; then
973
			rm -f ${OVA_TMP}/${OVFRAW}
974
		fi
975
		echo "Failed!" | tee -a ${LOGFILE}
976
		echo ">>> ERROR: Error creating temporary vmdk image. STOPPING!" | tee -a ${LOGFILE}
977
		print_error_pfS
978
	fi
979
	echo "Done!" | tee -a ${LOGFILE}
980

    
981
	# We don't need it anymore
982
	rm -f ${OVA_TMP}/${OVFUFS} >/dev/null 2>&1
983

    
984
	# Convert raw to vmdk
985
	echo -n ">>> Creating vmdk disk... " | tee -a ${LOGFILE}
986
	vmdktool -z9 -v ${OVA_TMP}/${OVFVMDK} ${OVA_TMP}/${OVFRAW}
987

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

    
1001
	rm -f ${OVA_TMP}/i${OVFRAW}
1002

    
1003
	ova_setup_ovf_template
1004

    
1005
	echo -n ">>> Writing final ova image... " | tee -a ${LOGFILE}
1006
	# Create OVA file for vmware
1007
	gtar -C ${OVA_TMP} -cpf ${OVAPATH} ${PRODUCT_NAME}.ovf ${OVFVMDK}
1008
	echo "Done!" | tee -a ${LOGFILE}
1009
	rm -f ${OVA_TMP}/${OVFVMDK} >/dev/null 2>&1
1010

    
1011
	echo ">>> OVA created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1012
}
1013

    
1014
# called from create_ova_image
1015
ova_setup_ovf_template() {
1016
	if [ ! -f ${OVFTEMPLATE} ]; then
1017
		echo ">>> ERROR: OVF template file (${OVFTEMPLATE}) not found."
1018
		print_error_pfS
1019
	fi
1020

    
1021
	#  OperatingSystemSection (${PRODUCT_NAME}.ovf)
1022
	#  42   FreeBSD 32-Bit
1023
	#  78   FreeBSD 64-Bit
1024
	if [ "${TARGET}" = "amd64" ]; then
1025
		local _os_id="78"
1026
		local _os_type="freebsd64Guest"
1027
		local _os_descr="FreeBSD 64-Bit"
1028
	elif [ "${TARGET}" = "i386" ]; then
1029
		local _os_id="42"
1030
		local _os_type="freebsdGuest"
1031
		local _os_descr="FreeBSD"
1032
	else
1033
		echo ">>> ERROR: Platform not supported for OVA (${TARGET})"
1034
		print_error_pfS
1035
	fi
1036

    
1037
	local POPULATED_SIZE=$(du -d0 -k $FINAL_CHROOT_DIR | cut -f1)
1038
	local POPULATED_SIZE_IN_BYTES=$((${POPULATED_SIZE}*1024))
1039
	local VMDK_FILE_SIZE=$(stat -f "%z" ${OVA_TMP}/${OVFVMDK})
1040

    
1041
	sed \
1042
		-e "s,%%VMDK_FILE_SIZE%%,${VMDK_FILE_SIZE},g" \
1043
		-e "s,%%VMDK_DISK_CAPACITY_IN_GB%%,${VMDK_DISK_CAPACITY_IN_GB},g" \
1044
		-e "s,%%POPULATED_SIZE_IN_BYTES%%,${POPULATED_SIZE_IN_BYTES},g" \
1045
		-e "s,%%OS_ID%%,${_os_id},g" \
1046
		-e "s,%%OS_TYPE%%,${_os_type},g" \
1047
		-e "s,%%OS_DESCR%%,${_os_descr},g" \
1048
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
1049
		-e "s,%%PRODUCT_NAME_SUFFIX%%,${PRODUCT_NAME_SUFFIX},g" \
1050
		-e "s,%%PRODUCT_VERSION%%,${PRODUCT_VERSION},g" \
1051
		-e "s,%%PRODUCT_URL%%,${PRODUCT_URL},g" \
1052
		-e "s#%%VENDOR_NAME%%#${VENDOR_NAME}#g" \
1053
		-e "s#%%OVF_INFO%%#${OVF_INFO}#g" \
1054
		-e "/^%%PRODUCT_LICENSE%%/r ${BUILDER_ROOT}/license.txt" \
1055
		-e "/^%%PRODUCT_LICENSE%%/d" \
1056
		${OVFTEMPLATE} > ${OVA_TMP}/${PRODUCT_NAME}.ovf
1057
}
1058

    
1059
# Cleans up previous builds
1060
clean_builder() {
1061
	# Clean out directories
1062
	echo ">>> Cleaning up previous build environment...Please wait!"
1063

    
1064
	staginareas_clean_each_run
1065

    
1066
	if [ -d "${STAGE_CHROOT_DIR}" ]; then
1067
		BASENAME=$(basename ${STAGE_CHROOT_DIR})
1068
		echo -n ">>> Cleaning ${STAGE_CHROOT_DIR} ..."
1069
		chflags -R noschg ${STAGE_CHROOT_DIR} 2>&1 >/dev/null
1070
		rm -rf ${STAGE_CHROOT_DIR}/* 2>/dev/null
1071
		echo "Done."
1072
	fi
1073

    
1074
	if [ -z "${NO_CLEAN_FREEBSD_OBJ}" -a -d "${FREEBSD_SRC_DIR}" ]; then
1075
		OBJTREE=$(env TARGET=${TARGET} TARGET_ARCH=${TARGET_ARCH} make -C ${FREEBSD_SRC_DIR} -V OBJTREE)
1076
		if [ -d "${OBJTREE}" ]; then
1077
			echo -n ">>> Cleaning FreeBSD objects dir staging..."
1078
			echo -n "."
1079
			chflags -R noschg ${OBJTREE} 2>&1 >/dev/null
1080
			echo -n "."
1081
			rm -rf ${OBJTREE}/*
1082
			echo "Done!"
1083
		fi
1084
		if [ -d "${KERNEL_BUILD_PATH}" ]; then
1085
			echo -n ">>> Cleaning previously built kernel stage area..."
1086
			rm -rf $KERNEL_BUILD_PATH/*
1087
			echo "Done!"
1088
		fi
1089
	fi
1090
	mkdir -p $KERNEL_BUILD_PATH
1091

    
1092
	echo -n ">>> Cleaning previously built images..."
1093
	rm -rf $IMAGES_FINAL_DIR/*
1094
	rm -rf $STAGINGAREA/*
1095
	echo "Done!"
1096

    
1097
	if [ -z "${NO_CLEAN_FREEBSD_SRC}" ]; then
1098
		if [ -d "$FREEBSD_SRC_DIR" ]; then
1099
			echo -n ">>> Ensuring $FREEBSD_SRC_DIR is clean..."
1100
			rm -rf ${FREEBSD_SRC_DIR}
1101
			echo "Done!"
1102
		fi
1103
	fi
1104

    
1105
	echo -n ">>> Cleaning previous builder logs..."
1106
	if [ -d "$BUILDER_LOGS" ]; then
1107
		rm -rf ${BUILDER_LOGS}
1108
	fi
1109
	mkdir -p ${BUILDER_LOGS}
1110

    
1111
	echo "Done!"
1112

    
1113
	echo ">>> Cleaning of builder environment has finished."
1114
}
1115

    
1116
clone_directory_contents() {
1117
	if [ ! -e "$2" ]; then
1118
		mkdir -p "$2"
1119
	fi
1120
	if [ ! -d "$1" -o ! -d "$2" ]; then
1121
		if [ -z "${LOGFILE}" ]; then
1122
			echo ">>> ERROR: Argument $1 supplied is not a directory!"
1123
		else
1124
			echo ">>> ERROR: Argument $1 supplied is not a directory!" | tee -a ${LOGFILE}
1125
		fi
1126
		print_error_pfS
1127
	fi
1128
	echo -n ">>> Using TAR to clone $1 to $2 ..."
1129
	tar -C ${1} -c -f - . | tar -C ${2} -x -p -f -
1130
	echo "Done!"
1131
}
1132

    
1133
clone_to_staging_area() {
1134
	# Clone everything to the final staging area
1135
	echo -n ">>> Cloning everything to ${STAGE_CHROOT_DIR} staging area..."
1136
	LOGFILE=${BUILDER_LOGS}/cloning.${TARGET}.log
1137

    
1138
	tar -C ${PRODUCT_SRC} -c -f - . | \
1139
		tar -C ${STAGE_CHROOT_DIR} -x -p -f -
1140

    
1141
	if [ "${PRODUCT_NAME}" != "pfSense" ]; then
1142
		mv ${STAGE_CHROOT_DIR}/usr/local/sbin/pfSense-upgrade \
1143
			${STAGE_CHROOT_DIR}/usr/local/sbin/${PRODUCT_NAME}-upgrade
1144
	fi
1145

    
1146
	if [ -f ${STAGE_CHROOT_DIR}/etc/master.passwd ]; then
1147
		chroot ${STAGE_CHROOT_DIR} pwd_mkdb /etc/master.passwd
1148
	fi
1149
	mkdir -p ${STAGE_CHROOT_DIR}/etc/mtree
1150
	mtree -Pcp ${STAGE_CHROOT_DIR}/var > ${STAGE_CHROOT_DIR}/etc/mtree/var.dist
1151
	mtree -Pcp ${STAGE_CHROOT_DIR}/etc > ${STAGE_CHROOT_DIR}/etc/mtree/etc.dist
1152
	if [ -d ${STAGE_CHROOT_DIR}/usr/local/etc ]; then
1153
		mtree -Pcp ${STAGE_CHROOT_DIR}/usr/local/etc > ${STAGE_CHROOT_DIR}/etc/mtree/localetc.dist
1154
	fi
1155

    
1156
	## Add buildtime and lastcommit information
1157
	# This is used for detecting updates.
1158
	echo "$BUILTDATESTRING" > $STAGE_CHROOT_DIR/etc/version.buildtime
1159
	# Record last commit info if it is available.
1160
	if [ -f $SCRATCHDIR/build_commit_info.txt ]; then
1161
		cp $SCRATCHDIR/build_commit_info.txt $STAGE_CHROOT_DIR/etc/version.lastcommit
1162
	fi
1163

    
1164
	local _exclude_files="${CORE_PKG_TMP}/base_exclude_files"
1165
	sed \
1166
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
1167
		-e "s,%%VERSION%%,${_version},g" \
1168
		${BUILDER_TOOLS}/templates/core_pkg/base/exclude_files \
1169
		> ${_exclude_files}
1170

    
1171
	mkdir -p ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR} >/dev/null 2>&1
1172

    
1173
	# Include a sample pkg stable conf to base
1174
	setup_pkg_repo \
1175
		${PKG_REPO_DEFAULT} \
1176
		${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/${PRODUCT_NAME}-repo.conf \
1177
		${TARGET} \
1178
		${TARGET_ARCH}
1179

    
1180
	mtree \
1181
		-c \
1182
		-k uid,gid,mode,size,flags,sha256digest \
1183
		-p ${STAGE_CHROOT_DIR} \
1184
		-X ${_exclude_files} \
1185
		> ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/base.mtree
1186
	tar \
1187
		-C ${STAGE_CHROOT_DIR} \
1188
		-cJf ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/base.txz \
1189
		-X ${_exclude_files} \
1190
		.
1191

    
1192
	local _share_repos_path="${SCRATCHDIR}/repo-tmp/${PRODUCT_SHARE_DIR}/pkg/repos"
1193
	rm -rf ${SCRATCHDIR}/repo-tmp >/dev/null 2>&1
1194
	mkdir -p ${_share_repos_path} >/dev/null 2>&1
1195

    
1196
	setup_pkg_repo \
1197
		${PKG_REPO_DEFAULT} \
1198
		${_share_repos_path}/${PRODUCT_NAME}-repo.conf \
1199
		${TARGET} \
1200
		${TARGET_ARCH}
1201

    
1202
	cp -f ${PKG_REPO_DEFAULT%%.conf}.descr ${_share_repos_path}
1203

    
1204
	# Add additional repos
1205
	for _template in ${PKG_REPO_BASE}/${PRODUCT_NAME}-repo-*.conf; do
1206
		_template_filename=$(basename ${_template})
1207
		setup_pkg_repo \
1208
			${_template} \
1209
			${_share_repos_path}/${_template_filename} \
1210
			${TARGET} \
1211
			${TARGET_ARCH}
1212
		cp -f ${_template%%.conf}.descr ${_share_repos_path}
1213
	done
1214

    
1215
	core_pkg_create repo "" ${CORE_PKG_VERSION} ${SCRATCHDIR}/repo-tmp
1216

    
1217
	core_pkg_create rc "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1218
	core_pkg_create base "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1219
	core_pkg_create base-nanobsd "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1220
	core_pkg_create default-config "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1221

    
1222
	local DEFAULTCONF=${STAGE_CHROOT_DIR}/conf.default/config.xml
1223

    
1224
	# Save current WAN and LAN if value
1225
	local _old_wan_if=$(xml sel -t -v "${XML_ROOTOBJ}/interfaces/wan/if" ${DEFAULTCONF})
1226
	local _old_lan_if=$(xml sel -t -v "${XML_ROOTOBJ}/interfaces/lan/if" ${DEFAULTCONF})
1227

    
1228
	# Change default interface names to match vmware driver
1229
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/wan/if" -v "vmx0" ${DEFAULTCONF}
1230
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/lan/if" -v "vmx1" ${DEFAULTCONF}
1231
	core_pkg_create default-config "vmware" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1232

    
1233
	# Restore default values to be used by serial package
1234
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/wan/if" -v "${_old_wan_if}" ${DEFAULTCONF}
1235
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/lan/if" -v "${_old_lan_if}" ${DEFAULTCONF}
1236

    
1237
	# Activate serial console in config.xml
1238
	xml ed -L -P -d "${XML_ROOTOBJ}/system/enableserial" ${DEFAULTCONF}
1239
	xml ed -P -s "${XML_ROOTOBJ}/system" -t elem -n "enableserial" \
1240
		${DEFAULTCONF} > ${DEFAULTCONF}.tmp
1241
	xml fo -t ${DEFAULTCONF}.tmp > ${DEFAULTCONF}
1242
	rm -f ${DEFAULTCONF}.tmp
1243

    
1244
	echo force > ${STAGE_CHROOT_DIR}/cf/conf/enableserial_force
1245

    
1246
	core_pkg_create default-config-serial "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1247

    
1248
	rm -f ${STAGE_CHROOT_DIR}/cf/conf/enableserial_force
1249
	rm -f ${STAGE_CHROOT_DIR}/cf/conf/config.xml
1250

    
1251
	# Make sure pkg is present
1252
	pkg_bootstrap ${STAGE_CHROOT_DIR}
1253

    
1254
	echo "Done!"
1255
}
1256

    
1257
create_final_staging_area() {
1258
	if [ -z "${FINAL_CHROOT_DIR}" ]; then
1259
		echo ">>> ERROR: FINAL_CHROOT_DIR is not set, cannot continue!" | tee -a ${LOGFILE}
1260
		print_error_pfS
1261
	fi
1262

    
1263
	if [ -d "${FINAL_CHROOT_DIR}" ]; then
1264
		echo -n ">>> Previous ${FINAL_CHROOT_DIR} detected cleaning up..." | tee -a ${LOGFILE}
1265
		chflags -R noschg ${FINAL_CHROOT_DIR} 2>&1 1>/dev/null
1266
		rm -rf ${FINAL_CHROOT_DIR}/* 2>&1 1>/dev/null
1267
		echo "Done." | tee -a ${LOGFILE}
1268
	fi
1269

    
1270
	echo ">>> Preparing Final image staging area: $(LC_ALL=C date)" 2>&1 | tee -a ${LOGFILE}
1271
	echo ">>> Cloning ${STAGE_CHROOT_DIR} to ${FINAL_CHROOT_DIR}" 2>&1 | tee -a ${LOGFILE}
1272
	clone_directory_contents ${STAGE_CHROOT_DIR} ${FINAL_CHROOT_DIR}
1273

    
1274
	if [ ! -f $FINAL_CHROOT_DIR/sbin/init ]; then
1275
		echo ">>> ERROR: Something went wrong during cloning -- Please verify!" 2>&1 | tee -a ${LOGFILE}
1276
		print_error_pfS
1277
	fi
1278
}
1279

    
1280
customize_stagearea_for_image() {
1281
	local _image_type="$1"
1282
	local _default_config="" # filled with $2 below
1283
	local _image_variant="$3"
1284

    
1285
	if [ -n "$2" ]; then
1286
		_default_config="$2"
1287
	elif [ "${_image_type}" = "nanobsd" -o \
1288
	     "${_image_type}" = "memstickserial" -o \
1289
	     "${_image_type}" = "memstickadi" ]; then
1290
		_default_config="default-config-serial"
1291
	elif [ "${_image_type}" = "ova" ]; then
1292
		_default_config="default-config-vmware"
1293
	else
1294
		_default_config="default-config"
1295
	fi
1296

    
1297
	# Prepare final stage area
1298
	create_final_staging_area
1299

    
1300
	pkg_chroot_add ${FINAL_CHROOT_DIR} rc
1301
	pkg_chroot_add ${FINAL_CHROOT_DIR} repo
1302

    
1303
	if [ "${_image_type}" = "nanobsd" -o \
1304
	     "${_image_type}" = "nanobsd-vga" ]; then
1305

    
1306
		mkdir -p ${FINAL_CHROOT_DIR}/root/var/db \
1307
			 ${FINAL_CHROOT_DIR}/root/var/cache \
1308
			 ${FINAL_CHROOT_DIR}/var/db/pkg \
1309
			 ${FINAL_CHROOT_DIR}/var/cache/pkg
1310
		mv -f ${FINAL_CHROOT_DIR}/var/db/pkg ${FINAL_CHROOT_DIR}/root/var/db
1311
		mv -f ${FINAL_CHROOT_DIR}/var/cache/pkg ${FINAL_CHROOT_DIR}/root/var/cache
1312
		ln -sf ../../root/var/db/pkg ${FINAL_CHROOT_DIR}/var/db/pkg
1313
		ln -sf ../../root/var/cache/pkg ${FINAL_CHROOT_DIR}/var/cache/pkg
1314

    
1315
		pkg_chroot_add ${FINAL_CHROOT_DIR} base-nanobsd
1316
	else
1317
		pkg_chroot_add ${FINAL_CHROOT_DIR} base
1318
	fi
1319

    
1320
	if [ "${_image_type}" = "iso" -o \
1321
	     "${_image_type}" = "memstick" -o \
1322
	     "${_image_type}" = "memstickserial" -o \
1323
	     "${_image_type}" = "memstickadi" ]; then
1324
		install_bsdinstaller
1325
		mkdir -p ${FINAL_CHROOT_DIR}/pkgs
1326
		cp ${CORE_PKG_ALL_PATH}/*default-config*.txz ${FINAL_CHROOT_DIR}/pkgs
1327
	fi
1328

    
1329
	pkg_chroot_add ${FINAL_CHROOT_DIR} ${_default_config}
1330

    
1331
	# XXX: Workaround to avoid pkg to complain regarding release
1332
	#      repo on first boot since packages are installed from
1333
	#      staging server during build phase
1334
	if [ -n "${USE_PKG_REPO_STAGING}" ]; then
1335
		_read_cmd="select value from repodata where key='packagesite'"
1336
		if [ -n "${_IS_RELEASE}" ]; then
1337
			local _tgt_server="${PKG_REPO_SERVER_RELEASE}"
1338
		else
1339
			local _tgt_server="${PKG_REPO_SERVER_DEVEL}"
1340
		fi
1341
		for _db in ${FINAL_CHROOT_DIR}/var/db/pkg/repo-*sqlite; do
1342
			_cur=$(/usr/local/bin/sqlite3 ${_db} "${_read_cmd}")
1343
			_new=$(echo "${_cur}" | sed -e "s,^${PKG_REPO_SERVER_STAGING},${_tgt_server},")
1344
			/usr/local/bin/sqlite3 ${_db} "update repodata set value='${_new}' where key='packagesite'"
1345
		done
1346
	fi
1347

    
1348
	if [ -n "$_image_variant" -a \
1349
	    -d ${BUILDER_TOOLS}/templates/custom_logos/${_image_variant} ]; then
1350
		mkdir -p ${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/custom_logos
1351
		cp -f \
1352
			${BUILDER_TOOLS}/templates/custom_logos/${_image_variant}/*.png \
1353
			${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/custom_logos
1354
	fi
1355
}
1356

    
1357
create_distribution_tarball() {
1358
	mkdir -p ${FINAL_CHROOT_DIR}/install
1359

    
1360
	tar -C ${FINAL_CHROOT_DIR} --exclude ./install --exclude ./pkgs -cJf ${FINAL_CHROOT_DIR}/install/${PRODUCT_NAME}.txz .
1361
}
1362

    
1363
create_iso_image() {
1364
	LOGFILE=${BUILDER_LOGS}/isoimage.${TARGET}
1365
	echo ">>> Building bootable ISO image for ${TARGET}" | tee -a ${LOGFILE}
1366
	if [ -z "${DEFAULT_KERNEL}" ]; then
1367
		echo ">>> ERROR: Could not identify DEFAULT_KERNEL to install on image!" | tee -a ${LOGFILE}
1368
		print_error_pfS
1369
	fi
1370

    
1371
	customize_stagearea_for_image "iso"
1372
	install_default_kernel ${DEFAULT_KERNEL}
1373

    
1374
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1375

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

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

    
1383
	create_distribution_tarball
1384

    
1385
	# Remove /rescue from iso since cd9660 cannot deal with hardlinks
1386
	rm -rf ${FINAL_CHROOT_DIR}/rescue
1387

    
1388
	makefs -t cd9660 -o bootimage="i386;${FINAL_CHROOT_DIR}/boot/cdboot" -o no-emul-boot -o rockridge \
1389
		-o label=${FSLABEL} -o publisher="${PRODUCT_NAME} project." $ISOPATH ${FINAL_CHROOT_DIR} 2>&1 >> ${LOGFILE}
1390
	if [ $? -ne 0 -o ! -f $ISOPATH ]; then
1391
		if [ -f ${ISOPATH} ]; then
1392
			rm -f $ISOPATH
1393
		fi
1394
		echo ">>> ERROR: Something wrong happened during ISO image creation. STOPPING!" | tee -a ${LOGFILE}
1395
		print_error_pfS
1396
	fi
1397
	gzip -qf $ISOPATH &
1398
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1399

    
1400
	echo ">>> ISO created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1401
}
1402

    
1403
create_memstick_image() {
1404
	local _variant="$1"
1405

    
1406
	LOGFILE=${BUILDER_LOGS}/memstick.${TARGET}
1407
	if [ "${MEMSTICKPATH}" = "" ]; then
1408
		echo ">>> MEMSTICKPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1409
		return
1410
	fi
1411

    
1412
	local _image_path=${MEMSTICKPATH}
1413
	if [ -n "${_variant}" ]; then
1414
		_image_path=$(echo "$_image_path" | \
1415
			sed "s/-memstick-/-memstick-${_variant}-/")
1416
		VARIANTIMAGES="${VARIANTIMAGES}${VARIANTIMAGES:+ }${_image_path}"
1417
	fi
1418

    
1419
	customize_stagearea_for_image "memstick" "" $_variant
1420
	install_default_kernel ${DEFAULT_KERNEL}
1421

    
1422
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1423

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

    
1428
	create_distribution_tarball
1429

    
1430
	makefs -B little -o label=${PRODUCT_NAME},version=2 ${_image_path} ${FINAL_CHROOT_DIR}
1431
	if [ $? -ne 0 ]; then
1432
		if [ -f ${_image_path} ]; then
1433
			rm -f $_image_path
1434
		fi
1435
		echo ">>> ERROR: Something wrong happened during MEMSTICK image creation. STOPPING!" | tee -a ${LOGFILE}
1436
		print_error_pfS
1437
	fi
1438
	MD=$(mdconfig -a -t vnode -f $_image_path)
1439
	# Just in case
1440
	trap "mdconfig -d -u ${MD}" 1 2 15 EXIT
1441
	gpart create -s BSD ${MD} 2>&1 >> ${LOGFILE}
1442
	gpart bootcode -b ${FINAL_CHROOT_DIR}/boot/boot ${MD} 2>&1 >> ${LOGFILE}
1443
	gpart add -t freebsd-ufs ${MD} 2>&1 >> ${LOGFILE}
1444
	trap "-" 1 2 15 EXIT
1445
	mdconfig -d -u ${MD} 2>&1 | tee -a ${LOGFILE}
1446
	gzip -qf $_image_path &
1447
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1448

    
1449
	echo ">>> MEMSTICK created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1450
}
1451

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

    
1459
	customize_stagearea_for_image "memstickserial"
1460
	install_default_kernel ${DEFAULT_KERNEL}
1461

    
1462
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1463

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

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

    
1469
	BOOTCONF=${FINAL_CHROOT_DIR}/boot.config
1470
	LOADERCONF=${FINAL_CHROOT_DIR}/boot/loader.conf
1471

    
1472
	echo ">>> Activating serial console..." 2>&1 | tee -a ${LOGFILE}
1473
	# Activate serial console in boot.config
1474
	if [ -f ${BOOTCONF} ]; then
1475
		sed -i "" '/-D$/d' ${BOOTCONF}
1476
	fi
1477
	echo "-S115200 -D" >> ${BOOTCONF}
1478

    
1479
	# Remove old console options if present.
1480
	[ -f "${LOADERCONF}" ] \
1481
		&& sed -i "" -Ee "/(console|boot_multicons|boot_serial)/d" ${LOADERCONF}
1482
	# Activate serial console+video console in loader.conf
1483
	echo 'boot_multicons="YES"' >>  ${LOADERCONF}
1484
	echo 'boot_serial="YES"' >> ${LOADERCONF}
1485
	echo 'console="comconsole,vidconsole"' >> ${LOADERCONF}
1486
	echo 'comconsole_speed="115200"' >> ${LOADERCONF}
1487

    
1488
	create_distribution_tarball
1489

    
1490
	makefs -B little -o label=${PRODUCT_NAME},version=2 ${MEMSTICKSERIALPATH} ${FINAL_CHROOT_DIR}
1491
	if [ $? -ne 0 ]; then
1492
		if [ -f ${MEMSTICKSERIALPATH} ]; then
1493
			rm -f $MEMSTICKSERIALPATH
1494
		fi
1495
		echo ">>> ERROR: Something wrong happened during MEMSTICKSERIAL image creation. STOPPING!" | tee -a ${LOGFILE}
1496
		print_error_pfS
1497
	fi
1498
	MD=$(mdconfig -a -t vnode -f $MEMSTICKSERIALPATH)
1499
	# Just in case
1500
	trap "mdconfig -d -u ${MD}" 1 2 15 EXIT
1501
	gpart create -s BSD ${MD} 2>&1 >> ${LOGFILE}
1502
	gpart bootcode -b ${FINAL_CHROOT_DIR}/boot/boot ${MD} 2>&1 >> ${LOGFILE}
1503
	gpart add -t freebsd-ufs ${MD} 2>&1 >> ${LOGFILE}
1504
	trap "-" 1 2 15 EXIT
1505
	mdconfig -d -u ${MD} 2>&1 >> ${LOGFILE}
1506
	gzip -qf $MEMSTICKSERIALPATH &
1507
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1508

    
1509
	echo ">>> MEMSTICKSERIAL created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1510
}
1511

    
1512
create_memstick_adi_image() {
1513
	LOGFILE=${BUILDER_LOGS}/memstickadi.${TARGET}
1514
	if [ "${MEMSTICKADIPATH}" = "" ]; then
1515
		echo ">>> MEMSTICKADIPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1516
		return
1517
	fi
1518

    
1519
	customize_stagearea_for_image "memstickadi"
1520
	install_default_kernel ${DEFAULT_KERNEL}
1521

    
1522
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1523

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

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

    
1529
	BOOTCONF=${FINAL_CHROOT_DIR}/boot.config
1530
	LOADERCONF=${FINAL_CHROOT_DIR}/boot/loader.conf
1531

    
1532
	echo ">>> Activating serial console..." 2>&1 | tee -a ${LOGFILE}
1533
	# Activate serial console in boot.config
1534
	if [ -f ${BOOTCONF} ]; then
1535
		sed -i "" '/-[Dh]$/d' ${BOOTCONF}
1536
	fi
1537
	echo "-S115200 -h" >> ${BOOTCONF}
1538

    
1539
	# Remove old console options if present.
1540
	[ -f "${LOADERCONF}" ] \
1541
		&& sed -i "" -Ee "/(console|boot_multicons|boot_serial|hint.uart)/d" ${LOADERCONF}
1542
	# Activate serial console+video console in loader.conf
1543
	echo 'boot_serial="YES"' >> ${LOADERCONF}
1544
	echo 'console="comconsole"' >> ${LOADERCONF}
1545
	echo 'comconsole_speed="115200"' >> ${LOADERCONF}
1546
	echo 'comconsole_port="0x2F8"' >> ${LOADERCONF}
1547
	echo 'hint.uart.0.flags="0x00"' >> ${LOADERCONF}
1548
	echo 'hint.uart.1.flags="0x10"' >> ${LOADERCONF}
1549

    
1550
	create_distribution_tarball
1551

    
1552
	makefs -B little -o label=${PRODUCT_NAME},version=2 ${MEMSTICKADIPATH} ${FINAL_CHROOT_DIR}
1553
	if [ $? -ne 0 ]; then
1554
		if [ -f ${MEMSTICKADIPATH} ]; then
1555
			rm -f $MEMSTICKADIPATH
1556
		fi
1557
		echo ">>> ERROR: Something wrong happened during MEMSTICKADI image creation. STOPPING!" | tee -a ${LOGFILE}
1558
		print_error_pfS
1559
	fi
1560
	MD=$(mdconfig -a -t vnode -f $MEMSTICKADIPATH)
1561
	# Just in case
1562
	trap "mdconfig -d -u ${MD}" 1 2 15 EXIT
1563
	gpart create -s BSD ${MD} 2>&1 >> ${LOGFILE}
1564
	gpart bootcode -b ${FINAL_CHROOT_DIR}/boot/boot ${MD} 2>&1 >> ${LOGFILE}
1565
	gpart add -t freebsd-ufs ${MD} 2>&1 >> ${LOGFILE}
1566
	trap "-" 1 2 15 EXIT
1567
	mdconfig -d -u ${MD} 2>&1 >> ${LOGFILE}
1568
	gzip -qf $MEMSTICKADIPATH &
1569
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1570

    
1571
	echo ">>> MEMSTICKADI created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1572
}
1573

    
1574
# Create pkg conf on desired place with desired arch/branch
1575
setup_pkg_repo() {
1576
	if [ -z "${4}" ]; then
1577
		return
1578
	fi
1579

    
1580
	local _template="${1}"
1581
	local _target="${2}"
1582
	local _arch="${3}"
1583
	local _target_arch="${4}"
1584
	local _staging="${5}"
1585

    
1586
	if [ -z "${_template}" -o ! -f "${_template}" ]; then
1587
		echo ">>> ERROR: It was not possible to find pkg conf template ${_template}"
1588
		print_error_pfS
1589
	fi
1590

    
1591
	if [ -n "${_staging}" -a -n "${USE_PKG_REPO_STAGING}" ]; then
1592
		local _pkg_repo_server_devel=${PKG_REPO_SERVER_STAGING}
1593
		local _pkg_repo_branch_devel=${PKG_REPO_BRANCH_STAGING}
1594
		local _pkg_repo_server_release=${PKG_REPO_SERVER_STAGING}
1595
		local _pkg_repo_branch_release=${PKG_REPO_BRANCH_STAGING}
1596
	else
1597
		local _pkg_repo_server_devel=${PKG_REPO_SERVER_DEVEL}
1598
		local _pkg_repo_branch_devel=${PKG_REPO_BRANCH_DEVEL}
1599
		local _pkg_repo_server_release=${PKG_REPO_SERVER_RELEASE}
1600
		local _pkg_repo_branch_release=${PKG_REPO_BRANCH_RELEASE}
1601
	fi
1602

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

    
1605
	sed \
1606
		-e "s/%%ARCH%%/${_target_arch}/" \
1607
		-e "s/%%PKG_REPO_BRANCH_DEVEL%%/${_pkg_repo_branch_devel}/g" \
1608
		-e "s/%%PKG_REPO_BRANCH_RELEASE%%/${_pkg_repo_branch_release}/g" \
1609
		-e "s,%%PKG_REPO_SERVER_DEVEL%%,${_pkg_repo_server_devel},g" \
1610
		-e "s,%%PKG_REPO_SERVER_RELEASE%%,${_pkg_repo_server_release},g" \
1611
		-e "s/%%PRODUCT_NAME%%/${PRODUCT_NAME}/g" \
1612
		${_template} \
1613
		> ${_target}
1614
}
1615

    
1616
# This routine ensures any ports / binaries that the builder
1617
# system needs are on disk and ready for execution.
1618
builder_setup() {
1619
	# If Product-builder is already installed, just leave
1620
	if pkg info -e -q ${PRODUCT_NAME}-builder; then
1621
		return
1622
	fi
1623

    
1624
	if [ ! -f ${PKG_REPO_PATH} ]; then
1625
		[ -d $(dirname ${PKG_REPO_PATH}) ] \
1626
			|| mkdir -p $(dirname ${PKG_REPO_PATH})
1627

    
1628
		update_freebsd_sources
1629

    
1630
		local _arch=$(uname -m)
1631
		setup_pkg_repo \
1632
			${PKG_REPO_DEFAULT} \
1633
			${PKG_REPO_PATH} \
1634
			${_arch} \
1635
			${_arch} \
1636
			"staging"
1637

    
1638
		# Use fingerprint keys from repo
1639
		sed -i '' -e "/fingerprints:/ s,\"/,\"${BUILDER_ROOT}/src/," \
1640
			${PKG_REPO_PATH}
1641
	fi
1642

    
1643
	pkg install ${PRODUCT_NAME}-builder
1644
}
1645

    
1646
# Updates FreeBSD sources
1647
update_freebsd_sources() {
1648
	if [ "${1}" = "full" ]; then
1649
		local _full=1
1650
		local _clone_params=""
1651
	else
1652
		local _full=0
1653
		local _clone_params="--depth 1 --single-branch"
1654
	fi
1655

    
1656
	if [ ! -d "${FREEBSD_SRC_DIR}" ]; then
1657
		mkdir -p ${FREEBSD_SRC_DIR}
1658
	fi
1659

    
1660
	if [ -n "${NO_BUILDWORLD}" -a -n "${NO_BUILDKERNEL}" ]; then
1661
		echo ">>> NO_BUILDWORLD and NO_BUILDKERNEL set, skipping update of freebsd sources" | tee -a ${LOGFILE}
1662
		return
1663
	fi
1664

    
1665
	echo -n ">>> Obtaining FreeBSD sources ${FREEBSD_BRANCH}..."
1666
	local _FREEBSD_BRANCH=${FREEBSD_BRANCH:-"devel"}
1667
	local _CLONE=1
1668

    
1669
	if [ -d "${FREEBSD_SRC_DIR}/.git" ]; then
1670
		CUR_BRANCH=$(cd ${FREEBSD_SRC_DIR} && git branch | grep '^\*' | cut -d' ' -f2)
1671
		if [ ${_full} -eq 0 -a "${CUR_BRANCH}" = "${_FREEBSD_BRANCH}" ]; then
1672
			_CLONE=0
1673
			( cd ${FREEBSD_SRC_DIR} && git clean -fd; git fetch origin; git reset --hard origin/${_FREEBSD_BRANCH} ) 2>&1 | grep -C3 -i -E 'error|fatal'
1674
		else
1675
			rm -rf ${FREEBSD_SRC_DIR}
1676
		fi
1677
	fi
1678

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

    
1683
	if [ ! -d "${FREEBSD_SRC_DIR}/.git" ]; then
1684
		echo ">>> ERROR: It was not possible to clone FreeBSD src repo"
1685
		print_error_pfS
1686
	fi
1687

    
1688
	if [ -n "${GIT_FREEBSD_COSHA1}" ]; then
1689
		( cd ${FREEBSD_SRC_DIR} && git checkout ${GIT_FREEBSD_COSHA1} ) 2>&1 | grep -C3 -i -E 'error|fatal'
1690
	fi
1691
	echo "Done!"
1692
}
1693

    
1694
pkg_chroot() {
1695
	local _root="${1}"
1696
	shift
1697

    
1698
	if [ $# -eq 0 ]; then
1699
		return -1
1700
	fi
1701

    
1702
	if [ -z "${_root}" -o "${_root}" = "/" -o ! -d "${_root}" ]; then
1703
		return -1
1704
	fi
1705

    
1706
	mkdir -p \
1707
		${SCRATCHDIR}/pkg_cache \
1708
		${_root}/var/cache/pkg \
1709
		${_root}/dev
1710

    
1711
	/sbin/mount -t nullfs ${SCRATCHDIR}/pkg_cache ${_root}/var/cache/pkg
1712
	/sbin/mount -t devfs devfs ${_root}/dev
1713
	cp -f /etc/resolv.conf ${_root}/etc/resolv.conf
1714
	touch ${BUILDER_LOGS}/install_pkg_install_ports.txt
1715
	script -aq ${BUILDER_LOGS}/install_pkg_install_ports.txt pkg -c ${_root} $@ >/dev/null 2>&1
1716
	local result=$?
1717
	rm -f ${_root}/etc/resolv.conf
1718
	/sbin/umount -f ${_root}/dev
1719
	/sbin/umount -f ${_root}/var/cache/pkg
1720

    
1721
	return $result
1722
}
1723

    
1724

    
1725
pkg_chroot_add() {
1726
	if [ -z "${1}" -o -z "${2}" ]; then
1727
		return 1
1728
	fi
1729

    
1730
	local _target="${1}"
1731
	local _pkg="$(get_pkg_name ${2}).txz"
1732

    
1733
	if [ ! -d "${_target}" ]; then
1734
		echo ">>> ERROR: Target dir ${_target} not found"
1735
		print_error_pfS
1736
	fi
1737

    
1738
	if [ ! -f ${CORE_PKG_ALL_PATH}/${_pkg} ]; then
1739
		echo ">>> ERROR: Package ${_pkg} not found"
1740
		print_error_pfS
1741
	fi
1742

    
1743
	cp ${CORE_PKG_ALL_PATH}/${_pkg} ${_target}
1744
	pkg_chroot ${_target} add /${_pkg}
1745
	rm -f ${_target}/${_pkg}
1746
}
1747

    
1748
pkg_bootstrap() {
1749
	local _root=${1:-"${STAGE_CHROOT_DIR}"}
1750

    
1751
	setup_pkg_repo \
1752
		${PKG_REPO_DEFAULT} \
1753
		${_root}${PKG_REPO_PATH} \
1754
		${TARGET} \
1755
		${TARGET_ARCH} \
1756
		"staging"
1757

    
1758
	pkg_chroot ${_root} bootstrap -f
1759
}
1760

    
1761
# This routine assists with installing various
1762
# freebsd ports files into the pfsense-fs staging
1763
# area.
1764
install_pkg_install_ports() {
1765
	local MAIN_PKG="${1}"
1766

    
1767
	if [ -z "${MAIN_PKG}" ]; then
1768
		MAIN_PKG=${PRODUCT_NAME}
1769
	fi
1770

    
1771
	echo ">>> Installing pkg repository in chroot (${STAGE_CHROOT_DIR})..."
1772

    
1773
	[ -d ${STAGE_CHROOT_DIR}/var/cache/pkg ] || \
1774
		mkdir -p ${STAGE_CHROOT_DIR}/var/cache/pkg
1775

    
1776
	[ -d ${SCRATCHDIR}/pkg_cache ] || \
1777
		mkdir -p ${SCRATCHDIR}/pkg_cache
1778

    
1779
	echo -n ">>> Installing built ports (packages) in chroot (${STAGE_CHROOT_DIR})... "
1780
	# First mark all packages as automatically installed
1781
	pkg_chroot ${STAGE_CHROOT_DIR} set -A 1 -a
1782
	# Install all necessary packages
1783
	if ! pkg_chroot ${STAGE_CHROOT_DIR} install ${MAIN_PKG} ${custom_package_list}; then
1784
		echo "Failed!"
1785
		print_error_pfS
1786
	fi
1787
	# Make sure required packages are set as non-automatic
1788
	pkg_chroot ${STAGE_CHROOT_DIR} set -A 0 pkg ${MAIN_PKG} ${custom_package_list}
1789
	# Remove unnecessary packages
1790
	pkg_chroot ${STAGE_CHROOT_DIR} autoremove
1791
	echo "Done!"
1792
}
1793

    
1794
install_bsdinstaller() {
1795
	local _params=""
1796

    
1797
	# Use staging repo on RELEASE
1798
	if [ -n "${_IS_RELEASE}" ]; then
1799
		mkdir -p ${FINAL_CHROOT_DIR}/tmp/pkg-repo
1800
		cp -f ${STAGE_CHROOT_DIR}${PKG_REPO_PATH} \
1801
			${FINAL_CHROOT_DIR}/tmp/pkg-repo
1802
		_params="--repo-conf-dir /tmp/pkg-repo "
1803
	fi
1804

    
1805
	echo ">>> Installing BSDInstaller in chroot (${FINAL_CHROOT_DIR})... (starting)"
1806
	pkg_chroot ${FINAL_CHROOT_DIR} ${_params}install -f bsdinstaller
1807
	sed -i '' -e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME}," \
1808
		  -e "s,%%PRODUCT_VERSION%%,${PRODUCT_VERSION}," \
1809
		  -e "s,%%ARCH%%,${TARGET}," \
1810
		  ${FINAL_CHROOT_DIR}/usr/local/share/dfuibe_lua/conf/pfSense.lua \
1811
		  ${FINAL_CHROOT_DIR}/usr/local/share/dfuibe_lua/conf/pfSense_rescue.lua
1812
	if [ -n "${_IS_RELEASE}" ]; then
1813
		rm -rf ${FINAL_CHROOT_DIR}/tmp/pkg-repo
1814
	fi
1815
	echo ">>> Installing BSDInstaller in chroot (${FINAL_CHROOT_DIR})... (finished)"
1816
}
1817

    
1818
staginareas_clean_each_run() {
1819
	echo -n ">>> Cleaning build directories: "
1820
	if [ -d "${FINAL_CHROOT_DIR}" ]; then
1821
		BASENAME=$(basename ${FINAL_CHROOT_DIR})
1822
		echo -n "$BASENAME "
1823
		chflags -R noschg ${FINAL_CHROOT_DIR} 2>&1 >/dev/null
1824
		rm -rf ${FINAL_CHROOT_DIR}/* 2>/dev/null
1825
	fi
1826
	echo "Done!"
1827
}
1828

    
1829
# Imported from FreeSBIE
1830
buildkernel() {
1831
	if [ -n "${NO_BUILDKERNEL}" ]; then
1832
		echo ">>> NO_BUILDKERNEL set, skipping build" | tee -a ${LOGFILE}
1833
		return
1834
	fi
1835

    
1836
	if [ -z "${KERNCONF}" ]; then
1837
		echo ">>> ERROR: No kernel configuration defined probably this is not what you want! STOPPING!" | tee -a ${LOGFILE}
1838
		print_error_pfS
1839
	fi
1840

    
1841
	if [ -n "${KERNELCONF}" ]; then
1842
		export KERNCONFDIR=$(dirname ${KERNELCONF})
1843
		export KERNCONF=$(basename ${KERNELCONF})
1844
	fi
1845

    
1846
	SRCCONFBASENAME=$(basename ${SRC_CONF})
1847
	echo ">>> KERNCONFDIR: ${KERNCONFDIR}"
1848
	echo ">>> ARCH:        ${TARGET}"
1849
	echo ">>> SRC_CONF:    ${SRCCONFBASENAME}"
1850

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

    
1856
# Imported from FreeSBIE
1857
installkernel() {
1858
	if [ -z "${KERNCONF}" ]; then
1859
		echo ">>> ERROR: No kernel configuration defined probably this is not what you want! STOPPING!" | tee -a ${LOGFILE}
1860
		print_error_pfS
1861
	fi
1862

    
1863
	if [ -n "${KERNELCONF}" ]; then
1864
		export KERNCONFDIR=$(dirname ${KERNELCONF})
1865
		export KERNCONF=$(basename ${KERNELCONF})
1866
	fi
1867

    
1868
	mkdir -p ${STAGE_CHROOT_DIR}/boot
1869
	makeargs="${MAKEJ_KERNEL} SRCCONF=${SRC_CONF} __MAKE_CONF=${MAKE_CONF} TARGET_ARCH=${TARGET_ARCH} TARGET=${TARGET} DESTDIR=${KERNEL_DESTDIR}"
1870
	echo ">>> Builder is running the command: script -aq $LOGFILE make ${makeargs} installkernel KERNCONF=${KERNCONF}"  | tee -a $LOGFILE
1871
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} installkernel KERNCONF=${KERNCONF} || print_error_pfS;) | egrep '^>>>'
1872
	gzip -f9 $KERNEL_DESTDIR/boot/kernel/kernel
1873
}
1874

    
1875
# Launch is ran first to setup a few variables that we need
1876
# Imported from FreeSBIE
1877
launch() {
1878
	if [ "$(id -u)" != "0" ]; then
1879
		echo "Sorry, this must be done as root."
1880
	fi
1881

    
1882
	echo ">>> Operation $0 has started at $(date)"
1883
}
1884

    
1885
finish() {
1886
	echo ">>> Operation $0 has ended at $(date)"
1887
}
1888

    
1889
pkg_repo_rsync() {
1890
	local _repo_path_param="${1}"
1891
	local _ignore_final_rsync="${2}"
1892

    
1893
	if [ -z "${_repo_path_param}" -o ! -d "${_repo_path_param}" ]; then
1894
		return
1895
	fi
1896

    
1897
	if [ -n "${SKIP_FINAL_RSYNC}" ]; then
1898
		_ignore_final_rsync="1"
1899
	fi
1900

    
1901
	# Sanitize path
1902
	_repo_path=$(realpath ${_repo_path_param})
1903

    
1904
	local _repo_dir=$(dirname ${_repo_path})
1905
	local _repo_base=$(basename ${_repo_path})
1906

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

    
1910
	if [ -z "${LOGFILE}" ]; then
1911
		local _logfile="/dev/null"
1912
	else
1913
		local _logfile="${LOGFILE}"
1914
	fi
1915

    
1916
	if [ -n "${PKG_REPO_SIGNING_COMMAND}" ]; then
1917

    
1918
		# Detect poudriere directory structure
1919
		if [ -L "${_repo_path}/.latest" ]; then
1920
			local _real_repo_path=$(readlink -f ${_repo_path}/.latest)
1921
		else
1922
			local _real_repo_path=${_repo_path}
1923
		fi
1924

    
1925
		echo -n ">>> Signing repository... " | tee -a ${_logfile}
1926
		############ ATTENTION ##############
1927
		#
1928
		# For some reason pkg-repo fail without / in the end of directory name
1929
		# so removing it will break command
1930
		#
1931
		# https://github.com/freebsd/pkg/issues/1364
1932
		#
1933
		if script -aq ${_logfile} pkg repo ${_real_repo_path}/ \
1934
		    signing_command: ${PKG_REPO_SIGNING_COMMAND} >/dev/null 2>&1; then
1935
			echo "Done!" | tee -a ${_logfile}
1936
		else
1937
			echo "Failed!" | tee -a ${_logfile}
1938
			echo ">>> ERROR: An error occurred trying to sign repo"
1939
			print_error_pfS
1940
		fi
1941

    
1942
		local _pkgfile="${_repo_path}/Latest/pkg.txz"
1943
		if [ -e ${_pkgfile} ]; then
1944
			echo -n ">>> Signing Latest/pkg.txz for bootstraping... " | tee -a ${_logfile}
1945

    
1946
			if sha256 -q ${_pkgfile} | ${PKG_REPO_SIGNING_COMMAND} \
1947
			    > ${_pkgfile}.sig 2>/dev/null; then
1948
				echo "Done!" | tee -a ${_logfile}
1949
			else
1950
				echo "Failed!" | tee -a ${_logfile}
1951
				echo ">>> ERROR: An error occurred trying to sign Latest/pkg.txz"
1952
				print_error_pfS
1953
			fi
1954
		fi
1955
	fi
1956

    
1957
	if [ -n "${DO_NOT_UPLOAD}" ]; then
1958
		return
1959
	fi
1960

    
1961
	# Make sure destination directory exist
1962
	ssh -p ${PKG_RSYNC_SSH_PORT} \
1963
		${PKG_RSYNC_USERNAME}@${PKG_RSYNC_HOSTNAME} \
1964
		"mkdir -p ${PKG_RSYNC_DESTDIR}"
1965

    
1966
	echo -n ">>> Sending updated repository to ${PKG_RSYNC_HOSTNAME}... " | tee -a ${_logfile}
1967
	if script -aq ${_logfile} rsync -Have "ssh -p ${PKG_RSYNC_SSH_PORT}" \
1968
		--timeout=60 --delete-delay ${_repo_path} \
1969
		${PKG_RSYNC_USERNAME}@${PKG_RSYNC_HOSTNAME}:${PKG_RSYNC_DESTDIR} >/dev/null 2>&1
1970
	then
1971
		echo "Done!" | tee -a ${_logfile}
1972
	else
1973
		echo "Failed!" | tee -a ${_logfile}
1974
		echo ">>> ERROR: An error occurred sending repo to remote hostname"
1975
		print_error_pfS
1976
	fi
1977

    
1978
	if [ -z "${USE_PKG_REPO_STAGING}" -o -n "${_ignore_final_rsync}" ]; then
1979
		return
1980
	fi
1981

    
1982
	if [ -n "${_IS_RELEASE}" -o "${_repo_path_param}" = "${CORE_PKG_PATH}" ]; then
1983
		# Send .real* directories first to prevent having a broken repo while transfer happens
1984
		local _cmd="rsync -Have \"ssh -p ${PKG_FINAL_RSYNC_SSH_PORT}\" \
1985
			--timeout=60 ${PKG_RSYNC_DESTDIR}/./${_repo_base%%-core}* \
1986
			--include=\"/*\" --include=\"*/.real*\" --include=\"*/.real*/*\" \
1987
			--exclude=\"*\" \
1988
			${PKG_FINAL_RSYNC_USERNAME}@${PKG_FINAL_RSYNC_HOSTNAME}:${PKG_FINAL_RSYNC_DESTDIR}"
1989

    
1990
		echo -n ">>> Sending updated packages to ${PKG_FINAL_RSYNC_HOSTNAME}... " | tee -a ${_logfile}
1991
		if script -aq ${_logfile} ssh -p ${PKG_RSYNC_SSH_PORT} \
1992
			${PKG_RSYNC_USERNAME}@${PKG_RSYNC_HOSTNAME} ${_cmd} >/dev/null 2>&1; then
1993
			echo "Done!" | tee -a ${_logfile}
1994
		else
1995
			echo "Failed!" | tee -a ${_logfile}
1996
			echo ">>> ERROR: An error occurred sending repo to final hostname"
1997
			print_error_pfS
1998
		fi
1999

    
2000
		_cmd="rsync -Have \"ssh -p ${PKG_FINAL_RSYNC_SSH_PORT}\" \
2001
			--timeout=60 --delete-delay ${PKG_RSYNC_DESTDIR}/./${_repo_base%%-core}* \
2002
			${PKG_FINAL_RSYNC_USERNAME}@${PKG_FINAL_RSYNC_HOSTNAME}:${PKG_FINAL_RSYNC_DESTDIR}"
2003

    
2004
		echo -n ">>> Sending updated repositories metadata to ${PKG_FINAL_RSYNC_HOSTNAME}... " | tee -a ${_logfile}
2005
		if script -aq ${_logfile} ssh -p ${PKG_RSYNC_SSH_PORT} \
2006
			${PKG_RSYNC_USERNAME}@${PKG_RSYNC_HOSTNAME} ${_cmd} >/dev/null 2>&1; then
2007
			echo "Done!" | tee -a ${_logfile}
2008
		else
2009
			echo "Failed!" | tee -a ${_logfile}
2010
			echo ">>> ERROR: An error occurred sending repo to final hostname"
2011
			print_error_pfS
2012
		fi
2013
	fi
2014
}
2015

    
2016
poudriere_create_patch() {
2017
	local _jail_patch="${SCRATCHDIR}/poudriere_jail.${POUDRIERE_BRANCH}.patch"
2018

    
2019
	if [ -z "${FREEBSD_PARENT_BRANCH}" ]; then
2020
		echo ">>> ERROR: FREEBSD_PARENT_BRANCH is not set"
2021
	fi
2022

    
2023
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2024

    
2025
	# Get FreeBSD source and apply patches
2026
	update_freebsd_sources full
2027

    
2028
	[ -f "${_jail_patch}" ] && \
2029
		rm -f "${_jail_patch}"
2030

    
2031
	# Create a big patch with all our changes to use on jail
2032
	( \
2033
		cd ${FREEBSD_SRC_DIR} && \
2034
		git diff $(git merge-base origin/${FREEBSD_PARENT_BRANCH} ${FREEBSD_BRANCH}) > ${_jail_patch}
2035
	) >/dev/null 2>&1
2036

    
2037
	# Check if patch was created
2038
	if [ ! -s "${_jail_patch}" ]; then
2039
		echo ">>> ERROR: Patch does not exist or is empty, aborting..." | tee -a ${LOGFILE}
2040
		print_error_pfS
2041
	fi
2042
}
2043

    
2044
poudriere_possible_archs() {
2045
	local _arch=$(uname -m)
2046
	local _archs="i386.i386"
2047

    
2048
	# If host is amd64, we'll create both repos, and if possible armv6
2049
	if [ "${_arch}" = "amd64" ]; then
2050
		_archs="amd64.amd64 ${_archs}"
2051

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

    
2056
			if binmiscctl lookup armv6 >/dev/null 2>&1; then
2057
				_archs="${_archs} arm.armv6"
2058
			fi
2059
		fi
2060
	fi
2061

    
2062
	if [ -n "${ARCH_LIST}" ]; then
2063
		local _found=0
2064
		for _desired_arch in ${ARCH_LIST}; do
2065
			_found=0
2066
			for _possible_arch in ${_archs}; do
2067
				if [ "${_desired_arch}" = "${_possible_arch}" ]; then
2068
					_found=1
2069
					break
2070
				fi
2071
			done
2072
			if [ ${_found} -eq 0 ]; then
2073
				echo ">>> ERROR: Impossible to build for arch: ${_desired_arch}"
2074
				print_error_pfS
2075
			fi
2076
		done
2077
		_archs="${ARCH_LIST}"
2078
	fi
2079

    
2080
	echo ${_archs}
2081
}
2082

    
2083
poudriere_jail_name() {
2084
	local _jail_arch="${1}"
2085

    
2086
	if [ -z "${_jail_arch}" ]; then
2087
		return 1
2088
	fi
2089

    
2090
	# Remove arch
2091
	echo "${PRODUCT_NAME}_${POUDRIERE_BRANCH}_${_jail_arch##*.}"
2092
}
2093

    
2094
poudriere_rename_ports() {
2095
	if [ "${PRODUCT_NAME}" = "pfSense" ]; then
2096
		return;
2097
	fi
2098

    
2099
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2100

    
2101
	local _ports_dir="/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}"
2102

    
2103
	echo -n ">>> Renaming product ports on ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
2104
	for d in $(find ${_ports_dir} -depth 2 -type d -name '*pfSense*'); do
2105
		local _pdir=$(dirname ${d})
2106
		local _pname=$(echo $(basename ${d}) | sed "s,pfSense,${PRODUCT_NAME},")
2107
		local _plist=""
2108

    
2109
		if [ -e ${_pdir}/${_pname} ]; then
2110
			rm -rf ${_pdir}/${_pname}
2111
		fi
2112

    
2113
		cp -r ${d} ${_pdir}/${_pname}
2114

    
2115
		if [ -f ${_pdir}/${_pname}/pkg-plist ]; then
2116
			_plist=${_pdir}/${_pname}/pkg-plist
2117
		fi
2118

    
2119
		sed -i '' -e "s,pfSense,${PRODUCT_NAME},g" \
2120
			  -e "s,https://www.pfsense.org,${PRODUCT_URL},g" \
2121
			  -e "/^MAINTAINER=/ s,^.*$,MAINTAINER=	${PRODUCT_EMAIL}," \
2122
			${_pdir}/${_pname}/Makefile \
2123
			${_pdir}/${_pname}/pkg-descr ${_plist}
2124

    
2125
		# PHP module is special
2126
		if echo "${_pname}" | grep -q "^php[0-9]*-${PRODUCT_NAME}-module"; then
2127
			local _product_capital=$(echo ${PRODUCT_NAME} | tr '[a-z]' '[A-Z]')
2128
			sed -i '' -e "s,PHP_PFSENSE,PHP_${_product_capital},g" \
2129
				  -e "s,PFSENSE_SHARED_LIBADD,${_product_capital}_SHARED_LIBADD,g" \
2130
				  -e "s,pfSense,${PRODUCT_NAME},g" \
2131
				  -e "s,${PRODUCT_NAME}\.c,pfSense.c,g" \
2132
				${_pdir}/${_pname}/files/config.m4
2133

    
2134
			sed -i '' -e "s,COMPILE_DL_PFSENSE,COMPILE_DL_${_product_capital}," \
2135
				  -e "s,pfSense_module_entry,${PRODUCT_NAME}_module_entry,g" \
2136
				  -e "/ZEND_GET_MODULE/ s,pfSense,${PRODUCT_NAME}," \
2137
				  -e "/PHP_PFSENSE_WORLD_EXTNAME/ s,pfSense,${PRODUCT_NAME}," \
2138
				${_pdir}/${_pname}/files/pfSense.c \
2139
				${_pdir}/${_pname}/files/php_pfSense.h
2140
		fi
2141

    
2142
		if [ -d ${_pdir}/${_pname}/files ]; then
2143
			for fd in $(find ${_pdir}/${_pname}/files -type d -name '*pfSense*'); do
2144
				local _fddir=$(dirname ${fd})
2145
				local _fdname=$(echo $(basename ${fd}) | sed "s,pfSense,${PRODUCT_NAME},")
2146

    
2147
				mv ${fd} ${_fddir}/${_fdname}
2148
			done
2149
		fi
2150
	done
2151
	echo "Done!" | tee -a ${LOGFILE}
2152
}
2153

    
2154
poudriere_create_ports_tree() {
2155
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2156

    
2157
	if ! poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
2158
		local _branch=""
2159
		if [ -z "${POUDRIERE_PORTS_GIT_URL}" ]; then
2160
			echo ">>> ERROR: POUDRIERE_PORTS_GIT_URL is not defined"
2161
			print_error_pfS
2162
		fi
2163
		if [ -n "${POUDRIERE_PORTS_GIT_BRANCH}" ]; then
2164
			_branch="-B ${POUDRIERE_PORTS_GIT_BRANCH}"
2165
		fi
2166
		echo -n ">>> Creating poudriere ports tree, it may take some time... " | tee -a ${LOGFILE}
2167
		if ! script -aq ${LOGFILE} poudriere ports -c -p "${POUDRIERE_PORTS_NAME}" -m git ${_branch} >/dev/null 2>&1; then
2168
			echo "" | tee -a ${LOGFILE}
2169
			echo ">>> ERROR: Error creating poudriere ports tree, aborting..." | tee -a ${LOGFILE}
2170
			print_error_pfS
2171
		fi
2172
		echo "Done!" | tee -a ${LOGFILE}
2173
		poudriere_rename_ports
2174
	fi
2175
}
2176

    
2177
poudriere_init() {
2178
	local _error=0
2179
	local _archs=$(poudriere_possible_archs)
2180
	local _jail_patch="${SCRATCHDIR}/poudriere_jail.${POUDRIERE_BRANCH}.patch"
2181

    
2182
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2183

    
2184
	# Sanity checks
2185
	if [ -z "${ZFS_TANK}" ]; then
2186
		echo ">>> ERROR: \$ZFS_TANK is empty" | tee -a ${LOGFILE}
2187
		error=1
2188
	fi
2189

    
2190
	if [ -z "${ZFS_ROOT}" ]; then
2191
		echo ">>> ERROR: \$ZFS_ROOT is empty" | tee -a ${LOGFILE}
2192
		error=1
2193
	fi
2194

    
2195
	if [ -z "${POUDRIERE_PORTS_NAME}" ]; then
2196
		echo ">>> ERROR: \$POUDRIERE_PORTS_NAME is empty" | tee -a ${LOGFILE}
2197
		error=1
2198
	fi
2199

    
2200
	if [ ${_error} -eq 1 ]; then
2201
		print_error_pfS
2202
	fi
2203

    
2204
	# Check if zpool exists
2205
	if ! zpool list ${ZFS_TANK} >/dev/null 2>&1; then
2206
		echo ">>> ERROR: ZFS tank ${ZFS_TANK} not found, please create it and try again..." | tee -a ${LOGFILE}
2207
		print_error_pfS
2208
	fi
2209

    
2210
	# Check if zfs rootfs exists
2211
	if ! zfs list ${ZFS_TANK}${ZFS_ROOT} >/dev/null 2>&1; then
2212
		echo -n ">>> Creating ZFS filesystem ${ZFS_TANK}${ZFS_ROOT}... "
2213
		if zfs create -o atime=off -o mountpoint=/usr/local${ZFS_ROOT} \
2214
		    ${ZFS_TANK}${ZFS_ROOT} >/dev/null 2>&1; then
2215
			echo "Done!"
2216
		else
2217
			echo "Failed!"
2218
			print_error_pfS
2219
		fi
2220
	fi
2221

    
2222
	# Make sure poudriere is installed
2223
	if ! pkg info --quiet poudriere; then
2224
		echo ">>> Installing poudriere..." | tee -a ${LOGFILE}
2225
		if ! pkg install poudriere >/dev/null 2>&1; then
2226
			echo ">>> ERROR: poudriere was not installed, aborting..." | tee -a ${LOGFILE}
2227
			print_error_pfS
2228
		fi
2229
	fi
2230

    
2231
	# Create poudriere.conf
2232
	if [ -z "${POUDRIERE_PORTS_GIT_URL}" ]; then
2233
		echo ">>> ERROR: POUDRIERE_PORTS_GIT_URL is not defined"
2234
		print_error_pfS
2235
	fi
2236
	echo ">>> Creating poudriere.conf" | tee -a ${LOGFILE}
2237
	cat <<EOF >/usr/local/etc/poudriere.conf
2238
ZPOOL=${ZFS_TANK}
2239
ZROOTFS=${ZFS_ROOT}
2240
RESOLV_CONF=/etc/resolv.conf
2241
BASEFS=/usr/local/poudriere
2242
USE_PORTLINT=no
2243
USE_TMPFS=yes
2244
NOLINUX=yes
2245
DISTFILES_CACHE=/usr/ports/distfiles
2246
CHECK_CHANGED_OPTIONS=yes
2247
CHECK_CHANGED_DEPS=yes
2248
ATOMIC_PACKAGE_REPOSITORY=yes
2249
COMMIT_PACKAGES_ON_FAILURE=no
2250
KEEP_OLD_PACKAGES=yes
2251
KEEP_OLD_PACKAGES_COUNT=5
2252
EOF
2253

    
2254
	# Create specific items conf
2255
	[ ! -d /usr/local/etc/poudriere.d ] \
2256
		&& mkdir -p /usr/local/etc/poudriere.d
2257

    
2258
	cat <<EOF >/usr/local/etc/poudriere.d/${POUDRIERE_PORTS_NAME}-poudriere.conf
2259
GIT_URL="${POUDRIERE_PORTS_GIT_URL}"
2260
EOF
2261

    
2262
	# Create DISTFILES_CACHE if it doesn't exist
2263
	if [ ! -d /usr/ports/distfiles ]; then
2264
		mkdir -p /usr/ports/distfiles
2265
	fi
2266

    
2267
	# Remove old jails
2268
	for jail_arch in ${_archs}; do
2269
		jail_name=$(poudriere_jail_name ${jail_arch})
2270

    
2271
		if poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
2272
			echo ">>> Poudriere jail ${jail_name} already exists, deleting it..." | tee -a ${LOGFILE}
2273
			poudriere jail -d -j "${jail_name}" >/dev/null 2>&1
2274
		fi
2275
	done
2276

    
2277
	# Remove old ports tree
2278
	if poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
2279
		echo ">>> Poudriere ports tree ${POUDRIERE_PORTS_NAME} already exists, deleting it..." | tee -a ${LOGFILE}
2280
		poudriere ports -d -p "${POUDRIERE_PORTS_NAME}"
2281
	fi
2282

    
2283
	poudriere_create_patch
2284

    
2285
	local native_xtools=""
2286
	# Now we are ready to create jails
2287
	for jail_arch in ${_archs}; do
2288
		jail_name=$(poudriere_jail_name ${jail_arch})
2289

    
2290
		if [ "${jail_arch}" = "arm.armv6" ]; then
2291
			native_xtools="-x"
2292
		else
2293
			native_xtools=""
2294
		fi
2295

    
2296
		echo -n ">>> Creating jail ${jail_name}, it may take some time... " | tee -a ${LOGFILE}
2297
		# XXX: Change -m to git when it's available in poudriere
2298
		if ! script -aq ${LOGFILE} poudriere jail -c -j "${jail_name}" -v ${FREEBSD_PARENT_BRANCH} \
2299
				-a ${jail_arch} -m svn -P ${_jail_patch} ${native_xtools} >/dev/null 2>&1; then
2300
			echo "" | tee -a ${LOGFILE}
2301
			echo ">>> ERROR: Error creating jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
2302
			print_error_pfS
2303
		fi
2304
		echo "Done!" | tee -a ${LOGFILE}
2305
	done
2306

    
2307
	poudriere_create_ports_tree
2308

    
2309
	echo ">>> Poudriere is now configured!" | tee -a ${LOGFILE}
2310
}
2311

    
2312
poudriere_update_jails() {
2313
	local _archs=$(poudriere_possible_archs)
2314
	local _jail_patch="${SCRATCHDIR}/poudriere_jail.${POUDRIERE_BRANCH}.patch"
2315

    
2316
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2317

    
2318
	poudriere_create_patch
2319

    
2320
	local native_xtools=""
2321
	for jail_arch in ${_archs}; do
2322
		jail_name=$(poudriere_jail_name ${jail_arch})
2323

    
2324
		local _create_or_update="-u"
2325
		local _create_or_update_text="Updating"
2326
		if ! poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
2327
			echo ">>> Poudriere jail ${jail_name} not found, creating..." | tee -a ${LOGFILE}
2328
			_create_or_update="-c -v ${FREEBSD_PARENT_BRANCH} -a ${jail_arch} -m svn"
2329
			_create_or_update_text="Creating"
2330
		fi
2331

    
2332
		if [ "${jail_arch}" = "arm.armv6" ]; then
2333
			native_xtools="-x"
2334
		else
2335
			native_xtools=""
2336
		fi
2337

    
2338
		echo -n ">>> ${_create_or_update_text} jail ${jail_name}, it may take some time... " | tee -a ${LOGFILE}
2339
		if ! script -aq ${LOGFILE} poudriere jail ${_create_or_update} -j "${jail_name}" -P ${_jail_patch} ${native_xtools} >/dev/null 2>&1; then
2340
			echo "" | tee -a ${LOGFILE}
2341
			echo ">>> ERROR: Error ${_create_or_update_text} jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
2342
			print_error_pfS
2343
		fi
2344
		echo "Done!" | tee -a ${LOGFILE}
2345
	done
2346
}
2347

    
2348
poudriere_update_ports() {
2349
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2350

    
2351
	# Create ports tree if necessary
2352
	if ! poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
2353
		poudriere_create_ports_tree
2354
	else
2355
		echo -n ">>> Resetting local changes on ports tree ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
2356
		script -aq ${LOGFILE} git -C "/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}" reset --hard >/dev/null 2>&1
2357
		script -aq ${LOGFILE} git -C "/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}" clean -fd >/dev/null 2>&1
2358
		echo "Done!" | tee -a ${LOGFILE}
2359
		echo -n ">>> Updating ports tree ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
2360
		script -aq ${LOGFILE} poudriere ports -u -p "${POUDRIERE_PORTS_NAME}" >/dev/null 2>&1
2361
		echo "Done!" | tee -a ${LOGFILE}
2362
		poudriere_rename_ports
2363
	fi
2364
}
2365

    
2366
poudriere_bulk() {
2367
	local _archs=$(poudriere_possible_archs)
2368

    
2369
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2370

    
2371
	if [ -z "${DO_NOT_UPLOAD}" -a -z "${PKG_RSYNC_HOSTNAME}" ]; then
2372
		echo ">>> ERROR: PKG_RSYNC_HOSTNAME is not set"
2373
		print_error_pfS
2374
	fi
2375

    
2376
	rm -f ${LOGFILE}
2377

    
2378
	poudriere_create_ports_tree
2379

    
2380
	[ -d /usr/local/etc/poudriere.d ] || \
2381
		mkdir -p /usr/local/etc/poudriere.d
2382

    
2383
	if [ -f "${BUILDER_TOOLS}/conf/pfPorts/make.conf" ]; then
2384
		cp -f "${BUILDER_TOOLS}/conf/pfPorts/make.conf" /usr/local/etc/poudriere.d/${POUDRIERE_PORTS_NAME}-make.conf
2385
	fi
2386

    
2387
	# Change version of pfSense meta ports for snapshots
2388
	if [ -z "${_IS_RELEASE}" ]; then
2389
		local _meta_pkg_version="$(echo "${PRODUCT_VERSION}" | sed 's,DEVELOPMENT,ALPHA,')-${DATESTRING}"
2390
		sed -i '' \
2391
			-e "/^DISTVERSION/ s,^.*,DISTVERSION=	${_meta_pkg_version}," \
2392
			-e "/^PORTREVISION=/d" \
2393
			/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}/security/${PRODUCT_NAME}/Makefile
2394
	fi
2395

    
2396
	for jail_arch in ${_archs}; do
2397
		jail_name=$(poudriere_jail_name ${jail_arch})
2398

    
2399
		if ! poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
2400
			echo ">>> Poudriere jail ${jail_name} not found, skipping..." | tee -a ${LOGFILE}
2401
			continue
2402
		fi
2403

    
2404
		if [ -f "${POUDRIERE_BULK}.${jail_arch}" ]; then
2405
			_ref_bulk="${POUDRIERE_BULK}.${jail_arch}"
2406
		else
2407
			_ref_bulk="${POUDRIERE_BULK}"
2408
		fi
2409

    
2410
		_bulk=${SCRATCHDIR}/poudriere_bulk.${POUDRIERE_BRANCH}
2411
		sed -e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" ${_ref_bulk} > ${_bulk}
2412

    
2413
		if ! poudriere bulk -f ${_bulk} -j ${jail_name} -p ${POUDRIERE_PORTS_NAME}; then
2414
			echo ">>> ERROR: Something went wrong..."
2415
			print_error_pfS
2416
		fi
2417

    
2418
		echo ">>> Cleaning up old packages from repo..."
2419
		if ! poudriere pkgclean -f ${_bulk} -j ${jail_name} -p ${POUDRIERE_PORTS_NAME} -y; then
2420
			echo ">>> ERROR: Something went wrong..."
2421
			print_error_pfS
2422
		fi
2423

    
2424
		pkg_repo_rsync "/usr/local/poudriere/data/packages/${jail_name}-${POUDRIERE_PORTS_NAME}"
2425
	done
2426
}
2427

    
2428
# This routine is called to write out to stdout
2429
# a string. The string is appended to $SNAPSHOTSLOGFILE
2430
# and we scp the log file to the builder host if
2431
# needed for the real time logging functions.
2432
snapshots_update_status() {
2433
	if [ -z "$1" ]; then
2434
		return
2435
	fi
2436
	if [ -z "${SNAPSHOTS}" -a -z "${POUDRIERE_SNAPSHOTS}" ]; then
2437
		return
2438
	fi
2439
	echo "$*"
2440
	echo "`date` -|- $*" >> $SNAPSHOTSLOGFILE
2441
	if [ -z "${DO_NOT_UPLOAD}" -a -n "${SNAPSHOTS_RSYNCIP}" ]; then
2442
		LU=$(cat $SNAPSHOTSLASTUPDATE 2>/dev/null)
2443
		CT=$(date "+%H%M%S")
2444
		# Only update every minute
2445
		if [ "$LU" != "$CT" ]; then
2446
			ssh ${SNAPSHOTS_RSYNCUSER}@${SNAPSHOTS_RSYNCIP} \
2447
				"mkdir -p ${SNAPSHOTS_RSYNCLOGS}"
2448
			scp -q $SNAPSHOTSLOGFILE \
2449
				${SNAPSHOTS_RSYNCUSER}@${SNAPSHOTS_RSYNCIP}:${SNAPSHOTS_RSYNCLOGS}/build.log
2450
			date "+%H%M%S" > $SNAPSHOTSLASTUPDATE
2451
		fi
2452
	fi
2453
}
2454

    
2455
# Copy the current log file to $filename.old on
2456
# the snapshot www server (real time logs)
2457
snapshots_rotate_logfile() {
2458
	if [ -z "${DO_NOT_UPLOAD}" -a -n "${SNAPSHOTS_RSYNCIP}" ]; then
2459
		scp -q $SNAPSHOTSLOGFILE \
2460
			${SNAPSHOTS_RSYNCUSER}@${SNAPSHOTS_RSYNCIP}:${SNAPSHOTS_RSYNCLOGS}/build.log.old
2461
	fi
2462

    
2463
	# Cleanup log file
2464
	rm -f $SNAPSHOTSLOGFILE;    touch $SNAPSHOTSLOGFILE
2465
	rm -f $SNAPSHOTSLASTUPDATE; touch $SNAPSHOTSLASTUPDATE
2466

    
2467
}
2468

    
2469
create_sha256() {
2470
	local _file="${1}"
2471

    
2472
	if [ ! -f "${_file}" ]; then
2473
		return 1
2474
	fi
2475

    
2476
	( \
2477
		cd $(dirname ${_file}) && \
2478
		sha256 $(basename ${_file}) > $(basename ${_file}).sha256 \
2479
	)
2480
}
2481

    
2482
snapshots_create_latest_symlink() {
2483
	local _image="${1}"
2484

    
2485
	if [ -z "${_image}" ]; then
2486
		return
2487
	fi
2488

    
2489
	if [ -z "${TIMESTAMP_SUFFIX}" ]; then
2490
		return
2491
	fi
2492

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

    
2497
	local _symlink=$(echo ${_image} | sed "s,${TIMESTAMP_SUFFIX},-latest,")
2498
	ln -sf $(basename ${_image}) ${_symlink}
2499
	ln -sf $(basename ${_image}).sha256 ${_symlink}.sha256
2500
}
2501

    
2502
snapshots_copy_to_staging_nanobsd() {
2503
	for NANOTYPE in nanobsd nanobsd-vga; do
2504
		for FILESIZE in ${1}; do
2505
			FILENAMEFULL="$(nanobsd_image_filename ${FILESIZE} ${NANOTYPE}).gz"
2506
			FILENAMEUPGRADE="$(nanobsd_image_filename ${FILESIZE} ${NANOTYPE} 1).gz"
2507
			mkdir -p $STAGINGAREA/nanobsd
2508
			mkdir -p $STAGINGAREA/nanobsdupdates
2509

    
2510
			cp -l $IMAGES_FINAL_DIR/$FILENAMEFULL $STAGINGAREA/nanobsd/ 2>/dev/null
2511
			cp -l $IMAGES_FINAL_DIR/$FILENAMEUPGRADE $STAGINGAREA/nanobsdupdates 2>/dev/null
2512

    
2513
			if [ -f $STAGINGAREA/nanobsd/$FILENAMEFULL ]; then
2514
				create_sha256 $STAGINGAREA/nanobsd/$FILENAMEFULL
2515
			fi
2516
			if [ -f $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE ]; then
2517
				create_sha256 $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE
2518
			fi
2519

    
2520
			# Copy NanoBSD auto update:
2521
			if [ -f $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE ]; then
2522
				cp -l $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE $STAGINGAREA/latest-${NANOTYPE}-$FILESIZE.img.gz 2>/dev/null
2523
				create_sha256 $STAGINGAREAA/latest-${NANOTYPE}-$FILESIZE.img.gz
2524
				# NOTE: Updates need a file with output similar to date output
2525
				# Use the file generated at start of snapshots_dobuilds() to be consistent on times
2526
				cp $BUILTDATESTRINGFILE $STAGINGAREA/version-${NANOTYPE}-$FILESIZE
2527
			fi
2528
		done
2529
	done
2530
}
2531

    
2532
snapshots_copy_to_staging_iso_updates() {
2533
	local _img=""
2534

    
2535
	for _img in ${ISOPATH} ${MEMSTICKPATH} ${MEMSTICKSERIALPATH} ${MEMSTICKADIPATH} ${VARIANTIMAGES}; do
2536
		if [ ! -f "${_img}.gz" ]; then
2537
			continue
2538
		fi
2539
		_img="${_img}.gz"
2540
		create_sha256 ${_img}
2541
		cp -l ${_img}* $STAGINGAREA/ 2>/dev/null
2542
		snapshots_create_latest_symlink ${STAGINGAREA}/$(basename ${_img})
2543
	done
2544

    
2545
	for _img in ${UPDATES_TARBALL_FILENAME} ${VARIANTUPDATES}; do
2546
		if [ -f "${_img}" ]; then
2547
			create_sha256 ${_img}
2548
			cp -l ${_img}* $STAGINGAREA/ 2>/dev/null
2549
			snapshots_create_latest_symlink ${STAGINGAREA}/$(basename ${_img})
2550
		fi
2551
	done
2552

    
2553
	if [ -f "${OVAPATH}" ]; then
2554
		mkdir -p ${STAGINGAREA}/virtualization
2555
		create_sha256 ${OVAPATH}
2556
		cp -l ${OVAPATH}* $STAGINGAREA/virtualization 2>/dev/null
2557
		snapshots_create_latest_symlink ${STAGINGAREA}/virtualization/$(basename ${OVAPATH})
2558
	fi
2559

    
2560
	# NOTE: Updates need a file with output similar to date output
2561
	# Use the file generated at start of snapshots_dobuilds() to be consistent on times
2562
	if [ -z "${_IS_RELEASE}" ]; then
2563
		cp $BUILTDATESTRINGFILE $STAGINGAREA/version 2>/dev/null
2564
	fi
2565
}
2566

    
2567
snapshots_scp_files() {
2568
	if [ -z "${RSYNC_COPY_ARGUMENTS}" ]; then
2569
		RSYNC_COPY_ARGUMENTS="-ave ssh --timeout=60 --bwlimit=${RSYNCKBYTELIMIT}" #--bwlimit=50
2570
	fi
2571

    
2572
	snapshots_update_status ">>> Copying core pkg repo to ${PKG_RSYNC_HOSTNAME}"
2573
	pkg_repo_rsync "${CORE_PKG_PATH}"
2574
	snapshots_update_status ">>> Finished copying core pkg repo"
2575

    
2576
	snapshots_update_status ">>> Copying files to ${RSYNCIP}"
2577

    
2578
	# Ensure directory(s) are available
2579
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/installer"
2580
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/updates"
2581
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/nanobsd"
2582
	if [ -d $STAGINGAREA/virtualization ]; then
2583
		ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/virtualization"
2584
	fi
2585
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/.updaters"
2586
	# ensure permissions are correct for r+w
2587
	ssh ${RSYNCUSER}@${RSYNCIP} "chmod -R ug+rw ${RSYNCPATH}/."
2588
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/${PRODUCT_NAME}${PRODUCT_NAME_SUFFIX}-*iso* \
2589
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/installer/
2590
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/${PRODUCT_NAME}${PRODUCT_NAME_SUFFIX}-memstick* \
2591
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/installer/
2592
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/${PRODUCT_NAME}${PRODUCT_NAME_SUFFIX}-*Update* \
2593
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/updates/
2594
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/nanobsd/* \
2595
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/nanobsd/
2596
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/nanobsdupdates/* \
2597
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/updates/
2598
	if [ -d $STAGINGAREA/virtualization ]; then
2599
		rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/virtualization/* \
2600
			${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/virtualization/
2601
	fi
2602

    
2603
	# Rather than copy these twice, use ln to link to the latest one.
2604

    
2605
	ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest.tgz"
2606
	ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest.tgz.sha256"
2607

    
2608
	LATESTFILENAME=$(basename ${UPDATES_TARBALL_FILENAME})
2609
	ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${LATESTFILENAME} \
2610
		${RSYNCPATH}/.updaters/latest.tgz"
2611
	ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${LATESTFILENAME}.sha256 \
2612
		${RSYNCPATH}/.updaters/latest.tgz.sha256"
2613

    
2614
	for i in ${FLASH_SIZE}
2615
	do
2616
		ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz"
2617
		ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz.sha256"
2618
		ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz"
2619
		ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz.sha256"
2620

    
2621
		FILENAMEUPGRADE="$(nanobsd_image_filename ${i} nanobsd 1).gz"
2622
		ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE} \
2623
			${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz"
2624
		ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE}.sha256 \
2625
			${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz.sha256"
2626

    
2627
		FILENAMEUPGRADE="$(nanobsd_image_filename ${i} nanobsd-vga 1).gz"
2628
		ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE} \
2629
			${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz"
2630
		ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE}.sha256 \
2631
			${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz.sha256"
2632
	done
2633

    
2634
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/version* \
2635
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/.updaters
2636
	snapshots_update_status ">>> Finished copying files."
2637
}
(2-2/3)