#!/bin/bash

: <<=cut

=head1 NAME 

 emc_vnx_file_stats - Plugin to monitor Basic, NFSv3 and NFSv4 statistics of 
 EMC VNX 5300 Unified Storage system's Datamovers

=head1 AUTHOR

 Evgeny Beysembaev <megabotva@gmail.com>

=head1 LICENSE

 GPLv2

=head1 MAGIC MARKERS

  #%# family=auto
  #%# capabilities=autoconf suggest

=head1 DESCRIPTION

 The plugin monitors basic statistics of EMC Unified Storage system Datamovers 
 and NFS statistics of EMC VNX5300 Unified Storage system. Probably it can 
 also be compatible with other Isilon or Celerra systems. It uses SSH to connect
 to Control Stations, then remotely executes '/nas/sbin/server_stats' and 
 fetches and parses data from it. It supports gathering data both from 
 active/active and active/passive Datamover configurations, ignoring offline or 
 standby Datamovers. 
 If all Datamovers are offline or absent, the plugin returns error.
 This plugin also automatically chooses Primary Control Station from the list by
  calling '/nasmcd/sbin/getreason' and '/nasmcd/sbin/t2slot'.
 
 At the moment data is gathered from the following statistics sources:
  * nfs.v3.op - Tons of timings about NFSv3 RPC calls
  * nfs.v4.op - Tons of timings about NFSv4 RPC calls
  * nfs.client - Here new Client addresses are rescanned and added automatically.
  * basic-std Statistics Group - Basic Statistics of Datamovers (eg. CPU, Memory
  etc.)
 
 It's quite easy to comment out unneeded data to make graphs less overloaded or
 to add new statistics sources.

 The plugin has been tested in the following Operating Environment (OE):
  File Version T7.1.76.4
  Block Revision 05.32.000.5.215

=head1 LIST OF GRAPHS

 These are Basic Datamover Graphs.
 Graph category CPU:
	EMC VNX 5300 Datamover CPU Util %
 Graph category Network:
	EMC VNX 5300 Datamover Network bytes over all interfaces
	EMC VNX 5300 Datamover Storage bytes over all interfaces
 Graph category Memory:
	EMC VNX 5300 Datamover Memory
	EMC VNX 5300 File Buffer Cache
	EMC VNX 5300 FileResolve

 These are NFS (v3,v4) Graphs.
 Graph category NFS:
	EMC VNX 5300 NFSv3 Calls per second
	EMC VNX 5300 NFSv3 uSeconds per call
	EMC VNX 5300 NFSv3 Op %
	EMC VNX 5300 NFSv4 Calls per second
	EMC VNX 5300 NFSv4 uSeconds per call
	EMC VNX 5300 NFSv4 Op %
	EMC VNX 5300 NFS Client Ops/s
	EMC VNX 5300 NFS Client B/s
	EMC VNX 5300 NFS Client Avg uSec/call

=head1 COMPATIBILITY

 The plugin has been written for being compatible with EMC VNX5300 Storage 
 system, as this is the only EMC storage which i have.
 By the way, i am pretty sure it can also work with other VNX1 storages, like
 VNX5100 and VNX5500.
 About VNX2 series, i don't know whether the plugin will be able to work with 
 them. Maybe it would need some corrections in command-line backend. The same
 situation is with other EMC systems, so i encourage you to try and fix the 
 plugin. 

