COM 对象函数上的 API 挂钩？

Question

您好，StackOverflowians，

正如此处所发现的那样，Windows 7 存在一个错误，即 Windows 资源管理器实例不会触发 DISPID_BEFORENAVIGATE2 事件。此事件允许 shell 扩展在导航即将发生时收到通知，并且（对我来说最重要的是）有机会取消导航。我已经寻找一种解决方法很长一段时间了，我想我找到了一个。但是，我想获得一些关于它的安全性的意见。

我最近经常使用 API 挂钩，并且已经使用它为我的扩展挂钩了一些函数。我注意到IShellBrowser中有一个函数 控制导航。起初我以为你不能挂钩这样的东西，但在阅读了 COM 对象的布局 我意识到，只需从任何活动实例的 vtable 中获取正确的函数指针就可以实现这一点。果然，这就像做梦一样。设置钩子后，所有资源管理器窗口中的所有导航都直接通过我的绕行函数运行，我可以根据它们的目标 pidl 决定是否拒绝它们。

所以我的问题是，有什么理由我不应该这样做吗？我从未听说过 API 挂钩用于挂钩 COM 对象函数。有没有什么情况下它不起作用？危险吗？ （至少比常规 API 挂钩要多）

相关代码如下。我正在使用 MinHook，这是一个简约的挂钩库，它使用经过验证的- 蹦床函数的真实方法。

typedef HRESULT (WINAPI *BROWSEOBJECT)(IShellBrowser*, PCUIDLIST_RELATIVE, UINT);
HRESULT WINAPI DetourBrowseObject(IShellBrowser* _this, PCUIDLIST_RELATIVE pidl, UINT wFlags);
BROWSEOBJECT fpBrowseObject = NULL;
BROWSEOBJECT ShellBrowser_BrowseObject = NULL;

bool Initialize() {
    if(MH_Initialize() != MH_OK) {
        return false;
    }

    // Get a reference to an existing IShellBrowser.  Any instance will do.
    // ShellBrowser enum code taken from The Old New Thing
    IShellWindows *psw;
    BOOL fFound = FALSE;
    if (SUCCEEDED(CoCreateInstance(CLSID_ShellWindows, NULL, CLSCTX_ALL, IID_IShellWindows, (void**)&psw))) {
        VARIANT v;
        V_VT(&v) = VT_I4;
        IDispatch  *pdisp;
        for (V_I4(&v) = 0; !fFound && psw->Item(v, &pdisp) == S_OK; V_I4(&v)++) {
            IWebBrowserApp *pwba;
            if (SUCCEEDED(pdisp->QueryInterface(IID_IWebBrowserApp, (void**)&pwba))) {
                IServiceProvider *psp;
                if (SUCCEEDED(pwba->QueryInterface(IID_IServiceProvider, (void**)&psp))) {
                    IShellBrowser *psb;
                    if (SUCCEEDED(psp->QueryService(SID_STopLevelBrowser,IID_IShellBrowser, (void**)&psb))) {
                        fFound = true;

                        // Grab the 11th entry in the VTable, which is BrowseObject
                        void** vtable = (*(void***)(psb));
                        ShellBrowser_BrowseObject = (BROWSEOBJECT)(vtable[11]);
                        psb->Release();
                    }
                    psp->Release();
                }
                pwba->Release();
            }
            pdisp->Release();
        }
        psw->Release();
    }

    if(fFound) {
        if(MH_CreateHook(ShellBrowser_BrowseObject, &DetourBrowseObject, reinterpret_cast<void**>(&fpBrowseObject)) != MH_OK) {
            return false;
        }
        if(MH_EnableHook(ShellBrowser_BrowseObject) != MH_OK) {
            return false;
        }
    }
    return true;
}

HRESULT WINAPI DetourBrowseObject(IShellBrowser* _this, PCUIDLIST_RELATIVE pidl, UINT wFlags) {
    if(NavigateIsOkay(pidl, wFlags)) {
        return fpBrowseObject(_this, pidl, wFlags);
    }
    else {
        return S_FALSE;
    }    
}

Answer 1

我从未听说过使用 API 挂钩
挂钩 COM 对象函数。

COM 对象的成员函数实际上并没有那么不同，如果您坚持通常的挂钩准则，实际上可以很好地挂钩。几年前，我必须连接专有 CRM 解决方案的 COM 组件才能将其连接到数据库服务器。该应用程序运行良好，并且已经稳定运行了好几年。