460 lines
19 KiB
Java
460 lines
19 KiB
Java
package net.ME1312.SubServers.Client.Sponge.Network;
|
|
|
|
import net.ME1312.SubServers.Client.Sponge.Event.SubNetworkConnectEvent;
|
|
import net.ME1312.SubServers.Client.Sponge.Event.SubNetworkDisconnectEvent;
|
|
import net.ME1312.SubServers.Client.Sponge.Library.Config.YAMLSection;
|
|
import net.ME1312.SubServers.Client.Sponge.Library.Exception.IllegalPacketException;
|
|
import net.ME1312.SubServers.Client.Sponge.Library.NamedContainer;
|
|
import net.ME1312.SubServers.Client.Sponge.Library.Util;
|
|
import net.ME1312.SubServers.Client.Sponge.Library.Version.Version;
|
|
import net.ME1312.SubServers.Client.Sponge.Network.Encryption.AES;
|
|
import net.ME1312.SubServers.Client.Sponge.Network.Packet.*;
|
|
import net.ME1312.SubServers.Client.Sponge.SubPlugin;
|
|
import org.msgpack.core.MessageInsufficientBufferException;
|
|
import org.msgpack.core.MessagePack;
|
|
import org.msgpack.core.MessagePacker;
|
|
import org.msgpack.core.MessageUnpacker;
|
|
import org.msgpack.value.Value;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.spongepowered.api.Sponge;
|
|
import org.yaml.snakeyaml.error.YAMLException;
|
|
|
|
import java.io.*;
|
|
import java.lang.reflect.InvocationTargetException;
|
|
import java.net.InetAddress;
|
|
import java.net.Socket;
|
|
import java.net.SocketException;
|
|
import java.util.*;
|
|
import java.util.concurrent.TimeUnit;
|
|
|
|
/**
|
|
* SubData Direct Client Class
|
|
*/
|
|
public final class SubDataClient {
|
|
private static HashMap<Class<? extends PacketOut>, NamedContainer<String, String>> pOut = new HashMap<Class<? extends PacketOut>, NamedContainer<String, String>>();
|
|
private static HashMap<String, HashMap<String, List<PacketIn>>> pIn = new HashMap<String, HashMap<String, List<PacketIn>>>();
|
|
private static HashMap<String, Cipher> ciphers = new HashMap<String, Cipher>();
|
|
private static boolean defaults = false;
|
|
protected static Logger log = LoggerFactory.getLogger("SubData");
|
|
private MessagePacker out;
|
|
private NamedContainer<Boolean, Socket> socket;
|
|
private String name;
|
|
private Cipher cipher;
|
|
private String password;
|
|
private SubPlugin plugin;
|
|
private LinkedList<NamedContainer<String, PacketOut>> queue;
|
|
|
|
/**
|
|
* SubServers Client Instance
|
|
*
|
|
* @param plugin SubPlugin
|
|
* @param name Server Name
|
|
* @param address Address
|
|
* @param port Port
|
|
* @param cipher Cipher
|
|
* @throws IOException
|
|
*/
|
|
public SubDataClient(SubPlugin plugin, String name, InetAddress address, int port, Cipher cipher) throws IOException {
|
|
if (Util.isNull(plugin, address, port)) throw new NullPointerException();
|
|
socket = new NamedContainer<>(false, new Socket(address, port));
|
|
this.plugin = plugin;
|
|
this.name = (name == null || name.length() > 0)?name:null;
|
|
this.out = MessagePack.newDefaultPacker(socket.get().getOutputStream());
|
|
this.queue = new LinkedList<NamedContainer<String, PacketOut>>();
|
|
this.password = plugin.config.get().getSection("Settings").getSection("SubData").getRawString("Password");
|
|
this.cipher = (cipher != null)?cipher:new Cipher() {
|
|
@Override
|
|
public String getName() {
|
|
return "NONE";
|
|
}
|
|
@Override
|
|
public Value encrypt(String key, YAMLSection data) {
|
|
return data.msgPack();
|
|
}
|
|
@Override
|
|
@SuppressWarnings("unchecked")
|
|
public YAMLSection decrypt(String key, Value data) {
|
|
return new YAMLSection(data.asMapValue());
|
|
}
|
|
};
|
|
|
|
if (!defaults) loadDefaults();
|
|
loop();
|
|
|
|
sendPacket(new NamedContainer<>(null, new PacketAuthorization(plugin, password)));
|
|
}
|
|
|
|
private void init() {
|
|
sendPacket(new PacketDownloadLang());
|
|
while (queue.size() != 0) {
|
|
sendPacket(queue.get(0));
|
|
queue.remove(0);
|
|
}
|
|
socket.rename(true);
|
|
Sponge.getEventManager().post(new SubNetworkConnectEvent(this));
|
|
}
|
|
|
|
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() {
|
|
defaults = true;
|
|
log = LoggerFactory.getLogger("SubData");
|
|
|
|
registerPacket(new PacketAuthorization(plugin, null), "SubData", "Authorization");
|
|
registerPacket(new PacketCommandServer(), "SubServers", "CommandServer");
|
|
registerPacket(new PacketCreateServer(), "SubServers", "CreateServer");
|
|
registerPacket(new PacketDownloadGroupInfo(), "SubServers", "DownloadGroupInfo");
|
|
registerPacket(new PacketDownloadHostInfo(), "SubServers", "DownloadHostInfo");
|
|
registerPacket(new PacketDownloadLang(plugin), "SubServers", "DownloadLang");
|
|
registerPacket(new PacketDownloadNetworkList(), "SubServers", "DownloadNetworkList");
|
|
registerPacket(new PacketDownloadPlatformInfo(), "SubServers", "DownloadPlatformInfo");
|
|
registerPacket(new PacketDownloadPlayerList(), "SubServers", "DownloadPlayerList");
|
|
registerPacket(new PacketDownloadProxyInfo(), "SubServers", "DownloadProxyInfo");
|
|
registerPacket(new PacketDownloadServerInfo(), "SubServers", "DownloadServerInfo");
|
|
registerPacket(new PacketInRunEvent(plugin), "SubServers", "RunEvent");
|
|
registerPacket(new PacketInReload(plugin), "SubServers", "Reload");
|
|
registerPacket(new PacketInReset(), "SubServers", "Reset");
|
|
registerPacket(new PacketLinkServer(plugin), "SubServers", "LinkServer");
|
|
registerPacket(new PacketRestartServer(), "SubServers", "RestartServer");
|
|
registerPacket(new PacketStartServer(), "SubServers", "StartServer");
|
|
registerPacket(new PacketStopServer(), "SubServers", "StopServer");
|
|
|
|
registerPacket(PacketAuthorization.class, "SubData", "Authorization");
|
|
registerPacket(PacketCommandServer.class, "SubServers", "CommandServer");
|
|
registerPacket(PacketCreateServer.class, "SubServers", "CreateServer");
|
|
registerPacket(PacketDownloadGroupInfo.class, "SubServers", "DownloadGroupInfo");
|
|
registerPacket(PacketDownloadHostInfo.class, "SubServers", "DownloadHostInfo");
|
|
registerPacket(PacketDownloadLang.class, "SubServers", "DownloadLang");
|
|
registerPacket(PacketDownloadNetworkList.class, "SubServers", "DownloadNetworkList");
|
|
registerPacket(PacketDownloadPlatformInfo.class, "SubServers", "DownloadPlatformInfo");
|
|
registerPacket(PacketDownloadPlayerList.class, "SubServers", "DownloadPlayerList");
|
|
registerPacket(PacketDownloadProxyInfo.class, "SubServers", "DownloadProxyInfo");
|
|
registerPacket(PacketDownloadServerInfo.class, "SubServers", "DownloadServerInfo");
|
|
registerPacket(PacketLinkServer.class, "SubServers", "LinkServer");
|
|
registerPacket(PacketRestartServer.class, "SubServers", "RestartServer");
|
|
registerPacket(PacketStartServer.class, "SubServers", "StartServer");
|
|
registerPacket(PacketStopServer.class, "SubServers", "StopServer");
|
|
}
|
|
|
|
private void loop() {
|
|
Sponge.getScheduler().createTaskBuilder().async().execute(() -> {
|
|
try {
|
|
MessageUnpacker in = MessagePack.newDefaultUnpacker(socket.get().getInputStream());
|
|
Value input;
|
|
while ((input = in.unpackValue()) != null) {
|
|
recieve(input);
|
|
}
|
|
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 instanceof MessageInsufficientBufferException)) e.printStackTrace();
|
|
try {
|
|
destroy(plugin.config.get().getSection("Settings").getSection("SubData").getInt("Reconnect", 30));
|
|
} catch (IOException e1) {
|
|
e1.printStackTrace();
|
|
}
|
|
}
|
|
}).submit(plugin);
|
|
}
|
|
|
|
private void recieve(Value input) {
|
|
try {
|
|
YAMLSection data = getCipher().decrypt(password, input);
|
|
for (PacketIn packet : decodePacket(data)) {
|
|
Sponge.getScheduler().createTaskBuilder().execute(() -> {
|
|
try {
|
|
packet.execute((data.contains("c")) ? data.getSection("c") : null);
|
|
} catch (Throwable e) {
|
|
new InvocationTargetException(e, "Exception while executing PacketIn").printStackTrace();
|
|
}
|
|
}).submit(plugin);
|
|
}
|
|
} catch (YAMLException e) {
|
|
new IllegalPacketException("Unknown Packet Format: " + input).printStackTrace();
|
|
} catch (IllegalPacketException e) {
|
|
e.printStackTrace();
|
|
} catch (Exception e) {
|
|
new InvocationTargetException(e, "Exception while decoding packet").printStackTrace();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Gets the Assigned Server Name
|
|
*
|
|
* @return Server Name
|
|
*/
|
|
public String getName() {
|
|
return name;
|
|
}
|
|
|
|
/**
|
|
* Gets the Client Socket
|
|
*
|
|
* @return Client Socket
|
|
*/
|
|
public Socket getClient() {
|
|
return socket.get();
|
|
}
|
|
|
|
/**
|
|
* 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.toUpperCase().replace('-', '_').replace(' ', '_'))) throw new IllegalStateException("Cipher already exists: " + handle);
|
|
ciphers.put(handle.toUpperCase().replace('-', '_').replace(' ', '_'), cipher);
|
|
}
|
|
|
|
/**
|
|
* Gets the Ciphers
|
|
*
|
|
* @return Cipher Map
|
|
*/
|
|
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.toUpperCase().replace('-', '_').replace(' ', '_'));
|
|
}
|
|
|
|
/**
|
|
* Register PacketIn to the Network
|
|
*
|
|
* @param packet PacketIn to register
|
|
* @param channel Packet Channel
|
|
* @param handle Handle to Bind
|
|
*/
|
|
public static void registerPacket(PacketIn packet, String channel, String handle) {
|
|
if (Util.isNull(packet, channel, handle)) throw new NullPointerException();
|
|
HashMap<String, List<PacketIn>> map = (pIn.keySet().contains(channel.toLowerCase()))?pIn.get(channel.toLowerCase()):new HashMap<String, List<PacketIn>>();
|
|
List<PacketIn> list = (map.keySet().contains(handle))?map.get(handle):new ArrayList<PacketIn>();
|
|
if (!list.contains(packet)) {
|
|
list.add(packet);
|
|
map.put(handle, list);
|
|
pIn.put(channel.toLowerCase(), map);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Unregister PacketIn from the Network
|
|
*
|
|
* @param channel Packet Channel
|
|
* @param packet PacketIn to unregister
|
|
*/
|
|
public static void unregisterPacket(String channel, PacketIn packet) {
|
|
if (Util.isNull(channel, packet)) throw new NullPointerException();
|
|
if (pIn.keySet().contains(channel.toLowerCase())) {
|
|
List<String> search = new ArrayList<String>();
|
|
search.addAll(pIn.get(channel.toLowerCase()).keySet());
|
|
for (String handle : search) if (pIn.get(channel.toLowerCase()).get(handle).contains(packet)) {
|
|
List<PacketIn> list = pIn.get(channel.toLowerCase()).get(handle);
|
|
list.remove(packet);
|
|
if (list.isEmpty()) {
|
|
pIn.get(channel.toLowerCase()).remove(handle);
|
|
if (pIn.get(channel.toLowerCase()).isEmpty()) pIn.remove(channel.toLowerCase());
|
|
} else {
|
|
pIn.get(channel.toLowerCase()).put(handle, list);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Register PacketOut to the Network
|
|
*
|
|
* @param packet PacketOut to register
|
|
* @param channel Packet Channel
|
|
* @param handle Handle to bind
|
|
*/
|
|
public static void registerPacket(Class<? extends PacketOut> packet, String channel, String handle) {
|
|
if (Util.isNull(packet, channel, handle)) throw new NullPointerException();
|
|
pOut.put(packet, new NamedContainer<String, String>(channel.toLowerCase(), handle));
|
|
}
|
|
|
|
/**
|
|
* Unregister PacketOut to the Network
|
|
*
|
|
* @param channel Packet Channel
|
|
* @param packet PacketOut to unregister
|
|
*/
|
|
public static void unregisterPacket(String channel, Class<? extends PacketOut> packet) {
|
|
if (Util.isNull(channel, packet)) throw new NullPointerException();
|
|
if (pOut.keySet().contains(packet) && pOut.get(packet).name().equalsIgnoreCase(channel)) pOut.remove(packet);
|
|
}
|
|
|
|
/**
|
|
* Grab PacketIn Instances via handle
|
|
*
|
|
* @param channel Packet Channel
|
|
* @param handle Handle
|
|
* @return PacketIn
|
|
*/
|
|
public static List<? extends PacketIn> getPacket(String channel, String handle) {
|
|
if (Util.isNull(channel, handle)) throw new NullPointerException();
|
|
return new ArrayList<PacketIn>(pIn.get(channel.toLowerCase()).get(handle));
|
|
}
|
|
|
|
/**
|
|
* Send Packet to Server
|
|
*
|
|
* @param packet Packet to send
|
|
*/
|
|
public void sendPacket(PacketOut packet) {
|
|
if (Util.isNull(packet)) throw new NullPointerException();
|
|
if (socket == null || !socket.name()) {
|
|
queue.add(new NamedContainer<>(null, packet));
|
|
} else {
|
|
sendPacket(new NamedContainer<>(null, packet));
|
|
}
|
|
}
|
|
|
|
private void sendPacket(NamedContainer<String, PacketOut> packet) {
|
|
try {
|
|
YAMLSection data = encodePacket(packet.get());
|
|
if (packet.name() != null) data.set("f", packet.name());
|
|
out.packValue(getCipher().encrypt(password, data));
|
|
out.flush();
|
|
} catch (Throwable e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 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();
|
|
if (socket.get() == null || !socket.name()) {
|
|
queue.add(new NamedContainer<>(location, packet));
|
|
} else {
|
|
sendPacket(new NamedContainer<>(location, packet));
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Broadcast packet to all Clients
|
|
*
|
|
* @param packet Packet to send
|
|
*/
|
|
public void broadcastPacket(PacketOut packet) {
|
|
forwardPacket(packet, "");
|
|
}
|
|
|
|
/**
|
|
* Encode PacketOut
|
|
*
|
|
* @param packet PacketOut
|
|
* @return JSON Formatted Packet
|
|
* @throws IllegalPacketException
|
|
*/
|
|
private static YAMLSection encodePacket(PacketOut packet) throws IllegalPacketException, InvocationTargetException {
|
|
YAMLSection data = new YAMLSection();
|
|
|
|
if (!pOut.keySet().contains(packet.getClass())) throw new IllegalPacketException("Unknown PacketOut Channel: " + packet.getClass().getCanonicalName());
|
|
if (packet.getVersion() == null) throw new NullPointerException("PacketOut Version cannot be null: " + packet.getClass().getCanonicalName());
|
|
|
|
try {
|
|
YAMLSection contents = packet.generate();
|
|
data.set("n", pOut.get(packet.getClass()).name());
|
|
data.set("h", pOut.get(packet.getClass()).get());
|
|
data.set("v", packet.getVersion());
|
|
if (contents != null) data.set("c", contents);
|
|
return data;
|
|
} catch (Throwable e) {
|
|
throw new InvocationTargetException(e, "Exception while encoding packet");
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Decode PacketIn
|
|
*
|
|
* @param data Data to Decode
|
|
* @return PacketIn
|
|
* @throws IllegalPacketException
|
|
* @throws InvocationTargetException
|
|
*/
|
|
private static List<PacketIn> decodePacket(YAMLSection data) throws IllegalPacketException, InvocationTargetException {
|
|
if (!data.contains("n") || !data.contains("h") || !data.contains("v")) throw new IllegalPacketException("Unknown Packet Format: " + data.toString());
|
|
if (!pIn.keySet().contains(data.getRawString("n")) || !pIn.get(data.getRawString("n")).keySet().contains(data.getRawString("h"))) throw new IllegalPacketException("Unknown PacketIn Channel: " + data.getRawString("n") + ':' + data.getRawString("h"));
|
|
|
|
List<PacketIn> list = new ArrayList<PacketIn>();
|
|
for (PacketIn packet : pIn.get(data.getRawString("n")).get(data.getRawString("h"))) {
|
|
if (packet.isCompatible(data.getVersion("v"))) {
|
|
list.add(packet);
|
|
} else {
|
|
new IllegalPacketException("Packet Version Mismatch in " + data.getRawString("h") + ": " + data.getRawString("v") + " -> " + packet.getVersion()).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();
|
|
if (socket.get() != null) {
|
|
final Socket socket = this.socket.get();
|
|
this.socket.set(null);
|
|
if (!socket.isClosed()) socket.close();
|
|
Sponge.getEventManager().post(new SubNetworkDisconnectEvent());
|
|
log.info("The SubData Connection was closed");
|
|
if (reconnect > 0) {
|
|
|
|
log.info("SubServers > Attempting to reconnect in " + reconnect + " seconds");
|
|
Sponge.getScheduler().createTaskBuilder().async().execute(new Runnable() {
|
|
@Override
|
|
public void run() {
|
|
try {
|
|
plugin.subdata = new SubDataClient(plugin, name, socket.getInetAddress(), socket.getPort(), cipher);
|
|
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) {
|
|
log.info("SubServers > Connection was unsuccessful, retrying in " + reconnect + " seconds");
|
|
Sponge.getScheduler().createTaskBuilder().async().execute(this).delay(reconnect, TimeUnit.SECONDS).submit(plugin);
|
|
}
|
|
}
|
|
}).delay(reconnect, TimeUnit.SECONDS).submit(plugin);
|
|
}
|
|
}
|
|
}
|
|
}
|