89 src/main/java/co/ryred/spongepls/SpongePlsPlugin.java
View file
@@ -36,25 +36,38 @@

package co.ryred.spongepls;

import de.inventivegames.packetlistener.handler.PacketHandler;
import co.ryred.spongepls.netty.BetterBossHandler;
import javassist.*;
import lombok.AccessLevel;
import lombok.Getter;
import lombok.Setter;
import net.md_5.bungee.BungeeCord;
import net.md_5.bungee.api.plugin.Plugin;
import net.md_5.bungee.config.Configuration;

import java.io.IOException;
import java.lang.reflect.Field;
import java.util.regex.Pattern;

/**
* @author Cory Redmond
* Created by acech_000 on 12/11/2015.
*/
public class SpongePlsPlugin extends Plugin
{

@Getter
@Setter(AccessLevel.PRIVATE)
public static SpongePlsPlugin __INSTANCE__ = null;

@Getter( AccessLevel.PACKAGE )
private SpongePlsConfig configuration;
private Class bossHandlerClass;

@Override
public void onLoad()
{
set__INSTANCE__( this );
this.configuration = new SpongePlsConfig( this );
}

@@ -64,7 +77,62 @@ public void onEnable()

getProxy().getPluginManager().registerCommand( this, new SpongePlsCommand( this ) );

PacketHandler.addHandler( new SpongePacketHandler( this ) );
try {
changeChild();
} catch ( Exception e ) {
e.printStackTrace();
}

}

private void changeChild() throws NoSuchFieldException, IllegalAccessException, NotFoundException, CannotCompileException, InstantiationException, IOException, ClassNotFoundException
{

//co.ryred.spongepls.SpongePlsPlugin.initChannel( ch );

ClassPool pool = ClassPool.getDefault();
pool.insertClassPath( new ClassClassPath( BetterBossHandler.class ) );
CtClass cc = pool.get( "net.md_5.bungee.netty.HandlerBoss" );

CtField field = new CtField( pool.get( "java.lang.Object" ), "invokableObj", cc );
field.setModifiers( Modifier.setPublic( Modifier.STATIC ) );
cc.addField( field );

CtMethod initMethod = cc.getDeclaredMethod( "setHandler", new CtClass[]{ pool.get( "net.md_5.bungee.netty.PacketHandler" ) } );
initMethod.insertAfter( "{"
+ "try {"
+ "if (invokableObj != null) {"
+ "java.lang.reflect.Method m = invokableObj.getClass().getDeclaredMethod(\"invokePls\", new java.lang.Class[] {net.md_5.bungee.netty.PacketHandler.class});"
+ "if (m != null) {"
+ "this.handler = (net.md_5.bungee.netty.PacketHandler) m.invoke(null, new Object[] { handler });"
+ "}"
+ "}"
+ "} catch (Exception ex) {"
+ "ex.printStackTrace();"
+ "this.handler = handler;"
+ "}"
+ "};" );

CustomClassLoader cl = new CustomClassLoader( BungeeCord.class.getClassLoader(), cc.toBytecode() );
this.bossHandlerClass = cl.loadClass( "net.md_5.bungee.netty.HandlerBoss" );

Field invokable = bossHandlerClass.getDeclaredField( "invokableObj" );
invokable.setAccessible( true );
invokable.set( null, new BetterBossHandler() );

}

@Override
public void onDisable()
{
set__INSTANCE__( null );

try {
Field invokable = bossHandlerClass.getDeclaredField( "invokableObj" );
invokable.setAccessible( true );
invokable.set( null, null );
} catch ( Exception ex ) {}

}

public Configuration getConfig() {
@@ -77,4 +145,21 @@ public void saveConfig() {
getConfiguration().saveConfig();
}

/**
* http://blog.janjonas.net/2012-03-06/java-test-string-match-wildcard-expression
*
* @param text Text to test
* @return True if the text matches the wildcard pattern
*/
public boolean isAllowed( String text )
{

for ( String pattern : getConfig().getStringList( "allowed-servers" ) ) {
if ( Pattern.compile( pattern, Pattern.CASE_INSENSITIVE ).matcher( text ).find() ) return true;
}

return false;

}

}
86 src/main/java/co/ryred/spongepls/netty/BetterBossHandler.java
View file
@@ -0,0 +1,86 @@
/*
* http://ryred.co/
* ace[at]ac3-servers.eu
*
* =================================================================
*
* Copyright (c) 2015, Cory Redmond
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* Redistributions of source code must retain the above copyright notice, this
* list of conditions and the following disclaimer.
*
* Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* Neither the name of SpongePls nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
*/

package co.ryred.spongepls.netty;

import co.ryred.spongepls.SpongePlsPlugin;
import net.md_5.bungee.BungeeCord;
import net.md_5.bungee.BungeeServerInfo;
import net.md_5.bungee.ServerConnector;
import net.md_5.bungee.UserConnection;
import net.md_5.bungee.api.ProxyServer;
import net.md_5.bungee.netty.PacketHandler;