=head1 CONFIGURATION

 The plugin uses SSH to connect to Control Stations. It's possible to use 
 'nasadmin' user, but it would be better if you create read-only global user by
 Unisphere Client. The user should have only Operator role.
 I created "operator" user but due to the fact that Control Stations already
 had one internal "operator" user, the new one was called "operator1". So be 
 careful. After that, copy .bash_profile from /home/nasadmin to a newly created
 /home/operator1
 
 On munin-node side choose a user which will be used to connect through SSH.
 Generally user "munin" is ok. Then, execute "sudo su munin -s /bin/bash", 
 "ssh-keygen" and "ssh-copy-id" to both Control Stations with newly created 
 user.
 
 Make a link from /usr/share/munin/plugins/emc_vnx_file_stats to 
 /etc/munin/plugins/. If you want to get NFS statistics, name the link as 
 "emc_vnx_file_nfs_stats_<NAME>", otherwise to get Basic Datamover statistics
 you have to name it "emc_vnx_file_basicdm_stats_<NAME>", where <NAME> is any
 arbitrary name of your storage system. The plugin will return <NAME> in its 
 answer as "host_name" field.

 For example, assume your storage system is called "VNX5300".
 Make a configuration file at 
 /etc/munin/plugin-conf.d/emc_vnx_file_stats_VNX5300
 
 [emc_vnx_file_*]
 user munin				
 env.username operator1				
 env.cs_addr 192.168.1.1 192.168.1.2		
 env.nas_servers server_2 server_3		

 Where: 
 user - SSH Client local user
 env.username - Remote user with Operator role
 env.cs_addr - Control Stations addresses
 env.nas_servers - This is the default value and can be omitted

=head1 HISTORY

 08.11.2016 - First Release
 17.11.2016 - NFSv4 support, Memory section
 16.12.2016 - Merged "NFS" and "Datamover Stats" plugins
 26.12.2016 - Compatibility with Munin coding style

=cut

export LANG=C

. "$MUNIN_LIBDIR/plugins/plugin.sh"

ssh_check_cmd() {
	ssh -q $username@$1 "/nasmcd/sbin/getreason | grep -w slot_\`/nasmcd/sbin/t2slot\` | cut -d- -f1"
}

check_conf () {
	if [ -z "$username" ]; then
		echo "No username ('username' environment variable)!"
		return 1
	fi

	if [ -z "$cs_addr" ]; then
		echo "No control station addresses ('cs_addr' environment variable)!"
		return 1
	fi

	#Choosing Cotrol Station. Code have to be "10"
	for CS in $cs_addr; do
		if [[ "10" -eq "$(ssh_check_cmd $CS)" ]]; then
			PRIMARY_CS=$CS
			break
		fi
	done

	if [ -z "$PRIMARY_CS" ]; then
		echo "No alive primary Control Station from list \"$cs_addr\"";
		return 1
	fi
	return 0
}

if [ "$1" = "autoconf" ]; then
	check_conf_ans=$(check_conf)
	if [ $? -eq 0 ]; then
		echo "yes"
	else
		echo "no ($check_conf_ans)"
	fi
	exit 0
fi

if [ "$1" = "suggest" ]; then
	echo "nfs_stats"
	echo "basicdm_stats"
	exit 0;
fi

STATSTYPE=$(echo "${0##*/}" | cut -d _ -f 1-5)
if [ "$STATSTYPE" = "emc_vnx_file_nfs_stats" ]; then STATSTYPE=NFS; 
elif [ "$STATSTYPE" = "emc_vnx_file_basicdm_stats" ]; then STATSTYPE=BASICDM;
else echo "Do not know what to do. Name the plugin as 'emc_vnx_file_nfs_stats_<HOSTNAME>' or 'emc_vnx_file_basicdm_stats_<HOSTNAME>'"; exit 1; fi
# echo $STATSTYPE

TARGET=$(echo "${0##*/}" | cut -d _ -f 6)
: ${nas_servers:="server_2 server_3"}

check_conf 1>&2 || exit 1

run_remote () {
	ssh -q "$username@$PRIMARY_CS" ". /home/""$username""/.bash_profile; $*"
}

echo "host_name ${TARGET}"

