Search in sources :

Example 11 with BackendInteractiveChatData

use of com.loohp.interactivechat.proxy.objectholders.BackendInteractiveChatData in project InteractiveChat by LOOHP.

the class ServerPingBungee method getPing.

@SuppressWarnings("deprecation")
public static CompletableFuture<ServerPingBungee> getPing(ServerInfo server) {
    CompletableFuture<ServerPingBungee> future = new CompletableFuture<>();
    new Thread(() -> {
        try (Socket socket = new Socket()) {
            SocketAddress address = server.getSocketAddress();
            if (address instanceof InetSocketAddress) {
                InetSocketAddress inetAddress = (InetSocketAddress) address;
                socket.connect(new InetSocketAddress(inetAddress.getHostName(), inetAddress.getPort()), 1500);
            } else {
                socket.connect(address, 1500);
            }
            if (socket.isConnected()) {
                DataInputStream input = new DataInputStream(socket.getInputStream());
                DataOutputStream output = new DataOutputStream(socket.getOutputStream());
                ByteArrayOutputStream buffer = new ByteArrayOutputStream();
                DataOutputStream out = new DataOutputStream(buffer);
                out.writeByte(0x00);
                DataStreamIO.writeVarInt(out, -1);
                DataStreamIO.writeString(out, Registry.PLUGIN_MESSAGING_PROTOCOL_IDENTIFIER, StandardCharsets.UTF_8);
                out.writeShort(25566);
                DataStreamIO.writeVarInt(out, 1);
                byte[] handshake = buffer.toByteArray();
                DataStreamIO.writeVarInt(output, handshake.length);
                output.write(handshake);
                buffer = new ByteArrayOutputStream();
                out = new DataOutputStream(buffer);
                out.writeByte(0x00);
                byte[] request = buffer.toByteArray();
                DataStreamIO.writeVarInt(output, request.length);
                output.write(request);
                DataStreamIO.readVarInt(input);
                int packetId = DataStreamIO.readVarInt(input);
                if (packetId == -1) {
                    throw new IOException("Premature end of stream.");
                }
                if (packetId != 0x00) {
                    // we want a status response
                    throw new IOException("Invalid packetID " + Integer.toHexString(packetId));
                }
                // length of json string
                int length = DataStreamIO.readVarInt(input);
                if (length == -1) {
                    throw new IOException("Premature end of stream.");
                }
                if (length == 0) {
                    throw new IOException("Invalid string length.");
                }
                byte[] in = new byte[length];
                input.readFully(in);
                String jsonStr = new String(in);
                boolean present;
                String version;
                MCVersion minecraftVersion;
                String exactMinecraftVersion;
                int protocol;
                try {
                    JSONObject json = (JSONObject) new JSONParser().parse(jsonStr);
                    Object descriptionObj = json.get("description");
                    JSONObject data;
                    if (descriptionObj instanceof JSONObject) {
                        JSONObject description = (JSONObject) json.get("description");
                        String descriptionAsStr = ChatColor.stripColor(ComponentSerializer.parse(description.toJSONString())[0].toPlainText());
                        data = (JSONObject) new JSONParser().parse(descriptionAsStr);
                    } else {
                        data = (JSONObject) new JSONParser().parse(StringEscapeUtils.unescapeJava(descriptionObj.toString()));
                    }
                    present = (boolean) data.get("present");
                    version = (String) data.get("version");
                    minecraftVersion = MCVersion.fromNumber((int) (long) data.get("minecraftVersion"));
                    exactMinecraftVersion = (String) data.get("exactMinecraftVersion");
                    protocol = data.containsKey("protocol") ? (int) (long) data.get("protocol") : 0;
                } catch (Exception e) {
                    present = false;
                    version = UNKNOWN_VERSION;
                    minecraftVersion = MCVersion.UNSUPPORTED;
                    exactMinecraftVersion = UNKNOWN_VERSION;
                    protocol = UNKNOWN_PROTOCOL;
                }
                long start = System.currentTimeMillis();
                buffer = new ByteArrayOutputStream();
                out = new DataOutputStream(buffer);
                out.writeByte(0x01);
                out.writeLong(start);
                byte[] ping = buffer.toByteArray();
                DataStreamIO.writeVarInt(output, ping.length);
                output.write(ping);
                DataStreamIO.readVarInt(input);
                packetId = DataStreamIO.readVarInt(input);
                int pong = (int) (System.currentTimeMillis() - start);
                if (packetId == -1) {
                    throw new IOException("Premature end of stream.");
                }
                if (packetId != 0x01) {
                    // we want a pong
                    throw new IOException("Invalid packetID " + Integer.toHexString(packetId));
                }
                future.complete(new ServerPingBungee(pong, present));
                BackendInteractiveChatData data = InteractiveChatBungee.serverInteractiveChatInfo.get(server.getName());
                if (data == null) {
                    InteractiveChatBungee.serverInteractiveChatInfo.put(server.getName(), new BackendInteractiveChatData(server.getName(), true, present, version, minecraftVersion, exactMinecraftVersion, pong, protocol));
                } else {
                    data.setPing(pong);
                    if (!data.isOnline()) {
                        data.setOnline(true);
                    }
                    if (data.hasInteractiveChat() != present) {
                        data.setInteractiveChat(present);
                    }
                    if (!data.getVersion().equals(version)) {
                        data.setVersion(version);
                    }
                    if (!data.getMinecraftVersion().equals(minecraftVersion)) {
                        data.setMinecraftVersion(minecraftVersion);
                    }
                    if (!data.getExactMinecraftVersion().equals(exactMinecraftVersion)) {
                        data.setExactMinecraftVersion(exactMinecraftVersion);
                    }
                    if (data.getProtocolVersion() != protocol) {
                        data.setProtocolVersion(protocol);
                    }
                }
            } else {
                future.complete(new ServerPingBungee(-1, false));
                BackendInteractiveChatData data = InteractiveChatBungee.serverInteractiveChatInfo.get(server.getName());
                if (data == null) {
                    InteractiveChatBungee.serverInteractiveChatInfo.put(server.getName(), new BackendInteractiveChatData(server.getName(), false, false, UNKNOWN_VERSION, MCVersion.UNSUPPORTED, UNKNOWN_VERSION, -1, UNKNOWN_PROTOCOL));
                } else {
                    data.setPing(-1);
                    if (data.isOnline()) {
                        data.setOnline(false);
                    }
                    if (data.hasInteractiveChat()) {
                        data.setInteractiveChat(false);
                    }
                    if (!data.getVersion().equals(UNKNOWN_VERSION)) {
                        data.setVersion(UNKNOWN_VERSION);
                    }
                    if (data.getProtocolVersion() != UNKNOWN_PROTOCOL) {
                        data.setProtocolVersion(UNKNOWN_PROTOCOL);
                    }
                }
            }
        } catch (IOException e) {
            future.complete(new ServerPingBungee(-1, false));
            BackendInteractiveChatData data = InteractiveChatBungee.serverInteractiveChatInfo.get(server.getName());
            if (data == null) {
                InteractiveChatBungee.serverInteractiveChatInfo.put(server.getName(), new BackendInteractiveChatData(server.getName(), false, false, UNKNOWN_VERSION, MCVersion.UNSUPPORTED, UNKNOWN_VERSION, -1, UNKNOWN_PROTOCOL));
            } else {
                data.setPing(-1);
                if (data.isOnline()) {
                    data.setOnline(false);
                }
                if (data.hasInteractiveChat()) {
                    data.setInteractiveChat(false);
                }
                if (!data.getVersion().equals(UNKNOWN_VERSION)) {
                    data.setVersion(UNKNOWN_VERSION);
                }
                if (data.getProtocolVersion() != UNKNOWN_PROTOCOL) {
                    data.setProtocolVersion(UNKNOWN_PROTOCOL);
                }
            }
        }
    }).start();
    return future;
}
Also used : InetSocketAddress(java.net.InetSocketAddress) DataOutputStream(java.io.DataOutputStream) BackendInteractiveChatData(com.loohp.interactivechat.proxy.objectholders.BackendInteractiveChatData) ByteArrayOutputStream(java.io.ByteArrayOutputStream) IOException(java.io.IOException) DataInputStream(java.io.DataInputStream) IOException(java.io.IOException) CompletableFuture(java.util.concurrent.CompletableFuture) MCVersion(com.loohp.interactivechat.utils.MCVersion) JSONObject(org.json.simple.JSONObject) JSONParser(org.json.simple.parser.JSONParser) JSONObject(org.json.simple.JSONObject) SocketAddress(java.net.SocketAddress) InetSocketAddress(java.net.InetSocketAddress) Socket(java.net.Socket)

