From 19f9f7c731f2da8474003af9f14ff4ed52cada7e Mon Sep 17 00:00:00 2001
From: Natalia Portillo <claunia@claunia.com>
Date: Wed, 17 Oct 2018 22:12:51 +0100
Subject: [PATCH] Implement image verification on GUI.

---
 .../.idea/contentModel.xml                    |   2 +
 DiscImageChef.Gui/Forms/frmImageVerify.xeto   | 101 +++++
 .../Forms/frmImageVerify.xeto.cs              | 418 ++++++++++++++++++
 DiscImageChef.Gui/Panels/pnlImageInfo.xeto    |   7 +-
 DiscImageChef.Gui/Panels/pnlImageInfo.xeto.cs |  14 +
 5 files changed, 541 insertions(+), 1 deletion(-)
 create mode 100644 DiscImageChef.Gui/Forms/frmImageVerify.xeto
 create mode 100644 DiscImageChef.Gui/Forms/frmImageVerify.xeto.cs

diff --git a/.idea/.idea.DiscImageChef/.idea/contentModel.xml b/.idea/.idea.DiscImageChef/.idea/contentModel.xml
index e5c3818e3..b77ee138a 100644
--- a/.idea/.idea.DiscImageChef/.idea/contentModel.xml
+++ b/.idea/.idea.DiscImageChef/.idea/contentModel.xml
@@ -1623,6 +1623,8 @@
           <e p="frmDump.xeto.cs" t="Include" />
           <e p="frmImageEntropy.xeto" t="Include" />
           <e p="frmImageEntropy.xeto.cs" t="Include" />
+          <e p="frmImageVerify.xeto" t="Include" />
+          <e p="frmImageVerify.xeto.cs" t="Include" />
           <e p="frmMain.xeto" t="Include" />
           <e p="frmMain.xeto.cs" t="Include" />
         </e>
