From 0f3373fdd0c3c002d68242c4bbe0c916b5378b12 Mon Sep 17 00:00:00 2001
From: Matt Joiner <anacrolix@gmail.com>
Date: Thu, 8 May 2025 15:50:56 +1000
Subject: [PATCH] Add missing return and tidy waitAvailable

---
 reader.go | 23 +++++++++++++----------
 1 file changed, 13 insertions(+), 10 deletions(-)

diff --git a/reader.go b/reader.go
index 0a1c2297..ab6df459 100644
--- a/reader.go
+++ b/reader.go
@@ -219,37 +219,40 @@ func init() {
 }
 
 // Wait until some data should be available to read. Tickles the client if it isn't. Returns how
-// much should be readable without blocking.
-func (r *reader) waitAvailable(ctx context.Context, pos, wanted int64, wait bool) (avail int64, err error) {
+// much should be readable without blocking. `block` is whether to block if nothing is available,
+// for successive reads for example.
+func (r *reader) waitAvailable(
+	ctx context.Context,
+	pos, wanted int64,
+	block bool,
+) (avail int64, err error) {
 	t := r.t
 	for {
-		r.t.cl.rLock()
+		t.cl.rLock()
 		avail = r.available(pos, wanted)
 		readerCond := t.piece(int((r.offset + pos) / t.info.PieceLength)).readerCond.Signaled()
-		r.t.cl.rUnlock()
+		t.cl.rUnlock()
 		if avail != 0 {
 			return
 		}
 		var dontWait <-chan struct{}
-		if !wait || wanted == 0 {
+		if !block || wanted == 0 {
 			dontWait = closedChan
 		}
 		select {
+		case <-readerCond:
+			continue
 		case <-r.t.closed.Done():
 			err = errTorrentClosed
-			return
 		case <-ctx.Done():
 			err = ctx.Err()
-			return
 		case <-r.t.dataDownloadDisallowed.On():
 			err = errors.New("torrent data downloading disabled")
 		case <-r.t.networkingEnabled.Off():
 			err = errors.New("torrent networking disabled")
-			return
 		case <-dontWait:
-			return
-		case <-readerCond:
 		}
+		return
 	}
 }
 
-- 
2.51.0