Implement Multipleslave

This patch implement multiple slave support.
Instead of having a maximum of one master and one slave in one docker
image, it is now possible to have multiple slave accross several docker
host.

For helping this change, a new boards.yaml format is introduced (See
README.md for details)
Note that tokens.yaml is also squashed in boards.yaml

6 files changed, 169 insertions, 1 deletions

diff --git a/lava-slave/Dockerfile b/lava-slave/Dockerfile
index b5e6385..be2dbaa 100644
--- a/lava-slave/Dockerfile
+++ b/lava-slave/Dockerfile
@@ -41,7 +41,7 @@ COPY configs/lava-slave /etc/lava-dispatcher/lava-slave
 COPY configs/tftpd-hpa /etc/default/tftpd-hpa
 
 COPY scripts/cu-loop /usr/local/bin/
-COPY conmux/* /etc/conmux/
+COPY conmux/ /etc/conmux/
 
 # Caution to not use any port between the Linux dynamic port range: 32768-60999
 RUN sed -i 's,XNBD_PORT_RANGE_MIN.*,XNBD_PORT_RANGE_MIN=61950,' /usr/lib/python2.7/dist-packages/lava_dispatcher/utils/constants.py
@@ -57,6 +57,24 @@ RUN rm /etc/apt/sources.list.d/testing.list
 COPY scripts/stop.sh .
 COPY scripts/start.sh .
 
+# lava-cli dependencies
+RUN apt-get -y install python3-setuptools python3-dev python3-zmq
+RUN git clone https://git.linaro.org/lava/lavacli.git /root/lavacli && cd /root/lavacli && git checkout v0.6 && python3 setup.py install
+
+COPY phyhostname /root/
+COPY scripts/setup.sh .
+
+COPY scripts/setdispatcherip.py /usr/local/bin/
+RUN chmod 755 /usr/local/bin/setdispatcherip.py
+
+COPY scripts/retire.sh /usr/local/bin/
+RUN chmod 755 /usr/local/bin/retire.sh
+
+COPY devices/ /root/devices/
+
+COPY scripts/extra_actions /root/
+RUN if [ -x /root/extra_actions ] ; then /root/extra_actions ; fi
+
 EXPOSE 69/udp 80
 
 CMD /start.sh
diff --git a/lava-slave/scripts/extra_actions b/lava-slave/scripts/extra_actions
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/lava-slave/scripts/extra_actions
diff --git a/lava-slave/scripts/retire.sh b/lava-slave/scripts/retire.sh
new file mode 100755
index 0000000..def6c8e
--- /dev/null
+++ b/lava-slave/scripts/retire.sh
@@ -0,0 +1,43 @@
+#!/bin/bash
+
+LAVA_MASTER_URI=$1
+
+if [ -z "$LAVA_MASTER_URI" ];then
+	echo "retire.sh: remove an offline worker"
+	echo "Usage: $0 LAVA_MASTER_URI"
+	echo "ERROR: Missing LAVA_MASTER_URI"
+	exit 11
+fi
+
+LAVACLIOPTS="--uri $LAVA_MASTER_URI"
+
+retire_worker() {
+	worker=$1
+	lavacli $LAVACLIOPTS workers list |grep -q $worker
+	if [ $? -eq 0 ];then
+		echo "Removing $worker"
+		lavacli $LAVACLIOPTS workers update $worker || exit $?
+	else
+		echo "SKIP: worker $worker does not exists"
+		return 0
+	fi
+	lavacli $LAVACLIOPTS devices list -a | grep '^\*' | cut -d' ' -f2 |
+	while read devicename
+	do
+		lavacli $LAVACLIOPTS devices show $devicename |grep -q "^worker.*$worker$"
+		if [ $? -eq 0 ];then
+			echo "Retire $devicename"
+			lavacli $LAVACLIOPTS devices update --health RETIRED --worker $worker $devicename || exit $?
+		fi
+	done
+	return 0
+}
+
+if [ -z "$2" ];then
+	for ww in $(ls devices/)
+	do
+		retire_worker $ww
+	done
+else
+	retire_worker $2
+fi
diff --git a/lava-slave/scripts/setdispatcherip.py b/lava-slave/scripts/setdispatcherip.py
new file mode 100644
index 0000000..a058cba
--- /dev/null
+++ b/lava-slave/scripts/setdispatcherip.py
@@ -0,0 +1,11 @@
+#!/usr/bin/env python
+
+import xmlrpclib
+import sys
+
+if len(sys.argv) < 4:
+    print("ERROR: Usage: %s URI workername dispatcherIP" % sys.argv[0])
+    sys.exit(1)
+
+server = xmlrpclib.ServerProxy("%s" % sys.argv[1])
+server.scheduler.workers.set_config("%s" % sys.argv[2], "dispatcher_ip: %s" % sys.argv[3])
diff --git a/lava-slave/scripts/setup.sh b/lava-slave/scripts/setup.sh
new file mode 100755
index 0000000..bf91c7a
--- /dev/null
+++ b/lava-slave/scripts/setup.sh
@@ -0,0 +1,93 @@
+#!/bin/bash
+
+if [ ! -e "/root/devices/$(hostname)" ];then
+	echo "Static slave for $LAVA_MASTER"
+	exit 0
+fi
+
+if [ -z "$LAVA_MASTER_URI" ];then
+	echo "ERROR: Missing LAVA_MASTER_URI"
+	exit 11
+fi
+
+echo "Dynamic slave for $LAVA_MASTER ($LAVA_MASTER_URI)"
+cd /root/lavacli
+LAVACLIOPTS="--uri $LAVA_MASTER_URI"
+
+# do a sort of ping for letting master to be up
+TIMEOUT=30
+while [ $TIMEOUT -ge 1 ];
+do
+	lavacli $LAVACLIOPTS device-types list 2>/dev/null >/dev/null
+	if [ $? -eq 0 ];then
+		TIMEOUT=0
+	else
+		echo "Wait for master...."
+		sleep 2	
+	fi
+	TIMEOUT=$(($TIMEOUT-1))
+done
+
+# This directory is used for storing device-types already added
+mkdir -p /root/.lavadocker/
+if [ -e /root/device-types ];then
+	for i in $(ls /root/device-types/*jinja2)
+	do
+		devicetype=$(basename $i |sed 's,.jinja2,,')
+		echo "Adding custom $devicetype"
+		lavacli $LAVACLIOPTS device-types list || exit $?
+		touch /root/.lavadocker/devicetype-$devicetype
+	done
+fi
+
+for worker in $(ls /root/devices/)
+do
+	lavacli $LAVACLIOPTS workers list |grep -q $worker
+	if [ $? -eq 0 ];then
+		echo "Remains of $worker, cleaning it"
+		/usr/local/bin/retire.sh $LAVA_MASTER_URI $worker
+		#lavacli $LAVACLIOPTS workers update $worker || exit $?
+	else
+		echo "Adding worker $worker"
+		lavacli $LAVACLIOPTS workers add --description "LAVA dispatcher on $(cat /root/phyhostname)" $worker || exit $?
+	fi
+	if [ ! -z "$LAVA_DISPATCHER_IP" ];then
+		echo "Add dispatcher_ip $LAVA_DISPATCHER_IP to $worker"
+		/usr/local/bin/setdispatcherip.py $LAVA_MASTER_URI $worker $LAVA_DISPATCHER_IP || exit $?
+	fi
+	for device in $(ls /root/devices/$worker/)
+	do
+		devicename=$(echo $device | sed 's,.jinja2,,')
+		devicetype=$(grep -h extends /root/devices/$worker/$device| grep -o '[a-zA-Z0-9_-]*.jinja2' | sed 's,.jinja2,,')
+		if [ -e /root/.lavadocker/devicetype-$devicetype ];then
+			echo "Skip devicetype $devicetype"
+		else
+			echo "Add devicetype $devicetype"
+			lavacli $LAVACLIOPTS device-types list | grep -q "$devicetype[[:space:]]"
+			if [ $? -eq 0 ];then
+				echo "Skip devicetype $devicetype"
+			else
+				lavacli $LAVACLIOPTS device-types add $devicetype || exit $?
+			fi
+			touch /root/.lavadocker/devicetype-$devicetype
+		fi
+		echo "Add device $devicename on $worker"
+		lavacli $LAVACLIOPTS devices list -a | grep -q $devicename
+		if [ $? -eq 0 ];then
+			echo "$devicename already present"
+			#verify if present on another worker
+			#TODO
+			lavacli $LAVACLIOPTS devices show $devicename |grep ^worker |grep -q $worker
+			if [ $? -ne 0 ];then
+				echo "ERROR: $devicename already present on another worker"
+				exit 1
+			fi
+			lavacli $LAVACLIOPTS devices update --worker $worker --health UNKNOWN $devicename || exit $?
+			# always reset the device dict in case of update of it
+			lavacli $LAVACLIOPTS devices dict set $devicename /root/devices/$worker/$device || exit $?
+		else
+			lavacli $LAVACLIOPTS devices add --type $devicetype --worker $worker $devicename || exit $?
+			lavacli $LAVACLIOPTS devices dict set $devicename /root/devices/$worker/$device || exit $?
+		fi
+	done
+done
diff --git a/lava-slave/scripts/start.sh b/lava-slave/scripts/start.sh
index 92e76bb..35e58a4 100755
--- a/lava-slave/scripts/start.sh
+++ b/lava-slave/scripts/start.sh
@@ -1,4 +1,7 @@
 #!/bin/bash
+
+/setup.sh || exit $?
+
 # Set LAVA Master IP
 if [[ -n "$LAVA_MASTER" ]]; then
 	sed -i -e "s/{LAVA_MASTER}/$LAVA_MASTER/g" /etc/lava-dispatcher/lava-slave