Aaru/DiscImageChef.Core/Devices/Dumping/ResumeSupport.cs

// /***************************************************************************
// The Disc Image Chef
// ----------------------------------------------------------------------------
//
// Filename       : ResumeSupport.cs
// Author(s)      : Natalia Portillo <claunia@claunia.com>
//
// Component      : Core algorithms.
//
// --[ Description ] ----------------------------------------------------------
//
//     Contains logic to support dump resuming.
//
// --[ 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 DiscImageChef.Metadata;
using Extents;
using Schemas;

namespace DiscImageChef.Core.Devices.Dumping
{
    static class ResumeSupport
    {
        internal static void Process(bool isLba, bool removable, ulong blocks, string Manufacturer, string Model,
                                   string Serial, Interop.PlatformID platform, ref Resume resume,
                                   ref DumpHardwareType currentTry, ref ExtentsULong extents)
        {
            if(resume != null)
            {
                if(!isLba) throw new NotImplementedException("Resuming CHS devices is currently not supported.");

                if(resume.Removable != removable)
                    throw new
                        Exception(string.Format("Resume file specifies a {0} device but you're requesting to dump a {1} device, not continuing...",
                                                resume.Removable ? "removable" : "non removable",
                                                removable ? "removable" : "non removable"));

                if(resume.LastBlock != blocks - 1)
                    throw new
                        Exception(string.Format("Resume file specifies a device with {0} blocks but you're requesting to dump one with {1} blocks, not continuing...",
                                                resume.LastBlock + 1, blocks));

                foreach(DumpHardwareType oldtry in resume.Tries)
                {
                    if(oldtry.Manufacturer != Manufacturer && !removable)
                        throw new
                            Exception(string.Format("Resume file specifies a device manufactured by {0} but you're requesting to dump one by {1}, not continuing...",
                                                    oldtry.Manufacturer, Manufacturer));

                    if(oldtry.Model != Model && !removable)
                        throw new
                            Exception(string.Format("Resume file specifies a device model {0} but you're requesting to dump model {1}, not continuing...",
                                                    oldtry.Model, Model));

                    if(oldtry.Serial != Serial && !removable)
                        throw new
                            Exception(string.Format("Resume file specifies a device with serial {0} but you're requesting to dump one with serial {1}, not continuing...",
                                                    oldtry.Serial, Serial));

                    if(oldtry.Software == null) throw new Exception("Found corrupt resume file, cannot continue...");

                    if(oldtry.Software.Name == "DiscImageChef" &&
                       oldtry.Software.OperatingSystem == platform.ToString() &&
                       oldtry.Software.Version == Version.GetVersion())
                    {
                        if(removable && (oldtry.Manufacturer != Manufacturer || oldtry.Model != Model ||
                                         oldtry.Serial != Serial)) continue;

                        currentTry = oldtry;
                        extents = ExtentsConverter.FromMetadata(currentTry.Extents);
                        break;
                    }
                }

                if(currentTry == null)
                {
                    currentTry = new DumpHardwareType
                    {
                        Software = Version.GetSoftwareType(platform),
                        Manufacturer = Manufacturer,
                        Model = Model,
                        Serial = Serial,
                    };
                    resume.Tries.Add(currentTry);
                    extents = new ExtentsULong();
                }
            }
            else
            {
                resume = new Resume
                {
                    Tries = new List<DumpHardwareType>(),
                    CreationDate = DateTime.UtcNow,
                    BadBlocks = new List<ulong>(),
                    LastBlock = blocks - 1
                };
                currentTry = new DumpHardwareType
                {
                    Software = Version.GetSoftwareType(platform),
                    Manufacturer = Manufacturer,
                    Model = Model,
                    Serial = Serial
                };
                resume.Tries.Add(currentTry);
                extents = new ExtentsULong();
                resume.Removable = removable;
            }
        }
    }
}