me.bigteddy98.slimeportal.protocol.handlers.ClientSideHandler.java Source code

Introduction

Here is the source code for me.bigteddy98.slimeportal.protocol.handlers.ClientSideHandler.java

Source

/* 
 * SlimePortal
 * Copyright (C) 2014 Sander Gielisse
 *
 *     This program 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.
 *
 *     This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package me.bigteddy98.slimeportal.protocol.handlers;

import io.netty.bootstrap.Bootstrap;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.channel.Channel;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelFutureListener;
import io.netty.channel.ChannelHandlerAdapter;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelOption;

import java.util.List;

import me.bigteddy98.slimeportal.SlimeLogger;
import me.bigteddy98.slimeportal.protocol.NetworkManager;
import me.bigteddy98.slimeportal.protocol.ServerboundConnectionInitializer;
import me.bigteddy98.slimeportal.protocol.packet.Packet;
import me.bigteddy98.slimeportal.protocol.packet.PacketDataWrapper;
import me.bigteddy98.slimeportal.protocol.packet.PacketReceiveEvent;

public class ClientSideHandler extends ChannelHandlerAdapter {

    public final String hostname;
    public final int port;
    public final NetworkManager networkManager = new NetworkManager(this);

    public volatile ServerboundConnectionInitializer serverboundConnectionInitializer;
    public volatile Channel incomingChannel;
    public volatile Channel outgoingChannel;

    public ClientSideHandler(String hostname, int port) {
        this.hostname = hostname;
        this.port = port;
    }

    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        this.networkManager.disconnect();
        super.channelInactive(ctx);
    }

    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        incomingChannel = ctx.channel();
        networkManager.clientsidePipeline = ctx.pipeline();

        Bootstrap bootstrab = new Bootstrap();
        bootstrab.group(incomingChannel.eventLoop());
        bootstrab.channel(ctx.channel().getClass());
        bootstrab.handler(serverboundConnectionInitializer = new ServerboundConnectionInitializer(networkManager,
                incomingChannel));
        bootstrab.option(ChannelOption.AUTO_READ, false);
        ChannelFuture f = bootstrab.connect(hostname, port);

        outgoingChannel = f.channel();
        f.addListener(new ChannelFutureListener() {
            @Override
            public void operationComplete(ChannelFuture future) throws Exception {
                if (future.isSuccess()) {
                    incomingChannel.read();
                } else {
                    incomingChannel.close();
                }
            }
        });
    }

    @Override
    public void channelRead(final ChannelHandlerContext ctx, Object msg) throws Exception {
        if (outgoingChannel.isActive()) {
            ByteBuf bufferClone = Unpooled.copiedBuffer((ByteBuf) msg);
            final List<Packet> packets = this.networkManager.handleServerBoundPackets(bufferClone);
            bufferClone.release();
            if (!packets.isEmpty()) {
                for (final Packet packet : packets) {
                    PacketReceiveEvent event = new PacketReceiveEvent();
                    packet.onReceive(this.networkManager, event);
                    if (event.isCancelled()) {
                        ctx.channel().read();
                        continue;
                    }

                    PacketDataWrapper excludingSize = new PacketDataWrapper(Unpooled.buffer());
                    excludingSize.writeVarInt(packet.getId());
                    packet.write(excludingSize);

                    final PacketDataWrapper includingSize = new PacketDataWrapper(Unpooled.buffer());
                    includingSize.writeVarInt(excludingSize.readableBytes());
                    includingSize.writeBytes(excludingSize.getBuffer());
                    excludingSize.getBuffer().release();

                    outgoingChannel.writeAndFlush(includingSize.getBuffer())
                            .addListener(new ChannelFutureListener() {
                                @Override
                                public void operationComplete(ChannelFuture future) throws Exception {
                                    packet.onSend(networkManager);
                                    if (future.isSuccess()) {
                                        ctx.channel().read();
                                    } else {
                                        future.channel().close();
                                    }
                                }
                            });
                }
                ((ByteBuf) msg).release();
            } else {
                outgoingChannel.writeAndFlush(msg).addListener(new ChannelFutureListener() {
                    @Override
                    public void operationComplete(ChannelFuture future) throws Exception {
                        if (future.isSuccess()) {
                            ctx.channel().read();
                        } else {
                            future.channel().close();
                        }
                    }
                });
            }
        }
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        SlimeLogger.exception(cause);
        super.exceptionCaught(ctx, cause);
    }
}