501 lines
18 KiB
C++
501 lines
18 KiB
C++
|
/*++
|
||
|
|
||
|
Copyright (c) 2000 Microsoft Corporation
|
||
|
|
||
|
Module Name:
|
||
|
|
||
|
cominterface.cpp
|
||
|
|
||
|
Abstract:
|
||
|
|
||
|
Activation context section contributor for COM interface proxy mapping.
|
||
|
|
||
|
Author:
|
||
|
|
||
|
Michael J. Grier (MGrier) 28-Mar-2000
|
||
|
|
||
|
Revision History:
|
||
|
|
||
|
--*/
|
||
|
|
||
|
#include "stdinc.h"
|
||
|
#include <windows.h>
|
||
|
#include <stdio.h>
|
||
|
#include "sxsp.h"
|
||
|
#include "sxsidp.h"
|
||
|
#include "fusionparser.h"
|
||
|
|
||
|
DECLARE_STD_ATTRIBUTE_NAME_DESCRIPTOR(baseInterface);
|
||
|
DECLARE_STD_ATTRIBUTE_NAME_DESCRIPTOR(clsid);
|
||
|
DECLARE_STD_ATTRIBUTE_NAME_DESCRIPTOR(iid);
|
||
|
DECLARE_STD_ATTRIBUTE_NAME_DESCRIPTOR(name);
|
||
|
DECLARE_STD_ATTRIBUTE_NAME_DESCRIPTOR(numMethods);
|
||
|
DECLARE_STD_ATTRIBUTE_NAME_DESCRIPTOR(proxyStubClsid32);
|
||
|
DECLARE_STD_ATTRIBUTE_NAME_DESCRIPTOR(tlbid);
|
||
|
|
||
|
/*
|
||
|
|
||
|
<file name="foo.dll">
|
||
|
<comInterfaceProxyStub iid="{iid}" tlbid="{tlbid}" numMethods="3" baseInterface="{iid}" name="IFoo"/>
|
||
|
</file>
|
||
|
|
||
|
*/
|
||
|
|
||
|
typedef struct _IID_ENTRY
|
||
|
{
|
||
|
_IID_ENTRY() { }
|
||
|
GUID m_clsid;
|
||
|
GUID m_iid;
|
||
|
ULONG m_nMethods;
|
||
|
GUID m_tlbid;
|
||
|
GUID m_iidBase;
|
||
|
CStringBuffer m_buffName;
|
||
|
bool m_fNumMethodsValid;
|
||
|
bool m_fBaseInterfaceValid;
|
||
|
|
||
|
private:
|
||
|
_IID_ENTRY(const _IID_ENTRY &);
|
||
|
void operator =(const _IID_ENTRY &);
|
||
|
} IID_ENTRY, *PIID_ENTRY;
|
||
|
|
||
|
BOOL
|
||
|
SxspComInterfaceRedirectionGuidSectionGenerationCallback(
|
||
|
PVOID Context,
|
||
|
ULONG Reason,
|
||
|
PVOID CallbackData
|
||
|
);
|
||
|
|
||
|
VOID
|
||
|
SxspComInterfaceRedirectionContributorCallback(
|
||
|
PACTCTXCTB_CALLBACK_DATA Data
|
||
|
)
|
||
|
{
|
||
|
FN_TRACE();
|
||
|
|
||
|
PGUID_SECTION_GENERATION_CONTEXT GSGenContext = (PGUID_SECTION_GENERATION_CONTEXT) Data->Header.ActCtxGenContext;
|
||
|
|
||
|
switch (Data->Header.Reason)
|
||
|
{
|
||
|
case ACTCTXCTB_CBREASON_ACTCTXGENBEGINNING:
|
||
|
Data->GenBeginning.Success = FALSE;
|
||
|
|
||
|
INTERNAL_ERROR_CHECK(GSGenContext == NULL);
|
||
|
|
||
|
// do everything if we are generating an activation context.
|
||
|
if (Data->Header.ManifestOperation == MANIFEST_OPERATION_GENERATE_ACTIVATION_CONTEXT)
|
||
|
{
|
||
|
IFW32FALSE_EXIT(
|
||
|
::SxsInitGuidSectionGenerationContext(
|
||
|
&GSGenContext,
|
||
|
ACTIVATION_CONTEXT_DATA_COM_INTERFACE_REDIRECTION_FORMAT_WHISTLER,
|
||
|
&::SxspComInterfaceRedirectionGuidSectionGenerationCallback,
|
||
|
NULL));
|
||
|
|
||
|
Data->Header.ActCtxGenContext = GSGenContext;
|
||
|
}
|
||
|
|
||
|
Data->GenBeginning.Success = TRUE;
|
||
|
break;
|
||
|
|
||
|
case ACTCTXCTB_CBREASON_ACTCTXGENENDED:
|
||
|
if (GSGenContext != NULL)
|
||
|
::SxsDestroyGuidSectionGenerationContext(GSGenContext);
|
||
|
break;
|
||
|
|
||
|
case ACTCTXCTB_CBREASON_ALLPARSINGDONE:
|
||
|
Data->AllParsingDone.Success = FALSE;
|
||
|
#if 0
|
||
|
// this call isn't implemented for guid sections ... yet
|
||
|
if (GSGenContext != NULL)
|
||
|
IFW32FALSE_EXIT(::SxsDoneModifyingGuidSectionGenerationContext(GSGenContext));
|
||
|
#endif
|
||
|
Data->AllParsingDone.Success = TRUE;
|
||
|
break;
|
||
|
|
||
|
case ACTCTXCTB_CBREASON_GETSECTIONSIZE:
|
||
|
Data->GetSectionSize.Success = FALSE;
|
||
|
// Someone shouldn't be asking for the section size if we
|
||
|
// are not generating an activation context.
|
||
|
// These two asserts should be equivalent...
|
||
|
INTERNAL_ERROR_CHECK(Data->Header.ManifestOperation == MANIFEST_OPERATION_GENERATE_ACTIVATION_CONTEXT);
|
||
|
INTERNAL_ERROR_CHECK(GSGenContext != NULL);
|
||
|
|
||
|
IFW32FALSE_EXIT(
|
||
|
::SxsGetGuidSectionGenerationContextSectionSize(
|
||
|
GSGenContext,
|
||
|
&Data->GetSectionSize.SectionSize));
|
||
|
|
||
|
Data->GetSectionSize.Success = TRUE;
|
||
|
|
||
|
break;
|
||
|
|
||
|
case ACTCTXCTB_CBREASON_ELEMENTPARSED:
|
||
|
Data->ElementParsed.Success = FALSE;
|
||
|
if (Data->Header.ManifestOperation == MANIFEST_OPERATION_GENERATE_ACTIVATION_CONTEXT)
|
||
|
{
|
||
|
ULONG MappedValue = 0;
|
||
|
bool fFound = false;
|
||
|
|
||
|
enum MappedValues
|
||
|
{
|
||
|
eAssemblyInterface = 1,
|
||
|
eAssemblyFileComClassInterface = 2,
|
||
|
};
|
||
|
|
||
|
#define PATH_MAP_ENTRY(_depth, _string, _mv) { (_depth), _string, NUMBER_OF(_string) - 1, (_mv) },
|
||
|
|
||
|
static const ELEMENT_PATH_MAP_ENTRY s_rgEntries[] =
|
||
|
{
|
||
|
PATH_MAP_ENTRY(2, L"urn:schemas-microsoft-com:asm.v1^assembly!urn:schemas-microsoft-com:asm.v1^comInterfaceExternalProxyStub", eAssemblyInterface)
|
||
|
PATH_MAP_ENTRY(3, L"urn:schemas-microsoft-com:asm.v1^assembly!urn:schemas-microsoft-com:asm.v1^file!urn:schemas-microsoft-com:asm.v1^comInterfaceProxyStub", eAssemblyFileComClassInterface)
|
||
|
};
|
||
|
|
||
|
#undef PATH_MAP_ENTRY
|
||
|
|
||
|
IFW32FALSE_EXIT(
|
||
|
::SxspProcessElementPathMap(
|
||
|
Data->ElementParsed.ParseContext,
|
||
|
s_rgEntries,
|
||
|
NUMBER_OF(s_rgEntries),
|
||
|
MappedValue,
|
||
|
fFound));
|
||
|
|
||
|
if (fFound)
|
||
|
{
|
||
|
switch (MappedValue)
|
||
|
{
|
||
|
default:
|
||
|
INTERNAL_ERROR_CHECK2(
|
||
|
FALSE,
|
||
|
"Invalid mapped value returned from SxspProcessElementPathMap");
|
||
|
|
||
|
case eAssemblyInterface:
|
||
|
{
|
||
|
bool fFound = false;
|
||
|
CStringBuffer TempBuffer;
|
||
|
GUID iid, tlbid, iidBase, clsid;
|
||
|
ULONG nMethods = 0;
|
||
|
SIZE_T cb;
|
||
|
bool fNumMethodsValid = false;
|
||
|
bool fBaseInterfaceValid = false;
|
||
|
|
||
|
IFW32FALSE_EXIT(
|
||
|
::SxspGetAttributeValue(
|
||
|
SXSP_GET_ATTRIBUTE_VALUE_FLAG_REQUIRED_ATTRIBUTE,
|
||
|
&s_AttributeName_iid,
|
||
|
&Data->ElementParsed,
|
||
|
fFound,
|
||
|
sizeof(iid),
|
||
|
&iid,
|
||
|
cb,
|
||
|
&::SxspValidateGuidAttribute,
|
||
|
0));
|
||
|
|
||
|
INTERNAL_ERROR_CHECK(fFound);
|
||
|
|
||
|
IFW32FALSE_EXIT(
|
||
|
::SxspGetAttributeValue(
|
||
|
0,
|
||
|
&s_AttributeName_proxyStubClsid32,
|
||
|
&Data->ElementParsed,
|
||
|
fFound,
|
||
|
sizeof(clsid),
|
||
|
&clsid,
|
||
|
cb,
|
||
|
&::SxspValidateGuidAttribute,
|
||
|
0));
|
||
|
|
||
|
if (!fFound)
|
||
|
clsid = iid;
|
||
|
|
||
|
IFW32FALSE_EXIT(::SxspGetAttributeValue(0, &s_AttributeName_numMethods, &Data->ElementParsed, fFound, sizeof(TempBuffer), &TempBuffer, cb, NULL, 0));
|
||
|
if (fFound)
|
||
|
{
|
||
|
IFW32FALSE_EXIT(CFusionParser::ParseULONG(nMethods, TempBuffer, TempBuffer.Cch()));
|
||
|
fNumMethodsValid = true;
|
||
|
}
|
||
|
|
||
|
tlbid = GUID_NULL;
|
||
|
IFW32FALSE_EXIT(
|
||
|
::SxspGetAttributeValue(
|
||
|
0,
|
||
|
&s_AttributeName_tlbid,
|
||
|
&Data->ElementParsed,
|
||
|
fFound,
|
||
|
sizeof(tlbid),
|
||
|
&tlbid,
|
||
|
cb,
|
||
|
&::SxspValidateGuidAttribute,
|
||
|
0));
|
||
|
|
||
|
IFW32FALSE_EXIT(
|
||
|
::SxspGetAttributeValue(
|
||
|
0,
|
||
|
&s_AttributeName_baseInterface,
|
||
|
&Data->ElementParsed,
|
||
|
fFound,
|
||
|
sizeof(iidBase),
|
||
|
&iidBase,
|
||
|
cb,
|
||
|
&::SxspValidateGuidAttribute,
|
||
|
0));
|
||
|
if (fFound)
|
||
|
fBaseInterfaceValid = true;
|
||
|
|
||
|
TempBuffer.Clear();
|
||
|
IFW32FALSE_EXIT(::SxspGetAttributeValue(0, &s_AttributeName_name, &Data->ElementParsed, fFound, sizeof(TempBuffer), &TempBuffer, cb, NULL, 0));
|
||
|
|
||
|
// Do more work if generating an activation context.
|
||
|
if (Data->Header.ManifestOperation == MANIFEST_OPERATION_GENERATE_ACTIVATION_CONTEXT)
|
||
|
{
|
||
|
PIID_ENTRY Entry = NULL;
|
||
|
|
||
|
IFALLOCFAILED_EXIT(Entry = new IID_ENTRY);
|
||
|
|
||
|
Entry->m_clsid = clsid;
|
||
|
Entry->m_tlbid = tlbid;
|
||
|
Entry->m_iid = iid;
|
||
|
Entry->m_iidBase = iidBase;
|
||
|
Entry->m_nMethods = nMethods;
|
||
|
Entry->m_fNumMethodsValid = fNumMethodsValid;
|
||
|
Entry->m_fBaseInterfaceValid = fBaseInterfaceValid;
|
||
|
IFW32FALSE_EXIT(Entry->m_buffName.Win32Assign(TempBuffer));
|
||
|
|
||
|
IFW32FALSE_EXIT(
|
||
|
::SxsAddGuidToGuidSectionGenerationContext(
|
||
|
(PGUID_SECTION_GENERATION_CONTEXT) Data->ElementParsed.Header.ActCtxGenContext,
|
||
|
&iid,
|
||
|
Entry,
|
||
|
Data->ElementParsed.AssemblyContext->AssemblyRosterIndex,
|
||
|
ERROR_SXS_DUPLICATE_IID));
|
||
|
}
|
||
|
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
case eAssemblyFileComClassInterface:
|
||
|
{
|
||
|
bool fFound = false;
|
||
|
CStringBuffer TempBuffer;
|
||
|
GUID iid, tlbid, iidBase;
|
||
|
ULONG nMethods = 0;
|
||
|
SIZE_T cb;
|
||
|
bool fNumMethodsValid = false;
|
||
|
bool fBaseInterfaceValid = false;
|
||
|
|
||
|
IFW32FALSE_EXIT(
|
||
|
::SxspGetAttributeValue(
|
||
|
SXSP_GET_ATTRIBUTE_VALUE_FLAG_REQUIRED_ATTRIBUTE,
|
||
|
&s_AttributeName_iid,
|
||
|
&Data->ElementParsed,
|
||
|
fFound,
|
||
|
sizeof(iid),
|
||
|
&iid,
|
||
|
cb,
|
||
|
&::SxspValidateGuidAttribute,
|
||
|
0));
|
||
|
|
||
|
INTERNAL_ERROR_CHECK(fFound);
|
||
|
|
||
|
IFW32FALSE_EXIT(::SxspGetAttributeValue(0, &s_AttributeName_numMethods, &Data->ElementParsed, fFound, sizeof(TempBuffer), &TempBuffer, cb, NULL, 0));
|
||
|
if (fFound)
|
||
|
{
|
||
|
IFW32FALSE_EXIT(CFusionParser::ParseULONG(nMethods, TempBuffer, TempBuffer.Cch()));
|
||
|
fNumMethodsValid = true;
|
||
|
}
|
||
|
|
||
|
tlbid = GUID_NULL;
|
||
|
IFW32FALSE_EXIT(
|
||
|
::SxspGetAttributeValue(
|
||
|
0,
|
||
|
&s_AttributeName_tlbid,
|
||
|
&Data->ElementParsed,
|
||
|
fFound,
|
||
|
sizeof(tlbid),
|
||
|
&tlbid,
|
||
|
cb,
|
||
|
&::SxspValidateGuidAttribute,
|
||
|
0));
|
||
|
|
||
|
IFW32FALSE_EXIT(
|
||
|
::SxspGetAttributeValue(
|
||
|
0,
|
||
|
&s_AttributeName_baseInterface,
|
||
|
&Data->ElementParsed,
|
||
|
fFound,
|
||
|
sizeof(iidBase),
|
||
|
&iidBase,
|
||
|
cb,
|
||
|
&::SxspValidateGuidAttribute,
|
||
|
0));
|
||
|
if (fFound)
|
||
|
fBaseInterfaceValid = true;
|
||
|
|
||
|
|
||
|
TempBuffer.Clear();
|
||
|
IFW32FALSE_EXIT(::SxspGetAttributeValue(0, &s_AttributeName_name, &Data->ElementParsed, fFound, sizeof(TempBuffer), &TempBuffer, cb, NULL, 0));
|
||
|
|
||
|
// Do more work if generating an activation context.
|
||
|
if (Data->Header.ManifestOperation == MANIFEST_OPERATION_GENERATE_ACTIVATION_CONTEXT)
|
||
|
{
|
||
|
PIID_ENTRY Entry = NULL;
|
||
|
|
||
|
IFALLOCFAILED_EXIT(Entry = new IID_ENTRY);
|
||
|
|
||
|
Entry->m_clsid = iid;
|
||
|
Entry->m_tlbid = tlbid;
|
||
|
Entry->m_iid = iid;
|
||
|
Entry->m_iidBase = iidBase;
|
||
|
Entry->m_nMethods = nMethods;
|
||
|
Entry->m_fNumMethodsValid = fNumMethodsValid;
|
||
|
Entry->m_fBaseInterfaceValid = fBaseInterfaceValid;
|
||
|
IFW32FALSE_EXIT(Entry->m_buffName.Win32Assign(TempBuffer));
|
||
|
|
||
|
IFW32FALSE_EXIT(
|
||
|
::SxsAddGuidToGuidSectionGenerationContext(
|
||
|
(PGUID_SECTION_GENERATION_CONTEXT) Data->ElementParsed.Header.ActCtxGenContext,
|
||
|
&iid,
|
||
|
Entry,
|
||
|
Data->ElementParsed.AssemblyContext->AssemblyRosterIndex,
|
||
|
ERROR_SXS_DUPLICATE_IID));
|
||
|
}
|
||
|
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
Data->ElementParsed.Success = TRUE;
|
||
|
|
||
|
break;
|
||
|
|
||
|
case ACTCTXCTB_CBREASON_GETSECTIONDATA:
|
||
|
Data->GetSectionData.Success = FALSE;
|
||
|
|
||
|
INTERNAL_ERROR_CHECK(GSGenContext != NULL);
|
||
|
INTERNAL_ERROR_CHECK(Data->Header.ManifestOperation == MANIFEST_OPERATION_GENERATE_ACTIVATION_CONTEXT);
|
||
|
|
||
|
IFW32FALSE_EXIT(
|
||
|
::SxsGetGuidSectionGenerationContextSectionData(
|
||
|
GSGenContext,
|
||
|
Data->GetSectionData.SectionSize,
|
||
|
Data->GetSectionData.SectionDataStart,
|
||
|
NULL));
|
||
|
|
||
|
Data->GetSectionData.Success = TRUE;
|
||
|
break;
|
||
|
}
|
||
|
Exit:
|
||
|
;
|
||
|
}
|
||
|
|
||
|
BOOL
|
||
|
SxspComInterfaceRedirectionGuidSectionGenerationCallback(
|
||
|
PVOID Context,
|
||
|
ULONG Reason,
|
||
|
PVOID CallbackData
|
||
|
)
|
||
|
{
|
||
|
BOOL fSuccess = FALSE;
|
||
|
FN_TRACE_WIN32(fSuccess);
|
||
|
|
||
|
switch (Reason)
|
||
|
{
|
||
|
default:
|
||
|
break;
|
||
|
|
||
|
case GUID_SECTION_GENERATION_CONTEXT_CALLBACK_REASON_ENTRYDELETED:
|
||
|
{
|
||
|
PGUID_SECTION_GENERATION_CONTEXT_CBDATA_ENTRYDELETED CBData = (PGUID_SECTION_GENERATION_CONTEXT_CBDATA_ENTRYDELETED) CallbackData;
|
||
|
PIID_ENTRY Entry = (PIID_ENTRY) CBData->DataContext;
|
||
|
|
||
|
if (Entry != NULL)
|
||
|
FUSION_DELETE_SINGLETON(Entry);
|
||
|
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
case GUID_SECTION_GENERATION_CONTEXT_CALLBACK_REASON_GETDATASIZE:
|
||
|
{
|
||
|
PGUID_SECTION_GENERATION_CONTEXT_CBDATA_GETDATASIZE CBData = (PGUID_SECTION_GENERATION_CONTEXT_CBDATA_GETDATASIZE) CallbackData;
|
||
|
PIID_ENTRY Entry = (PIID_ENTRY) CBData->DataContext;
|
||
|
|
||
|
CBData->DataSize = sizeof(ACTIVATION_CONTEXT_DATA_COM_INTERFACE_REDIRECTION);
|
||
|
|
||
|
if (Entry != NULL)
|
||
|
{
|
||
|
if (Entry->m_buffName.Cch() != 0)
|
||
|
{
|
||
|
CBData->DataSize += ((Entry->m_buffName.Cch() + 1) * sizeof(WCHAR));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
case GUID_SECTION_GENERATION_CONTEXT_CALLBACK_REASON_GETDATA:
|
||
|
{
|
||
|
PGUID_SECTION_GENERATION_CONTEXT_CBDATA_GETDATA CBData = (PGUID_SECTION_GENERATION_CONTEXT_CBDATA_GETDATA) CallbackData;
|
||
|
PACTIVATION_CONTEXT_DATA_COM_INTERFACE_REDIRECTION Info;
|
||
|
PIID_ENTRY Entry = (PIID_ENTRY) CBData->DataContext;
|
||
|
PWSTR Cursor;
|
||
|
|
||
|
SIZE_T BytesLeft = CBData->BufferSize;
|
||
|
SIZE_T BytesWritten = 0;
|
||
|
|
||
|
Info = (PACTIVATION_CONTEXT_DATA_COM_INTERFACE_REDIRECTION) CBData->Buffer;
|
||
|
|
||
|
if (BytesLeft < sizeof(ACTIVATION_CONTEXT_DATA_COM_INTERFACE_REDIRECTION))
|
||
|
{
|
||
|
::FusionpSetLastWin32Error(ERROR_INSUFFICIENT_BUFFER);
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
BytesWritten += sizeof(ACTIVATION_CONTEXT_DATA_COM_INTERFACE_REDIRECTION);
|
||
|
BytesLeft -= sizeof(ACTIVATION_CONTEXT_DATA_COM_INTERFACE_REDIRECTION);
|
||
|
|
||
|
Cursor = (PWSTR) (Info + 1);
|
||
|
|
||
|
Info->Size = sizeof(ACTIVATION_CONTEXT_DATA_COM_INTERFACE_REDIRECTION);
|
||
|
Info->Flags = 0;
|
||
|
Info->ProxyStubClsid32 = Entry->m_clsid;
|
||
|
|
||
|
if (Entry->m_fNumMethodsValid)
|
||
|
{
|
||
|
Info->Flags |= ACTIVATION_CONTEXT_DATA_COM_INTERFACE_REDIRECTION_FLAG_NUM_METHODS_VALID;
|
||
|
Info->NumMethods = Entry->m_nMethods;
|
||
|
}
|
||
|
else
|
||
|
Info->NumMethods = 0;
|
||
|
|
||
|
Info->TypeLibraryId = Entry->m_tlbid;
|
||
|
|
||
|
if (Entry->m_fBaseInterfaceValid)
|
||
|
{
|
||
|
Info->Flags |= ACTIVATION_CONTEXT_DATA_COM_INTERFACE_REDIRECTION_FLAG_BASE_INTERFACE_VALID;
|
||
|
Info->BaseInterface = Entry->m_iidBase;
|
||
|
}
|
||
|
else
|
||
|
Info->BaseInterface = GUID_NULL;
|
||
|
|
||
|
IFW32FALSE_EXIT(
|
||
|
Entry->m_buffName.Win32CopyIntoBuffer(
|
||
|
&Cursor,
|
||
|
&BytesLeft,
|
||
|
&BytesWritten,
|
||
|
Info,
|
||
|
&Info->NameOffset,
|
||
|
&Info->NameLength));
|
||
|
|
||
|
CBData->BytesWritten = BytesWritten;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
fSuccess = TRUE;
|
||
|
|
||
|
Exit:
|
||
|
return fSuccess;
|
||
|
}
|
||
|
|