6

问候 StackOverflowians,

正如在此处发现的那样,Windows 7 存在一个错误,在该错误中,Windows Explorer 实例不会触发 DISPID_BEFORENAVIGATE2 事件。此事件允许在导航即将发生时通知 shell 扩展,并且(对我来说最重要的是)有机会取消导航。很长一段时间以来,我一直在寻找一种解决方法,我想我找到了。但是,我想就它的安全性提出一些意见。

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

所以我的问题是,有什么理由我不应该这样做吗?我从未听说过用于挂钩 COM 对象函数的 API 挂钩。有没有它不起作用的情况?危险吗?(至少比常规的 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;
    }    
}
4

1 回答 1

2

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

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

于 2010-09-11T22:18:25.730 回答