#! /bin/bash

usage()
{
  echo "$(basename $0) <options>"
  echo "-h, --help                print help message"
  echo "-v, --verbose             print more info"
}

case $1 in
  "-h" | "--help")
    usage
    exit 0
    ;;
esac

if (( $# > 1 )); then
  usage
  exit -1
fi

if (( $# == 1 )) && [ "$1" != "-v" ]; then
  usage
  exit -1
fi

ibdevs=$(ls /sys/class/infiniband/)

pushd /sys/class/net > /dev/null
devs=$(ls -d e* i* 2> /dev/null)
popd > /dev/null
if [ "x$devs" == "x" ]; then
  # no relevant devices - quit immediately
  exit
fi

for d in $devs; do
  if [ -f /sys/class/net/$d/dev_id ]; then
    oldstyle=n
    break
  fi
done

function print_line()
{
  echo "$1 port $2 <===> $3"
}

function find_guid()
{
  ibdevs=$(ls /sys/class/infiniband/)
  for ibdev in $ibdevs; do
    ports=$(ls /sys/class/infiniband/$ibdev/ports/)
    for port in $ports; do
      gids=$(ls /sys/class/infiniband/$ibdev/ports/$port/gids)
      for gid in $gids; do
        pguid=$(cat /sys/class/infiniband/$ibdev/ports/$port/gids/$gid | cut -b 21- | sed -e 's/://g')
        if [ x$pguid == x$1 ]; then
          print_line $ibdev $port $2
        fi
      done
    done
  done
}

function find_mac()
{
  ibdevs=$(ls /sys/class/infiniband/)
  for ibdev in $ibdevs; do
    type=$(cat /sys/class/infiniband/$ibdev/node_type|cut -d ' ' -f 2)
    ports=$(ls /sys/class/infiniband/$ibdev/ports/)
    for port in $ports; do
      gids=$(ls /sys/class/infiniband/$ibdev/ports/$port/gids)
      for gid in $gids; do
        if [[ "$type" = "RNIC" ]]; then
          pmac=$(cat /sys/class/infiniband/$ibdev/ports/$port/gids/$gid | cut -b 1-14|sed -e 's/://g')
        else
          first=$(cat /sys/class/infiniband/$ibdev/ports/$port/gids/$gid | cut -b 21-22)
          first=$(( first ^ 2 ))
          first=$(printf "%02x" $first)
          second=$(cat /sys/class/infiniband/$ibdev/ports/$port/gids/$gid | cut -b 21- | sed -e 's/://g' | cut -b 3-6)
          third=$(cat /sys/class/infiniband/$ibdev/ports/$port/gids/$gid | cut -b 21- | sed -e 's/://g' | cut -b 11-)
          pmac=$first$second$third
        fi
        if [ x$pmac == x$1 ]; then
          print_line $ibdev $port $2
        fi
      done
    done
  done
}

if [ "x$oldstyle" == "xn" ]; then
  for d in $ibdevs; do
    ibrsc=$(cat /sys/class/infiniband/$d/device/resource)
    eths=$(ls /sys/class/net/)
    for eth in $eths; do
      filepath_resource=/sys/class/net/$eth/device/resource

      if [ -f $filepath_resource ]; then
        ethrsc=$(cat $filepath_resource)
        if [ "x$ethrsc" == "x$ibrsc" ]; then
          filepath_devid=/sys/class/net/$eth/dev_id
          filepath_devport=/sys/class/net/$eth/dev_port
          if [ -f $filepath_devid ]; then
            port1=0
            if [ -f $filepath_devport ]; then
              port1=$(cat $filepath_devport)
              port1=$(printf "%d" $port1)
            fi

            port=$(cat $filepath_devid)
            port=$(printf "%d" $port)
            if [ $port1 -gt $port ]; then
              port=$port1
            fi

            port=$(( port + 1 ))

            filepath_carrier=/sys/class/net/$eth/carrier

            if [ -f $filepath_carrier ]; then
              link_state=$(cat $filepath_carrier 2> /dev/null)
              if (( link_state == 1 )); then
                link_state="Up"
              else
                link_state="Down"
              fi
            else
              link_state="NA"
            fi

            if [ "$1" == "-v" ]; then
              filepath_portstate=/sys/class/infiniband/$d/ports/$port/state
              filepath_deviceid=/sys/class/infiniband/$d/device/device
              filepath_fwver=/sys/class/infiniband/$d/fw_ver
              filepath_vpd=/sys/class/infiniband/$d/device/vpd

              # read port state
              if [ -f $filepath_portstate ]; then
                ibstate=$(printf "%-6s" "$(cat $filepath_portstate | gawk '{print $2}')")
              else
                ibstate="na"
              fi

              # read device
              if [ -f $filepath_deviceid ]; then
                devid=$(printf "mt%d" "$(cat $filepath_deviceid)")
              else
                devid="na"
              fi

              # read fw version
              if [ -f $filepath_fwver ]; then
                fwver=$(cat $filepath_fwver)
              else
                fwver="na"
              fi

              # read device description and part id from the vpd
              if [ -f $filepath_vpd ]; then
                tmp=$ifs
                ifs=":"
                vpd_content=`cat $filepath_vpd`
                devdesc=$(printf "%-15s" "$(echo $vpd_content | strings | head -1)")
                partid=$(printf "%-11s" "$(echo $vpd_content | strings | head -4 | tail -1 | gawk '{print $1}')")
                ifs=$tmp
              else
                devdesc=""
                partid="na"
              fi

              echo "$d ($devid - $partid) $devdesc fw $fwver port $port ($ibstate) ==> $eth ($link_state)"
            else
              echo "$d port $port ==> $eth ($link_state)"
            fi
          fi
        fi
      fi
    done
  done
else
##########################
### old style
##########################
  ifcs=$(ifconfig -a | egrep '^eth|^ib' | gawk '{print $1}')

  for ifc in $ifcs; do
    len=$(cat /sys/class/net/$ifc/addr_len)
    if (( len == 20 )); then
      guid=$(cat /sys/class/net/$ifc/address | cut -b 37- | sed -e 's/://g')
      find_guid $guid $ifc
    elif (( len == 6)); then
      mac=$(cat /sys/class/net/$ifc/address | sed -e 's/://g')
      find_mac $mac $ifc
    fi
  done
fi