Cleanup mappingservice.yangmodel
[lispflowmapping.git] / mappingservice / southbound / src / main / java / org / opendaylight / lispflowmapping / southbound / LispSouthboundPlugin.java
index e26ffb4f53bf784f3a75dc893415089b82900a43..5e220b4ffb5e25d49547dc7f3c0bf5a59b86d8d2 100644 (file)
@@ -16,60 +16,73 @@ import java.net.InetSocketAddress;
 import java.net.SocketException;
 import java.net.SocketTimeoutException;
 import java.nio.ByteBuffer;
-import java.util.concurrent.Future;
 
-import org.eclipse.osgi.framework.console.CommandProvider;
-import org.opendaylight.controller.sal.binding.api.AbstractBindingAwareProvider;
+import org.apache.commons.lang3.exception.ExceptionUtils;
+import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.ProviderContext;
+import org.opendaylight.controller.sal.binding.api.BindingAwareProvider;
 import org.opendaylight.controller.sal.binding.api.NotificationProviderService;
-import org.opendaylight.lispflowmapping.implementation.serializer.LispMessage;
-import org.opendaylight.lispflowmapping.implementation.serializer.MapNotifySerializer;
-import org.opendaylight.lispflowmapping.implementation.serializer.MapReplySerializer;
-import org.opendaylight.lispflowmapping.implementation.serializer.MapRequestSerializer;
+import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
+import org.opendaylight.lispflowmapping.lisp.type.LispMessage;
 import org.opendaylight.lispflowmapping.southbound.lisp.ILispSouthboundService;
 import org.opendaylight.lispflowmapping.southbound.lisp.LispSouthboundService;
 import org.opendaylight.lispflowmapping.southbound.lisp.LispXtrSouthboundService;
 import org.opendaylight.lispflowmapping.type.sbplugin.IConfigLispSouthboundPlugin;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.LispflowmappingService;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.SendMapNotifyInput;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.SendMapReplyInput;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.SendMapRequestInput;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.transportaddress.TransportAddress;
-import org.opendaylight.yangtools.yang.common.RpcResult;
-import org.osgi.framework.BundleContext;
-import org.osgi.framework.FrameworkUtil;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.LispProtoService;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.transportaddress.TransportAddress;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 import com.google.common.net.InetAddresses;
 
