Merge "BUG-6118: making the OFentityListener aware of the InJeopardy() flag"
[openflowplugin.git] / openflowplugin-impl / src / main / java / org / opendaylight / openflowplugin / impl / connection / listener / HandshakeListenerImpl.java
1 /**
2  * Copyright (c) 2015 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
9 package org.opendaylight.openflowplugin.impl.connection.listener;
10
11 import com.google.common.util.concurrent.FutureCallback;
12 import com.google.common.util.concurrent.Futures;
13 import com.google.common.util.concurrent.JdkFutureAdapters;
14 import com.google.common.util.concurrent.ListenableFuture;
15 import javax.annotation.Nullable;
16 import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
17 import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionStatus;
18 import org.opendaylight.openflowplugin.api.openflow.connection.HandshakeContext;
19 import org.opendaylight.openflowplugin.api.openflow.device.handlers.DeviceConnectedHandler;
20 import org.opendaylight.openflowplugin.api.openflow.md.core.HandshakeListener;
21 import org.opendaylight.openflowplugin.impl.statistics.ofpspecific.SessionStatistics;
22 import org.opendaylight.openflowplugin.openflow.md.util.InventoryDataServiceUtil;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.BarrierInput;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.BarrierInputBuilder;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.BarrierOutput;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetFeaturesOutput;
27 import org.opendaylight.yangtools.yang.common.RpcResult;
28 import org.slf4j.Logger;
29 import org.slf4j.LoggerFactory;
30
31 /**
32  *
33  */
34 public class HandshakeListenerImpl implements HandshakeListener {
35
36     private static final Logger LOG = LoggerFactory.getLogger(HandshakeListenerImpl.class);
37
38     private final ConnectionContext connectionContext;
39     private final DeviceConnectedHandler deviceConnectedHandler;
40     private HandshakeContext handshakeContext;
41
42     /**
43      * @param connectionContext
44      * @param deviceConnectedHandler
45      */
46     public HandshakeListenerImpl(final ConnectionContext connectionContext, final DeviceConnectedHandler deviceConnectedHandler) {
47         this.connectionContext = connectionContext;
48         this.deviceConnectedHandler = deviceConnectedHandler;
49     }
50
51     @Override
52     public void onHandshakeSuccessful(final GetFeaturesOutput featureOutput, final Short version) {
53         if (LOG.isDebugEnabled()) {
54             LOG.debug("handshake succeeded: {}", connectionContext.getConnectionAdapter().getRemoteAddress());
55         }
56         this.handshakeContext.close();
57         connectionContext.changeStateToWorking();
58         connectionContext.setFeatures(featureOutput);
59         connectionContext.setNodeId(InventoryDataServiceUtil.nodeIdFromDatapathId(featureOutput.getDatapathId()));
60         connectionContext.handshakeSuccessful();
61
62         // fire barrier in order to sweep all handshake and posthandshake messages before continue
63         final ListenableFuture<RpcResult<BarrierOutput>> barrier = fireBarrier(version, 0L);
64         Futures.addCallback(barrier, addBarrierCallback());
65     }
66
67     private FutureCallback<RpcResult<BarrierOutput>> addBarrierCallback() {
68         return new FutureCallback<RpcResult<BarrierOutput>>() {
69             @Override
70             public void onSuccess(@Nullable final RpcResult<BarrierOutput> result) {
71                 if (LOG.isDebugEnabled()) {
72                     LOG.debug("succeeded by getting sweep barrier after post-handshake for device {}", connectionContext.getNodeId().getValue());
73                 }
74                 try {
75                     ConnectionStatus connectionStatusResult = deviceConnectedHandler.deviceConnected(connectionContext);
76                     if (!ConnectionStatus.MAY_CONTINUE.equals(connectionStatusResult)) {
77                         connectionContext.closeConnection(true);
78                     }
79                     SessionStatistics.countEvent(connectionContext.getNodeId().toString(),
80                             SessionStatistics.ConnectionStatus.CONNECTION_CREATED);
81                 } catch (final Exception e) {
82                     LOG.error("ConnectionContext initial processing failed: ", e);
83                     SessionStatistics.countEvent(connectionContext.getNodeId().toString(),
84                             SessionStatistics.ConnectionStatus.CONNECTION_DISCONNECTED_BY_OFP);
85                     connectionContext.closeConnection(true);
86                 }
87             }
88
89             @Override
90             public void onFailure(final Throwable t) {
91                 LOG.error("failed to get sweep barrier after post-handshake for device {}", connectionContext.getNodeId());
92                 connectionContext.closeConnection(false);
93             }
94         };
95     }
96
97     private ListenableFuture<RpcResult<BarrierOutput>> fireBarrier(final Short version, final long xid) {
98         final BarrierInput barrierInput = new BarrierInputBuilder()
99                 .setXid(xid)
100                 .setVersion(version)
101                 .build();
102         return JdkFutureAdapters.listenInPoolThread(
103                 this.connectionContext.getConnectionAdapter().barrier(barrierInput));
104     }
105
106     @Override
107     public void onHandshakeFailure() {
108         if (LOG.isDebugEnabled()) {
109             LOG.debug("handshake failed: {}", this.connectionContext.getConnectionAdapter().getRemoteAddress());
110         }
111         this.handshakeContext.close();
112         this.connectionContext.closeConnection(false);
113     }
114
115     @Override
116     public void setHandshakeContext(final HandshakeContext handshakeContext) {
117         this.handshakeContext = handshakeContext;
118     }
119 }