博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理

如何获得一个COM接口的事件源的GUID

Posted on 2010-12-24 23:11  一桶浆糊  阅读(1944)  评论(0编辑  收藏  举报

最近坛子有人问起怎样从一个HTML元素接口获取它的连接点的DIID,这个问题本来不是个问题,用OleView看看组件的类型库信息,马上就能得到它的事件接口GUID,问题在于当得到一个 IHTMLElement 指针时,它到底是那种元素类型?每种元素类型的连接点接口是不同的,所以必须要动态获取事件GUID才能挂接到组件上。

想起我正在做的界面项目,已经实现了动态获取功能,就共享一下源码吧。这段源码封装到了一个类的成员函数里面,现在原封不动地贴出来,有少量内容跟目标无关,但不影响代码的阅读和理解,关键注释已经有了。代码用了 4 种方案,一个不行就试另一个。

好了,废话不多说,上菜!

HRESULT CDuiActiveXEvent::GetEventIID( IUnknown* pUnk, IID* piid )
{
	HRESULT hr = E_FAIL;
	if (pUnk==NULL) return E_INVALIDARG;
	if (piid==NULL) return E_POINTER;
	*piid = IID_NULL;

	// 1. 尝试IProvideClassInfo2
	CComQIPtr<IProvideClassInfo2> spPci(pUnk);
	if (spPci.p && SUCCEEDED(spPci->GetGUID(GUIDKIND_DEFAULT_SOURCE_DISP_IID, piid)))
	{
		_assert(*piid != IID_NULL);
		return S_OK;
	}

	// 2. IProvideClassInfo
	CComPtr<ITypeInfo> spTypeInfo;
	if (spPci.p == NULL)
		hr = pUnk->QueryInterface(IID_IProvideClassInfo, (void**)&spPci.p);
	if (spPci.p)
		hr = spPci->GetClassInfo(&spTypeInfo);

	// 3. IDispatch
	if (spTypeInfo.p == NULL)
	{
		CComQIPtr<IDispatch> spDisp(pUnk);
		if (spDisp.p)
			hr = spDisp->GetTypeInfo(0, 0, &spTypeInfo);
	}

	if (spTypeInfo.p)
	{
		CComPtr<ITypeLib> spTypeLib;
		hr = spTypeInfo->GetContainingTypeLib(&spTypeLib, 0);
		if (SUCCEEDED(hr))
		{
			// 首先找到接口对应的CLSID,其实不需要找,直接用CDuiActiveX::m_clsid 即可,这里作为验证
			CLSID clsid = CLSID_NULL;
			CComQIPtr<IPersist> spPersist(pUnk);
			if (spPersist.p)
			{
				hr = spPersist->GetClassID(&clsid);
				if (SUCCEEDED(hr))
				{
					_assert(clsid!=CLSID_NULL);
					_assert(clsid==m_pOwner->m_clsid);
				}
			}
			if (clsid==CLSID_NULL)
				clsid = m_pOwner->m_clsid;

			CComPtr<ITypeInfo> tiClass;
			hr = spTypeLib->GetTypeInfoOfGuid(clsid, &tiClass);
			if (SUCCEEDED(hr))
			{
				TYPEATTR* attr=NULL;
				hr = tiClass->GetTypeAttr(&attr);
				if (SUCCEEDED(hr))
				{
					for (WORD j=0; j<attr->cImplTypes; j++)
					{
						int nType;
						hr = tiClass->GetImplTypeFlags(j, &nType);
						if (SUCCEEDED(hr) && nType == (IMPLTYPEFLAG_FDEFAULT | IMPLTYPEFLAG_FSOURCE))
						{
							// found!!
							HREFTYPE hRef;
							hr = tiClass->GetRefTypeOfImplType(j, &hRef);
							if (SUCCEEDED(hr))
							{
								CComPtr<ITypeInfo> ti2;
								hr = tiClass->GetRefTypeInfo(hRef, &ti2);
								if (SUCCEEDED(hr))
								{
									TYPEATTR* pAttrIF;
									hr = ti2->GetTypeAttr(&pAttrIF);
									if (pAttrIF != NULL)
									{
										Checked::memcpy_s(piid, sizeof(GUID), &pAttrIF->guid, sizeof(GUID));
										ti2->ReleaseTypeAttr(pAttrIF);
									}
								}
							}
							break;
						}
					}
					tiClass->ReleaseTypeAttr(attr);
				}
			}
		}
	}

	// 4. IConnectionPoint
	if (FAILED(hr))
	{
		CComQIPtr<IConnectionPointContainer> cpc = pUnk;
		if (cpc)
		{
			CComPtr<IEnumConnectionPoints> ecp;
			hr = cpc->EnumConnectionPoints(&ecp);
			if (SUCCEEDED(hr))
			{
				CComPtr<IConnectionPoint> cp;
				hr = ecp->Next(1, &cp, NULL);
				if (SUCCEEDED(hr))
				{
					return cp->GetConnectionInterface(piid);
				}
			}
		}
	}
	return hr;
}