if [ "$1" = "config" ] ; then
	for server in $nas_servers; do
		run_remote nas_server -i $server | grep -q 'type *= nas'
		if [ "$?" != 0 ]  ; then continue; fi
		nas_server_ok=TRUE
		
		if [ $STATSTYPE = "BASICDM" ] ; then
			cat <<-EOF 
			multigraph emc_vnx_cpu_percent
			graph_title EMC VNX 5300 Datamover CPU Util %
			graph_vlabel %
			graph_category cpu
			graph_scale no
			graph_args --upper-limit 100 -l 0
			${server}_cpuutil.min 0
			${server}_cpuutil.label $server CPU util. in %.

			multigraph emc_vnx_network_b
			graph_title EMC VNX 5300 Datamover Network bytes over all interfaces
			graph_vlabel B/s recv. (-) / sent (+)
			graph_category network
			graph_args --base 1000
			${server}_net_in.graph no
			${server}_net_in.label none
			${server}_net_out.label $server B/s
			${server}_net_out.negative ${server}_net_in
			${server}_net_out.draw AREA

			multigraph emc_vnx_storage_b
			graph_title EMC VNX 5300 Datamover Storage bytes over all interfaces
			graph_vlabel B/s recv. (-) / sent (+)
			graph_category network
			graph_args --base 1000
			${server}_stor_read.graph no
			${server}_stor_read.label none
			${server}_stor_write.label $server B/s
			${server}_stor_write.negative ${server}_stor_read
			${server}_stor_write.draw AREA

			multigraph emc_vnx_memory
			graph_title EMC VNX 5300 Datamover Memory
			graph_vlabel MiB
			graph_category memory
			graph_args --base 1024
			graph_order ${server}_used ${server}_free ${server}_total ${server}_freebuffer ${server}_encumbered
			${server}_used.label ${server} Used
			${server}_free.label ${server} Free
			${server}_free.draw STACK
			${server}_total.label ${server} Total
			${server}_freebuffer.label ${server} Free Buffer
			${server}_encumbered.label ${server} Encumbered
						
			multigraph emc_vnx_filecache
			graph_title EMC VNX 5300 File Buffer Cache
			graph_vlabel per second
			graph_category memory
			graph_args --base 1000
			graph_order ${server}_highw_hits ${server}_loww_hits ${server}_w_hits ${server}_hits ${server}_lookups
			${server}_highw_hits.label High Watermark Hits
			${server}_loww_hits.label Low Watermark Hits
			${server}_loww_hits.draw STACK
			${server}_w_hits.label Watermark Hits
			${server}_hits.label Hits
			${server}_lookups.label Lookups
				
			multigraph emc_vnx_fileresolve
			graph_title EMC VNX 5300 FileResolve
			graph_vlabel Entries
			graph_category memory
			graph_args --base 1000
			${server}_dropped.label Dropped Entries
			${server}_max.label Max Limit
			${server}_used.label Used Entries
			EOF
		fi
		if [ $STATSTYPE = "NFS" ] ; then
#nfs.v3.op data
			member_elements_by_line=$(run_remote server_stats "$server" -info nfs.v3.op | grep member_elements | sed -ne 's/^.*= //p')
			IFS=',' read -ra graphs <<< "$member_elements_by_line"
			cat <<-EOF 
			multigraph vnx_emc_v3_calls_s
			graph_title EMC VNX 5300 NFSv3 Calls per second
			graph_vlabel Calls
			graph_category nfs
			graph_args --base 1000
			EOF
			for graph in "${graphs[@]}"; do
				field=$(echo "$graph" | cut -d '.' -f4 )
				echo "${server}_$field.label $server $field"
			done

			cat <<-EOF
			
			multigraph vnx_emc_v3_usec_call
			graph_title EMC VNX 5300 NFSv3 uSeconds per call
			graph_vlabel uSec / call
			graph_category nfs
			graph_args --base 1000
			EOF
			for graph in "${graphs[@]}"; do
				field=$(echo "$graph" | cut -d '.' -f4 )
				echo "${server}_$field.label $server $field"
			done
			cat <<-EOF

			multigraph vnx_emc_v3_op_percent
			graph_title EMC VNX 5300 NFSv3 Op %
			graph_vlabel %
			graph_scale no
			graph_category nfs
			EOF
			for graph in "${graphs[@]}"; do
				field=$(echo "$graph" | cut -d '.' -f4 )
				echo "${server}_$field.label $server $field"
				echo "${server}_$field.min 0"
			done
			graphs=()
