152 lines
5.8 KiB
C#
152 lines
5.8 KiB
C#
|
/*
|
|||
|
Copyright (C) 2011 de4dot@gmail.com
|
|||
|
|
|||
|
This file is part of de4dot.
|
|||
|
|
|||
|
de4dot is free software: you can redistribute it and/or modify
|
|||
|
it under the terms of the GNU General Public License as published by
|
|||
|
the Free Software Foundation, either version 3 of the License, or
|
|||
|
(at your option) any later version.
|
|||
|
|
|||
|
de4dot is distributed in the hope that it will be useful,
|
|||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|||
|
GNU General Public License for more details.
|
|||
|
|
|||
|
You should have received a copy of the GNU General Public License
|
|||
|
along with de4dot. If not, see <http://www.gnu.org/licenses/>.
|
|||
|
*/
|
|||
|
|
|||
|
using System;
|
|||
|
using System.IO;
|
|||
|
|
|||
|
namespace de4dot.PE {
|
|||
|
using MVT = MetadataVarType;
|
|||
|
|
|||
|
class MetadataTables {
|
|||
|
BinaryReader reader;
|
|||
|
Metadata metadata;
|
|||
|
byte heapOffsetSizes;
|
|||
|
MetadataType[] metadataTypes = new MetadataType[64];
|
|||
|
|
|||
|
public MetadataTables(BinaryReader reader, Metadata metadata) {
|
|||
|
this.reader = reader;
|
|||
|
this.metadata = metadata;
|
|||
|
init();
|
|||
|
}
|
|||
|
|
|||
|
public MetadataType getMetadataType(MetadataIndex index) {
|
|||
|
return metadataTypes[(int)index];
|
|||
|
}
|
|||
|
|
|||
|
void seek(uint fileOffset) {
|
|||
|
reader.BaseStream.Position = fileOffset;
|
|||
|
}
|
|||
|
|
|||
|
static MetadataVarType[] metadataVarType = new MetadataVarType[] {
|
|||
|
MVT.byte2, MVT.stringIndex, MVT.guidIndex, MVT.guidIndex, MVT.guidIndex, MVT.end, // 0
|
|||
|
MVT.resolutionScope, MVT.stringIndex, MVT.stringIndex, MVT.end, // 1
|
|||
|
MVT.byte4, MVT.stringIndex, MVT.stringIndex, MVT.typeDefOrRef, MVT.fieldIndex, MVT.methodDefIndex, MVT.end, // 2
|
|||
|
MVT.end, // 3
|
|||
|
MVT.byte2, MVT.stringIndex, MVT.blobIndex, MVT.end, // 4
|
|||
|
MVT.end, // 5
|
|||
|
MVT.byte4, MVT.byte2, MVT.byte2, MVT.stringIndex, MVT.blobIndex, MVT.paramIndex, MVT.end, // 6
|
|||
|
MVT.end, // 7
|
|||
|
MVT.byte2, MVT.byte2, MVT.stringIndex, MVT.end, // 8
|
|||
|
MVT.typeDefIndex, MVT.typeDefOrRef, MVT.end, // 9
|
|||
|
MVT.memberRefParent, MVT.stringIndex, MVT.blobIndex, MVT.end, // 10
|
|||
|
MVT.byte1, MVT.byte1, MVT.hasConstant, MVT.blobIndex, MVT.end, // 11
|
|||
|
MVT.hasCustomAttribute, MVT.customAttributeType, MVT.blobIndex, MVT.end,// 12
|
|||
|
MVT.hasFieldMarshal, MVT.blobIndex, MVT.end, // 13
|
|||
|
MVT.byte2, MVT.hasDeclSecurity, MVT.blobIndex, MVT.end, // 14
|
|||
|
MVT.byte2, MVT.byte4, MVT.typeDefIndex, MVT.end, // 15
|
|||
|
MVT.byte4, MVT.fieldIndex, MVT.end, // 16
|
|||
|
MVT.blobIndex, MVT.end, // 17
|
|||
|
MVT.typeDefIndex, MVT.eventIndex, MVT.end, // 18
|
|||
|
MVT.end, // 19
|
|||
|
MVT.byte2, MVT.stringIndex, MVT.typeDefOrRef, MVT.end, // 20
|
|||
|
MVT.typeDefIndex, MVT.propertyIndex, MVT.end, // 21
|
|||
|
MVT.end, // 22
|
|||
|
MVT.byte2, MVT.stringIndex, MVT.blobIndex, MVT.end, // 23
|
|||
|
MVT.byte2, MVT.methodDefIndex, MVT.hasSemantics, MVT.end, // 24
|
|||
|
MVT.typeDefIndex, MVT.methodDefOrRef, MVT.methodDefOrRef, MVT.end, // 25
|
|||
|
MVT.stringIndex, MVT.end, // 26
|
|||
|
MVT.blobIndex, MVT.end, // 27
|
|||
|
MVT.byte2, MVT.memberForwarded, MVT.stringIndex, MVT.moduleRefIndex, MVT.end, // 28
|
|||
|
MVT.byte4, MVT.fieldIndex, MVT.end, // 29
|
|||
|
MVT.end, // 30
|
|||
|
MVT.end, // 31
|
|||
|
MVT.byte4, MVT.byte2, MVT.byte2, MVT.byte2, MVT.byte2, MVT.byte4, MVT.blobIndex, MVT.stringIndex, MVT.stringIndex, MVT.end, // 32
|
|||
|
MVT.byte4, MVT.end, // 33
|
|||
|
MVT.byte4, MVT.byte4, MVT.byte4, MVT.end, // 34
|
|||
|
MVT.byte2, MVT.byte2, MVT.byte2, MVT.byte2, MVT.byte4, MVT.blobIndex, MVT.stringIndex, MVT.stringIndex, MVT.blobIndex, MVT.end, // 35
|
|||
|
MVT.byte4, MVT.assemblyRefIndex, MVT.end, // 36
|
|||
|
MVT.byte4, MVT.byte4, MVT.byte4, MVT.assemblyRefIndex, MVT.end, // 37
|
|||
|
MVT.byte4, MVT.stringIndex, MVT.blobIndex, MVT.end, // 38
|
|||
|
MVT.byte4, MVT.byte4, MVT.stringIndex, MVT.stringIndex, MVT.implementation, MVT.end,// 39
|
|||
|
MVT.byte4, MVT.byte4, MVT.stringIndex, MVT.implementation, MVT.end, // 40
|
|||
|
MVT.typeDefIndex, MVT.typeDefIndex, MVT.end, // 41
|
|||
|
MVT.byte2, MVT.byte2, MVT.typeOrMethodDef, MVT.stringIndex, MVT.end, // 42
|
|||
|
MVT.end, // 43
|
|||
|
MVT.genericParamIndex, MVT.typeDefOrRef, MVT.end, // 44
|
|||
|
MVT.end, // 45
|
|||
|
MVT.end, // 46
|
|||
|
MVT.end, // 47
|
|||
|
MVT.end, // 48
|
|||
|
MVT.end, // 49
|
|||
|
MVT.end, // 50
|
|||
|
MVT.end, // 51
|
|||
|
MVT.end, // 52
|
|||
|
MVT.end, // 53
|
|||
|
MVT.end, // 54
|
|||
|
MVT.end, // 55
|
|||
|
MVT.end, // 56
|
|||
|
MVT.end, // 57
|
|||
|
MVT.end, // 58
|
|||
|
MVT.end, // 59
|
|||
|
MVT.end, // 60
|
|||
|
MVT.end, // 61
|
|||
|
MVT.end, // 62
|
|||
|
MVT.end, // 63
|
|||
|
|
|||
|
MVT.stop
|
|||
|
};
|
|||
|
|
|||
|
void init() {
|
|||
|
var streamTable = metadata.getStream("#~");
|
|||
|
if (streamTable == null)
|
|||
|
throw new ApplicationException("Could not find #~ stream");
|
|||
|
|
|||
|
seek(streamTable.Offset);
|
|||
|
reader.ReadUInt32(); // reserved
|
|||
|
reader.ReadUInt16(); // major + minor version
|
|||
|
heapOffsetSizes = reader.ReadByte();
|
|||
|
reader.ReadByte(); // always 1
|
|||
|
ulong validMask = reader.ReadUInt64();
|
|||
|
reader.ReadUInt64(); // sorted
|
|||
|
|
|||
|
var numRows = new uint[64];
|
|||
|
for (int i = 0; validMask != 0; i++, validMask >>= 1) {
|
|||
|
if ((validMask & 1) != 0)
|
|||
|
numRows[i] = reader.ReadUInt32();
|
|||
|
}
|
|||
|
|
|||
|
var builder = new MetadataTypeBuilder(heapOffsetSizes, numRows);
|
|||
|
uint fileOffset = (uint)reader.BaseStream.Position;
|
|||
|
for (int i = 0, j = 0; ; i++) {
|
|||
|
if (metadataVarType[i] == MVT.end) {
|
|||
|
var mdType = builder.create();
|
|||
|
mdType.rows = numRows[j];
|
|||
|
mdType.fileOffset = fileOffset;
|
|||
|
fileOffset += mdType.rows * mdType.totalSize;
|
|||
|
metadataTypes[j++] = mdType;
|
|||
|
}
|
|||
|
else if (metadataVarType[i] == MVT.stop)
|
|||
|
break;
|
|||
|
else
|
|||
|
builder.field(metadataVarType[i]);
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
}
|