#!/usr/bin/perl
#
# Copyright (c) 2007-2008 Voltaire, Inc. All rights reserved.
# Copyright (c) 2006 The Regents of the University of California.
#
# This software is available to you under a choice of one of two
# licenses.  You may choose to be licensed under the terms of the GNU
# General Public License (GPL) Version 2, available from the file
# COPYING in the main directory of this source tree, or the
# OpenIB.org BSD license below:
#
#     Redistribution and use in source and binary forms, with or
#     without modification, are permitted provided that the following
#     conditions are met:
#
#      - Redistributions of source code must retain the above
#        copyright notice, this list of conditions and the following
#        disclaimer.
#
#      - Redistributions in binary form must reproduce the above
#        copyright notice, this list of conditions and the following
#        disclaimer in the documentation and/or other materials
#        provided with the distribution.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
# BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
# ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.
#

use strict;

use Getopt::Std;
use IBswcountlimits;

my $return_code = 0;

sub usage_and_exit
{
	my $prog = $_[0];
	print "Usage: $prog [-Rh]\n";
	print
"   Validate LIDs and GUIDs (check for zero and duplicates) in the local subnet\n";
	print "   -h This help message\n";
	print
"   -R Recalculate ibnetdiscover information (Default is to reuse ibnetdiscover output)\n";
	print "   -C <ca_name> use selected Channel Adaptor name for queries\n";
	print "   -P <ca_port> use selected channel adaptor port for queries\n";
	exit 2;
}

my $argv0          = `basename $0`;
my $regenerate_map = undef;
my $ca_name        = "";
my $ca_port        = "";

chomp $argv0;
if (!getopts("hRC:P:"))          { usage_and_exit $argv0; }
if (defined $Getopt::Std::opt_h) { usage_and_exit $argv0; }
if (defined $Getopt::Std::opt_R) { $regenerate_map = $Getopt::Std::opt_R; }
if (defined $Getopt::Std::opt_C) { $ca_name        = $Getopt::Std::opt_C; }
if (defined $Getopt::Std::opt_P) { $ca_port        = $Getopt::Std::opt_P; }

sub validate_non_zero_lid
{
	my ($lid)      = shift(@_);
	my ($nodeguid) = shift(@_);
	my ($nodetype) = shift(@_);

	if ($lid eq 0) {
		print "LID 0 found for $nodetype NodeGUID $nodeguid\n";
		return 1;
	}
	return 0;
}

sub validate_non_zero_guid
{
	my ($lid)      = shift(@_);
	my ($guid)     = shift(@_);
	my ($nodetype) = shift(@_);

	if ($guid eq 0x0) {
		print "$nodetype GUID 0x0 found with LID $lid\n";
		return 1;
	}
	return 0;
}

$insert_lid::lids           = undef;
$insert_nodeguid::nodeguids = undef;
$insert_portguid::portguids = undef;

sub insert_lid
{
	my ($lid)      = shift(@_);
	my ($nodeguid) = shift(@_);
	my ($nodetype) = shift(@_);
	my $rec        = undef;
	my $status     = "";

	$status = validate_non_zero_lid($lid, $nodeguid, $nodetype);
	if ($status eq 0) {
		if (defined($insert_lid::lids{$lid})) {
			print
"LID $lid already defined for NodeGUID $insert_lid::lids{$lid}->{nodeguid}\n";
			$return_code = 1;
		} else {
			$rec = {lid => $lid, nodeguid => $nodeguid};
			$insert_lid::lids{$lid} = $rec;
		}
	} else {
		$return_code = $status;
	}
}

sub insert_nodeguid
{
	my ($lid)      = shift(@_);
	my ($nodeguid) = shift(@_);
	my ($nodetype) = shift(@_);
	my $rec        = undef;
	my $status     = "";

	$status = validate_non_zero_guid($lid, $nodeguid, $nodetype);
	if ($status eq 0) {
		if (defined($insert_nodeguid::nodeguids{$nodeguid})) {
			print
"NodeGUID $nodeguid already defined for LID $insert_nodeguid::nodeguids{$nodeguid}->{lid}\n";
			$return_code = 1;
		} else {
			$rec = {lid => $lid, nodeguid => $nodeguid};
			$insert_nodeguid::nodeguids{$nodeguid} = $rec;
		}
	} else {
		$return_code = $status;
	}
}