-public class LispSouthboundPlugin extends AbstractBindingAwareProvider implements IConfigLispSouthboundPlugin, CommandProvider, LispflowmappingService {
-    protected static final Logger logger = LoggerFactory.getLogger(LispSouthboundPlugin.class);
+public class LispSouthboundPlugin implements IConfigLispSouthboundPlugin, AutoCloseable, BindingAwareProvider {
+    protected static final Logger LOG = LoggerFactory.getLogger(LispSouthboundPlugin.class);
 
     private static Object startLock = new Object();
     private LispIoThread lispThread;
     private LispIoThread xtrThread;
     private LispSouthboundService lispSouthboundService;
     private LispXtrSouthboundService lispXtrSouthboundService;
+    private NotificationProviderService notificationService;
+    private RpcProviderRegistry rpcRegistry;
+    private BindingAwareBroker broker;
     private volatile DatagramSocket socket = null;
-    private final String MAP_NOTIFY = "MapNotify";
-    private final String MAP_REPlY = "MapReply";
-    private final String MAP_REQUEST = "MapRequest";
     private volatile String bindingAddress = null;
-    private volatile boolean alreadyInit = false;
     private volatile int xtrPort = LispMessage.XTR_PORT_NUM;
     private volatile boolean listenOnXtrPort = false;
-
+    private BindingAwareBroker.RpcRegistration<LispProtoService> controlPlaneRpc;
     private DatagramSocket xtrSocket;
 
-    private void registerWithOSGIConsole() {
-        BundleContext bundleContext = FrameworkUtil.getBundle(this.getClass()).getBundleContext();
-        bundleContext.registerService(CommandProvider.class.getName(), this, null);
-        bundleContext.registerService(IConfigLispSouthboundPlugin.class.getName(), this, null);
+    public void init() {
+        LOG.info("LISP (RFC6830) Mapping Service is up!");
+        final LfmControlPlaneRpc lfmCpRpc = new LfmControlPlaneRpc(this);
+
+        controlPlaneRpc = rpcRegistry.addRpcImplementation(LispProtoService.class, lfmCpRpc);
+        broker.registerProvider(this);
+
+        synchronized (startLock) {
+            lispSouthboundService = new LispSouthboundService();
+            lispXtrSouthboundService = new LispXtrSouthboundService();
+            lispSouthboundService.setNotificationProvider(this.notificationService);
+            lispXtrSouthboundService.setNotificationProvider(this.notificationService);
+            LOG.trace("Provider Session initialized");
+            if (bindingAddress == null) {
+                setLispAddress("0.0.0.0");
+            }
+            LOG.info("LISP (RFC6830) Mapping Service is up!");
+        }
     }
 
-    protected void stopImpl(BundleContext context) {
-        unloadActions();
+    public void setNotificationProviderService(NotificationProviderService notificationService) {
+        this.notificationService = notificationService;
+    }
+
+    public void setRpcRegistryDependency(RpcProviderRegistry rpcRegistry) {
+        this.rpcRegistry = rpcRegistry;
+    }
+
+    public void setBindingAwareBroker(BindingAwareBroker broker) {
+        this.broker = broker;
     }
 
     private void unloadActions() {
@@ -80,17 +93,14 @@ public class LispSouthboundPlugin extends AbstractBindingAwareProvider implement
         lispXtrSouthboundService = null;
         lispThread = null;
         xtrThread = null;
-        logger.info("LISP (RFC6830) Mapping Service is down!");
+        bindingAddress = null;
+        LOG.info("LISP (RFC6830) Mapping Service is down!");
         try {
             Thread.sleep(1100);
         } catch (InterruptedException e) {
         }
     }
 
-    public void destroy() {
-        unloadActions();
-    }
-
     private class LispIoThread extends Thread {
         private volatile boolean shouldRun;
         private volatile DatagramSocket threadSocket = null;
@@ -110,13 +120,13 @@ public class LispSouthboundPlugin extends AbstractBindingAwareProvider implement
 
             int lispReceiveTimeout = 1000;
 
-            logger.info("LISP (RFC6830) Mapping Service is running and listening on address: " + bindingAddress + " port: "
-                    + threadSocket.getLocalPort());
+            LOG.info("LISP (RFC6830) Mapping Service is running and listening on address: " + bindingAddress
+                    + " port: " + threadSocket.getLocalPort());
             try {
 
                 threadSocket.setSoTimeout(lispReceiveTimeout);
             } catch (SocketException e) {
-                logger.error("Cannot open socket on UDP port " + threadSocket.getLocalPort(), e);
+                LOG.error("Cannot open socket on UDP port " + threadSocket.getLocalPort(), e);
                 return;
             }
 
@@ -125,24 +135,24 @@ public class LispSouthboundPlugin extends AbstractBindingAwareProvider implement
                 DatagramPacket packet = new DatagramPacket(buffer, buffer.length);
                 try {
                     threadSocket.receive(packet);
-                    logger.trace("Received a packet!");
+                    LOG.trace("Received a packet!");
                 } catch (SocketTimeoutException ste) {
                     continue;
                 } catch (IOException e) {
-                    logger.warn("IO Exception while trying to recieve packet", e);
+                    LOG.warn("IO Exception while trying to recieve packet", e);
                 }
-                logger.trace(String.format("Handling packet from {%s}:{%d} (len={%d})", packet.getAddress().getHostAddress(), packet.getPort(),
-                        packet.getLength()));
+                LOG.trace(String.format("Handling packet from {%s}:{%d} (len={%d})", packet.getAddress()
+                        .getHostAddress(), packet.getPort(), packet.getLength()));
 
                 try {
                     this.service.handlePacket(packet);
                 } catch (Exception e) {
-                    logger.warn("Error while handling packet", e);
+                    LOG.warn("Error while handling packet", e);
                 }
             }
 
             threadSocket.close();
-            logger.trace("Socket closed");
+            LOG.trace("Socket closed");
             running = false;
         }
 
@@ -169,17 +179,24 @@ public class LispSouthboundPlugin extends AbstractBindingAwareProvider implement
     }
 
     private void startIOThread() {
+        if (socket != null) {
+            while (!socket.isClosed()) {
+                try {
+                    Thread.sleep(500);
+                } catch (InterruptedException e) {
+                }
+            }
+        }
         try {
             socket = new DatagramSocket(new InetSocketAddress(bindingAddress, LispMessage.PORT_NUM));
             lispThread = new LispIoThread(socket, lispSouthboundService);
             lispThread.start();
-            logger.info("LISP (RFC6830) Mapping Service Southbound Plugin is up!");
+            LOG.info("LISP (RFC6830) Mapping Service Southbound Plugin is up!");
             if (listenOnXtrPort) {
                 restartXtrThread();
             }
         } catch (SocketException e) {
-            logger.error("couldn't start socket {}", e.getMessage());
-            e.printStackTrace();
+            LOG.error("couldn't start socket: {}", ExceptionUtils.getStackTrace(e));
         }
     }
 
@@ -189,62 +206,34 @@ public class LispSouthboundPlugin extends AbstractBindingAwareProvider implement
             xtrSocket = new DatagramSocket(new InetSocketAddress(bindingAddress, xtrPort));
             xtrThread = new LispIoThread(xtrSocket, lispXtrSouthboundService);
             xtrThread.start();
-            logger.info("xTR Southbound Plugin is up!");
+            LOG.info("xTR Southbound Plugin is up!");
         } catch (SocketException e) {
-            logger.warn("failed to start xtr thread: {}", e.getMessage());
-        }
-    }
-
-    public void onSessionInitiated(ProviderContext session) {
-        logger.info("LISP (RFC6830) Mapping Service is up!");
-        synchronized (startLock) {
-            if (!alreadyInit) {
-                alreadyInit = true;
-                lispSouthboundService = new LispSouthboundService();
-                lispXtrSouthboundService = new LispXtrSouthboundService();
-                registerWithOSGIConsole();
-                registerRPCs(session);
-                logger.trace("Provider Session initialized");
-                if (bindingAddress == null) {
-                    setLispAddress("0.0.0.0");
-                }
-            }
-
+            LOG.warn("failed to start xtr thread: {}", ExceptionUtils.getStackTrace(e));
         }
     }
 