/**
* @author Cory Redmond
* Created by acech_000 on 13/11/2015.
*/
public class BetterBossHandler
{

public static PacketHandler invokePls( PacketHandler handler )
{

if ( BungeeCord.getInstance().getConfig().isIpForward() && handler instanceof ServerConnector ) {

ServerConnector sc = ( (ServerConnector) handler );
try {
if ( SpongePlsPlugin.get__INSTANCE__().isAllowed( WrappedServerConnector.getTarget( sc ).getName() ) ) {

ProxyServer bungee = WrappedServerConnector.getBungee( sc );
UserConnection uc = WrappedServerConnector.getUserConnection( sc );
BungeeServerInfo si = WrappedServerConnector.getTarget( sc );

WrappedServerConnector wsc = new WrappedServerConnector( bungee, uc, si );

SpongePlsPlugin.get__INSTANCE__().getLogger().info( "Using the Forge Server Connector for " + uc.getName() );
SpongePlsPlugin.get__INSTANCE__().getLogger().info( " for the target \"" + si.getName() + "\"." );

return wsc;

}
} catch ( Exception e ) {
e.printStackTrace();
return handler;
}

}

return handler;

}

}
147 src/main/java/co/ryred/spongepls/netty/WrappedServerConnector.java
View file
@@ -0,0 +1,147 @@
/*
* http://ryred.co/
* ace[at]ac3-servers.eu
*
* =================================================================
*
* Copyright (c) 2015, Cory Redmond
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* Redistributions of source code must retain the above copyright notice, this
* list of conditions and the following disclaimer.
*
* Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* Neither the name of SpongePls nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
*/

package co.ryred.spongepls.netty;

import net.md_5.bungee.BungeeCord;
import net.md_5.bungee.BungeeServerInfo;
import net.md_5.bungee.ServerConnector;
import net.md_5.bungee.UserConnection;
import net.md_5.bungee.api.ProxyServer;
import net.md_5.bungee.connection.LoginResult;
import net.md_5.bungee.forge.ForgeServerHandler;
import net.md_5.bungee.netty.ChannelWrapper;
import net.md_5.bungee.protocol.Protocol;
import net.md_5.bungee.protocol.packet.Handshake;

import java.lang.reflect.Field;

/**
* @author Cory Redmond
* Created by acech_000 on 13/11/2015.
*/
public class WrappedServerConnector extends ServerConnector
{
public WrappedServerConnector( ProxyServer bungee, UserConnection user, BungeeServerInfo target )
{
super( bungee, user, target );
}

public static UserConnection getUserConnection( ServerConnector sc ) throws NoSuchFieldException, IllegalAccessException
{

Field field = ServerConnector.class.getDeclaredField( "user" );
field.setAccessible( true );
return ( (UserConnection) field.get( sc ) );

}

public static BungeeServerInfo getTarget( ServerConnector sc ) throws NoSuchFieldException, IllegalAccessException
{

Field field = ServerConnector.class.getDeclaredField( "target" );
field.setAccessible( true );
return ( (BungeeServerInfo) field.get( sc ) );

}

public static ProxyServer getBungee( ServerConnector sc ) throws NoSuchFieldException, IllegalAccessException
{

Field field = ServerConnector.class.getDeclaredField( "bungee" );
field.setAccessible( true );
return ( (ProxyServer) field.get( sc ) );

}

public static void setHandler( ServerConnector sc, ForgeServerHandler handler ) throws NoSuchFieldException, IllegalAccessException
{

Field field = ServerConnector.class.getDeclaredField( "handshakeHandler" );
field.setAccessible( true );
field.set( sc, handler );

}

public static void setChannel( ServerConnector sc, ChannelWrapper ch ) throws NoSuchFieldException, IllegalAccessException
{

Field field = ServerConnector.class.getDeclaredField( "ch" );
field.setAccessible( true );
field.set( sc, ch );

}

@Override
public void connected( ChannelWrapper ch ) throws Exception
{

setChannel( this, ch );

UserConnection user = getUserConnection( this );
BungeeServerInfo target = getTarget( this );

setHandler( this, new ForgeServerHandler( user, ch, target ) );
Handshake originalHandshake = user.getPendingConnection().getHandshake();
Handshake copiedHandshake = new Handshake( originalHandshake.getProtocolVersion(), originalHandshake.getHost(), originalHandshake.getPort(), 2 );

String newHost = copiedHandshake.getHost() + "\00" + user.getAddress().getHostString() + "\00" + user.getUUID();

LoginResult profile = user.getPendingConnection().getLoginProfile();
if ( profile != null && profile.getProperties() != null && profile.getProperties().length > 0 ) {
newHost += "\00" + BungeeCord.getInstance().gson.toJson( profile.getProperties() );
}

// In order to be able to support Forge servers that use IP forwarding, we change the original host string from:
// <host><extradata>
// to
// <host><BungeeData>\00|<extradata>
// Implementations can then simply split on the string "\00|" and take the first chunk of the string and
// do their normal processing, and may ignore the extra data if they so wish.
if ( !user.getExtraDataInHandshake().isEmpty() ) {
newHost += "\00|" + user.getExtraDataInHandshake();
}

copiedHandshake.setHost( newHost );

ch.write( copiedHandshake );

ch.setProtocol( Protocol.LOGIN );
ch.write( user.getPendingConnection().getLoginRequest() );

}

}