枚举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
}

 

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