Prepare the YANG models for RESTCONF
[lispflowmapping.git] / mappingservice / southbound / src / main / java / org / opendaylight / lispflowmapping / southbound / lisp / LispSouthboundService.java
1 /*
2  * Copyright (c) 2014 Contextream, Inc. 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
9 package org.opendaylight.lispflowmapping.southbound.lisp;
10
11 import java.net.DatagramPacket;
12 import java.net.InetAddress;
13 import java.nio.ByteBuffer;
14
15 import org.opendaylight.controller.sal.binding.api.NotificationProviderService;
16 import org.opendaylight.lispflowmapping.implementation.serializer.LispMessage;
17 import org.opendaylight.lispflowmapping.implementation.serializer.LispMessageEnum;
18 import org.opendaylight.lispflowmapping.implementation.serializer.MapRegisterSerializer;
19 import org.opendaylight.lispflowmapping.implementation.serializer.MapRequestSerializer;
20 import org.opendaylight.lispflowmapping.implementation.util.ByteUtil;
21 import org.opendaylight.lispflowmapping.implementation.util.LispNotificationHelper;
22 import org.opendaylight.lispflowmapping.implementation.util.MapRequestUtil;
23 import org.opendaylight.lispflowmapping.southbound.lisp.exception.LispMalformedPacketException;
24 import org.opendaylight.lispflowmapping.southbound.lisp.network.PacketHeader;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.AddMappingBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.MapRegister;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.MapRequest;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.RequestMappingBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.transportaddress.TransportAddressBuilder;
30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.PortNumber;
31 import org.slf4j.Logger;
32 import org.slf4j.LoggerFactory;
33
34 public class LispSouthboundService implements ILispSouthboundService {
35     private NotificationProviderService notificationProvider;
36     protected static final Logger LOG = LoggerFactory.getLogger(LispSouthboundService.class);
37
38     public void setNotificationProvider(NotificationProviderService nps) {
39         this.notificationProvider = nps;
40     }
41
42     public void handlePacket(DatagramPacket packet) {
43         ByteBuffer inBuffer = ByteBuffer.wrap(packet.getData(), 0, packet.getLength());
44         int type = ByteUtil.getUnsignedByte(inBuffer, LispMessage.Pos.TYPE) >> 4;
45         Object lispType = LispMessageEnum.valueOf((byte) (type));
46         if (lispType == LispMessageEnum.EncapsulatedControlMessage) {
47             LOG.trace("Received packet of type EncapsulatedControlMessage");
48             handleEncapsulatedControlMessage(inBuffer, packet.getAddress());
49         } else if (lispType == LispMessageEnum.MapRequest) {
50             LOG.trace("Received packet of type MapRequest");
51             handleMapRequest(inBuffer, packet.getPort());
52         } else if (lispType == LispMessageEnum.MapRegister) {
53             LOG.trace("Received packet of type MapRegister");
54             handleMapRegister(inBuffer, packet.getAddress(), packet.getPort());
55         } else {
56             LOG.warn("Received unknown LISP control packet (type " + ((lispType != null) ? lispType : type) + ")");
57             LOG.trace("Buffer: " + ByteUtil.bytesToHex(packet.getData(), packet.getLength()));
58         }
59     }
60
61     private void handleEncapsulatedControlMessage(ByteBuffer inBuffer, InetAddress sourceAddress) {
62         try {
63             handleMapRequest(inBuffer, extractEncapsulatedSourcePort(inBuffer));
64         } catch (RuntimeException re) {
65             throw new LispMalformedPacketException("Couldn't deserialize Map-Request (len=" + inBuffer.capacity() + ")", re);
66         }
67     }
68
69     private void handleMapRequest(ByteBuffer inBuffer, int port) {
70         try {
71             MapRequest request = MapRequestSerializer.getInstance().deserialize(inBuffer);
72             InetAddress finalSourceAddress = MapRequestUtil.selectItrRloc(request);
73             if (finalSourceAddress == null) {
74                 throw new LispMalformedPacketException("Couldn't deserialize Map-Request, no ITR Rloc found!");
75             }
76
77             RequestMappingBuilder requestMappingBuilder = new RequestMappingBuilder();
78             requestMappingBuilder.setMapRequest(LispNotificationHelper.convertMapRequest(request));
79             TransportAddressBuilder transportAddressBuilder = new TransportAddressBuilder();
80             transportAddressBuilder.setIpAddress(LispNotificationHelper.getIpAddressFromInetAddress(finalSourceAddress));
81             transportAddressBuilder.setPort(new PortNumber(port));
82             requestMappingBuilder.setTransportAddress(transportAddressBuilder.build());
83             if (notificationProvider != null) {
84                 notificationProvider.publish(requestMappingBuilder.build());
85                 LOG.trace("MapRequest was published!");
86             } else {
87                 LOG.warn("Notification Provider is null!");
88             }
89         } catch (RuntimeException re) {
90             throw new LispMalformedPacketException("Couldn't deserialize Map-Request (len=" + inBuffer.capacity() + ")", re);
91         }
92     }
93
94     private int extractEncapsulatedSourcePort(ByteBuffer inBuffer) {
95         try {
96             inBuffer.position(PacketHeader.Length.LISP_ENCAPSULATION);
97             int ipType = (inBuffer.get() >> 4);
98             if (ipType == 4) {
99                 inBuffer.position(inBuffer.position() + PacketHeader.Length.IPV4 - 1);
100             } else if (ipType == 6) {
101                 inBuffer.position(inBuffer.position() + PacketHeader.Length.IPV6_NO_EXT - 1);
102             } else {
103                 throw new LispMalformedPacketException("Couldn't deserialize Map-Request: inner packet has unknown IP version: " + ipType);
104             }
105
106             int encapsulatedSourcePort = inBuffer.getShort() & 0xFFFF;
107             inBuffer.position(inBuffer.position() + PacketHeader.Length.UDP - 2);
108             return encapsulatedSourcePort;
109         } catch (RuntimeException re) {
110             throw new LispMalformedPacketException("Couldn't deserialize Map-Request (len=" + inBuffer.capacity() + ")", re);
111         }
112     }
113
114     private void handleMapRegister(ByteBuffer inBuffer, InetAddress sourceAddress, int port) {
115         try {
116             MapRegister mapRegister = MapRegisterSerializer.getInstance().deserialize(inBuffer);
117             AddMappingBuilder addMappingBuilder = new AddMappingBuilder();
118             addMappingBuilder.setMapRegister(LispNotificationHelper.convertMapRegister(mapRegister));
119             TransportAddressBuilder transportAddressBuilder = new TransportAddressBuilder();
120             transportAddressBuilder.setIpAddress(LispNotificationHelper.getIpAddressFromInetAddress(sourceAddress));
121             transportAddressBuilder.setPort(new PortNumber(port));
122             addMappingBuilder.setTransportAddress(transportAddressBuilder.build());
123             if (notificationProvider != null) {
124                 notificationProvider.publish(addMappingBuilder.build());
125                 LOG.trace("MapRegister was published!");
126             } else {
127                 LOG.warn("Notification Provider is null!");
128             }
129         } catch (RuntimeException re) {
130             throw new LispMalformedPacketException("Couldn't deserialize Map-Register (len=" + inBuffer.capacity() + ")", re);
131         }
132     }
133 }