dlss
/
FedP2P
4
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix Torrent.Drop to wait for storage close without client lock

This commit is contained in:
Matt Joiner 2021-09-10 14:14:57 +10:00
parent d957502528
commit 7378d50a3c
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
client.go
View File

@ -1258,15 +1258,13 @@ func useTorrentSource(ctx context.Context, source string, t *Torrent) (err error
return t.MergeSpec(TorrentSpecFromMetaInfo(&mi))
}
func (cl *Client) dropTorrent(infoHash metainfo.Hash) (err error) {
func (cl *Client) dropTorrent(infoHash metainfo.Hash, wg *sync.WaitGroup) (err error) {
t, ok := cl.torrents[infoHash]
if !ok {
err = fmt.Errorf("no such torrent")
return
}
var wg sync.WaitGroup
defer wg.Wait()
err = t.close(&wg)
err = t.close(wg)
if err != nil {
panic(err)
}

5
t.go
View File

@ -5,6 +5,7 @@ import (
"strings"
"github.com/anacrolix/missinggo/pubsub"
"github.com/anacrolix/sync"
"github.com/anacrolix/torrent/metainfo"
)
@ -95,9 +96,11 @@ func (t *Torrent) PieceBytesMissing(piece int) int64 {
// this. No data corruption can, or should occur to either the torrent's data,
// or connected peers.
func (t *Torrent) Drop() {
var wg sync.WaitGroup
defer wg.Wait()
t.cl.lock()
defer t.cl.unlock()
t.cl.dropTorrent(t.infoHash)
t.cl.dropTorrent(t.infoHash, &wg)
}
// Number of bytes of the entire torrent we have completed. This is the sum of