public static void TestClone()
{
UnrecognizedHeaderBlock unrecognizedHeaderBlock = new UnrecognizedHeaderBlock(HeaderBlockType.Unrecognized, new byte[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16 });
UnrecognizedHeaderBlock cloned = (UnrecognizedHeaderBlock)unrecognizedHeaderBlock.Clone();
Assert.That(unrecognizedHeaderBlock, Is.Not.SameAs(cloned), "The clone should not be the same reference.");
Assert.That(cloned.GetDataBlockBytes(), Is.EquivalentTo(unrecognizedHeaderBlock.GetDataBlockBytes()), "The clone should have equivalent data block bytes as the original.");
}
private void ParseHeaderBlock(HeaderBlockType headerBlockType, byte[] dataBlock)
{
bool isFirst = CurrentItemType == AxCryptItemType.MagicGuid;
CurrentItemType = AxCryptItemType.HeaderBlock;
if (headerBlockType == HeaderBlockType.Preamble)
{
if (!isFirst)
{
throw new FileFormatException("Preamble can only be first.", ErrorStatus.FileFormatError);
}
CurrentHeaderBlock = new PreambleHeaderBlock(dataBlock);
_sendDataToHmacStream = true;
return;
}
else
{
if (isFirst)
{
throw new FileFormatException("Preamble must be first.", ErrorStatus.FileFormatError);
}
}
switch (headerBlockType)
{
case HeaderBlockType.Version:
CurrentHeaderBlock = new VersionHeaderBlock(dataBlock);
break;
case HeaderBlockType.KeyWrap1:
CurrentHeaderBlock = new KeyWrap1HeaderBlock(dataBlock);
break;
case HeaderBlockType.KeyWrap2:
CurrentHeaderBlock = new KeyWrap2HeaderBlock(dataBlock);
break;
case HeaderBlockType.IdTag:
CurrentHeaderBlock = new IdTagHeaderBlock(dataBlock);
break;
case HeaderBlockType.Data:
CurrentHeaderBlock = new DataHeaderBlock(dataBlock);
break;
case HeaderBlockType.FileNameInfo:
CurrentHeaderBlock = new FileNameInfoHeaderBlock(dataBlock);
break;
case HeaderBlockType.EncryptionInfo:
CurrentHeaderBlock = new EncryptionInfoHeaderBlock(dataBlock);
break;
case HeaderBlockType.CompressionInfo:
CurrentHeaderBlock = new CompressionInfoHeaderBlock(dataBlock);
break;
case HeaderBlockType.FileInfo:
CurrentHeaderBlock = new FileInfoHeaderBlock(dataBlock);
break;
case HeaderBlockType.Compression:
CurrentHeaderBlock = new CompressionHeaderBlock(dataBlock);
break;
case HeaderBlockType.UnicodeFileNameInfo:
CurrentHeaderBlock = new UnicodeFileNameInfoHeaderBlock(dataBlock);
break;
case HeaderBlockType.Encrypted:
case HeaderBlockType.None:
case HeaderBlockType.Any:
throw new FileFormatException("Illegal header block type.", ErrorStatus.FileFormatError);
default:
CurrentHeaderBlock = new UnrecognizedHeaderBlock(headerBlockType, dataBlock);
break;
}
if (_sendDataToHmacStream)
{
CurrentHeaderBlock.Write(_hmacBufferStream);
}
return;
}
private void ParseHeaderBlock(HeaderBlockType headerBlockType, byte[] dataBlock)
{
bool isFirst = CurrentItemType == AxCryptItemType.MagicGuid;
CurrentItemType = AxCryptItemType.HeaderBlock;
if (headerBlockType == HeaderBlockType.Preamble)
{
if (!isFirst)
{
throw new FileFormatException("Preamble can only be first.", ErrorStatus.FileFormatError);
}
CurrentHeaderBlock = new PreambleHeaderBlock(dataBlock);
_sendDataToHmacStream = true;
return;
}
else
{
if (isFirst)
{
throw new FileFormatException("Preamble must be first.", ErrorStatus.FileFormatError);
}
}
switch (headerBlockType)
{
case HeaderBlockType.Version:
CurrentHeaderBlock = new VersionHeaderBlock(dataBlock);
break;
case HeaderBlockType.KeyWrap1:
CurrentHeaderBlock = new KeyWrap1HeaderBlock(dataBlock);
break;
case HeaderBlockType.KeyWrap2:
CurrentHeaderBlock = new KeyWrap2HeaderBlock(dataBlock);
break;
case HeaderBlockType.IdTag:
CurrentHeaderBlock = new IdTagHeaderBlock(dataBlock);
break;
case HeaderBlockType.Data:
CurrentHeaderBlock = new DataHeaderBlock(dataBlock);
break;
case HeaderBlockType.FileNameInfo:
CurrentHeaderBlock = new FileNameInfoHeaderBlock(dataBlock);
break;
case HeaderBlockType.EncryptionInfo:
CurrentHeaderBlock = new EncryptionInfoHeaderBlock(dataBlock);
break;
case HeaderBlockType.CompressionInfo:
CurrentHeaderBlock = new CompressionInfoHeaderBlock(dataBlock);
break;
case HeaderBlockType.FileInfo:
CurrentHeaderBlock = new FileInfoHeaderBlock(dataBlock);
break;
case HeaderBlockType.Compression:
CurrentHeaderBlock = new CompressionHeaderBlock(dataBlock);
break;
case HeaderBlockType.UnicodeFileNameInfo:
CurrentHeaderBlock = new UnicodeFileNameInfoHeaderBlock(dataBlock);
break;
case HeaderBlockType.Encrypted:
case HeaderBlockType.None:
case HeaderBlockType.Any:
throw new FileFormatException("Illegal header block type.", ErrorStatus.FileFormatError);
default:
CurrentHeaderBlock = new UnrecognizedHeaderBlock(headerBlockType, dataBlock);
break;
}
if (_sendDataToHmacStream)
{
CurrentHeaderBlock.Write(_hmacBufferStream);
}
return;
}
public static void TestUnrecognizedHeaderBlock()
{
using (MemoryStream inputStream = new MemoryStream())
{
AxCrypt1Guid.Write(inputStream);
PreambleHeaderBlock preambleHeaderBlock = new PreambleHeaderBlock();
preambleHeaderBlock.Write(inputStream);
UnrecognizedHeaderBlock unrecognizedHeaderBlock = new UnrecognizedHeaderBlock(HeaderBlockType.Unrecognized, new byte[0]);
unrecognizedHeaderBlock.Write(inputStream);
inputStream.Position = 0;
using (AxCryptReader axCryptReader = AxCryptReader.Create(inputStream))
{
Assert.That(axCryptReader.Read(), Is.True, "We should be able to read the Guid");
Assert.That(axCryptReader.CurrentItemType, Is.EqualTo(AxCryptItemType.MagicGuid), "We're expecting to have found a MagicGuid");
Assert.That(axCryptReader.Read(), Is.True, "We should be able to read the next HeaderBlock");
Assert.That(axCryptReader.CurrentItemType, Is.EqualTo(AxCryptItemType.HeaderBlock), "We're expecting to have found a HeaderBlock");
Assert.That(axCryptReader.CurrentHeaderBlock.HeaderBlockType, Is.EqualTo(HeaderBlockType.Preamble), "We're expecting to have found a Preamble specifically");
Assert.That(axCryptReader.Read(), Is.True, "We should be able to read the next HeaderBlock");
Assert.That(axCryptReader.CurrentItemType, Is.EqualTo(AxCryptItemType.HeaderBlock), "We're expecting to have found a HeaderBlock");
Assert.That(axCryptReader.CurrentHeaderBlock.HeaderBlockType, Is.EqualTo(HeaderBlockType.Unrecognized), "We're expecting to have found an unrecognized block specifically");
}
}
}
