枚举COM指针所能查询的接口

鄙人最近从事COM开发,基本上得到一COM接口就要QueryInterface另外一个接口,都能获取到哪些接口就要查文档了。如果有个函数传入一接口指针,能输出其支持的所有接口那就好了。接口的注册表项位于HKEY_CLASSES_ROOT\Interface中,每个子项的名称都是以接口IID命名,其默认值就是接口的字符串名称。

先写个函数查注册表中接口IID对应的接口名称:

/*
* 获取COM接口的名称,返回获取成功与否
* 接口名为GUID注册表项默认的值
*/
template
<int t_nSize>
ATLINLINE BOOL GetInterfaceName( REFIID riid, TCHAR (
&szName)[t_nSize] )
{
    BOOL  bRet  
=  FALSE;
#ifdef   USER_ATL_
    
//CRegKey版本
    CRegKey  regKey;
    OLECHAR szGuid[
64];
    ::StringFromGUID2(riid, szGuid, 
sizeof(szGuid)/sizeof(OLECHAR));
    TCHAR szSubKey[
128];
    wsprintf( szSubKey, _T(
"Interface\\%s"), COLE2T(szGuid) );
    
if( ERROR_SUCCESS == regKey.Open( HKEY_CLASSES_ROOT, szSubKey,  KEY_QUERY_VALUE ) )
    {
        ULONG cb 
= 127;
        
if( ERROR_SUCCESS == regKey.QueryStringValue( NULL, szName, &cb) )
        {
            bRet  
=  TRUE;
        }
        regKey.Close();
    }
#else
    
//API版本:
    OLECHAR szGuid[64];
    ::StringFromGUID2(riid, szGuid, 
sizeof(szGuid)/sizeof(OLECHAR));
    TCHAR szSubKey[
128];
    wsprintf( szSubKey, _T(
"Interface\\%s"), COLE2T(szGuid) );

    HKEY hKey;
    
if( ERROR_SUCCESS == ::RegOpenKeyEx( HKEY_CLASSES_ROOT, szSubKey, 0, KEY_QUERY_VALUE, &hKey) )
    {
        DWORD dwType;
        DWORD dwSize 
= t_nSize * sizeof(TCHAR);
        
if( ERROR_SUCCESS == ::RegQueryValueEx(hKey, NULL, NULL, &dwType, reinterpret_cast<LPBYTE>(szName), &dwSize) )
        {
            bRet  
=  TRUE;
        }
        ::RegCloseKey(hKey);
    }
#endif
    
return bRet;
}

读注册表分别用了API版本和ATL的CRegKey版本(大家不要BS,鄙人一向信奉多练练手没害处)。

之后遍历注册表项HKEY_CLASSES_ROOT\Interface,获取每个接口的IID,如果传入的COM接口QueryInterface这个接口成功,则支持该接口,输出其接口名称。

 

/*
* 枚举接口所有名称
*/
ATLINLINE 
void ScanInterface( IUnknown *pUnk )
{
    
if( pUnk==NULL ) return;

#ifdef   USER_ATL_
    CRegKey  regKey;

    
if( ERROR_SUCCESS == regKey.Open( HKEY_CLASSES_ROOT, _T("Interface"), KEY_QUERY_VALUE | KEY_ENUMERATE_SUB_KEYS ) )
    {
        TCHAR szGuid[
128];
        DWORD dwIndex 
= 0;
        DWORD dwSize 
= _countof(szGuid) * sizeof(TCHAR);
        
while( ERROR_SUCCESS == regKey.EnumKey( dwIndex, szGuid, &dwSize ) ) 
        {
            IID iid;
            ::IIDFromString( CT2OLE(szGuid), 
&iid );

            CComPtr
<IUnknown> spUnk;
            
if( SUCCEEDED( pUnk->QueryInterface( iid, (LPVOID*)&spUnk ) ) ) 
            {
                TCHAR szName[
128];
                
if( GetInterfaceName( iid, szName ) )
                {
                    AtlTrace( _T(
"%s\n"), szName );
                }
            }
            dwSize 
= _countof(szGuid) * sizeof(TCHAR);
            dwIndex
++;
        }

        regKey.Close();
    }
#else
    HKEY hKey;
    
if( ERROR_SUCCESS == ::RegOpenKeyEx( HKEY_CLASSES_ROOT, _T("Interface"), 0, KEY_QUERY_VALUE | KEY_ENUMERATE_SUB_KEYS, &hKey) ) 
    {
        TCHAR   szGuid[
128];
        DWORD   dwIndex 
= 0;
        DWORD   dwSize 
= _countof(szGuid) * sizeof(TCHAR);

        
while( ERROR_SUCCESS == ::RegEnumKeyEx(hKey, dwIndex, szGuid, &dwSize, NULL, NULL,NULL, NULL ) ) 
        {
            IID iid;
            ::IIDFromString( CT2OLE(szGuid), 
&iid);

            CComPtr
<IUnknown> spUnk;
            
if( SUCCEEDED( pUnk->QueryInterface( iid, (LPVOID*)&spUnk ) ) ) 
            {
                TCHAR szName[
128];
                
if( GetInterfaceName( iid, szName ) )
                {
                    AtlTrace( _T(
"%s\n"), szName );
                }
            }

            dwSize 
= _countof(szGuid) * sizeof(TCHAR);
            dwIndex
++;
        }
        ::RegCloseKey(hKey);
    }

#endif
}

 

至此已经完成,当然这种方案缺点在于还有很多定制的接口并没有在注册表中注册,因此不够彻底。不过有胜于无了

posted @ 2009-11-09 23:51  孤竹君  阅读(946)  评论(0编辑  收藏  举报