From 56b21e0d8398a9ac988b18102391ef59b368642e Mon Sep 17 00:00:00 2001 From: morkt Date: Sat, 5 May 2018 01:58:57 +0400 Subject: [PATCH] implemented MPF2 archives. --- ArcFormats/Apricot/ArcDAT.cs | 208 +++++++++++++++++++++++++++++++++++ ArcFormats/ArcFormats.csproj | 1 + 2 files changed, 209 insertions(+) create mode 100644 ArcFormats/Apricot/ArcDAT.cs diff --git a/ArcFormats/Apricot/ArcDAT.cs b/ArcFormats/Apricot/ArcDAT.cs new file mode 100644 index 00000000..3d7e1485 --- /dev/null +++ b/ArcFormats/Apricot/ArcDAT.cs @@ -0,0 +1,208 @@ +//! \file ArcDAT.cs +//! \date 2018 May 04 +//! \brief Apricot resource archive. +// +// Copyright (C) 2018 by morkt +// +// Permission is hereby granted, free of charge, to any person obtaining a copy +// of this software and associated documentation files (the "Software"), to +// deal in the Software without restriction, including without limitation the +// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or +// sell copies of the Software, and to permit persons to whom the Software is +// furnished to do so, subject to the following conditions: +// +// The above copyright notice and this permission notice shall be included in +// all copies or substantial portions of the Software. +// +// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING +// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS +// IN THE SOFTWARE. +// + +using System; +using System.Collections.Generic; +using System.ComponentModel.Composition; +using System.IO; +using System.Text; +using GameRes.Compression; + +namespace GameRes.Formats.Apricot +{ + [Export(typeof(ArchiveFormat))] + public class Mpf2Opener : ArchiveFormat + { + public override string Tag { get { return "DAT/MPF2"; } } + public override string Description { get { return "Apricot resource archive"; } } + public override uint Signature { get { return 0x3246504D; } } // 'MPF2' + public override bool IsHierarchic { get { return true; } } + public override bool CanWrite { get { return false; } } + + public override ArcFile TryOpen (ArcView file) + { + var arc_list = new List(); + long max_offset = file.MaxOffset; + for (int i = 1; i < 100; ++i) + { + var part_name = Path.ChangeExtension (file.Name, string.Format ("a{0:D02}", i)); + if (!VFS.FileExists (part_name)) + break; + var part = VFS.FindFile (part_name); + arc_list.Add (part); + max_offset += part.Size; + } + uint index_length = file.View.ReadUInt32 (8); + uint data_offset = file.View.ReadUInt32 (0x10); + using (var zindex = file.CreateStream (0x20, index_length)) + using (var uindex = new ZLibStream (zindex, CompressionMode.Decompress)) + using (var index = new BinaryStream (uindex, file.Name)) + { + var buffer = new byte[500]; + var dir = new List(); + while (index.PeekByte() != -1) + { + int entry_length = index.ReadInt32(); + if (entry_length <= 528) + return null; + bool is_deleted = index.ReadUInt32() != 0; + var entry = new PackedEntry(); + entry.Offset = index.ReadInt64() + data_offset; + index.ReadUInt32(); + entry.Size = index.ReadUInt32(); + entry.UnpackedSize = index.ReadUInt32(); + entry.IsPacked = entry.Size != entry.UnpackedSize; + index.Read (buffer, 0, 500); + int name_length = index.Read (buffer, 0, entry_length - 528); + if (!is_deleted && entry.CheckPlacement (max_offset)) + { + entry.Name = Encoding.Unicode.GetString (buffer, 0, name_length); + entry.Type = FormatCatalog.Instance.GetTypeFromName (entry.Name); + dir.Add (entry); + } + } + return new Mpf2Archive (file, this, dir, arc_list); + } + } + + public override Stream OpenEntry (ArcFile arc, Entry entry) + { + var mpf = (Mpf2Archive)arc; + var input = mpf.ViewChain.CreateStream (entry.Offset, entry.Size); + var pent = entry as PackedEntry; + if (pent != null && pent.IsPacked) + input = new ZLibStream (input, CompressionMode.Decompress); + return input; + } + } + + internal class Mpf2Archive : ArcFile + { + ArcViewChain m_parts; + + public ArcViewChain ViewChain { get { return m_parts; } } + + public Mpf2Archive (ArcView arc, ArchiveFormat impl, ICollection dir, IReadOnlyList arc_list) + : base (arc, impl, dir) + { + m_parts = new ArcViewChain (arc, arc_list); + } + + bool m_disposed = false; + protected override void Dispose (bool disposing) + { + if (disposing && !m_disposed) + { + if (m_parts != null) + m_parts.Dispose(); + m_disposed = true; + } + } + } + + internal sealed class ArcViewChain : IDisposable + { + ArcView m_file; + IReadOnlyList m_parts; + + public IEnumerable Parts { + get { + yield return m_file; + if (m_parts != null) + foreach (var part in m_parts) + yield return part; + } + } + + public ArcViewChain (ArcView file, IReadOnlyList arc_list) + { + var parts = new List (arc_list.Count); + try + { + foreach (var arc_entry in arc_list) + { + var arc_file = VFS.OpenView (arc_entry); + parts.Add (arc_file); + } + } + catch + { + foreach (var part in parts) + part.Dispose(); + throw; + } + m_file = file; + m_parts = parts; + } + + public Stream CreateStream (long offset, uint size) + { + Stream input = null; + try + { + long part_offset = 0; + long entry_start = offset; + long entry_end = offset + size; + foreach (var part in Parts) + { + long part_end_offset = part_offset + part.MaxOffset; + if (entry_start < part_end_offset) + { + uint part_size = (uint)Math.Min (entry_end - entry_start, part_end_offset - entry_start); + var entry_part = part.CreateStream (entry_start - part_offset, part_size); + if (input != null) + input = new ConcatStream (input, entry_part); + else + input = entry_part; + entry_start += part_size; + if (entry_start >= entry_end) + break; + } + part_offset = part_end_offset; + } + return input ?? Stream.Null; + } + catch + { + if (input != null) + input.Dispose(); + throw; + } + } + + bool m_disposed = false; + public void Dispose () + { + if (m_disposed) + return; + if (m_parts != null) + { + foreach (var arc in m_parts) + arc.Dispose(); + } + m_disposed = true; + } + } +} diff --git a/ArcFormats/ArcFormats.csproj b/ArcFormats/ArcFormats.csproj index feb66610..0459f7b3 100644 --- a/ArcFormats/ArcFormats.csproj +++ b/ArcFormats/ArcFormats.csproj @@ -113,6 +113,7 @@ +