refactor: Shrink buffer if subsequent requests can work with smaller …

…request sizes
pojntfx · Oct 5, 2023 · b9d19c2 · b9d19c2
1 parent 7183cde
commit b9d19c2
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/pkg/server/nbd.go b/pkg/server/nbd.go
@@ -330,10 +330,7 @@ n:
 	}
 
 	// Transmission
-	var (
-		b                 = []byte{}
-		maxObservedLength = uint32(0)
-	)
+	b := []byte{}
 	for {
 		var requestHeader protocol.TransmissionRequestHeader
 		if err := binary.Read(conn, binary.BigEndian, &requestHeader); err != nil {
@@ -349,10 +346,8 @@ n:
 			return ErrInvalidBlocksize
 		}
 
-		if length > maxObservedLength {
+		if length != uint32(len(b)) {
 			b = make([]byte, length)
-
-			maxObservedLength = length
 		}
 
 		switch requestHeader.Type {