Project

General

Profile

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

    
31
if [ -z "${IMAGES_FINAL_DIR}" -o "${IMAGES_FINAL_DIR}" = "/" ]; then
32
	echo "IMAGES_FINAL_DIR is not defined"
33
	print_error_pfS
34
fi
35

    
36
kldload filemon >/dev/null 2>&1
37

    
38
lc() {
39
	echo "${1}" | tr '[[:upper:]]' '[[:lower:]]'
40
}
41

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

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

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

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

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

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

    
89
	${BUILDER_SCRIPTS}/create_core_pkg.sh \
90
		-t "${_template_path}" \
91
		-f "${_flavor}" \
92
		-v "${_version}" \
93
		-r "${_root}" \
94
		-F "${_filter}" \
95
		-d "${CORE_PKG_REAL_PATH}/All" \
96
		|| print_error_pfS
97
}
98

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

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

    
129
get_pkg_name() {
130
	echo "${PRODUCT_NAME}-${1}-${CORE_PKG_VERSION}"
131
}
132

    
133
# This routine builds all related kernels
134
build_all_kernels() {
135
	# Set KERNEL_BUILD_PATH if it has not been set
136
	if [ -z "${KERNEL_BUILD_PATH}" ]; then
137
		KERNEL_BUILD_PATH=$SCRATCHDIR/kernels
138
		echo ">>> KERNEL_BUILD_PATH has not been set. Setting to ${KERNEL_BUILD_PATH}!"
139
	fi
140

    
141
	[ -d "${KERNEL_BUILD_PATH}" ] \
142
		&& rm -rf ${KERNEL_BUILD_PATH}
143

    
144
	# Build embedded kernel
145
	for BUILD_KERNEL in $BUILD_KERNELS; do
146
		unset KERNCONF
147
		unset KERNEL_DESTDIR
148
		unset KERNEL_NAME
149
		export KERNCONF=$BUILD_KERNEL
150
		export KERNEL_DESTDIR="$KERNEL_BUILD_PATH/$BUILD_KERNEL"
151
		export KERNEL_NAME=${BUILD_KERNEL}
152

    
153
		LOGFILE="${BUILDER_LOGS}/kernel.${KERNCONF}.${TARGET}.log"
154
		echo ">>> Building $BUILD_KERNEL kernel."  | tee -a ${LOGFILE}
155

    
156
		if [ -n "${NO_BUILDKERNEL}" -a -f "${CORE_PKG_ALL_PATH}/$(get_pkg_name kernel-${KERNEL_NAME}).txz" ]; then
157
			echo ">>> NO_BUILDKERNEL set, skipping build" | tee -a ${LOGFILE}
158
			continue
159
		fi
160

    
161
		buildkernel
162

    
163
		echo ">>> Staging $BUILD_KERNEL kernel..." | tee -a ${LOGFILE}
164
		installkernel
165

    
166
		ensure_kernel_exists $KERNEL_DESTDIR
167

    
168
		echo ">>> Creating pkg of $KERNEL_NAME-debug kernel to staging area..."  | tee -a ${LOGFILE}
169
		core_pkg_create kernel-debug ${KERNEL_NAME} ${CORE_PKG_VERSION} ${KERNEL_DESTDIR} \*.symbols
170
		find ${KERNEL_DESTDIR} -name '*.symbols' -type f -delete
171

    
172
		echo ">>> Creating pkg of $KERNEL_NAME kernel to staging area..."  | tee -a ${LOGFILE}
173
		core_pkg_create kernel ${KERNEL_NAME} ${CORE_PKG_VERSION} ${KERNEL_DESTDIR}
174

    
175
		rm -rf $KERNEL_DESTDIR 2>&1 1>/dev/null
176
	done
177
}
178

    
179
install_default_kernel() {
180
	if [ -z "${1}" ]; then
181
		echo ">>> ERROR: install_default_kernel called without a kernel config name"| tee -a ${LOGFILE}
182
		print_error_pfS
183
	fi
184

    
185
	export KERNEL_NAME="${1}"
186

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

    
189
	# Copy kernel package to chroot, otherwise pkg won't find it to install
190
	if ! pkg_chroot_add ${FINAL_CHROOT_DIR} kernel-${KERNEL_NAME}; then
191
		echo ">>> ERROR: Error installing kernel package $(get_pkg_name kernel-${KERNEL_NAME}).txz" | tee -a ${LOGFILE}
192
		print_error_pfS
193
	fi
194

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

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

    
220
	unset KERNEL_NAME
221
}
222

    
223
# This builds FreeBSD (make buildworld)
224
# Imported from FreeSBIE
225
make_world() {
226
	LOGFILE=${BUILDER_LOGS}/buildworld.${TARGET}
227
	echo ">>> LOGFILE set to $LOGFILE." | tee -a ${LOGFILE}
228
	if [ -n "${NO_BUILDWORLD}" ]; then
229
		echo ">>> NO_BUILDWORLD set, skipping build" | tee -a ${LOGFILE}
230
		return
231
	fi
232

    
233
	echo ">>> $(LC_ALL=C date) - Starting build world for ${TARGET} architecture..." | tee -a ${LOGFILE}
234
	script -aq $LOGFILE ${BUILDER_SCRIPTS}/build_freebsd.sh -K -s ${FREEBSD_SRC_DIR} \
235
		|| print_error_pfS
236
	echo ">>> $(LC_ALL=C date) - Finished build world for ${TARGET} architecture..." | tee -a ${LOGFILE}
237

    
238
	LOGFILE=${BUILDER_LOGS}/installworld.${TARGET}
239
	echo ">>> LOGFILE set to $LOGFILE." | tee -a ${LOGFILE}
240

    
241
	[ -d "${INSTALLER_CHROOT_DIR}" ] \
242
		|| mkdir -p ${INSTALLER_CHROOT_DIR}
243

    
244
	echo ">>> Installing world with bsdinstall for ${TARGET} architecture..." | tee -a ${LOGFILE}
245
	script -aq $LOGFILE ${BUILDER_SCRIPTS}/install_freebsd.sh -i -K \
246
		-s ${FREEBSD_SRC_DIR} \
247
		-d ${INSTALLER_CHROOT_DIR} \
248
		|| print_error_pfS
249

    
250
	echo ">>> Installing world without bsdinstall for ${TARGET} architecture..." | tee -a ${LOGFILE}
251
	script -aq $LOGFILE ${BUILDER_SCRIPTS}/install_freebsd.sh -K \
252
		-s ${FREEBSD_SRC_DIR} \
253
		-d ${STAGE_CHROOT_DIR} \
254
		|| print_error_pfS
255

    
256
	# XXX It must go to the scripts
257
	[ -d "${STAGE_CHROOT_DIR}/usr/local/bin" ] \
258
		|| mkdir -p ${STAGE_CHROOT_DIR}/usr/local/bin
259
	makeargs="DESTDIR=${STAGE_CHROOT_DIR}"
260
	echo ">>> Building and installing crypto tools and athstats for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
261
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/crypto ${makeargs} clean all install || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
262
	# XXX FIX IT
263
#	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} clean all install || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
264
	echo ">>> Building and installing crypto tools and athstats for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
265

    
266
	unset makeargs
267
}
268

    
269
nanobsd_image_filename() {
270
	local _size="$1"
271
	local _type="$2"
272

    
273
	echo "$NANOBSD_IMG_TEMPLATE" | sed \
274
		-e "s,%%SIZE%%,${_size},g" \
275
		-e "s,%%TYPE%%,${_type},g"
276

    
277
	return 0
278
}
279

    
280
# This routine originated in nanobsd.sh
281
nanobsd_set_flash_details () {
282
	a1=$(echo $1 | tr '[:upper:]' '[:lower:]')
283

    
284
	# Source:
285
	#	SanDisk CompactFlash Memory Card
286
	#	Product Manual
287
	#	Version 10.9
288
	#	Document No. 20-10-00038
289
	#	April 2005
290
	# Table 2-7
291
	# NB: notice math error in SDCFJ-4096-388 line.
292
	#
293
	case "${a1}" in
294
		2048|2048m|2048mb|2g)
295
			NANO_MEDIASIZE=$((1989999616/512))
296
			;;
297
		4096|4096m|4096mb|4g)
298
			NANO_MEDIASIZE=$((3989999616/512))
299
			;;
300
		8192|8192m|8192mb|8g)
301
			NANO_MEDIASIZE=$((7989999616/512))
302
			;;
303
		16384|16384m|16384mb|16g)
304
			NANO_MEDIASIZE=$((15989999616/512))
305
			;;
306
		*)
307
			echo "Unknown Flash capacity"
308
			exit 2
309
			;;
310
	esac
311

    
312
	NANO_HEADS=16
313
	NANO_SECTS=63
314

    
315
	echo ">>> [nanoo] $1"
316
	echo ">>> [nanoo] NANO_MEDIASIZE: $NANO_MEDIASIZE"
317
	echo ">>> [nanoo] NANO_HEADS: $NANO_HEADS"
318
	echo ">>> [nanoo] NANO_SECTS: $NANO_SECTS"
319
	echo ">>> [nanoo] NANO_BOOT0CFG: $NANO_BOOT0CFG"
