public void CompletePartialTest()
{
Piece piece;
peer.IsChoking = false;
peer.AmInterested = true;
peer.BitField.SetAll(true);
RequestMessage message = picker.PickPiece(peer, peers);
Assert.IsTrue(picker.ValidatePiece(peer, message.PieceIndex, message.StartOffset, message.RequestLength, out piece), "#1");
picker.CancelRequests(peer);
for (int i = 0; i < piece.BlockCount; i++)
{
message = picker.PickPiece(peer, peers);
Piece p;
Assert.IsTrue(picker.ValidatePiece(peer, message.PieceIndex, message.StartOffset, message.RequestLength, out p), "#2." + i);
}
Assert.IsTrue(piece.AllBlocksRequested, "#3");
Assert.IsTrue(piece.AllBlocksReceived, "#4");
}
public void PieceDataReceived(PeerId peer, PieceMessage message)
{
Piece piece;
if (picker.ValidatePiece(peer, message.PieceIndex, message.StartOffset, message.RequestLength, out piece))
{
PeerId id = peer;
TorrentManager manager = id.TorrentManager;
Block block = piece.Blocks [message.StartOffset / Piece.BlockSize];
long offset = (long)message.PieceIndex * id.TorrentManager.Torrent.PieceLength + message.StartOffset;
id.LastBlockReceived = DateTime.Now;
id.TorrentManager.PieceManager.RaiseBlockReceived(new BlockEventArgs(manager, block, piece, id));
id.TorrentManager.Engine.DiskManager.QueueWrite(manager, offset, message.Data, message.RequestLength, delegate {
piece.Blocks[message.StartOffset / Piece.BlockSize].Written = true;
ClientEngine.BufferManager.FreeBuffer(ref message.Data);
// If we haven't written all the pieces to disk, there's no point in hash checking
if (!piece.AllBlocksWritten)
{
return;
}
Logger.Info("Piece downloaded: {0}", piece.Index);
// Hashcheck the piece as we now have all the blocks.
id.Engine.DiskManager.BeginGetHash(id.TorrentManager, piece.Index, delegate(object o) {
byte[] hash = (byte[])o;
bool result = hash == null ? false : id.TorrentManager.Torrent.Pieces.IsValid(hash, piece.Index);
id.TorrentManager.Bitfield[message.PieceIndex] = result;
ClientEngine.MainLoop.Queue(delegate
{
id.TorrentManager.PieceManager.UnhashedPieces[piece.Index] = false;
id.TorrentManager.HashedPiece(new PieceHashedEventArgs(id.TorrentManager, piece.Index, result));
List <PeerId> peers = new List <PeerId>(piece.Blocks.Length);
for (int i = 0; i < piece.Blocks.Length; i++)
{
if (piece.Blocks[i].RequestedOff != null && !peers.Contains(piece.Blocks[i].RequestedOff))
{
peers.Add(piece.Blocks[i].RequestedOff);
}
}
for (int i = 0; i < peers.Count; i++)
{
if (peers[i].Connection != null)
{
peers[i].Peer.HashedPiece(result);
if (peers [i].Peer.TotalHashFails == 5)
{
peers[i].ConnectionManager.CleanupSocket(id, "Too many hash fails");
}
}
}
// If the piece was successfully hashed, enqueue a new "have" message to be sent out
if (result)
{
id.TorrentManager.finishedPieces.Enqueue(piece.Index);
}
});
});
});
if (piece.AllBlocksReceived)
{
this.unhashedPieces[message.PieceIndex] = true;
}
}
else
{
}
}
public virtual bool ValidatePiece(PeerId peer, int pieceIndex, int startOffset, int length, out Piece piece)
{
CheckOverriden();
return(picker.ValidatePiece(peer, pieceIndex, startOffset, length, out piece));
}