#nfs.v4.op data
			member_elements_by_line=$(run_remote server_stats "$server" -info nfs.v4.op | grep member_elements | sed -ne 's/^.*= //p')
			IFS=',' read -ra graphs <<< "$member_elements_by_line"
			cat <<-EOF
			multigraph vnx_emc_v4_calls_s
			graph_title EMC VNX 5300 NFSv4 Calls per second
			graph_vlabel Calls
			graph_category nfs
			graph_args --base 1000
			EOF
			for graph in "${graphs[@]}"; do
				field=$(echo "$graph" | cut -d '.' -f4 )
				echo "${server}_$field.label $server $field"
			done

			cat <<-EOF

			multigraph vnx_emc_v4_usec_call
			graph_title EMC VNX 5300 NFSv4 uSeconds per call
			graph_vlabel uSec / call
			graph_category nfs
			graph_args --base 1000
			EOF
			for graph in "${graphs[@]}"; do
				field=$(echo "$graph" | cut -d '.' -f4 )
				echo "${server}_$field.label $server $field"
			done
			cat <<-EOF
			
			multigraph vnx_emc_v4_op_percent
			graph_title EMC VNX 5300 NFSv4 Op %
			graph_vlabel %
			graph_scale no
			graph_category nfs
			EOF
			for graph in "${graphs[@]}"; do
				field=$(echo "$graph" | cut -d '.' -f4 )
				echo "${server}_$field.label $server $field"
				echo "${server}_$field.min 0"
			done

#nfs.client data
#							 Total    Read     Write   Suspicious   Total    Read     Write      Avg   
#                                                      Ops/s    Ops/s    Ops/s    Ops diff    KiB/s    KiB/s    KiB/s   uSec/call
			member_elements_by_line=$(run_remote server_stats server_2 -monitor nfs.client -count 1 -terminationsummary no -titles never | sed -ne 's/^.*id=//p' | cut -d' ' -f1)
			readarray graphs_array <<< "$member_elements_by_line"
			cat <<-EOF

			multigraph vnx_emc_nfs_client_ops_s
			graph_title EMC VNX 5300 NFS Client Ops/s
			graph_vlabel Ops/s
			graph_category nfs
			EOF
			echo -n "graph_order "
			for graph in "${graphs_array[@]}"; do
				field="$(clean_fieldname "$graph")"
				echo -n "${server}_${field}_r ${server}_${field}_w ${server}_${field}_t ${server}_${field}_s "
			done
			echo " "
			for graph in "${graphs_array[@]}"; do
				field="$(clean_fieldname "$graph")"
				echo "${server}_${field}_r.label $server $field Read Ops/s"
				echo "${server}_${field}_w.label $server $field Write Ops/s"
				echo "${server}_${field}_w.draw STACK"
				echo "${server}_${field}_t.label $server $field Total Ops/s"
				echo "${server}_${field}_s.label $server $field Suspicious Ops diff"
			done

			cat <<-EOF

			multigraph vnx_emc_nfs_client_b_s
			graph_title EMC VNX 5300 NFS Client B/s
			graph_vlabel B/s
			graph_category nfs
			EOF
			echo -n "graph_order "
			for graph in "${graphs_array[@]}"; do
				field="$(clean_fieldname "$graph")"
				echo -n "${server}_${field}_r ${server}_${field}_w ${server}_${field}_t "
			done
			echo " "
			for graph in "${graphs_array[@]}"; do
				field="$(clean_fieldname "$graph")"
				echo "${server}_${field}_r.label $server $field Read B/s"
				echo "${server}_${field}_w.label $server $field Write B/s"
				echo "${server}_${field}_w.draw STACK"
				echo "${server}_${field}_t.label $server $field Total B/s"
			done

			cat <<-EOF

			multigraph vnx_emc_nfs_client_avg_usec
			graph_title EMC VNX 5300 NFS Client Avg uSec/call
			graph_vlabel uSec/call
			graph_category nfs
			EOF
			for graph in "${graphs_array[@]}"; do
				field="$(clean_fieldname "$graph")"
				echo "${server}_${field}.label $server $field Avg uSec/call"
			done
		fi
	done
	if [ -z $nas_server_ok ]; then
		echo "No active data movers!" 1>&2
	fi
	exit 0
