From e7e7be65ecbd9284965d9f1ed39b23586449e8d7 Mon Sep 17 00:00:00 2001
From: Ross Philipson <philipsonr@ainfosec.com>
Date: Fri, 14 Oct 2016 15:47:57 -0400
Subject: [PATCH 1/2] [build] Add cleanup script.

OXT-507

Signed-off-by: Ross Philipson <philipsonr@ainfosec.com>
---
 build-scripts/clean.sh | 140 +++++++++++++++++++++++++++++++++++++++++
 build-scripts/setup.sh |   3 +
 2 files changed, 143 insertions(+)
 create mode 100755 build-scripts/clean.sh

diff --git a/build-scripts/clean.sh b/build-scripts/clean.sh
new file mode 100755
index 000000000..a3617fbaf
--- /dev/null
+++ b/build-scripts/clean.sh
@@ -0,0 +1,140 @@
+#!/bin/bash -e
+#
+# OpenXT build script.
+# Software license: see accompanying LICENSE file.
+#
+# Copyright (c) 2016 Assured Information Security, Inc.
+#
+# Contributions by Ross Philipson <philipsonr@ainfosec.com>
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+#
+# Invocation:
+# Used to cleanup a single build or all builds.
+#
+# N.B. this script starts outside the containers in the OUT OF CONTAINER
+# block. It then sets the variables in the IN CONTAINER part and sends
+# the modified script to each container to run the IN CONTAINER block.
+# Finally back out in the OUT OF CONTAINER block it cleans up the local
+# build dirs in xt-builds.
+
+BDLIST=
+
+#------------ IN CONTAINER ------------
+
+IN_CONTAINER=%IN_CONTAINER%
+C_BUILD_DIR=%C_BUILD_DIR%
+C_CLEAN_ALL=%C_CLEAN_ALL%
+
+if [ "$IN_CONTAINER" = "y" ]; then
+    if [ -n "$C_BUILD_DIR" ]; then
+        echo "Cleaning up: $C_BUILD_DIR"
+        rm -rf $C_BUILD_DIR
+        exit 0
+    fi
+
+    BDLIST=`ls | grep -e "^[0-9]\{6\}-[0-9]\+$"`
+    for i in ${BDLIST[@]}; do
+        echo "Cleaning up: $i"
+        rm -rf $i
+    done
+    exit 0
+fi
+
+#------------ OUT OF CONTAINER ------------
+
+BUILD_DIR=""
+CLEAN_ALL=""
+
+usage() {
+    cat >&2 <<EOF
+usage: $0 [-h] [-n build]
+  -h: Help
+  -n: Remove specified build
+  -A: Clean all builds
+EOF
+    exit $1
+}
+
+while getopts "hn:A" opt; do
+    case $opt in
+        h)
+            usage 0
+            ;;
+        n)
+            BUILD_DIR="${OPTARG}"
+            ;;
+        A)
+            CLEAN_ALL="y"
+            ;;
+        \?)
+            usage 1
+            ;;
+    esac
+done
+
+if [ -n "$BUILD_DIR" ] && [ -n "$CLEAN_ALL" ]; then
+    echo "Can't specify both -n and -A"
+    usage 2
+fi
+
+if [ -z "$BUILD_DIR" ] && [ -z "$CLEAN_ALL" ]; then
+    echo "Nothing to do??"
+    usage 2
+fi
+
+CONTAINER_USER=%CONTAINER_USER%
+SUBNET_PREFIX=%SUBNET_PREFIX%
+BUILD_USER="$(whoami)"
+BUILD_USER_ID="$(id -u ${BUILD_USER})"
+BUILD_USER_HOME="$(eval echo ~${BUILD_USER})"
+IP_C=$(( 150 + ${BUILD_USER_ID} % 100 ))
+
+clean_container() {
+    NUMBER=$1           # 01
+    NAME=$2             # oe
+
+    CONTAINER_IP="${SUBNET_PREFIX}.${IP_C}.1${NUMBER}"
+
+    if [ -d $NAME ]; then
+        echo "Cleaning build $NUMBER for container: $NAME"
+    else
+        echo "Cleaning all builds for container: $NAME"
+    fi
+
+    cat clean.sh | \
+        sed -e "s|\%IN_CONTAINER\%|y|" \
+            -e "s|\%C_BUILD_DIR%|${BUILD_DIR}|" \
+            -e "s|\%C_CLEAN_ALL\%|${CLEAN_ALL}|" |\
+        ssh -t -t -i "${BUILD_USER_HOME}"/ssh-key/openxt \
+            -oStrictHostKeyChecking=no ${CONTAINER_USER}@${CONTAINER_IP}
+}
+
+clean_container "01" "oe"
+clean_container "02" "debian"
+clean_container "03" "centos"
+
+# Cleanup local stuffs
+if [ -n "$BUILD_DIR" ]; then
+    echo "Cleaning up: $BUILD_DIR"
+    rm -rf xt-builds/${BUILD_DIR}
+    exit 0
+fi
+
+BDLIST=`ls xt-builds | grep -e "^[0-9]\{6\}-[0-9]\+$"`
+for i in ${BDLIST[@]}; do
+    echo "Cleaning up: xt-builds/$i"
+    rm -rf xt-builds/$i
+done
diff --git a/build-scripts/setup.sh b/build-scripts/setup.sh
index c49ca3bf3..7c0cbc759 100755
--- a/build-scripts/setup.sh
+++ b/build-scripts/setup.sh
@@ -279,11 +279,14 @@ fi
 
 # Copy the main build scripts to the home directory of the user
 cp -f build.sh "${BUILD_USER_HOME}/"
