Examples with NettyInputWrapper com.github.dirtpowered.dirtmv.data.protocol.io.NettyInputWrapper used on opensource projects

Example 6 with NettyInputWrapper

use of com.github.dirtpowered.dirtmv.data.protocol.io.NettyInputWrapper in project DirtMultiversion by DirtPowered.

the class ProtocolRelease51To39 method registerTranslators.

@Override
public void registerTranslators() {
    // login
    addTranslator(0x01, PacketDirection.TO_CLIENT, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            ProtocolStorage storage = session.getStorage();
            if (!storage.hasObject(DimensionTracker.class)) {
                storage.set(DimensionTracker.class, new DimensionTracker());
            }
            DimensionTracker dimensionTracker = session.getStorage().get(DimensionTracker.class);
            dimensionTracker.setDimension(data.read(Type.BYTE, 3));
            return data;
        }
    });
    // respawn
    addTranslator(0x09, PacketDirection.TO_CLIENT, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            DimensionTracker dimensionTracker = session.getStorage().get(DimensionTracker.class);
            dimensionTracker.setDimension(data.read(Type.INT, 0));
            return data;
        }
    });
    // server ping request
    addTranslator(0xFE, PacketDirection.TO_SERVER, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            // empty
            return PacketUtil.createPacket(0xFE, new TypeHolder[0]);
        }
    });
    // kick disconnect
    addTranslator(0xFF, PacketDirection.TO_CLIENT, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            String reason = data.read(Type.STRING, 0);
            // TODO: fix issues with detecting protocol state on post-netty versions
            if (reason.split("\u00a7").length != 3) {
                return data;
            }
            Configuration configuration = session.getMain().getConfiguration();
            // old to new format
            return PacketUtil.createPacket(0xFF, new TypeHolder[] { set(Type.STRING, transformMotd(reason, configuration)) });
        }
    });
    // handshake
    addTranslator(0x02, PacketDirection.TO_SERVER, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            return PacketUtil.createPacket(0x02, new TypeHolder[] { set(Type.BYTE, (byte) 39), data.read(1), data.read(2), data.read(3) });
        }
    });
    // client settings
    addTranslator(0xCC, PacketDirection.TO_SERVER, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            return PacketUtil.createPacket(0xCC, new TypeHolder[] { data.read(0), data.read(1), data.read(2), data.read(3) });
        }
    });
    // map data
    addTranslator(0x83, PacketDirection.TO_CLIENT, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            byte[] mapData = data.read(Type.BYTE_BYTE_ARRAY, 2);
            if (mapData[0] == 1) {
                for (int i = 0; i < (mapData.length - 1) / 3; ++i) {
                    final byte icon = (byte) (mapData[i * 3 + 1] % 16);
                    final byte centerX = mapData[i * 3 + 2];
                    final byte centerZ = mapData[i * 3 + 3];
                    final byte iconRotation = (byte) (mapData[i * 3 + 1] / 16);
                    mapData[i * 3 + 1] = (byte) (icon << 4 | iconRotation & 15);
                    mapData[i * 3 + 2] = centerX;
                    mapData[i * 3 + 3] = centerZ;
                }
            }
            return PacketUtil.createPacket(0x83, new TypeHolder[] { data.read(0), data.read(1), set(Type.UNSIGNED_SHORT_BYTE_ARRAY, mapData) });
        }
    });
    // update time
    addTranslator(0x04, PacketDirection.TO_CLIENT, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            return PacketUtil.createPacket(0x04, new TypeHolder[] { data.read(0), data.read(0) });
        }
    });
    // vehicle spawn
    addTranslator(0x17, PacketDirection.TO_CLIENT, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            return PacketUtil.createPacket(0x17, new TypeHolder[] { data.read(0), data.read(1), data.read(2), data.read(3), data.read(4), // yaw
            set(Type.BYTE, (byte) 0), // pitch
            set(Type.BYTE, (byte) 0), data.read(5) });
        }
    });
    // item spawn
    addTranslator(0x15, PacketDirection.TO_CLIENT, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            PacketData vehicleSpawn = PacketUtil.createPacket(0x17, new TypeHolder[] { data.read(0), set(Type.BYTE, (byte) 2), data.read(4), data.read(5), data.read(6), data.read(7), data.read(8), set(Type.MOTION, new Motion(0, (short) 0, (short) 0, (short) 0)) });
            short itemId = data.read(Type.SHORT, 1);
            byte amount = data.read(Type.BYTE, 2);
            short itemData = data.read(Type.SHORT, 3);
            ItemStack itemStack = new ItemStack(itemId, amount, itemData, CompoundBinaryTag.empty());
            List<WatchableObject> metadata = Collections.singletonList(new WatchableObject(MetadataType.ITEM, 10, itemStack));
            PacketData itemMetadata = PacketUtil.createPacket(0x28, new TypeHolder[] { data.read(0), set(Type.V1_4R_METADATA, metadata.toArray(new WatchableObject[0])) });
            session.sendPacket(vehicleSpawn, PacketDirection.TO_CLIENT, getFrom());
            session.sendPacket(itemMetadata, PacketDirection.TO_CLIENT, getFrom());
            return cancel();
        }
    });
    // mob spawn
    addTranslator(0x18, PacketDirection.TO_CLIENT, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            session.sendPacket(data, PacketDirection.TO_CLIENT, getFrom());
            byte type = data.read(Type.BYTE, 1);
            int itemId = 0;
            if (type == 51)
                // bow
                itemId = 261;
            if (type == 57)
                // golden sword
                itemId = 283;
            if (itemId > 0) {
                ItemStack itemStack = new ItemStack(itemId, 1, 0, null);
                PacketData entityEquipment = PacketUtil.createPacket(0x05, new TypeHolder[] { data.read(0), set(Type.SHORT, (short) 0), set(Type.V1_3R_ITEM, itemStack) });
                session.sendPacket(entityEquipment, PacketDirection.TO_CLIENT, getFrom());
            }
            return cancel();
        }
    });
    // custom payload
    addTranslator(0xFA, PacketDirection.TO_CLIENT, new PacketTranslator() {

        @SneakyThrows
        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            String channel = data.read(Type.STRING, 0);
            byte[] payload = data.read(Type.SHORT_BYTE_ARRAY, 1);
            if (channel.equals("MC|TrList")) {
                // TODO: Custom Payload reader
                NettyInputWrapper buf = new NettyInputWrapper(Unpooled.wrappedBuffer(payload));
                NettyOutputWrapper fixedPayload = new NettyOutputWrapper(Unpooled.buffer());
                fixedPayload.writeInt(buf.readInt());
                short size = buf.readUnsignedByte();
                fixedPayload.writeByte(size);
                for (int i = 0; i < size; i++) {
                    V1_3_1RProtocol.ITEM.write(set(Type.V1_3R_ITEM, V1_3_1RProtocol.ITEM.read(buf)), fixedPayload);
                    V1_3_1RProtocol.ITEM.write(set(Type.V1_3R_ITEM, V1_3_1RProtocol.ITEM.read(buf)), fixedPayload);
                    boolean b = buf.readBoolean();
                    fixedPayload.writeBoolean(b);
                    if (b) {
                        V1_3_1RProtocol.ITEM.write(set(Type.V1_3R_ITEM, V1_3_1RProtocol.ITEM.read(buf)), fixedPayload);
                    }
                    fixedPayload.writeBoolean(false);
                }
                return PacketUtil.createPacket(0xFA, new TypeHolder[] { data.read(0), set(Type.SHORT_BYTE_ARRAY, fixedPayload.array()) });
            }
            return data;
        }
    });
    // level sound
    addTranslator(0x3E, PacketDirection.TO_CLIENT, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            String soundName = data.read(Type.STRING, 0);
            String newSoundName = soundRemapper.getNewSoundName(soundName);
            if (newSoundName.isEmpty())
                return cancel();
            return PacketUtil.createPacket(0x3E, new TypeHolder[] { set(Type.STRING, newSoundName), data.read(1), data.read(2), data.read(3), data.read(4), data.read(5) });
        }
    });
    // door change
    addTranslator(0x3D, PacketDirection.TO_CLIENT, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            return PacketUtil.createPacket(0x3D, new TypeHolder[] { data.read(0), data.read(1), data.read(2), data.read(3), data.read(4), set(Type.BOOLEAN, false) });
        }
    });
    // chunk bulk
    addTranslator(0x38, PacketDirection.TO_CLIENT, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            V1_3_4ChunkBulk oldChunk = data.read(Type.V1_3CHUNK_BULK, 0);
            DimensionTracker tracker = session.getStorage().get(DimensionTracker.class);
            oldChunk.setSkylight(tracker.getDimension() == 0);
            return PacketUtil.createPacket(0x38, new TypeHolder[] { set(Type.V1_4CHUNK_BULK, oldChunk) });
        }
    });
    // block place
    addTranslator(0x0F, PacketDirection.TO_SERVER, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            ItemStack item = data.read(Type.V1_3R_ITEM, 4);
            if (item == null)
                return data;
            int itemId = item.getItemId();
            if (itemId >= 298 && itemId <= 317) {
                return PacketUtil.createPacket(0x66, new TypeHolder[] { set(Type.BYTE, (byte) 0), set(Type.SHORT, (short) 0), set(Type.BYTE, (byte) 0), set(Type.SHORT, (short) 0), set(Type.BYTE, (byte) 0), // fake item
                set(Type.V1_3R_ITEM, new ItemStack(256, 0, 0, null)) });
            }
            return data;
        }
    });
    // creative item get
    addTranslator(0x6B, ProtocolState.PLAY, PacketDirection.TO_SERVER, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            ItemStack item = data.read(Type.V1_3R_ITEM, 1);
            boolean notNull = item != null;
            if (notNull && !creativeTab.exists(item.getItemId())) {
                // replace all unknown items to stone
                item.setItemId(1);
                item.setData(0);
            }
            return PacketUtil.createPacket(0x6B, new TypeHolder[] { data.read(0), set(Type.V1_3R_ITEM, item) });
        }
    });
    // spawn painting
    addTranslator(0x19, PacketDirection.TO_CLIENT, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            int direction = data.read(Type.INT, 5);
            int correctedDirection = 0;
            switch(direction) {
                case 0:
                    correctedDirection = 2;
                    break;
                case 1:
                    correctedDirection = 1;
                    break;
                case 2:
                    break;
                case 3:
                    correctedDirection = 3;
                    break;
            }
            return PacketUtil.createPacket(0x19, new TypeHolder[] { data.read(0), data.read(1), data.read(2), data.read(3), data.read(4), set(Type.INT, correctedDirection) });
        }
    });
    // player look move
    addTranslator(0x0D, PacketDirection.TO_SERVER, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            ProtocolStorage storage = session.getStorage();
            if (!storage.hasObject(BlockStorage.class)) {
                return data;
            }
            double x = data.read(Type.DOUBLE, 0);
            double y = data.read(Type.DOUBLE, 1);
            double originalStance = data.read(Type.DOUBLE, 2);
            double z = data.read(Type.DOUBLE, 3);
            MovementTranslator.updateBoundingBox(session, new Location(x, y, z));
            Location loc = MovementTranslator.correctPosition(session, x, y, z);
            return PacketUtil.createPacket(0x0D, new TypeHolder[] { set(Type.DOUBLE, loc.getX()), set(Type.DOUBLE, loc.getY()), // stance
            set(Type.DOUBLE, loc.getY() + (originalStance - y)), set(Type.DOUBLE, loc.getZ()), data.read(4), data.read(5), data.read(6) });
        }
    });
    // player position
    addTranslator(0x0B, PacketDirection.TO_SERVER, new PacketTranslator() {

        @Override
        public PacketData translate(ServerSession session, PacketData data) {
            ProtocolStorage storage = session.getStorage();
            if (!storage.hasObject(BlockStorage.class)) {
                return data;
            }
            double x = data.read(Type.DOUBLE, 0);
            double y = data.read(Type.DOUBLE, 1);
            double z = data.read(Type.DOUBLE, 3);
            MovementTranslator.updateBoundingBox(session, new Location(x, y, z));
            Location loc = MovementTranslator.correctPosition(session, x, y, z);
            return PacketUtil.createPacket(0x0B, new TypeHolder[] { set(Type.DOUBLE, loc.getX()), set(Type.DOUBLE, loc.getY()), set(Type.DOUBLE, loc.getY() + 1.6200000047683716D), set(Type.DOUBLE, loc.getZ()), data.read(4) });
        }
    });
}