// /***************************************************************************
// Aaru Data Preservation Suite
// ----------------------------------------------------------------------------
//
// Filename       : ATA.cs
// Author(s)      : Natalia Portillo <claunia@claunia.com>
//
// Component      : Core algorithms.
//
// --[ Description ] ----------------------------------------------------------
//
//     Creates reports from ATA devices.
//
// --[ 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-2022 Natalia Portillo
// ****************************************************************************/

namespace Aaru.Core.Devices.Report;

using System;
using Aaru.CommonTypes.Metadata;
using Aaru.Console;
using Aaru.Decoders.ATA;
using Aaru.Devices;
using global::Spectre.Console;
using Identify = Aaru.CommonTypes.Structs.Devices.ATA.Identify;

public sealed partial class DeviceReport
{
    /// <summary>Creates a report for media inserted into an ATA device</summary>
    /// <returns>Media report</returns>
    public TestedMedia ReportAtaMedia()
    {
        var                    sense      = true;
        AtaErrorRegistersChs   errorChs   = new();
        AtaErrorRegistersLba28 errorLba   = new();
        AtaErrorRegistersLba48 errorLba48 = new();
        byte[]                 buffer     = Array.Empty<byte>();
        byte[]                 readBuf    = Array.Empty<byte>();

        var mediaTest = new TestedMedia
        {
            MediumTypeName = AnsiConsole.Ask<string>("Please write a description of the media type and press enter: "),
            Model = AnsiConsole.Ask<string>("Please write the media model and press enter: "),
            MediaIsRecognized = true
        };

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Querying ATA IDENTIFY...").IsIndeterminate();
            _dev.AtaIdentify(out buffer, out _, _dev.Timeout, out _);
        });

        mediaTest.IdentifyData   = ClearIdentify(buffer);
        mediaTest.IdentifyDevice = Identify.Decode(buffer);

        if(mediaTest.IdentifyDevice.HasValue)
        {
            Identify.IdentifyDevice ataId = mediaTest.IdentifyDevice.Value;

            if(ataId.UnformattedBPT != 0)
                mediaTest.UnformattedBPT = ataId.UnformattedBPT;

            if(ataId.UnformattedBPS != 0)
                mediaTest.UnformattedBPS = ataId.UnformattedBPS;

            if(ataId.Cylinders       > 0 &&
               ataId.Heads           > 0 &&
               ataId.SectorsPerTrack > 0)
            {
                mediaTest.CHS = new Chs
                {
                    Cylinders = ataId.Cylinders,
                    Heads     = ataId.Heads,
                    Sectors   = ataId.SectorsPerTrack
                };

                mediaTest.Blocks = (ulong)(ataId.Cylinders * ataId.Heads * ataId.SectorsPerTrack);
            }

            if(ataId.CurrentCylinders       > 0 &&
               ataId.CurrentHeads           > 0 &&
               ataId.CurrentSectorsPerTrack > 0)
            {
                mediaTest.CurrentCHS = new Chs
                {
                    Cylinders = ataId.CurrentCylinders,
                    Heads     = ataId.CurrentHeads,
                    Sectors   = ataId.CurrentSectorsPerTrack
                };

                if(mediaTest.Blocks == 0)
                    mediaTest.Blocks =
                        (ulong)(ataId.CurrentCylinders * ataId.CurrentHeads * ataId.CurrentSectorsPerTrack);
            }

            if(ataId.Capabilities.HasFlag(Identify.CapabilitiesBit.LBASupport))
            {
                mediaTest.LBASectors = ataId.LBASectors;
                mediaTest.Blocks     = ataId.LBASectors;
            }

            if(ataId.CommandSet2.HasFlag(Identify.CommandSetBit2.LBA48))
            {
                mediaTest.LBA48Sectors = ataId.LBA48Sectors;
                mediaTest.Blocks       = ataId.LBA48Sectors;
            }

            if(ataId.NominalRotationRate != 0x0000 &&
               ataId.NominalRotationRate != 0xFFFF)
                if(ataId.NominalRotationRate == 0x0001)
                    mediaTest.SolidStateDevice = true;
                else
                {
                    mediaTest.SolidStateDevice    = false;
                    mediaTest.NominalRotationRate = ataId.NominalRotationRate;
                }

            uint logicalSectorSize;
            uint physicalSectorSize;

            if((ataId.PhysLogSectorSize & 0x8000) == 0x0000 &&
               (ataId.PhysLogSectorSize & 0x4000) == 0x4000)
            {
                if((ataId.PhysLogSectorSize & 0x1000) == 0x1000)
                    if(ataId.LogicalSectorWords <= 255 ||
                       ataId.LogicalAlignment   == 0xFFFF)
                        logicalSectorSize = 512;
                    else
                        logicalSectorSize = ataId.LogicalSectorWords * 2;
                else
                    logicalSectorSize = 512;

                if((ataId.PhysLogSectorSize & 0x2000) == 0x2000)
                    physicalSectorSize = (uint)(logicalSectorSize * ((1 << ataId.PhysLogSectorSize) & 0xF));
                else
                    physicalSectorSize = logicalSectorSize;
            }
            else
            {
                logicalSectorSize  = 512;
                physicalSectorSize = 512;
            }

            mediaTest.BlockSize = logicalSectorSize;

            if(physicalSectorSize != logicalSectorSize)
            {
                mediaTest.PhysicalBlockSize = physicalSectorSize;

                if((ataId.LogicalAlignment & 0x8000) == 0x0000 &&
                   (ataId.LogicalAlignment & 0x4000) == 0x4000)
                    mediaTest.LogicalAlignment = (ushort)(ataId.LogicalAlignment & 0x3FFF);
            }

            if(ataId.EccBytes != 0x0000 &&
               ataId.EccBytes != 0xFFFF)
                mediaTest.LongBlockSize = logicalSectorSize + ataId.EccBytes;

            if(ataId.UnformattedBPS > logicalSectorSize &&
               (!(ataId.EccBytes != 0x0000 && ataId.EccBytes != 0xFFFF) || mediaTest.LongBlockSize == 516))
                mediaTest.LongBlockSize = ataId.UnformattedBPS;

            if(ataId.CommandSet3.HasFlag(Identify.CommandSetBit3.MustBeSet)    &&
               !ataId.CommandSet3.HasFlag(Identify.CommandSetBit3.MustBeClear) &&
               ataId.EnabledCommandSet3.HasFlag(Identify.CommandSetBit3.MediaSerial))
            {
                mediaTest.CanReadMediaSerial = true;

                if(!string.IsNullOrWhiteSpace(ataId.MediaManufacturer))
                    mediaTest.Manufacturer = ataId.MediaManufacturer;
            }

            var checkCorrectRead = BitConverter.ToUInt64(buffer, 0);

            Spectre.ProgressSingleSpinner(ctx =>
            {
                ctx.AddTask("Trying READ SECTOR(S) in CHS mode...").IsIndeterminate();
                sense = _dev.Read(out readBuf, out errorChs, false, 0, 0, 1, 1, _dev.Timeout, out _);
            });

            mediaTest.SupportsReadSectors = !sense && (errorChs.Status & 0x01) != 0x01 && errorChs.Error == 0 &&
                                            readBuf.Length                     > 0;

            AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                       sense, errorChs.Status, errorChs.Error, readBuf.Length);

            mediaTest.ReadSectorsData = readBuf;

            Spectre.ProgressSingleSpinner(ctx =>
            {
                ctx.AddTask("Trying READ SECTOR(S) RETRY in CHS mode...").IsIndeterminate();
                sense = _dev.Read(out readBuf, out errorChs, true, 0, 0, 1, 1, _dev.Timeout, out _);
            });

            mediaTest.SupportsReadRetry = !sense && (errorChs.Status & 0x01) != 0x01 && errorChs.Error == 0 &&
                                          readBuf.Length                     > 0;

            AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                       sense, errorChs.Status, errorChs.Error, readBuf.Length);

            mediaTest.ReadSectorsRetryData = readBuf;

            Spectre.ProgressSingleSpinner(ctx =>
            {
                ctx.AddTask("Trying READ DMA in CHS mode...").IsIndeterminate();
                sense = _dev.ReadDma(out readBuf, out errorChs, false, 0, 0, 1, 1, _dev.Timeout, out _);
            });

            mediaTest.SupportsReadDma = !sense && (errorChs.Status & 0x01) != 0x01 && errorChs.Error == 0 &&
                                        readBuf.Length                     > 0;

            AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                       sense, errorChs.Status, errorChs.Error, readBuf.Length);

            mediaTest.ReadDmaData = readBuf;

            Spectre.ProgressSingleSpinner(ctx =>
            {
                ctx.AddTask("Trying READ DMA RETRY in CHS mode...").IsIndeterminate();
                sense = _dev.ReadDma(out readBuf, out errorChs, true, 0, 0, 1, 1, _dev.Timeout, out _);
            });

            mediaTest.SupportsReadDmaRetry = !sense && (errorChs.Status & 0x01) != 0x01 && errorChs.Error == 0 &&
                                             readBuf.Length                     > 0;

            AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                       sense, errorChs.Status, errorChs.Error, readBuf.Length);

            mediaTest.ReadDmaRetryData = readBuf;

            Spectre.ProgressSingleSpinner(ctx =>
            {
                ctx.AddTask("Trying SEEK in CHS mode...").IsIndeterminate();
                sense = _dev.Seek(out errorChs, 0, 0, 1, _dev.Timeout, out _);
            });

            mediaTest.SupportsSeek = !sense && (errorChs.Status & 0x01) != 0x01 && errorChs.Error == 0;

            AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}", sense,
                                       errorChs.Status, errorChs.Error);

            Spectre.ProgressSingleSpinner(ctx =>
            {
                ctx.AddTask("Trying READ SECTOR(S) in LBA mode...").IsIndeterminate();
                sense = _dev.Read(out readBuf, out errorLba, false, 0, 1, _dev.Timeout, out _);
            });

            mediaTest.SupportsReadLba = !sense && (errorLba.Status & 0x01) != 0x01 && errorLba.Error == 0 &&
                                        readBuf.Length                     > 0;

            AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                       sense, errorChs.Status, errorChs.Error, readBuf.Length);

            mediaTest.ReadLbaData = readBuf;

            Spectre.ProgressSingleSpinner(ctx =>
            {
                ctx.AddTask("Trying READ SECTOR(S) RETRY in LBA mode...").IsIndeterminate();
                sense = _dev.Read(out readBuf, out errorLba, true, 0, 1, _dev.Timeout, out _);
            });

            mediaTest.SupportsReadRetryLba = !sense && (errorLba.Status & 0x01) != 0x01 && errorLba.Error == 0 &&
                                             readBuf.Length                     > 0;

            AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                       sense, errorChs.Status, errorChs.Error, readBuf.Length);

            mediaTest.ReadRetryLbaData = readBuf;

            Spectre.ProgressSingleSpinner(ctx =>
            {
                ctx.AddTask("Trying READ DMA in LBA mode...").IsIndeterminate();
                sense = _dev.ReadDma(out readBuf, out errorLba, false, 0, 1, _dev.Timeout, out _);
            });

            mediaTest.SupportsReadDmaLba = !sense && (errorLba.Status & 0x01) != 0x01 && errorLba.Error == 0 &&
                                           readBuf.Length                     > 0;

            AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                       sense, errorChs.Status, errorChs.Error, readBuf.Length);

            mediaTest.ReadDmaLbaData = readBuf;

            Spectre.ProgressSingleSpinner(ctx =>
            {
                ctx.AddTask("Trying READ DMA RETRY in LBA mode...").IsIndeterminate();
                sense = _dev.ReadDma(out readBuf, out errorLba, true, 0, 1, _dev.Timeout, out _);
            });

            mediaTest.SupportsReadDmaRetryLba = !sense && (errorLba.Status & 0x01) != 0x01 && errorLba.Error == 0 &&
                                                readBuf.Length                     > 0;

            AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                       sense, errorChs.Status, errorChs.Error, readBuf.Length);

            mediaTest.ReadDmaRetryLbaData = readBuf;

            Spectre.ProgressSingleSpinner(ctx =>
            {
                ctx.AddTask("Trying SEEK in LBA mode...").IsIndeterminate();
                sense = _dev.Seek(out errorLba, 0, _dev.Timeout, out _);
            });

            mediaTest.SupportsSeekLba = !sense && (errorLba.Status & 0x01) != 0x01 && errorLba.Error == 0;

            AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}", sense,
                                       errorChs.Status, errorChs.Error);

            Spectre.ProgressSingleSpinner(ctx =>
            {
                ctx.AddTask("Trying READ SECTOR(S) in LBA48 mode...").IsIndeterminate();
                sense = _dev.Read(out readBuf, out AtaErrorRegistersLba48 errorLba48, 0, 1, _dev.Timeout, out _);
            });

            mediaTest.SupportsReadLba48 = !sense && (errorLba48.Status & 0x01) != 0x01 && errorLba48.Error == 0 &&
                                          readBuf.Length                       > 0;

            AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                       sense, errorChs.Status, errorChs.Error, readBuf.Length);

            mediaTest.ReadLba48Data = readBuf;

            Spectre.ProgressSingleSpinner(ctx =>
            {
                ctx.AddTask("Trying READ DMA in LBA48 mode...").IsIndeterminate();
                sense = _dev.ReadDma(out readBuf, out errorLba48, 0, 1, _dev.Timeout, out _);
            });

            mediaTest.SupportsReadDmaLba48 = !sense && (errorLba48.Status & 0x01) != 0x01 && errorLba48.Error == 0 &&
                                             readBuf.Length                       > 0;

            AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                       sense, errorChs.Status, errorChs.Error, readBuf.Length);

            mediaTest.ReadDmaLba48Data = readBuf;

            // Send SET FEATURES before sending READ LONG commands, retrieve IDENTIFY again and
            // check if ECC size changed. Sector is set to 1 because without it most drives just return
            // CORRECTABLE ERROR for this command.
            _dev.SetFeatures(out _, AtaFeatures.EnableReadLongVendorLength, 0, 0, 1, 0, _dev.Timeout, out _);

            _dev.AtaIdentify(out buffer, out _, _dev.Timeout, out _);

            if(Identify.Decode(buffer).HasValue)
            {
                ataId = Identify.Decode(buffer).Value;

                if(ataId.EccBytes != 0x0000 &&
                   ataId.EccBytes != 0xFFFF)
                    mediaTest.LongBlockSize = logicalSectorSize + ataId.EccBytes;

                if(ataId.UnformattedBPS > logicalSectorSize &&
                   (!(ataId.EccBytes != 0x0000 && ataId.EccBytes != 0xFFFF) || mediaTest.LongBlockSize == 516))
                    mediaTest.LongBlockSize = ataId.UnformattedBPS;
            }

            Spectre.ProgressSingleSpinner(ctx =>
            {
                ctx.AddTask("Trying READ LONG in CHS mode...").IsIndeterminate();

                sense = _dev.ReadLong(out readBuf, out errorChs, false, 0, 0, 1, mediaTest.LongBlockSize ?? 0,
                                      _dev.Timeout, out _);
            });

            mediaTest.SupportsReadLong = !sense && (errorChs.Status & 0x01) != 0x01 && errorChs.Error == 0 &&
                                         readBuf.Length > 0 && BitConverter.ToUInt64(readBuf, 0) != checkCorrectRead;

            AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                       sense, errorChs.Status, errorChs.Error, readBuf.Length);

            mediaTest.ReadLongData = readBuf;

            Spectre.ProgressSingleSpinner(ctx =>
            {
                ctx.AddTask("Trying READ LONG RETRY in CHS mode...").IsIndeterminate();

                sense = _dev.ReadLong(out readBuf, out errorChs, true, 0, 0, 1, mediaTest.LongBlockSize ?? 0,
                                      _dev.Timeout, out _);
            });

            mediaTest.SupportsReadLongRetry = !sense && (errorChs.Status & 0x01) != 0x01 && errorChs.Error == 0 &&
                                              readBuf.Length > 0 && BitConverter.ToUInt64(readBuf, 0) !=
                                              checkCorrectRead;

            AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                       sense, errorChs.Status, errorChs.Error, readBuf.Length);

            mediaTest.ReadLongRetryData = readBuf;

            Spectre.ProgressSingleSpinner(ctx =>
            {
                ctx.AddTask("Trying READ LONG in LBA mode...").IsIndeterminate();

                sense = _dev.ReadLong(out readBuf, out errorLba, false, 0, mediaTest.LongBlockSize ?? 0, _dev.Timeout,
                                      out _);
            });

            mediaTest.SupportsReadLongLba = !sense && (errorLba.Status & 0x01) != 0x01 && errorLba.Error == 0 &&
                                            readBuf.Length > 0 && BitConverter.ToUInt64(readBuf, 0) != checkCorrectRead;

            AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                       sense, errorChs.Status, errorChs.Error, readBuf.Length);

            mediaTest.ReadLongLbaData = readBuf;

            Spectre.ProgressSingleSpinner(ctx =>
            {
                ctx.AddTask("Trying READ LONG RETRY in LBA mode...").IsIndeterminate();

                sense = _dev.ReadLong(out readBuf, out errorLba, true, 0, mediaTest.LongBlockSize ?? 0, _dev.Timeout,
                                      out _);
            });

            mediaTest.SupportsReadLongRetryLba = !sense && (errorLba.Status & 0x01) != 0x01 && errorLba.Error == 0 &&
                                                 readBuf.Length > 0 && BitConverter.ToUInt64(readBuf, 0) !=
                                                 checkCorrectRead;

            AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                       sense, errorChs.Status, errorChs.Error, readBuf.Length);

            mediaTest.ReadLongRetryLbaData = readBuf;
        }
        else
            mediaTest.MediaIsRecognized = false;

        return mediaTest;
    }

    /// <summary>Creates a report of an ATA device</summary>
    public TestedMedia ReportAta(Identify.IdentifyDevice ataId)
    {
        var                    sense        = true;
        byte[]                 readBuf      = Array.Empty<byte>();
        AtaErrorRegistersChs   errorChs     = new();
        AtaErrorRegistersLba28 errorLba     = new();
        AtaErrorRegistersLba48 errorLba48   = new();
        var                    capabilities = new TestedMedia();

        if(ataId.UnformattedBPT != 0)
            capabilities.UnformattedBPT = ataId.UnformattedBPT;

        if(ataId.UnformattedBPS != 0)
            capabilities.UnformattedBPS = ataId.UnformattedBPS;

        if(ataId.Cylinders       > 0 &&
           ataId.Heads           > 0 &&
           ataId.SectorsPerTrack > 0)
        {
            capabilities.CHS = new Chs
            {
                Cylinders = ataId.Cylinders,
                Heads     = ataId.Heads,
                Sectors   = ataId.SectorsPerTrack
            };

            capabilities.Blocks = (ulong)(ataId.Cylinders * ataId.Heads * ataId.SectorsPerTrack);
        }

        if(ataId.CurrentCylinders       > 0 &&
           ataId.CurrentHeads           > 0 &&
           ataId.CurrentSectorsPerTrack > 0)
        {
            capabilities.CurrentCHS = new Chs
            {
                Cylinders = ataId.CurrentCylinders,
                Heads     = ataId.CurrentHeads,
                Sectors   = ataId.CurrentSectorsPerTrack
            };

            capabilities.Blocks = (ulong)(ataId.CurrentCylinders * ataId.CurrentHeads * ataId.CurrentSectorsPerTrack);
        }

        if(ataId.Capabilities.HasFlag(Identify.CapabilitiesBit.LBASupport))
        {
            capabilities.LBASectors = ataId.LBASectors;
            capabilities.Blocks     = ataId.LBASectors;
        }

        if(ataId.CommandSet2.HasFlag(Identify.CommandSetBit2.LBA48))
        {
            capabilities.LBA48Sectors = ataId.LBA48Sectors;
            capabilities.Blocks       = ataId.LBA48Sectors;
        }

        if(ataId.NominalRotationRate != 0x0000 &&
           ataId.NominalRotationRate != 0xFFFF)
            if(ataId.NominalRotationRate == 0x0001)
                capabilities.SolidStateDevice = true;
            else
            {
                capabilities.SolidStateDevice    = false;
                capabilities.NominalRotationRate = ataId.NominalRotationRate;
            }

        uint logicalSectorSize;
        uint physicalSectorSize;

        if((ataId.PhysLogSectorSize & 0x8000) == 0x0000 &&
           (ataId.PhysLogSectorSize & 0x4000) == 0x4000)
        {
            if((ataId.PhysLogSectorSize & 0x1000) == 0x1000)
                if(ataId.LogicalSectorWords <= 255 ||
                   ataId.LogicalAlignment   == 0xFFFF)
                    logicalSectorSize = 512;
                else
                    logicalSectorSize = ataId.LogicalSectorWords * 2;
            else
                logicalSectorSize = 512;

            if((ataId.PhysLogSectorSize & 0x2000) == 0x2000)
                physicalSectorSize = logicalSectorSize * (uint)Math.Pow(2, ataId.PhysLogSectorSize & 0xF);
            else
                physicalSectorSize = logicalSectorSize;
        }
        else
        {
            logicalSectorSize  = 512;
            physicalSectorSize = 512;
        }

        capabilities.BlockSize = logicalSectorSize;

        if(physicalSectorSize != logicalSectorSize)
        {
            capabilities.PhysicalBlockSize = physicalSectorSize;

            if((ataId.LogicalAlignment & 0x8000) == 0x0000 &&
               (ataId.LogicalAlignment & 0x4000) == 0x4000)
                capabilities.LogicalAlignment = (ushort)(ataId.LogicalAlignment & 0x3FFF);
        }

        if(ataId.EccBytes != 0x0000 &&
           ataId.EccBytes != 0xFFFF)
            capabilities.LongBlockSize = logicalSectorSize + ataId.EccBytes;

        if(ataId.UnformattedBPS > logicalSectorSize &&
           (!(ataId.EccBytes != 0x0000 && ataId.EccBytes != 0xFFFF) || capabilities.LongBlockSize == 516))
            capabilities.LongBlockSize = ataId.UnformattedBPS;

        if(ataId.CommandSet3.HasFlag(Identify.CommandSetBit3.MustBeSet)    &&
           !ataId.CommandSet3.HasFlag(Identify.CommandSetBit3.MustBeClear) &&
           ataId.EnabledCommandSet3.HasFlag(Identify.CommandSetBit3.MediaSerial))
        {
            capabilities.CanReadMediaSerial = true;

            if(!string.IsNullOrWhiteSpace(ataId.MediaManufacturer))
                capabilities.Manufacturer = ataId.MediaManufacturer;
        }

        ulong checkCorrectRead = 0;

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Trying READ SECTOR(S) in CHS mode...").IsIndeterminate();
            sense = _dev.Read(out readBuf, out errorChs, false, 0, 0, 1, 1, _dev.Timeout, out _);
        });

        capabilities.SupportsReadSectors = !sense && (errorChs.Status & 0x01) != 0x01 && errorChs.Error == 0 &&
                                           readBuf.Length                     > 0;

        AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                   sense, errorChs.Status, errorChs.Error, readBuf.Length);

        capabilities.ReadSectorsData = readBuf;

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Trying READ SECTOR(S) RETRY in CHS mode...").IsIndeterminate();
            sense = _dev.Read(out readBuf, out errorChs, true, 0, 0, 1, 1, _dev.Timeout, out _);
        });

        capabilities.SupportsReadRetry =
            !sense && (errorChs.Status & 0x01) != 0x01 && errorChs.Error == 0 && readBuf.Length > 0;

        AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                   sense, errorChs.Status, errorChs.Error, readBuf.Length);

        capabilities.ReadSectorsRetryData = readBuf;

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Trying READ DMA in CHS mode...").IsIndeterminate();
            sense = _dev.ReadDma(out readBuf, out errorChs, false, 0, 0, 1, 1, _dev.Timeout, out _);
        });

        capabilities.SupportsReadDma =
            !sense && (errorChs.Status & 0x01) != 0x01 && errorChs.Error == 0 && readBuf.Length > 0;

        AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                   sense, errorChs.Status, errorChs.Error, readBuf.Length);

        capabilities.ReadDmaData = readBuf;

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Trying READ DMA RETRY in CHS mode...").IsIndeterminate();
            sense = _dev.ReadDma(out readBuf, out errorChs, true, 0, 0, 1, 1, _dev.Timeout, out _);
        });

        capabilities.SupportsReadDmaRetry = !sense && (errorChs.Status & 0x01) != 0x01 && errorChs.Error == 0 &&
                                            readBuf.Length                     > 0;

        AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                   sense, errorChs.Status, errorChs.Error, readBuf.Length);

        capabilities.ReadDmaRetryData = readBuf;

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Trying SEEK in CHS mode...").IsIndeterminate();
            sense = _dev.Seek(out errorChs, 0, 0, 1, _dev.Timeout, out _);
        });

        capabilities.SupportsSeek = !sense && (errorChs.Status & 0x01) != 0x01 && errorChs.Error == 0;

        AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}", sense,
                                   errorChs.Status, errorChs.Error);

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Trying READ SECTOR(S) in LBA mode...").IsIndeterminate();
            sense = _dev.Read(out readBuf, out errorLba, false, 0, 1, _dev.Timeout, out _);
        });

        capabilities.SupportsReadLba =
            !sense && (errorLba.Status & 0x01) != 0x01 && errorLba.Error == 0 && readBuf.Length > 0;

        AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                   sense, errorLba.Status, errorLba.Error, readBuf.Length);

        capabilities.ReadLbaData = readBuf;

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Trying READ SECTOR(S) RETRY in LBA mode...").IsIndeterminate();
            sense = _dev.Read(out readBuf, out errorLba, true, 0, 1, _dev.Timeout, out _);
        });

        capabilities.SupportsReadRetryLba = !sense && (errorLba.Status & 0x01) != 0x01 && errorLba.Error == 0 &&
                                            readBuf.Length                     > 0;

        AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                   sense, errorLba.Status, errorLba.Error, readBuf.Length);

        capabilities.ReadRetryLbaData = readBuf;

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Trying READ DMA in LBA mode...").IsIndeterminate();
            sense = _dev.ReadDma(out readBuf, out errorLba, false, 0, 1, _dev.Timeout, out _);
        });

        capabilities.SupportsReadDmaLba = !sense && (errorLba.Status & 0x01) != 0x01 && errorLba.Error == 0 &&
                                          readBuf.Length                     > 0;

        AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                   sense, errorLba.Status, errorLba.Error, readBuf.Length);

        capabilities.ReadDmaLbaData = readBuf;

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Trying READ DMA RETRY in LBA mode...").IsIndeterminate();
            sense = _dev.ReadDma(out readBuf, out errorLba, true, 0, 1, _dev.Timeout, out _);
        });

        capabilities.SupportsReadDmaRetryLba =
            !sense && (errorLba.Status & 0x01) != 0x01 && errorLba.Error == 0 && readBuf.Length > 0;

        AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                   sense, errorLba.Status, errorLba.Error, readBuf.Length);

        capabilities.ReadDmaRetryLbaData = readBuf;

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Trying SEEK in LBA mode...").IsIndeterminate();
            sense = _dev.Seek(out errorLba, 0, _dev.Timeout, out _);
        });

        capabilities.SupportsSeekLba = !sense && (errorLba.Status & 0x01) != 0x01 && errorLba.Error == 0;

        AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}", sense,
                                   errorLba.Status, errorLba.Error);

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Trying READ SECTOR(S) in LBA48 mode...").IsIndeterminate();
            sense = _dev.Read(out readBuf, out errorLba48, 0, 1, _dev.Timeout, out _);
        });

        capabilities.SupportsReadLba48 = !sense && (errorLba48.Status & 0x01) != 0x01 && errorLba48.Error == 0 &&
                                         readBuf.Length                       > 0;

        AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                   sense, errorLba48.Status, errorLba48.Error, readBuf.Length);

        capabilities.ReadLba48Data = readBuf;

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Trying READ DMA in LBA48 mode...").IsIndeterminate();
            sense = _dev.ReadDma(out readBuf, out errorLba48, 0, 1, _dev.Timeout, out _);
        });

        capabilities.SupportsReadDmaLba48 = !sense && (errorLba48.Status & 0x01) != 0x01 && errorLba48.Error == 0 &&
                                            readBuf.Length                       > 0;

        AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                   sense, errorLba48.Status, errorLba48.Error, readBuf.Length);

        capabilities.ReadDmaLba48Data = readBuf;

        // Send SET FEATURES before sending READ LONG commands, retrieve IDENTIFY again and
        // check if ECC size changed. Sector is set to 1 because without it most drives just return
        // CORRECTABLE ERROR for this command.
        _dev.SetFeatures(out _, AtaFeatures.EnableReadLongVendorLength, 0, 0, 1, 0, _dev.Timeout, out _);

        _dev.AtaIdentify(out byte[] buffer, out _, _dev.Timeout, out _);

        if(Identify.Decode(buffer).HasValue)
        {
            ataId = Identify.Decode(buffer).Value;

            if(ataId.EccBytes != 0x0000 &&
               ataId.EccBytes != 0xFFFF)
                capabilities.LongBlockSize = logicalSectorSize + ataId.EccBytes;

            if(ataId.UnformattedBPS > logicalSectorSize &&
               (!(ataId.EccBytes != 0x0000 && ataId.EccBytes != 0xFFFF) || capabilities.LongBlockSize == 516))
                capabilities.LongBlockSize = ataId.UnformattedBPS;
        }

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Trying READ LONG in CHS mode...").IsIndeterminate();

            sense = _dev.ReadLong(out readBuf, out errorChs, false, 0, 0, 1, capabilities.LongBlockSize ?? 0,
                                  _dev.Timeout, out _);
        });

        capabilities.SupportsReadLong = !sense && (errorChs.Status & 0x01) != 0x01 && errorChs.Error == 0 &&
                                        readBuf.Length > 0 && BitConverter.ToUInt64(readBuf, 0) != checkCorrectRead;

        AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                   sense, errorChs.Status, errorChs.Error, readBuf.Length);

        capabilities.ReadLongData = readBuf;

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Trying READ LONG RETRY in CHS mode...").IsIndeterminate();

            sense = _dev.ReadLong(out readBuf, out errorChs, true, 0, 0, 1, capabilities.LongBlockSize ?? 0,
                                  _dev.Timeout, out _);
        });

        capabilities.SupportsReadLongRetry = !sense && (errorChs.Status & 0x01) != 0x01 && errorChs.Error == 0 &&
                                             readBuf.Length                     > 0     &&
                                             BitConverter.ToUInt64(readBuf, 0)  != checkCorrectRead;

        AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                   sense, errorChs.Status, errorChs.Error, readBuf.Length);

        capabilities.ReadLongRetryData = readBuf;

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Trying READ LONG in LBA mode...").IsIndeterminate();

            sense = _dev.ReadLong(out readBuf, out errorLba, false, 0, capabilities.LongBlockSize ?? 0, _dev.Timeout,
                                  out _);
        });

        capabilities.SupportsReadLongLba = !sense && (errorLba.Status & 0x01) != 0x01 && errorLba.Error == 0 &&
                                           readBuf.Length > 0 && BitConverter.ToUInt64(readBuf, 0) != checkCorrectRead;

        AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                   sense, errorLba.Status, errorLba.Error, readBuf.Length);

        capabilities.ReadLongLbaData = readBuf;

        Spectre.ProgressSingleSpinner(ctx =>
        {
            ctx.AddTask("Trying READ LONG RETRY in LBA mode...").IsIndeterminate();

            sense = _dev.ReadLong(out readBuf, out errorLba, true, 0, capabilities.LongBlockSize ?? 0, _dev.Timeout,
                                  out _);
        });

        capabilities.SupportsReadLongRetryLba = !sense && (errorLba.Status & 0x01) != 0x01 && errorLba.Error == 0 &&
                                                readBuf.Length > 0 && BitConverter.ToUInt64(readBuf, 0) !=
                                                checkCorrectRead;

        AaruConsole.DebugWriteLine("ATA Report", "Sense = {0}, Status = 0x{1:X2}, Error = 0x{2:X2}, Length = {3}",
                                   sense, errorLba.Status, errorLba.Error, readBuf.Length);

        capabilities.ReadLongRetryLbaData = readBuf;

        return capabilities;
    }

    /// <summary>Clear serial numbers and other private fields from an IDENTIFY ATA DEVICE response</summary>
    /// <param name="buffer">IDENTIFY ATA DEVICE response</param>
    /// <returns>IDENTIFY ATA DEVICE response without the private fields</returns>
    public static byte[] ClearIdentify(byte[] buffer)
    {
        var empty = new byte[512];

        Array.Copy(empty, 0, buffer, 20, 20);
        Array.Copy(empty, 0, buffer, 216, 8);
        Array.Copy(empty, 0, buffer, 224, 8);
        Array.Copy(empty, 0, buffer, 352, 40);

        return buffer;
    }
}