net.ieldor.network.codec.handshake.HandshakeDecoder.java Source code

Introduction

Here is the source code for net.ieldor.network.codec.handshake.HandshakeDecoder.java

Source

/*
 * This file is part of Ieldor.
 *
 * Ieldor is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Ieldor is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Ieldor.  If not, see <http://www.gnu.org/licenses/>.
 */
package net.ieldor.network.codec.handshake;

import net.ieldor.network.codec.login.LoginDataEncoder;
import net.ieldor.network.codec.login.LoginDecoder;
import net.ieldor.network.codec.login.LoginEncoder;
import net.ieldor.network.codec.messages.HandshakeMessage;
import net.ieldor.network.codec.messages.LoginResponse;
import net.ieldor.network.codec.update.UpdateDecoder;
import net.ieldor.network.codec.update.UpdateEncoder;
import net.ieldor.network.codec.update.UpdateStatusEncoder;
import net.ieldor.network.codec.update.XorEncoder;
import net.ieldor.network.codec.worldlist.WorldListDecoder;
import net.ieldor.network.codec.worldlist.WorldListEncoder;
import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundByteHandlerAdapter;

/**
 * An {@link ChannelInboundByteHandlerAdapter} that is used to handle the
 * handshake procedure of the server->client connection.
 * 
 * @author Thomas Le Godais <thomaslegodais@live.com>
 * 
 */
public class HandshakeDecoder extends ChannelInboundByteHandlerAdapter {

    /* (non-Javadoc)
     * @see io.netty.channel.ChannelInboundByteHandlerAdapter#inboundBufferUpdated(io.netty.channel.ChannelHandlerContext, io.netty.buffer.ByteBuf)
     */
    @Override
    public void inboundBufferUpdated(ChannelHandlerContext ctx, ByteBuf in) throws Exception {
        if (!in.readable())
            return;
        ctx.pipeline().remove(HandshakeDecoder.class);

        int incomingOpcode = in.readByte() & 0xFF;
        //System.out.println("Received handshake opcode: "+incomingOpcode);
        HandshakeState handshakeState = HandshakeState.forId(incomingOpcode);

        if (handshakeState == null)
            return;
        switch (handshakeState) {
        case HANDSHAKE_UPDATE:
            ctx.pipeline().addFirst(new UpdateEncoder(), new UpdateStatusEncoder(), new XorEncoder(),
                    new UpdateDecoder());
            break;
        case HANDSHAKE_WORLD_LIST:
            ctx.pipeline().addFirst(new WorldListEncoder(), new WorldListDecoder());
            break;
        case HANDSHAKE_LOGIN:
            ctx.pipeline().addFirst(new LoginEncoder(), new LoginDataEncoder());
            ctx.pipeline().addFirst("decoder", new LoginDecoder());
            ctx.write(new LoginResponse(0));
            break;
        default:
            break;
        }

        ctx.nextInboundMessageBuffer().add(new HandshakeMessage(handshakeState));
        ctx.fireInboundBufferUpdated();

        if (in.readable()) {
            ChannelHandlerContext head = ctx.pipeline().firstContext();
            head.nextInboundByteBuffer().writeBytes(in.readBytes(in.readableBytes()));
            head.fireInboundBufferUpdated();
        }
    }
}