enviarAFilas.py

# Copyright 2011-2012 James McCauley
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at:
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""
l2 learning switch modificado del codigo original de James McCauley
Este esqueleto permite realizar manejo de ancho de banda y calidad de servicio a traves de filas
Antes de iniciar la aplicacion del controlador, las filas deberian estar creadas

"""

from pox.core import core
import pox.openflow.libopenflow_01 as of
from pox.lib.util import dpid_to_str
from pox.lib.util import str_to_bool
import pox.lib.packet.ethernet as pkt
from pox.lib.addresses import IPAddr, EthAddr
import time
import datetime

log = core.getLogger()
ip_servidor = "10.0.0.4"

# We don't want to flood immediately when a switch connects.
# Can be overriden on commandline.
_flood_delay = 0

class LearningSwitch (object):
  """
  The learning switch "brain" associated with a single OpenFlow switch.

  When we see a packet, we'd like to output it on a port which will
  eventually lead to the destination.  To accomplish this, we build a
  table that maps addresses to ports.

  We populate the table by observing traffic.  When we see a packet
  from some source coming from some port, we know that source is out
  that port.

  When we want to forward traffic, we look up the desintation in our
  table.  If we don't know the port, we simply send the message out
  all ports except the one it came in on.  (In the presence of loops,
  this is bad!).

  In short, our algorithm looks like this:

  For each packet from the switch:
  1) Use source address and switch port to update address/port table
  2) Is transparent = False and either Ethertype is LLDP or the packet's
     destination address is a Bridge Filtered address?
     Yes:
        2a) Drop packet -- don't forward link-local traffic (LLDP, 802.1x)
            DONE
  3) Is destination multicast?
     Yes:
        3a) Flood the packet
            DONE
  4) Port for destination address in our address/port table?
     No:
        4a) Flood the packet
            DONE
  5) Is output port the same as input port?
     Yes:
        5a) Drop packet and similar ones for a while
  6) Install flow table entry in the switch so that this
     flow goes out the appopriate port
     6a) Send the packet out appropriate port
  """
  def __init__ (self, connection, transparent):
    # Switch we'll be adding L2 learning switch capabilities to
    self.connection = connection
    self.transparent = transparent

    # Our table
    self.macToPort = {}

    # We want to hear PacketIn messages, so we listen
    # to the connection
    connection.addListeners(self)

    # We just use this to know when to log a helpful message
    self.hold_down_expired = _flood_delay == 0

    #log.debug("Initializing LearningSwitch, transparent=%s",
    #          str(self.transparent))

  def enviar_a_fila(self, event, port):

    # Este metodo en particular, envia hacia una fila determinada por el ultimo octeto de la IP de origen, si la direccion es 10.0.0.1 enviara a la fila 1
    # note que tomamos el parametro "port" que nos indica el puerto de destino 

    packet = event.parsed
    ip_packet = packet.payload
    ip_origen = ip_packet.srcip

    hora1 = datetime.strftime("06:00:00", "%X").time()
    hora2 = datetime.strftime("18:00:00", "%X").time()
    hora_act = datetime.now().time()

    print hora_act
    if hora_act > hora1 and hora_act < hora2:
      print "Hora actual 1 " + hora_act
      id_fila = 1
    else:
      print "Hora actual 2 " + hora_act
      id_fila = 2

    #id_fila = str(ip_origen).split(".")[3]
    print "La fila seria: ", id_fila

    msg = of.ofp_flow_mod()
    msg.match = of.ofp_match.from_packet(packet, event.port)
    msg.idle_timeout = 10
    msg.hard_timeout = 30

    # La unica diferencia respecto a un "output" normal, es que agregamos el id de la fila
    msg.actions.append(of.ofp_action_enqueue(port = port, queue_id=int(id_fila)))
    msg.data = event.ofp # 6a
    self.connection.send(msg)

  def _handle_PacketIn (self, event):
    """
    Handle packet in messages from the switch to implement above algorithm.
    """

    packet = event.parsed

    def flood (message = None):
      """ Floods the packet """
      msg = of.ofp_packet_out()
      if time.time() - self.connection.connect_time >= _flood_delay:
        # Only flood if we've been connected for a little while...

        if self.hold_down_expired is False:
          # Oh yes it is!
          self.hold_down_expired = True
          log.info("%s: Flood hold-down expired -- flooding",
              dpid_to_str(event.dpid))

        if message is not None: log.debug(message)
        #log.debug("%i: flood %s -> %s", event.dpid,packet.src,packet.dst)
        # OFPP_FLOOD is optional; on some switches you may need to change
        # this to OFPP_ALL.
        msg.actions.append(of.ofp_action_output(port = of.OFPP_FLOOD))
      else:
        pass
        #log.info("Holding down flood for %s", dpid_to_str(event.dpid))
      msg.data = event.ofp
      msg.in_port = event.port
      self.connection.send(msg)

    def drop (duration = None):
      """
      Drops this packet and optionally installs a flow to continue
      dropping similar ones for a while
      """
      if duration is not None:
        if not isinstance(duration, tuple):
          duration = (duration,duration)
        msg = of.ofp_flow_mod()
        msg.match = of.ofp_match.from_packet(packet)
        msg.idle_timeout = duration[0]
        msg.hard_timeout = duration[1]
        msg.buffer_id = event.ofp.buffer_id
        self.connection.send(msg)
      elif event.ofp.buffer_id is not None:
        msg = of.ofp_packet_out()
        msg.buffer_id = event.ofp.buffer_id
        msg.in_port = event.port
        self.connection.send(msg)

    self.macToPort[packet.src] = event.port # 1

    if not self.transparent: # 2
      if packet.type == packet.LLDP_TYPE or packet.dst.isBridgeFiltered():
        drop() # 2a
        return

    # Este es el lugar apropiado para ingresar nuestra funcion personalizada, porque ya se realizaron los chequeos que le permiten a l2 enrutar paquetes
    # Note el siguiente if, es imporante porque en este evento llegan TODOS los paquetes que pasen por la red (incluyendo paquetes ARP que no son IP)
    # Por esta razon, solo aplicamos reglas a los paquetes IP 
    # Nota: Este if no aplicaria para el proyecto que tiene el proyecto "arp defense" por obvias razones
    if packet.dst.is_multicast:
      flood() # 3a
    else:
      if packet.dst not in self.macToPort: # 4
        flood("Port for %s unknown -- flooding" % (packet.dst,)) # 4a
      else:
        port = self.macToPort[packet.dst]
        if port == event.port: # 5
          # 5a
          log.warning("Same port for packet from %s -> %s on %s.%s.  Drop."
              % (packet.src, packet.dst, dpid_to_str(event.dpid), port))
          drop(10)
          return
        # 6

        if packet.type == pkt.IP_TYPE:

          # El evento packet_in recibe un objeto llamado "evento", este evento cont$
          # 1. El puerto por el que ingresa el paquete, que es necesario para reali$
          # 2. El paquete que produjo el evento, lo que hace este codigo es parsear$
          #    si debe o no bloquear el flujo
          ip_packet = packet.payload
          # El paquete que recibimos es de tipo "Ethernet", asi que su carga util s$
          # El paquete ip_packet ya es un paquete IP, por lo tal posee un campo que$
          ip_destino = ip_packet.dstip
          print "IP Destino: ", ip_destino
          # Note la funcion IPAddr, se usa para manejar direcciones IP, en este cas$
          if (ip_destino == IPAddr(ip_servidor)):
                # Solo estamos enviando hacia las filas el trafico de salida hacia el servidor
                print "Enviar a fila"
                self.enviar_a_fila(event, port)
		return
        else:
          print "Not IP packet"

        log.debug("installing flow for %s.%i -> %s.%i" %
                  (packet.src, event.port, packet.dst, port))
        msg = of.ofp_flow_mod()
        msg.match = of.ofp_match.from_packet(packet, event.port)
        msg.idle_timeout = 10
        msg.hard_timeout = 30
        msg.actions.append(of.ofp_action_output(port = port))
        msg.data = event.ofp # 6a
        self.connection.send(msg)


class l2_learning (object):
  """
  Waits for OpenFlow switches to connect and makes them learning switches.
  """
  def __init__ (self, transparent):
    core.openflow.addListeners(self)
    self.transparent = transparent

  def _handle_ConnectionUp (self, event):
    log.debug("Connection %s" % (event.connection,))
    LearningSwitch(event.connection, self.transparent)


def launch (transparent=False, hold_down=_flood_delay):
  """
  Starts an L2 learning switch.
  """
  try:
    global _flood_delay
    _flood_delay = int(str(hold_down), 10)
    assert _flood_delay >= 0
  except:
    raise RuntimeError("Expected hold-down to be a number")

  core.registerNew(l2_learning, str_to_bool(transparent))