#
# $Id$
# $Revision$
#

module Msf
class Plugin::AutoAddRoute < Msf::Plugin
  include Msf::SessionEvent
  def name; 'auto_add_route'; end

  def desc
    "Adds routes for any new subnets whenever a session opens"
  end

  def on_session_open(session)
    return if not session.type == 'meterpreter'
    session.load_stdapi
    sb = Rex::Socket::SwitchBoard.instance
    session.net.config.each_route { |route|
      # Remove multicast and loopback interfaces
      next if route.subnet =~ /^(224\.|127\.)/
      next if route.subnet == '0.0.0.0'
      next if route.netmask == '255.255.255.255'
      if not sb.route_exists?(route.subnet, route.netmask)
        print_status("AutoAddRoute: Routing new subnet #{route.subnet}/#{route.netmask} through session #{session.sid}")
        sb.add_route(route.subnet, route.netmask, session)
      end
    }
  end

  def initialize(framework, opts)
    super
    self.framework.events.add_session_subscriber(self)
  end

  def cleanup
    self.framework.events.remove_session_subscriber(self)
  end

end
end