Example 12 with BackendInteractiveChatData

use of com.loohp.interactivechat.proxy.objectholders.BackendInteractiveChatData in project InteractiveChat by LOOHP.

the class InteractiveChatBungee method onBungeeChat.

@EventHandler(priority = EventPriority.HIGH)
public void onBungeeChat(ChatEvent event) {
    if (event.isCancelled()) {
        return;
    }
    ProxiedPlayer player = (ProxiedPlayer) event.getSender();
    UUID uuid = player.getUniqueId();
    String message = event.getMessage();
    String newMessage = event.getMessage();
    boolean hasInteractiveChat = false;
    Server server = player.getServer();
    if (server != null) {
        BackendInteractiveChatData data = serverInteractiveChatInfo.get(server.getInfo().getName());
        if (data != null) {
            hasInteractiveChat = data.hasInteractiveChat();
        }
    }
    boolean usage = false;
    outer: for (List<ICPlaceholder> serverPlaceholders : placeholderList.values()) {
        for (ICPlaceholder icplaceholder : serverPlaceholders) {
            Matcher matcher = icplaceholder.getKeyword().matcher(message);
            if (matcher.find()) {
                int start = matcher.start();
                if ((start < 1 || message.charAt(start - 1) != '\\') || (start > 1 && message.charAt(start - 1) == '\\' && message.charAt(start - 2) == '\\')) {
                    usage = true;
                    break outer;
                }
            }
        }
    }
    if (newMessage.startsWith("/")) {
        if (usage && hasInteractiveChat) {
            for (String parsecommand : InteractiveChatBungee.parseCommands) {
                if (newMessage.matches(parsecommand)) {
                    String command = newMessage.trim();
                    outer: for (List<ICPlaceholder> serverPlaceholders : placeholderList.values()) {
                        for (ICPlaceholder icplaceholder : serverPlaceholders) {
                            Pattern placeholder = icplaceholder.getKeyword();
                            Matcher matcher = placeholder.matcher(command);
                            if (matcher.find()) {
                                int start = matcher.start();
                                if ((start < 1 || command.charAt(start - 1) != '\\') || (start > 1 && command.charAt(start - 1) == '\\' && command.charAt(start - 2) == '\\')) {
                                    String uuidmatch = "<cmd=" + uuid + ":" + Registry.ID_ESCAPE_PATTERN.matcher(command.substring(matcher.start(), matcher.end())).replaceAll("\\>") + ":>";
                                    command = command.substring(0, matcher.start()) + uuidmatch + command.substring(matcher.end());
                                    if (command.length() > 256) {
                                        command = command.substring(0, 256);
                                    }
                                    event.setMessage(command);
                                    break outer;
                                }
                            }
                        }
                    }
                    break;
                }
            }
        }
    } else {
        if (usage && InteractiveChatBungee.useAccurateSenderFinder && hasInteractiveChat) {
            outer: for (List<ICPlaceholder> serverPlaceholders : placeholderList.values()) {
                for (ICPlaceholder icplaceholder : serverPlaceholders) {
                    Pattern placeholder = icplaceholder.getKeyword();
                    Matcher matcher = placeholder.matcher(message);
                    if (matcher.find()) {
                        int start = matcher.start();
                        if ((start < 1 || message.charAt(start - 1) != '\\') || (start > 1 && message.charAt(start - 1) == '\\' && message.charAt(start - 2) == '\\')) {
                            String uuidmatch = "<chat=" + uuid + ":" + Registry.ID_ESCAPE_PATTERN.matcher(message.substring(matcher.start(), matcher.end())).replaceAll("\\>") + ":>";
                            message = message.substring(0, matcher.start()) + uuidmatch + message.substring(matcher.end());
                            if (message.length() > 256) {
                                message = message.substring(0, 256);
                            }
                            event.setMessage(message);
                            break outer;
                        }
                    }
                }
            }
        }
        ProxyServer.getInstance().getScheduler().schedule(plugin, () -> {
            Map<String, Long> messages = forwardedMessages.get(uuid);
            if (messages != null && messages.remove(newMessage) != null) {
                try {
                    PluginMessageSendingBungee.sendMessagePair(uuid, newMessage);
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }, 100, TimeUnit.MILLISECONDS);
    }
}
Also used : ProxiedPlayer(net.md_5.bungee.api.connection.ProxiedPlayer) Pattern(java.util.regex.Pattern) Server(net.md_5.bungee.api.connection.Server) ProxyServer(net.md_5.bungee.api.ProxyServer) Matcher(java.util.regex.Matcher) BackendInteractiveChatData(com.loohp.interactivechat.proxy.objectholders.BackendInteractiveChatData) IOException(java.io.IOException) ICPlaceholder(com.loohp.interactivechat.objectholders.ICPlaceholder) AtomicLong(java.util.concurrent.atomic.AtomicLong) List(java.util.List) ArrayList(java.util.ArrayList) UUID(java.util.UUID) EventHandler(net.md_5.bungee.event.EventHandler)

Aggregations

BackendInteractiveChatData (com.loohp.interactivechat.proxy.objectholders.BackendInteractiveChatData)12 IOException (java.io.IOException)8 ArrayList (java.util.ArrayList)6 ICPlaceholder (com.loohp.interactivechat.objectholders.ICPlaceholder)4 Player (com.velocitypowered.api.proxy.Player)4 List (java.util.List)4 UUID (java.util.UUID)4 AtomicLong (java.util.concurrent.atomic.AtomicLong)4 Matcher (java.util.regex.Matcher)4 Pattern (java.util.regex.Pattern)4 ProxiedPlayer (net.md_5.bungee.api.connection.ProxiedPlayer)4 ParsePlayer (com.loohp.interactivechat.objectholders.CustomPlaceholder.ParsePlayer)3 ConnectedPlayer (com.velocitypowered.proxy.connection.client.ConnectedPlayer)3 Component (net.kyori.adventure.text.Component)3 JSONObject (org.json.simple.JSONObject)3 ByteArrayDataOutput (com.google.common.io.ByteArrayDataOutput)2 MCVersion (com.loohp.interactivechat.utils.MCVersion)2 Subscribe (com.velocitypowered.api.event.Subscribe)2 ChannelDuplexHandler (io.netty.channel.ChannelDuplexHandler)2 ChannelHandlerContext (io.netty.channel.ChannelHandlerContext)2