sub validate_portguid
{
	my ($portguid)  = shift(@_);
	my ($nodeguid)  = shift(@_);

	if (($nodeguid ne $portguid)
		&& defined($insert_nodeguid::nodeguids{$portguid})) {
		print "PortGUID $portguid is an invalid duplicate of a NodeGUID\n";
		$return_code = 1;
	}
}

sub insert_portguid
{
	my ($lid)       = shift(@_);
	my ($portguid)  = shift(@_);
	my ($nodetype)  = shift(@_);
	my ($nodeguid)  = shift(@_);
	my $rec         = undef;
	my $status      = "";

	$status = validate_non_zero_guid($lid, $portguid, $nodetype);
	if ($status eq 0) {
		if (defined($insert_portguid::portguids{$portguid})) {
			print
"PortGUID $portguid already defined for LID $insert_portguid::portguids{$portguid}->{lid}\n";
			$return_code = 1;
		} else {
			$rec = {lid => $lid, portguid => $portguid};
			$insert_portguid::portguids{$portguid} = $rec;
			validate_portguid($portguid, $nodeguid);
		}
	} else {
		$return_code = $status;
	}
}

sub main
{
	my $cache_file = get_cache_file($ca_name, $ca_port);

	if ($regenerate_map || !(-f "$cache_file")) {
		generate_ibnetdiscover_topology($ca_name, $ca_port);
	}
	open IBNET_TOPO, "<$cache_file"
	  or die "Failed to open ibnet topology: $!\n";

	my $nodetype  = "";
	my $nodeguid  = "";
	my $portguid  = "";
	my $lid       = "";
	my $line      = "";
	my $firstport = "";

	while ($line = <IBNET_TOPO>) {

		if ($line =~ /^caguid=(.*)/ || $line =~ /^rtguid=(.*)/) {
			$nodeguid = $1;
			$nodetype = "";
		}

		if ($line =~ /^switchguid=(.*)/) {
			$nodeguid = $1;
			$portguid = "";
			$nodetype = "";
		}
		if ($line =~ /^switchguid=(.*)\((.*)\)/) {
			$nodeguid = $1;
			$portguid = "0x" . $2;
		}

		if ($line =~ /^Switch.*\"S-(.*)\"\s+# (.*) port.* lid (\d+) .*/) {
			$nodetype  = "switch";
			$firstport = "yes";
			$lid       = $3;
			insert_lid($lid, $nodeguid, $nodetype);
			insert_nodeguid($lid, $nodeguid, $nodetype);
			if ($portguid ne "") {
				insert_portguid($lid, $portguid, $nodetype, $nodeguid);
			}
		}
		if ($line =~ /^Ca.*/) {
			$nodetype  = "ca";
			$firstport = "yes";
		}
		if ($line =~ /^Rt.*/) {
			$nodetype  = "router";
			$firstport = "yes";
		}

		if ($nodetype eq "ca" || $nodetype eq "router") {
			if ($line =~ /"S-(.*)\# lid (\d+) .*/) {
				$lid = $2;
				insert_lid($lid, $nodeguid, $nodetype);
				if ($firstport eq "yes") {
					insert_nodeguid($lid, $nodeguid, $nodetype);
					$firstport = "no";
				}
			}
			if ($line =~ /^.*"H-(.*)\# lid (\d+) .*/) {
				$lid = $2;
				insert_lid($lid, $nodeguid, $nodetype);
				if ($firstport eq "yes") {
					insert_nodeguid($lid, $nodeguid, $nodetype);
					$firstport = "no";
				}
			}
			if ($line =~ /^.*"R-(.*)\# lid (\d+) .*/) {
				$lid = $2;
				insert_lid($lid, $nodeguid, $nodetype);
				if ($firstport eq "yes") {
					insert_nodeguid($lid, $nodeguid, $nodetype);
					$firstport = "no";
				}
			}
			if ($line =~ /^\[(\d+)\]\((.*)\)/) {
				$portguid = "0x" . $2;
				insert_portguid($lid, $portguid, $nodetype, $nodeguid);
			}
		}

	}

	close IBNET_TOPO;
}
main;

exit ($return_code);