diff --git a/DiscImageChef.Gui/Forms/frmImageVerify.xeto b/DiscImageChef.Gui/Forms/frmImageVerify.xeto
new file mode 100644
index 000000000..4bed4c106
--- /dev/null
+++ b/DiscImageChef.Gui/Forms/frmImageVerify.xeto
@@ -0,0 +1,101 @@
+<?xml version="1.0" encoding="UTF-8"?><!--
+// /***************************************************************************
+// The Disc Image Chef
+// ============================================================================
+//
+// Filename       : frmImageVerify.xeto
+// Author(s)      : Natalia Portillo <claunia@claunia.com>
+//
+// Component      : Main window.
+//
+// ==[ Description ] ==========================================================
+//
+//     Defines the structure for the image verification GUI 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-2018 Natalia Portillo
+// ****************************************************************************/
+-->
+<Form xmlns="http://schema.picoe.ca/eto.forms" xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
+      Title="Verify image" ClientSize="600, 450" Padding="10">
+    <StackLayout Orientation="Vertical" HorizontalContentAlignment="Stretch" VerticalContentAlignment="Stretch">
+        <StackLayout Orientation="Vertical" HorizontalContentAlignment="Stretch" VerticalContentAlignment="Stretch"
+                     ID="stkOptions">
+            <CheckBox Text="Verify disc image if supported." ID="chkVerifyImage" Checked="True"/>
+            <CheckBox Text="Verify all sectors if supported." ID="chkVerifySectors" Checked="True"/>
+        </StackLayout>
+        <StackLayout Orientation="Vertical" HorizontalContentAlignment="Stretch" VerticalContentAlignment="Stretch"
+                     ID="stkResults" Visible="False">
+            <StackLayout Orientation="Horizontal" HorizontalContentAlignment="Stretch"
+                         VerticalContentAlignment="Stretch" ID="stkSectorResults" Visible="False">
+                <StackLayout Orientation="Vertical" HorizontalContentAlignment="Stretch"
+                             VerticalContentAlignment="Stretch" ID="stkSectorErrors" Visible="False">
+                    <GroupBox ID="grpSectorErrors">
+                        <TreeGridView ID="treeSectorErrors"/>
+                    </GroupBox>
+                </StackLayout>
+                <StackLayout Orientation="Vertical" HorizontalContentAlignment="Stretch"
+                             VerticalContentAlignment="Stretch" ID="stkSectorUnknowns" Visible="False">
+                    <GroupBox ID="grpSectorsUnknowns">
+                        <TreeGridView ID="treeSectorsUnknowns"/>
+                    </GroupBox>
+                </StackLayout>
+            </StackLayout>
+            <Label ID="lblImageResult" Visible="False"/>
+            <Label ID="lblSectorsErrorsAll" Visible="False"/>
+            <Label ID="lblSectorsUnknownAll" Visible="False"/>
+            <GroupBox ID="grpSectorSummary" Visible="False">
+                <StackLayout Orientation="Vertical" HorizontalContentAlignment="Stretch"
+                             VerticalContentAlignment="Stretch">
+                    <Label ID="lblTotalSectors"/>
+                    <Label ID="lblTotalSectorErrors"/>
+                    <Label ID="lblTotalSectorUnknowns"/>
+                    <Label ID="lblTotalSectorErrorsUnknowns"/>
+                </StackLayout>
+            </GroupBox>
+        </StackLayout>
+        <StackLayout Orientation="Vertical" HorizontalContentAlignment="Stretch" VerticalContentAlignment="Stretch"
+                     ID="stkProgress" Visible="False">
+            <StackLayout Orientation="Vertical" HorizontalContentAlignment="Stretch" VerticalContentAlignment="Stretch"
+                         ID="stkProgress1">
+                <StackLayoutItem HorizontalAlignment="Center" Expand="True">
+                    <Label ID="lblProgress"/>
+                </StackLayoutItem>
+                <StackLayoutItem HorizontalAlignment="Center" Expand="True">
+                    <ProgressBar ID="prgProgress"/>
+                </StackLayoutItem>
+            </StackLayout>
+            <StackLayout Orientation="Vertical" HorizontalContentAlignment="Stretch" VerticalContentAlignment="Stretch"
+                         ID="stkProgress2">
+                <StackLayoutItem HorizontalAlignment="Center" Expand="True">
+                    <Label ID="lblProgress2"/>
+                </StackLayoutItem>
+                <StackLayoutItem HorizontalAlignment="Center" Expand="True">
+                    <ProgressBar ID="prgProgress2"/>
+                </StackLayoutItem>
+            </StackLayout>
+        </StackLayout>
+        <StackLayoutItem HorizontalAlignment="Right" Expand="True">
+            <StackLayout Orientation="Horizontal" HorizontalContentAlignment="Right" VerticalContentAlignment="Bottom">
+                <Button ID="btnStart" Text="Start" Click="OnBtnStart"/>
+                <Button ID="btnClose" Text="Close" Click="OnBtnClose"/>
+                <Button ID="btnStop" Text="Stop" Enabled="False" Visible="False" Click="OnBtnStop"/>
+            </StackLayout>
+        </StackLayoutItem>
+    </StackLayout>
+</Form>
\ No newline at end of file
diff --git a/DiscImageChef.Gui/Forms/frmImageVerify.xeto.cs b/DiscImageChef.Gui/Forms/frmImageVerify.xeto.cs
new file mode 100644
index 000000000..4fc5403c1
--- /dev/null
+++ b/DiscImageChef.Gui/Forms/frmImageVerify.xeto.cs
@@ -0,0 +1,418 @@
+// /***************************************************************************
+// The Disc Image Chef
+// ----------------------------------------------------------------------------
+//
+// Filename       : frmImageVerify.xeto.cs
+// Author(s)      : Natalia Portillo <claunia@claunia.com>
+//
+// Component      : Image entropy calculation window.
+//
+// --[ Description ] ----------------------------------------------------------
+//
+//     Implements verifying media image.
+//
+// --[ 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-2018 Natalia Portillo
+// ****************************************************************************/
+
+using System;
+using System.Collections.Generic;
+using System.Threading;
+using DiscImageChef.CommonTypes.Interfaces;
+using DiscImageChef.CommonTypes.Structs;
+using DiscImageChef.Console;
+using DiscImageChef.Core;
+using Eto.Forms;
+using Eto.Serialization.Xaml;
+
+namespace DiscImageChef.Gui.Forms
+{
+    public class frmImageVerify : Form
+    {
+        bool        cancel;
+        IMediaImage inputFormat;
+
+        public frmImageVerify(IMediaImage inputFormat)
+        {
+            this.inputFormat = inputFormat;
+            XamlReader.Load(this);
+            cancel = false;
+        }
+
+        protected void OnBtnStart(object sender, EventArgs e)
+        {
+            chkVerifyImage.Enabled   = false;
+            chkVerifySectors.Enabled = false;
+            btnClose.Visible         = false;
+            btnStart.Visible         = false;
+            btnStop.Visible          = true;
+            stkProgress.Visible      = true;
+            lblProgress2.Visible     = false;
+
+            new Thread(DoWork).Start();
+        }
+
+        void DoWork()
+        {
+            bool? correctDisc    = null;
+            long  totalSectors   = 0;
+            long  errorSectors   = 0;
+            long  correctSectors = 0;
+            long  unknownSectors = 0;
+            bool  formatHasTracks;
+
+            try { formatHasTracks = inputFormat.Tracks?.Count > 0; }
+            catch { formatHasTracks = false; }
+
+            // Setup progress bars
+            Application.Instance.Invoke(() =>
+            {
+                stkProgress.Visible  = true;
+                prgProgress.MaxValue = 0;
+
+                if(chkVerifyImage.Checked == true || chkVerifySectors.Checked == true) prgProgress.MaxValue = 1;
+
+                if(formatHasTracks) prgProgress.MaxValue += inputFormat.Tracks.Count;
+                else
+                {
+                    if(chkVerifySectors.Checked == true)
+                    {
+                        prgProgress.MaxValue = 2;
+                        prgProgress2.Visible = false;
+                        lblProgress2.Visible = false;
+                    }
+                    else
+                    {
+                        prgProgress2.Visible = true;
+                        lblProgress2.Visible = true;
+                    }
+                }
+
+                prgProgress.MaxValue++;
+            });
+
+            if(chkVerifyImage.Checked == true)
+            {
+                Application.Instance.Invoke(() =>
+                {
+                    lblProgress.Text = "Checking media image...";
+                    if(chkVerifySectors.Checked == true) prgProgress.Value = 1;
+                    else prgProgress.Indeterminate                         = true;
+
+                    prgProgress2.Indeterminate = true;
+                });
+
+                DateTime startCheck      = DateTime.UtcNow;
+                bool?    discCheckStatus = inputFormat.VerifyMediaImage();
+                DateTime endCheck        = DateTime.UtcNow;
+
+                TimeSpan checkTime = endCheck - startCheck;
+
+                Application.Instance.Invoke(() =>
+                {
+                    lblImageResult.Visible = true;
+                    switch(discCheckStatus)
+                    {
+                        case true:
+                            lblImageResult.Text = "Disc image checksums are correct";
+                            break;
+                        case false:
+                            lblImageResult.Text = "Disc image checksums are incorrect";
+                            break;
+                        case null:
+                            lblImageResult.Text = "Disc image does not contain checksums";
+                            break;
+                    }
+                });
+
+                correctDisc = discCheckStatus;
+                DicConsole.VerboseWriteLine("Checking disc image checksums took {0} seconds", checkTime.TotalSeconds);
+            }
+
+            if(chkVerifySectors.Checked == true)
+            {
+                DateTime    startCheck;
+                DateTime    endCheck;
+                List<ulong> failingLbas = new List<ulong>();
+                List<ulong> unknownLbas = new List<ulong>();
+
+                Application.Instance.Invoke(() =>
+                {
+                    lblProgress2.Visible       = true;
+                    prgProgress2.Indeterminate = false;
+                    prgProgress2.MaxValue      = (int)(inputFormat.Info.Sectors / 512);
+                    btnStop.Enabled            = true;
+                });
+
+                if(formatHasTracks)
+                {
+                    List<Track> inputTracks      = inputFormat.Tracks;
+                    ulong       currentSectorAll = 0;
+
+                    startCheck = DateTime.UtcNow;
+                    foreach(Track currentTrack in inputFormat.Tracks)
+                    {
+                        Application.Instance.Invoke(() =>
+                        {
+                            lblProgress.Text =
+                                $"Verifying track {currentTrack.TrackSequence} of {inputFormat.Tracks.Count}";
+                            prgProgress.Value++;
+                        });
+
+                        ulong remainingSectors = currentTrack.TrackEndSector - currentTrack.TrackStartSector;
+                        ulong currentSector    = 0;
+
+                        while(remainingSectors > 0)
+                        {
+                            if(cancel)
+                            {
+                                Application.Instance.Invoke(() =>
+                                {
+                                    btnClose.Visible = true;
+                                    btnStart.Visible = false;
+                                    btnStop.Visible  = false;
+                                });
+                                return;
+                            }
+
+                            ulong all = currentSectorAll;
+                            Application.Instance.Invoke(() =>
+                            {
+                                prgProgress2.Value = (int)(all / 512);
+                                lblProgress2.Text =
+                                    $"Checking sector {all} of {inputFormat.Info.Sectors}, on track {currentTrack.TrackSequence}";
+                            });
+
+                            List<ulong> tempfailingLbas;
+                            List<ulong> tempunknownLbas;
+
+                            if(remainingSectors < 512)
+                                inputFormat.VerifySectors(currentSector, (uint)remainingSectors,
+                                                          currentTrack.TrackSequence, out tempfailingLbas,
+                                                          out tempunknownLbas);
+                            else
+                                inputFormat.VerifySectors(currentSector, 512, currentTrack.TrackSequence,
+                                                          out tempfailingLbas, out tempunknownLbas);
+
+                            failingLbas.AddRange(tempfailingLbas);
+
+                            unknownLbas.AddRange(tempunknownLbas);
+
+                            if(remainingSectors < 512)
+                            {
+                                currentSector    += remainingSectors;
+                                currentSectorAll += remainingSectors;
+                                remainingSectors =  0;
+                            }
+                            else
+                            {
+                                currentSector    += 512;
+                                currentSectorAll += 512;
+                                remainingSectors -= 512;
+                            }
+                        }
+                    }
+
+                    endCheck = DateTime.UtcNow;
+                }
+                else
+                {
+                    ulong remainingSectors = inputFormat.Info.Sectors;
+                    ulong currentSector    = 0;
+
+                    startCheck = DateTime.UtcNow;
+                    while(remainingSectors > 0)
+                    {
+                        if(cancel)
+                        {
+                            Application.Instance.Invoke(() =>
+                            {
+                                btnClose.Visible = true;
+                                btnStart.Visible = false;
+                                btnStop.Visible  = false;
+                            });
+                            return;
+                        }
+
+                        ulong sector = currentSector;
+                        Application.Instance.Invoke(() =>
+                        {
+                            prgProgress2.Value = (int)(sector / 512);
+                            lblProgress2.Text  = $"Checking sector {sector} of {inputFormat.Info.Sectors}";
+                        });
+
+                        List<ulong> tempfailingLbas;
+                        List<ulong> tempunknownLbas;
+
+                        if(remainingSectors < 512)
+                            inputFormat.VerifySectors(currentSector, (uint)remainingSectors, out tempfailingLbas,
+                                                      out tempunknownLbas);
+                        else inputFormat.VerifySectors(currentSector, 512, out tempfailingLbas, out tempunknownLbas);
+
+                        failingLbas.AddRange(tempfailingLbas);
+
+                        unknownLbas.AddRange(tempunknownLbas);
+
+                        if(remainingSectors < 512)
+                        {
+                            currentSector    += remainingSectors;
+                            remainingSectors =  0;
+                        }
+                        else
+                        {
+                            currentSector    += 512;
+                            remainingSectors -= 512;
+                        }
+                    }
+
+                    endCheck = DateTime.UtcNow;
+                }
+
+                TimeSpan checkTime = endCheck - startCheck;
+
+                DicConsole.VerboseWriteLine("Checking sector checksums took {0} seconds", checkTime.TotalSeconds);
+
+                Application.Instance.Invoke(() =>
+                {
+                    if(failingLbas.Count > 0)
+                    {
+                        if(failingLbas.Count == (int)inputFormat.Info.Sectors)
+                        {
+                            lblSectorsErrorsAll.Visible = true;
+                            lblSectorsErrorsAll.Text    = "All sectors contain errors";
+                        }
+                        else
+                        {
+                            grpSectorErrors.Text    = "LBAs with error:";
+                            grpSectorErrors.Visible = true;
+
+                            TreeGridItemCollection errorList = new TreeGridItemCollection();
+
+                            treeSectorErrors.Columns.Add(new GridColumn
+                            {
+                                HeaderText = "LBA", DataCell = new TextBoxCell(0)
+                            });
+
+                            treeSectorErrors.AllowMultipleSelection = false;
+                            treeSectorErrors.ShowHeader             = false;
+                            treeSectorErrors.DataStore              = errorList;
+
+                            foreach(ulong t in failingLbas) errorList.Add(new TreeGridItem {Values = new object[] {t}});
+                        }
+                    }
+
+                    if(unknownLbas.Count > 0)
+                    {
+                        if(unknownLbas.Count == (int)inputFormat.Info.Sectors)
+                        {
+                            lblSectorsErrorsAll.Visible = true;
+                            lblSectorsErrorsAll.Text    = "All sectors contain errors";
+                        }
+                        else
+                        {
+                            grpSectorsUnknowns.Text    = "LBAs with error:";
+                            grpSectorsUnknowns.Visible = true;
+
+                            TreeGridItemCollection unknownList = new TreeGridItemCollection();
+
+                            treeSectorsUnknowns.Columns.Add(new GridColumn
+                            {
+                                HeaderText = "LBA", DataCell = new TextBoxCell(0)
+                            });
+
+                            treeSectorsUnknowns.AllowMultipleSelection = false;
+                            treeSectorsUnknowns.ShowHeader             = false;
+                            treeSectorsUnknowns.DataStore              = unknownList;
+
+                            foreach(ulong t in unknownLbas)
+                                unknownList.Add(new TreeGridItem {Values = new object[] {t}});
+                        }
+                    }
+
+                    grpSectorSummary.Visible    = true;
+                    lblTotalSectors.Text        = $"Total sectors........... {inputFormat.Info.Sectors}";
+                    lblTotalSectorErrors.Text   = $"Total errors............ {failingLbas.Count}";
+                    lblTotalSectorUnknowns.Text = $"Total unknowns.......... {unknownLbas.Count}";
+                    lblTotalSectorErrorsUnknowns.Text =
+                        $"Total errors+unknowns... {failingLbas.Count + unknownLbas.Count}";
+                });
+
+                totalSectors   = (long)inputFormat.Info.Sectors;
+                errorSectors   = failingLbas.Count;
+                unknownSectors = unknownLbas.Count;
+                correctSectors = totalSectors - errorSectors - unknownSectors;
+            }
+
+            Statistics.AddCommand("verify");
+            Statistics.AddVerify(correctDisc, correctSectors, errorSectors, unknownSectors, totalSectors);
+
+            Application.Instance.Invoke(() =>
+            {
+                stkOptions.Visible  = false;
+                stkResults.Visible  = true;
+                stkProgress.Visible = false;
+                btnStart.Visible    = false;
+                btnStop.Visible     = false;
+                btnClose.Visible    = true;
+            });
+        }
+
+        protected void OnBtnClose(object sender, EventArgs e)
+        {
+            Close();
+        }
+
+        protected void OnBtnStop(object sender, EventArgs e)
+        {
+            cancel          = true;
+            btnStop.Enabled = false;
+        }
+
+        #region XAML IDs
+        StackLayout  stkOptions;
+        CheckBox     chkVerifyImage;
+        CheckBox     chkVerifySectors;
+        StackLayout  stkResults;
+        StackLayout  stkSectorResults;
+        StackLayout  stkSectorErrors;
+        GroupBox     grpSectorErrors;
+        TreeGridView treeSectorErrors;
+        StackLayout  stkSectorUnknowns;
+        GroupBox     grpSectorsUnknowns;
+        TreeGridView treeSectorsUnknowns;
+        Label        lblImageResult;
+        Label        lblSectorsErrorsAll;
+        Label        lblSectorsUnknownAll;
+        GroupBox     grpSectorSummary;
+        Label        lblTotalSectors;
+        Label        lblTotalSectorErrors;
+        Label        lblTotalSectorUnknowns;
+        Label        lblTotalSectorErrorsUnknowns;
+        StackLayout  stkProgress;
+        StackLayout  stkProgress1;
+        Label        lblProgress;
+        ProgressBar  prgProgress;
+        StackLayout  stkProgress2;
+        Label        lblProgress2;
+        ProgressBar  prgProgress2;
+        Button       btnStart;
+        Button       btnClose;
+        Button       btnStop;
+        #endregion
+    }
+}
\ No newline at end of file
diff --git a/DiscImageChef.Gui/Panels/pnlImageInfo.xeto b/DiscImageChef.Gui/Panels/pnlImageInfo.xeto
index 2bb57d538..648469254 100644
--- a/DiscImageChef.Gui/Panels/pnlImageInfo.xeto
+++ b/DiscImageChef.Gui/Panels/pnlImageInfo.xeto
@@ -111,6 +111,11 @@
                 </StackLayout>
             </TabPage>
         </TabControl>
