Project

General

Profile

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

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

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

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

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

    
70
	echo -n ">>> Creating core packages repository... "
71
	if pkg repo -q "${CORE_PKG_PATH}"; then
72
		echo "Done!"
73
	else
74
		echo "Failed!"
75
		print_error_pfS
76
	fi
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
	[ -d "${CORE_PKG_TMP}" ] \
88
		&& rm -rf ${CORE_PKG_TMP}
89

    
90
	local _templates_path=${BUILDER_TOOLS}/templates/core_pkg/${_template}
91
	local _template_metadir=${_templates_path}/metadir
92
	local _metadir=${CORE_PKG_TMP}/${_template}_metadir
93

    
94
	if [ ! -d ${_template_metadir} ]; then
95
		echo "ERROR: Template dir not found for pkg ${_template}"
96
		exit
97
	fi
98

    
99
	mkdir -p ${CORE_PKG_TMP}
100

    
101
	cp -r ${_template_metadir} ${_metadir}
102

    
103
	local _manifest=${_metadir}/+MANIFEST
104
	local _plist=${CORE_PKG_TMP}/${_template}_plist
105
	local _exclude_plist=${CORE_PKG_TMP}/${_template}_exclude_plist
106

    
107
	if [ -f "${_templates_path}/pkg-plist" ]; then
108
		cp ${_templates_path}/pkg-plist ${_plist}
109
	else
110
		if [ -n "${_filter}" ]; then
111
			_filter="-name ${_filter}"
112
		fi
113
		(cd ${_root} && find . ${_filter} -type f -or -type l | sed 's,^.,,' | sort -u) > ${_plist}
114
	fi
115

    
116
	if [ -f "${_templates_path}/exclude_plist" ]; then
117
		cp ${_templates_path}/exclude_plist ${_exclude_plist}
118
	else
119
		touch ${_exclude_plist}
120
	fi
121

    
122
	sed \
123
		-i '' \
124
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
125
		-e "s,%%PRODUCT_URL%%,${PRODUCT_URL},g" \
126
		-e "s,%%FLAVOR%%,${_flavor},g" \
127
		-e "s,%%VERSION%%,${_version},g" \
