private void WriteRecord(int i, BiosPartitionRecord newRecord)
{
_diskData.Position = 0;
byte[] bootSector = Utilities.ReadFully(_diskData, Utilities.SectorSize);
newRecord.WriteTo(bootSector, 0x01BE + (i * 16));
_diskData.Position = 0;
_diskData.Write(bootSector, 0, bootSector.Length);
}
private static BiosPartitionRecord[] ReadPrimaryRecords(byte[] bootSector)
{
BiosPartitionRecord[] records = new BiosPartitionRecord[4];
for (int i = 0; i < 4; ++i)
{
records[i] = new BiosPartitionRecord(bootSector, 0x01BE + (i * 0x10), 0, i);
}
return(records);
}
public BiosPartitionRecord[] GetPartitions()
{
List <BiosPartitionRecord> result = new List <BiosPartitionRecord>();
uint partPos = _firstSector;
while (partPos != 0)
{
_disk.Position = ((long)partPos) * Utilities.SectorSize;
byte[] sector = Utilities.ReadFully(_disk, Utilities.SectorSize);
if (sector[510] != 0x55 || sector[511] != 0xAA)
{
throw new IOException("Invalid extended partition sector");
}
uint nextPartPos = 0;
for (int offset = 0x1BE; offset <= 0x1EE; offset += 0x10)
{
BiosPartitionRecord thisPart = new BiosPartitionRecord(sector, offset, partPos, -1);
if (thisPart.StartCylinder != 0 || thisPart.StartHead != 0 || thisPart.StartSector != 0)
{
if (thisPart.PartitionType != 0x05 && thisPart.PartitionType != 0x0F)
{
result.Add(thisPart);
}
else
{
nextPartPos = _firstSector + thisPart.LBAStart;
}
}
}
partPos = nextPartPos;
}
return(result.ToArray());
}
/// <summary>
/// Gets all of the disk ranges containing partition table data.
/// </summary>
/// <returns>Set of stream extents, indicated as byte offset from the start of the disk.</returns>
public IEnumerable <StreamExtent> GetMetadataDiskExtents()
{
List <StreamExtent> extents = new List <StreamExtent>();
uint partPos = _firstSector;
while (partPos != 0)
{
extents.Add(new StreamExtent(((long)partPos) * Utilities.SectorSize, Utilities.SectorSize));
_disk.Position = ((long)partPos) * Utilities.SectorSize;
byte[] sector = Utilities.ReadFully(_disk, Utilities.SectorSize);
if (sector[510] != 0x55 || sector[511] != 0xAA)
{
throw new IOException("Invalid extended partition sector");
}
uint nextPartPos = 0;
for (int offset = 0x1BE; offset <= 0x1EE; offset += 0x10)
{
BiosPartitionRecord thisPart = new BiosPartitionRecord(sector, offset, partPos, -1);
if (thisPart.StartCylinder != 0 || thisPart.StartHead != 0 || thisPart.StartSector != 0)
{
if (thisPart.PartitionType == 0x05 || thisPart.PartitionType == 0x0F)
{
nextPartPos = _firstSector + thisPart.LBAStart;
}
}
}
partPos = nextPartPos;
}
return(extents);
}
/// <summary>
/// Updates the CHS fields in partition records to reflect a new BIOS geometry.
/// </summary>
/// <param name="geometry">The disk's new BIOS geometry.</param>
/// <remarks>The partitions are not relocated to a cylinder boundary, just the CHS fields are updated on the
/// assumption the LBA fields are definitive.</remarks>
public void UpdateBiosGeometry(Geometry geometry)
{
_diskData.Position = 0;
byte[] bootSector = Utilities.ReadFully(_diskData, Utilities.SectorSize);
BiosPartitionRecord[] records = ReadPrimaryRecords(bootSector);
for (int i = 0; i < records.Length; ++i)
{
BiosPartitionRecord record = records[i];
if (record.IsValid)
{
ChsAddress newStartAddress = geometry.ToChsAddress(record.LBAStartAbsolute);
if (newStartAddress.Cylinder > 1023)
{
newStartAddress = new ChsAddress(1023, geometry.HeadsPerCylinder - 1, geometry.SectorsPerTrack);
}
ChsAddress newEndAddress = geometry.ToChsAddress(record.LBAStartAbsolute + record.LBALength - 1);
if (newEndAddress.Cylinder > 1023)
{
newEndAddress = new ChsAddress(1023, geometry.HeadsPerCylinder - 1, geometry.SectorsPerTrack);
}
record.StartCylinder = (ushort)newStartAddress.Cylinder;
record.StartHead = (byte)newStartAddress.Head;
record.StartSector = (byte)newStartAddress.Sector;
record.EndCylinder = (ushort)newEndAddress.Cylinder;
record.EndHead = (byte)newEndAddress.Head;
record.EndSector = (byte)newEndAddress.Sector;
WriteRecord(i, record);
}
}
_diskGeometry = geometry;
}
internal BiosPartitionInfo(BiosPartitionTable table, BiosPartitionRecord record)
{
_table = table;
_record = record;
}
private BiosPartitionRecord[] GetExtendedRecords(BiosPartitionRecord r)
{
return(new BiosExtendedPartitionTable(_diskData, r.LBAStart).GetPartitions());
}
private static bool IsExtendedPartition(BiosPartitionRecord r)
{
return(r.PartitionType == BiosPartitionTypes.Extended || r.PartitionType == BiosPartitionTypes.ExtendedLba);
}
internal SparseStream Open(BiosPartitionRecord record)
{
return(new SubStream(_diskData, Ownership.None, ((long)record.LBAStartAbsolute) * _diskGeometry.BytesPerSector, ((long)record.LBALength) * _diskGeometry.BytesPerSector));
}
/// <summary>
/// Creates a new Primary Partition, specified by Logical Block Addresses.
/// </summary>
/// <param name="first">The LBA address of the first sector (inclusive).</param>
/// <param name="last">The LBA address of the last sector (inclusive).</param>
/// <param name="type">The BIOS (MBR) type of the new partition.</param>
/// <param name="markActive">Whether to mark the partition active (bootable).</param>
/// <returns>The index of the new partition.</returns>
public int CreatePrimaryBySector(long first, long last, byte type, bool markActive)
{
if (first >= last)
{
throw new ArgumentException("The first sector in a partition must be before the last");
}
if ((last + 1) * _diskGeometry.BytesPerSector > _diskData.Length)
{
throw new ArgumentOutOfRangeException("last", last, "The last sector extends beyond the end of the disk");
}
BiosPartitionRecord[] existing = GetPrimaryRecords();
BiosPartitionRecord newRecord = new BiosPartitionRecord();
ChsAddress startAddr = _diskGeometry.ToChsAddress(first);
ChsAddress endAddr = _diskGeometry.ToChsAddress(last);
// Because C/H/S addresses can max out at lower values than the LBA values,
// the special tuple (1023, 254, 63) is used.
if (startAddr.Cylinder > 1023)
{
startAddr = new ChsAddress(1023, 254, 63);
}
if (endAddr.Cylinder > 1023)
{
endAddr = new ChsAddress(1023, 254, 63);
}
newRecord.StartCylinder = (ushort)startAddr.Cylinder;
newRecord.StartHead = (byte)startAddr.Head;
newRecord.StartSector = (byte)startAddr.Sector;
newRecord.EndCylinder = (ushort)endAddr.Cylinder;
newRecord.EndHead = (byte)endAddr.Head;
newRecord.EndSector = (byte)endAddr.Sector;
newRecord.LBAStart = (uint)first;
newRecord.LBALength = (uint)(last - first + 1);
newRecord.PartitionType = type;
newRecord.Status = (byte)(markActive ? 0x80 : 0x00);
// First check for overlap with existing partition...
foreach (var r in existing)
{
if (Utilities.RangesOverlap((uint)first, (uint)last + 1, r.LBAStartAbsolute, r.LBAStartAbsolute + r.LBALength))
{
throw new IOException("New partition overlaps with existing partition");
}
}
// Now look for empty partition
for (int i = 0; i < 4; ++i)
{
if (!existing[i].IsValid)
{
WriteRecord(i, newRecord);
return(i);
}
}
throw new IOException("No primary partition slots available");
}