root/samrb/trunk/SimpleAuthManager.rb

#!/usr/bin/ruby
#
# Copyright (c) 2010-2012 WINLAB, Rutgers University, USA
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# 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.
#
#

require 'yaml'
require 'socket'      # Sockets are in standard library
require 'rubygems'
require 'open4'
require 'orderedhash'
require 'snmp'

$sClientStore = 'WimaxClient.yaml'
$sClick = '/usr/local/bin/click'
$sClickConf = 'Wimax.click'
$sNullMac = '00:00:00:00:00:00'

# Insert the list of client mac addresses, 
# and their corresponding IP addresses in ips varaible below.
$ips = {   
'00:aa:bb:cc:dd:ee' => '10.41.0.10',
}

# Insert the network specific entries below:
$def_gw = "10.41.0.1"    # Holds the IP of the default gateway.
$net_mask = "255.255.0.0" # Holds the netmask
$def_ip = "10.41.0.254"   # This is a default IP used for a client whose entry is not found in the client MAC listing described above.

$sHostname = 'localhost'
$dRcvPort = 54321
$dSndPort = 54322

class Client 
  attr_accessor :mac
  attr_accessor :ul
  attr_accessor :dl
  attr_accessor :ip
  attr_accessor :vlan

  def initialize(mac,ipa)
    @mac = mac
    @ip = ipa
    @ul = nil
    @dl = nil
    @vlan = nil
  end

end

#hClients.delete(mac)
#

def AddClient(mac)
  # Check if MAC address exists already

#  SNMP::Manager.open(:Host => "ofwireless.stanford.edu",:Version => ::SNMPv2c, :Community => 'public') do |snmp|
#    x=snmp.trap_v2(100,
#     
#    'localhost',
#    :enterpriseSpecific, #Generic Trap Type
#    [] 
#  )

  if ($hClients.has_key?(mac)) then
    # If it does just remove it's tunnels
    $hClients[mac].dl = nil
    $hClients[mac].ul = nil
  else 
    # if not create it
    if ($ips.has_key?(mac)) then
      $hClients[mac] = Client.new(mac,$ips[mac])
      puts "New client [#{mac}] = #{@ip}"
    else
      puts "Unknown mac= "+mac
      $hClients[mac] = Client.new(mac,$def_ip)
    end
#    # and then store it in persistent storage
#    open($sClientStore, 'w') { |f| YAML.dump($hClients, f) }
  end
end

def addGRE(mac,ch,gre)
  # Check if MAC address exists already
  AddClient(mac) unless ($hClients.has_key?(mac))
  if (ch == '1') then
    $hClients[mac].ul = gre
  else 
    $hClients[mac].dl = gre
  end
end

def delGRE(mac,ch,gre)
  # Check if MAC address exists already
  if ($hClients.has_key?(mac)) then
    if (ch == '1') then
      $hClients[mac].ul = nil
    else 
      $hClients[mac].dl = nil
    end
  end
end

def CreateClickConfiguration(file,clients)

  file << "
switch :: EtherSwitch;
eth0_queue :: Queue;
FromDevice(eth0, PROMISC 1) -> [0]switch;
switch[0] -> eth0_queue -> ToDevice(eth0);
"
  puts "Writting to a file"
  i = 1
  clients.each { |mac,client|
    next unless (client.ul != nil and client.dl != nil);
    file << "// ---  Client #{i} -------- //"
    file << "
AddressInfo(c_#{i} #{client.ip} #{client.mac});
arr_#{i} :: ARPResponder(c_#{i});
arq_#{i} :: ARPQuerier(c_#{i});
"
    file << "
Script(write arq_#{i}.gateway #{$def_gw}, write arq_#{i}.netmask #{$net_mask} )
"
    file << "
ulgre_#{i} :: FromDevice(#{client.ul});
dlgre_#{i} :: ToDevice(#{client.dl});
"
    file << "
switch[#{i}] -> cf_#{i} :: Classifier(12/0806 20/0001, 12/0806 20/0002, -);
cf_#{i}[0] -> arr_#{i} -> [#{i}]switch
cf_#{i}[1] -> [1]arq_#{i}
cf_#{i}[2] -> Strip(14) -> dlgreq_#{i} :: Queue -> dlgre_#{i};
// Switch output //
ulgre_#{i} -> GetIPAddress(16) -> arq_#{i} -> [#{i}]switch;
"
    i += 1
  }

end

def KillProcess(proc)
   pipe = IO.popen("pkill -9  #{proc}")
   pipe.readlines.each { |line|
   }
end

def RunClick(conf)
  if $oClickThread and $oClickThread.alive? then 
     puts "Killing click PID=#{$oClickThread.pid}"
     Process.kill(9, $oClickThread.pid)
     $oClickThread.kill
  end
  $oClickMonitor.kill if $oClickMonitor and $oClickMonitor.alive?

  stdin,stdout,stderr = '','',''
  $oClickThread = Open4::bg("#{$sClick} #{conf}",0=>stdin, 1=>stdout, 2=>stderr)
  puts "Started click PID=#{$oClickThread.pid}"
  $oClickMoitor = Thread.new {
    loop {
      puts stderr unless stderr.gets.nil?
      puts stdout unless stdout.gets.nil?
      sleep(1)
    }
  }
end

def RestartClick()
  if File.exist?($sClickConf+".bak") then File.delete($sClickConf+".bak") end
  if File.exist?($sClickConf) then File.rename($sClickConf,$sClickConf+".bak") end
  open($sClickConf, 'w') { |f| CreateClickConfiguration(f,$hClients); f.close }
  RunClick($sClickConf) 
end

def AsngwRcvr
  # Connect to the ASN daemon
  r = UDPSocket.open
  r.bind(0, $dRcvPort)

  loop { 
    line = r.recvfrom(100)[0]    # Read line from the ASN 
    args = line.split(' ')
    case args[0]
      when /^MS_REG/ then
        AddClient(args[1])                                      # Nee client
        UDPSocket.open.send("ALLOW", 0, $sHostname, $dSndPort)  # Accept it unconditionaly
        puts "Added client: "+args[1]
      when /^MS_DEL/ then
        delGRE(args[1],args[2],args[3])
        puts "Del GRE "+args[1]+" "+args[2]+" "+args[3]
      when /^MS_GRE/ then
        addGRE(args[1],args[2],args[3])
        puts "Add GRE "+args[1]+" "+args[2]+" "+args[3]
        if args[2] == "2" then RestartClick() end
      else puts "Unknown command: "+line
    end 
 }
end

## do we have anything stored?
#if File.exist?($sClientStore) then
#  # if we do load the state
#  $hClients = open($sClientStore) { |f| YAML.load(f) }
#else
#  # otherwise initialise the hash
#  $hClients = OrderedHash.new
#  # and persistant storage
#  open($sClientStore, 'w') { |f| YAML.dump($hClients, f) }
#end

KillProcess("click")

$hClients  = OrderedHash.new
$oClickThread = nil
$oClickMonitor = nil

$hGREs = {}
# Lets check if there are tunnels already up 
ifc = IO.popen("/sbin/ifconfig -a | grep gre")
ifc.each {  |gre| 
  $hGREs[gre.scan(/greAnc_\d+/)[0]] = 1 
}
# now let's find mobiles that are assigned to these
if !($hGREs.empty?)
  File.open("/etc/asnctrl_gre.conf").each { | line|
    mac,dir,tunnel,des = line.split(" ")
    next if !($hGREs.has_key?(mac))
    AddGre(mac,dir,tunnel)
  }
end
RestartClick()
AsngwRcvr()