-    private void registerRPCs(ProviderContext session) {
-        try {
-            lispSouthboundService.setNotificationProvider(session.getSALService(NotificationProviderService.class));
-            lispXtrSouthboundService.setNotificationProvider(session.getSALService(NotificationProviderService.class));
-            session.addRpcImplementation(LispflowmappingService.class, this);
-        } catch (Exception e) {
-            logger.error(e.getMessage(), e);
-        }
-    }
-
-    private void handleSerializedLispBuffer(TransportAddress address, ByteBuffer outBuffer, String packetType) {
+    public void handleSerializedLispBuffer(TransportAddress address, ByteBuffer outBuffer, String packetType) {
         DatagramPacket packet = new DatagramPacket(outBuffer.array(), outBuffer.limit());
         packet.setPort(address.getPort().getValue());
-        InetAddress ip = InetAddresses.forString(address.getIpAddress().getIpv4Address().getValue());
+        InetAddress ip = InetAddresses.forString(new String(address.getIpAddress().getValue()));
         packet.setAddress(ip);
         try {
-            if (logger.isDebugEnabled()) {
-                logger.trace("Sending " + packetType + " on port " + address.getPort().getValue() + " to address: " + ip);
+            if (LOG.isDebugEnabled()) {
+                LOG.trace("Sending " + packetType + " on port " + address.getPort().getValue() + " to address: " + ip);
             }
             socket.send(packet);
         } catch (IOException e) {
-            logger.warn("Failed to send " + packetType, e);
+            LOG.warn("Failed to send " + packetType, e);
         }
     }
 
     public void setLispAddress(String address) {
         synchronized (startLock) {
             if (bindingAddress != null && bindingAddress.equals(address)) {
-                logger.trace("configured lisp binding address didn't change.");
+                LOG.trace("configured lisp binding address didn't change.");
             } else {
                 String action = (bindingAddress == null ? "Setting" : "Resetting");
-                logger.trace(action + " lisp binding address to: " + address);
+                LOG.trace(action + " lisp binding address to: " + address);
                 bindingAddress = address;
                 if (lispThread != null) {
                     lispThread.stopRunning();
@@ -273,50 +262,14 @@ public class LispSouthboundPlugin extends AbstractBindingAwareProvider implement
         }
     }
 
-    @Override
-    public Future<RpcResult<Void>> sendMapNotify(SendMapNotifyInput mapNotifyInput) {
-        logger.trace("sendMapNotify called!!");
-        if (mapNotifyInput != null) {
-            ByteBuffer outBuffer = MapNotifySerializer.getInstance().serialize(mapNotifyInput.getMapNotify());
-            handleSerializedLispBuffer(mapNotifyInput.getTransportAddress(), outBuffer, MAP_NOTIFY);
-        } else {
-            logger.warn("MapNotify was null");
-        }
-        return null;
-    }
-
-    @Override
-    public Future<RpcResult<Void>> sendMapReply(SendMapReplyInput mapReplyInput) {
-        logger.trace("sendMapReply called!!");
-        if (mapReplyInput != null) {
-            ByteBuffer outBuffer = MapReplySerializer.getInstance().serialize(mapReplyInput.getMapReply());
-            handleSerializedLispBuffer(mapReplyInput.getTransportAddress(), outBuffer, MAP_REPlY);
-        } else {
-            logger.warn("MapReply was null");
-        }
-        return null;
-    }
-
-    @Override
-    public Future<RpcResult<Void>> sendMapRequest(SendMapRequestInput mapRequestInput) {
-        logger.trace("sendMapRequest called!!");
-        if (mapRequestInput != null) {
-            ByteBuffer outBuffer = MapRequestSerializer.getInstance().serialize(mapRequestInput.getMapRequest());
-            handleSerializedLispBuffer(mapRequestInput.getTransportAddress(), outBuffer, MAP_REQUEST);
-        } else {
-            logger.debug("MapRequest was null");
-        }
-        return null;
-    }
-
     @Override
     public void shouldListenOnXtrPort(boolean shouldListenOnXtrPort) {
         listenOnXtrPort = shouldListenOnXtrPort;
         if (listenOnXtrPort) {
-            logger.debug("restarting xtr thread");
+            LOG.debug("restarting xtr thread");
             restartXtrThread();
         } else {
-            logger.debug("terminating thread");
+            LOG.debug("terminating thread");
             stopXtrThread();
         }
     }
@@ -328,4 +281,15 @@ public class LispSouthboundPlugin extends AbstractBindingAwareProvider implement
             restartXtrThread();
         }
     }
+
+    @Override
+    public void close() throws Exception {
+        unloadActions();
+        controlPlaneRpc.close();
+    }
+
+    @Override
+    public void onSessionInitiated(ProviderContext session) {
+        LOG.debug("LispSouthboundPlugin Provider Session Initiated");
+    }
 }