private bool LogicalPartitionLayout()
{
// Try to resize logical to fit inside newly created extended
var percentCounter = -.1;
var upSizeLock = new Dictionary <string, long>();
var logicalPartLayoutVerified = false;
while (!logicalPartLayoutVerified)
{
percentCounter += .1;
var isError = false;
LogicalPartitions.Clear();
double totalExtendedPercentage = 0;
var partCounter = -1;
foreach (var part in _imageSchema.HardDrives[HdNumberToGet].Partitions)
{
partCounter++;
if (part.Type.ToLower() != "logical")
{
continue;
}
var clientPartition = new ClientPartition
{
IsBoot = BootPart == part.Number,
Number = part.Number,
Start = part.Start,
Type = part.Type,
FsId = part.FsId,
Uuid = part.Uuid,
Guid = part.Guid,
FsType = part.FsType
};
var logicalPartitionHelper = new ServiceClientPartition(_imageProfile).Partition(HdNumberToGet,
partCounter, _newHdSize);
var percentOfExtendedForThisPartition = (double)logicalPartitionHelper.MinSizeBlk /
ExtendedPartitionHelper.AgreedSizeBlk;
var tmpClientPartitionSizeBlk = logicalPartitionHelper.MinSizeBlk;
if (upSizeLock.ContainsKey(part.Number))
{
tmpClientPartitionSizeBlk = upSizeLock[part.Number];
}
else
{
if (logicalPartitionHelper.IsDynamicSize)
{
clientPartition.SizeIsDynamic = true;
var percentOfOrigDrive = Convert.ToInt64(part.Size) /
(double)
Convert.ToInt64(_imageSchema.HardDrives[HdNumberToGet].Size);
if (Convert.ToInt64(NewHdBlk * percentOfOrigDrive) < logicalPartitionHelper.MinSizeBlk)
{
//This will never work because each loop only gets smaller
tmpClientPartitionSizeBlk =
Convert.ToInt64(NewHdBlk * (percentOfOrigDrive + percentCounter / 100));
if (logicalPartitionHelper.MinSizeBlk < tmpClientPartitionSizeBlk)
{
upSizeLock.Add(part.Number, tmpClientPartitionSizeBlk);
}
}
else
{
tmpClientPartitionSizeBlk = percentOfOrigDrive - percentCounter / 100 <= 0
? Convert.ToInt64(NewHdBlk * percentOfOrigDrive)
: Convert.ToInt64(NewHdBlk * (percentOfOrigDrive - percentCounter / 100));
percentOfExtendedForThisPartition = (double)tmpClientPartitionSizeBlk /
ExtendedPartitionHelper.AgreedSizeBlk;
}
}
}
if (logicalPartitionHelper.MinSizeBlk > tmpClientPartitionSizeBlk)
{
isError = true;
break;
}
totalExtendedPercentage += percentOfExtendedForThisPartition;
clientPartition.Size = tmpClientPartitionSizeBlk;
LogicalPartitions.Add(clientPartition);
if (logicalPartitionHelper.PartitionHasVolumeGroup)
{
logicalPartitionHelper.VolumeGroupHelper.AgreedPvSizeBlk = tmpClientPartitionSizeBlk;
VolumeGroupHelpers.Add(logicalPartitionHelper.VolumeGroupHelper);
}
}
//Could not determine a partition layout that works with this hard drive
if (isError && percentCounter > 99)
{
log.Error(JsonConvert.SerializeObject(PrimaryAndExtendedPartitions));
log.Error(JsonConvert.SerializeObject(LogicalPartitions));
return(false);
}
//This partition size doesn't work, continuation of break from earlier
if (isError)
{
continue;
}
if (totalExtendedPercentage <= 1)
{
long totalAllocatedBlk = 0;
var dynamicPartitionCount = 0;
//If totalPercentage is too far below 1 try to increase size of available resizable partitions
if (totalExtendedPercentage < .98)
{
foreach (var partition in LogicalPartitions)
{
totalAllocatedBlk += Convert.ToInt64(partition.Size);
if (partition.SizeIsDynamic)
{
dynamicPartitionCount++;
}
}
var totalUnallocated = ExtendedPartitionHelper.AgreedSizeBlk - totalAllocatedBlk;
if (dynamicPartitionCount > 0)
{
foreach (
var partition in LogicalPartitions.Where(partition => partition.SizeIsDynamic))
{
partition.Size =
partition.Size + totalUnallocated / dynamicPartitionCount;
}
}
}
logicalPartLayoutVerified = true;
}
//Theoretically should never hit this, but added to prevent infinite loop
if (percentCounter > 100)
{
return(false);
}
}
return(true);
}
private bool PrimaryAndExtendedPartitionLayout()
{
//Try to determine a layout for each primary or extended partition that will be able to fit logical partitions
//or logical volumes in. Also if the partition is logical and is the physical volume for a volume group determine
// a size that will work for all logical volumes
ExtendedPartitionHelper = new ServiceClientPartition(_imageProfile).ExtendedPartition(HdNumberToGet,
_newHdSize);
var upSizeLock = new Dictionary <string, long>();
var percentCounter = -.1;
var partitionLayoutVerified = false;
while (!partitionLayoutVerified)
{
//percentCounter++;
percentCounter += .1;
var isError = false;
double totalHdPercentage = 0;
PrimaryAndExtendedPartitions.Clear();
VolumeGroupHelpers.Clear();
FirstPartitionStartSector = Convert.ToInt64(_imageSchema.HardDrives[HdNumberToGet].Partitions[0].Start);
var partCounter = -1;
foreach (var schemaPartition in _imageSchema.HardDrives[HdNumberToGet].Partitions)
{
partCounter++;
//Determine what sector the first partition should start at
if (Convert.ToInt64(schemaPartition.Start) < FirstPartitionStartSector)
{
FirstPartitionStartSector = Convert.ToInt64(schemaPartition.Start);
}
if (!schemaPartition.Active)
{
continue;
}
if (schemaPartition.Type.ToLower() == "logical")
{
continue;
}
var clientPartition = new ClientPartition
{
IsBoot = BootPart == schemaPartition.Number,
Number = schemaPartition.Number,
Start = schemaPartition.Start,
Type = schemaPartition.Type,
FsId = schemaPartition.FsId,
Uuid = schemaPartition.Uuid,
Guid = schemaPartition.Guid,
FsType = schemaPartition.FsType
};
var partitionHelper = new ServiceClientPartition(_imageProfile).Partition(HdNumberToGet, partCounter,
_newHdSize);
var percentOfHdForThisPartition = (double)partitionHelper.MinSizeBlk / NewHdBlk;
var tmpClientPartitionSizeBlk = partitionHelper.MinSizeBlk;
if (partitionHelper.IsDynamicSize)
{
clientPartition.SizeIsDynamic = true;
var percentOfOrigDrive = Convert.ToInt64(schemaPartition.Size) /
(double)Convert.ToInt64(_imageSchema.HardDrives[HdNumberToGet].Size);
//Change the resized partition size based off original percentage and percentCounter loop
//This is the active part of the loop that lowers the partition size based on each iteration
//If a partition's used space is almost maxed out to the size of the partition this can cause
//problems with calculations.
if (upSizeLock.ContainsKey(schemaPartition.Number))
{
tmpClientPartitionSizeBlk = upSizeLock[schemaPartition.Number];
}
else
{
if (Convert.ToInt64(NewHdBlk * percentOfOrigDrive) < partitionHelper.MinSizeBlk)
{
//This will never work because each loop only gets smaller
tmpClientPartitionSizeBlk =
Convert.ToInt64(NewHdBlk * (percentOfOrigDrive + percentCounter / 100));
if (partitionHelper.MinSizeBlk < tmpClientPartitionSizeBlk)
{
upSizeLock.Add(schemaPartition.Number, tmpClientPartitionSizeBlk);
}
}
else
{
tmpClientPartitionSizeBlk = percentOfOrigDrive - percentCounter / 100 <= 0
? Convert.ToInt64(NewHdBlk * percentOfOrigDrive)
: Convert.ToInt64(NewHdBlk * (percentOfOrigDrive - percentCounter / 100));
}
}
//Add the percent of this partition used to the total percent used to make sure we don't go over
//100% of the size of the new drive.
//Each logical partition requires and extra 1 mb added to the size of the extended partition.
if (clientPartition.Type.ToLower() == "extended")
{
percentOfHdForThisPartition = ((double)tmpClientPartitionSizeBlk +
(1048576 / LbsByte * ExtendedPartitionHelper.LogicalCount +
1048576 / LbsByte)) /
NewHdBlk;
}
else
{
percentOfHdForThisPartition = (double)tmpClientPartitionSizeBlk / NewHdBlk;
}
}
if (partitionHelper.MinSizeBlk > tmpClientPartitionSizeBlk)
{
isError = true;
break;
}
if (clientPartition.Type.ToLower() == "extended")
{
ExtendedPartitionHelper.AgreedSizeBlk = tmpClientPartitionSizeBlk;
}
if (partitionHelper.PartitionHasVolumeGroup)
{
partitionHelper.VolumeGroupHelper.AgreedPvSizeBlk = tmpClientPartitionSizeBlk;
VolumeGroupHelpers.Add(partitionHelper.VolumeGroupHelper);
}
clientPartition.Size = tmpClientPartitionSizeBlk;
totalHdPercentage += percentOfHdForThisPartition;
PrimaryAndExtendedPartitions.Add(clientPartition);
}
//Could not determine a partition layout that works with this hard drive
if (isError && percentCounter > 99)
{
return(false);
}
//This partition size doesn't work, continuation of break from earlier
if (isError)
{
continue;
}
DebugStatus += percentCounter + "\r\n";
if (totalHdPercentage <= 1)
{
//if totalPercentage is < 1 then layout has been verified to work
partitionLayoutVerified = true;
//If totalPercentage is too far below 1 try to increase size of available resizable partitions
long totalAllocatedBlk = 0;
var dynamicPartitionCount = 0;
if (totalHdPercentage < .98)
{
foreach (var partition in PrimaryAndExtendedPartitions)
{
totalAllocatedBlk += Convert.ToInt64(partition.Size);
if (partition.SizeIsDynamic)
{
dynamicPartitionCount++;
}
}
var totalUnallocated = NewHdBlk - totalAllocatedBlk;
if (dynamicPartitionCount > 0)
{
foreach (
var partition in
PrimaryAndExtendedPartitions.Where(partition => partition.SizeIsDynamic))
{
partition.Size = partition.Size + totalUnallocated / dynamicPartitionCount;
if (partition.Type.ToLower() == "extended")
{
ExtendedPartitionHelper.AgreedSizeBlk = Convert.ToInt64(partition.Size);
}
for (var i = 0; i < VolumeGroupHelpers.Count(); i++)
{
if (_imageSchema.HardDrives[HdNumberToGet].Name + partition.Number ==
VolumeGroupHelpers[i].Pv)
{
VolumeGroupHelpers[i].AgreedPvSizeBlk = Convert.ToInt64(partition.Size);
}
}
}
}
}
}
//Theoretically should never hit this, but added to prevent infinite loop
if (percentCounter > 100)
{
return(false);
}
}
return(true);
}
public string GeneratePartitionScript()
{
var imageProfile = BLL.ImageProfile.ReadProfile(profileId);
ImageSchema = new ClientPartitionHelper(imageProfile).GetImageSchema();
string partitionScript = null;
var clientSchema = new ClientPartition(HdNumberToGet,NewHdSize,imageProfile).GenerateClientSchema();
if (clientSchema == null) return "failed";
if (TaskType == "debug")
{
partitionScript = clientSchema.DebugStatus;
if (clientSchema.PrimaryAndExtendedPartitions.Count == 0)
return partitionScript;
try
{
clientSchema.ExtendedPartitionHelper.AgreedSizeBlk = clientSchema.ExtendedPartitionHelper.AgreedSizeBlk * 512 / 1024 / 1024;
}
catch
{
// ignored
}
foreach (var p in clientSchema.PrimaryAndExtendedPartitions)
p.Size = p.Size * 512 / 1024 / 1024;
foreach (var p in clientSchema.LogicalPartitions)
p.Size = p.Size * 512 / 1024 / 1024;
foreach (var p in clientSchema.LogicalVolumes)
p.Size = p.Size * 512 / 1024 / 1024;
}
//Create Menu
if (ImageSchema.HardDrives[HdNumberToGet].Table.ToLower() == "mbr")
{
var counter = 0;
var partCount = clientSchema.PrimaryAndExtendedPartitions.Count;
string partitionCommands;
partitionCommands = "fdisk " + ClientHd + " &>>/tmp/clientlog.log <<FDISK\r\n";
if (Convert.ToInt32(clientSchema.PrimaryAndExtendedPartitions[0].Start) < 2048)
partitionCommands += "c\r\n";
foreach (var part in clientSchema.PrimaryAndExtendedPartitions)
{
counter++;
partitionCommands += "n\r\n";
switch (part.Type)
{
case "primary":
partitionCommands += "p\r\n";
break;
case "extended":
partitionCommands += "e\r\n";
break;
}
partitionCommands += part.Number + "\r\n";
if (counter == 1)
partitionCommands += clientSchema.FirstPartitionStartSector + "\r\n";
else
partitionCommands += "\r\n";
if (part.Type == "extended")
partitionCommands += "+" + (Convert.ToInt64(clientSchema.ExtendedPartitionHelper.AgreedSizeBlk) - 1) + "\r\n";
else //FDISK seems to include the starting sector in size so we need to subtract 1
partitionCommands += "+" + (Convert.ToInt64(part.Size) - 1) + "\r\n";
partitionCommands += "t\r\n";
if (counter == 1)
partitionCommands += part.FsId + "\r\n";
else
{
partitionCommands += part.Number + "\r\n";
partitionCommands += part.FsId + "\r\n";
}
if ((counter == 1 && part.IsBoot) || clientSchema.PrimaryAndExtendedPartitions.Count == 1)
partitionCommands += "a\r\n";
if (counter != 1 && part.IsBoot)
{
partitionCommands += "a\r\n";
partitionCommands += part.Number + "\r\n";
}
if ((counter != partCount || clientSchema.LogicalPartitions.Count != 0)) continue;
partitionCommands += "w\r\n";
partitionCommands += "FDISK\r\n";
}
var logicalCounter = 0;
foreach (var logicalPart in clientSchema.LogicalPartitions)
{
logicalCounter++;
partitionCommands += "n\r\n";
if (clientSchema.PrimaryAndExtendedPartitions.Count < 4)
partitionCommands += "l\r\n";
partitionCommands += "\r\n";
if (TaskType == "debug")
partitionCommands += "+" + (Convert.ToInt64(logicalPart.Size) - (logicalCounter * 1)) + "\r\n";
else
partitionCommands += "+" + (Convert.ToInt64(logicalPart.Size) - (logicalCounter * 2049)) + "\r\n";
partitionCommands += "t\r\n";
partitionCommands += logicalPart.Number + "\r\n";
partitionCommands += logicalPart.FsId + "\r\n";
if (logicalPart.IsBoot)
{
partitionCommands += "a\r\n";
partitionCommands += logicalPart.Number + "\r\n";
}
if (logicalCounter != clientSchema.LogicalPartitions.Count) continue;
partitionCommands += "w\r\n";
partitionCommands += "FDISK\r\n";
}
partitionScript += partitionCommands;
}
else
{
var counter = 0;
var partCount = clientSchema.PrimaryAndExtendedPartitions.Count;
var partitionCommands = "gdisk " + ClientHd + " &>>/tmp/clientlog.log <<GDISK\r\n";
bool isApple = false;
foreach (var part in clientSchema.PrimaryAndExtendedPartitions)
{
if (part.FsType.Contains("hfs"))
{
isApple = true;
break;
}
}
if (clientSchema.FirstPartitionStartSector < 2048 && isApple) //osx cylinder boundary is 8
{
partitionCommands += "x\r\nl\r\n8\r\nm\r\n";
}
foreach (var part in clientSchema.PrimaryAndExtendedPartitions)
{
counter++;
partitionCommands += "n\r\n";
partitionCommands += part.Number + "\r\n";
if (counter == 1)
partitionCommands += clientSchema.FirstPartitionStartSector + "\r\n";
else
partitionCommands += "\r\n";
//GDISK seems to NOT include the starting sector in size so don't subtract 1 like in FDISK
partitionCommands += "+" + Convert.ToInt64(part.Size) + "\r\n";
partitionCommands += part.FsId + "\r\n";
if ((counter != partCount)) continue;
partitionCommands += "w\r\n";
partitionCommands += "y\r\n";
partitionCommands += "GDISK\r\n";
}
partitionScript += partitionCommands;
}
foreach (var part in from part in ImageSchema.HardDrives[HdNumberToGet].Partitions
where part.Active
where part.VolumeGroup != null
where part.VolumeGroup.LogicalVolumes != null
select part)
{
partitionScript += "echo \"pvcreate -u " + part.Uuid + " --norestorefile -yf " +
ClientHd + partitionPrefix + part.VolumeGroup.PhysicalVolume[part.VolumeGroup.PhysicalVolume.Length - 1] +
"\" >>/tmp/lvmcommands \r\n";
partitionScript += "echo \"vgcreate " + part.VolumeGroup.Name + " " + ClientHd + partitionPrefix +
part.VolumeGroup.PhysicalVolume[part.VolumeGroup.PhysicalVolume.Length - 1] + " -yf" +
"\" >>/tmp/lvmcommands \r\n";
partitionScript += "echo \"" + part.VolumeGroup.Uuid + "\" >>/tmp/vg-" + part.VolumeGroup.Name +
" \r\n";
foreach (var lv in part.VolumeGroup.LogicalVolumes)
{
foreach (var rlv in clientSchema.LogicalVolumes)
{
if (lv.Name != rlv.Name || lv.VolumeGroup != rlv.Vg) continue;
if (TaskType == "debug")
{
partitionScript += "echo \"lvcreate --yes -L " +
rlv.Size + "mb -n " +
rlv.Name + " " + rlv.Vg +
"\" >>/tmp/lvmcommands \r\n";
}
else
{
partitionScript += "echo \"lvcreate --yes -L " +
((Convert.ToInt64(rlv.Size) - 8192)) + "s -n " +
rlv.Name + " " + rlv.Vg +
"\" >>/tmp/lvmcommands \r\n";
}
partitionScript += "echo \"" + rlv.Uuid + "\" >>/tmp/" + rlv.Vg +
"-" + rlv.Name + "\r\n";
}
}
partitionScript += "echo \"vgcfgbackup -f /tmp/lvm-" + part.VolumeGroup.Name +
"\" >>/tmp/lvmcommands\r\n";
}
return partitionScript;
}