vct-container.sh

#!/bin/bash
#This script is used to create an updated vct container

function tear_down_vct_container(){
	echo "Verify running $VCT_CONTAINER_DIR..."
    if lxc-info -n $VCT_CONTAINER_DIR | grep -q RUNNING; then
		echo "Stopping LXC..."
        lxc-stop -n $VCT_CONTAINER_DIR;
    fi
}

function extract_vct(){
    # fetch/copy the latest VCT container
	if [[ ! -f dl/$VCT_CONTAINER ]]; then
		echo "Downloading $VCT_CONTAINER"
		wget -q --no-check-certificate $VCT_CONTAINER_URL -O dl/$VCT_CONTAINER
	fi
	if [[ ! -f dl/$VCT_CONTAINER ]]; then
		echo "Could not download $VCT_CONTAINER"
		exit 1
	fi
	
    echo "Unpacking..."
	rm -rf $VCT_CONTAINER_DIR/vct
	tar -C $VCT_CONTAINER_DIR --numeric-owner -xJf dl/$VCT_CONTAINER
	
	cd $VCT_CONTAINER_DIR
	echo "Patching..."
	rm -rf .pc # remove old quilt information, if any
	QUILT_PATCHES="../vct/patches" quilt push -a -v
	cd -
	
}

function update_vct() {
	echo "Updating VCT to $VCT_HASH..."
	CDDIR=$VCT_CONTAINER_DIR/vct/rootfs/home/vct/confine-dist
	mv $CDDIR/utils/vct/vct.conf.overrides /tmp/
	rm -rf $VCT_CONTAINER_DIR/vct/rootfs/home/vct/confine-dist
    git clone http://git.confine-project.eu/confine.git $CDDIR
	cd $CDDIR
	git checkout $VCT_HASH
	mv /tmp/vct.conf.overrides ./utils/vct/
	
	VCT_HASH=$(git rev-parse --short $VCT_HASH)
	if ! [ ${#NODEFIRMWARE_HASH} -eq 40 ]; then
		NODEFIRMWARE_HASH=$(git rev-parse --short $NODEFIRMWARE_HASH)
		NODEFIRMWARE_HASH_LONG=$(git rev-parse $NODEFIRMWARE_HASH)
	fi
	cd -
}

function start_vct() {
    echo "Starting LXC..."
    cd $VCT_CONTAINER_DIR
	lxc-start --name $VCT_CONTAINER_DIR -f vct/config -s lxc.rootfs=$(pwd)/vct/rootfs -o vct.log -d
	cd -
}

function clean_vct() {
	echo "Cleaning VCT installation..."
    ssh -i ./sshkey/id_rsa -o StrictHostKeyChecking=no vct@${VCT_IP} '/home/vct/confine-dist/utils/vct/vct_system_cleanup && sudo rm -rf /var/lib/vct' > $VCT_CONTAINER_DIR/vct_system_cleanup.log 2>&1
}

function configure_inet() {
	ssh -i ./sshkey/id_rsa -o StrictHostKeyChecking=no vct@${VCT_IP} "sudo ip addr add ${IPV4PREFIX}2/24 dev eth0 && sudo ip route add default via ${IPV4PREFIX}1 && ping -c 5 8.8.8.8";
}

# function network_vct() {
# 	ssh -i ./sshkey/id_rsa -o StrictHostKeyChecking=no root@${VCT_IP} "ifconfig eth0 $VCT_IP_PUBLIC netmask 255.255.255.224; route add default gw 143.129.77.158; ping -c 1 8.8.8.8"
# }

function install_vct(){
	echo "Installing VCT..."
    ssh -i ./sshkey/id_rsa -o StrictHostKeyChecking=no vct@${VCT_IP} '/home/vct/confine-dist/utils/vct/vct_system_install && sudo apt-get clean' > $VCT_CONTAINER_DIR/vct_system_install.log 2>&1
}

function init_vct(){
	echo "Initialising VCT..."
	ssh -i ./sshkey/id_rsa -o StrictHostKeyChecking=no vct@${VCT_IP} '/home/vct/confine-dist/utils/vct/vct_system_init' > $VCT_CONTAINER_DIR/vct_system_init.log 2>&1
}

function build_node_vct() {
	echo "Building node firmware..."
    ssh -i ./sshkey/id_rsa -o StrictHostKeyChecking=no vct@${VCT_IP} '/home/vct/confine-dist/utils/vct/vct_build_node_base_image'
}

function tar_xz_vct() {
	echo "Packaging VCT lXC..."
    cd $VCT_CONTAINER_DIR
    #clean up
    quilt pop -a -v -f
    #tar
    id=$(date +%Y%m%d_%H%M%S);
	name=vct-container,vct$VCT_HASH,controller$CONTROLLER_HASH,nodefw$NODEFIRMWARE_HASH.tar.xz
	echo "Compressing with tar xz..."
    tar -c --xz -f ./$name vct
    cd -
	mv $VCT_CONTAINER_DIR/$name ./dl/
	echo $name
}

function update_controller() {
	echo "Updating controller to $CONTROLLER_HASH"
	CONTROLLER_VERSION=$(ssh -i ./sshkey/id_rsa -o StrictHostKeyChecking=no vct@${VCT_IP} 'python /home/vct/confine-dist/utils/vct/server/manage.py controllerversion')
	rm -rf $VCT_CONTAINER_DIR/vct/rootfs/usr/local/lib/python2.7/dist-packages/controller
	git clone http://git.confine-project.eu/confine/controller.git $VCT_CONTAINER_DIR/vct/rootfs/home/vct/confine-controller
	cd $VCT_CONTAINER_DIR/vct/rootfs/home/vct/confine-controller
	git checkout $CONTROLLER_HASH
	CONTROLLER_HASH=$(git rev-parse --short $CONTROLLER_HASH)
	cd -
	echo "/home/vct/confine-controller" > $VCT_CONTAINER_DIR/vct/rootfs/usr/local/lib/python2.7/dist-packages/controller.pth
	# use the newest controller-admin
	cp $VCT_CONTAINER_DIR/vct/rootfs/home/vct/confine-controller/controller/bin/controller-admin.sh $VCT_CONTAINER_DIR/vct/rootfs/usr/local/bin/
	ssh -i ./sshkey/id_rsa -o StrictHostKeyChecking=no vct@${VCT_IP} 'cd /home/vct/confine-dist/utils/vct/server/ && sudo python ./manage.py postupgradecontroller --from=$CONTROLLER_VERSION' > $VCT_CONTAINER_DIR/vct_update_controller.log 2>&1
}


function install_node_firmware() {
	echo "Updating node firmware to $NODEFIRMWARE_HASH"
	for branch in testing master; do
		URL=http://builds.confine-project.eu/confine/openwrt/x86/$branch-builds/$NODEFIRMWARE_HASH_LONG/images/CONFINE-openwrt-$branch-latest.img.gz
		if wget -q --spider $URL; then
#			echo VCT_NODE_TEMPLATE_URL=\"$URL\" >> $VCT_CONTAINER_DIR/vct/rootfs/home/vct/confine-dist/utils/vct/vct.conf.overrides
			#this should be changed
			wget -q $URL -O $VCT_CONTAINER_DIR/vct/rootfs/var/lib/vct/downloads/CONFINE-owrt-master-atom-20130925-1614.img.gz
			break;
		fi
	done
}

function build_vct() {
	VCT_CONTAINER=vct,20131104.tar.xz
	VCT_CONTAINER_URL=https://media.confine-project.eu/vct-container/$VCT_CONTAINER
	VCT_CONTAINER_DIR="vct-container"
	
	VCT_HASH=$1
	CONTROLLER_HASH=$2
	NODEFIRMWARE_HASH=$3
	
	. ./host.sh
	configure_network
	configure_masquerade
	
	mkdir -p $VCT_CONTAINER_DIR

	tear_down_vct_container
	extract_vct
	start_vct
	sleep 10
	clean_vct
	update_vct
	#network_vct
	configure_inet
	install_vct
	update_controller
	install_node_firmware
	init_vct
	tear_down_vct_container
	tar_xz_vct
}

function build_vct_testing() {
	build_vct origin/testing origin/master origin/testing
}
LXC_NETWORK_LINK=${LXC_NETWORK_LINK:-"vmbridge"};
IPPREFIX=${IPPREFIX:-"fdf6:1e51:5f7b:b50c::"};
VCT_IP=${IPPREFIX}2
VCT_IP_PUBLIC=${VCT_IP_PUBLIC:-143.129.77.140}
OUT_IFACE=${OUT_IFACE:-"eth0"}