useTorrentSources: fix http request resource leak (#562)

This commit is contained in:
YenForYang 2021-09-09 07:39:12 -05:00 committed by GitHub
parent 62c6fbc8f8
commit 1fc6093383
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 22 additions and 18 deletions

View File

@ -1214,24 +1214,22 @@ func (t *Torrent) MergeSpec(spec *TorrentSpec) error {
}
func useTorrentSources(sources []string, t *Torrent) {
for _, s := range sources {
go func(s string) {
err := useTorrentSource(s, t)
if err != nil {
// TODO: bind context to the lifetime of *Torrent so that it's cancelled if the torrent closes
ctx := context.Background()
for i := 0; i < len(sources); i += 1 {
s := sources[i]
go func() {
if err := useTorrentSource(ctx, s, t); err != nil {
t.logger.WithDefaultLevel(log.Warning).Printf("using torrent source %q: %v", s, err)
} else {
t.logger.Printf("successfully used source %q", s)
}
}(s)
}()
}
}
func useTorrentSource(source string, t *Torrent) error {
req, err := http.NewRequest(http.MethodGet, source, nil)
if err != nil {
panic(err)
}
ctx, cancel := context.WithCancel(context.Background())
func useTorrentSource(ctx context.Context, source string, t *Torrent) (err error) {
ctx, cancel := context.WithCancel(ctx)
defer cancel()
go func() {
select {
@ -1241,21 +1239,27 @@ func useTorrentSource(source string, t *Torrent) error {
}
cancel()
}()
req = req.WithContext(ctx)
resp, err := http.DefaultClient.Do(req)
if err != nil {
return err
var req *http.Request
if req, err = http.NewRequestWithContext(ctx, http.MethodGet, source, nil); err != nil {
panic(err)
}
mi, err := metainfo.Load(resp.Body)
var resp *http.Response
if resp, err = http.DefaultClient.Do(req); err != nil {
return
}
var mi metainfo.MetaInfo
err = bencode.NewDecoder(resp.Body).Decode(&mi)
resp.Body.Close()
if err != nil {
if ctx.Err() != nil {
return nil
}
return err
return
}
return t.MergeSpec(TorrentSpecFromMetaInfo(mi))
return t.MergeSpec(TorrentSpecFromMetaInfo(&mi))
}
func (cl *Client) dropTorrent(infoHash metainfo.Hash) (err error) {
t, ok := cl.torrents[infoHash]
if !ok {