#!/usr/bin/perl
package xCAT::SwitchHandler;

BEGIN
{
    $::XCATROOT = $ENV{'XCATROOT'} ? $ENV{'XCATROOT'} : '/opt/xcat';
}
use lib "$::XCATROOT/lib/perl";
use strict;
use xCAT::Table;
use xCAT::Utils;
use xCAT::MsgUtils;
use xCAT::MacMap;
use IO::Select;
use IO::Handle;
use Sys::Syslog;
use Data::Dumper;
use POSIX qw/WNOHANG/;
use SNMP;

my $sysDescr = '.1.3.6.1.2.1.1.1';

sub new {
    my $self  = {};
    my $proto = shift;
    my $class = ref($proto) || $proto;
    $self->{switch} = shift;

    bless($self, $class);
    return $self;
}


sub fill_sessionparms {
    my $self      = shift;
    my %args      = @_;
    my $community = $args{community};
    $self->{sessionparms} = $args{sessionents};
    if ($self->{sessionparms}->{snmpversion}) {
        if ($self->{sessionparms}->{snmpversion} =~ /3/) { #clean up to accept things like v3 or ver3 or 3, whatever.
            $self->{sessionparms}->{snmpversion} = 3;
            unless ($self->{sessionparms}->{auth}) {
                $self->{sessionparms}->{auth} = 'md5'; #Default to md5 auth if not specified but using v3
            }
        } elsif ($self->{sessionparms}->{snmpversion} =~ /2/) {
            $self->{sessionparms}->{snmpversion} = 2;
        } else {
            $self->{sessionparms}->{snmpversion} = 1; #Default to lowest common denominator, snmpv1
        }
    }
    unless (defined $self->{sessionparms}->{password}) { #if no password set, inherit the community
        $self->{sessionparms}->{password} = $community;
    }
}



sub setoid {
    my $session = shift;
    my $oid     = shift;
    my $offset  = shift;
    my $value   = shift;
    my $type    = shift;
    unless ($type) { $type = 'INTEGER'; }
    my $varbind = new SNMP::Varbind([ $oid, $offset, $value, $type ]);
    my $data = $session->set($varbind);
    if ($session->{ErrorStr}) { return (1, $session->{ErrorStr}); }
    return 0, $varbind;
}

#---------------------------------------------------------

=head3  getsnmpsession
   It gets an snmp session appropriate for a switch using the switches table 
   for guidance on the hows.
   Arguments: vlan=> $vid if needed for community string indexing (optional)
=cut

