Project

General

Profile

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

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

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

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

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

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

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

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

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

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

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

    
114
	mkdir -p ${CORE_PKG_TMP}
115

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

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

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

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

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

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

    
155
	mkdir -p ${CORE_PKG_REAL_PATH}/All
156
	if ! pkg create -o ${CORE_PKG_REAL_PATH}/All -p ${_plist} -r ${_root} -m ${_metadir}; then
157
		echo ">>> ERROR: Error building package ${_template} ${_flavor}"
158
		print_error_pfS
159
	fi
160
}
161

    
162
# This routine will output that something went wrong
163
print_error_pfS() {
164
	echo
165
	echo "####################################"
166
	echo "Something went wrong, check errors!" >&2
167
	echo "####################################"
168
	echo
169
	echo "NOTE: a lot of times you can run './build.sh --clean-builder' to resolve."
170
	echo
171
	if [ "$1" != "" ]; then
172
		echo $1
173
	fi
174
	[ -n "${LOGFILE}" -a -f "${LOGFILE}" ] && \
175
		echo "Log saved on ${LOGFILE}" && \
176
		tail -n20 ${LOGFILE} >&2
177
	echo
178
	kill $$
179
	exit 1
180
}
181

    
182
prestage_on_ram_setup() {
183
	[ -d "${STAGE_CHROOT_DIR}" ] \
184
		|| mkdir -p ${STAGE_CHROOT_DIR}
185
	[ -d "${FINAL_CHROOT_DIR}" ] \
186
		|| mkdir -p ${FINAL_CHROOT_DIR}
187

    
188
	_AVAIL_MEM=$(($(sysctl -n hw.usermem) / 1024 / 1024))
189
	if [ $_AVAIL_MEM -lt 2000 ]; then
190
		echo ">>> Builder has less than 2GiB RAM skipping memory disks"
191
		return
192
	else
193
		echo "######################################################################################"
194
		echo
195
		echo ">>> Builder has more than 2GiB RAM enabling memory disks"
196
		echo ">>> WARNING: Remember to remove these memory disks by running $0 --disable-memorydisks"
197
		echo
198
		echo "######################################################################################"
199
	fi
200

    
201
	if df /dev/ufs/prestagebacking >/dev/null 2>&1; then
202
		echo ">>> Detected preexisting memory disk enabled for ${STAGE_CHROOT_DIR}."
203
	else
204
		mdconfig -a -t swap -u 10001 -s ${MEMORYDISK_SIZE}
205
		newfs -L prestagebacking -U /dev/md10001
206
		mount /dev/ufs/prestagebacking ${STAGE_CHROOT_DIR}
207
	fi
208

    
209
	if df /dev/ufs/finalstagebacking >/dev/null 2>&1; then
210
		echo ">>> Detected preexisting memory disk enabled for ${FINAL_CHROOT_DIR}."
211
	else
212
		mdconfig -a -t swap -u 10002 -s ${MEMORYDISK_SIZE}
213
		newfs -L finalstagebacking -U /dev/md10002
214
		mount /dev/ufs/finalstagebacking ${FINAL_CHROOT_DIR}
215
	fi
216
}
217

    
218
prestage_on_ram_cleanup() {
219
	if [ -c /dev/md10001 ]; then
220
		if [ -d ${STAGE_CHROOT_DIR} ]; then
221
			umount ${STAGE_CHROOT_DIR}
222
		fi
223
		mdconfig -d -u 10001
224
	fi
225
	if [ -c /dev/md10002 ]; then
226
		if [ -d ${FINAL_CHROOT_DIR} ]; then
227
			umount ${FINAL_CHROOT_DIR}
228
		fi
229
		mdconfig -d -u 10002
230
	fi
231
}
232

    
233
# This routine will verify that the kernel has been
234
# installed OK to the staging area.
235
ensure_kernel_exists() {
236
	if [ ! -f "$1/boot/kernel/kernel.gz" ]; then
237
		echo ">>> ERROR: Could not locate $1/boot/kernel.gz"
238
		print_error_pfS
239
	fi
240
	KERNEL_SIZE=$(stat -f "%z" $1/boot/kernel/kernel.gz)
241
	if [ "$KERNEL_SIZE" -lt 3500 ]; then
242
		echo ">>> ERROR: Kernel $1/boot/kernel.gz appears to be smaller than it should be: $KERNEL_SIZE"
243
		print_error_pfS
244
	fi
245
}
246

    
247
get_pkg_name() {
248
	echo "${PRODUCT_NAME}-${1}-${CORE_PKG_VERSION}"
249
}
250

    
251
# This routine builds all related kernels
252
build_all_kernels() {
253
	# Set KERNEL_BUILD_PATH if it has not been set
254
	if [ -z "${KERNEL_BUILD_PATH}" ]; then
255
		KERNEL_BUILD_PATH=$SCRATCHDIR/kernels
256
		echo ">>> KERNEL_BUILD_PATH has not been set. Setting to ${KERNEL_BUILD_PATH}!"
257
	fi
258

    
259
	[ -d "${KERNEL_BUILD_PATH}" ] \
260
		&& rm -rf ${KERNEL_BUILD_PATH}
261

    
262
	# Build embedded kernel
263
	for BUILD_KERNEL in $BUILD_KERNELS; do
264
		unset KERNCONF
265
		unset KERNEL_DESTDIR
266
		unset KERNELCONF
267
		unset KERNEL_NAME
268
		export KERNCONF=$BUILD_KERNEL
269
		export KERNEL_DESTDIR="$KERNEL_BUILD_PATH/$BUILD_KERNEL"
270
		export KERNELCONF="${FREEBSD_SRC_DIR}/sys/${TARGET}/conf/$BUILD_KERNEL"
271
		export KERNEL_NAME=${BUILD_KERNEL}
272

    
273
		LOGFILE="${BUILDER_LOGS}/kernel.${KERNCONF}.${TARGET}.log"
274
		echo ">>> Building $BUILD_KERNEL kernel."  | tee -a ${LOGFILE}
275

    
276
		if [ ! -e "${FREEBSD_SRC_DIR}/sys/${TARGET}/conf/${BUILD_KERNEL}" ]; then
277
			echo ">>> ERROR: Could not find $KERNELCONF"
278
			print_error_pfS
279
		fi
280

    
281
		if [ -n "${NO_BUILDKERNEL}" -a -f "${CORE_PKG_REAL_PATH}/All/$(get_pkg_name kernel-${KERNEL_NAME}).txz" ]; then
282
			echo ">>> NO_BUILDKERNEL set, skipping build" | tee -a ${LOGFILE}
283
			continue
284
		fi
285

    
286
		export SRC_CONF=${SRC_CONF}
287
		buildkernel
288

    
289
		echo ">>> Staging $BUILD_KERNEL kernel..." | tee -a ${LOGFILE}
290
		installkernel
291

    
292
		ensure_kernel_exists $KERNEL_DESTDIR
293

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

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

    
301
		rm -rf $KERNEL_DESTDIR 2>&1 1>/dev/null
302

    
303
		echo ".Done" | tee -a ${LOGFILE}
304
	done
305
}
306

    
307
install_default_kernel() {
308
	if [ -z "${1}" ]; then
309
		echo ">>> ERROR: install_default_kernel called without a kernel config name"| tee -a ${LOGFILE}
310
		print_error_pfS
311
	fi
312

    
313
	export KERNEL_NAME="${1}"
314

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

    
317
	# Copy kernel package to chroot, otherwise pkg won't find it to install
318
	if ! pkg_chroot_add ${FINAL_CHROOT_DIR} kernel-${KERNEL_NAME}; then
319
		echo ">>> ERROR: Error installing kernel package $(get_pkg_name kernel-${KERNEL_NAME}).txz" | tee -a ${LOGFILE}
320
		print_error_pfS
321
	fi
322

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

    
326
	if [ ! -f $FINAL_CHROOT_DIR/boot/kernel/kernel.gz ]; then
327
		echo ">>> ERROR: No kernel installed on $FINAL_CHROOT_DIR and the resulting image will be unusable. STOPPING!" | tee -a ${LOGFILE}
328
		print_error_pfS
329
	fi
330
	mkdir -p $FINAL_CHROOT_DIR/pkgs
331
	if [ -z "${2}" -o -n "${INSTALL_EXTRA_KERNELS}" ]; then
332
		cp ${CORE_PKG_REAL_PATH}/All/$(get_pkg_name kernel-${KERNEL_NAME}).txz $FINAL_CHROOT_DIR/pkgs
333
		if [ -n "${INSTALL_EXTRA_KERNELS}" ]; then
334
			for _EXTRA_KERNEL in $INSTALL_EXTRA_KERNELS; do
335
				_EXTRA_KERNEL_PATH=${CORE_PKG_REAL_PATH}/All/$(get_pkg_name kernel-${_EXTRA_KERNEL}).txz
336
				if [ -f "${_EXTRA_KERNEL_PATH}" ]; then
337
					echo -n ". adding ${_EXTRA_KERNEL_PATH} on image /pkgs folder"
338
					cp ${_EXTRA_KERNEL_PATH} $FINAL_CHROOT_DIR/pkgs
339
				else
340
					echo ">>> ERROR: Requested kernel $(get_pkg_name kernel-${_EXTRA_KERNEL}).txz was not found to be put on image /pkgs folder!"
341
					print_error_pfS
342
				fi
343
			done
344
		fi
345
	fi
346
	echo "Done." | tee -a ${LOGFILE}
347

    
348
	unset KERNEL_NAME
349
}
350

    
351
# Creates a full update file
352
create_Full_update_tarball() {
353
	mkdir -p $UPDATESDIR
354

    
355
	customize_stagearea_for_image "fullupdate"
356
	install_default_kernel ${DEFAULT_KERNEL}
357

    
358
	rm -rf ${FINAL_CHROOT_DIR}/cf
359
	rm -rf ${FINAL_CHROOT_DIR}/conf
360
	rm -f ${FINAL_CHROOT_DIR}/etc/dh-parameters.*
361
	rm -f ${FINAL_CHROOT_DIR}/etc/rc.conf
362
	rm -f ${FINAL_CHROOT_DIR}/etc/pwd.db 2>/dev/null
363
	rm -f ${FINAL_CHROOT_DIR}/etc/group 2>/dev/null
364
	rm -f ${FINAL_CHROOT_DIR}/etc/spwd.db 2>/dev/null
365
	rm -f ${FINAL_CHROOT_DIR}/etc/passwd 2>/dev/null
366
	rm -f ${FINAL_CHROOT_DIR}/etc/master.passwd 2>/dev/null
367
	rm -f ${FINAL_CHROOT_DIR}/etc/fstab 2>/dev/null
368
	rm -f ${FINAL_CHROOT_DIR}/etc/bogons 2>/dev/null
369
	# Remove loader.conf and friends.  Ticket #560
370
	rm ${FINAL_CHROOT_DIR}/boot/loader.conf 2>/dev/null
371
	rm ${FINAL_CHROOT_DIR}/boot/loader.conf.local 2>/dev/null
372

    
373
	# Old systems will run (pre|post)_upgrade_command from /tmp
374
	if [ -f ${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/pre_upgrade_command ]; then
375
		cp -p \
376
			${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/pre_upgrade_command \
377
			${FINAL_CHROOT_DIR}/tmp
378
	fi
379
	if [ -f ${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/post_upgrade_command ]; then
380
		cp -p \
381
			${FINAL_CHROOT_DIR}${PRODUCT_SHARE_DIR}/post_upgrade_command \
382
			${FINAL_CHROOT_DIR}/tmp
383
	fi
384

    
385
	echo ">>> Creating ${UPDATES_TARBALL_FILENAME} ..." | tee -a ${LOGFILE}
386
	tar --exclude=./dev -czPf ${UPDATES_TARBALL_FILENAME} -C ${FINAL_CHROOT_DIR} .
387
}
388

    
389
# Outputs various set variables aka env
390
print_flags() {
391

    
392
	echo
393
	printf "             Product version: %s\n" $PRODUCT_VERSION
394
	printf "                   Stage DIR: %s\n" $STAGE_CHROOT_DIR
395
	printf "                 Updates dir: %s\n" $UPDATESDIR
396
	printf " Image Preparation Stage DIR: %s\n" $FINAL_CHROOT_DIR
397
	printf "                  Source DIR: %s\n" $FREEBSD_SRC_DIR
398
	printf "          FreeBSD repository: %s\n" $FREEBSD_REPO_BASE
399
	printf "          FreeBSD-src branch: %s\n" $FREEBSD_BRANCH
400
	printf "     FreeBSD original branch: %s\n" $FREEBSD_PARENT_BRANCH
401
	printf "               BUILD_KERNELS: %s\n" $BUILD_KERNELS
402
	printf "           Git Branch or Tag: %s\n" $GIT_REPO_BRANCH_OR_TAG
403
	printf "            MODULES_OVERRIDE: %s\n" $MODULES_OVERRIDE
404
	printf "    VMDK_DISK_CAPACITY_IN_GB: %s\n" $VMDK_DISK_CAPACITY_IN_GB
405
	printf "                 OVFTEMPLATE: %s\n" $OVFTEMPLATE
406
	printf "                     OVFVMDK: %s\n" $OVFVMDK
407
	printf "                    SRC_CONF: %s\n" $SRC_CONF
408
	printf "                     ISOPATH: %s\n" $ISOPATH
409
	printf "                MEMSTICKPATH: %s\n" $MEMSTICKPATH
410
	printf "    UPDATES_TARBALL_FILENAME: %s\n" $UPDATES_TARBALL_FILENAME
411
if [ -n "$SHOW_ENV" ]; then
412
	for LINE in $(env | egrep -v '(terminal|PASS|NAME|USER|SSH|GROUP|HOST)'); do
413
		echo "SHOW_ENV: $LINE"
414
	done
415
fi
416
	echo
417
}
418

    
419
# This builds FreeBSD (make buildworld)
420
# Imported from FreeSBIE
421
make_world() {
422

    
423
	LOGFILE=${BUILDER_LOGS}/buildworld.${TARGET}
424
	if [ -n "${NO_BUILDWORLD}" ]; then
425
		echo ">>> NO_BUILDWORLD set, skipping build" | tee -a ${LOGFILE}
426
		return
427
	fi
428

    
429
	# Set SRC_CONF variable if it's not already set.
430
	if [ -z "${SRC_CONF}" ]; then
431
		echo ">>> SRC_CONF is unset make sure this is what you want!" | tee -a ${LOGFILE}
432
	else
433
		echo ">>> Setting SRC_CONF to $SRC_CONF" | tee -a ${LOGFILE}
434
	fi
435

    
436
	# Set default parameters
437
	export MAKE_ARGS="${MAKEJ_WORLD} __MAKE_CONF=${MAKE_CONF} SRCCONF=${SRC_CONF} TARGET=${TARGET} TARGET_ARCH=${TARGET_ARCH}"
438

    
439
	echo ">>> LOGFILE set to $LOGFILE." | tee -a ${LOGFILE}
440
	makeargs="${MAKE_ARGS}"
441
	echo ">>> Building world for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
442
	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}
443
	(env LOCAL_ITOOLS="${EXTRA_TOOLS}" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} -DNO_CLEAN ${makeargs} buildworld || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
444
	echo ">>> Building world for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
445

    
446
	LOGFILE=${BUILDER_LOGS}/installworld.${TARGET}
447
	echo ">>> LOGFILE set to $LOGFILE." | tee -a ${LOGFILE}
448
	# Create if cleaned up
449
	makeargs="${MAKE_ARGS} DESTDIR=${STAGE_CHROOT_DIR} WITHOUT_TOOLCHAIN=1"
450
	echo ">>> Installing world for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
451
	echo ">>> Builder is running the command: env LOCAL_ITOOLS=\"${EXTRA_TOOLS}\" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} installworld" | tee -a ${LOGFILE}
452
	(env LOCAL_ITOOLS="${EXTRA_TOOLS}" script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} installworld || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
453
	echo ">>> Installing world for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
454

    
455
	makeargs="${MAKE_ARGS} DESTDIR=${STAGE_CHROOT_DIR}"
456
	echo ">>> Distribution world for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
457
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} distribution " | tee -a ${LOGFILE}
458
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} distribution  || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
459
	echo ">>> Distribution world for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
460

    
461
	[ -d "${STAGE_CHROOT_DIR}/usr/local/bin" ] \
462
		|| mkdir -p ${STAGE_CHROOT_DIR}/usr/local/bin
463
	makeargs="${MAKE_ARGS} DESTDIR=${STAGE_CHROOT_DIR}"
464
	echo ">>> Building and installing crypto tools and athstats for ${TARGET} architecture... (Starting - $(LC_ALL=C date))" | tee -a ${LOGFILE}
465
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/crypto ${makeargs} clean all install " | tee -a ${LOGFILE}
466
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/crypto ${makeargs} clean all install || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
467
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} clean" | tee -a ${LOGFILE}
468
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} clean || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
469
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} all" | tee -a ${LOGFILE}
470
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} all || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
471
	echo ">>> Builder is running the command: script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} install" | tee -a ${LOGFILE}