-        <Button ID="btnEntropy" Text="Calculate entropy" Click="OnBtnEntropy"/>
+        <StackLayoutItem HorizontalAlignment="Right" Expand="True">
+            <StackLayout Orientation="Horizontal">
+                <Button ID="btnEntropy" Text="Calculate entropy" Click="OnBtnEntropy"/>
+                <Button ID="btnVerify" Text="Verify" Click="OnBtnVerify"/>
+            </StackLayout>
+        </StackLayoutItem>
     </StackLayout>
 </Panel>
\ No newline at end of file
diff --git a/DiscImageChef.Gui/Panels/pnlImageInfo.xeto.cs b/DiscImageChef.Gui/Panels/pnlImageInfo.xeto.cs
index 1beb9a16d..f024b7f71 100644
--- a/DiscImageChef.Gui/Panels/pnlImageInfo.xeto.cs
+++ b/DiscImageChef.Gui/Panels/pnlImageInfo.xeto.cs
@@ -56,6 +56,7 @@ namespace DiscImageChef.Gui.Panels
     public class pnlImageInfo : Panel
     {
         frmImageEntropy frmImageEntropy;
+        frmImageVerify  frmImageVerify;
         IMediaImage     imageFormat;
 
         public pnlImageInfo(string imagePath, IFilter filter, IMediaImage imageFormat)
@@ -798,6 +799,19 @@ namespace DiscImageChef.Gui.Panels
             frmImageEntropy.Show();
         }
 
+        protected void OnBtnVerify(object sender, EventArgs e)
+        {
+            if(frmImageVerify != null)
+            {
+                frmImageVerify.Show();
+                return;
+            }
+
+            frmImageVerify        =  new frmImageVerify(imageFormat);
+            frmImageVerify.Closed += (s, ea) => { frmImageVerify = null; };
+            frmImageVerify.Show();
+        }
+
         #region XAML controls
         #pragma warning disable 169
         #pragma warning disable 649