status-im/torrent
2
0
Fork 0
mirror of synced 2025-02-23 22:28:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Limit udp tracker server request concurrency

Browse Source
This commit is contained in:
Matt Joiner 2023-01-03 00:14:21 +11:00
parent f38629d354
commit 6baf8dcb99
No known key found for this signature in database
GPG Key ID: 6B990B8185E7F782
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tracker/udp/server/server.go
View File

@ -206,13 +206,24 @@ func RunSimple(ctx context.Context, s *Server, pc net.PacketConn, family udp.Add
ctx, cancel := context.WithCancel(ctx) ctx, cancel := context.WithCancel(ctx)
defer cancel() defer cancel()
var b [1500]byte var b [1500]byte
// Limit concurrent handled requests.
sem := make(chan struct{}, 1000)
for { for {
n, addr, err := pc.ReadFrom(b[:]) n, addr, err := pc.ReadFrom(b[:])
if err != nil { if err != nil {
return err return err
} }
select {
case <-ctx.Done():
return ctx.Err()
default:
log.Printf("dropping request from %v: concurrency limit reached", addr)
continue
case sem <- struct{}{}:
}
b := append([]byte(nil), b[:n]...) b := append([]byte(nil), b[:n]...)
go func() { go func() {
defer func() { <-sem }()
err := s.HandleRequest(ctx, family, addr, b) err := s.HandleRequest(ctx, family, addr, b)
if err != nil { if err != nil {
log.Printf("error handling %v byte request from %v: %v", n, addr, err) log.Printf("error handling %v byte request from %v: %v", n, addr, err)