org.maodian.flyingcat.netty.handler.XmppXMLStreamHandler.java Source code

Introduction

Here is the source code for org.maodian.flyingcat.netty.handler.XmppXMLStreamHandler.java

Source

/*
 * Copyright 2013 - 2013 Cole Wen
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.maodian.flyingcat.netty.handler;

import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelFutureListener;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundMessageHandlerAdapter;

import java.lang.invoke.MethodHandles;

import org.apache.commons.lang3.StringUtils;
import org.maodian.flyingcat.xmpp.state.StanzaError;
import org.maodian.flyingcat.xmpp.state.XmppContext;
import org.maodian.flyingcat.xmpp.state.XmppError;
import org.maodian.flyingcat.xmpp.state.XmppException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author Cole Wen
 * 
 */
public class XmppXMLStreamHandler extends ChannelInboundMessageHandlerAdapter<String> {
    private static final Logger log = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
    private XmppContext xmppContext;

    // true if the </stream:stream> is sent first by server
    private boolean initCloseingStream = false;

    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        xmppContext.setNettyChannelHandlerContext(ctx);
        super.channelActive(ctx);
    }

    /*
     * (non-Javadoc)
     * 
     * @see
     * io.netty.channel.ChannelInboundMessageHandlerAdapter#messageReceived(io
     * .netty.channel.ChannelHandlerContext, java.lang.Object)
     */
    @Override
    public void messageReceived(ChannelHandlerContext ctx, String msg) throws Exception {
        // discard xml declaration
        if (StringUtils.startsWith(msg, "<?xml ")) {
            return;
        }

        // deal with </stream:stream>
        if (StringUtils.contains(msg, ":stream") && StringUtils.contains(msg, "</")) {
            xmppContext.destroy();
            if (initCloseingStream) {
                log.info("Close Stream and underhood socket due to requested by server");
                ctx.channel().close();
            } else if (ctx.channel().isOpen()) {
                ctx.write("</stream:stream>").addListener(new ChannelFutureListener() {

                    @Override
                    public void operationComplete(ChannelFuture future) throws Exception {
                        log.info("Close Stream and underhood socket due to requested by client");
                        future.channel().close();
                    }
                });
            } else {
                log.info(
                        "Won't respond client's close stream request since the channel has been closed (may because ssl has been closed)");
            }
            return;
        }

        xmppContext.parseXML(msg);
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        log.error("Catch exceptions", cause);
        if (cause instanceof XmppException) {
            XmppException xmppException = (XmppException) cause;
            XmppError error = xmppException.getXmppError();
            if (error instanceof StanzaError) {
                ctx.write(error.toXML()).addListener(ChannelFutureListener.CLOSE_ON_FAILURE);
            } else {
                xmppContext.destroy();
                StringBuilder xml = new StringBuilder(xmppException.getXmppError().toXML())
                        .append("</stream:stream>");
                initCloseingStream = true;
                ctx.write(xml.toString()).addListener(ChannelFutureListener.CLOSE_ON_FAILURE);
                log.error("Close the XMPP Stream due to error", cause);
            }
            return;
        }
        xmppContext.destroy();
        String xml = "</stream:stream>";
        initCloseingStream = true;
        ctx.write(xml).addListener(ChannelFutureListener.CLOSE_ON_FAILURE);
        log.error("Close the XMPP Stream due to error");
        super.exceptionCaught(ctx, cause);
    }

    public void setXmppContext(XmppContext xmppContext) {
        this.xmppContext = xmppContext;
    }

}