320
}
321

    
322
# This routine originated in nanobsd.sh
323
create_nanobsd_diskimage () {
324
	if [ -z "${1}" ]; then
325
		echo ">>> ERROR: Type of image has not been specified"
326
		print_error_pfS
327
	fi
328
	if [ -z "${2}" ]; then
329
		echo ">>> ERROR: Size of image has not been specified"
330
		print_error_pfS
331
	fi
332

    
333
	if [ "${1}" = "nanobsd" ]; then
334
		# It's serial
335
		export NANO_BOOTLOADER="boot/boot0sio"
336
	elif [ "${1}" = "nanobsd-vga" ]; then
337
		# It's vga
338
		export NANO_BOOTLOADER="boot/boot0"
339
	else
340
		echo ">>> ERROR: Type of image to create unknown"
341
		print_error_pfS
342
	fi
343

    
344
	if [ -z "${2}" ]; then
345
		echo ">>> ERROR: Media size(s) not specified."
346
		print_error_pfS
347
	fi
348

    
349
	if [ -z "${2}" ]; then
350
		echo ">>> ERROR: FLASH_SIZE is not set."
351
		print_error_pfS
352
	fi
353

    
354
	LOGFILE=${BUILDER_LOGS}/${1}.${TARGET}
355
	# Prepare folder to be put in image
356
	customize_stagearea_for_image "${1}"
357
	install_default_kernel ${DEFAULT_KERNEL} "no"
358

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

    
361
	local BOOTCONF=${FINAL_CHROOT_DIR}/boot.config
362
	local LOADERCONF=${FINAL_CHROOT_DIR}/boot/loader.conf
363

    
364
	if [ "${1}" = "nanobsd" ]; then
365
		# Tell loader to use serial console early.
366
		echo "-S115200 -h" >> ${BOOTCONF}
367

    
368
		# Remove old console options if present.
369
		[ -f "${LOADERCONF}" ] \
370
			&& sed -i "" -Ee "/(console|boot_multicons|boot_serial|hint.uart)/d" ${LOADERCONF}
371
		# Activate serial console+video console in loader.conf
372
		echo 'loader_color="NO"' >> ${LOADERCONF}
373
		echo 'beastie_disable="YES"' >> ${LOADERCONF}
374
		echo 'boot_serial="YES"' >> ${LOADERCONF}
375
		echo 'console="comconsole"' >> ${LOADERCONF}
376
		echo 'comconsole_speed="115200"' >> ${LOADERCONF}
377
	fi
378
	echo 'autoboot_delay="5"' >> ${LOADERCONF}
379

    
380
	# Old systems will run (pre|post)_upgrade_command from /tmp
381
	if [ -f ${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/pre_upgrade_command ]; then
382
		cp -p \
383
			${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/pre_upgrade_command \
384
			${FINAL_CHROOT_DIR}/tmp
385
	fi
386
	if [ -f ${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/post_upgrade_command ]; then
387
		cp -p \
388
			${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/post_upgrade_command \
389
			${FINAL_CHROOT_DIR}/tmp
390
	fi
391

    
392
	mkdir -p ${IMAGES_FINAL_DIR}/nanobsd
393

    
394
	for _NANO_MEDIASIZE in ${2}; do
395
		if [ -z "${_NANO_MEDIASIZE}" ]; then
396
			continue;
397
		fi
398

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

    
402
		IMG="${IMAGES_FINAL_DIR}/nanobsd/$(nanobsd_image_filename ${_NANO_MEDIASIZE} ${1})"
403

    
404
		nanobsd_set_flash_details ${_NANO_MEDIASIZE}
405

    
406
		# These are defined in FlashDevice and on builder_default.sh
407
		echo $NANO_MEDIASIZE \
408
			$NANO_IMAGES \
409
			$NANO_SECTS \
410
			$NANO_HEADS \
411
			$NANO_CODESIZE \
412
			$NANO_CONFSIZE \
413
			$NANO_DATASIZE |
414
awk '
415
{
416
	printf "# %s\n", $0
417

    
418
	# size of cylinder in sectors
419
	cs = $3 * $4
420

    
421
	# number of full cylinders on media
422
	cyl = int ($1 / cs)
423

    
424
	# output fdisk geometry spec, truncate cyls to 1023
425
	if (cyl <= 1023)
426
		print "g c" cyl " h" $4 " s" $3
427
	else
428
		print "g c" 1023 " h" $4 " s" $3
429

    
430
	if ($7 > 0) {
431
		# size of data partition in full cylinders
432
		dsl = int (($7 + cs - 1) / cs)
433
	} else {
434
		dsl = 0;
435
	}
436

    
437
	# size of config partition in full cylinders
438
	csl = int (($6 + cs - 1) / cs)
439

    
440
	if ($5 == 0) {
441
		# size of image partition(s) in full cylinders
442
		isl = int ((cyl - dsl - csl) / $2)
443
	} else {
444
		isl = int (($5 + cs - 1) / cs)
445
	}
446

    
447
	# First image partition start at second track
448
	print "p 1 165 " $3, isl * cs - $3
449
	c = isl * cs;
450

    
451
	# Second image partition (if any) also starts offset one
452
	# track to keep them identical.
453
	if ($2 > 1) {
454
		print "p 2 165 " $3 + c, isl * cs - $3
455
		c += isl * cs;
456
	}
457

    
458
	# Config partition starts at cylinder boundary.
459
	print "p 3 165 " c, csl * cs
460
	c += csl * cs
461

    
462
	# Data partition (if any) starts at cylinder boundary.
463
	if ($7 > 0) {
464
		print "p 4 165 " c, dsl * cs
465
	} else if ($7 < 0 && $1 > c) {
466
		print "p 4 165 " c, $1 - c
467
	} else if ($1 < c) {
468
		print "Disk space overcommitted by", \
469
		    c - $1, "sectors" > "/dev/stderr"
470
		exit 2
471
	}
472

    
473
	# Force slice 1 to be marked active. This is necessary
474
	# for booting the image from a USB device to work.
475
	print "a 1"
476
}
477
	' > ${SCRATCHDIR}/_.fdisk
478

    
479
		MNT=${SCRATCHDIR}/_.mnt
480
		mkdir -p ${MNT}
481

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

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

    
488
		fdisk -i -f ${SCRATCHDIR}/_.fdisk ${MD} 2>&1 >> ${LOGFILE}
489
		fdisk ${MD} 2>&1 >> ${LOGFILE}
490

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

    
493
		# Create first image
494
		bsdlabel -m i386 -w -B -b ${FINAL_CHROOT_DIR}/boot/boot ${MD}s1 2>&1 >> ${LOGFILE}
495
		bsdlabel -m i386 ${MD}s1 2>&1 >> ${LOGFILE}
496
		local _label=$(lc ${PRODUCT_NAME})
497
		newfs -L ${_label}0 ${NANO_NEWFS} /dev/${MD}s1a 2>&1 >> ${LOGFILE}
498
		mount /dev/ufs/${_label}0 ${MNT}
499
		if [ $? -ne 0 ]; then
500
			echo ">>> ERROR: Something wrong happened during mount of first slice image creation. STOPPING!" | tee -a ${LOGFILE}
501
			print_error_pfS
502
		fi
503
		# Consider the unmounting as well
504
		trap "umount /dev/ufs/${_label}0; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
505

    
506
		clone_directory_contents ${FINAL_CHROOT_DIR} ${MNT}
507

    
508
		# Set NanoBSD image size
509
		echo "${_NANO_MEDIASIZE}" > ${MNT}/etc/nanosize.txt
510

    
511
		echo "/dev/ufs/${_label}0 / ufs ro,sync,noatime 1 1" > ${MNT}/etc/fstab
512
		if [ $NANO_CONFSIZE -gt 0 ] ; then
513
			echo "/dev/ufs/cf /cf ufs ro,sync,noatime 1 1" >> ${MNT}/etc/fstab
514
		fi
515

    
516
		umount ${MNT}
517
		# Restore the original trap
518
		trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
519

    
520
		# Setting NANO_IMAGES to 1 and NANO_INIT_IMG2 will tell
521
		# NanoBSD to only create one partition.  We default to 2
522
		# partitions in case anything happens to the first the
523
		# operator can boot from the 2nd and should be OK.
524

    
525
		# Before just going to use dd for duplicate think!
526
		# The images are created as sparse so lets take advantage
527
		# of that by just exec some commands.
528
		if [ $NANO_IMAGES -gt 1 -a $NANO_INIT_IMG2 -gt 0 ] ; then
529
			# Duplicate to second image (if present)
530
			echo ">>> Creating NanoBSD second slice by duplicating first slice." | tee -a ${LOGFILE}
531
			# Create second image
532
			dd if=/dev/${MD}s1 of=/dev/${MD}s2 conv=sparse bs=64k 2>&1 >> ${LOGFILE}
533
			tunefs -L ${_label}1 /dev/${MD}s2a 2>&1 >> ${LOGFILE}
534
			mount /dev/ufs/${_label}1 ${MNT}
535
			if [ $? -ne 0 ]; then
536
				echo ">>> ERROR: Something wrong happened during mount of second slice image creation. STOPPING!" | tee -a ${LOGFILE}
537
				print_error_pfS
538
			fi
539
			# Consider the unmounting as well
540
			trap "umount /dev/ufs/${_label}1; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
541

    
542
			echo "/dev/ufs/${_label}1 / ufs ro,sync,noatime 1 1" > ${MNT}/etc/fstab
543
			if [ $NANO_CONFSIZE -gt 0 ] ; then
544
				echo "/dev/ufs/cf /cf ufs ro,sync,noatime 1 1" >> ${MNT}/etc/fstab
545
			fi
546

    
547
			umount ${MNT}
548
			# Restore the trap back
549
			trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
550
		fi
551

    
552
		# Create Data slice, if any.
553
		# Note the changing of the variable to NANO_CONFSIZE
554
		# from NANO_DATASIZE.  We also added glabel support
555
		# and populate the Product configuration from the /cf
556
		# directory located in FINAL_CHROOT_DIR
557
		if [ $NANO_CONFSIZE -gt 0 ] ; then
558
			echo ">>> Creating /cf area to hold config.xml"
559
			newfs -L cf ${NANO_NEWFS} /dev/${MD}s3 2>&1 >> ${LOGFILE}
560
			# Mount data partition and copy contents of /cf
561
			# Can be used later to create custom default config.xml while building
562
			mount /dev/ufs/cf ${MNT}
563
			if [ $? -ne 0 ]; then
564
				echo ">>> ERROR: Something wrong happened during mount of cf slice image creation. STOPPING!" | tee -a ${LOGFILE}
565
				print_error_pfS
566
			fi
567
			# Consider the unmounting as well
568
			trap "umount /dev/ufs/cf; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
569

    
570
			clone_directory_contents ${FINAL_CHROOT_DIR}/cf ${MNT}
571

    
572
			umount ${MNT}
573
			# Restore the trap back
574
			trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
575
		else
576
			">>> [nanoo] NANO_CONFSIZE is not set. Not adding a /conf partition.. You sure about this??" | tee -a ${LOGFILE}
577
		fi
578

    
579
		mdconfig -d -u $MD
580
		# Restore default action
581
		trap "-" 1 2 15 EXIT
582

    
583
		# Check each image and ensure that they are over
584
		# 3 megabytes.  If either image is under 20 megabytes
585
		# in size then error out.
586
		IMGSIZE=$(stat -f "%z" ${IMG})
587
		CHECKSIZE="20040710"
588
		if [ "$IMGSIZE" -lt "$CHECKSIZE" ]; then
589
			echo ">>> ERROR: Something went wrong when building NanoBSD.  The image size is under 20 megabytes!" | tee -a ${LOGFILE}
590
			print_error_pfS
591
		fi
592

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

    
596
		gzip -qf $IMG &
597
		_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
598
	done
599

    
600
	unset IMG
601
	unset IMGSIZE
602
}
603

    
604
# This routine creates a ova image that contains
605
# a ovf and vmdk file. These files can be imported
606
# right into vmware or virtual box.
607
# (and many other emulation platforms)
608
# http://www.vmware.com/pdf/ovf_whitepaper_specification.pdf
609
create_ova_image() {
610
	# XXX create a .ovf php creator that you can pass:
611
	#     1. populatedSize
612
	#     2. license
613
	#     3. product name
614
	#     4. version
615
	#     5. number of network interface cards
616
	#     6. allocationUnits
617
	#     7. capacity
618
	#     8. capacityAllocationUnits
619

    
620
	LOGFILE=${BUILDER_LOGS}/ova.${TARGET}.log
621

    
622
	if [ -d "${OVA_TMP}" ]; then
623
		chflags -R noschg ${OVA_TMP}
624
		rm -rf ${OVA_TMP}
625
	fi
626

    
627
	mkdir -p $(dirname ${OVAPATH})
628

    
629
	local _mntdir=${OVA_TMP}/mnt
630
	mkdir -p ${_mntdir}
631

    
632
	if [ -z "${OVA_SWAP_PART_SIZE_IN_GB}" -o "${OVA_SWAP_PART_SIZE_IN_GB}" = "0" ]; then
633
		# first partition size (freebsd-ufs)
634
		local OVA_FIRST_PART_SIZE_IN_GB=${VMDK_DISK_CAPACITY_IN_GB}
635
		# Calculate real first partition size, removing 128 blocks (65536 bytes) beginning/loader
636
		local OVA_FIRST_PART_SIZE=$((${OVA_FIRST_PART_SIZE_IN_GB}*1024*1024*1024-65536))
637
		# Unset swap partition size variable
638
		unset OVA_SWAP_PART_SIZE
639
		# Parameter used by mkimg
640
		unset OVA_SWAP_PART_PARAM
641
	else
642
		# first partition size (freebsd-ufs)
643
		local OVA_FIRST_PART_SIZE_IN_GB=$((VMDK_DISK_CAPACITY_IN_GB-OVA_SWAP_PART_SIZE_IN_GB))
644
		# Use first partition size in g
645
		local OVA_FIRST_PART_SIZE="${OVA_FIRST_PART_SIZE_IN_GB}g"
646
		# Calculate real swap size, removing 128 blocks (65536 bytes) beginning/loader
647
		local OVA_SWAP_PART_SIZE=$((${OVA_SWAP_PART_SIZE_IN_GB}*1024*1024*1024-65536))
648
		# Parameter used by mkimg
649
		local OVA_SWAP_PART_PARAM="-p freebsd-swap/swap0::${OVA_SWAP_PART_SIZE}"
650
	fi
651

    
652
	# Prepare folder to be put in image
653
	customize_stagearea_for_image "ova"
654
	install_default_kernel ${DEFAULT_KERNEL} "no"
655

    
656
	# Fill fstab
657
	echo ">>> Installing platform specific items..." | tee -a ${LOGFILE}
658
	echo "/dev/gpt/${PRODUCT_NAME}	/	ufs		rw	1	1" > ${FINAL_CHROOT_DIR}/etc/fstab
659
	if [ -n "${OVA_SWAP_PART_SIZE}" ]; then
660
		echo "/dev/gpt/swap0	none	swap	sw	0	0" >> ${FINAL_CHROOT_DIR}/etc/fstab
661
	fi
662

    
663
	# Create / partition
664
	echo -n ">>> Creating / partition... " | tee -a ${LOGFILE}
665
	truncate -s ${OVA_FIRST_PART_SIZE} ${OVA_TMP}/${OVFUFS}
666
	local _md=$(mdconfig -a -f ${OVA_TMP}/${OVFUFS})
667
	trap "mdconfig -d -u ${_md}; return" 1 2 15 EXIT
668

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

    
671
	if ! mount /dev/${_md} ${_mntdir} 2>&1 >>${LOGFILE}; then
672
		echo "Failed!" | tee -a ${LOGFILE}
673
		echo ">>> ERROR: Error mounting temporary vmdk image. STOPPING!" | tee -a ${LOGFILE}
674
		print_error_pfS
675
	fi
676
	trap "umount ${_mntdir}; mdconfig -d -u ${_md}; return" 1 2 15 EXIT
677

    
678
	echo "Done!" | tee -a ${LOGFILE}
679

    
680
	clone_directory_contents ${FINAL_CHROOT_DIR} ${_mntdir}
681

    
682
	sync
683
	umount ${_mntdir} 2>&1 >>${LOGFILE}
684
	mdconfig -d -u ${_md}
685
	trap "-" 1 2 15 EXIT
686

    
687
	# Create raw disk
688
	echo -n ">>> Creating raw disk... " | tee -a ${LOGFILE}
689
	mkimg \
690
		-s gpt \
691
		-f raw \
692
		-b ${FINAL_CHROOT_DIR}/boot/pmbr \
693
		-p freebsd-boot:=${FINAL_CHROOT_DIR}/boot/gptboot \
694
		-p freebsd-ufs/${PRODUCT_NAME}:=${OVA_TMP}/${OVFUFS} \
695
		${OVA_SWAP_PART_PARAM} \
696
		-o ${OVA_TMP}/${OVFRAW} 2>&1 >> ${LOGFILE}
697

    
698
	if [ $? -ne 0 -o ! -f ${OVA_TMP}/${OVFRAW} ]; then
699
		if [ -f ${OVA_TMP}/${OVFUFS} ]; then
700
			rm -f ${OVA_TMP}/${OVFUFS}
701
		fi
702
		if [ -f ${OVA_TMP}/${OVFRAW} ]; then
703
			rm -f ${OVA_TMP}/${OVFRAW}
704
		fi
705
		echo "Failed!" | tee -a ${LOGFILE}
706
		echo ">>> ERROR: Error creating temporary vmdk image. STOPPING!" | tee -a ${LOGFILE}
707
		print_error_pfS
708
	fi
709
	echo "Done!" | tee -a ${LOGFILE}
710

    
711
	# We don't need it anymore
712
	rm -f ${OVA_TMP}/${OVFUFS} >/dev/null 2>&1
713

    
714
	# Convert raw to vmdk
715
	echo -n ">>> Creating vmdk disk... " | tee -a ${LOGFILE}
716
	vmdktool -z9 -v ${OVA_TMP}/${OVFVMDK} ${OVA_TMP}/${OVFRAW}
717

    
718
	if [ $? -ne 0 -o ! -f ${OVA_TMP}/${OVFVMDK} ]; then
719
		if [ -f ${OVA_TMP}/${OVFRAW} ]; then
720
			rm -f ${OVA_TMP}/${OVFRAW}
721
		fi
722
		if [ -f ${OVA_TMP}/${OVFVMDK} ]; then
723
			rm -f ${OVA_TMP}/${OVFVMDK}
724
		fi
725
		echo "Failed!" | tee -a ${LOGFILE}
726
		echo ">>> ERROR: Error creating vmdk image. STOPPING!" | tee -a ${LOGFILE}
727
		print_error_pfS
728
	fi
729
	echo "Done!" | tee -a ${LOGFILE}
730

    
731
	rm -f ${OVA_TMP}/${OVFRAW}
732

    
733
	ova_setup_ovf_template
734

    
735
	echo -n ">>> Writing final ova image... " | tee -a ${LOGFILE}
736
	# Create OVA file for vmware
737
	gtar -C ${OVA_TMP} -cpf ${OVAPATH} ${PRODUCT_NAME}.ovf ${OVFVMDK}
738
	echo "Done!" | tee -a ${LOGFILE}
739
	rm -f ${OVA_TMP}/${OVFVMDK} >/dev/null 2>&1
740

    
741
	echo ">>> OVA created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
742
}
743

    
744
# called from create_ova_image
745
ova_setup_ovf_template() {
746
	if [ ! -f ${OVFTEMPLATE} ]; then
747
		echo ">>> ERROR: OVF template file (${OVFTEMPLATE}) not found."
748
		print_error_pfS
749
	fi
750

    
751
	#  OperatingSystemSection (${PRODUCT_NAME}.ovf)
752
	#  42   FreeBSD 32-Bit
753
	#  78   FreeBSD 64-Bit
754
	if [ "${TARGET}" = "amd64" ]; then
755
		local _os_id="78"
756
		local _os_type="freebsd64Guest"
757
		local _os_descr="FreeBSD 64-Bit"
758
	else
759
		echo ">>> ERROR: Platform not supported for OVA (${TARGET})"
760
		print_error_pfS
761
	fi
762

    
763
	local POPULATED_SIZE=$(du -d0 -k $FINAL_CHROOT_DIR | cut -f1)
764
	local POPULATED_SIZE_IN_BYTES=$((${POPULATED_SIZE}*1024))
765
	local VMDK_FILE_SIZE=$(stat -f "%z" ${OVA_TMP}/${OVFVMDK})
766

    
767
	sed \
768
		-e "s,%%VMDK_FILE_SIZE%%,${VMDK_FILE_SIZE},g" \
769
		-e "s,%%VMDK_DISK_CAPACITY_IN_GB%%,${VMDK_DISK_CAPACITY_IN_GB},g" \
770
		-e "s,%%POPULATED_SIZE_IN_BYTES%%,${POPULATED_SIZE_IN_BYTES},g" \
771
		-e "s,%%OS_ID%%,${_os_id},g" \
772
		-e "s,%%OS_TYPE%%,${_os_type},g" \
773
		-e "s,%%OS_DESCR%%,${_os_descr},g" \
774
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
775
		-e "s,%%PRODUCT_NAME_SUFFIX%%,${PRODUCT_NAME_SUFFIX},g" \
776
		-e "s,%%PRODUCT_VERSION%%,${PRODUCT_VERSION},g" \
777
		-e "s,%%PRODUCT_URL%%,${PRODUCT_URL},g" \
778
		-e "s#%%VENDOR_NAME%%#${VENDOR_NAME}#g" \
779
		-e "s#%%OVF_INFO%%#${OVF_INFO}#g" \
780
		-e "/^%%PRODUCT_LICENSE%%/r ${BUILDER_ROOT}/LICENSE" \
781
		-e "/^%%PRODUCT_LICENSE%%/d" \
782
		${OVFTEMPLATE} > ${OVA_TMP}/${PRODUCT_NAME}.ovf
783
}
784

    
785
# Cleans up previous builds
786
clean_builder() {
787
	# Clean out directories
788
	echo ">>> Cleaning up previous build environment...Please wait!"
789

    
790
	staginareas_clean_each_run
791

    
792
	if [ -d "${STAGE_CHROOT_DIR}" ]; then
793
		echo -n ">>> Cleaning ${STAGE_CHROOT_DIR}... "
794
		chflags -R noschg ${STAGE_CHROOT_DIR} 2>&1 >/dev/null
795
		rm -rf ${STAGE_CHROOT_DIR}/* 2>/dev/null
796
		echo "Done."
797
	fi
798

    
799
	if [ -d "${INSTALLER_CHROOT_DIR}" ]; then
800
		echo -n ">>> Cleaning ${INSTALLER_CHROOT_DIR}... "
801
		chflags -R noschg ${INSTALLER_CHROOT_DIR} 2>&1 >/dev/null
802
		rm -rf ${INSTALLER_CHROOT_DIR}/* 2>/dev/null
803
		echo "Done."
804
	fi
805

    
806
	if [ -z "${NO_CLEAN_FREEBSD_OBJ}" -a -d "${FREEBSD_SRC_DIR}" ]; then
807
		OBJTREE=$(make -C ${FREEBSD_SRC_DIR} -V OBJTREE)
808
		if [ -d "${OBJTREE}" ]; then
809
			echo -n ">>> Cleaning FreeBSD objects dir staging..."
810
			echo -n "."
811
			chflags -R noschg ${OBJTREE} 2>&1 >/dev/null
812
			echo -n "."
813
			rm -rf ${OBJTREE}/*
814
			echo "Done!"
815
		fi
816
		if [ -d "${KERNEL_BUILD_PATH}" ]; then
817
			echo -n ">>> Cleaning previously built kernel stage area..."
818
			rm -rf $KERNEL_BUILD_PATH/*
819
			echo "Done!"
820
		fi
821
	fi
822
	mkdir -p $KERNEL_BUILD_PATH
823

    
824
	echo -n ">>> Cleaning previously built images..."
825
	rm -rf $IMAGES_FINAL_DIR/*
826
	echo "Done!"
827

    
828
	echo -n ">>> Cleaning previous builder logs..."
829
	if [ -d "$BUILDER_LOGS" ]; then
830
		rm -rf ${BUILDER_LOGS}
831
	fi
832
	mkdir -p ${BUILDER_LOGS}
833

    
834
	echo "Done!"
835

    
836
	echo ">>> Cleaning of builder environment has finished."
837
}
838

    
839
clone_directory_contents() {
840
	if [ ! -e "$2" ]; then
841
		mkdir -p "$2"
842
	fi
843
	if [ ! -d "$1" -o ! -d "$2" ]; then
844
		if [ -z "${LOGFILE}" ]; then
845
			echo ">>> ERROR: Argument $1 supplied is not a directory!"
846
		else
847
			echo ">>> ERROR: Argument $1 supplied is not a directory!" | tee -a ${LOGFILE}
848
		fi
849
		print_error_pfS
850
	fi
851
	echo -n ">>> Using TAR to clone $1 to $2 ..."
852
	tar -C ${1} -c -f - . | tar -C ${2} -x -p -f -
853
	echo "Done!"
854
}
855

    
856
clone_to_staging_area() {
857
	# Clone everything to the final staging area
858
	echo -n ">>> Cloning everything to ${STAGE_CHROOT_DIR} staging area..."
859
	LOGFILE=${BUILDER_LOGS}/cloning.${TARGET}.log
860

    
861
	tar -C ${PRODUCT_SRC} -c -f - . | \
862
		tar -C ${STAGE_CHROOT_DIR} -x -p -f -
863

    
864
	if [ "${PRODUCT_NAME}" != "pfSense" ]; then
865
		mv ${STAGE_CHROOT_DIR}/usr/local/sbin/pfSense-upgrade \
866
			${STAGE_CHROOT_DIR}/usr/local/sbin/${PRODUCT_NAME}-upgrade
867
	fi
868

    
869
	mkdir -p ${STAGE_CHROOT_DIR}/etc/mtree
870
	mtree -Pcp ${STAGE_CHROOT_DIR}/var > ${STAGE_CHROOT_DIR}/etc/mtree/var.dist
871
	mtree -Pcp ${STAGE_CHROOT_DIR}/etc > ${STAGE_CHROOT_DIR}/etc/mtree/etc.dist
872
	if [ -d ${STAGE_CHROOT_DIR}/usr/local/etc ]; then
873
		mtree -Pcp ${STAGE_CHROOT_DIR}/usr/local/etc > ${STAGE_CHROOT_DIR}/etc/mtree/localetc.dist
874
	fi
875

    
876
	## Add buildtime and lastcommit information
877
	# This is used for detecting updates.
878
	echo "$BUILTDATESTRING" > $STAGE_CHROOT_DIR/etc/version.buildtime
879
	# Record last commit info if it is available.
880
	if [ -f $SCRATCHDIR/build_commit_info.txt ]; then
881
		cp $SCRATCHDIR/build_commit_info.txt $STAGE_CHROOT_DIR/etc/version.lastcommit
882
	fi
883

    
884
	local _exclude_files="${SCRATCHDIR}/base_exclude_files"
885
	sed \
886
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
887
		-e "s,%%VERSION%%,${_version},g" \
888
		${BUILDER_TOOLS}/templates/core_pkg/base/exclude_files \
889
		> ${_exclude_files}
890

    
891
	mkdir -p ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR} >/dev/null 2>&1
892

    
893
	# Include a sample pkg stable conf to base
894
	setup_pkg_repo \
895
		${PKG_REPO_DEFAULT} \
896
		${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/${PRODUCT_NAME}-repo.conf \
897
		${TARGET} \
898
		${TARGET_ARCH}
899

    
900
	mtree \
901
		-c \
902
		-k uid,gid,mode,size,flags,sha256digest \
903
		-p ${STAGE_CHROOT_DIR} \
904
		-X ${_exclude_files} \
905
		> ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/base.mtree
906
	tar \
907
		-C ${STAGE_CHROOT_DIR} \
908
		-cJf ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/base.txz \
909
		-X ${_exclude_files} \
910
		.
911

    
912
	local _share_repos_path="${SCRATCHDIR}/repo-tmp/${PRODUCT_SHARE_DIR}/pkg/repos"
913
	rm -rf ${SCRATCHDIR}/repo-tmp >/dev/null 2>&1
914
	mkdir -p ${_share_repos_path} >/dev/null 2>&1
915

    
916
	setup_pkg_repo \
917
		${PKG_REPO_DEFAULT} \
918
		${_share_repos_path}/${PRODUCT_NAME}-repo.conf \
919
		${TARGET} \
920
		${TARGET_ARCH}
921

    
922
	cp -f ${PKG_REPO_DEFAULT%%.conf}.descr ${_share_repos_path}
923

    
924
	# Add additional repos
925
	for _template in ${PKG_REPO_BASE}/${PRODUCT_NAME}-repo-*.conf; do
926
		_template_filename=$(basename ${_template})
927
		setup_pkg_repo \
928
			${_template} \
929
			${_share_repos_path}/${_template_filename} \
930
			${TARGET} \
931
			${TARGET_ARCH}
932
		cp -f ${_template%%.conf}.descr ${_share_repos_path}
933
	done
934

    
935
	core_pkg_create repo "" ${CORE_PKG_VERSION} ${SCRATCHDIR}/repo-tmp
936

    
937
	core_pkg_create rc "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
938
	core_pkg_create base "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
939
	core_pkg_create base-nanobsd "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
940
	core_pkg_create default-config "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
941

    
942
	local DEFAULTCONF=${STAGE_CHROOT_DIR}/conf.default/config.xml
943

    
944
	# Save current WAN and LAN if value
945
	local _old_wan_if=$(xml sel -t -v "${XML_ROOTOBJ}/interfaces/wan/if" ${DEFAULTCONF})
946
	local _old_lan_if=$(xml sel -t -v "${XML_ROOTOBJ}/interfaces/lan/if" ${DEFAULTCONF})
947

    
948
	# Change default interface names to match vmware driver
949
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/wan/if" -v "vmx0" ${DEFAULTCONF}
950
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/lan/if" -v "vmx1" ${DEFAULTCONF}
951
	core_pkg_create default-config "vmware" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
952

    
953
	# Restore default values to be used by serial package
954
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/wan/if" -v "${_old_wan_if}" ${DEFAULTCONF}
955
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/lan/if" -v "${_old_lan_if}" ${DEFAULTCONF}
956

    
957
	# Activate serial console in config.xml
958
	xml ed -L -P -d "${XML_ROOTOBJ}/system/enableserial" ${DEFAULTCONF}
959
	xml ed -P -s "${XML_ROOTOBJ}/system" -t elem -n "enableserial" \
960
		${DEFAULTCONF} > ${DEFAULTCONF}.tmp
961
	xml fo -t ${DEFAULTCONF}.tmp > ${DEFAULTCONF}
962
	rm -f ${DEFAULTCONF}.tmp
963

    
964
	echo force > ${STAGE_CHROOT_DIR}/cf/conf/enableserial_force
965

    
966
	core_pkg_create default-config-serial "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
967

    
968
	rm -f ${STAGE_CHROOT_DIR}/cf/conf/enableserial_force
969
	rm -f ${STAGE_CHROOT_DIR}/cf/conf/config.xml
970

    
971
	# Make sure pkg is present
972
	pkg_bootstrap ${STAGE_CHROOT_DIR}
973

    
974
	echo "Done!"
975
}
976

    
977
create_final_staging_area() {
978
	if [ -z "${FINAL_CHROOT_DIR}" ]; then
979
		echo ">>> ERROR: FINAL_CHROOT_DIR is not set, cannot continue!" | tee -a ${LOGFILE}
980
		print_error_pfS
981
	fi
982

    
983
	if [ -d "${FINAL_CHROOT_DIR}" ]; then
984
		echo -n ">>> Previous ${FINAL_CHROOT_DIR} detected cleaning up..." | tee -a ${LOGFILE}
985
		chflags -R noschg ${FINAL_CHROOT_DIR} 2>&1 1>/dev/null
986
		rm -rf ${FINAL_CHROOT_DIR}/* 2>&1 1>/dev/null
987
		echo "Done." | tee -a ${LOGFILE}
988
	fi
989

    
990
	echo ">>> Preparing Final image staging area: $(LC_ALL=C date)" 2>&1 | tee -a ${LOGFILE}
991
	echo ">>> Cloning ${STAGE_CHROOT_DIR} to ${FINAL_CHROOT_DIR}" 2>&1 | tee -a ${LOGFILE}
992
	clone_directory_contents ${STAGE_CHROOT_DIR} ${FINAL_CHROOT_DIR}
993

    
994
	if [ ! -f $FINAL_CHROOT_DIR/sbin/init ]; then
995
		echo ">>> ERROR: Something went wrong during cloning -- Please verify!" 2>&1 | tee -a ${LOGFILE}
996
		print_error_pfS
997
	fi
998
}
999

    
1000
customize_stagearea_for_image() {
1001
	local _image_type="$1"
1002
	local _default_config="" # filled with $2 below
1003
	local _image_variant="$3"
1004

    
1005
	if [ -n "$2" ]; then
1006
		_default_config="$2"
1007
	elif [ "${_image_type}" = "nanobsd" -o \
1008
	     "${_image_type}" = "memstickserial" -o \
1009
	     "${_image_type}" = "memstickadi" ]; then
1010
		_default_config="default-config-serial"
1011
	elif [ "${_image_type}" = "ova" ]; then
1012
		_default_config="default-config-vmware"
1013
	else
1014
		_default_config="default-config"
1015
	fi
1016

    
1017
	# Prepare final stage area
1018
	create_final_staging_area
1019

    
1020
	pkg_chroot_add ${FINAL_CHROOT_DIR} rc
1021
	pkg_chroot_add ${FINAL_CHROOT_DIR} repo
1022

    
1023
	if [ "${_image_type}" = "nanobsd" -o \
1024
	     "${_image_type}" = "nanobsd-vga" ]; then
1025

    
1026
		mkdir -p ${FINAL_CHROOT_DIR}/root/var/db \
1027
			 ${FINAL_CHROOT_DIR}/root/var/cache \
1028
			 ${FINAL_CHROOT_DIR}/var/db/pkg \
1029
			 ${FINAL_CHROOT_DIR}/var/cache/pkg
1030
		mv -f ${FINAL_CHROOT_DIR}/var/db/pkg ${FINAL_CHROOT_DIR}/root/var/db
1031
		mv -f ${FINAL_CHROOT_DIR}/var/cache/pkg ${FINAL_CHROOT_DIR}/root/var/cache
1032
		ln -sf ../../root/var/db/pkg ${FINAL_CHROOT_DIR}/var/db/pkg
1033
		ln -sf ../../root/var/cache/pkg ${FINAL_CHROOT_DIR}/var/cache/pkg
1034

    
1035
		pkg_chroot_add ${FINAL_CHROOT_DIR} base-nanobsd
1036
	else
1037
		pkg_chroot_add ${FINAL_CHROOT_DIR} base
1038
	fi
1039

    
1040
	if [ "${_image_type}" = "iso" -o \
1041
	     "${_image_type}" = "memstick" -o \
1042
	     "${_image_type}" = "memstickserial" -o \
1043
	     "${_image_type}" = "memstickadi" ]; then
1044
		mkdir -p ${FINAL_CHROOT_DIR}/pkgs
1045
		cp ${CORE_PKG_ALL_PATH}/*default-config*.txz ${FINAL_CHROOT_DIR}/pkgs
1046
	fi
1047

    
1048
	pkg_chroot_add ${FINAL_CHROOT_DIR} ${_default_config}
1049

    
1050
	# XXX: Workaround to avoid pkg to complain regarding release
1051
	#      repo on first boot since packages are installed from
1052
	#      staging server during build phase
1053
	if [ -n "${USE_PKG_REPO_STAGING}" ]; then
1054
		_read_cmd="select value from repodata where key='packagesite'"
1055
		if [ -n "${_IS_RELEASE}" ]; then
1056
			local _tgt_server="${PKG_REPO_SERVER_RELEASE}"
1057
		else
1058
			local _tgt_server="${PKG_REPO_SERVER_DEVEL}"
1059
		fi
1060
		for _db in ${FINAL_CHROOT_DIR}/var/db/pkg/repo-*sqlite; do
1061
			_cur=$(/usr/local/bin/sqlite3 ${_db} "${_read_cmd}")
1062
			_new=$(echo "${_cur}" | sed -e "s,^${PKG_REPO_SERVER_STAGING},${_tgt_server},")
1063
			/usr/local/bin/sqlite3 ${_db} "update repodata set value='${_new}' where key='packagesite'"
1064
		done
1065
	fi
1066

    
1067
	if [ -n "$_image_variant" -a \
1068
	    -d ${BUILDER_TOOLS}/templates/custom_logos/${_image_variant} ]; then
1069
		mkdir -p ${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/custom_logos
1070
		cp -f \
1071
			${BUILDER_TOOLS}/templates/custom_logos/${_image_variant}/*.png \
1072
			${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/custom_logos
1073
	fi
1074
}
1075

    
1076
create_distribution_tarball() {
1077
	mkdir -p ${INSTALLER_CHROOT_DIR}/usr/freebsd-dist
1078

    
1079
	tar -C ${FINAL_CHROOT_DIR} --exclude ./install --exclude ./pkgs \
1080
		-cJf ${INSTALLER_CHROOT_DIR}/usr/freebsd-dist/base.txz .
1081

    
1082
	(cd ${INSTALLER_CHROOT_DIR}/usr/freebsd-dist && \
1083
		sh ${FREEBSD_SRC_DIR}/release/scripts/make-manifest.sh base.txz) \
1084
		> ${INSTALLER_CHROOT_DIR}/usr/freebsd-dist/MANIFEST
1085
}
1086

    
1087
create_iso_image() {
1088
	LOGFILE=${BUILDER_LOGS}/isoimage.${TARGET}
1089

    
1090
	if [ -z "${ISOPATH}" ]; then
1091
		echo ">>> ISOPATH is empty skipping generation of ISO image!" | tee -a ${LOGFILE}
1092
		return
1093
	fi
1094

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

    
1097
	mkdir -p $(dirname ${ISOPATH})
1098

    
1099
	customize_stagearea_for_image "iso"
1100
	install_default_kernel ${DEFAULT_KERNEL}
1101

    
1102
	BOOTCONF=${INSTALLER_CHROOT_DIR}/boot.config
1103
	LOADERCONF=${INSTALLER_CHROOT_DIR}/boot/loader.conf
1104

    
1105
	rm -f ${LOADERCONF} ${BOOTCONF} >/dev/null 2>&1
1106

    
1107
	touch ${FINAL_CHROOT_DIR}/boot/loader.conf
1108

    
1109
	create_distribution_tarball
1110

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

    
1113
	sh ${FREEBSD_SRC_DIR}/release/${TARGET}/mkisoimages.sh -b \
1114
		${FSLABEL} \
1115
		${ISOPATH} \
1116
		${INSTALLER_CHROOT_DIR}
1117

    
1118
	if [ ! -f "${ISOPATH}" ]; then
1119
		echo "ERROR! ISO image was not built"
1120
		print_error_pfS
1121
	fi
1122

    
1123
	gzip -qf $ISOPATH &
1124
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1125

    
1126
	echo ">>> ISO created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1127
}
1128

    
1129
create_memstick_image() {
1130
	local _variant="$1"
1131

    
1132
	LOGFILE=${BUILDER_LOGS}/memstick.${TARGET}
1133
	if [ "${MEMSTICKPATH}" = "" ]; then
1134
		echo ">>> MEMSTICKPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1135
		return
1136
	fi
1137

    
1138
	mkdir -p $(dirname ${MEMSTICKPATH})
1139

    
1140
	local _image_path=${MEMSTICKPATH}
1141
	if [ -n "${_variant}" ]; then
1142
		_image_path=$(echo "$_image_path" | \
1143
			sed "s/-memstick-/-memstick-${_variant}-/")
1144
		VARIANTIMAGES="${VARIANTIMAGES}${VARIANTIMAGES:+ }${_image_path}"
1145
	fi
1146

    
1147
	customize_stagearea_for_image "memstick" "" $_variant
1148
	install_default_kernel ${DEFAULT_KERNEL}
1149

    
1150
	echo ">>> Creating memstick to ${_image_path}." 2>&1 | tee -a ${LOGFILE}
1151
	echo "kern.cam.boot_delay=10000" >> ${FINAL_CHROOT_DIR}/boot/loader.conf.local
1152

    
1153
	BOOTCONF=${INSTALLER_CHROOT_DIR}/boot.config
1154
	LOADERCONF=${INSTALLER_CHROOT_DIR}/boot/loader.conf
1155

    
1156
	rm -f ${LOADERCONF} ${BOOTCONF} >/dev/null 2>&1
1157

    
1158
	touch ${FINAL_CHROOT_DIR}/boot/loader.conf
1159

    
1160
	create_distribution_tarball
1161

    
1162
	sh ${FREEBSD_SRC_DIR}/release/${TARGET}/make-memstick.sh \
1163
		${INSTALLER_CHROOT_DIR} \
1164
		${_image_path}
1165

    
1166
	if [ ! -f "${_image_path}" ]; then
1167
		echo "ERROR! memstick image was not built"
1168
		print_error_pfS
1169
	fi
1170

    
1171
	gzip -qf $_image_path &
1172
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1173

    
1174
	echo ">>> MEMSTICK created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1175
}
1176

    
1177
create_memstick_serial_image() {
1178
	LOGFILE=${BUILDER_LOGS}/memstickserial.${TARGET}
1179
	if [ "${MEMSTICKSERIALPATH}" = "" ]; then
1180
		echo ">>> MEMSTICKSERIALPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1181
		return
1182
	fi
1183

    
1184
	mkdir -p $(dirname ${MEMSTICKSERIALPATH})
1185

    
1186
	customize_stagearea_for_image "memstickserial"
1187
	install_default_kernel ${DEFAULT_KERNEL}
1188

    
1189
	echo ">>> Creating serial memstick to ${MEMSTICKSERIALPATH}." 2>&1 | tee -a ${LOGFILE}
1190
	echo "kern.cam.boot_delay=10000" >> ${FINAL_CHROOT_DIR}/boot/loader.conf.local
1191

    
1192
	BOOTCONF=${INSTALLER_CHROOT_DIR}/boot.config
1193
	LOADERCONF=${INSTALLER_CHROOT_DIR}/boot/loader.conf
1194

    
1195
	echo ">>> Activating serial console..." 2>&1 | tee -a ${LOGFILE}
1196
	echo "-S115200 -D" > ${BOOTCONF}
1197

    
1198
	# Activate serial console+video console in loader.conf
1199
	echo 'boot_multicons="YES"' >  ${LOADERCONF}
1200
	echo 'boot_serial="YES"' >> ${LOADERCONF}
1201
	echo 'console="comconsole,vidconsole"' >> ${LOADERCONF}
1202
	echo 'comconsole_speed="115200"' >> ${LOADERCONF}
1203

    
1204
	cat ${BOOTCONF} >> ${FINAL_CHROOT_DIR}/boot.config
1205
	cat ${LOADERCONF} >> ${FINAL_CHROOT_DIR}/boot/loader.conf
1206

    
1207
	create_distribution_tarball
1208

    
1209
	sh ${FREEBSD_SRC_DIR}/release/${TARGET}/make-memstick.sh \
1210
		${INSTALLER_CHROOT_DIR} \
1211
		${MEMSTICKSERIALPATH}
1212

    
1213
	if [ ! -f "${MEMSTICKSERIALPATH}" ]; then
1214
		echo "ERROR! memstick serial image was not built"
1215
		print_error_pfS
1216
	fi
1217

    
1218
	gzip -qf $MEMSTICKSERIALPATH &
1219
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1220

    
1221
	echo ">>> MEMSTICKSERIAL created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1222
}
1223

    
1224
create_memstick_adi_image() {
1225
	LOGFILE=${BUILDER_LOGS}/memstickadi.${TARGET}
1226
	if [ "${MEMSTICKADIPATH}" = "" ]; then
1227
		echo ">>> MEMSTICKADIPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1228
		return
1229
	fi
1230

    
1231
	mkdir -p $(dirname ${MEMSTICKADIPATH})
1232

    
1233
	customize_stagearea_for_image "memstickadi"
1234
	install_default_kernel ${DEFAULT_KERNEL}
1235

    
1236
	echo ">>> Creating serial memstick to ${MEMSTICKADIPATH}." 2>&1 | tee -a ${LOGFILE}
1237
	echo "kern.cam.boot_delay=10000" >> ${FINAL_CHROOT_DIR}/boot/loader.conf.local
1238

    
1239
	BOOTCONF=${INSTALLER_CHROOT_DIR}/boot.config
1240
	LOADERCONF=${INSTALLER_CHROOT_DIR}/boot/loader.conf
1241

    
1242
	echo ">>> Activating serial console..." 2>&1 | tee -a ${LOGFILE}
1243
	echo "-S115200 -h" > ${BOOTCONF}
1244

    
1245
	# Activate serial console+video console in loader.conf
1246
	echo 'boot_serial="YES"' > ${LOADERCONF}
1247
	echo 'console="comconsole"' >> ${LOADERCONF}
1248
	echo 'comconsole_speed="115200"' >> ${LOADERCONF}
1249
	echo 'comconsole_port="0x2F8"' >> ${LOADERCONF}
1250
	echo 'hint.uart.0.flags="0x00"' >> ${LOADERCONF}
1251
	echo 'hint.uart.1.flags="0x10"' >> ${LOADERCONF}
1252

    
1253
	cat ${BOOTCONF} >> ${FINAL_CHROOT_DIR}/boot.config
1254
	cat ${LOADERCONF} >> ${FINAL_CHROOT_DIR}/boot/loader.conf
1255

    
1256
	create_distribution_tarball
1257

    
1258
	sh ${FREEBSD_SRC_DIR}/release/${TARGET}/make-memstick.sh \
1259
		${INSTALLER_CHROOT_DIR} \
1260
		${MEMSTICKADIPATH}
1261

    
1262
	if [ ! -f "${MEMSTICKADIPATH}" ]; then
1263
		echo "ERROR! memstick ADI image was not built"
1264
		print_error_pfS
1265
	fi
1266

    
1267
	gzip -qf $MEMSTICKADIPATH &
1268
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1269

    
1270
	echo ">>> MEMSTICKADI created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1271
}
1272

    
1273
# Create pkg conf on desired place with desired arch/branch
1274
setup_pkg_repo() {
1275
	if [ -z "${4}" ]; then
1276
		return
1277
	fi
1278

    
1279
	local _template="${1}"
1280
	local _target="${2}"
1281
	local _arch="${3}"
1282
	local _target_arch="${4}"
1283
	local _staging="${5}"
1284

    
1285
	if [ -z "${_template}" -o ! -f "${_template}" ]; then
1286
		echo ">>> ERROR: It was not possible to find pkg conf template ${_template}"
1287
		print_error_pfS
1288
	fi
1289

    
1290
	if [ -n "${_staging}" -a -n "${USE_PKG_REPO_STAGING}" ]; then
1291
		local _pkg_repo_server_devel=${PKG_REPO_SERVER_STAGING}
1292
		local _pkg_repo_branch_devel=${PKG_REPO_BRANCH_STAGING}
1293
		local _pkg_repo_server_release=${PKG_REPO_SERVER_STAGING}
1294
		local _pkg_repo_branch_release=${PKG_REPO_BRANCH_STAGING}
1295
	else
1296
		local _pkg_repo_server_devel=${PKG_REPO_SERVER_DEVEL}
1297
		local _pkg_repo_branch_devel=${PKG_REPO_BRANCH_DEVEL}
1298
		local _pkg_repo_server_release=${PKG_REPO_SERVER_RELEASE}
1299
		local _pkg_repo_branch_release=${PKG_REPO_BRANCH_RELEASE}
1300
	fi
1301

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

    
1304
	sed \
1305
		-e "s/%%ARCH%%/${_target_arch}/" \
1306
		-e "s/%%PKG_REPO_BRANCH_DEVEL%%/${_pkg_repo_branch_devel}/g" \
1307
		-e "s/%%PKG_REPO_BRANCH_RELEASE%%/${_pkg_repo_branch_release}/g" \
1308
		-e "s,%%PKG_REPO_SERVER_DEVEL%%,${_pkg_repo_server_devel},g" \
1309
		-e "s,%%PKG_REPO_SERVER_RELEASE%%,${_pkg_repo_server_release},g" \
1310
		-e "s,%%POUDRIERE_PORTS_NAME%%,${POUDRIERE_PORTS_NAME},g" \
1311
		-e "s/%%PRODUCT_NAME%%/${PRODUCT_NAME}/g" \
1312
		${_template} \
1313
		> ${_target}
1314
}
1315

    
1316
# This routine ensures any ports / binaries that the builder
1317
# system needs are on disk and ready for execution.
1318
builder_setup() {
1319
	# If Product-builder is already installed, just leave
1320
	if pkg info -e -q ${PRODUCT_NAME}-builder; then
1321
		return
1322
	fi
1323

    
1324
	if [ ! -f ${PKG_REPO_PATH} ]; then
1325
		[ -d $(dirname ${PKG_REPO_PATH}) ] \
1326
			|| mkdir -p $(dirname ${PKG_REPO_PATH})
1327

    
1328
		update_freebsd_sources
1329

    
1330
		local _arch=$(uname -m)
1331
		setup_pkg_repo \
1332
			${PKG_REPO_DEFAULT} \
1333
			${PKG_REPO_PATH} \
1334
			${_arch} \
1335
			${_arch} \
1336
			"staging"
1337

    
1338
		# Use fingerprint keys from repo
1339
		sed -i '' -e "/fingerprints:/ s,\"/,\"${BUILDER_ROOT}/src/," \
1340
			${PKG_REPO_PATH}
1341
	fi
1342

    
1343
	pkg install ${PRODUCT_NAME}-builder
1344
}
1345

    
1346
# Updates FreeBSD sources
1347
update_freebsd_sources() {
1348
	if [ "${1}" = "full" ]; then
1349
		local _full=1
1350
		local _clone_params=""
1351
	else
1352
		local _full=0
1353
		local _clone_params="--depth 1 --single-branch"
1354
	fi
1355

    
1356
	if [ -n "${NO_BUILDWORLD}" -a -n "${NO_BUILDKERNEL}" ]; then
1357
		echo ">>> NO_BUILDWORLD and NO_BUILDKERNEL set, skipping update of freebsd sources" | tee -a ${LOGFILE}
1358
		return
1359
	fi
1360

    
1361
	echo ">>> Obtaining FreeBSD sources (${FREEBSD_BRANCH})..."
1362
	${BUILDER_SCRIPTS}/git_checkout.sh \
1363
		-r ${FREEBSD_REPO_BASE} \
1364
		-d ${FREEBSD_SRC_DIR} \
1365
		-b ${FREEBSD_BRANCH}
1366

    
1367
	if [ $? -ne 0 -o ! -d "${FREEBSD_SRC_DIR}/.git" ]; then
1368
		echo ">>> ERROR: It was not possible to clone FreeBSD src repo"
1369
		print_error_pfS
1370
	fi
1371

    
1372
	if [ -n "${GIT_FREEBSD_COSHA1}" ]; then
1373
		echo -n ">>> Checking out desired commit (${GIT_FREEBSD_COSHA1})... "
1374
		( git -C  ${FREEBSD_SRC_DIR} checkout ${GIT_FREEBSD_COSHA1} ) 2>&1 | \
1375
			grep -C3 -i -E 'error|fatal'
1376
		echo "Done!"
1377
	fi
1378
}
1379

    
1380
pkg_chroot() {
1381
	local _root="${1}"
1382
	shift
1383

    
1384
	if [ $# -eq 0 ]; then
1385
		return -1
1386
	fi
1387

    
1388
	if [ -z "${_root}" -o "${_root}" = "/" -o ! -d "${_root}" ]; then
1389
		return -1
1390
	fi
1391

    
1392
	mkdir -p \
1393
		${SCRATCHDIR}/pkg_cache \
1394
		${_root}/var/cache/pkg \
1395
		${_root}/dev
1396

    
1397
	/sbin/mount -t nullfs ${SCRATCHDIR}/pkg_cache ${_root}/var/cache/pkg
1398
	/sbin/mount -t devfs devfs ${_root}/dev
1399
	cp -f /etc/resolv.conf ${_root}/etc/resolv.conf
1400
	touch ${BUILDER_LOGS}/install_pkg_install_ports.txt
1401
	script -aq ${BUILDER_LOGS}/install_pkg_install_ports.txt pkg -c ${_root} $@ >/dev/null 2>&1
1402
	local result=$?
1403
	rm -f ${_root}/etc/resolv.conf
1404
	/sbin/umount -f ${_root}/dev
1405
	/sbin/umount -f ${_root}/var/cache/pkg
1406

    
1407
	return $result
1408
}
1409

    
1410

    
1411
pkg_chroot_add() {
1412
	if [ -z "${1}" -o -z "${2}" ]; then
1413
		return 1
1414
	fi
1415

    
1416
	local _target="${1}"
1417
	local _pkg="$(get_pkg_name ${2}).txz"
1418

    
1419
	if [ ! -d "${_target}" ]; then
1420
		echo ">>> ERROR: Target dir ${_target} not found"
1421
		print_error_pfS
1422
	fi
1423

    
1424
	if [ ! -f ${CORE_PKG_ALL_PATH}/${_pkg} ]; then
1425
		echo ">>> ERROR: Package ${_pkg} not found"
1426
		print_error_pfS
1427
	fi
1428

    
1429
	cp ${CORE_PKG_ALL_PATH}/${_pkg} ${_target}
1430
	pkg_chroot ${_target} add /${_pkg}
1431
	rm -f ${_target}/${_pkg}
1432
}
1433

    
1434
pkg_bootstrap() {
1435
	local _root=${1:-"${STAGE_CHROOT_DIR}"}
1436

    
1437
	setup_pkg_repo \
1438
		${PKG_REPO_DEFAULT} \
1439
		${_root}${PKG_REPO_PATH} \
1440
		${TARGET} \
1441
		${TARGET_ARCH} \
1442
		"staging"
1443

    
1444
	pkg_chroot ${_root} bootstrap -f
1445
}
1446

    
1447
# This routine assists with installing various
1448
# freebsd ports files into the pfsense-fs staging
1449
# area.
1450
install_pkg_install_ports() {
1451
	local MAIN_PKG="${1}"
1452

    
1453
	if [ -z "${MAIN_PKG}" ]; then
1454
		MAIN_PKG=${PRODUCT_NAME}
1455
	fi
1456

    
1457
	echo ">>> Installing pkg repository in chroot (${STAGE_CHROOT_DIR})..."
1458

    
1459
	[ -d ${STAGE_CHROOT_DIR}/var/cache/pkg ] || \
1460
		mkdir -p ${STAGE_CHROOT_DIR}/var/cache/pkg
1461

    
1462
	[ -d ${SCRATCHDIR}/pkg_cache ] || \
1463
		mkdir -p ${SCRATCHDIR}/pkg_cache
1464

    
1465
	echo -n ">>> Installing built ports (packages) in chroot (${STAGE_CHROOT_DIR})... "
1466
	# First mark all packages as automatically installed
1467
	pkg_chroot ${STAGE_CHROOT_DIR} set -A 1 -a
1468
	# Install all necessary packages
1469
	if ! pkg_chroot ${STAGE_CHROOT_DIR} install ${MAIN_PKG} ${custom_package_list}; then
1470
		echo "Failed!"
1471
		print_error_pfS
1472
	fi
1473
	# Make sure required packages are set as non-automatic
1474
	pkg_chroot ${STAGE_CHROOT_DIR} set -A 0 pkg ${MAIN_PKG} ${custom_package_list}
1475
	# Remove unnecessary packages
1476
	pkg_chroot ${STAGE_CHROOT_DIR} autoremove
1477
	echo "Done!"
1478
}
1479

    
1480
staginareas_clean_each_run() {
1481
	echo -n ">>> Cleaning build directories: "
1482
	if [ -d "${FINAL_CHROOT_DIR}" ]; then
1483
		BASENAME=$(basename ${FINAL_CHROOT_DIR})
1484
		echo -n "$BASENAME "
1485
		chflags -R noschg ${FINAL_CHROOT_DIR} 2>&1 >/dev/null
1486
		rm -rf ${FINAL_CHROOT_DIR}/* 2>/dev/null
1487
	fi
1488
	echo "Done!"
1489
}
1490

    
1491
# Imported from FreeSBIE
1492
buildkernel() {
1493
	local _kernconf=${1:-${KERNCONF}}
1494

    
1495
	if [ -n "${NO_BUILDKERNEL}" ]; then
1496
		echo ">>> NO_BUILDKERNEL set, skipping build" | tee -a ${LOGFILE}
1497
		return
1498
	fi
1499

    
1500
	if [ -z "${_kernconf}" ]; then
1501
		echo ">>> ERROR: No kernel configuration defined probably this is not what you want! STOPPING!" | tee -a ${LOGFILE}
1502
		print_error_pfS
1503
	fi
1504

    
1505
	local _old_kernconf=${KERNCONF}
1506
	export KERNCONF=${_kernconf}
1507

    
1508
	echo ">>> $(LC_ALL=C date) - Starting build kernel for ${TARGET} architecture..." | tee -a ${LOGFILE}
1509
	script -aq $LOGFILE ${BUILDER_SCRIPTS}/build_freebsd.sh -W -s ${FREEBSD_SRC_DIR} \
1510
		|| print_error_pfS
1511
	echo ">>> $(LC_ALL=C date) - Finished build kernel for ${TARGET} architecture..." | tee -a ${LOGFILE}
1512

    
1513
	export KERNCONF=${_old_kernconf}
1514
}
1515

    
1516
# Imported from FreeSBIE
1517
installkernel() {
1518
	local _destdir=${1:-${KERNEL_DESTDIR}}
1519
	local _kernconf=${2:-${KERNCONF}}
1520

    
1521
	if [ -z "${_kernconf}" ]; then
1522
		echo ">>> ERROR: No kernel configuration defined probably this is not what you want! STOPPING!" | tee -a ${LOGFILE}
1523
		print_error_pfS
1524
	fi
1525

    
1526
	local _old_kernconf=${KERNCONF}
1527
	export KERNCONF=${_kernconf}
1528

    
1529
	mkdir -p ${STAGE_CHROOT_DIR}/boot
1530
	echo ">>> Installing kernel (${_kernconf}) for ${TARGET} architecture..." | tee -a ${LOGFILE}
1531
	script -aq $LOGFILE ${BUILDER_SCRIPTS}/install_freebsd.sh -W -D -z \
1532
		-s ${FREEBSD_SRC_DIR} \
1533
		-d ${_destdir} \
1534
		|| print_error_pfS
1535

    
1536
	export KERNCONF=${_old_kernconf}
1537
}
1538

    
1539
# Launch is ran first to setup a few variables that we need
1540
# Imported from FreeSBIE
1541
launch() {
1542
	if [ "$(id -u)" != "0" ]; then
1543
		echo "Sorry, this must be done as root."
1544
	fi
1545

    
1546
	echo ">>> Operation $0 has started at $(date)"
1547
}
1548

    
1549
finish() {
1550
	echo ">>> Operation $0 has ended at $(date)"
1551
}
1552

    
1553
pkg_repo_rsync() {
1554
	local _repo_path_param="${1}"
1555
	local _ignore_final_rsync="${2}"
1556

    
1557
	if [ -z "${_repo_path_param}" -o ! -d "${_repo_path_param}" ]; then
1558
		return
1559
	fi
1560

    
1561
	if [ -n "${SKIP_FINAL_RSYNC}" ]; then
1562
		_ignore_final_rsync="1"
1563
	fi
1564

    
1565
	# Sanitize path
1566
	_repo_path=$(realpath ${_repo_path_param})
1567

    
1568
	local _repo_dir=$(dirname ${_repo_path})
1569
	local _repo_base=$(basename ${_repo_path})
1570

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

    
1574
	if [ -z "${LOGFILE}" ]; then
1575
		local _logfile="/dev/null"
1576
	else
1577
		local _logfile="${LOGFILE}"
1578
	fi
1579

    
1580
	if [ -n "${PKG_REPO_SIGNING_COMMAND}" ]; then
1581

    
1582
		# Detect poudriere directory structure
1583
		if [ -L "${_repo_path}/.latest" ]; then
1584
			local _real_repo_path=$(readlink -f ${_repo_path}/.latest)
1585
		else
1586
			local _real_repo_path=${_repo_path}
1587
		fi
1588

    
1589
		echo -n ">>> Signing repository... " | tee -a ${_logfile}
1590
		############ ATTENTION ##############
1591
		#
1592
		# For some reason pkg-repo fail without / in the end of directory name
1593
		# so removing it will break command
1594
		#
1595
		# https://github.com/freebsd/pkg/issues/1364
1596
		#
1597
		if script -aq ${_logfile} pkg repo ${_real_repo_path}/ \
1598
		    signing_command: ${PKG_REPO_SIGNING_COMMAND} >/dev/null 2>&1; then
1599
			echo "Done!" | tee -a ${_logfile}
1600
		else
1601
			echo "Failed!" | tee -a ${_logfile}
1602
			echo ">>> ERROR: An error occurred trying to sign repo"
1603
			print_error_pfS
1604
		fi
1605

    
1606
		local _pkgfile="${_repo_path}/Latest/pkg.txz"
1607
		if [ -e ${_pkgfile} ]; then
1608
			echo -n ">>> Signing Latest/pkg.txz for bootstraping... " | tee -a ${_logfile}
1609

    
1610
			if sha256 -q ${_pkgfile} | ${PKG_REPO_SIGNING_COMMAND} \
1611
			    > ${_pkgfile}.sig 2>/dev/null; then
1612
				echo "Done!" | tee -a ${_logfile}
1613
			else
1614
				echo "Failed!" | tee -a ${_logfile}
1615
				echo ">>> ERROR: An error occurred trying to sign Latest/pkg.txz"
1616
				print_error_pfS
1617
			fi
1618
		fi
1619
	fi
1620

    
1621
	if [ -n "${DO_NOT_UPLOAD}" ]; then
1622
		return
1623
	fi
1624

    
1625
	# Make sure destination directory exist
1626
	ssh -p ${PKG_RSYNC_SSH_PORT} \
1627
		${PKG_RSYNC_USERNAME}@${PKG_RSYNC_HOSTNAME} \
1628
		"mkdir -p ${PKG_RSYNC_DESTDIR}"
1629

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

    
1642
	if [ -z "${USE_PKG_REPO_STAGING}" -o -n "${_ignore_final_rsync}" ]; then
1643
		return
1644
	fi
1645

    
1646
	if [ -n "${_IS_RELEASE}" -o "${_repo_path_param}" = "${CORE_PKG_PATH}" ]; then
1647
		# Send .real* directories first to prevent having a broken repo while transfer happens
1648
		local _cmd="rsync -Have \"ssh -p ${PKG_FINAL_RSYNC_SSH_PORT}\" \
1649
			--timeout=60 ${PKG_RSYNC_DESTDIR}/./${_repo_base%%-core}* \
1650
			--include=\"/*\" --include=\"*/.real*\" --include=\"*/.real*/***\" \
1651
			--exclude=\"*\" \
1652
			${PKG_FINAL_RSYNC_USERNAME}@${PKG_FINAL_RSYNC_HOSTNAME}:${PKG_FINAL_RSYNC_DESTDIR}"
1653

    
1654
		echo -n ">>> Sending updated packages to ${PKG_FINAL_RSYNC_HOSTNAME}... " | tee -a ${_logfile}
1655
		if script -aq ${_logfile} ssh -p ${PKG_RSYNC_SSH_PORT} \
1656
			${PKG_RSYNC_USERNAME}@${PKG_RSYNC_HOSTNAME} ${_cmd} >/dev/null 2>&1; then
1657
			echo "Done!" | tee -a ${_logfile}
1658
		else
1659
			echo "Failed!" | tee -a ${_logfile}
1660
			echo ">>> ERROR: An error occurred sending repo to final hostname"
1661
			print_error_pfS
1662
		fi
1663

    
1664
		_cmd="rsync -Have \"ssh -p ${PKG_FINAL_RSYNC_SSH_PORT}\" \
1665
			--timeout=60 --delete-delay ${PKG_RSYNC_DESTDIR}/./${_repo_base%%-core}* \
1666
			${PKG_FINAL_RSYNC_USERNAME}@${PKG_FINAL_RSYNC_HOSTNAME}:${PKG_FINAL_RSYNC_DESTDIR}"
1667

    
1668
		echo -n ">>> Sending updated repositories metadata to ${PKG_FINAL_RSYNC_HOSTNAME}... " | tee -a ${_logfile}
1669
		if script -aq ${_logfile} ssh -p ${PKG_RSYNC_SSH_PORT} \
1670
			${PKG_RSYNC_USERNAME}@${PKG_RSYNC_HOSTNAME} ${_cmd} >/dev/null 2>&1; then
1671
			echo "Done!" | tee -a ${_logfile}
1672
		else
1673
			echo "Failed!" | tee -a ${_logfile}
1674
			echo ">>> ERROR: An error occurred sending repo to final hostname"
1675
			print_error_pfS
1676
		fi
1677
	fi
1678
}
1679

    
1680
poudriere_possible_archs() {
1681
	local _arch=$(uname -m)
1682
	local _archs=""
1683

    
1684
	# If host is amd64, we'll create both repos, and if possible armv6
1685
	if [ "${_arch}" = "amd64" ]; then
1686
		_archs="amd64.amd64"
1687

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

    
1692
			if binmiscctl lookup armv6 >/dev/null 2>&1; then
1693
				_archs="${_archs} arm.armv6"
1694
			fi
1695
		fi
1696
	fi
1697

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

    
1716
	echo ${_archs}
1717
}
1718

    
1719
poudriere_jail_name() {
1720
	local _jail_arch="${1}"
1721

    
1722
	if [ -z "${_jail_arch}" ]; then
1723
		return 1
1724
	fi
1725

    
1726
	# Remove arch
1727
	echo "${PRODUCT_NAME}_${POUDRIERE_BRANCH}_${_jail_arch##*.}"
1728
}
1729

    
1730
poudriere_rename_ports() {
1731
	if [ "${PRODUCT_NAME}" = "pfSense" ]; then
1732
		return;
1733
	fi
1734

    
1735
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1736

    
1737
	local _ports_dir="/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}"
1738

    
1739
	echo -n ">>> Renaming product ports on ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
1740
	for d in $(find ${_ports_dir} -depth 2 -type d -name '*pfSense*'); do
1741
		local _pdir=$(dirname ${d})
1742
		local _pname=$(echo $(basename ${d}) | sed "s,pfSense,${PRODUCT_NAME},")
1743
		local _plist=""
1744

    
1745
		if [ -e ${_pdir}/${_pname} ]; then
1746
			rm -rf ${_pdir}/${_pname}
1747
		fi
1748

    
1749
		cp -r ${d} ${_pdir}/${_pname}
1750

    
1751
		if [ -f ${_pdir}/${_pname}/pkg-plist ]; then
1752
			_plist=${_pdir}/${_pname}/pkg-plist
1753
		fi
1754

    
1755
		sed -i '' -e "s,pfSense,${PRODUCT_NAME},g" \
1756
			  -e "s,https://www.pfsense.org,${PRODUCT_URL},g" \
1757
			  -e "/^MAINTAINER=/ s,^.*$,MAINTAINER=	${PRODUCT_EMAIL}," \
1758
			${_pdir}/${_pname}/Makefile \
1759
			${_pdir}/${_pname}/pkg-descr ${_plist}
1760

    
1761
		# PHP module is special
1762
		if echo "${_pname}" | grep -q "^php[0-9]*-${PRODUCT_NAME}-module"; then
1763
			local _product_capital=$(echo ${PRODUCT_NAME} | tr '[a-z]' '[A-Z]')
1764
			sed -i '' -e "s,PHP_PFSENSE,PHP_${_product_capital},g" \
1765
				  -e "s,PFSENSE_SHARED_LIBADD,${_product_capital}_SHARED_LIBADD,g" \
1766
				  -e "s,pfSense,${PRODUCT_NAME},g" \
1767
				  -e "s,${PRODUCT_NAME}\.c,pfSense.c,g" \
1768
				${_pdir}/${_pname}/files/config.m4
1769

    
1770
			sed -i '' -e "s,COMPILE_DL_PFSENSE,COMPILE_DL_${_product_capital}," \
1771
				  -e "s,pfSense_module_entry,${PRODUCT_NAME}_module_entry,g" \
1772
				  -e "/ZEND_GET_MODULE/ s,pfSense,${PRODUCT_NAME}," \
1773
				  -e "/PHP_PFSENSE_WORLD_EXTNAME/ s,pfSense,${PRODUCT_NAME}," \
1774
				${_pdir}/${_pname}/files/pfSense.c \
1775
				${_pdir}/${_pname}/files/php_pfSense.h
1776
		fi
1777

    
1778
		if [ -d ${_pdir}/${_pname}/files ]; then
1779
			for fd in $(find ${_pdir}/${_pname}/files -type d -name '*pfSense*'); do
1780
				local _fddir=$(dirname ${fd})
1781
				local _fdname=$(echo $(basename ${fd}) | sed "s,pfSense,${PRODUCT_NAME},")
1782

    
1783
				mv ${fd} ${_fddir}/${_fdname}
1784
			done
1785
		fi
1786
	done
1787
	echo "Done!" | tee -a ${LOGFILE}
1788
}
1789

    
1790
poudriere_create_ports_tree() {
1791
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1792

    
1793
	if ! poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
1794
		local _branch=""
1795
		if [ -z "${POUDRIERE_PORTS_GIT_URL}" ]; then
1796
			echo ">>> ERROR: POUDRIERE_PORTS_GIT_URL is not defined"
1797
			print_error_pfS
1798
		fi
1799
		if [ -n "${POUDRIERE_PORTS_GIT_BRANCH}" ]; then
1800
			_branch="-B ${POUDRIERE_PORTS_GIT_BRANCH}"
1801
		fi
1802
		echo -n ">>> Creating poudriere ports tree, it may take some time... " | tee -a ${LOGFILE}
1803
		if ! script -aq ${LOGFILE} poudriere ports -c -p "${POUDRIERE_PORTS_NAME}" -m git -U ${POUDRIERE_PORTS_GIT_URL} ${_branch} >/dev/null 2>&1; then
1804
			echo "" | tee -a ${LOGFILE}
1805
			echo ">>> ERROR: Error creating poudriere ports tree, aborting..." | tee -a ${LOGFILE}
1806
			print_error_pfS
1807
		fi
1808
		echo "Done!" | tee -a ${LOGFILE}
1809
		poudriere_rename_ports
1810
	fi
1811
}
1812

    
1813
poudriere_init() {
1814
	local _error=0
1815
	local _archs=$(poudriere_possible_archs)
1816

    
1817
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1818

    
1819
	# Sanity checks
1820
	if [ -z "${ZFS_TANK}" ]; then
1821
		echo ">>> ERROR: \$ZFS_TANK is empty" | tee -a ${LOGFILE}
1822
		error=1
1823
	fi
1824

    
1825
	if [ -z "${ZFS_ROOT}" ]; then
1826
		echo ">>> ERROR: \$ZFS_ROOT is empty" | tee -a ${LOGFILE}
1827
		error=1
1828
	fi
1829

    
1830
	if [ -z "${POUDRIERE_PORTS_NAME}" ]; then
1831
		echo ">>> ERROR: \$POUDRIERE_PORTS_NAME is empty" | tee -a ${LOGFILE}
1832
		error=1
1833
	fi
1834

    
1835
	if [ ${_error} -eq 1 ]; then
1836
		print_error_pfS
1837
	fi
1838

    
1839
	# Check if zpool exists
1840
	if ! zpool list ${ZFS_TANK} >/dev/null 2>&1; then
1841
		echo ">>> ERROR: ZFS tank ${ZFS_TANK} not found, please create it and try again..." | tee -a ${LOGFILE}
1842
		print_error_pfS
1843
	fi
1844

    
1845
	# Check if zfs rootfs exists
1846
	if ! zfs list ${ZFS_TANK}${ZFS_ROOT} >/dev/null 2>&1; then
1847
		echo -n ">>> Creating ZFS filesystem ${ZFS_TANK}${ZFS_ROOT}... "
1848
		if zfs create -o atime=off -o mountpoint=/usr/local${ZFS_ROOT} \
1849
		    ${ZFS_TANK}${ZFS_ROOT} >/dev/null 2>&1; then
1850
			echo "Done!"
1851
		else
1852
			echo "Failed!"
1853
			print_error_pfS
1854
		fi
1855
	fi
1856

    
1857
	# Make sure poudriere is installed
1858
	if ! pkg info --quiet poudriere-devel; then
1859
		echo ">>> Installing poudriere-devel..." | tee -a ${LOGFILE}
1860
		if ! pkg install poudriere-devel >/dev/null 2>&1; then
1861
			echo ">>> ERROR: poudriere-devel was not installed, aborting..." | tee -a ${LOGFILE}
1862
			print_error_pfS
1863
		fi
1864
	fi
1865

    
1866
	# Create poudriere.conf
1867
	if [ -z "${POUDRIERE_PORTS_GIT_URL}" ]; then
1868
		echo ">>> ERROR: POUDRIERE_PORTS_GIT_URL is not defined"
1869
		print_error_pfS
1870
	fi
1871
	echo ">>> Creating poudriere.conf" | tee -a ${LOGFILE}
1872
	cat <<EOF >/usr/local/etc/poudriere.conf
1873
ZPOOL=${ZFS_TANK}
1874
ZROOTFS=${ZFS_ROOT}
1875
RESOLV_CONF=/etc/resolv.conf
1876
BASEFS=/usr/local/poudriere
1877
USE_PORTLINT=no
1878
USE_TMPFS=yes
1879
NOLINUX=yes
1880
DISTFILES_CACHE=/usr/ports/distfiles
1881
CHECK_CHANGED_OPTIONS=yes
1882
CHECK_CHANGED_DEPS=yes
1883
ATOMIC_PACKAGE_REPOSITORY=yes
1884
COMMIT_PACKAGES_ON_FAILURE=no
1885
KEEP_OLD_PACKAGES=yes
1886
KEEP_OLD_PACKAGES_COUNT=5
1887
EOF
1888

    
1889
	# Create specific items conf
1890
	[ ! -d /usr/local/etc/poudriere.d ] \
1891
		&& mkdir -p /usr/local/etc/poudriere.d
1892

    
1893
	# Create DISTFILES_CACHE if it doesn't exist
1894
	if [ ! -d /usr/ports/distfiles ]; then
1895
		mkdir -p /usr/ports/distfiles
1896
	fi
1897

    
1898
	# Remove old jails
1899
	for jail_arch in ${_archs}; do
1900
		jail_name=$(poudriere_jail_name ${jail_arch})
1901

    
1902
		if poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
1903
			echo ">>> Poudriere jail ${jail_name} already exists, deleting it..." | tee -a ${LOGFILE}
1904
			poudriere jail -d -j "${jail_name}" >/dev/null 2>&1
1905
		fi
1906
	done
1907

    
1908
	# Remove old ports tree
1909
	if poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
1910
		echo ">>> Poudriere ports tree ${POUDRIERE_PORTS_NAME} already exists, deleting it..." | tee -a ${LOGFILE}
1911
		poudriere ports -d -p "${POUDRIERE_PORTS_NAME}"
1912
	fi
1913

    
1914
	local native_xtools=""
1915
	# Now we are ready to create jails
1916
	for jail_arch in ${_archs}; do
1917
		jail_name=$(poudriere_jail_name ${jail_arch})
1918

    
1919
		if [ "${jail_arch}" = "arm.armv6" ]; then
1920
			native_xtools="-x"
1921
		else
1922
			native_xtools=""
1923
		fi
1924

    
1925
		echo -n ">>> Creating jail ${jail_name}, it may take some time... " | tee -a ${LOGFILE}
1926
		if ! script -aq ${LOGFILE} poudriere jail -c -j "${jail_name}" -v ${FREEBSD_BRANCH} \
1927
				-a ${jail_arch} -m git -U ${FREEBSD_REPO_BASE_POUDRIERE} ${native_xtools} >/dev/null 2>&1; then
1928
			echo "" | tee -a ${LOGFILE}
1929
			echo ">>> ERROR: Error creating jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
1930
			print_error_pfS
1931
		fi
1932
		echo "Done!" | tee -a ${LOGFILE}
1933
	done
1934

    
1935
	poudriere_create_ports_tree
1936

    
1937
	echo ">>> Poudriere is now configured!" | tee -a ${LOGFILE}
1938
}
1939

    
1940
poudriere_update_jails() {
1941
	local _archs=$(poudriere_possible_archs)
1942

    
1943
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1944

    
1945
	local native_xtools=""
1946
	for jail_arch in ${_archs}; do
1947
		jail_name=$(poudriere_jail_name ${jail_arch})
1948

    
1949
		local _create_or_update="-u"
1950
		local _create_or_update_text="Updating"
1951
		if ! poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
1952
			echo ">>> Poudriere jail ${jail_name} not found, creating..." | tee -a ${LOGFILE}
1953
			_create_or_update="-c -v ${FREEBSD_BRANCH} -a ${jail_arch} -m git -U ${FREEBSD_REPO_BASE_POUDRIERE}"
1954
			_create_or_update_text="Creating"
1955
		fi
1956

    
1957
		if [ "${jail_arch}" = "arm.armv6" ]; then
1958
			native_xtools="-x"
1959
		else
1960
			native_xtools=""
1961
		fi
1962

    
1963
		echo -n ">>> ${_create_or_update_text} jail ${jail_name}, it may take some time... " | tee -a ${LOGFILE}
1964
		if ! script -aq ${LOGFILE} poudriere jail ${_create_or_update} -j "${jail_name}" ${native_xtools} >/dev/null 2>&1; then
1965
			echo "" | tee -a ${LOGFILE}
1966
			echo ">>> ERROR: Error ${_create_or_update_text} jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
1967
			print_error_pfS
1968
		fi
1969
		echo "Done!" | tee -a ${LOGFILE}
1970
	done
1971
}
1972

    
1973
poudriere_update_ports() {
1974
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1975

    
1976
	# Create ports tree if necessary
1977
	if ! poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
1978
		poudriere_create_ports_tree
1979
	else
1980
		echo -n ">>> Resetting local changes on ports tree ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
1981
		script -aq ${LOGFILE} git -C "/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}" reset --hard >/dev/null 2>&1
1982
		script -aq ${LOGFILE} git -C "/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}" clean -fd >/dev/null 2>&1
1983
		echo "Done!" | tee -a ${LOGFILE}
1984
		echo -n ">>> Updating ports tree ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
1985
		script -aq ${LOGFILE} poudriere ports -u -p "${POUDRIERE_PORTS_NAME}" >/dev/null 2>&1
1986
		echo "Done!" | tee -a ${LOGFILE}
1987
		poudriere_rename_ports
1988
	fi
1989
}
1990

    
1991
poudriere_bulk() {
1992
	local _archs=$(poudriere_possible_archs)
1993

    
1994
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1995

    
1996
	if [ -z "${DO_NOT_UPLOAD}" -a -z "${PKG_RSYNC_HOSTNAME}" ]; then
1997
		echo ">>> ERROR: PKG_RSYNC_HOSTNAME is not set"
1998
		print_error_pfS
1999
	fi
2000

    
2001
	rm -f ${LOGFILE}
2002

    
2003
	poudriere_create_ports_tree
2004

    
2005
	[ -d /usr/local/etc/poudriere.d ] || \
2006
		mkdir -p /usr/local/etc/poudriere.d
2007

    
2008
	if [ -f "${BUILDER_TOOLS}/conf/pfPorts/make.conf" ]; then
2009
		cp -f "${BUILDER_TOOLS}/conf/pfPorts/make.conf" /usr/local/etc/poudriere.d/${POUDRIERE_PORTS_NAME}-make.conf
2010
	fi
2011

    
2012
	# Change version of pfSense meta ports for snapshots
2013
	if [ -z "${_IS_RELEASE}" ]; then
2014
		local _meta_pkg_version="$(echo "${PRODUCT_VERSION}" | sed 's,DEVELOPMENT,ALPHA,')-${DATESTRING}"
2015
		sed -i '' \
2016
			-e "/^DISTVERSION/ s,^.*,DISTVERSION=	${_meta_pkg_version}," \
2017
			-e "/^PORTREVISION=/d" \
2018
			/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}/security/${PRODUCT_NAME}/Makefile
2019
	fi
2020

    
2021
	for jail_arch in ${_archs}; do
2022
		jail_name=$(poudriere_jail_name ${jail_arch})
2023

    
2024
		if ! poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
2025
			echo ">>> Poudriere jail ${jail_name} not found, skipping..." | tee -a ${LOGFILE}
2026
			continue
2027
		fi
2028

    
2029
		if [ -f "${POUDRIERE_BULK}.${jail_arch}" ]; then
2030
			_ref_bulk="${POUDRIERE_BULK}.${jail_arch}"
2031
		else
2032
			_ref_bulk="${POUDRIERE_BULK}"
2033
		fi
2034

    
2035
		_bulk=${SCRATCHDIR}/poudriere_bulk.${POUDRIERE_BRANCH}
2036
		sed -e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" ${_ref_bulk} > ${_bulk}
2037

    
2038
		if ! poudriere bulk -f ${_bulk} -j ${jail_name} -p ${POUDRIERE_PORTS_NAME}; then
2039
			echo ">>> ERROR: Something went wrong..."
2040
			print_error_pfS
2041
		fi
2042

    
2043
		echo ">>> Cleaning up old packages from repo..."
2044
		if ! poudriere pkgclean -f ${_bulk} -j ${jail_name} -p ${POUDRIERE_PORTS_NAME} -y; then
2045
			echo ">>> ERROR: Something went wrong..."
2046
			print_error_pfS
2047
		fi
2048

    
2049
		pkg_repo_rsync "/usr/local/poudriere/data/packages/${jail_name}-${POUDRIERE_PORTS_NAME}"
2050
	done
2051
}
2052

    
2053
# This routine is called to write out to stdout
2054
# a string. The string is appended to $SNAPSHOTSLOGFILE
2055
snapshots_update_status() {
2056
	if [ -z "$1" ]; then
2057
		return
2058
	fi
2059
	if [ -z "${SNAPSHOTS}" -a -z "${POUDRIERE_SNAPSHOTS}" ]; then
2060
		return
2061
	fi
2062
	echo "$*"
2063
	echo "`date` -|- $*" >> $SNAPSHOTSLOGFILE
2064
}
2065

    
2066
create_sha256() {
2067
	local _file="${1}"
2068

    
2069
	if [ ! -f "${_file}" ]; then
2070
		return 1
2071
	fi
2072

    
2073
	( \
2074
		cd $(dirname ${_file}) && \
2075
		sha256 $(basename ${_file}) > $(basename ${_file}).sha256 \
2076
	)
2077
}
2078

    
2079
snapshots_create_latest_symlink() {
2080
	local _image="${1}"
2081

    
2082
	if [ -z "${_image}" ]; then
2083
		return
2084
	fi
2085

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

    
2090
	if [ ! -f "${_image}" ]; then
2091
		return
2092
	fi
2093

    
2094
	local _symlink=$(echo ${_image} | sed "s,${TIMESTAMP_SUFFIX},-latest,")
2095
	ln -sf $(basename ${_image}) ${_symlink}
2096
	ln -sf $(basename ${_image}).sha256 ${_symlink}.sha256
2097
}
2098

    
2099
snapshots_create_sha256() {
2100
	local _img=""
2101

    
2102
	for _img in ${ISOPATH} ${MEMSTICKPATH} ${MEMSTICKSERIALPATH} ${MEMSTICKADIPATH} ${OVAPATH} ${VARIANTIMAGES}; do
2103
		if [ -f "${_img}.gz" ]; then
2104
			_img="${_img}.gz"
2105
		fi
2106
		if [ ! -f "${_img}" ]; then
2107
			continue
2108
		fi
2109
		create_sha256 ${_img}
2110
		snapshots_create_latest_symlink ${_img}
2111
	done
2112

    
2113
	for NANOTYPE in nanobsd nanobsd-vga; do
2114
		for FILESIZE in ${FLASH_SIZE}; do
2115
			FILENAMEFULL="$(nanobsd_image_filename ${FILESIZE} ${NANOTYPE}).gz"
2116

    
2117
			if [ -f $IMAGES_FINAL_DIR/nanobsd/$FILENAMEFULL ]; then
2118
				create_sha256 $IMAGES_FINAL_DIR/nanobsd/$FILENAMEFULL
2119
			fi
2120
		done
2121
	done
2122
}
2123

    
2124
snapshots_scp_files() {
2125
	if [ -z "${RSYNC_COPY_ARGUMENTS}" ]; then
2126
		RSYNC_COPY_ARGUMENTS="-ave ssh --timeout=60 --bwlimit=${RSYNCKBYTELIMIT}" #--bwlimit=50
2127
	fi
2128

    
2129
	snapshots_update_status ">>> Copying core pkg repo to ${PKG_RSYNC_HOSTNAME}"
2130
	pkg_repo_rsync "${CORE_PKG_PATH}"
2131
	snapshots_update_status ">>> Finished copying core pkg repo"
2132

    
2133
	snapshots_update_status ">>> Copying files to ${RSYNCIP}"
2134

    
2135
	# Ensure directory(s) are available
2136
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/installer"
2137
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/nanobsd"
2138
	if [ -d $IMAGES_FINAL_DIR/virtualization ]; then
2139
		ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/virtualization"
2140
	fi
2141
	# ensure permissions are correct for r+w
2142
	ssh ${RSYNCUSER}@${RSYNCIP} "chmod -R ug+rw ${RSYNCPATH}/."
2143
	rsync $RSYNC_COPY_ARGUMENTS $IMAGES_FINAL_DIR/installer/* \
2144
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/installer/
2145
	rsync $RSYNC_COPY_ARGUMENTS $IMAGES_FINAL_DIR/nanobsd/* \
2146
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/nanobsd/
2147
	if [ -d $IMAGES_FINAL_DIR/virtualization ]; then
2148
		rsync $RSYNC_COPY_ARGUMENTS $IMAGES_FINAL_DIR/virtualization/* \
2149
			${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/virtualization/
2150
	fi
2151

    
2152
	snapshots_update_status ">>> Finished copying files."
2153
}
(2-2/3)