using System;
#if NET40_OR_GREATER || NETCOREAPP
using System.Collections.Concurrent;
#endif
using System.Collections.Generic;
using System.Linq;
using System.Text;
using SabreTools.Matching;
using SabreTools.Serialization.Wrappers;

namespace BinaryObjectScanner.Protection
{
    /// <summary>
    /// SafeCast is in the same family of protections as SafeDisc, and appears to mainly be for license management, and doesn't appear to affect the mastering of the disc in any way.
    /// Although SafeCast is most commonly used in non-game software, there is one game that comes with both SafeDisc and SafeCast protections (Redump entry 83145).
    /// Macrovision bought the company C-Dilla and created SafeCast based on C-Dilla's existing products (https://web.archive.org/web/20030212040047/http://www.auditmypc.com/freescan/readingroom/cdilla.asp).
    /// There are multiple different versions of SafeCast out there.
    /// Deep dive of MechWarrior 4 and its expansions, which use SafeDisc, possibly SafeDisc LT, and SafeCast: https://digipres.club/@TheRogueArchivist/110224192068908590
    /// SafeCast ESD: https://web.archive.org/web/20000306044246/http://www.macrovision.com/safecast_ESD.html
    /// SafeCast ESD Demo: https://web.archive.org/web/20010417215236/http://www.macrovision.com:80/demos/SafeCast_ESD.exe
    /// SafeCast Gold: https://web.archive.org/web/20000129071444/http://www.macrovision.com/scp_gold.html
    /// SafeCast LM: https://web.archive.org/web/20000128224337/http://www.macrovision.com/safecast_LM.html
    /// SafeCast UNSORTED samples:
    /// https://www.virustotal.com/gui/file/220b8eec64d43fd854a48f28efa915c158a06f3675cbf0ef52796750984d9e1c/details
    /// https://groups.google.com/g/alt.english.usage/c/kcBzeqXgE-M
    /// https://www.shouldiremoveit.com/aerosofts-FDC-Live-Cockpit-64651-program.aspx
    /// https://archive.org/details/danceejay4
    /// https://archive.org/details/ejay_nestle_trial
    /// https://archive.org/details/eJayXtremeSoundtraxx
    /// https://community.ptc.com/t5/Mathcad/SafeCast/td-p/25233
    /// SafeCast resources: 
    /// http://web.archive.org/web/20010417222834/http://www.macrovision.com/press_rel3_17_99.html (Press release introducing SafeCast)
    /// https://web.archive.org/web/20000129013431/http://www.macrovision.com/safecast_faq.html (SafeCast FAQ)
    /// https://web.archive.org/web/20040223025801/http://www.macrovision.com/products/legacy_products/safecast/safecast_cdilla_faq.shtml
    /// https://web.archive.org/web/20031204024544mp_/http://www.macrovision.com/products/safecast/index.shtml
    /// https://web.archive.org/web/20010417222834/http://www.macrovision.com/press_rel3_17_99.html
    /// https://www.extremetech.com/computing/53394-turbotax-so-what-do-i-do-now/4
    /// https://web.archive.org/web/20031013085038/http://www.pestpatrol.com/PestInfo/c/c-dilla.asp
    /// </summary>
    public partial class Macrovision
    {
        /// <inheritdoc cref="Interfaces.INewExecutableCheck.CheckNewExecutable(string, NewExecutable, bool)"/>
        internal string? SafeCastCheckNewExecutable(string file, NewExecutable nex, bool includeDebug)
        {
            // Check for the CDAC01AA name string.
            if (nex.Model.ResidentNameTable != null)
            {
                bool cdac01aaNameFound = nex.Model.ResidentNameTable
                .Select(rnte => rnte?.NameString == null ? string.Empty : Encoding.ASCII.GetString(rnte.NameString))
                .Any(s => s.Contains("CDAC01AA"));

                if (cdac01aaNameFound)
                    return "SafeCast";
            }

            // TODO: Don't read entire file
            var data = nex.ReadArbitraryRange();
            if (data == null)
                return null;

            var neMatchSets = new List<ContentMatchSet>
            {
                // SafeCast
                // Found as the Product Name in "cdac01aa.dll" from IA item "ejay_nestle_trial". Windows 10 appears to incorrectly truncate this to "SafeCas" in File Explorer.
                new(new byte?[] { 0x53, 0x61, 0x66, 0x65, 0x43, 0x61, 0x73, 0x74 }, "SafeCast"),
            };

            return MatchUtil.GetFirstMatch(file, data, neMatchSets, includeDebug);
        }

