/// <summary>
/// Dispose method
/// </summary>
/// <param name="disposing"><c>true</c> if called by <see cref="Dispose()"/></param>
protected virtual void Dispose(bool disposing)
{
if (!disposing)
{
return;
}
Dispose(peImage);
Dispose(stringsStream);
Dispose(usStream);
Dispose(blobStream);
Dispose(guidStream);
Dispose(tablesStream);
var as2 = allStreams;
if (as2 != null)
{
foreach (var stream in as2.GetSafeEnumerable())
{
Dispose(stream);
}
}
peImage = null;
cor20Header = null;
mdHeader = null;
stringsStream = null;
usStream = null;
blobStream = null;
guidStream = null;
tablesStream = null;
allStreams = null;
fieldRidToTypeDefRid = null;
methodRidToTypeDefRid = null;
typeDefRidToNestedClasses = null;
}
/// <summary>
/// Creates empty heap objects if they're not present in the metadata
/// </summary>
protected void InitializeNonExistentHeaps()
{
if (stringsStream == null)
{
stringsStream = new StringsStream();
}
if (usStream == null)
{
usStream = new USStream();
}
if (blobStream == null)
{
blobStream = new BlobStream();
}
if (guidStream == null)
{
guidStream = new GuidStream();
}
}
/// <inheritdoc/>
protected override void InitializeInternal()
{
IImageStream imageStream = null;
DotNetStream dns = null;
try
{
var mdRva = cor20Header.MetaData.VirtualAddress;
foreach (var sh in mdHeader.StreamHeaders)
{
var rva = mdRva + sh.Offset;
imageStream = peImage.CreateStream(rva, sh.StreamSize);
switch (sh.Name.ToUpperInvariant())
{
case "#STRINGS":
if (stringsStream == null)
{
stringsStream = new StringsStream(imageStream, sh);
imageStream = null;
allStreams.Add(stringsStream);
continue;
}
break;
case "#US":
if (usStream == null)
{
usStream = new USStream(imageStream, sh);
imageStream = null;
allStreams.Add(usStream);
continue;
}
break;
case "#BLOB":
if (blobStream == null)
{
blobStream = new BlobStream(imageStream, sh);
imageStream = null;
allStreams.Add(blobStream);
continue;
}
break;
case "#GUID":
if (guidStream == null)
{
guidStream = new GuidStream(imageStream, sh);
imageStream = null;
allStreams.Add(guidStream);
continue;
}
break;
case "#~": // Only if #Schema is used
case "#-":
if (tablesStream == null)
{
tablesStream = new TablesStream(imageStream, sh);
imageStream = null;
allStreams.Add(tablesStream);
continue;
}
break;
}
dns = new DotNetStream(imageStream, sh);
imageStream = null;
allStreams.Add(dns);
dns = null;
}
}
finally
{
if (imageStream != null)
{
imageStream.Dispose();
}
if (dns != null)
{
dns.Dispose();
}
}
if (tablesStream == null)
{
throw new BadImageFormatException("Missing MD stream");
}
tablesStream.Initialize(peImage);
// The pointer tables are used iff row count != 0
hasFieldPtr = !tablesStream.FieldPtrTable.IsEmpty;
hasMethodPtr = !tablesStream.MethodPtrTable.IsEmpty;
hasParamPtr = !tablesStream.ParamPtrTable.IsEmpty;
hasEventPtr = !tablesStream.EventPtrTable.IsEmpty;
hasPropertyPtr = !tablesStream.PropertyPtrTable.IsEmpty;
hasDeletedRows = tablesStream.HasDelete;
}
/// <inheritdoc/>
protected override void InitializeInternal()
{
var hotHeapVersion = GetHotHeapVersion(peImage.FileName, mdHeader.VersionString);
IImageStream imageStream = null, fullStream = null;
DotNetStream dns = null;
List <HotStream> hotStreams = null;
HotStream hotStream = null;
var newAllStreams = new List <DotNetStream>(allStreams);
try
{
var mdRva = cor20Header.MetaData.VirtualAddress;
for (int i = mdHeader.StreamHeaders.Count - 1; i >= 0; i--)
{
var sh = mdHeader.StreamHeaders[i];
var rva = mdRva + sh.Offset;
var fileOffset = peImage.ToFileOffset(rva);
imageStream = peImage.CreateStream(fileOffset, sh.StreamSize);
switch (sh.Name)
{
case "#Strings":
if (stringsStream == null)
{
stringsStream = new StringsStream(imageStream, sh);
imageStream = null;
newAllStreams.Add(stringsStream);
continue;
}
break;
case "#US":
if (usStream == null)
{
usStream = new USStream(imageStream, sh);
imageStream = null;
newAllStreams.Add(usStream);
continue;
}
break;
case "#Blob":
if (blobStream == null)
{
blobStream = new BlobStream(imageStream, sh);
imageStream = null;
newAllStreams.Add(blobStream);
continue;
}
break;
case "#GUID":
if (guidStream == null)
{
guidStream = new GuidStream(imageStream, sh);
imageStream = null;
newAllStreams.Add(guidStream);
continue;
}
break;
case "#~":
if (tablesStream == null)
{
tablesStream = new TablesStream(imageStream, sh);
imageStream = null;
newAllStreams.Add(tablesStream);
continue;
}
break;
case "#!":
if (hotStreams == null)
{
hotStreams = new List <HotStream>();
}
fullStream = peImage.CreateFullStream();
hotStream = HotStream.Create(hotHeapVersion, imageStream, sh, fullStream, fileOffset);
fullStream = null;
hotStreams.Add(hotStream);
newAllStreams.Add(hotStream);
hotStream = null;
imageStream = null;
continue;
}
dns = new DotNetStream(imageStream, sh);
imageStream = null;
newAllStreams.Add(dns);
dns = null;
}
}
finally
{
if (imageStream != null)
{
imageStream.Dispose();
}
if (fullStream != null)
{
fullStream.Dispose();
}
if (dns != null)
{
dns.Dispose();
}
if (hotStream != null)
{
hotStream.Dispose();
}
newAllStreams.Reverse();
allStreams = ThreadSafeListCreator.MakeThreadSafe(newAllStreams);
}
if (tablesStream == null)
{
throw new BadImageFormatException("Missing MD stream");
}
if (hotStreams != null)
{
hotStreams.Reverse();
InitializeHotStreams(hotStreams);
}
tablesStream.Initialize(peImage);
}