8cf0536bdcf7c21d1f8a2b0d03dc1f0a72d34b84
[openflowplugin.git] / openflowjava / openflow-protocol-impl / src / main / java / org / opendaylight / openflowjava / protocol / impl / core / TcpChannelInitializer.java
1 /*
2  * Copyright (c) 2013 Pantheon Technologies s.r.o. 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.openflowjava.protocol.impl.core;
10
11 import io.netty.channel.Channel;
12 import io.netty.channel.group.DefaultChannelGroup;
13 import io.netty.channel.socket.SocketChannel;
14 import io.netty.handler.ssl.SslHandler;
15 import io.netty.util.concurrent.Future;
16 import java.net.InetAddress;
17 import java.util.Arrays;
18 import java.util.Iterator;
19 import java.util.List;
20 import java.util.concurrent.TimeUnit;
21 import javax.net.ssl.SSLEngine;
22 import org.opendaylight.openflowjava.protocol.impl.core.connection.ConnectionAdapterFactory;
23 import org.opendaylight.openflowjava.protocol.impl.core.connection.ConnectionAdapterFactoryImpl;
24 import org.opendaylight.openflowjava.protocol.impl.core.connection.ConnectionFacade;
25 import org.slf4j.Logger;
26 import org.slf4j.LoggerFactory;
27
28 /**
29  * Initializes TCP / TLS channel.
30  * @author michal.polkorab
31  */
32 public class TcpChannelInitializer extends ProtocolChannelInitializer<SocketChannel> {
33
34     private static final Logger LOG = LoggerFactory.getLogger(TcpChannelInitializer.class);
35     private final DefaultChannelGroup allChannels;
36     private final ConnectionAdapterFactory connectionAdapterFactory;
37
38     /**
39      * Default constructor.
40      */
41     public TcpChannelInitializer() {
42         this(new DefaultChannelGroup("netty-receiver", null), new ConnectionAdapterFactoryImpl());
43     }
44
45     /**
46      * Testing constructor.
47      */
48     protected TcpChannelInitializer(final DefaultChannelGroup channelGroup,
49             final ConnectionAdapterFactory connAdaptorFactory) {
50         allChannels = channelGroup;
51         connectionAdapterFactory = connAdaptorFactory;
52     }
53
54     @Override
55     @SuppressWarnings("checkstyle:IllegalCatch")
56     protected void initChannel(final SocketChannel ch) {
57         if (ch.remoteAddress() != null) {
58             final InetAddress switchAddress = ch.remoteAddress().getAddress();
59             final int port = ch.localAddress().getPort();
60             final int remotePort = ch.remoteAddress().getPort();
61             LOG.debug("Incoming connection from (remote address): {}:{} --> :{}",
62                 switchAddress.toString(), remotePort, port);
63
64             if (!getSwitchConnectionHandler().accept(switchAddress)) {
65                 ch.disconnect();
66                 LOG.debug("Incoming connection rejected");
67                 return;
68             }
69         }
70         LOG.debug("Incoming connection accepted - building pipeline");
71         allChannels.add(ch);
72         ConnectionFacade connectionFacade = null;
73         connectionFacade = connectionAdapterFactory.createConnectionFacade(ch, null, useBarrier());
74         try {
75             LOG.debug("Calling OF plugin: {}", getSwitchConnectionHandler());
76             getSwitchConnectionHandler().onSwitchConnected(connectionFacade);
77             connectionFacade.checkListeners();
78             ch.pipeline().addLast(PipelineHandlers.IDLE_HANDLER.name(),
79                     new IdleHandler(getSwitchIdleTimeout(), TimeUnit.MILLISECONDS));
80             boolean tlsPresent = false;
81
82             // If this channel is configured to support SSL it will only support SSL
83             if (getTlsConfiguration() != null) {
84                 tlsPresent = true;
85                 final SslContextFactory sslFactory = new SslContextFactory(getTlsConfiguration());
86                 final SSLEngine engine = sslFactory.getServerContext().createSSLEngine();
87                 engine.setNeedClientAuth(true);
88                 engine.setUseClientMode(false);
89                 List<String> suitesList = getTlsConfiguration().getCipherSuites();
90                 if (suitesList != null && !suitesList.isEmpty()) {
91                     LOG.debug("Requested Cipher Suites are: {}", suitesList);
92                     String[] suites = suitesList.toArray(new String[suitesList.size()]);
93                     engine.setEnabledCipherSuites(suites);
94                     LOG.debug("Cipher suites enabled in SSLEngine are: {}",
95                             Arrays.toString(engine.getEnabledCipherSuites()));
96                 }
97                 final SslHandler ssl = new SslHandler(engine);
98                 final Future<Channel> handshakeFuture = ssl.handshakeFuture();
99                 final ConnectionFacade finalConnectionFacade = connectionFacade;
100                 handshakeFuture.addListener(future -> finalConnectionFacade.fireConnectionReadyNotification());
101                 ch.pipeline().addLast(PipelineHandlers.SSL_HANDLER.name(), ssl);
102             }
103             ch.pipeline().addLast(PipelineHandlers.OF_FRAME_DECODER.name(),
104                     new OFFrameDecoder(connectionFacade, tlsPresent));
105             ch.pipeline().addLast(PipelineHandlers.OF_VERSION_DETECTOR.name(), new OFVersionDetector());
106             final OFDecoder ofDecoder = new OFDecoder();
107             ofDecoder.setDeserializationFactory(getDeserializationFactory());
108             ch.pipeline().addLast(PipelineHandlers.OF_DECODER.name(), ofDecoder);
109             final OFEncoder ofEncoder = new OFEncoder();
110             ofEncoder.setSerializationFactory(getSerializationFactory());
111             ch.pipeline().addLast(PipelineHandlers.OF_ENCODER.name(), ofEncoder);
112             ch.pipeline().addLast(PipelineHandlers.DELEGATING_INBOUND_HANDLER.name(),
113                     new DelegatingInboundHandler(connectionFacade));
114             if (!tlsPresent) {
115                 connectionFacade.fireConnectionReadyNotification();
116             }
117         } catch (RuntimeException e) {
118             LOG.warn("Failed to initialize channel", e);
119             ch.close();
120         }
121     }
122
123     /**
124      * Returns the connection iterator.
125      *
126      * @return iterator through active connections
127      */
128     public Iterator<Channel> getConnectionIterator() {
129         return allChannels.iterator();
130     }
131
132     /**
133      * Returns the number of active channels.
134      *
135      * @return amount of active channels
136      */
137     public int size() {
138         return allChannels.size();
139     }
140 }