sendnrw/netbird
2
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2026-10-26 14:29:08 +01:00
Code Issues Packages Projects Releases Wiki Activity

add sleep on login and log

Browse Source
This commit is contained in:
Maycon Santos
2025-05-08 10:12:58 +02:00
parent 89729d85df
commit 1046342e2c
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
management/server/grpcserver.go
+3
View File
@@ -3,6 +3,7 @@ package server
import ( import (
"context" "context"
"fmt" "fmt"
"math/rand/v2"
"net" "net"
"net/netip" "net/netip"
"os" "os"
@@ -455,6 +456,8 @@ func (s *GRPCServer) parseRequest(ctx context.Context, req *proto.EncryptedMessa
func (s *GRPCServer) Login(ctx context.Context, req *proto.EncryptedMessage) (*proto.EncryptedMessage, error) { func (s *GRPCServer) Login(ctx context.Context, req *proto.EncryptedMessage) (*proto.EncryptedMessage, error) {
limiter, _ := s.loginLimiterStore.LoadOrStore(req.WgPubKey, rate.NewLimiter(s.loginPeerLimit, 1)) limiter, _ := s.loginLimiterStore.LoadOrStore(req.WgPubKey, rate.NewLimiter(s.loginPeerLimit, 1))
if !limiter.(*rate.Limiter).Allow() { if !limiter.(*rate.Limiter).Allow() {
time.Sleep(time.Millisecond * time.Duration(rand.IntN(10)*100))
log.WithContext(ctx).Warnf("rate limit exceeded for %s", req.WgPubKey)
return nil, status.Errorf(codes.Internal, "temp rate limit reached") return nil, status.Errorf(codes.Internal, "temp rate limit reached")
} }