Files
Aaru/Aaru.Gui/ViewModels/Windows/ImageChecksumViewModel.cs
2020-07-22 13:20:25 +01:00

700 lines
25 KiB
C#

// /***************************************************************************
// Aaru Data Preservation Suite
// ----------------------------------------------------------------------------
//
// Filename : ImageChecksumViewModel.cs
// Author(s) : Natalia Portillo <claunia@claunia.com>
//
// Component : GUI view models.
//
// --[ Description ] ----------------------------------------------------------
//
// View model and code for the image checksum window.
//
// --[ License ] --------------------------------------------------------------
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General public License for more details.
//
// You should have received a copy of the GNU General public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
//
// ----------------------------------------------------------------------------
// Copyright © 2011-2020 Natalia Portillo
// ****************************************************************************/
using System;
using System.Collections.ObjectModel;
using System.Reactive;
using System.Threading;
using Aaru.CommonTypes.Interfaces;
using Aaru.CommonTypes.Structs;
using Aaru.Console;
using Aaru.Core;
using Aaru.Gui.Models;
using Avalonia.Controls;
using Avalonia.Threading;
using ReactiveUI;
using Schemas;
namespace Aaru.Gui.ViewModels.Windows
{
public sealed class ImageChecksumViewModel : ViewModelBase
{
// How many sectors to read at once
const uint SECTORS_TO_READ = 256;
readonly IMediaImage _inputFormat;
readonly Window _view;
bool _adler32Checked;
bool _cancel;
bool _checksumMediaChecked;
bool _checksumTracksChecked;
bool _checksumTracksVisible;
bool _closeCommandEnabled;
bool _closeCommandVisible;
bool _crc16Checked;
bool _crc32Checked;
bool _crc64Checked;
bool _fletcher16Checked;
bool _fletcher32Checked;
bool _md5Checked;
bool _mediaChecksumsVisible;
bool _optionsEnabled;
bool _progress1Visible;
double _progress2Max;
string _progress2Text;
double _progress2Value;
bool _progress2Visible;
double _progressMax;
string _progressText;
double _progressValue;
bool _progressVisible;
bool _resultsVisible;
bool _sha1Checked;
bool _sha256Checked;
bool _sha384Checked;
bool _sha512Checked;
bool _spamsumChecked;
bool _startCommandEnabled;
bool _startCommandVisible;
bool _stopCommandEnabled;
bool _stopCommandVisible;
string _title;
bool _trackChecksumsVisible;
public ImageChecksumViewModel(IMediaImage inputFormat, Window view)
{
_view = view;
_cancel = false;
_inputFormat = inputFormat;
ChecksumTracksChecked = ChecksumTracksVisible;
OptionsEnabled = true;
ChecksumMediaChecked = true;
ChecksumTracksChecked = true;
Adler32Checked = true;
Crc16Checked = true;
Crc32Checked = true;
Md5Checked = true;
Sha1Checked = true;
SpamsumChecked = true;
TrackChecksums = new ObservableCollection<ChecksumModel>();
MediaChecksums = new ObservableCollection<ChecksumModel>();
StartCommand = ReactiveCommand.Create(ExecuteStartCommand);
CloseCommand = ReactiveCommand.Create(ExecuteCloseCommand);
StopCommand = ReactiveCommand.Create(ExecuteStopCommand);
StopCommandVisible = false;
StartCommandVisible = true;
CloseCommandVisible = true;
StopCommandEnabled = true;
StartCommandEnabled = true;
CloseCommandEnabled = true;
try
{
ChecksumTracksVisible = (inputFormat as IOpticalMediaImage)?.Tracks?.Count > 0;
}
catch
{
ChecksumTracksVisible = false;
}
}
public string Title
{
get => _title;
set => this.RaiseAndSetIfChanged(ref _title, value);
}
public bool OptionsEnabled
{
get => _optionsEnabled;
set => this.RaiseAndSetIfChanged(ref _optionsEnabled, value);
}
public bool ChecksumMediaChecked
{
get => _checksumMediaChecked;
set => this.RaiseAndSetIfChanged(ref _checksumMediaChecked, value);
}
public bool ChecksumTracksChecked
{
get => _checksumTracksChecked;
set => this.RaiseAndSetIfChanged(ref _checksumTracksChecked, value);
}
public bool Adler32Checked
{
get => _adler32Checked;
set => this.RaiseAndSetIfChanged(ref _adler32Checked, value);
}
public bool Crc16Checked
{
get => _crc16Checked;
set => this.RaiseAndSetIfChanged(ref _crc16Checked, value);
}
public bool Crc32Checked
{
get => _crc32Checked;
set => this.RaiseAndSetIfChanged(ref _crc32Checked, value);
}
public bool Crc64Checked
{
get => _crc64Checked;
set => this.RaiseAndSetIfChanged(ref _crc64Checked, value);
}
public bool Fletcher16Checked
{
get => _fletcher16Checked;
set => this.RaiseAndSetIfChanged(ref _fletcher16Checked, value);
}
public bool Fletcher32Checked
{
get => _fletcher32Checked;
set => this.RaiseAndSetIfChanged(ref _fletcher32Checked, value);
}
public bool Md5Checked
{
get => _md5Checked;
set => this.RaiseAndSetIfChanged(ref _md5Checked, value);
}
public bool Sha1Checked
{
get => _sha1Checked;
set => this.RaiseAndSetIfChanged(ref _sha1Checked, value);
}
public bool Sha256Checked
{
get => _sha256Checked;
set => this.RaiseAndSetIfChanged(ref _sha256Checked, value);
}
public bool Sha384Checked
{
get => _sha384Checked;
set => this.RaiseAndSetIfChanged(ref _sha384Checked, value);
}
public bool Sha512Checked
{
get => _sha512Checked;
set => this.RaiseAndSetIfChanged(ref _sha512Checked, value);
}
public bool SpamsumChecked
{
get => _spamsumChecked;
set => this.RaiseAndSetIfChanged(ref _spamsumChecked, value);
}
public bool ResultsVisible
{
get => _resultsVisible;
set => this.RaiseAndSetIfChanged(ref _resultsVisible, value);
}
public bool TrackChecksumsVisible
{
get => _trackChecksumsVisible;
set => this.RaiseAndSetIfChanged(ref _trackChecksumsVisible, value);
}
public bool MediaChecksumsVisible
{
get => _mediaChecksumsVisible;
set => this.RaiseAndSetIfChanged(ref _mediaChecksumsVisible, value);
}
public bool ProgressVisible
{
get => _progressVisible;
set => this.RaiseAndSetIfChanged(ref _progressVisible, value);
}
public bool Progress1Visible
{
get => _progress1Visible;
set => this.RaiseAndSetIfChanged(ref _progress1Visible, value);
}
public string ProgressText
{
get => _progressText;
set => this.RaiseAndSetIfChanged(ref _progressText, value);
}
public double ProgressMax
{
get => _progressMax;
set => this.RaiseAndSetIfChanged(ref _progressMax, value);
}
public double ProgressValue
{
get => _progressValue;
set => this.RaiseAndSetIfChanged(ref _progressValue, value);
}
public bool Progress2Visible
{
get => _progress2Visible;
set => this.RaiseAndSetIfChanged(ref _progress2Visible, value);
}
public string Progress2Text
{
get => _progress2Text;
set => this.RaiseAndSetIfChanged(ref _progress2Text, value);
}
public double Progress2Max
{
get => _progress2Max;
set => this.RaiseAndSetIfChanged(ref _progress2Max, value);
}
public double Progress2Value
{
get => _progress2Value;
set => this.RaiseAndSetIfChanged(ref _progress2Value, value);
}
public bool StartCommandEnabled
{
get => _startCommandEnabled;
set => this.RaiseAndSetIfChanged(ref _startCommandEnabled, value);
}
public bool StartCommandVisible
{
get => _startCommandVisible;
set => this.RaiseAndSetIfChanged(ref _startCommandVisible, value);
}
public bool CloseCommandEnabled
{
get => _closeCommandEnabled;
set => this.RaiseAndSetIfChanged(ref _closeCommandEnabled, value);
}
public bool CloseCommandVisible
{
get => _closeCommandVisible;
set => this.RaiseAndSetIfChanged(ref _closeCommandVisible, value);
}
public bool StopCommandEnabled
{
get => _stopCommandEnabled;
set => this.RaiseAndSetIfChanged(ref _stopCommandEnabled, value);
}
public bool StopCommandVisible
{
get => _stopCommandVisible;
set => this.RaiseAndSetIfChanged(ref _stopCommandVisible, value);
}
public bool ChecksumTracksVisible
{
get => _checksumTracksVisible;
set => this.RaiseAndSetIfChanged(ref _stopCommandVisible, value);
}
public ObservableCollection<ChecksumModel> TrackChecksums { get; }
public ObservableCollection<ChecksumModel> MediaChecksums { get; }
public ReactiveCommand<Unit, Unit> StartCommand { get; }
public ReactiveCommand<Unit, Unit> CloseCommand { get; }
public ReactiveCommand<Unit, Unit> StopCommand { get; }
void ExecuteStartCommand()
{
OptionsEnabled = false;
CloseCommandVisible = false;
StartCommandVisible = false;
StopCommandVisible = true;
ProgressVisible = true;
Progress1Visible = true;
Progress2Visible = false;
new Thread(DoWork)
{
Priority = ThreadPriority.BelowNormal
}.Start();
}
void ExecuteCloseCommand() => _view.Close();
internal void ExecuteStopCommand()
{
_cancel = true;
StopCommandEnabled = false;
}
async void DoWork()
{
var opticalMediaImage = _inputFormat as IOpticalMediaImage;
bool formatHasTracks = false;
if(opticalMediaImage != null)
try
{
formatHasTracks = opticalMediaImage.Tracks?.Count > 0;
}
catch
{
formatHasTracks = false;
}
// Setup progress bars
await Dispatcher.UIThread.InvokeAsync(() =>
{
ProgressVisible = true;
Progress1Visible = true;
Progress2Visible = true;
ProgressMax = 1;
Progress2Max = (int)(_inputFormat.Info.Sectors / SECTORS_TO_READ);
if(formatHasTracks &&
ChecksumTracksChecked &&
opticalMediaImage != null)
ProgressMax += opticalMediaImage.Tracks.Count;
else
{
ProgressMax = 2;
Progress2Visible = false;
}
});
var enabledChecksums = new EnableChecksum();
if(Adler32Checked)
enabledChecksums |= EnableChecksum.Adler32;
if(Crc16Checked)
enabledChecksums |= EnableChecksum.Crc16;
if(Crc32Checked)
enabledChecksums |= EnableChecksum.Crc32;
if(Crc64Checked)
enabledChecksums |= EnableChecksum.Crc64;
if(Md5Checked)
enabledChecksums |= EnableChecksum.Md5;
if(Sha1Checked)
enabledChecksums |= EnableChecksum.Sha1;
if(Sha256Checked)
enabledChecksums |= EnableChecksum.Sha256;
if(Sha384Checked)
enabledChecksums |= EnableChecksum.Sha384;
if(Sha512Checked)
enabledChecksums |= EnableChecksum.Sha512;
if(SpamsumChecked)
enabledChecksums |= EnableChecksum.SpamSum;
if(Fletcher16Checked)
enabledChecksums |= EnableChecksum.Fletcher16;
if(Fletcher32Checked)
enabledChecksums |= EnableChecksum.Fletcher32;
Checksum mediaChecksum = null;
if(opticalMediaImage?.Tracks != null)
try
{
Checksum trackChecksum = null;
if(ChecksumMediaChecked)
mediaChecksum = new Checksum(enabledChecksums);
ulong previousTrackEnd = 0;
foreach(Track currentTrack in opticalMediaImage.Tracks)
{
await Dispatcher.UIThread.InvokeAsync(() =>
{
ProgressText =
$"Hashing track {currentTrack.TrackSequence} of {opticalMediaImage.Tracks.Count}";
ProgressValue++;
});
if(currentTrack.TrackStartSector - previousTrackEnd != 0 && ChecksumMediaChecked)
for(ulong i = previousTrackEnd + 1; i < currentTrack.TrackStartSector; i++)
{
ulong sector = i;
await Dispatcher.UIThread.InvokeAsync(() =>
{
Progress2Value = (int)(sector / SECTORS_TO_READ);
Progress2Text = $"Hashing track-less sector {sector}";
});
byte[] hiddenSector = opticalMediaImage.ReadSector(i);
mediaChecksum?.Update(hiddenSector);
}
AaruConsole.DebugWriteLine("Checksum command",
"Track {0} starts at sector {1} and ends at sector {2}",
currentTrack.TrackSequence, currentTrack.TrackStartSector,
currentTrack.TrackEndSector);
if(ChecksumTracksChecked)
trackChecksum = new Checksum(enabledChecksums);
ulong sectors = (currentTrack.TrackEndSector - currentTrack.TrackStartSector) + 1;
ulong doneSectors = 0;
while(doneSectors < sectors)
{
if(_cancel)
{
await Dispatcher.UIThread.InvokeAsync(() =>
{
CloseCommandVisible = true;
StartCommandVisible = false;
StopCommandVisible = false;
});
return;
}
byte[] sector;
if(sectors - doneSectors >= SECTORS_TO_READ)
{
sector = opticalMediaImage.ReadSectors(doneSectors, SECTORS_TO_READ,
currentTrack.TrackSequence);
ulong doneSectorsToInvoke = doneSectors;
await Dispatcher.UIThread.InvokeAsync(() =>
{
Progress2Value = (int)(doneSectorsToInvoke / SECTORS_TO_READ);
Progress2Text =
$"Hashing sectors {doneSectorsToInvoke} to {doneSectorsToInvoke + SECTORS_TO_READ} of track {currentTrack.TrackSequence}";
});
doneSectors += SECTORS_TO_READ;
}
else
{
sector = opticalMediaImage.ReadSectors(doneSectors, (uint)(sectors - doneSectors),
currentTrack.TrackSequence);
ulong doneSectorsToInvoke = doneSectors;
await Dispatcher.UIThread.InvokeAsync(() =>
{
Progress2Value = (int)(doneSectorsToInvoke / SECTORS_TO_READ);
Progress2Text =
$"Hashing sectors {doneSectorsToInvoke} to {doneSectorsToInvoke + (sectors - doneSectorsToInvoke)} of track {currentTrack.TrackSequence}";
});
doneSectors += sectors - doneSectors;
}
if(ChecksumMediaChecked)
mediaChecksum?.Update(sector);
if(ChecksumTracksChecked)
trackChecksum?.Update(sector);
}
await Dispatcher.UIThread.InvokeAsync(() =>
{
if(ChecksumTracksChecked != true)
return;
if(trackChecksum == null)
return;
foreach(ChecksumType chk in trackChecksum.End())
TrackChecksums.Add(new ChecksumModel
{
Track = currentTrack.TrackSequence.ToString(),
Algorithm = chk.type.ToString(),
Hash = chk.Value
});
});
previousTrackEnd = currentTrack.TrackEndSector;
}
if(opticalMediaImage.Info.Sectors - previousTrackEnd != 0 && ChecksumMediaChecked)
for(ulong i = previousTrackEnd + 1; i < opticalMediaImage.Info.Sectors; i++)
{
ulong sector = i;
await Dispatcher.UIThread.InvokeAsync(() =>
{
Progress2Value = (int)(sector / SECTORS_TO_READ);
Progress2Text = $"Hashing track-less sector {sector}";
});
byte[] hiddenSector = opticalMediaImage.ReadSector(i);
mediaChecksum?.Update(hiddenSector);
}
await Dispatcher.UIThread.InvokeAsync(() =>
{
if(mediaChecksum == null)
return;
foreach(ChecksumType chk in mediaChecksum.End())
MediaChecksums.Add(new ChecksumModel
{
Algorithm = chk.type.ToString(),
Hash = chk.Value
});
});
}
catch(Exception ex)
{
AaruConsole.DebugWriteLine("Could not get tracks because {0}", ex.Message);
AaruConsole.WriteLine("Unable to get separate tracks, not checksumming them");
}
else
{
await Dispatcher.UIThread.InvokeAsync(() =>
{
Progress1Visible = false;
});
mediaChecksum = new Checksum(enabledChecksums);
ulong doneSectors = 0;
while(doneSectors < _inputFormat.Info.Sectors)
{
if(_cancel)
{
await Dispatcher.UIThread.InvokeAsync(() =>
{
CloseCommandVisible = true;
StartCommandVisible = false;
StopCommandVisible = false;
});
return;
}
byte[] sector;
if(_inputFormat.Info.Sectors - doneSectors >= SECTORS_TO_READ)
{
sector = _inputFormat.ReadSectors(doneSectors, SECTORS_TO_READ);
ulong doneSectorsToInvoke = doneSectors;
await Dispatcher.UIThread.InvokeAsync(() =>
{
Progress2Value = (int)(doneSectorsToInvoke / SECTORS_TO_READ);
Progress2Text =
$"Hashing sectors {doneSectorsToInvoke} to {doneSectorsToInvoke + SECTORS_TO_READ}";
});
doneSectors += SECTORS_TO_READ;
}
else
{
sector = _inputFormat.ReadSectors(doneSectors, (uint)(_inputFormat.Info.Sectors - doneSectors));
ulong doneSectorsToInvoke = doneSectors;
await Dispatcher.UIThread.InvokeAsync(() =>
{
Progress2Value = (int)(doneSectorsToInvoke / SECTORS_TO_READ);
Progress2Text =
$"Hashing sectors {doneSectorsToInvoke} to {doneSectorsToInvoke + (_inputFormat.Info.Sectors - doneSectorsToInvoke)}";
});
doneSectors += _inputFormat.Info.Sectors - doneSectors;
}
mediaChecksum.Update(sector);
}
await Dispatcher.UIThread.InvokeAsync(() =>
{
foreach(ChecksumType chk in mediaChecksum.End())
MediaChecksums.Add(new ChecksumModel
{
Algorithm = chk.type.ToString(),
Hash = chk.Value
});
});
}
if(ChecksumTracksChecked)
await Dispatcher.UIThread.InvokeAsync(() =>
{
TrackChecksumsVisible = true;
});
if(ChecksumMediaChecked)
await Dispatcher.UIThread.InvokeAsync(() =>
{
MediaChecksumsVisible = true;
});
Statistics.AddCommand("checksum");
await Dispatcher.UIThread.InvokeAsync(() =>
{
OptionsEnabled = false;
ResultsVisible = true;
ProgressVisible = false;
StartCommandVisible = false;
StopCommandVisible = false;
CloseCommandVisible = true;
});
}
}
}