String info = jedis.info(); for (String s : info.split("\r\n")) { if (s.startsWith("redis_version:")) { String version = s.split(":")[1]; getLogger().info("{} <- redis version", version); if (!RedisUtil.isRedisVersionRight(version)) { getLogger().error("Your version of Redis ({}) is not at least version 6.0 RedisBungee requires a newer version of Redis.", version); throw new RuntimeException("Unsupported Redis version detected"); } long uuidCacheSize = jedis.hlen("uuid-cache"); if (uuidCacheSize > 750000) { getLogger().info("Looks like you have a really big UUID cache! Run https://www.spigotmc.org/resources/redisbungeecleaner.8505/ as soon as possible."); } break; } } return null; } @Override public Void clusterJedisTask(JedisCluster jedisCluster) { // This is more portable than INFO

try { getRedisClusterTimeScript = luaManager.createScript(IOUtil.readInputStreamAsString(getResourceAsStream("lua/get_cluster_time.lua"))); } catch (JedisException e) { throw new RuntimeException("possible not supported redis version", e); } String info = (String) luaManager.createScript(IOUtil.readInputStreamAsString(getResourceAsStream("lua/get_cluster_info.lua"))).eval(Collections.singletonList("0"), Collections.emptyList()); for (String s : info.split("\r\n")) { if (s.startsWith("redis_version:")) { String version = s.split(":")[1]; getLogger().info("{} <- redis version", version); if (!RedisUtil.isRedisVersionRight(version)) { getLogger().error("Your version of Redis ({}) is not at least version 6.0 RedisBungee requires a newer version of Redis.", version); throw new RuntimeException("Unsupported Redis version detected"); } long uuidCacheSize = jedisCluster.hlen("uuid-cache"); if (uuidCacheSize > 750000) { getLogger().info("Looks like you have a really big UUID cache! Run https://www.spigotmc.org/resources/redisbungeecleaner.8505/ as soon as possible."); } break; } } return null; } }.execute(); getLogger().info("lua manager was loaded"); // check if this proxy is recovering from a crash and start heart the beat. new RedisTask(api) { @Override public Void jedisTask(Jedis jedis) { Path crashFile = getDataFolder().resolve("restarted_from_crash.txt"); if (Files.exists(crashFile)) { try { Files.delete(crashFile); } catch (IOException e) { throw new RuntimeException(e); } getLogger().info("crash file was deleted"); } else if (jedis.hexists("heartbeats", configuration.getProxyId())) { try { long value = Long.parseLong(jedis.hget("heartbeats", configuration.getProxyId())); long redisTime = getRedisTime(jedis.time()); if (redisTime < value + 20) { getLogger().error("You have launched a possible impostor Velocity / Bungeecord instance. Another instance is already running."); getLogger().error("For data consistency reasons, RedisBungee will now disable itself."); getLogger().error("If this instance is coming up from a crash, create a file in your RedisBungee plugins directory with the name 'restarted_from_crash.txt' and RedisBungee will not perform this check."); throw new RuntimeException("Possible impostor instance!"); } } catch (NumberFormatException ignored) { } } return null; } @Override public Void clusterJedisTask(JedisCluster jedisCluster) { Path crashFile = getDataFolder().resolve("restarted_from_crash.txt"); if (Files.exists(crashFile)) { try { Files.delete(crashFile); } catch (IOException e) { throw new RuntimeException(e); } getLogger().info("crash file was deleted"); } else if (jedisCluster.hexists("heartbeats", configuration.getProxyId())) { try { long value = Long.parseLong(jedisCluster.hget("heartbeats", configuration.getProxyId())); long redisTime = getRedisClusterTime(); if (redisTime < value + 20) { getLogger().error("You have launched a possible impostor Velocity / Bungeecord instance. Another instance is already running."); getLogger().error("For data consistency reasons, RedisBungee will now disable itself."); getLogger().error("If this instance is coming up from a crash, create a file in your RedisBungee plugins directory with the name 'restarted_from_crash.txt' and RedisBungee will not perform this check."); throw new RuntimeException("Possible impostor instance!"); } } catch (NumberFormatException ignored) { } } return null; } }.execute(); uuidTranslator = new UUIDTranslator(this); dataManager = new VelocityDataManager(this); psl = new PubSubListener(this); this.httpClient = new OkHttpClient(); Dispatcher dispatcher = new Dispatcher(Executors.newFixedThreadPool(6)); this.httpClient.setDispatcher(dispatcher); NameFetcher.setHttpClient(httpClient); UUIDFetcher.setHttpClient(httpClient); // keeping this lol new RedisBungee(api); } @Override public RedisBungeeConfiguration getConfiguration() { return this.configuration; } @Override public int getCount() { return this.globalPlayerCount.get(); } @Override public int getCurrentCount() { return new RedisTask(api) { @Override public Long jedisTask(Jedis jedis) { long total = 0; long redisTime = getRedisTime(jedis.time()); Map heartBeats = jedis.hgetAll("heartbeats"); for (Map.Entry stringStringEntry : heartBeats.entrySet()) { String k = stringStringEntry.getKey(); String v = stringStringEntry.getValue(); long heartbeatTime = Long.parseLong(v); if (heartbeatTime + 30 >= redisTime) { total = total + jedis.scard("proxy:" + k + ":usersOnline"); } } return total; } @Override public Long clusterJedisTask(JedisCluster jedisCluster) { long total = 0; long redisTime = getRedisClusterTime(); Map heartBeats = jedisCluster.hgetAll("heartbeats"); for (Map.Entry stringStringEntry : heartBeats.entrySet()) { String k = stringStringEntry.getKey(); String v = stringStringEntry.getValue(); long heartbeatTime = Long.parseLong(v); if (heartbeatTime + 30 >= redisTime) { total = total + jedisCluster.scard("proxy:" + k + ":usersOnline"); } } return total; } }.execute().intValue(); } @Override public Set getLocalPlayersAsUuidStrings() { ImmutableSet.Builder builder = ImmutableSet.builder(); for (Player player : getProxy().getAllPlayers()) { builder.add(player.getUniqueId().toString()); } return builder.build(); } @Override public AbstractDataManager getDataManager() { return this.dataManager; } @Override public Set getPlayers() { return new RedisTask>(api) { @Override public Set jedisTask(Jedis jedis) { ImmutableSet.Builder setBuilder = ImmutableSet.builder(); try { List keys = new ArrayList<>(); for (String i : getProxiesIds()) { keys.add("proxy:" + i + ":usersOnline"); } if (!keys.isEmpty()) { Set users = jedis.sunion(keys.toArray(new String[0])); if (users != null && !users.isEmpty()) { for (String user : users) { try { setBuilder = setBuilder.add(UUID.fromString(user)); } catch (IllegalArgumentException ignored) { } } } } } catch (JedisConnectionException e) { // Redis server has disappeared! getLogger().error("Unable to get connection from pool - did your Redis server go away?", e); throw new RuntimeException("Unable to get all players online", e); } return setBuilder.build(); } @Override public Set clusterJedisTask(JedisCluster jedisCluster) { ImmutableSet.Builder setBuilder = ImmutableSet.builder(); try { List keys = new ArrayList<>(); for (String i : getProxiesIds()) { keys.add("proxy:" + i + ":usersOnline"); } if (!keys.isEmpty()) { Set users = jedisCluster.sunion(keys.toArray(new String[0])); if (users != null && !users.isEmpty()) { for (String user : users) { try { setBuilder = setBuilder.add(UUID.fromString(user)); } catch (IllegalArgumentException ignored) { } } } } } catch (JedisConnectionException e) { // Redis server has disappeared! getLogger().error("Unable to get connection from pool - did your Redis server go away?", e); throw new RuntimeException("Unable to get all players online", e); } return setBuilder.build(); } }.execute(); } @Override public Summoner getSummoner() { return this.jedisSummoner; } @Override public RedisBungeeAPI getApi() { return this.api; } @Override public UUIDTranslator getUuidTranslator() { return this.uuidTranslator; } @Override public Multimap serversToPlayers() { try { return serverToPlayersCache.get(SERVER_TO_PLAYERS_KEY, new RedisTask>(api) { @Override public Multimap jedisTask(Jedis jedis) { ImmutableMultimap.Builder builder = ImmutableMultimap.builder(); for (String serverId : getProxiesIds()) { Set players = jedis.smembers("proxy:" + serverId + ":usersOnline"); for (String player : players) { String playerServer = jedis.hget("player:" + player, "server"); if (playerServer == null) { continue; } builder.put(playerServer, UUID.fromString(player)); } } return builder.build(); } @Override public Multimap clusterJedisTask(JedisCluster jedisCluster) { ImmutableMultimap.Builder builder = ImmutableMultimap.builder(); for (String serverId : getProxiesIds()) { Set players = jedisCluster.smembers("proxy:" + serverId + ":usersOnline"); for (String player : players) { String playerServer = jedisCluster.hget("player:" + player, "server"); if (playerServer == null) { continue; } builder.put(playerServer, UUID.fromString(player)); } } return builder.build(); } }); } catch (ExecutionException e) { throw new RuntimeException(e); } } @Override public Set getPlayersOnProxy(String proxyId) { checkArgument(getProxiesIds().contains(proxyId), proxyId + " is not a valid proxy ID"); return new RedisTask>(api) { @Override public Set jedisTask(Jedis jedis) { Set users = jedis.smembers("proxy:" + proxyId + ":usersOnline"); ImmutableSet.Builder builder = ImmutableSet.builder(); for (String user : users) { builder.add(UUID.fromString(user)); } return builder.build(); } @Override public Set clusterJedisTask(JedisCluster jedisCluster) { Set users = jedisCluster.smembers("proxy:" + proxyId + ":usersOnline"); ImmutableSet.Builder builder = ImmutableSet.builder(); for (String user : users) { builder.add(UUID.fromString(user)); } return builder.build(); } }.execute(); } @Override public void sendProxyCommand(String serverId, String command) { checkArgument(getProxiesIds().contains(serverId) || serverId.equals("allservers"), "proxyId is invalid"); sendChannelMessage("redisbungee-" + serverId, command); } @Override public List getProxiesIds() { return proxiesIds; } @Override public List getCurrentProxiesIds(boolean nag, boolean lagged) { return new RedisTask>(api) { @Override public List jedisTask(Jedis jedis) { try { long time = getRedisTime(jedis.time()); int nagTime = 0; if (nag) { nagTime = nagAboutServers.decrementAndGet(); if (nagTime <= 0) { nagAboutServers.set(10); } } ImmutableList.Builder servers = ImmutableList.builder(); Map heartbeats = jedis.hgetAll("heartbeats"); for (Map.Entry entry : heartbeats.entrySet()) { try { long stamp = Long.parseLong(entry.getValue()); if (lagged ? time >= stamp + 30 : time <= stamp + 30) servers.add(entry.getKey()); else if (nag && nagTime <= 0) { getLogger().warn("{} is {} seconds behind! (Time not synchronized or server down?) and was removed from heartbeat.", entry.getKey(), (time - stamp)); jedis.hdel("heartbeats", entry.getKey()); } } catch (NumberFormatException ignored) { } } return servers.build(); } catch (JedisConnectionException e) { getLogger().error("Unable to fetch server IDs", e); return Collections.singletonList(configuration.getProxyId()); } } @Override public List clusterJedisTask(JedisCluster jedisCluster) { try { long time = getRedisClusterTime(); int nagTime = 0; if (nag) { nagTime = nagAboutServers.decrementAndGet(); if (nagTime <= 0) { nagAboutServers.set(10); } } ImmutableList.Builder servers = ImmutableList.builder(); Map heartbeats = jedisCluster.hgetAll("heartbeats"); for (Map.Entry entry : heartbeats.entrySet()) { try { long stamp = Long.parseLong(entry.getValue()); if (lagged ? time >= stamp + 30 : time <= stamp + 30) servers.add(entry.getKey()); else if (nag && nagTime <= 0) { getLogger().warn("{} is {} seconds behind! (Time not synchronized or server down?) and was removed from heartbeat.", entry.getKey(), (time - stamp)); jedisCluster.hdel("heartbeats", entry.getKey()); } } catch (NumberFormatException ignored) { } } return servers.build(); } catch (JedisConnectionException e) { getLogger().error("Unable to fetch server IDs", e); return Collections.singletonList(configuration.getProxyId()); } } }.execute(); } @Override public PubSubListener getPubSubListener() { return this.psl; } @Override public void sendChannelMessage(String channel, String message) { new RedisTask(api) { @Override public Void jedisTask(Jedis jedis) { try { jedis.publish(channel, message); } catch (JedisConnectionException e) { // Redis server has disappeared! getLogger().error("Unable to get connection from pool - did your Redis server go away?", e); throw new RuntimeException("Unable to publish channel message", e); } return null; } @Override public Void clusterJedisTask(JedisCluster jedisCluster) { try { jedisCluster.publish(channel, message); } catch (JedisConnectionException e) { // Redis server has disappeared! getLogger().error("Unable to get connection from pool - did your Redis server go away?", e); throw new RuntimeException("Unable to publish channel message", e); } return null; } }.execute(); } @Override public void executeAsync(Runnable runnable) { this.getProxy().getScheduler().buildTask(this, runnable).schedule(); } @Override public void executeAsyncAfter(Runnable runnable, TimeUnit timeUnit, int time) { this.getProxy().getScheduler().buildTask(this, runnable).delay(time, timeUnit).schedule(); } @Override public void callEvent(Object event) { this.getProxy().getEventManager().fireAndForget(event); } @Override public boolean isOnlineMode() { return this.getProxy().getConfiguration().isOnlineMode(); } @Override public void logInfo(String msg) { this.getLogger().info(msg); } @Override public void logWarn(String msg) { this.getLogger().warn(msg); } @Override public void logFatal(String msg) { this.getLogger().error(msg); } @Override public Player getPlayer(UUID uuid) { return this.getProxy().getPlayer(uuid).orElse(null); } @Override public Player getPlayer(String name) { return this.getProxy().getPlayer(name).orElse(null); } @Override public UUID getPlayerUUID(String player) { return this.getProxy().getPlayer(player).map(Player::getUniqueId).orElse(null); } @Override public String getPlayerName(UUID player) { return this.getProxy().getPlayer(player).map(Player::getUsername).orElse(null); } @Override public String getPlayerServerName(Player player) { return player.getCurrentServer().map(serverConnection -> serverConnection.getServerInfo().getName()).orElse(null); } @Override public boolean isPlayerOnAServer(Player player) { return player.getCurrentServer().isPresent(); } @Override public InetAddress getPlayerIp(Player player) { return player.getRemoteAddress().getAddress(); } @Override public void sendProxyCommand(String cmd) { checkArgument(getProxiesIds().contains(this.configuration.getProxyId()) || this.configuration.getProxyId().equals("allservers"), "proxyId is invalid"); sendChannelMessage("redisbungee-" + this.configuration.getProxyId(), cmd); } @Override public long getRedisTime(List timeRes) { return Long.parseLong(timeRes.get(0)); } @Override public void initialize() { updateProxyIds(); // start heartbeat task heartbeatTask = getProxy().getScheduler().buildTask(this, new HeartbeatTask(this, this.globalPlayerCount)).repeat(HeartbeatTask.INTERVAL, HeartbeatTask.REPEAT_INTERVAL_TIME_UNIT).schedule(); getProxy().getEventManager().register(this, new RedisBungeeVelocityListener(this, configuration.getExemptAddresses())); getProxy().getEventManager().register(this, dataManager); getProxy().getScheduler().buildTask(this, psl).schedule(); RedisTask integrityCheckRedisTask = new RedisTask(api) { @Override public Void jedisTask(Jedis jedis) { try { Set players = getLocalPlayersAsUuidStrings(); Set playersInRedis = jedis.smembers("proxy:" + configuration.getProxyId() + ":usersOnline"); List lagged = getCurrentProxiesIds(false, true); // Clean up lagged players. for (String s : lagged) { Set laggedPlayers = jedis.smembers("proxy:" + s + ":usersOnline"); jedis.del("proxy:" + s + ":usersOnline"); if (!laggedPlayers.isEmpty()) { getLogger().info("Cleaning up lagged proxy {} ({} players)...", s, laggedPlayers.size()); for (String laggedPlayer : laggedPlayers) { GenericPlayerUtils.cleanUpPlayer(laggedPlayer, jedis, true); } } } Set absentLocally = new HashSet<>(playersInRedis); absentLocally.removeAll(players); Set absentInRedis = new HashSet<>(players); absentInRedis.removeAll(playersInRedis); for (String member : absentLocally) { boolean found = false; for (String proxyId : getProxiesIds()) { if (proxyId.equals(configuration.getProxyId())) continue; if (jedis.sismember("proxy:" + proxyId + ":usersOnline", member)) { // Just clean up the set. found = true; break; } } if (!found) { GenericPlayerUtils.cleanUpPlayer(member, jedis, false); getLogger().warn("Player found in set that was not found locally and globally: {}", member); } else { jedis.srem("proxy:" + configuration.getProxyId() + ":usersOnline", member); getLogger().warn("Player found in set that was not found locally, but is on another proxy: {}", member); } } Pipeline pipeline = jedis.pipelined(); for (String player : absentInRedis) { // Player not online according to Redis but not BungeeCord. getLogger().warn("Player {} is on the proxy but not in Redis.", player); Player playerProxied = getProxy().getPlayer(UUID.fromString(player)).orElse(null); if (playerProxied == null) continue; // We'll deal with it later. VelocityPlayerUtils.createPlayer(playerProxied, pipeline, false); } pipeline.sync(); } catch (Throwable e) { getLogger().error("Unable to fix up stored player data", e); } return null; } @Override public Void clusterJedisTask(JedisCluster jedisCluster) { try { Set players = getLocalPlayersAsUuidStrings(); Set playersInRedis = jedisCluster.smembers("proxy:" + configuration.getProxyId() + ":usersOnline"); List lagged = getCurrentProxiesIds(false, true); // Clean up lagged players. for (String s : lagged) { Set laggedPlayers = jedisCluster.smembers("proxy:" + s + ":usersOnline"); jedisCluster.del("proxy:" + s + ":usersOnline"); if (!laggedPlayers.isEmpty()) { getLogger().info("Cleaning up lagged proxy {} ({} players)...", s, laggedPlayers.size()); for (String laggedPlayer : laggedPlayers) { GenericPlayerUtils.cleanUpPlayer(laggedPlayer, jedisCluster, true); } } } Set absentLocally = new HashSet<>(playersInRedis); absentLocally.removeAll(players); Set absentInRedis = new HashSet<>(players); absentInRedis.removeAll(playersInRedis); for (String member : absentLocally) { boolean found = false; for (String proxyId : getProxiesIds()) { if (proxyId.equals(configuration.getProxyId())) continue; if (jedisCluster.sismember("proxy:" + proxyId + ":usersOnline", member)) { // Just clean up the set. found = true; break; } } if (!found) { GenericPlayerUtils.cleanUpPlayer(member, jedisCluster, false); getLogger().warn("Player found in set that was not found locally and globally: {}", member); } else { jedisCluster.srem("proxy:" + configuration.getProxyId() + ":usersOnline", member); getLogger().warn("Player found in set that was not found locally, but is on another proxy: {}", member); } } for (String player : absentInRedis) { // Player not online according to Redis but not BungeeCord. getLogger().warn("Player {} is on the proxy but not in Redis.", player); Player playerProxied = getProxy().getPlayer(UUID.fromString(player)).orElse(null); if (playerProxied == null) continue; // We'll deal with it later. VelocityPlayerUtils.createPlayer(playerProxied, jedisCluster, false); } } catch (Throwable e) { getLogger().error("Unable to fix up stored player data", e); } return null; } }; integrityCheck = getProxy().getScheduler().buildTask(this, integrityCheckRedisTask::execute).repeat(30, TimeUnit.SECONDS).schedule(); // register plugin messages IDENTIFIERS.forEach(getProxy().getChannelRegistrar()::register); // register legacy commands if (configuration.doRegisterLegacyCommands()) { // Override Velocity commands if (configuration.doOverrideBungeeCommands()) { getProxy().getCommandManager().register("glist", new RedisBungeeCommands.GlistCommand(this), "redisbungee", "rglist"); } getProxy().getCommandManager().register("sendtoall", new RedisBungeeCommands.SendToAll(this), "rsendtoall"); getProxy().getCommandManager().register("serverid", new RedisBungeeCommands.ServerId(this), "rserverid"); getProxy().getCommandManager().register("serverids", new RedisBungeeCommands.ServerIds(this)); getProxy().getCommandManager().register("pproxy", new RedisBungeeCommands.PlayerProxyCommand(this)); getProxy().getCommandManager().register("plist", new RedisBungeeCommands.PlistCommand(this), "rplist"); getProxy().getCommandManager().register("lastseen", new RedisBungeeCommands.LastSeenCommand(this), "rlastseen"); getProxy().getCommandManager().register("ip", new RedisBungeeCommands.IpCommand(this), "playerip", "rip", "rplayerip"); getProxy().getCommandManager().register("find", new RedisBungeeCommands.FindCommand(this), "rfind"); } } @Override public void stop() { // Poison the PubSub listener if (psl != null) { psl.poison(); } if (integrityCheck != null) { integrityCheck.cancel(); } if (heartbeatTask != null) { heartbeatTask.cancel(); } new RedisTask(api) { @Override public Void jedisTask(Jedis jedis) { jedis.hdel("heartbeats", configuration.getProxyId()); if (jedis.scard("proxy:" + configuration.getProxyId() + ":usersOnline") > 0) { Set players = jedis.smembers("proxy:" + configuration.getProxyId() + ":usersOnline"); for (String member : players) GenericPlayerUtils.cleanUpPlayer(member, jedis, true); } return null; } @Override public Void clusterJedisTask(JedisCluster jedisCluster) { jedisCluster.hdel("heartbeats", configuration.getProxyId()); if (jedisCluster.scard("proxy:" + configuration.getProxyId() + ":usersOnline") > 0) { Set players = jedisCluster.smembers("proxy:" + configuration.getProxyId() + ":usersOnline"); for (String member : players) GenericPlayerUtils.cleanUpPlayer(member, jedisCluster, true); } return null; } }.execute(); try { this.jedisSummoner.close(); } catch (IOException e) { throw new RuntimeException(e); } this.httpClient.getDispatcher().getExecutorService().shutdown(); try { this.httpClient.getDispatcher().getExecutorService().awaitTermination(20, TimeUnit.SECONDS); } catch (InterruptedException e) { throw new RuntimeException(e); } } @Override public void onConfigLoad(RedisBungeeConfiguration configuration, Summoner summoner, RedisBungeeMode mode) { this.jedisSummoner = summoner; this.configuration = configuration; this.redisBungeeMode = mode; } @Override public void kickPlayer(UUID playerUniqueId, String message) { // first handle on origin proxy if player not found publish the payload if (!dataManager.handleKick(playerUniqueId, message)) { new RedisTask(api) { @Override public Void jedisTask(Jedis jedis) { PayloadUtils.kickPlayerPayload(playerUniqueId, message, jedis); return null; } @Override public Void clusterJedisTask(JedisCluster jedisCluster) { PayloadUtils.kickPlayerPayload(playerUniqueId, message, jedisCluster); return null; } }.execute(); } } @Override public void kickPlayer(String playerName, String message) { // fetch the uuid UUID playerUUID = this.uuidTranslator.getTranslatedUuid(playerName, true); kickPlayer(playerUUID, message); } @Override public RedisBungeeMode getRedisBungeeMode() { return this.redisBungeeMode; } @Override public Long getRedisClusterTime() { return getRedisTime((List) this.getRedisClusterTimeScript.eval(Collections.singletonList("0"), Collections.emptyList())); } @Override public void updateProxyIds() { this.proxiesIds = this.getCurrentProxiesIds(true, false); } @Subscribe public void proxyInit(ProxyInitializeEvent event) { initialize(); } @Subscribe public void proxyShutdownEvent(ProxyShutdownEvent event) { stop(); } @Override public Object createPlayerChangedNetworkEvent(UUID uuid, String previousServer, String server) { return new PlayerChangedServerNetworkEvent(uuid, previousServer, server); } @Override public Object createPlayerJoinedNetworkEvent(UUID uuid) { return new PlayerJoinedNetworkEvent(uuid); } @Override public Object createPlayerLeftNetworkEvent(UUID uuid) { return new PlayerLeftNetworkEvent(uuid); } @Override public Object createPubSubEvent(String channel, String message) { return new PubSubMessageEvent(channel, message); } public ProxyServer getProxy() { return server; } public Logger getLogger() { return logger; } public Path getDataFolder() { return this.dataFolder; } public InputStream getResourceAsStream(String name) { return this.getClass().getClassLoader().getResourceAsStream(name); } }