Project

General

Profile

Download (72.3 KB) Statistics
| Branch: | Tag: | Revision:
1
#!/bin/sh
2
#
3
# builder_common.sh
4
#
5
# part of pfSense (https://www.pfsense.org)
6
# Copyright (c) 2004-2013 BSD Perimeter
7
# Copyright (c) 2013-2016 Electric Sheep Fencing
8
# Copyright (c) 2014-2024 Rubicon Communications, LLC (Netgate)
9
# All rights reserved.
10
#
11
# FreeSBIE portions of the code
12
# Copyright (c) 2005 Dario Freni
13
# and copied from FreeSBIE project
14
# All rights reserved.
15
#
16
# Licensed under the Apache License, Version 2.0 (the "License");
17
# you may not use this file except in compliance with the License.
18
# You may obtain a copy of the License at
19
#
20
# http://www.apache.org/licenses/LICENSE-2.0
21
#
22
# Unless required by applicable law or agreed to in writing, software
23
# distributed under the License is distributed on an "AS IS" BASIS,
24
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
25
# See the License for the specific language governing permissions and
26
# limitations under the License.
27

    
28
if [ -z "${IMAGES_FINAL_DIR}" -o "${IMAGES_FINAL_DIR}" = "/" ]; then
29
	echo "IMAGES_FINAL_DIR is not defined"
30
	print_error_pfS
31
fi
32

    
33
kldload filemon >/dev/null 2>&1
34

    
35
lc() {
36
	echo "${1}" | tr '[[:upper:]]' '[[:lower:]]'
37
}
38

    
39
git_last_commit() {
40
	export CURRENT_COMMIT=$(git -C ${BUILDER_ROOT} log -1 --format='%H')
41
	export CURRENT_AUTHOR=$(git -C ${BUILDER_ROOT} log -1 --format='%an')
42
	echo ">>> Last known commit $CURRENT_AUTHOR - $CURRENT_COMMIT"
43
	echo "$CURRENT_COMMIT" > $SCRATCHDIR/build_commit_info.txt
44
}
45

    
46
# Create core pkg repository
47
core_pkg_create_repo() {
48
	if [ ! -d "${CORE_PKG_REAL_PATH}/All" ]; then
49
		return
50
	fi
51

    
52
	############ ATTENTION ##############
53
	#
54
	# For some reason pkg-repo fail without / in the end of directory name
55
	# so removing it will break command
56
	#
57
	# https://github.com/freebsd/pkg/issues/1364
58
	#
59
	echo -n ">>> Creating core packages repository... "
60
	if pkg repo -q "${CORE_PKG_REAL_PATH}/"; then
61
		echo "Done!"
62
	else
63
		echo "Failed!"
64
		print_error_pfS
65
	fi
66

    
67
	# Use the same directory structure as poudriere does to avoid
68
	# breaking snapshot repositories during rsync
69
	ln -sf $(basename ${CORE_PKG_REAL_PATH}) ${CORE_PKG_PATH}/.latest
70
	ln -sf .latest/All ${CORE_PKG_ALL_PATH}
71
	ln -sf .latest/digests.txz ${CORE_PKG_PATH}/digests.txz
72
	ln -sf .latest/meta.conf ${CORE_PKG_PATH}/meta.conf
73
	ln -sf .latest/meta.txz ${CORE_PKG_PATH}/meta.txz
74
	ln -sf .latest/packagesite.txz ${CORE_PKG_PATH}/packagesite.txz
75
}
76

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

    
86
	local _template_path=${BUILDER_TOOLS}/templates/core_pkg/${_template}
87

    
88
	# Use default pkg repo to obtain ABI and ALTABI
89
	local _abi=$(sed -e "s/%%ARCH%%/${TARGET_ARCH}/g" \
90
	    ${PKG_REPO_DEFAULT%%.conf}.abi)
91
	local _altabi_arch=$(get_altabi_arch ${TARGET_ARCH})
92
	local _altabi=$(sed -e "s/%%ARCH%%/${_altabi_arch}/g" \
93
	    ${PKG_REPO_DEFAULT%%.conf}.altabi)
94

    
95
	${BUILDER_SCRIPTS}/create_core_pkg.sh \
96
		-t "${_template_path}" \
97
		-f "${_flavor}" \
98
		-v "${_version}" \
99
		-r "${_root}" \
100
		-s "${_findroot}" \
101
		-F "${_filter}" \
102
		-d "${CORE_PKG_REAL_PATH}/All" \
103
		-a "${_abi}" \
104
		-A "${_altabi}" \
105
		|| print_error_pfS
