check integrity

baarayy · May 16, 2024 · c296134 · c296134
1 parent c83bf18
commit c296134
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/p2p/p2p.go b/p2p/p2p.go
@@ -1,6 +1,9 @@
 package p2p
 
 import (
+	"bytes"
+	"crypto/sha1"
+	"fmt"
 	"magnetflow/client"
 	"magnetflow/message"
 	"magnetflow/peers"
@@ -104,3 +107,11 @@ func attemptDownloadPiece(c *client.Client, pw *pieceWork) ([]byte, error) {
 	}
 	return state.buf, nil
 }
+
+func checkIntegrity(pw *pieceWork, buf []byte) error {
+	hash := sha1.Sum(buf)
+	if !bytes.Equal(hash[:], pw.hash[:]) {
+		return fmt.Errorf("Piece %d failed integrity check", pw.index)
+	}
+	return nil
+}