1 files changed, 190 insertions, 0 deletions
diff --git a/resources/libraries/bash/function/setup.sh b/resources/libraries/bash/function/setup.sh
new file mode 100644
index 0000000000..0b863aa17a
--- /dev/null
+++ b/resources/libraries/bash/function/setup.sh
@@ -0,0 +1,190 @@
+#!/usr/bin/env bash
+
+# Copyright (c) 2018 Cisco and/or its affiliates.
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at:
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+set -exuo pipefail
+
+
+function ansible_host () {
+    # Setup host via ansible playbook.
+    #
+    # Variable read:
+    # - ${HOST} - Server production IP address (Linux).
+
+    set -exuo pipefail
+
+    pushd "${TOOLS_DIR}"/testbed-setup/ansible || die "Pushd failed!"
+    ssh-copy-id -o StrictHostKeyChecking=no testuser@"${HOST}"
+    ansible-playbook --vault-id vault_pass --extra-vars '@vault.yml' \
+        --inventory production site.yaml --limit ${HOST} || {
+        die "Failed to ansible host!"
+    }
+    popd || die "Popd failed!"
+}
+
+
+function installed () {
+    # Check if the given utility is installed. Fail if not installed.
+    #
+    # Arguments:
+    # - ${1} - Utility to check.
+    # Returns:
+    # - 0 - If command is installed.
+    # - 1 - If command is not installed.
+
+    set -exuo pipefail
+
+    command -v "${1}"
+}
+
+
+function pxe_cimc () {
+    # Reboot server with next boot set to PXE, disables PXE after server is UP
+    # to prevent endless loop.
+    #
+    # Variable read:
+    # - ${HOST} - Server production IP address (Linux).
+    # - ${MGMT} - Server management IP address (IPMI/CIMC).
+    # - ${USER} - CIMC user.
+    # - ${PASS} - CIMC pass.
+
+    set -exuo pipefail
+
+    pushd "${TOOLS_DIR}"/testbed-setup/cimc || die "Pushd failed!"
+    ./cimc.py -u "${USER}" -p "${PASS}" "${MGMT}" --debug -pxe || {
+        die "Failed to send the PXE reboot command!"
+    }
+    for i in $(seq 1 500); do
+        warn "Waiting for server to become reachable ... " || die
+        if pingable "${HOST}"; then
+            ./cimc.py -u "${USER}" -p "${PASS}" "${MGMT}" --debug -hdd || {
+                die "Failed to send the HDD command!"
+            }
+            ssh-keygen -f "/home/testuser/.ssh/known_hosts" -R "${HOST}" || {
+                die "Failed to remove obsolete SSH key!"
+            }
+            warn "Server reachable, PXE running!" || die
+            break
+        fi
+    done
+    popd || die "Popd failed!"
+}
+
+
+function pxe_host () {
+    # Reboot host into PXE mode and detect once it is up.
+    #
+    # Variable read:
+    # - ${@} - All script arguments from command line.
+    # Variable set:
+    # - ${HOST} - Server production IP address (Linux).
+    # - ${MGMT} - Server management IP address (IPMI/CIMC).
+    # - ${USER} - User.
+    # - ${PASS} - Pass.
+
+    set -exuo pipefail
+
+    if ! installed ipmitool; then
+        die "ipmitool not present. Please install before continue!"
+    fi
+
+    HOST=${1}
+    MGMT=${2}
+    USER=${3}
+    PASS=${4}
+
+    ipmi="ipmitool -I lanplus -H ${MGMT} -U ${USER} -P ${PASS} chassis status"
+    cimc="./cimc.py -u ${USER} -p ${PASS} ${MGMT} --mac-table"
+    pushd "${TOOLS_DIR}"/testbed-setup/cimc || die "Pushd failed!"
+    if ${ipmi}; then
+        warn "This is IMPI managed server!"
+        pxe_supermicro || die
+    elif ${cimc}; then
+        warn "This is CIMC managed server!"
+        pxe_cimc || die
+    else
+        die "Unknown management console!"
+    fi
+    popd || die "Popd failed!"
+
+    wait_for_ssh || die
+}
+
+
+function pxe_supermicro () {
+    # Reboot server with next boot set to PXE.
+    #
+    # Variable read:
+    # - ${HOST} - Server production IP address (Linux).
+    # - ${MGMT} - Server management IP address (IPMI/CIMC).
+    # - ${USER} - IPMI user.
+    # - ${PASS} - IPMI pass.
+
+    set -exuo pipefail
+
+    if ! installed ipmitool; then
+        die "ipmitool not present. Please install before continue!"
+    fi
+
+    cmd="chassis bootdev pxe"
+    ipmitool -I lanplus -H "${MGMT}" -U "${USER}" -P "${PASS}" "${cmd}" || {
+        die "Failed to send the ${cmd} command!"
+    }
+    cmd="power reset"
+    ipmitool -I lanplus -H "${MGMT}" -U "${USER}" -P "${PASS}" "${cmd}" || {
+        die "Failed to send the ${cmd} command!"
+    }
+    for i in $(seq 1 500); do
+        warn "Waiting for server to become reachable ... " || die
+        if pingable "${HOST}"; then
+            ssh-keygen -f "/home/testuser/.ssh/known_hosts" -R "${HOST}" || {
+                die "Failed to remove obsolete SSH key!"
+            }
+            warn "Server reachable, PXE running!" || die
+            break
+        fi
+    done
+}
+
+
+function pingable () {
+    # Check if the host is reachable over ping. Fail if not reachable.
+    #
+    # Arguments:
+    # - ${1} - Host to check.
+    # Returns:
+    # - 0 - If host is reachable over ping.
+    # - 1 - If host is not reachable over ping.
+
+    set -exuo pipefail
+
+    ping -q -c 1 "${1}" 2>&1 > /dev/null
+}
+
+
+function wait_for_ssh () {
+    # Check if the host is reachable over SSH. Fail if not reachable.
+    #
+    # Variable read:
+    # - ${HOST} - Server production IP address (Linux).
+
+    set -exuo pipefail
+
+    while ! ssh "${HOST}"
+    do
+        warn "SSH not yet reachable, trying again!"
+        sleep 2
+    done
+    warn "SSH reachable!"
+}