|
|
@@ -18,10 +18,16 @@
|
|
|
using System;
|
|
|
using System.IO;
|
|
|
using System.Collections;
|
|
|
+using System.Collections.Generic;
|
|
|
|
|
|
|
|
|
namespace QUT.PERWAPI
|
|
|
{
|
|
|
+ /// <summary>
|
|
|
+ /// (Unmanaged) Resource Elements consist of PEResourceDirectories
|
|
|
+ /// or PEResourceData elements. Resource directories may be nested up
|
|
|
+ /// to three deep sorted on Type, Name and Language in that order.
|
|
|
+ /// </summary>
|
|
|
public abstract class PEResourceElement
|
|
|
{
|
|
|
|
|
|
@@ -44,116 +50,221 @@ namespace QUT.PERWAPI
|
|
|
|
|
|
protected internal abstract uint Size();
|
|
|
|
|
|
- protected internal abstract void Write(BinaryWriter dest, uint RVA);
|
|
|
+ /// <summary>
|
|
|
+ /// Write out the unmanaged resource data.
|
|
|
+ /// </summary>
|
|
|
+ /// <param name="dest">The Binary Writer</param>
|
|
|
+ /// <param name="baseOffset">File position at start of .rsrc section</param>
|
|
|
+ /// <param name="RVA">RVA of .rsrc section when loaded</param>
|
|
|
+ protected internal abstract void Write(BinaryWriter dest, uint baseOffset, uint currentOffset, uint RVA);
|
|
|
+
|
|
|
+ protected internal long offset;
|
|
|
+
|
|
|
+ protected const uint HeaderSize = 16;
|
|
|
+ protected const uint EntrySize = 8;
|
|
|
}
|
|
|
|
|
|
- public class PEResourceDirectory : PEResourceElement
|
|
|
- {
|
|
|
- public PEResourceDirectory() { }
|
|
|
+ /// <summary>
|
|
|
+ /// ResourceDirectory entries, as defined in Winnt.h
|
|
|
+ /// as type struct _IMAGE_RESOURCE_DIRECTORY.
|
|
|
+ /// </summary>
|
|
|
+ public class PEResourceDirectory : PEResourceElement {
|
|
|
|
|
|
- private uint date = 0;
|
|
|
- private ushort majver = 1;
|
|
|
- private ushort minver = 0;
|
|
|
+ private uint date = 0;
|
|
|
+ private ushort majver = 1;
|
|
|
+ private ushort minver = 0;
|
|
|
|
|
|
- public uint Date { get { return date; } set { date = value; } }
|
|
|
- public ushort MajVer { get { return majver; } set { majver = value; } }
|
|
|
- public ushort MinVer { get { return minver; } set { minver = value; } }
|
|
|
+ public uint Date { get { return date; } set { date = value; } }
|
|
|
+ public ushort MajVer { get { return majver; } set { majver = value; } }
|
|
|
+ public ushort MinVer { get { return minver; } set { minver = value; } }
|
|
|
|
|
|
- private ArrayList subitems = new ArrayList();
|
|
|
+ //private ArrayList subItems = new ArrayList();
|
|
|
+ private List<PEResourceElement> elements = new List<PEResourceElement>();
|
|
|
|
|
|
- public PEResourceElement this[int i] { get { return (PEResourceElement)subitems[i]; } }
|
|
|
+ public int Count() { return elements.Count; }
|
|
|
|
|
|
- public int Count() { return subitems.Count; }
|
|
|
+ /// <summary>
|
|
|
+ /// Programmatically create unmanaged resource.
|
|
|
+ /// </summary>
|
|
|
+ public PEResourceDirectory() { }
|
|
|
|
|
|
- public bool HasData()
|
|
|
- {
|
|
|
- return subitems.Count > 0;
|
|
|
+
|
|
|
+ /// <summary>
|
|
|
+ /// Read unmanged resource directory structure from PE-file.
|
|
|
+ /// </summary>
|
|
|
+ /// <param name="reader"></param>
|
|
|
+ internal void PopulateResourceDirectory(PEReader reader, long baseOffset) {
|
|
|
+ PEResourceElement resElement = null;
|
|
|
+ PEResourceDirectory resDirectory;
|
|
|
+ PEResourceData resData;
|
|
|
+
|
|
|
+ int junk = reader.ReadInt32(); // Must be zero.
|
|
|
+ this.date = reader.ReadUInt32(); // Time stamp.
|
|
|
+ this.majver = reader.ReadUInt16();
|
|
|
+ this.minver = reader.ReadUInt16();
|
|
|
+
|
|
|
+ int numNameEntries = reader.ReadUInt16(); // Number of named entries.
|
|
|
+ int numIdntEntries = reader.ReadUInt16(); // Number of ID entries.
|
|
|
+ for (int i = 0; i < numNameEntries; i++) {
|
|
|
+ uint nameOrId = reader.ReadUInt32();
|
|
|
+ uint elemOfst = reader.ReadUInt32();
|
|
|
+ if ((elemOfst & 0x80000000) != 0) // Element is subdirectory.
|
|
|
+ resElement = new PEResourceDirectory();
|
|
|
+ else
|
|
|
+ resElement = new PEResourceData();
|
|
|
+ resElement.Name = ReadName(reader, baseOffset + nameOrId & 0x7fffffff);
|
|
|
+ resElement.offset = baseOffset + (long)(elemOfst & 0x7fffffff);
|
|
|
+ this.AddElement(resElement);
|
|
|
+ }
|
|
|
+ // Read the Ident entries.
|
|
|
+ for (int i = 0; i < numIdntEntries; i++) {
|
|
|
+ uint nameOrId = reader.ReadUInt32();
|
|
|
+ uint elemOfst = reader.ReadUInt32();
|
|
|
+ if ((elemOfst & 0x80000000) != 0) // Element is subdirectory.
|
|
|
+ resElement = new PEResourceDirectory();
|
|
|
+ else
|
|
|
+ resElement = new PEResourceData();
|
|
|
+ resElement.Id = (ushort)nameOrId;
|
|
|
+ resElement.offset = baseOffset + (long)(elemOfst & 0x7fffffff);
|
|
|
+ this.AddElement(resElement);
|
|
|
}
|
|
|
+ // Now recurse to get subdirectories/the real data.
|
|
|
+ foreach (PEResourceElement elem in this.elements) {
|
|
|
+ if ((resDirectory = elem as PEResourceDirectory) != null) {
|
|
|
+ reader.BaseStream.Seek(resDirectory.offset, SeekOrigin.Begin);
|
|
|
+ resDirectory.PopulateResourceDirectory(reader, baseOffset);
|
|
|
+ }
|
|
|
+ else if ((resData = elem as PEResourceData) != null) {
|
|
|
+ reader.BaseStream.Seek(resData.offset, SeekOrigin.Begin);
|
|
|
+ resData.PopulateResourceData(reader, baseOffset);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
|
|
|
- public void AddElement(PEResourceElement el)
|
|
|
- {
|
|
|
- subitems.Add(el);
|
|
|
+ private string ReadName(BinaryReader rdr, long offset) {
|
|
|
+ long savedPos = rdr.BaseStream.Position;
|
|
|
+ rdr.BaseStream.Seek(offset, SeekOrigin.Begin);
|
|
|
+ ushort nLength = rdr.ReadUInt16();
|
|
|
+ char[] name = new char[nLength];
|
|
|
+ for (int i = 0; i < nLength; i++)
|
|
|
+ name[i] = (char)rdr.ReadUInt16();
|
|
|
+ return new string(name);
|
|
|
+ }
|
|
|
+
|
|
|
+ public bool HasData() {
|
|
|
+ return elements.Count > 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ public void AddElement(PEResourceElement el) {
|
|
|
+ //subItems.Add(el);
|
|
|
+ elements.Add(el);
|
|
|
+ }
|
|
|
+
|
|
|
+ /// <summary>
|
|
|
+ /// Total file-space size of all child elements
|
|
|
+ /// </summary>
|
|
|
+ private uint subSize;
|
|
|
+
|
|
|
+ /// <summary>
|
|
|
+ /// File-space needed for all names of this directory
|
|
|
+ /// </summary>
|
|
|
+ private uint nameSize;
|
|
|
+
|
|
|
+ /// <summary>
|
|
|
+ /// File-space taken up by this directory
|
|
|
+ /// </summary>
|
|
|
+ private uint dirSize;
|
|
|
+
|
|
|
+ /// <summary>
|
|
|
+ /// Number of named elements. These come first in list.
|
|
|
+ /// </summary>
|
|
|
+ private uint numNamed;
|
|
|
+
|
|
|
+ private uint numIds { get { return (uint)elements.Count - numNamed; } }
|
|
|
+
|
|
|
+ protected internal override uint Size() {
|
|
|
+ nameSize = 0;
|
|
|
+ numNamed = 0;
|
|
|
+ subSize = 0;
|
|
|
+ //for (int i = 0; i < subItems.Count; i++)
|
|
|
+ foreach (PEResourceElement elem in this.elements) {
|
|
|
+ subSize += elem.Size();
|
|
|
+ if (elem.Name != null) {
|
|
|
+ nameSize += 2 + (uint)elem.Name.Length * 2;
|
|
|
+ numNamed++;
|
|
|
+ }
|
|
|
}
|
|
|
- private uint subsize, namesize, dirsize, numnamed;
|
|
|
- protected internal override uint Size()
|
|
|
- {
|
|
|
- namesize = 0;
|
|
|
- numnamed = 0;
|
|
|
- subsize = 0;
|
|
|
- for (int i = 0; i < subitems.Count; i++)
|
|
|
- {
|
|
|
- PEResourceElement el = (PEResourceElement)subitems[i];
|
|
|
- subsize += el.Size();
|
|
|
- if (el.Name != null)
|
|
|
- {
|
|
|
- namesize += 2 + (uint)el.Name.Length * 2;
|
|
|
- numnamed++;
|
|
|
- }
|
|
|
- }
|
|
|
- dirsize = (uint)subitems.Count * 8 + 16;
|
|
|
- return dirsize + namesize + subsize;
|
|
|
+ dirSize = (uint)elements.Count * EntrySize + HeaderSize;
|
|
|
+ return dirSize + nameSize + subSize;
|
|
|
+ }
|
|
|
+
|
|
|
+ /// <summary>
|
|
|
+ /// Write out the unmanaged resource rooted at this directory.
|
|
|
+ /// </summary>
|
|
|
+ /// <param name="dest">The Binary Writer</param>
|
|
|
+ /// <param name="RVA">RVA of this .rsrc section</param>
|
|
|
+ internal void Write(BinaryWriter dest, uint RVA) {
|
|
|
+ Size();
|
|
|
+ dest.Flush();
|
|
|
+ uint baseOffset = (uint)dest.BaseStream.Position;
|
|
|
+ this.Write(dest, baseOffset, 0, RVA);
|
|
|
+ }
|
|
|
+
|
|
|
+ protected internal override void Write(BinaryWriter dest, uint baseOffset, uint currentOffset, uint RVA) {
|
|
|
+ uint nameOffset = currentOffset + this.dirSize;
|
|
|
+ uint targetOffset = currentOffset + this.dirSize;
|
|
|
+ dest.Write((uint)0); // characteristics
|
|
|
+ dest.Write(date); // datetime
|
|
|
+ dest.Write(majver);
|
|
|
+ dest.Write(minver);
|
|
|
+ dest.Write((ushort)numNamed);
|
|
|
+ dest.Write((ushort)numIds);
|
|
|
+ currentOffset += HeaderSize;
|
|
|
+
|
|
|
+ // Write out the named items.
|
|
|
+ foreach (PEResourceElement elem in elements) {
|
|
|
+ if (elem.Name != null) {
|
|
|
+ dest.Write((uint)(nameOffset | 0x80000000));
|
|
|
+ if (elem is PEResourceDirectory)
|
|
|
+ dest.Write((uint)(targetOffset | 0x80000000));
|
|
|
+ else
|
|
|
+ dest.Write((uint)targetOffset);
|
|
|
+ nameOffset += 2 + (uint)elem.Name.Length * 2;
|
|
|
+ targetOffset += (uint)elem.Size();
|
|
|
+ currentOffset += EntrySize;
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
- protected internal override void Write(BinaryWriter dest, uint RVA)
|
|
|
- {
|
|
|
- Size();
|
|
|
- dest.Flush();
|
|
|
- uint startnameoffset = (uint)dest.BaseStream.Position + (uint)dirsize;
|
|
|
- uint curritemoffset = startnameoffset + (uint)namesize;
|
|
|
- dest.Write((uint)0); // characteristics
|
|
|
- dest.Write(date); // datetime
|
|
|
- dest.Write(majver);
|
|
|
- dest.Write(minver);
|
|
|
- dest.Write((ushort)numnamed);
|
|
|
- dest.Write((ushort)(subitems.Count - numnamed));
|
|
|
-
|
|
|
- uint currnameoffset = startnameoffset;
|
|
|
- for (int i = 0; i < subitems.Count; i++)
|
|
|
- {
|
|
|
- PEResourceElement el = (PEResourceElement)subitems[i];
|
|
|
- if (el.Name != null)
|
|
|
- {
|
|
|
- dest.Write((uint)(currnameoffset | 0x80000000));
|
|
|
- if (el is PEResourceDirectory)
|
|
|
- dest.Write((uint)(curritemoffset | 0x80000000));
|
|
|
- else
|
|
|
- dest.Write((uint)curritemoffset);
|
|
|
- currnameoffset += 2 + (uint)el.Name.Length * 2;
|
|
|
- }
|
|
|
- curritemoffset += el.Size();
|
|
|
- }
|
|
|
- curritemoffset = startnameoffset + namesize;
|
|
|
- for (int i = 0; i < subitems.Count; i++)
|
|
|
- {
|
|
|
- PEResourceElement el = (PEResourceElement)subitems[i];
|
|
|
- if (el.Name == null)
|
|
|
- {
|
|
|
- dest.Write(el.Id);
|
|
|
- if (el is PEResourceDirectory)
|
|
|
- dest.Write((uint)(curritemoffset | 0x80000000));
|
|
|
- else
|
|
|
- dest.Write((uint)curritemoffset);
|
|
|
- }
|
|
|
- curritemoffset += el.Size();
|
|
|
- }
|
|
|
- for (int i = 0; i < subitems.Count; i++)
|
|
|
- {
|
|
|
- PEResourceElement el = (PEResourceElement)subitems[i];
|
|
|
- string s = el.Name;
|
|
|
- if (s != null)
|
|
|
- {
|
|
|
- dest.Write((ushort)s.Length);
|
|
|
- byte[] b = System.Text.Encoding.Unicode.GetBytes(s);
|
|
|
- dest.Write(b);
|
|
|
- }
|
|
|
- }
|
|
|
- for (int i = 0; i < subitems.Count; i++)
|
|
|
- {
|
|
|
- PEResourceElement el = (PEResourceElement)subitems[i];
|
|
|
- el.Write(dest, RVA);
|
|
|
- }
|
|
|
+ // Write out the items with ID.
|
|
|
+ foreach (PEResourceElement elem in elements) {
|
|
|
+ if (elem.Name == null) {
|
|
|
+ dest.Write(elem.Id);
|
|
|
+ if (elem is PEResourceDirectory)
|
|
|
+ dest.Write((uint)(targetOffset | 0x80000000));
|
|
|
+ else
|
|
|
+ dest.Write((uint)targetOffset);
|
|
|
+ currentOffset += EntrySize;
|
|
|
+ targetOffset += elem.Size();
|
|
|
+ }
|
|
|
+ }
|
|
|
|
|
|
+ // Write out the name strings.
|
|
|
+ foreach (PEResourceElement elem in elements) {
|
|
|
+ string s = elem.Name;
|
|
|
+ if (s != null) {
|
|
|
+ dest.Write((ushort)s.Length);
|
|
|
+ byte[] b = System.Text.Encoding.Unicode.GetBytes(s);
|
|
|
+ dest.Write(b);
|
|
|
+ }
|
|
|
}
|
|
|
+ currentOffset += this.nameSize;
|
|
|
+
|
|
|
+ // Now recurse to the children.
|
|
|
+ foreach (PEResourceElement elem in elements) {
|
|
|
+ elem.Write(dest, baseOffset, currentOffset, RVA);
|
|
|
+ currentOffset += elem.Size();
|
|
|
+ }
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
public class PEResourceData : PEResourceElement
|
|
|
@@ -171,10 +282,23 @@ namespace QUT.PERWAPI
|
|
|
return 16 + (uint)Data.Length;
|
|
|
}
|
|
|
|
|
|
- protected internal override void Write(BinaryWriter dest, uint RVA)
|
|
|
+ /// <summary>
|
|
|
+ /// Read the binary data from the PE file.
|
|
|
+ /// </summary>
|
|
|
+ /// <param name="reader"></param>
|
|
|
+ internal void PopulateResourceData(PEReader reader, long baseOffset) {
|
|
|
+ uint dataRVA = reader.ReadUInt32();
|
|
|
+ int dataLength = reader.ReadInt32();
|
|
|
+ this.codepage = reader.ReadInt32();
|
|
|
+ uint junk = reader.ReadUInt32(); // Must be zero.
|
|
|
+ reader.BaseStream.Seek(reader.GetOffset(dataRVA), SeekOrigin.Begin);
|
|
|
+ data = new byte[dataLength];
|
|
|
+ int numberRead = reader.BaseStream.Read(data, 0, dataLength);
|
|
|
+ }
|
|
|
+
|
|
|
+ protected internal override void Write(BinaryWriter dest, uint baseOffset, uint currentOffset, uint RVA)
|
|
|
{
|
|
|
- dest.Flush();
|
|
|
- dest.Write((uint)(dest.BaseStream.Position + 16) + RVA);
|
|
|
+ dest.Write((uint)(currentOffset + HeaderSize) + RVA);
|
|
|
dest.Write((uint)data.Length);
|
|
|
dest.Write((uint)codepage);
|
|
|
dest.Write((uint)0);
|
k_john_gough_cp