vpnservice/natservice/natservice-impl/src/main/java/org/opendaylight/netvirt/natservice/internal/NaptPacketInHandler.java

   1 /*
   2  * Copyright (c) 2016 Ericsson India Global Services Pvt Ltd. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.netvirt.natservice.internal;
   9
  10 import org.opendaylight.controller.liblldp.NetUtils;
  11 import org.opendaylight.genius.mdsalutil.MetaDataUtil;
  12 import org.opendaylight.genius.mdsalutil.NwConstants;
  13 import org.opendaylight.genius.mdsalutil.packet.Ethernet;
  14 import org.opendaylight.genius.mdsalutil.packet.IPv4;
  15 import org.opendaylight.genius.mdsalutil.packet.TCP;
  16 import org.opendaylight.genius.mdsalutil.packet.UDP;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.packet.service.rev130709.PacketProcessingListener;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.packet.service.rev130709.PacketReceived;
  19 import org.slf4j.Logger;
  20 import org.slf4j.LoggerFactory;
  21
  22 import java.math.BigInteger;
  23 import java.util.HashSet;
  24 import com.google.common.primitives.Ints;
  25
  26 public class NaptPacketInHandler implements PacketProcessingListener {
  27
  28     private static final Logger LOG = LoggerFactory.getLogger(NaptPacketInHandler.class);
  29     private final static HashSet<String> incomingPacketMap = new HashSet<>();
  30     private final EventDispatcher naptEventdispatcher;
  31
  32     public NaptPacketInHandler(EventDispatcher eventDispatcher) {
  33         this.naptEventdispatcher = eventDispatcher;
  34     }
  35
  36     @Override
  37     public void onPacketReceived(PacketReceived packetReceived) {
  38         String internalIPAddress = "";
  39         int portNumber = 0;
  40         long routerId = 0L;
  41         NAPTEntryEvent.Operation operation = NAPTEntryEvent.Operation.ADD;
  42         NAPTEntryEvent.Protocol protocol;
  43
  44         Short tableId = packetReceived.getTableId().getValue();
  45
  46         LOG.trace("packet: {}, tableId {}", packetReceived, tableId);
  47
  48         if (tableId == NwConstants.OUTBOUND_NAPT_TABLE) {
  49             LOG.debug("NAT Service : NAPTPacketInHandler Packet for Outbound NAPT Table");
  50             byte[] inPayload = packetReceived.getPayload();
  51             Ethernet ethPkt = new Ethernet();
  52             if (inPayload != null) {
  53                 try {
  54                     ethPkt.deserialize(inPayload, 0, inPayload.length * NetUtils.NumBitsInAByte);
  55                 } catch (Exception e) {
  56                     LOG.warn("Failed to decode Packet", e);
  57                     return;
  58                 }
  59                 if (ethPkt.getPayload() instanceof IPv4) {
  60                     IPv4 ipPkt = (IPv4) ethPkt.getPayload();
  61                     byte[] ipSrc = Ints.toByteArray(ipPkt.getSourceAddress());
  62
  63                     internalIPAddress = NatUtil.toStringIpAddress(ipSrc, LOG);
  64                     LOG.trace("Retrieved internalIPAddress {}", internalIPAddress);
  65                     if (ipPkt.getPayload() instanceof TCP) {
  66                         TCP tcpPkt = (TCP) ipPkt.getPayload();
  67                         portNumber = tcpPkt.getSourcePort();
  68                         if(portNumber < 0){
  69                             portNumber = 32767 + portNumber + 32767 + 2;
  70                             LOG.trace("Retrieved and extracted TCP portNumber {}", portNumber);
  71                         }
  72                         protocol = NAPTEntryEvent.Protocol.TCP;
  73                         LOG.trace("Retrieved TCP portNumber {}", portNumber);
  74                     } else if (ipPkt.getPayload() instanceof UDP) {
  75                         UDP udpPkt = (UDP) ipPkt.getPayload();
  76                         portNumber = udpPkt.getSourcePort();
  77                         if(portNumber < 0){
  78                             portNumber = 32767 + portNumber + 32767 + 2;
  79                             LOG.trace("Retrieved and extracted UDP portNumber {}", portNumber);
  80                         }
  81                         protocol = NAPTEntryEvent.Protocol.UDP;
  82                         LOG.trace("Retrieved UDP portNumber {}", portNumber);
  83                     } else {
  84                         LOG.error("Incoming Packet is neither TCP or UDP packet");
  85                         return;
  86                     }
  87                 } else {
  88                     LOG.error("Incoming Packet is not IPv4 packet");
  89                     return;
  90                 }
  91
  92                 if(internalIPAddress != null) {
  93                     String sourceIPPortKey = internalIPAddress + ":" + portNumber;
  94                     LOG.debug("NAT Service : sourceIPPortKey {} mapping maintained in the map", sourceIPPortKey);
  95                     if (!incomingPacketMap.contains(sourceIPPortKey)) {
  96                         incomingPacketMap.add(internalIPAddress + portNumber);
  97                         LOG.trace("NAT Service : Processing new Packet");
  98                         BigInteger metadata = packetReceived.getMatch().getMetadata().getMetadata();
  99                         routerId = MetaDataUtil.getNatRouterIdFromMetadata(metadata);
 100                         if( routerId <= 0) {
 101                             LOG.error("NAT Service : Router ID is invalid");
 102                             return;
 103                         }
 104                         //send to Event Queue
 105                         LOG.trace("NAT Service : Creating NaptEvent for routerId {} and sourceIp {} and Port {}", routerId,
 106                                 internalIPAddress, portNumber);
 107                         NAPTEntryEvent naptEntryEvent = new NAPTEntryEvent(internalIPAddress,portNumber,routerId,
 108                                 operation,protocol, packetReceived, false);
 109                         naptEventdispatcher.addNaptEvent(naptEntryEvent);
 110                         LOG.trace("NAT Service : PacketInHandler sent event to NaptEventHandler");
 111                     } else {
 112                         LOG.trace("NAT Service : Packet already processed");
 113                         NAPTEntryEvent naptEntryEvent = new NAPTEntryEvent(internalIPAddress,portNumber,routerId,
 114                                 operation,protocol, packetReceived, true);
 115                         LOG.trace("NAT Service : PacketInHandler sent event to NaptEventHandler");
 116                     }
 117                 }else {
 118                     LOG.error("Nullpointer exception in retrieving internalIPAddress");
 119                 }
 120             }
 121         }else {
 122             LOG.trace("Packet is not from the Outbound NAPT table");
 123         }
 124     }
 125
 126     public void removeIncomingPacketMap(String sourceIPPortKey) {
 127         incomingPacketMap.remove(sourceIPPortKey);
 128         LOG.debug("NAT Service : sourceIPPortKey {} mapping is removed from map", sourceIPPortKey);
 129     }
 130 }