diff options
Diffstat (limited to 'src/md/compiler/newmerger.h')
| -rw-r--r-- | src/md/compiler/newmerger.h | 256 |
1 files changed, 256 insertions, 0 deletions
diff --git a/src/md/compiler/newmerger.h b/src/md/compiler/newmerger.h new file mode 100644 index 0000000000..fc89ab7f61 --- /dev/null +++ b/src/md/compiler/newmerger.h @@ -0,0 +1,256 @@ +// Licensed to the .NET Foundation under one or more agreements. +// The .NET Foundation licenses this file to you under the MIT license. +// See the LICENSE file in the project root for more information. +//***************************************************************************** +// NewMerger.h +// + +// +// Contains utility code for MD directory +// +//***************************************************************************** +#ifndef __NEWMERGER__h__ +#define __NEWMERGER__h__ + +class RegMeta; + +class MDTOKENMAP; + +// module-level awareness of Security critical annotions +typedef BYTE InputScopeSecurityCriticalStatus; +#define ISSCS_Unknown 0x0 +#define ISSCS_SecurityCritical 0x1 +#define ISSCS_SecurityCriticalEverything (ISSCS_SecurityCritical | 0x2) +#define ISSCS_SecurityCriticalExplicit (ISSCS_SecurityCritical) +#define ISSCS_SecurityTransparent 0x4 +#define ISSCS_SecurityTreatAsSafe 0x8 +#define ISSCS_SECURITYCRITICAL_LEGACY (ISSCS_SecurityCriticalEverything | ISSCS_SecurityTreatAsSafe) +#define ISSCS_SECURITYCRITICAL_FLAGS (ISSCS_SecurityCriticalEverything | ISSCS_SecurityTransparent) + +//********************************************************************* +// MergeImportData +//********************************************************************* +class MergeImportData +{ +public: + RegMeta *m_pRegMetaImport; + IUnknown *m_pHandler; + IMapToken *m_pHostMapToken; + MDTOKENMAP *m_pMDTokenMap; + MergeImportData *m_pNextImportData; + + mdMemberRef m_tkSuppressMergeCheckCtor; // caches the SuppressMergeCheckAttribute's .ctor token + mdMemberRef m_tkHandleProcessCorruptedStateCtor; // caches the memberRef token to HandleProcessCorruptedStateExceptionsAttribute's .ctor token + + // import contains assembly-level SecurityTransparent or SecurityCritical + InputScopeSecurityCriticalStatus m_isscsSecurityCriticalStatus; +#if _DEBUG + int m_iImport; // debug only. This is the ith import for merge. +#endif // _DEBUG +}; + +//********************************************************************* +// MergeTypeData +//********************************************************************* +struct MergeTypeData +{ + ULONG m_cMethods; + ULONG m_cFields; + ULONG m_cEvents; + ULONG m_cProperties; + BOOL m_bSuppressMergeCheck; +}; + + +//********************************************************************* +// Class to handle merge +//********************************************************************* +class NEWMERGER +{ + friend class RegMeta; +public: + NEWMERGER(); + ~NEWMERGER(); + + HRESULT Init(RegMeta *pRegMetaDest); + + HRESULT AddImport( + IMetaDataImport2 *pImport, // [IN] The scope to be merged. + IMapToken *pHostMapToken, // [IN] Host IMapToken interface to receive token remap notification + IUnknown *pHandler); // [IN] An object to receive to receive error notification. + + HRESULT Merge(MergeFlags flags, CorRefToDefCheck optimizeRefToDef); + +protected: + + CMiniMdRW *GetMiniMdEmit(); + + HRESULT InitMergeTypeData(); + + HRESULT MergeTypeDefNamesOnly(); + HRESULT MergeModuleRefs(); + HRESULT MergeAssemblyRefs(); + HRESULT MergeTypeRefs(); + HRESULT CompleteMergeTypeDefs(); + + HRESULT CopyTypeDefPartially( + TypeDefRec *pRecEmit, // [IN] the emit record to fill + CMiniMdRW *pMiniMdImport, // [IN] the importing scope + TypeDefRec *pRecImp); // [IN] the record to import + + // helpers for merging tables + HRESULT MergeModule( ); + HRESULT MergeTypeDefChildren(); + HRESULT MergeInterfaceImpls( ); + HRESULT MergeMemberRefs( ); + HRESULT MergePinvoke(); + + HRESULT MergeConstants( ); + HRESULT MergeCustomAttributes( ); + HRESULT MergeFieldMarshals( ); + HRESULT MergeDeclSecuritys( ); + HRESULT MergeClassLayouts( ); + HRESULT MergeFieldLayouts( ); + HRESULT MergeFieldRVAs(); + HRESULT MergeMethodImpls( ); + HRESULT MergeStandAloneSigs(); + HRESULT MergeMethodSpecs(); + HRESULT MergeTypeSpecs(); + HRESULT MergeSourceFiles( ); + HRESULT MergeBlocks( ); + HRESULT MergeScopes( ); + HRESULT MergeLocalVariables( ); + HRESULT MergeStrings( ); + + HRESULT MergeAssembly(); + HRESULT MergeFiles(); + HRESULT MergeExportedTypes(); + HRESULT MergeManifestResources(); + + // helpers for SecurityCritical-related merging + InputScopeSecurityCriticalStatus CheckInputScopeIsCritical(MergeImportData* pImportData, HRESULT& hr); + HRESULT RetrieveStandardSecurityCriticalMetaData( + mdAssemblyRef& tkMscorlib, + mdTypeRef& securityEnum, + BYTE*& rgSigBytesSecurityCriticalEverythingCtor, + DWORD& dwSigEverythingSize, + BYTE*& rgSigBytesSecurityCriticalExplicitCtor, + DWORD& dwSigExplicitSize); + + HRESULT MergeSecurityCriticalModuleLevelAttributes( + MergeImportData* pImportData, + mdToken tkParentImp, TOKENREC* pTypeRec, + mdToken mrSecurityTreatAsSafeAttributeCtor, + mdToken mrSecurityTransparentAttributeCtor, + mdToken mrSecurityCriticalExplicitAttributeCtor, + mdToken mrSecurityCriticalEverythingAttributeCtor); + HRESULT MergeSecurityCriticalAttributes(); + + // copy over a interfaceimpl record + HRESULT CopyInterfaceImpl( + InterfaceImplRec *pRecEmit, // [IN] the emit record to fill + MergeImportData *pImportData, // [IN] the importing context + InterfaceImplRec *pRecImp); // [IN] the record to import + + // verification helpers + HRESULT VerifyMethods(MergeImportData *pImportData, mdTypeDef tdImp, mdTypeDef tdEmit); + HRESULT VerifyFields(MergeImportData *pImportData, mdTypeDef tdImp, mdTypeDef tdEmit); + HRESULT VerifyEvents(MergeImportData *pImportData, mdTypeDef tdImp, mdTypeDef tdEmit); + HRESULT VerifyProperties(MergeImportData *pImportData, mdTypeDef tdImp, mdTypeDef tdEmit); + HRESULT VerifyParams(MergeImportData *pImportData, mdMethodDef mdImp, mdMethodDef mdEmit); + HRESULT VerifyGenericParams(MergeImportData *pImportData, mdTypeDef tdImp, mdTypeDef tdEmit); + HRESULT VerifyGenericParamConstraints(MergeImportData *pImportData, mdGenericParam gpImp, mdGenericParam gpEmit); + + // Copy helpers + HRESULT CopyMethods(MergeImportData *pImportData, mdTypeDef tdImp, mdTypeDef tdEmit); + HRESULT CopyFields(MergeImportData *pImportData, mdTypeDef tdImp, mdTypeDef tdEmit); + HRESULT CopyEvents(MergeImportData *pImportData, mdTypeDef tdImp, mdTypeDef tdEmit); + HRESULT CopyProperties(MergeImportData *pImportData, mdTypeDef tdImp, mdTypeDef tdEmit); + HRESULT CopyParams(MergeImportData *pImportData, mdMethodDef mdImp, mdMethodDef mdEmit); + HRESULT CopyGenericParams(MergeImportData *pImportData, mdToken tkImp, mdToken tkEmit); + HRESULT CopyGenericParamConstraints(MergeImportData *pImportData, mdGenericParam gpImp, mdGenericParam gpEmit); + + HRESULT CopyMethod( + MergeImportData *pImportData, // [IN] import scope + MethodRec *pRecImp, // [IN] the record to import + MethodRec *pRecEmit); // [IN] the emit record to fill + + HRESULT CopyField( + MergeImportData *pImportData, // [IN] import scope + FieldRec *pRecImp, // [IN] the record to import + FieldRec *pRecEmit); // [IN] the emit record to fill + + HRESULT CopyEvent( + MergeImportData *pImportData, // [IN] import scope + EventRec *pRecImp, // [IN] the record to import + EventRec *pRecEmit); // [IN] the emit record to fill + + HRESULT CopyProperty( + MergeImportData *pImportData, // [IN] import scope + PropertyRec *pRecImp, // [IN] the record to import + PropertyRec *pRecEmit); // [IN] the emit record to fill + + HRESULT CopyParam( + MergeImportData *pImportData, // [IN] import scope + ParamRec *pRecImp, // [IN] the record to import + ParamRec *pRecEmit); // [IN] the emit record to fill + + HRESULT CopyMethodSemantics( + MergeImportData *pImportData, + mdToken tkImport, // Event or property in the import scope + mdToken tkEmit); // corresponding event or property in the emitting scope + + HRESULT VerifyMethod( + MergeImportData *pImportData, + mdMethodDef mdImp, // [IN] the emit record to fill + mdMethodDef mdEmit); // [IN] the record to import + + HRESULT OnError(HRESULT hr, MergeImportData *pImportData, mdToken token); + +private: + RegMeta *m_pRegMetaEmit; + MergeImportData *m_pImportDataList; + MergeImportData **m_pImportDataTail; + MergeFlags m_dwMergeFlags; + BOOL m_fDupCheck; + CorRefToDefCheck m_optimizeRefToDef; + // the combined value of the Security Critical input scopes (e.g. UNION of each scope's attributes) + // if ANY of the scopes have a bit set, then we must do some merging + InputScopeSecurityCriticalStatus m_isscsSecurityCritical; + // the common values of the Security Critical input scopes (e.g. INTERSECTION of each scope's attributes) + // if all scopes have the same bit set, then we can emit one bit at the final output scope + InputScopeSecurityCriticalStatus m_isscsSecurityCriticalAllScopes; + + CDynArray<MergeTypeData> m_rMTDs; +#if _DEBUG + int m_iImport; // debug only. To count how many import scopes to be merged. +#endif // _DEBUG +}; + + +#define CheckContinuableErrorEx(EXPR, HANDLER, TOKEN) \ +{ \ + HRESULT hrOnErr, hrExpr; \ + hrExpr = EXPR; \ + \ + hrOnErr = OnError(hrExpr, HANDLER, TOKEN); \ + if (hrOnErr != S_OK) \ + { \ + if (hrOnErr == S_FALSE) \ + { \ + hr = hrExpr; \ + } \ + else if (SUCCEEDED(hrOnErr)) \ + { \ + hr = E_UNEXPECTED; \ + } \ + else if (FAILED(hrOnErr)) \ + { \ + hr = hrOnErr; \ + } \ + IfFailGo(hr); \ + } \ +} + + +#endif // __NEWMERGER__h__ |