public void Destream_can_read_streamed_blocks()
{
var blocks = new HashBlock[5];
for (int index = 0; index < blocks.Length; index++)
{
blocks[index] = new HashBlock
{
Checksum = (uint)index,
Length = index * 2,
Offset = 9223372036854775807L,
Hash = new byte[16]
};
new Random().NextBytes(blocks[index].Hash);
}
var ms = new MemoryStream();
HashBlockStreamer.Stream(blocks, ms);
ms.Seek(0, SeekOrigin.Begin);
HashBlock[] newBlocks = HashBlockStreamer.Destream(ms);
Assert.AreEqual(blocks.Length, newBlocks.Length);
for (int index = 0; index < newBlocks.Length; index++)
{
HashBlock newBlock = newBlocks[index];
HashBlock oldBlock = blocks[index];
Assert.AreEqual(oldBlock.Checksum, newBlock.Checksum);
Assert.AreEqual(oldBlock.Length, newBlock.Length);
Assert.AreEqual(oldBlock.Offset, newBlock.Offset);
Assert.IsTrue(oldBlock.Hash.SequenceEqual(newBlock.Hash));
}
}
static void Main(string[] args)
{
int blockSize = 512;
// Compute hashes
IEnumerable <HashBlock> hashBlocksFromReceiver;
using (FileStream sourceStream = File.Open("../../dest.bmp", FileMode.Open))
{
hashBlocksFromReceiver = new HashBlockGenerator(new RollingChecksum(),
new HashAlgorithmWrapper <MD5>(MD5.Create()),
blockSize).ProcessStream(sourceStream).ToArray();
}
// Stream the hash blocks
var hashBlockStream = new MemoryStream();
HashBlockStreamer.Stream(hashBlocksFromReceiver, hashBlockStream);
// Receive the hash block stream
hashBlockStream.Seek(0, SeekOrigin.Begin);
Console.Out.WriteLine("Hash block stream length: {0}", hashBlockStream.Length);
hashBlocksFromReceiver = HashBlockStreamer.Destream(hashBlockStream);
// Compute deltas
var deltaStream = new MemoryStream();
using (FileStream fileStream = File.Open("../../source.bmp", FileMode.Open))
{
var deltaGen = new DeltaGenerator(new RollingChecksum(), new HashAlgorithmWrapper <MD5>(MD5.Create()));
deltaGen.Initialize(blockSize, hashBlocksFromReceiver);
IEnumerable <IDelta> deltas = deltaGen.GetDeltas(fileStream);
deltaGen.Statistics.Dump();
fileStream.Seek(0, SeekOrigin.Begin);
var streamer = new DeltaStreamer();
streamer.Send(deltas, fileStream, deltaStream);
Console.Out.WriteLine("Delta stream length: {0}", deltaStream.Length);
}
// Rewind the delta stream (obviously wouldn't apply from a network pipe)
deltaStream.Seek(0, SeekOrigin.Begin);
// Reconstruct file
using (FileStream sourceStream = File.Open("../../dest.bmp", FileMode.Open))
{
using (FileStream outStream = File.Open("../../reconstructed.bmp", FileMode.Create))
{
var streamer = new DeltaStreamer();
streamer.Receive(deltaStream, sourceStream, outStream);
outStream.Close();
}
}
}
public void Stream_writes_valid_byte_length()
{
var blocks = new HashBlock[5];
for (int index = 0; index < blocks.Length; index++)
{
blocks[index] = new HashBlock {
Hash = new byte[16]
};
}
var ms = new MemoryStream();
HashBlockStreamer.Stream(blocks, ms);
Assert.AreEqual(blocks.Length * 32 + 4, ms.Length);
}
public void Stream_writes_valid_length()
{
var blocks = new HashBlock[5];
for (int index = 0; index < blocks.Length; index++)
{
blocks[index] = new HashBlock {
Hash = new byte[16]
};
}
var ms = new MemoryStream();
HashBlockStreamer.Stream(blocks, ms);
ms.Seek(0, SeekOrigin.Begin);
var buf = new byte[4];
ms.Read(buf, 0, 4);
Assert.AreEqual((uint)blocks.Length, BitConverter.ToUInt32(buf, 0));
}