472
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR}/tools/tools/ath/athstats ${makeargs} install || print_error_pfS;) | egrep '^>>>' | tee -a ${LOGFILE}
473
	echo ">>> Building and installing crypto tools and athstats for ${TARGET} architecture... (Finished - $(LC_ALL=C date))" | tee -a ${LOGFILE}
474

    
475
	unset makeargs
476
}
477

    
478
nanobsd_image_filename() {
479
	local _size="$1"
480
	local _type="$2"
481
	local _upgrade="$3"
482

    
483
	if [ -z "$_upgrade" ]; then
484
		local _template=${NANOBSD_IMG_TEMPLATE}
485
	else
486
		local _template=${NANOBSD_UPGRADE_TEMPLATE}
487
	fi
488

    
489
	echo "$_template" | sed \
490
		-e "s,%%SIZE%%,${_size},g" \
491
		-e "s,%%TYPE%%,${_type},g"
492

    
493
	return 0
494
}
495

    
496
# This routine originated in nanobsd.sh
497
nanobsd_set_flash_details () {
498
	a1=$(echo $1 | tr '[:upper:]' '[:lower:]')
499

    
500
	# Source:
501
	#	SanDisk CompactFlash Memory Card
502
	#	Product Manual
503
	#	Version 10.9
504
	#	Document No. 20-10-00038
505
	#	April 2005
506
	# Table 2-7
507
	# NB: notice math error in SDCFJ-4096-388 line.
508
	#
509
	case "${a1}" in
510
		2048|2048m|2048mb|2g)
511
			NANO_MEDIASIZE=$((1989999616/512))
512
			;;
513
		4096|4096m|4096mb|4g)
514
			NANO_MEDIASIZE=$((3989999616/512))
515
			;;
516
		8192|8192m|8192mb|8g)
517
			NANO_MEDIASIZE=$((7989999616/512))
518
			;;
519
		16384|16384m|16384mb|16g)
520
			NANO_MEDIASIZE=$((15989999616/512))
521
			;;
522
		*)
523
			echo "Unknown Flash capacity"
524
			exit 2
525
			;;
526
	esac
527

    
528
	NANO_HEADS=16
529
	NANO_SECTS=63
530

    
531
	echo ">>> [nanoo] $1"
532
	echo ">>> [nanoo] NANO_MEDIASIZE: $NANO_MEDIASIZE"
533
	echo ">>> [nanoo] NANO_HEADS: $NANO_HEADS"
534
	echo ">>> [nanoo] NANO_SECTS: $NANO_SECTS"
535
	echo ">>> [nanoo] NANO_BOOT0CFG: $NANO_BOOT0CFG"
