wabbajack/Wabbajack.VirtualFileSystem/SevenZipExtractor/ArchiveFile.cs

using System;
using System.Collections.Generic;
using System.Globalization;
using System.IO;
using System.Linq;
using System.Runtime.InteropServices;
using System.Threading.Tasks;
using Wabbajack.Common;
using Wabbajack.Common.FileSignatures;

namespace Wabbajack.VirtualFileSystem.SevenZipExtractor
{
    public class ArchiveFile : IDisposable
    {
        private SevenZipHandle _sevenZipHandle;
        internal IInArchive _archive;
        public InStreamWrapper _archiveStream;
        private IList<Entry> _entries;

        private static readonly AbsolutePath LibraryFilePath = @"Extractors\7z.dll".RelativeTo(AbsolutePath.EntryPoint);
        private static SignatureChecker _checker = new SignatureChecker(Formats.FileTypeGuidMapping.Keys.ToArray());

        public static async Task<ArchiveFile> Open(Stream archiveStream, Definitions.FileType format)
        {
            var self = new ArchiveFile();
            self.InitializeAndValidateLibrary();
            self._archive = self._sevenZipHandle.CreateInArchive(Formats.FileTypeGuidMapping[format]);

            if (!FileExtractor2.FavorPerfOverRAM)
            {
                self.SetCompressionProperties(new Dictionary<string, string>() {{"mt", "off"}});
            }


            self._archiveStream = new InStreamWrapper(archiveStream);
            return self;
        }

        /// <summary>
        /// Sets the compression properties
        /// </summary>
        private void SetCompressionProperties(Dictionary<string, string> CustomParameters)
        {


            {
                ISetProperties setter;
                try
                {
                    setter = (ISetProperties)_archive;
                }
                catch (InvalidCastException _)
                {
                    return;
                }

                var names = new List<IntPtr>(1 + CustomParameters.Count);
                var values = new List<PropVariant>(1 + CustomParameters.Count);
                //var sp = new SecurityPermission(SecurityPermissionFlag.UnmanagedCode);
                //sp.Demand();

                #region Initialize compression properties

                names.Add(Marshal.StringToBSTR("x"));
                values.Add(new PropVariant());

                foreach (var pair in CustomParameters)
                {

                    names.Add(Marshal.StringToBSTR(pair.Key));
                    var pv = new PropVariant();

                    #region List of parameters to cast as integers

                    var integerParameters = new HashSet<string>
                    {
                        "fb",
                        "pass",
                        "o",
                        "yx",
                        "a",
                        "mc",
                        "lc",
                        "lp",
                        "pb",
                        "cp"
                    };

                    #endregion

                    if (integerParameters.Contains(pair.Key))
                    {
                        pv.VarType = VarEnum.VT_UI4;
                        pv.UInt32Value = Convert.ToUInt32(pair.Value, CultureInfo.InvariantCulture);
                    }
                    else
                    {
                        pv.VarType = VarEnum.VT_BSTR;
                        pv.pointerValue = Marshal.StringToBSTR(pair.Value);
                    }

                    values.Add(pv);
                }

                #endregion

                #region Set compression level

                var clpv = values[0];
                clpv.VarType = VarEnum.VT_UI4;
                clpv.UInt32Value = 0;

                values[0] = clpv;

                #endregion

                var namesHandle = GCHandle.Alloc(names.ToArray(), GCHandleType.Pinned);
                var valuesHandle = GCHandle.Alloc(values.ToArray(), GCHandleType.Pinned);

                try
                {
                    setter?.SetProperties(namesHandle.AddrOfPinnedObject(), valuesHandle.AddrOfPinnedObject(),
                        names.Count);
                }
                finally
                {
                    namesHandle.Free();
                    valuesHandle.Free();
                }

            }
        }
        
        public IList<Entry> Entries
        {
            get
            {
                if (this._entries != null)
                {
                    return this._entries;
                }

                ulong checkPos = 32 * 1024;
                int open = this._archive.Open(this._archiveStream, ref checkPos, null);

                if (open != 0)
                {
                    throw new Exception("Unable to open archive");
                }

                uint itemsCount = this._archive.GetNumberOfItems();

                this._entries = new List<Entry>();

                for (uint fileIndex = 0; fileIndex < itemsCount; fileIndex++)
                {
                    var entry = GetEntry(fileIndex);
                    this._entries.Add(entry);
                }

                return this._entries;
            }
        }

