| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287 |
- //
- // AssemblyReader.cs
- //
- // Author:
- // Jb Evain (jbevain@gmail.com)
- //
- // Copyright (c) 2008 - 2011 Jb Evain
- //
- // Permission is hereby granted, free of charge, to any person obtaining
- // a copy of this software and associated documentation files (the
- // "Software"), to deal in the Software without restriction, including
- // without limitation the rights to use, copy, modify, merge, publish,
- // distribute, sublicense, and/or sell copies of the Software, and to
- // permit persons to whom the Software is furnished to do so, subject to
- // the following conditions:
- //
- // The above copyright notice and this permission notice shall be
- // included in all copies or substantial portions of the Software.
- //
- // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
- // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
- // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
- // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
- // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
- // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
- // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
- //
- using System;
- using System.Collections.Generic;
- using System.IO;
- using System.Text;
- using Mono.Collections.Generic;
- using Mono.Cecil.Cil;
- using Mono.Cecil.Metadata;
- using Mono.Cecil.PE;
- using RVA = System.UInt32;
- namespace Mono.Cecil
- {
- abstract class ModuleReader
- {
- readonly protected Image image;
- readonly protected ModuleDefinition module;
- protected ModuleReader(Image image, ReadingMode mode)
- {
- this.image = image;
- this.module = new ModuleDefinition(image);
- this.module.ReadingMode = mode;
- }
- protected abstract void ReadModule();
- protected void ReadModuleManifest(MetadataReader reader)
- {
- reader.Populate(module);
- ReadAssembly(reader);
- }
- void ReadAssembly(MetadataReader reader)
- {
- var name = reader.ReadAssemblyNameDefinition();
- if (name == null)
- {
- module.kind = ModuleKind.NetModule;
- return;
- }
- var assembly = new AssemblyDefinition();
- assembly.Name = name;
- module.assembly = assembly;
- assembly.main_module = module;
- }
- public static ModuleDefinition CreateModuleFrom(Image image, ReaderParameters parameters)
- {
- var module = ReadModule(image, parameters);
- ReadSymbols(module, parameters);
- if (parameters.AssemblyResolver != null)
- module.assembly_resolver = parameters.AssemblyResolver;
- if (parameters.MetadataResolver != null)
- module.metadata_resolver = parameters.MetadataResolver;
- return module;
- }
- static void ReadSymbols(ModuleDefinition module, ReaderParameters parameters)
- {
- var symbol_reader_provider = parameters.SymbolReaderProvider;
- if (symbol_reader_provider == null && parameters.ReadSymbols)
- symbol_reader_provider = SymbolProvider.GetPlatformReaderProvider();
- if (symbol_reader_provider != null)
- {
- module.SymbolReaderProvider = symbol_reader_provider;
- var reader = parameters.SymbolStream != null
- ? symbol_reader_provider.GetSymbolReader(module, parameters.SymbolStream)
- : symbol_reader_provider.GetSymbolReader(module, module.FullyQualifiedName);
- module.ReadSymbols(reader);
- }
- }
- static ModuleDefinition ReadModule(Image image, ReaderParameters parameters)
- {
- var reader = CreateModuleReader(image, parameters.ReadingMode);
- reader.ReadModule();
- return reader.module;
- }
- static ModuleReader CreateModuleReader(Image image, ReadingMode mode)
- {
- switch (mode)
- {
- case ReadingMode.Immediate:
- return new ImmediateModuleReader(image);
- case ReadingMode.Deferred:
- return new DeferredModuleReader(image);
- default:
- throw new ArgumentException();
- }
- }
- }
- sealed class ImmediateModuleReader : ModuleReader
- {
- public ImmediateModuleReader(Image image)
- : base(image, ReadingMode.Immediate)
- {
- }
- protected override void ReadModule()
- {
- this.module.Read(this.module, (module, reader) =>
- {
- ReadModuleManifest(reader);
- ReadModule(module);
- return module;
- });
- }
- public static void ReadModule(ModuleDefinition module)
- {
- if (module.HasAssemblyReferences)
- Read(module.AssemblyReferences);
- if (module.HasResources)
- Read(module.Resources);
- if (module.HasModuleReferences)
- Read(module.ModuleReferences);
- if (module.HasTypes)
- ReadTypes(module.Types);
- if (module.HasExportedTypes)
- Read(module.ExportedTypes);
- if (module.HasCustomAttributes)
- Read(module.CustomAttributes);
- var assembly = module.Assembly;
- if (assembly == null)
- return;
- if (assembly.HasCustomAttributes)
- Read(assembly.CustomAttributes);
- if (assembly.HasSecurityDeclarations)
- Read(assembly.SecurityDeclarations);
- }
- static void ReadTypes(Collection<TypeDefinition> types)
- {
- for (int i = 0; i < types.Count; i++)
- ReadType(types[i]);
- }
- static void ReadType(TypeDefinition type)
- {
- ReadGenericParameters(type);
- if (type.HasInterfaces)
- Read(type.Interfaces);
- if (type.HasNestedTypes)
- ReadTypes(type.NestedTypes);
- if (type.HasLayoutInfo)
- Read(type.ClassSize);
- if (type.HasFields)
- ReadFields(type);
- if (type.HasMethods)
- ReadMethods(type);
- if (type.HasProperties)
- ReadProperties(type);
- if (type.HasEvents)
- ReadEvents(type);
- ReadSecurityDeclarations(type);
- ReadCustomAttributes(type);
- }
- static void ReadGenericParameters(IGenericParameterProvider provider)
- {
- if (!provider.HasGenericParameters)
- return;
- var parameters = provider.GenericParameters;
- for (int i = 0; i < parameters.Count; i++)
- {
- var parameter = parameters[i];
- if (parameter.HasConstraints)
- Read(parameter.Constraints);
- if (parameter.HasCustomAttributes)
- Read(parameter.CustomAttributes);
- }
- }
- static void ReadSecurityDeclarations(ISecurityDeclarationProvider provider)
- {
- if (provider.HasSecurityDeclarations)
- Read(provider.SecurityDeclarations);
- }
- static void ReadCustomAttributes(ICustomAttributeProvider provider)
- {
- if (provider.HasCustomAttributes)
- Read(provider.CustomAttributes);
- }
- static void ReadFields(TypeDefinition type)
- {
- var fields = type.Fields;
- for (int i = 0; i < fields.Count; i++)
- {
- var field = fields[i];
- if (field.HasConstant)
- Read(field.Constant);
- if (field.HasLayoutInfo)
- Read(field.Offset);
- if (field.RVA > 0)
- Read(field.InitialValue);
- if (field.HasMarshalInfo)
- Read(field.MarshalInfo);
- ReadCustomAttributes(field);
- }
- }
- static void ReadMethods(TypeDefinition type)
- {
- var methods = type.Methods;
- for (int i = 0; i < methods.Count; i++)
- {
- var method = methods[i];
- ReadGenericParameters(method);
- if (method.HasParameters)
- ReadParameters(method);
- if (method.HasOverrides)
- Read(method.Overrides);
- if (method.IsPInvokeImpl)
- Read(method.PInvokeInfo);
- ReadSecurityDeclarations(method);
- ReadCustomAttributes(method);
- var return_type = method.MethodReturnType;
- if (return_type.HasConstant)
- Read(return_type.Constant);
- if (return_type.HasMarshalInfo)
- Read(return_type.MarshalInfo);
- ReadCustomAttributes(return_type);
- }
- }
- static void ReadParameters(MethodDefinition method)
- {
- var parameters = method.Parameters;
- for (int i = 0; i < parameters.Count; i++)
- {
- var parameter = parameters[i];
- if (parameter.HasConstant)
- Read(parameter.Constant);
- if (parameter.HasMarshalInfo)
- Read(parameter.MarshalInfo);
- ReadCustomAttributes(parameter);
- }
- }
- static void ReadProperties(TypeDefinition type)
- {
- var properties = type.Properties;
- for (int i = 0; i < properties.Count; i++)
- {
- var property = properties[i];
- Read(property.GetMethod);
- if (property.HasConstant)
- Read(property.Constant);
- ReadCustomAttributes(property);
- }
- }
- static void ReadEvents(TypeDefinition type)
- {
- var events = type.Events;
- for (int i = 0; i < events.Count; i++)
- {
- var @event = events[i];
- Read(@event.AddMethod);
- ReadCustomAttributes(@event);
- }
- }
- static void Read(object collection)
- {
- }
- }
- sealed class DeferredModuleReader : ModuleReader
- {
- public DeferredModuleReader(Image image)
- : base(image, ReadingMode.Deferred)
- {
- }
- protected override void ReadModule()
- {
- this.module.Read(this.module, (module, reader) =>
- {
- ReadModuleManifest(reader);
- return module;
- });
- }
- }
- sealed class MetadataReader : ByteBuffer
- {
- readonly internal Image image;
- readonly internal ModuleDefinition module;
- readonly internal MetadataSystem metadata;
- internal IGenericContext context;
- internal CodeReader code;
- uint Position
- {
- get { return (uint)base.position; }
- set { base.position = (int)value; }
- }
- public MetadataReader(ModuleDefinition module)
- : base(module.Image.MetadataSection.Data)
- {
- this.image = module.Image;
- this.module = module;
- this.metadata = module.MetadataSystem;
- this.code = new CodeReader(image.MetadataSection, this);
- }
- int GetCodedIndexSize(CodedIndex index)
- {
- return image.GetCodedIndexSize(index);
- }
- uint ReadByIndexSize(int size)
- {
- if (size == 4)
- return ReadUInt32();
- else
- return ReadUInt16();
- }
- byte[] ReadBlob()
- {
- var blob_heap = image.BlobHeap;
- if (blob_heap == null)
- {
- position += 2;
- return Empty<byte>.Array;
- }
- return blob_heap.Read(ReadBlobIndex());
- }
- byte[] ReadBlob(uint signature)
- {
- var blob_heap = image.BlobHeap;
- if (blob_heap == null)
- return Empty<byte>.Array;
- return blob_heap.Read(signature);
- }
- uint ReadBlobIndex()
- {
- var blob_heap = image.BlobHeap;
- return ReadByIndexSize(blob_heap != null ? blob_heap.IndexSize : 2);
- }
- string ReadString()
- {
- return image.StringHeap.Read(ReadByIndexSize(image.StringHeap.IndexSize));
- }
- uint ReadStringIndex()
- {
- return ReadByIndexSize(image.StringHeap.IndexSize);
- }
- uint ReadTableIndex(Table table)
- {
- return ReadByIndexSize(image.GetTableIndexSize(table));
- }
- MetadataToken ReadMetadataToken(CodedIndex index)
- {
- return Mixin.GetMetadataToken(index, ReadByIndexSize(GetCodedIndexSize(index)));
- }
- int MoveTo(Table table)
- {
- var info = image.TableHeap[table];
- if (info.Length != 0)
- Position = info.Offset;
- return (int)info.Length;
- }
- bool MoveTo(Table table, uint row)
- {
- var info = image.TableHeap[table];
- var length = info.Length;
- if (length == 0 || row > length)
- return false;
- Position = info.Offset + (info.RowSize * (row - 1));
- return true;
- }
- public AssemblyNameDefinition ReadAssemblyNameDefinition()
- {
- if (MoveTo(Table.Assembly) == 0)
- return null;
- var name = new AssemblyNameDefinition();
- name.HashAlgorithm = (AssemblyHashAlgorithm)ReadUInt32();
- PopulateVersionAndFlags(name);
- name.PublicKey = ReadBlob();
- PopulateNameAndCulture(name);
- return name;
- }
- public ModuleDefinition Populate(ModuleDefinition module)
- {
- if (MoveTo(Table.Module) == 0)
- return module;
- Advance(2); // Generation
- module.Name = ReadString();
- module.Mvid = image.GuidHeap.Read(ReadByIndexSize(image.GuidHeap.IndexSize));
- return module;
- }
- void InitializeAssemblyReferences()
- {
- if (metadata.AssemblyReferences != null)
- return;
- int length = MoveTo(Table.AssemblyRef);
- var references = metadata.AssemblyReferences = new AssemblyNameReference[length];
- for (uint i = 0; i < length; i++)
- {
- var reference = new AssemblyNameReference();
- reference.token = new MetadataToken(TokenType.AssemblyRef, i + 1);
- PopulateVersionAndFlags(reference);
- var key_or_token = ReadBlob();
- if (reference.HasPublicKey)
- reference.PublicKey = key_or_token;
- else
- reference.PublicKeyToken = key_or_token;
- PopulateNameAndCulture(reference);
- reference.Hash = ReadBlob();
- references[i] = reference;
- }
- }
- public Collection<AssemblyNameReference> ReadAssemblyReferences()
- {
- InitializeAssemblyReferences();
- return new Collection<AssemblyNameReference>(metadata.AssemblyReferences);
- }
- public MethodDefinition ReadEntryPoint()
- {
- if (module.Image.EntryPointToken == 0)
- return null;
- var token = new MetadataToken(module.Image.EntryPointToken);
- return GetMethodDefinition(token.RID);
- }
- public Collection<ModuleDefinition> ReadModules()
- {
- var modules = new Collection<ModuleDefinition>(1);
- modules.Add(this.module);
- int length = MoveTo(Table.File);
- for (uint i = 1; i <= length; i++)
- {
- var attributes = (FileAttributes)ReadUInt32();
- var name = ReadString();
- ReadBlobIndex();
- if (attributes != FileAttributes.ContainsMetaData)
- continue;
- var parameters = new ReaderParameters
- {
- ReadingMode = module.ReadingMode,
- SymbolReaderProvider = module.SymbolReaderProvider,
- };
- modules.Add(ModuleDefinition.ReadModule(
- GetModuleFileName(name), parameters));
- }
- return modules;
- }
- string GetModuleFileName(string name)
- {
- if (module.FullyQualifiedName == null)
- throw new NotSupportedException();
- var path = Path.GetDirectoryName(module.FullyQualifiedName);
- return Path.Combine(path, name);
- }
- void InitializeModuleReferences()
- {
- if (metadata.ModuleReferences != null)
- return;
- int length = MoveTo(Table.ModuleRef);
- var references = metadata.ModuleReferences = new ModuleReference[length];
- for (uint i = 0; i < length; i++)
- {
- var reference = new ModuleReference(ReadString());
- reference.token = new MetadataToken(TokenType.ModuleRef, i + 1);
- references[i] = reference;
- }
- }
- public Collection<ModuleReference> ReadModuleReferences()
- {
- InitializeModuleReferences();
- return new Collection<ModuleReference>(metadata.ModuleReferences);
- }
- public bool HasFileResource()
- {
- int length = MoveTo(Table.File);
- if (length == 0)
- return false;
- for (uint i = 1; i <= length; i++)
- if (ReadFileRecord(i).Col1 == FileAttributes.ContainsNoMetaData)
- return true;
- return false;
- }
- public Collection<Resource> ReadResources()
- {
- int length = MoveTo(Table.ManifestResource);
- var resources = new Collection<Resource>(length);
- for (int i = 1; i <= length; i++)
- {
- var offset = ReadUInt32();
- var flags = (ManifestResourceAttributes)ReadUInt32();
- var name = ReadString();
- var implementation = ReadMetadataToken(CodedIndex.Implementation);
- Resource resource;
- if (implementation.RID == 0)
- {
- resource = new EmbeddedResource(name, flags, offset, this);
- }
- else if (implementation.TokenType == TokenType.AssemblyRef)
- {
- resource = new AssemblyLinkedResource(name, flags)
- {
- Assembly = (AssemblyNameReference)GetTypeReferenceScope(implementation),
- };
- }
- else if (implementation.TokenType == TokenType.File)
- {
- var file_record = ReadFileRecord(implementation.RID);
- resource = new LinkedResource(name, flags)
- {
- File = file_record.Col2,
- hash = ReadBlob(file_record.Col3)
- };
- }
- else
- throw new NotSupportedException();
- resources.Add(resource);
- }
- return resources;
- }
- Row<FileAttributes, string, uint> ReadFileRecord(uint rid)
- {
- var position = this.position;
- if (!MoveTo(Table.File, rid))
- throw new ArgumentException();
- var record = new Row<FileAttributes, string, uint>(
- (FileAttributes)ReadUInt32(),
- ReadString(),
- ReadBlobIndex());
- this.position = position;
- return record;
- }
- public MemoryStream GetManagedResourceStream(uint offset)
- {
- var rva = image.Resources.VirtualAddress;
- var section = image.GetSectionAtVirtualAddress(rva);
- var position = (rva - section.VirtualAddress) + offset;
- var buffer = section.Data;
- var length = buffer[position]
- | (buffer[position + 1] << 8)
- | (buffer[position + 2] << 16)
- | (buffer[position + 3] << 24);
- return new MemoryStream(buffer, (int)position + 4, length);
- }
- void PopulateVersionAndFlags(AssemblyNameReference name)
- {
- name.Version = new Version(
- ReadUInt16(),
- ReadUInt16(),
- ReadUInt16(),
- ReadUInt16());
- name.Attributes = (AssemblyAttributes)ReadUInt32();
- }
- void PopulateNameAndCulture(AssemblyNameReference name)
- {
- name.Name = ReadString();
- name.Culture = ReadString();
- }
- public TypeDefinitionCollection ReadTypes()
- {
- InitializeTypeDefinitions();
- var mtypes = metadata.Types;
- var type_count = mtypes.Length - metadata.NestedTypes.Count;
- var types = new TypeDefinitionCollection(module, type_count);
- for (int i = 0; i < mtypes.Length; i++)
- {
- var type = mtypes[i];
- if (IsNested(type.Attributes))
- continue;
- types.Add(type);
- }
- if (image.HasTable(Table.MethodPtr) || image.HasTable(Table.FieldPtr))
- CompleteTypes();
- return types;
- }
- void CompleteTypes()
- {
- var types = metadata.Types;
- for (int i = 0; i < types.Length; i++)
- {
- var type = types[i];
- InitializeCollection(type.Fields);
- InitializeCollection(type.Methods);
- }
- }
- void InitializeTypeDefinitions()
- {
- if (metadata.Types != null)
- return;
- InitializeNestedTypes();
- InitializeFields();
- InitializeMethods();
- int length = MoveTo(Table.TypeDef);
- var types = metadata.Types = new TypeDefinition[length];
- for (uint i = 0; i < length; i++)
- {
- if (types[i] != null)
- continue;
- types[i] = ReadType(i + 1);
- }
- }
- static bool IsNested(TypeAttributes attributes)
- {
- switch (attributes & TypeAttributes.VisibilityMask)
- {
- case TypeAttributes.NestedAssembly:
- case TypeAttributes.NestedFamANDAssem:
- case TypeAttributes.NestedFamily:
- case TypeAttributes.NestedFamORAssem:
- case TypeAttributes.NestedPrivate:
- case TypeAttributes.NestedPublic:
- return true;
- default:
- return false;
- }
- }
- public bool HasNestedTypes(TypeDefinition type)
- {
- uint[] mapping;
- InitializeNestedTypes();
- if (!metadata.TryGetNestedTypeMapping(type, out mapping))
- return false;
- return mapping.Length > 0;
- }
- public Collection<TypeDefinition> ReadNestedTypes(TypeDefinition type)
- {
- InitializeNestedTypes();
- uint[] mapping;
- if (!metadata.TryGetNestedTypeMapping(type, out mapping))
- return new MemberDefinitionCollection<TypeDefinition>(type);
- var nested_types = new MemberDefinitionCollection<TypeDefinition>(type, mapping.Length);
- for (int i = 0; i < mapping.Length; i++)
- {
- var nested_type = GetTypeDefinition(mapping[i]);
- if (nested_type != null)
- nested_types.Add(nested_type);
- }
- metadata.RemoveNestedTypeMapping(type);
- return nested_types;
- }
- void InitializeNestedTypes()
- {
- if (metadata.NestedTypes != null)
- return;
- var length = MoveTo(Table.NestedClass);
- metadata.NestedTypes = new Dictionary<uint, uint[]>(length);
- metadata.ReverseNestedTypes = new Dictionary<uint, uint>(length);
- if (length == 0)
- return;
- for (int i = 1; i <= length; i++)
- {
- var nested = ReadTableIndex(Table.TypeDef);
- var declaring = ReadTableIndex(Table.TypeDef);
- AddNestedMapping(declaring, nested);
- }
- }
- void AddNestedMapping(uint declaring, uint nested)
- {
- metadata.SetNestedTypeMapping(declaring, AddMapping(metadata.NestedTypes, declaring, nested));
- metadata.SetReverseNestedTypeMapping(nested, declaring);
- }
- static TValue[] AddMapping<TKey, TValue>(Dictionary<TKey, TValue[]> cache, TKey key, TValue value)
- {
- TValue[] mapped;
- if (!cache.TryGetValue(key, out mapped))
- {
- mapped = new[] { value };
- return mapped;
- }
- var new_mapped = new TValue[mapped.Length + 1];
- Array.Copy(mapped, new_mapped, mapped.Length);
- new_mapped[mapped.Length] = value;
- return new_mapped;
- }
- TypeDefinition ReadType(uint rid)
- {
- if (!MoveTo(Table.TypeDef, rid))
- return null;
- var attributes = (TypeAttributes)ReadUInt32();
- var name = ReadString();
- var @namespace = ReadString();
- var type = new TypeDefinition(@namespace, name, attributes);
- type.token = new MetadataToken(TokenType.TypeDef, rid);
- type.scope = module;
- type.module = module;
- metadata.AddTypeDefinition(type);
- this.context = type;
- type.BaseType = GetTypeDefOrRef(ReadMetadataToken(CodedIndex.TypeDefOrRef));
- type.fields_range = ReadFieldsRange(rid);
- type.methods_range = ReadMethodsRange(rid);
- if (IsNested(attributes))
- type.DeclaringType = GetNestedTypeDeclaringType(type);
- return type;
- }
- TypeDefinition GetNestedTypeDeclaringType(TypeDefinition type)
- {
- uint declaring_rid;
- if (!metadata.TryGetReverseNestedTypeMapping(type, out declaring_rid))
- return null;
- metadata.RemoveReverseNestedTypeMapping(type);
- return GetTypeDefinition(declaring_rid);
- }
- Range ReadFieldsRange(uint type_index)
- {
- return ReadListRange(type_index, Table.TypeDef, Table.Field);
- }
- Range ReadMethodsRange(uint type_index)
- {
- return ReadListRange(type_index, Table.TypeDef, Table.Method);
- }
- Range ReadListRange(uint current_index, Table current, Table target)
- {
- var list = new Range();
- list.Start = ReadTableIndex(target);
- uint next_index;
- var current_table = image.TableHeap[current];
- if (current_index == current_table.Length)
- next_index = image.TableHeap[target].Length + 1;
- else
- {
- var position = Position;
- Position += (uint)(current_table.RowSize - image.GetTableIndexSize(target));
- next_index = ReadTableIndex(target);
- Position = position;
- }
- list.Length = next_index - list.Start;
- return list;
- }
- public Row<short, int> ReadTypeLayout(TypeDefinition type)
- {
- InitializeTypeLayouts();
- Row<ushort, uint> class_layout;
- var rid = type.token.RID;
- if (!metadata.ClassLayouts.TryGetValue(rid, out class_layout))
- return new Row<short, int>(Mixin.NoDataMarker, Mixin.NoDataMarker);
- type.PackingSize = (short)class_layout.Col1;
- type.ClassSize = (int)class_layout.Col2;
- metadata.ClassLayouts.Remove(rid);
- return new Row<short, int>((short)class_layout.Col1, (int)class_layout.Col2);
- }
- void InitializeTypeLayouts()
- {
- if (metadata.ClassLayouts != null)
- return;
- int length = MoveTo(Table.ClassLayout);
- var class_layouts = metadata.ClassLayouts = new Dictionary<uint, Row<ushort, uint>>(length);
- for (uint i = 0; i < length; i++)
- {
- var packing_size = ReadUInt16();
- var class_size = ReadUInt32();
- var parent = ReadTableIndex(Table.TypeDef);
- class_layouts.Add(parent, new Row<ushort, uint>(packing_size, class_size));
- }
- }
- public TypeReference GetTypeDefOrRef(MetadataToken token)
- {
- return (TypeReference)LookupToken(token);
- }
- public TypeDefinition GetTypeDefinition(uint rid)
- {
- InitializeTypeDefinitions();
- var type = metadata.GetTypeDefinition(rid);
- if (type != null)
- return type;
- return ReadTypeDefinition(rid);
- }
- TypeDefinition ReadTypeDefinition(uint rid)
- {
- if (!MoveTo(Table.TypeDef, rid))
- return null;
- return ReadType(rid);
- }
- void InitializeTypeReferences()
- {
- if (metadata.TypeReferences != null)
- return;
- metadata.TypeReferences = new TypeReference[image.GetTableLength(Table.TypeRef)];
- }
- public TypeReference GetTypeReference(string scope, string full_name)
- {
- InitializeTypeReferences();
- var length = metadata.TypeReferences.Length;
- for (uint i = 1; i <= length; i++)
- {
- var type = GetTypeReference(i);
- if (type.FullName != full_name)
- continue;
- if (string.IsNullOrEmpty(scope))
- return type;
- if (type.Scope.Name == scope)
- return type;
- }
- return null;
- }
- TypeReference GetTypeReference(uint rid)
- {
- InitializeTypeReferences();
- var type = metadata.GetTypeReference(rid);
- if (type != null)
- return type;
- return ReadTypeReference(rid);
- }
- TypeReference ReadTypeReference(uint rid)
- {
- if (!MoveTo(Table.TypeRef, rid))
- return null;
- TypeReference declaring_type = null;
- IMetadataScope scope;
- var scope_token = ReadMetadataToken(CodedIndex.ResolutionScope);
- var name = ReadString();
- var @namespace = ReadString();
- var type = new TypeReference(
- @namespace,
- name,
- module,
- null);
- type.token = new MetadataToken(TokenType.TypeRef, rid);
- metadata.AddTypeReference(type);
- if (scope_token.TokenType == TokenType.TypeRef)
- {
- declaring_type = GetTypeDefOrRef(scope_token);
- scope = declaring_type != null
- ? declaring_type.Scope
- : module;
- }
- else
- scope = GetTypeReferenceScope(scope_token);
- type.scope = scope;
- type.DeclaringType = declaring_type;
- MetadataSystem.TryProcessPrimitiveTypeReference(type);
- return type;
- }
- IMetadataScope GetTypeReferenceScope(MetadataToken scope)
- {
- switch (scope.TokenType)
- {
- case TokenType.AssemblyRef:
- InitializeAssemblyReferences();
- return metadata.AssemblyReferences[(int)scope.RID - 1];
- case TokenType.ModuleRef:
- InitializeModuleReferences();
- return metadata.ModuleReferences[(int)scope.RID - 1];
- case TokenType.Module:
- return module;
- default:
- throw new NotSupportedException();
- }
- }
- public IEnumerable<TypeReference> GetTypeReferences()
- {
- InitializeTypeReferences();
- var length = image.GetTableLength(Table.TypeRef);
- var type_references = new TypeReference[length];
- for (uint i = 1; i <= length; i++)
- type_references[i - 1] = GetTypeReference(i);
- return type_references;
- }
- TypeReference GetTypeSpecification(uint rid)
- {
- if (!MoveTo(Table.TypeSpec, rid))
- return null;
- var reader = ReadSignature(ReadBlobIndex());
- var type = reader.ReadTypeSignature();
- if (type.token.RID == 0)
- type.token = new MetadataToken(TokenType.TypeSpec, rid);
- return type;
- }
- SignatureReader ReadSignature(uint signature)
- {
- return new SignatureReader(signature, this);
- }
- public bool HasInterfaces(TypeDefinition type)
- {
- InitializeInterfaces();
- MetadataToken[] mapping;
- return metadata.TryGetInterfaceMapping(type, out mapping);
- }
- public Collection<TypeReference> ReadInterfaces(TypeDefinition type)
- {
- InitializeInterfaces();
- MetadataToken[] mapping;
- if (!metadata.TryGetInterfaceMapping(type, out mapping))
- return new Collection<TypeReference>();
- var interfaces = new Collection<TypeReference>(mapping.Length);
- this.context = type;
- for (int i = 0; i < mapping.Length; i++)
- interfaces.Add(GetTypeDefOrRef(mapping[i]));
- metadata.RemoveInterfaceMapping(type);
- return interfaces;
- }
- void InitializeInterfaces()
- {
- if (metadata.Interfaces != null)
- return;
- int length = MoveTo(Table.InterfaceImpl);
- metadata.Interfaces = new Dictionary<uint, MetadataToken[]>(length);
- for (int i = 0; i < length; i++)
- {
- var type = ReadTableIndex(Table.TypeDef);
- var @interface = ReadMetadataToken(CodedIndex.TypeDefOrRef);
- AddInterfaceMapping(type, @interface);
- }
- }
- void AddInterfaceMapping(uint type, MetadataToken @interface)
- {
- metadata.SetInterfaceMapping(type, AddMapping(metadata.Interfaces, type, @interface));
- }
- public Collection<FieldDefinition> ReadFields(TypeDefinition type)
- {
- var fields_range = type.fields_range;
- if (fields_range.Length == 0)
- return new MemberDefinitionCollection<FieldDefinition>(type);
- var fields = new MemberDefinitionCollection<FieldDefinition>(type, (int)fields_range.Length);
- this.context = type;
- if (!MoveTo(Table.FieldPtr, fields_range.Start))
- {
- if (!MoveTo(Table.Field, fields_range.Start))
- return fields;
- for (uint i = 0; i < fields_range.Length; i++)
- ReadField(fields_range.Start + i, fields);
- }
- else
- ReadPointers(Table.FieldPtr, Table.Field, fields_range, fields, ReadField);
- return fields;
- }
- void ReadField(uint field_rid, Collection<FieldDefinition> fields)
- {
- var attributes = (FieldAttributes)ReadUInt16();
- var name = ReadString();
- var signature = ReadBlobIndex();
- var field = new FieldDefinition(name, attributes, ReadFieldType(signature));
- field.token = new MetadataToken(TokenType.Field, field_rid);
- metadata.AddFieldDefinition(field);
- if (IsDeleted(field))
- return;
- fields.Add(field);
- }
- void InitializeFields()
- {
- if (metadata.Fields != null)
- return;
- metadata.Fields = new FieldDefinition[image.GetTableLength(Table.Field)];
- }
- TypeReference ReadFieldType(uint signature)
- {
- var reader = ReadSignature(signature);
- const byte field_sig = 0x6;
- if (reader.ReadByte() != field_sig)
- throw new NotSupportedException();
- return reader.ReadTypeSignature();
- }
- public int ReadFieldRVA(FieldDefinition field)
- {
- InitializeFieldRVAs();
- var rid = field.token.RID;
- RVA rva;
- if (!metadata.FieldRVAs.TryGetValue(rid, out rva))
- return 0;
- var size = GetFieldTypeSize(field.FieldType);
- if (size == 0 || rva == 0)
- return 0;
- metadata.FieldRVAs.Remove(rid);
- field.InitialValue = GetFieldInitializeValue(size, rva);
- return (int)rva;
- }
- byte[] GetFieldInitializeValue(int size, RVA rva)
- {
- var section = image.GetSectionAtVirtualAddress(rva);
- if (section == null)
- return Empty<byte>.Array;
- var value = new byte[size];
- Buffer.BlockCopy(section.Data, (int)(rva - section.VirtualAddress), value, 0, size);
- return value;
- }
- static int GetFieldTypeSize(TypeReference type)
- {
- int size = 0;
- switch (type.etype)
- {
- case ElementType.Boolean:
- case ElementType.U1:
- case ElementType.I1:
- size = 1;
- break;
- case ElementType.U2:
- case ElementType.I2:
- case ElementType.Char:
- size = 2;
- break;
- case ElementType.U4:
- case ElementType.I4:
- case ElementType.R4:
- size = 4;
- break;
- case ElementType.U8:
- case ElementType.I8:
- case ElementType.R8:
- size = 8;
- break;
- case ElementType.Ptr:
- case ElementType.FnPtr:
- size = IntPtr.Size;
- break;
- case ElementType.CModOpt:
- case ElementType.CModReqD:
- return GetFieldTypeSize(((IModifierType)type).ElementType);
- default:
- var field_type = Mixin.CheckedResolve(type);
- if (field_type.HasLayoutInfo)
- size = field_type.ClassSize;
- break;
- }
- return size;
- }
- void InitializeFieldRVAs()
- {
- if (metadata.FieldRVAs != null)
- return;
- int length = MoveTo(Table.FieldRVA);
- var field_rvas = metadata.FieldRVAs = new Dictionary<uint, uint>(length);
- for (int i = 0; i < length; i++)
- {
- var rva = ReadUInt32();
- var field = ReadTableIndex(Table.Field);
- field_rvas.Add(field, rva);
- }
- }
- public int ReadFieldLayout(FieldDefinition field)
- {
- InitializeFieldLayouts();
- var rid = field.token.RID;
- uint offset;
- if (!metadata.FieldLayouts.TryGetValue(rid, out offset))
- return Mixin.NoDataMarker;
- metadata.FieldLayouts.Remove(rid);
- return (int)offset;
- }
- void InitializeFieldLayouts()
- {
- if (metadata.FieldLayouts != null)
- return;
- int length = MoveTo(Table.FieldLayout);
- var field_layouts = metadata.FieldLayouts = new Dictionary<uint, uint>(length);
- for (int i = 0; i < length; i++)
- {
- var offset = ReadUInt32();
- var field = ReadTableIndex(Table.Field);
- field_layouts.Add(field, offset);
- }
- }
- public bool HasEvents(TypeDefinition type)
- {
- InitializeEvents();
- Range range;
- if (!metadata.TryGetEventsRange(type, out range))
- return false;
- return range.Length > 0;
- }
- public Collection<EventDefinition> ReadEvents(TypeDefinition type)
- {
- InitializeEvents();
- Range range;
- if (!metadata.TryGetEventsRange(type, out range))
- return new MemberDefinitionCollection<EventDefinition>(type);
- var events = new MemberDefinitionCollection<EventDefinition>(type, (int)range.Length);
- metadata.RemoveEventsRange(type);
- if (range.Length == 0)
- return events;
- this.context = type;
- if (!MoveTo(Table.EventPtr, range.Start))
- {
- if (!MoveTo(Table.Event, range.Start))
- return events;
- for (uint i = 0; i < range.Length; i++)
- ReadEvent(range.Start + i, events);
- }
- else
- ReadPointers(Table.EventPtr, Table.Event, range, events, ReadEvent);
- return events;
- }
- void ReadEvent(uint event_rid, Collection<EventDefinition> events)
- {
- var attributes = (EventAttributes)ReadUInt16();
- var name = ReadString();
- var event_type = GetTypeDefOrRef(ReadMetadataToken(CodedIndex.TypeDefOrRef));
- var @event = new EventDefinition(name, attributes, event_type);
- @event.token = new MetadataToken(TokenType.Event, event_rid);
- if (IsDeleted(@event))
- return;
- events.Add(@event);
- }
- void InitializeEvents()
- {
- if (metadata.Events != null)
- return;
- int length = MoveTo(Table.EventMap);
- metadata.Events = new Dictionary<uint, Range>(length);
- for (uint i = 1; i <= length; i++)
- {
- var type_rid = ReadTableIndex(Table.TypeDef);
- Range events_range = ReadEventsRange(i);
- metadata.AddEventsRange(type_rid, events_range);
- }
- }
- Range ReadEventsRange(uint rid)
- {
- return ReadListRange(rid, Table.EventMap, Table.Event);
- }
- public bool HasProperties(TypeDefinition type)
- {
- InitializeProperties();
- Range range;
- if (!metadata.TryGetPropertiesRange(type, out range))
- return false;
- return range.Length > 0;
- }
- public Collection<PropertyDefinition> ReadProperties(TypeDefinition type)
- {
- InitializeProperties();
- Range range;
- if (!metadata.TryGetPropertiesRange(type, out range))
- return new MemberDefinitionCollection<PropertyDefinition>(type);
- metadata.RemovePropertiesRange(type);
- var properties = new MemberDefinitionCollection<PropertyDefinition>(type, (int)range.Length);
- if (range.Length == 0)
- return properties;
- this.context = type;
- if (!MoveTo(Table.PropertyPtr, range.Start))
- {
- if (!MoveTo(Table.Property, range.Start))
- return properties;
- for (uint i = 0; i < range.Length; i++)
- ReadProperty(range.Start + i, properties);
- }
- else
- ReadPointers(Table.PropertyPtr, Table.Property, range, properties, ReadProperty);
- return properties;
- }
- void ReadProperty(uint property_rid, Collection<PropertyDefinition> properties)
- {
- var attributes = (PropertyAttributes)ReadUInt16();
- var name = ReadString();
- var signature = ReadBlobIndex();
- var reader = ReadSignature(signature);
- const byte property_signature = 0x8;
- var calling_convention = reader.ReadByte();
- if ((calling_convention & property_signature) == 0)
- throw new NotSupportedException();
- var has_this = (calling_convention & 0x20) != 0;
- reader.ReadCompressedUInt32(); // count
- var property = new PropertyDefinition(name, attributes, reader.ReadTypeSignature());
- property.HasThis = has_this;
- property.token = new MetadataToken(TokenType.Property, property_rid);
- if (IsDeleted(property))
- return;
- properties.Add(property);
- }
- void InitializeProperties()
- {
- if (metadata.Properties != null)
- return;
- int length = MoveTo(Table.PropertyMap);
- metadata.Properties = new Dictionary<uint, Range>(length);
- for (uint i = 1; i <= length; i++)
- {
- var type_rid = ReadTableIndex(Table.TypeDef);
- var properties_range = ReadPropertiesRange(i);
- metadata.AddPropertiesRange(type_rid, properties_range);
- }
- }
- Range ReadPropertiesRange(uint rid)
- {
- return ReadListRange(rid, Table.PropertyMap, Table.Property);
- }
- MethodSemanticsAttributes ReadMethodSemantics(MethodDefinition method)
- {
- InitializeMethodSemantics();
- Row<MethodSemanticsAttributes, MetadataToken> row;
- if (!metadata.Semantics.TryGetValue(method.token.RID, out row))
- return MethodSemanticsAttributes.None;
- var type = method.DeclaringType;
- switch (row.Col1)
- {
- case MethodSemanticsAttributes.AddOn:
- GetEvent(type, row.Col2).add_method = method;
- break;
- case MethodSemanticsAttributes.Fire:
- GetEvent(type, row.Col2).invoke_method = method;
- break;
- case MethodSemanticsAttributes.RemoveOn:
- GetEvent(type, row.Col2).remove_method = method;
- break;
- case MethodSemanticsAttributes.Getter:
- GetProperty(type, row.Col2).get_method = method;
- break;
- case MethodSemanticsAttributes.Setter:
- GetProperty(type, row.Col2).set_method = method;
- break;
- case MethodSemanticsAttributes.Other:
- switch (row.Col2.TokenType)
- {
- case TokenType.Event:
- {
- var @event = GetEvent(type, row.Col2);
- if (@event.other_methods == null)
- @event.other_methods = new Collection<MethodDefinition>();
- @event.other_methods.Add(method);
- break;
- }
- case TokenType.Property:
- {
- var property = GetProperty(type, row.Col2);
- if (property.other_methods == null)
- property.other_methods = new Collection<MethodDefinition>();
- property.other_methods.Add(method);
- break;
- }
- default:
- throw new NotSupportedException();
- }
- break;
- default:
- throw new NotSupportedException();
- }
- metadata.Semantics.Remove(method.token.RID);
- return row.Col1;
- }
- static EventDefinition GetEvent(TypeDefinition type, MetadataToken token)
- {
- if (token.TokenType != TokenType.Event)
- throw new ArgumentException();
- return GetMember(type.Events, token);
- }
- static PropertyDefinition GetProperty(TypeDefinition type, MetadataToken token)
- {
- if (token.TokenType != TokenType.Property)
- throw new ArgumentException();
- return GetMember(type.Properties, token);
- }
- static TMember GetMember<TMember>(Collection<TMember> members, MetadataToken token) where TMember : IMemberDefinition
- {
- for (int i = 0; i < members.Count; i++)
- {
- var member = members[i];
- if (member.MetadataToken == token)
- return member;
- }
- throw new ArgumentException();
- }
- void InitializeMethodSemantics()
- {
- if (metadata.Semantics != null)
- return;
- int length = MoveTo(Table.MethodSemantics);
- var semantics = metadata.Semantics = new Dictionary<uint, Row<MethodSemanticsAttributes, MetadataToken>>(0);
- for (uint i = 0; i < length; i++)
- {
- var attributes = (MethodSemanticsAttributes)ReadUInt16();
- var method_rid = ReadTableIndex(Table.Method);
- var association = ReadMetadataToken(CodedIndex.HasSemantics);
- semantics[method_rid] = new Row<MethodSemanticsAttributes, MetadataToken>(attributes, association);
- }
- }
- public PropertyDefinition ReadMethods(PropertyDefinition property)
- {
- ReadAllSemantics(property.DeclaringType);
- return property;
- }
- public EventDefinition ReadMethods(EventDefinition @event)
- {
- ReadAllSemantics(@event.DeclaringType);
- return @event;
- }
- public MethodSemanticsAttributes ReadAllSemantics(MethodDefinition method)
- {
- ReadAllSemantics(method.DeclaringType);
- return method.SemanticsAttributes;
- }
- void ReadAllSemantics(TypeDefinition type)
- {
- var methods = type.Methods;
- for (int i = 0; i < methods.Count; i++)
- {
- var method = methods[i];
- if (method.sem_attrs_ready)
- continue;
- method.sem_attrs = ReadMethodSemantics(method);
- method.sem_attrs_ready = true;
- }
- }
- Range ReadParametersRange(uint method_rid)
- {
- return ReadListRange(method_rid, Table.Method, Table.Param);
- }
- public Collection<MethodDefinition> ReadMethods(TypeDefinition type)
- {
- var methods_range = type.methods_range;
- if (methods_range.Length == 0)
- return new MemberDefinitionCollection<MethodDefinition>(type);
- var methods = new MemberDefinitionCollection<MethodDefinition>(type, (int)methods_range.Length);
- if (!MoveTo(Table.MethodPtr, methods_range.Start))
- {
- if (!MoveTo(Table.Method, methods_range.Start))
- return methods;
- for (uint i = 0; i < methods_range.Length; i++)
- ReadMethod(methods_range.Start + i, methods);
- }
- else
- ReadPointers(Table.MethodPtr, Table.Method, methods_range, methods, ReadMethod);
- return methods;
- }
- void ReadPointers<TMember>(Table ptr, Table table, Range range, Collection<TMember> members, Action<uint, Collection<TMember>> reader)
- where TMember : IMemberDefinition
- {
- for (uint i = 0; i < range.Length; i++)
- {
- MoveTo(ptr, range.Start + i);
- var rid = ReadTableIndex(table);
- MoveTo(table, rid);
- reader(rid, members);
- }
- }
- static bool IsDeleted(IMemberDefinition member)
- {
- return member.IsSpecialName && member.Name == "_Deleted";
- }
- void InitializeMethods()
- {
- if (metadata.Methods != null)
- return;
- metadata.Methods = new MethodDefinition[image.GetTableLength(Table.Method)];
- }
- void ReadMethod(uint method_rid, Collection<MethodDefinition> methods)
- {
- var method = new MethodDefinition();
- method.rva = ReadUInt32();
- method.ImplAttributes = (MethodImplAttributes)ReadUInt16();
- method.Attributes = (MethodAttributes)ReadUInt16();
- method.Name = ReadString();
- method.token = new MetadataToken(TokenType.Method, method_rid);
- if (IsDeleted(method))
- return;
- methods.Add(method); // attach method
- var signature = ReadBlobIndex();
- var param_range = ReadParametersRange(method_rid);
- this.context = method;
- ReadMethodSignature(signature, method);
- metadata.AddMethodDefinition(method);
- if (param_range.Length == 0)
- return;
- var position = base.position;
- ReadParameters(method, param_range);
- base.position = position;
- }
- void ReadParameters(MethodDefinition method, Range param_range)
- {
- if (!MoveTo(Table.ParamPtr, param_range.Start))
- {
- if (!MoveTo(Table.Param, param_range.Start))
- return;
- for (uint i = 0; i < param_range.Length; i++)
- ReadParameter(param_range.Start + i, method);
- }
- else
- ReadParameterPointers(method, param_range);
- }
- void ReadParameterPointers(MethodDefinition method, Range range)
- {
- for (uint i = 0; i < range.Length; i++)
- {
- MoveTo(Table.ParamPtr, range.Start + i);
- var rid = ReadTableIndex(Table.Param);
- MoveTo(Table.Param, rid);
- ReadParameter(rid, method);
- }
- }
- void ReadParameter(uint param_rid, MethodDefinition method)
- {
- var attributes = (ParameterAttributes)ReadUInt16();
- var sequence = ReadUInt16();
- var name = ReadString();
- var parameter = sequence == 0
- ? method.MethodReturnType.Parameter
- : method.Parameters[sequence - 1];
- parameter.token = new MetadataToken(TokenType.Param, param_rid);
- parameter.Name = name;
- parameter.Attributes = attributes;
- }
- void ReadMethodSignature(uint signature, IMethodSignature method)
- {
- var reader = ReadSignature(signature);
- reader.ReadMethodSignature(method);
- }
- public PInvokeInfo ReadPInvokeInfo(MethodDefinition method)
- {
- InitializePInvokes();
- Row<PInvokeAttributes, uint, uint> row;
- var rid = method.token.RID;
- if (!metadata.PInvokes.TryGetValue(rid, out row))
- return null;
- metadata.PInvokes.Remove(rid);
- return new PInvokeInfo(
- row.Col1,
- image.StringHeap.Read(row.Col2),
- module.ModuleReferences[(int)row.Col3 - 1]);
- }
- void InitializePInvokes()
- {
- if (metadata.PInvokes != null)
- return;
- int length = MoveTo(Table.ImplMap);
- var pinvokes = metadata.PInvokes = new Dictionary<uint, Row<PInvokeAttributes, uint, uint>>(length);
- for (int i = 1; i <= length; i++)
- {
- var attributes = (PInvokeAttributes)ReadUInt16();
- var method = ReadMetadataToken(CodedIndex.MemberForwarded);
- var name = ReadStringIndex();
- var scope = ReadTableIndex(Table.File);
- if (method.TokenType != TokenType.Method)
- continue;
- pinvokes.Add(method.RID, new Row<PInvokeAttributes, uint, uint>(attributes, name, scope));
- }
- }
- public bool HasGenericParameters(IGenericParameterProvider provider)
- {
- InitializeGenericParameters();
- Range range;
- if (!metadata.TryGetGenericParameterRange(provider, out range))
- return false;
- return range.Length > 0;
- }
- public Collection<GenericParameter> ReadGenericParameters(IGenericParameterProvider provider)
- {
- InitializeGenericParameters();
- Range range;
- if (!metadata.TryGetGenericParameterRange(provider, out range)
- || !MoveTo(Table.GenericParam, range.Start))
- return new GenericParameterCollection(provider);
- metadata.RemoveGenericParameterRange(provider);
- var generic_parameters = new GenericParameterCollection(provider, (int)range.Length);
- for (uint i = 0; i < range.Length; i++)
- {
- ReadUInt16(); // index
- var flags = (GenericParameterAttributes)ReadUInt16();
- ReadMetadataToken(CodedIndex.TypeOrMethodDef);
- var name = ReadString();
- var parameter = new GenericParameter(name, provider);
- parameter.token = new MetadataToken(TokenType.GenericParam, range.Start + i);
- parameter.Attributes = flags;
- generic_parameters.Add(parameter);
- }
- return generic_parameters;
- }
- void InitializeGenericParameters()
- {
- if (metadata.GenericParameters != null)
- return;
- metadata.GenericParameters = InitializeRanges(
- Table.GenericParam, () =>
- {
- Advance(4);
- var next = ReadMetadataToken(CodedIndex.TypeOrMethodDef);
- ReadStringIndex();
- return next;
- });
- }
- Dictionary<MetadataToken, Range> InitializeRanges(Table table, Func<MetadataToken> get_next)
- {
- int length = MoveTo(table);
- var ranges = new Dictionary<MetadataToken, Range>(length);
- if (length == 0)
- return ranges;
- MetadataToken owner = MetadataToken.Zero;
- Range range = new Range(1, 0);
- for (uint i = 1; i <= length; i++)
- {
- var next = get_next();
- if (i == 1)
- {
- owner = next;
- range.Length++;
- }
- else if (next != owner)
- {
- if (owner.RID != 0)
- ranges.Add(owner, range);
- range = new Range(i, 1);
- owner = next;
- }
- else
- range.Length++;
- }
- if (owner != MetadataToken.Zero && !ranges.ContainsKey(owner))
- ranges.Add(owner, range);
- return ranges;
- }
- public bool HasGenericConstraints(GenericParameter generic_parameter)
- {
- InitializeGenericConstraints();
- MetadataToken[] mapping;
- if (!metadata.TryGetGenericConstraintMapping(generic_parameter, out mapping))
- return false;
- return mapping.Length > 0;
- }
- public Collection<TypeReference> ReadGenericConstraints(GenericParameter generic_parameter)
- {
- InitializeGenericConstraints();
- MetadataToken[] mapping;
- if (!metadata.TryGetGenericConstraintMapping(generic_parameter, out mapping))
- return new Collection<TypeReference>();
- var constraints = new Collection<TypeReference>(mapping.Length);
- this.context = (IGenericContext)generic_parameter.Owner;
- for (int i = 0; i < mapping.Length; i++)
- constraints.Add(GetTypeDefOrRef(mapping[i]));
- metadata.RemoveGenericConstraintMapping(generic_parameter);
- return constraints;
- }
- void InitializeGenericConstraints()
- {
- if (metadata.GenericConstraints != null)
- return;
- var length = MoveTo(Table.GenericParamConstraint);
- metadata.GenericConstraints = new Dictionary<uint, MetadataToken[]>(length);
- for (int i = 1; i <= length; i++)
- AddGenericConstraintMapping(
- ReadTableIndex(Table.GenericParam),
- ReadMetadataToken(CodedIndex.TypeDefOrRef));
- }
- void AddGenericConstraintMapping(uint generic_parameter, MetadataToken constraint)
- {
- metadata.SetGenericConstraintMapping(
- generic_parameter,
- AddMapping(metadata.GenericConstraints, generic_parameter, constraint));
- }
- public bool HasOverrides(MethodDefinition method)
- {
- InitializeOverrides();
- MetadataToken[] mapping;
- if (!metadata.TryGetOverrideMapping(method, out mapping))
- return false;
- return mapping.Length > 0;
- }
- public Collection<MethodReference> ReadOverrides(MethodDefinition method)
- {
- InitializeOverrides();
- MetadataToken[] mapping;
- if (!metadata.TryGetOverrideMapping(method, out mapping))
- return new Collection<MethodReference>();
- var overrides = new Collection<MethodReference>(mapping.Length);
- this.context = method;
- for (int i = 0; i < mapping.Length; i++)
- overrides.Add((MethodReference)LookupToken(mapping[i]));
- metadata.RemoveOverrideMapping(method);
- return overrides;
- }
- void InitializeOverrides()
- {
- if (metadata.Overrides != null)
- return;
- var length = MoveTo(Table.MethodImpl);
- metadata.Overrides = new Dictionary<uint, MetadataToken[]>(length);
- for (int i = 1; i <= length; i++)
- {
- ReadTableIndex(Table.TypeDef);
- var method = ReadMetadataToken(CodedIndex.MethodDefOrRef);
- if (method.TokenType != TokenType.Method)
- throw new NotSupportedException();
- var @override = ReadMetadataToken(CodedIndex.MethodDefOrRef);
- AddOverrideMapping(method.RID, @override);
- }
- }
- void AddOverrideMapping(uint method_rid, MetadataToken @override)
- {
- metadata.SetOverrideMapping(
- method_rid,
- AddMapping(metadata.Overrides, method_rid, @override));
- }
- public MethodBody ReadMethodBody(MethodDefinition method)
- {
- return code.ReadMethodBody(method);
- }
- public CallSite ReadCallSite(MetadataToken token)
- {
- if (!MoveTo(Table.StandAloneSig, token.RID))
- return null;
- var signature = ReadBlobIndex();
- var call_site = new CallSite();
- ReadMethodSignature(signature, call_site);
- call_site.MetadataToken = token;
- return call_site;
- }
- public VariableDefinitionCollection ReadVariables(MetadataToken local_var_token)
- {
- if (!MoveTo(Table.StandAloneSig, local_var_token.RID))
- return null;
- var reader = ReadSignature(ReadBlobIndex());
- const byte local_sig = 0x7;
- if (reader.ReadByte() != local_sig)
- throw new NotSupportedException();
- var count = reader.ReadCompressedUInt32();
- if (count == 0)
- return null;
- var variables = new VariableDefinitionCollection((int)count);
- for (int i = 0; i < count; i++)
- variables.Add(new VariableDefinition(reader.ReadTypeSignature()));
- return variables;
- }
- public IMetadataTokenProvider LookupToken(MetadataToken token)
- {
- var rid = token.RID;
- if (rid == 0)
- return null;
- IMetadataTokenProvider element;
- var position = this.position;
- var context = this.context;
- switch (token.TokenType)
- {
- case TokenType.TypeDef:
- element = GetTypeDefinition(rid);
- break;
- case TokenType.TypeRef:
- element = GetTypeReference(rid);
- break;
- case TokenType.TypeSpec:
- element = GetTypeSpecification(rid);
- break;
- case TokenType.Field:
- element = GetFieldDefinition(rid);
- break;
- case TokenType.Method:
- element = GetMethodDefinition(rid);
- break;
- case TokenType.MemberRef:
- element = GetMemberReference(rid);
- break;
- case TokenType.MethodSpec:
- element = GetMethodSpecification(rid);
- break;
- default:
- return null;
- }
- this.position = position;
- this.context = context;
- return element;
- }
- public FieldDefinition GetFieldDefinition(uint rid)
- {
- InitializeTypeDefinitions();
- var field = metadata.GetFieldDefinition(rid);
- if (field != null)
- return field;
- return LookupField(rid);
- }
- FieldDefinition LookupField(uint rid)
- {
- var type = metadata.GetFieldDeclaringType(rid);
- if (type == null)
- return null;
- InitializeCollection(type.Fields);
- return metadata.GetFieldDefinition(rid);
- }
- public MethodDefinition GetMethodDefinition(uint rid)
- {
- InitializeTypeDefinitions();
- var method = metadata.GetMethodDefinition(rid);
- if (method != null)
- return method;
- return LookupMethod(rid);
- }
- MethodDefinition LookupMethod(uint rid)
- {
- var type = metadata.GetMethodDeclaringType(rid);
- if (type == null)
- return null;
- InitializeCollection(type.Methods);
- return metadata.GetMethodDefinition(rid);
- }
- MethodSpecification GetMethodSpecification(uint rid)
- {
- if (!MoveTo(Table.MethodSpec, rid))
- return null;
- var element_method = (MethodReference)LookupToken(
- ReadMetadataToken(CodedIndex.MethodDefOrRef));
- var signature = ReadBlobIndex();
- var method_spec = ReadMethodSpecSignature(signature, element_method);
- method_spec.token = new MetadataToken(TokenType.MethodSpec, rid);
- return method_spec;
- }
- MethodSpecification ReadMethodSpecSignature(uint signature, MethodReference method)
- {
- var reader = ReadSignature(signature);
- const byte methodspec_sig = 0x0a;
- var call_conv = reader.ReadByte();
- if (call_conv != methodspec_sig)
- throw new NotSupportedException();
- var instance = new GenericInstanceMethod(method);
- reader.ReadGenericInstanceSignature(method, instance);
- return instance;
- }
- MemberReference GetMemberReference(uint rid)
- {
- InitializeMemberReferences();
- var member = metadata.GetMemberReference(rid);
- if (member != null)
- return member;
- member = ReadMemberReference(rid);
- if (member != null && !member.ContainsGenericParameter)
- metadata.AddMemberReference(member);
- return member;
- }
- MemberReference ReadMemberReference(uint rid)
- {
- if (!MoveTo(Table.MemberRef, rid))
- return null;
- var token = ReadMetadataToken(CodedIndex.MemberRefParent);
- var name = ReadString();
- var signature = ReadBlobIndex();
- MemberReference member;
- switch (token.TokenType)
- {
- case TokenType.TypeDef:
- case TokenType.TypeRef:
- case TokenType.TypeSpec:
- member = ReadTypeMemberReference(token, name, signature);
- break;
- case TokenType.Method:
- member = ReadMethodMemberReference(token, name, signature);
- break;
- default:
- throw new NotSupportedException();
- }
- member.token = new MetadataToken(TokenType.MemberRef, rid);
- return member;
- }
- MemberReference ReadTypeMemberReference(MetadataToken type, string name, uint signature)
- {
- var declaring_type = GetTypeDefOrRef(type);
- if (!declaring_type.IsArray)
- this.context = declaring_type;
- var member = ReadMemberReferenceSignature(signature, declaring_type);
- member.Name = name;
- return member;
- }
- MemberReference ReadMemberReferenceSignature(uint signature, TypeReference declaring_type)
- {
- var reader = ReadSignature(signature);
- const byte field_sig = 0x6;
- if (reader.buffer[reader.position] == field_sig)
- {
- reader.position++;
- var field = new FieldReference();
- field.DeclaringType = declaring_type;
- field.FieldType = reader.ReadTypeSignature();
- return field;
- }
- else
- {
- var method = new MethodReference();
- method.DeclaringType = declaring_type;
- reader.ReadMethodSignature(method);
- return method;
- }
- }
- MemberReference ReadMethodMemberReference(MetadataToken token, string name, uint signature)
- {
- var method = GetMethodDefinition(token.RID);
- this.context = method;
- var member = ReadMemberReferenceSignature(signature, method.DeclaringType);
- member.Name = name;
- return member;
- }
- void InitializeMemberReferences()
- {
- if (metadata.MemberReferences != null)
- return;
- metadata.MemberReferences = new MemberReference[image.GetTableLength(Table.MemberRef)];
- }
- public IEnumerable<MemberReference> GetMemberReferences()
- {
- InitializeMemberReferences();
- var length = image.GetTableLength(Table.MemberRef);
- var type_system = module.TypeSystem;
- var context = new MethodReference(string.Empty, type_system.Void);
- context.DeclaringType = new TypeReference(string.Empty, string.Empty, module, type_system.Corlib);
- var member_references = new MemberReference[length];
- for (uint i = 1; i <= length; i++)
- {
- this.context = context;
- member_references[i - 1] = GetMemberReference(i);
- }
- return member_references;
- }
- void InitializeConstants()
- {
- if (metadata.Constants != null)
- return;
- var length = MoveTo(Table.Constant);
- var constants = metadata.Constants = new Dictionary<MetadataToken, Row<ElementType, uint>>(length);
- for (uint i = 1; i <= length; i++)
- {
- var type = (ElementType)ReadUInt16();
- var owner = ReadMetadataToken(CodedIndex.HasConstant);
- var signature = ReadBlobIndex();
- constants.Add(owner, new Row<ElementType, uint>(type, signature));
- }
- }
- public object ReadConstant(IConstantProvider owner)
- {
- InitializeConstants();
- Row<ElementType, uint> row;
- if (!metadata.Constants.TryGetValue(owner.MetadataToken, out row))
- return Mixin.NoValue;
- metadata.Constants.Remove(owner.MetadataToken);
- switch (row.Col1)
- {
- case ElementType.Class:
- case ElementType.Object:
- return null;
- case ElementType.String:
- return ReadConstantString(ReadBlob(row.Col2));
- default:
- return ReadConstantPrimitive(row.Col1, row.Col2);
- }
- }
- static string ReadConstantString(byte[] blob)
- {
- var length = blob.Length;
- if ((length & 1) == 1)
- length--;
- return Encoding.Unicode.GetString(blob, 0, length);
- }
- object ReadConstantPrimitive(ElementType type, uint signature)
- {
- var reader = ReadSignature(signature);
- return reader.ReadConstantSignature(type);
- }
- void InitializeCustomAttributes()
- {
- if (metadata.CustomAttributes != null)
- return;
- metadata.CustomAttributes = InitializeRanges(
- Table.CustomAttribute, () =>
- {
- var next = ReadMetadataToken(CodedIndex.HasCustomAttribute);
- ReadMetadataToken(CodedIndex.CustomAttributeType);
- ReadBlobIndex();
- return next;
- });
- }
- public bool HasCustomAttributes(ICustomAttributeProvider owner)
- {
- InitializeCustomAttributes();
- Range range;
- if (!metadata.TryGetCustomAttributeRange(owner, out range))
- return false;
- return range.Length > 0;
- }
- public Collection<CustomAttribute> ReadCustomAttributes(ICustomAttributeProvider owner)
- {
- InitializeCustomAttributes();
- Range range;
- if (!metadata.TryGetCustomAttributeRange(owner, out range)
- || !MoveTo(Table.CustomAttribute, range.Start))
- return new Collection<CustomAttribute>();
- var custom_attributes = new Collection<CustomAttribute>((int)range.Length);
- for (int i = 0; i < range.Length; i++)
- {
- ReadMetadataToken(CodedIndex.HasCustomAttribute);
- var constructor = (MethodReference)LookupToken(
- ReadMetadataToken(CodedIndex.CustomAttributeType));
- var signature = ReadBlobIndex();
- custom_attributes.Add(new CustomAttribute(signature, constructor));
- }
- metadata.RemoveCustomAttributeRange(owner);
- return custom_attributes;
- }
- public byte[] ReadCustomAttributeBlob(uint signature)
- {
- return ReadBlob(signature);
- }
- public void ReadCustomAttributeSignature(CustomAttribute attribute)
- {
- var reader = ReadSignature(attribute.signature);
- if (!reader.CanReadMore())
- return;
- if (reader.ReadUInt16() != 0x0001)
- throw new InvalidOperationException();
- var constructor = attribute.Constructor;
- if (constructor.HasParameters)
- reader.ReadCustomAttributeConstructorArguments(attribute, constructor.Parameters);
- if (!reader.CanReadMore())
- return;
- var named = reader.ReadUInt16();
- if (named == 0)
- return;
- reader.ReadCustomAttributeNamedArguments(named, ref attribute.fields, ref attribute.properties);
- }
- void InitializeMarshalInfos()
- {
- if (metadata.FieldMarshals != null)
- return;
- var length = MoveTo(Table.FieldMarshal);
- var marshals = metadata.FieldMarshals = new Dictionary<MetadataToken, uint>(length);
- for (int i = 0; i < length; i++)
- {
- var token = ReadMetadataToken(CodedIndex.HasFieldMarshal);
- var signature = ReadBlobIndex();
- if (token.RID == 0)
- continue;
- marshals.Add(token, signature);
- }
- }
- public bool HasMarshalInfo(IMarshalInfoProvider owner)
- {
- InitializeMarshalInfos();
- return metadata.FieldMarshals.ContainsKey(owner.MetadataToken);
- }
- public MarshalInfo ReadMarshalInfo(IMarshalInfoProvider owner)
- {
- InitializeMarshalInfos();
- uint signature;
- if (!metadata.FieldMarshals.TryGetValue(owner.MetadataToken, out signature))
- return null;
- var reader = ReadSignature(signature);
- metadata.FieldMarshals.Remove(owner.MetadataToken);
- return reader.ReadMarshalInfo();
- }
- void InitializeSecurityDeclarations()
- {
- if (metadata.SecurityDeclarations != null)
- return;
- metadata.SecurityDeclarations = InitializeRanges(
- Table.DeclSecurity, () =>
- {
- ReadUInt16();
- var next = ReadMetadataToken(CodedIndex.HasDeclSecurity);
- ReadBlobIndex();
- return next;
- });
- }
- public bool HasSecurityDeclarations(ISecurityDeclarationProvider owner)
- {
- InitializeSecurityDeclarations();
- Range range;
- if (!metadata.TryGetSecurityDeclarationRange(owner, out range))
- return false;
- return range.Length > 0;
- }
- public Collection<SecurityDeclaration> ReadSecurityDeclarations(ISecurityDeclarationProvider owner)
- {
- InitializeSecurityDeclarations();
- Range range;
- if (!metadata.TryGetSecurityDeclarationRange(owner, out range)
- || !MoveTo(Table.DeclSecurity, range.Start))
- return new Collection<SecurityDeclaration>();
- var security_declarations = new Collection<SecurityDeclaration>((int)range.Length);
- for (int i = 0; i < range.Length; i++)
- {
- var action = (SecurityAction)ReadUInt16();
- ReadMetadataToken(CodedIndex.HasDeclSecurity);
- var signature = ReadBlobIndex();
- security_declarations.Add(new SecurityDeclaration(action, signature, module));
- }
- metadata.RemoveSecurityDeclarationRange(owner);
- return security_declarations;
- }
- public byte[] ReadSecurityDeclarationBlob(uint signature)
- {
- return ReadBlob(signature);
- }
- public void ReadSecurityDeclarationSignature(SecurityDeclaration declaration)
- {
- var signature = declaration.signature;
- var reader = ReadSignature(signature);
- if (reader.buffer[reader.position] != '.')
- {
- ReadXmlSecurityDeclaration(signature, declaration);
- return;
- }
- reader.position++;
- var count = reader.ReadCompressedUInt32();
- var attributes = new Collection<SecurityAttribute>((int)count);
- for (int i = 0; i < count; i++)
- attributes.Add(reader.ReadSecurityAttribute());
- declaration.security_attributes = attributes;
- }
- void ReadXmlSecurityDeclaration(uint signature, SecurityDeclaration declaration)
- {
- var blob = ReadBlob(signature);
- var attributes = new Collection<SecurityAttribute>(1);
- var attribute = new SecurityAttribute(
- module.TypeSystem.LookupType("System.Security.Permissions", "PermissionSetAttribute"));
- attribute.properties = new Collection<CustomAttributeNamedArgument>(1);
- attribute.properties.Add(
- new CustomAttributeNamedArgument(
- "XML",
- new CustomAttributeArgument(
- module.TypeSystem.String,
- Encoding.Unicode.GetString(blob, 0, blob.Length))));
- attributes.Add(attribute);
- declaration.security_attributes = attributes;
- }
- public Collection<ExportedType> ReadExportedTypes()
- {
- var length = MoveTo(Table.ExportedType);
- if (length == 0)
- return new Collection<ExportedType>();
- var exported_types = new Collection<ExportedType>(length);
- for (int i = 1; i <= length; i++)
- {
- var attributes = (TypeAttributes)ReadUInt32();
- var identifier = ReadUInt32();
- var name = ReadString();
- var @namespace = ReadString();
- var implementation = ReadMetadataToken(CodedIndex.Implementation);
- ExportedType declaring_type = null;
- IMetadataScope scope = null;
- switch (implementation.TokenType)
- {
- case TokenType.AssemblyRef:
- case TokenType.File:
- scope = GetExportedTypeScope(implementation);
- break;
- case TokenType.ExportedType:
- // FIXME: if the table is not properly sorted
- declaring_type = exported_types[(int)implementation.RID - 1];
- break;
- }
- var exported_type = new ExportedType(@namespace, name, module, scope)
- {
- Attributes = attributes,
- Identifier = (int)identifier,
- DeclaringType = declaring_type,
- };
- exported_type.token = new MetadataToken(TokenType.ExportedType, i);
- exported_types.Add(exported_type);
- }
- return exported_types;
- }
- IMetadataScope GetExportedTypeScope(MetadataToken token)
- {
- var position = this.position;
- IMetadataScope scope;
- switch (token.TokenType)
- {
- case TokenType.AssemblyRef:
- InitializeAssemblyReferences();
- scope = metadata.AssemblyReferences[(int)token.RID - 1];
- break;
- case TokenType.File:
- InitializeModuleReferences();
- scope = GetModuleReferenceFromFile(token);
- break;
- default:
- throw new NotSupportedException();
- }
- this.position = position;
- return scope;
- }
- ModuleReference GetModuleReferenceFromFile(MetadataToken token)
- {
- if (!MoveTo(Table.File, token.RID))
- return null;
- ReadUInt32();
- var file_name = ReadString();
- var modules = module.ModuleReferences;
- ModuleReference reference;
- for (int i = 0; i < modules.Count; i++)
- {
- reference = modules[i];
- if (reference.Name == file_name)
- return reference;
- }
- reference = new ModuleReference(file_name);
- modules.Add(reference);
- return reference;
- }
- static void InitializeCollection(object o)
- {
- }
- }
- sealed class SignatureReader : ByteBuffer
- {
- readonly MetadataReader reader;
- readonly uint start, sig_length;
- TypeSystem TypeSystem
- {
- get { return reader.module.TypeSystem; }
- }
- public SignatureReader(uint blob, MetadataReader reader)
- : base(reader.buffer)
- {
- this.reader = reader;
- MoveToBlob(blob);
- this.sig_length = ReadCompressedUInt32();
- this.start = (uint)position;
- }
- void MoveToBlob(uint blob)
- {
- position = (int)(reader.image.BlobHeap.Offset + blob);
- }
- MetadataToken ReadTypeTokenSignature()
- {
- return Mixin.GetMetadataToken(CodedIndex.TypeDefOrRef, ReadCompressedUInt32());
- }
- GenericParameter GetGenericParameter(GenericParameterType type, uint var)
- {
- var context = reader.context;
- int index = (int)var;
- if (context == null)
- return GetUnboundGenericParameter(type, index);
- IGenericParameterProvider provider;
- switch (type)
- {
- case GenericParameterType.Type:
- provider = context.Type;
- break;
- case GenericParameterType.Method:
- provider = context.Method;
- break;
- default:
- throw new NotSupportedException();
- }
- if (!context.IsDefinition)
- CheckGenericContext(provider, index);
- if (index >= provider.GenericParameters.Count)
- return GetUnboundGenericParameter(type, index);
- return provider.GenericParameters[index];
- }
- GenericParameter GetUnboundGenericParameter(GenericParameterType type, int index)
- {
- return new GenericParameter(index, type, reader.module);
- }
- static void CheckGenericContext(IGenericParameterProvider owner, int index)
- {
- var owner_parameters = owner.GenericParameters;
- for (int i = owner_parameters.Count; i <= index; i++)
- owner_parameters.Add(new GenericParameter(owner));
- }
- public void ReadGenericInstanceSignature(IGenericParameterProvider provider, IGenericInstance instance)
- {
- var arity = ReadCompressedUInt32();
- if (!provider.IsDefinition)
- CheckGenericContext(provider, (int)arity - 1);
- var instance_arguments = instance.GenericArguments;
- for (int i = 0; i < arity; i++)
- instance_arguments.Add(ReadTypeSignature());
- }
- ArrayType ReadArrayTypeSignature()
- {
- var array = new ArrayType(ReadTypeSignature());
- var rank = ReadCompressedUInt32();
- var sizes = new uint[ReadCompressedUInt32()];
- for (int i = 0; i < sizes.Length; i++)
- sizes[i] = ReadCompressedUInt32();
- var low_bounds = new int[ReadCompressedUInt32()];
- for (int i = 0; i < low_bounds.Length; i++)
- low_bounds[i] = ReadCompressedInt32();
- array.Dimensions.Clear();
- for (int i = 0; i < rank; i++)
- {
- int? lower = null, upper = null;
- if (i < low_bounds.Length)
- lower = low_bounds[i];
- if (i < sizes.Length)
- upper = lower + (int)sizes[i] - 1;
- array.Dimensions.Add(new ArrayDimension(lower, upper));
- }
- return array;
- }
- TypeReference GetTypeDefOrRef(MetadataToken token)
- {
- return reader.GetTypeDefOrRef(token);
- }
- public TypeReference ReadTypeSignature()
- {
- return ReadTypeSignature((ElementType)ReadByte());
- }
- TypeReference ReadTypeSignature(ElementType etype)
- {
- switch (etype)
- {
- case ElementType.ValueType:
- {
- var value_type = GetTypeDefOrRef(ReadTypeTokenSignature());
- value_type.IsValueType = true;
- return value_type;
- }
- case ElementType.Class:
- return GetTypeDefOrRef(ReadTypeTokenSignature());
- case ElementType.Ptr:
- return new PointerType(ReadTypeSignature());
- case ElementType.FnPtr:
- {
- var fptr = new FunctionPointerType();
- ReadMethodSignature(fptr);
- return fptr;
- }
- case ElementType.ByRef:
- return new ByReferenceType(ReadTypeSignature());
- case ElementType.Pinned:
- return new PinnedType(ReadTypeSignature());
- case ElementType.SzArray:
- return new ArrayType(ReadTypeSignature());
- case ElementType.Array:
- return ReadArrayTypeSignature();
- case ElementType.CModOpt:
- return new OptionalModifierType(
- GetTypeDefOrRef(ReadTypeTokenSignature()), ReadTypeSignature());
- case ElementType.CModReqD:
- return new RequiredModifierType(
- GetTypeDefOrRef(ReadTypeTokenSignature()), ReadTypeSignature());
- case ElementType.Sentinel:
- return new SentinelType(ReadTypeSignature());
- case ElementType.Var:
- return GetGenericParameter(GenericParameterType.Type, ReadCompressedUInt32());
- case ElementType.MVar:
- return GetGenericParameter(GenericParameterType.Method, ReadCompressedUInt32());
- case ElementType.GenericInst:
- {
- var is_value_type = ReadByte() == (byte)ElementType.ValueType;
- var element_type = GetTypeDefOrRef(ReadTypeTokenSignature());
- var generic_instance = new GenericInstanceType(element_type);
- ReadGenericInstanceSignature(element_type, generic_instance);
- if (is_value_type)
- {
- generic_instance.IsValueType = true;
- element_type.GetElementType().IsValueType = true;
- }
- return generic_instance;
- }
- case ElementType.Object: return TypeSystem.Object;
- case ElementType.Void: return TypeSystem.Void;
- case ElementType.TypedByRef: return TypeSystem.TypedReference;
- case ElementType.I: return TypeSystem.IntPtr;
- case ElementType.U: return TypeSystem.UIntPtr;
- default: return GetPrimitiveType(etype);
- }
- }
- public void ReadMethodSignature(IMethodSignature method)
- {
- var calling_convention = ReadByte();
- const byte has_this = 0x20;
- const byte explicit_this = 0x40;
- if ((calling_convention & has_this) != 0)
- {
- method.HasThis = true;
- calling_convention = (byte)(calling_convention & ~has_this);
- }
- if ((calling_convention & explicit_this) != 0)
- {
- method.ExplicitThis = true;
- calling_convention = (byte)(calling_convention & ~explicit_this);
- }
- method.CallingConvention = (MethodCallingConvention)calling_convention;
- var generic_context = method as MethodReference;
- if (generic_context != null && !generic_context.DeclaringType.IsArray)
- reader.context = generic_context;
- if ((calling_convention & 0x10) != 0)
- {
- var arity = ReadCompressedUInt32();
- if (generic_context != null && !generic_context.IsDefinition)
- CheckGenericContext(generic_context, (int)arity - 1);
- }
- var param_count = ReadCompressedUInt32();
- method.MethodReturnType.ReturnType = ReadTypeSignature();
- if (param_count == 0)
- return;
- Collection<ParameterDefinition> parameters;
- var method_ref = method as MethodReference;
- if (method_ref != null)
- parameters = method_ref.parameters = new ParameterDefinitionCollection(method, (int)param_count);
- else
- parameters = method.Parameters;
- for (int i = 0; i < param_count; i++)
- parameters.Add(new ParameterDefinition(ReadTypeSignature()));
- }
- public object ReadConstantSignature(ElementType type)
- {
- return ReadPrimitiveValue(type);
- }
- public void ReadCustomAttributeConstructorArguments(CustomAttribute attribute, Collection<ParameterDefinition> parameters)
- {
- var count = parameters.Count;
- if (count == 0)
- return;
- attribute.arguments = new Collection<CustomAttributeArgument>(count);
- for (int i = 0; i < count; i++)
- attribute.arguments.Add(
- ReadCustomAttributeFixedArgument(parameters[i].ParameterType));
- }
- CustomAttributeArgument ReadCustomAttributeFixedArgument(TypeReference type)
- {
- if (type.IsArray)
- return ReadCustomAttributeFixedArrayArgument((ArrayType)type);
- return ReadCustomAttributeElement(type);
- }
- public void ReadCustomAttributeNamedArguments(ushort count, ref Collection<CustomAttributeNamedArgument> fields, ref Collection<CustomAttributeNamedArgument> properties)
- {
- for (int i = 0; i < count; i++)
- ReadCustomAttributeNamedArgument(ref fields, ref properties);
- }
- void ReadCustomAttributeNamedArgument(ref Collection<CustomAttributeNamedArgument> fields, ref Collection<CustomAttributeNamedArgument> properties)
- {
- var kind = ReadByte();
- var type = ReadCustomAttributeFieldOrPropType();
- var name = ReadUTF8String();
- Collection<CustomAttributeNamedArgument> container;
- switch (kind)
- {
- case 0x53:
- container = GetCustomAttributeNamedArgumentCollection(ref fields);
- break;
- case 0x54:
- container = GetCustomAttributeNamedArgumentCollection(ref properties);
- break;
- default:
- throw new NotSupportedException();
- }
- container.Add(new CustomAttributeNamedArgument(name, ReadCustomAttributeFixedArgument(type)));
- }
- static Collection<CustomAttributeNamedArgument> GetCustomAttributeNamedArgumentCollection(ref Collection<CustomAttributeNamedArgument> collection)
- {
- if (collection != null)
- return collection;
- return collection = new Collection<CustomAttributeNamedArgument>();
- }
- CustomAttributeArgument ReadCustomAttributeFixedArrayArgument(ArrayType type)
- {
- var length = ReadUInt32();
- if (length == 0xffffffff)
- return new CustomAttributeArgument(type, null);
- if (length == 0)
- return new CustomAttributeArgument(type, Empty<CustomAttributeArgument>.Array);
- var arguments = new CustomAttributeArgument[length];
- var element_type = type.ElementType;
- for (int i = 0; i < length; i++)
- arguments[i] = ReadCustomAttributeElement(element_type);
- return new CustomAttributeArgument(type, arguments);
- }
- CustomAttributeArgument ReadCustomAttributeElement(TypeReference type)
- {
- if (type.IsArray)
- return ReadCustomAttributeFixedArrayArgument((ArrayType)type);
- return new CustomAttributeArgument(
- type,
- type.etype == ElementType.Object
- ? ReadCustomAttributeElement(ReadCustomAttributeFieldOrPropType())
- : ReadCustomAttributeElementValue(type));
- }
- object ReadCustomAttributeElementValue(TypeReference type)
- {
- var etype = type.etype;
- switch (etype)
- {
- case ElementType.String:
- return ReadUTF8String();
- case ElementType.None:
- if (Mixin.IsTypeOf(type,"System", "Type"))
- return ReadTypeReference();
- return ReadCustomAttributeEnum(type);
- default:
- return ReadPrimitiveValue(etype);
- }
- }
- object ReadPrimitiveValue(ElementType type)
- {
- switch (type)
- {
- case ElementType.Boolean:
- return ReadByte() == 1;
- case ElementType.I1:
- return (sbyte)ReadByte();
- case ElementType.U1:
- return ReadByte();
- case ElementType.Char:
- return (char)ReadUInt16();
- case ElementType.I2:
- return ReadInt16();
- case ElementType.U2:
- return ReadUInt16();
- case ElementType.I4:
- return ReadInt32();
- case ElementType.U4:
- return ReadUInt32();
- case ElementType.I8:
- return ReadInt64();
- case ElementType.U8:
- return ReadUInt64();
- case ElementType.R4:
- return ReadSingle();
- case ElementType.R8:
- return ReadDouble();
- default:
- throw new NotImplementedException(type.ToString());
- }
- }
- TypeReference GetPrimitiveType(ElementType etype)
- {
- switch (etype)
- {
- case ElementType.Boolean:
- return TypeSystem.Boolean;
- case ElementType.Char:
- return TypeSystem.Char;
- case ElementType.I1:
- return TypeSystem.SByte;
- case ElementType.U1:
- return TypeSystem.Byte;
- case ElementType.I2:
- return TypeSystem.Int16;
- case ElementType.U2:
- return TypeSystem.UInt16;
- case ElementType.I4:
- return TypeSystem.Int32;
- case ElementType.U4:
- return TypeSystem.UInt32;
- case ElementType.I8:
- return TypeSystem.Int64;
- case ElementType.U8:
- return TypeSystem.UInt64;
- case ElementType.R4:
- return TypeSystem.Single;
- case ElementType.R8:
- return TypeSystem.Double;
- case ElementType.String:
- return TypeSystem.String;
- default:
- throw new NotImplementedException(etype.ToString());
- }
- }
- TypeReference ReadCustomAttributeFieldOrPropType()
- {
- var etype = (ElementType)ReadByte();
- switch (etype)
- {
- case ElementType.Boxed:
- return TypeSystem.Object;
- case ElementType.SzArray:
- return new ArrayType(ReadCustomAttributeFieldOrPropType());
- case ElementType.Enum:
- return ReadTypeReference();
- case ElementType.Type:
- return TypeSystem.LookupType("System", "Type");
- default:
- return GetPrimitiveType(etype);
- }
- }
- public TypeReference ReadTypeReference()
- {
- return TypeParser.ParseType(reader.module, ReadUTF8String());
- }
- object ReadCustomAttributeEnum(TypeReference enum_type)
- {
- var type = Mixin.CheckedResolve(enum_type);
- if (!type.IsEnum)
- throw new ArgumentException();
- return ReadCustomAttributeElementValue(Mixin.GetEnumUnderlyingType(type));
- }
- public SecurityAttribute ReadSecurityAttribute()
- {
- var attribute = new SecurityAttribute(ReadTypeReference());
- ReadCompressedUInt32();
- ReadCustomAttributeNamedArguments(
- (ushort)ReadCompressedUInt32(),
- ref attribute.fields,
- ref attribute.properties);
- return attribute;
- }
- public MarshalInfo ReadMarshalInfo()
- {
- var native = ReadNativeType();
- switch (native)
- {
- case NativeType.Array:
- {
- var array = new ArrayMarshalInfo();
- if (CanReadMore())
- array.element_type = ReadNativeType();
- if (CanReadMore())
- array.size_parameter_index = (int)ReadCompressedUInt32();
- if (CanReadMore())
- array.size = (int)ReadCompressedUInt32();
- if (CanReadMore())
- array.size_parameter_multiplier = (int)ReadCompressedUInt32();
- return array;
- }
- case NativeType.SafeArray:
- {
- var array = new SafeArrayMarshalInfo();
- if (CanReadMore())
- array.element_type = ReadVariantType();
- return array;
- }
- case NativeType.FixedArray:
- {
- var array = new FixedArrayMarshalInfo();
- if (CanReadMore())
- array.size = (int)ReadCompressedUInt32();
- if (CanReadMore())
- array.element_type = ReadNativeType();
- return array;
- }
- case NativeType.FixedSysString:
- {
- var sys_string = new FixedSysStringMarshalInfo();
- if (CanReadMore())
- sys_string.size = (int)ReadCompressedUInt32();
- return sys_string;
- }
- case NativeType.CustomMarshaler:
- {
- var marshaler = new CustomMarshalInfo();
- var guid_value = ReadUTF8String();
- marshaler.guid = !string.IsNullOrEmpty(guid_value) ? new Guid(guid_value) : Guid.Empty;
- marshaler.unmanaged_type = ReadUTF8String();
- marshaler.managed_type = ReadTypeReference();
- marshaler.cookie = ReadUTF8String();
- return marshaler;
- }
- default:
- return new MarshalInfo(native);
- }
- }
- NativeType ReadNativeType()
- {
- return (NativeType)ReadByte();
- }
- VariantType ReadVariantType()
- {
- return (VariantType)ReadByte();
- }
- string ReadUTF8String()
- {
- if (buffer[position] == 0xff)
- {
- position++;
- return null;
- }
- var length = (int)ReadCompressedUInt32();
- if (length == 0)
- return string.Empty;
- var @string = Encoding.UTF8.GetString(buffer, position,
- buffer[position + length - 1] == 0 ? length - 1 : length);
- position += length;
- return @string;
- }
- public bool CanReadMore()
- {
- return position - start < sig_length;
- }
- }
- }
|