在Windows 7中的shell新上下文菜单中添加非文件类型

#include<windows.h>
#include <shlobj.h>

class CMyContextMenu : public IShellExtInit, public IContextMenu
{
public:
    // IUnknown methods
    STDMETHODIMP QueryInterface(REFIID riid, void **ppvObject);
    STDMETHODIMP_(ULONG) AddRef(void);
    STDMETHODIMP_(ULONG) Release(void);

    // IShellExtInit methods
    STDMETHODIMP Initialize(LPCITEMIDLIST pidlFolder, LPDATAOBJECT pDataObj, HKEY hKeyProgID);

    // IContextMenu methods
    STDMETHODIMP GetCommandString(UINT_PTR idCmd, UINT uFlags, UINT *pwReserved, LPSTR pszName, UINT cchMax);
    STDMETHODIMP InvokeCommand(LPCMINVOKECOMMANDINFO pici);
    STDMETHODIMP QueryContextMenu(HMENU hMenu, UINT indexMenu, UINT idCmdFirst, UINT idCmdLast, UINT uFlags);

private:
    LONG m_cRef;
};

// Implement IUnknown methods
STDMETHODIMP CMyContextMenu::QueryInterface(REFIID riid, void **ppvObject)
{
    if (IsEqualIID(riid, IID_IUnknown) || IsEqualIID(riid, IID_IShellExtInit) || IsEqualIID(riid, IID_IContextMenu))
    {
        *ppvObject = this;
        AddRef();
        return S_OK;
    }

    *ppvObject = NULL;
    return E_NOINTERFACE;
}

STDMETHODIMP_(ULONG) CMyContextMenu::AddRef(void)
{
    return InterlockedIncrement(&m_cRef);
}

STDMETHODIMP_(ULONG) CMyContextMenu::Release(void)
{
    LONG cRef = InterlockedDecrement(&m_cRef);
    if (cRef == 0)
    {
        delete this;
    }

    return cRef;
}

// Implement IShellExtInit methods
STDMETHODIMP CMyContextMenu::Initialize(LPCITEMIDLIST pidlFolder, LPDATAOBJECT pDataObj, HKEY hKeyProgID)
{
    // TODO: Initialize the shell extension
    return S_OK;
}

// Implement IContextMenu methods
STDMETHODIMP CMyContextMenu::GetCommandString(UINT_PTR idCmd, UINT uFlags, UINT *pwReserved, LPSTR pszName, UINT cchMax)
{
    // TODO: Return the command string for the specified command
    return E_NOTIMPL;
}

STDMETHODIMP CMyContextMenu::InvokeCommand(LPCMINVOKECOMMANDINFO pici)
{
    // TODO: Execute the specified command
    return E_NOTIMPL;
}

STDMETHODIMP CMyContextMenu::QueryContextMenu(HMENU hMenu, UINT indexMenu, UINT idCmdFirst, UINT idCmdLast, UINT uFlags)
{
    // TODO: Add the custom menu item to the context menu
    return E_NOTIMPL;
}

// DLL export functions
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppvOut)
{
    if (IsEqualCLSID(rclsid, CLSID_MyContextMenu))
    {
        CMyContextMenu *pMyContextMenu = new CMyContextMenu;
        if (pMyContextMenu == NULL)
        {
            return E_OUTOFMEMORY;
        }

        HRESULT hr = pMyContextMenu->QueryInterface(riid, ppvOut);
        pMyContextMenu->Release();
        return hr;
    }

    return CLASS_E_CLASSNOTAVAILABLE;
}

STDAPI DllCanUnloadNow(void)
{
    return S_OK;
}

STDAPI DllRegisterServer(void)
{
    // TODO: Register the shell extension in the registry
    return S_OK;
}

STDAPI DllUnregisterServer(void)
{
    // TODO: Unregister the shell extension from the registry
    return S_OK;
}