        /// <inheritdoc cref="Interfaces.IPortableExecutableCheck.CheckPortableExecutable(string, PortableExecutable, bool)"/>
        internal string? SafeCastCheckPortableExecutable(string file, PortableExecutable pex, bool includeDebug)
        {
            // Get the sections from the executable, if possible
            var sections = pex.Model.SectionTable;
            if (sections == null)
                return null;

            // TODO: Investigate import hint/name table entry "CdaSysInstall"
            // TODO: Investigate string table entries: "CDWP02DG", "CDWP02DG", "CDWS02DG"

            // Get the import directory table, if it exists
            if (pex.Model.ImportTable?.ImportDirectoryTable != null)
            {
                if (pex.Model.ImportTable.ImportDirectoryTable.Any(idte => idte?.Name != null && idte.Name.Equals("CdaC14BA.dll", StringComparison.OrdinalIgnoreCase)))
                    return "SafeCast";
            }

            // Get the dialog box resources
            // Found in "CDAC21BA.DLL" in Redump entry 95524.
            var resource = pex.FindDialogByTitle("SafeCast API");
            if (resource.Any())
                return "SafeCast";

            // Get the .data/DATA section strings, if they exist
            var strs = pex.GetFirstSectionStrings(".data") ?? pex.GetFirstSectionStrings("DATA");
            if (strs != null)
            {
                // Found in "DJMixStation\DJMixStation.exe" in IA item "ejay_nestle_trial".
                if (strs.Any(s => s.Contains("SOFTWARE\\C-Dilla\\SafeCast")))
                    return "SafeCast";
            }

            // Found in "32bit\Tax02\cdac14ba.dll" in IA item "TurboTax Deluxe Tax Year 2002 for Wndows (2.00R)(Intuit)(2002)(352282)".
            var name = pex.FileDescription;
            if (name?.Equals("SafeCast2", StringComparison.OrdinalIgnoreCase) == true)
                return "SafeCast";

            // Found in "cdac01ba.dll" from IA item "ejay_nestle_trial".
            // TODO: Figure out a reasonable way to parse version.
            if (name?.Equals("CdaC01BA", StringComparison.OrdinalIgnoreCase) == true)
                return $"SafeCast";

            // Found in "C2CDEL.EXE" in IA item "britney-spears-special-edition-cd-rom".
            if (name?.Equals("32-bit SafeCast Copy To Clear Delete", StringComparison.OrdinalIgnoreCase) == true)
                return $"SafeCast";

            // Found in "C2C.DLL" in IA item "britney-spears-special-edition-cd-rom".
            if (name?.Equals("32-bit SafeCast Shell Copy To Clear DLL", StringComparison.OrdinalIgnoreCase) == true)
                return $"SafeCast";

            // Found in "SCRfrsh.exe" in Redump entry 102979.
            if (name?.Equals("32-bit SafeCast Toolkit", StringComparison.OrdinalIgnoreCase) == true)
                return $"SafeCast {pex.FileVersion}";

            // Found in "CDAC14BA.DLL" in Redump entry 95524.
            if (name?.Equals("32-bit SafeCast Anchor Installer", StringComparison.OrdinalIgnoreCase) == true)
                return $"SafeCast";

            // Found in "CDAC21BA.DLL" in Redump entry 95524.
            if (name?.Equals("32-bit CdaC20BA", StringComparison.OrdinalIgnoreCase) == true)
                return $"SafeCast";

            // Found in hidden resource of "32bit\Tax02\cdac14ba.dll" in IA item "TurboTax Deluxe Tax Year 2002 for Wndows (2.00R)(Intuit)(2002)(352282)".
            name = pex.ProductName;
            if (name?.Equals("SafeCast Windows NT", StringComparison.OrdinalIgnoreCase) == true)
                return "SafeCast";

            // Found in "cdac01ba.dll" from IA item "ejay_nestle_trial".
            if (name?.Equals("SafeCast", StringComparison.OrdinalIgnoreCase) == true)
                return "SafeCast";

            // Check for CDSHARE/DISAG_SH sections

            return null;
        }