536
}
537

    
538
# This routine originated in nanobsd.sh
539
create_nanobsd_diskimage () {
540
	if [ -z "${1}" ]; then
541
		echo ">>> ERROR: Type of image has not been specified"
542
		print_error_pfS
543
	fi
544
	if [ -z "${2}" ]; then
545
		echo ">>> ERROR: Size of image has not been specified"
546
		print_error_pfS
547
	fi
548

    
549
	if [ "${1}" = "nanobsd" ]; then
550
		# It's serial
551
		export NANO_BOOTLOADER="boot/boot0sio"
552
	elif [ "${1}" = "nanobsd-vga" ]; then
553
		# It's vga
554
		export NANO_BOOTLOADER="boot/boot0"
555
	else
556
		echo ">>> ERROR: Type of image to create unknown"
557
		print_error_pfS
558
	fi
559

    
560
	if [ -z "${2}" ]; then
561
		echo ">>> ERROR: Media size(s) not specified."
562
		print_error_pfS
563
	fi
564

    
565
	if [ -z "${2}" ]; then
566
		echo ">>> ERROR: FLASH_SIZE is not set."
567
		print_error_pfS
568
	fi
569

    
570
	LOGFILE=${BUILDER_LOGS}/${1}.${TARGET}
571
	# Prepare folder to be put in image
572
	customize_stagearea_for_image "${1}"
573
	install_default_kernel ${DEFAULT_KERNEL} "no"
574

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

    
577
	echo "nanobsd" > $FINAL_CHROOT_DIR/etc/platform
578

    
579
	local BOOTCONF=${FINAL_CHROOT_DIR}/boot.config
580
	local LOADERCONF=${FINAL_CHROOT_DIR}/boot/loader.conf
581

    
582
	if [ "${1}" = "nanobsd" ]; then
583
		# Tell loader to use serial console early.
584
		echo "-S115200 -h" >> ${BOOTCONF}
585

    
586
		# Remove old console options if present.
587
		[ -f "${LOADERCONF}" ] \
588
			&& sed -i "" -Ee "/(console|boot_multicons|boot_serial|hint.uart)/d" ${LOADERCONF}
589
		# Activate serial console+video console in loader.conf
590
		echo 'loader_color="NO"' >> ${LOADERCONF}
591
		echo 'beastie_disable="YES"' >> ${LOADERCONF}
592
		echo 'boot_serial="YES"' >> ${LOADERCONF}
593
		echo 'console="comconsole"' >> ${LOADERCONF}
594
		echo 'comconsole_speed="115200"' >> ${LOADERCONF}
595
	fi
596
	echo 'autoboot_delay="5"' >> ${LOADERCONF}
597

    
598
	for _NANO_MEDIASIZE in ${2}; do
599
		if [ -z "${_NANO_MEDIASIZE}" ]; then
600
			continue;
601
		fi
602

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

    
606
		IMG="${IMAGES_FINAL_DIR}/$(nanobsd_image_filename ${_NANO_MEDIASIZE} ${1})"
607
		IMGUPDATE="${IMAGES_FINAL_DIR}/$(nanobsd_image_filename ${_NANO_MEDIASIZE} ${1} 1)"
608

    
609
		nanobsd_set_flash_details ${_NANO_MEDIASIZE}
610

    
611
		# These are defined in FlashDevice and on builder_default.sh
612
		echo $NANO_MEDIASIZE \
613
			$NANO_IMAGES \
614
			$NANO_SECTS \
615
			$NANO_HEADS \
616
			$NANO_CODESIZE \
617
			$NANO_CONFSIZE \
618
			$NANO_DATASIZE |
619
awk '
620
{
621
	printf "# %s\n", $0
622

    
623
	# size of cylinder in sectors
624
	cs = $3 * $4
625

    
626
	# number of full cylinders on media
627
	cyl = int ($1 / cs)
628

    
629
	# output fdisk geometry spec, truncate cyls to 1023
630
	if (cyl <= 1023)
631
		print "g c" cyl " h" $4 " s" $3
632
	else
633
		print "g c" 1023 " h" $4 " s" $3
634

    
635
	if ($7 > 0) {
636
		# size of data partition in full cylinders
637
		dsl = int (($7 + cs - 1) / cs)
638
	} else {
639
		dsl = 0;
640
	}
641

    
642
	# size of config partition in full cylinders
643
	csl = int (($6 + cs - 1) / cs)
644

    
645
	if ($5 == 0) {
646
		# size of image partition(s) in full cylinders
647
		isl = int ((cyl - dsl - csl) / $2)
648
	} else {
649
		isl = int (($5 + cs - 1) / cs)
650
	}
651

    
652
	# First image partition start at second track
653
	print "p 1 165 " $3, isl * cs - $3
654
	c = isl * cs;
655

    
656
	# Second image partition (if any) also starts offset one
657
	# track to keep them identical.
658
	if ($2 > 1) {
659
		print "p 2 165 " $3 + c, isl * cs - $3
660
		c += isl * cs;
661
	}
662

    
663
	# Config partition starts at cylinder boundary.
664
	print "p 3 165 " c, csl * cs
665
	c += csl * cs
666

    
667
	# Data partition (if any) starts at cylinder boundary.
668
	if ($7 > 0) {
669
		print "p 4 165 " c, dsl * cs
670
	} else if ($7 < 0 && $1 > c) {
671
		print "p 4 165 " c, $1 - c
672
	} else if ($1 < c) {
673
		print "Disk space overcommitted by", \
674
		    c - $1, "sectors" > "/dev/stderr"
675
		exit 2
676
	}
677

    
678
	# Force slice 1 to be marked active. This is necessary
679
	# for booting the image from a USB device to work.
680
	print "a 1"
681
}
682
	' > ${IMAGES_FINAL_DIR}/_.fdisk
683

    
684
		MNT=${IMAGES_FINAL_DIR}/_.mnt
685
		mkdir -p ${MNT}
686

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

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

    
693
		fdisk -i -f ${IMAGES_FINAL_DIR}/_.fdisk ${MD} 2>&1 >> ${LOGFILE}
694
		fdisk ${MD} 2>&1 >> ${LOGFILE}
695

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

    
698
		# Create first image
699
		bsdlabel -m i386 -w -B -b ${FINAL_CHROOT_DIR}/boot/boot ${MD}s1 2>&1 >> ${LOGFILE}
700
		bsdlabel -m i386 ${MD}s1 2>&1 >> ${LOGFILE}
701
		local _label=$(lc ${PRODUCT_NAME})
702
		newfs -L ${_label}0 ${NANO_NEWFS} /dev/${MD}s1a 2>&1 >> ${LOGFILE}
703
		mount /dev/ufs/${_label}0 ${MNT}
704
		if [ $? -ne 0 ]; then
705
			echo ">>> ERROR: Something wrong happened during mount of first slice image creation. STOPPING!" | tee -a ${LOGFILE}
706
			print_error_pfS
707
		fi
708
		# Consider the unmounting as well
709
		trap "umount /dev/ufs/${_label}0; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
710

    
711
		clone_directory_contents ${FINAL_CHROOT_DIR} ${MNT}
712

    
713
		# Set NanoBSD image size
714
		echo "${_NANO_MEDIASIZE}" > ${MNT}/etc/nanosize.txt
715

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

    
721
		umount ${MNT}
722
		# Restore the original trap
723
		trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
724

    
725
		# Setting NANO_IMAGES to 1 and NANO_INIT_IMG2 will tell
726
		# NanoBSD to only create one partition.  We default to 2
727
		# partitions in case anything happens to the first the
728
		# operator can boot from the 2nd and should be OK.
729

    
730
		# Before just going to use dd for duplicate think!
731
		# The images are created as sparse so lets take advantage
732
		# of that by just exec some commands.
733
		if [ $NANO_IMAGES -gt 1 -a $NANO_INIT_IMG2 -gt 0 ] ; then
734
			# Duplicate to second image (if present)
735
			echo ">>> Creating NanoBSD second slice by duplicating first slice." | tee -a ${LOGFILE}
736
			# Create second image
737
			dd if=/dev/${MD}s1 of=/dev/${MD}s2 conv=sparse bs=64k 2>&1 >> ${LOGFILE}
738
			tunefs -L ${_label}1 /dev/${MD}s2a 2>&1 >> ${LOGFILE}
739
			mount /dev/ufs/${_label}1 ${MNT}
740
			if [ $? -ne 0 ]; then
741
				echo ">>> ERROR: Something wrong happened during mount of second slice image creation. STOPPING!" | tee -a ${LOGFILE}
742
				print_error_pfS
743
			fi
744
			# Consider the unmounting as well
745
			trap "umount /dev/ufs/${_label}1; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
746

    
747
			echo "/dev/ufs/${_label}1 / ufs ro,sync,noatime 1 1" > ${MNT}/etc/fstab
748
			if [ $NANO_CONFSIZE -gt 0 ] ; then
749
				echo "/dev/ufs/cf /cf ufs ro,sync,noatime 1 1" >> ${MNT}/etc/fstab
750
			fi
751

    
752
			umount ${MNT}
753
			# Restore the trap back
754
			trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
755
		fi
756

    
757
		# Create Data slice, if any.
758
		# Note the changing of the variable to NANO_CONFSIZE
759
		# from NANO_DATASIZE.  We also added glabel support
760
		# and populate the Product configuration from the /cf
761
		# directory located in FINAL_CHROOT_DIR
762
		if [ $NANO_CONFSIZE -gt 0 ] ; then
763
			echo ">>> Creating /cf area to hold config.xml"
764
			newfs -L cf ${NANO_NEWFS} /dev/${MD}s3 2>&1 >> ${LOGFILE}
765
			# Mount data partition and copy contents of /cf
766
			# Can be used later to create custom default config.xml while building
767
			mount /dev/ufs/cf ${MNT}
768
			if [ $? -ne 0 ]; then
769
				echo ">>> ERROR: Something wrong happened during mount of cf slice image creation. STOPPING!" | tee -a ${LOGFILE}
770
				print_error_pfS
771
			fi
772
			# Consider the unmounting as well
773
			trap "umount /dev/ufs/cf; mdconfig -d -u ${MD}; return" 1 2 15 EXIT
774

    
775
			clone_directory_contents ${FINAL_CHROOT_DIR}/cf ${MNT}
776

    
777
			umount ${MNT}
778
			# Restore the trap back
779
			trap "mdconfig -d -u ${MD}; return" 1 2 15 EXIT
780
		else
781
			">>> [nanoo] NANO_CONFSIZE is not set. Not adding a /conf partition.. You sure about this??" | tee -a ${LOGFILE}
782
		fi
783

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

    
787
		mdconfig -d -u $MD
788
		# Restore default action
789
		trap "-" 1 2 15 EXIT
790

    
791
		# Check each image and ensure that they are over
792
		# 3 megabytes.  If either image is under 20 megabytes
793
		# in size then error out.
794
		IMGSIZE=$(stat -f "%z" ${IMG})
795
		IMGUPDATESIZE=$(stat -f "%z" ${IMGUPDATE})
796
		CHECKSIZE="20040710"
797
		if [ "$IMGSIZE" -lt "$CHECKSIZE" ]; then
798
			echo ">>> ERROR: Something went wrong when building NanoBSD.  The image size is under 20 megabytes!" | tee -a ${LOGFILE}
799
			print_error_pfS
800
		fi
801
		if [ "$IMGUPDATESIZE" -lt "$CHECKSIZE" ]; then
802
			echo ">>> ERROR: Something went wrong when building NanoBSD upgrade image.  The image size is under 20 megabytes!" | tee -a ${LOGFILE}
803
			print_error_pfS
804
		fi
805

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

    
809
		gzip -qf $IMG &
810
		_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
811
		gzip -qf $IMGUPDATE &
812
		_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
813
	done
814

    
815
	unset IMG
816
	unset IMGUPDATE
817
	unset IMGUPDATESIZE
818
	unset IMGSIZE
819

    
820
	ls -lah $IMAGES_FINAL_DIR
821
}
822

    
823
# This routine creates a ova image that contains
824
# a ovf and vmdk file. These files can be imported
825
# right into vmware or virtual box.
826
# (and many other emulation platforms)
827
# http://www.vmware.com/pdf/ovf_whitepaper_specification.pdf
828
create_ova_image() {
829
	# XXX create a .ovf php creator that you can pass:
830
	#     1. populatedSize
831
	#     2. license
832
	#     3. product name
833
	#     4. version
834
	#     5. number of network interface cards
835
	#     6. allocationUnits
836
	#     7. capacity
837
	#     8. capacityAllocationUnits
838

    
839
	LOGFILE=${BUILDER_LOGS}/ova.${TARGET}.log
840

    
841
	[ -d "${OVA_TMP}" ] \
842
		&& rm -rf ${OVA_TMP}
843

    
844
	mkdir -p ${OVA_TMP}
845

    
846
	if [ -z "${OVA_SWAP_PART_SIZE_IN_GB}" -o "${OVA_SWAP_PART_SIZE_IN_GB}" = "0" ]; then
847
		# first partition size (freebsd-ufs)
848
		local OVA_FIRST_PART_SIZE_IN_GB=${VMDK_DISK_CAPACITY_IN_GB}
849
		# Calculate real first partition size, removing 128 blocks (65536 bytes) beginning/loader
850
		local OVA_FIRST_PART_SIZE=$((${OVA_FIRST_PART_SIZE_IN_GB}*1024*1024*1024-65536))
851
		# Unset swap partition size variable
852
		unset OVA_SWAP_PART_SIZE
853
		# Parameter used by mkimg
854
		unset OVA_SWAP_PART_PARAM
855
	else
856
		# first partition size (freebsd-ufs)
857
		local OVA_FIRST_PART_SIZE_IN_GB=$((VMDK_DISK_CAPACITY_IN_GB-OVA_SWAP_PART_SIZE_IN_GB))
858
		# Use first partition size in g
859
		local OVA_FIRST_PART_SIZE="${OVA_FIRST_PART_SIZE_IN_GB}g"
860
		# Calculate real swap size, removing 128 blocks (65536 bytes) beginning/loader
861
		local OVA_SWAP_PART_SIZE=$((${OVA_SWAP_PART_SIZE_IN_GB}*1024*1024*1024-65536))
862
		# Parameter used by mkimg
863
		local OVA_SWAP_PART_PARAM="-p freebsd-swap/swap0::${OVA_SWAP_PART_SIZE}"
864
	fi
865

    
866
	# Prepare folder to be put in image
867
	customize_stagearea_for_image "ova"
868
	install_default_kernel ${DEFAULT_KERNEL} "no"
869

    
870
	# Fill fstab
871
	echo ">>> Installing platform specific items..." | tee -a ${LOGFILE}
872
	echo "/dev/gpt/${PRODUCT_NAME}	/	ufs		rw	1	1" > ${FINAL_CHROOT_DIR}/etc/fstab
873
	if [ -n "${OVA_SWAP_PART_SIZE}" ]; then
874
		echo "/dev/gpt/swap0	none	swap	sw	0	0" >> ${FINAL_CHROOT_DIR}/etc/fstab
875
	fi
876

    
877
	# Create / partition
878
	echo -n ">>> Creating / partition... " | tee -a ${LOGFILE}
879
	makefs \
880
		-B little \
881
		-o label=${PRODUCT_NAME},version=2 \
882
		-s ${OVA_FIRST_PART_SIZE} \
883
		${OVA_TMP}/${OVFUFS} \
884
		${FINAL_CHROOT_DIR} 2>&1 >> ${LOGFILE}
885

    
886
	if [ $? -ne 0 -o ! -f ${OVA_TMP}/${OVFUFS} ]; then
887
		if [ -f ${OVA_TMP}/${OVFUFS} ]; then
888
			rm -f ${OVA_TMP}/${OVFUFS}
889
		fi
890
		echo "Failed!" | tee -a ${LOGFILE}
891
		echo ">>> ERROR: Error creating vmdk / partition. STOPPING!" | tee -a ${LOGFILE}
892
		print_error_pfS
893
	fi
894
	echo "Done!" | tee -a ${LOGFILE}
895

    
896
	echo -n ">>> Enabling SUJ on recently created disk... " | tee -a ${LOGFILE}
897
	if ! tunefs -j enable ${OVA_TMP}/${OVFUFS} 2>&1 >>${LOGFILE}; then
898
		echo "Failed!" | tee -a ${LOGFILE}
899
		echo ">>> ERROR: Error enabling SUJ on disk. STOPPING!" | tee -a ${LOGFILE}
900
		print_error_pfS
901
	fi
902
	echo "Done!" | tee -a ${LOGFILE}
903

    
904
	# Create raw disk
905
	echo -n ">>> Creating raw disk... " | tee -a ${LOGFILE}
906
	mkimg \
907
		-s gpt \
908
		-f raw \
909
		-b /boot/pmbr \
910
		-p freebsd-boot:=/boot/gptboot \
911
		-p freebsd-ufs/${PRODUCT_NAME}:=${OVA_TMP}/${OVFUFS} \
912
		${OVA_SWAP_PART_PARAM} \
913
		-o ${OVA_TMP}/${OVFRAW} 2>&1 >> ${LOGFILE}
914

    
915
	if [ $? -ne 0 -o ! -f ${OVA_TMP}/${OVFRAW} ]; then
916
		if [ -f ${OVA_TMP}/${OVFUFS} ]; then
917
			rm -f ${OVA_TMP}/${OVFUFS}
918
		fi
919
		if [ -f ${OVA_TMP}/${OVFRAW} ]; then
920
			rm -f ${OVA_TMP}/${OVFRAW}
921
		fi
922
		echo "Failed!" | tee -a ${LOGFILE}
923
		echo ">>> ERROR: Error creating temporary vmdk image. STOPPING!" | tee -a ${LOGFILE}
924
		print_error_pfS
925
	fi
926
	echo "Done!" | tee -a ${LOGFILE}
927

    
928
	# We don't need it anymore
929
	rm -f ${OVA_TMP}/${OVFUFS} >/dev/null 2>&1
930

    
931
	# Convert raw to vmdk
932
	echo -n ">>> Creating vmdk disk... " | tee -a ${LOGFILE}
933
	vmdktool -z9 -v ${OVA_TMP}/${OVFVMDK} ${OVA_TMP}/${OVFRAW}
934

    
935
	if [ $? -ne 0 -o ! -f ${OVA_TMP}/${OVFVMDK} ]; then
936
		if [ -f ${OVA_TMP}/${OVFRAW} ]; then
937
			rm -f ${OVA_TMP}/${OVFRAW}
938
		fi
939
		if [ -f ${OVA_TMP}/${OVFVMDK} ]; then
940
			rm -f ${OVA_TMP}/${OVFVMDK}
941
		fi
942
		echo "Failed!" | tee -a ${LOGFILE}
943
		echo ">>> ERROR: Error creating vmdk image. STOPPING!" | tee -a ${LOGFILE}
944
		print_error_pfS
945
	fi
946
	echo "Done!" | tee -a ${LOGFILE}
947

    
948
	rm -f ${OVA_TMP}/i${OVFRAW}
949

    
950
	ova_setup_ovf_template
951

    
952
	echo -n ">>> Writing final ova image... " | tee -a ${LOGFILE}
953
	# Create OVA file for vmware
954
	gtar -C ${OVA_TMP} -cpf ${OVAPATH} ${PRODUCT_NAME}.ovf ${OVFVMDK}
955
	echo "Done!" | tee -a ${LOGFILE}
956
	rm -f ${OVA_TMP}/${OVFVMDK} >/dev/null 2>&1
957

    
958
	echo ">>> OVA created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
959
}
960

    
961
# called from create_ova_image
962
ova_setup_ovf_template() {
963
	if [ ! -f ${OVFTEMPLATE} ]; then
964
		echo ">>> ERROR: OVF template file (${OVFTEMPLATE}) not found."
965
		print_error_pfS
966
	fi
967

    
968
	#  OperatingSystemSection (${PRODUCT_NAME}.ovf)
969
	#  42   FreeBSD 32-Bit
970
	#  78   FreeBSD 64-Bit
971
	if [ "${TARGET}" = "amd64" ]; then
972
		local _os_id="78"
973
		local _os_type="freebsd64Guest"
974
		local _os_descr="FreeBSD 64-Bit"
975
	elif [ "${TARGET}" = "i386" ]; then
976
		local _os_id="42"
977
		local _os_type="freebsdGuest"
978
		local _os_descr="FreeBSD"
979
	else
980
		echo ">>> ERROR: Platform not supported for OVA (${TARGET})"
981
		print_error_pfS
982
	fi
983

    
984
	local POPULATED_SIZE=$(du -d0 -k $FINAL_CHROOT_DIR | cut -f1)
985
	local POPULATED_SIZE_IN_BYTES=$((${POPULATED_SIZE}*1024))
986
	local VMDK_FILE_SIZE=$(stat -f "%z" ${OVA_TMP}/${OVFVMDK})
987

    
988
	sed \
989
		-e "s,%%VMDK_FILE_SIZE%%,${VMDK_FILE_SIZE},g" \
990
		-e "s,%%VMDK_DISK_CAPACITY_IN_GB%%,${VMDK_DISK_CAPACITY_IN_GB},g" \
991
		-e "s,%%POPULATED_SIZE_IN_BYTES%%,${POPULATED_SIZE_IN_BYTES},g" \
992
		-e "s,%%OS_ID%%,${_os_id},g" \
993
		-e "s,%%OS_TYPE%%,${_os_type},g" \
994
		-e "s,%%OS_DESCR%%,${_os_descr},g" \
995
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
996
		-e "s,%%PRODUCT_NAME_SUFFIX%%,${PRODUCT_NAME_SUFFIX},g" \
997
		-e "s,%%PRODUCT_VERSION%%,${PRODUCT_VERSION},g" \
998
		-e "s,%%PRODUCT_URL%%,${PRODUCT_URL},g" \
999
		-e "s#%%VENDOR_NAME%%#${VENDOR_NAME}#g" \
1000
		-e "s#%%OVF_INFO%%#${OVF_INFO}#g" \
1001
		-e "/^%%PRODUCT_LICENSE%%/r ${BUILDER_ROOT}/license.txt" \
1002
		-e "/^%%PRODUCT_LICENSE%%/d" \
1003
		${OVFTEMPLATE} > ${OVA_TMP}/${PRODUCT_NAME}.ovf
1004
}
1005

    
1006
# Cleans up previous builds
1007
clean_builder() {
1008
	# Clean out directories
1009
	echo ">>> Cleaning up previous build environment...Please wait!"
1010

    
1011
	staginareas_clean_each_run
1012

    
1013
	if [ -d "${STAGE_CHROOT_DIR}" ]; then
1014
		BASENAME=$(basename ${STAGE_CHROOT_DIR})
1015
		echo -n ">>> Cleaning ${STAGE_CHROOT_DIR} ..."
1016
		chflags -R noschg ${STAGE_CHROOT_DIR} 2>&1 >/dev/null
1017
		rm -rf ${STAGE_CHROOT_DIR}/* 2>/dev/null
1018
		echo "Done."
1019
	fi
1020

    
1021
	if [ -z "${NO_CLEAN_FREEBSD_OBJ}" -a -d "${FREEBSD_SRC_DIR}" ]; then
1022
		OBJTREE=$(env TARGET=${TARGET} TARGET_ARCH=${TARGET_ARCH} make -C ${FREEBSD_SRC_DIR} -V OBJTREE)
1023
		if [ -d "${OBJTREE}" ]; then
1024
			echo -n ">>> Cleaning FreeBSD objects dir staging..."
1025
			echo -n "."
1026
			chflags -R noschg ${OBJTREE} 2>&1 >/dev/null
1027
			echo -n "."
1028
			rm -rf ${OBJTREE}/*
1029
			echo "Done!"
1030
		fi
1031
		if [ -d "${KERNEL_BUILD_PATH}" ]; then
1032
			echo -n ">>> Cleaning previously built kernel stage area..."
1033
			rm -rf $KERNEL_BUILD_PATH/*
1034
			echo "Done!"
1035
		fi
1036
	fi
1037
	mkdir -p $KERNEL_BUILD_PATH
1038

    
1039
	echo -n ">>> Cleaning previously built images..."
1040
	rm -rf $IMAGES_FINAL_DIR/*
1041
	rm -rf $STAGINGAREA/*
1042
	echo "Done!"
1043

    
1044
	if [ -z "${NO_CLEAN_FREEBSD_SRC}" ]; then
1045
		if [ -d "$FREEBSD_SRC_DIR" ]; then
1046
			echo -n ">>> Ensuring $FREEBSD_SRC_DIR is clean..."
1047
			rm -rf ${FREEBSD_SRC_DIR}
1048
			echo "Done!"
1049
		fi
1050
	fi
1051

    
1052
	echo -n ">>> Cleaning previous builder logs..."
1053
	if [ -d "$BUILDER_LOGS" ]; then
1054
		rm -rf ${BUILDER_LOGS}
1055
	fi
1056
	mkdir -p ${BUILDER_LOGS}
1057

    
1058
	echo "Done!"
1059

    
1060
	echo ">>> Cleaning of builder environment has finished."
1061
}
1062

    
1063
clone_directory_contents() {
1064
	if [ ! -d "$1" -o ! -d "$2" ]; then
1065
		if [ -z "${LOGFILE}" ]; then
1066
			echo ">>> ERROR: Argument $1 supplied is not a directory!"
1067
		else
1068
			echo ">>> ERROR: Argument $1 supplied is not a directory!" | tee -a ${LOGFILE}
1069
		fi
1070
		print_error_pfS
1071
	fi
1072
	echo -n ">>> Using TAR to clone $1 to $2 ..."
1073
	tar -C ${1} -c -f - . | tar -C ${2} -x -p -f -
1074
	echo "Done!"
1075
}
1076

    
1077
clone_to_staging_area() {
1078
	# Clone everything to the final staging area
1079
	echo -n ">>> Cloning everything to ${STAGE_CHROOT_DIR} staging area..."
1080
	LOGFILE=${BUILDER_LOGS}/cloning.${TARGET}.log
1081

    
1082
	tar -C ${PRODUCT_SRC} -c -f - . | \
1083
		tar -C ${STAGE_CHROOT_DIR} -x -p -f -
1084

    
1085
	if [ "${PRODUCT_NAME}" != "pfSense" ]; then
1086
		mv ${STAGE_CHROOT_DIR}/usr/local/sbin/pfSense-upgrade \
1087
			${STAGE_CHROOT_DIR}/usr/local/sbin/${PRODUCT_NAME}-upgrade
1088
	fi
1089

    
1090
	if [ -f ${STAGE_CHROOT_DIR}/etc/master.passwd ]; then
1091
		chroot ${STAGE_CHROOT_DIR} pwd_mkdb /etc/master.passwd
1092
	fi
1093
	mkdir -p ${STAGE_CHROOT_DIR}/etc/mtree
1094
	mtree -Pcp ${STAGE_CHROOT_DIR}/var > ${STAGE_CHROOT_DIR}/etc/mtree/var.dist
1095
	mtree -Pcp ${STAGE_CHROOT_DIR}/etc > ${STAGE_CHROOT_DIR}/etc/mtree/etc.dist
1096
	if [ -d ${STAGE_CHROOT_DIR}/usr/local/etc ]; then
1097
		mtree -Pcp ${STAGE_CHROOT_DIR}/usr/local/etc > ${STAGE_CHROOT_DIR}/etc/mtree/localetc.dist
1098
	fi
1099

    
1100
	## Add buildtime and lastcommit information
1101
	# This is used for detecting updates.
1102
	echo "$BUILTDATESTRING" > $STAGE_CHROOT_DIR/etc/version.buildtime
1103
	# Record last commit info if it is available.
1104
	if [ -f $SCRATCHDIR/build_commit_info.txt ]; then
1105
		cp $SCRATCHDIR/build_commit_info.txt $STAGE_CHROOT_DIR/etc/version.lastcommit
1106
	fi
1107

    
1108
	local _exclude_files="${CORE_PKG_TMP}/base_exclude_files"
1109
	sed \
1110
		-e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" \
1111
		-e "s,%%VERSION%%,${_version},g" \
1112
		${BUILDER_TOOLS}/templates/core_pkg/base/exclude_files \
1113
		> ${_exclude_files}
1114

    
1115
	mkdir -p ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR} >/dev/null 2>&1
1116

    
1117
	# Include a sample pkg stable conf to base
1118
	setup_pkg_repo \
1119
		${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/${PRODUCT_NAME}-repo.conf \
1120
		${TARGET} \
1121
		${TARGET_ARCH} \
1122
		${PKG_REPO_CONF_BRANCH} \
1123
		"release"
1124

    
1125
	# Include a sample pkg devel conf to base
1126
	setup_pkg_repo \
1127
		${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/${PRODUCT_NAME}-repo-devel.conf \
1128
		${TARGET} \
1129
		${TARGET_ARCH} \
1130
		${PKG_REPO_CONF_BRANCH}
1131

    
1132
	mtree \
1133
		-c \
1134
		-k uid,gid,mode,size,flags,sha256digest \
1135
		-p ${STAGE_CHROOT_DIR} \
1136
		-X ${_exclude_files} \
1137
		> ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/base.mtree
1138
	tar \
1139
		-C ${STAGE_CHROOT_DIR} \
1140
		-cJf ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/base.txz \
1141
		-X ${_exclude_files} \
1142
		.
1143

    
1144
	mkdir -p $(dirname ${STAGE_CHROOT_DIR}${PKG_REPO_PATH}) >/dev/null 2>&1
1145

    
1146
	# Create repo and repo-devel packages
1147
	cp -f ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/${PRODUCT_NAME}-repo.conf \
1148
		${STAGE_CHROOT_DIR}${PKG_REPO_PATH}
1149

    
1150
	core_pkg_create repo "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1151

    
1152
	cp -f ${STAGE_CHROOT_DIR}${PRODUCT_SHARE_DIR}/${PRODUCT_NAME}-repo-devel.conf \
1153
		${STAGE_CHROOT_DIR}${PKG_REPO_PATH}
1154

    
1155
	core_pkg_create repo-devel "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1156

    
1157
	rm -f ${STAGE_CHROOT_DIR}${PKG_REPO_PATH}
1158

    
1159
	core_pkg_create rc "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1160
	core_pkg_create base "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1161
	core_pkg_create base-nanobsd "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1162
	core_pkg_create default-config "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1163

    
1164
	local DEFAULTCONF=${STAGE_CHROOT_DIR}/conf.default/config.xml
1165

    
1166
	# Save current WAN and LAN if value
1167
	local _old_wan_if=$(xml sel -t -v "${XML_ROOTOBJ}/interfaces/wan/if" ${DEFAULTCONF})
1168
	local _old_lan_if=$(xml sel -t -v "${XML_ROOTOBJ}/interfaces/lan/if" ${DEFAULTCONF})
1169

    
1170
	# Change default interface names to match vmware driver
1171
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/wan/if" -v "vmx0" ${DEFAULTCONF}
1172
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/lan/if" -v "vmx1" ${DEFAULTCONF}
1173
	core_pkg_create default-config "vmware" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1174

    
1175
	# Restore default values to be used by serial package
1176
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/wan/if" -v "${_old_wan_if}" ${DEFAULTCONF}
1177
	xml ed -P -L -u "${XML_ROOTOBJ}/interfaces/lan/if" -v "${_old_lan_if}" ${DEFAULTCONF}
1178

    
1179
	# Activate serial console in config.xml
1180
	xml ed -L -P -d "${XML_ROOTOBJ}/system/enableserial" ${DEFAULTCONF}
1181
	xml ed -P -s "${XML_ROOTOBJ}/system" -t elem -n "enableserial" \
1182
		${DEFAULTCONF} > ${DEFAULTCONF}.tmp
1183
	xml fo -t ${DEFAULTCONF}.tmp > ${DEFAULTCONF}
1184
	rm -f ${DEFAULTCONF}.tmp
1185

    
1186
	echo force > ${STAGE_CHROOT_DIR}/cf/conf/enableserial_force
1187

    
1188
	core_pkg_create default-config-serial "" ${CORE_PKG_VERSION} ${STAGE_CHROOT_DIR}
1189

    
1190
	rm -f ${STAGE_CHROOT_DIR}/cf/conf/enableserial_force
1191
	rm -f ${STAGE_CHROOT_DIR}/cf/conf/config.xml
1192

    
1193
	# Make sure pkg is present
1194
	pkg_bootstrap ${STAGE_CHROOT_DIR}
1195

    
1196
	echo "Done!"
1197
}
1198

    
1199
create_final_staging_area() {
1200
	if [ -z "${FINAL_CHROOT_DIR}" ]; then
1201
		echo ">>> ERROR: FINAL_CHROOT_DIR is not set, cannot continue!" | tee -a ${LOGFILE}
1202
		print_error_pfS
1203
	fi
1204

    
1205
	if [ -d "${FINAL_CHROOT_DIR}" ]; then
1206
		echo -n ">>> Previous ${FINAL_CHROOT_DIR} detected cleaning up..." | tee -a ${LOGFILE}
1207
		chflags -R noschg ${FINAL_CHROOT_DIR} 2>&1 1>/dev/null
1208
		rm -rf ${FINAL_CHROOT_DIR}/* 2>&1 1>/dev/null
1209
		echo "Done." | tee -a ${LOGFILE}
1210
	fi
1211

    
1212
	echo ">>> Preparing Final image staging area: $(LC_ALL=C date)" 2>&1 | tee -a ${LOGFILE}
1213
	echo ">>> Cloning ${STAGE_CHROOT_DIR} to ${FINAL_CHROOT_DIR}" 2>&1 | tee -a ${LOGFILE}
1214
	clone_directory_contents ${STAGE_CHROOT_DIR} ${FINAL_CHROOT_DIR}
1215

    
1216
	if [ ! -f $FINAL_CHROOT_DIR/sbin/init ]; then
1217
		echo ">>> ERROR: Something went wrong during cloning -- Please verify!" 2>&1 | tee -a ${LOGFILE}
1218
		print_error_pfS
1219
	fi
1220
}
1221

    
1222
customize_stagearea_for_image() {
1223
	local _image_type="$1"
1224
	local _default_config=""
1225

    
1226
	if [ -n "$2" ]; then
1227
		_default_config="$2"
1228
	elif [ "${_image_type}" = "nanobsd" -o \
1229
	     "${_image_type}" = "memstickserial" -o \
1230
	     "${_image_type}" = "memstickadi" ]; then
1231
		_default_config="default-config-serial"
1232
	elif [ "${_image_type}" = "ova" ]; then
1233
		_default_config="default-config-vmware"
1234
	else
1235
		_default_config="default-config"
1236
	fi
1237

    
1238
	# Prepare final stage area
1239
	create_final_staging_area
1240

    
1241
	pkg_chroot_add ${FINAL_CHROOT_DIR} rc
1242

    
1243
	if [ "${_image_type}" = "nanobsd" -o \
1244
	     "${_image_type}" = "nanobsd-vga" ]; then
1245

    
1246
		mkdir -p ${FINAL_CHROOT_DIR}/root/var/db \
1247
			 ${FINAL_CHROOT_DIR}/root/var/cache \
1248
			 ${FINAL_CHROOT_DIR}/var/db/pkg \
1249
			 ${FINAL_CHROOT_DIR}/var/cache/pkg
1250
		mv -f ${FINAL_CHROOT_DIR}/var/db/pkg ${FINAL_CHROOT_DIR}/root/var/db
1251
		mv -f ${FINAL_CHROOT_DIR}/var/cache/pkg ${FINAL_CHROOT_DIR}/root/var/cache
1252
		ln -sf ../../root/var/db/pkg ${FINAL_CHROOT_DIR}/var/db/pkg
1253
		ln -sf ../../root/var/cache/pkg ${FINAL_CHROOT_DIR}/var/cache/pkg
1254

    
1255
		pkg_chroot_add ${FINAL_CHROOT_DIR} base-nanobsd
1256
	else
1257
		pkg_chroot_add ${FINAL_CHROOT_DIR} base
1258
	fi
1259

    
1260
	if [ -n "${IS_RELEASE}" ]; then
1261
		pkg_chroot_add ${FINAL_CHROOT_DIR} repo
1262
	else
1263
		pkg_chroot_add ${FINAL_CHROOT_DIR} repo-devel
1264
	fi
1265

    
1266
	if [ "${_image_type}" = "iso" -o \
1267
	     "${_image_type}" = "memstick" -o \
1268
	     "${_image_type}" = "memstickserial" -o \
1269
	     "${_image_type}" = "memstickadi" ]; then
1270
		install_bsdinstaller
1271
		mkdir -p ${FINAL_CHROOT_DIR}/pkgs
1272
		cp ${CORE_PKG_REAL_PATH}/All/*default-config*.txz ${FINAL_CHROOT_DIR}/pkgs
1273
	fi
1274

    
1275
	pkg_chroot_add ${FINAL_CHROOT_DIR} ${_default_config}
1276
}
1277

    
1278
create_distribution_tarball() {
1279
	mkdir -p ${FINAL_CHROOT_DIR}/install
1280

    
1281
	tar -C ${FINAL_CHROOT_DIR} --exclude ./install --exclude ./pkgs -cJf ${FINAL_CHROOT_DIR}/install/${PRODUCT_NAME}.txz .
1282
}
1283

    
1284
create_iso_image() {
1285
	LOGFILE=${BUILDER_LOGS}/isoimage.${TARGET}
1286
	echo ">>> Building bootable ISO image for ${TARGET}" | tee -a ${LOGFILE}
1287
	if [ -z "${DEFAULT_KERNEL}" ]; then
1288
		echo ">>> ERROR: Could not identify DEFAULT_KERNEL to install on image!" | tee -a ${LOGFILE}
1289
		print_error_pfS
1290
	fi
1291

    
1292
	customize_stagearea_for_image "iso"
1293
	install_default_kernel ${DEFAULT_KERNEL}
1294

    
1295
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1296

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

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

    
1304
	create_distribution_tarball
1305

    
1306
	# Remove /rescue from iso since cd9660 cannot deal with hardlinks
1307
	rm -rf ${FINAL_CHROOT_DIR}/rescue
1308

    
1309
	makefs -t cd9660 -o bootimage="i386;${FINAL_CHROOT_DIR}/boot/cdboot" -o no-emul-boot -o rockridge \
1310
		-o label=${FSLABEL} -o publisher="${PRODUCT_NAME} project." $ISOPATH ${FINAL_CHROOT_DIR} 2>&1 >> ${LOGFILE}
1311
	if [ $? -ne 0 -o ! -f $ISOPATH ]; then
1312
		if [ -f ${ISOPATH} ]; then
1313
			rm -f $ISOPATH
1314
		fi
1315
		echo ">>> ERROR: Something wrong happened during ISO image creation. STOPPING!" | tee -a ${LOGFILE}
1316
		print_error_pfS
1317
	fi
1318
	gzip -qf $ISOPATH &
1319
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1320

    
1321
	echo ">>> ISO created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1322
}
1323

    
1324
create_memstick_image() {
1325

    
1326
	LOGFILE=${BUILDER_LOGS}/memstick.${TARGET}
1327
	if [ "${MEMSTICKPATH}" = "" ]; then
1328
		echo ">>> MEMSTICKPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1329
		return
1330
	fi
1331

    
1332
	customize_stagearea_for_image "memstick"
1333
	install_default_kernel ${DEFAULT_KERNEL}
1334

    
1335
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1336

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

    
1341
	create_distribution_tarball
1342

    
1343
	makefs -B little -o label=${PRODUCT_NAME},version=2 ${MEMSTICKPATH} ${FINAL_CHROOT_DIR}
1344
	if [ $? -ne 0 ]; then
1345
		if [ -f ${MEMSTICKPATH} ]; then
1346
			rm -f $MEMSTICKPATH
1347
		fi
1348
		echo ">>> ERROR: Something wrong happened during MEMSTICK image creation. STOPPING!" | tee -a ${LOGFILE}
1349
		print_error_pfS
1350
	fi
1351
	MD=$(mdconfig -a -t vnode -f $MEMSTICKPATH)
1352
	# Just in case
1353
	trap "mdconfig -d -u ${MD}" 1 2 15 EXIT
1354
	gpart create -s BSD ${MD} 2>&1 >> ${LOGFILE}
1355
	gpart bootcode -b ${FINAL_CHROOT_DIR}/boot/boot ${MD} 2>&1 >> ${LOGFILE}
1356
	gpart add -t freebsd-ufs ${MD} 2>&1 >> ${LOGFILE}
1357
	trap "-" 1 2 15 EXIT
1358
	mdconfig -d -u ${MD} 2>&1 | tee -a ${LOGFILE}
1359
	gzip -qf $MEMSTICKPATH &
1360
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1361

    
1362
	echo ">>> MEMSTICK created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1363
}
1364

    
1365
create_memstick_serial_image() {
1366
	LOGFILE=${BUILDER_LOGS}/memstickserial.${TARGET}
1367
	if [ "${MEMSTICKSERIALPATH}" = "" ]; then
1368
		echo ">>> MEMSTICKSERIALPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1369
		return
1370
	fi
1371

    
1372
	customize_stagearea_for_image "memstickserial"
1373
	install_default_kernel ${DEFAULT_KERNEL}
1374

    
1375
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1376

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

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

    
1382
	BOOTCONF=${FINAL_CHROOT_DIR}/boot.config
1383
	LOADERCONF=${FINAL_CHROOT_DIR}/boot/loader.conf
1384

    
1385
	echo ">>> Activating serial console..." 2>&1 | tee -a ${LOGFILE}
1386
	# Activate serial console in boot.config
1387
	if [ -f ${BOOTCONF} ]; then
1388
		sed -i "" '/-D$/d' ${BOOTCONF}
1389
	fi
1390
	echo "-S115200 -D" >> ${BOOTCONF}
1391

    
1392
	# Remove old console options if present.
1393
	[ -f "${LOADERCONF}" ] \
1394
		&& sed -i "" -Ee "/(console|boot_multicons|boot_serial)/d" ${LOADERCONF}
1395
	# Activate serial console+video console in loader.conf
1396
	echo 'boot_multicons="YES"' >>  ${LOADERCONF}
1397
	echo 'boot_serial="YES"' >> ${LOADERCONF}
1398
	echo 'console="comconsole,vidconsole"' >> ${LOADERCONF}
1399
	echo 'comconsole_speed="115200"' >> ${LOADERCONF}
1400

    
1401
	create_distribution_tarball
1402

    
1403
	makefs -B little -o label=${PRODUCT_NAME},version=2 ${MEMSTICKSERIALPATH} ${FINAL_CHROOT_DIR}
1404
	if [ $? -ne 0 ]; then
1405
		if [ -f ${MEMSTICKSERIALPATH} ]; then
1406
			rm -f $MEMSTICKSERIALPATH
1407
		fi
1408
		echo ">>> ERROR: Something wrong happened during MEMSTICKSERIAL image creation. STOPPING!" | tee -a ${LOGFILE}
1409
		print_error_pfS
1410
	fi
1411
	MD=$(mdconfig -a -t vnode -f $MEMSTICKSERIALPATH)
1412
	# Just in case
1413
	trap "mdconfig -d -u ${MD}" 1 2 15 EXIT
1414
	gpart create -s BSD ${MD} 2>&1 >> ${LOGFILE}
1415
	gpart bootcode -b ${FINAL_CHROOT_DIR}/boot/boot ${MD} 2>&1 >> ${LOGFILE}
1416
	gpart add -t freebsd-ufs ${MD} 2>&1 >> ${LOGFILE}
1417
	trap "-" 1 2 15 EXIT
1418
	mdconfig -d -u ${MD} 2>&1 >> ${LOGFILE}
1419
	gzip -qf $MEMSTICKSERIALPATH &
1420
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1421

    
1422
	echo ">>> MEMSTICKSERIAL created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1423
}
1424

    
1425
create_memstick_adi_image() {
1426
	LOGFILE=${BUILDER_LOGS}/memstickadi.${TARGET}
1427
	if [ "${MEMSTICKADIPATH}" = "" ]; then
1428
		echo ">>> MEMSTICKADIPATH is empty skipping generation of memstick image!" | tee -a ${LOGFILE}
1429
		return
1430
	fi
1431

    
1432
	customize_stagearea_for_image "memstickadi"
1433
	install_default_kernel ${DEFAULT_KERNEL}
1434

    
1435
	echo cdrom > $FINAL_CHROOT_DIR/etc/platform
1436

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

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

    
1442
	BOOTCONF=${FINAL_CHROOT_DIR}/boot.config
1443
	LOADERCONF=${FINAL_CHROOT_DIR}/boot/loader.conf
1444

    
1445
	echo ">>> Activating serial console..." 2>&1 | tee -a ${LOGFILE}
1446
	# Activate serial console in boot.config
1447
	if [ -f ${BOOTCONF} ]; then
1448
		sed -i "" '/-[Dh]$/d' ${BOOTCONF}
1449
	fi
1450
	echo "-S115200 -h" >> ${BOOTCONF}
1451

    
1452
	# Remove old console options if present.
1453
	[ -f "${LOADERCONF}" ] \
1454
		&& sed -i "" -Ee "/(console|boot_multicons|boot_serial|hint.uart)/d" ${LOADERCONF}
1455
	# Activate serial console+video console in loader.conf
1456
	echo 'boot_serial="YES"' >> ${LOADERCONF}
1457
	echo 'console="comconsole"' >> ${LOADERCONF}
1458
	echo 'comconsole_speed="115200"' >> ${LOADERCONF}
1459
	echo 'comconsole_port="0x2F8"' >> ${LOADERCONF}
1460
	echo 'hint.uart.0.flags="0x00"' >> ${LOADERCONF}
1461
	echo 'hint.uart.1.flags="0x10"' >> ${LOADERCONF}
1462

    
1463
	create_distribution_tarball
1464

    
1465
	makefs -B little -o label=${PRODUCT_NAME},version=2 ${MEMSTICKADIPATH} ${FINAL_CHROOT_DIR}
1466
	if [ $? -ne 0 ]; then
1467
		if [ -f ${MEMSTICKADIPATH} ]; then
1468
			rm -f $MEMSTICKADIPATH
1469
		fi
1470
		echo ">>> ERROR: Something wrong happened during MEMSTICKADI image creation. STOPPING!" | tee -a ${LOGFILE}
1471
		print_error_pfS
1472
	fi
1473
	MD=$(mdconfig -a -t vnode -f $MEMSTICKADIPATH)
1474
	# Just in case
1475
	trap "mdconfig -d -u ${MD}" 1 2 15 EXIT
1476
	gpart create -s BSD ${MD} 2>&1 >> ${LOGFILE}
1477
	gpart bootcode -b ${FINAL_CHROOT_DIR}/boot/boot ${MD} 2>&1 >> ${LOGFILE}
1478
	gpart add -t freebsd-ufs ${MD} 2>&1 >> ${LOGFILE}
1479
	trap "-" 1 2 15 EXIT
1480
	mdconfig -d -u ${MD} 2>&1 >> ${LOGFILE}
1481
	gzip -qf $MEMSTICKADIPATH &
1482
	_bg_pids="${_bg_pids}${_bg_pids:+ }$!"
1483

    
1484
	echo ">>> MEMSTICKADI created: $(LC_ALL=C date)" | tee -a ${LOGFILE}
1485
}
1486

    
1487
# Create pkg conf on desired place with desired arch/branch
1488
setup_pkg_repo() {
1489
	if [ -z "${4}" ]; then
1490
		return
1491
	fi
1492

    
1493
	local _target="${1}"
1494
	local _arch="${2}"
1495
	local _target_arch="${3}"
1496
	local _branch="${4}"
1497
	local _release="${5}"
1498

    
1499
	if [ -n "${_release}" ]; then
1500
		local _template="${PKG_REPO_TEMPLATE}"
1501
	else
1502
		local _template="${PKG_REPO_DEVEL_TEMPLATE}"
1503
	fi
1504

    
1505
	if [ -z "${_template}" -o ! -f "${_template}" ]; then
1506
		echo ">>> ERROR: It was not possible to find pkg conf template ${_template}"
1507
		print_error_pfS
1508
	fi
1509

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

    
1512
	sed \
1513
		-e "s/%%ARCH%%/${_arch}_${_target_arch}/" \
1514
		-e "s/%%GIT_REPO_BRANCH_OR_TAG%%/${_branch}/g" \
1515
		-e "s,%%PKG_REPO_SERVER%%,${PKG_REPO_SERVER},g" \
1516
		-e "s/%%PRODUCT_NAME%%/${PRODUCT_NAME}/g" \
1517
		${_template} \
1518
		> ${_target}
1519
}
1520

    
1521
# This routine ensures any ports / binaries that the builder
1522
# system needs are on disk and ready for execution.
1523
builder_setup() {
1524
	# If Product-builder is already installed, just leave
1525
	if pkg info -e -q ${PRODUCT_NAME}-builder; then
1526
		return
1527
	fi
1528

    
1529
	if [ ! -f ${PKG_REPO_PATH} ]; then
1530
		[ -d $(dirname ${PKG_REPO_PATH}) ] \
1531
			|| mkdir -p $(dirname ${PKG_REPO_PATH})
1532

    
1533
		update_freebsd_sources
1534

    
1535
		local _arch=$(uname -m)
1536
		setup_pkg_repo \
1537
			${PKG_REPO_PATH} \
1538
			${_arch} \
1539
			${_arch} \
1540
			${PKG_REPO_CONF_BRANCH} \
1541
			${IS_RELEASE}
1542

    
1543
		# Use fingerprint keys from repo
1544
		sed -i '' -e "/fingerprints:/ s,\"/,\"${BUILDER_ROOT}/src/," \
1545
			${PKG_REPO_PATH}
1546
	fi
1547

    
1548
	pkg install ${PRODUCT_NAME}-builder
1549
}
1550

    
1551
# Updates FreeBSD sources
1552
update_freebsd_sources() {
1553
	if [ "${1}" = "full" ]; then
1554
		local _full=1
1555
		local _clone_params=""
1556
	else
1557
		local _full=0
1558
		local _clone_params="--depth 1 --single-branch"
1559
	fi
1560

    
1561
	if [ ! -d "${FREEBSD_SRC_DIR}" ]; then
1562
		mkdir -p ${FREEBSD_SRC_DIR}
1563
	fi
1564

    
1565
	if [ -n "${NO_BUILDWORLD}" -a -n "${NO_BUILDKERNEL}" ]; then
1566
		echo ">>> NO_BUILDWORLD and NO_BUILDKERNEL set, skipping update of freebsd sources" | tee -a ${LOGFILE}
1567
		return
1568
	fi
1569

    
1570
	echo -n ">>> Obtaining FreeBSD sources ${FREEBSD_BRANCH}..."
1571
	local _FREEBSD_BRANCH=${FREEBSD_BRANCH:-"devel"}
1572
	local _CLONE=1
1573

    
1574
	if [ -d "${FREEBSD_SRC_DIR}/.git" ]; then
1575
		CUR_BRANCH=$(cd ${FREEBSD_SRC_DIR} && git branch | grep '^\*' | cut -d' ' -f2)
1576
		if [ ${_full} -eq 0 -a "${CUR_BRANCH}" = "${_FREEBSD_BRANCH}" ]; then
1577
			_CLONE=0
1578
			( cd ${FREEBSD_SRC_DIR} && git clean -fd; git fetch origin; git reset --hard origin/${_FREEBSD_BRANCH} ) 2>&1 | grep -C3 -i -E 'error|fatal'
1579
		else
1580
			rm -rf ${FREEBSD_SRC_DIR}
1581
		fi
1582
	fi
1583

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

    
1588
	if [ ! -d "${FREEBSD_SRC_DIR}/.git" ]; then
1589
		echo ">>> ERROR: It was not possible to clone FreeBSD src repo"
1590
		print_error_pfS
1591
	fi
1592

    
1593
	if [ -n "${GIT_FREEBSD_COSHA1}" ]; then
1594
		( cd ${FREEBSD_SRC_DIR} && git checkout ${GIT_FREEBSD_COSHA1} ) 2>&1 | grep -C3 -i -E 'error|fatal'
1595
	fi
1596
	echo "Done!"
1597
}
1598

    
1599
pkg_chroot() {
1600
	local _root="${1}"
1601
	shift
1602

    
1603
	if [ $# -eq 0 ]; then
1604
		return -1
1605
	fi
1606

    
1607
	if [ -z "${_root}" -o "${_root}" = "/" -o ! -d "${_root}" ]; then
1608
		return -1
1609
	fi
1610

    
1611
	mkdir -p \
1612
		${SCRATCHDIR}/pkg_cache \
1613
		${_root}/var/cache/pkg \
1614
		${_root}/dev
1615

    
1616
	/sbin/mount -t nullfs ${SCRATCHDIR}/pkg_cache ${_root}/var/cache/pkg
1617
	/sbin/mount -t devfs devfs ${_root}/dev
1618
	cp -f /etc/resolv.conf ${_root}/etc/resolv.conf
1619
	touch ${BUILDER_LOGS}/install_pkg_install_ports.txt
1620
	script -aq ${BUILDER_LOGS}/install_pkg_install_ports.txt pkg -c ${_root} $@ >/dev/null 2>&1
1621
	rm -f ${_root}/etc/resolv.conf
1622
	/sbin/umount -f ${_root}/dev
1623
	/sbin/umount -f ${_root}/var/cache/pkg
1624
}
1625

    
1626

    
1627
pkg_chroot_add() {
1628
	if [ -z "${1}" -o -z "${2}" ]; then
1629
		return 1
1630
	fi
1631

    
1632
	local _target="${1}"
1633
	local _pkg="$(get_pkg_name ${2}).txz"
1634

    
1635
	if [ ! -d "${_target}" ]; then
1636
		echo ">>> ERROR: Target dir ${_target} not found"
1637
		print_error_pfS
1638
	fi
1639

    
1640
	if [ ! -f ${CORE_PKG_REAL_PATH}/All/${_pkg} ]; then
1641
		echo ">>> ERROR: Package ${_pkg} not found"
1642
		print_error_pfS
1643
	fi
1644

    
1645
	cp ${CORE_PKG_REAL_PATH}/All/${_pkg} ${_target}
1646
	pkg_chroot ${_target} add /${_pkg}
1647
	rm -f ${_target}/${_pkg}
1648
}
1649

    
1650
pkg_bootstrap() {
1651
	local _root=${1:-"${STAGE_CHROOT_DIR}"}
1652

    
1653
	setup_pkg_repo \
1654
		${_root}${PKG_REPO_PATH} \
1655
		${TARGET} \
1656
		${TARGET_ARCH} \
1657
		${PKG_REPO_CONF_BRANCH} \
1658
		${IS_RELEASE}
1659

    
1660
	pkg_chroot ${_root} bootstrap -f
1661
}
1662

    
1663
# This routine assists with installing various
1664
# freebsd ports files into the pfsense-fs staging
1665
# area.
1666
install_pkg_install_ports() {
1667
	local MAIN_PKG="${1}"
1668

    
1669
	if [ -z "${MAIN_PKG}" ]; then
1670
		MAIN_PKG=${PRODUCT_NAME}
1671
	fi
1672

    
1673
	echo ">>> Installing pkg repository in chroot (${STAGE_CHROOT_DIR})..."
1674

    
1675
	[ -d ${STAGE_CHROOT_DIR}/var/cache/pkg ] || \
1676
		mkdir -p ${STAGE_CHROOT_DIR}/var/cache/pkg
1677

    
1678
	[ -d ${SCRATCHDIR}/pkg_cache ] || \
1679
		mkdir -p ${SCRATCHDIR}/pkg_cache
1680

    
1681
	echo ">>> Installing built ports (packages) in chroot (${STAGE_CHROOT_DIR})... (starting)"
1682
	# First mark all packages as automatically installed
1683
	pkg_chroot ${STAGE_CHROOT_DIR} set -A 1 -a
1684
	# Install all necessary packages
1685
	pkg_chroot ${STAGE_CHROOT_DIR} install ${MAIN_PKG} ${custom_package_list}
1686
	# Make sure required packages are set as non-automatic
1687
	pkg_chroot ${STAGE_CHROOT_DIR} set -A 0 pkg ${MAIN_PKG} ${custom_package_list}
1688
	# Remove unnecessary packages
1689
	pkg_chroot ${STAGE_CHROOT_DIR} autoremove
1690
	echo ">>> Installing built ports (packages) in chroot (${STAGE_CHROOT_DIR})... (finshied)"
1691
}
1692

    
1693
install_bsdinstaller() {
1694
	echo ">>> Installing BSDInstaller in chroot (${FINAL_CHROOT_DIR})... (starting)"
1695
	pkg_chroot ${FINAL_CHROOT_DIR} install -f bsdinstaller
1696
	sed -i '' -e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME}," \
1697
		  -e "s,%%PRODUCT_VERSION%%,${PRODUCT_VERSION}," \
1698
		  -e "s,%%ARCH%%,${TARGET}," \
1699
		  ${FINAL_CHROOT_DIR}/usr/local/share/dfuibe_lua/conf/pfSense.lua \
1700
		  ${FINAL_CHROOT_DIR}/usr/local/share/dfuibe_lua/conf/pfSense_rescue.lua
1701
	echo ">>> Installing BSDInstaller in chroot (${FINAL_CHROOT_DIR})... (finished)"
1702
}
1703

    
1704
staginareas_clean_each_run() {
1705
	echo -n ">>> Cleaning build directories: "
1706
	if [ -d "${FINAL_CHROOT_DIR}" ]; then
1707
		BASENAME=$(basename ${FINAL_CHROOT_DIR})
1708
		echo -n "$BASENAME "
1709
		chflags -R noschg ${FINAL_CHROOT_DIR} 2>&1 >/dev/null
1710
		rm -rf ${FINAL_CHROOT_DIR}/* 2>/dev/null
1711
	fi
1712
	echo "Done!"
1713
}
1714

    
1715
# Imported from FreeSBIE
1716
buildkernel() {
1717
	if [ -n "${NO_BUILDKERNEL}" ]; then
1718
		echo ">>> NO_BUILDKERNEL set, skipping build" | tee -a ${LOGFILE}
1719
		return
1720
	fi
1721

    
1722
	if [ -z "${KERNCONF}" ]; then
1723
		echo ">>> ERROR: No kernel configuration defined probably this is not what you want! STOPPING!" | tee -a ${LOGFILE}
1724
		print_error_pfS
1725
	fi
1726

    
1727
	if [ -n "${KERNELCONF}" ]; then
1728
		export KERNCONFDIR=$(dirname ${KERNELCONF})
1729
		export KERNCONF=$(basename ${KERNELCONF})
1730
	fi
1731

    
1732
	SRCCONFBASENAME=$(basename ${SRC_CONF})
1733
	echo ">>> KERNCONFDIR: ${KERNCONFDIR}"
1734
	echo ">>> ARCH:        ${TARGET}"
1735
	echo ">>> SRC_CONF:    ${SRCCONFBASENAME}"
1736

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

    
1742
# Imported from FreeSBIE
1743
installkernel() {
1744
	if [ -z "${KERNCONF}" ]; then
1745
		echo ">>> ERROR: No kernel configuration defined probably this is not what you want! STOPPING!" | tee -a ${LOGFILE}
1746
		print_error_pfS
1747
	fi
1748

    
1749
	if [ -n "${KERNELCONF}" ]; then
1750
		export KERNCONFDIR=$(dirname ${KERNELCONF})
1751
		export KERNCONF=$(basename ${KERNELCONF})
1752
	fi
1753

    
1754
	mkdir -p ${STAGE_CHROOT_DIR}/boot
1755
	makeargs="${MAKEJ_KERNEL} SRCCONF=${SRC_CONF} __MAKE_CONF=${MAKE_CONF} TARGET_ARCH=${TARGET_ARCH} TARGET=${TARGET} DESTDIR=${KERNEL_DESTDIR}"
1756
	echo ">>> Builder is running the command: script -aq $LOGFILE make ${makeargs} installkernel KERNCONF=${KERNCONF}"  | tee -a $LOGFILE
1757
	(script -aq $LOGFILE make -C ${FREEBSD_SRC_DIR} ${makeargs} installkernel KERNCONF=${KERNCONF} || print_error_pfS;) | egrep '^>>>'
1758
	gzip -f9 $KERNEL_DESTDIR/boot/kernel/kernel
1759
}
1760

    
1761
# Launch is ran first to setup a few variables that we need
1762
# Imported from FreeSBIE
1763
launch() {
1764
	if [ "$(id -u)" != "0" ]; then
1765
		echo "Sorry, this must be done as root."
1766
	fi
1767

    
1768
	echo ">>> Operation $0 has started at $(date)"
1769
}
1770

    
1771
finish() {
1772
	echo ">>> Operation $0 has ended at $(date)"
1773
}
1774

    
1775
pkg_repo_rsync() {
1776
	local _repo_path="${1}"
1777

    
1778
	if [ -z "${_repo_path}" -o ! -d "${_repo_path}" ]; then
1779
		return
1780
	fi
1781

    
1782
	if [ -z "${LOGFILE}" ]; then
1783
		local _logfile="/dev/null"
1784
	else
1785
		local _logfile="${LOGFILE}"
1786
	fi
1787

    
1788
	if [ -n "${PKG_REPO_SIGNING_COMMAND}" ]; then
1789

    
1790
		# Detect poudriere directory structure
1791
		if [ -L "${_repo_path}/.latest" ]; then
1792
			local _real_repo_path=$(readlink -f ${_repo_path}/.latest)
1793
		else
1794
			local _real_repo_path=${_repo_path}
1795
		fi
1796

    
1797
		echo -n ">>> Signing repository... " | tee -a ${_logfile}
1798
		############ ATTENTION ##############
1799
		#
1800
		# For some reason pkg-repo fail without / in the end of directory name
1801
		# so removing it will break command
1802
		#
1803
		# https://github.com/freebsd/pkg/issues/1364
1804
		#
1805
		if script -aq ${_logfile} pkg repo ${_real_repo_path}/ \
1806
		    signing_command: ${PKG_REPO_SIGNING_COMMAND} >/dev/null 2>&1; then
1807
			echo "Done!" | tee -a ${_logfile}
1808
		else
1809
			echo "Failed!" | tee -a ${_logfile}
1810
			echo ">>> ERROR: An error occurred trying to sign repo"
1811
			print_error_pfS
1812
		fi
1813

    
1814
		local _pkgfile="${_repo_path}/Latest/pkg.txz"
1815
		if [ -e ${_pkgfile} ]; then
1816
			echo -n ">>> Signing Latest/pkg.txz for bootstraping... " | tee -a ${_logfile}
1817

    
1818
			if sha256 -q ${_pkgfile} | ${PKG_REPO_SIGNING_COMMAND} \
1819
			    > ${_pkgfile}.sig 2>/dev/null; then
1820
				echo "Done!" | tee -a ${_logfile}
1821
			else
1822
				echo "Failed!" | tee -a ${_logfile}
1823
				echo ">>> ERROR: An error occurred trying to sign Latest/pkg.txz"
1824
				print_error_pfS
1825
			fi
1826
		fi
1827
	fi
1828

    
1829
	if [ -n "${DO_NOT_UPLOAD}" ]; then
1830
		return
1831
	fi
1832

    
1833
	echo -n ">>> Sending updated repository to ${PKG_RSYNC_HOSTNAME}... " | tee -a ${_logfile}
1834
	if script -aq ${_logfile} rsync -ave "ssh -p ${PKG_RSYNC_SSH_PORT}" \
1835
		--timeout=60 --delete-delay ${_repo_path} \
1836
		${PKG_RSYNC_USERNAME}@${PKG_RSYNC_HOSTNAME}:${PKG_RSYNC_DESTDIR} >/dev/null 2>&1
1837
	then
1838
		echo "Done!" | tee -a ${_logfile}
1839
	else
1840
		echo "Failed!" | tee -a ${_logfile}
1841
		echo ">>> ERROR: An error occurred sending repo to remote hostname"
1842
		print_error_pfS
1843
	fi
1844
}
1845

    
1846
poudriere_create_patch() {
1847
	local _jail_patch="${SCRATCHDIR}/poudriere_jail.${GIT_REPO_BRANCH_OR_TAG}.patch"
1848

    
1849
	if [ -z "${FREEBSD_PARENT_BRANCH}" ]; then
1850
		echo ">>> ERROR: FREEBSD_PARENT_BRANCH is not set"
1851
	fi
1852

    
1853
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1854

    
1855
	# Get FreeBSD source and apply patches
1856
	update_freebsd_sources full
1857

    
1858
	[ -f "${_jail_patch}" ] && \
1859
		rm -f "${_jail_patch}"
1860

    
1861
	# Create a big patch with all our changes to use on jail
1862
	( \
1863
		cd ${FREEBSD_SRC_DIR} && \
1864
		git diff $(git merge-base origin/${FREEBSD_PARENT_BRANCH} ${FREEBSD_BRANCH}) > ${_jail_patch}
1865
	) >/dev/null 2>&1
1866

    
1867
	# Check if patch was created
1868
	if [ ! -s "${_jail_patch}" ]; then
1869
		echo ">>> ERROR: Patch does not exist or is empty, aborting..." | tee -a ${LOGFILE}
1870
		print_error_pfS
1871
	fi
1872
}
1873

    
1874
poudriere_possible_archs() {
1875
	local _arch=$(uname -m)
1876
	local _archs="i386.i386"
1877

    
1878
	# If host is amd64, we'll create both repos, and if possible armv6
1879
	if [ "${_arch}" = "amd64" ]; then
1880
		_archs="amd64.amd64 ${_archs}"
1881

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

    
1886
			if binmiscctl lookup armv6 >/dev/null 2>&1; then
1887
				_archs="${_archs} arm.armv6"
1888
			fi
1889
		fi
1890
	fi
1891

    
1892
	if [ -n "${ARCH_LIST}" ]; then
1893
		local _found=0
1894
		for _desired_arch in ${ARCH_LIST}; do
1895
			_found=0
1896
			for _possible_arch in ${_archs}; do
1897
				if [ "${_desired_arch}" = "${_possible_arch}" ]; then
1898
					_found=1
1899
					break
1900
				fi
1901
			done
1902
			if [ ${_found} -eq 0 ]; then
1903
				echo ">>> ERROR: Impossible to build for arch: ${_desired_arch}"
1904
				print_error_pfS
1905
			fi
1906
		done
1907
		_archs="${ARCH_LIST}"
1908
	fi
1909

    
1910
	echo ${_archs}
1911
}
1912

    
1913
poudriere_jail_name() {
1914
	local _jail_arch="${1}"
1915

    
1916
	if [ -z "${_jail_arch}" ]; then
1917
		return 1
1918
	fi
1919

    
1920
	# Poudriere doesn't like periods in jail names
1921
	_jail_arch=$(echo "${_jail_arch}" | tr '.' '_')
1922

    
1923
	echo "${PRODUCT_NAME}_${GIT_REPO_BRANCH_OR_TAG}_${_jail_arch}"
1924
}
1925

    
1926
poudriere_rename_ports() {
1927
	if [ "${PRODUCT_NAME}" = "pfSense" ]; then
1928
		return;
1929
	fi
1930

    
1931
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1932

    
1933
	local _ports_dir="/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}"
1934

    
1935
	echo -n ">>> Renaming product ports on ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
1936
	for d in $(find ${_ports_dir} -depth 2 -type d -name '*pfSense*'); do
1937
		local _pdir=$(dirname ${d})
1938
		local _pname=$(echo $(basename ${d}) | sed "s,pfSense,${PRODUCT_NAME},")
1939
		local _plist=""
1940

    
1941
		if [ -e ${_pdir}/${_pname} ]; then
1942
			rm -rf ${_pdir}/${_pname}
1943
		fi
1944

    
1945
		cp -r ${d} ${_pdir}/${_pname}
1946

    
1947
		if [ -f ${_pdir}/${_pname}/pkg-plist ]; then
1948
			_plist=${_pdir}/${_pname}/pkg-plist
1949
		fi
1950

    
1951
		sed -i '' -e "s,pfSense,${PRODUCT_NAME},g" \
1952
			  -e "s,https://www.pfsense.org,${PRODUCT_URL},g" \
1953
			  -e "/^MAINTAINER=/ s,^.*$,MAINTAINER=	${PRODUCT_EMAIL}," \
1954
			${_pdir}/${_pname}/Makefile \
1955
			${_pdir}/${_pname}/pkg-descr ${_plist}
1956

    
1957
		# PHP module is special
1958
		if echo "${_pname}" | grep -q "^php[0-9]*-${PRODUCT_NAME}-module"; then
1959
			local _product_capital=$(echo ${PRODUCT_NAME} | tr '[a-z]' '[A-Z]')
1960
			sed -i '' -e "s,PHP_PFSENSE,PHP_${_product_capital},g" \
1961
				  -e "s,PFSENSE_SHARED_LIBADD,${_product_capital}_SHARED_LIBADD,g" \
1962
				  -e "s,pfSense,${PRODUCT_NAME},g" \
1963
				  -e "s,${PRODUCT_NAME}\.c,pfSense.c,g" \
1964
				${_pdir}/${_pname}/files/config.m4
1965

    
1966
			sed -i '' -e "s,COMPILE_DL_PFSENSE,COMPILE_DL_${_product_capital}," \
1967
				  -e "s,pfSense_module_entry,${PRODUCT_NAME}_module_entry,g" \
1968
				  -e "/ZEND_GET_MODULE/ s,pfSense,${PRODUCT_NAME}," \
1969
				  -e "/PHP_PFSENSE_WORLD_EXTNAME/ s,pfSense,${PRODUCT_NAME}," \
1970
				${_pdir}/${_pname}/files/pfSense.c \
1971
				${_pdir}/${_pname}/files/php_pfSense.h
1972
		fi
1973

    
1974
		if [ -d ${_pdir}/${_pname}/files ]; then
1975
			for fd in $(find ${_pdir}/${_pname}/files -type d -name '*pfSense*'); do
1976
				local _fddir=$(dirname ${fd})
1977
				local _fdname=$(echo $(basename ${fd}) | sed "s,pfSense,${PRODUCT_NAME},")
1978

    
1979
				mv ${fd} ${_fddir}/${_fdname}
1980
			done
1981
		fi
1982
	done
1983
	echo "Done!" | tee -a ${LOGFILE}
1984
}
1985

    
1986
poudriere_create_ports_tree() {
1987
	LOGFILE=${BUILDER_LOGS}/poudriere.log
1988

    
1989
	if ! poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
1990
		local _branch=""
1991
		if [ -z "${POUDRIERE_PORTS_GIT_URL}" ]; then
1992
			echo ">>> ERROR: POUDRIERE_PORTS_GIT_URL is not defined"
1993
			print_error_pfS
1994
		fi
1995
		if [ -n "${POUDRIERE_PORTS_GIT_BRANCH}" ]; then
1996
			_branch="-B ${POUDRIERE_PORTS_GIT_BRANCH}"
1997
		fi
1998
		echo -n ">>> Creating poudriere ports tree, it may take some time... " | tee -a ${LOGFILE}
1999
		if ! script -aq ${LOGFILE} poudriere ports -c -p "${POUDRIERE_PORTS_NAME}" -m git ${_branch} >/dev/null 2>&1; then
2000
			echo "" | tee -a ${LOGFILE}
2001
			echo ">>> ERROR: Error creating poudriere ports tree, aborting..." | tee -a ${LOGFILE}
2002
			print_error_pfS
2003
		fi
2004
		echo "Done!" | tee -a ${LOGFILE}
2005
		poudriere_rename_ports
2006
	fi
2007
}
2008

    
2009
poudriere_init() {
2010
	local _error=0
2011
	local _archs=$(poudriere_possible_archs)
2012
	local _jail_patch="${SCRATCHDIR}/poudriere_jail.${GIT_REPO_BRANCH_OR_TAG}.patch"
2013

    
2014
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2015

    
2016
	# Sanity checks
2017
	if [ -z "${ZFS_TANK}" ]; then
2018
		echo ">>> ERROR: \$ZFS_TANK is empty" | tee -a ${LOGFILE}
2019
		error=1
2020
	fi
2021

    
2022
	if [ -z "${ZFS_ROOT}" ]; then
2023
		echo ">>> ERROR: \$ZFS_ROOT is empty" | tee -a ${LOGFILE}
2024
		error=1
2025
	fi
2026

    
2027
	if [ -z "${POUDRIERE_PORTS_NAME}" ]; then
2028
		echo ">>> ERROR: \$POUDRIERE_PORTS_NAME is empty" | tee -a ${LOGFILE}
2029
		error=1
2030
	fi
2031

    
2032
	if [ ${_error} -eq 1 ]; then
2033
		print_error_pfS
2034
	fi
2035

    
2036
	# Check if zpool exists
2037
	if ! zpool list ${ZFS_TANK} >/dev/null 2>&1; then
2038
		echo ">>> ERROR: ZFS tank ${ZFS_TANK} not found, please create it and try again..." | tee -a ${LOGFILE}
2039
		print_error_pfS
2040
	fi
2041

    
2042
	# Check if zfs rootfs exists
2043
	if ! zfs list ${ZFS_TANK}${ZFS_ROOT} >/dev/null 2>&1; then
2044
		echo -n ">>> Creating ZFS filesystem ${ZFS_TANK}${ZFS_ROOT}... "
2045
		if zfs create -o atime=off -o mountpoint=/usr/local${ZFS_ROOT} \
2046
		    ${ZFS_TANK}${ZFS_ROOT} >/dev/null 2>&1; then
2047
			echo "Done!"
2048
		else
2049
			echo "Failed!"
2050
			print_error_pfS
2051
		fi
2052
	fi
2053

    
2054
	# Make sure poudriere is installed
2055
	if ! pkg info --quiet poudriere; then
2056
		echo ">>> Installing poudriere..." | tee -a ${LOGFILE}
2057
		if ! pkg install poudriere >/dev/null 2>&1; then
2058
			echo ">>> ERROR: poudriere was not installed, aborting..." | tee -a ${LOGFILE}
2059
			print_error_pfS
2060
		fi
2061
	fi
2062

    
2063
	# Create poudriere.conf
2064
	if [ -z "${POUDRIERE_PORTS_GIT_URL}" ]; then
2065
		echo ">>> ERROR: POUDRIERE_PORTS_GIT_URL is not defined"
2066
		print_error_pfS
2067
	fi
2068
	echo ">>> Creating poudriere.conf" | tee -a ${LOGFILE}
2069
	cat <<EOF >/usr/local/etc/poudriere.conf
2070
ZPOOL=${ZFS_TANK}
2071
ZROOTFS=${ZFS_ROOT}
2072
RESOLV_CONF=/etc/resolv.conf
2073
BASEFS=/usr/local/poudriere
2074
USE_PORTLINT=no
2075
USE_TMPFS=yes
2076
NOLINUX=yes
2077
DISTFILES_CACHE=/usr/ports/distfiles
2078
CHECK_CHANGED_OPTIONS=yes
2079
CHECK_CHANGED_DEPS=yes
2080
ATOMIC_PACKAGE_REPOSITORY=yes
2081
COMMIT_PACKAGES_ON_FAILURE=no
2082
EOF
2083

    
2084
	# Create specific items conf
2085
	[ ! -d /usr/local/etc/poudriere.d ] \
2086
		&& mkdir -p /usr/local/etc/poudriere.d
2087

    
2088
	cat <<EOF >/usr/local/etc/poudriere.d/${POUDRIERE_PORTS_NAME}-poudriere.conf
2089
GIT_URL="${POUDRIERE_PORTS_GIT_URL}"
2090
EOF
2091

    
2092
	# Create DISTFILES_CACHE if it doesn't exist
2093
	if [ ! -d /usr/ports/distfiles ]; then
2094
		mkdir -p /usr/ports/distfiles
2095
	fi
2096

    
2097
	# Remove old jails
2098
	for jail_arch in ${_archs}; do
2099
		jail_name=$(poudriere_jail_name ${jail_arch})
2100

    
2101
		if poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
2102
			echo ">>> Poudriere jail ${jail_name} already exists, deleting it..." | tee -a ${LOGFILE}
2103
			poudriere jail -d -j "${jail_name}" >/dev/null 2>&1
2104
		fi
2105
	done
2106

    
2107
	# Remove old ports tree
2108
	if poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
2109
		echo ">>> Poudriere ports tree ${POUDRIERE_PORTS_NAME} already exists, deleting it..." | tee -a ${LOGFILE}
2110
		poudriere ports -d -p "${POUDRIERE_PORTS_NAME}"
2111
	fi
2112

    
2113
	poudriere_create_patch
2114

    
2115
	local native_xtools=""
2116
	# Now we are ready to create jails
2117
	for jail_arch in ${_archs}; do
2118
		jail_name=$(poudriere_jail_name ${jail_arch})
2119

    
2120
		if [ "${jail_arch}" = "arm.armv6" ]; then
2121
			native_xtools="-x"
2122
		else
2123
			native_xtools=""
2124
		fi
2125

    
2126
		echo -n ">>> Creating jail ${jail_name}, it may take some time... " | tee -a ${LOGFILE}
2127
		# XXX: Change -m to git when it's available in poudriere
2128
		if ! script -aq ${LOGFILE} poudriere jail -c -j "${jail_name}" -v ${FREEBSD_PARENT_BRANCH} \
2129
				-a ${jail_arch} -m svn -P ${_jail_patch} ${native_xtools} >/dev/null 2>&1; then
2130
			echo "" | tee -a ${LOGFILE}
2131
			echo ">>> ERROR: Error creating jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
2132
			print_error_pfS
2133
		fi
2134
		echo "Done!" | tee -a ${LOGFILE}
2135
	done
2136

    
2137
	poudriere_create_ports_tree
2138

    
2139
	echo ">>> Poudriere is now configured!" | tee -a ${LOGFILE}
2140
}
2141

    
2142
poudriere_update_jails() {
2143
	local _archs=$(poudriere_possible_archs)
2144
	local _jail_patch="${SCRATCHDIR}/poudriere_jail.${GIT_REPO_BRANCH_OR_TAG}.patch"
2145

    
2146
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2147

    
2148
	poudriere_create_patch
2149

    
2150
	local native_xtools=""
2151
	for jail_arch in ${_archs}; do
2152
		jail_name=$(poudriere_jail_name ${jail_arch})
2153

    
2154
		local _create_or_update="-u"
2155
		local _create_or_update_text="Updating"
2156
		if ! poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
2157
			echo ">>> Poudriere jail ${jail_name} not found, creating..." | tee -a ${LOGFILE}
2158
			_create_or_update="-c -v ${FREEBSD_PARENT_BRANCH} -a ${jail_arch} -m svn"
2159
			_create_or_update_text="Creating"
2160
		fi
2161

    
2162
		if [ "${jail_arch}" = "arm.armv6" ]; then
2163
			native_xtools="-x"
2164
		else
2165
			native_xtools=""
2166
		fi
2167

    
2168
		echo -n ">>> ${_create_or_update_text} jail ${jail_name}, it may take some time... " | tee -a ${LOGFILE}
2169
		if ! script -aq ${LOGFILE} poudriere jail ${_create_or_update} -j "${jail_name}" -P ${_jail_patch} ${native_xtools} >/dev/null 2>&1; then
2170
			echo "" | tee -a ${LOGFILE}
2171
			echo ">>> ERROR: Error ${_create_or_update_text} jail ${jail_name}, aborting..." | tee -a ${LOGFILE}
2172
			print_error_pfS
2173
		fi
2174
		echo "Done!" | tee -a ${LOGFILE}
2175
	done
2176
}
2177

    
2178
poudriere_update_ports() {
2179
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2180

    
2181
	# Create ports tree if necessary
2182
	if ! poudriere ports -l | grep -q -E "^${POUDRIERE_PORTS_NAME}[[:blank:]]"; then
2183
		poudriere_create_ports_tree
2184
	else
2185
		echo -n ">>> Resetting local changes on ports tree ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
2186
		script -aq ${LOGFILE} git -C "/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}" reset --hard >/dev/null 2>&1
2187
		script -aq ${LOGFILE} git -C "/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}" clean -fd >/dev/null 2>&1
2188
		echo "Done!" | tee -a ${LOGFILE}
2189
		echo -n ">>> Updating ports tree ${POUDRIERE_PORTS_NAME}... " | tee -a ${LOGFILE}
2190
		script -aq ${LOGFILE} poudriere ports -u -p "${POUDRIERE_PORTS_NAME}" >/dev/null 2>&1
2191
		echo "Done!" | tee -a ${LOGFILE}
2192
		poudriere_rename_ports
2193
	fi
2194
}
2195

    
2196
poudriere_bulk() {
2197
	local _archs=$(poudriere_possible_archs)
2198

    
2199
	LOGFILE=${BUILDER_LOGS}/poudriere.log
2200

    
2201
	if [ -z "${DO_NOT_UPLOAD}" -a -z "${PKG_RSYNC_HOSTNAME}" ]; then
2202
		echo ">>> ERROR: PKG_RSYNC_HOSTNAME is not set"
2203
		print_error_pfS
2204
	fi
2205

    
2206
	poudriere_create_ports_tree
2207

    
2208
	[ -d /usr/local/etc/poudriere.d ] || \
2209
		mkdir -p /usr/local/etc/poudriere.d
2210

    
2211
	if [ -f "${BUILDER_TOOLS}/conf/pfPorts/make.conf" ]; then
2212
		cp -f "${BUILDER_TOOLS}/conf/pfPorts/make.conf" /usr/local/etc/poudriere.d/${POUDRIERE_PORTS_NAME}-make.conf
2213
	fi
2214

    
2215
	# Change version of pfSense meta ports for snapshots
2216
	if [ -z "${_IS_RELEASE}" ]; then
2217
		local _meta_pkg_version="$(echo "${PRODUCT_VERSION}" | sed 's,DEVELOPMENT,ALPHA,')-${DATESTRING}"
2218
		sed -i '' -e "/^DISTVERSION/ s,^.*,DISTVERSION=	${_meta_pkg_version}," \
2219
			/usr/local/poudriere/ports/${POUDRIERE_PORTS_NAME}/security/${PRODUCT_NAME}/Makefile
2220
	fi
2221

    
2222
	for jail_arch in ${_archs}; do
2223
		jail_name=$(poudriere_jail_name ${jail_arch})
2224

    
2225
		if ! poudriere jail -i -j "${jail_name}" >/dev/null 2>&1; then
2226
			echo ">>> Poudriere jail ${jail_name} not found, skipping..." | tee -a ${LOGFILE}
2227
			continue
2228
		fi
2229

    
2230
		if [ -f "${POUDRIERE_BULK}.${jail_arch}" ]; then
2231
			_ref_bulk="${POUDRIERE_BULK}.${jail_arch}"
2232
		else
2233
			_ref_bulk="${POUDRIERE_BULK}"
2234
		fi
2235

    
2236
		_bulk=${SCRATCHDIR}/poudriere_bulk.${GIT_REPO_BRANCH_OR_TAG}
2237
		sed -e "s,%%PRODUCT_NAME%%,${PRODUCT_NAME},g" ${_ref_bulk} > ${_bulk}
2238

    
2239
		if ! poudriere bulk -f ${_bulk} -j ${jail_name} -p ${POUDRIERE_PORTS_NAME}; then
2240
			echo ">>> ERROR: Something went wrong..."
2241
			print_error_pfS
2242
		fi
2243

    
2244
		echo ">>> Cleaning up old packages from repo..."
2245
		if ! poudriere pkgclean -f ${_bulk} -j ${jail_name} -p ${POUDRIERE_PORTS_NAME} -y; then
2246
			echo ">>> ERROR: Something went wrong..."
2247
			print_error_pfS
2248
		fi
2249

    
2250
		# ./ is intentional, it's an rsync trick to make it chdir to directory before sending it
2251
		pkg_repo_rsync "/usr/local/poudriere/data/packages/./${jail_name}-${POUDRIERE_PORTS_NAME}"
2252
	done
2253
}
2254

    
2255
# This routine is called to write out to stdout
2256
# a string. The string is appended to $SNAPSHOTSLOGFILE
2257
# and we scp the log file to the builder host if
2258
# needed for the real time logging functions.
2259
snapshots_update_status() {
2260
	if [ -z "$1" ]; then
2261
		return
2262
	fi
2263
	if [ -z "${SNAPSHOTS}" -a -z "${POUDRIERE_SNAPSHOTS}" ]; then
2264
		return
2265
	fi
2266
	echo "$*"
2267
	echo "`date` -|- $*" >> $SNAPSHOTSLOGFILE
2268
	if [ -z "${DO_NOT_UPLOAD}" -a -n "${SNAPSHOTS_RSYNCIP}" ]; then
2269
		LU=$(cat $SNAPSHOTSLASTUPDATE 2>/dev/null)
2270
		CT=$(date "+%H%M%S")
2271
		# Only update every minute
2272
		if [ "$LU" != "$CT" ]; then
2273
			ssh ${SNAPSHOTS_RSYNCUSER}@${SNAPSHOTS_RSYNCIP} \
2274
				"mkdir -p ${SNAPSHOTS_RSYNCLOGS}"
2275
			scp -q $SNAPSHOTSLOGFILE \
2276
				${SNAPSHOTS_RSYNCUSER}@${SNAPSHOTS_RSYNCIP}:${SNAPSHOTS_RSYNCLOGS}/build.log
2277
			date "+%H%M%S" > $SNAPSHOTSLASTUPDATE
2278
		fi
2279
	fi
2280
}
2281

    
2282
# Copy the current log file to $filename.old on
2283
# the snapshot www server (real time logs)
2284
snapshots_rotate_logfile() {
2285
	if [ -z "${DO_NOT_UPLOAD}" -a -n "${SNAPSHOTS_RSYNCIP}" ]; then
2286
		scp -q $SNAPSHOTSLOGFILE \
2287
			${SNAPSHOTS_RSYNCUSER}@${SNAPSHOTS_RSYNCIP}:${SNAPSHOTS_RSYNCLOGS}/build.log.old
2288
	fi
2289

    
2290
	# Cleanup log file
2291
	rm -f $SNAPSHOTSLOGFILE;    touch $SNAPSHOTSLOGFILE
2292
	rm -f $SNAPSHOTSLASTUPDATE; touch $SNAPSHOTSLASTUPDATE
2293

    
2294
}
2295

    
2296
snapshots_create_latest_symlink() {
2297
	local _image="${1}"
2298

    
2299
	if [ -z "${_image}" ]; then
2300
		return
2301
	fi
2302

    
2303
	if [ -z "${TIMESTAMP_SUFFIX}" ]; then
2304
		return
2305
	fi
2306

    
2307
	if [ -f "${_image}.gz" ]; then
2308
		local _image_fixed="${_image}.gz"
2309
	elif [ -f "${_image}" ]; then
2310
		local _image_fixed=${_image}
2311
	else
2312
		return
2313
	fi
2314

    
2315
	local _symlink=$(echo ${_image_fixed} | sed "s,${TIMESTAMP_SUFFIX},-latest,")
2316
	ln -sf $(basename ${_image_fixed}) ${_symlink}
2317
	ln -sf $(basename ${_image}).sha256 ${_symlink}.sha256
2318
}
2319

    
2320
snapshots_copy_to_staging_nanobsd() {
2321
	for NANOTYPE in nanobsd nanobsd-vga; do
2322
		for FILESIZE in ${1}; do
2323
			FILENAMEFULL="$(nanobsd_image_filename ${FILESIZE} ${NANOTYPE}).gz"
2324
			FILENAMEUPGRADE="$(nanobsd_image_filename ${FILESIZE} ${NANOTYPE} 1).gz"
2325
			mkdir -p $STAGINGAREA/nanobsd
2326
			mkdir -p $STAGINGAREA/nanobsdupdates
2327

    
2328
			cp -l $IMAGES_FINAL_DIR/$FILENAMEFULL $STAGINGAREA/nanobsd/ 2>/dev/null
2329
			cp -l $IMAGES_FINAL_DIR/$FILENAMEUPGRADE $STAGINGAREA/nanobsdupdates 2>/dev/null
2330

    
2331
			if [ -f $STAGINGAREA/nanobsd/$FILENAMEFULL ]; then
2332
				sha256 $STAGINGAREA/nanobsd/$FILENAMEFULL > $STAGINGAREA/nanobsd/$FILENAMEFULL.sha256 2>/dev/null
2333
			fi
2334
			if [ -f $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE ]; then
2335
				sha256 $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE > $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE.sha256 2>/dev/null
2336
			fi
2337

    
2338
			# Copy NanoBSD auto update:
2339
			if [ -f $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE ]; then
2340
				cp -l $STAGINGAREA/nanobsdupdates/$FILENAMEUPGRADE $STAGINGAREA/latest-${NANOTYPE}-$FILESIZE.img.gz 2>/dev/null
2341
				sha256 $STAGINGAREA/latest-${NANOTYPE}-$FILESIZE.img.gz > $STAGINGAREA/latest-${NANOTYPE}-$FILESIZE.img.gz.sha256 2>/dev/null
2342
				# NOTE: Updates need a file with output similar to date output
2343
				# Use the file generated at start of snapshots_dobuilds() to be consistent on times
2344
				cp $BUILTDATESTRINGFILE $STAGINGAREA/version-${NANOTYPE}-$FILESIZE
2345
			fi
2346
		done
2347
	done
2348
}
2349

    
2350
snapshots_copy_to_staging_iso_updates() {
2351
	local _img=""
2352

    
2353
	for _img in ${ISOPATH} ${MEMSTICKPATH} ${MEMSTICKSERIALPATH} ${MEMSTICKADIPATH}; do
2354
		if [ ! -f "${_img}.gz" ]; then
2355
			continue
2356
		fi
2357
		sha256 ${_img}.gz > ${_img}.sha256
2358
		cp -l ${_img}* $STAGINGAREA/ 2>/dev/null
2359
		snapshots_create_latest_symlink ${STAGINGAREA}/$(basename ${_img})
2360
	done
2361

    
2362
	if [ -f "${UPDATES_TARBALL_FILENAME}" ]; then
2363
		sha256 ${UPDATES_TARBALL_FILENAME} > ${UPDATES_TARBALL_FILENAME}.sha256
2364
		cp -l ${UPDATES_TARBALL_FILENAME}* $STAGINGAREA/ 2>/dev/null
2365
		snapshots_create_latest_symlink ${STAGINGAREA}/$(basename ${UPDATES_TARBALL_FILENAME})
2366
	fi
2367

    
2368
	if [ -f "${OVAPATH}" ]; then
2369
		mkdir -p ${STAGINGAREA}/virtualization
2370
		sha256 ${OVAPATH} > ${OVAPATH}.sha256
2371
		cp -l ${OVAPATH}* $STAGINGAREA/virtualization 2>/dev/null
2372
		snapshots_create_latest_symlink ${STAGINGAREA}/virtualization/$(basename ${OVAPATH})
2373
	fi
2374

    
2375
	# NOTE: Updates need a file with output similar to date output
2376
	# Use the file generated at start of snapshots_dobuilds() to be consistent on times
2377
	if [ -z "${_IS_RELEASE}" ]; then
2378
		cp $BUILTDATESTRINGFILE $STAGINGAREA/version 2>/dev/null
2379
	fi
2380
}
2381

    
2382
snapshots_scp_files() {
2383
	if [ -z "${RSYNC_COPY_ARGUMENTS}" ]; then
2384
		RSYNC_COPY_ARGUMENTS="-ave ssh --timeout=60 --bwlimit=${RSYNCKBYTELIMIT}" #--bwlimit=50
2385
	fi
2386

    
2387
	snapshots_update_status ">>> Copying core pkg repo to ${PKG_RSYNC_HOSTNAME}"
2388
	# Add ./ before last directory, it's an rsync trick to make it chdir to parent directory before sending
2389
	pkg_repo_rsync $(echo "${CORE_PKG_PATH}" | sed -E 's,/$,,; s,/([^/]*)$,/./\1,')
2390
	snapshots_update_status ">>> Finished copying core pkg repo"
2391

    
2392
	snapshots_update_status ">>> Copying files to ${RSYNCIP}"
2393

    
2394
	# Ensure directory(s) are available
2395
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/installer"
2396
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/updates"
2397
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/nanobsd"
2398
	if [ -d $STAGINGAREA/virtualization ]; then
2399
		ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/virtualization"
2400
	fi
2401
	ssh ${RSYNCUSER}@${RSYNCIP} "mkdir -p ${RSYNCPATH}/.updaters"
2402
	# ensure permissions are correct for r+w
2403
	ssh ${RSYNCUSER}@${RSYNCIP} "chmod -R ug+rw ${RSYNCPATH}/."
2404
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/${PRODUCT_NAME}${PRODUCT_NAME_SUFFIX}-*iso* \
2405
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/installer/
2406
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/${PRODUCT_NAME}${PRODUCT_NAME_SUFFIX}-memstick* \
2407
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/installer/
2408
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/${PRODUCT_NAME}${PRODUCT_NAME_SUFFIX}-*Update* \
2409
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/updates/
2410
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/nanobsd/* \
2411
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/nanobsd/
2412
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/nanobsdupdates/* \
2413
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/updates/
2414
	if [ -d $STAGINGAREA/virtualization ]; then
2415
		rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/virtualization/* \
2416
			${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/virtualization/
2417
	fi
2418

    
2419
	# Rather than copy these twice, use ln to link to the latest one.
2420

    
2421
	ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest.tgz"
2422
	ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest.tgz.sha256"
2423

    
2424
	LATESTFILENAME=$(basename ${UPDATES_TARBALL_FILENAME})
2425
	ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${LATESTFILENAME} \
2426
		${RSYNCPATH}/.updaters/latest.tgz"
2427
	ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${LATESTFILENAME}.sha256 \
2428
		${RSYNCPATH}/.updaters/latest.tgz.sha256"
2429

    
2430
	for i in ${FLASH_SIZE}
2431
	do
2432
		ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz"
2433
		ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz.sha256"
2434
		ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz"
2435
		ssh ${RSYNCUSER}@${RSYNCIP} "rm -f ${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz.sha256"
2436

    
2437
		FILENAMEUPGRADE="$(nanobsd_image_filename ${i} nanobsd 1).gz"
2438
		ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE} \
2439
			${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz"
2440
		ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE}.sha256 \
2441
			${RSYNCPATH}/.updaters/latest-nanobsd-${i}.img.gz.sha256"
2442

    
2443
		FILENAMEUPGRADE="$(nanobsd_image_filename ${i} nanobsd-vga 1).gz"
2444
		ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE} \
2445
			${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz"
2446
		ssh ${RSYNCUSER}@${RSYNCIP} "ln -s ${RSYNCPATH}/updates/${FILENAMEUPGRADE}.sha256 \
2447
			${RSYNCPATH}/.updaters/latest-nanobsd-vga-${i}.img.gz.sha256"
2448
	done
2449

    
2450
	rsync $RSYNC_COPY_ARGUMENTS $STAGINGAREA/version* \
2451
		${RSYNCUSER}@${RSYNCIP}:${RSYNCPATH}/.updaters
2452
	snapshots_update_status ">>> Finished copying files."
2453
}
(2-2/3)