mirror of
https://github.com/crskycode/GARbro.git
synced 2026-07-08 01:30:18 +08:00
(MGPK): added encryption query widget.
This commit is contained in:
morkt
207
ArcFormats/MangaGamer/ArcMGPK.cs
Normal file
207
ArcFormats/MangaGamer/ArcMGPK.cs
Normal file
@@ -0,0 +1,207 @@
|
||||
//! \file ArcMGPK.cs
|
||||
//! \date Mon Nov 03 20:03:36 2014
|
||||
//! \brief MGPK archive format.
|
||||
//
|
||||
// Copyright (C) 2014-2016 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.Formats.Properties;
|
||||
using GameRes.Formats.Strings;
|
||||
using GameRes.Utility;
|
||||
|
||||
namespace GameRes.Formats.Mg
|
||||
{
|
||||
internal class MgArchive : ArcFile
|
||||
{
|
||||
public readonly byte[] Key;
|
||||
|
||||
public MgArchive (ArcView arc, ArchiveFormat impl, ICollection<Entry> dir, byte[] key)
|
||||
: base (arc, impl, dir)
|
||||
{
|
||||
Key = key;
|
||||
}
|
||||
}
|
||||
|
||||
public class MgOptions : ResourceOptions
|
||||
{
|
||||
public byte[] Key;
|
||||
}
|
||||
|
||||
[Serializable]
|
||||
public class MgScheme : ResourceScheme
|
||||
{
|
||||
public Dictionary<string, byte[]> KnownKeys;
|
||||
}
|
||||
|
||||
[Export(typeof(ArchiveFormat))]
|
||||
public class MgpkOpener : ArchiveFormat
|
||||
{
|
||||
public override string Tag { get { return "MGPK"; } }
|
||||
public override string Description { get { return "MG resource archive"; } }
|
||||
public override uint Signature { get { return 0x4b50474d; } } // MGPK
|
||||
public override bool IsHierarchic { get { return false; } }
|
||||
public override bool CanCreate { get { return false; } }
|
||||
|
||||
public MgpkOpener ()
|
||||
{
|
||||
Extensions = new string[] { "pac" };
|
||||
}
|
||||
|
||||
public override ArcFile TryOpen (ArcView file)
|
||||
{
|
||||
int count = file.View.ReadInt32 (8);
|
||||
if (count > 0xFFFFF)
|
||||
return null;
|
||||
long cur_offset = 0x0C;
|
||||
var dir = new List<Entry> (count);
|
||||
bool has_encrypted = false;
|
||||
for (int i = 0; i < count; ++i)
|
||||
{
|
||||
uint name_length = file.View.ReadByte (cur_offset);
|
||||
string name = file.View.ReadString (cur_offset+1, name_length, Encoding.UTF8);
|
||||
var entry = FormatCatalog.Instance.Create<Entry> (name);
|
||||
entry.Offset = file.View.ReadUInt32 (cur_offset+0x20);
|
||||
entry.Size = file.View.ReadUInt32 (cur_offset+0x24);
|
||||
if (!entry.CheckPlacement (file.MaxOffset))
|
||||
return null;
|
||||
has_encrypted = has_encrypted
|
||||
|| name.EndsWith (".png", StringComparison.InvariantCultureIgnoreCase)
|
||||
|| name.EndsWith (".txt", StringComparison.InvariantCultureIgnoreCase);
|
||||
dir.Add (entry);
|
||||
cur_offset += 0x30;
|
||||
}
|
||||
if (has_encrypted && KnownKeys.Count > 0)
|
||||
{
|
||||
var options = Query<MgOptions> (arcStrings.ArcEncryptedNotice);
|
||||
if (options.Key != null)
|
||||
return new MgArchive (file, this, dir, options.Key);
|
||||
}
|
||||
return new ArcFile (file, this, dir);
|
||||
}
|
||||
|
||||
public override Stream OpenEntry (ArcFile arc, Entry entry)
|
||||
{
|
||||
var mgarc = arc as MgArchive;
|
||||
Stream input = arc.File.CreateStream (entry.Offset, entry.Size);
|
||||
if (null == mgarc || null == mgarc.Key)
|
||||
return input;
|
||||
using (input)
|
||||
{
|
||||
byte[] data = new byte[entry.Size];
|
||||
input.Read (data, 0, data.Length);
|
||||
data = DecryptBlock (data, mgarc.Key);
|
||||
if (entry.Name.EndsWith (".txt", StringComparison.InvariantCultureIgnoreCase))
|
||||
return DecompressStream (data);
|
||||
return new MemoryStream (data);
|
||||
}
|
||||
}
|
||||
|
||||
private byte[] DecryptBlock (byte[] input, byte[] key)
|
||||
{
|
||||
key = (byte[])key.Clone();
|
||||
for (int i = 0; i < input.Length; i++)
|
||||
{
|
||||
input[i] ^= key[i % key.Length];
|
||||
key[i % key.Length] += 27;
|
||||
}
|
||||
return input;
|
||||
}
|
||||
|
||||
private Stream DecompressStream (byte[] input)
|
||||
{
|
||||
byte[] output = new byte[input.Length * 2];
|
||||
int output_size = lzf_decompress (input, ref output);
|
||||
return new MemoryStream (output, 0, output_size);
|
||||
}
|
||||
|
||||
private static int lzf_decompress (byte[] input, ref byte[] output)
|
||||
{
|
||||
int src = 0;
|
||||
int dst = 0;
|
||||
while (src < input.Length)
|
||||
{
|
||||
int count = input[src++];
|
||||
if (count < 32)
|
||||
{
|
||||
++count;
|
||||
if (dst + count > output.Length)
|
||||
{
|
||||
Array.Resize (ref output, Math.Max (checked(output.Length * 2), dst + count));
|
||||
}
|
||||
Buffer.BlockCopy (input, src, output, dst, count);
|
||||
src += count;
|
||||
dst += count;
|
||||
}
|
||||
else
|
||||
{
|
||||
int offset = (count & 31) << 8;
|
||||
count >>= 5;
|
||||
if (7 == count)
|
||||
{
|
||||
count += input[src++];
|
||||
}
|
||||
count += 2;
|
||||
offset += input[src++] + 1;
|
||||
if (offset > dst)
|
||||
throw new InvalidFormatException();
|
||||
if (dst + count > output.Length)
|
||||
{
|
||||
Array.Resize (ref output, Math.Max (checked(output.Length * 2), dst + count));
|
||||
}
|
||||
Binary.CopyOverlapped (output, dst-offset, dst, count);
|
||||
dst += count;
|
||||
}
|
||||
}
|
||||
return dst;
|
||||
}
|
||||
|
||||
public override ResourceOptions GetDefaultOptions ()
|
||||
{
|
||||
return new MgOptions { Key = GetKey (Settings.Default.MGPKTitle) };
|
||||
}
|
||||
|
||||
public override object GetAccessWidget ()
|
||||
{
|
||||
return new GUI.WidgetMGPK();
|
||||
}
|
||||
|
||||
public static byte[] GetKey (string title)
|
||||
{
|
||||
byte[] key;
|
||||
if (string.IsNullOrEmpty (title) || !KnownKeys.TryGetValue (title, out key))
|
||||
return null;
|
||||
return key;
|
||||
}
|
||||
|
||||
public static Dictionary<string, byte[]> KnownKeys = new Dictionary<string, byte[]>();
|
||||
|
||||
public override ResourceScheme Scheme
|
||||
{
|
||||
get { return new MgScheme { KnownKeys = KnownKeys }; }
|
||||
set { KnownKeys = ((MgScheme)value).KnownKeys; }
|
||||
}
|
||||
}
|
||||
}
|
||||
7
ArcFormats/MangaGamer/WidgetMGPK.xaml
Normal file
7
ArcFormats/MangaGamer/WidgetMGPK.xaml
Normal file
@@ -0,0 +1,7 @@
|
||||
<StackPanel x:Class="GameRes.Formats.GUI.WidgetMGPK"
|
||||
xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
|
||||
xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
|
||||
xmlns:p="clr-namespace:GameRes.Formats.Properties">
|
||||
<ComboBox Name="Title" Width="180" ItemsSource="{Binding}"
|
||||
SelectedValue="{Binding Source={x:Static p:Settings.Default}, Path=MGPKTitle, Mode=TwoWay}"/>
|
||||
</StackPanel>
|
||||
22
ArcFormats/MangaGamer/WidgetMGPK.xaml.cs
Normal file
22
ArcFormats/MangaGamer/WidgetMGPK.xaml.cs
Normal file
@@ -0,0 +1,22 @@
|
||||
using System.Linq;
|
||||
using System.Windows.Controls;
|
||||
using GameRes.Formats.Mg;
|
||||
using GameRes.Formats.Strings;
|
||||
|
||||
namespace GameRes.Formats.GUI
|
||||
{
|
||||
/// <summary>
|
||||
/// Interaction logic for WidgetMGPK.xaml
|
||||
/// </summary>
|
||||
public partial class WidgetMGPK : StackPanel
|
||||
{
|
||||
public WidgetMGPK()
|
||||
{
|
||||
InitializeComponent();
|
||||
var keys = new string[] { arcStrings.ArcNoEncryption };
|
||||
Title.ItemsSource = keys.Concat (MgpkOpener.KnownKeys.Keys.OrderBy (x => x));
|
||||
if (-1 == Title.SelectedIndex)
|
||||
Title.SelectedIndex = 0;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user