Project

General

Profile

Download (61.9 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-2020 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.txz ${CORE_PKG_PATH}/meta.txz
73
	ln -sf .latest/packagesite.txz ${CORE_PKG_PATH}/packagesite.txz
74
}
75

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

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

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

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

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

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

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

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

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

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

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

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

    
169
		buildkernel
170

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

    
174
		ensure_kernel_exists $KERNEL_DESTDIR
175

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

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

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

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

    
194
	export KERNEL_NAME="${1}"
195

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

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

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

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

    
229
	unset KERNEL_NAME
230
}
231

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

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

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

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

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

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

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

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

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

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

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

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

    
305
	unset makeargs
306
}
307

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

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

    
326
	local _mntdir=${OVA_TMP}/mnt
327

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

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

    
346
	mkdir -p ${_mntdir}
347

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

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

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

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

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

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

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

    
396
	clone_directory_contents ${FINAL_CHROOT_DIR} ${_mntdir}
397

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

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

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

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

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

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

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

    
450
	ova_setup_ovf_template
451

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

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

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

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

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

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

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

    
507
	staginareas_clean_each_run
508

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

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

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

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

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

    
551
	echo "Done!"
552

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

    
652
	core_pkg_create default-config-serial "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
653

    
654
	rm -f ${STAGE_CHROOT_DIR}/cf/conf/enableserial_force
655
	rm -f ${STAGE_CHROOT_DIR}/cf/conf/config.xml
656

    
657
	# Make sure pkg is present
658
	pkg_bootstrap ${STAGE_CHROOT_DIR}
659

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

    
670
	echo "Done!"
671
}
672

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

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

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

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

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

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

    
712
	# Prepare final stage area
713
	create_final_staging_area
714

    
715
	pkg_chroot_add ${FINAL_CHROOT_DIR} rc
716
	pkg_chroot_add ${FINAL_CHROOT_DIR} base
717

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

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

    
730
	pkg_chroot_add ${FINAL_CHROOT_DIR} ${_default_config}
