+#!/usr/bin/env bash
+
+# SSH authentication script for cluster edition
+# This script will create a single key pair, which is then
+# propagated throughout the entire cluster.
+# There are two options for setup; temporary setup
+# persistent setup. If no options are specified, and the script
+# is only given ip addresses or host names, it will default to
+# the temporary setup. An ssh directory is then created in
+# /tmp/mn/ssh on each node, and mounted with the keys over the
+# user's ssh directory. This setup can easily be torn down by running
+# clustersetup with the -c option.
+# If the -p option is used, the setup will be persistent. In this
+# case, the key pair will be be distributed directly to each node's
+# ssh directory, but will be called cluster_key. An option to
+# specify this key for use will be added to the config file in each
+# user's ssh directory.
+
+
+set -e
+num_options=0
+persistent=false
+showHelp=false
+clean=false
+declare -a hosts=()
+user=$(whoami)
+SSHDIR=/tmp/mn/ssh
+USERDIR=$HOME/.ssh
+usage="./clustersetup.sh [ -p|h|c ] [ host1 ] [ host2 ] ...\n
+ Authenticate yourself and other cluster nodes to each other
+ via ssh for mininet cluster edition. By default, we use a
+ temporary ssh setup. An ssh directory is mounted over
+ $USERDIR on each machine in the cluster.
+
+ -h: display this help
+ -p: create a persistent ssh setup. This will add
+ new ssh keys and known_hosts to each nodes
+ $USERDIR directory
+ -c: method to clean up a temporary ssh setup.
+ Any hosts taken as arguments will be cleaned
+ "
+
+persistentSetup() {
+ echo "***creating key pair"
+ ssh-keygen -t rsa -C "Cluster_Edition_Key" -f $USERDIR/cluster_key -N '' # &> /dev/null
+ cat $USERDIR/cluster_key.pub >> $USERDIR/authorized_keys
+ echo "***configuring ssh"
+ echo "IdentityFile $USERDIR/cluster_key" >> $USERDIR/config
+ echo "IdentityFile $USERDIR/id_rsa" >> $USERDIR/config
+
+ for host in $hosts; do
+ echo "***copying public key to $host"
+ ssh-copy-id -i $USERDIR/cluster_key.pub $user@$host &> /dev/null
+ echo "***copying key pair to remote host"
+ scp $USERDIR/cluster_key $user@$host:$USERDIR
+ scp $USERDIR/cluster_key.pub $user@$host:$USERDIR
+ echo "***configuring remote host"
+ ssh -o ForwardAgent=yes $user@$host "
+ echo 'IdentityFile $USERDIR/cluster_key' >> $USERDIR/config
+ echo 'IdentityFile $USERDIR/id_rsa' >> $USERDIR/config"
+ done
+
+ for host in $hosts; do
+ echo "***copying known_hosts to $host"
+ scp $USERDIR/known_hosts $user@$host:$USERDIR/cluster_known_hosts
+ ssh $user@$host "
+ cat $USERDIR/cluster_known_hosts >> $USERDIR/known_hosts
+ rm $USERDIR/cluster_known_hosts"
+ done
+}
+
+tempSetup() {
+
+ echo "***creating temporary ssh directory"
+ mkdir -p $SSHDIR
+ echo "***creating key pair"
+ ssh-keygen -t rsa -C "Cluster_Edition_Key" -f $SSHDIR/id_rsa -N '' &> /dev/null
+
+ echo "***mounting temporary ssh directory"
+ sudo mount --bind $SSHDIR $USERDIR
+ cp $SSHDIR/id_rsa.pub $SSHDIR/authorized_keys
+
+ for host in $hosts; do
+ echo "***copying public key to $host"
+ ssh-copy-id $user@$host &> /dev/null
+ echo "***mounting remote temporary ssh directory for $host"
+ ssh -o ForwardAgent=yes $user@$host "
+ mkdir -p $SSHDIR
+ cp $USERDIR/authorized_keys $SSHDIR/authorized_keys
+ sudo mount --bind $SSHDIR $USERDIR"
+ echo "***copying key pair to $host"
+ scp $SSHDIR/{id_rsa,id_rsa.pub} $user@$host:$SSHDIR
+ done
+
+ for host in $hosts; do
+ echo "***copying known_hosts to $host"
+ scp $SSHDIR/known_hosts $user@$host:$SSHDIR
+ done
+}
+
+cleanup() {
+
+ for host in $hosts; do
+ echo "***cleaning up $host"
+ ssh $user@$host "sudo umount $USERDIR
+ sudo rm -rf $SSHDIR"
+ done
+
+ echo "**unmounting local directories"
+ sudo umount $USERDIR
+ echo "***removing temporary ssh directory"
+ sudo rm -rf $SSHDIR
+ echo "done!"
+
+}
+
+
+if [ $# -eq 0 ]; then
+ echo "ERROR: No Arguments"
+ echo "$usage"
+ exit
+else
+ while getopts 'hpc' OPTION
+ do
+ ((num_options+=1))
+ case $OPTION in
+ h) showHelp=true;;
+ p) persistent=true;;
+ c) clean=true;;
+ ?) showHelp=true;;
+ esac
+ done
+ shift $(($OPTIND - 1))
+fi
+
+if [ "$num_options" -gt 1 ]; then
+ echo "ERROR: Too Many Options"
+ echo "$usage"
+ exit
+fi
+
+if $showHelp; then
+ echo "$usage"
+ exit
+fi
+
+for i in "$@"; do
+ output=$(getent ahostsv4 "$i")
+ if [ -z "$output" ]; then
+ echo '***WARNING: could not find hostname "$i"'
+ echo ""
+ else
+ hosts+="$i "
+ fi
+done
+
+if $clean; then
+ cleanup
+ exit
+fi
+
+echo "***authenticating to:"
+for host in $hosts; do
+ echo "$host"
+done
+
+echo
+
+if $persistent; then
+ echo '***Setting up persistent SSH configuration between all nodes'
+ persistentSetup
+ echo $'\n*** Sucessfully set up ssh throughout the cluster!'
+
+else
+ echo '*** Setting up temporary SSH configuration between all nodes'
+ tempSetup
+ echo $'\n***Finished temporary setup. When you are done with your cluster'
+ echo $' session, tear down the SSH connections with'
+ echo $' ./clustersetup.sh -c '$hosts''
+fi
+
+echo