        /// <inheritdoc cref="Interfaces.IPathCheck.CheckDirectoryPath(string, IEnumerable{string})"/>
#if NET20 || NET35
        internal Queue<string> SafeCastCheckDirectoryPath(string path, IEnumerable<string>? files)
#else
        internal ConcurrentQueue<string> SafeCastCheckDirectoryPath(string path, IEnumerable<string>? files)
#endif
        {
            var matchers = new List<PathMatchSet>
            {
                // Found in IA item "britney-spears-special-edition-cd-rom".
                new(new List<PathMatch>
                {
                    new FilePathMatch("C2C.16"),
                    new FilePathMatch("C2C.DLL"),
                    new FilePathMatch("C2CDEL.16"),
                    new FilePathMatch("C2CDEL.EXE"),
                }, "SafeCast"),

                // Found in IA item "ejay_nestle_trial".
                new(new FilePathMatch("cdac01aa.dll"), "SafeCast"),
                new(new FilePathMatch("cdac01ba.dll"), "SafeCast"),

                // Found in multiple versions of SafeCast, including Redump entries 83145 and 95524, as well as IA item "TurboTax_Deluxe_Tax_Year_2002_for_Wndows_2.00R_Intuit_2002_352282".
                new(new FilePathMatch("cdac14ba.dll"), "SafeCast"),

                // Found in Redump entry 83145.
                new(new FilePathMatch("CDAC21BA.DLL"), "SafeCast"),

                // Found in Redump entry 102979.
                new(new FilePathMatch("SCRfrsh.exe"), "SafeCast"),

                // Found in Redump entries 26211 and 95524.
                new(new FilePathMatch("SCSHD.CSA"), "SafeCast"),

                // Found in Redump entries 95524.
                new(new FilePathMatch("SCSHD.EXE"), "SafeCast"),

                // Found in IA item "TurboTax_Deluxe_Tax_Year_2002_for_Wndows_2.00R_Intuit_2002_352282".
                new(new FilePathMatch("CDAC15BA.SYS"), "SafeCast"),

                // Found in "cdac14ba.dll" in IA item "TurboTax_Deluxe_Tax_Year_2002_for_Wndows_2.00R_Intuit_2002_352282".
                new(new FilePathMatch("CDAC13BA.EXE"), "SafeCast"),
            };

            return MatchUtil.GetAllMatches(files, matchers, any: false);
        }

        /// <inheritdoc cref="Interfaces.IPathCheck.CheckFilePath(string)"/>
        internal string? SafeCastCheckFilePath(string path)
        {
            var matchers = new List<PathMatchSet>
            {
                // Found in IA item "ejay_nestle_trial".
                new(new FilePathMatch("cdac01aa.dll"), "SafeCast"),
                new(new FilePathMatch("cdac01ba.dll"), "SafeCast"),

                new(new FilePathMatch("cdac11ba.exe"), "SafeCast"),

                // Found in multiple versions of SafeCast, including Redump entry 83145 and IA item "TurboTax_Deluxe_Tax_Year_2002_for_Wndows_2.00R_Intuit_2002_352282".
                new(new FilePathMatch("cdac14ba.dll"), "SafeCast"),

                // Found in Redump entry 83145.
                new(new FilePathMatch("CDAC21BA.DLL"), "SafeCast"),

                // Found in Redump entry 102979.
                new(new FilePathMatch("SCRfrsh.exe"), "SafeCast"),
                
                // Found in Redump entries 26211 and 95524.
                new(new FilePathMatch("SCSHD.CSA"), "SafeCast"),

                // Found in Redump entries 95524.
                new(new FilePathMatch("SCSHD.EXE"), "SafeCast"),

                // Found in IA item "TurboTax_Deluxe_Tax_Year_2002_for_Wndows_2.00R_Intuit_2002_352282".
                new(new FilePathMatch("CDAC15BA.SYS"), "SafeCast"),

                // Found in "cdac14ba.dll" in IA item "TurboTax_Deluxe_Tax_Year_2002_for_Wndows_2.00R_Intuit_2002_352282".
                new(new FilePathMatch("CDAC13BA.EXE"), "SafeCast"),
            };

            return MatchUtil.GetFirstMatch(path, matchers, any: true);
        }
    }
}
