Files
App-Installer-For-Windows-8…/PriFileFormat/HierarchicalSchemaSection.cs
2026-04-09 22:16:47 +08:00

291 lines
11 KiB
C#

using System;
using System.Collections.Generic;
using System.IO;
using System.Text;
using System.Runtime.InteropServices;
namespace PriFileFormat
{
public class HierarchicalSchemaSection: Section
{
public HierarchicalSchemaVersionInfo Version { get; private set; }
public string UniqueName { get; private set; }
public string Name { get; private set; }
public IReadOnlyList <ResourceMapScope> Scopes { get; private set; }
public IReadOnlyList <ResourceMapItem> Items { get; private set; }
bool extendedVersion;
internal const string Identifier1 = "[mrm_hschema] \0";
internal const string Identifier2 = "[mrm_hschemaex] ";
internal HierarchicalSchemaSection (PriFile priFile, bool extendedVersion) : base (extendedVersion ? Identifier2 : Identifier1, priFile)
{
this.extendedVersion = extendedVersion;
}
protected override bool ParseSectionContent (BinaryReader binaryReader)
{
if (binaryReader.BaseStream.Length == 0)
{
Version = null;
UniqueName = null;
Name = null;
Scopes = Array.Empty<ResourceMapScope> ();
Items = Array.Empty<ResourceMapItem> ();
return true;
}
binaryReader.ExpectUInt16 (1);
ushort uniqueNameLength = binaryReader.ReadUInt16 ();
ushort nameLength = binaryReader.ReadUInt16 ();
binaryReader.ExpectUInt16 (0);
bool extendedHNames;
if (extendedVersion)
{
extendedHNames = false;
string extstr = new string (binaryReader.ReadChars (16));
if (extstr == "[def_hnamesx] \0") extendedHNames = true;
else if (extstr == "[def_hnames] \0") extendedHNames = false;
else throw new InvalidDataException ();
}
else extendedHNames = false;
// hierarchical schema version info
ushort majorVersion = binaryReader.ReadUInt16 ();
ushort minorVersion = binaryReader.ReadUInt16 ();
binaryReader.ExpectUInt32 (0);
uint checksum = binaryReader.ReadUInt32 ();
uint numScopes = binaryReader.ReadUInt32 ();
uint numItems = binaryReader.ReadUInt32 ();
Version = new HierarchicalSchemaVersionInfo (majorVersion, minorVersion, checksum, numScopes, numItems);
UniqueName = binaryReader.ReadNullTerminatedString (Encoding.Unicode);
Name = binaryReader.ReadNullTerminatedString (Encoding.Unicode);
if (UniqueName.Length != uniqueNameLength - 1 || Name.Length != nameLength - 1) throw new InvalidDataException ();
binaryReader.ExpectUInt16 (0);
ushort maxFullPathLength = binaryReader.ReadUInt16 ();
binaryReader.ExpectUInt16 (0);
binaryReader.ExpectUInt32 (numScopes + numItems);
binaryReader.ExpectUInt32 (numScopes);
binaryReader.ExpectUInt32 (numItems);
uint unicodeDataLength = binaryReader.ReadUInt32 ();
binaryReader.ReadUInt32 (); // meaning unknown
if (extendedHNames) binaryReader.ReadUInt32 (); // meaning unknown
List <ScopeAndItemInfo> scopeAndItemInfos = new List<ScopeAndItemInfo> ((int)(numScopes + numItems));
for (int i = 0; i < numScopes + numItems; i++)
{
ushort parent = binaryReader.ReadUInt16 ();
ushort fullPathLength = binaryReader.ReadUInt16 ();
char uppercaseFirstChar = (char)binaryReader.ReadUInt16 ();
byte nameLength2 = binaryReader.ReadByte ();
byte flags = binaryReader.ReadByte ();
uint nameOffset = binaryReader.ReadUInt16 () | (uint)((flags & 0xF) << 16);
ushort index = binaryReader.ReadUInt16 ();
bool isScope = (flags & 0x10) != 0;
bool nameInAscii = (flags & 0x20) != 0;
scopeAndItemInfos.Add (new ScopeAndItemInfo (parent, fullPathLength, isScope, nameInAscii, nameOffset, index));
}
List <ScopeExInfo> scopeExInfos = new List <ScopeExInfo> ((int)numScopes);
for (int i = 0; i < numScopes; i++)
{
ushort scopeIndex = binaryReader.ReadUInt16 ();
ushort childCount = binaryReader.ReadUInt16 ();
ushort firstChildIndex = binaryReader.ReadUInt16 ();
binaryReader.ExpectUInt16 (0);
scopeExInfos.Add (new ScopeExInfo (scopeIndex, childCount, firstChildIndex));
}
ushort [] itemIndexPropertyToIndex = new ushort [numItems];
for (int i = 0; i < numItems; i++) itemIndexPropertyToIndex [i] = binaryReader.ReadUInt16 ();
long unicodeDataOffset = binaryReader.BaseStream.Position;
long asciiDataOffset = binaryReader.BaseStream.Position + unicodeDataLength * 2;
ResourceMapScope [] scopes = new ResourceMapScope [numScopes];
ResourceMapItem [] items = new ResourceMapItem [numItems];
for (int i = 0; i < numScopes + numItems; i++)
{
long pos;
if (scopeAndItemInfos [i].NameInAscii) pos = asciiDataOffset + scopeAndItemInfos [i].NameOffset;
else pos = unicodeDataOffset + scopeAndItemInfos [i].NameOffset * 2;
binaryReader.BaseStream.Seek (pos, SeekOrigin.Begin);
string name;
if (scopeAndItemInfos [i].FullPathLength != 0) name = binaryReader.ReadNullTerminatedString (scopeAndItemInfos [i].NameInAscii ? Encoding.ASCII : Encoding.Unicode);
else name = string.Empty;
ushort index = scopeAndItemInfos [i].Index;
if (scopeAndItemInfos [i].IsScope)
{
if (scopes [index] != null) throw new InvalidDataException ();
scopes [index] = new ResourceMapScope (index, null, name);
}
else
{
if (items [index] != null) throw new InvalidDataException ();
items [index] = new ResourceMapItem (index, null, name);
}
}
for (int i = 0; i < numScopes + numItems; i++)
{
ushort index = scopeAndItemInfos [i].Index;
ushort parent = scopeAndItemInfos [scopeAndItemInfos [i].Parent].Index;
if (parent != 0xFFFF)
if (scopeAndItemInfos [i].IsScope)
{
if (parent != index) scopes [index].Parent = scopes [parent];
}
else items [index].Parent = scopes [parent];
}
for (int i = 0; i < numScopes; i++)
{
List <ResourceMapEntry> children = new List<ResourceMapEntry> (scopeExInfos [i].ChildCount);
for (int j = 0; j < scopeExInfos [i].ChildCount; j++)
{
ScopeAndItemInfo saiInfo = scopeAndItemInfos [scopeExInfos [i].FirstChildIndex + j];
if (saiInfo.IsScope) children.Add (scopes [saiInfo.Index]);
else children.Add (items [saiInfo.Index]);
}
scopes [i].Children = children;
}
Scopes = scopes;
Items = items;
//if (checksum != ComputeHierarchicalSchemaVersionInfoChecksum())
// throw new Exception();
return true;
}
private struct ScopeAndItemInfo
{
public ushort Parent;
public ushort FullPathLength;
public bool IsScope;
public bool NameInAscii;
public uint NameOffset;
public ushort Index;
public ScopeAndItemInfo (ushort parent, ushort fullPathLength, bool isScope, bool nameInAscii, uint nameOffset, ushort index)
{
Parent = parent;
FullPathLength = fullPathLength;
IsScope = isScope;
NameInAscii = nameInAscii;
NameOffset = nameOffset;
Index = index;
}
}
private struct ScopeExInfo
{
public ushort ScopeIndex;
public ushort ChildCount;
public ushort FirstChildIndex;
public ScopeExInfo (ushort scopeIndex, ushort childCount, ushort firstChildIndex)
{
ScopeIndex = scopeIndex;
ChildCount = childCount;
FirstChildIndex = firstChildIndex;
}
}
~HierarchicalSchemaSection ()
{
return;
Version = null;
foreach (var item in Items) { item.Parent = null; }
foreach (var scope in Scopes) { scope.Parent = null; }
Scopes = null;
Items = null;
}
// Checksum computation is buggy for some files
//private uint ComputeHierarchicalSchemaVersionInfoChecksum()
//{
// CRC32 crc32 = new CRC32();
// StringChecksum(crc32, UniqueName);
// StringChecksum(crc32, Name);
// crc32.TransformBlock(BitConverter.GetBytes(Version.MajorVersion), 0, 2, new byte[2], 0);
// crc32.TransformBlock(BitConverter.GetBytes(Version.MinorVersion), 0, 2, new byte[2], 0);
// crc32.TransformBlock(BitConverter.GetBytes(0), 0, 4, new byte[4], 0);
// crc32.TransformBlock(BitConverter.GetBytes(0), 0, 4, new byte[4], 0);
// crc32.TransformBlock(BitConverter.GetBytes(1), 0, 4, new byte[4], 0);
// crc32.TransformBlock(BitConverter.GetBytes(Scopes.Count), 0, 4, new byte[4], 0);
// foreach (ResourceMapScope scope in Scopes)
// StringChecksum(crc32, scope.FullName.Replace('\\', '/').TrimStart('/'));
// crc32.TransformBlock(BitConverter.GetBytes(0), 0, 4, new byte[4], 0);
// crc32.TransformBlock(BitConverter.GetBytes(0), 0, 4, new byte[4], 0);
// crc32.TransformBlock(BitConverter.GetBytes(1), 0, 4, new byte[4], 0);
// crc32.TransformBlock(BitConverter.GetBytes(Items.Count), 0, 4, new byte[4], 0);
// foreach (ResourceMapItem item in Items)
// StringChecksum(crc32, item.FullName.Replace('\\', '/').TrimStart('/'));
// return crc32.Result;
//}
//private void StringChecksum(CRC32 crc32, string s)
//{
// if (s == null)
// {
// byte[] data = new byte[8];
// crc32.TransformBlock(data, 0, data.Length, new byte[data.Length], 0);
// }
// else
// {
// byte[] data = Encoding.Unicode.GetBytes(s.ToLowerInvariant() + '\0');
// byte[] l = BitConverter.GetBytes(data.Length);
// crc32.TransformBlock(l, 0, l.Length, new byte[l.Length], 0);
// crc32.TransformBlock(data, 0, data.Length, new byte[data.Length], 0);
// }
//}
}
public class HierarchicalSchemaVersionInfo
{
public ushort MajorVersion { get; }
public ushort MinorVersion { get; }
public uint Checksum { get; }
public uint NumScopes { get; }
public uint NumItems { get; }
internal HierarchicalSchemaVersionInfo (ushort majorVersion, ushort minorVersion, uint checksum, uint numScopes, uint numItems)
{
MajorVersion = majorVersion;
MinorVersion = minorVersion;
Checksum = checksum;
NumScopes = numScopes;
NumItems = numItems;
}
}
public class ResourceMapEntry
{
public ushort Index { get; }
public ResourceMapScope Parent { get; internal set; }
public string Name { get; }
internal ResourceMapEntry (ushort index, ResourceMapScope parent, string name)
{
Index = index;
Parent = parent;
Name = name;
}
string fullName;
public string FullName
{
get
{
if (fullName == null)
if (Parent == null) fullName = Name;
else fullName = Parent.FullName + "\\" + Name;
return fullName;
}
}
~ResourceMapEntry () { Parent = null; }
}
public class ResourceMapScope: ResourceMapEntry
{
internal ResourceMapScope (ushort index, ResourceMapScope parent, string name) : base (index, parent, name) {}
public IReadOnlyList <ResourceMapEntry> Children { get; internal set; }
public override string ToString ()
{
return $"Scope {Index} {FullName} ({Children.Count} children)";
}
~ResourceMapScope () { Children = null; }
}
public class ResourceMapItem: ResourceMapEntry
{
internal ResourceMapItem (ushort index, ResourceMapScope parent, string name) : base (index, parent, name) {}
public override string ToString ()
{
return $"Item {Index} {FullName}";
}
}
}