#------------------------------------------------------------
sub getsnmpsession {
    my $self      = shift;
    my $community = shift;
    my $vlanid    = shift;
    my $switch    = $self->{switch};
    my $session;
    my $sessionparams;

    if (!$community) { $community = "private"; }

    $self->{sitetab} = xCAT::Table->new('site');
    my $tmp = $self->{sitetab}->getAttribs({ key => 'snmpc' }, 'value');
    if ($tmp and $tmp->{value}) { $community = $tmp->{value} }

    my $switchestab = xCAT::Table->new('switches', -create => 0);
    my $ent = $switchestab->getNodeAttribs($switch, [qw(switch snmpversion username password privacy auth)]);
    if ($ent) {
        $self->fill_sessionparms(community => $community, sessionents => $ent);
    }

    $sessionparams = $self->{sessionparms};

    my $snmpver = '1';
    if ($sessionparams) {
        $snmpver   = $sessionparams->{snmpversion};
        $community = $sessionparams->{password};
    }

    if ($snmpver ne '3') {
        if ($vlanid) { $community .= '@' . $vlanid; }
        $session = new SNMP::Session(
            DestHost   => $switch,
            Version    => $snmpver,
            Community  => $community,
            UseNumeric => 1
        );
    } else {    #we have snmp3
        my %args = (
            DestHost   => $switch,
            SecName    => $sessionparams->{username},
            AuthProto  => uc($sessionparams->{auth}),
            AuthPass   => $community,
            Version    => $snmpver,
            SecLevel   => 'authNoPriv',
            UseNumeric => 1
        );
        if ($vlanid) { $args{Context} = "vlan-" . $vlanid; }
        if ($sessionparams->{privacy}) {
            $args{SecLevel}  = 'authPriv';
            $args{PrivProto} = uc($sessionparams->{privacy});
            $args{PrivPass}  = $community;
            $args{Retries}   = 4;
            $args{Timeout}   = 1500000;
        }

        #print "args=" . Dumper(%args) . "\n";
        $session = new SNMP::Session(%args);
    }

    #print "switch=$switch\n";
    if (!$session) { return $session; }

    #get the the switch brand name
    my $tmp = xCAT::MacMap::walkoid($session, "$sysDescr", silentfail => 1);
    my $swbrand;
    if ($tmp->{0}) {

        #print "Desc=" . $tmp->{0} . "\n";
        my @switch_plugins = glob("$::XCATROOT/lib/perl/xCAT_plugin/vlan/*.pm");
        foreach my $fn (@switch_plugins) {
            $fn =~ /.*\/([^\/]*).pm$/;
            my $module = $1;

            #print "fn=$fn,modele=$module\n";
            if (!eval { require "$fn" }) {
                xCAT::MsgUtils->message("S", "Cannot load module $fn");
                next;
            } else {
                no strict 'refs';
                my $filter = ${ "xCAT_plugin::vlan::" . $module . "::" }{filter_string}->();
                my $descr = $tmp->{0};
                if ($descr =~ /$filter/) {
                    $self->{module} = $module;

                    #print "found it:$module\n";
                    last;
                }
            }
        }
    }

    if (!exists($self->{module})) {
        $self->{session} = 0;
        return 0;
    } else {
        $self->{session} = $session;
        return $session;
    }
}



#--------------------------------------------------------------

=head3 get_vlan_ids    

  It gets the existing vlan IDs for the switch.
  Returns:  an array containing all the vlan ids for the switch
=cut

#-------------------------------------------------------------
sub get_vlan_ids {
    my $self    = shift;
    my $session = $self->{session};
    if (!$self->{session}) {
        $session = $self->getsnmpsession();
    }
    unless ($session) { xCAT::MsgUtils->message("S", "Failed to communicate with " . $self->{switch} . " or find a plugin module for the switch."); return; }

    no strict 'refs';
    return ${ "xCAT_plugin::vlan::" . $self->{module} . "::" }{get_vlan_ids}->($session);
}


#--------------------------------------------------------------

=head3 get_vlanids_for_ports   
    It returns a hash pointer that contains the vlan id for each given port.
=cut

#-------------------------------------------------------------
sub get_vlanids_for_ports {
    my $self  = shift;
    my @ports = @_;

    my $session = $self->{session};
    if (!$self->{session}) {
        $session = $self->getsnmpsession();
    }
    unless ($session) { xCAT::MsgUtils->message("S", "Failed to communicate with " . $self->{switch} . " or find a plugin module for the switch."); return; }

    no strict 'refs';
    return ${ "xCAT_plugin::vlan::" . $self->{module} . "::" }{get_vlanids_for_ports}->($session, @ports);

}


#--------------------------------------------------------------

=head3 create_vlan   
    Creates a new vlan on the switch
    Returns an array. (erorcode, errormsg). When errorcode=0, means no error.
=cut

#-------------------------------------------------------------
sub create_vlan {
    my $self      = shift;
    my $vlan_id   = shift;
    my $vlan_name = "xcat_vlan_" . $vlan_id;

    #print "create vlan get called.\n";

    my $session = $self->{session};
    if (!$self->{session}) {
        $session = $self->getsnmpsession();
    }

    #my $session  = $self->getsnmpsession();
    unless ($session) { xCAT::MsgUtils->message("S", "Failed to communicate with " . $self->{switch} . " or find a plugin module for the switch."); return; }

    #print Dumper($self->{sessionparms});

    no strict 'refs';
    return ${ "xCAT_plugin::vlan::" . $self->{module} . "::" }{create_vlan}->($session, $vlan_id);
}

