openflowplugin/src/main/java/org/opendaylight/openflowplugin/openflow/md/core/sal/SalRegistrationManager.java

   1 /**
   2  * Copyright (c) 2013 Cisco Systems, 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 package org.opendaylight.openflowplugin.openflow.md.core.sal;
   9
  10 import java.math.BigInteger;
  11 import java.net.Inet4Address;
  12 import java.net.Inet6Address;
  13 import java.net.InetAddress;
  14 import java.net.InetSocketAddress;
  15 import java.util.Collection;
  16 import java.util.concurrent.Future;
  17
  18 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.ProviderContext;
  19 import org.opendaylight.controller.sal.binding.api.NotificationProviderService;
  20 import org.opendaylight.controller.sal.binding.api.data.DataProviderService;
  21 import org.opendaylight.openflowplugin.openflow.md.ModelDrivenSwitch;
  22 import org.opendaylight.openflowplugin.openflow.md.core.session.OFSessionUtil;
  23 import org.opendaylight.openflowplugin.openflow.md.core.session.SessionContext;
  24 import org.opendaylight.openflowplugin.openflow.md.core.session.SessionListener;
  25 import org.opendaylight.openflowplugin.openflow.md.core.session.SessionManager;
  26 import org.opendaylight.openflowplugin.openflow.md.core.session.SwitchSessionKeyOF;
  27 import org.opendaylight.openflowplugin.openflow.md.lldp.LLDPSpeaker;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
  29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
  30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv6Address;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeUpdated;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeUpdatedBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.GetNodeIpAddressInput;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.GetNodeIpAddressInputBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.GetNodeIpAddressOutput;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.GetNodeIpAddressOutputBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRemoved;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRemovedBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeUpdated;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeUpdatedBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetFeaturesOutput;
  47 import org.opendaylight.yangtools.concepts.CompositeObjectRegistration;
  48 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  49 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.InstanceIdentifierBuilder;
  50 import org.opendaylight.yangtools.yang.common.RpcError;
  51 import org.opendaylight.yangtools.yang.common.RpcResult;
  52 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
  53 import org.slf4j.Logger;
  54 import org.slf4j.LoggerFactory;
  55
  56 import com.google.common.util.concurrent.Futures;
  57
  58 /**
  59  * session and inventory listener implementation
  60  */
  61 public class SalRegistrationManager implements SessionListener, AutoCloseable {
  62
  63     private final static Logger LOG = LoggerFactory.getLogger(SalRegistrationManager.class);
  64
  65     private ProviderContext providerContext;
  66
  67     private NotificationProviderService publishService;
  68
  69     private DataProviderService dataService;
  70
  71     private SwitchFeaturesUtil swFeaturesUtil;
  72
  73     public SalRegistrationManager() {
  74         swFeaturesUtil = SwitchFeaturesUtil.getInstance();
  75     }
  76
  77     public NotificationProviderService getPublishService() {
  78         return publishService;
  79     }
  80
  81     public void setPublishService(NotificationProviderService publishService) {
  82         this.publishService = publishService;
  83     }
  84
  85     public ProviderContext getProviderContext() {
  86         return providerContext;
  87     }
  88
  89     public void onSessionInitiated(ProviderContext session) {
  90         LOG.debug("onSessionInitiated");
  91         this.providerContext = session;
  92         this.publishService = session.getSALService(NotificationProviderService.class);
  93         this.dataService = session.getSALService(DataProviderService.class);
  94         // We register as listener for Session Manager
  95         getSessionManager().registerSessionListener(this);
  96         getSessionManager().setNotificationProviderService(publishService);
  97         getSessionManager().setDataProviderService(dataService);
  98         LOG.debug("SalRegistrationManager initialized");
  99     }
 100
 101     @Override
 102     public void onSessionAdded(SwitchSessionKeyOF sessionKey, SessionContext context) {
 103         GetFeaturesOutput features = context.getFeatures();
 104         BigInteger datapathId = features.getDatapathId();
 105         InstanceIdentifier<Node> identifier = identifierFromDatapathId(datapathId);
 106         NodeRef nodeRef = new NodeRef(identifier);
 107         NodeId nodeId = nodeIdFromDatapathId(datapathId);
 108         ModelDrivenSwitchImpl ofSwitch = new ModelDrivenSwitchImpl(nodeId, identifier, context);
 109         LLDPSpeaker.getInstance().addModelDrivenSwitch(identifier, ofSwitch);
 110         CompositeObjectRegistration<ModelDrivenSwitch> registration = ofSwitch.register(providerContext);
 111         context.setProviderRegistration(registration);
 112
 113         LOG.debug("ModelDrivenSwitch for {} registered to MD-SAL.", datapathId.toString());
 114
 115         publishService.publish(nodeAdded(ofSwitch, features,nodeRef));
 116     }
 117
 118     @Override
 119     public void onSessionRemoved(SessionContext context) {
 120         GetFeaturesOutput features = context.getFeatures();
 121         BigInteger datapathId = features.getDatapathId();
 122         InstanceIdentifier<Node> identifier = identifierFromDatapathId(datapathId);
 123         NodeRef nodeRef = new NodeRef(identifier);
 124         NodeRemoved nodeRemoved = nodeRemoved(nodeRef);
 125         LLDPSpeaker.getInstance().removeModelDrivenSwitch(identifier);
 126         CompositeObjectRegistration<ModelDrivenSwitch> registration = context.getProviderRegistration();
 127         registration.close();
 128
 129         LOG.debug("ModelDrivenSwitch for {} unregistered from MD-SAL.", datapathId.toString());
 130         publishService.publish(nodeRemoved);
 131     }
 132
 133     private NodeUpdated nodeAdded(ModelDrivenSwitch sw, GetFeaturesOutput features, NodeRef nodeRef) {
 134         NodeUpdatedBuilder builder = new NodeUpdatedBuilder();
 135         builder.setId(sw.getNodeId());
 136         builder.setNodeRef(nodeRef);
 137         FlowCapableNodeUpdatedBuilder builder2 = new FlowCapableNodeUpdatedBuilder();
 138         builder2.setIpAddress(getIpAddressOf(sw));
 139         builder2.setSwitchFeatures(swFeaturesUtil.buildSwitchFeatures(features));
 140         builder.addAugmentation(FlowCapableNodeUpdated.class, builder2.build());
 141         return builder.build();
 142     }
 143
 144     private IpAddress getIpAddressOf(ModelDrivenSwitch sw) {
 145         SessionContext sessionContext = sw.getSessionContext();
 146         if (!sessionContext.isValid()) {
 147             LOG.warn("IP address of the node {} cannot be obtained. Session is not valid.", sw.getNodeId());
 148             return null;
 149         }
 150         InetSocketAddress remoteAddress = sessionContext.getPrimaryConductor().getConnectionAdapter()
 151                 .getRemoteAddress();
 152         if (remoteAddress == null) {
 153             LOG.warn("IP address of the node {} cannot be obtained. No connection with switch.", sw.getNodeId());
 154             return null;
 155         }
 156         return resolveIpAddress(remoteAddress.getAddress());
 157     }
 158
 159     private static IpAddress resolveIpAddress(InetAddress address) {
 160         String hostAddress = address.getHostAddress();
 161         if (address instanceof Inet4Address) {
 162             return new IpAddress(new Ipv4Address(hostAddress));
 163         }
 164         if (address instanceof Inet6Address) {
 165             return new IpAddress(new Ipv6Address(hostAddress));
 166         }
 167         throw new IllegalArgumentException("Unsupported IP address type!");
 168     }
 169
 170     private NodeRemoved nodeRemoved(NodeRef nodeRef) {
 171         NodeRemovedBuilder builder = new NodeRemovedBuilder();
 172         builder.setNodeRef(nodeRef);
 173         return builder.build();
 174     }
 175
 176     public static InstanceIdentifier<Node> identifierFromDatapathId(BigInteger datapathId) {
 177         NodeKey nodeKey = nodeKeyFromDatapathId(datapathId);
 178         InstanceIdentifierBuilder<Node> builder = InstanceIdentifier.builder(Nodes.class).child(Node.class,nodeKey);
 179         return builder.toInstance();
 180     }
 181
 182     public static NodeKey nodeKeyFromDatapathId(BigInteger datapathId) {
 183         return new NodeKey(nodeIdFromDatapathId(datapathId));
 184     }
 185
 186     public static NodeId nodeIdFromDatapathId(BigInteger datapathId) {
 187         // FIXME: Convert to textual representation of datapathID
 188         String current = datapathId.toString();
 189         return new NodeId("openflow:" + current);
 190     }
 191
 192     public SessionManager getSessionManager() {
 193         return OFSessionUtil.getSessionManager();
 194     }
 195
 196     @Override
 197     public void close() {
 198         LOG.debug("close");
 199         dataService = null;
 200         providerContext = null;
 201         publishService = null;
 202     }
 203 }