#!/usr/bin/perl -w
#
# check_thecus.pl
# - A nagios plugin to monitor a THECUS NAS applicance, esp. N16000, esp. N16000 (2.01)
#
# Usage:
#  check_thecus.pl --user=<username> --password=<password> --host=<host>
# 
# sceleton from http://exchange.nagios.org/directory/Plugins/Hardware/Storage-Systems/SAN-and-NAS/Thecus-NAS-Health/details,
# adapted for N16000.
#
# It checks for:
#  - RAID healthiness
#  - chassis temperature
#  - CPU temperature
#  - CPU load
#  - Disk bad_block
#  - Disk recognition
#  
#  License: MIT
#  Patrick Kirsch <pkirsch@zscho.de>, 2012

use strict;
use Getopt::Long;
use LWP;
use LWP::UserAgent;
use HTTP::Cookies;
use JSON;
use Data::Dumper;

my $ua = new LWP::UserAgent;
$ua->agent('check_thecus Nagios plugin');
$ua->cookie_jar( { } );

my $user = 'admin';
my $pass = 'admin';
my $host = '127.0.0.1';

GetOptions("user=s" => \$user, "password=s" => \$pass, "host=s" => \$host);

if ($host eq '127.0.0.1') {
	$host = shift;
}

if($host eq '127.0.0.1') {
	die "Usage: $0 [--user <username>] [--password <password>] --host <host>\n";
}

if ($host =~ /\$/) {
	$host =~ s/\$//g;
}

my $res = $ua->post("http://$host/adm/login.php", [ 'eplang'=>'english', 'p_pass' => $pass, 'p_user' => $user, 'action' => 'login', 'option' => 'com',  'username' => $user, 'pwd' => $pass ]);

if($res->is_error()) {
	print "CRITICAL: Thecus login failed (" . $res->status_line . ")\n";
	exit 2;
}

my $aboutRes = $ua->get("http://$host/adm/getmain.php");


if(!$aboutRes->is_success()) {
	print "CRITICAL: Could not fetch status information (" . $aboutRes->status_line . ")\n";
	exit 2;
}

if(!$aboutRes =~ m/^\<html\>/s) {
	print "UNKNOWN: Could not log in to Thecus NAS";
	exit 3;
}


my $sysStatRes = $ua->get("http://$host/adm/getmain.php?fun=systatus&update=1&262");


if(!$sysStatRes->is_success()) {
	print "CRITICAL: Could not fetch status information (" . $sysStatRes->status_line . ")\n";
	exit 2;
}

my $sysStatObj = decode_json($sysStatRes->content());


# cpu_temp $sysStatObj->{'cup_temp1'}, cpu_load  $sysStatObj->{'cpu_loading'}, chassis temp  $sysStatObj->{'sys_temp'}[0]->{'temp'}

# checkVal 
# @param check_name, e.g. cpu_tmep
# @param current value, e.g. 50
# @param warning threshold, e.g. 60
# @param critical, threshold, e.g. 70;
#
# @return nothing
sub checkVal {
	(my @args) = (@_);

	if ($args[1] > $args[3]) {
		print "CRITICAL: $args[0] | current $args[1] > $args[3]\n";
		exit 2;
	}
	if ($args[1] > $args[2]) {
		print "WARNING: $args[0] | current $args[1] > $args[2]\n";
		exit 1;
	}

}

&checkVal('cpu_temp', $sysStatObj->{'cup_temp1'}, 50 , 70);
&checkVal('cpu_load', $sysStatObj->{'cpu_loading'}, 4 , 8);
&checkVal('chassis_temp', $sysStatObj->{'sys_temp'}[0]->{'temp'}, 55 , 70);

#print Dumper($sysStatObj);

# RAID
# #####
my $sysStatRaid = $ua->get("http://$host/adm/getmain.php?fun=raid&action=getraidlist");

if(!$sysStatRaid->is_success()) {
	print "CRITICAL: Could not fetch raid information (" . $sysStatRaid->status_line . ")\n";
	exit 2;
}

my $sysRaidObj = decode_json($sysStatRaid->content());

#print Dumper( $sysRaidObj );
for my $raid (@{$sysRaidObj->{'raid_list'}}) {
	if ($raid->{'raid_status'} ne 'Healthy') {
		print "CRITICAL: RAID id:$raid->{'raid_id'} is not marked as HEALTHY anymore\n";
		exit 2;
	}
}

# Disks
# #####
my $sysStatDisks = $ua->get("http://$host/adm/getmain.php?fun=disks&update=1");

if(!$sysStatDisks->is_success()) {
	print "CRITICAL: Could not fetch disk information (" . $sysStatDisks->status_line . ")\n";
	exit 2;
}

my $sysDiskObj = decode_json($sysStatDisks->content());

#print Dumper( $sysDiskObj );exit;
for my $tray (@{$sysDiskObj->{'disk_data'}}) {
	if ($tray->{'s_status'} ne 'N/A' && $tray->{'linkrate'} !~ /SATA\s+6Gb/) {
		print "WARNING: link rate runs with slower speed\n";
		exit 1;
	}
	if ($tray->{'s_status'} ne 'N/A' && $tray->{'badblock'} ne '0') {
		print "CRITICAL: tray:$tray->{'trayno'} disk $tray->{'model'} has bad blocks\n";
		exit 2;
	}
}

#
# else everything is fine
print "OK: Thecus NAS running normally|cpuLoad=$sysStatObj->{'cpu_loading'}, cpuTemp=$sysStatObj->{'cup_temp1'}, RAID: OK\n";
exit 0;