106
}
107

    
108
# This routine will output that something went wrong
109
print_error_pfS() {
110
	echo
111
	echo "####################################"
112
	echo "Something went wrong, check errors!" >&2
113
	echo "####################################"
114
	echo
115
	echo "NOTE: a lot of times you can run './build.sh --clean-builder' to resolve."
116
	echo
117
	[ -n "${LOGFILE}" -a -f "${LOGFILE}" ] && \
118
		echo "Log saved on ${LOGFILE}" && \
119
	echo
120
	kill $$
121
	exit 1
122
}
123

    
124
# This routine will verify that the kernel has been
125
# installed OK to the staging area.
126
ensure_kernel_exists() {
127
	if [ ! -f "$1/boot/kernel/kernel.gz" ]; then
128
		echo ">>> ERROR: Could not locate $1/boot/kernel.gz"
129
		print_error_pfS
130
	fi
131
	KERNEL_SIZE=$(stat -f "%z" $1/boot/kernel/kernel.gz)
132
	if [ "$KERNEL_SIZE" -lt 3500 ]; then
133
		echo ">>> ERROR: Kernel $1/boot/kernel.gz appears to be smaller than it should be: $KERNEL_SIZE"
134
		print_error_pfS
135
	fi
136
}
137

    
138
get_pkg_name() {
139
	echo "${PRODUCT_NAME}-${1}-${CORE_PKG_VERSION}"
140
}
141

    
142
# This routine builds all related kernels
143
build_all_kernels() {
144
	# Set KERNEL_BUILD_PATH if it has not been set
145
	if [ -z "${KERNEL_BUILD_PATH}" ]; then
146
		KERNEL_BUILD_PATH=$SCRATCHDIR/kernels
147
		echo ">>> KERNEL_BUILD_PATH has not been set. Setting to ${KERNEL_BUILD_PATH}!"
148
	fi
149

    
150
	[ -d "${KERNEL_BUILD_PATH}" ] \
151
		&& rm -rf ${KERNEL_BUILD_PATH}
152

    
153
	# Build embedded kernel
154
	for BUILD_KERNEL in $BUILD_KERNELS; do
155
		unset KERNCONF
156
		unset KERNEL_DESTDIR
157
		unset KERNEL_NAME
158
		export KERNCONF=$BUILD_KERNEL
159
		export KERNEL_DESTDIR="$KERNEL_BUILD_PATH/$BUILD_KERNEL"
160
		export KERNEL_NAME=${BUILD_KERNEL}
161

    
162
		LOGFILE="${BUILDER_LOGS}/kernel.${KERNCONF}.${TARGET}.log"
163
		echo ">>> Building $BUILD_KERNEL kernel."  | tee -a ${LOGFILE}
164

    
165
		if [ -n "${NO_BUILDKERNEL}" -a -f "${CORE_PKG_ALL_PATH}/$(get_pkg_name kernel-${KERNEL_NAME}).txz" ]; then
166
			echo ">>> NO_BUILDKERNEL set, skipping build" | tee -a ${LOGFILE}
167
			continue
168
		fi
169

    
170
		buildkernel
171

    
172
		echo ">>> Staging $BUILD_KERNEL kernel..." | tee -a ${LOGFILE}
173
		installkernel
174

    
175
		ensure_kernel_exists $KERNEL_DESTDIR
176

    
177
		echo ">>> Creating pkg of $KERNEL_NAME-debug kernel to staging area..."  | tee -a ${LOGFILE}
178
		core_pkg_create kernel-debug ${KERNEL_NAME} ${CORE_PKG_VERSION} ${KERNEL_DESTDIR} \
179
		    "./usr/lib/debug/boot" \*.debug
180
		rm -rf ${KERNEL_DESTDIR}/usr
181

    
182
		echo ">>> Creating pkg of $KERNEL_NAME kernel to staging area..."  | tee -a ${LOGFILE}
183
		core_pkg_create kernel ${KERNEL_NAME} ${CORE_PKG_VERSION} ${KERNEL_DESTDIR} "./boot/kernel ./boot/modules"
184

    
185
		rm -rf $KERNEL_DESTDIR 2>&1 1>/dev/null
186
	done
187
}
188

    
189
install_default_kernel() {
190
	if [ -z "${1}" ]; then
191
		echo ">>> ERROR: install_default_kernel called without a kernel config name"| tee -a ${LOGFILE}
192
		print_error_pfS
193
	fi
194

    
195
	export KERNEL_NAME="${1}"
196

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

    
199
	# Copy kernel package to chroot, otherwise pkg won't find it to install
200
	if ! pkg_chroot_add ${FINAL_CHROOT_DIR} kernel-${KERNEL_NAME}; then
201
		echo ">>> ERROR: Error installing kernel package $(get_pkg_name kernel-${KERNEL_NAME}).txz" | tee -a ${LOGFILE}
202
		print_error_pfS
203
	fi
204

    
205
	# Set kernel pkg as vital to avoid user end up removing it for any reason
206
	pkg_chroot ${FINAL_CHROOT_DIR} set -v 1 -y $(get_pkg_name kernel-${KERNEL_NAME})
207

    
208
	if [ ! -f $FINAL_CHROOT_DIR/boot/kernel/kernel.gz ]; then
209
		echo ">>> ERROR: No kernel installed on $FINAL_CHROOT_DIR and the resulting image will be unusable. STOPPING!" | tee -a ${LOGFILE}
210
		print_error_pfS
211
	fi
212
	mkdir -p $FINAL_CHROOT_DIR/pkgs
213
	if [ -z "${2}" -o -n "${INSTALL_EXTRA_KERNELS}" ]; then
214
		cp ${CORE_PKG_ALL_PATH}/$(get_pkg_name kernel-${KERNEL_NAME}).txz $FINAL_CHROOT_DIR/pkgs
215
		if [ -n "${INSTALL_EXTRA_KERNELS}" ]; then
216
			for _EXTRA_KERNEL in $INSTALL_EXTRA_KERNELS; do
217
				_EXTRA_KERNEL_PATH=${CORE_PKG_ALL_PATH}/$(get_pkg_name kernel-${_EXTRA_KERNEL}).txz
218
				if [ -f "${_EXTRA_KERNEL_PATH}" ]; then
219
					echo -n ". adding ${_EXTRA_KERNEL_PATH} on image /pkgs folder"
220
					cp ${_EXTRA_KERNEL_PATH} $FINAL_CHROOT_DIR/pkgs
221
				else
222
					echo ">>> ERROR: Requested kernel $(get_pkg_name kernel-${_EXTRA_KERNEL}).txz was not found to be put on image /pkgs folder!"
223
					print_error_pfS
224
				fi
225
			done
226
		fi
227
	fi
228
	echo "Done." | tee -a ${LOGFILE}
229

    
230
	unset KERNEL_NAME
231
}
232

    
233
# This builds FreeBSD (make buildworld)
234
# Imported from FreeSBIE
235
make_world() {
236
	LOGFILE=${BUILDER_LOGS}/buildworld.${TARGET}
237
	echo ">>> LOGFILE set to $LOGFILE." | tee -a ${LOGFILE}
238
	if [ -n "${NO_BUILDWORLD}" ]; then
239
		echo ">>> NO_BUILDWORLD set, skipping build" | tee -a ${LOGFILE}
240
		return
241
	fi
242

    
243
	echo ">>> $(LC_ALL=C date) - Starting build world for ${TARGET} architecture..." | tee -a ${LOGFILE}
244
	script -aq $LOGFILE ${BUILDER_SCRIPTS}/build_freebsd.sh -K -s ${FREEBSD_SRC_DIR} \
245
		|| print_error_pfS
246
	echo ">>> $(LC_ALL=C date) - Finished build world for ${TARGET} architecture..." | tee -a ${LOGFILE}
247

    
248
	LOGFILE=${BUILDER_LOGS}/installworld.${TARGET}
249
	echo ">>> LOGFILE set to $LOGFILE." | tee -a ${LOGFILE}
250

    
251
	[ -d "${INSTALLER_CHROOT_DIR}" ] \
252
		|| mkdir -p ${INSTALLER_CHROOT_DIR}
253

    
254
	echo ">>> Installing world with bsdinstall for ${TARGET} architecture..." | tee -a ${LOGFILE}
255
	script -aq $LOGFILE ${BUILDER_SCRIPTS}/install_freebsd.sh -i -K \
256
		-s ${FREEBSD_SRC_DIR} \
257
		-d ${INSTALLER_CHROOT_DIR} \
258
		|| print_error_pfS
259

    
260
	# Copy additional installer scripts
261
	install -o root -g wheel -m 0755 ${BUILDER_TOOLS}/installer/*.sh \
262
		${INSTALLER_CHROOT_DIR}/root
263

    
264
	# XXX set root password since we don't have nullok enabled
265
	pw -R ${INSTALLER_CHROOT_DIR} usermod root -w yes
266

    
267
	echo ">>> Installing world without bsdinstall for ${TARGET} architecture..." | tee -a ${LOGFILE}
268
	script -aq $LOGFILE ${BUILDER_SCRIPTS}/install_freebsd.sh -K \
269
		-s ${FREEBSD_SRC_DIR} \
270
		-d ${STAGE_CHROOT_DIR} \
271
		|| print_error_pfS
272

    
273
	# Use the builder cross compiler from obj to produce the final binary.
274
	BUILD_CC="${MAKEOBJDIRPREFIX}${FREEBSD_SRC_DIR}/${TARGET}.${TARGET_ARCH}/tmp/usr/bin/cc"
275

    
276
	[ -f "${BUILD_CC}" ] || print_error_pfS
277

    
278
	# XXX It must go to the scripts
279
	[ -d "${STAGE_CHROOT_DIR}/usr/local/bin" ] \
280
		|| mkdir -p ${STAGE_CHROOT_DIR}/usr/local/bin
281
	makeargs="CC=${BUILD_CC} DESTDIR=${STAGE_CHROOT_DIR}"
282
	echo ">>> Building and installing crypto tools and athstats for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
283
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/crypto ${makeargs} clean all install || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
284
	# XXX FIX IT
285
#	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} clean all install || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
286
	echo ">>> Building and installing crypto tools and athstats for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
287

    
288
	if [ "${PRODUCT_NAME}" = "pfSense" -a -n "${GNID_REPO_BASE}" ]; then
289
		echo ">>> Building gnid... " | tee -a ${LOGFILE}
290
		(\
291
			cd ${GNID_SRC_DIR} && \
292
			make \
293
				CC=${BUILD_CC} \
294
				INCLUDE_DIR=${GNID_INCLUDE_DIR} \
295
				LIBCRYPTO_DIR=${GNID_LIBCRYPTO_DIR} \
296
			clean gnid \
297
		) || print_error_pfS
298
		install -o root -g wheel -m 0700 ${GNID_SRC_DIR}/gnid \
299
			${STAGE_CHROOT_DIR}/usr/sbin \
300
			|| print_error_pfS
301
		install -o root -g wheel -m 0700 ${GNID_SRC_DIR}/gnid \
302
			${INSTALLER_CHROOT_DIR}/usr/sbin \
303
			|| print_error_pfS
304
	fi
305

    
306
	unset makeargs
307
}
308

    
309
# This routine creates a ova image that contains
310
# a ovf and vmdk file. These files can be imported
311
# right into vmware or virtual box.
312
# (and many other emulation platforms)
313
# http://www.vmware.com/pdf/ovf_whitepaper_specification.pdf
314
create_ova_image() {
315
	# XXX create a .ovf php creator that you can pass:
316
	#     1. populatedSize
317
	#     2. license
318
	#     3. product name
319
	#     4. version
320
	#     5. number of network interface cards
321
	#     6. allocationUnits
322
	#     7. capacity
323
	#     8. capacityAllocationUnits
324

    
325
	LOGFILE=${BUILDER_LOGS}/ova.${TARGET}.log
326

    
327
	local _mntdir=${OVA_TMP}/mnt
328

    
329
	if [ -d "${_mntdir}" ]; then
330
		local _dev
331
		# XXX Root cause still didn't found but it doesn't umount
332
		#     properly on looped builds and then require this extra
333
		#     check
334
		while true; do
335
			_dev=$(mount -p ${_mntdir} 2>/dev/null | awk '{print $1}')
336
			[ $? -ne 0 -o -z "${_dev}" ] \
337
				&& break
338
			umount -f ${_mntdir}
339
			mdconfig -d -u ${_dev#/dev/}
340
		done
341
		chflags -R noschg ${OVA_TMP}
342
		rm -rf ${OVA_TMP}
343
	fi
344

    
345
	mkdir -p $(dirname ${OVAPATH})
346

    
347
	mkdir -p ${_mntdir}
348

    
349
	if [ -z "${OVA_SWAP_PART_SIZE_IN_GB}" -o "${OVA_SWAP_PART_SIZE_IN_GB}" = "0" ]; then
350
		# first partition size (freebsd-ufs)
351
		local OVA_FIRST_PART_SIZE_IN_GB=${VMDK_DISK_CAPACITY_IN_GB}
352
		# Calculate real first partition size, removing 256 blocks (131072 bytes) beginning/loader
353
		local OVA_FIRST_PART_SIZE=$((${OVA_FIRST_PART_SIZE_IN_GB}*1024*1024*1024-131072))
354
		# Unset swap partition size variable
355
		unset OVA_SWAP_PART_SIZE
356
		# Parameter used by mkimg
357
		unset OVA_SWAP_PART_PARAM
358
	else
359
		# first partition size (freebsd-ufs)
360
		local OVA_FIRST_PART_SIZE_IN_GB=$((VMDK_DISK_CAPACITY_IN_GB-OVA_SWAP_PART_SIZE_IN_GB))
361
		# Use first partition size in g
362
		local OVA_FIRST_PART_SIZE="${OVA_FIRST_PART_SIZE_IN_GB}g"
363
		# Calculate real swap size, removing 256 blocks (131072 bytes) beginning/loader
364
		local OVA_SWAP_PART_SIZE=$((${OVA_SWAP_PART_SIZE_IN_GB}*1024*1024*1024-131072))
365
		# Parameter used by mkimg
366
		local OVA_SWAP_PART_PARAM="-p freebsd-swap/swap0::${OVA_SWAP_PART_SIZE}"
367
	fi
368

    
369
	# Prepare folder to be put in image
370
	customize_stagearea_for_image "ova"
371
	install_default_kernel ${DEFAULT_KERNEL} "no"
372

    
373
	# Fill fstab
374
	echo ">>> Installing platform specific items..." | tee -a ${LOGFILE}
375
	echo "/dev/gpt/${PRODUCT_NAME}	/	ufs		rw	1	1" > ${FINAL_CHROOT_DIR}/etc/fstab
376
	if [ -n "${OVA_SWAP_PART_SIZE}" ]; then
377
		echo "/dev/gpt/swap0	none	swap	sw	0	0" >> ${FINAL_CHROOT_DIR}/etc/fstab
378
	fi
379

    
380
	# Create / partition
381
	echo -n ">>> Creating / partition... " | tee -a ${LOGFILE}
382
	truncate -s ${OVA_FIRST_PART_SIZE} ${OVA_TMP}/${OVFUFS}
383
	local _md=$(mdconfig -a -f ${OVA_TMP}/${OVFUFS})
384
	trap "mdconfig -d -u ${_md}; return" 1 2 15 EXIT
385

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

    
388
	if ! mount /dev/${_md} ${_mntdir} 2>&1 >>${LOGFILE}; then
389
		echo "Failed!" | tee -a ${LOGFILE}
390
		echo ">>> ERROR: Error mounting temporary vmdk image. STOPPING!" | tee -a ${LOGFILE}
391
		print_error_pfS
392
	fi
393
	trap "sync; sleep 3; umount ${_mntdir} || umount -f ${_mntdir}; mdconfig -d -u ${_md}; return" 1 2 15 EXIT
394

    
395
	echo "Done!" | tee -a ${LOGFILE}
396

    
397
	clone_directory_contents ${FINAL_CHROOT_DIR} ${_mntdir}
398

    
399
	sync
400
	sleep 3
401
	umount ${_mntdir} || umount -f ${_mntdir} >>${LOGFILE} 2>&1
402
	mdconfig -d -u ${_md}
403
	trap "-" 1 2 15 EXIT
404

    
405
	# Create raw disk
406
	echo -n ">>> Creating raw disk... " | tee -a ${LOGFILE}
407
	mkimg \
408
		-s gpt \
409
		-f raw \
410
		-b ${FINAL_CHROOT_DIR}/boot/pmbr \
411
		-p freebsd-boot:=${FINAL_CHROOT_DIR}/boot/gptboot \
412
		-p freebsd-ufs/${PRODUCT_NAME}:=${OVA_TMP}/${OVFUFS} \
413
		${OVA_SWAP_PART_PARAM} \
414
		-o ${OVA_TMP}/${OVFRAW} 2>&1 >> ${LOGFILE}
415

    
416
	if [ $? -ne 0 -o ! -f ${OVA_TMP}/${OVFRAW} ]; then
417
		if [ -f ${OVA_TMP}/${OVFUFS} ]; then
418
			rm -f ${OVA_TMP}/${OVFUFS}
419
		fi
420
		if [ -f ${OVA_TMP}/${OVFRAW} ]; then
421
			rm -f ${OVA_TMP}/${OVFRAW}
422
		fi
423
		echo "Failed!" | tee -a ${LOGFILE}
424
		echo ">>> ERROR: Error creating temporary vmdk image. STOPPING!" | tee -a ${LOGFILE}
425
		print_error_pfS
426
	fi
427
	echo "Done!" | tee -a ${LOGFILE}
428

    
429
	# We don't need it anymore
430
	rm -f ${OVA_TMP}/${OVFUFS} >/dev/null 2>&1
431

    
432
	# Convert raw to vmdk
433
	echo -n ">>> Creating vmdk disk... " | tee -a ${LOGFILE}
434
	vmdktool -z9 -v ${OVA_TMP}/${OVFVMDK} ${OVA_TMP}/${OVFRAW}
435

    
436
	if [ $? -ne 0 -o ! -f ${OVA_TMP}/${OVFVMDK} ]; then
437
		if [ -f ${OVA_TMP}/${OVFRAW} ]; then
438
			rm -f ${OVA_TMP}/${OVFRAW}
439
		fi
440
		if [ -f ${OVA_TMP}/${OVFVMDK} ]; then
441
			rm -f ${OVA_TMP}/${OVFVMDK}
442
		fi
443
		echo "Failed!" | tee -a ${LOGFILE}
444
		echo ">>> ERROR: Error creating vmdk image. STOPPING!" | tee -a ${LOGFILE}
445
		print_error_pfS
446
	fi
447
	echo "Done!" | tee -a ${LOGFILE}
448

    
449
	rm -f ${OVA_TMP}/${OVFRAW}
450

    
451
	ova_setup_ovf_template
452

    
453
	echo -n ">>> Writing final ova image... " | tee -a ${LOGFILE}
454
	# Create OVA file for vmware
455
	gtar -C ${OVA_TMP} -cpf ${OVAPATH} ${PRODUCT_NAME}.ovf ${OVFVMDK}
456
	echo "Done!" | tee -a ${LOGFILE}
457
	rm -f ${OVA_TMP}/${OVFVMDK} >/dev/null 2>&1
458

    
459
	echo ">>> OVA created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
460
}
461

    
462
# called from create_ova_image
463
ova_setup_ovf_template() {
464
	if [ ! -f ${OVFTEMPLATE} ]; then
465
		echo ">>> ERROR: OVF template file (${OVFTEMPLATE}) not found."
466
		print_error_pfS
467
	fi
468

    
469
	#  OperatingSystemSection (${PRODUCT_NAME}.ovf)
470
	#  42   FreeBSD 32-Bit
471
	#  78   FreeBSD 64-Bit
472
	if [ "${TARGET}" = "amd64" ]; then
473
		local _os_id="78"
474
		local _os_type="freebsd64Guest"
475
		local _os_descr="FreeBSD 64-Bit"
476
	else
477
		echo ">>> ERROR: Platform not supported for OVA (${TARGET})"
478
		print_error_pfS
479
	fi
480

    
481
	local POPULATED_SIZE=$(du -d0 -k $FINAL_CHROOT_DIR | cut -f1)
482
	local POPULATED_SIZE_IN_BYTES=$((${POPULATED_SIZE}*1024))
483
	local VMDK_FILE_SIZE=$(stat -f "%z" ${OVA_TMP}/${OVFVMDK})
484

    
485
	sed \
486
		-e "s,%%VMDK_FILE_SIZE%%,${VMDK_FILE_SIZE},g" \
487
		-e "s,%%VMDK_DISK_CAPACITY_IN_GB%%,${VMDK_DISK_CAPACITY_IN_GB},g" \
488
		-e "s,%%POPULATED_SIZE_IN_BYTES%%,${POPULATED_SIZE_IN_BYTES},g" \
489
		-e "s,%%OS_ID%%,${_os_id},g" \
490
		-e "s,%%OS_TYPE%%,${_os_type},g" \
491
		-e "s,%%OS_DESCR%%,${_os_descr},g" \
492
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
493
		-e "s,%%PRODUCT_NAME_SUFFIX%%,${PRODUCT_NAME_SUFFIX},g" \
494
		-e "s,%%PRODUCT_VERSION%%,${PRODUCT_VERSION},g" \
495
		-e "s,%%PRODUCT_URL%%,${PRODUCT_URL},g" \
496
		-e "s#%%VENDOR_NAME%%#${VENDOR_NAME}#g" \
497
		-e "s#%%OVF_INFO%%#${OVF_INFO}#g" \
498
		-e "/^%%PRODUCT_LICENSE%%/r ${BUILDER_ROOT}/LICENSE" \
499
		-e "/^%%PRODUCT_LICENSE%%/d" \
500
		${OVFTEMPLATE} > ${OVA_TMP}/${PRODUCT_NAME}.ovf
501
}
502

    
503
# Cleans up previous builds
504
clean_builder() {
505
	# Clean out directories
506
	echo ">>> Cleaning up previous build environment...Please wait!"
507

    
508
	staginareas_clean_each_run
509

    
510
	if [ -d "${STAGE_CHROOT_DIR}" ]; then
511
		echo -n ">>> Cleaning ${STAGE_CHROOT_DIR}... "
512
		chflags -R noschg ${STAGE_CHROOT_DIR} 2>&1 >/dev/null
513
		rm -rf ${STAGE_CHROOT_DIR}/* 2>/dev/null
514
		echo "Done."
515
	fi
516

    
517
	if [ -d "${INSTALLER_CHROOT_DIR}" ]; then
518
		echo -n ">>> Cleaning ${INSTALLER_CHROOT_DIR}... "
519
		chflags -R noschg ${INSTALLER_CHROOT_DIR} 2>&1 >/dev/null
520
		rm -rf ${INSTALLER_CHROOT_DIR}/* 2>/dev/null
521
		echo "Done."
522
	fi
523

    
524
	if [ -z "${NO_CLEAN_FREEBSD_OBJ}" -a -d "${FREEBSD_SRC_DIR}" ]; then
525
		OBJTREE=$(make -C ${FREEBSD_SRC_DIR} -V OBJTREE)
526
		if [ -d "${OBJTREE}" ]; then
527
			echo -n ">>> Cleaning FreeBSD objects dir staging..."
528
			echo -n "."
529
			chflags -R noschg ${OBJTREE} 2>&1 >/dev/null
530
			echo -n "."
531
			rm -rf ${OBJTREE}/*
532
			echo "Done!"
533
		fi
534
		if [ -d "${KERNEL_BUILD_PATH}" ]; then
535
			echo -n ">>> Cleaning previously built kernel stage area..."
536
			rm -rf $KERNEL_BUILD_PATH/*
537
			echo "Done!"
538
		fi
539
	fi
540
	mkdir -p $KERNEL_BUILD_PATH
541

    
542
	echo -n ">>> Cleaning previously built images..."
543
	rm -rf $IMAGES_FINAL_DIR/*
544
	echo "Done!"
545

    
546
	echo -n ">>> Cleaning previous builder logs..."
547
	if [ -d "$BUILDER_LOGS" ]; then
548
		rm -rf ${BUILDER_LOGS}
549
	fi
550
	mkdir -p ${BUILDER_LOGS}
551

    
552
	echo "Done!"
553

    
554
	echo ">>> Cleaning of builder environment has finished."
555
}
556

    
557
clone_directory_contents() {
558
	if [ ! -e "$2" ]; then
559
		mkdir -p "$2"
560
	fi
561
	if [ ! -d "$1" -o ! -d "$2" ]; then
562
		if [ -z "${LOGFILE}" ]; then
563
			echo ">>> ERROR: Argument $1 supplied is not a directory!"
564
		else
565
			echo ">>> ERROR: Argument $1 supplied is not a directory!" | tee -a ${LOGFILE}
566
		fi
567
		print_error_pfS
568
	fi
569
	echo -n ">>> Using TAR to clone $1 to $2 ..."
570
	tar -C ${1} -c -f - . | tar -C ${2} -x -p -f -
571
	echo "Done!"
572
}
573

    
574
clone_to_staging_area() {
575
	# Clone everything to the final staging area
576
	echo -n ">>> Cloning everything to ${STAGE_CHROOT_DIR} staging area..."
577
	LOGFILE=${BUILDER_LOGS}/cloning.${TARGET}.log
578

    
579
	tar -C ${PRODUCT_SRC} -c -f - . | \
580
		tar -C ${STAGE_CHROOT_DIR} -x -p -f -
581

    
582
	mkdir -p ${STAGE_CHROOT_DIR}/etc/mtree
583
	mtree -Pcp ${STAGE_CHROOT_DIR}/var > ${STAGE_CHROOT_DIR}/etc/mtree/var.dist
584
	mtree -Pcp ${STAGE_CHROOT_DIR}/etc > ${STAGE_CHROOT_DIR}/etc/mtree/etc.dist
585
	if [ -d ${STAGE_CHROOT_DIR}/usr/local/etc ]; then
586
		mtree -Pcp ${STAGE_CHROOT_DIR}/usr/local/etc > ${STAGE_CHROOT_DIR}/etc/mtree/localetc.dist
587
	fi
588

    
589
	## Add buildtime and lastcommit information
590
	# This is used for detecting updates.
591
	echo "$BUILTDATESTRING" > $STAGE_CHROOT_DIR/etc/version.buildtime
592
	# Record last commit info if it is available.
593
	if [ -f $SCRATCHDIR/build_commit_info.txt ]; then
594
		cp $SCRATCHDIR/build_commit_info.txt $STAGE_CHROOT_DIR/etc/version.lastcommit
595
	fi
596

    
597
	local _exclude_files="${SCRATCHDIR}/base_exclude_files"
598
	sed \
599
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
600
		-e "s,%%VERSION%%,${_version},g" \
601
		${BUILDER_TOOLS}/templates/core_pkg/base/exclude_files \
602
		> ${_exclude_files}
603

    
604
	mkdir -p ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR} >/dev/null 2>&1
605

    
606
	# Include a sample pkg stable conf to base
607
	setup_pkg_repo \
608
		${PKG_REPO_DEFAULT} \
609
		${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/${PRODUCT_NAME}-repo.conf \
610
		${TARGET} \
611
		${TARGET_ARCH}
612

    
613
	mtree \
614
		-c \
615
		-k uid,gid,mode,size,flags,sha256digest \
616
		-p ${STAGE_CHROOT_DIR} \
617
		-X ${_exclude_files} \
618
		> ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/base.mtree
619
	tar \
620
		-C ${STAGE_CHROOT_DIR} \
621
		-cJf ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/base.txz \
622
		-X ${_exclude_files} \
623
		.
624

    
625
	core_pkg_create rc "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
626
	core_pkg_create base "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
627
	core_pkg_create default-config "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
628

    
629
	local DEFAULTCONF=${STAGE_CHROOT_DIR}/conf.default/config.xml
630

    
631
	# Save current WAN and LAN if value
632
	local _old_wan_if=$(xml sel -t -v "${XML_ROOTOBJ}/interfaces/wan/if" ${DEFAULTCONF})
633
	local _old_lan_if=$(xml sel -t -v "${XML_ROOTOBJ}/interfaces/lan/if" ${DEFAULTCONF})
634

    
635
	# Change default interface names to match vmware driver
636
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/wan/if" -v "vmx0" ${DEFAULTCONF}
637
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/lan/if" -v "vmx1" ${DEFAULTCONF}
638
	core_pkg_create default-config "vmware" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
639

    
640
	# Restore default values to be used by serial package
641
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/wan/if" -v "${_old_wan_if}" ${DEFAULTCONF}
642
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/lan/if" -v "${_old_lan_if}" ${DEFAULTCONF}
643

    
644
	# Activate serial console in config.xml
645
	xml ed -L -P -d "${XML_ROOTOBJ}/system/enableserial" ${DEFAULTCONF}
646
	xml ed -P -s "${XML_ROOTOBJ}/system" -t elem -n "enableserial" \
647
		${DEFAULTCONF} > ${DEFAULTCONF}.tmp
648
	xml fo -t ${DEFAULTCONF}.tmp > ${DEFAULTCONF}
649
	rm -f ${DEFAULTCONF}.tmp
650

    
651
	echo force > ${STAGE_CHROOT_DIR}/cf/conf/enableserial_force
652

    
653
	core_pkg_create default-config-serial "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
654
	core_pkg_create default-config "bhyve" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
655

    
656
	rm -f ${STAGE_CHROOT_DIR}/cf/conf/enableserial_force
657
	rm -f ${STAGE_CHROOT_DIR}/cf/conf/config.xml
658

    
659
	# Make sure pkg is present
660
	pkg_bootstrap ${STAGE_CHROOT_DIR}
661

    
662
	# Make sure correct repo is available on tmp dir
663
	mkdir -p ${STAGE_CHROOT_DIR}/tmp/pkg/pkg-repos
664
	setup_pkg_repo \
665
		${PKG_REPO_BUILD} \
666
		${STAGE_CHROOT_DIR}/tmp/pkg/pkg-repos/repo.conf \
667
		${TARGET} \
668
		${TARGET_ARCH} \
669
		staging \
670
		${STAGE_CHROOT_DIR}/tmp/pkg/pkg.conf
671

    
672
	echo "Done!"
673
}
674

    
675
create_final_staging_area() {
676
	if [ -z "${FINAL_CHROOT_DIR}" ]; then
677
		echo ">>> ERROR: FINAL_CHROOT_DIR is not set, cannot continue!" | tee -a ${LOGFILE}
678
		print_error_pfS
679
	fi
680

    
681
	if [ -d "${FINAL_CHROOT_DIR}" ]; then
682
		echo -n ">>> Previous ${FINAL_CHROOT_DIR} detected cleaning up..." | tee -a ${LOGFILE}
683
		chflags -R noschg ${FINAL_CHROOT_DIR} 2>&1 1>/dev/null
684
		rm -rf ${FINAL_CHROOT_DIR}/* 2>&1 1>/dev/null
685
		echo "Done." | tee -a ${LOGFILE}
686
	fi
687

    
688
	echo ">>> Preparing Final image staging area: $(LC_ALL=C date)" 2>&1 | tee -a ${LOGFILE}
689
	echo ">>> Cloning ${STAGE_CHROOT_DIR} to ${FINAL_CHROOT_DIR}" 2>&1 | tee -a ${LOGFILE}
690
	clone_directory_contents ${STAGE_CHROOT_DIR} ${FINAL_CHROOT_DIR}
691

    
692
	if [ ! -f $FINAL_CHROOT_DIR/sbin/init ]; then
693
		echo ">>> ERROR: Something went wrong during cloning -- Please verify!" 2>&1 | tee -a ${LOGFILE}
694
		print_error_pfS
695
	fi
696
}
697

    
698
customize_stagearea_for_image() {
699
	local _image_type="$1"
700
	local _default_config="" # filled with $2 below
701
	local _image_variant="$3"
702

    
703
	if [ -n "$2" ]; then
704
		_default_config="$2"
705
	elif [ "${_image_type}" = "memstickserial" -o \
706
	     "${_image_type}" = "memstickadi" ]; then
707
		_default_config="default-config-serial"
708
	elif [ "${_image_type}" = "ova" ]; then
709
		_default_config="default-config-vmware"
710
	else
711
		_default_config="default-config"
712
	fi
713

    
714
	# Prepare final stage area
715
	create_final_staging_area
716

    
717
	pkg_chroot_add ${FINAL_CHROOT_DIR} rc
718
	pkg_chroot_add ${FINAL_CHROOT_DIR} base
719

    
720
	# Set base/rc pkgs as vital to avoid user end up removing it for any reason
721
	pkg_chroot ${FINAL_CHROOT_DIR} set -v 1 -y $(get_pkg_name rc)
722
	pkg_chroot ${FINAL_CHROOT_DIR} set -v 1 -y $(get_pkg_name base)
723

    
724
	if [ "${_image_type}" = "iso" -o \
725
	     "${_image_type}" = "memstick" -o \
726
	     "${_image_type}" = "memstickserial" -o \
727
	     "${_image_type}" = "memstickadi" ]; then
728
		mkdir -p ${FINAL_CHROOT_DIR}/pkgs
729
		cp ${CORE_PKG_ALL_PATH}/*default-config*.txz ${FINAL_CHROOT_DIR}/pkgs
730
	fi
731

    
732
	pkg_chroot_add ${FINAL_CHROOT_DIR} ${_default_config}
733

    
734
	# XXX: Workaround to avoid pkg to complain regarding release
735
	#      repo on first boot since packages are installed from
736
	#      staging server during build phase
737
	if [ -n "${USE_PKG_REPO_STAGING}" ]; then
738
		_read_cmd="select value from repodata where key='packagesite'"
739
		if [ -n "${_IS_RELEASE}" -o -n "${_IS_RC}" ]; then
740
			local _tgt_server="${PKG_REPO_SERVER_RELEASE}"
741
		else
742
			local _tgt_server="${PKG_REPO_SERVER_DEVEL}"
743
		fi
744
		for _db in ${FINAL_CHROOT_DIR}/var/db/pkg/repo-*sqlite; do
745
			_cur=$(/usr/local/bin/sqlite3 ${_db} "${_read_cmd}")
746
			_new=$(echo "${_cur}" | sed -e "s,^${PKG_REPO_SERVER_STAGING},${_tgt_server},")
747
			/usr/local/bin/sqlite3 ${_db} "update repodata set value='${_new}' where key='packagesite'"
748
		done
749
	fi
750

    
751
	if [ -n "$_image_variant" -a \
752
	    -d ${BUILDER_TOOLS}/templates/custom_logos/${_image_variant} ]; then
753
		mkdir -p ${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/custom_logos
754
		cp -f \
755
			${BUILDER_TOOLS}/templates/custom_logos/${_image_variant}/*.svg \
756
			${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/custom_logos
757
		cp -f \
758
			${BUILDER_TOOLS}/templates/custom_logos/${_image_variant}/*.css \
759
			${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/custom_logos
760
	fi
761

    
762
	# Remove temporary repo conf
763
	rm -rf ${FINAL_CHROOT_DIR}/tmp/pkg
764
}
765

    
766
create_distribution_tarball() {
767
	mkdir -p ${INSTALLER_CHROOT_DIR}/usr/freebsd-dist
768

    
769
	echo -n ">>> Creating distribution tarball... " | tee -a ${LOGFILE}
770
	tar -C ${FINAL_CHROOT_DIR} --exclude ./pkgs \
771
		-cJf ${INSTALLER_CHROOT_DIR}/usr/freebsd-dist/base.txz .
772
	echo "Done!" | tee -a ${LOGFILE}
773

    
774
	echo -n ">>> Creating manifest... " | tee -a ${LOGFILE}
775
	(cd ${INSTALLER_CHROOT_DIR}/usr/freebsd-dist && \
776
		sh ${FREEBSD_SRC_DIR}/release/scripts/make-manifest.sh base.txz) \
777
		> ${INSTALLER_CHROOT_DIR}/usr/freebsd-dist/MANIFEST
778
	echo "Done!" | tee -a ${LOGFILE}
779
}
780

    
781
create_iso_image() {
782
	local _variant="$1"
783

    
784
	LOGFILE=${BUILDER_LOGS}/isoimage.${TARGET}
785

    
786
	if [ -z "${ISOPATH}" ]; then
787
		echo ">>> ISOPATH is empty skipping generation of ISO image!" | tee -a ${LOGFILE}
788
		return
789
	fi
790

    
791
	echo ">>> Building bootable ISO image for ${TARGET}" | tee -a ${LOGFILE}
792

    
793
	mkdir -p $(dirname ${ISOPATH})
794

    
795
	local _image_path=${ISOPATH}
796
	if [ -n "${_variant}" ]; then
797
		_image_path=$(echo "$_image_path" | \
798
			sed "s/${PRODUCT_NAME_SUFFIX}-/&${_variant}-/")
799
		VARIANTIMAGES="${VARIANTIMAGES}${VARIANTIMAGES:+ }${_image_path}"
800
	fi
801

    
802
	customize_stagearea_for_image "iso" "" $_variant
803
	install_default_kernel ${DEFAULT_KERNEL}
804

    
805
	BOOTCONF=${INSTALLER_CHROOT_DIR}/boot.config
806
	LOADERCONF=${INSTALLER_CHROOT_DIR}/boot/loader.conf
807

    
808
	rm -f ${LOADERCONF} ${BOOTCONF} >/dev/null 2>&1
809
	echo 'autoboot_delay="3"' > ${LOADERCONF}
810
	echo 'kern.cam.boot_delay=10000' >> ${LOADERCONF}
811
	cat ${LOADERCONF} > ${FINAL_CHROOT_DIR}/boot/loader.conf
812

    
813
	create_distribution_tarball
814

    
815
	FSLABEL=$(echo ${PRODUCT_NAME} | tr '[:lower:]' '[:upper:]')
816

    
817
	sh ${FREEBSD_SRC_DIR}/release/${TARGET}/mkisoimages.sh -b \
818
		${FSLABEL} \
819
		${_image_path} \
820
		${INSTALLER_CHROOT_DIR}
821

    
822
	if [ ! -f "${_image_path}" ]; then
823
		echo "ERROR! ISO image was not built"
824
		print_error_pfS
825
	fi
826

    
827
	gzip -qf $_image_path &
828
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
829

    
830
	echo ">>> ISO created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
831
}
832

    
833
create_memstick_image() {
834
	local _variant="$1"
835

    
836
	LOGFILE=${BUILDER_LOGS}/memstick.${TARGET}
837
	if [ "${MEMSTICKPATH}" = "" ]; then
838
		echo ">>> MEMSTICKPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
839
		return
840
	fi
841

    
842
	mkdir -p $(dirname ${MEMSTICKPATH})
843

    
844
	local _image_path=${MEMSTICKPATH}
845
	if [ -n "${_variant}" ]; then
846
		_image_path=$(echo "$_image_path" | \
847
			sed "s/-memstick-/-memstick-${_variant}-/")
848
		VARIANTIMAGES="${VARIANTIMAGES}${VARIANTIMAGES:+ }${_image_path}"
849
	fi
850

    
851
	customize_stagearea_for_image "memstick" "" $_variant
852
	install_default_kernel ${DEFAULT_KERNEL}
853

    
854
	echo ">>> Creating memstick to ${_image_path}." 2>&1 | tee -a ${LOGFILE}
855

    
856
	BOOTCONF=${INSTALLER_CHROOT_DIR}/boot.config
857
	LOADERCONF=${INSTALLER_CHROOT_DIR}/boot/loader.conf
858

    
859
	rm -f ${LOADERCONF} ${BOOTCONF} >/dev/null 2>&1
860

    
861
	echo 'autoboot_delay="3"' > ${LOADERCONF}
862
	echo 'kern.cam.boot_delay=10000' >> ${LOADERCONF}
863
	echo 'boot_serial="NO"' >> ${LOADERCONF}
864
	cat ${LOADERCONF} > ${FINAL_CHROOT_DIR}/boot/loader.conf
865

    
866
	create_distribution_tarball
867

    
868
	FSLABEL=$(echo ${PRODUCT_NAME} | tr '[:lower:]' '[:upper:]')
869

    
870
	sh ${FREEBSD_SRC_DIR}/release/${TARGET}/mkisoimages.sh -b \
871
		${FSLABEL} \
872
		${_image_path} \
873
		${INSTALLER_CHROOT_DIR}
874

    
875
	if [ ! -f "${_image_path}" ]; then
876
		echo "ERROR! memstick image was not built"
877
		print_error_pfS
878
	fi
879

    
880
	gzip -qf $_image_path &
881
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
882

    
883
	echo ">>> MEMSTICK created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
884
}
885

    
886
create_memstick_serial_image() {
887
	LOGFILE=${BUILDER_LOGS}/memstickserial.${TARGET}
888
	if [ "${MEMSTICKSERIALPATH}" = "" ]; then
889
		echo ">>> MEMSTICKSERIALPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
890
		return
891
	fi
892

    
893
	mkdir -p $(dirname ${MEMSTICKSERIALPATH})
894

    
895
	customize_stagearea_for_image "memstickserial"
896
	install_default_kernel ${DEFAULT_KERNEL}
897

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

    
900
	BOOTCONF=${INSTALLER_CHROOT_DIR}/boot.config
901
	LOADERCONF=${INSTALLER_CHROOT_DIR}/boot/loader.conf
902

    
903
	echo ">>> Activating serial console..." 2>&1 | tee -a ${LOGFILE}
904
	echo "-S115200 -D" > ${BOOTCONF}
905

    
906
	# Activate serial console+video console in loader.conf
907
	echo 'autoboot_delay="3"' > ${LOADERCONF}
908
	echo 'kern.cam.boot_delay=10000' >> ${LOADERCONF}
909
	echo 'boot_multicons="YES"' >> ${LOADERCONF}
910
	echo 'boot_serial="YES"' >> ${LOADERCONF}
911
	echo 'console="comconsole,vidconsole"' >> ${LOADERCONF}
912
	echo 'comconsole_speed="115200"' >> ${LOADERCONF}
913

    
914
	cat ${BOOTCONF} >> ${FINAL_CHROOT_DIR}/boot.config
915
	cat ${LOADERCONF} >> ${FINAL_CHROOT_DIR}/boot/loader.conf
916

    
917
	create_distribution_tarball
918

    
919
	sh ${FREEBSD_SRC_DIR}/release/${TARGET}/make-memstick.sh \
920
		${INSTALLER_CHROOT_DIR} \
921
		${MEMSTICKSERIALPATH}
922

    
923
	if [ ! -f "${MEMSTICKSERIALPATH}" ]; then
924
		echo "ERROR! memstick serial image was not built"
925
		print_error_pfS
926
	fi
927

    
928
	gzip -qf $MEMSTICKSERIALPATH &
929
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
930

    
931
	echo ">>> MEMSTICKSERIAL created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
932
}
933

    
934
create_memstick_adi_image() {
935
	LOGFILE=${BUILDER_LOGS}/memstickadi.${TARGET}
936
	if [ "${MEMSTICKADIPATH}" = "" ]; then
937
		echo ">>> MEMSTICKADIPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
938
		return
939
	fi
940

    
941
	mkdir -p $(dirname ${MEMSTICKADIPATH})
942

    
943
	customize_stagearea_for_image "memstickadi"
944
	install_default_kernel ${DEFAULT_KERNEL}
945

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

    
948
	BOOTCONF=${INSTALLER_CHROOT_DIR}/boot.config
949
	LOADERCONF=${INSTALLER_CHROOT_DIR}/boot/loader.conf
950

    
951
	echo ">>> Activating serial console..." 2>&1 | tee -a ${LOGFILE}
952
	echo "-S115200 -h" > ${BOOTCONF}
953

    
954
	# Activate serial console+video console in loader.conf
955
	echo 'autoboot_delay="3"' > ${LOADERCONF}
956
	echo 'kern.cam.boot_delay=10000' >> ${LOADERCONF}
957
	echo 'boot_serial="YES"' >> ${LOADERCONF}
958
	echo 'console="comconsole"' >> ${LOADERCONF}
959
	echo 'comconsole_speed="115200"' >> ${LOADERCONF}
960
	echo 'comconsole_port="0x2F8"' >> ${LOADERCONF}
961
	echo 'hint.uart.0.flags="0x00"' >> ${LOADERCONF}
962
	echo 'hint.uart.1.flags="0x10"' >> ${LOADERCONF}
963

    
964
	cat ${BOOTCONF} >> ${FINAL_CHROOT_DIR}/boot.config
965
	cat ${LOADERCONF} >> ${FINAL_CHROOT_DIR}/boot/loader.conf
966

    
967
	create_distribution_tarball
968

    
969
	sh ${FREEBSD_SRC_DIR}/release/${TARGET}/make-memstick.sh \
970
		${INSTALLER_CHROOT_DIR} \
971
		${MEMSTICKADIPATH}
972

    
973
	if [ ! -f "${MEMSTICKADIPATH}" ]; then
974
		echo "ERROR! memstick ADI image was not built"
975
		print_error_pfS
976
	fi
977

    
978
	gzip -qf $MEMSTICKADIPATH &
979
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
980

    
981
	echo ">>> MEMSTICKADI created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
982
}
983

    
984
get_altabi_arch() {
985
	local _target_arch="$1"
986

    
987
	if [ "${_target_arch}" = "amd64" ]; then
988
		echo "x86:64"
989
	elif [ "${_target_arch}" = "i386" ]; then
990
		echo "x86:32"
991
	elif [ "${_target_arch}" = "armv7" ]; then
992
		echo "32:el:eabi:softfp"
993
	else
994
		echo ">>> ERROR: Invalid arch"
995
		print_error_pfS
996
	fi
997
}
998

    
999
# Create pkg conf on desired place with desired arch/branch
1000
setup_pkg_repo() {
1001
	if [ -z "${4}" ]; then
1002
		return
1003
	fi
1004

    
1005
	local _template="${1}"
1006
	local _target="${2}"
1007
	local _arch="${3}"
1008
	local _target_arch="${4}"
1009
	local _staging="${5}"
1010
	local _pkg_conf="${6}"
1011
	local _mirror_type="srv"
1012
	local _signature_type="fingerprints"
1013

    
1014
	if [ -z "${_template}" -o ! -f "${_template}" ]; then
1015
		echo ">>> ERROR: It was not possible to find pkg conf template ${_template}"
1016
		print_error_pfS
1017
	fi
1018

    
1019
	if [ -n "${_staging}" -a -n "${USE_PKG_REPO_STAGING}" ]; then
1020
		local _pkg_repo_server_devel=${PKG_REPO_SERVER_STAGING}
1021
		local _pkg_repo_branch_devel=${PKG_REPO_BRANCH_STAGING}
1022
		local _pkg_repo_server_release=${PKG_REPO_SERVER_STAGING}
1023
		local _pkg_repo_branch_release=${PKG_REPO_BRANCH_STAGING}
1024
	else
1025
		local _pkg_repo_server_devel=${PKG_REPO_SERVER_DEVEL}
1026
		local _pkg_repo_branch_devel=${PKG_REPO_BRANCH_DEVEL}
1027
		local _pkg_repo_server_release=${PKG_REPO_SERVER_RELEASE}
1028
		local _pkg_repo_branch_release=${PKG_REPO_BRANCH_RELEASE}
1029
	fi
1030

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

    
1033
	sed \
1034
		-e "s/%%ARCH%%/${_target_arch}/" \
1035
		-e "s/%%MIRROR_TYPE%%/${_mirror_type}/" \
1036
		-e "s/%%PKG_REPO_BRANCH_DEVEL%%/${_pkg_repo_branch_devel}/g" \
1037
		-e "s/%%PKG_REPO_BRANCH_RELEASE%%/${_pkg_repo_branch_release}/g" \
1038
		-e "s,%%PKG_REPO_SERVER_DEVEL%%,${_pkg_repo_server_devel},g" \
1039
		-e "s,%%PKG_REPO_SERVER_RELEASE%%,${_pkg_repo_server_release},g" \
1040
		-e "s,%%POUDRIERE_PORTS_NAME%%,${POUDRIERE_PORTS_NAME},g" \
1041
		-e "s/%%PRODUCT_NAME%%/${PRODUCT_NAME}/g" \
1042
		-e "s/%%REPO_BRANCH_PREFIX%%/${REPO_PATH_PREFIX}/g" \
1043
		-e "s/%%SIGNATURE_TYPE%%/${_signature_type}/" \
1044
		${_template} \
1045
		> ${_target}
1046

    
1047
	local ALTABI_ARCH=$(get_altabi_arch ${_target_arch})
1048

    
1049
	ABI=$(cat ${_template%%.conf}.abi 2>/dev/null \
1050
	    | sed -e "s/%%ARCH%%/${_target_arch}/g")
1051
	ALTABI=$(cat ${_template%%.conf}.altabi 2>/dev/null \
1052
	    | sed -e "s/%%ARCH%%/${ALTABI_ARCH}/g")
1053

    
1054
	if [ -n "${_pkg_conf}" -a -n "${ABI}" -a -n "${ALTABI}" ]; then
1055
		mkdir -p $(dirname ${_pkg_conf})
1056
		echo "ABI=${ABI}" > ${_pkg_conf}
1057
		echo "ALTABI=${ALTABI}" >> ${_pkg_conf}
1058
	fi
1059
}
1060

    
1061
depend_check() {
1062
	for _pkg in ${BUILDER_PKG_DEPENDENCIES}; do
1063
		if ! pkg info -e ${_pkg}; then
1064
			echo "Missing dependency (${_pkg})."
1065
			print_error_pfS
1066
		fi
1067
	done
1068
}
1069

    
1070
# This routine ensures any ports / binaries that the builder
1071
# system needs are on disk and ready for execution.
1072
builder_setup() {
1073
	# If Product-builder is already installed, just leave
1074
	if pkg info -e -q ${PRODUCT_NAME}-builder; then
1075
		return
1076
	fi
1077

    
1078
	if [ ! -f ${PKG_REPO_PATH} ]; then
1079
		[ -d $(dirname ${PKG_REPO_PATH}) ] \
1080
			|| mkdir -p $(dirname ${PKG_REPO_PATH})
1081

    
1082
		update_freebsd_sources
1083

    
1084
		local _arch=$(uname -m)
1085
		setup_pkg_repo \
1086
			${PKG_REPO_BUILD} \
1087
			${PKG_REPO_PATH} \
1088
			${_arch} \
1089
			${_arch} \
1090
			"staging"
1091

    
1092
		# Use fingerprint keys from repo
1093
		sed -i '' -e "/fingerprints:/ s,\"/,\"${BUILDER_ROOT}/src/," \
1094
			${PKG_REPO_PATH}
1095
	fi
1096

    
1097
	pkg install ${PRODUCT_NAME}-builder
1098
}
1099

    
1100
# Updates FreeBSD sources
1101
update_freebsd_sources() {
1102
	if [ "${1}" = "full" ]; then
1103
		local _full=1
1104
		local _clone_params=""
1105
	else
1106
		local _full=0
1107
		local _clone_params="--depth 1 --single-branch"
1108
	fi
1109

    
1110
	if [ -n "${NO_BUILDWORLD}" -a -n "${NO_BUILDKERNEL}" ]; then
1111
		echo ">>> NO_BUILDWORLD and NO_BUILDKERNEL set, skipping update of freebsd sources" | tee -a ${LOGFILE}
1112
		return
1113
	fi
1114

    
1115
	echo ">>> Obtaining FreeBSD sources (${FREEBSD_BRANCH})..."
1116
	${BUILDER_SCRIPTS}/git_checkout.sh \
1117
		-r ${FREEBSD_REPO_BASE} \
1118
		-d ${FREEBSD_SRC_DIR} \
1119
		-b ${FREEBSD_BRANCH}
1120

    
1121
	if [ $? -ne 0 -o ! -d "${FREEBSD_SRC_DIR}/.git" ]; then
1122
		echo ">>> ERROR: It was not possible to clone FreeBSD src repo"
1123
		print_error_pfS
1124
	fi
1125

    
1126
	if [ -n "${GIT_FREEBSD_COSHA1}" ]; then
1127
		echo -n ">>> Checking out desired commit (${GIT_FREEBSD_COSHA1})... "
1128
		( git -C  ${FREEBSD_SRC_DIR} checkout ${GIT_FREEBSD_COSHA1} ) 2>&1 | \
1129
			grep -C3 -i -E 'error|fatal'
1130
		echo "Done!"
1131
	fi
1132

    
1133
	if [ "${PRODUCT_NAME}" = "pfSense" -a -n "${GNID_REPO_BASE}" ]; then
1134
		echo ">>> Obtaining gnid sources..."
1135
		${BUILDER_SCRIPTS}/git_checkout.sh \
1136
			-r ${GNID_REPO_BASE} \
1137
			-d ${GNID_SRC_DIR} \
1138
			-b ${GNID_BRANCH}
1139
	fi
1140
}
1141

    
1142
pkg_chroot() {
1143
	local _root="${1}"
1144
	shift
1145

    
1146
	if [ $# -eq 0 ]; then
1147
		return -1
1148
	fi
1149

    
1150
	if [ -z "${_root}" -o "${_root}" = "/" -o ! -d "${_root}" ]; then
1151
		return -1
1152
	fi
1153

    
1154
	mkdir -p \
1155
		${SCRATCHDIR}/pkg_cache \
1156
		${_root}/var/cache/pkg \
1157
		${_root}/dev
1158

    
1159
	/sbin/mount -t nullfs ${SCRATCHDIR}/pkg_cache ${_root}/var/cache/pkg
1160
	/sbin/mount -t devfs devfs ${_root}/dev
1161
	cp -f /etc/resolv.conf ${_root}/etc/resolv.conf
1162
	touch ${BUILDER_LOGS}/install_pkg_install_ports.txt
1163
	local _params=""
1164
	if [ -f "${_root}/tmp/pkg/pkg-repos/repo.conf" ]; then
1165
		_params="--repo-conf-dir /tmp/pkg/pkg-repos "
1166
	fi
1167
	if [ -f "${_root}/tmp/pkg/pkg.conf" ]; then
1168
		_params="${_params} --config /tmp/pkg/pkg.conf "
1169
	fi
1170
	script -aq ${BUILDER_LOGS}/install_pkg_install_ports.txt \
1171
		chroot ${_root} pkg ${_params}$@ >/dev/null 2>&1
1172
	local result=$?
1173
	rm -f ${_root}/etc/resolv.conf
1174
	/sbin/umount -f ${_root}/dev
1175
	/sbin/umount -f ${_root}/var/cache/pkg
1176

    
1177
	return $result
1178
}
1179

    
1180

    
1181
pkg_chroot_add() {
1182
	if [ -z "${1}" -o -z "${2}" ]; then
1183
		return 1
1184
	fi
1185

    
1186
	local _target="${1}"
1187
	local _pkg="$(get_pkg_name ${2}).txz"
1188

    
1189
	if [ ! -d "${_target}" ]; then
1190
		echo ">>> ERROR: Target dir ${_target} not found"
1191
		print_error_pfS
1192
	fi
1193

    
1194
	if [ ! -f ${CORE_PKG_ALL_PATH}/${_pkg} ]; then
1195
		echo ">>> ERROR: Package ${_pkg} not found"
1196
		print_error_pfS
1197
	fi
1198

    
1199
	cp ${CORE_PKG_ALL_PATH}/${_pkg} ${_target}
1200
	pkg_chroot ${_target} add /${_pkg}
1201
	rm -f ${_target}/${_pkg}
1202
}
1203

    
1204
pkg_bootstrap() {
1205
	local _root=${1:-"${STAGE_CHROOT_DIR}"}
1206

    
1207
	setup_pkg_repo \
1208
		${PKG_REPO_BUILD} \
1209
		${_root}${PKG_REPO_PATH} \
1210
		${TARGET} \
1211
		${TARGET_ARCH} \
1212
		"staging"
1213

    
1214
	pkg_chroot ${_root} bootstrap -f
1215
}
1216

    
1217
# This routine assists with installing various
1218
# freebsd ports files into the pfsense-fs staging
1219
# area.
1220
install_pkg_install_ports() {
1221
	local MAIN_PKG="${1}"
1222

    
1223
	if [ -z "${MAIN_PKG}" ]; then
1224
		MAIN_PKG=${PRODUCT_NAME}
1225
	fi
1226

    
1227
	echo ">>> Installing pkg repository in chroot (${STAGE_CHROOT_DIR})..."
1228

    
1229
	[ -d ${STAGE_CHROOT_DIR}/var/cache/pkg ] || \
1230
		mkdir -p ${STAGE_CHROOT_DIR}/var/cache/pkg
1231

    
1232
	[ -d ${SCRATCHDIR}/pkg_cache ] || \
1233
		mkdir -p ${SCRATCHDIR}/pkg_cache
1234

    
1235
	echo -n ">>> Installing built ports (packages) in chroot (${STAGE_CHROOT_DIR})... "
1236
	# First mark all packages as automatically installed
1237
	pkg_chroot ${STAGE_CHROOT_DIR} set -A 1 -a
1238
	# Install all necessary packages
1239
	if ! pkg_chroot ${STAGE_CHROOT_DIR} install ${MAIN_PKG} ${custom_package_list}; then
1240
		echo "Failed!"
1241
		print_error_pfS
1242
	fi
1243
	# Make sure required packages are set as non-automatic
1244
	pkg_chroot ${STAGE_CHROOT_DIR} set -A 0 pkg ${MAIN_PKG} ${custom_package_list}
1245
	# pkg and MAIN_PKG are vital
1246
	pkg_chroot ${STAGE_CHROOT_DIR} set -y -v 1 pkg ${MAIN_PKG}
1247
	# Remove unnecessary packages
1248
	pkg_chroot ${STAGE_CHROOT_DIR} autoremove
1249
	echo "Done!"
1250
}
1251

    
1252
staginareas_clean_each_run() {
1253
	echo -n ">>> Cleaning build directories: "
1254
	if [ -d "${FINAL_CHROOT_DIR}" ]; then
1255
		BASENAME=$(basename ${FINAL_CHROOT_DIR})
1256
		echo -n "$BASENAME "
1257
		chflags -R noschg ${FINAL_CHROOT_DIR} 2>&1 >/dev/null
1258
		rm -rf ${FINAL_CHROOT_DIR}/* 2>/dev/null
1259
	fi
1260
	echo "Done!"
1261
}
1262

    
1263
# Imported from FreeSBIE
1264
buildkernel() {
1265
	local _kernconf=${1:-${KERNCONF}}
1266

    
1267
	if [ -n "${NO_BUILDKERNEL}" ]; then
1268
		echo ">>> NO_BUILDKERNEL set, skipping build" | tee -a ${LOGFILE}
1269
		return
1270
	fi
1271

    
1272
	if [ -z "${_kernconf}" ]; then
1273
		echo ">>> ERROR: No kernel configuration defined probably this is not what you want! STOPPING!" | tee -a ${LOGFILE}
1274
		print_error_pfS
1275
	fi
1276

    
1277
	local _old_kernconf=${KERNCONF}
1278
	export KERNCONF=${_kernconf}
1279

    
1280
	echo ">>> $(LC_ALL=C date) - Starting build kernel for ${TARGET} architecture..." | tee -a ${LOGFILE}
1281
	script -aq $LOGFILE ${BUILDER_SCRIPTS}/build_freebsd.sh -W -s ${FREEBSD_SRC_DIR} \
1282
		|| print_error_pfS
1283
	echo ">>> $(LC_ALL=C date) - Finished build kernel for ${TARGET} architecture..." | tee -a ${LOGFILE}
1284

    
1285
	export KERNCONF=${_old_kernconf}
1286
}
1287

    
1288
# Imported from FreeSBIE
1289
installkernel() {
1290
	local _destdir=${1:-${KERNEL_DESTDIR}}
1291
	local _kernconf=${2:-${KERNCONF}}
1292

    
1293
	if [ -z "${_kernconf}" ]; then
1294
		echo ">>> ERROR: No kernel configuration defined probably this is not what you want! STOPPING!" | tee -a ${LOGFILE}
1295
		print_error_pfS
1296
	fi
1297

    
1298
	local _old_kernconf=${KERNCONF}
1299
	export KERNCONF=${_kernconf}
1300

    
1301
	mkdir -p ${STAGE_CHROOT_DIR}/boot
1302
	echo ">>> Installing kernel (${_kernconf}) for ${TARGET} architecture..." | tee -a ${LOGFILE}
1303
	script -aq $LOGFILE ${BUILDER_SCRIPTS}/install_freebsd.sh -W -D -z \
1304
		-s ${FREEBSD_SRC_DIR} \
1305
		-d ${_destdir} \
1306
		|| print_error_pfS
1307

    
1308
	export KERNCONF=${_old_kernconf}
1309
}
1310

    
1311
# Launch is ran first to setup a few variables that we need
1312
# Imported from FreeSBIE
1313
launch() {
1314
	if [ "$(id -u)" != "0" ]; then
1315
		echo "Sorry, this must be done as root."
1316
	fi
1317

    
1318
	echo ">>> Operation $0 has started at $(date)"
1319
}
1320

    
1321
finish() {
1322
	echo ">>> Operation $0 has ended at $(date)"
1323
}
1324

    
1325
pkg_repo_rsync() {
1326
	local _repo_path_param="${1}"
1327
	local _ignore_final_rsync="${2}"
1328
	local _aws_sync_cmd="aws s3 sync --quiet --exclude '.real*/*' --exclude '.latest/*'"
1329

    
1330
	if [ -z "${_repo_path_param}" -o ! -d "${_repo_path_param}" ]; then
1331
		return
1332
	fi
1333

    
1334
	if [ -n "${SKIP_FINAL_RSYNC}" ]; then
1335
		_ignore_final_rsync="1"
1336
	fi
1337

    
1338
	# Sanitize path
1339
	_repo_path=$(realpath ${_repo_path_param})
1340

    
1341
	local _repo_dir=$(dirname ${_repo_path})
1342
	local _repo_base=$(basename ${_repo_path})
1343

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

    
1347
	if [ -z "${LOGFILE}" ]; then
1348
		local _logfile="/dev/null"
1349
	else
1350
		local _logfile="${LOGFILE}"
1351
	fi
1352

    
1353
	if [ -n "${PKG_REPO_SIGNING_COMMAND}" -a -z "${DO_NOT_SIGN_PKG_REPO}" ]; then
1354
		# Detect poudriere directory structure
1355
		if [ -L "${_repo_path}/.latest" ]; then
1356
			local _real_repo_path=$(readlink -f ${_repo_path}/.latest)
1357
		else
1358
			local _real_repo_path=${_repo_path}
1359
		fi
1360

    
1361
		echo -n ">>> Signing repository... " | tee -a ${_logfile}
1362
		############ ATTENTION ##############
1363
		#
1364
		# For some reason pkg-repo fail without / in the end of directory name
1365
		# so removing it will break command
1366
		#
1367
		# https://github.com/freebsd/pkg/issues/1364
1368
		#
1369
		if script -aq ${_logfile} pkg repo ${_real_repo_path}/ \
1370
		    signing_command: ${PKG_REPO_SIGNING_COMMAND} >/dev/null 2>&1; then
1371
			echo "Done!" | tee -a ${_logfile}
1372
		else
1373
			echo "Failed!" | tee -a ${_logfile}
1374
			echo ">>> ERROR: An error occurred trying to sign repo"
1375
			print_error_pfS
1376
		fi
1377

    
1378
		local _pkgfile="${_repo_path}/Latest/pkg.pkg"
1379
		if [ -e ${_pkgfile} ]; then
1380
			echo -n ">>> Signing Latest/pkg.pkg for bootstrapping... " | tee -a ${_logfile}
1381

    
1382
			if sha256 -q ${_pkgfile} | ${PKG_REPO_SIGNING_COMMAND} \
1383
			    > ${_pkgfile}.sig 2>/dev/null; then
1384
				# XXX Temporary workaround to create link to pkg sig
1385
				[ -e ${_repo_path}/Latest/pkg.txz ] && \
1386
					ln -sf pkg.pkg.sig ${_repo_path}/Latest/pkg.txz.sig
1387
				echo "Done!" | tee -a ${_logfile}
1388
			else
1389
				echo "Failed!" | tee -a ${_logfile}
1390
				echo ">>> ERROR: An error occurred trying to sign Latest/pkg.txz"
1391
				print_error_pfS
1392
			fi
1393
		fi
1394
	fi
1395

    
1396
	if [ -z "${UPLOAD}" ]; then
1397
		return
1398
	fi
1399

    
1400
	local _pkg_rsync_site
1401
	for _pkg_rsync_site in ${PKG_RSYNC_HOSTS}; do
1402
		eval _pkg_rsync_hostname=\$PKG_RSYNC_HOSTNAME_$_pkg_rsync_site
1403
		if [ -z "${_pkg_rsync_hostname}" ]; then
1404
			echo "PKG_RSYNC_HOSTNAME_$_pkg_rsync_site is empty, skipping.."
1405
			continue
1406
		fi
1407
		# Make sure destination directory exist
1408
		ssh -o StrictHostKeyChecking=no -p ${PKG_RSYNC_SSH_PORT} \
1409
			${PKG_RSYNC_USERNAME}@${_pkg_rsync_hostname} \
1410
			"mkdir -p ${PKG_RSYNC_DESTDIR}"
1411

    
1412
		echo -n ">>> Sending updated repository to ${_pkg_rsync_hostname}... " | tee -a ${_logfile}
1413
		if script -aq ${_logfile} rsync -Have "ssh -o StrictHostKeyChecking=no -p ${PKG_RSYNC_SSH_PORT}" \
1414
			--timeout=60 --delete-delay ${_repo_path} \
1415
			${PKG_RSYNC_USERNAME}@${_pkg_rsync_hostname}:${PKG_RSYNC_DESTDIR} >> ${BUILDER_LOGS}/rsync.log 2>&1
1416
		then
1417
			echo "Done!" | tee -a ${_logfile}
1418
		else
1419
			echo "Failed!" | tee -a ${_logfile}
1420
			echo ">>> ERROR: An error occurred sending repo to remote hostname"
1421
			print_error_pfS
1422
		fi
1423

    
1424
		if [ -z "${USE_PKG_REPO_STAGING}" -o -n "${_ignore_final_rsync}" ]; then
1425
			return
1426
		fi
1427

    
1428
		if [ -n "${_IS_RELEASE}" -o "${_repo_path_param}" = "${CORE_PKG_PATH}" ]; then
1429
			local _pkg_final_rsync_hostname
1430
			eval _pkg_final_rsync_hostname=\$PKG_FINAL_RSYNC_HOSTNAME_$_pkg_rsync_site
1431
			if [ -z "${_pkg_final_rsync_hostname}" ]; then
1432
				_pkg_final_rsync_hostname="$_pkg_rsync_hostname"
1433
			fi
1434

    
1435
			# Send .real* directories first to prevent having a broken repo while transfer happens
1436
			local _cmd="rsync -Have \"ssh -o StrictHostKeyChecking=no -p ${PKG_FINAL_RSYNC_SSH_PORT}\" \
1437
				--timeout=60 ${PKG_RSYNC_DESTDIR}/./${_repo_base%%-core}* \
1438
				--include=\"/*\" --include=\"*/.real*\" --include=\"*/.real*/***\" \
1439
				--exclude=\"*\" \
1440
				${PKG_FINAL_RSYNC_USERNAME}@${_pkg_final_rsync_hostname}:${PKG_FINAL_RSYNC_DESTDIR}"
1441

    
1442
			echo -n ">>> Sending updated packages to ${_pkg_final_rsync_hostname}... " | tee -a ${_logfile}
1443
			if script -aq ${_logfile} ssh -o StrictHostKeyChecking=no -p ${PKG_RSYNC_SSH_PORT} \
1444
				${PKG_RSYNC_USERNAME}@${_pkg_rsync_hostname} ${_cmd} >> ${BUILDER_LOGS}/rsync.log 2>&1; then
1445
				echo "Done!" | tee -a ${_logfile}
1446
			else
1447
				echo "Failed!" | tee -a ${_logfile}
1448
				echo ">>> ERROR: An error occurred sending repo to final hostname"
1449
				print_error_pfS
1450
			fi
1451

    
1452
			_cmd="rsync -Have \"ssh -o StrictHostKeyChecking=no -p ${PKG_FINAL_RSYNC_SSH_PORT}\" \
1453
				--timeout=60 --delete-delay ${PKG_RSYNC_DESTDIR}/./${_repo_base%%-core}* \
1454
				${PKG_FINAL_RSYNC_USERNAME}@${_pkg_final_rsync_hostname}:${PKG_FINAL_RSYNC_DESTDIR}"
1455

    
1456
			echo -n ">>> Sending updated repositories metadata to ${_pkg_final_rsync_hostname}... " | tee -a ${_logfile}
1457
			if script -aq ${_logfile} ssh -o StrictHostKeyChecking=no -p ${PKG_RSYNC_SSH_PORT} \
1458
				${PKG_RSYNC_USERNAME}@${_pkg_rsync_hostname} ${_cmd} >> ${BUILDER_LOGS}/rsync.log 2>&1; then
1459
				echo "Done!" | tee -a ${_logfile}
1460
			else
1461
				echo "Failed!" | tee -a ${_logfile}
1462
				echo ">>> ERROR: An error occurred sending repo to final hostname"
1463
				print_error_pfS
1464
			fi
1465

    
1466
			if [ -z "${PKG_FINAL_S3_PATH}" ]; then
1467
				continue
1468
			fi
1469

    
1470
			local _repos=$(ssh -o StrictHostKeyChecking=no -p ${PKG_FINAL_RSYNC_SSH_PORT} \
1471
			    ${PKG_FINAL_RSYNC_USERNAME}@${_pkg_final_rsync_hostname} \
1472
			    "ls -1d ${PKG_FINAL_RSYNC_DESTDIR}/${_repo_base%%-core}*")
1473
			for _repo in ${_repos}; do
1474
				echo -n ">>> Sending updated packages to AWS ${PKG_FINAL_S3_PATH}... " | tee -a ${_logfile}
1475
				if script -aq ${_logfile} ssh -o StrictHostKeyChecking=no -p ${PKG_FINAL_RSYNC_SSH_PORT} \
1476
				    ${PKG_FINAL_RSYNC_USERNAME}@${_pkg_final_rsync_hostname} \
1477
				    "${_aws_sync_cmd} ${_repo} ${PKG_FINAL_S3_PATH}/$(basename ${_repo})"; then
1478
					echo "Done!" | tee -a ${_logfile}
1479
				else
1480
					echo "Failed!" | tee -a ${_logfile}
1481
					echo ">>> ERROR: An error occurred sending files to AWS S3"
1482
					print_error_pfS
1483
				fi
1484
				echo -n ">>> Cleaning up packages at AWS ${PKG_FINAL_S3_PATH}... " | tee -a ${_logfile}
1485
				if script -aq ${_logfile} ssh -o StrictHostKeyChecking=no -p ${PKG_FINAL_RSYNC_SSH_PORT} \
1486
				    ${PKG_FINAL_RSYNC_USERNAME}@${_pkg_final_rsync_hostname} \
1487
				    "${_aws_sync_cmd} --delete ${_repo} ${PKG_FINAL_S3_PATH}/$(basename ${_repo})"; then
1488
					echo "Done!" | tee -a ${_logfile}
1489
				else
1490
					echo "Failed!" | tee -a ${_logfile}
1491
					echo ">>> ERROR: An error occurred sending files to AWS S3"
1492
					print_error_pfS
1493
				fi
1494
			done
1495
		fi
1496
	done
1497
}
1498

    
1499
poudriere_possible_archs() {
1500
	local _arch=$(uname -m)
1501
	local _archs=""
1502

    
1503
	# If host is amd64, we'll create both repos, and if possible armv7
1504
	if [ "${_arch}" = "amd64" ]; then
1505
		_archs="amd64.amd64"
1506

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

    
1511
			if binmiscctl lookup armv7 >/dev/null 2>&1; then
1512
				_archs="${_archs} arm.armv7"
1513
			fi
1514
		fi
1515
	fi
1516

    
1517
	if [ -n "${ARCH_LIST}" ]; then
1518
		local _found=0
1519
		for _desired_arch in ${ARCH_LIST}; do
1520
			_found=0
1521
			for _possible_arch in ${_archs}; do
1522
				if [ "${_desired_arch}" = "${_possible_arch}" ]; then
1523
					_found=1
1524
					break
1525
				fi
1526
			done
1527
			if [ ${_found} -eq 0 ]; then
1528
				echo ">>> ERROR: Impossible to build for arch: ${_desired_arch}"
1529
				print_error_pfS
1530
			fi
1531
		done
1532
		_archs="${ARCH_LIST}"
1533
	fi
1534

    
1535
	echo ${_archs}
1536
}
1537

    
1538
poudriere_jail_name() {
1539
	local _jail_arch="${1}"
1540

    
1541
	if [ -z "${_jail_arch}" ]; then
1542
		return 1
1543
	fi
1544

    
1545
	# Remove arch
1546
	echo "${PRODUCT_NAME}_${POUDRIERE_BRANCH}_${_jail_arch##*.}"
1547
}
1548

    
1549
poudriere_rename_ports() {
1550
	if [ "${PRODUCT_NAME}" = "pfSense" ]; then
1551
		return;
1552
	fi
1553

    
1554
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1555

    
1556
	local _ports_dir="/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}"
1557

    
1558
	echo -n ">>> Renaming product ports on ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
1559
	for d in $(find ${_ports_dir} -depth 2 -type d -name '*pfSense*'); do
1560
		local _pdir=$(dirname ${d})
1561
		local _pname=$(echo $(basename ${d}) | sed "s,pfSense,${PRODUCT_NAME},")
1562
		local _plist=""
1563
		local _pdescr=""
1564

    
1565
		if [ -e ${_pdir}/${_pname} ]; then
1566
			rm -rf ${_pdir}/${_pname}
1567
		fi
1568

    
1569
		cp -r ${d} ${_pdir}/${_pname}
1570

    
1571
		if [ -f ${_pdir}/${_pname}/pkg-plist ]; then
1572
			_plist=${_pdir}/${_pname}/pkg-plist
1573
		fi
1574

    
1575
		if [ -f ${_pdir}/${_pname}/pkg-descr ]; then
1576
			_pdescr=${_pdir}/${_pname}/pkg-descr
1577
		fi
1578

    
1579
		sed -i '' -e "s,pfSense,${PRODUCT_NAME},g" \
1580
			  -e "s,https://www.pfsense.org,${PRODUCT_URL},g" \
1581
			  -e "/^MAINTAINER=/ s,^.*$,MAINTAINER=	${PRODUCT_EMAIL}," \
1582
			${_pdir}/${_pname}/Makefile ${_pdescr} ${_plist}
1583

    
1584
		# PHP module is special
1585
		if echo "${_pname}" | grep -q "^php[0-9]*-${PRODUCT_NAME}-module"; then
1586
			local _product_capital=$(echo ${PRODUCT_NAME} | tr '[a-z]' '[A-Z]')
1587
			sed -i '' -e "s,PHP_PFSENSE,PHP_${_product_capital},g" \
1588
				  -e "s,PFSENSE_SHARED_LIBADD,${_product_capital}_SHARED_LIBADD,g" \
1589
				  -e "s,pfSense,${PRODUCT_NAME},g" \
1590
				  -e "s,pfSense.c,${PRODUCT_NAME}\.c,g" \
1591
				${_pdir}/${_pname}/files/config.m4
1592

    
1593
			sed -i '' -e "s,COMPILE_DL_PFSENSE,COMPILE_DL_${_product_capital}," \
1594
				  -e "s,pfSense_module_entry,${PRODUCT_NAME}_module_entry,g" \
1595
				  -e "s,php_pfSense.h,php_${PRODUCT_NAME}\.h,g" \
1596
				  -e "/ZEND_GET_MODULE/ s,pfSense,${PRODUCT_NAME}," \
1597
				  -e "/PHP_PFSENSE_WORLD_EXTNAME/ s,pfSense,${PRODUCT_NAME}," \
1598
				${_pdir}/${_pname}/files/pfSense.c \
1599
				${_pdir}/${_pname}/files/dummynet.c \
1600
				${_pdir}/${_pname}/files/php_pfSense.h
1601
		fi
1602

    
1603
		if [ -d ${_pdir}/${_pname}/files ]; then
1604
			for fd in $(find ${_pdir}/${_pname}/files -name '*pfSense*'); do
1605
				local _fddir=$(dirname ${fd})
1606
				local _fdname=$(echo $(basename ${fd}) | sed "s,pfSense,${PRODUCT_NAME},")
1607

    
1608
				mv ${fd} ${_fddir}/${_fdname}
1609
			done
1610
		fi
1611
	done
1612
	echo "Done!" | tee -a ${LOGFILE}
1613
}
1614

    
1615
poudriere_create_ports_tree() {
1616
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1617

    
1618
	if ! poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
1619
		local _branch=""
1620
		if [ -z "${POUDRIERE_PORTS_GIT_URL}" ]; then
1621
			echo ">>> ERROR: POUDRIERE_PORTS_GIT_URL is not defined"
1622
			print_error_pfS
1623
		fi
1624
		if [ -n "${POUDRIERE_PORTS_GIT_BRANCH}" ]; then
1625
			_branch="${POUDRIERE_PORTS_GIT_BRANCH}"
1626
		fi
1627
		echo -n ">>> Creating poudriere ports tree, it may take some time... " | tee -a ${LOGFILE}
1628
		if [ "${AWS}" = 1 ]; then
1629
			set -e
1630
			script -aq ${LOGFILE} poudriere ports -c -p "${POUDRIERE_PORTS_NAME}" -m none
1631
			script -aq ${LOGFILE} zfs create ${ZFS_TANK}/poudriere/ports/${POUDRIERE_PORTS_NAME}
1632

    
1633
			# If S3 doesn't contain stashed ports tree, create one
1634
			if ! aws_exec s3 ls s3://pfsense-engineering-build-pkg/${FLAVOR}-ports.tz >/dev/null 2>&1; then
1635
				mkdir ${SCRATCHDIR}/${FLAVOR}-ports
1636
				${BUILDER_SCRIPTS}/git_checkout.sh \
1637
				    -r ${POUDRIERE_PORTS_GIT_URL} \
1638
				    -d ${SCRATCHDIR}/${FLAVOR}-ports \
1639
				    -b ${POUDRIERE_PORTS_GIT_BRANCH}
1640

    
1641
				tar --zstd -C ${SCRATCHDIR} -cf ${FLAVOR}-ports.tz ${FLAVOR}-ports
1642
				aws_exec s3 cp ${FLAVOR}-ports.tz s3://pfsense-engineering-build-pkg/${FLAVOR}-ports.tz --no-progress
1643
			else
1644
				# Download local copy of the ports tree stashed in S3
1645
				echo ">>>  Downloading cached copy of the ports tree from S3.." | tee -a ${LOGFILE}
1646
				aws_exec s3 cp s3://pfsense-engineering-build-pkg/${FLAVOR}-ports.tz . --no-progress
1647
			fi
1648

    
1649
			script -aq ${LOGFILE} tar --strip-components 1 -xf ${FLAVOR}-ports.tz -C /usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}
