From 6baf8dcb9989cb22b20893b20347be010475f0b4 Mon Sep 17 00:00:00 2001
From: Matt Joiner <anacrolix@gmail.com>
Date: Tue, 3 Jan 2023 00:14:21 +1100
Subject: [PATCH] Limit udp tracker server request concurrency

---
 tracker/udp/server/server.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/tracker/udp/server/server.go b/tracker/udp/server/server.go
index c686a23d..6f0fb438 100644
--- a/tracker/udp/server/server.go
+++ b/tracker/udp/server/server.go
@@ -206,13 +206,24 @@ func RunSimple(ctx context.Context, s *Server, pc net.PacketConn, family udp.Add
 	ctx, cancel := context.WithCancel(ctx)
 	defer cancel()
 	var b [1500]byte
+	// Limit concurrent handled requests.
+	sem := make(chan struct{}, 1000)
 	for {
 		n, addr, err := pc.ReadFrom(b[:])
 		if err != nil {
 			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]...)
 		go func() {
+			defer func() { <-sem }()
 			err := s.HandleRequest(ctx, family, addr, b)
 			if err != nil {
 				log.Printf("error handling %v byte request from %v: %v", n, addr, err)