ToxicMushroom/mare.server
1
0
Fork 0
mirror of https://github.com/Caraxi/mare.server.git synced 2025-12-18 16:14:16 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

try onconnected rate limiting

Browse source
This commit is contained in:
Stanley Dimant 2022-08-03 23:34:52 +02:00
parent 831029a244
commit 74efb5eb6a
1 changed files with 25 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

31
MareSynchronosServer/MareSynchronosServer/Throttling/SignalRLimitFilter.cs
View file

@ -1,4 +1,5 @@
using AspNetCoreRateLimit; using AspNetCoreRateLimit;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.SignalR; using Microsoft.AspNetCore.SignalR;
using Microsoft.Extensions.Options; using Microsoft.Extensions.Options;
using System; using System;
@ -8,19 +9,19 @@ namespace MareSynchronosServer.Throttling;
public class SignalRLimitFilter : IHubFilter public class SignalRLimitFilter : IHubFilter
{ {
private readonly IRateLimitProcessor _processor; private readonly IRateLimitProcessor _processor;
private readonly IHttpContextAccessor accessor;
public SignalRLimitFilter( public SignalRLimitFilter(
IOptions<IpRateLimitOptions> options, IProcessingStrategy processing, IRateLimitCounterStore counterStore, IOptions<IpRateLimitOptions> options, IProcessingStrategy processing, IIpPolicyStore policyStore, IHttpContextAccessor accessor)
IRateLimitConfiguration rateLimitConfiguration, IIpPolicyStore policyStore)
{ {
_processor = new IpRateLimitProcessor(options?.Value, policyStore, processing); _processor = new IpRateLimitProcessor(options?.Value, policyStore, processing);
this.accessor = accessor;
} }
public async ValueTask<object> InvokeMethodAsync( public async ValueTask<object> InvokeMethodAsync(
HubInvocationContext invocationContext, Func<HubInvocationContext, ValueTask<object>> next) HubInvocationContext invocationContext, Func<HubInvocationContext, ValueTask<object>> next)
{ {
var httpContext = invocationContext.Context.GetHttpContext(); var ip = accessor.GetIpAddress();
var ip = httpContext.Connection.RemoteIpAddress.ToString();
var client = new ClientRequestIdentity var client = new ClientRequestIdentity
{ {
ClientIp = ip, ClientIp = ip,
@ -44,9 +45,27 @@ public class SignalRLimitFilter : IHubFilter
} }
// Optional method // Optional method
public Task OnConnectedAsync(HubLifetimeContext context, Func<HubLifetimeContext, Task> next) public async Task OnConnectedAsync(HubLifetimeContext context, Func<HubLifetimeContext, Task> next)
{ {
return next(context); var ip = accessor.GetIpAddress();
var client = new ClientRequestIdentity
{
ClientIp = ip,
Path = "Connect",
HttpVerb = "ws",
};
foreach (var rule in await _processor.GetMatchingRulesAsync(client))
{
var counter = await _processor.ProcessRequestAsync(client, rule);
Console.WriteLine("time: {0}, count: {1}", counter.Timestamp, counter.Count);
if (counter.Count > rule.Limit)
{
var retry = counter.Timestamp.RetryAfterFrom(rule);
throw new HubException($"rate limit {retry}");
}
}
await next(context);
} }
// Optional method // Optional method