View Javadoc
1   /*
2    * This file is part of Waarp Project (named also Waarp or GG).
3    *
4    *  Copyright (c) 2019, Waarp SAS, and individual contributors by the @author
5    *  tags. See the COPYRIGHT.txt in the distribution for a full listing of
6    * individual contributors.
7    *
8    *  All Waarp Project is free software: you can redistribute it and/or
9    * modify it under the terms of the GNU General Public License as published by
10   * the Free Software Foundation, either version 3 of the License, or (at your
11   * option) any later version.
12   *
13   * Waarp is distributed in the hope that it will be useful, but WITHOUT ANY
14   * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
15   * A PARTICULAR PURPOSE. See the GNU General Public License for more details.
16   *
17   *  You should have received a copy of the GNU General Public License along with
18   * Waarp . If not, see <http://www.gnu.org/licenses/>.
19   */
20  package org.waarp.ftp.core.control.ftps;
21  
22  import io.netty.channel.Channel;
23  import io.netty.channel.ChannelHandlerContext;
24  import org.waarp.common.crypto.ssl.WaarpSslUtility;
25  import org.waarp.common.logging.WaarpLogger;
26  import org.waarp.common.logging.WaarpLoggerFactory;
27  import org.waarp.ftp.core.control.NetworkHandler;
28  import org.waarp.ftp.core.session.FtpSession;
29  
30  /**
31   *
32   */
33  public class SslNetworkHandler extends NetworkHandler {
34    /**
35     * Internal Logger
36     */
37    private static final WaarpLogger logger =
38        WaarpLoggerFactory.getLogger(SslNetworkHandler.class);
39  
40    /**
41     * @param session
42     */
43    public SslNetworkHandler(final FtpSession session) {
44      super(session);
45    }
46  
47    @Override
48    public void channelRegistered(final ChannelHandlerContext ctx)
49        throws Exception {
50      final Channel channel = ctx.channel();
51      logger.debug("Add channel to ssl {}", channel);
52      WaarpSslUtility.addSslOpenedChannel(channel);
53      getFtpSession().prepareSsl();
54      super.channelRegistered(ctx);
55    }
56  
57    /**
58     * To be extended to inform of an error to SNMP support
59     *
60     * @param error1
61     * @param error2
62     */
63    @Override
64    protected void callForSnmp(final String error1, final String error2) {
65      // ignore
66    }
67  
68    @Override
69    public void channelActive(final ChannelHandlerContext ctx) throws Exception {
70      if (!WaarpSslUtility.waitForHandshake(ctx.channel())) {
71        callForSnmp("SSL Connection Error", "During Ssl Handshake");
72        getFtpSession().setSsl(false);
73        return;
74      } else {
75        getFtpSession().setSsl(true);
76      }
77      super.channelActive(ctx);
78    }
79  
80  }