(Ai5Win): generalized index reader.

This commit is contained in:
morkt 2017-12-08 14:42:20 +04:00
parent a5d76f7b29
commit 5e1e4338f1

View File

@ -77,42 +77,51 @@ namespace GameRes.Formats.Elf
int count = file.View.ReadInt32 (0); int count = file.View.ReadInt32 (0);
if (!IsSaneCount (count)) if (!IsSaneCount (count))
return null; return null;
var reader = new IndexReader (file, count); var reader = new Ai5ArcIndexReader (file, count);
foreach (var scheme in KnownSchemes.Values) var dir = reader.TrySchemes (KnownSchemes.Values);
{
try
{
var dir = reader.Read (scheme);
if (dir != null) if (dir != null)
return new ArcFile (file, this, dir); return new ArcFile (file, this, dir);
}
catch { /* ignore parse errors */ }
}
return null; return null;
} }
public override Stream OpenEntry (ArcFile arc, Entry entry) public override Stream OpenEntry (ArcFile arc, Entry entry)
{ {
var input = arc.File.CreateStream (entry.Offset, entry.Size); var input = arc.File.CreateStream (entry.Offset, entry.Size);
if (entry.Name.HasAnyOfExtensions ("mes", "lib")) if (entry.Name.HasAnyOfExtensions ("mes", "lib", "a", "a6", "msk"))
return new LzssStream (input); return new LzssStream (input);
return input; return input;
} }
}
internal class IndexReader internal class Ai5ArcIndexReader
{ {
ArcView m_file; protected ArcView m_file;
int m_count; protected int m_count;
List<Entry> m_dir; protected List<Entry> m_dir;
byte[] m_name_buf = new byte[0x100]; protected byte[] m_name_buf = new byte[0x100];
public IndexReader (ArcView file, int count) public Ai5ArcIndexReader (ArcView file, int count)
{ {
m_file = file; m_file = file;
m_count = count; m_count = count;
m_dir = new List<Entry> (m_count); m_dir = new List<Entry> (m_count);
} }
public List<Entry> TrySchemes (IEnumerable<ArcIndexScheme> schemes)
{
foreach (var scheme in schemes)
{
try
{
var dir = Read (scheme);
if (dir != null)
return dir;
}
catch { /* ignore parse errors */ }
}
return null;
}
public List<Entry> Read (ArcIndexScheme scheme) public List<Entry> Read (ArcIndexScheme scheme)
{ {
if (scheme.NameLength > m_name_buf.Length) if (scheme.NameLength > m_name_buf.Length)
@ -125,18 +134,9 @@ namespace GameRes.Formats.Elf
for (int i = 0; i < m_count; ++i) for (int i = 0; i < m_count; ++i)
{ {
m_file.View.Read (index_offset, m_name_buf, 0, (uint)scheme.NameLength); m_file.View.Read (index_offset, m_name_buf, 0, (uint)scheme.NameLength);
int n; string name = DecryptName (scheme);
for (n = 0; n < m_name_buf.Length; ++n) if (null == name)
{
m_name_buf[n] ^= scheme.NameKey;
if (0 == m_name_buf[n])
break;
if (m_name_buf[n] < 0x20)
return null; return null;
}
if (0 == n)
return null;
string name = Encodings.cp932.GetString (m_name_buf, 0, n);
index_offset += scheme.NameLength; index_offset += scheme.NameLength;
var entry = FormatCatalog.Instance.Create<Entry> (name); var entry = FormatCatalog.Instance.Create<Entry> (name);
entry.Size = m_file.View.ReadUInt32 (index_offset) ^ scheme.SizeKey; entry.Size = m_file.View.ReadUInt32 (index_offset) ^ scheme.SizeKey;
@ -148,76 +148,24 @@ namespace GameRes.Formats.Elf
} }
return m_dir; return m_dir;
} }
}
/* internal string DecryptName (ArcIndexScheme scheme)
internal class IndexReader
{ {
ArcView m_file; int n;
int m_count; for (n = 0; n < m_name_buf.Length; ++n)
byte[] m_first = new byte[0x108];
public const int MinNameLength = 0x10;
public const int MaxNameLength = 0x100;
public IndexReader (ArcView file)
{ {
m_file = file; if (n > scheme.NameLength)
m_count = m_file.View.ReadInt32 (0); return null;
m_file.View.Read (4, m_first, 0, m_first.Length); m_name_buf[n] ^= scheme.NameKey;
} if (0 == m_name_buf[n])
break;
ArcIndexScheme m_scheme = new ArcIndexScheme(); if (m_name_buf[n] < 0x20)
public ArcIndexScheme Parse ()
{
if (m_count <= 0 || m_count > 0xfffff)
return null; return null;
uint supposed_first_offset = (uint)(m_count * (name_length + 8));
uint first_size = LittleEndian.ToUInt32 (first_entry, name_length);
uint first_offset = LittleEndian.ToUInt32 (first_entry, name_length+4);
uint supposed_offset_key = first_offset ^ supposed_first_offset;
int last_index_offset = 4 + (m_count - 1) * (name_length + 8);
uint last_size = m_file.View.ReadUInt32 (last_index_offset + name_length);
uint last_offset = m_file.View.ReadUInt32 (last_index_offset + name_length + 4);
last_offset ^= supposed_offset_key;
} }
if (n != 0)
bool ParseFirstEntry (int name_length) return Encodings.cp932.GetString (m_name_buf, 0, n);
{ else
int index_offset = 4; return null;
}
public byte NameKey { get; private set; }
int GuessNameLength (int initial)
{
int name_pos = initial;
byte sym;
do
{
do
{
sym = first_entry[name_pos++];
}
while (name_pos < MaxNameLength && sym != first_entry[name_pos]);
if (MaxNameLength == name_pos)
return 0;
while (name_pos < MaxNameLength && sym == first_entry[name_pos])
{
++name_pos;
}
if (MaxNameLength == name_pos && sym == first_entry[name_pos] && sym == first_entry[name_pos+1])
return 0;
}
while (name_pos < MinNameLength || 0 != (name_pos & 1));
NameKey = sym;
return name_pos;
} }
} }
*/
}
} }