dlss
/
FedP2P
4
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix crash if peer sends out of bounds HAVE message

This commit is contained in:
Matt Joiner 2015-08-04 01:17:32 +10:00
parent f69c51bea3
commit f3317322f8
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
client.go
View File

@ -1249,7 +1249,7 @@ func (t *torrent) initRequestOrdering(c *connection) {
}
}
func (me *Client) peerGotPiece(t *torrent, c *connection, piece int) {
func (me *Client) peerGotPiece(t *torrent, c *connection, piece int) error {
if !c.peerHasAll {
if t.haveInfo() {
if c.PeerPieces == nil {
@ -1260,12 +1260,16 @@ func (me *Client) peerGotPiece(t *torrent, c *connection, piece int) {
c.PeerPieces = append(c.PeerPieces, false)
}
}
if piece >= len(c.PeerPieces) {
return errors.New("peer got out of range piece index")
}
c.PeerPieces[piece] = true
}
if t.wantPiece(piece) {
t.connPendPiece(c, piece)
me.replenishConnRequests(t, c)
}
return nil
}
func (me *Client) peerUnchoked(torrent *torrent, conn *connection) {