/// <summary>
/// Writes a sequence of bytes to the current stream and advances the current position within this stream by the number of bytes written.
/// </summary>
/// <param name="buffer">An array of bytes. This method copies <paramref name="count"/> bytes from <paramref name="buffer"/> to the current stream.</param>
/// <param name="offset">The zero-based byte offset in <paramref name="buffer"/> at which to begin copying bytes to the current stream.</param>
/// <param name="count">The number of bytes to be written to the current stream.</param>
public override void Write(byte[] buffer, int offset, int count)
{
ValidateSize(count);
if (count == 0)
{
return;
}
//get the current sector
var currentSector = CurrentSector;
if (_encryptor == null)
{
_encryptor = _xts.CreateEncryptor();
}
//encrypt the sector
int transformedCount = _encryptor.TransformBlock(buffer, offset, count, _tempBuffer, 0, currentSector);
//Console.WriteLine("Encrypting sector {0}", currentSector);
//write it to the base stream
base.Write(_tempBuffer, 0, transformedCount);
}
/// <summary>
/// Reads a sequence of bytes from the current stream and advances the position within the stream by the number of bytes read.
/// </summary>
/// <returns>The total number of bytes read into the buffer. This can be less than the number of bytes requested if that many bytes are not currently available, or zero (0) if the end of the stream has been reached.</returns>
/// <param name="buffer">An array of bytes. When this method returns, the buffer contains the specified byte array with the values between <paramref name="offset"/> and (<paramref name="offset"/> + <paramref name="count"/> - 1) replaced by the bytes read from the current source. </param>
/// <param name="offset">The zero-based byte offset in <paramref name="buffer"/> at which to begin storing the data read from the current stream.</param>
/// <param name="count">The maximum number of bytes to be read from the current stream.</param>
public override int Read(byte[] buffer, int offset, int count)
{
ValidateSize(count);
//get the current sector
var currentSector = CurrentSector;
//read the sector from the base stream
var ret = base.Read(_tempBuffer, 0, count);
if (ret == 0)
{
return(0);
}
if (_decryptor == null)
{
_decryptor = _xts.CreateDecryptor();
}
//decrypt the sector
var retV = _decryptor.TransformBlock(_tempBuffer, 0, ret, buffer, offset, currentSector);
//Console.WriteLine("Decrypting sector {0} == {1} bytes", currentSector, retV);
return(retV);
}
/// <summary>
/// Creates a new stream
/// </summary>
/// <param name="baseStream">The base stream</param>
/// <param name="xts">The xts transform</param>
/// <param name="sectorSize">Sector size</param>
public XTSWriteOnlyStream(Stream baseStream, Xts xts, int sectorSize = DefaultSectorSize)
{
_baseStream = baseStream;
_encryptor = xts.CreateEncryptor();
_sectorSize = sectorSize;
_encriptedBuffer = new byte[sectorSize];
_sectorBuffer = new byte[sectorSize];
}
public XTSReadOnlyStream(Stream baseStream, Xts xts, int sectorSize, ulong startSector, int trimStart, uint trimEnd) : base(baseStream, xts, sectorSize)
{
_baseStream = baseStream;
_tempBuffer = new byte[SectorSize];
_decryptor = xts.CreateDecryptor();
_currentSector = startSector;
Length = baseStream.Length - trimStart - trimEnd;
if (trimStart > 0)
{
int read = ReadExactly(_baseStream, _tempBuffer, sectorSize);
_decryptor.TransformBlock(_tempBuffer, 0, SectorSize, _tempBuffer, 0, _currentSector);
_tempBufferCount = read;
_tempBufferPos = trimStart;
_currentSector++;
}
}