123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274 |
- using DiskAccessLibrary;
- using ISCSIConsole.Mods;
- using System;
- using System.Collections.Generic;
- using System.IO;
- using System.Threading;
- using System.Windows.Forms;
- using static DiskAccessLibrary.Mod.Consts;
- namespace ISCSIConsole
- {
- public partial class CreateDiskImageForm : BaseForm
- {
- private bool m_isWorking;
- private FileDialog _imageFileDialog;
- private FileDialog _snapshotFileDialog;
- public DiskImage DiskImage { get; private set; }
- private bool SnapshotControlEnabled
- {
- set
- {
- SnapshotLabel.Enabled = value;
- SnapshotTextBox.Enabled = value;
- SnapshotButton.Enabled = value;
- BlockSizeUpDown.Enabled = value;
- KbLabel.Enabled = value;
- AheadLabel.Enabled = value;
- ImageSizeUpDown.Enabled = !value;
- }
- }
- private bool NoBaseImageControlEnable
- {
- set
- {
- ImageFileLabel.Enabled = !value;
- ImageFileTextBox.Enabled = !value;
- ImageFileButton.Enabled = !value;
- ImageSizeUpDown.Enabled = value;
- }
- }
- public CreateDiskImageForm()
- {
- InitializeComponent();
- }
- private void CreateDiskImageForm_Shown(object sender, EventArgs e)
- {
- FormatComboBox.DataSource = Enum.GetValues(typeof(DiskImageFormat));
- }
- private void FormatComboBox_SelectedIndexChanged(object sender, EventArgs e)
- {
- var items = new List<DiskImageType>();
- switch ((DiskImageFormat)FormatComboBox.SelectedValue)
- {
- case DiskImageFormat.BlockDifferencingDiskImage:
- items.Add(DiskImageType.Differencing);
- items.Add(DiskImageType.Dynamic);
- SnapshotControlEnabled = true;
- _imageFileDialog = new OpenFileDialog
- {
- CheckFileExists = true,
- Filter = "Raw Disk Image (*.img)|*.img"
- };
- _snapshotFileDialog = new SaveFileDialog
- {
- Filter = "Block Differencing Disk Image (*.bdd)|*.bdd"
- };
- break;
- case DiskImageFormat.RawDiskImage:
- items.Add(DiskImageType.Fixed);
- SnapshotControlEnabled = false;
- _imageFileDialog = new SaveFileDialog
- {
- Filter = "Raw Disk Image (*.img)|*.img"
- };
- break;
- case DiskImageFormat.VirtualHardDisk:
- items.Add(DiskImageType.Dynamic);
- items.Add(DiskImageType.Fixed);
- SnapshotControlEnabled = false;
- _imageFileDialog = new SaveFileDialog
- {
- Filter = "VirtualHardDisk File (*.vhd)|*.vhd"
- };
- break;
- default:
- throw new ArgumentOutOfRangeException();
- }
- TypeComboBox.DataSource = items;
- }
- private void TypeComboBox_SelectedIndexChanged(object sender, EventArgs e)
- {
- var format = (DiskImageFormat)FormatComboBox.SelectedValue;
- if (format == DiskImageFormat.BlockDifferencingDiskImage)
- {
- switch ((DiskImageType)TypeComboBox.SelectedValue)
- {
- case DiskImageType.Dynamic:
- NoBaseImageControlEnable = true;
- break;
- default:
- NoBaseImageControlEnable = false;
- break;
- }
- }
- }
- private void ImageFileButton_Click(object sender, EventArgs e)
- {
- if (_imageFileDialog.ShowDialog() == DialogResult.OK)
- {
- ImageFileTextBox.Text = _imageFileDialog.FileName;
- if ((DiskImageFormat)FormatComboBox.SelectedValue == DiskImageFormat.BlockDifferencingDiskImage)
- {
- var length=IoUtility.GetRealFileSize(_imageFileDialog.FileName);
- // ReSharper disable once PossibleLossOfFraction
- ImageSizeUpDown.Value = length / MegaByte;
- BlockSizeUpDown_SelectedItemChanged(null, null);
- }
- }
- }
- private void SnapshotButton_Click(object sender, EventArgs e)
- {
- if (_snapshotFileDialog.ShowDialog() == DialogResult.OK)
- {
- SnapshotTextBox.Text = _snapshotFileDialog.FileName;
- }
- }
- private void ImageSizeUpDown_ValueChanged(object sender, EventArgs e)
- {
- UpdateSnapshotInfo();
- }
- private void BlockSizeUpDown_SelectedItemChanged(object sender, EventArgs e)
- {
- UpdateSnapshotInfo();
- }
- private void UpdateSnapshotInfo()
- {
- var format = (DiskImageFormat)FormatComboBox.SelectedValue;
- var type = (DiskImageType)TypeComboBox.SelectedValue;
- var blockSize = int.Parse(BlockSizeUpDown.Text) * KiloByte;
- if (format == DiskImageFormat.BlockDifferencingDiskImage && type == DiskImageType.Dynamic)
- {
- AheadLabel.Text =
- $"Init Size:{BddInfo.CalcHeaderSize((long)(ImageSizeUpDown.Value * MegaByte), blockSize, out var blocks) / KiloByte:N0} KB" +
- $", {blocks:N0} Blocks.";
- }
- else
- {
- AheadLabel.Text =
- $"Init Size:{BddInfo.CalcHeaderSize(ImageFileTextBox.Text, blockSize, out var blocks) / KiloByte:N0} KB" +
- $", {blocks:N0} Blocks.";
- }
- }
- private void btnOK_Click(object sender, EventArgs e)
- {
- var format = (DiskImageFormat)FormatComboBox.SelectedValue;
- var type = (DiskImageType)TypeComboBox.SelectedValue;
- var imgPath = ImageFileTextBox.Text;
- var imgSize = (long)ImageSizeUpDown.Value * MegaByte;
- var snapshotPath = SnapshotTextBox.Text;
- var blkSize = int.Parse(BlockSizeUpDown.Text) * KiloByte;
- if ((format != DiskImageFormat.BlockDifferencingDiskImage && type != DiskImageType.Dynamic) && imgPath == string.Empty)
- {
- MessageBox.Show("Please choose image file location", "Error");
- return;
- }
- if (format == DiskImageFormat.BlockDifferencingDiskImage
- && string.IsNullOrWhiteSpace(SnapshotTextBox.Text))
- {
- MessageBox.Show("Please choose snapshot file location", "Error");
- return;
- }
- m_isWorking = true;
- new Thread(delegate ()
- {
- DiskImage diskImage;
- try
- {
- diskImage = DiskImageCreator.CreateImage(format, type, imgPath, imgSize, snapshotPath, blkSize);
- }
- catch (Exception ex)
- {
- Invoke((MethodInvoker)delegate
- {
- MessageBox.Show("Failed to create the disk: " + ex.Message, "Error");
- ImageFileTextBox.Enabled = true;
- ImageFileButton.Enabled = true;
- ImageSizeUpDown.Enabled = true;
- btnOK.Enabled = true;
- btnCancel.Enabled = true;
- });
- m_isWorking = false;
- return;
- }
- var isLocked = diskImage.ExclusiveLock();
- if (!isLocked)
- {
- Invoke((MethodInvoker)delegate
- {
- MessageBox.Show("Cannot lock the disk image for exclusive access", "Error");
- ImageFileTextBox.Enabled = true;
- ImageFileButton.Enabled = true;
- ImageSizeUpDown.Enabled = true;
- btnOK.Enabled = true;
- btnCancel.Enabled = true;
- m_isWorking = false;
- });
- return;
- }
- DiskImage = diskImage;
- Invoke((MethodInvoker)delegate
- {
- m_isWorking = false;
- DialogResult = DialogResult.OK;
- Close();
- });
- }).Start();
- ImageFileTextBox.Enabled = false;
- ImageFileButton.Enabled = false;
- ImageSizeUpDown.Enabled = false;
- btnOK.Enabled = false;
- btnCancel.Enabled = false;
- }
- private void btnCancel_Click(object sender, EventArgs e)
- {
- DialogResult = DialogResult.Cancel;
- Close();
- }
- private void CreateDiskImageForm_FormClosing(object sender, FormClosingEventArgs e)
- {
- if (m_isWorking)
- {
- e.Cancel = true;
- MessageBox.Show("Please wait until the creation of the disk image is completed.", "Error");
- }
- }
- }
- }
|