mirror of
https://github.com/ReneLergner/WPinternals.git
synced 2026-06-19 13:50:11 +10:00
Initial commit - WPinternals 2.6
This commit is contained in:
Rene Lergner
@@ -0,0 +1,319 @@
|
||||
//
|
||||
// Copyright (c) 2008-2011, Kenneth Bell
|
||||
//
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a
|
||||
// copy of this software and associated documentation files (the "Software"),
|
||||
// to deal in the Software without restriction, including without limitation
|
||||
// the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
||||
// and/or sell copies of the Software, and to permit persons to whom the
|
||||
// Software is furnished to do so, subject to the following conditions:
|
||||
//
|
||||
// The above copyright notice and this permission notice shall be included in
|
||||
// all copies or substantial portions of the Software.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
||||
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
||||
// DEALINGS IN THE SOFTWARE.
|
||||
//
|
||||
|
||||
namespace DiscUtils.LogicalDiskManager
|
||||
{
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Globalization;
|
||||
using System.IO;
|
||||
using DiscUtils.Partitions;
|
||||
|
||||
internal class DynamicDiskGroup : IDiagnosticTraceable
|
||||
{
|
||||
private DiskGroupRecord _record;
|
||||
private Dictionary<Guid, DynamicDisk> _disks;
|
||||
private Database _database;
|
||||
|
||||
internal DynamicDiskGroup(VirtualDisk disk)
|
||||
{
|
||||
_disks = new Dictionary<Guid, DynamicDisk>();
|
||||
|
||||
DynamicDisk dynDisk = new DynamicDisk(disk);
|
||||
_database = dynDisk.Database;
|
||||
_disks.Add(dynDisk.Id, dynDisk);
|
||||
_record = dynDisk.Database.GetDiskGroup(dynDisk.GroupId);
|
||||
}
|
||||
|
||||
public void Add(VirtualDisk disk)
|
||||
{
|
||||
DynamicDisk dynDisk = new DynamicDisk(disk);
|
||||
_disks.Add(dynDisk.Id, dynDisk);
|
||||
}
|
||||
|
||||
#region IDiagnosticTraceable Members
|
||||
|
||||
public void Dump(TextWriter writer, string linePrefix)
|
||||
{
|
||||
writer.WriteLine(linePrefix + "DISK GROUP (" + _record.Name + ")");
|
||||
writer.WriteLine(linePrefix + " Name: " + _record.Name);
|
||||
writer.WriteLine(linePrefix + " Flags: 0x" + (_record.Flags & 0xFFF0).ToString("X4", CultureInfo.InvariantCulture));
|
||||
writer.WriteLine(linePrefix + " Database Id: " + _record.Id);
|
||||
writer.WriteLine(linePrefix + " Guid: " + _record.GroupGuidString);
|
||||
writer.WriteLine();
|
||||
|
||||
writer.WriteLine(linePrefix + " DISKS");
|
||||
foreach (var disk in _database.Disks)
|
||||
{
|
||||
writer.WriteLine(linePrefix + " DISK (" + disk.Name + ")");
|
||||
writer.WriteLine(linePrefix + " Name: " + disk.Name);
|
||||
writer.WriteLine(linePrefix + " Flags: 0x" + (disk.Flags & 0xFFF0).ToString("X4", CultureInfo.InvariantCulture));
|
||||
writer.WriteLine(linePrefix + " Database Id: " + disk.Id);
|
||||
writer.WriteLine(linePrefix + " Guid: " + disk.DiskGuidString);
|
||||
|
||||
DynamicDisk dynDisk;
|
||||
if (_disks.TryGetValue(new Guid(disk.DiskGuidString), out dynDisk))
|
||||
{
|
||||
writer.WriteLine(linePrefix + " PRIVATE HEADER");
|
||||
dynDisk.Dump(writer, linePrefix + " ");
|
||||
}
|
||||
}
|
||||
|
||||
writer.WriteLine(linePrefix + " VOLUMES");
|
||||
foreach (var vol in _database.Volumes)
|
||||
{
|
||||
writer.WriteLine(linePrefix + " VOLUME (" + vol.Name + ")");
|
||||
writer.WriteLine(linePrefix + " Name: " + vol.Name);
|
||||
writer.WriteLine(linePrefix + " BIOS Type: " + vol.BiosType.ToString("X2", CultureInfo.InvariantCulture) + " [" + BiosPartitionTypes.ToString(vol.BiosType) + "]");
|
||||
writer.WriteLine(linePrefix + " Flags: 0x" + (vol.Flags & 0xFFF0).ToString("X4", CultureInfo.InvariantCulture));
|
||||
writer.WriteLine(linePrefix + " Database Id: " + vol.Id);
|
||||
writer.WriteLine(linePrefix + " Guid: " + vol.VolumeGuid);
|
||||
writer.WriteLine(linePrefix + " State: " + vol.ActiveString);
|
||||
writer.WriteLine(linePrefix + " Drive Hint: " + vol.MountHint);
|
||||
writer.WriteLine(linePrefix + " Num Components: " + vol.ComponentCount);
|
||||
writer.WriteLine(linePrefix + " Link Id: " + vol.PartitionComponentLink);
|
||||
|
||||
writer.WriteLine(linePrefix + " COMPONENTS");
|
||||
foreach (var cmpnt in _database.GetVolumeComponents(vol.Id))
|
||||
{
|
||||
writer.WriteLine(linePrefix + " COMPONENT (" + cmpnt.Name + ")");
|
||||
writer.WriteLine(linePrefix + " Name: " + cmpnt.Name);
|
||||
writer.WriteLine(linePrefix + " Flags: 0x" + (cmpnt.Flags & 0xFFF0).ToString("X4", CultureInfo.InvariantCulture));
|
||||
writer.WriteLine(linePrefix + " Database Id: " + cmpnt.Id);
|
||||
writer.WriteLine(linePrefix + " State: " + cmpnt.StatusString);
|
||||
writer.WriteLine(linePrefix + " Mode: " + cmpnt.MergeType);
|
||||
writer.WriteLine(linePrefix + " Num Extents: " + cmpnt.NumExtents);
|
||||
writer.WriteLine(linePrefix + " Link Id: " + cmpnt.LinkId);
|
||||
writer.WriteLine(linePrefix + " Stripe Size: " + cmpnt.StripeSizeSectors + " (Sectors)");
|
||||
writer.WriteLine(linePrefix + " Stripe Stride: " + cmpnt.StripeStride);
|
||||
|
||||
writer.WriteLine(linePrefix + " EXTENTS");
|
||||
foreach (var extent in _database.GetComponentExtents(cmpnt.Id))
|
||||
{
|
||||
writer.WriteLine(linePrefix + " EXTENT (" + extent.Name + ")");
|
||||
writer.WriteLine(linePrefix + " Name: " + extent.Name);
|
||||
writer.WriteLine(linePrefix + " Flags: 0x" + (extent.Flags & 0xFFF0).ToString("X4", CultureInfo.InvariantCulture));
|
||||
writer.WriteLine(linePrefix + " Database Id: " + extent.Id);
|
||||
writer.WriteLine(linePrefix + " Disk Offset: " + extent.DiskOffsetLba + " (Sectors)");
|
||||
writer.WriteLine(linePrefix + " Volume Offset: " + extent.OffsetInVolumeLba + " (Sectors)");
|
||||
writer.WriteLine(linePrefix + " Size: " + extent.SizeLba + " (Sectors)");
|
||||
writer.WriteLine(linePrefix + " Component Id: " + extent.ComponentId);
|
||||
writer.WriteLine(linePrefix + " Disk Id: " + extent.DiskId);
|
||||
writer.WriteLine(linePrefix + " Link Id: " + extent.PartitionComponentLink);
|
||||
writer.WriteLine(linePrefix + " Interleave Order: " + extent.InterleaveOrder);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
|
||||
internal DynamicVolume[] GetVolumes()
|
||||
{
|
||||
List<DynamicVolume> vols = new List<DynamicVolume>();
|
||||
foreach (var record in _database.GetVolumes())
|
||||
{
|
||||
vols.Add(new DynamicVolume(this, record.VolumeGuid));
|
||||
}
|
||||
|
||||
return vols.ToArray();
|
||||
}
|
||||
|
||||
internal VolumeRecord GetVolume(Guid volume)
|
||||
{
|
||||
return _database.GetVolume(volume);
|
||||
}
|
||||
|
||||
internal LogicalVolumeStatus GetVolumeStatus(ulong volumeId)
|
||||
{
|
||||
return GetVolumeStatus(_database.GetVolume(volumeId));
|
||||
}
|
||||
|
||||
internal SparseStream OpenVolume(ulong volumeId)
|
||||
{
|
||||
return OpenVolume(_database.GetVolume(volumeId));
|
||||
}
|
||||
|
||||
private static int CompareExtentOffsets(ExtentRecord x, ExtentRecord y)
|
||||
{
|
||||
if (x.OffsetInVolumeLba > y.OffsetInVolumeLba)
|
||||
{
|
||||
return 1;
|
||||
}
|
||||
else if (x.OffsetInVolumeLba < y.OffsetInVolumeLba)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
private static int CompareExtentInterleaveOrder(ExtentRecord x, ExtentRecord y)
|
||||
{
|
||||
if (x.InterleaveOrder > y.InterleaveOrder)
|
||||
{
|
||||
return 1;
|
||||
}
|
||||
else if (x.InterleaveOrder < y.InterleaveOrder)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
private static LogicalVolumeStatus WorstOf(LogicalVolumeStatus x, LogicalVolumeStatus y)
|
||||
{
|
||||
return (LogicalVolumeStatus)Math.Max((int)x, (int)y);
|
||||
}
|
||||
|
||||
private LogicalVolumeStatus GetVolumeStatus(VolumeRecord volume)
|
||||
{
|
||||
int numFailed = 0;
|
||||
ulong numOK = 0;
|
||||
LogicalVolumeStatus worst = LogicalVolumeStatus.Healthy;
|
||||
foreach (var cmpnt in _database.GetVolumeComponents(volume.Id))
|
||||
{
|
||||
LogicalVolumeStatus cmpntStatus = GetComponentStatus(cmpnt);
|
||||
worst = WorstOf(worst, cmpntStatus);
|
||||
if (cmpntStatus == LogicalVolumeStatus.Failed)
|
||||
{
|
||||
numFailed++;
|
||||
}
|
||||
else
|
||||
{
|
||||
numOK++;
|
||||
}
|
||||
}
|
||||
|
||||
if (numOK < 1)
|
||||
{
|
||||
return LogicalVolumeStatus.Failed;
|
||||
}
|
||||
else if (numOK == volume.ComponentCount)
|
||||
{
|
||||
return worst;
|
||||
}
|
||||
else
|
||||
{
|
||||
return LogicalVolumeStatus.FailedRedundancy;
|
||||
}
|
||||
}
|
||||
|
||||
private LogicalVolumeStatus GetComponentStatus(ComponentRecord cmpnt)
|
||||
{
|
||||
// NOTE: no support for RAID, so either valid or failed...
|
||||
LogicalVolumeStatus status = LogicalVolumeStatus.Healthy;
|
||||
|
||||
foreach (var extent in _database.GetComponentExtents(cmpnt.Id))
|
||||
{
|
||||
DiskRecord disk = _database.GetDisk(extent.DiskId);
|
||||
if (!_disks.ContainsKey(new Guid(disk.DiskGuidString)))
|
||||
{
|
||||
status = LogicalVolumeStatus.Failed;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
return status;
|
||||
}
|
||||
|
||||
private SparseStream OpenExtent(ExtentRecord extent)
|
||||
{
|
||||
DiskRecord disk = _database.GetDisk(extent.DiskId);
|
||||
|
||||
DynamicDisk diskObj = _disks[new Guid(disk.DiskGuidString)];
|
||||
|
||||
return new SubStream(diskObj.Content, Ownership.None, (diskObj.DataOffset + extent.DiskOffsetLba) * Sizes.Sector, extent.SizeLba * Sizes.Sector);
|
||||
}
|
||||
|
||||
private SparseStream OpenComponent(ComponentRecord component)
|
||||
{
|
||||
if (component.MergeType == ExtentMergeType.Concatenated)
|
||||
{
|
||||
List<ExtentRecord> extents = new List<ExtentRecord>(_database.GetComponentExtents(component.Id));
|
||||
extents.Sort(CompareExtentOffsets);
|
||||
|
||||
// Sanity Check...
|
||||
long pos = 0;
|
||||
foreach (var extent in extents)
|
||||
{
|
||||
if (extent.OffsetInVolumeLba != pos)
|
||||
{
|
||||
throw new IOException("Volume extents are non-contiguous");
|
||||
}
|
||||
|
||||
pos += extent.SizeLba;
|
||||
}
|
||||
|
||||
List<SparseStream> streams = new List<SparseStream>();
|
||||
foreach (var extent in extents)
|
||||
{
|
||||
streams.Add(OpenExtent(extent));
|
||||
}
|
||||
|
||||
return new ConcatStream(Ownership.Dispose, streams.ToArray());
|
||||
}
|
||||
else if (component.MergeType == ExtentMergeType.Interleaved)
|
||||
{
|
||||
List<ExtentRecord> extents = new List<ExtentRecord>(_database.GetComponentExtents(component.Id));
|
||||
extents.Sort(CompareExtentInterleaveOrder);
|
||||
|
||||
List<SparseStream> streams = new List<SparseStream>();
|
||||
foreach (var extent in extents)
|
||||
{
|
||||
streams.Add(OpenExtent(extent));
|
||||
}
|
||||
|
||||
return new StripedStream(component.StripeSizeSectors * Sizes.Sector, Ownership.Dispose, streams.ToArray());
|
||||
}
|
||||
else
|
||||
{
|
||||
throw new NotImplementedException("Unknown component mode: " + component.MergeType);
|
||||
}
|
||||
}
|
||||
|
||||
private SparseStream OpenVolume(VolumeRecord volume)
|
||||
{
|
||||
List<SparseStream> cmpntStreams = new List<SparseStream>();
|
||||
foreach (var component in _database.GetVolumeComponents(volume.Id))
|
||||
{
|
||||
if (GetComponentStatus(component) == LogicalVolumeStatus.Healthy)
|
||||
{
|
||||
cmpntStreams.Add(OpenComponent(component));
|
||||
}
|
||||
}
|
||||
|
||||
if (cmpntStreams.Count < 1)
|
||||
{
|
||||
throw new IOException("Volume with no associated or healthy components");
|
||||
}
|
||||
else if (cmpntStreams.Count == 1)
|
||||
{
|
||||
return cmpntStreams[0];
|
||||
}
|
||||
else
|
||||
{
|
||||
return new MirrorStream(Ownership.Dispose, cmpntStreams.ToArray());
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user