+cp -f clean.sh "${BUILD_USER_HOME}/"
 cp -f fetch.sh "${BUILD_USER_HOME}/"
 cp -f ../version "${BUILD_USER_HOME}/"
 cp -f *.layer "${BUILD_USER_HOME}/"
 sed -i "s|\%CONTAINER_USER\%|${CONTAINER_USER}|" ${BUILD_USER_HOME}/build.sh
 sed -i "s|\%SUBNET_PREFIX\%|${SUBNET_PREFIX}|" ${BUILD_USER_HOME}/build.sh
+sed -i "s|\%CONTAINER_USER\%|${CONTAINER_USER}|" ${BUILD_USER_HOME}/clean.sh
+sed -i "s|\%SUBNET_PREFIX\%|${SUBNET_PREFIX}|" ${BUILD_USER_HOME}/clean.sh
 sed -i "s|\%GIT_ROOT_PATH\%|${GIT_ROOT_PATH}|" ${BUILD_USER_HOME}/fetch.sh
 chown ${BUILD_USER}:${BUILD_USER} ${BUILD_USER_HOME}/build.sh
 chown ${BUILD_USER}:${BUILD_USER} ${BUILD_USER_HOME}/fetch.sh

From 39af3c825b600df19207cfe3c42974cbf25232b3 Mon Sep 17 00:00:00 2001
From: Ross Philipson <philipsonr@ainfosec.com>
Date: Fri, 14 Oct 2016 16:39:02 -0400
Subject: [PATCH 2/2] [build] clean.sh - various fixes from comments on PR

OXT-507

Signed-off-by: Ross Philipson <philipsonr@ainfosec.com>
---
 build-scripts/clean.sh | 13 +++++++++----
 build-scripts/setup.sh |  1 +
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/build-scripts/clean.sh b/build-scripts/clean.sh
index a3617fbaf..b25c2dbe5 100755
--- a/build-scripts/clean.sh
+++ b/build-scripts/clean.sh
@@ -45,11 +45,16 @@ if [ "$IN_CONTAINER" = "y" ]; then
         exit 0
     fi
 
+    if [ -z "$C_CLEAN_ALL" ]; then
+        exit 0
+    fi
+
     BDLIST=`ls | grep -e "^[0-9]\{6\}-[0-9]\+$"`
-    for i in ${BDLIST[@]}; do
+    for i in $BDLIST; do
         echo "Cleaning up: $i"
         rm -rf $i
     done
+
     exit 0
 fi
 
@@ -60,7 +65,7 @@ CLEAN_ALL=""
 
 usage() {
     cat >&2 <<EOF
-usage: $0 [-h] [-n build]
+usage: $0 [-h] [-n build] [-A]
   -h: Help
   -n: Remove specified build
   -A: Clean all builds
@@ -118,7 +123,7 @@ clean_container() {
         sed -e "s|\%IN_CONTAINER\%|y|" \
             -e "s|\%C_BUILD_DIR%|${BUILD_DIR}|" \
             -e "s|\%C_CLEAN_ALL\%|${CLEAN_ALL}|" |\
-        ssh -t -t -i "${BUILD_USER_HOME}"/ssh-key/openxt \
+        ssh -i "${BUILD_USER_HOME}"/ssh-key/openxt \
             -oStrictHostKeyChecking=no ${CONTAINER_USER}@${CONTAINER_IP}
 }
 
@@ -134,7 +139,7 @@ if [ -n "$BUILD_DIR" ]; then
 fi
 
 BDLIST=`ls xt-builds | grep -e "^[0-9]\{6\}-[0-9]\+$"`
-for i in ${BDLIST[@]}; do
+for i in $BDLIST; do
     echo "Cleaning up: xt-builds/$i"
     rm -rf xt-builds/$i
 done
diff --git a/build-scripts/setup.sh b/build-scripts/setup.sh
index 7c0cbc759..d54306d5d 100755
--- a/build-scripts/setup.sh
+++ b/build-scripts/setup.sh
@@ -289,6 +289,7 @@ sed -i "s|\%CONTAINER_USER\%|${CONTAINER_USER}|" ${BUILD_USER_HOME}/clean.sh
 sed -i "s|\%SUBNET_PREFIX\%|${SUBNET_PREFIX}|" ${BUILD_USER_HOME}/clean.sh
 sed -i "s|\%GIT_ROOT_PATH\%|${GIT_ROOT_PATH}|" ${BUILD_USER_HOME}/fetch.sh
 chown ${BUILD_USER}:${BUILD_USER} ${BUILD_USER_HOME}/build.sh
+chown ${BUILD_USER}:${BUILD_USER} ${BUILD_USER_HOME}/clean.sh
 chown ${BUILD_USER}:${BUILD_USER} ${BUILD_USER_HOME}/fetch.sh
 chown ${BUILD_USER}:${BUILD_USER} ${BUILD_USER_HOME}/version
 chown ${BUILD_USER}:${BUILD_USER} ${BUILD_USER_HOME}/*.layer