status-im
/
torrent
mirror of
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Move peerGotPiece

This commit is contained in:
Matt Joiner 2016-03-21 14:17:15 +11:00
parent 125c864a8c
commit 1508e02935
2 changed files with 20 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
client.go
View File

@ -1147,26 +1147,6 @@ func (me *Client) sendInitialMessages(conn *connection, torrent *torrent) {
} }
} }
func (me *Client) peerGotPiece(t *torrent, c *connection, piece int) error {
if !c.peerHasAll {
if t.haveInfo() {
if c.PeerPieces == nil {
c.PeerPieces = make([]bool, t.numPieces())
}
} else {
for piece >= len(c.PeerPieces) {
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
}
c.updatePiecePriority(piece)
return nil
}
func (me *Client) peerUnchoked(torrent *torrent, conn *connection) { func (me *Client) peerUnchoked(torrent *torrent, conn *connection) {
conn.updateRequests() conn.updateRequests()
} }

20
connection.go
View File

@ -625,3 +625,23 @@ func (c *connection) discardPieceInclination() {
c.t.putPieceInclination(c.pieceInclination) c.t.putPieceInclination(c.pieceInclination)
c.pieceInclination = nil c.pieceInclination = nil
} }
func (me *Client) peerGotPiece(t *torrent, c *connection, piece int) error {
if !c.peerHasAll {
if t.haveInfo() {
if c.PeerPieces == nil {
c.PeerPieces = make([]bool, t.numPieces())
}
} else {
for piece >= len(c.PeerPieces) {
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
}
c.updatePiecePriority(piece)
return nil
}