org.dragonet.net.inf.mcpe.NetworkHandler.java Source code

Introduction

Here is the source code for org.dragonet.net.inf.mcpe.NetworkHandler.java

Source

/*
 * GNU LESSER GENERAL PUBLIC LICENSE
 *                       Version 3, 29 June 2007
 *
 * Copyright (C) 2007 Free Software Foundation, Inc. <http://fsf.org/>
 * Everyone is permitted to copy and distribute verbatim copies
 * of this license document, but changing it is not allowed.
 *
 * You can view LICENCE file for details. 
 *
 * @author The Dragonet Team
 */
package org.dragonet.net.inf.mcpe;

import org.dragonet.net.inf.mcpe.MCPESession;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.net.DatagramPacket;
import java.net.InetSocketAddress;
import java.net.SocketAddress;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import lombok.Getter;
import org.apache.commons.lang.ArrayUtils;
import org.dragonet.net.RaknetConstants;
import org.dragonet.net.SessionManager;
import org.dragonet.net.packet.RaknetDataPacket;
import org.dragonet.net.packet.ServerInfoPacket;
import org.dragonet.utilities.io.PEBinaryReader;
import org.dragonet.utilities.io.PEBinaryWriter;

public class NetworkHandler {

    @Getter
    private SessionManager manager;

    public final static long serverID = 0x0000000012345678L;

    @Getter
    private NonBlockUDPSocket udp;

    private final ConcurrentMap<String, PENetworkClient> clients = new ConcurrentHashMap<>();

    public NetworkHandler(SessionManager manager, InetSocketAddress address) throws Exception {
        this.manager = manager;
        this.udp = new NonBlockUDPSocket(manager, address);
        this.udp.start();
    }

    public void onTick() {
        DatagramPacket packet = null;
        while ((packet = this.udp.receive()) != null) {
            this.processPacket(packet);
        }
    }

    public void remove(SocketAddress sockAddr) {
        clients.remove(sockAddr.toString());
    }

    private void processPacket(DatagramPacket packet) {
        try {
            PEBinaryReader reader = new PEBinaryReader(new ByteArrayInputStream(packet.getData()));
            int raknetPID = reader.readByte() & 0xFF;
            switch (raknetPID) {
            case RaknetConstants.ID_PING_OPEN_CONNECTIONS:
                ServerInfoPacket pkReply = new ServerInfoPacket();
                pkReply.time = reader.readLong();
                pkReply.serverID = serverID;
                pkReply.serverName = this.getManager().getServer().getServer().getName();
                pkReply.encode();
                this.udp.send(pkReply.getData(), packet.getSocketAddress());
                break;
            case RaknetConstants.ID_OPEN_CONNECTION_REQUEST_1:
                reader.read(16); //MAGIC
                reader.readByte(); //RakNet Protocol
                short mtu = (short) ((packet.getLength() - 18) & 0xFFFF);
                ByteArrayOutputStream bos = new ByteArrayOutputStream();
                PEBinaryWriter writer = new PEBinaryWriter(bos);
                writer.writeByte(RaknetConstants.ID_OPEN_CONNECTION_REPLY_1);
                writer.write(RaknetConstants.magic);
                writer.writeLong(NetworkHandler.serverID);
                writer.writeByte((byte) 0x00);
                writer.writeShort(mtu);
                this.udp.send(bos.toByteArray(), packet.getSocketAddress());
                break;
            case RaknetConstants.ID_OPEN_CONNECTION_REQUEST_2:
                reader.read(16);
                reader.readByte();
                reader.readInt();
                reader.readShort();
                short clientMTU = reader.readShort();
                long clientID = reader.readLong();
                ByteArrayOutputStream bos8 = new ByteArrayOutputStream();
                PEBinaryWriter writer8 = new PEBinaryWriter(bos8);
                writer8.writeByte(RaknetConstants.ID_OPEN_CONNECTION_REPLY_2);
                writer8.write(RaknetConstants.magic);
                writer8.writeLong(NetworkHandler.serverID);
                writer8.writeAddress(packet.getAddress(), (short) packet.getPort());
                writer8.writeShort(clientMTU);
                writer8.writeByte((byte) 0x00);
                this.send(bos8.toByteArray(), packet.getSocketAddress());
                PENetworkClient session = new PENetworkClient(this, packet.getSocketAddress(), clientID, clientMTU);
                clients.put(packet.getSocketAddress().toString(), session);
                //this.server.getServer().getSessionRegistry().add(session);
                break;
            case 0x80:
            case 0x81:
            case 0x82:
            case 0x83:
            case 0x84:
            case 0x85:
            case 0x86:
            case 0x87:
            case 0x88:
            case 0x89:
            case 0x8A:
            case 0x8B:
            case 0x8C:
            case 0x8D:
            case 0x8E:
            case 0x8F:
                if (this.manager.getSessions().containsKey(getClientKey(packet.getSocketAddress()))) {
                    RaknetDataPacket dataPacket = new RaknetDataPacket(
                            ArrayUtils.subarray(packet.getData(), 1, packet.getLength()));
                    dataPacket.decode();
                    clients.get(getClientKey(packet.getSocketAddress())).processDataPacket(dataPacket);
                }
                break;
            case 0xC0:
                if (this.manager.getSessions().containsKey(getClientKey(packet.getSocketAddress()))) {
                    clients.get(getClientKey(packet.getSocketAddress()))
                            .processACKPacket(ArrayUtils.subarray(packet.getData(), 1, packet.getData().length));
                }
                break;
            case 0xA0:
                if (this.manager.getSessions().containsKey(getClientKey(packet.getSocketAddress()))) {
                    clients.get(getClientKey(packet.getSocketAddress()))
                            .processNACKPacket(ArrayUtils.subarray(packet.getData(), 1, packet.getData().length));
                }
                break;
            }
        } catch (IOException e) {
        }
    }

    public void send(byte[] buffer, SocketAddress remoteAddr) {
        this.udp.send(buffer, remoteAddr);
    }

    private String getClientKey(SocketAddress addr) {
        return "MCPE-" + addr.toString();
    }
}