fi

for server in $nas_servers; do
	run_remote nas_server -i $server | grep -q 'type *= nas'
	if [ "$?" != 0 ]  ; then continue; fi
	nas_server_ok=TRUE
	if [ $STATSTYPE = "BASICDM" ] ; then
	member_elements_by_line=$(run_remote server_stats "$server" -count 1 -terminationsummary no -titles never | grep '^[^[:space:]]')
	IFS=$' ' read -ra graphs <<< "$member_elements_by_line"

	echo "multigraph emc_vnx_cpu_percent"
	echo "${server}_cpuutil.value ${graphs[1]}"

	echo -e "\nmultigraph emc_vnx_network_b"
	echo "${server}_net_in.value $((${graphs[2]} * 1000))"
	echo "${server}_net_out.value $((${graphs[3]} * 1000))"

	echo -e "\nmultigraph emc_vnx_storage_b"
	echo "${server}_stor_read.value $((${graphs[4]} * 1000))"
	echo "${server}_stor_write.value $((${graphs[5]} * 1000))"

	member_elements_by_line=$(run_remote server_stats "$server" -monitor kernel.memory -count 1 -terminationsummary no -titles never | grep '^[^[:space:]]')
	IFS=$' ' read -ra graphs <<< "$member_elements_by_line"

	echo -e "\nmultigraph emc_vnx_memory"
	echo "${server}_total.value $((${graphs[14]} / 1024))"
	echo "${server}_used.value $((${graphs[15]} / 1024))"
	echo "${server}_free.value $((${graphs[12]} / 1024))"
	echo "${server}_freebuffer.value $((${graphs[1]} / 1024))"
	echo "${server}_encumbered.value $((${graphs[8]} / 1024))"

	echo -e "\nmultigraph emc_vnx_filecache"
	echo "${server}_highw_hits.value ${graphs[2]}"
	echo "${server}_loww_hits.value ${graphs[6]}"
	echo "${server}_w_hits.value ${graphs[7]}"
	echo "${server}_hits.value ${graphs[4]}"
	echo "${server}_lookups.value ${graphs[5]}"

	echo -e "\nmultigraph emc_vnx_fileresolve"
	echo "${server}_dropped.value ${graphs[9]}"
	echo "${server}_max.value ${graphs[10]}"
	echo "${server}_used.value ${graphs[11]}"


	fi
	if [ $STATSTYPE = "NFS" ] ; then
#nfs.v3.op data
	member_elements_by_line=$(run_remote server_stats "$server" -monitor nfs.v3.op -count 1 -terminationsummary no -titles never | sed -ne 's/^.*v3/v3/p')
	NUMCOL=5
	LINES=$(wc -l <<< "$member_elements_by_line")
	while IFS=$'\n' read -ra graphs ; do
		elements_array+=( $graphs )
	done <<< "$member_elements_by_line"

	echo "multigraph vnx_emc_v3_calls_s"
	for ((i=0; i<$((LINES)); i++ )); do
		echo "${server}_${elements_array[i*$NUMCOL]}".value "${elements_array[i*$NUMCOL+1]}"
	done

	echo -e "\nmultigraph vnx_emc_v3_usec_call"
	for ((i=0; i<$((LINES)); i++ )); do
		echo "${server}_${elements_array[i*$NUMCOL]}".value "${elements_array[i*$NUMCOL+3]}"
	done

	echo -e "\nmultigraph vnx_emc_v3_op_percent"
	for ((i=0; i<$((LINES)); i++ )); do
		echo "${server}_${elements_array[i*$NUMCOL]}".value "${elements_array[i*$NUMCOL+4]}"
	done

	elements_array=()