#--------------------------------------------------------------

=head3 add_ports_to_vlan   
    Adds the given ports to the existing vlan
    Returns an array. (erorcode, errormsg). When errorcode=0, means no error.
=cut

#-------------------------------------------------------------
sub add_ports_to_vlan {
    my $self     = shift;
    my $vlan_id  = shift;
    my $portmode = shift;
    my @ports    = @_;

    #print "vlan=$vlan_id, ports=@ports\n";

    my $session = $self->{session};
    if (!$self->{session}) {
        $session = $self->getsnmpsession();
    }

    #my $session  = $self->getsnmpsession();
    unless ($session) { xCAT::MsgUtils->message("S", "Failed to communicate with " . $self->{switch} . " or find a plugin module for the switch."); return; }

    no strict 'refs';
    return ${ "xCAT_plugin::vlan::" . $self->{module} . "::" }{add_ports_to_vlan}->($session, $vlan_id, $portmode, @ports);
}

#-------------------------------------------------------

=head3  add_crossover_ports_to_vlan
  It enables the vlan on the cross-over links.
    Returns an array. (erorcode, errormsg). When errorcode=0, means no error.
=cut

#-------------------------------------------------------
sub add_crossover_ports_to_vlan {
    my $self     = shift;
    my $vlan_id  = shift;
    my @switches = @_;

    if (@switches == 0) { return (0, ""); }

    my $session = $self->{session};
    if (!$self->{session}) {
        $session = $self->getsnmpsession();
    }

    #my $session  = $self->getsnmpsession();
    unless ($session) { xCAT::MsgUtils->message("S", "Failed to communicate with " . $self->{switch} . " or find a plugin module for the switch."); return; }

    no strict 'refs';
    return ${ "xCAT_plugin::vlan::" . $self->{module} . "::" }{add_crossover_ports_to_vlan}->($session, $vlan_id, $self->{switch}, @switches);
}

#--------------------------------------------------------------

=head3 remove_vlan   
    Remove a vlan from the switch
    Returns an array. (erorcode, errormsg). When errorcode=0, means no error.
=cut

#-------------------------------------------------------------
sub remove_vlan {
    my $self    = shift;
    my $vlan_id = shift;

    my $session = $self->{session};
    if (!$self->{session}) {
        $session = $self->getsnmpsession();
    }

    #my $session  = $self->getsnmpsession();
    unless ($session) { xCAT::MsgUtils->message("S", "Failed to communicate with " . $self->{switch} . " or find a plugin module for the switch."); return; }

    no strict 'refs';
    return ${ "xCAT_plugin::vlan::" . $self->{module} . "::" }{remove_vlan}->($session, $vlan_id);
}

#--------------------------------------------------------------

=head3 remove_ports_from_vlan  
    Remove ports from a vlan
    Returns an array. (erorcode, errormsg). When errorcode=0, means no error.
=cut

#-------------------------------------------------------------
sub remove_ports_from_vlan {
    my $self    = shift;
    my $vlan_id = shift;
    my @ports   = @_;

    my $session = $self->{session};
    if (!$self->{session}) {
        $session = $self->getsnmpsession();
    }

    #my $session  = $self->getsnmpsession();
    unless ($session) { xCAT::MsgUtils->message("S", "Failed to communicate with " . $self->{switch} . " or find a plugin module for the switch."); return; }

    no strict 'refs';
    return ${ "xCAT_plugin::vlan::" . $self->{module} . "::" }{remove_ports_from_vlan}->($session, $vlan_id, @ports);
}



1;