#!/bin/bash

# Copyright 2017 Open Networking Foundation (ONF)
#
# Please refer questions to either the onos test mailing list at <onos-test@onosproject.org>,
# the System Testing Plans and Results wiki page at <https://wiki.onosproject.org/x/voMg>,
# or the System Testing Guide page at <https://wiki.onosproject.org/x/WYQg>
#
#     TestON 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.
#
#     TestON 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 TestON.  If not, see <http://www.gnu.org/licenses/>.

# ------------------------------------------------------------------------
# This script is a workaround for the key collision issue when loging into
# onos cli from both test-station and onos cells with identical user names.
# It copies both private and public keys of localhost (test-station) to
# all remote hosts specified in your cell.
# It also backs up the original keys of the remote hosts before
# overwriting them.
# Setting up passwordless login is recommended before running the script.
# ------------------------------------------------------------------------

ADDRESSES=()
# Check if OC1 to OC7 exist
for i in $(seq 1 7); do
    if [ -n "$OC$i" ]; then
        TEMP=OC$i
        ADDRESSES+=(${!TEMP})
    fi
done

# Check if OCN exists
[ -n "$OCN" ] && ADDRESSES+=($OCN)

# Copy keys to remote hosts
for address in "${ADDRESSES[@]}"; do
    echo "Backing up remote keys on" $address
    ssh sdn@$address "cd ~/.ssh; \
    cp id_rsa id_rsa.old; \
    cp id_rsa.pub id_rsa.pub.old; "

    echo "Copying keys to" $address
    scp ~/.ssh/id_rsa sdn@$address:~/.ssh/
    scp ~/.ssh/id_rsa.pub sdn@$address:~/.ssh/
done