From aa8ef0ed768a6f9f25c01248489a1e7aa5bb22a8 Mon Sep 17 00:00:00 2001
From: Jesse Li <jessejesse123@gmail.com>
Date: Tue, 24 Dec 2019 10:19:40 -0500
Subject: [PATCH] Close connection after handling error

Fixes segfault when connection refused
---
 p2p/p2p.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/p2p/p2p.go b/p2p/p2p.go
index 67548a5..56ac638 100644
--- a/p2p/p2p.go
+++ b/p2p/p2p.go
@@ -29,10 +29,10 @@ type Downloader struct {
 // Download downloads a torrent
 func (d *Downloader) Download() error {
 	conn, err := d.Peers[0].connect(d.PeerID, d.InfoHash)
-	defer conn.Close()
 	if err != nil {
 		return err
 	}
+	defer conn.Close()
 	h, err := d.handshake(conn)
 	if err != nil {
 		return err