start downloader worker

baarayy · May 17, 2024 · e2fa142 · e2fa142
1 parent 7e12e01
commit e2fa142
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 3 deletions.
diff --git a/client/client.go b/client/client.go
@@ -96,7 +96,7 @@ func (c *Client) SendUnchoke() error {
 	return err
 }
 
-func (c *Client) sendHave(index int) error {
+func (c *Client) SendHave(index int) error {
 	msg := message.FormatHave(index)
 	_, err := c.Conn.Write(msg.Serialize())
 	return err

diff --git a/p2p/p2p.go b/p2p/p2p.go
@@ -125,8 +125,29 @@ func (t *Torrent) startDownloadWorker(peer peers.Peer, workQueue chan *pieceWork
 	}
 	defer c.Conn.Close()
 	log.Printf("Completed handshake with %s\n", peer)
-	
+
 	c.SendUnchoke()
 	c.SendInterested()
-
+
+	for pw := range workQueue {
+		if !c.Bitfield.HasPiece(pw.index) {
+			workQueue <- pw
+			continue
+		}
+
+		buf, err := attemptDownloadPiece(c, pw)
+		if err != nil {
+			log.Println("Error downloading piece:", err)
+			workQueue <- pw
+			return
+		}
+		err = checkIntegrity(pw, buf)
+		if err != nil {
+			log.Println("Error, Integrity check failed piece:", err)
+			workQueue <- pw
+			return
+		}
+		c.SendHave(pw.index)
+	}
+
 }