68 lines
2.8 KiB
Java
68 lines
2.8 KiB
Java
/*
|
|
* This file is part of ViaVersion - https://github.com/ViaVersion/ViaVersion
|
|
* Copyright (C) 2016-2021 ViaVersion and contributors
|
|
*
|
|
* 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 com.viaversion.viaversion.velocity.handlers;
|
|
|
|
import com.viaversion.viaversion.api.connection.UserConnection;
|
|
import com.viaversion.viaversion.connection.UserConnectionImpl;
|
|
import com.viaversion.viaversion.protocol.ProtocolPipelineImpl;
|
|
import io.netty.channel.Channel;
|
|
import io.netty.channel.ChannelInitializer;
|
|
|
|
import java.lang.reflect.Method;
|
|
|
|
public class VelocityChannelInitializer extends ChannelInitializer<Channel> {
|
|
public static final String MINECRAFT_ENCODER = "minecraft-encoder";
|
|
public static final String MINECRAFT_DECODER = "minecraft-decoder";
|
|
public static final String VIA_ENCODER = "via-encoder";
|
|
public static final String VIA_DECODER = "via-decoder";
|
|
public static final Object COMPRESSION_ENABLED_EVENT;
|
|
private static final Method INIT_CHANNEL;
|
|
|
|
private final ChannelInitializer<?> original;
|
|
private final boolean clientSide;
|
|
|
|
static {
|
|
try {
|
|
INIT_CHANNEL = ChannelInitializer.class.getDeclaredMethod("initChannel", Channel.class);
|
|
INIT_CHANNEL.setAccessible(true);
|
|
|
|
Class<?> eventClass = Class.forName("com.velocitypowered.proxy.protocol.VelocityConnectionEvent");
|
|
COMPRESSION_ENABLED_EVENT = eventClass.getDeclaredField("COMPRESSION_ENABLED").get(null);
|
|
} catch (ReflectiveOperationException e) {
|
|
throw new RuntimeException(e);
|
|
}
|
|
}
|
|
|
|
public VelocityChannelInitializer(ChannelInitializer<?> original, boolean clientSide) {
|
|
this.original = original;
|
|
this.clientSide = clientSide;
|
|
}
|
|
|
|
@Override
|
|
protected void initChannel(Channel channel) throws Exception {
|
|
INIT_CHANNEL.invoke(original, channel);
|
|
|
|
UserConnection user = new UserConnectionImpl(channel, clientSide);
|
|
new ProtocolPipelineImpl(user);
|
|
|
|
// We need to add a separated handler because Velocity uses pipeline().get(MINECRAFT_DECODER)
|
|
channel.pipeline().addBefore(MINECRAFT_ENCODER, VIA_ENCODER, new VelocityEncodeHandler(user));
|
|
channel.pipeline().addBefore(MINECRAFT_DECODER, VIA_DECODER, new VelocityDecodeHandler(user));
|
|
}
|
|
}
|