881f697adf85d0f8cdf50ec12844723688c9af46
[openflowjava.git] / 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 io.netty.util.concurrent.GenericFutureListener;
17 import java.net.InetAddress;
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 LOGGER = LoggerFactory
35             .getLogger(TcpChannelInitializer.class);
36     private final DefaultChannelGroup allChannels;
37     private final ConnectionAdapterFactory connectionAdapterFactory;
38
39     /**
40      * default ctor
41      */
42     public TcpChannelInitializer() {
43         this( new DefaultChannelGroup("netty-receiver", null), new ConnectionAdapterFactoryImpl() );
44     }
45
46     /**
47      * Testing Constructor
48      *
49      */
50     protected TcpChannelInitializer( final DefaultChannelGroup channelGroup, final ConnectionAdapterFactory connAdaptorFactory ) {
51         allChannels = channelGroup ;
52         connectionAdapterFactory = connAdaptorFactory ;
53     }
54
55     @Override
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             LOGGER.debug("Incoming connection from (remote address): {}:{} --> :{}",
62                             switchAddress.toString(), remotePort, port);
63
64             if (!getSwitchConnectionHandler().accept(switchAddress)) {
65                 ch.disconnect();
66                 LOGGER.debug("Incoming connection rejected");
67                 return;
68             }
69         }
70         LOGGER.debug("Incoming connection accepted - building pipeline");
71         allChannels.add(ch);
72         ConnectionFacade connectionFacade = null;
73         connectionFacade = connectionAdapterFactory.createConnectionFacade(ch, null, useBarrier());
74         try {
75             LOGGER.debug("calling plugin: {}", getSwitchConnectionHandler());
76             getSwitchConnectionHandler().onSwitchConnected(connectionFacade);
77             connectionFacade.checkListeners();
78             ch.pipeline().addLast(PipelineHandlers.IDLE_HANDLER.name(), new IdleHandler(getSwitchIdleTimeout(), TimeUnit.MILLISECONDS));
79             boolean tlsPresent = false;
80
81             // If this channel is configured to support SSL it will only support SSL
82             if (getTlsConfiguration() != null) {
83                 tlsPresent = true;
84                 final SslContextFactory sslFactory = new SslContextFactory(getTlsConfiguration());
85                 final SSLEngine engine = sslFactory.getServerContext().createSSLEngine();
86                 engine.setNeedClientAuth(true);
87                 engine.setUseClientMode(false);
88                 List<String> suitesList = getTlsConfiguration().getCipherSuites();
89                 if (suitesList != null && !suitesList.isEmpty()) {
90                     LOGGER.debug("Requested Cipher Suites are: {}", suitesList);
91                     String[] suites = suitesList.toArray(new String[suitesList.size()]);
92                     engine.setEnabledCipherSuites(suites);
93                     LOGGER.debug("Cipher suites enabled in SSLEngine are: {}", engine.getEnabledCipherSuites().toString());
94                 }
95                 final SslHandler ssl = new SslHandler(engine);
96                 final Future<Channel> handshakeFuture = ssl.handshakeFuture();
97                 final ConnectionFacade finalConnectionFacade = connectionFacade;
98                 handshakeFuture.addListener(new GenericFutureListener<Future<? super Channel>>() {
99                     @Override
100                     public void operationComplete(final Future<? super Channel> future) throws Exception {
101                         finalConnectionFacade.fireConnectionReadyNotification();
102                     }
103                 });
104                 ch.pipeline().addLast(PipelineHandlers.SSL_HANDLER.name(), ssl);
105             }
106             ch.pipeline().addLast(PipelineHandlers.OF_FRAME_DECODER.name(),
107                     new OFFrameDecoder(connectionFacade, tlsPresent));
108             ch.pipeline().addLast(PipelineHandlers.OF_VERSION_DETECTOR.name(), new OFVersionDetector());
109             final OFDecoder ofDecoder = new OFDecoder();
110             ofDecoder.setDeserializationFactory(getDeserializationFactory());
111             ch.pipeline().addLast(PipelineHandlers.OF_DECODER.name(), ofDecoder);
112             final OFEncoder ofEncoder = new OFEncoder();
113             ofEncoder.setSerializationFactory(getSerializationFactory());
114             ch.pipeline().addLast(PipelineHandlers.OF_ENCODER.name(), ofEncoder);
115             ch.pipeline().addLast(PipelineHandlers.DELEGATING_INBOUND_HANDLER.name(), new DelegatingInboundHandler(connectionFacade));
116             if (!tlsPresent) {
117                 connectionFacade.fireConnectionReadyNotification();
118             }
119         } catch (final Exception e) {
120             LOGGER.warn("Failed to initialize channel", e);
121             ch.close();
122         }
123     }
124
125     /**
126      * @return iterator through active connections
127      */
128     public Iterator<Channel> getConnectionIterator() {
129         return allChannels.iterator();
130     }
131
132     /**
133      * @return amount of active channels
134      */
135     public int size() {
136         return allChannels.size();
137     }
138 }