2017-08-22 15:02:23 +02:00
package net.ME1312.SubServers.Sync.Network ;
2017-09-24 05:19:22 +02:00
import net.ME1312.SubServers.Sync.Event.SubNetworkConnectEvent ;
2017-08-22 15:02:23 +02:00
import net.ME1312.SubServers.Sync.Event.SubNetworkDisconnectEvent ;
import net.ME1312.SubServers.Sync.Library.Exception.IllegalPacketException ;
import net.ME1312.SubServers.Sync.Library.NamedContainer ;
import net.ME1312.SubServers.Sync.Library.Util ;
import net.ME1312.SubServers.Sync.Library.Version.Version ;
2018-01-05 23:30:01 +01:00
import net.ME1312.SubServers.Sync.Network.Encryption.AES ;
2017-08-22 15:02:23 +02:00
import net.ME1312.SubServers.Sync.Network.Packet.* ;
import net.ME1312.SubServers.Sync.SubPlugin ;
import org.json.JSONException ;
import org.json.JSONObject ;
import java.io.BufferedReader ;
import java.io.IOException ;
import java.io.InputStreamReader ;
import java.io.PrintWriter ;
import java.lang.reflect.InvocationTargetException ;
import java.net.InetAddress ;
import java.net.Socket ;
import java.net.SocketException ;
2017-11-22 22:58:33 +01:00
import java.nio.charset.StandardCharsets ;
2017-08-22 15:02:23 +02:00
import java.util.* ;
import java.util.concurrent.TimeUnit ;
/ * *
* SubData Direct Client Class
* /
public final class SubDataClient {
private static HashMap < Class < ? extends PacketOut > , String > pOut = new HashMap < Class < ? extends PacketOut > , String > ( ) ;
private static HashMap < String , List < PacketIn > > pIn = new HashMap < String , List < PacketIn > > ( ) ;
2018-01-05 21:37:23 +01:00
private static HashMap < String , Cipher > ciphers = new HashMap < String , Cipher > ( ) ;
2017-08-22 15:02:23 +02:00
private static boolean defaults = false ;
private PrintWriter writer ;
2017-09-24 05:19:22 +02:00
private NamedContainer < Boolean , Socket > socket ;
2018-01-05 21:37:23 +01:00
private Cipher cipher ;
2017-08-22 15:02:23 +02:00
private SubPlugin plugin ;
private LinkedList < NamedContainer < String , PacketOut > > queue ;
/ * *
* SubServers Client Instance
*
* @param plugin SubPlugin
* @param address Address
* @param port Port
2018-01-05 21:37:23 +01:00
* @param cipher Cipher
2017-08-22 15:02:23 +02:00
* @throws IOException
* /
2018-01-05 21:37:23 +01:00
public SubDataClient ( SubPlugin plugin , InetAddress address , int port , Cipher cipher ) throws IOException {
2017-08-22 15:02:23 +02:00
if ( Util . isNull ( plugin , address , port ) ) throw new NullPointerException ( ) ;
2017-09-24 05:19:22 +02:00
socket = new NamedContainer < > ( false , new Socket ( address , port ) ) ;
2017-08-22 15:02:23 +02:00
this . plugin = plugin ;
2017-09-24 05:19:22 +02:00
this . writer = new PrintWriter ( socket . get ( ) . getOutputStream ( ) , true ) ;
2018-01-05 21:37:23 +01:00
this . cipher = ( cipher ! = null ) ? cipher : new Cipher ( ) {
@Override
public String getName ( ) {
return " NONE " ;
}
@Override
public byte [ ] encrypt ( String key , JSONObject data ) throws Exception {
return data . toString ( ) . getBytes ( StandardCharsets . UTF_8 ) ;
}
@Override
public JSONObject decrypt ( String key , byte [ ] data ) throws Exception {
return new JSONObject ( new String ( data , StandardCharsets . UTF_8 ) ) ;
}
} ;
2017-08-22 15:02:23 +02:00
this . queue = new LinkedList < NamedContainer < String , PacketOut > > ( ) ;
if ( ! defaults ) loadDefaults ( ) ;
loop ( ) ;
2017-09-24 05:19:22 +02:00
sendPacket ( new NamedContainer < > ( null , new PacketAuthorization ( plugin ) ) ) ;
}
private void init ( ) {
plugin . subdata . sendPacket ( new PacketDownloadLang ( plugin ) ) ;
2018-01-02 23:29:25 +01:00
plugin . subdata . sendPacket ( new PacketDownloadProxyInfo ( proxy - > plugin . subdata . sendPacket ( new PacketDownloadServerList ( null , null , json - > {
2017-12-10 01:30:06 +01:00
if ( plugin . lastReload ! = proxy . getJSONObject ( " subservers " ) . getLong ( " last-reload " ) ) {
System . out . println ( " SubServers > Resetting Server Data " ) ;
plugin . servers . clear ( ) ;
plugin . lastReload = proxy . getJSONObject ( " subservers " ) . getLong ( " last-reload " ) ;
}
2017-09-24 05:19:22 +02:00
for ( String host : json . getJSONObject ( " hosts " ) . keySet ( ) ) {
for ( String subserver : json . getJSONObject ( " hosts " ) . getJSONObject ( host ) . getJSONObject ( " servers " ) . keySet ( ) ) {
2017-12-10 01:30:06 +01:00
plugin . merge ( subserver , json . getJSONObject ( " hosts " ) . getJSONObject ( host ) . getJSONObject ( " servers " ) . getJSONObject ( subserver ) , true ) ;
2017-09-24 05:19:22 +02:00
}
}
for ( String server : json . getJSONObject ( " servers " ) . keySet ( ) ) {
2017-12-10 01:30:06 +01:00
plugin . merge ( server , json . getJSONObject ( " servers " ) . getJSONObject ( server ) , false ) ;
2017-08-22 15:02:23 +02:00
}
2017-12-10 01:30:06 +01:00
} ) ) ) ) ;
2017-09-24 05:19:22 +02:00
while ( queue . size ( ) ! = 0 ) {
sendPacket ( queue . get ( 0 ) ) ;
queue . remove ( 0 ) ;
}
socket . rename ( true ) ;
plugin . getPluginManager ( ) . callEvent ( new SubNetworkConnectEvent ( this ) ) ;
2017-08-22 15:02:23 +02:00
}
2018-01-05 21:37:23 +01:00
static {
addCipher ( " AES " , new AES ( 128 ) ) ;
addCipher ( " AES_128 " , new AES ( 128 ) ) ;
addCipher ( " AES_192 " , new AES ( 192 ) ) ;
addCipher ( " AES_256 " , new AES ( 256 ) ) ;
} private void loadDefaults ( ) {
2017-08-22 15:02:23 +02:00
defaults = true ;
registerPacket ( new PacketAuthorization ( plugin ) , " Authorization " ) ;
registerPacket ( new PacketCommandServer ( ) , " SubCommandServer " ) ;
registerPacket ( new PacketCreateServer ( ) , " SubCreateServer " ) ;
registerPacket ( new PacketDownloadHostInfo ( ) , " SubDownloadHostInfo " ) ;
registerPacket ( new PacketDownloadLang ( plugin ) , " SubDownloadLang " ) ;
registerPacket ( new PacketDownloadNetworkList ( ) , " SubDownloadNetworkList " ) ;
registerPacket ( new PacketDownloadPlayerList ( ) , " SubDownloadPlayerList " ) ;
2017-12-10 01:30:06 +01:00
registerPacket ( new PacketDownloadProxyInfo ( ) , " SubDownloadProxyInfo " ) ;
2017-08-22 15:02:23 +02:00
registerPacket ( new PacketDownloadServerInfo ( ) , " SubDownloadServerInfo " ) ;
registerPacket ( new PacketDownloadServerList ( ) , " SubDownloadServerList " ) ;
registerPacket ( new PacketInRunEvent ( ) , " SubRunEvent " ) ;
registerPacket ( new PacketInReset ( ) , " SubReset " ) ;
registerPacket ( new PacketStartServer ( ) , " SubStartServer " ) ;
registerPacket ( new PacketStopServer ( ) , " SubStopServer " ) ;
registerPacket ( PacketAuthorization . class , " Authorization " ) ;
registerPacket ( PacketCommandServer . class , " SubCommandServer " ) ;
registerPacket ( PacketCreateServer . class , " SubCreateServer " ) ;
registerPacket ( PacketDownloadHostInfo . class , " SubDownloadHostInfo " ) ;
registerPacket ( PacketDownloadLang . class , " SubDownloadLang " ) ;
registerPacket ( PacketDownloadNetworkList . class , " SubDownloadNetworkList " ) ;
registerPacket ( PacketDownloadPlayerList . class , " SubDownloadPlayerList " ) ;
2017-12-10 01:30:06 +01:00
registerPacket ( PacketDownloadProxyInfo . class , " SubDownloadProxyInfo " ) ;
2017-08-22 15:02:23 +02:00
registerPacket ( PacketDownloadServerInfo . class , " SubDownloadServerInfo " ) ;
registerPacket ( PacketDownloadServerList . class , " SubDownloadServerList " ) ;
registerPacket ( PacketStartServer . class , " SubStartServer " ) ;
registerPacket ( PacketStopServer . class , " SubStopServer " ) ;
}
private void loop ( ) {
new Thread ( ( ) - > {
try {
2017-09-24 05:19:22 +02:00
BufferedReader in = new BufferedReader ( new InputStreamReader ( socket . get ( ) . getInputStream ( ) ) ) ;
2017-08-22 15:02:23 +02:00
String input ;
while ( ( input = in . readLine ( ) ) ! = null ) {
try {
2018-01-05 21:37:23 +01:00
JSONObject json = cipher . decrypt ( plugin . config . get ( ) . getSection ( " Settings " ) . getSection ( " SubData " ) . getRawString ( " Password " ) , Base64 . getDecoder ( ) . decode ( input ) ) ;
2017-08-22 15:02:23 +02:00
for ( PacketIn packet : decodePacket ( json ) ) {
try {
packet . execute ( ( json . keySet ( ) . contains ( " c " ) ) ? json . getJSONObject ( " c " ) : null ) ;
} catch ( Throwable e ) {
new InvocationTargetException ( e , " Exception while executing PacketIn " ) . printStackTrace ( ) ;
}
}
} catch ( JSONException e ) {
2018-01-05 21:37:23 +01:00
new IllegalPacketException ( " Unknown Packet Format: " + input ) . printStackTrace ( ) ;
2017-08-22 15:02:23 +02:00
} catch ( IllegalPacketException e ) {
e . printStackTrace ( ) ;
} catch ( Exception e ) {
new InvocationTargetException ( e , " Exception while decoding packet " ) . printStackTrace ( ) ;
}
}
try {
destroy ( plugin . config . get ( ) . getSection ( " Settings " ) . getSection ( " SubData " ) . getInt ( " Reconnect " , 30 ) ) ;
} catch ( IOException e1 ) {
e1 . printStackTrace ( ) ;
}
} catch ( Exception e ) {
if ( ! ( e instanceof SocketException ) ) e . printStackTrace ( ) ;
try {
destroy ( plugin . config . get ( ) . getSection ( " Settings " ) . getSection ( " SubData " ) . getInt ( " Reconnect " , 30 ) ) ;
} catch ( IOException e1 ) {
e1 . printStackTrace ( ) ;
}
}
} ) . start ( ) ;
}
/ * *
* Gets the Server Socket
*
* @return Server Socket
* /
public Socket getClient ( ) {
2017-09-24 05:19:22 +02:00
return socket . get ( ) ;
2017-08-22 15:02:23 +02:00
}
/ * *
2018-01-05 21:37:23 +01:00
* Add a Cipher for use by SubData
*
* @param cipher Cipher to Add
* @param handle Handle to Bind
* /
public static void addCipher ( String handle , Cipher cipher ) {
if ( Util . isNull ( cipher ) ) throw new NullPointerException ( ) ;
if ( ! ciphers . keySet ( ) . contains ( handle . toLowerCase ( ) . replace ( '-' , '_' ) . replace ( ' ' , '_' ) ) )
ciphers . put ( handle . toLowerCase ( ) . replace ( '-' , '_' ) . replace ( ' ' , '_' ) , cipher ) ;
}
/ * *
* Gets the Ciphers
2017-08-22 15:02:23 +02:00
*
2018-01-05 21:37:23 +01:00
* @return Cipher Map
2017-08-22 15:02:23 +02:00
* /
2018-01-05 21:37:23 +01:00
public static Map < String , Cipher > getCiphers ( ) {
return new TreeMap < > ( ciphers ) ;
}
/ * *
* Gets the Client ' s Cipher
*
* @return Cipher
* /
public Cipher getCipher ( ) {
return cipher ;
}
/ * *
* Gets a Cipher by Handle
*
* @param handle Handle
* @return Cipher
* /
public static Cipher getCipher ( String handle ) {
return getCiphers ( ) . get ( handle . toLowerCase ( ) . replace ( '-' , '_' ) . replace ( ' ' , '_' ) ) ;
2017-08-22 15:02:23 +02:00
}
/ * *
* Register PacketIn to the Network
*
* @param packet PacketIn to register
* @param handle Handle to Bind
* /
public static void registerPacket ( PacketIn packet , String handle ) {
if ( Util . isNull ( packet , handle ) ) throw new NullPointerException ( ) ;
2017-11-22 22:58:33 +01:00
List < PacketIn > list = ( pIn . keySet ( ) . contains ( handle . toLowerCase ( ) ) ) ? pIn . get ( handle . toLowerCase ( ) ) : new ArrayList < PacketIn > ( ) ;
2017-08-22 15:02:23 +02:00
if ( ! list . contains ( packet ) ) {
list . add ( packet ) ;
2017-11-22 22:58:33 +01:00
pIn . put ( handle . toLowerCase ( ) , list ) ;
2017-08-22 15:02:23 +02:00
}
}
/ * *
* Unregister PacketIn from the Network
*
* @param packet PacketIn to unregister
* /
public static void unregisterPacket ( PacketIn packet ) {
if ( Util . isNull ( packet ) ) throw new NullPointerException ( ) ;
List < String > search = new ArrayList < String > ( ) ;
search . addAll ( pIn . keySet ( ) ) ;
2017-11-22 22:58:33 +01:00
for ( String handle : search ) if ( pIn . get ( handle . toLowerCase ( ) ) . contains ( packet ) ) {
List < PacketIn > list = pIn . get ( handle . toLowerCase ( ) ) ;
2017-08-22 15:02:23 +02:00
list . remove ( packet ) ;
if ( list . isEmpty ( ) ) {
2017-11-22 22:58:33 +01:00
pIn . remove ( handle . toLowerCase ( ) ) ;
2017-08-22 15:02:23 +02:00
} else {
2017-11-22 22:58:33 +01:00
pIn . put ( handle . toLowerCase ( ) , list ) ;
2017-08-22 15:02:23 +02:00
}
}
}
/ * *
* Register PacketOut to the Network
*
* @param packet PacketOut to register
* @param handle Handle to bind
* /
public static void registerPacket ( Class < ? extends PacketOut > packet , String handle ) {
if ( Util . isNull ( packet , handle ) ) throw new NullPointerException ( ) ;
2017-11-22 22:58:33 +01:00
pOut . put ( packet , handle . toLowerCase ( ) ) ;
2017-08-22 15:02:23 +02:00
}
/ * *
* Unregister PacketOut to the Network
*
* @param packet PacketOut to unregister
* /
public static void unregisterPacket ( Class < ? extends PacketOut > packet ) {
if ( Util . isNull ( packet ) ) throw new NullPointerException ( ) ;
pOut . remove ( packet ) ;
}
/ * *
* Grab PacketIn Instances via handle
*
* @param handle Handle
* @return PacketIn
* /
public static List < ? extends PacketIn > getPacket ( String handle ) {
if ( Util . isNull ( handle ) ) throw new NullPointerException ( ) ;
2017-11-22 22:58:33 +01:00
return new ArrayList < PacketIn > ( pIn . get ( handle . toLowerCase ( ) ) ) ;
2017-08-22 15:02:23 +02:00
}
/ * *
* Send Packet to Server
*
* @param packet Packet to send
* /
public void sendPacket ( PacketOut packet ) {
if ( Util . isNull ( packet ) ) throw new NullPointerException ( ) ;
2017-09-24 05:19:22 +02:00
if ( socket . get ( ) = = null | | ! socket . name ( ) ) {
2017-08-22 15:02:23 +02:00
queue . add ( new NamedContainer < > ( null , packet ) ) ;
} else {
2017-09-24 05:19:22 +02:00
sendPacket ( new NamedContainer < > ( null , packet ) ) ;
}
}
private void sendPacket ( NamedContainer < String , PacketOut > packet ) {
try {
JSONObject json = encodePacket ( packet . get ( ) ) ;
if ( packet . name ( ) ! = null ) json . put ( " f " , packet . name ( ) ) ;
2018-01-05 21:37:23 +01:00
writer . println ( Base64 . getEncoder ( ) . encodeToString ( cipher . encrypt ( plugin . config . get ( ) . getSection ( " Settings " ) . getSection ( " SubData " ) . getRawString ( " Password " ) , json ) ) ) ;
2017-09-24 05:19:22 +02:00
} catch ( Throwable e ) {
e . printStackTrace ( ) ;
2017-08-22 15:02:23 +02:00
}
}
/ * *
* Forward Packet to Client
*
* @param packet Packet to send
* @param location Where to send
* /
public void forwardPacket ( PacketOut packet , String location ) {
if ( Util . isNull ( packet , location ) ) throw new NullPointerException ( ) ;
2017-09-24 05:19:22 +02:00
if ( socket . get ( ) = = null | | ! socket . name ( ) ) {
2017-08-22 15:02:23 +02:00
queue . add ( new NamedContainer < > ( location , packet ) ) ;
} else {
2017-09-24 05:19:22 +02:00
sendPacket ( new NamedContainer < > ( null , packet ) ) ;
2017-08-22 15:02:23 +02:00
}
}
/ * *
* Broadcast packet to all Clients
*
* @param packet Packet to send
* /
public void broadcastPacket ( PacketOut packet ) {
forwardPacket ( packet , " " ) ;
}
/ * *
* JSON Encode PacketOut
*
* @param packet PacketOut
* @return JSON Formatted Packet
* @throws IllegalPacketException
* /
private static JSONObject encodePacket ( PacketOut packet ) throws IllegalPacketException , InvocationTargetException {
JSONObject json = new JSONObject ( ) ;
if ( ! pOut . keySet ( ) . contains ( packet . getClass ( ) ) ) throw new IllegalPacketException ( " Unknown PacketOut Channel: " + packet . getClass ( ) . getCanonicalName ( ) ) ;
if ( packet . getVersion ( ) . toString ( ) = = null ) throw new NullPointerException ( " PacketOut Version cannot be null: " + packet . getClass ( ) . getCanonicalName ( ) ) ;
try {
JSONObject contents = packet . generate ( ) ;
json . put ( " h " , pOut . get ( packet . getClass ( ) ) ) ;
json . put ( " v " , packet . getVersion ( ) . toString ( ) ) ;
if ( contents ! = null ) json . put ( " c " , contents ) ;
return json ;
} catch ( Throwable e ) {
throw new InvocationTargetException ( e , " Exception while encoding packet " ) ;
}
}
/ * *
* JSON Decode PacketIn
*
* @param json JSON to Decode
* @return PacketIn
* @throws IllegalPacketException
* @throws InvocationTargetException
* /
private static List < PacketIn > decodePacket ( JSONObject json ) throws IllegalPacketException , InvocationTargetException {
if ( ! json . keySet ( ) . contains ( " h " ) | | ! json . keySet ( ) . contains ( " v " ) ) throw new IllegalPacketException ( " Unknown Packet Format: " + json . toString ( ) ) ;
if ( ! pIn . keySet ( ) . contains ( json . getString ( " h " ) ) ) throw new IllegalPacketException ( " Unknown PacketIn Channel: " + json . getString ( " h " ) ) ;
List < PacketIn > list = new ArrayList < PacketIn > ( ) ;
for ( PacketIn packet : pIn . get ( json . getString ( " h " ) ) ) {
if ( new Version ( json . getString ( " v " ) ) . equals ( packet . getVersion ( ) ) ) {
list . add ( packet ) ;
} else {
new IllegalPacketException ( " Packet Version Mismatch in " + json . getString ( " h " ) + " : " + json . getString ( " v " ) + " -> " + packet . getVersion ( ) . toString ( ) ) . printStackTrace ( ) ;
}
}
return list ;
}
/ * *
* Drops All Connections and Stops the SubData Listener
*
* @throws IOException
* /
public void destroy ( int reconnect ) throws IOException {
if ( Util . isNull ( reconnect ) ) throw new NullPointerException ( ) ;
2017-09-24 05:19:22 +02:00
if ( socket . get ( ) ! = null ) {
final Socket socket = this . socket . get ( ) ;
this . socket . set ( null ) ;
2017-08-22 15:02:23 +02:00
if ( ! socket . isClosed ( ) ) socket . close ( ) ;
plugin . getPluginManager ( ) . callEvent ( new SubNetworkDisconnectEvent ( ) ) ;
System . out . println ( " SubServers > The SubData Connection was closed " ) ;
if ( reconnect > 0 ) {
System . out . println ( " SubServers > Attempting to reconnect in " + reconnect + " seconds " ) ;
Timer timer = new Timer ( ) ;
timer . scheduleAtFixedRate ( new TimerTask ( ) {
@Override
public void run ( ) {
try {
2018-01-05 21:37:23 +01:00
plugin . subdata = new SubDataClient ( plugin , socket . getInetAddress ( ) , socket . getPort ( ) , cipher ) ;
2017-08-22 15:02:23 +02:00
timer . cancel ( ) ;
while ( queue . size ( ) ! = 0 ) {
if ( queue . get ( 0 ) . name ( ) ! = null ) {
plugin . subdata . forwardPacket ( queue . get ( 0 ) . get ( ) , queue . get ( 0 ) . name ( ) ) ;
} else {
plugin . subdata . sendPacket ( queue . get ( 0 ) . get ( ) ) ;
}
queue . remove ( 0 ) ;
}
} catch ( IOException e ) {
System . out . println ( " SubServers > Connection was unsuccessful, retrying in " + reconnect + " seconds " ) ;
}
}
} , TimeUnit . SECONDS . toMillis ( reconnect ) , TimeUnit . SECONDS . toMillis ( reconnect ) ) ;
}
}
}
}