731

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

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

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

    
764
create_distribution_tarball() {
765
	mkdir -p ${INSTALLER_CHROOT_DIR}/usr/freebsd-dist
766

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

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

    
779
create_iso_image() {
780
	local _variant="$1"
781

    
782
	LOGFILE=${BUILDER_LOGS}/isoimage.${TARGET}
783

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

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

    
791
	mkdir -p $(dirname ${ISOPATH})
792

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

    
800
	customize_stagearea_for_image "iso" "" $_variant
801
	install_default_kernel ${DEFAULT_KERNEL}
802

    
803
	BOOTCONF=${INSTALLER_CHROOT_DIR}/boot.config
804
	LOADERCONF=${INSTALLER_CHROOT_DIR}/boot/loader.conf
805

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

    
811
	create_distribution_tarball
812

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

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

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

    
825
	gzip -qf $_image_path &
826
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
827

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

    
831
create_memstick_image() {
832
	local _variant="$1"
833

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

    
840
	mkdir -p $(dirname ${MEMSTICKPATH})
841

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

    
849
	customize_stagearea_for_image "memstick" "" $_variant
850
	install_default_kernel ${DEFAULT_KERNEL}
851

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

    
854
	BOOTCONF=${INSTALLER_CHROOT_DIR}/boot.config
855
	LOADERCONF=${INSTALLER_CHROOT_DIR}/boot/loader.conf
856

    
857
	rm -f ${LOADERCONF} ${BOOTCONF} >/dev/null 2>&1
858

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

    
864
	create_distribution_tarball
865

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

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

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

    
878
	gzip -qf $_image_path &
879
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
880

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

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

    
891
	mkdir -p $(dirname ${MEMSTICKSERIALPATH})
892

    
893
	customize_stagearea_for_image "memstickserial"
894
	install_default_kernel ${DEFAULT_KERNEL}
895

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

    
898
	BOOTCONF=${INSTALLER_CHROOT_DIR}/boot.config
899
	LOADERCONF=${INSTALLER_CHROOT_DIR}/boot/loader.conf
900

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

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

    
912
	cat ${BOOTCONF} >> ${FINAL_CHROOT_DIR}/boot.config
913
	cat ${LOADERCONF} >> ${FINAL_CHROOT_DIR}/boot/loader.conf
914

    
915
	create_distribution_tarball
916

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

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

    
926
	gzip -qf $MEMSTICKSERIALPATH &
927
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
928

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

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

    
939
	mkdir -p $(dirname ${MEMSTICKADIPATH})
940

    
941
	customize_stagearea_for_image "memstickadi"
942
	install_default_kernel ${DEFAULT_KERNEL}
943

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

    
946
	BOOTCONF=${INSTALLER_CHROOT_DIR}/boot.config
947
	LOADERCONF=${INSTALLER_CHROOT_DIR}/boot/loader.conf
948

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

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

    
962
	cat ${BOOTCONF} >> ${FINAL_CHROOT_DIR}/boot.config
963
	cat ${LOADERCONF} >> ${FINAL_CHROOT_DIR}/boot/loader.conf
964

    
965
	create_distribution_tarball
966

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

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

    
976
	gzip -qf $MEMSTICKADIPATH &
977
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
978

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

    
982
get_altabi_arch() {
983
	local _target_arch="$1"
984

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

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

    
1003
	local _template="${1}"
1004
	local _target="${2}"
1005
	local _arch="${3}"
1006
	local _target_arch="${4}"
1007
	local _staging="${5}"
1008
	local _pkg_conf="${6}"
1009

    
1010
	if [ -z "${_template}" -o ! -f "${_template}" ]; then
1011
		echo ">>> ERROR: It was not possible to find pkg conf template ${_template}"
1012
		print_error_pfS
1013
	fi
1014

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

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

    
1029
	sed \
1030
		-e "s/%%ARCH%%/${_target_arch}/" \
1031
		-e "s/%%PKG_REPO_BRANCH_DEVEL%%/${_pkg_repo_branch_devel}/g" \
1032
		-e "s/%%PKG_REPO_BRANCH_RELEASE%%/${_pkg_repo_branch_release}/g" \
1033
		-e "s,%%PKG_REPO_SERVER_DEVEL%%,${_pkg_repo_server_devel},g" \
1034
		-e "s,%%PKG_REPO_SERVER_RELEASE%%,${_pkg_repo_server_release},g" \
1035
		-e "s,%%POUDRIERE_PORTS_NAME%%,${POUDRIERE_PORTS_NAME},g" \
1036
		-e "s/%%PRODUCT_NAME%%/${PRODUCT_NAME}/g" \
1037
		-e "s/%%REPO_BRANCH_PREFIX%%/${REPO_BRANCH_PREFIX}/g" \
1038
		${_template} \
1039
		> ${_target}
1040

    
1041
	local ALTABI_ARCH=$(get_altabi_arch ${_target_arch})
1042

    
1043
	ABI=$(cat ${_template%%.conf}.abi 2>/dev/null \
1044
	    | sed -e "s/%%ARCH%%/${_target_arch}/g")
1045
	ALTABI=$(cat ${_template%%.conf}.altabi 2>/dev/null \
1046
	    | sed -e "s/%%ARCH%%/${ALTABI_ARCH}/g")
1047

    
1048
	if [ -n "${_pkg_conf}" -a -n "${ABI}" -a -n "${ALTABI}" ]; then
1049
		mkdir -p $(dirname ${_pkg_conf})
1050
		echo "ABI=${ABI}" > ${_pkg_conf}
1051
		echo "ALTABI=${ALTABI}" >> ${_pkg_conf}
1052
	fi
1053
}
1054

    
1055
depend_check() {
1056
	for _pkg in ${BUILDER_PKG_DEPENDENCIES}; do
1057
		if ! pkg info -e ${_pkg}; then
1058
			echo "Missing dependency (${_pkg})."
1059
			print_error_pfS
1060
		fi
1061
	done
1062
}
1063

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

    
1072
	if [ ! -f ${PKG_REPO_PATH} ]; then
1073
		[ -d $(dirname ${PKG_REPO_PATH}) ] \
1074
			|| mkdir -p $(dirname ${PKG_REPO_PATH})
1075

    
1076
		update_freebsd_sources
1077

    
1078
		local _arch=$(uname -m)
1079
		setup_pkg_repo \
1080
			${PKG_REPO_BUILD} \
1081
			${PKG_REPO_PATH} \
1082
			${_arch} \
1083
			${_arch} \
1084
			"staging"
1085

    
1086
		# Use fingerprint keys from repo
1087
		sed -i '' -e "/fingerprints:/ s,\"/,\"${BUILDER_ROOT}/src/," \
1088
			${PKG_REPO_PATH}
1089
	fi
1090

    
1091
	pkg install ${PRODUCT_NAME}-builder
1092
}
1093

    
1094
# Updates FreeBSD sources
1095
update_freebsd_sources() {
1096
	if [ "${1}" = "full" ]; then
1097
		local _full=1
1098
		local _clone_params=""
1099
	else
1100
		local _full=0
1101
		local _clone_params="--depth 1 --single-branch"
1102
	fi
1103

    
1104
	if [ -n "${NO_BUILDWORLD}" -a -n "${NO_BUILDKERNEL}" ]; then
1105
		echo ">>> NO_BUILDWORLD and NO_BUILDKERNEL set, skipping update of freebsd sources" | tee -a ${LOGFILE}
1106
		return
1107
	fi
1108

    
1109
	echo ">>> Obtaining FreeBSD sources (${FREEBSD_BRANCH})..."
1110
	${BUILDER_SCRIPTS}/git_checkout.sh \
1111
		-r ${FREEBSD_REPO_BASE} \
1112
		-d ${FREEBSD_SRC_DIR} \
1113
		-b ${FREEBSD_BRANCH}
1114

    
1115
	if [ $? -ne 0 -o ! -d "${FREEBSD_SRC_DIR}/.git" ]; then
1116
		echo ">>> ERROR: It was not possible to clone FreeBSD src repo"
1117
		print_error_pfS
1118
	fi
1119

    
1120
	if [ -n "${GIT_FREEBSD_COSHA1}" ]; then
1121
		echo -n ">>> Checking out desired commit (${GIT_FREEBSD_COSHA1})... "
1122
		( git -C  ${FREEBSD_SRC_DIR} checkout ${GIT_FREEBSD_COSHA1} ) 2>&1 | \
1123
			grep -C3 -i -E 'error|fatal'
1124
		echo "Done!"
1125
	fi
1126

    
1127
	if [ "${PRODUCT_NAME}" = "pfSense" -a -n "${GNID_REPO_BASE}" ]; then
1128
		echo ">>> Obtaining gnid sources..."
1129
		${BUILDER_SCRIPTS}/git_checkout.sh \
1130
			-r ${GNID_REPO_BASE} \
1131
			-d ${GNID_SRC_DIR} \
1132
			-b ${GNID_BRANCH}
1133
	fi
1134
}
1135

    
1136
pkg_chroot() {
1137
	local _root="${1}"
1138
	shift
1139

    
1140
	if [ $# -eq 0 ]; then
1141
		return -1
1142
	fi
1143

    
1144
	if [ -z "${_root}" -o "${_root}" = "/" -o ! -d "${_root}" ]; then
1145
		return -1
1146
	fi
1147

    
1148
	mkdir -p \
1149
		${SCRATCHDIR}/pkg_cache \
1150
		${_root}/var/cache/pkg \
1151
		${_root}/dev
1152

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

    
1171
	return $result
1172
}
1173

    
1174

    
1175
pkg_chroot_add() {
1176
	if [ -z "${1}" -o -z "${2}" ]; then
1177
		return 1
1178
	fi
1179

    
1180
	local _target="${1}"
1181
	local _pkg="$(get_pkg_name ${2}).txz"
1182

    
1183
	if [ ! -d "${_target}" ]; then
1184
		echo ">>> ERROR: Target dir ${_target} not found"
1185
		print_error_pfS
1186
	fi
1187

    
1188
	if [ ! -f ${CORE_PKG_ALL_PATH}/${_pkg} ]; then
1189
		echo ">>> ERROR: Package ${_pkg} not found"
1190
		print_error_pfS
1191
	fi
1192

    
1193
	cp ${CORE_PKG_ALL_PATH}/${_pkg} ${_target}
1194
	pkg_chroot ${_target} add /${_pkg}
1195
	rm -f ${_target}/${_pkg}
1196
}
1197

    
1198
pkg_bootstrap() {
1199
	local _root=${1:-"${STAGE_CHROOT_DIR}"}
1200

    
1201
	setup_pkg_repo \
1202
		${PKG_REPO_BUILD} \
1203
		${_root}${PKG_REPO_PATH} \
1204
		${TARGET} \
1205
		${TARGET_ARCH} \
1206
		"staging"
1207

    
1208
	pkg_chroot ${_root} bootstrap -f
1209
}
1210

    
1211
# This routine assists with installing various
1212
# freebsd ports files into the pfsense-fs staging
1213
# area.
1214
install_pkg_install_ports() {
1215
	local MAIN_PKG="${1}"
1216

    
1217
	if [ -z "${MAIN_PKG}" ]; then
1218
		MAIN_PKG=${PRODUCT_NAME}
1219
	fi
1220

    
1221
	echo ">>> Installing pkg repository in chroot (${STAGE_CHROOT_DIR})..."
1222

    
1223
	[ -d ${STAGE_CHROOT_DIR}/var/cache/pkg ] || \
1224
		mkdir -p ${STAGE_CHROOT_DIR}/var/cache/pkg
1225

    
1226
	[ -d ${SCRATCHDIR}/pkg_cache ] || \
1227
		mkdir -p ${SCRATCHDIR}/pkg_cache
1228

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

    
1246
staginareas_clean_each_run() {
1247
	echo -n ">>> Cleaning build directories: "
1248
	if [ -d "${FINAL_CHROOT_DIR}" ]; then
1249
		BASENAME=$(basename ${FINAL_CHROOT_DIR})
1250
		echo -n "$BASENAME "
1251
		chflags -R noschg ${FINAL_CHROOT_DIR} 2>&1 >/dev/null
1252
		rm -rf ${FINAL_CHROOT_DIR}/* 2>/dev/null
1253
	fi
1254
	echo "Done!"
1255
}
1256

    
1257
# Imported from FreeSBIE
1258
buildkernel() {
1259
	local _kernconf=${1:-${KERNCONF}}
1260

    
1261
	if [ -n "${NO_BUILDKERNEL}" ]; then
1262
		echo ">>> NO_BUILDKERNEL set, skipping build" | tee -a ${LOGFILE}
1263
		return
1264
	fi
1265

    
1266
	if [ -z "${_kernconf}" ]; then
1267
		echo ">>> ERROR: No kernel configuration defined probably this is not what you want! STOPPING!" | tee -a ${LOGFILE}
1268
		print_error_pfS
1269
	fi
1270

    
1271
	local _old_kernconf=${KERNCONF}
1272
	export KERNCONF=${_kernconf}
1273

    
1274
	echo ">>> $(LC_ALL=C date) - Starting build kernel for ${TARGET} architecture..." | tee -a ${LOGFILE}
1275
	script -aq $LOGFILE ${BUILDER_SCRIPTS}/build_freebsd.sh -W -s ${FREEBSD_SRC_DIR} \
1276
		|| print_error_pfS
1277
	echo ">>> $(LC_ALL=C date) - Finished build kernel for ${TARGET} architecture..." | tee -a ${LOGFILE}
1278

    
1279
	export KERNCONF=${_old_kernconf}
1280
}
1281

    
1282
# Imported from FreeSBIE
1283
installkernel() {
1284
	local _destdir=${1:-${KERNEL_DESTDIR}}
1285
	local _kernconf=${2:-${KERNCONF}}
1286

    
1287
	if [ -z "${_kernconf}" ]; then
1288
		echo ">>> ERROR: No kernel configuration defined probably this is not what you want! STOPPING!" | tee -a ${LOGFILE}
1289
		print_error_pfS
1290
	fi
1291

    
1292
	local _old_kernconf=${KERNCONF}
1293
	export KERNCONF=${_kernconf}
1294

    
1295
	mkdir -p ${STAGE_CHROOT_DIR}/boot
1296
	echo ">>> Installing kernel (${_kernconf}) for ${TARGET} architecture..." | tee -a ${LOGFILE}
1297
	script -aq $LOGFILE ${BUILDER_SCRIPTS}/install_freebsd.sh -W -D -z \
1298
		-s ${FREEBSD_SRC_DIR} \
1299
		-d ${_destdir} \
1300
		|| print_error_pfS
1301

    
1302
	export KERNCONF=${_old_kernconf}
1303
}
1304

    
1305
# Launch is ran first to setup a few variables that we need
1306
# Imported from FreeSBIE
1307
launch() {
1308
	if [ "$(id -u)" != "0" ]; then
1309
		echo "Sorry, this must be done as root."
1310
	fi
1311

    
1312
	echo ">>> Operation $0 has started at $(date)"
1313
}
1314

    
1315
finish() {
1316
	echo ">>> Operation $0 has ended at $(date)"
1317
}
1318

    
1319
pkg_repo_rsync() {
1320
	local _repo_path_param="${1}"
1321
	local _ignore_final_rsync="${2}"
1322
	local _aws_sync_cmd="aws s3 sync --quiet --exclude '.real*/*' --exclude '.latest/*'"
1323

    
1324
	if [ -z "${_repo_path_param}" -o ! -d "${_repo_path_param}" ]; then
1325
		return
1326
	fi
1327

    
1328
	if [ -n "${SKIP_FINAL_RSYNC}" ]; then
1329
		_ignore_final_rsync="1"
1330
	fi
1331

    
1332
	# Sanitize path
1333
	_repo_path=$(realpath ${_repo_path_param})
1334

    
1335
	local _repo_dir=$(dirname ${_repo_path})
1336
	local _repo_base=$(basename ${_repo_path})
1337

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

    
1341
	if [ -z "${LOGFILE}" ]; then
1342
		local _logfile="/dev/null"
1343
	else
1344
		local _logfile="${LOGFILE}"
1345
	fi
1346

    
1347
	if [ -n "${PKG_REPO_SIGNING_COMMAND}" -a -z "${DO_NOT_SIGN_PKG_REPO}" ]; then
1348
		# Detect poudriere directory structure
1349
		if [ -L "${_repo_path}/.latest" ]; then
1350
			local _real_repo_path=$(readlink -f ${_repo_path}/.latest)
1351
		else
1352
			local _real_repo_path=${_repo_path}
1353
		fi
1354

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

    
1372
		local _pkgfile="${_repo_path}/Latest/pkg.txz"
1373
		if [ -e ${_pkgfile} ]; then
1374
			echo -n ">>> Signing Latest/pkg.txz for bootstraping... " | tee -a ${_logfile}
1375

    
1376
			if sha256 -q ${_pkgfile} | ${PKG_REPO_SIGNING_COMMAND} \
1377
			    > ${_pkgfile}.sig 2>/dev/null; then
1378
				echo "Done!" | tee -a ${_logfile}
1379
			else
1380
				echo "Failed!" | tee -a ${_logfile}
1381
				echo ">>> ERROR: An error occurred trying to sign Latest/pkg.txz"
1382
				print_error_pfS
1383
			fi
1384
		fi
1385
	fi
1386

    
1387
	if [ -z "${UPLOAD}" ]; then
1388
		return
1389
	fi
1390

    
1391
	for _pkg_rsync_hostname in ${PKG_RSYNC_HOSTNAME}; do
1392
		# Make sure destination directory exist
1393
		ssh -p ${PKG_RSYNC_SSH_PORT} \
1394
			${PKG_RSYNC_USERNAME}@${_pkg_rsync_hostname} \
1395
			"mkdir -p ${PKG_RSYNC_DESTDIR}"
1396

    
1397
		echo -n ">>> Sending updated repository to ${_pkg_rsync_hostname}... " | tee -a ${_logfile}
1398
		if script -aq ${_logfile} rsync -Have "ssh -p ${PKG_RSYNC_SSH_PORT}" \
1399
			--timeout=60 --delete-delay ${_repo_path} \
1400
			${PKG_RSYNC_USERNAME}@${_pkg_rsync_hostname}:${PKG_RSYNC_DESTDIR} >/dev/null 2>&1
1401
		then
1402
			echo "Done!" | tee -a ${_logfile}
1403
		else
1404
			echo "Failed!" | tee -a ${_logfile}
1405
			echo ">>> ERROR: An error occurred sending repo to remote hostname"
1406
			print_error_pfS
1407
		fi
1408

    
1409
		if [ -z "${USE_PKG_REPO_STAGING}" -o -n "${_ignore_final_rsync}" ]; then
1410
			return
1411
		fi
1412

    
1413
		if [ -n "${_IS_RELEASE}" -o "${_repo_path_param}" = "${CORE_PKG_PATH}" ]; then
1414
			for _pkg_final_rsync_hostname in ${PKG_FINAL_RSYNC_HOSTNAME}; do
1415
				# Send .real* directories first to prevent having a broken repo while transfer happens
1416
				local _cmd="rsync -Have \"ssh -p ${PKG_FINAL_RSYNC_SSH_PORT}\" \
1417
					--timeout=60 ${PKG_RSYNC_DESTDIR}/./${_repo_base%%-core}* \
1418
					--include=\"/*\" --include=\"*/.real*\" --include=\"*/.real*/***\" \
1419
					--exclude=\"*\" \
1420
					${PKG_FINAL_RSYNC_USERNAME}@${_pkg_final_rsync_hostname}:${PKG_FINAL_RSYNC_DESTDIR}"
1421

    
1422
				echo -n ">>> Sending updated packages to ${_pkg_final_rsync_hostname}... " | tee -a ${_logfile}
1423
				if script -aq ${_logfile} ssh -p ${PKG_RSYNC_SSH_PORT} \
1424
					${PKG_RSYNC_USERNAME}@${_pkg_rsync_hostname} ${_cmd} >/dev/null 2>&1; then
1425
					echo "Done!" | tee -a ${_logfile}
1426
				else
1427
					echo "Failed!" | tee -a ${_logfile}
1428
					echo ">>> ERROR: An error occurred sending repo to final hostname"
1429
					print_error_pfS
1430
				fi
1431

    
1432
				_cmd="rsync -Have \"ssh -p ${PKG_FINAL_RSYNC_SSH_PORT}\" \
1433
					--timeout=60 --delete-delay ${PKG_RSYNC_DESTDIR}/./${_repo_base%%-core}* \
1434
					${PKG_FINAL_RSYNC_USERNAME}@${_pkg_final_rsync_hostname}:${PKG_FINAL_RSYNC_DESTDIR}"
1435

    
1436
				echo -n ">>> Sending updated repositories metadata to ${_pkg_final_rsync_hostname}... " | tee -a ${_logfile}
1437
				if script -aq ${_logfile} ssh -p ${PKG_RSYNC_SSH_PORT} \
1438
					${PKG_RSYNC_USERNAME}@${_pkg_rsync_hostname} ${_cmd} >/dev/null 2>&1; then
1439
					echo "Done!" | tee -a ${_logfile}
1440
				else
1441
					echo "Failed!" | tee -a ${_logfile}
1442
					echo ">>> ERROR: An error occurred sending repo to final hostname"
1443
					print_error_pfS
1444
				fi
1445

    
1446
				if [ -z "${PKG_FINAL_S3_PATH}" ]; then
1447
					continue
1448
				fi
1449

    
1450
				local _repos=$(ssh -p ${PKG_FINAL_RSYNC_SSH_PORT} \
1451
				    ${PKG_FINAL_RSYNC_USERNAME}@${_pkg_final_rsync_hostname} \
1452
				    "ls -1d ${PKG_FINAL_RSYNC_DESTDIR}/${_repo_base%%-core}*")
1453
				for _repo in ${_repos}; do
1454
					echo -n ">>> Sending updated packages to AWS ${PKG_FINAL_S3_PATH}... " | tee -a ${_logfile}
1455
					if script -aq ${_logfile} ssh -p ${PKG_FINAL_RSYNC_SSH_PORT} \
1456
					    ${PKG_FINAL_RSYNC_USERNAME}@${_pkg_final_rsync_hostname} \
1457
					    "${_aws_sync_cmd} ${_repo} ${PKG_FINAL_S3_PATH}/$(basename ${_repo})"; then
1458
						echo "Done!" | tee -a ${_logfile}
1459
					else
1460
						echo "Failed!" | tee -a ${_logfile}
1461
						echo ">>> ERROR: An error occurred sending files to AWS S3"
1462
						print_error_pfS
1463
					fi
1464
					echo -n ">>> Cleaning up packages at AWS ${PKG_FINAL_S3_PATH}... " | tee -a ${_logfile}
1465
					if script -aq ${_logfile} ssh -p ${PKG_FINAL_RSYNC_SSH_PORT} \
1466
					    ${PKG_FINAL_RSYNC_USERNAME}@${_pkg_final_rsync_hostname} \
1467
					    "${_aws_sync_cmd} --delete ${_repo} ${PKG_FINAL_S3_PATH}/$(basename ${_repo})"; then
1468
						echo "Done!" | tee -a ${_logfile}
1469
					else
1470
						echo "Failed!" | tee -a ${_logfile}
1471
						echo ">>> ERROR: An error occurred sending files to AWS S3"
1472
						print_error_pfS
1473
					fi
1474
				done
1475
			done
1476
		fi
1477
	done
1478
}
1479

    
1480
poudriere_possible_archs() {
1481
	local _arch=$(uname -m)
1482
	local _archs=""
1483

    
1484
	# If host is amd64, we'll create both repos, and if possible armv7
1485
	if [ "${_arch}" = "amd64" ]; then
1486
		_archs="amd64.amd64"
1487

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

    
1492
			if binmiscctl lookup armv7 >/dev/null 2>&1; then
1493
				_archs="${_archs} arm.armv7"
1494
			fi
1495
		fi
1496
	fi
1497

    
1498
	if [ -n "${ARCH_LIST}" ]; then
1499
		local _found=0
1500
		for _desired_arch in ${ARCH_LIST}; do
1501
			_found=0
1502
			for _possible_arch in ${_archs}; do
1503
				if [ "${_desired_arch}" = "${_possible_arch}" ]; then
1504
					_found=1
1505
					break
1506
				fi
1507
			done
1508
			if [ ${_found} -eq 0 ]; then
1509
				echo ">>> ERROR: Impossible to build for arch: ${_desired_arch}"
1510
				print_error_pfS
1511
			fi
1512
		done
1513
		_archs="${ARCH_LIST}"
1514
	fi
1515

    
1516
	echo ${_archs}
1517
}
1518

    
1519
poudriere_jail_name() {
1520
	local _jail_arch="${1}"
1521

    
1522
	if [ -z "${_jail_arch}" ]; then
1523
		return 1
1524
	fi
1525

    
1526
	# Remove arch
1527
	echo "${PRODUCT_NAME}_${POUDRIERE_BRANCH}_${_jail_arch##*.}"
1528
}
1529

    
1530
poudriere_rename_ports() {
1531
	if [ "${PRODUCT_NAME}" = "pfSense" ]; then
1532
		return;
1533
	fi
1534

    
1535
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1536

    
1537
	local _ports_dir="/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}"
1538

    
1539
	echo -n ">>> Renaming product ports on ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
1540
	for d in $(find ${_ports_dir} -depth 2 -type d -name '*pfSense*'); do
1541
		local _pdir=$(dirname ${d})
1542
		local _pname=$(echo $(basename ${d}) | sed "s,pfSense,${PRODUCT_NAME},")
1543
		local _plist=""
1544
		local _pdescr=""
1545

    
1546
		if [ -e ${_pdir}/${_pname} ]; then
1547
			rm -rf ${_pdir}/${_pname}
1548
		fi
1549

    
1550
		cp -r ${d} ${_pdir}/${_pname}
1551

    
1552
		if [ -f ${_pdir}/${_pname}/pkg-plist ]; then
1553
			_plist=${_pdir}/${_pname}/pkg-plist
1554
		fi
1555

    
1556
		if [ -f ${_pdir}/${_pname}/pkg-descr ]; then
1557
			_pdescr=${_pdir}/${_pname}/pkg-descr
1558
		fi
1559

    
1560
		sed -i '' -e "s,pfSense,${PRODUCT_NAME},g" \
1561
			  -e "s,https://www.pfsense.org,${PRODUCT_URL},g" \
1562
			  -e "/^MAINTAINER=/ s,^.*$,MAINTAINER=	${PRODUCT_EMAIL}," \
1563
			${_pdir}/${_pname}/Makefile ${_pdescr} ${_plist}
1564

    
1565
		# PHP module is special
1566
		if echo "${_pname}" | grep -q "^php[0-9]*-${PRODUCT_NAME}-module"; then
1567
			local _product_capital=$(echo ${PRODUCT_NAME} | tr '[a-z]' '[A-Z]')
1568
			sed -i '' -e "s,PHP_PFSENSE,PHP_${_product_capital},g" \
1569
				  -e "s,PFSENSE_SHARED_LIBADD,${_product_capital}_SHARED_LIBADD,g" \
1570
				  -e "s,pfSense,${PRODUCT_NAME},g" \
1571
				  -e "s,pfSense.c,${PRODUCT_NAME}\.c,g" \
1572
				${_pdir}/${_pname}/files/config.m4
1573

    
1574
			sed -i '' -e "s,COMPILE_DL_PFSENSE,COMPILE_DL_${_product_capital}," \
1575
				  -e "s,pfSense_module_entry,${PRODUCT_NAME}_module_entry,g" \
1576
				  -e "s,php_pfSense.h,php_${PRODUCT_NAME}\.h,g" \
1577
				  -e "/ZEND_GET_MODULE/ s,pfSense,${PRODUCT_NAME}," \
1578
				  -e "/PHP_PFSENSE_WORLD_EXTNAME/ s,pfSense,${PRODUCT_NAME}," \
1579
				${_pdir}/${_pname}/files/pfSense.c \
1580
				${_pdir}/${_pname}/files/dummynet.c \
1581
				${_pdir}/${_pname}/files/php_pfSense.h
1582
		fi
1583

    
1584
		if [ -d ${_pdir}/${_pname}/files ]; then
1585
			for fd in $(find ${_pdir}/${_pname}/files -name '*pfSense*'); do
1586
				local _fddir=$(dirname ${fd})
1587
				local _fdname=$(echo $(basename ${fd}) | sed "s,pfSense,${PRODUCT_NAME},")
1588

    
1589
				mv ${fd} ${_fddir}/${_fdname}
1590
			done
1591
		fi
1592
	done
1593
	echo "Done!" | tee -a ${LOGFILE}
1594
}
1595

    
1596
poudriere_create_ports_tree() {
1597
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1598

    
1599
	if ! poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
1600
		local _branch=""
1601
		if [ -z "${POUDRIERE_PORTS_GIT_URL}" ]; then
1602
			echo ">>> ERROR: POUDRIERE_PORTS_GIT_URL is not defined"
1603
			print_error_pfS
1604
		fi
1605
		if [ -n "${POUDRIERE_PORTS_GIT_BRANCH}" ]; then
1606
			_branch="-B ${POUDRIERE_PORTS_GIT_BRANCH}"
1607
		fi
1608
		echo -n ">>> Creating poudriere ports tree, it may take some time... " | tee -a ${LOGFILE}
1609
		if ! script -aq ${LOGFILE} poudriere ports -c -p "${POUDRIERE_PORTS_NAME}" -m git -U ${POUDRIERE_PORTS_GIT_URL} ${_branch} >/dev/null 2>&1; then
1610
			echo "" | tee -a ${LOGFILE}
1611
			echo ">>> ERROR: Error creating poudriere ports tree, aborting..." | tee -a ${LOGFILE}
1612
			print_error_pfS
1613
		fi
1614
		echo "Done!" | tee -a ${LOGFILE}
1615
		poudriere_rename_ports
1616
	fi
1617
}
1618

    
1619
poudriere_init() {
1620
	local _error=0
1621
	local _archs=$(poudriere_possible_archs)
1622

    
1623
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1624

    
1625
	# Sanity checks
1626
	if [ -z "${ZFS_TANK}" ]; then
1627
		echo ">>> ERROR: \$ZFS_TANK is empty" | tee -a ${LOGFILE}
1628
		error=1
1629
	fi
1630

    
1631
	if [ -z "${ZFS_ROOT}" ]; then
1632
		echo ">>> ERROR: \$ZFS_ROOT is empty" | tee -a ${LOGFILE}
1633
		error=1
1634
	fi
1635

    
1636
	if [ -z "${POUDRIERE_PORTS_NAME}" ]; then
1637
		echo ">>> ERROR: \$POUDRIERE_PORTS_NAME is empty" | tee -a ${LOGFILE}
1638
		error=1
1639
	fi
1640

    
1641
	if [ ${_error} -eq 1 ]; then
1642
		print_error_pfS
1643
	fi
1644

    
1645
	# Check if zpool exists
1646
	if ! zpool list ${ZFS_TANK} >/dev/null 2>&1; then
1647
		echo ">>> ERROR: ZFS tank ${ZFS_TANK} not found, please create it and try again..." | tee -a ${LOGFILE}
1648
		print_error_pfS
1649
	fi
1650

    
1651
	# Check if zfs rootfs exists
1652
	if ! zfs list ${ZFS_TANK}${ZFS_ROOT} >/dev/null 2>&1; then
1653
		echo -n ">>> Creating ZFS filesystem ${ZFS_TANK}${ZFS_ROOT}... "
1654
		if zfs create -o atime=off -o mountpoint=/usr/local${ZFS_ROOT} \
1655
		    ${ZFS_TANK}${ZFS_ROOT} >/dev/null 2>&1; then
1656
			echo "Done!"
1657
		else
1658
			echo "Failed!"
1659
			print_error_pfS
1660
		fi
1661
	fi
1662

    
1663
	# Make sure poudriere is installed
1664
	if [ ! -f /usr/local/bin/poudriere ]; then
1665
		echo ">>> Installing poudriere..." | tee -a ${LOGFILE}
1666
		if ! pkg install poudriere >/dev/null 2>&1; then
1667
			echo ">>> ERROR: poudriere was not installed, aborting..." | tee -a ${LOGFILE}
1668
			print_error_pfS
1669
		fi
1670
	fi
1671

    
1672
	# Create poudriere.conf
1673
	if [ -z "${POUDRIERE_PORTS_GIT_URL}" ]; then
1674
		echo ">>> ERROR: POUDRIERE_PORTS_GIT_URL is not defined"
1675
		print_error_pfS
1676
	fi
1677

    
1678
	# PARALLEL_JOBS us ncpu / 4 for best performance
1679
	local _parallel_jobs=$(sysctl -qn hw.ncpu)
1680
	_parallel_jobs=$((_parallel_jobs / 4))
1681

    
1682
	echo ">>> Creating poudriere.conf" | tee -a ${LOGFILE}
1683
	cat <<EOF >/usr/local/etc/poudriere.conf
1684
ZPOOL=${ZFS_TANK}
1685
ZROOTFS=${ZFS_ROOT}
1686
RESOLV_CONF=/etc/resolv.conf
1687
BASEFS=/usr/local/poudriere
1688
USE_PORTLINT=no
1689
USE_TMPFS=yes
1690
NOLINUX=yes
1691
DISTFILES_CACHE=/usr/ports/distfiles
1692
CHECK_CHANGED_OPTIONS=yes
1693
CHECK_CHANGED_DEPS=yes
1694
ATOMIC_PACKAGE_REPOSITORY=yes
1695
COMMIT_PACKAGES_ON_FAILURE=no
1696
KEEP_OLD_PACKAGES=yes
1697
KEEP_OLD_PACKAGES_COUNT=5
1698
ALLOW_MAKE_JOBS=yes
1699
PARALLEL_JOBS=${_parallel_jobs}
1700
EOF
1701

    
1702
	if pkg info -e ccache; then
1703
	cat <<EOF >>/usr/local/etc/poudriere.conf
1704
CCACHE_DIR=/var/cache/ccache
1705
EOF
1706
	fi
1707

    
1708
	# Create specific items conf
1709
	[ ! -d /usr/local/etc/poudriere.d ] \
1710
		&& mkdir -p /usr/local/etc/poudriere.d
1711

    
1712
	# Create DISTFILES_CACHE if it doesn't exist
1713
	if [ ! -d /usr/ports/distfiles ]; then
1714
		mkdir -p /usr/ports/distfiles
1715
	fi
1716

    
1717
	# Remove old jails
1718
	for jail_arch in ${_archs}; do
1719
		jail_name=$(poudriere_jail_name ${jail_arch})
1720

    
1721
		if poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
1722
			echo ">>> Poudriere jail ${jail_name} already exists, deleting it..." | tee -a ${LOGFILE}
1723
			poudriere jail -d -j "${jail_name}"
1724
		fi
1725
	done
1726

    
1727
	# Remove old ports tree
1728
	if poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
1729
		echo ">>> Poudriere ports tree ${POUDRIERE_PORTS_NAME} already exists, deleting it..." | tee -a ${LOGFILE}
1730
		poudriere ports -d -p "${POUDRIERE_PORTS_NAME}"
1731
	fi
1732

    
1733
	local native_xtools=""
1734
	# Now we are ready to create jails
1735
	for jail_arch in ${_archs}; do
1736
		jail_name=$(poudriere_jail_name ${jail_arch})
1737

    
1738
		if [ "${jail_arch}" = "arm.armv7" ]; then
1739
			native_xtools="-x"
1740
		else
1741
			native_xtools=""
1742
		fi
1743

    
1744
		echo -n ">>> Creating jail ${jail_name}, it may take some time... " | tee -a ${LOGFILE}
1745
		if ! script -aq ${LOGFILE} poudriere jail -c -j "${jail_name}" -v ${FREEBSD_BRANCH} \
1746
				-a ${jail_arch} -m git -U ${FREEBSD_REPO_BASE_POUDRIERE} ${native_xtools} >/dev/null 2>&1; then
1747
			echo "" | tee -a ${LOGFILE}
1748
			echo ">>> ERROR: Error creating jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
1749
			print_error_pfS
1750
		fi
1751
		echo "Done!" | tee -a ${LOGFILE}
1752
	done
1753

    
1754
	poudriere_create_ports_tree
1755

    
1756
	echo ">>> Poudriere is now configured!" | tee -a ${LOGFILE}
1757
}
1758

    
1759
poudriere_update_jails() {
1760
	local _archs=$(poudriere_possible_archs)
1761

    
1762
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1763

    
1764
	local native_xtools=""
1765
	for jail_arch in ${_archs}; do
1766
		jail_name=$(poudriere_jail_name ${jail_arch})
1767

    
1768
		local _create_or_update="-u"
1769
		local _create_or_update_text="Updating"
1770
		if ! poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
1771
			echo ">>> Poudriere jail ${jail_name} not found, creating..." | tee -a ${LOGFILE}
1772
			_create_or_update="-c -v ${FREEBSD_BRANCH} -a ${jail_arch} -m git -U ${FREEBSD_REPO_BASE_POUDRIERE}"
1773
			_create_or_update_text="Creating"
1774
		fi
1775

    
1776
		if [ "${jail_arch}" = "arm.armv7" ]; then
1777
			native_xtools="-x"
1778
		else
1779
			native_xtools=""
1780
		fi
1781

    
1782
		echo -n ">>> ${_create_or_update_text} jail ${jail_name}, it may take some time... " | tee -a ${LOGFILE}
1783
		if ! script -aq ${LOGFILE} poudriere jail ${_create_or_update} -j "${jail_name}" ${native_xtools} >/dev/null 2>&1; then
1784
			echo "" | tee -a ${LOGFILE}
1785
			echo ">>> ERROR: Error ${_create_or_update_text} jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
1786
			print_error_pfS
1787
		fi
1788
		echo "Done!" | tee -a ${LOGFILE}
1789
	done
1790
}
1791

    
1792
poudriere_update_ports() {
1793
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1794

    
1795
	# Create ports tree if necessary
1796
	if ! poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
1797
		poudriere_create_ports_tree
1798
	else
1799
		echo -n ">>> Resetting local changes on ports tree ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
1800
		script -aq ${LOGFILE} git -C "/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}" reset --hard >/dev/null 2>&1
1801
		script -aq ${LOGFILE} git -C "/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}" clean -fd >/dev/null 2>&1
1802
		echo "Done!" | tee -a ${LOGFILE}
1803
		echo -n ">>> Updating ports tree ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
1804
		script -aq ${LOGFILE} poudriere ports -u -p "${POUDRIERE_PORTS_NAME}" >/dev/null 2>&1
1805
		echo "Done!" | tee -a ${LOGFILE}
1806
		poudriere_rename_ports
1807
	fi
1808
}
1809

    
1810
poudriere_bulk() {
1811
	local _archs=$(poudriere_possible_archs)
1812
	local _makeconf
1813

    
1814
	# Create DISTFILES_CACHE if it doesn't exist
1815
	if [ ! -d /usr/ports/distfiles ]; then
1816
		mkdir -p /usr/ports/distfiles
1817
	fi
1818

    
1819
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1820

    
1821
	if [ -n "${UPLOAD}" -a -z "${PKG_RSYNC_HOSTNAME}" ]; then
1822
		echo ">>> ERROR: PKG_RSYNC_HOSTNAME is not set"
1823
		print_error_pfS
1824
	fi
1825

    
1826
	rm -f ${LOGFILE}
1827

    
1828
	poudriere_create_ports_tree
1829

    
1830
	[ -d /usr/local/etc/poudriere.d ] || \
1831
		mkdir -p /usr/local/etc/poudriere.d
1832

    
1833
	_makeconf=/usr/local/etc/poudriere.d/${POUDRIERE_PORTS_NAME}-make.conf
1834
	if [ -f "${BUILDER_TOOLS}/conf/pfPorts/make.conf" ]; then
1835
		cp -f "${BUILDER_TOOLS}/conf/pfPorts/make.conf" ${_makeconf}
1836
	fi
1837

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

    
1840
PKG_REPO_BRANCH_DEVEL=${PKG_REPO_BRANCH_DEVEL}
1841
PKG_REPO_BRANCH_RELEASE=${PKG_REPO_BRANCH_RELEASE}
1842
PKG_REPO_SERVER_DEVEL=${PKG_REPO_SERVER_DEVEL}
1843
PKG_REPO_SERVER_RELEASE=${PKG_REPO_SERVER_RELEASE}
1844
POUDRIERE_PORTS_NAME=${POUDRIERE_PORTS_NAME}
1845
PFSENSE_DEFAULT_REPO=${PFSENSE_DEFAULT_REPO}
1846
PRODUCT_NAME=${PRODUCT_NAME}
1847
REPO_BRANCH_PREFIX=${REPO_BRANCH_PREFIX}
1848
EOF
1849

    
1850
	local _value=""
1851
	for jail_arch in ${_archs}; do
1852
		eval "_value=\${PKG_REPO_BRANCH_DEVEL_${jail_arch##*.}}"
1853
		if [ -n "${_value}" ]; then
1854
			echo "PKG_REPO_BRANCH_DEVEL_${jail_arch##*.}=${_value}" \
1855
				>> ${_makeconf}
1856
		fi
1857
		eval "_value=\${PKG_REPO_BRANCH_RELEASE_${jail_arch##*.}}"
1858
		if [ -n "${_value}" ]; then
1859
			echo "PKG_REPO_BRANCH_RELEASE_${jail_arch##*.}=${_value}" \
1860
				>> ${_makeconf}
1861
		fi
1862
		eval "_value=\${PKG_REPO_SERVER_DEVEL_${jail_arch##*.}}"
1863
		if [ -n "${_value}" ]; then
1864
			echo "PKG_REPO_SERVER_DEVEL_${jail_arch##*.}=${_value}" \
1865
				>> ${_makeconf}
1866
		fi
1867
		eval "_value=\${PKG_REPO_SERVER_RELEASE_${jail_arch##*.}}"
1868
		if [ -n "${_value}" ]; then
1869
			echo "PKG_REPO_SERVER_RELEASE_${jail_arch##*.}=${_value}" \
1870
				>> ${_makeconf}
1871
		fi
1872
	done
1873

    
1874
	# Change version of pfSense meta ports for snapshots
1875
	if [ -z "${_IS_RELEASE}" ]; then
1876
		local _meta_pkg_version="$(echo "${PRODUCT_VERSION}" | sed 's,DEVELOPMENT,ALPHA,')-${DATESTRING}"
1877
		sed -i '' \
1878
			-e "/^DISTVERSION/ s,^.*,DISTVERSION=	${_meta_pkg_version}," \
1879
			-e "/^PORTREVISION=/d" \
1880
			/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}/security/${PRODUCT_NAME}/Makefile \
1881
			/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}/sysutils/${PRODUCT_NAME}-repo/Makefile
1882
	fi
1883

    
1884
	# Copy over pkg repo templates to pfSense-repo
1885
	mkdir -p /usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}/sysutils/${PRODUCT_NAME}-repo/files
1886
	cp -f ${PKG_REPO_BASE}/* \
1887
		/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}/sysutils/${PRODUCT_NAME}-repo/files
1888

    
1889
	for jail_arch in ${_archs}; do
1890
		jail_name=$(poudriere_jail_name ${jail_arch})
1891

    
1892
		if ! poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
1893
			echo ">>> Poudriere jail ${jail_name} not found, skipping..." | tee -a ${LOGFILE}
1894
			continue
1895
		fi
1896

    
1897
		_ref_bulk=${SCRATCHDIR}/poudriere_bulk.${POUDRIERE_BRANCH}.ref.${jail_arch}
1898
		rm -rf ${_ref_bulk} ${_ref_bulk}.tmp
1899
		touch ${_ref_bulk}.tmp
1900
		if [ -f "${POUDRIERE_BULK}.${jail_arch#*.}" ]; then
1901
			cat "${POUDRIERE_BULK}.${jail_arch#*.}" >> ${_ref_bulk}.tmp
1902
		fi
1903
		if [ -f "${POUDRIERE_BULK}" ]; then
1904
			cat "${POUDRIERE_BULK}" >> ${_ref_bulk}.tmp
1905
		fi
1906
		cat ${_ref_bulk}.tmp | sort -u > ${_ref_bulk}
1907

    
1908
		_bulk=${SCRATCHDIR}/poudriere_bulk.${POUDRIERE_BRANCH}.${jail_arch}
1909
		sed -e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" ${_ref_bulk} > ${_bulk}
1910

    
1911
		local _exclude_bulk="${POUDRIERE_BULK}.exclude.${jail_arch}"
1912
		if [ -f "${_exclude_bulk}" ]; then
1913
			mv ${_bulk} ${_bulk}.tmp
1914
			sed -e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" ${_exclude_bulk} > ${_bulk}.exclude
1915
			cat ${_bulk}.tmp ${_bulk}.exclude | sort | uniq -u > ${_bulk}
1916
			rm -f ${_bulk}.tmp ${_bulk}.exclude
1917
		fi
1918

    
1919
		if ! poudriere bulk -f ${_bulk} -j ${jail_name} -p ${POUDRIERE_PORTS_NAME}; then
1920
			echo ">>> ERROR: Something went wrong..."
1921
			print_error_pfS
1922
		fi
1923

    
1924
		echo ">>> Cleaning up old packages from repo..."
1925
		if ! poudriere pkgclean -f ${_bulk} -j ${jail_name} -p ${POUDRIERE_PORTS_NAME} -y; then
1926
			echo ">>> ERROR: Something went wrong..."
1927
			print_error_pfS
1928
		fi
1929

    
1930
		pkg_repo_rsync "/usr/local/poudriere/data/packages/${jail_name}-${POUDRIERE_PORTS_NAME}"
1931
	done
1932
}
1933

    
1934
# This routine is called to write out to stdout
1935
# a string. The string is appended to $SNAPSHOTSLOGFILE
1936
snapshots_update_status() {
1937
	if [ -z "$1" ]; then
1938
		return
1939
	fi
1940
	if [ -z "${SNAPSHOTS}" -a -z "${POUDRIERE_SNAPSHOTS}" ]; then
1941
		return
1942
	fi
1943
	echo "$*"
1944
	echo "`date` -|- $*" >> $SNAPSHOTSLOGFILE
1945
}
1946

    
1947
create_sha256() {
1948
	local _file="${1}"
1949

    
1950
	if [ ! -f "${_file}" ]; then
1951
		return 1
1952
	fi
1953

    
1954
	( \
1955
		cd $(dirname ${_file}) && \
1956
		sha256 $(basename ${_file}) > $(basename ${_file}).sha256 \
1957
	)
1958
}
1959

    
1960
snapshots_create_latest_symlink() {
1961
	local _image="${1}"
1962

    
1963
	if [ -z "${_image}" ]; then
1964
		return
1965
	fi
1966

    
1967
	if [ -z "${TIMESTAMP_SUFFIX}" ]; then
1968
		return
1969
	fi
1970

    
1971
	if [ ! -f "${_image}" ]; then
1972
		return
1973
	fi
1974

    
1975
	local _symlink=$(echo ${_image} | sed "s,${TIMESTAMP_SUFFIX},-latest,")
1976
	ln -sf $(basename ${_image}) ${_symlink}
1977
	ln -sf $(basename ${_image}).sha256 ${_symlink}.sha256
1978
}
1979

    
1980
snapshots_create_sha256() {
1981
	local _img=""
1982

    
1983
	for _img in ${ISOPATH} ${MEMSTICKPATH} ${MEMSTICKSERIALPATH} ${MEMSTICKADIPATH} ${OVAPATH} ${VARIANTIMAGES}; do
1984
		if [ -f "${_img}.gz" ]; then
1985
			_img="${_img}.gz"
1986
		fi
1987
		if [ ! -f "${_img}" ]; then
1988
			continue
1989
		fi
1990
		create_sha256 ${_img}
1991
		snapshots_create_latest_symlink ${_img}
1992
	done
1993
}
1994

    
1995
snapshots_scp_files() {
1996
	if [ -z "${RSYNC_COPY_ARGUMENTS}" ]; then
1997
		RSYNC_COPY_ARGUMENTS="-ave ssh --timeout=60"
1998
	fi
1999

    
2000
	snapshots_update_status ">>> Copying core pkg repo to ${PKG_RSYNC_HOSTNAME}"
2001
	pkg_repo_rsync "${CORE_PKG_PATH}"
2002
	snapshots_update_status ">>> Finished copying core pkg repo"
2003

    
2004
	for _rsyncip in ${RSYNCIP}; do
2005
		snapshots_update_status ">>> Copying files to ${_rsyncip}"
2006

    
2007
		# Ensure directory(s) are available
2008
		ssh ${RSYNCUSER}@${_rsyncip} "mkdir -p ${RSYNCPATH}/installer"
2009
		if [ -d $IMAGES_FINAL_DIR/virtualization ]; then
2010
			ssh ${RSYNCUSER}@${_rsyncip} "mkdir -p ${RSYNCPATH}/virtualization"
2011
		fi
2012
		# ensure permissions are correct for r+w
2013
		ssh ${RSYNCUSER}@${_rsyncip} "chmod -R ug+rw ${RSYNCPATH}/."
2014
		rsync $RSYNC_COPY_ARGUMENTS $IMAGES_FINAL_DIR/installer/* \
2015
			${RSYNCUSER}@${_rsyncip}:${RSYNCPATH}/installer/
2016
		if [ -d $IMAGES_FINAL_DIR/virtualization ]; then
2017
			rsync $RSYNC_COPY_ARGUMENTS $IMAGES_FINAL_DIR/virtualization/* \
2018
				${RSYNCUSER}@${_rsyncip}:${RSYNCPATH}/virtualization/
2019
		fi
2020

    
2021
		snapshots_update_status ">>> Finished copying files."
2022
	done
2023
}
(2-2/3)