        internal Entry GetEntry(uint fileIndex)
        {
            string fileName = this.GetProperty<string>(fileIndex, ItemPropId.kpidPath);
            bool isFolder = this.GetProperty<bool>(fileIndex, ItemPropId.kpidIsFolder);
            bool isEncrypted = this.GetProperty<bool>(fileIndex, ItemPropId.kpidEncrypted);
            ulong size = this.GetProperty<ulong>(fileIndex, ItemPropId.kpidSize);
            ulong packedSize = this.GetProperty<ulong>(fileIndex, ItemPropId.kpidPackedSize);
            DateTime creationTime = this.GetPropertySafe<DateTime>(fileIndex, ItemPropId.kpidCreationTime);
            DateTime lastWriteTime = this.GetPropertySafe<DateTime>(fileIndex, ItemPropId.kpidLastWriteTime);
            DateTime lastAccessTime = this.GetPropertySafe<DateTime>(fileIndex, ItemPropId.kpidLastAccessTime);
            uint crc = this.GetPropertySafe<uint>(fileIndex, ItemPropId.kpidCRC);
            uint attributes = this.GetPropertySafe<uint>(fileIndex, ItemPropId.kpidAttributes);
            string comment = this.GetPropertySafe<string>(fileIndex, ItemPropId.kpidComment);
            string hostOS = this.GetPropertySafe<string>(fileIndex, ItemPropId.kpidHostOS);
            string method = this.GetPropertySafe<string>(fileIndex, ItemPropId.kpidMethod);

            bool isSplitBefore = this.GetPropertySafe<bool>(fileIndex, ItemPropId.kpidSplitBefore);
            bool isSplitAfter = this.GetPropertySafe<bool>(fileIndex, ItemPropId.kpidSplitAfter);

            var entry = new Entry(this._archive, fileIndex)
            {
                FileName = fileName,
                IsFolder = isFolder,
                IsEncrypted = isEncrypted,
                Size = size,
                PackedSize = packedSize,
                CreationTime = creationTime,
                LastWriteTime = lastWriteTime,
                LastAccessTime = lastAccessTime,
                CRC = crc,
                Attributes = attributes,
                Comment = comment,
                HostOS = hostOS,
                Method = method,
                IsSplitBefore = isSplitBefore,
                IsSplitAfter = isSplitAfter
            };
            return entry;
        }

        private T GetPropertySafe<T>(uint fileIndex, ItemPropId name)
        {
            try
            {
                return this.GetProperty<T>(fileIndex, name);
            }
            catch (InvalidCastException)
            {
                return default(T);
            }
        }

        private T GetProperty<T>(uint fileIndex, ItemPropId name)
        {
            PropVariant propVariant = new PropVariant();
            this._archive.GetProperty(fileIndex, name, ref propVariant);
            object value = propVariant.GetObject();

            if (propVariant.VarType == VarEnum.VT_EMPTY)
            {
                propVariant.Clear();
                return default(T);
            }

            propVariant.Clear();

            if (value == null)
            {
                return default(T);
            }

            Type type = typeof(T);
            bool isNullable = type.IsGenericType && type.GetGenericTypeDefinition() == typeof(Nullable<>);
            Type underlyingType = isNullable ? Nullable.GetUnderlyingType(type) : type;

            T result = (T)Convert.ChangeType(value.ToString(), underlyingType);

            return result;
        }

        private void InitializeAndValidateLibrary()
        {
            try
            {
                this._sevenZipHandle = new SevenZipHandle((string)LibraryFilePath);
            }
            catch (Exception e)
            {
                throw new Exception("Unable to initialize SevenZipHandle", e);
            }
        }

        ~ArchiveFile()
        {
            this.Dispose(false);
        }

        protected void Dispose(bool disposing)
        {
            if (this._archiveStream != null)
            {
                this._archiveStream.Dispose();
            }

            if (this._archive != null)
            {
                Marshal.ReleaseComObject(this._archive);
            }

            if (this._sevenZipHandle != null)
            {
                this._sevenZipHandle.Dispose();
            }
        }

        public void Dispose()
        {
            this.Dispose(true);
            GC.SuppressFinalize(this);
        }
    }
}