1650
			# Update the ports tree
1651
			(
1652
				cd /usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}
1653
				echo ">>>  Updating cached copy of the ports tree from git.." | tee -a ${LOGFILE}
1654
				script -aq ${LOGFILE} git pull
1655
				script -aq ${LOGFILE} git checkout ${_branch}
1656
			)
1657
			set +e
1658
		else
1659
			if ! script -aq ${LOGFILE} poudriere ports -c -p "${POUDRIERE_PORTS_NAME}" -m git -U ${POUDRIERE_PORTS_GIT_URL} -B ${_branch} >/dev/null 2>&1; then
1660
				echo "" | tee -a ${LOGFILE}
1661
				echo ">>> ERROR: Error creating poudriere ports tree, aborting..." | tee -a ${LOGFILE}
1662
				print_error_pfS
1663
			fi
1664
		fi
1665
		echo "Done!" | tee -a ${LOGFILE}
1666
		poudriere_rename_ports
1667
	fi
1668
}
1669

    
1670
poudriere_init() {
1671
	local _error=0
1672
	local _archs=$(poudriere_possible_archs)
1673

    
1674
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1675

    
1676
	# Sanity checks
1677
	if [ -z "${ZFS_TANK}" ]; then
1678
		echo ">>> ERROR: \$ZFS_TANK is empty" | tee -a ${LOGFILE}
1679
		error=1
1680
	fi
1681

    
1682
	if [ -z "${ZFS_ROOT}" ]; then
1683
		echo ">>> ERROR: \$ZFS_ROOT is empty" | tee -a ${LOGFILE}
1684
		error=1
1685
	fi
1686

    
1687
	if [ -z "${POUDRIERE_PORTS_NAME}" ]; then
1688
		echo ">>> ERROR: \$POUDRIERE_PORTS_NAME is empty" | tee -a ${LOGFILE}
1689
		error=1
1690
	fi
1691

    
1692
	if [ ${_error} -eq 1 ]; then
1693
		print_error_pfS
1694
	fi
1695

    
1696
	# Check if zpool exists
1697
	if ! zpool list ${ZFS_TANK} >/dev/null 2>&1; then
1698
		echo ">>> ERROR: ZFS tank ${ZFS_TANK} not found, please create it and try again..." | tee -a ${LOGFILE}
1699
		print_error_pfS
1700
	fi
1701

    
1702
	# Check if zfs rootfs exists
1703
	if ! zfs list ${ZFS_TANK}${ZFS_ROOT} >/dev/null 2>&1; then
1704
		echo -n ">>> Creating ZFS filesystem ${ZFS_TANK}${ZFS_ROOT}... "
1705
		if zfs create -o atime=off -o mountpoint=/usr/local${ZFS_ROOT} \
1706
		    ${ZFS_TANK}${ZFS_ROOT} >/dev/null 2>&1; then
1707
			echo "Done!"
1708
		else
1709
			echo "Failed!"
1710
			print_error_pfS
1711
		fi
1712
	fi
1713

    
1714
	# Make sure poudriere is installed
1715
	if [ ! -f /usr/local/bin/poudriere ]; then
1716
		echo ">>> Installing poudriere..." | tee -a ${LOGFILE}
1717
		if ! pkg install poudriere >/dev/null 2>&1; then
1718
			echo ">>> ERROR: poudriere was not installed, aborting..." | tee -a ${LOGFILE}
1719
			print_error_pfS
1720
		fi
1721
	fi
1722

    
1723
	# Create poudriere.conf
1724
	if [ -z "${POUDRIERE_PORTS_GIT_URL}" ]; then
1725
		echo ">>> ERROR: POUDRIERE_PORTS_GIT_URL is not defined"
1726
		print_error_pfS
1727
	fi
1728

    
1729
	# PARALLEL_JOBS us ncpu / 4 for best performance
1730
	local _parallel_jobs=$(sysctl -qn hw.ncpu)
1731
	_parallel_jobs=$((_parallel_jobs / 4))
1732

    
1733
	echo ">>> Creating poudriere.conf" | tee -a ${LOGFILE}
1734
	cat <<EOF >/usr/local/etc/poudriere.conf
1735
ZPOOL=${ZFS_TANK}
1736
ZROOTFS=${ZFS_ROOT}
1737
RESOLV_CONF=/etc/resolv.conf
1738
BASEFS=/usr/local/poudriere
1739
USE_PORTLINT=no
1740
USE_TMPFS=yes
1741
NOLINUX=yes
1742
DISTFILES_CACHE=/usr/ports/distfiles
1743
CHECK_CHANGED_OPTIONS=yes
1744
CHECK_CHANGED_DEPS=yes
1745
ATOMIC_PACKAGE_REPOSITORY=yes
1746
COMMIT_PACKAGES_ON_FAILURE=no
1747
ALLOW_MAKE_JOBS=yes
1748
PARALLEL_JOBS=${_parallel_jobs}
1749
EOF
1750

    
1751
	if pkg info -e ccache; then
1752
	cat <<EOF >>/usr/local/etc/poudriere.conf
1753
CCACHE_DIR=/var/cache/ccache
1754
EOF
1755
	fi
1756

    
1757
	# Create specific items conf
1758
	[ ! -d /usr/local/etc/poudriere.d ] \
1759
		&& mkdir -p /usr/local/etc/poudriere.d
1760

    
1761
	# Create DISTFILES_CACHE if it doesn't exist
1762
	if [ ! -d /usr/ports/distfiles ]; then
1763
		mkdir -p /usr/ports/distfiles
1764
	fi
1765

    
1766
	if [ "${AWS}" = 1 ]; then
1767
		# Find the distfiles cache for our branch, but fall back to devel cache if it does not exist
1768
		if [ "${FLAVOR}" = "Plus" ]; then
1769
			DEFAULT_BRANCH="plus-devel"
1770
		else
1771
			DEFAULT_BRANCH="devel"
1772
		fi
1773

    
1774
		if [ "${POUDRIERE_PORTS_GIT_BRANCH}" = "${DEFAULT_BRANCH}" ]; then
1775
			DISTFILES="${FLAVOR}-${POUDRIERE_PORTS_GIT_BRANCH}-distfiles"
1776
		else
1777
			if aws_exec s3 ls s3://pfsense-engineering-build-pkg/${FLAVOR}-${POUDRIERE_PORTS_GIT_BRANCH}-distfiles.tar >/dev/null 2>&1; then
1778
				DISTFILES="${FLAVOR}-${POUDRIERE_PORTS_GIT_BRANCH}-distfiles"
1779
			else
1780
				DISTFILES="${FLAVOR}-${DEFAULT_BRANCH}-distfiles"
1781
				echo ">>> ${FLAVOR}-${POUDRIERE_PORTS_GIT_BRANCH}-distfiles.tar, not found."
1782
				echo ">>> Falling back to ${DISTFILES}.tar"
1783
			fi
1784
		fi
1785

    
1786
		if aws_exec s3 ls s3://pfsense-engineering-build-pkg/${DISTFILES}.tar >/dev/null 2>&1; then
1787
			# Download a copy of the distfiles from S3
1788
			echo ">>> Downloading distfile cache ${DISTFILES} from S3.." | tee -a ${LOGFILE}
1789
			aws_exec s3 cp s3://pfsense-engineering-build-pkg/${DISTFILES}.tar . --no-progress
1790
			script -aq ${LOGFILE} tar -xf ${DISTFILES}.tar -C /usr/ports/distfiles
1791
			# Save a list of distfiles
1792
			find /usr/ports/distfiles > pre-build-distfile-list
1793
		else
1794
			echo ">>> No distfile cache found, all distfiles will be fetched."
1795
			touch pre-build-distfile-list
1796
		fi
1797
	fi
1798

    
1799
	# Remove old jails
1800
	for jail_arch in ${_archs}; do
1801
		jail_name=$(poudriere_jail_name ${jail_arch})
1802

    
1803
		if poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
1804
			echo ">>> Poudriere jail ${jail_name} already exists, deleting it..." | tee -a ${LOGFILE}
1805
			poudriere jail -d -j "${jail_name}"
1806
		fi
1807
	done
1808

    
1809
	# Remove old ports tree
1810
	if poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
1811
		echo ">>> Poudriere ports tree ${POUDRIERE_PORTS_NAME} already exists, deleting it..." | tee -a ${LOGFILE}
1812
		poudriere ports -d -p "${POUDRIERE_PORTS_NAME}"
1813
		if [ "${AWS}" = 1 ]; then
1814
			for d in `zfs list -o name`; do
1815
				if [ "${d}" = "${ZFS_TANK}/poudriere/ports/${POUDRIERE_PORTS_NAME}" ]; then
1816
					script -aq ${LOGFILE} zfs destroy ${ZFS_TANK}/poudriere/ports/${POUDRIERE_PORTS_NAME}
1817
				fi
1818
			done
1819
		fi
1820
	fi
1821

    
1822
	local native_xtools=""
1823
	# Now we are ready to create jails
1824
	for jail_arch in ${_archs}; do
1825
		jail_name=$(poudriere_jail_name ${jail_arch})
1826

    
1827
		if [ "${jail_arch}" = "arm.armv7" ]; then
1828
			native_xtools="-x"
1829
		else
1830
			native_xtools=""
1831
		fi
1832

    
1833
		echo ">>> Creating jail ${jail_name}, it may take some time... " | tee -a ${LOGFILE}
1834
		if [ "${AWS}" = "1" ]; then
1835
			mkdir objs
1836
			echo ">>> Downloading prebuilt release objs from s3://pfsense-engineering-build-freebsd-obj-tarballs/${FLAVOR}/${FREEBSD_BRANCH}/ ..." | tee -a ${LOGFILE}
1837
			# Download prebuilt release tarballs from previous job
1838
			aws_exec s3 cp s3://pfsense-engineering-build-freebsd-obj-tarballs/${FLAVOR}/${FREEBSD_BRANCH}/LATEST-${jail_arch} objs --no-progress
1839
			SRC_COMMIT=`cat objs/LATEST-${jail_arch}`
1840
			aws_exec s3 cp s3://pfsense-engineering-build-freebsd-obj-tarballs/${FLAVOR}/${FREEBSD_BRANCH}/MANIFEST-${jail_arch}-${SRC_COMMIT} objs --no-progress
1841
			ln -s MANIFEST-${jail_arch}-${SRC_COMMIT} objs/MANIFEST
1842
			for i in base doc kernel src tests; do
1843
				if [ ! -f objs/${i}-${jail_arch}-${SRC_COMMIT}.txz ]; then
1844
					aws_exec s3 cp s3://pfsense-engineering-build-freebsd-obj-tarballs/${FLAVOR}/${FREEBSD_BRANCH}/${i}-${jail_arch}-${SRC_COMMIT}.txz objs --no-progress
1845
					ln -s ${i}-${jail_arch}-${SRC_COMMIT}.txz objs/${i}.txz
1846
				fi
1847
			done
1848

    
1849
			if ! script -aq ${LOGFILE} poudriere jail -c -j "${jail_name}" -v ${FREEBSD_BRANCH} \
1850
					-a ${jail_arch} -m url=file://${PWD}/objs >/dev/null 2>&1; then
1851
				echo "" | tee -a ${LOGFILE}
1852
				echo ">>> ERROR: Error creating jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
1853
				print_error_pfS
1854
			fi
1855

    
1856
			# Download a cached pkg repo from S3
1857
			OLDIFS=${IFS}
1858
			IFS=$'\n'
1859
			echo ">>> Downloading cached pkgs for ${jail_arch} from S3.." | tee -a ${LOGFILE}
1860
			if aws_exec s3 ls s3://pfsense-engineering-build-pkg/${FLAVOR}-${POUDRIERE_PORTS_GIT_BRANCH}-pkgs-${jail_arch}.tar >/dev/null 2>&1; then
1861
				aws_exec s3 cp s3://pfsense-engineering-build-pkg/${FLAVOR}-${POUDRIERE_PORTS_GIT_BRANCH}-pkgs-${jail_arch}.tar . --no-progress
1862
				[ ! -d /usr/local/poudriere/data/packages/${jail_name}-${POUDRIERE_PORTS_NAME} ] && mkdir -p /usr/local/poudriere/data/packages/${jail_name}-${POUDRIERE_PORTS_NAME}
1863
				echo "Extracting ${FLAVOR}-${POUDRIERE_PORTS_GIT_BRANCH}-pkgs-${jail_arch}.tar to /usr/local/poudriere/data/packages/${jail_name}-${POUDRIERE_PORTS_NAME}" | tee -a ${LOGFILE}
1864
				[ ! -d /usr/local/poudriere/data/packages/${jail_name}-${POUDRIERE_PORTS_NAME} ] && mkdir /usr/local/poudriere/data/packages/${jail_name}-${POUDRIERE_PORTS_NAME}
1865
				script -aq ${LOGFILE} tar -xf ${FLAVOR}-${POUDRIERE_PORTS_GIT_BRANCH}-pkgs-${jail_arch}.tar -C /usr/local/poudriere/data/packages/${jail_name}-${POUDRIERE_PORTS_NAME}
1866
				# Save a list of pkgs
1867
				cd /usr/local/poudriere/data/packages/${jail_name}-${POUDRIERE_PORTS_NAME}/.latest
1868
				find . > ${WORKSPACE}/pre-build-pkg-list-${jail_arch}
1869
				cd ${WORKSPACE}
1870
			else
1871
				touch pre-build-pkg-list-${jail_arch}
1872
			fi
1873
			IFS=${OLDIFS}
1874
		else
1875
			if ! script -aq ${LOGFILE} poudriere jail -c -j "${jail_name}" -v ${FREEBSD_BRANCH} \
1876
					-a ${jail_arch} -m git -U ${FREEBSD_REPO_BASE_POUDRIERE} ${native_xtools} >/dev/null 2>&1; then
1877
				echo "" | tee -a ${LOGFILE}
1878
				echo ">>> ERROR: Error creating jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
1879
				print_error_pfS
1880
			fi
1881
		fi
1882
		echo "Done!" | tee -a ${LOGFILE}
1883
	done
1884

    
1885
	poudriere_create_ports_tree
1886

    
1887
	echo ">>> Poudriere is now configured!" | tee -a ${LOGFILE}
1888
}
1889

    
1890
poudriere_update_jails() {
1891
	local _archs=$(poudriere_possible_archs)
1892

    
1893
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1894

    
1895
	local native_xtools=""
1896
	for jail_arch in ${_archs}; do
1897
		jail_name=$(poudriere_jail_name ${jail_arch})
1898

    
1899
		local _create_or_update="-u"
1900
		local _create_or_update_text="Updating"
1901
		if ! poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
1902
			echo ">>> Poudriere jail ${jail_name} not found, creating..." | tee -a ${LOGFILE}
1903
			_create_or_update="-c -v ${FREEBSD_BRANCH} -a ${jail_arch} -m git -U ${FREEBSD_REPO_BASE_POUDRIERE}"
1904
			_create_or_update_text="Creating"
1905
		fi
1906

    
1907
		if [ "${jail_arch}" = "arm.armv7" ]; then
1908
			native_xtools="-x"
1909
		else
1910
			native_xtools=""
1911
		fi
1912

    
1913
		echo -n ">>> ${_create_or_update_text} jail ${jail_name}, it may take some time... " | tee -a ${LOGFILE}
1914
		if ! script -aq ${LOGFILE} poudriere jail ${_create_or_update} -j "${jail_name}" ${native_xtools} >/dev/null 2>&1; then
1915
			echo "" | tee -a ${LOGFILE}
1916
			echo ">>> ERROR: Error ${_create_or_update_text} jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
1917
			print_error_pfS
1918
		fi
1919
		echo "Done!" | tee -a ${LOGFILE}
1920
	done
1921
}
1922

    
1923
poudriere_update_ports() {
1924
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1925

    
1926
	# Create ports tree if necessary
1927
	if ! poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
1928
		poudriere_create_ports_tree
1929
	else
1930
		echo -n ">>> Resetting local changes on ports tree ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
1931
		script -aq ${LOGFILE} git -C "/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}" reset --hard >/dev/null 2>&1
1932
		script -aq ${LOGFILE} git -C "/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}" clean -fd >/dev/null 2>&1
1933
		echo "Done!" | tee -a ${LOGFILE}
1934
		echo -n ">>> Updating ports tree ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
1935
		script -aq ${LOGFILE} poudriere ports -u -p "${POUDRIERE_PORTS_NAME}" >/dev/null 2>&1
1936
		echo "Done!" | tee -a ${LOGFILE}
1937
		poudriere_rename_ports
1938
	fi
1939
}
1940

    
1941
save_logs_to_s3() {
1942
	# Save a copy of the past few logs into S3
1943
	DATE=`date +%Y%m%d-%H%M%S`
1944
	script -aq ${LOGFILE} tar --zstd -cf pkg-logs-${jail_arch}-${DATE}.tar -C /usr/local/poudriere/data/logs/bulk/${jail_name}-${POUDRIERE_PORTS_NAME}/latest/ .
1945
	aws_exec s3 cp pkg-logs-${jail_arch}-${DATE}.tar s3://pfsense-engineering-build-pkg/logs/ --no-progress
1946
	echo ">>> Uploading pkg-logs-${jail_arch}-${DATE}.tar to s3" | tee -a ${LOGFILE}
1947
	OLDIFS=${IFS}
1948
	IFS=$'\n'
1949
	local _logtemp=$( mktemp /tmp/loglist.XXXXX )
1950
	for i in $(aws_exec s3 ls s3://pfsense-engineering-build-pkg/logs/); do
1951
		echo ${i} | awk '{print $4}' | grep pkg-logs-${jail_arch} | tr -d '\r' >> ${_logtemp}
1952
	done
1953
	# keep at least ~30 days of logs, plus some extra for one off runs
1954
	local _maxlogs=45
1955
	local _curlogs=0
1956
	_curlogs=$( wc -l ${_logtemp} | awk '{print $1}' )
1957
	if [ ${_curlogs} -gt ${_maxlogs} ]; then
1958
		local _extralogs=$(( ${_curlogs} - ${_maxlogs} ))
1959
		for _last in $( head -${_extralogs} ${_logtemp} ); do
1960
			aws_exec s3 rm s3://pfsense-engineering-build-pkg/logs/${_last}
1961
		done
1962
	fi
1963
	IFS=${OLDIFS}
1964
}
1965

    
1966
save_pkgs_to_s3() {
1967
	cd /usr/local/poudriere/data/packages/${jail_name}-${POUDRIERE_PORTS_NAME}/.latest
1968
	find . > ${WORKSPACE}/post-build-pkg-list-${jail_arch}
1969
	cd ${WORKSPACE}
1970
	diff pre-build-pkg-list-${jail_arch} post-build-pkg-list-${jail_arch} > /dev/null
1971
	if [ $? = 1 ]; then
1972
		echo ">>> Saving a copy of the package repo into S3..." | tee -a ${LOGFILE}
1973
		[ -f ${FLAVOR}-${POUDRIERE_PORTS_GIT_BRANCH}-pkgs-${jail_arch}.tar ] && rm ${FLAVOR}-${POUDRIERE_PORTS_GIT_BRANCH}-pkgs-${jail_arch}.tar
1974
		script -aq ${LOGFILE} tar -cf ${FLAVOR}-${POUDRIERE_PORTS_GIT_BRANCH}-pkgs-${jail_arch}.tar -C /usr/local/poudriere/data/packages/${jail_name}-${POUDRIERE_PORTS_NAME} .
1975
		aws_exec s3 cp ${FLAVOR}-${POUDRIERE_PORTS_GIT_BRANCH}-pkgs-${jail_arch}.tar s3://pfsense-engineering-build-pkg/ --no-progress
1976
	else
1977
		echo ">>> No pkgs different, not saving to S3..." | tee -a ${LOGFILE}
1978
	fi
1979
	save_logs_to_s3
1980
}
1981

    
1982
aws_exec() {
1983
	script -aq ${LOGFILE} \
1984
	    env AWS_ACCESS_KEY_ID=${AWS_ACCESS_KEY_ID} \
1985
	    AWS_SECRET_ACCESS_KEY=${AWS_SECRET_ACCESS_KEY} \
1986
	    AWS_DEFAULT_REGION=us-east-2 \
1987
	    AWS_DEFAULT_OUTPUT=text \
1988
	    aws $@
1989
	return $?
1990
}
1991

    
1992
poudriere_bulk() {
1993
	local _archs=$(poudriere_possible_archs)
1994
	local _makeconf
1995

    
1996
	# Create DISTFILES_CACHE if it doesn't exist
1997
	if [ ! -d /usr/ports/distfiles ]; then
1998
		mkdir -p /usr/ports/distfiles
1999
	fi
2000

    
2001
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2002

    
2003
	local _pkg_rsync_site_count=0
2004
	for _pkg_rsync_site in ${PKG_RSYNC_HOSTS}; do
2005
		eval _pkg_rsync_hostname=\$PKG_RSYNC_HOSTNAME_$_pkg_rsync_site
2006
		[ -n "${_pkg_rsync_hostname}" ] && \
2007
			_pkg_rsync_site_count=$(( ${_pkg_rsync_site_count} + 1 ))
2008
	done
2009

    
2010
	if [ -n "${UPLOAD}" ] && [ "${_pkg_rsync_site_count}" -eq 0 ]; then
2011
		echo ">>> ERROR: PKG_RSYNC_HOSTS is not set"
2012
		print_error_pfS
2013
	fi
2014

    
2015
	rm -f ${LOGFILE}
2016

    
2017
	poudriere_create_ports_tree
2018

    
2019
	[ -d /usr/local/etc/poudriere.d ] || \
2020
		mkdir -p /usr/local/etc/poudriere.d
2021

    
2022
	_makeconf=/usr/local/etc/poudriere.d/${POUDRIERE_PORTS_NAME}-make.conf
2023
	if [ -f "${BUILDER_TOOLS}/conf/pfPorts/make.conf" ]; then
2024
		sed -e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
2025
		    -e "s,%%PRODUCT_VERSION%%,${PRODUCT_VERSION},g" \
2026
		    "${BUILDER_TOOLS}/conf/pfPorts/make.conf" > ${_makeconf}
2027
	fi
2028

    
2029
	cat <<EOF >>/usr/local/etc/poudriere.d/${POUDRIERE_PORTS_NAME}-make.conf
2030

    
2031
PKG_REPO_BRANCH_DEVEL=${PKG_REPO_BRANCH_DEVEL}
2032
PKG_REPO_BRANCH_NEXT=${PKG_REPO_BRANCH_NEXT}
2033
PKG_REPO_BRANCH_RELEASE=${PKG_REPO_BRANCH_RELEASE}
2034
PKG_REPO_BRANCH_PREVIOUS=${PKG_REPO_BRANCH_PREVIOUS}
2035
PKG_REPO_SERVER_DEVEL=${PKG_REPO_SERVER_DEVEL}
2036
PKG_REPO_SERVER_RELEASE=${PKG_REPO_SERVER_RELEASE}
2037
POUDRIERE_PORTS_NAME=${POUDRIERE_PORTS_NAME}
2038
PFSENSE_DEFAULT_REPO=${PFSENSE_DEFAULT_REPO}
2039
PRODUCT_NAME=${PRODUCT_NAME}
2040
REPO_BRANCH_PREFIX=${REPO_PATH_PREFIX}
2041
EOF
2042

    
2043
	local _value=""
2044
	for jail_arch in ${_archs}; do
2045
		eval "_value=\${PKG_REPO_BRANCH_DEVEL_${jail_arch##*.}}"
2046
		if [ -n "${_value}" ]; then
2047
			echo "PKG_REPO_BRANCH_DEVEL_${jail_arch##*.}=${_value}" \
2048
				>> ${_makeconf}
2049
		fi
2050
		eval "_value=\${PKG_REPO_BRANCH_RELEASE_${jail_arch##*.}}"
2051
		if [ -n "${_value}" ]; then
2052
			echo "PKG_REPO_BRANCH_RELEASE_${jail_arch##*.}=${_value}" \
2053
				>> ${_makeconf}
2054
		fi
2055
		eval "_value=\${PKG_REPO_SERVER_DEVEL_${jail_arch##*.}}"
2056
		if [ -n "${_value}" ]; then
2057
			echo "PKG_REPO_SERVER_DEVEL_${jail_arch##*.}=${_value}" \
2058
				>> ${_makeconf}
2059
		fi
2060
		eval "_value=\${PKG_REPO_SERVER_RELEASE_${jail_arch##*.}}"
2061
		if [ -n "${_value}" ]; then
2062
			echo "PKG_REPO_SERVER_RELEASE_${jail_arch##*.}=${_value}" \
2063
				>> ${_makeconf}
2064
		fi
2065
	done
2066

    
2067
	# Change version of pfSense meta ports for snapshots
2068
	if [ -z "${_IS_RELEASE}" ]; then
2069
		local _meta_pkg_version="$(echo "${PRODUCT_VERSION}" | sed 's,DEVELOPMENT,ALPHA,')-${DATESTRING}"
2070
		sed -i '' \
2071
			-e "/^DISTVERSION/ s,^.*,DISTVERSION=	${_meta_pkg_version}," \
2072
			-e "/^PORTREVISION=/d" \
2073
			/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}/security/${PRODUCT_NAME}/Makefile \
2074
			/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}/sysutils/${PRODUCT_NAME}-repo/Makefile
