/Mono.Cecil/AssemblyReader.cs

http://github.com/jbevain/cecil · C# · 3887 lines · 2977 code · 897 blank · 13 comment · 599 complexity · d42c832854fc52a9bb4626201d3ae696 MD5 · raw file

Large files are truncated click here to view the full file

  1. //
  2. // Author:
  3. // Jb Evain (jbevain@gmail.com)
  4. //
  5. // Copyright (c) 2008 - 2015 Jb Evain
  6. // Copyright (c) 2008 - 2011 Novell, Inc.
  7. //
  8. // Licensed under the MIT/X11 license.
  9. //
  10. using System;
  11. using System.Collections.Generic;
  12. using System.IO;
  13. using System.IO.Compression;
  14. using System.Text;
  15. using Mono.Collections.Generic;
  16. using Mono.Cecil.Cil;
  17. using Mono.Cecil.Metadata;
  18. using Mono.Cecil.PE;
  19. using RVA = System.UInt32;
  20. namespace Mono.Cecil {
  21. abstract class ModuleReader {
  22. readonly protected ModuleDefinition module;
  23. protected ModuleReader (Image image, ReadingMode mode)
  24. {
  25. this.module = new ModuleDefinition (image);
  26. this.module.ReadingMode = mode;
  27. }
  28. protected abstract void ReadModule ();
  29. public abstract void ReadSymbols (ModuleDefinition module);
  30. protected void ReadModuleManifest (MetadataReader reader)
  31. {
  32. reader.Populate (module);
  33. ReadAssembly (reader);
  34. }
  35. void ReadAssembly (MetadataReader reader)
  36. {
  37. var name = reader.ReadAssemblyNameDefinition ();
  38. if (name == null) {
  39. module.kind = ModuleKind.NetModule;
  40. return;
  41. }
  42. var assembly = new AssemblyDefinition ();
  43. assembly.Name = name;
  44. module.assembly = assembly;
  45. assembly.main_module = module;
  46. }
  47. public static ModuleDefinition CreateModule (Image image, ReaderParameters parameters)
  48. {
  49. var reader = CreateModuleReader (image, parameters.ReadingMode);
  50. var module = reader.module;
  51. if (parameters.assembly_resolver != null)
  52. module.assembly_resolver = Disposable.NotOwned (parameters.assembly_resolver);
  53. if (parameters.metadata_resolver != null)
  54. module.metadata_resolver = parameters.metadata_resolver;
  55. if (parameters.metadata_importer_provider != null)
  56. module.metadata_importer = parameters.metadata_importer_provider.GetMetadataImporter (module);
  57. if (parameters.reflection_importer_provider != null)
  58. module.reflection_importer = parameters.reflection_importer_provider.GetReflectionImporter (module);
  59. GetMetadataKind (module, parameters);
  60. reader.ReadModule ();
  61. ReadSymbols (module, parameters);
  62. reader.ReadSymbols (module);
  63. if (parameters.ReadingMode == ReadingMode.Immediate)
  64. module.MetadataSystem.Clear ();
  65. return module;
  66. }
  67. static void ReadSymbols (ModuleDefinition module, ReaderParameters parameters)
  68. {
  69. var symbol_reader_provider = parameters.SymbolReaderProvider;
  70. if (symbol_reader_provider == null && parameters.ReadSymbols)
  71. symbol_reader_provider = new DefaultSymbolReaderProvider ();
  72. if (symbol_reader_provider != null) {
  73. module.SymbolReaderProvider = symbol_reader_provider;
  74. var reader = parameters.SymbolStream != null
  75. ? symbol_reader_provider.GetSymbolReader (module, parameters.SymbolStream)
  76. : symbol_reader_provider.GetSymbolReader (module, module.FileName);
  77. if (reader != null) {
  78. try {
  79. module.ReadSymbols (reader, parameters.ThrowIfSymbolsAreNotMatching);
  80. } catch (Exception) {
  81. reader.Dispose ();
  82. throw;
  83. }
  84. }
  85. }
  86. if (module.Image.HasDebugTables ())
  87. module.ReadSymbols (new PortablePdbReader (module.Image, module));
  88. }
  89. static void GetMetadataKind (ModuleDefinition module, ReaderParameters parameters)
  90. {
  91. if (!parameters.ApplyWindowsRuntimeProjections) {
  92. module.MetadataKind = MetadataKind.Ecma335;
  93. return;
  94. }
  95. var runtime_version = module.RuntimeVersion;
  96. if (!runtime_version.Contains ("WindowsRuntime"))
  97. module.MetadataKind = MetadataKind.Ecma335;
  98. else if (runtime_version.Contains ("CLR"))
  99. module.MetadataKind = MetadataKind.ManagedWindowsMetadata;
  100. else
  101. module.MetadataKind = MetadataKind.WindowsMetadata;
  102. }
  103. static ModuleReader CreateModuleReader (Image image, ReadingMode mode)
  104. {
  105. switch (mode) {
  106. case ReadingMode.Immediate:
  107. return new ImmediateModuleReader (image);
  108. case ReadingMode.Deferred:
  109. return new DeferredModuleReader (image);
  110. default:
  111. throw new ArgumentException ();
  112. }
  113. }
  114. }
  115. sealed class ImmediateModuleReader : ModuleReader {
  116. bool resolve_attributes;
  117. public ImmediateModuleReader (Image image)
  118. : base (image, ReadingMode.Immediate)
  119. {
  120. }
  121. protected override void ReadModule ()
  122. {
  123. this.module.Read (this.module, (module, reader) => {
  124. ReadModuleManifest (reader);
  125. ReadModule (module, resolve_attributes: true);
  126. });
  127. }
  128. public void ReadModule (ModuleDefinition module, bool resolve_attributes)
  129. {
  130. this.resolve_attributes = resolve_attributes;
  131. if (module.HasAssemblyReferences)
  132. Mixin.Read (module.AssemblyReferences);
  133. if (module.HasResources)
  134. Mixin.Read (module.Resources);
  135. if (module.HasModuleReferences)
  136. Mixin.Read (module.ModuleReferences);
  137. if (module.HasTypes)
  138. ReadTypes (module.Types);
  139. if (module.HasExportedTypes)
  140. Mixin.Read (module.ExportedTypes);
  141. ReadCustomAttributes (module);
  142. var assembly = module.Assembly;
  143. if (assembly == null)
  144. return;
  145. ReadCustomAttributes (assembly);
  146. ReadSecurityDeclarations (assembly);
  147. }
  148. void ReadTypes (Collection<TypeDefinition> types)
  149. {
  150. for (int i = 0; i < types.Count; i++)
  151. ReadType (types [i]);
  152. }
  153. void ReadType (TypeDefinition type)
  154. {
  155. ReadGenericParameters (type);
  156. if (type.HasInterfaces)
  157. ReadInterfaces (type);
  158. if (type.HasNestedTypes)
  159. ReadTypes (type.NestedTypes);
  160. if (type.HasLayoutInfo)
  161. Mixin.Read (type.ClassSize);
  162. if (type.HasFields)
  163. ReadFields (type);
  164. if (type.HasMethods)
  165. ReadMethods (type);
  166. if (type.HasProperties)
  167. ReadProperties (type);
  168. if (type.HasEvents)
  169. ReadEvents (type);
  170. ReadSecurityDeclarations (type);
  171. ReadCustomAttributes (type);
  172. }
  173. void ReadInterfaces (TypeDefinition type)
  174. {
  175. var interfaces = type.Interfaces;
  176. for (int i = 0; i < interfaces.Count; i++)
  177. ReadCustomAttributes (interfaces [i]);
  178. }
  179. void ReadGenericParameters (IGenericParameterProvider provider)
  180. {
  181. if (!provider.HasGenericParameters)
  182. return;
  183. var parameters = provider.GenericParameters;
  184. for (int i = 0; i < parameters.Count; i++) {
  185. var parameter = parameters [i];
  186. if (parameter.HasConstraints)
  187. ReadGenericParameterConstraints (parameter);
  188. ReadCustomAttributes (parameter);
  189. }
  190. }
  191. void ReadGenericParameterConstraints (GenericParameter parameter)
  192. {
  193. var constraints = parameter.Constraints;
  194. for (int i = 0; i < constraints.Count; i++)
  195. ReadCustomAttributes (constraints [i]);
  196. }
  197. void ReadSecurityDeclarations (ISecurityDeclarationProvider provider)
  198. {
  199. if (!provider.HasSecurityDeclarations)
  200. return;
  201. var security_declarations = provider.SecurityDeclarations;
  202. if (!resolve_attributes)
  203. return;
  204. for (int i = 0; i < security_declarations.Count; i++) {
  205. var security_declaration = security_declarations [i];
  206. Mixin.Read (security_declaration.SecurityAttributes);
  207. }
  208. }
  209. void ReadCustomAttributes (ICustomAttributeProvider provider)
  210. {
  211. if (!provider.HasCustomAttributes)
  212. return;
  213. var custom_attributes = provider.CustomAttributes;
  214. if (!resolve_attributes)
  215. return;
  216. for (int i = 0; i < custom_attributes.Count; i++) {
  217. var custom_attribute = custom_attributes [i];
  218. Mixin.Read (custom_attribute.ConstructorArguments);
  219. }
  220. }
  221. void ReadFields (TypeDefinition type)
  222. {
  223. var fields = type.Fields;
  224. for (int i = 0; i < fields.Count; i++) {
  225. var field = fields [i];
  226. if (field.HasConstant)
  227. Mixin.Read (field.Constant);
  228. if (field.HasLayoutInfo)
  229. Mixin.Read (field.Offset);
  230. if (field.RVA > 0)
  231. Mixin.Read (field.InitialValue);
  232. if (field.HasMarshalInfo)
  233. Mixin.Read (field.MarshalInfo);
  234. ReadCustomAttributes (field);
  235. }
  236. }
  237. void ReadMethods (TypeDefinition type)
  238. {
  239. var methods = type.Methods;
  240. for (int i = 0; i < methods.Count; i++) {
  241. var method = methods [i];
  242. ReadGenericParameters (method);
  243. if (method.HasParameters)
  244. ReadParameters (method);
  245. if (method.HasOverrides)
  246. Mixin.Read (method.Overrides);
  247. if (method.IsPInvokeImpl)
  248. Mixin.Read (method.PInvokeInfo);
  249. ReadSecurityDeclarations (method);
  250. ReadCustomAttributes (method);
  251. var return_type = method.MethodReturnType;
  252. if (return_type.HasConstant)
  253. Mixin.Read (return_type.Constant);
  254. if (return_type.HasMarshalInfo)
  255. Mixin.Read (return_type.MarshalInfo);
  256. ReadCustomAttributes (return_type);
  257. }
  258. }
  259. void ReadParameters (MethodDefinition method)
  260. {
  261. var parameters = method.Parameters;
  262. for (int i = 0; i < parameters.Count; i++) {
  263. var parameter = parameters [i];
  264. if (parameter.HasConstant)
  265. Mixin.Read (parameter.Constant);
  266. if (parameter.HasMarshalInfo)
  267. Mixin.Read (parameter.MarshalInfo);
  268. ReadCustomAttributes (parameter);
  269. }
  270. }
  271. void ReadProperties (TypeDefinition type)
  272. {
  273. var properties = type.Properties;
  274. for (int i = 0; i < properties.Count; i++) {
  275. var property = properties [i];
  276. Mixin.Read (property.GetMethod);
  277. if (property.HasConstant)
  278. Mixin.Read (property.Constant);
  279. ReadCustomAttributes (property);
  280. }
  281. }
  282. void ReadEvents (TypeDefinition type)
  283. {
  284. var events = type.Events;
  285. for (int i = 0; i < events.Count; i++) {
  286. var @event = events [i];
  287. Mixin.Read (@event.AddMethod);
  288. ReadCustomAttributes (@event);
  289. }
  290. }
  291. public override void ReadSymbols (ModuleDefinition module)
  292. {
  293. if (module.symbol_reader == null)
  294. return;
  295. ReadTypesSymbols (module.Types, module.symbol_reader);
  296. }
  297. void ReadTypesSymbols (Collection<TypeDefinition> types, ISymbolReader symbol_reader)
  298. {
  299. for (int i = 0; i < types.Count; i++) {
  300. var type = types [i];
  301. if (type.HasNestedTypes)
  302. ReadTypesSymbols (type.NestedTypes, symbol_reader);
  303. if (type.HasMethods)
  304. ReadMethodsSymbols (type, symbol_reader);
  305. }
  306. }
  307. void ReadMethodsSymbols (TypeDefinition type, ISymbolReader symbol_reader)
  308. {
  309. var methods = type.Methods;
  310. for (int i = 0; i < methods.Count; i++) {
  311. var method = methods [i];
  312. if (method.HasBody && method.token.RID != 0 && method.debug_info == null)
  313. method.debug_info = symbol_reader.Read (method);
  314. }
  315. }
  316. }
  317. sealed class DeferredModuleReader : ModuleReader {
  318. public DeferredModuleReader (Image image)
  319. : base (image, ReadingMode.Deferred)
  320. {
  321. }
  322. protected override void ReadModule ()
  323. {
  324. this.module.Read (this.module, (_, reader) => ReadModuleManifest (reader));
  325. }
  326. public override void ReadSymbols (ModuleDefinition module)
  327. {
  328. }
  329. }
  330. sealed class MetadataReader : ByteBuffer {
  331. readonly internal Image image;
  332. readonly internal ModuleDefinition module;
  333. readonly internal MetadataSystem metadata;
  334. internal CodeReader code;
  335. internal IGenericContext context;
  336. readonly MetadataReader metadata_reader;
  337. public MetadataReader (ModuleDefinition module)
  338. : base (module.Image.TableHeap.data)
  339. {
  340. this.image = module.Image;
  341. this.module = module;
  342. this.metadata = module.MetadataSystem;
  343. this.code = new CodeReader (this);
  344. }
  345. public MetadataReader (Image image, ModuleDefinition module, MetadataReader metadata_reader)
  346. : base (image.TableHeap.data)
  347. {
  348. this.image = image;
  349. this.module = module;
  350. this.metadata = module.MetadataSystem;
  351. this.metadata_reader = metadata_reader;
  352. }
  353. int GetCodedIndexSize (CodedIndex index)
  354. {
  355. return image.GetCodedIndexSize (index);
  356. }
  357. uint ReadByIndexSize (int size)
  358. {
  359. if (size == 4)
  360. return ReadUInt32 ();
  361. else
  362. return ReadUInt16 ();
  363. }
  364. byte [] ReadBlob ()
  365. {
  366. var blob_heap = image.BlobHeap;
  367. if (blob_heap == null) {
  368. position += 2;
  369. return Empty<byte>.Array;
  370. }
  371. return blob_heap.Read (ReadBlobIndex ());
  372. }
  373. byte [] ReadBlob (uint signature)
  374. {
  375. var blob_heap = image.BlobHeap;
  376. if (blob_heap == null)
  377. return Empty<byte>.Array;
  378. return blob_heap.Read (signature);
  379. }
  380. uint ReadBlobIndex ()
  381. {
  382. var blob_heap = image.BlobHeap;
  383. return ReadByIndexSize (blob_heap != null ? blob_heap.IndexSize : 2);
  384. }
  385. void GetBlobView (uint signature, out byte [] blob, out int index, out int count)
  386. {
  387. var blob_heap = image.BlobHeap;
  388. if (blob_heap == null) {
  389. blob = null;
  390. index = count = 0;
  391. return;
  392. }
  393. blob_heap.GetView (signature, out blob, out index, out count);
  394. }
  395. string ReadString ()
  396. {
  397. return image.StringHeap.Read (ReadByIndexSize (image.StringHeap.IndexSize));
  398. }
  399. uint ReadStringIndex ()
  400. {
  401. return ReadByIndexSize (image.StringHeap.IndexSize);
  402. }
  403. Guid ReadGuid ()
  404. {
  405. return image.GuidHeap.Read (ReadByIndexSize (image.GuidHeap.IndexSize));
  406. }
  407. uint ReadTableIndex (Table table)
  408. {
  409. return ReadByIndexSize (image.GetTableIndexSize (table));
  410. }
  411. MetadataToken ReadMetadataToken (CodedIndex index)
  412. {
  413. return index.GetMetadataToken (ReadByIndexSize (GetCodedIndexSize (index)));
  414. }
  415. int MoveTo (Table table)
  416. {
  417. var info = image.TableHeap [table];
  418. if (info.Length != 0)
  419. this.position = (int) info.Offset;
  420. return (int) info.Length;
  421. }
  422. bool MoveTo (Table table, uint row)
  423. {
  424. var info = image.TableHeap [table];
  425. var length = info.Length;
  426. if (length == 0 || row > length)
  427. return false;
  428. this.position = (int) (info.Offset + (info.RowSize * (row - 1)));
  429. return true;
  430. }
  431. public AssemblyNameDefinition ReadAssemblyNameDefinition ()
  432. {
  433. if (MoveTo (Table.Assembly) == 0)
  434. return null;
  435. var name = new AssemblyNameDefinition ();
  436. name.HashAlgorithm = (AssemblyHashAlgorithm) ReadUInt32 ();
  437. PopulateVersionAndFlags (name);
  438. name.PublicKey = ReadBlob ();
  439. PopulateNameAndCulture (name);
  440. return name;
  441. }
  442. public ModuleDefinition Populate (ModuleDefinition module)
  443. {
  444. if (MoveTo (Table.Module) == 0)
  445. return module;
  446. Advance (2); // Generation
  447. module.Name = ReadString ();
  448. module.Mvid = ReadGuid ();
  449. return module;
  450. }
  451. void InitializeAssemblyReferences ()
  452. {
  453. if (metadata.AssemblyReferences != null)
  454. return;
  455. int length = MoveTo (Table.AssemblyRef);
  456. var references = metadata.AssemblyReferences = new AssemblyNameReference [length];
  457. for (uint i = 0; i < length; i++) {
  458. var reference = new AssemblyNameReference ();
  459. reference.token = new MetadataToken (TokenType.AssemblyRef, i + 1);
  460. PopulateVersionAndFlags (reference);
  461. var key_or_token = ReadBlob ();
  462. if (reference.HasPublicKey)
  463. reference.PublicKey = key_or_token;
  464. else
  465. reference.PublicKeyToken = key_or_token;
  466. PopulateNameAndCulture (reference);
  467. reference.Hash = ReadBlob ();
  468. references [i] = reference;
  469. }
  470. }
  471. public Collection<AssemblyNameReference> ReadAssemblyReferences ()
  472. {
  473. InitializeAssemblyReferences ();
  474. var references = new Collection<AssemblyNameReference> (metadata.AssemblyReferences);
  475. if (module.IsWindowsMetadata ())
  476. module.Projections.AddVirtualReferences (references);
  477. return references;
  478. }
  479. public MethodDefinition ReadEntryPoint ()
  480. {
  481. if (module.Image.EntryPointToken == 0)
  482. return null;
  483. var token = new MetadataToken (module.Image.EntryPointToken);
  484. return GetMethodDefinition (token.RID);
  485. }
  486. public Collection<ModuleDefinition> ReadModules ()
  487. {
  488. var modules = new Collection<ModuleDefinition> (1);
  489. modules.Add (this.module);
  490. int length = MoveTo (Table.File);
  491. for (uint i = 1; i <= length; i++) {
  492. var attributes = (FileAttributes) ReadUInt32 ();
  493. var name = ReadString ();
  494. ReadBlobIndex ();
  495. if (attributes != FileAttributes.ContainsMetaData)
  496. continue;
  497. var parameters = new ReaderParameters {
  498. ReadingMode = module.ReadingMode,
  499. SymbolReaderProvider = module.SymbolReaderProvider,
  500. AssemblyResolver = module.AssemblyResolver
  501. };
  502. modules.Add (ModuleDefinition.ReadModule (
  503. GetModuleFileName (name), parameters));
  504. }
  505. return modules;
  506. }
  507. string GetModuleFileName (string name)
  508. {
  509. if (module.FileName == null)
  510. throw new NotSupportedException ();
  511. var path = Path.GetDirectoryName (module.FileName);
  512. return Path.Combine (path, name);
  513. }
  514. void InitializeModuleReferences ()
  515. {
  516. if (metadata.ModuleReferences != null)
  517. return;
  518. int length = MoveTo (Table.ModuleRef);
  519. var references = metadata.ModuleReferences = new ModuleReference [length];
  520. for (uint i = 0; i < length; i++) {
  521. var reference = new ModuleReference (ReadString ());
  522. reference.token = new MetadataToken (TokenType.ModuleRef, i + 1);
  523. references [i] = reference;
  524. }
  525. }
  526. public Collection<ModuleReference> ReadModuleReferences ()
  527. {
  528. InitializeModuleReferences ();
  529. return new Collection<ModuleReference> (metadata.ModuleReferences);
  530. }
  531. public bool HasFileResource ()
  532. {
  533. int length = MoveTo (Table.File);
  534. if (length == 0)
  535. return false;
  536. for (uint i = 1; i <= length; i++)
  537. if (ReadFileRecord (i).Col1 == FileAttributes.ContainsNoMetaData)
  538. return true;
  539. return false;
  540. }
  541. public Collection<Resource> ReadResources ()
  542. {
  543. int length = MoveTo (Table.ManifestResource);
  544. var resources = new Collection<Resource> (length);
  545. for (int i = 1; i <= length; i++) {
  546. var offset = ReadUInt32 ();
  547. var flags = (ManifestResourceAttributes) ReadUInt32 ();
  548. var name = ReadString ();
  549. var implementation = ReadMetadataToken (CodedIndex.Implementation);
  550. Resource resource;
  551. if (implementation.RID == 0) {
  552. resource = new EmbeddedResource (name, flags, offset, this);
  553. } else if (implementation.TokenType == TokenType.AssemblyRef) {
  554. resource = new AssemblyLinkedResource (name, flags) {
  555. Assembly = (AssemblyNameReference) GetTypeReferenceScope (implementation),
  556. };
  557. } else if (implementation.TokenType == TokenType.File) {
  558. var file_record = ReadFileRecord (implementation.RID);
  559. resource = new LinkedResource (name, flags) {
  560. File = file_record.Col2,
  561. hash = ReadBlob (file_record.Col3)
  562. };
  563. } else
  564. continue;
  565. resources.Add (resource);
  566. }
  567. return resources;
  568. }
  569. Row<FileAttributes, string, uint> ReadFileRecord (uint rid)
  570. {
  571. var position = this.position;
  572. if (!MoveTo (Table.File, rid))
  573. throw new ArgumentException ();
  574. var record = new Row<FileAttributes, string, uint> (
  575. (FileAttributes) ReadUInt32 (),
  576. ReadString (),
  577. ReadBlobIndex ());
  578. this.position = position;
  579. return record;
  580. }
  581. public byte [] GetManagedResource (uint offset)
  582. {
  583. return image.GetReaderAt (image.Resources.VirtualAddress, offset, (o, reader) => {
  584. reader.Advance ((int) o);
  585. return reader.ReadBytes (reader.ReadInt32 ());
  586. }) ?? Empty<byte>.Array;
  587. }
  588. void PopulateVersionAndFlags (AssemblyNameReference name)
  589. {
  590. name.Version = new Version (
  591. ReadUInt16 (),
  592. ReadUInt16 (),
  593. ReadUInt16 (),
  594. ReadUInt16 ());
  595. name.Attributes = (AssemblyAttributes) ReadUInt32 ();
  596. }
  597. void PopulateNameAndCulture (AssemblyNameReference name)
  598. {
  599. name.Name = ReadString ();
  600. name.Culture = ReadString ();
  601. }
  602. public TypeDefinitionCollection ReadTypes ()
  603. {
  604. InitializeTypeDefinitions ();
  605. var mtypes = metadata.Types;
  606. var type_count = mtypes.Length - metadata.NestedTypes.Count;
  607. var types = new TypeDefinitionCollection (module, type_count);
  608. for (int i = 0; i < mtypes.Length; i++) {
  609. var type = mtypes [i];
  610. if (IsNested (type.Attributes))
  611. continue;
  612. types.Add (type);
  613. }
  614. if (image.HasTable (Table.MethodPtr) || image.HasTable (Table.FieldPtr))
  615. CompleteTypes ();
  616. return types;
  617. }
  618. void CompleteTypes ()
  619. {
  620. var types = metadata.Types;
  621. for (int i = 0; i < types.Length; i++) {
  622. var type = types [i];
  623. Mixin.Read (type.Fields);
  624. Mixin.Read (type.Methods);
  625. }
  626. }
  627. void InitializeTypeDefinitions ()
  628. {
  629. if (metadata.Types != null)
  630. return;
  631. InitializeNestedTypes ();
  632. InitializeFields ();
  633. InitializeMethods ();
  634. int length = MoveTo (Table.TypeDef);
  635. var types = metadata.Types = new TypeDefinition [length];
  636. for (uint i = 0; i < length; i++) {
  637. if (types [i] != null)
  638. continue;
  639. types [i] = ReadType (i + 1);
  640. }
  641. if (module.IsWindowsMetadata ()) {
  642. for (uint i = 0; i < length; i++) {
  643. WindowsRuntimeProjections.Project (types [i]);
  644. }
  645. }
  646. }
  647. static bool IsNested (TypeAttributes attributes)
  648. {
  649. switch (attributes & TypeAttributes.VisibilityMask) {
  650. case TypeAttributes.NestedAssembly:
  651. case TypeAttributes.NestedFamANDAssem:
  652. case TypeAttributes.NestedFamily:
  653. case TypeAttributes.NestedFamORAssem:
  654. case TypeAttributes.NestedPrivate:
  655. case TypeAttributes.NestedPublic:
  656. return true;
  657. default:
  658. return false;
  659. }
  660. }
  661. public bool HasNestedTypes (TypeDefinition type)
  662. {
  663. Collection<uint> mapping;
  664. InitializeNestedTypes ();
  665. if (!metadata.TryGetNestedTypeMapping (type, out mapping))
  666. return false;
  667. return mapping.Count > 0;
  668. }
  669. public Collection<TypeDefinition> ReadNestedTypes (TypeDefinition type)
  670. {
  671. InitializeNestedTypes ();
  672. Collection<uint> mapping;
  673. if (!metadata.TryGetNestedTypeMapping (type, out mapping))
  674. return new MemberDefinitionCollection<TypeDefinition> (type);
  675. var nested_types = new MemberDefinitionCollection<TypeDefinition> (type, mapping.Count);
  676. for (int i = 0; i < mapping.Count; i++) {
  677. var nested_type = GetTypeDefinition (mapping [i]);
  678. if (nested_type != null)
  679. nested_types.Add (nested_type);
  680. }
  681. metadata.RemoveNestedTypeMapping (type);
  682. return nested_types;
  683. }
  684. void InitializeNestedTypes ()
  685. {
  686. if (metadata.NestedTypes != null)
  687. return;
  688. var length = MoveTo (Table.NestedClass);
  689. metadata.NestedTypes = new Dictionary<uint, Collection<uint>> (length);
  690. metadata.ReverseNestedTypes = new Dictionary<uint, uint> (length);
  691. if (length == 0)
  692. return;
  693. for (int i = 1; i <= length; i++) {
  694. var nested = ReadTableIndex (Table.TypeDef);
  695. var declaring = ReadTableIndex (Table.TypeDef);
  696. AddNestedMapping (declaring, nested);
  697. }
  698. }
  699. void AddNestedMapping (uint declaring, uint nested)
  700. {
  701. metadata.SetNestedTypeMapping (declaring, AddMapping (metadata.NestedTypes, declaring, nested));
  702. metadata.SetReverseNestedTypeMapping (nested, declaring);
  703. }
  704. static Collection<TValue> AddMapping<TKey, TValue> (Dictionary<TKey, Collection<TValue>> cache, TKey key, TValue value)
  705. {
  706. Collection<TValue> mapped;
  707. if (!cache.TryGetValue (key, out mapped)) {
  708. mapped = new Collection<TValue> ();
  709. }
  710. mapped.Add (value);
  711. return mapped;
  712. }
  713. TypeDefinition ReadType (uint rid)
  714. {
  715. if (!MoveTo (Table.TypeDef, rid))
  716. return null;
  717. var attributes = (TypeAttributes) ReadUInt32 ();
  718. var name = ReadString ();
  719. var @namespace = ReadString ();
  720. var type = new TypeDefinition (@namespace, name, attributes);
  721. type.token = new MetadataToken (TokenType.TypeDef, rid);
  722. type.scope = module;
  723. type.module = module;
  724. metadata.AddTypeDefinition (type);
  725. this.context = type;
  726. type.BaseType = GetTypeDefOrRef (ReadMetadataToken (CodedIndex.TypeDefOrRef));
  727. type.fields_range = ReadListRange (rid, Table.TypeDef, Table.Field);
  728. type.methods_range = ReadListRange (rid, Table.TypeDef, Table.Method);
  729. if (IsNested (attributes))
  730. type.DeclaringType = GetNestedTypeDeclaringType (type);
  731. return type;
  732. }
  733. TypeDefinition GetNestedTypeDeclaringType (TypeDefinition type)
  734. {
  735. uint declaring_rid;
  736. if (!metadata.TryGetReverseNestedTypeMapping (type, out declaring_rid))
  737. return null;
  738. metadata.RemoveReverseNestedTypeMapping (type);
  739. return GetTypeDefinition (declaring_rid);
  740. }
  741. Range ReadListRange (uint current_index, Table current, Table target)
  742. {
  743. var list = new Range ();
  744. var start = ReadTableIndex (target);
  745. if (start == 0)
  746. return list;
  747. uint next_index;
  748. var current_table = image.TableHeap [current];
  749. if (current_index == current_table.Length)
  750. next_index = image.TableHeap [target].Length + 1;
  751. else {
  752. var position = this.position;
  753. this.position += (int) (current_table.RowSize - image.GetTableIndexSize (target));
  754. next_index = ReadTableIndex (target);
  755. this.position = position;
  756. }
  757. list.Start = start;
  758. list.Length = next_index - start;
  759. return list;
  760. }
  761. public Row<short, int> ReadTypeLayout (TypeDefinition type)
  762. {
  763. InitializeTypeLayouts ();
  764. Row<ushort, uint> class_layout;
  765. var rid = type.token.RID;
  766. if (!metadata.ClassLayouts.TryGetValue (rid, out class_layout))
  767. return new Row<short, int> (Mixin.NoDataMarker, Mixin.NoDataMarker);
  768. type.PackingSize = (short) class_layout.Col1;
  769. type.ClassSize = (int) class_layout.Col2;
  770. metadata.ClassLayouts.Remove (rid);
  771. return new Row<short, int> ((short) class_layout.Col1, (int) class_layout.Col2);
  772. }
  773. void InitializeTypeLayouts ()
  774. {
  775. if (metadata.ClassLayouts != null)
  776. return;
  777. int length = MoveTo (Table.ClassLayout);
  778. var class_layouts = metadata.ClassLayouts = new Dictionary<uint, Row<ushort, uint>> (length);
  779. for (uint i = 0; i < length; i++) {
  780. var packing_size = ReadUInt16 ();
  781. var class_size = ReadUInt32 ();
  782. var parent = ReadTableIndex (Table.TypeDef);
  783. class_layouts.Add (parent, new Row<ushort, uint> (packing_size, class_size));
  784. }
  785. }
  786. public TypeReference GetTypeDefOrRef (MetadataToken token)
  787. {
  788. return (TypeReference) LookupToken (token);
  789. }
  790. public TypeDefinition GetTypeDefinition (uint rid)
  791. {
  792. InitializeTypeDefinitions ();
  793. var type = metadata.GetTypeDefinition (rid);
  794. if (type != null)
  795. return type;
  796. type = ReadTypeDefinition (rid);
  797. if (module.IsWindowsMetadata ())
  798. WindowsRuntimeProjections.Project (type);
  799. return type;
  800. }
  801. TypeDefinition ReadTypeDefinition (uint rid)
  802. {
  803. if (!MoveTo (Table.TypeDef, rid))
  804. return null;
  805. return ReadType (rid);
  806. }
  807. void InitializeTypeReferences ()
  808. {
  809. if (metadata.TypeReferences != null)
  810. return;
  811. metadata.TypeReferences = new TypeReference [image.GetTableLength (Table.TypeRef)];
  812. }
  813. public TypeReference GetTypeReference (string scope, string full_name)
  814. {
  815. InitializeTypeReferences ();
  816. var length = metadata.TypeReferences.Length;
  817. for (uint i = 1; i <= length; i++) {
  818. var type = GetTypeReference (i);
  819. if (type.FullName != full_name)
  820. continue;
  821. if (string.IsNullOrEmpty (scope))
  822. return type;
  823. if (type.Scope.Name == scope)
  824. return type;
  825. }
  826. return null;
  827. }
  828. TypeReference GetTypeReference (uint rid)
  829. {
  830. InitializeTypeReferences ();
  831. var type = metadata.GetTypeReference (rid);
  832. if (type != null)
  833. return type;
  834. return ReadTypeReference (rid);
  835. }
  836. TypeReference ReadTypeReference (uint rid)
  837. {
  838. if (!MoveTo (Table.TypeRef, rid))
  839. return null;
  840. TypeReference declaring_type = null;
  841. IMetadataScope scope;
  842. var scope_token = ReadMetadataToken (CodedIndex.ResolutionScope);
  843. var name = ReadString ();
  844. var @namespace = ReadString ();
  845. var type = new TypeReference (
  846. @namespace,
  847. name,
  848. module,
  849. null);
  850. type.token = new MetadataToken (TokenType.TypeRef, rid);
  851. metadata.AddTypeReference (type);
  852. if (scope_token.TokenType == TokenType.TypeRef) {
  853. if (scope_token.RID != rid) {
  854. declaring_type = GetTypeDefOrRef (scope_token);
  855. scope = declaring_type != null
  856. ? declaring_type.Scope
  857. : module;
  858. } else // obfuscated typeref row pointing to self
  859. scope = module;
  860. } else
  861. scope = GetTypeReferenceScope (scope_token);
  862. type.scope = scope;
  863. type.DeclaringType = declaring_type;
  864. MetadataSystem.TryProcessPrimitiveTypeReference (type);
  865. if (type.Module.IsWindowsMetadata ())
  866. WindowsRuntimeProjections.Project (type);
  867. return type;
  868. }
  869. IMetadataScope GetTypeReferenceScope (MetadataToken scope)
  870. {
  871. if (scope.TokenType == TokenType.Module)
  872. return module;
  873. IMetadataScope[] scopes;
  874. switch (scope.TokenType) {
  875. case TokenType.AssemblyRef:
  876. InitializeAssemblyReferences ();
  877. scopes = metadata.AssemblyReferences;
  878. break;
  879. case TokenType.ModuleRef:
  880. InitializeModuleReferences ();
  881. scopes = metadata.ModuleReferences;
  882. break;
  883. default:
  884. throw new NotSupportedException ();
  885. }
  886. var index = scope.RID - 1;
  887. if (index < 0 || index >= scopes.Length)
  888. return null;
  889. return scopes [index];
  890. }
  891. public IEnumerable<TypeReference> GetTypeReferences ()
  892. {
  893. InitializeTypeReferences ();
  894. var length = image.GetTableLength (Table.TypeRef);
  895. var type_references = new TypeReference [length];
  896. for (uint i = 1; i <= length; i++)
  897. type_references [i - 1] = GetTypeReference (i);
  898. return type_references;
  899. }
  900. TypeReference GetTypeSpecification (uint rid)
  901. {
  902. if (!MoveTo (Table.TypeSpec, rid))
  903. return null;
  904. var reader = ReadSignature (ReadBlobIndex ());
  905. var type = reader.ReadTypeSignature ();
  906. if (type.token.RID == 0)
  907. type.token = new MetadataToken (TokenType.TypeSpec, rid);
  908. return type;
  909. }
  910. SignatureReader ReadSignature (uint signature)
  911. {
  912. return new SignatureReader (signature, this);
  913. }
  914. public bool HasInterfaces (TypeDefinition type)
  915. {
  916. InitializeInterfaces ();
  917. Collection<Row<uint, MetadataToken>> mapping;
  918. return metadata.TryGetInterfaceMapping (type, out mapping);
  919. }
  920. public InterfaceImplementationCollection ReadInterfaces (TypeDefinition type)
  921. {
  922. InitializeInterfaces ();
  923. Collection<Row<uint, MetadataToken>> mapping;
  924. if (!metadata.TryGetInterfaceMapping (type, out mapping))
  925. return new InterfaceImplementationCollection (type);
  926. var interfaces = new InterfaceImplementationCollection (type, mapping.Count);
  927. this.context = type;
  928. for (int i = 0; i < mapping.Count; i++) {
  929. interfaces.Add (
  930. new InterfaceImplementation (
  931. GetTypeDefOrRef (mapping [i].Col2),
  932. new MetadataToken(TokenType.InterfaceImpl, mapping [i].Col1)));
  933. }
  934. metadata.RemoveInterfaceMapping (type);
  935. return interfaces;
  936. }
  937. void InitializeInterfaces ()
  938. {
  939. if (metadata.Interfaces != null)
  940. return;
  941. int length = MoveTo (Table.InterfaceImpl);
  942. metadata.Interfaces = new Dictionary<uint, Collection<Row<uint, MetadataToken>>> (length);
  943. for (uint i = 1; i <= length; i++) {
  944. var type = ReadTableIndex (Table.TypeDef);
  945. var @interface = ReadMetadataToken (CodedIndex.TypeDefOrRef);
  946. AddInterfaceMapping (type, new Row<uint, MetadataToken> (i, @interface));
  947. }
  948. }
  949. void AddInterfaceMapping (uint type, Row<uint, MetadataToken> @interface)
  950. {
  951. metadata.SetInterfaceMapping (type, AddMapping (metadata.Interfaces, type, @interface));
  952. }
  953. public Collection<FieldDefinition> ReadFields (TypeDefinition type)
  954. {
  955. var fields_range = type.fields_range;
  956. if (fields_range.Length == 0)
  957. return new MemberDefinitionCollection<FieldDefinition> (type);
  958. var fields = new MemberDefinitionCollection<FieldDefinition> (type, (int) fields_range.Length);
  959. this.context = type;
  960. if (!MoveTo (Table.FieldPtr, fields_range.Start)) {
  961. if (!MoveTo (Table.Field, fields_range.Start))
  962. return fields;
  963. for (uint i = 0; i < fields_range.Length; i++)
  964. ReadField (fields_range.Start + i, fields);
  965. } else
  966. ReadPointers (Table.FieldPtr, Table.Field, fields_range, fields, ReadField);
  967. return fields;
  968. }
  969. void ReadField (uint field_rid, Collection<FieldDefinition> fields)
  970. {
  971. var attributes = (FieldAttributes) ReadUInt16 ();
  972. var name = ReadString ();
  973. var signature = ReadBlobIndex ();
  974. var field = new FieldDefinition (name, attributes, ReadFieldType (signature));
  975. field.token = new MetadataToken (TokenType.Field, field_rid);
  976. metadata.AddFieldDefinition (field);
  977. if (IsDeleted (field))
  978. return;
  979. fields.Add (field);
  980. if (module.IsWindowsMetadata ())
  981. WindowsRuntimeProjections.Project (field);
  982. }
  983. void InitializeFields ()
  984. {
  985. if (metadata.Fields != null)
  986. return;
  987. metadata.Fields = new FieldDefinition [image.GetTableLength (Table.Field)];
  988. }
  989. TypeReference ReadFieldType (uint signature)
  990. {
  991. var reader = ReadSignature (signature);
  992. const byte field_sig = 0x6;
  993. if (reader.ReadByte () != field_sig)
  994. throw new NotSupportedException ();
  995. return reader.ReadTypeSignature ();
  996. }
  997. public int ReadFieldRVA (FieldDefinition field)
  998. {
  999. InitializeFieldRVAs ();
  1000. var rid = field.token.RID;
  1001. RVA rva;
  1002. if (!metadata.FieldRVAs.TryGetValue (rid, out rva))
  1003. return 0;
  1004. var size = GetFieldTypeSize (field.FieldType);
  1005. if (size == 0 || rva == 0)
  1006. return 0;
  1007. metadata.FieldRVAs.Remove (rid);
  1008. field.InitialValue = GetFieldInitializeValue (size, rva);
  1009. return (int) rva;
  1010. }
  1011. byte [] GetFieldInitializeValue (int size, RVA rva)
  1012. {
  1013. return image.GetReaderAt (rva, size, (s, reader) => reader.ReadBytes (s)) ?? Empty<byte>.Array;
  1014. }
  1015. static int GetFieldTypeSize (TypeReference type)
  1016. {
  1017. int size = 0;
  1018. switch (type.etype) {
  1019. case ElementType.Boolean:
  1020. case ElementType.U1:
  1021. case ElementType.I1:
  1022. size = 1;
  1023. break;
  1024. case ElementType.U2:
  1025. case ElementType.I2:
  1026. case ElementType.Char:
  1027. size = 2;
  1028. break;
  1029. case ElementType.U4:
  1030. case ElementType.I4:
  1031. case ElementType.R4:
  1032. size = 4;
  1033. break;
  1034. case ElementType.U8:
  1035. case ElementType.I8:
  1036. case ElementType.R8:
  1037. size = 8;
  1038. break;
  1039. case ElementType.Ptr:
  1040. case ElementType.FnPtr:
  1041. size = IntPtr.Size;
  1042. break;
  1043. case ElementType.CModOpt:
  1044. case ElementType.CModReqD:
  1045. return GetFieldTypeSize (((IModifierType) type).ElementType);
  1046. default:
  1047. var field_type = type.Resolve ();
  1048. if (field_type != null && field_type.HasLayoutInfo)
  1049. size = field_type.ClassSize;
  1050. break;
  1051. }
  1052. return size;
  1053. }
  1054. void InitializeFieldRVAs ()
  1055. {
  1056. if (metadata.FieldRVAs != null)
  1057. return;
  1058. int length = MoveTo (Table.FieldRVA);
  1059. var field_rvas = metadata.FieldRVAs = new Dictionary<uint, uint> (length);
  1060. for (int i = 0; i < length; i++) {
  1061. var rva = ReadUInt32 ();
  1062. var field = ReadTableIndex (Table.Field);
  1063. field_rvas.Add (field, rva);
  1064. }
  1065. }
  1066. public int ReadFieldLayout (FieldDefinition field)
  1067. {
  1068. InitializeFieldLayouts ();
  1069. var rid = field.token.RID;
  1070. uint offset;
  1071. if (!metadata.FieldLayouts.TryGetValue (rid, out offset))
  1072. return Mixin.NoDataMarker;
  1073. metadata.FieldLayouts.Remove (rid);
  1074. return (int) offset;
  1075. }
  1076. void InitializeFieldLayouts ()
  1077. {
  1078. if (metadata.FieldLayouts != null)
  1079. return;
  1080. int length = MoveTo (Table.FieldLayout);
  1081. var field_layouts = metadata.FieldLayouts = new Dictionary<uint, uint> (length);
  1082. for (int i = 0; i < length; i++) {
  1083. var offset = ReadUInt32 ();
  1084. var field = ReadTableIndex (Table.Field);
  1085. field_layouts.Add (field, offset);
  1086. }
  1087. }
  1088. public bool HasEvents (TypeDefinition type)
  1089. {
  1090. InitializeEvents ();
  1091. Range range;
  1092. if (!metadata.TryGetEventsRange (type, out range))
  1093. return false;
  1094. return range.Length > 0;
  1095. }
  1096. public Collection<EventDefinition> ReadEvents (TypeDefinition type)
  1097. {
  1098. InitializeEvents ();
  1099. Range range;
  1100. if (!metadata.TryGetEventsRange (type, out range))
  1101. return new MemberDefinitionCollection<EventDefinition> (type);
  1102. var events = new MemberDefinitionCollection<EventDefinition> (type, (int) range.Length);
  1103. metadata.RemoveEventsRange (type);
  1104. if (range.Length == 0)
  1105. return events;
  1106. this.context = type;
  1107. if (!MoveTo (Table.EventPtr, range.Start)) {
  1108. if (!MoveTo (Table.Event, range.Start))
  1109. return events;
  1110. for (uint i = 0; i < range.Length; i++)
  1111. ReadEvent (range.Start + i, events);
  1112. } else
  1113. ReadPointers (Table.EventPtr, Table.Event, range, events, ReadEvent);
  1114. return events;
  1115. }
  1116. void ReadEvent (uint event_rid, Collection<EventDefinition> events)
  1117. {
  1118. var attributes = (EventAttributes) ReadUInt16 ();
  1119. var name = ReadString ();
  1120. var event_type = GetTypeDefOrRef (ReadMetadataToken (CodedIndex.TypeDefOrRef));
  1121. var @event = new EventDefinition (name, attributes, event_type);
  1122. @event.token = new MetadataToken (TokenType.Event, event_rid);
  1123. if (IsDeleted (@event))
  1124. return;
  1125. events.Add (@event);
  1126. }
  1127. void InitializeEvents ()
  1128. {
  1129. if (metadata.Events != null)
  1130. return;
  1131. int length = MoveTo (Table.EventMap);
  1132. metadata.Events = new Dictionary<uint, Range> (length);
  1133. for (uint i = 1; i <= length; i++) {
  1134. var type_rid = ReadTableIndex (Table.TypeDef);
  1135. Range events_range = ReadListRange (i, Table.EventMap, Table.Event);
  1136. metadata.AddEventsRange (type_rid, events_range);
  1137. }
  1138. }
  1139. public bool HasProperties (TypeDefinition type)
  1140. {
  1141. InitializeProperties ();
  1142. Range range;
  1143. if (!metadata.TryGetPropertiesRange (type, out range))
  1144. return false;
  1145. return range.Length > 0;
  1146. }
  1147. public Collection<PropertyDefinition> ReadProperties (TypeDefinition type)
  1148. {
  1149. InitializeProperties ();
  1150. Range range;
  1151. if (!metadata.TryGetPropertiesRange (type, out range))
  1152. return new MemberDefinitionCollection<PropertyDefinition> (type);
  1153. metadata.RemovePropertiesRange (type);
  1154. var properties = new MemberDefinitionCollection<PropertyDefinition> (type, (int) range.Length);
  1155. if (range.Length == 0)
  1156. return properties;
  1157. this.context = type;
  1158. if (!MoveTo (Table.PropertyPtr, range.Start)) {
  1159. if (!MoveTo (Table.Property, range.Start))
  1160. return properties;
  1161. for (uint i = 0; i < range.Length; i++)
  1162. ReadProperty (range.Start + i, properties);
  1163. } else
  1164. ReadPointers (Table.PropertyPtr, Table.Property, range, properties, ReadProperty);
  1165. return properties;
  1166. }
  1167. void ReadProperty (uint property_rid, Collection<PropertyDefinition> properties)
  1168. {
  1169. var attributes = (PropertyAttributes) ReadUInt16 ();
  1170. var name = ReadString ();
  1171. var signature = ReadBlobIndex ();
  1172. var reader = ReadSignature (signature);
  1173. const byte property_signature = 0x8;
  1174. var calling_convention = reader.ReadByte ();
  1175. if ((calling_convention & property_signature) == 0)
  1176. throw new NotSupportedException ();
  1177. var has_this = (calling_convention & 0x20) != 0;
  1178. reader.ReadCompressedUInt32 (); // count
  1179. var property = new PropertyDefinition (name, attributes, reader.ReadTypeSignature ());
  1180. property.HasThis = has_this;
  1181. property.token = new MetadataToken (TokenType.Property, property_rid);
  1182. if (IsDeleted (property))
  1183. return;
  1184. properties.Add (property);
  1185. }
  1186. void InitializeProperties ()
  1187. {
  1188. if (metadata.Properties != null)
  1189. return;
  1190. int length = MoveTo (Table.PropertyMap);
  1191. metadata.Properties = new Dictionary<uint, Range> (length);
  1192. for (uint i = 1; i <= length; i++) {
  1193. var type_rid = ReadTableIndex (Table.TypeDef);
  1194. var properties_range = ReadListRange (i, Table.PropertyMap, Table.Property);
  1195. metadata.AddPropertiesRange (type_rid, properties_range);
  1196. }
  1197. }
  1198. MethodSemanticsAttributes ReadMethodSemantics (MethodDefinition method)
  1199. {
  1200. InitializeMethodSemantics ();
  1201. Row<MethodSemanticsAttributes, MetadataToken> row;
  1202. if (!metadata.Semantics.TryGetValue (method.token.RID, out row))
  1203. return MethodSemanticsAttributes.None;
  1204. var type = method.DeclaringType;
  1205. switch (row.Col1) {
  1206. case MethodSemanticsAttributes.AddOn:
  1207. GetEvent (type, row.Col2).add_method = method;
  1208. break;
  1209. case MethodSemanticsAttributes.Fire:
  1210. GetEvent (type, row.Col2).invoke_method = method;
  1211. break;
  1212. case MethodSemanticsAttributes.RemoveOn:
  1213. GetEvent (type, row.Col2).remove_method = method;
  1214. break;
  1215. case MethodSemanticsAttributes.Getter:
  1216. GetProperty (type, row.Col2).get_method = method;
  1217. break;
  1218. case MethodSemanticsAttributes.Setter:
  1219. GetProperty (type, row.Col2).set_method = method;
  1220. break;
  1221. case MethodSemanticsAttributes.Other:
  1222. switch (row.Col2.TokenType) {
  1223. case TokenType.Event: {
  1224. var @event = GetEvent (type, row.Col2);
  1225. if (@event.other_methods == null)
  1226. @event.other_methods = new Collection<MethodDefinition> ();
  1227. @event.other_methods.Add (method);
  1228. break;
  1229. }
  1230. case TokenType.Property: {
  1231. var property = GetProperty (type, row.Col2);
  1232. if (property.other_methods == null)
  1233. property.other_methods = new Collection<MethodDefinition> ();
  1234. property.other_methods.Add (method);
  1235. break;
  1236. }
  1237. default:
  1238. throw new NotSupportedException ();
  1239. }
  1240. break;
  1241. default:
  1242. throw new NotSupportedException ();
  1243. }
  1244. metadata.Semantics.Remove (method.token.RID);
  1245. return row.Col1;
  1246. }
  1247. static EventDefinition GetEvent (TypeDefinition type, MetadataToken token)
  1248. {
  1249. if (token.TokenType != TokenType.Event)
  1250. throw new ArgumentException ();
  1251. return GetMember (type.Events, token);
  1252. }
  1253. static PropertyDefinition GetProperty (TypeDefinition type, MetadataToken token)
  1254. {
  1255. if (token.TokenType != TokenType.Property)
  1256. throw new ArgumentException ();
  1257. return GetMember (type.Properties, token);
  1258. }
  1259. static TMember GetMember<TMember> (Collection<TMember> members, MetadataToken token) where TMember : IMemberDefinition
  1260. {
  1261. for (int i = 0; i < members.Count; i++) {
  1262. var member = members [i];
  1263. if (member.MetadataToken == token)
  1264. return member;
  1265. }
  1266. throw new ArgumentException ();
  1267. }
  1268. void InitializeMethodSemantics ()
  1269. {
  1270. if (metadata.Semantics != null)
  1271. return;
  1272. int length = MoveTo (Table.MethodSemantics);
  1273. var semantics = metadata.Semantics = new Dictionary<uint, Row<MethodSemanticsAttributes, MetadataToken>> (0);
  1274. for (uint i = 0; i < length; i++) {
  1275. var attributes = (MethodSemanticsAttributes) ReadUInt16 ();
  1276. var method_rid = ReadTableIndex (Table.Method);
  1277. var association = ReadMetadataToken (CodedIndex.HasSemantics);
  1278. semantics [method_rid] = new Row<MethodSemanticsAttributes, MetadataToken> (attributes, association);
  1279. }
  1280. }
  1281. public void ReadMethods (PropertyDefinition property)
  1282. {
  1283. ReadAllSemantics (property.DeclaringType);
  1284. }
  1285. public void ReadMethods (EventDefinition @event)
  1286. {
  1287. ReadAllSemantics (@event.DeclaringType);
  1288. }
  1289. public void ReadAllSemantics (MethodDefinition method)
  1290. {
  1291. ReadAllSemantics (method.DeclaringType);
  1292. }
  1293. void ReadAllSemantics (TypeDefinition type)
  1294. {
  1295. var methods = type.Methods;
  1296. for (int i = 0; i < methods.Count; i++) {
  1297. var method = methods [i];
  1298. if (method.sem_attrs_ready)
  1299. continue;
  1300. method.sem_attrs = ReadMethodSemantics (method);
  1301. method.sem_attrs_ready = true;
  1302. }
  1303. }
  1304. public Collection<MethodDefinition> ReadMethods (TypeDefinition type)
  1305. {
  1306. var methods_range = type.methods_range;
  1307. if (methods_range.Length == 0)
  1308. return new MemberDefinitionCollection<MethodDefinition> (type);
  1309. var methods = new MemberDefinitionCollection<MethodDefinition> (type, (int) methods_range.Length);
  1310. if (!MoveTo (Table.MethodPtr, methods_range.Start)) {
  1311. if (!MoveTo (Table.Method, methods_range.Start))
  1312. return methods;
  1313. for (uint i = 0; i < methods_range.Length; i++)
  1314. ReadMethod (methods_range.Start + i, methods);
  1315. } else
  1316. ReadPointers (Table.MethodPtr, Table.Method, methods_range, methods, ReadMethod);
  1317. return methods;
  1318. }
  1319. void ReadPointers<TMember> (Table ptr, Table table, Range range, Collection<TMember> members, Action<uint, Collection<TMember>> reader)
  1320. where TMember : IMemberDefinition
  1321. {
  1322. for (uint i = 0; i < range.Length; i++) {
  1323. MoveTo (ptr, range.Start + i);
  1324. var rid = ReadTableIndex (table);
  1325. MoveTo (table, rid);
  1326. reader (rid, members);
  1327. }
  1328. }
  1329. static bool IsDeleted (IMemberDefinition member)
  1330. {
  1331. return member.IsSpecialName && member.Name == "_Deleted";
  1332. }
  1333. void InitializeMethods ()
  1334. {
  1335. if (metadata.Methods != null)
  1336. return;
  1337. metadata.Methods = new MethodDefinition [image.GetTableLength (Table.Method)];
  1338. }
  1339. void ReadMethod (uint method_rid, Collection<MethodDefinition> methods)
  1340. {
  1341. var method = new MethodDefinition ();
  1342. method.rva = ReadUInt32 ();
  1343. method.ImplAttributes = (MethodImplAttributes) ReadUInt16 ();
  1344. method.Attributes = (MethodAttributes) ReadUInt16 ();
  1345. method.Name = ReadString ();
  1346. method.token = new MetadataToken (TokenType.Method, method_rid);
  1347. if (IsDeleted (method))
  1348. return;
  1349. methods.Add (method); // attach method
  1350. var signature = ReadBlobIndex ();
  1351. var param_range = ReadListRange (method_rid, Table.Method, Table.Param);
  1352. this.context = method;
  1353. ReadMethodSignature (signature, method);
  1354. metadata.AddMethodDefinition (method);
  1355. if (param_range.Length != 0) {
  1356. var position = base.position;
  1357. ReadParameters (method, param_range);
  1358. base.position = position;
  1359. }
  1360. if (module.IsWindowsMetadata ())
  1361. WindowsRuntimeProjections.Project (method);
  1362. }
  1363. void ReadParameters (MethodDefinition method, Range param_range)
  1364. {
  1365. if (!MoveTo (Table.ParamPtr, param_range.Start)) {
  1366. if (!MoveTo (Table.Param, param_range.Start))
  1367. return;
  1368. for (uint i = 0; i < param_range.Length; i++)
  1369. ReadParameter (param_range.Start + i, method);
  1370. } else
  1371. ReadParameterPointers (method, param_range);
  1372. }
  1373. void ReadParameterPointers (MethodDefinition method, Range range)
  1374. {
  1375. for (uint i = 0; i < range.Length; i++) {
  1376. MoveTo (Table.ParamPtr, range.Start + i);
  1377. var rid = ReadTableIndex (Table.Param);
  1378. MoveTo (Table.Param, rid);
  1379. ReadParameter (rid, method);
  1380. }
  1381. }
  1382. void ReadParameter (uint param_rid, MethodDefinition method)
  1383. {
  1384. var attributes = (ParameterAttributes) ReadUInt16 ();
  1385. var sequence = ReadUInt16 ();
  1386. var name = ReadString ();
  1387. var parameter = sequence == 0
  1388. ? method.MethodReturnType.Parameter
  1389. : method.Parameters [sequence - 1];
  1390. parameter.token = new MetadataToken (TokenType.Param, param_rid);
  1391. parameter.Name = name;
  1392. parameter.Attributes = attributes;
  1393. }
  1394. void ReadMethodSignature (uint signature, IMethodSignature method)
  1395. {
  1396. var reader = ReadSignature (signature);
  1397. reader.ReadMethodSignature (method);
  1398. }
  1399. public PInvokeInfo ReadPInvokeInfo (MethodDefinition method)
  1400. {
  1401. InitializePInvokes ();
  1402. Row<PInvokeAttributes, uint, uint> row;
  1403. var rid = method.token.RID;
  1404. if (!metadata.PInvokes.TryGetValue (rid, out row))
  1405. return null;
  1406. metadata.PInvokes.Remove (rid);
  1407. return new PInvokeInfo (
  1408. row.Col1,
  1409. image.StringHeap.Read (row.Col2),
  1410. module.ModuleReferences [(int) row.Col3 - 1]);
  1411. }
  1412. void InitializePInvokes ()
  1413. {
  1414. if (metadata.PInvokes != null)
  1415. return;
  1416. int length = MoveTo (Table.ImplMap);
  1417. var pinvokes = metadata.PInvokes = new Dictionary<uint, Row<PInvokeAttributes, uint, uint>> (length);
  1418. for (int i = 1; i <= length; i++) {
  1419. var attributes = (PInvokeAttributes) ReadUInt16 ();
  1420. var method = ReadMetadataToken (CodedIndex.MemberForwarded);
  1421. var name = ReadStringIndex ();
  1422. var scope = ReadTableIndex (Table.File);
  1423. if (method.TokenType != TokenType.Method)
  1424. continue;
  1425. pinvokes.Add (method.RID, new Row<PInvokeAttributes, uint, uint> (attributes, name, scope));
  1426. }
  1427. }
  1428. public bool HasGenericParameters (IGenericParameterProvider provider)
  1429. {
  1430. InitializeGenericParameters ();
  1431. Range [] ranges;
  1432. if (!metadata.TryGetGenericParameterRanges (provider, out ranges))
  1433. return false;
  1434. return RangesSize (ranges) > 0;
  1435. }
  1436. public Collection<GenericParameter> ReadGenericParameters (IGenericParameterProvider provider)
  1437. {
  1438. InitializeGenericParameters ();
  1439. Range [] ranges;
  1440. if (!metadata.TryGetGenericParameterRanges (provider, out ranges))
  1441. return new GenericParameterCollection (provider);
  1442. metadata.RemoveGenericParameterRange (provider);
  1443. var generic_parameters = new GenericParameterCollection (provider, RangesSize (ranges));
  1444. for (int i = 0; i < ranges.Length; i++)
  1445. ReadGenericParametersRange (ranges [i], provider, generic_parameters);
  1446. return generic_parameters;
  1447. }
  1448. void ReadGenericParametersRange (Range range, IGenericParameterProvider provider, GenericParameterCollection generic_parameters)
  1449. {
  1450. if (!MoveTo (Table.GenericParam, range.Start))
  1451. return;
  1452. for (uint i = 0; i < range.Length; i++) {
  1453. ReadUInt16 (); // index
  1454. var flags = (GenericParameterAttributes) ReadUInt16 ();
  1455. ReadMetadataToken (CodedIndex.TypeOrMethodDef);
  1456. var name = ReadString ();
  1457. var parameter = new GenericParameter (name, provider);
  1458. parameter.token = new MetadataToken (TokenType.GenericParam, range.Start + i);
  1459. parameter.Attributes = flags;
  1460. generic_parameters.Add (parameter);
  1461. }
  1462. }
  1463. void InitializeGenericParameters ()
  1464. {
  1465. if (metadata.GenericParameters != null)
  1466. return;
  1467. metadata.GenericParameters = InitializeRanges (
  1468. Table.GenericParam, () => {
  1469. Advance (4);
  1470. var next = ReadMetadataToken (CodedIndex.TypeOrMethodDef);
  1471. ReadStringIndex ();
  1472. return next;
  1473. });
  1474. }
  1475. Dictionary<MetadataToken, Range []> InitializeRanges (Table table, Func<MetadataToken> get_next)
  1476. {
  1477. int length = MoveTo (table);
  1478. var ranges = new Dictionary<MetadataToken, Range []> (length);
  1479. if (length == 0)
  1480. return ranges;
  1481. MetadataToken owner = MetadataToken.Zero;
  1482. Range range = new Range (1, 0);
  1483. for (uint i = 1; i <= length; i++) {
  1484. var next = get_next ();
  1485. if (i == 1) {
  1486. owner = next;
  1487. range.Length++;
  1488. } else if (next != owner) {
  1489. AddRange (ranges, owner, range);
  1490. range = new Range (i, 1);
  1491. owner = next;
  1492. } else
  1493. range.Length++;
  1494. }
  1495. AddRange (ranges, owner, range);
  1496. return ranges;
  1497. }
  1498. static void AddRange (Dictionary<MetadataToken, Range []> ranges, MetadataToken owner, Range range)
  1499. {
  1500. if (owner.RID == 0)
  1501. return;
  1502. Range [] slots;
  1503. if (!ranges.TryGetValue (owner, out slots)) {
  1504. ranges.Add (owner, new [] { range });
  1505. return;
  1506. }
  1507. ranges [owner] = slots.Add(range);
  1508. }
  1509. public bool HasGenericConstraints (GenericParameter generic_parameter)
  1510. {
  1511. InitializeGenericConstraints ();
  1512. Collection<Row<uint, MetadataToken>> mapping;
  1513. if (!metadata.TryGetGenericConstraintMapping (generic_parameter, out mapping))
  1514. return false;
  1515. return mapping.Count > 0;
  1516. }
  1517. public GenericParameterConst