#nfs.v4.op data
	member_elements_by_line=$(run_remote server_stats "$server" -monitor nfs.v4.op -count 1 -terminationsummary no -titles never | sed -ne 's/^.*v4/v4/p')
	NUMCOL=5
	LINES=$(wc -l <<< "$member_elements_by_line")
	while IFS=$'\n' read -ra graphs ; do
		elements_array+=( $graphs )
	done <<< "$member_elements_by_line"

	echo -e "\nmultigraph vnx_emc_v4_calls_s"
	for ((i=0; i<$((LINES)); i++ )); do
		echo "${server}_${elements_array[i*$NUMCOL]}".value "${elements_array[i*$NUMCOL+1]}"
	done

	echo -e "\nmultigraph vnx_emc_v4_usec_call"
	for ((i=0; i<$((LINES)); i++ )); do
		echo "${server}_${elements_array[i*$NUMCOL]}".value "${elements_array[i*$NUMCOL+3]}"
	done

	echo -e "\nmultigraph vnx_emc_v4_op_percent"
	for ((i=0; i<$((LINES)); i++ )); do
		echo "${server}_${elements_array[i*$NUMCOL]}".value "${elements_array[i*$NUMCOL+4]}"
	done

	elements_array=()

#nfs.client data
	echo -e "\nmultigraph vnx_emc_nfs_client_ops_s"
 	member_elements_by_line=$(run_remote server_stats server_2 -monitor nfs.client -count 1 -terminationsummary no -titles never | sed -ne 's/^.*id=//p')
	NUMCOL=9
	LINES=$(wc -l <<< "$member_elements_by_line")
	while IFS=$'\n' read -ra graphs; do
		elements_array+=($graphs)
	done  <<< "$member_elements_by_line"
	for (( i=0; i<$((LINES)); i++ )); do
		client="$(clean_fieldname "${element[i*$NUMCOL]}")"
		echo "${server}_${client}_r".value "${elements_array[$i*$NUMCOL+2]}"
		echo "${server}_${client}_w".value "${elements_array[$i*$NUMCOL+3]}"
		echo "${server}_${client}_t".value "${elements_array[$i*$NUMCOL+1]}"
		echo "${server}_${client}_s".value "${elements_array[$i*$NUMCOL+4]}"
        done
	echo -e "\nmultigraph vnx_emc_nfs_client_b_s"
	for (( i=0; i<$((LINES)); i++ )); do
		client="$(clean_fieldname "${element[i*$NUMCOL]}")"
		echo "${server}_${client}_r".value "$((${elements_array[$i*$NUMCOL+6]} * 1024))"
		echo "${server}_${client}_w".value "$((${elements_array[$i*$NUMCOL+7]} * 1024))"
		echo "${server}_${client}_t".value "$((${elements_array[$i*$NUMCOL+5]} * 1024))"
        done
	echo -e "\nmultigraph vnx_emc_nfs_client_avg_usec"
	for (( i=0; i<$((LINES)); i++ )); do
		client="$(clean_fieldname "${element[i*$NUMCOL]}")"
		echo "${server}_${client}".value "${elements_array[$i*$NUMCOL+8]}"
	done
	fi
done
if [ -z $nas_server_ok ]; then
	echo "No active data movers!" 1>&2
fi
exit 0