ci/util.sh

   1 #!/usr/bin/env bash
   2 # Utility script used to interact with a deployment
   3 # @author Tim Rozet (trozet@redhat.com)
   4
   5 VALID_CMDS="undercloud overcloud opendaylight debug-stack mock-detached -h --help"
   6 SSH_OPTIONS=(-o StrictHostKeyChecking=no -o GlobalKnownHostsFile=/dev/null -o UserKnownHostsFile=/dev/null -o LogLevel=error)
   7
   8 ##connects to undercloud
   9 ##params: user to login with, command to execute on undercloud (optional)
  10 function undercloud_connect {
  11   local user=$1
  12
  13   if [ -z "$1" ]; then
  14     echo "Missing required argument: user to login as to undercloud"
  15     return 1
  16   fi
  17
  18   if [ -z "$2" ]; then
  19     ssh ${SSH_OPTIONS[@]} ${user}@$(get_undercloud_ip)
  20   else
  21     ssh ${SSH_OPTIONS[@]} -T ${user}@$(get_undercloud_ip) "$2"
  22   fi
  23 }
  24
  25 ##outputs the Undercloud's IP address
  26 ##params: none
  27 function get_undercloud_ip {
  28   echo $(arp -an | grep $(virsh domiflist undercloud | grep default |\
  29     awk '{print $5}') | grep -Eo "[0-9]+\.[0-9]+\.[0-9]+\.[0-9]+")
  30 }
  31
  32 ##connects to overcloud nodes
  33 ##params: node to login to, command to execute on overcloud (optional)
  34 function overcloud_connect {
  35   local node
  36   local node_output
  37   local node_ip
  38
  39   if [ -z "$1" ]; then
  40     echo "Missing required argument: overcloud node to login to"
  41     return 1
  42   elif ! echo "$1" | grep -E "(controller|compute)[0-9]+" > /dev/null; then
  43     echo "Invalid argument: overcloud node to login to must be in the format: \
  44 controller<number> or compute<number>"
  45     return 1
  46   fi
  47
  48   node_output=$(undercloud_connect "stack" "source stackrc; nova list")
  49   node=$(echo "$1" | sed -E 's/([a-zA-Z]+)([0-9]+)/\1-\2/')
  50
  51   node_ip=$(echo "$node_output" | grep "$node" | grep -Eo "[0-9]+\.[0-9]+\.[0-9]+\.[0-9]+")
  52
  53   if [ "$node_ip" == "" ]; then
  54     echo -e "Unable to find IP for ${node} in \n${node_output}"
  55     return 1
  56   fi
  57
  58   if [ -z "$2" ]; then
  59     ssh ${SSH_OPTIONS[@]} heat-admin@${node_ip}
  60   else
  61     ssh ${SSH_OPTIONS[@]} -T heat-admin@${node_ip} "$2"
  62   fi
  63 }
  64
  65 ##connects to opendaylight karaf console
  66 ##params: None
  67 function opendaylight_connect {
  68   local opendaylight_ip
  69   opendaylight_ip=$(undercloud_connect "stack" "cat overcloudrc | grep SDN_CONTROLLER_IP | grep -Eo [0-9]+\.[0-9]+\.[0-9]+\.[0-9]+")
  70
  71   if [ "$opendaylight_ip" == "" ]; then
  72     echo -e "Unable to find IP for OpenDaylight in overcloudrc"
  73     return 1
  74   else
  75     echo -e "Connecting to ODL Karaf console.  Default password is 'karaf'"
  76   fi
  77
  78   ssh -p 8101 ${SSH_OPTIONS[@]} karaf@${opendaylight_ip}
  79 }
  80
  81 ##outputs heat stack deployment failures
  82 ##params: none
  83 function debug_stack {
  84   source ~/stackrc
  85   openstack stack failures list overcloud --long
  86 }
  87
  88 resolve_cmd() {
  89   local given=$1
  90   shift
  91   local list=($*)
  92   local inv=(${list[*]##${given}*})
  93   local OIFS=$IFS; IFS='|'; local pat="${inv[*]}"; IFS=$OIFS
  94   shopt -s extglob
  95   echo "${list[*]##+($pat)}"
  96   shopt -u extglob
  97 }
  98
  99 display_usage() {
 100   echo -e "Usage:\n$0 subcommand [ arguments ]\n"
 101   echo -e "Arguments:\n"
 102   echo -e "   undercloud [ user [ command ] ]   Connect to Undercloud VM as user and optionally execute a command"
 103   echo -e "                                     user    Optional: Defaults to 'stack'"
 104   echo -e "                                     command Optional: Defaults to none"
 105   echo -e ""
 106   echo -e "   opendaylight                      Connect to OpenDaylight Karaf console"
 107   echo -e ""
 108   echo -e "   overcloud  [ node [ command ] ]   Connect to an Overcloud node and optionally execute a command"
 109   echo -e "                                     node    Required: in format controller|compute<number>.  Example: controller0"
 110   echo -e "                                     command Optional: Defaults to none"
 111   echo -e ""
 112   echo -e "   debug-stack                       Print parsed deployment failures to stdout"
 113   echo -e ""
 114   echo -e "   mock-detached on | off            Add firewall rules to the jump host to mock a detached deployment\n"
 115 }
 116
 117 ##translates the command line argument
 118 ##params: $@ the entire command line is passed
 119 ##usage: parse_cmd_line() "$@"
 120 parse_cmdline() {
 121   local match
 122
 123   match=($(resolve_cmd $1 $VALID_CMDS))
 124   if [ ${#match[*]} -gt 1 ]; then
 125     echo "$1 is ambiguous, possible matches: ${match[*]}" >&2
 126     exit 1
 127   elif [ ${#match[*]} -lt 1 ]; then
 128     echo "$1 is not a recognized command.  Use -h to see acceptable list" >&2
 129     exit 1
 130   else
 131     match=$(echo $match | tr -d ' ')
 132   fi
 133
 134   case "$match" in
 135         -h|--help)
 136                 display_usage
 137                 exit 0
 138             ;;
 139         undercloud)
 140                 if [ -z "$2" ]; then
 141                   # connect as stack by default
 142                   undercloud_connect stack
 143                 elif [ -z "$3" ]; then
 144                   undercloud_connect "$2"
 145                 else
 146                   undercloud_connect "$2" "$3"
 147                 fi
 148                 exit 0
 149             ;;
 150         overcloud)
 151                 if [ -z "$2" ]; then
 152                   overcloud_connect
 153                 elif [ -z "$3" ]; then
 154                   overcloud_connect "$2"
 155                 else
 156                   overcloud_connect "$2" "$3"
 157                 fi
 158                 exit 0
 159             ;;
 160         opendaylight)
 161                 opendaylight_connect
 162                 exit 0
 163             ;;
 164         debug-stack)
 165                 undercloud_connect stack "$(typeset -f debug_stack); debug_stack"
 166                 exit 0
 167             ;;
 168         mock-detached)
 169                 if [ "$2" == "on" ]; then
 170                     echo "Ensuring we can talk to gerrit.opnfv.org"
 171                     iptables -A OUTPUT -p tcp -d gerrit.opnfv.org --dport 443 -j ACCEPT
 172                     echo "Blocking output http (80) traffic"
 173                     iptables -A OUTPUT -p tcp --dport 80 -j REJECT
 174                     iptables -A FORWARD -p tcp --dport 80 -j REJECT
 175                     echo "Blocking output https (443) traffic"
 176                     iptables -A OUTPUT -p tcp --dport 443 -j REJECT
 177                     iptables -A FORWARD -p tcp --dport 443 -j REJECT
 178                     echo "Blocking output dns (53) traffic"
 179                     iptables -A FORWARD -p tcp --dport 53 -j REJECT
 180                 elif [ "$2" == "off" ]; then
 181                     echo "Cleaning gerrit.opnfv.org specific rule"
 182                     iptables -D OUTPUT -p tcp -d gerrit.opnfv.org --dport 443 -j ACCEPT
 183                     echo "Allowing output http (80) traffic"
 184                     iptables -D OUTPUT -p tcp --dport 80 -j REJECT
 185                     iptables -D FORWARD -p tcp --dport 80 -j REJECT
 186                     echo "Allowing output https (443) traffic"
 187                     iptables -D OUTPUT -p tcp --dport 443 -j REJECT
 188                     iptables -D FORWARD -p tcp --dport 443 -j REJECT
 189                     echo "Allowing output dns (53) traffic"
 190                     iptables -D OUTPUT -p tcp --dport 53 -j REJECT
 191                     iptables -D FORWARD -p tcp --dport 53 -j REJECT
 192                 else
 193                     display_usage
 194                 fi
 195                 exit 0
 196             ;;
 197         *)
 198                 echo -e "\n\nThis script is used to interact with Apex deployments\n\n"
 199                 echo "Use -h to display help"
 200                 exit 1
 201             ;;
 202   esac
 203 }
 204
 205
 206 main() {
 207   parse_cmdline "$@"
 208 }
 209
 210 main "$@"