2075
	fi
2076

    
2077
	# Copy over pkg repo templates to pfSense-repo
2078
	mkdir -p /usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}/sysutils/${PRODUCT_NAME}-repo/files
2079
	cp -f ${PKG_REPO_BASE}/* \
2080
		/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}/sysutils/${PRODUCT_NAME}-repo/files
2081

    
2082
	for jail_arch in ${_archs}; do
2083
		jail_name=$(poudriere_jail_name ${jail_arch})
2084

    
2085
		if ! poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
2086
			echo ">>> Poudriere jail ${jail_name} not found, skipping..." | tee -a ${LOGFILE}
2087
			continue
2088
		fi
2089

    
2090
		_ref_bulk=${SCRATCHDIR}/poudriere_bulk.${POUDRIERE_BRANCH}.ref.${jail_arch}
2091
		rm -rf ${_ref_bulk} ${_ref_bulk}.tmp
2092
		touch ${_ref_bulk}.tmp
2093
		if [ -f "${POUDRIERE_BULK}.${jail_arch#*.}" ]; then
2094
			cat "${POUDRIERE_BULK}.${jail_arch#*.}" >> ${_ref_bulk}.tmp
2095
		fi
2096
		if [ -f "${POUDRIERE_BULK}" ]; then
2097
			cat "${POUDRIERE_BULK}" >> ${_ref_bulk}.tmp
2098
		fi
2099
		cat ${_ref_bulk}.tmp | sort -u > ${_ref_bulk}
2100

    
2101
		_bulk=${SCRATCHDIR}/poudriere_bulk.${POUDRIERE_BRANCH}.${jail_arch}
2102
		sed -e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" ${_ref_bulk} > ${_bulk}
2103

    
2104
		local _exclude_bulk="${POUDRIERE_BULK}.exclude.${jail_arch}"
2105
		if [ -f "${_exclude_bulk}" ]; then
2106
			mv ${_bulk} ${_bulk}.tmp
2107
			sed -e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" ${_exclude_bulk} > ${_bulk}.exclude
2108
			cat ${_bulk}.tmp ${_bulk}.exclude | sort | uniq -u > ${_bulk}
2109
			rm -f ${_bulk}.tmp ${_bulk}.exclude
2110
		fi
2111

    
2112
		echo ">>> Poudriere bulk started at `date "+%Y/%m/%d %H:%M:%S"` for ${jail_arch}"
2113
		if ! poudriere bulk -f ${_bulk} -j ${jail_name} -p ${POUDRIERE_PORTS_NAME}; then
2114
			echo ">>> ERROR: Something went wrong..."
2115
			if [ "${AWS}" = 1 ]; then
2116
				save_pkgs_to_s3
2117
			fi
2118
			print_error_pfS
2119
		fi
2120
		echo ">>> Poudriere bulk complated at `date "+%Y/%m/%d %H:%M:%S"` for ${jail_arch}"
2121

    
2122
		echo ">>> Cleaning up old packages from repo..."
2123
		if ! poudriere pkgclean -f ${_bulk} -j ${jail_name} -p ${POUDRIERE_PORTS_NAME} -y; then
2124
			echo ">>> ERROR: Something went wrong..."
2125
			print_error_pfS
2126
		fi
2127

    
2128
		if [ "${AWS}" = 1 ]; then
2129
			save_pkgs_to_s3
2130
		fi
2131

    
2132
		pkg_repo_rsync "/usr/local/poudriere/data/packages/${jail_name}-${POUDRIERE_PORTS_NAME}"
2133
	done
2134

    
2135
	if [ "${AWS}" = 1 ]; then
2136
		echo ">>> Run poudriere distclean to prune old distfiles..." | tee -a ${LOGFILE}
2137
		if ! poudriere distclean -f ${_bulk} -p ${POUDRIERE_PORTS_NAME} -n; then
2138
			echo ">>> ERROR: Something went wrong..."
2139
			print_error_pfS
2140
		fi
2141
		echo ">>> Save a copy of the distfiles into S3..." | tee -a ${LOGFILE}
2142
		# Save a copy of the distfiles from S3
2143
		find /usr/ports/distfiles > post-build-distfile-list
2144
		diff pre-build-distfile-list post-build-distfile-list > /dev/null
2145
		if [ $? -eq 1 ]; then
2146
			rm -f ${FLAVOR}-${POUDRIERE_PORTS_GIT_BRANCH}-distfiles.tar
2147
			script -aq ${LOGFILE} tar -cf ${FLAVOR}-${POUDRIERE_PORTS_GIT_BRANCH}-distfiles.tar -C /usr/ports/distfiles .
2148
			aws_exec s3 cp ${FLAVOR}-${POUDRIERE_PORTS_GIT_BRANCH}-distfiles.tar s3://pfsense-engineering-build-pkg/ --no-progress
2149
		fi
2150
	fi
2151
}
2152

    
2153
# This routine is called to write out to stdout
2154
# a string. The string is appended to $SNAPSHOTSLOGFILE
2155
snapshots_update_status() {
2156
	if [ -z "$1" ]; then
2157
		return
2158
	fi
2159
	if [ -z "${SNAPSHOTS}" -a -z "${POUDRIERE_SNAPSHOTS}" ]; then
2160
		return
2161
	fi
2162
	echo "$*"
2163
	echo "`date` -|- $*" >> $SNAPSHOTSLOGFILE
2164
}
2165

    
2166
create_sha256() {
2167
	local _file="${1}"
2168

    
2169
	if [ ! -f "${_file}" ]; then
2170
		return 1
2171
	fi
2172

    
2173
	( \
2174
		cd $(dirname ${_file}) && \
2175
		sha256 $(basename ${_file}) > $(basename ${_file}).sha256 \
2176
	)
2177
}
2178

    
2179
snapshots_create_latest_symlink() {
2180
	local _image="${1}"
2181

    
2182
	if [ -z "${_image}" ]; then
2183
		return
2184
	fi
2185

    
2186
	if [ -z "${TIMESTAMP_SUFFIX}" ]; then
2187
		return
2188
	fi
2189

    
2190
	if [ ! -f "${_image}" ]; then
2191
		return
2192
	fi
2193

    
2194
	local _symlink=$(echo ${_image} | sed "s,${TIMESTAMP_SUFFIX},-latest,")
2195
	ln -sf $(basename ${_image}) ${_symlink}
2196
	ln -sf $(basename ${_image}).sha256 ${_symlink}.sha256
2197
}
2198

    
2199
snapshots_create_sha256() {
2200
	local _img=""
2201

    
2202
	for _img in ${ISOPATH} ${MEMSTICKPATH} ${MEMSTICKSERIALPATH} ${MEMSTICKADIPATH} ${OVAPATH} ${VARIANTIMAGES}; do
2203
		if [ -f "${_img}.gz" ]; then
2204
			_img="${_img}.gz"
2205
		fi
2206
		if [ ! -f "${_img}" ]; then
2207
			continue
2208
		fi
2209
		create_sha256 ${_img}
2210
		snapshots_create_latest_symlink ${_img}
2211
	done
2212
}
2213

    
2214
snapshots_scp_files() {
2215
	if [ -z "${RSYNC_COPY_ARGUMENTS}" ]; then
2216
		RSYNC_COPY_ARGUMENTS="-Have \"ssh -o StrictHostKeyChecking=no\" --timeout=60"
2217
	fi
2218

    
2219
	snapshots_update_status ">>> Copying core pkg repo to ${PKG_RSYNC_HOSTNAME}"
2220
	pkg_repo_rsync "${CORE_PKG_PATH}"
2221
	snapshots_update_status ">>> Finished copying core pkg repo"
2222

    
2223
	for _rsyncip in ${RSYNCIP}; do
2224
		snapshots_update_status ">>> Copying files to ${_rsyncip}"
2225

    
2226
		# Ensure directory(s) are available
2227
		ssh -o StrictHostKeyChecking=no ${RSYNCUSER}@${_rsyncip} "mkdir -p ${RSYNCPATH}/installer"
2228
		if [ -d $IMAGES_FINAL_DIR/virtualization ]; then
2229
			ssh -o StrictHostKeyChecking=no ${RSYNCUSER}@${_rsyncip} "mkdir -p ${RSYNCPATH}/virtualization"
2230
		fi
2231
		# ensure permissions are correct for r+w
2232
		ssh -o StrictHostKeyChecking=no ${RSYNCUSER}@${_rsyncip} "chmod -R ug+rw ${RSYNCPATH}/."
2233
		rsync $RSYNC_COPY_ARGUMENTS $IMAGES_FINAL_DIR/installer/* \
2234
			${RSYNCUSER}@${_rsyncip}:${RSYNCPATH}/installer/
2235
		if [ -d $IMAGES_FINAL_DIR/virtualization ]; then
2236
			rsync $RSYNC_COPY_ARGUMENTS $IMAGES_FINAL_DIR/virtualization/* \
2237
				${RSYNCUSER}@${_rsyncip}:${RSYNCPATH}/virtualization/
2238
		fi
2239

    
2240
		snapshots_update_status ">>> Finished copying files."
2241
	done
2242
}
(2-2/3)