Open RakServerRateLimiter for extension

Author: Alemiz112

transport-raknet/src/main/java/org/cloudburstmc/netty/handler/codec/raknet/server/RakServerRateLimiter.java

@@ -65,12 +65,12 @@ public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
         this.rateLimitMap.clear();
     }
 
-    private void onRakTick() {
+    protected void onRakTick() {
         this.rateLimitMap.clear();
         this.globalCounter.set(0);
     }
 
-    private void onBlockedTick() {
+    protected void onBlockedTick() {
         long currTime = System.currentTimeMillis();
 
         RakServerMetrics metrics = this.channel.config().getMetrics();
@@ -126,6 +126,14 @@ public void removeException(InetAddress address) {
         this.exceptions.remove(address);
     }
 
+    public Collection<InetAddress> getExceptions() {
+        return Collections.unmodifiableCollection(this.exceptions);
+    }
+
+    protected int getAddressMaxPacketCount(InetAddress address) {
+        return this.channel.config().getPacketLimit();
+    }
+
     @Override
     protected void channelRead0(ChannelHandlerContext ctx, DatagramPacket datagram) throws Exception {
         if (this.globalCounter.incrementAndGet() > this.channel.config().getGlobalPacketLimit()) {
@@ -141,7 +149,7 @@ protected void channelRead0(ChannelHandlerContext ctx, DatagramPacket datagram)
         }
 
         AtomicInteger counter = this.rateLimitMap.computeIfAbsent(address, a -> new AtomicInteger());
-        if (counter.incrementAndGet() > this.channel.config().getPacketLimit() &&
+        if (counter.incrementAndGet() > this.getAddressMaxPacketCount(address) &&
                 this.blockAddress(address, 10, TimeUnit.SECONDS)) {
             log.warn("[{}] Blocked because packet limit was reached", address);
         } else {