go/torrent-client
1
0
Fork 0
mirror of https://github.com/veggiedefender/torrent-client.git synced 2025-11-06 09:29:16 +02:00
Code Issues Releases Activity

Check infohash in handshake

Browse Source
This commit is contained in:
Jesse Li
2020-01-02 12:26:54 -05:00
parent b3ee9f4117
commit b3beb5ed2a
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
p2p/client.go
View File

@@ -2,6 +2,7 @@ package p2p
import ( import (
"bufio" "bufio"
"bytes"
"fmt" "fmt"
"net" "net"
"strconv" "strconv"
@@ -39,6 +40,9 @@ func completeHandshake(conn net.Conn, r *bufio.Reader, infohash, peerID [20]byte
if err != nil { if err != nil {
return nil, err return nil, err
} }
if !bytes.Equal(res.InfoHash[:], infohash[:]) {
return nil, fmt.Errorf("Expected infohash %x but got %x", res.InfoHash, infohash)
}
return res, nil return res, nil
} }