128
		${_metadir}/* \
129
		${_plist} \
130
		${exclude_plist}
131

    
132
	if [ -f "${_exclude_plist}" ]; then
133
		sort -u ${_exclude_plist} > ${_plist}.exclude
134
		mv ${_plist} ${_plist}.tmp
135
		comm -23 ${_plist}.tmp ${_plist}.exclude > ${_plist}
136
		rm -f ${_plist}.tmp ${plist}.exclude
137
	fi
138

    
139
	mkdir -p ${CORE_PKG_PATH}/All
140
	if ! pkg create -o ${CORE_PKG_PATH}/All -p ${_plist} -r ${_root} -m ${_metadir}; then
141
		echo ">>> ERROR: Error building package ${_template} ${_flavor}"
142
		print_error_pfS
143
	fi
144
}
145

    
146
# This routine will output that something went wrong
147
print_error_pfS() {
148
	echo
149
	echo "####################################"
150
	echo "Something went wrong, check errors!" >&2
151
	echo "####################################"
152
	echo
153
	echo "NOTE: a lot of times you can run './build.sh --clean-builder' to resolve."
154
	echo
155
	if [ "$1" != "" ]; then
156
		echo $1
157
	fi
158
	[ -n "${LOGFILE}" -a -f "${LOGFILE}" ] && \
159
		echo "Log saved on ${LOGFILE}" && \
160
		tail -n20 ${LOGFILE} >&2
161
	echo
162
	kill $$
163
	exit 1
164
}
165

    
166
prestage_on_ram_setup() {
167
	[ -d "${STAGE_CHROOT_DIR}" ] \
168
		|| mkdir -p ${STAGE_CHROOT_DIR}
169
	[ -d "${FINAL_CHROOT_DIR}" ] \
170
		|| mkdir -p ${FINAL_CHROOT_DIR}
171

    
172
	_AVAIL_MEM=$(($(sysctl -n hw.usermem) / 1024 / 1024))
173
	if [ $_AVAIL_MEM -lt 2000 ]; then
174
		echo ">>> Builder has less than 2GiB RAM skipping memory disks"
175
		return
176
	else
177
		echo "######################################################################################"
178
		echo
179
		echo ">>> Builder has more than 2GiB RAM enabling memory disks"
180
		echo ">>> WARNING: Remember to remove these memory disks by running $0 --disable-memorydisks"
181
		echo
182
		echo "######################################################################################"
183
	fi
184

    
185
	if df /dev/ufs/prestagebacking >/dev/null 2>&1; then
186
		echo ">>> Detected preexisting memory disk enabled for ${STAGE_CHROOT_DIR}."
187
	else
188
		mdconfig -a -t swap -u 10001 -s ${MEMORYDISK_SIZE}
189
		newfs -L prestagebacking -U /dev/md10001
190
		mount /dev/ufs/prestagebacking ${STAGE_CHROOT_DIR}
191
	fi
192

    
193
	if df /dev/ufs/finalstagebacking >/dev/null 2>&1; then
194
		echo ">>> Detected preexisting memory disk enabled for ${FINAL_CHROOT_DIR}."
195
	else
196
		mdconfig -a -t swap -u 10002 -s ${MEMORYDISK_SIZE}
197
		newfs -L finalstagebacking -U /dev/md10002
198
		mount /dev/ufs/finalstagebacking ${FINAL_CHROOT_DIR}
199
	fi
200
}
201

    
202
prestage_on_ram_cleanup() {
203
	if [ -c /dev/md10001 ]; then
204
		if [ -d ${STAGE_CHROOT_DIR} ]; then
205
			umount ${STAGE_CHROOT_DIR}
206
		fi
207
		mdconfig -d -u 10001
208
	fi
209
	if [ -c /dev/md10002 ]; then
210
		if [ -d ${FINAL_CHROOT_DIR} ]; then
211
			umount ${FINAL_CHROOT_DIR}
212
		fi
213
		mdconfig -d -u 10002
214
	fi
215
}
216

    
217
# This routine will verify that the kernel has been
218
# installed OK to the staging area.
219
ensure_kernel_exists() {
220
	if [ ! -f "$1/boot/kernel/kernel.gz" ]; then
221
		echo ">>> ERROR: Could not locate $1/boot/kernel.gz"
222
		print_error_pfS
223
	fi
224
	KERNEL_SIZE=$(stat -f "%z" $1/boot/kernel/kernel.gz)
225
	if [ "$KERNEL_SIZE" -lt 3500 ]; then
226
		echo ">>> ERROR: Kernel $1/boot/kernel.gz appears to be smaller than it should be: $KERNEL_SIZE"
227
		print_error_pfS
228
	fi
229
}
230

    
231
get_pkg_name() {
232
	echo "${PRODUCT_NAME}-${1}-${CORE_PKG_VERSION}"
233
}
234

    
235
# This routine builds all related kernels
236
build_all_kernels() {
237
	# Set KERNEL_BUILD_PATH if it has not been set
238
	if [ -z "${KERNEL_BUILD_PATH}" ]; then
239
		KERNEL_BUILD_PATH=$SCRATCHDIR/kernels
240
		echo ">>> KERNEL_BUILD_PATH has not been set. Setting to ${KERNEL_BUILD_PATH}!"
241
	fi
242

    
243
	[ -d "${KERNEL_BUILD_PATH}" ] \
244
		&& rm -rf ${KERNEL_BUILD_PATH}
245

    
246
	# Build embedded kernel
247
	for BUILD_KERNEL in $BUILD_KERNELS; do
248
		unset KERNCONF
249
		unset KERNEL_DESTDIR
250
		unset KERNELCONF
251
		unset KERNEL_NAME
252
		export KERNCONF=$BUILD_KERNEL
253
		export KERNEL_DESTDIR="$KERNEL_BUILD_PATH/$BUILD_KERNEL"
254
		export KERNELCONF="${FREEBSD_SRC_DIR}/sys/${TARGET}/conf/$BUILD_KERNEL"
255
		export KERNEL_NAME=${BUILD_KERNEL}
256

    
257
		LOGFILE="${BUILDER_LOGS}/kernel.${KERNCONF}.${TARGET}.log"
258
		echo ">>> Building $BUILD_KERNEL kernel."  | tee -a ${LOGFILE}
259

    
260
		if [ ! -e "${FREEBSD_SRC_DIR}/sys/${TARGET}/conf/${BUILD_KERNEL}" ]; then
261
			echo ">>> ERROR: Could not find $KERNELCONF"
262
			print_error_pfS
263
		fi
264

    
265
		if [ -n "${NO_BUILDKERNEL}" -a -f "${CORE_PKG_PATH}/All/$(get_pkg_name kernel-${KERNEL_NAME}).txz" ]; then
266
			echo ">>> NO_BUILDKERNEL set, skipping build" | tee -a ${LOGFILE}
267
			continue
268
		fi
269

    
270
		export SRC_CONF=${SRC_CONF}
271
		buildkernel
272

    
273
		echo ">>> Staging $BUILD_KERNEL kernel..." | tee -a ${LOGFILE}
274
		installkernel
275

    
276
		ensure_kernel_exists $KERNEL_DESTDIR
277

    
278
		echo -n ">>> Creating pkg of $KERNEL_NAME-debug kernel to staging area..."  | tee -a ${LOGFILE}
279
		core_pkg_create kernel-debug ${KERNEL_NAME} ${CORE_PKG_VERSION} ${KERNEL_DESTDIR} \*.symbols
280
		find ${KERNEL_DESTDIR} -name '*.symbols' -type f -delete
281

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

    
285
		rm -rf $KERNEL_DESTDIR 2>&1 1>/dev/null
286

    
287
		echo ".Done" | tee -a ${LOGFILE}
288
	done
289
}
290

    
291
install_default_kernel() {
292
	if [ -z "${1}" ]; then
293
		echo ">>> ERROR: install_default_kernel called without a kernel config name"| tee -a ${LOGFILE}
294
		print_error_pfS
295
	fi
296

    
297
	export KERNEL_NAME="${1}"
298

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

    
301
	# Copy kernel package to chroot, otherwise pkg won't find it to install
302
	if ! pkg_chroot_add ${FINAL_CHROOT_DIR} kernel-${KERNEL_NAME}; then
303
		echo ">>> ERROR: Error installing kernel package $(get_pkg_name kernel-${KERNEL_NAME}).txz" | tee -a ${LOGFILE}
304
		print_error_pfS
305
	fi
306

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

    
310
	if [ ! -f $FINAL_CHROOT_DIR/boot/kernel/kernel.gz ]; then
311
		echo ">>> ERROR: No kernel installed on $FINAL_CHROOT_DIR and the resulting image will be unusable. STOPPING!" | tee -a ${LOGFILE}
312
		print_error_pfS
313
	fi
314
	mkdir -p $FINAL_CHROOT_DIR/pkgs
315
	if [ -z "${2}" -o -n "${INSTALL_EXTRA_KERNELS}" ]; then
316
		cp ${CORE_PKG_PATH}/All/$(get_pkg_name kernel-${KERNEL_NAME}).txz $FINAL_CHROOT_DIR/pkgs
317
		if [ -n "${INSTALL_EXTRA_KERNELS}" ]; then
318
			for _EXTRA_KERNEL in $INSTALL_EXTRA_KERNELS; do
319
				_EXTRA_KERNEL_PATH=${CORE_PKG_PATH}/All/$(get_pkg_name kernel-${_EXTRA_KERNEL}).txz
320
				if [ -f "${_EXTRA_KERNEL_PATH}" ]; then
321
					echo -n ". adding ${_EXTRA_KERNEL_PATH} on image /pkgs folder"
322
					cp ${_EXTRA_KERNEL_PATH} $FINAL_CHROOT_DIR/pkgs
323
				else
324
					echo ">>> ERROR: Requested kernel $(get_pkg_name kernel-${_EXTRA_KERNEL}).txz was not found to be put on image /pkgs folder!"
325
					print_error_pfS
326
				fi
327
			done
328
		fi
329
	fi
330
	echo "Done." | tee -a ${LOGFILE}
331

    
332
	unset KERNEL_NAME
333
}
334

    
335
# Creates a full update file
336
create_Full_update_tarball() {
337
	mkdir -p $UPDATESDIR
338

    
339
	customize_stagearea_for_image "fullupdate"
340
	install_default_kernel ${DEFAULT_KERNEL}
341

    
342
	rm -rf ${FINAL_CHROOT_DIR}/cf
343
	rm -rf ${FINAL_CHROOT_DIR}/conf
344
	rm -f ${FINAL_CHROOT_DIR}/etc/rc.conf
345
	rm -f ${FINAL_CHROOT_DIR}/etc/pwd.db 2>/dev/null
346
	rm -f ${FINAL_CHROOT_DIR}/etc/group 2>/dev/null
347
	rm -f ${FINAL_CHROOT_DIR}/etc/spwd.db 2>/dev/null
348
	rm -f ${FINAL_CHROOT_DIR}/etc/passwd 2>/dev/null
349
	rm -f ${FINAL_CHROOT_DIR}/etc/master.passwd 2>/dev/null
350
	rm -f ${FINAL_CHROOT_DIR}/etc/fstab 2>/dev/null
351
	rm -f ${FINAL_CHROOT_DIR}/etc/bogons 2>/dev/null
352
	# Remove loader.conf and friends.  Ticket #560
353
	rm ${FINAL_CHROOT_DIR}/boot/loader.conf 2>/dev/null
354
	rm ${FINAL_CHROOT_DIR}/boot/loader.conf.local 2>/dev/null
355

    
356
	# Old systems will run (pre|post)_upgrade_command from /tmp
357
	if [ -f ${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/pre_upgrade_command ]; then
358
		cp -p \
359
			${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/pre_upgrade_command \
360
			${FINAL_CHROOT_DIR}/tmp
361
	fi
362
	if [ -f ${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/post_upgrade_command ]; then
363
		cp -p \
364
			${FINAL_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/post_upgrade_command \
365
			${FINAL_CHROOT_DIR}/tmp
366
	fi
367

    
368
	echo ">>> Creating ${UPDATES_TARBALL_FILENAME} ..." | tee -a ${LOGFILE}
369
	tar --exclude=./dev -czPf ${UPDATES_TARBALL_FILENAME} -C ${FINAL_CHROOT_DIR} .
370
}
371

    
372
# Outputs various set variables aka env
373
print_flags() {
374

    
375
	echo
376
	printf "             Product version: %s\n" $PRODUCT_VERSION
377
	printf "                   Stage DIR: %s\n" $STAGE_CHROOT_DIR
378
	printf "                 Updates dir: %s\n" $UPDATESDIR
379
	printf " Image Preparation Stage DIR: %s\n" $FINAL_CHROOT_DIR
380
	printf "                  Source DIR: %s\n" $FREEBSD_SRC_DIR
381
	printf "          FreeBSD repository: %s\n" $FREEBSD_REPO_BASE
382
	printf "          FreeBSD-src branch: %s\n" $FREEBSD_BRANCH
383
	printf "     FreeBSD original branch: %s\n" $FREEBSD_PARENT_BRANCH
384
	printf "               BUILD_KERNELS: %s\n" $BUILD_KERNELS
385
	printf "           Git Branch or Tag: %s\n" $GIT_REPO_BRANCH_OR_TAG
386
	printf "            MODULES_OVERRIDE: %s\n" $MODULES_OVERRIDE
387
	printf "    VMDK_DISK_CAPACITY_IN_GB: %s\n" $VMDK_DISK_CAPACITY_IN_GB
388
	printf "   OVA_FIRST_PART_SIZE_IN_GB: %s\n" $OVA_FIRST_PART_SIZE_IN_GB
389
	printf "    OVA_SWAP_PART_SIZE_IN_GB: %s\n" $OVA_SWAP_PART_SIZE_IN_GB
390
	printf "                 OVFTEMPLATE: %s\n" $OVFTEMPLATE
391
	printf "                     OVFVMDK: %s\n" $OVFVMDK
392
	printf "                    SRC_CONF: %s\n" $SRC_CONF
393
	printf "                     ISOPATH: %s\n" $ISOPATH
394
	printf "                MEMSTICKPATH: %s\n" $MEMSTICKPATH
395
	printf "    UPDATES_TARBALL_FILENAME: %s\n" $UPDATES_TARBALL_FILENAME
396
if [ -n "$SHOW_ENV" ]; then
397
	for LINE in $(env | egrep -v '(terminal|PASS|NAME|USER|SSH|GROUP|HOST)'); do
398
		echo "SHOW_ENV: $LINE"
399
	done
400
fi
401
	echo
402
}
403

    
404
# This builds FreeBSD (make buildworld)
405
# Imported from FreeSBIE
406
make_world() {
407

    
408
	LOGFILE=${BUILDER_LOGS}/buildworld.${TARGET}
409
	if [ -n "${NO_BUILDWORLD}" ]; then
410
		echo ">>> NO_BUILDWORLD set, skipping build" | tee -a ${LOGFILE}
411
		return
412
	fi
413

    
414
	# Set SRC_CONF variable if it's not already set.
415
	if [ -z "${SRC_CONF}" ]; then
416
		echo ">>> SRC_CONF is unset make sure this is what you want!" | tee -a ${LOGFILE}
417
	else
418
		echo ">>> Setting SRC_CONF to $SRC_CONF" | tee -a ${LOGFILE}
419
	fi
420

    
421
	# Set default parameters
422
	export MAKE_ARGS="${MAKEJ_WORLD} __MAKE_CONF=${MAKE_CONF} SRCCONF=${SRC_CONF} TARGET=${TARGET} TARGET_ARCH=${TARGET_ARCH}"
423

    
424
	echo ">>> LOGFILE set to $LOGFILE." | tee -a ${LOGFILE}
425
	makeargs="${MAKE_ARGS}"
426
	echo ">>> Building world for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
427
	echo ">>> Builder is running the command: env LOCAL_ITOOLS=\"${EXTRA_TOOLS}\" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} -DNO_CLEAN ${makeargs} buildworld" | tee -a ${LOGFILE}
428
	(env LOCAL_ITOOLS="${EXTRA_TOOLS}" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} -DNO_CLEAN ${makeargs} buildworld || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
429
	echo ">>> Building world for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
430

    
431
	LOGFILE=${BUILDER_LOGS}/installworld.${TARGET}
432
	echo ">>> LOGFILE set to $LOGFILE." | tee -a ${LOGFILE}
433
	# Create if cleaned up
434
	makeargs="${MAKE_ARGS} DESTDIR=${STAGE_CHROOT_DIR} WITHOUT_TOOLCHAIN=1"
435
	echo ">>> Installing world for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
436
	echo ">>> Builder is running the command: env LOCAL_ITOOLS=\"${EXTRA_TOOLS}\" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} installworld" | tee -a ${LOGFILE}
437
	(env LOCAL_ITOOLS="${EXTRA_TOOLS}" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} installworld || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
438
	echo ">>> Installing world for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
439

    
440
	makeargs="${MAKE_ARGS} DESTDIR=${STAGE_CHROOT_DIR}"
441
	echo ">>> Distribution world for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
442
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} distribution " | tee -a ${LOGFILE}
443
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} distribution  || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
444
	echo ">>> Distribution world for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
445

    
446
	[ -d "${STAGE_CHROOT_DIR}/usr/local/bin" ] \
447
		|| mkdir -p ${STAGE_CHROOT_DIR}/usr/local/bin
448
	makeargs="${MAKE_ARGS} DESTDIR=${STAGE_CHROOT_DIR}"
449
	echo ">>> Building and installing crypto tools and athstats for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
450
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/crypto ${makeargs} clean all install " | tee -a ${LOGFILE}
451
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/crypto ${makeargs} clean all install || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
452
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} clean" | tee -a ${LOGFILE}
453
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} clean || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
454
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} all" | tee -a ${LOGFILE}
455
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} all || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
456
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} install" | tee -a ${LOGFILE}
457
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} install || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
458
	echo ">>> Building and installing crypto tools and athstats for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
459

    
460
	unset makeargs
461
}
462

    
463
# This routine originated in nanobsd.sh
464
nanobsd_set_flash_details () {
465
	a1=$(echo $1 | tr '[:upper:]' '[:lower:]')
466

    
467
	# Source:
468
	#	SanDisk CompactFlash Memory Card
469
	#	Product Manual
470
	#	Version 10.9
471
	#	Document No. 20-10-00038
472
	#	April 2005
473
	# Table 2-7
474
	# NB: notice math error in SDCFJ-4096-388 line.
475
	#
476
	case "${a1}" in
477
		1024|1024m|1024mb|1g)
478
			NANO_MEDIASIZE=$((997129216/512))
479
			;;
480
		2048|2048m|2048mb|2g)
481
			NANO_MEDIASIZE=$((1989999616/512))
482
			;;
483
		4096|4096m|4096mb|4g)
484
			NANO_MEDIASIZE=$((3989999616/512))
485
			;;
486
		8192|8192m|8192mb|8g)
487
			NANO_MEDIASIZE=$((7989999616/512))
488
			;;
489
		16384|16384m|16384mb|16g)
490
			NANO_MEDIASIZE=$((15989999616/512))
491
			;;
492
		*)
493
			echo "Unknown Flash capacity"
494
			exit 2
495
			;;
496
	esac
497

    
498
	NANO_HEADS=16
499
	NANO_SECTS=63
500

    
501
	echo ">>> [nanoo] $1"
502
	echo ">>> [nanoo] NANO_MEDIASIZE: $NANO_MEDIASIZE"
503
	echo ">>> [nanoo] NANO_HEADS: $NANO_HEADS"
504
	echo ">>> [nanoo] NANO_SECTS: $NANO_SECTS"
505
	echo ">>> [nanoo] NANO_BOOT0CFG: $NANO_BOOT0CFG"
506
}
507

    
508
# This routine originated in nanobsd.sh
509
create_nanobsd_diskimage () {
510
	if [ -z "${1}" ]; then
511
		echo ">>> ERROR: Type of image has not been specified"
512
		print_error_pfS
513
	fi
514
	if [ -z "${2}" ]; then
515
		echo ">>> ERROR: Size of image has not been specified"
516
		print_error_pfS
517
	fi
518

    
519
	if [ "${1}" = "nanobsd" ]; then
520
		# It's serial
521
		export NANO_BOOTLOADER="boot/boot0sio"
522
	elif [ "${1}" = "nanobsd-vga" ]; then
523
		# It's vga
524
		export NANO_BOOTLOADER="boot/boot0"
525
	else
526
		echo ">>> ERROR: Type of image to create unknown"
527
		print_error_pfS
528
	fi
529

    
530
	if [ -z "${2}" ]; then
531
		echo ">>> ERROR: Media size(s) not specified."
532
		print_error_pfS
533
	fi
534

    
535
	if [ -z "${2}" ]; then
536
		echo ">>> ERROR: FLASH_SIZE is not set."
537
		print_error_pfS
538
	fi
539

    
540
	LOGFILE=${BUILDER_LOGS}/${1}.${TARGET}
541
	# Prepare folder to be put in image
542
	customize_stagearea_for_image "${1}"
543
	install_default_kernel ${DEFAULT_KERNEL} "no"
544

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

    
547
	echo "nanobsd" > $FINAL_CHROOT_DIR/etc/platform
548

    
549
	local BOOTCONF=${FINAL_CHROOT_DIR}/boot.config
550
	local LOADERCONF=${FINAL_CHROOT_DIR}/boot/loader.conf
551

    
552
	if [ "${1}" = "nanobsd" ]; then
553
		# Tell loader to use serial console early.
554
		echo "-S115200 -h" >> ${BOOTCONF}
555

    
556
		# Remove old console options if present.
557
		[ -f "${LOADERCONF}" ] \
558
			&& sed -i "" -Ee "/(console|boot_multicons|boot_serial|hint.uart)/d" ${LOADERCONF}
559
		# Activate serial console+video console in loader.conf
560
		echo 'loader_color="NO"' >> ${LOADERCONF}
561
		echo 'beastie_disable="YES"' >> ${LOADERCONF}
562
		echo 'boot_serial="YES"' >> ${LOADERCONF}
563
		echo 'console="comconsole"' >> ${LOADERCONF}
564
		echo 'comconsole_speed="115200"' >> ${LOADERCONF}
565
	fi
566
	echo 'autoboot_delay="5"' >> ${LOADERCONF}
567

    
568
	for _NANO_MEDIASIZE in ${2}; do
569
		if [ -z "${_NANO_MEDIASIZE}" ]; then
570
			continue;
571
		fi
572

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

    
576
		IMG="${IMAGES_FINAL_DIR}/${PRODUCT_NAME}-${PRODUCT_VERSION}-${_NANO_MEDIASIZE}-${TARGET}-${1}${TIMESTAMP_SUFFIX}.img"
577
		IMGUPDATE="${IMAGES_FINAL_DIR}/${PRODUCT_NAME}-${PRODUCT_VERSION}-${_NANO_MEDIASIZE}-${TARGET}-${1}-upgrade${TIMESTAMP_SUFFIX}.img"
578

    
579
		nanobsd_set_flash_details ${_NANO_MEDIASIZE}
580

    
581
		# These are defined in FlashDevice and on builder_default.sh
582
		echo $NANO_MEDIASIZE \
583
			$NANO_IMAGES \
584
			$NANO_SECTS \
585
			$NANO_HEADS \
586
			$NANO_CODESIZE \
587
			$NANO_CONFSIZE \
588
			$NANO_DATASIZE |
589
awk '
590
{
591
	printf "# %s\n", $0
592

    
593
	# size of cylinder in sectors
594
	cs = $3 * $4
595

    
596
	# number of full cylinders on media
597
	cyl = int ($1 / cs)
598

    
599
	# output fdisk geometry spec, truncate cyls to 1023
600
	if (cyl <= 1023)
601
		print "g c" cyl " h" $4 " s" $3
602
	else
603
		print "g c" 1023 " h" $4 " s" $3
604

    
605
	if ($7 > 0) {
606
		# size of data partition in full cylinders
607
		dsl = int (($7 + cs - 1) / cs)
608
	} else {
609
		dsl = 0;
610
	}
611

    
612
	# size of config partition in full cylinders
613
	csl = int (($6 + cs - 1) / cs)
614

    
615
	if ($5 == 0) {
616
		# size of image partition(s) in full cylinders
617
		isl = int ((cyl - dsl - csl) / $2)
618
	} else {
619
		isl = int (($5 + cs - 1) / cs)
620
	}
621

    
622
	# First image partition start at second track
623
	print "p 1 165 " $3, isl * cs - $3
624
	c = isl * cs;
625

    
626
	# Second image partition (if any) also starts offset one
627
	# track to keep them identical.
628
	if ($2 > 1) {
629
		print "p 2 165 " $3 + c, isl * cs - $3
630
		c += isl * cs;
631
	}
632

    
633
	# Config partition starts at cylinder boundary.
634
	print "p 3 165 " c, csl * cs
635
	c += csl * cs
636

    
637
	# Data partition (if any) starts at cylinder boundary.
638
	if ($7 > 0) {
639
		print "p 4 165 " c, dsl * cs
640
	} else if ($7 < 0 && $1 > c) {
641
		print "p 4 165 " c, $1 - c
642
	} else if ($1 < c) {
643
		print "Disk space overcommitted by", \
644
		    c - $1, "sectors" > "/dev/stderr"
645
		exit 2
646
	}
647

    
648
	# Force slice 1 to be marked active. This is necessary
649
	# for booting the image from a USB device to work.
650
	print "a 1"
651
}
652
	' > ${IMAGES_FINAL_DIR}/_.fdisk
653

    
654
		MNT=${IMAGES_FINAL_DIR}/_.mnt
655
		mkdir -p ${MNT}
656

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

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

    
663
		fdisk -i -f ${IMAGES_FINAL_DIR}/_.fdisk ${MD} 2>&1 >> ${LOGFILE}
664
		fdisk ${MD} 2>&1 >> ${LOGFILE}
665

    
666
		boot0cfg -B -b ${FINAL_CHROOT_DIR}/${NANO_BOOTLOADER} ${NANO_BOOT0CFG} ${MD} 2>&1 >> ${LOGFILE}
667

    
668
		# Create first image
669
		bsdlabel -m i386 -w -B -b ${FINAL_CHROOT_DIR}/boot/boot ${MD}s1 2>&1 >> ${LOGFILE}
670
		bsdlabel -m i386 ${MD}s1 2>&1 >> ${LOGFILE}
671
		local _label=$(lc ${PRODUCT_NAME})
672
		newfs -L ${_label}0 ${NANO_NEWFS} /dev/${MD}s1a 2>&1 >> ${LOGFILE}
673
		mount /dev/ufs/${_label}0 ${MNT}
674
		if [ $? -ne 0 ]; then
675
			echo ">>> ERROR: Something wrong happened during mount of first slice image creation. STOPPING!" | tee -a ${LOGFILE}
676
			print_error_pfS
677
		fi
678
		# Consider the unmounting as well
679
		trap "umount /dev/ufs/${_label}0; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
680

    
681
		clone_directory_contents ${FINAL_CHROOT_DIR} ${MNT}
682

    
683
		# Set NanoBSD image size
684
		echo "${_NANO_MEDIASIZE}" > ${MNT}/etc/nanosize.txt
685
		rm -f $MNT/cf/conf/* 2>/dev/null
686

    
687
		echo "/dev/ufs/${_label}0 / ufs ro,sync,noatime 1 1" > ${MNT}/etc/fstab
688
		if [ $NANO_CONFSIZE -gt 0 ] ; then
689
			echo "/dev/ufs/cf /cf ufs ro,sync,noatime 1 1" >> ${MNT}/etc/fstab
690
		fi
691

    
692
		umount ${MNT}
693
		# Restore the original trap
694
		trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
695

    
696
		# Setting NANO_IMAGES to 1 and NANO_INIT_IMG2 will tell
697
		# NanoBSD to only create one partition.  We default to 2
698
		# partitions in case anything happens to the first the
699
		# operator can boot from the 2nd and should be OK.
700

    
701
		# Before just going to use dd for duplicate think!
702
		# The images are created as sparse so lets take advantage
703
		# of that by just exec some commands.
704
		if [ $NANO_IMAGES -gt 1 -a $NANO_INIT_IMG2 -gt 0 ] ; then
705
			# Duplicate to second image (if present)
706
			echo ">>> Creating NanoBSD second slice by duplicating first slice." | tee -a ${LOGFILE}
707
			# Create second image
708
			dd if=/dev/${MD}s1 of=/dev/${MD}s2 conv=sparse bs=64k 2>&1 >> ${LOGFILE}
709
			tunefs -L ${_label}1 /dev/${MD}s2a 2>&1 >> ${LOGFILE}
710
			mount /dev/ufs/${_label}1 ${MNT}
711
			if [ $? -ne 0 ]; then
712
				echo ">>> ERROR: Something wrong happened during mount of second slice image creation. STOPPING!" | tee -a ${LOGFILE}
713
				print_error_pfS
714
			fi
715
			# Consider the unmounting as well
716
			trap "umount /dev/ufs/${_label}1; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
717

    
718
			echo "/dev/ufs/${_label}1 / ufs ro,sync,noatime 1 1" > ${MNT}/etc/fstab
719
			if [ $NANO_CONFSIZE -gt 0 ] ; then
720
				echo "/dev/ufs/cf /cf ufs ro,sync,noatime 1 1" >> ${MNT}/etc/fstab
721
			fi
722

    
723
			umount ${MNT}
724
			# Restore the trap back
725
			trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
726
		fi
727

    
728
		# Create Data slice, if any.
729
		# Note the changing of the variable to NANO_CONFSIZE
730
		# from NANO_DATASIZE.  We also added glabel support
731
		# and populate the Product configuration from the /cf
732
		# directory located in FINAL_CHROOT_DIR
733
		if [ $NANO_CONFSIZE -gt 0 ] ; then
734
			echo ">>> Creating /cf area to hold config.xml"
735
			newfs -L cf ${NANO_NEWFS} /dev/${MD}s3 2>&1 >> ${LOGFILE}
736
			# Mount data partition and copy contents of /cf
737
			# Can be used later to create custom default config.xml while building
738
			mount /dev/ufs/cf ${MNT}
739
			if [ $? -ne 0 ]; then
740
				echo ">>> ERROR: Something wrong happened during mount of cf slice image creation. STOPPING!" | tee -a ${LOGFILE}
741
				print_error_pfS
742
			fi
743
			# Consider the unmounting as well
744
			trap "umount /dev/ufs/cf; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
745

    
746
			clone_directory_contents ${FINAL_CHROOT_DIR}/cf ${MNT}
747

    
748
			umount ${MNT}
749
			# Restore the trap back
750
			trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
751
		else
752
			">>> [nanoo] NANO_CONFSIZE is not set. Not adding a /conf partition.. You sure about this??" | tee -a ${LOGFILE}
753
		fi
754

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

    
758
		mdconfig -d -u $MD
759
		# Restore default action
760
		trap "-" 1 2 15 EXIT
761

    
762
		# Check each image and ensure that they are over
763
		# 3 megabytes.  If either image is under 20 megabytes
764
		# in size then error out.
765
		IMGSIZE=$(stat -f "%z" ${IMG})
766
		IMGUPDATESIZE=$(stat -f "%z" ${IMGUPDATE})
767
		CHECKSIZE="20040710"
768
		if [ "$IMGSIZE" -lt "$CHECKSIZE" ]; then
769
			echo ">>> ERROR: Something went wrong when building NanoBSD.  The image size is under 20 megabytes!" | tee -a ${LOGFILE}
770
			print_error_pfS
771
		fi
772
		if [ "$IMGUPDATESIZE" -lt "$CHECKSIZE" ]; then
773
			echo ">>> ERROR: Something went wrong when building NanoBSD upgrade image.  The image size is under 20 megabytes!" | tee -a ${LOGFILE}
774
			print_error_pfS
775
		fi
776

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

    
780
		gzip -qf $IMG &
781
		_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
782
		gzip -qf $IMGUPDATE &
783
		_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
784
	done
785

    
786
	unset IMG
787
	unset IMGUPDATE
788
	unset IMGUPDATESIZE
789
	unset IMGSIZE
790

    
791
	ls -lah $IMAGES_FINAL_DIR
792
}
793

    
794
# This routine creates a ova image that contains
795
# a ovf and vmdk file. These files can be imported
796
# right into vmware or virtual box.
797
# (and many other emulation platforms)
798
# http://www.vmware.com/pdf/ovf_whitepaper_specification.pdf
799
create_ova_image() {
800
	# XXX create a .ovf php creator that you can pass:
801
	#     1. populatedSize
802
	#     2. license
803
	#     3. product name
804
	#     4. version
805
	#     5. number of network interface cards
806
	#     6. allocationUnits
807
	#     7. capacity
808
	#     8. capacityAllocationUnits
809

    
810
	LOGFILE=${BUILDER_LOGS}/ova.${TARGET}.log
811

    
812
	[ -d "${OVA_TMP}" ] \
813
		&& rm -rf ${OVA_TMP}
814

    
815
	mkdir -p ${OVA_TMP}
816

    
817
	# Prepare folder to be put in image
818
	customize_stagearea_for_image "ova"
819
	install_default_kernel ${DEFAULT_KERNEL} "no"
820

    
821
	# Fill fstab
822
	echo ">>> Installing platform specific items..." | tee -a ${LOGFILE}
823
	echo "/dev/gpt/${PRODUCT_NAME}	/	ufs		rw	0	0" > ${FINAL_CHROOT_DIR}/etc/fstab
824
	echo "/dev/gpt/swap0	none	swap	sw	0	0" >> ${FINAL_CHROOT_DIR}/etc/fstab
825

    
826
	# Create / partition
827
	echo -n ">>> Creating / partition... " | tee -a ${LOGFILE}
828
	makefs \
829
		-B little \
830
		-o label=${PRODUCT_NAME} \
831
		-s ${OVA_FIRST_PART_SIZE_IN_GB}g \
832
		${OVA_TMP}/${OVFUFS} \
833
		${FINAL_CHROOT_DIR} 2>&1 >> ${LOGFILE}
834

    
835
	if [ $? -ne 0 -o ! -f ${OVA_TMP}/${OVFUFS} ]; then
836
		if [ -f ${OVA_TMP}/${OVFUFS} ]; then
837
			rm -f ${OVA_TMP}/${OVFUFS}
838
		fi
839
		echo "Failed!" | tee -a ${LOGFILE}
840
		echo ">>> ERROR: Error creating vmdk / partition. STOPPING!" | tee -a ${LOGFILE}
841
		print_error_pfS
842
	fi
843
	echo "Done!" | tee -a ${LOGFILE}
844

    
845
	# Create raw disk
846
	echo -n ">>> Creating raw disk... " | tee -a ${LOGFILE}
847
	mkimg \
848
		-s gpt \
849
		-f raw \
850
		-b /boot/pmbr \
851
		-p freebsd-boot:=/boot/gptboot \
852
		-p freebsd-ufs/${PRODUCT_NAME}:=${OVA_TMP}/${OVFUFS} \
853
		-p freebsd-swap/swap0::${OVA_SWAP_PART_SIZE} \
854
		-o ${OVA_TMP}/${OVFRAW} 2>&1 >> ${LOGFILE}
855

    
856
	if [ $? -ne 0 -o ! -f ${OVA_TMP}/${OVFRAW} ]; then
857
		if [ -f ${OVA_TMP}/${OVFUFS} ]; then
858
			rm -f ${OVA_TMP}/${OVFUFS}
859
		fi
860
		if [ -f ${OVA_TMP}/${OVFRAW} ]; then
861
			rm -f ${OVA_TMP}/${OVFRAW}
862
		fi
863
		echo "Failed!" | tee -a ${LOGFILE}
864
		echo ">>> ERROR: Error creating temporary vmdk image. STOPPING!" | tee -a ${LOGFILE}
865
		print_error_pfS
866
	fi
867
	echo "Done!" | tee -a ${LOGFILE}
868

    
869
	# We don't need it anymore
870
	rm -f ${OVA_TMP}/${OVFUFS} >/dev/null 2>&1
871

    
872
	# Convert raw to vmdk
873
	echo -n ">>> Creating vmdk disk... " | tee -a ${LOGFILE}
874
	vmdktool -z9 -v ${OVA_TMP}/${OVFVMDK} ${OVA_TMP}/${OVFRAW}
875

    
876
	if [ $? -ne 0 -o ! -f ${OVA_TMP}/${OVFVMDK} ]; then
877
		if [ -f ${OVA_TMP}/${OVFRAW} ]; then
878
			rm -f ${OVA_TMP}/${OVFRAW}
879
		fi
880
		if [ -f ${OVA_TMP}/${OVFVMDK} ]; then
881
			rm -f ${OVA_TMP}/${OVFVMDK}
882
		fi
883
		echo "Failed!" | tee -a ${LOGFILE}
884
		echo ">>> ERROR: Error creating vmdk image. STOPPING!" | tee -a ${LOGFILE}
885
		print_error_pfS
886
	fi
887
	echo "Done!" | tee -a ${LOGFILE}
888

    
889
	rm -f ${OVA_TMP}/i${OVFRAW}
890

    
891
	ova_setup_ovf_template
892

    
893
	echo -n ">>> Writing final ova image... " | tee -a ${LOGFILE}
894
	# Create OVA file for vmware
895
	gtar -C ${OVA_TMP} -cpf ${OVAPATH} ${PRODUCT_NAME}.ovf ${OVFVMDK}
896
	echo "Done!" | tee -a ${LOGFILE}
897
	rm -f ${OVA_TMP}/${OVFVMDK} >/dev/null 2>&1
898

    
899
	echo ">>> OVA created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
900
}
901

    
902
# called from create_ova_image
903
ova_setup_ovf_template() {
904
	if [ ! -f ${OVFTEMPLATE} ]; then
905
		echo ">>> ERROR: OVF template file (${OVFTEMPLATE}) not found."
906
		print_error_pfS
907
	fi
908

    
909
	#  OperatingSystemSection (${PRODUCT_NAME}.ovf)
910
	#  42   FreeBSD 32-Bit
911
	#  78   FreeBSD 64-Bit
912
	if [ "${TARGET}" = "amd64" ]; then
913
		local _os_id="78"
914
		local _os_type="freebsd64Guest"
915
		local _os_descr="FreeBSD 64-Bit"
916
	elif [ "${TARGET}" = "i386" ]; then
917
		local _os_id="42"
918
		local _os_type="freebsdGuest"
919
		local _os_descr="FreeBSD"
920
	else
921
		echo ">>> ERROR: Platform not supported for OVA (${TARGET})"
922
		print_error_pfS
923
	fi
924

    
925
	local POPULATED_SIZE=$(du -d0 -k $FINAL_CHROOT_DIR | cut -f1)
926
	local POPULATED_SIZE_IN_BYTES=$((${POPULATED_SIZE}*1024))
927
	local VMDK_FILE_SIZE=$(stat -f "%z" ${OVA_TMP}/${OVFVMDK})
928

    
929
	sed \
930
		-e "s,%%VMDK_FILE_SIZE%%,${VMDK_FILE_SIZE},g" \
931
		-e "s,%%VMDK_DISK_CAPACITY_IN_GB%%,${VMDK_DISK_CAPACITY_IN_GB},g" \
932
		-e "s,%%POPULATED_SIZE_IN_BYTES%%,${POPULATED_SIZE_IN_BYTES},g" \
933
		-e "s,%%OS_ID%%,${_os_id},g" \
934
		-e "s,%%OS_TYPE%%,${_os_type},g" \
935
		-e "s,%%OS_DESCR%%,${_os_descr},g" \
936
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
937
		-e "s,%%PRODUCT_VERSION%%,${PRODUCT_VERSION},g" \
938
		-e "s,%%PRODUCT_URL%%,${PRODUCT_URL},g" \
939
		-e "/^%%PRODUCT_LICENSE%%/r ${BUILDER_ROOT}/license.txt" \
940
		-e "/^%%PRODUCT_LICENSE%%/d" \
941
		${OVFTEMPLATE} > ${OVA_TMP}/${PRODUCT_NAME}.ovf
942
}
943

    
944
# Cleans up previous builds
945
clean_builder() {
946
	# Clean out directories
947
	echo ">>> Cleaning up previous build environment...Please wait!"
948

    
949
	staginareas_clean_each_run
950

    
951
	if [ -d "${STAGE_CHROOT_DIR}" ]; then
952
		BASENAME=$(basename ${STAGE_CHROOT_DIR})
953
		echo -n ">>> Cleaning ${STAGE_CHROOT_DIR} ..."
954
		chflags -R noschg ${STAGE_CHROOT_DIR} 2>&1 >/dev/null
955
		rm -rf ${STAGE_CHROOT_DIR}/* 2>/dev/null
956
		echo "Done."
957
	fi
958

    
959
	if [ -z "${NO_CLEAN_FREEBSD_OBJ}" -a -d "${FREEBSD_SRC_DIR}" ]; then
960
		OBJTREE=$(env TARGET=${TARGET} TARGET_ARCH=${TARGET_ARCH} make -C ${FREEBSD_SRC_DIR} -V OBJTREE)
961
		if [ -d "${OBJTREE}" ]; then
962
			echo -n ">>> Cleaning FreeBSD objects dir staging..."
963
			echo -n "."
964
			chflags -R noschg ${OBJTREE} 2>&1 >/dev/null
965
			echo -n "."
966
			rm -rf ${OBJTREE}/*
967
			echo "Done!"
968
		fi
969
		if [ -d "${KERNEL_BUILD_PATH}" ]; then
970
			echo -n ">>> Cleaning previously built kernel stage area..."
971
			rm -rf $KERNEL_BUILD_PATH/*
972
			echo "Done!"
973
		fi
974
	fi
975
	mkdir -p $KERNEL_BUILD_PATH
976

    
977
	echo -n ">>> Cleaning previously built images..."
978
	rm -rf $IMAGES_FINAL_DIR/*
979
	rm -rf $STAGINGAREA/*
980
	echo "Done!"
981

    
982
	if [ -z "${NO_CLEAN_FREEBSD_SRC}" ]; then
983
		if [ -d "$FREEBSD_SRC_DIR" ]; then
984
			echo -n ">>> Ensuring $FREEBSD_SRC_DIR is clean..."
985
			rm -rf ${FREEBSD_SRC_DIR}
986
			echo "Done!"
987
		fi
988
	fi
989

    
990
	echo -n ">>> Cleaning previous builder logs..."
991
	if [ -d "$BUILDER_LOGS" ]; then
992
		rm -rf ${BUILDER_LOGS}
993
	fi
994
	mkdir -p ${BUILDER_LOGS}
995

    
996
	echo "Done!"
997

    
998
	echo ">>> Cleaning of builder environment has finished."
999
}
1000

    
1001
clone_directory_contents() {
1002
	if [ ! -d "$1" -o ! -d "$2" ]; then
1003
		if [ -z "${LOGFILE}" ]; then
1004
			echo ">>> ERROR: Argument $1 supplied is not a directory!"
1005
		else
1006
			echo ">>> ERROR: Argument $1 supplied is not a directory!" | tee -a ${LOGFILE}
1007
		fi
1008
		print_error_pfS
1009
	fi
1010
	echo -n ">>> Using TAR to clone $1 to $2 ..."
1011
	tar -C ${1} -c -f - . | tar -C ${2} -x -p -f -
1012
	echo "Done!"
1013
}
1014

    
1015
clone_to_staging_area() {
1016
	# Clone everything to the final staging area
1017
	echo -n ">>> Cloning everything to ${STAGE_CHROOT_DIR} staging area..."
1018
	LOGFILE=${BUILDER_LOGS}/cloning.${TARGET}.log
1019

    
1020
	tar -C ${PRODUCT_SRC} -c -f - . | \
1021
		tar -C ${STAGE_CHROOT_DIR} -x -p -f -
1022

    
1023
	if [ -f ${STAGE_CHROOT_DIR}/etc/master.passwd ]; then
1024
		chroot ${STAGE_CHROOT_DIR} cap_mkdb /etc/master.passwd
1025
		chroot ${STAGE_CHROOT_DIR} pwd_mkdb /etc/master.passwd
1026
	fi
1027
	mkdir -p ${STAGE_CHROOT_DIR}/etc/mtree
1028
	mtree -Pcp ${STAGE_CHROOT_DIR}/var > ${STAGE_CHROOT_DIR}/etc/mtree/var.dist
1029
	mtree -Pcp ${STAGE_CHROOT_DIR}/etc > ${STAGE_CHROOT_DIR}/etc/mtree/etc.dist
1030
	if [ -d ${STAGE_CHROOT_DIR}/usr/local/etc ]; then
1031
		mtree -Pcp ${STAGE_CHROOT_DIR}/usr/local/etc > ${STAGE_CHROOT_DIR}/etc/mtree/localetc.dist
1032
	fi
1033

    
1034
	## Add buildtime and lastcommit information
1035
	# This is used for detecting updates.
1036
	echo "$BUILTDATESTRING" > $STAGE_CHROOT_DIR/etc/version.buildtime
1037
	# Record last commit info if it is available.
1038
	if [ -f $SCRATCHDIR/build_commit_info.txt ]; then
1039
		cp $SCRATCHDIR/build_commit_info.txt $STAGE_CHROOT_DIR/etc/version.lastcommit
1040
	fi
1041

    
1042
	local _exclude_files="${CORE_PKG_TMP}/base_exclude_files"
1043
	sed \
1044
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
1045
		-e "s,%%FLAVOR%%,${_flavor},g" \
1046
		-e "s,%%VERSION%%,${_version},g" \
1047
		${BUILDER_TOOLS}/templates/core_pkg/base/exclude_files \
1048
		> ${_exclude_files}
1049

    
1050
	mkdir -p ${STAGE_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME} >/dev/null 2>&1
1051
	mtree \
1052
		-c \
1053
		-k uid,gid,mode,size,flags,sha256digest \
1054
		-p ${STAGE_CHROOT_DIR} \
1055
		-X ${_exclude_files} \
1056
		> ${STAGE_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/base.mtree
1057
	tar \
1058
		-C ${STAGE_CHROOT_DIR} \
1059
		-cJf ${STAGE_CHROOT_DIR}/usr/local/share/${PRODUCT_NAME}/base.txz \
1060
		-X ${_exclude_files} \
1061
		.
1062

    
1063
	core_pkg_create base "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1064
	core_pkg_create base-nanobsd "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1065
	core_pkg_create default-config "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1066

    
1067
	local DEFAULTCONF=${STAGE_CHROOT_DIR}/conf.default/config.xml
1068

    
1069
	# Change default interface names to match vmware driver
1070
	sed -i '' -e 's,em0,vmx0,' -e 's,em1,vmx1,' ${DEFAULTCONF}
1071
	core_pkg_create default-config-vmware "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1072

    
1073
	# Restore default values to be used by serial package
1074
	sed -i '' -e 's,vmx0,em0,' -e 's,vmx1,em1,' ${DEFAULTCONF}
1075

    
1076
	# Activate serial console in config.xml
1077
	# If it was there before, clear the setting to be sure we don't add it twice.
1078
	sed -i "" -e "/		<enableserial\/>/d" ${DEFAULTCONF}
1079
	# Enable serial in the config
1080
	sed -i "" -e "s/	<\/system>/		<enableserial\/>\\$(echo -e \\\n)	<\/system>/" ${DEFAULTCONF}
1081

    
1082
	echo force > ${STAGE_CHROOT_DIR}/cf/conf/enableserial_force
1083

    
1084
	core_pkg_create default-config-serial "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1085

    
1086
	rm -f ${STAGE_CHROOT_DIR}/cf/conf/enableserial_force
1087
	rm -f ${STAGE_CHROOT_DIR}/cf/conf/config.xml
1088

    
1089
	# Make sure pkg is present
1090
	pkg_bootstrap ${STAGE_CHROOT_DIR}
1091

    
1092
	echo "Done!"
1093
}
1094

    
1095
create_final_staging_area() {
1096
	if [ -z "${FINAL_CHROOT_DIR}" ]; then
1097
		echo ">>> ERROR: FINAL_CHROOT_DIR is not set, cannot continue!" | tee -a ${LOGFILE}
1098
		print_error_pfS
1099
	fi
1100

    
1101
	if [ -d "${FINAL_CHROOT_DIR}" ]; then
1102
		echo -n ">>> Previous ${FINAL_CHROOT_DIR} detected cleaning up..." | tee -a ${LOGFILE}
1103
		chflags -R noschg ${FINAL_CHROOT_DIR} 2>&1 1>/dev/null
1104
		rm -rf ${FINAL_CHROOT_DIR}/* 2>&1 1>/dev/null
1105
		echo "Done." | tee -a ${LOGFILE}
1106
	fi
1107

    
1108
	echo ">>> Preparing Final image staging area: $(LC_ALL=C date)" 2>&1 | tee -a ${LOGFILE}
1109
	echo ">>> Cloning ${STAGE_CHROOT_DIR} to ${FINAL_CHROOT_DIR}" 2>&1 | tee -a ${LOGFILE}
1110
	clone_directory_contents ${STAGE_CHROOT_DIR} ${FINAL_CHROOT_DIR}
1111

    
1112
	if [ ! -f $FINAL_CHROOT_DIR/sbin/init ]; then
1113
		echo ">>> ERROR: Something went wrong during cloning -- Please verify!" 2>&1 | tee -a ${LOGFILE}
1114
		print_error_pfS
1115
	fi
1116
}
1117

    
1118
customize_stagearea_for_image() {
1119
	# Prepare final stage area
1120
	create_final_staging_area
1121

    
1122
	if [ "${1}" = "nanobsd" -o \
1123
	     "${1}" = "nanobsd-vga" ]; then
1124

    
1125
		mkdir -p ${FINAL_CHROOT_DIR}/root/var/db \
1126
			 ${FINAL_CHROOT_DIR}/root/var/cache \
1127
			 ${FINAL_CHROOT_DIR}/var/db/pkg \
1128
			 ${FINAL_CHROOT_DIR}/var/cache/pkg
1129
		mv -f ${FINAL_CHROOT_DIR}/var/db/pkg ${FINAL_CHROOT_DIR}/root/var/db
1130
		mv -f ${FINAL_CHROOT_DIR}/var/cache/pkg ${FINAL_CHROOT_DIR}/root/var/cache
1131
		ln -sf ../../root/var/db/pkg ${FINAL_CHROOT_DIR}/var/db/pkg
1132
		ln -sf ../../root/var/cache/pkg ${FINAL_CHROOT_DIR}/var/cache/pkg
1133

    
1134
		pkg_chroot_add ${FINAL_CHROOT_DIR} base-nanobsd
1135
	else
1136
		pkg_chroot_add ${FINAL_CHROOT_DIR} base
1137
	fi
1138

    
1139
	if [ "${1}" = "iso" -o \
1140
	     "${1}" = "memstick" -o \
1141
	     "${1}" = "memstickserial" -o \
1142
	     "${1}" = "memstickadi" ]; then
1143
		install_bsdinstaller
1144
		mkdir -p ${FINAL_CHROOT_DIR}/pkgs
1145
		cp ${CORE_PKG_PATH}/All/*default-config*.txz ${FINAL_CHROOT_DIR}/pkgs
1146
	fi
1147

    
1148
	if [ "${1}" = "nanobsd" -o \
1149
	     "${1}" = "memstickserial" -o \
1150
	     "${1}" = "memstickadi" ]; then
1151
		pkg_chroot_add ${FINAL_CHROOT_DIR} default-config-serial
1152
	elif [ "${1}" = "ova" ]; then
1153
		pkg_chroot_add ${FINAL_CHROOT_DIR} default-config-vmware
1154
	else
1155
		pkg_chroot_add ${FINAL_CHROOT_DIR} default-config
1156
	fi
1157
}
1158

    
1159
create_distribution_tarball() {
1160
	mkdir -p ${FINAL_CHROOT_DIR}/install
1161

    
1162
	tar -C ${FINAL_CHROOT_DIR} --exclude ./install --exclude ./pkgs -cJf ${FINAL_CHROOT_DIR}/install/${PRODUCT_NAME}.txz .
1163
}
1164

    
1165
create_iso_image() {
1166
	LOGFILE=${BUILDER_LOGS}/isoimage.${TARGET}
1167
	echo ">>> Building bootable ISO image for ${TARGET}" | tee -a ${LOGFILE}
1168
	if [ -z "${DEFAULT_KERNEL}" ]; then
1169
		echo ">>> ERROR: Could not identify DEFAULT_KERNEL to install on image!" | tee -a ${LOGFILE}
1170
		print_error_pfS
1171
	fi
1172

    
1173
	customize_stagearea_for_image "iso"
1174
	install_default_kernel ${DEFAULT_KERNEL}
1175

    
1176
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1177

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

    
1181
	# This check is for supporting create memstick/ova images
1182
	echo -n ">>> Running command: script -aq $LOGFILE makefs -t cd9660 -o bootimage=\"i386;${FINAL_CHROOT_DIR}/boot/cdboot \"-o no-emul-boot -o rockridge " | tee -a ${LOGFILE}
1183
	echo "-o label=${FSLABEL} -o publisher=\"${PRODUCT_NAME} project.\" $ISOPATH ${FINAL_CHROOT_DIR}" | tee -a ${LOGFILE}
1184

    
1185
	create_distribution_tarball
1186

    
1187
	# Remove /rescue from iso since cd9660 cannot deal with hardlinks
1188
	rm -rf ${FINAL_CHROOT_DIR}/rescue
1189

    
1190
	makefs -t cd9660 -o bootimage="i386;${FINAL_CHROOT_DIR}/boot/cdboot" -o no-emul-boot -o rockridge \
1191
		-o label=${FSLABEL} -o publisher="${PRODUCT_NAME} project." $ISOPATH ${FINAL_CHROOT_DIR} 2>&1 >> ${LOGFILE}
1192
	if [ $? -ne 0 -o ! -f $ISOPATH ]; then
1193
		if [ -f ${ISOPATH} ]; then
1194
			rm -f $ISOPATH
1195
		fi
1196
		echo ">>> ERROR: Something wrong happened during ISO image creation. STOPPING!" | tee -a ${LOGFILE}
1197
		print_error_pfS
1198
	fi
1199
	gzip -qf $ISOPATH &
1200
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1201

    
1202
	echo ">>> ISO created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1203
}
1204

    
1205
create_memstick_image() {
1206

    
1207
	LOGFILE=${BUILDER_LOGS}/memstick.${TARGET}
1208
	if [ "${MEMSTICKPATH}" = "" ]; then
1209
		echo ">>> MEMSTICKPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1210
		return
1211
	fi
1212

    
1213
	if [ ! -d ${FINAL_CHROOT_DIR}/boot ]; then
1214
		customize_stagearea_for_image "memstick"
1215
		install_default_kernel ${DEFAULT_KERNEL}
1216
	fi
1217

    
1218
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1219

    
1220
	echo ">>> Creating memstick to ${MEMSTICKPATH}." 2>&1 | tee -a ${LOGFILE}
1221
	echo "/dev/ufs/${PRODUCT_NAME} / ufs ro 0 0" > ${FINAL_CHROOT_DIR}/etc/fstab
1222
	echo "kern.cam.boot_delay=10000" >> ${FINAL_CHROOT_DIR}/boot/loader.conf.local
1223

    
1224
	create_distribution_tarball
1225

    
1226
	makefs -B little -o label=${PRODUCT_NAME} ${MEMSTICKPATH} ${FINAL_CHROOT_DIR}
1227
	if [ $? -ne 0 ]; then
1228
		if [ -f ${MEMSTICKPATH} ]; then
1229
			rm -f $MEMSTICKPATH
1230
		fi
1231
		echo ">>> ERROR: Something wrong happened during MEMSTICK image creation. STOPPING!" | tee -a ${LOGFILE}
1232
		print_error_pfS
1233
	fi
1234
	MD=$(mdconfig -a -t vnode -f $MEMSTICKPATH)
1235
	# Just in case
1236
	trap "mdconfig -d -u ${MD}" 1 2 15 EXIT
1237
	gpart create -s BSD ${MD} 2>&1 >> ${LOGFILE}|
1238
	gpart bootcode -b ${FINAL_CHROOT_DIR}/boot/boot ${MD} 2>&1 >> ${LOGFILE}
1239
	gpart add -t freebsd-ufs ${MD} 2>&1 >> ${LOGFILE}
1240
	trap "-" 1 2 15 EXIT
1241
	mdconfig -d -u ${MD} 2>&1 | tee -a ${LOGFILE}
1242
	gzip -qf $MEMSTICKPATH &
1243
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1244

    
1245
	echo ">>> MEMSTICK created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1246
}
1247

    
1248
create_memstick_serial_image() {
1249
	LOGFILE=${BUILDER_LOGS}/memstickserial.${TARGET}
1250
	if [ "${MEMSTICKSERIALPATH}" = "" ]; then
1251
		echo ">>> MEMSTICKSERIALPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1252
		return
1253
	fi
1254

    
1255
	if [ ! -d ${FINAL_CHROOT_DIR}/boot ]; then
1256
		customize_stagearea_for_image "memstickserial"
1257
		install_default_kernel ${DEFAULT_KERNEL}
1258
	fi
1259

    
1260
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1261

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

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

    
1267
	BOOTCONF=${FINAL_CHROOT_DIR}/boot.config
1268
	LOADERCONF=${FINAL_CHROOT_DIR}/boot/loader.conf
1269

    
1270
	echo ">>> Activating serial console..." 2>&1 | tee -a ${LOGFILE}
1271
	# Activate serial console in boot.config
1272
	if [ -f ${BOOTCONF} ]; then
1273
		sed -i "" '/-D$/d' ${BOOTCONF}
1274
	fi
1275
	echo "-S115200 -D" >> ${BOOTCONF}
1276

    
1277
	# Remove old console options if present.
1278
	[ -f "${LOADERCONF}" ] \
1279
		&& sed -i "" -Ee "/(console|boot_multicons|boot_serial)/d" ${LOADERCONF}
1280
	# Activate serial console+video console in loader.conf
1281
	echo 'boot_multicons="YES"' >>  ${LOADERCONF}
1282
	echo 'boot_serial="YES"' >> ${LOADERCONF}
1283
	echo 'console="comconsole,vidconsole"' >> ${LOADERCONF}
1284
	echo 'comconsole_speed="115200"' >> ${LOADERCONF}
1285

    
1286
	create_distribution_tarball
1287

    
1288
	makefs -B little -o label=${PRODUCT_NAME} ${MEMSTICKSERIALPATH} ${FINAL_CHROOT_DIR}
1289
	if [ $? -ne 0 ]; then
1290
		if [ -f ${MEMSTICKSERIALPATH} ]; then
1291
			rm -f $MEMSTICKSERIALPATH
1292
		fi
1293
		echo ">>> ERROR: Something wrong happened during MEMSTICKSERIAL image creation. STOPPING!" | tee -a ${LOGFILE}
1294
		print_error_pfS
1295
	fi
1296
	MD=$(mdconfig -a -t vnode -f $MEMSTICKSERIALPATH)
1297
	# Just in case
1298
	trap "mdconfig -d -u ${MD}" 1 2 15 EXIT
1299
	gpart create -s BSD ${MD} 2>&1 >> ${LOGFILE}
1300
	gpart bootcode -b ${FINAL_CHROOT_DIR}/boot/boot ${MD} 2>&1 >> ${LOGFILE}
1301
	gpart add -t freebsd-ufs ${MD} 2>&1 >> ${LOGFILE}
1302
	trap "-" 1 2 15 EXIT
1303
	mdconfig -d -u ${MD} 2>&1 >> ${LOGFILE}
1304
	gzip -qf $MEMSTICKSERIALPATH &
1305
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1306

    
1307
	echo ">>> MEMSTICKSERIAL created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1308
}
1309

    
1310
create_memstick_adi_image() {
1311
	LOGFILE=${BUILDER_LOGS}/memstickadi${TARGET}
1312
	if [ "${MEMSTICKADIPATH}" = "" ]; then
1313
		echo ">>> MEMSTICKADIPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1314
		return
1315
	fi
1316

    
1317
	if [ ! -d ${FINAL_CHROOT_DIR}/boot ]; then
1318
		customize_stagearea_for_image "memstickadi"
1319
		install_default_kernel ${DEFAULT_KERNEL}
1320
	fi
1321

    
1322
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1323

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

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

    
1329
	BOOTCONF=${FINAL_CHROOT_DIR}/boot.config
1330
	LOADERCONF=${FINAL_CHROOT_DIR}/boot/loader.conf
1331

    
1332
	echo ">>> Activating serial console..." 2>&1 | tee -a ${LOGFILE}
1333
	# Activate serial console in boot.config
1334
	if [ -f ${BOOTCONF} ]; then
1335
		sed -i "" '/-[Dh]$/d' ${BOOTCONF}
1336
	fi
1337
	echo "-S115200 -h" >> ${BOOTCONF}
1338

    
1339
	# Remove old console options if present.
1340
	[ -f "${LOADERCONF}" ] \
1341
		&& sed -i "" -Ee "/(console|boot_multicons|boot_serial|hint.uart)/d" ${LOADERCONF}
1342
	# Activate serial console+video console in loader.conf
1343
	echo 'boot_serial="YES"' >> ${LOADERCONF}
1344
	echo 'console="comconsole"' >> ${LOADERCONF}
1345
	echo 'comconsole_speed="115200"' >> ${LOADERCONF}
1346
	echo 'comconsole_port="0x2F8"' >> ${LOADERCONF}
1347
	echo 'hint.uart.0.flags="0x00"' >> ${LOADERCONF}
1348
	echo 'hint.uart.1.flags="0x10"' >> ${LOADERCONF}
1349

    
1350
	create_distribution_tarball
1351

    
1352
	makefs -B little -o label=${PRODUCT_NAME} ${MEMSTICKADIPATH} ${FINAL_CHROOT_DIR}
1353
	if [ $? -ne 0 ]; then
1354
		if [ -f ${MEMSTICKADIPATH} ]; then
1355
			rm -f $MEMSTICKADIPATH
1356
		fi
1357
		echo ">>> ERROR: Something wrong happened during MEMSTICKADI image creation. STOPPING!" | tee -a ${LOGFILE}
1358
		print_error_pfS
1359
	fi
1360
	MD=$(mdconfig -a -t vnode -f $MEMSTICKADIPATH)
1361
	# Just in case
1362
	trap "mdconfig -d -u ${MD}" 1 2 15 EXIT
1363
	gpart create -s BSD ${MD} 2>&1 >> ${LOGFILE}
1364
	gpart bootcode -b ${FINAL_CHROOT_DIR}/boot/boot ${MD} 2>&1 >> ${LOGFILE}
1365
	gpart add -t freebsd-ufs ${MD} 2>&1 >> ${LOGFILE}
1366
	trap "-" 1 2 15 EXIT
1367
	mdconfig -d -u ${MD} 2>&1 >> ${LOGFILE}
1368
	gzip -qf $MEMSTICKADIPATH &
1369
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1370

    
1371
	echo ">>> MEMSTICKADI created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1372
}
1373

    
1374
# Create pkg conf on desired place with desired arch/branch
1375
setup_pkg_repo() {
1376
	if [ -z "${4}" ]; then
1377
		return
1378
	fi
1379

    
1380
	local _target="${1}"
1381
	local _arch="${2}"
1382
	local _target_arch="${3}"
1383
	local _branch="${4}"
1384

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

    
1387
	sed \
1388
		-e "s/%%ARCH%%/${_arch}_${_target_arch}/" \
1389
		-e "s/%%GIT_REPO_BRANCH_OR_TAG%%/${_branch}/g" \
1390
		-e "s,%%PKG_REPO_SERVER%%,${PKG_REPO_SERVER},g" \
1391
		-e "s/%%PRODUCT_NAME%%/${PRODUCT_NAME}/g" \
1392
		${FREEBSD_SRC_DIR}/release/pkg_repos/${PRODUCT_NAME}.conf.template \
1393
		> ${_target}
1394
}
1395

    
1396
# This routine ensures any ports / binaries that the builder
1397
# system needs are on disk and ready for execution.
1398
builder_setup() {
1399
	# If Product-builder is already installed, just leave
1400
	if pkg info -e -q ${PRODUCT_NAME}-builder; then
1401
		return
1402
	fi
1403

    
1404
	if [ ! -f /usr/local/etc/pkg/repos/${PRODUCT_NAME}.conf ]; then
1405
		[ -d /usr/local/etc/pkg/repos ] \
1406
			|| mkdir -p /usr/local/etc/pkg/repos
1407

    
1408
		update_freebsd_sources
1409

    
1410
		local _arch=$(uname -m)
1411
		setup_pkg_repo /usr/local/etc/pkg/repos/${PRODUCT_NAME}.conf ${_arch} ${_arch} ${PKG_REPO_CONF_BRANCH}
1412
	fi
1413

    
1414
	pkg install ${PRODUCT_NAME}-builder
1415
}
1416

    
1417
# Updates FreeBSD sources
1418
update_freebsd_sources() {
1419
	if [ "${1}" = "full" ]; then
1420
		local _full=1
1421
		local _clone_params=""
1422
	else
1423
		local _full=0
1424
		local _clone_params="--depth 1 --single-branch"
1425
	fi
1426

    
1427
	if [ ! -d "${FREEBSD_SRC_DIR}" ]; then
1428
		mkdir -p ${FREEBSD_SRC_DIR}
1429
	fi
1430

    
1431
	if [ -n "${NO_BUILDWORLD}" -a -n "${NO_BUILDKERNEL}" ]; then
1432
		echo ">>> NO_BUILDWORLD and NO_BUILDKERNEL set, skipping update of freebsd sources" | tee -a ${LOGFILE}
1433
		return
1434
	fi
1435

    
1436
	echo -n ">>> Obtaining FreeBSD sources ${FREEBSD_BRANCH}..."
1437
	local _FREEBSD_BRANCH=${FREEBSD_BRANCH:-"devel"}
1438
	local _CLONE=1
1439

    
1440
	if [ -d "${FREEBSD_SRC_DIR}/.git" ]; then
1441
		CUR_BRANCH=$(cd ${FREEBSD_SRC_DIR} && git branch | grep '^\*' | cut -d' ' -f2)
1442
		if [ ${_full} -eq 0 -a "${CUR_BRANCH}" = "${_FREEBSD_BRANCH}" ]; then
1443
			_CLONE=0
1444
			( cd ${FREEBSD_SRC_DIR} && git clean -fd; git fetch origin; git reset --hard origin/${_FREEBSD_BRANCH} ) 2>&1 | grep -C3 -i -E 'error|fatal'
1445
		else
1446
			rm -rf ${FREEBSD_SRC_DIR}
1447
		fi
1448
	fi
1449

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

    
1454
	if [ ! -d "${FREEBSD_SRC_DIR}/.git" ]; then
1455
		echo ">>> ERROR: It was not possible to clone FreeBSD src repo"
1456
		print_error_pfS
1457
	fi
1458

    
1459
	if [ -n "${GIT_FREEBSD_COSHA1}" ]; then
1460
		( cd ${FREEBSD_SRC_DIR} && git checkout ${GIT_FREEBSD_COSHA1} ) 2>&1 | grep -C3 -i -E 'error|fatal'
1461
	fi
1462
	echo "Done!"
1463
}
1464

    
1465
pkg_chroot() {
1466
	local _root="${1}"
1467
	shift
1468

    
1469
	if [ $# -eq 0 ]; then
1470
		return -1
1471
	fi
1472

    
1473
	if [ -z "${_root}" -o "${_root}" = "/" -o ! -d "${_root}" ]; then
1474
		return -1
1475
	fi
1476

    
1477
	mkdir -p \
1478
		${SCRATCHDIR}/pkg_cache \
1479
		${_root}/var/cache/pkg \
1480
		${_root}/dev
1481

    
1482
	/sbin/mount -t nullfs ${SCRATCHDIR}/pkg_cache ${_root}/var/cache/pkg
1483
	/sbin/mount -t devfs devfs ${_root}/dev
1484
	cp -f /etc/resolv.conf ${_root}/etc/resolv.conf
1485
	touch ${BUILDER_LOGS}/install_pkg_install_ports.txt
1486
	script -aq ${BUILDER_LOGS}/install_pkg_install_ports.txt pkg -c ${_root} $@ >/dev/null 2>&1
1487
	rm -f ${_root}/etc/resolv.conf
1488
	/sbin/umount -f ${_root}/dev
1489
	/sbin/umount -f ${_root}/var/cache/pkg
1490
}
1491

    
1492

    
1493
pkg_chroot_add() {
1494
	if [ -z "${1}" -o -z "${2}" ]; then
1495
		return 1
1496
	fi
1497

    
1498
	local _target="${1}"
1499
	local _pkg="$(get_pkg_name ${2}).txz"
1500

    
1501
	if [ ! -d "${_target}" ]; then
1502
		echo ">>> ERROR: Target dir ${_target} not found"
1503
		print_error_pfS
1504
	fi
1505

    
1506
	if [ ! -f ${CORE_PKG_PATH}/All/${_pkg} ]; then
1507
		echo ">>> ERROR: Package ${_pkg} not found"
1508
		print_error_pfS
1509
	fi
1510

    
1511
	cp ${CORE_PKG_PATH}/All/${_pkg} ${_target}
1512
	pkg_chroot ${_target} add /${_pkg}
1513
	rm -f ${_target}/${_pkg}
1514
}
1515

    
1516
pkg_bootstrap() {
1517
	local _root=${1:-"${STAGE_CHROOT_DIR}"}
1518

    
1519
	setup_pkg_repo ${_root}/usr/local/etc/pkg/repos/${PRODUCT_NAME}.conf ${TARGET} ${TARGET_ARCH} ${PKG_REPO_CONF_BRANCH}
1520

    
1521
	pkg_chroot ${_root} bootstrap -f
1522
}
1523

    
1524
# This routine assists with installing various
1525
# freebsd ports files into the pfsenese-fs staging
1526
# area.
1527
install_pkg_install_ports() {
1528
	local MAIN_PKG="${1}"
1529

    
1530
	if [ -z "${MAIN_PKG}" ]; then
1531
		MAIN_PKG=${PRODUCT_NAME}
1532
	fi
1533

    
1534
	echo ">>> Installing pkg repository in chroot (${STAGE_CHROOT_DIR})..."
1535

    
1536
	[ -d ${STAGE_CHROOT_DIR}/var/cache/pkg ] || \
1537
		mkdir -p ${STAGE_CHROOT_DIR}/var/cache/pkg
1538

    
1539
	[ -d ${SCRATCHDIR}/pkg_cache ] || \
1540
		mkdir -p ${SCRATCHDIR}/pkg_cache
1541

    
1542
	echo ">>> Installing built ports (packages) in chroot (${STAGE_CHROOT_DIR})... (starting)"
1543
	pkg_chroot ${STAGE_CHROOT_DIR} install ${MAIN_PKG} ${custom_package_list}
1544
	pkg_chroot ${STAGE_CHROOT_DIR} autoremove
1545
	echo ">>> Installing built ports (packages) in chroot (${STAGE_CHROOT_DIR})... (finshied)"
1546
}
1547

    
1548
install_bsdinstaller() {
1549
	echo ">>> Installing BSDInstaller in chroot (${FINAL_CHROOT_DIR})... (starting)"
1550
	pkg_chroot ${FINAL_CHROOT_DIR} install -f bsdinstaller
1551
	sed -i '' -e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME}," \
1552
		  -e "s,%%PRODUCT_VERSION%%,${PRODUCT_VERSION}," \
1553
		  -e "s,%%ARCH%%,${TARGET}," \
1554
		  ${FINAL_CHROOT_DIR}/usr/local/share/dfuibe_lua/conf/pfSense.lua \
1555
		  ${FINAL_CHROOT_DIR}/usr/local/share/dfuibe_lua/conf/pfSense_rescue.lua
1556
	echo ">>> Installing BSDInstaller in chroot (${FINAL_CHROOT_DIR})... (finished)"
1557
}
1558

    
1559
staginareas_clean_each_run() {
1560
	echo -n ">>> Cleaning build directories: "
1561
	if [ -d "${FINAL_CHROOT_DIR}" ]; then
1562
		BASENAME=$(basename ${FINAL_CHROOT_DIR})
1563
		echo -n "$BASENAME "
1564
		chflags -R noschg ${FINAL_CHROOT_DIR} 2>&1 >/dev/null
1565
		rm -rf ${FINAL_CHROOT_DIR}/* 2>/dev/null
1566
	fi
1567
	echo "Done!"
1568
}
1569

    
1570
# Imported from FreeSBIE
1571
buildkernel() {
1572
	if [ -n "${NO_BUILDKERNEL}" ]; then
1573
		echo ">>> NO_BUILDKERNEL set, skipping build" | tee -a ${LOGFILE}
1574
		return
1575
	fi
1576

    
1577
	if [ -z "${KERNCONF}" ]; then
1578
		echo ">>> ERROR: No kernel configuration defined probably this is not what you want! STOPPING!" | tee -a ${LOGFILE}
1579
		print_error_pfS
1580
	fi
1581

    
1582
	if [ -n "${KERNELCONF}" ]; then
1583
		export KERNCONFDIR=$(dirname ${KERNELCONF})
1584
		export KERNCONF=$(basename ${KERNELCONF})
1585
	fi
1586

    
1587
	SRCCONFBASENAME=$(basename ${SRC_CONF})
1588
	echo ">>> KERNCONFDIR: ${KERNCONFDIR}"
1589
	echo ">>> ARCH:        ${TARGET}"
1590
	echo ">>> SRC_CONF:    ${SRCCONFBASENAME}"
1591

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

    
1597
# Imported from FreeSBIE
1598
installkernel() {
1599
	if [ -z "${KERNCONF}" ]; then
1600
		echo ">>> ERROR: No kernel configuration defined probably this is not what you want! STOPPING!" | tee -a ${LOGFILE}
1601
		print_error_pfS
1602
	fi
1603

    
1604
	if [ -n "${KERNELCONF}" ]; then
1605
		export KERNCONFDIR=$(dirname ${KERNELCONF})
1606
		export KERNCONF=$(basename ${KERNELCONF})
1607
	fi
1608

    
1609
	mkdir -p ${STAGE_CHROOT_DIR}/boot
1610
	makeargs="${MAKEJ_KERNEL} SRCCONF=${SRC_CONF} __MAKE_CONF=${MAKE_CONF} TARGET_ARCH=${TARGET_ARCH} TARGET=${TARGET} DESTDIR=${KERNEL_DESTDIR}"
1611
	echo ">>> Builder is running the command: script -aq $LOGFILE make ${makeargs} installkernel KERNCONF=${KERNCONF}"  | tee -a $LOGFILE
1612
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} installkernel KERNCONF=${KERNCONF} || print_error_pfS;) | egrep '^>>>'
1613
	gzip -f9 $KERNEL_DESTDIR/boot/kernel/kernel
1614
}
1615

    
1616
# Launch is ran first to setup a few variables that we need
1617
# Imported from FreeSBIE
1618
launch() {
1619
	if [ "$(id -u)" != "0" ]; then
1620
		echo "Sorry, this must be done as root."
1621
	fi
1622

    
1623
	echo ">>> Operation $0 has started at $(date)"
1624
}
1625

    
1626
finish() {
1627
	echo ">>> Operation $0 has ended at $(date)"
1628
}
1629

    
1630
pkg_repo_rsync() {
1631
	local _repo_path="${1}"
1632

    
1633
	if [ -n "${DO_NOT_UPLOAD}" -o -z "${_repo_path}" -o ! -d "${_repo_path}" ]; then
1634
		return
1635
	fi
1636

    
1637
	if [ -z "${LOGFILE}" ]; then
1638
		local _logfile="/dev/null"
1639
	else
1640
		local _logfile="${LOGFILE}"
1641
	fi
1642

    
1643
	echo -n ">>> Sending updated repository to ${PKG_RSYNC_HOSTNAME}... " | tee -a ${_logfile}
1644
	if script -aq ${_logfile} rsync -ave "ssh -p ${PKG_RSYNC_SSH_PORT}" \
1645
		--timeout=60 --delete-delay ${_repo_path} \
1646
		${PKG_RSYNC_USERNAME}@${PKG_RSYNC_HOSTNAME}:${PKG_RSYNC_DESTDIR} >/dev/null 2>&1
1647
	then
1648
		echo "Done!" | tee -a ${_logfile}
1649
	else
1650
		echo "Failed!" | tee -a ${_logfile}
1651
		echo ">>> ERROR: An error occurred sending repo to remote hostname"
1652
		print_error_pfS
1653
	fi
1654
}
1655

    
1656
poudriere_create_patch() {
1657
	local _jail_patch="${SCRATCHDIR}/poudriere_jail.${GIT_REPO_BRANCH_OR_TAG}.patch"
1658

    
1659
	if [ -z "${FREEBSD_PARENT_BRANCH}" ]; then
1660
		echo ">>> ERROR: FREEBSD_PARENT_BRANCH is not set"
1661
	fi
1662

    
1663
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1664

    
1665
	# Get FreeBSD source and apply patches
1666
	update_freebsd_sources full
1667

    
1668
	[ -f "${_jail_patch}" ] && \
1669
		rm -f "${_jail_patch}"
1670

    
1671
	# Create a big patch with all our changes to use on jail
1672
	( \
1673
		cd ${FREEBSD_SRC_DIR} && \
1674
		git diff $(git merge-base origin/${FREEBSD_PARENT_BRANCH} ${FREEBSD_BRANCH}) > ${_jail_patch}
1675
	) >/dev/null 2>&1
1676

    
1677
	# Check if patch was created
1678
	if [ ! -s "${_jail_patch}" ]; then
1679
		echo ">>> ERROR: Patch does not exist or is empty, aborting..." | tee -a ${LOGFILE}
1680
		print_error_pfS
1681
	fi
1682
}
1683

    
1684
poudriere_possible_archs() {
1685
	local _arch=$(uname -m)
1686
	local _archs="i386.i386"
1687

    
1688
	# IF host is amd64, we'll create both repos, and if possible armv6
1689
	if [ "${_arch}" = "amd64" ]; then
1690
		_archs="amd64.amd64 ${_archs}"
1691

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

    
1696
			if binmiscctl lookup armv6 >/dev/null 2>&1; then
1697
				_archs="${_archs} arm.armv6"
1698
			fi
1699
		fi
1700
	fi
1701

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

    
1720
	echo ${_archs}
1721
}
1722

    
1723
poudriere_jail_name() {
1724
	local _jail_arch="${1}"
1725

    
1726
	if [ -z "${_jail_arch}" ]; then
1727
		return 1
1728
	fi
1729

    
1730
	# Poudriere doesn't like periods in jail names
1731
	_jail_arch=$(echo "${_jail_arch}" | tr '.' '_')
1732

    
1733
	echo "${PRODUCT_NAME}_${GIT_REPO_BRANCH_OR_TAG}_${_jail_arch}"
1734
}
1735

    
1736
poudriere_create_ports_tree() {
1737
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1738

    
1739
	if ! poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
1740
		local _branch=""
1741
		if [ -z "${POUDRIERE_PORTS_GIT_URL}" ]; then
1742
			echo ">>> ERROR: POUDRIERE_PORTS_GIT_URL is not defined"
1743
			print_error_pfS
1744
		fi
1745
		if [ -n "${POUDRIERE_PORTS_GIT_BRANCH}" ]; then
1746
			_branch="-B ${POUDRIERE_PORTS_GIT_BRANCH}"
1747
		fi
1748
		echo -n ">>> Creating poudriere ports tree, it may take some time... " | tee -a ${LOGFILE}
1749
		if ! script -aq ${LOGFILE} poudriere ports -c -p "${POUDRIERE_PORTS_NAME}" -m git ${_branch} >/dev/null 2>&1; then
1750
			echo "" | tee -a ${LOGFILE}
1751
			echo ">>> ERROR: Error creating poudriere ports tree, aborting..." | tee -a ${LOGFILE}
1752
			print_error_pfS
1753
		fi
1754
		echo "Done!" | tee -a ${LOGFILE}
1755
	fi
1756
}
1757

    
1758
poudriere_init() {
1759
	local _error=0
1760
	local _archs=$(poudriere_possible_archs)
1761
	local _jail_patch="${SCRATCHDIR}/poudriere_jail.${GIT_REPO_BRANCH_OR_TAG}.patch"
1762

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

    
1765
	# Sanity checks
1766
	if [ -z "${ZFS_TANK}" ]; then
1767
		echo ">>> ERROR: \$ZFS_TANK is empty" | tee -a ${LOGFILE}
1768
		error=1
1769
	fi
1770

    
1771
	if [ -z "${ZFS_ROOT}" ]; then
1772
		echo ">>> ERROR: \$ZFS_ROOT is empty" | tee -a ${LOGFILE}
1773
		error=1
1774
	fi
1775

    
1776
	if [ -z "${POUDRIERE_PORTS_NAME}" ]; then
1777
		echo ">>> ERROR: \$POUDRIERE_PORTS_NAME is empty" | tee -a ${LOGFILE}
1778
		error=1
1779
	fi
1780

    
1781
	if [ ${_error} -eq 1 ]; then
1782
		print_error_pfS
1783
	fi
1784

    
1785
	# Check if zpool exists
1786
	if ! zpool list ${ZFS_TANK} >/dev/null 2>&1; then
1787
		echo ">>> ERROR: ZFS tank ${ZFS_TANK} not found, please create it and try again..." | tee -a ${LOGFILE}
1788
		print_error_pfS
1789
	fi
1790

    
1791
	# Check if zfs rootfs exists
1792
	if ! zfs list ${ZFS_TANK}${ZFS_ROOT} >/dev/null 2>&1; then
1793
		echo ">>> ERROR: ZFS filesystem ${ZFS_TANK}${ZFS_ROOT} not found, please create it and try again..." | tee -a ${LOGFILE}
1794
		print_error_pfS
1795
	fi
1796

    
1797
	# Make sure poudriere is installed
1798
	if ! pkg info --quiet poudriere; then
1799
		echo ">>> Installing poudriere..." | tee -a ${LOGFILE}
1800
		if ! pkg install poudriere >/dev/null 2>&1; then
1801
			echo ">>> ERROR: poudriere was not installed, aborting..." | tee -a ${LOGFILE}
1802
			print_error_pfS
1803
		fi
1804
	fi
1805

    
1806
	# Create poudriere.conf
1807
	if [ -z "${POUDRIERE_PORTS_GIT_URL}" ]; then
1808
		echo ">>> ERROR: POUDRIERE_PORTS_GIT_URL is not defined"
1809
		print_error_pfS
1810
	fi
1811
	echo ">>> Creating poudriere.conf" | tee -a ${LOGFILE}
1812
	cat <<EOF >/usr/local/etc/poudriere.conf
1813
ZPOOL=${ZFS_TANK}
1814
ZROOTFS=${ZFS_ROOT}
1815
RESOLV_CONF=/etc/resolv.conf
1816
BASEFS=/usr/local/poudriere
1817
USE_PORTLINT=no
1818
USE_TMPFS=yes
1819
NOLINUX=yes
1820
DISTFILES_CACHE=/usr/ports/distfiles
1821
CHECK_CHANGED_OPTIONS=yes
1822
CHECK_CHANGED_DEPS=yes
1823
ATOMIC_PACKAGE_REPOSITORY=yes
1824
COMMIT_PACKAGES_ON_FAILURE=no
1825
GIT_URL="${POUDRIERE_PORTS_GIT_URL}"
1826
EOF
1827

    
1828
	# Remove old jails
1829
	for jail_arch in ${_archs}; do
1830
		jail_name=$(poudriere_jail_name ${jail_arch})
1831

    
1832
		if poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
1833
			echo ">>> Poudriere jail ${jail_name} already exists, deleting it..." | tee -a ${LOGFILE}
1834
			poudriere jail -d -j "${jail_name}" >/dev/null 2>&1
1835
		fi
1836
	done
1837

    
1838
	# Remove old ports tree
1839
	if poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
1840
		echo ">>> Poudriere ports tree ${POUDRIERE_PORTS_NAME} already exists, deleting it..." | tee -a ${LOGFILE}
1841
		poudriere ports -d -p "${POUDRIERE_PORTS_NAME}"
1842
	fi
1843

    
1844
	poudriere_create_patch
1845

    
1846
	local native_xtools=""
1847
	# Now we are ready to create jails
1848
	for jail_arch in ${_archs}; do
1849
		jail_name=$(poudriere_jail_name ${jail_arch})
1850

    
1851
		if [ "${jail_arch}" = "arm.armv6" ]; then
1852
			native_xtools="-x"
1853
		else
1854
			native_xtools=""
1855
		fi
1856

    
1857
		echo -n ">>> Creating jail ${jail_name}, it may take some time... " | tee -a ${LOGFILE}
1858
		# XXX: Change -m to git when it's available in poudriere
1859
		if ! script -aq ${LOGFILE} poudriere jail -c -j "${jail_name}" -v ${FREEBSD_PARENT_BRANCH} \
1860
				-a ${jail_arch} -m svn -P ${_jail_patch} ${native_xtools} >/dev/null 2>&1; then
1861
			echo "" | tee -a ${LOGFILE}
1862
			echo ">>> ERROR: Error creating jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
1863
			print_error_pfS
1864
		fi
1865
		echo "Done!" | tee -a ${LOGFILE}
1866
	done
1867

    
1868
	poudriere_create_ports_tree
1869

    
1870
	echo ">>> Poudriere is now configured!" | tee -a ${LOGFILE}
1871
}
1872

    
1873
poudriere_update_jails() {
1874
	local _archs=$(poudriere_possible_archs)
1875
	local _jail_patch="${SCRATCHDIR}/poudriere_jail.${GIT_REPO_BRANCH_OR_TAG}.patch"
1876

    
1877
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1878

    
1879
	poudriere_create_patch
1880

    
1881
	local native_xtools=""
1882
	for jail_arch in ${_archs}; do
1883
		jail_name=$(poudriere_jail_name ${jail_arch})
1884

    
1885
		if ! poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
1886
			echo ">>> Poudriere jail ${jail_name} not found, skipping..." | tee -a ${LOGFILE}
1887
			continue
1888
		fi
1889

    
1890
		if [ "${jail_arch}" = "arm.armv6" ]; then
1891
			native_xtools="-x"
1892
		else
1893
			native_xtools=""
1894
		fi
1895

    
1896
		echo -n ">>> Updating jail ${jail_name}, it may take some time... " | tee -a ${LOGFILE}
1897
		if ! script -aq ${LOGFILE} poudriere jail -u -j "${jail_name}" -P ${_jail_patch} ${native_xtools} >/dev/null 2>&1; then
1898
			echo "" | tee -a ${LOGFILE}
1899
			echo ">>> ERROR: Error updating jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
1900
			print_error_pfS
1901
		fi
1902
		echo "Done!" | tee -a ${LOGFILE}
1903
	done
1904
}
1905

    
1906
poudriere_update_ports() {
1907
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1908

    
1909
	# Create ports tree if necessary
1910
	if ! poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
1911
		poudriere_create_ports_tree
1912
	else
1913
		echo -n ">>> Reseting local changes on ports tree ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
1914
		script -aq ${LOGFILE} git -C "/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}" reset --hard >/dev/null 2>&1
1915
		script -aq ${LOGFILE} git -C "/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}" clean -fd >/dev/null 2>&1
1916
		echo "Done!" | tee -a ${LOGFILE}
1917
		echo -n ">>> Updating ports tree ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
1918
		script -aq ${LOGFILE} poudriere ports -u -p "${POUDRIERE_PORTS_NAME}" >/dev/null 2>&1
1919
		echo "Done!" | tee -a ${LOGFILE}
1920
	fi
1921
}
1922

    
1923
poudriere_bulk() {
1924
	local _archs=$(poudriere_possible_archs)
1925

    
1926
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1927

    
1928
	if [ -z "${DO_NOT_UPLOAD}" -a -z "${PKG_RSYNC_HOSTNAME}" ]; then
1929
		echo ">>> ERROR: PKG_RSYNC_HOSTNAME is not set"
1930
		print_error_pfS
1931
	fi
1932

    
1933
	poudriere_create_ports_tree
1934

    
1935
	[ -d /usr/local/etc/poudriere.d ] || \
1936
		mkdir -p /usr/local/etc/poudriere.d
1937

    
1938
	if [ -f "${BUILDER_TOOLS}/conf/pfPorts/make.conf" ]; then
1939
		cp -f "${BUILDER_TOOLS}/conf/pfPorts/make.conf" /usr/local/etc/poudriere.d/${POUDRIERE_PORTS_NAME}-make.conf
1940
	fi
1941

    
1942
	# Change version of pfSense meta ports for snapshots
1943
	if [ -z "${_IS_RELEASE}" ]; then
1944
		for meta_pkg in ${PRODUCT_NAME} ${PRODUCT_NAME}-vmware; do
1945
			local _meta_pkg_version="$(echo "${PRODUCT_VERSION}" | sed 's,DEVELOPMENT,ALPHA,')-${DATESTRING}"
1946
			sed -i '' -e "/^DISTVERSION/ s,^.*,DISTVERSION=	${_meta_pkg_version}," \
1947
				/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}/security/${meta_pkg}/Makefile
1948
		done
1949
	fi
1950

    
1951
	for jail_arch in ${_archs}; do
1952
		jail_name=$(poudriere_jail_name ${jail_arch})
1953

    
1954
		if ! poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
1955
			echo ">>> Poudriere jail ${jail_name} not found, skipping..." | tee -a ${LOGFILE}
1956
			continue
1957
		fi
1958

    
1959
		if [ -f "${POUDRIERE_BULK}.${jail_arch}" ]; then
1960
			_ref_bulk="${POUDRIERE_BULK}.${jail_arch}"
1961
		else
1962
			_ref_bulk="${POUDRIERE_BULK}"
1963
		fi
1964

    
1965
		_bulk=${SCRATCHDIR}/poudriere_bulk.${GIT_REPO_BRANCH_OR_TAG}
1966
		sed -e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" ${_ref_bulk} > ${_bulk}
1967

    
1968
		if ! poudriere bulk -f ${_bulk} -j ${jail_name} -p ${POUDRIERE_PORTS_NAME}; then
1969
			echo ">>> ERROR: Something went wrong..."
1970
			print_error_pfS
1971
		fi
1972

    
1973
		echo ">>> Cleaning up old packages from repo..."
1974
		if ! poudriere pkgclean -f ${_bulk} -j ${jail_name} -p ${POUDRIERE_PORTS_NAME} -y; then
1975
			echo ">>> ERROR: Something went wrong..."
1976
			print_error_pfS
1977
		fi
1978

    
1979
		# ./ is intentional, it's a rsync trick to make it chdir to directory before send it
1980
		pkg_repo_rsync "/usr/local/poudriere/data/packages/./${jail_name}-${POUDRIERE_PORTS_NAME}"
1981
	done
1982
}
1983

    
1984
# This routine is called to write out to stdout
1985
# a string. The string is appended to $SNAPSHOTSLOGFILE
1986
# and we scp the log file to the builder host if
1987
# needed for the real time logging functions.
1988
snapshots_update_status() {
1989
	if [ -z "${SNAPSHOTS}" -o -z "$1" ]; then
1990
		return
1991
	fi
1992
	echo $1
1993
	echo "`date` -|- $1" >> $SNAPSHOTSLOGFILE
1994
	if [ -z "${DO_NOT_UPLOAD}" -a -n "${RSYNCIP}" ]; then
1995
		LU=`cat $SNAPSHOTSLASTUPDATE`
1996
		CT=`date "+%H%M%S"`
1997
		# Only update every minute
1998
		if [ "$LU" != "$CT" ]; then
1999
			ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCLOGS}"
2000
			scp -q $SNAPSHOTSLOGFILE ${RSYNCUSER}@${RSYNCIP}:${RSYNCLOGS}/build.log
2001
			date "+%H%M%S" > $SNAPSHOTSLASTUPDATE
2002
		fi
2003
	fi
2004
}
2005

    
2006
# Copy the current log file to $filename.old on
2007
# the snapshot www server (real time logs)
2008
snapshots_rotate_logfile() {
2009
	if [ -z "${DO_NOT_UPLOAD}" -a -n "${RSYNCIP}" ]; then
2010
		scp -q $SNAPSHOTSLOGFILE ${RSYNCUSER}@${RSYNCIP}:${RSYNCLOGS}/build.log.old
2011
	fi
2012

    
2013
	# Cleanup log file
2014
	rm -f $SNAPSHOTSLOGFILE;    touch $SNAPSHOTSLOGFILE
2015
	rm -f $SNAPSHOTSLASTUPDATE; touch $SNAPSHOTSLASTUPDATE
2016

    
2017
}
2018

    
2019
snapshots_create_latest_symlink() {
2020
	local _image="${1}"
2021

    
2022
	if [ -z "${_image}" ]; then
2023
		return
2024
	fi
2025

    
2026
	if [ -z "${TIMESTAMP_SUFFIX}" ]; then
2027
		return
2028
	fi
2029

    
2030
	if [ -f "${_image}.gz" ]; then
2031
		local _image_fixed="${_image}.gz"
2032
	elif [ -f "${_image}" ]; then
2033
		local _image_fixed=${_image}
2034
	else
2035
		return
2036
	fi
2037

    
2038
	local _symlink=$(echo ${_image_fixed} | sed "s,${TIMESTAMP_SUFFIX},-latest,")
2039
	ln -sf $(basename ${_image_fixed}) ${_symlink}
2040
	ln -sf $(basename ${_image}).sha256 ${_symlink}.sha256
2041
}
2042

    
2043
snapshots_copy_to_staging_nanobsd() {
2044
	for NANOTYPE in nanobsd nanobsd-vga; do
2045
		for FILESIZE in ${1}; do
2046
			FILENAMEFULL="${PRODUCT_NAME}-${PRODUCT_VERSION}-${FILESIZE}-${TARGET}-${NANOTYPE}${TIMESTAMP_SUFFIX}.img.gz"
2047
			FILENAMEUPGRADE="${PRODUCT_NAME}-${PRODUCT_VERSION}-${FILESIZE}-${TARGET}-${NANOTYPE}-upgrade${TIMESTAMP_SUFFIX}.img.gz"
2048
			mkdir -p $STAGINGAREA/nanobsd
2049
			mkdir -p $STAGINGAREA/nanobsdupdates
2050

    
2051
			cp -l $IMAGES_FINAL_DIR/$FILENAMEFULL $STAGINGAREA/nanobsd/ 2>/dev/null
2052
			cp -l $IMAGES_FINAL_DIR/$FILENAMEUPGRADE $STAGINGAREA/nanobsdupdates 2>/dev/null
2053

    
2054
			if [ -f $STAGINGAREA/nanobsd/$FILENAMEFULL ]; then
2055
				sha256 $STAGINGAREA/nanobsd/$FILENAMEFULL > $STAGINGAREA/nanobsd/$FILENAMEFULL.sha256 2>/dev/null
2056
			fi
2057
			if [ -f $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE ]; then
2058
				sha256 $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE > $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE.sha256 2>/dev/null
2059
			fi
2060

    
2061
			# Copy NanoBSD auto update:
2062
			if [ -f $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE ]; then
2063
				cp -l $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE $STAGINGAREA/latest-${NANOTYPE}-$FILESIZE.img.gz 2>/dev/null
2064
				sha256 $STAGINGAREA/latest-${NANOTYPE}-$FILESIZE.img.gz > $STAGINGAREA/latest-${NANOTYPE}-$FILESIZE.img.gz.sha256 2>/dev/null
2065
				# NOTE: Updates need a file with output similar to date output
2066
				# Use the file generated at start of snapshots_dobuilds() to be consistent on times
2067
				cp $BUILTDATESTRINGFILE $STAGINGAREA/version-${NANOTYPE}-$FILESIZE
2068
			fi
2069
		done
2070
	done
2071
}
2072

    
2073
snapshots_copy_to_staging_iso_updates() {
2074
	local _img=""
2075

    
2076
	for _img in ${ISOPATH} ${MEMSTICKPATH} ${MEMSTICKSERIALPATH} ${UPDATES_TARBALL_FILENAME}; do
2077
		if [ ! -f "${_img}.gz" ]; then
2078
			continue
2079
		fi
2080
		sha256 ${_img}.gz > ${_img}.sha256
2081
		cp -l ${_img}* $STAGINGAREA/ 2>/dev/null
2082
		snapshots_create_latest_symlink ${STAGINGAREA}/$(basename ${_img})
2083
	done
2084

    
2085
	if [ "${TARGET}" = "amd64" -a -f "${MEMSTICKADIPATH}.gz" ]; then
2086
		sha256 ${MEMSTICKADIPATH}.gz > ${MEMSTICKADIPATH}.sha256
2087
		cp -l ${MEMSTICKADIPATH}* $STAGINGAREA/ 2>/dev/null
2088
		snapshots_create_latest_symlink ${STAGINGAREA}/$(basename ${MEMSTICKADIPATH})
2089
	fi
2090

    
2091
	# NOTE: Updates need a file with output similar to date output
2092
	# Use the file generated at start of snapshots_dobuilds() to be consistent on times
2093
	if [ -z "${_IS_RELEASE}" ]; then
2094
		cp $BUILTDATESTRINGFILE $STAGINGAREA/version 2>/dev/null
2095
	fi
2096
}
2097

    
2098
snapshots_scp_files() {
2099
	if [ -z "${RSYNC_COPY_ARGUMENTS}" ]; then
2100
		RSYNC_COPY_ARGUMENTS="-ave ssh --timeout=60 --bwlimit=${RSYNCKBYTELIMIT}" #--bwlimit=50
2101
	fi
2102

    
2103
	snapshots_update_status ">>> Copying core pkg repo to ${PKG_RSYNC_HOSTNAME}"
2104
	# Add ./ before last directory, it's rsync trick to make it chdir to parent directory before send
2105
	pkg_repo_rsync $(echo "${CORE_PKG_PATH}" | sed -E 's,/$,,; s,/([^/]*)$,/./\1,')
2106
	snapshots_update_status ">>> Finished copying core pkg repo"
2107

    
2108
	snapshots_update_status ">>> Copying files to ${RSYNCIP}"
2109

    
2110
	# Ensure directory(s) are available
2111
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/livecd_installer"
2112
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/updates"
2113
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/nanobsd"
2114
	if [ -d $STAGINGAREA/virtualization ]; then
2115
		ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/virtualization"
2116
	fi
2117
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/.updaters"
2118
	# ensure permissions are correct for r+w
2119
	ssh ${RSYNCUSER}@${RSYNCIP} "chmod -R ug+rw ${RSYNCPATH}/."
2120
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/${PRODUCT_NAME}-*iso* \
2121
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/livecd_installer/
2122
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/${PRODUCT_NAME}-memstick* \
2123
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/livecd_installer/
2124
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/${PRODUCT_NAME}-*Update* \
2125
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/updates/
2126
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/nanobsd/* \
2127
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/nanobsd/
2128
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/nanobsdupdates/* \
2129
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/updates/
2130
	if [ -d $STAGINGAREA/virtualization ]; then
2131
		rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/virtualization/* \
2132
			${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/virtualization/
2133
	fi
2134

    
2135
	# Rather than copy these twice, use ln to link to the latest one.
2136

    
2137
	ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest.tgz"
2138
	ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest.tgz.sha256"
2139

    
2140
	LATESTFILENAME=$(basename ${UPDATES_TARBALL_FILENAME})
2141
	ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${LATESTFILENAME} \
2142
		${RSYNCPATH}/.updaters/latest.tgz"
2143
	ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${LATESTFILENAME}.sha256 \
2144
		${RSYNCPATH}/.updaters/latest.tgz.sha256"
2145

    
2146
	for i in ${FLASH_SIZE}
2147
	do
2148
		ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz"
2149
		ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz.sha256"
2150
		ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz"
2151
		ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz.sha256"
2152

    
2153
		FILENAMEUPGRADE="${PRODUCT_NAME}-${PRODUCT_VERSION}-${i}-${TARGET}-nanobsd-upgrade${TIMESTAMP_SUFFIX}.img.gz"
2154
		ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE} \
2155
			${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz"
2156
		ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE}.sha256 \
2157
			${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz.sha256"
2158

    
2159
		FILENAMEUPGRADE="${PRODUCT_NAME}-${PRODUCT_VERSION}-${i}-${TARGET}-nanobsd-vga-upgrade${TIMESTAMP_SUFFIX}.img.gz"
2160
		ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE} \
2161
			${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz"
2162
		ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE}.sha256 \
2163
			${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz.sha256"
2164
	done
2165

    
2166
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/version* \
2167
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/.updaters
2168
	snapshots_update_status ">>> Finished copying files."
2169
}
(2-2/3)