static internal async Task <RegionTableEntry> SetupAddRegion(this SessionRepository repository, Guid sessionId, Guid regionId, Guid continentId, String regionName, UInt32 cardValue)
{
var dataTable = repository.GetTableForSessionData(sessionId);
RegionTableEntry newRegionEntry = new RegionTableEntry(sessionId, regionId, continentId, regionName, cardValue);
TableOperation insertOperation = TableOperation.InsertOrReplace(newRegionEntry);
await dataTable.ExecuteAsync(insertOperation);
return(newRegionEntry);
}
public async Task IntegrationTestCreateRegion()
{
// Arrange
RegionRepository repository = new RegionRepository(DevelopmentStorageAccountConnectionString, String.Empty);
Guid dummySessionId = new Guid("74720766-452A-40AD-8A61-FEF07E8573C9");
Guid dummyRegionId = new Guid("024D1E45-EF34-4AB1-840D-79229CCDE8C3");
Guid dummyContinentId = new Guid("DE167712-0CE6-455C-83EA-CB2A6936F1BE");
List <Guid> dummyConnections = new List <Guid> {
new Guid("0533203F-13F2-4863-B528-17F53D279E19"), new Guid("4A9779D0-0727-4AD9-AD66-17AE9AF9BE02")
};
var dataTable = TableClient.SetupSessionDataTable(dummySessionId);
// Act
using (IBatchOperationHandle batchOperation = new BatchOperationHandle(SessionRepository.GetTableForSessionData(TableClient, dummySessionId)))
{
repository.CreateRegion(batchOperation, dummySessionId, dummyRegionId, dummyContinentId, "DummyRegion", dummyConnections, 3);
}
// Assert
TableOperation operation = TableOperation.Retrieve <RegionTableEntry>(dummySessionId.ToString(), "Region_" + dummyRegionId.ToString());
TableResult result = await dataTable.ExecuteAsync(operation);
Assert.IsNotNull(result.Result);
Assert.IsInstanceOfType(result.Result, typeof(RegionTableEntry));
RegionTableEntry resultStronglyTyped = result.Result as RegionTableEntry;
Assert.AreEqual(dummySessionId, resultStronglyTyped.SessionId);
Assert.AreEqual(dummyRegionId, resultStronglyTyped.RegionId);
Assert.AreEqual(dummyContinentId, resultStronglyTyped.ContinentId);
Assert.AreEqual("DummyRegion", resultStronglyTyped.Name);
Assert.AreEqual(String.Empty, resultStronglyTyped.OwnerId);
Assert.AreEqual(3U, resultStronglyTyped.CardValue);
Assert.AreEqual(0, resultStronglyTyped.StoredTroopCount);
Assert.IsTrue(resultStronglyTyped.ETag.Length > 0);
Assert.IsTrue(resultStronglyTyped.ConnectedRegions.Contains(dummyConnections[0]));
TableOperation cardOperation = TableOperation.Retrieve <CardTableEntry>(dummySessionId.ToString(), "Card_" + dummyRegionId.ToString());
result = await dataTable.ExecuteAsync(cardOperation);
Assert.IsNotNull(result.Result);
Assert.IsInstanceOfType(result.Result, typeof(CardTableEntry));
CardTableEntry cardStronglyTyped = result.Result as CardTableEntry;
Assert.AreEqual(dummySessionId, cardStronglyTyped.SessionId);
Assert.AreEqual(dummyRegionId, cardStronglyTyped.RegionId);
Assert.AreEqual(3U, cardStronglyTyped.Value);
Assert.AreEqual(String.Empty, cardStronglyTyped.OwnerId);
Assert.AreEqual(CardTableEntry.State.Unowned, cardStronglyTyped.OwnerState);
}
public void CommitTroopsToPhase(IBatchOperationHandle batchOperationHandleInterface, IRegionData sourceRegion, UInt32 troopsToCommit)
{
BatchOperationHandle batchOperationHandle = batchOperationHandleInterface as BatchOperationHandle;
RegionTableEntry region = sourceRegion as RegionTableEntry;
if (region != null && region.TroopCount > region.TroopsCommittedToPhase + troopsToCommit)
{
region.StoredTroopsCommittedToPhase += (int)troopsToCommit;
batchOperationHandle.BatchOperation.Replace(region);
}
else
{
throw new InvalidOperationException();
}
}
public void CreateRegion(IBatchOperationHandle batchOperationHandleInterface, Guid sessionId, Guid regionId, Guid continentId, String name, IEnumerable <Guid> connectedRegions, UInt32 cardValue)
{
BatchOperationHandle batchOperationHandle = batchOperationHandleInterface as BatchOperationHandle;
// Create a new table entry
RegionTableEntry newRegion = new RegionTableEntry(sessionId, regionId, continentId, name, cardValue);
newRegion.IsValid();
newRegion.SetConnectedRegions(connectedRegions);
batchOperationHandle.BatchOperation.Insert(newRegion);
// Create a new card for this region
CardTableEntry newCard = new CardTableEntry(sessionId, regionId);
newCard.IsValid();
newCard.ValueRaw = (Int32)cardValue;
batchOperationHandle.BatchOperation.Insert(newCard);
}
public void AssignRegionOwnership(IBatchOperationHandle batchOperationHandleInterface, IEnumerable <IRegionData> regions, Dictionary <Guid, OwnershipChange> ownershipChanges)
{
BatchOperationHandle batchOperationHandle = batchOperationHandleInterface as BatchOperationHandle;
// Get NationTableEntry for every player that needs updating
var updateOperations = from ownershipChange in ownershipChanges
join region in regions.Cast <RegionTableEntry>() on ownershipChange.Key equals region.RegionId
select new { Region = region, NewOwner = ownershipChange.Value.UserId, NewTroopCount = (Int32)ownershipChange.Value.TroopCount };
// Modify entries as required
foreach (var operation in updateOperations)
{
RegionTableEntry regionEntry = operation.Region;
regionEntry.IsValid();
regionEntry.OwnerId = operation.NewOwner;
regionEntry.StoredTroopCount = operation.NewTroopCount;
regionEntry.StoredTroopsCommittedToPhase = 0;
batchOperationHandle.BatchOperation.Replace(regionEntry);
}
}
public bool Open(IFilter imageFilter)
{
Stream stream = imageFilter.GetDataForkStream();
stream.Seek(0, SeekOrigin.Begin);
if (stream.Length < 512)
{
return(false);
}
byte[] vhdxIdB = new byte[Marshal.SizeOf <Identifier>()];
stream.Read(vhdxIdB, 0, Marshal.SizeOf <Identifier>());
_id = Marshal.ByteArrayToStructureLittleEndian <Identifier>(vhdxIdB);
if (_id.signature != VHDX_SIGNATURE)
{
return(false);
}
_imageInfo.Application = Encoding.Unicode.GetString(_id.creator);
stream.Seek(64 * 1024, SeekOrigin.Begin);
byte[] vHdrB = new byte[Marshal.SizeOf <Header>()];
stream.Read(vHdrB, 0, Marshal.SizeOf <Header>());
_vHdr = Marshal.ByteArrayToStructureLittleEndian <Header>(vHdrB);
if (_vHdr.Signature != VHDX_HEADER_SIG)
{
stream.Seek(128 * 1024, SeekOrigin.Begin);
vHdrB = new byte[Marshal.SizeOf <Header>()];
stream.Read(vHdrB, 0, Marshal.SizeOf <Header>());
_vHdr = Marshal.ByteArrayToStructureLittleEndian <Header>(vHdrB);
if (_vHdr.Signature != VHDX_HEADER_SIG)
{
throw new ImageNotSupportedException("VHDX header not found");
}
}
stream.Seek(192 * 1024, SeekOrigin.Begin);
byte[] vRegTableB = new byte[Marshal.SizeOf <RegionTableHeader>()];
stream.Read(vRegTableB, 0, Marshal.SizeOf <RegionTableHeader>());
_vRegHdr = Marshal.ByteArrayToStructureLittleEndian <RegionTableHeader>(vRegTableB);
if (_vRegHdr.signature != VHDX_REGION_SIG)
{
stream.Seek(256 * 1024, SeekOrigin.Begin);
vRegTableB = new byte[Marshal.SizeOf <RegionTableHeader>()];
stream.Read(vRegTableB, 0, Marshal.SizeOf <RegionTableHeader>());
_vRegHdr = Marshal.ByteArrayToStructureLittleEndian <RegionTableHeader>(vRegTableB);
if (_vRegHdr.signature != VHDX_REGION_SIG)
{
throw new ImageNotSupportedException("VHDX region table not found");
}
}
_vRegs = new RegionTableEntry[_vRegHdr.entries];
for (int i = 0; i < _vRegs.Length; i++)
{
byte[] vRegB = new byte[System.Runtime.InteropServices.Marshal.SizeOf(_vRegs[i])];
stream.Read(vRegB, 0, System.Runtime.InteropServices.Marshal.SizeOf(_vRegs[i]));
_vRegs[i] = Marshal.ByteArrayToStructureLittleEndian <RegionTableEntry>(vRegB);
if (_vRegs[i].guid == _batGuid)
{
_batOffset = (long)_vRegs[i].offset;
}
else if (_vRegs[i].guid == _metadataGuid)
{
_metadataOffset = (long)_vRegs[i].offset;
}
else if ((_vRegs[i].flags & REGION_FLAGS_REQUIRED) == REGION_FLAGS_REQUIRED)
{
throw new
ImageNotSupportedException($"Found unsupported and required region Guid {_vRegs[i].guid}, not proceeding with image.");
}
}
if (_batOffset == 0)
{
throw new Exception("BAT not found, cannot continue.");
}
if (_metadataOffset == 0)
{
throw new Exception("Metadata not found, cannot continue.");
}
uint fileParamsOff = 0, vdSizeOff = 0, p83Off = 0, logOff = 0, physOff = 0, parentOff = 0;
stream.Seek(_metadataOffset, SeekOrigin.Begin);
byte[] metTableB = new byte[Marshal.SizeOf <MetadataTableHeader>()];
stream.Read(metTableB, 0, Marshal.SizeOf <MetadataTableHeader>());
_vMetHdr = Marshal.ByteArrayToStructureLittleEndian <MetadataTableHeader>(metTableB);
_vMets = new MetadataTableEntry[_vMetHdr.entries];
for (int i = 0; i < _vMets.Length; i++)
{
byte[] vMetB = new byte[System.Runtime.InteropServices.Marshal.SizeOf(_vMets[i])];
stream.Read(vMetB, 0, System.Runtime.InteropServices.Marshal.SizeOf(_vMets[i]));
_vMets[i] = Marshal.ByteArrayToStructureLittleEndian <MetadataTableEntry>(vMetB);
if (_vMets[i].itemId == _fileParametersGuid)
{
fileParamsOff = _vMets[i].offset;
}
else if (_vMets[i].itemId == _virtualDiskSizeGuid)
{
vdSizeOff = _vMets[i].offset;
}
else if (_vMets[i].itemId == _page83DataGuid)
{
p83Off = _vMets[i].offset;
}
else if (_vMets[i].itemId == _logicalSectorSizeGuid)
{
logOff = _vMets[i].offset;
}
else if (_vMets[i].itemId == _physicalSectorSizeGuid)
{
physOff = _vMets[i].offset;
}
else if (_vMets[i].itemId == _parentLocatorGuid)
{
parentOff = _vMets[i].offset;
}
else if ((_vMets[i].flags & METADATA_FLAGS_REQUIRED) == METADATA_FLAGS_REQUIRED)
{
throw new
ImageNotSupportedException($"Found unsupported and required metadata Guid {_vMets[i].itemId}, not proceeding with image.");
}
}
byte[] tmp;
if (fileParamsOff != 0)
{
stream.Seek(fileParamsOff + _metadataOffset, SeekOrigin.Begin);
tmp = new byte[8];
stream.Read(tmp, 0, 8);
_vFileParms = new FileParameters
{
blockSize = BitConverter.ToUInt32(tmp, 0),
flags = BitConverter.ToUInt32(tmp, 4)
};
}
else
{
throw new Exception("File parameters not found.");
}
if (vdSizeOff != 0)
{
stream.Seek(vdSizeOff + _metadataOffset, SeekOrigin.Begin);
tmp = new byte[8];
stream.Read(tmp, 0, 8);
_virtualDiskSize = BitConverter.ToUInt64(tmp, 0);
}
else
{
throw new Exception("Virtual disk size not found.");
}
if (p83Off != 0)
{
stream.Seek(p83Off + _metadataOffset, SeekOrigin.Begin);
tmp = new byte[16];
stream.Read(tmp, 0, 16);
_page83Data = new Guid(tmp);
}
if (logOff != 0)
{
stream.Seek(logOff + _metadataOffset, SeekOrigin.Begin);
tmp = new byte[4];
stream.Read(tmp, 0, 4);
_logicalSectorSize = BitConverter.ToUInt32(tmp, 0);
}
else
{
throw new Exception("Logical sector size not found.");
}
if (physOff != 0)
{
stream.Seek(physOff + _metadataOffset, SeekOrigin.Begin);
tmp = new byte[4];
stream.Read(tmp, 0, 4);
_physicalSectorSize = BitConverter.ToUInt32(tmp, 0);
}
else
{
throw new Exception("Physical sector size not found.");
}
if (parentOff != 0 &&
(_vFileParms.flags & FILE_FLAGS_HAS_PARENT) == FILE_FLAGS_HAS_PARENT)
{
stream.Seek(parentOff + _metadataOffset, SeekOrigin.Begin);
byte[] vParHdrB = new byte[Marshal.SizeOf <ParentLocatorHeader>()];
stream.Read(vParHdrB, 0, Marshal.SizeOf <ParentLocatorHeader>());
_vParHdr = Marshal.ByteArrayToStructureLittleEndian <ParentLocatorHeader>(vParHdrB);
if (_vParHdr.locatorType != _parentTypeVhdxGuid)
{
throw new
ImageNotSupportedException($"Found unsupported and required parent locator type {_vParHdr.locatorType}, not proceeding with image.");
}
_vPars = new ParentLocatorEntry[_vParHdr.keyValueCount];
for (int i = 0; i < _vPars.Length; i++)
{
byte[] vParB = new byte[System.Runtime.InteropServices.Marshal.SizeOf(_vPars[i])];
stream.Read(vParB, 0, System.Runtime.InteropServices.Marshal.SizeOf(_vPars[i]));
_vPars[i] = Marshal.ByteArrayToStructureLittleEndian <ParentLocatorEntry>(vParB);
}
}
else if ((_vFileParms.flags & FILE_FLAGS_HAS_PARENT) == FILE_FLAGS_HAS_PARENT)
{
throw new Exception("Parent locator not found.");
}
if ((_vFileParms.flags & FILE_FLAGS_HAS_PARENT) == FILE_FLAGS_HAS_PARENT &&
_vParHdr.locatorType == _parentTypeVhdxGuid)
{
_parentImage = new Vhdx();
bool parentWorks = false;
foreach (ParentLocatorEntry parentEntry in _vPars)
{
stream.Seek(parentEntry.keyOffset + _metadataOffset, SeekOrigin.Begin);
byte[] tmpKey = new byte[parentEntry.keyLength];
stream.Read(tmpKey, 0, tmpKey.Length);
string entryType = Encoding.Unicode.GetString(tmpKey);
IFilter parentFilter;
if (string.Compare(entryType, RELATIVE_PATH_KEY, StringComparison.OrdinalIgnoreCase) == 0)
{
stream.Seek(parentEntry.valueOffset + _metadataOffset, SeekOrigin.Begin);
byte[] tmpVal = new byte[parentEntry.valueLength];
stream.Read(tmpVal, 0, tmpVal.Length);
string entryValue = Encoding.Unicode.GetString(tmpVal);
try
{
parentFilter =
new FiltersList().GetFilter(Path.Combine(imageFilter.GetParentFolder(), entryValue));
if (parentFilter != null &&
_parentImage.Open(parentFilter))
{
parentWorks = true;
break;
}
}
catch
{
parentWorks = false;
}
string relEntry = Path.Combine(Path.GetDirectoryName(imageFilter.GetPath()), entryValue);
try
{
parentFilter =
new FiltersList().GetFilter(Path.Combine(imageFilter.GetParentFolder(), relEntry));
if (parentFilter == null ||
!_parentImage.Open(parentFilter))
{
continue;
}
parentWorks = true;
break;
}
catch
{
// ignored
}
}
else if (string.Compare(entryType, VOLUME_PATH_KEY, StringComparison.OrdinalIgnoreCase) == 0 ||
string.Compare(entryType, ABSOLUTE_WIN32_PATH_KEY, StringComparison.OrdinalIgnoreCase) == 0)
{
stream.Seek(parentEntry.valueOffset + _metadataOffset, SeekOrigin.Begin);
byte[] tmpVal = new byte[parentEntry.valueLength];
stream.Read(tmpVal, 0, tmpVal.Length);
string entryValue = Encoding.Unicode.GetString(tmpVal);
try
{
parentFilter =
new FiltersList().GetFilter(Path.Combine(imageFilter.GetParentFolder(), entryValue));
if (parentFilter == null ||
!_parentImage.Open(parentFilter))
{
continue;
}
parentWorks = true;
break;
}
catch
{
// ignored
}
}
}
if (!parentWorks)
{
throw new Exception("Image is differential but parent cannot be opened.");
}
_hasParent = true;
}
_chunkRatio = (long)((Math.Pow(2, 23) * _logicalSectorSize) / _vFileParms.blockSize);
_dataBlocks = _virtualDiskSize / _vFileParms.blockSize;
if (_virtualDiskSize % _vFileParms.blockSize > 0)
{
_dataBlocks++;
}
long batEntries;
if (_hasParent)
{
long sectorBitmapBlocks = (long)_dataBlocks / _chunkRatio;
if (_dataBlocks % (ulong)_chunkRatio > 0)
{
sectorBitmapBlocks++;
}
_sectorBitmapPointers = new ulong[sectorBitmapBlocks];
batEntries = sectorBitmapBlocks * (_chunkRatio - 1);
}
else
{
batEntries = (long)(_dataBlocks + ((_dataBlocks - 1) / (ulong)_chunkRatio));
}
AaruConsole.DebugWriteLine("VHDX plugin", "Reading BAT");
long readChunks = 0;
_blockAllocationTable = new ulong[_dataBlocks];
byte[] batB = new byte[batEntries * 8];
stream.Seek(_batOffset, SeekOrigin.Begin);
stream.Read(batB, 0, batB.Length);
ulong skipSize = 0;
for (ulong i = 0; i < _dataBlocks; i++)
{
if (readChunks == _chunkRatio)
{
if (_hasParent)
{
_sectorBitmapPointers[skipSize / 8] = BitConverter.ToUInt64(batB, (int)((i * 8) + skipSize));
}
readChunks = 0;
skipSize += 8;
}
else
{
_blockAllocationTable[i] = BitConverter.ToUInt64(batB, (int)((i * 8) + skipSize));
readChunks++;
}
}
if (_hasParent)
{
AaruConsole.DebugWriteLine("VHDX plugin", "Reading Sector Bitmap");
var sectorBmpMs = new MemoryStream();
foreach (ulong pt in _sectorBitmapPointers)
{
switch (pt & BAT_FLAGS_MASK)
{
case SECTOR_BITMAP_NOT_PRESENT:
sectorBmpMs.Write(new byte[1048576], 0, 1048576);
break;
case SECTOR_BITMAP_PRESENT:
stream.Seek((long)((pt & BAT_FILE_OFFSET_MASK) * 1048576), SeekOrigin.Begin);
byte[] bmp = new byte[1048576];
stream.Read(bmp, 0, bmp.Length);
sectorBmpMs.Write(bmp, 0, bmp.Length);
break;
default:
if ((pt & BAT_FLAGS_MASK) != 0)
{
throw new
ImageNotSupportedException($"Unsupported sector bitmap block flags (0x{pt & BAT_FLAGS_MASK:X16}) found, not proceeding.");
}
break;
}
}
_sectorBitmap = sectorBmpMs.ToArray();
sectorBmpMs.Close();
}
_maxBlockCache = (int)(MAX_CACHE_SIZE / _vFileParms.blockSize);
_maxSectorCache = (int)(MAX_CACHE_SIZE / _logicalSectorSize);
_imageStream = stream;
_sectorCache = new Dictionary <ulong, byte[]>();
_blockCache = new Dictionary <ulong, byte[]>();
_imageInfo.CreationTime = imageFilter.GetCreationTime();
_imageInfo.LastModificationTime = imageFilter.GetLastWriteTime();
_imageInfo.MediaTitle = Path.GetFileNameWithoutExtension(imageFilter.GetFilename());
_imageInfo.SectorSize = _logicalSectorSize;
_imageInfo.XmlMediaType = XmlMediaType.BlockMedia;
_imageInfo.MediaType = MediaType.GENERIC_HDD;
_imageInfo.ImageSize = _virtualDiskSize;
_imageInfo.Sectors = _imageInfo.ImageSize / _imageInfo.SectorSize;
_imageInfo.DriveSerialNumber = _page83Data.ToString();
// TODO: Separate image application from version, need several samples.
_imageInfo.Cylinders = (uint)(_imageInfo.Sectors / 16 / 63);
_imageInfo.Heads = 16;
_imageInfo.SectorsPerTrack = 63;
return(true);
}
