47 lines
1.4 KiB
C#
47 lines
1.4 KiB
C#
|
using System;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
using System.Linq;
|
|||
|
|
|
|||
|
|
namespace SharpChat {
|
|||
|
|
public enum ChatRateLimitState {
|
|||
|
|
None,
|
|||
|
|
Warning,
|
|||
|
|
Kick,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public class ChatRateLimiter {
|
|||
|
|
private const int FLOOD_PROTECTION_AMOUNT = 30;
|
|||
|
|
private const int FLOOD_PROTECTION_THRESHOLD = 10;
|
|||
|
|
|
|||
|
|
private readonly Queue<DateTimeOffset> TimePoints = new Queue<DateTimeOffset>();
|
|||
|
|
|
|||
|
|
public ChatRateLimitState State {
|
|||
|
|
get {
|
|||
|
|
lock (TimePoints) {
|
|||
|
|
if (TimePoints.Count == FLOOD_PROTECTION_AMOUNT) {
|
|||
|
|
if ((TimePoints.Last() - TimePoints.First()).TotalSeconds <= FLOOD_PROTECTION_THRESHOLD)
|
|||
|
|
return ChatRateLimitState.Kick;
|
|||
|
|
|
|||
|
|
if ((TimePoints.Last() - TimePoints.Skip(5).First()).TotalSeconds <= FLOOD_PROTECTION_THRESHOLD)
|
|||
|
|
return ChatRateLimitState.Warning;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return ChatRateLimitState.None;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void AddTimePoint(DateTimeOffset? dto = null) {
|
|||
|
|
if (!dto.HasValue)
|
|||
|
|
dto = DateTimeOffset.Now;
|
|||
|
|
|
|||
|
|
lock (TimePoints) {
|
|||
|
|
if (TimePoints.Count >= FLOOD_PROTECTION_AMOUNT)
|
|||
|
|
TimePoints.Dequeue();
|
|||
|
|
|
|||
|
|
TimePoints.Enqueue(dto.Value);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|