⭐ 欢迎来到虫虫下载站! | 📦 资源下载 📁 资源专辑 ℹ️ 关于我们
⭐ 虫虫下载站

📄 startmnu.c

📁 ReactOS是一些高手根据Windows XP的内核编写出的类XP。内核实现机理和API函数调用几乎相同。甚至可以兼容XP的程序。喜欢研究系统内核的人可以看一看。
💻 C
📖 第 1 页 / 共 2 页
字号:
/*
 * ReactOS Explorer
 *
 * Copyright 2006 - 2007 Thomas Weidenmueller <w3seek@reactos.org>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#include <precomp.h>

/*
 * Start menu button context menu
 */

typedef struct _STARTMNU_CTMENU_CTX
{
    IContextMenu *pcm;
    LPITEMIDLIST pidl;
} STARTMNU_CTMENU_CTX, *PSTARTMNU_CTMENU_CTX;

static HMENU
CreateStartContextMenu(IN HWND hWndOwner,
                       IN PVOID *ppcmContext,
                       IN PVOID Context  OPTIONAL);

static VOID
OnStartContextMenuCommand(IN HWND hWndOwner,
                          IN UINT uiCmdId,
                          IN PVOID pcmContext  OPTIONAL,
                          IN PVOID Context  OPTIONAL);

const TRAYWINDOW_CTXMENU StartMenuBtnCtxMenu = {
    CreateStartContextMenu,
    OnStartContextMenuCommand
};

static HMENU
CreateContextMenuFromShellFolderPidl(IN HWND hWndOwner,
                                     IN OUT IShellFolder *psf,
                                     IN OUT LPITEMIDLIST pidl,
                                     OUT IContextMenu **ppcm)
{
    IContextMenu *pcm;
    HRESULT hRet;
    HMENU hPopup;

    hRet = IShellFolder_GetUIObjectOf(psf,
                                      hWndOwner,
                                      1,
                                      (LPCITEMIDLIST*)&pidl, /* FIXME: shouldn't need a typecast! */
                                      &IID_IContextMenu,
                                      NULL,
                                      (PVOID*)&pcm);
    if (SUCCEEDED(hRet))
    {
        hPopup = CreatePopupMenu();

        if (hPopup != NULL)
        {
            hRet = IContextMenu_QueryContextMenu(pcm,
                                                 hPopup,
                                                 0,
                                                 ID_SHELL_CMD_FIRST,
                                                 ID_SHELL_CMD_LAST,
                                                 CMF_VERBSONLY);

            if (SUCCEEDED(hRet))
            {
                *ppcm = pcm;
                return hPopup;
            }

            IContextMenu_Release(pcm);
            DestroyMenu(hPopup);
        }
    }

    return NULL;
}

static VOID
OnStartContextMenuCommand(IN HWND hWndOwner,
                          IN UINT uiCmdId,
                          IN PVOID pcmContext  OPTIONAL,
                          IN PVOID Context  OPTIONAL)
{
    PSTARTMNU_CTMENU_CTX psmcmc = (PSTARTMNU_CTMENU_CTX)pcmContext;

    if (uiCmdId != 0)
    {
        switch (uiCmdId)
        {
            case ID_SHELL_CMD_FIRST ... ID_SHELL_CMD_LAST:
            {
                CMINVOKECOMMANDINFO cmici = {0};
                CHAR szDir[MAX_PATH];

                /* Setup and invoke the shell command */
                cmici.cbSize = sizeof(cmici);
                cmici.hwnd = hWndOwner;
                cmici.lpVerb = (LPCSTR)MAKEINTRESOURCE(uiCmdId - ID_SHELL_CMD_FIRST);
                cmici.nShow = SW_NORMAL;

                /* FIXME: Support Unicode!!! */
                if (SHGetPathFromIDListA(psmcmc->pidl,
                                         szDir))
                {
                    cmici.lpDirectory = szDir;
                }

                IContextMenu_InvokeCommand(psmcmc->pcm,
                                           &cmici);
                break;
            }

            default:
                ITrayWindow_ExecContextMenuCmd((ITrayWindow *)Context,
                                               uiCmdId);
                break;
        }
    }

    IContextMenu_Release(psmcmc->pcm);

    HeapFree(hProcessHeap,
             0,
             psmcmc);
}

static VOID
AddStartContextMenuItems(IN HWND hWndOwner,
                         IN HMENU hPopup)
{
    TCHAR szBuf[MAX_PATH];
    HRESULT hRet;

    /* Add the "Open All Users" menu item */
    if (LoadString(hExplorerInstance,
                   IDS_PROPERTIES,
                   szBuf,
                   sizeof(szBuf) / sizeof(szBuf[0])))
    {
        AppendMenu(hPopup,
                   MF_STRING,
                   ID_SHELL_CMD_PROPERTIES,
                   szBuf);
    }

    if (!SHRestricted(REST_NOCOMMONGROUPS))
    {
        /* Check if we should add menu items for the common start menu */
        hRet = SHGetFolderPath(hWndOwner,
                               CSIDL_COMMON_STARTMENU,
                               NULL,
                               SHGFP_TYPE_CURRENT,
                               szBuf);
        if (SUCCEEDED(hRet) && hRet != S_FALSE)
        {
            /* The directory exists, but only show the items if the
               user can actually make any changes to the common start
               menu. This is most likely only the case if the user
               has administrative rights! */
            if (IsUserAnAdmin())
            {
                AppendMenu(hPopup,
                           MF_SEPARATOR,
                           0,
                           NULL);

                /* Add the "Open All Users" menu item */
                if (LoadString(hExplorerInstance,
                               IDS_OPEN_ALL_USERS,
                               szBuf,
                               sizeof(szBuf) / sizeof(szBuf[0])))
                {
                    AppendMenu(hPopup,
                               MF_STRING,
                               ID_SHELL_CMD_OPEN_ALL_USERS,
                               szBuf);
                }

                /* Add the "Explore All Users" menu item */
                if (LoadString(hExplorerInstance,
                               IDS_EXPLORE_ALL_USERS,
                               szBuf,
                               sizeof(szBuf) / sizeof(szBuf[0])))
                {
                    AppendMenu(hPopup,
                               MF_STRING,
                               ID_SHELL_CMD_EXPLORE_ALL_USERS,
                               szBuf);
                }
            }
        }
    }
}

static HMENU
CreateStartContextMenu(IN HWND hWndOwner,
                       IN PVOID *ppcmContext,
                       IN PVOID Context  OPTIONAL)
{
    LPITEMIDLIST pidlStart, pidlLast;
    IShellFolder *psfStart, *psfDesktop;
    IContextMenu *pcm;
    HRESULT hRet;
    HMENU hPopup;

    pidlStart = SHCloneSpecialIDList(hWndOwner,
                                     CSIDL_STARTMENU,
                                     TRUE);

    if (pidlStart != NULL)
    {
        pidlLast = ILClone(ILFindLastID(pidlStart));
        ILRemoveLastID(pidlStart);

        if (pidlLast != NULL)
        {
            hRet = SHGetDesktopFolder(&psfDesktop);
            if (SUCCEEDED(hRet))
            {
                hRet = IShellFolder_BindToObject(psfDesktop,
                                                 pidlStart,
                                                 NULL,
                                                 (REFIID)&IID_IShellFolder, /* FIXME: Shouldn't require a typecast */
                                                 (PVOID*)&psfStart);
                if (SUCCEEDED(hRet))
                {
                    hPopup = CreateContextMenuFromShellFolderPidl(hWndOwner,
                                                                  psfStart,
                                                                  pidlLast,
                                                                  &pcm);

                    if (hPopup != NULL)
                    {
                        PSTARTMNU_CTMENU_CTX psmcmc;

                        psmcmc = HeapAlloc(hProcessHeap,
                                           0,
                                           sizeof(*psmcmc));
                        if (psmcmc != NULL)
                        {
                            psmcmc->pcm = pcm;
                            psmcmc->pidl = pidlLast;

                            AddStartContextMenuItems(hWndOwner,
                                                     hPopup);

                            *((PSTARTMNU_CTMENU_CTX*)ppcmContext) = psmcmc;
                            return hPopup;
                        }
                        else
                        {
                            IContextMenu_Release(pcm);

                            DestroyMenu(hPopup);
                            hPopup = NULL;
                        }
                    }

                    IShellFolder_Release(psfStart);
                }

                IShellFolder_Release(psfDesktop);
            }

            ILFree(pidlLast);
        }

        ILFree(pidlStart);
    }

    return NULL;
}

/*****************************************************************************
 ** IStartMenuSite ***********************************************************
 *****************************************************************************/

static const IStartMenuSiteVtbl IStartMenuSiteImpl_Vtbl;
static const IServiceProviderVtbl IServiceProviderImpl_Vtbl;
static const IStartMenuCallbackVtbl IStartMenuCallbackImpl_Vtbl;
static const IOleCommandTargetVtbl IOleCommandTargetImpl_Vtbl;

typedef struct
{
    const IStartMenuSiteVtbl *lpVtbl;
    const IServiceProviderVtbl *lpServiceProviderVtbl;
    const IStartMenuCallbackVtbl *lpStartMenuCallbackVtbl;
    const IOleCommandTargetVtbl *lpOleCommandTargetVtbl;
    LONG Ref;

    ITrayWindow *Tray;
} IStartMenuSiteImpl;

static IUnknown *
IUnknown_from_IStartMenuSiteImpl(IStartMenuSiteImpl *This)
{
    return (IUnknown *)&This->lpVtbl;
}

IMPL_CASTS(IStartMenuSite, IStartMenuSite, lpVtbl)
IMPL_CASTS(IServiceProvider, IStartMenuSite, lpServiceProviderVtbl)
IMPL_CASTS(IStartMenuCallback, IStartMenuSite, lpStartMenuCallbackVtbl)
IMPL_CASTS(IOleCommandTarget, IStartMenuSite, lpOleCommandTargetVtbl)

/*******************************************************************/

static ULONG STDMETHODCALLTYPE
IStartMenuSiteImpl_AddRef(IN OUT IStartMenuSite *iface)
{
    IStartMenuSiteImpl *This = IStartMenuSiteImpl_from_IStartMenuSite(iface);

    return InterlockedIncrement(&This->Ref);
}

static VOID
IStartMenuSiteImpl_Free(IN OUT IStartMenuSiteImpl *This)
{
    HeapFree(hProcessHeap,
             0,
             This);
}

static ULONG STDMETHODCALLTYPE
IStartMenuSiteImpl_Release(IN OUT IStartMenuSite *iface)
{
    IStartMenuSiteImpl *This = IStartMenuSiteImpl_from_IStartMenuSite(iface);
    ULONG Ret;

    Ret = InterlockedDecrement(&This->Ref);

    if (Ret == 0)
        IStartMenuSiteImpl_Free(This);

    return Ret;
}

static HRESULT STDMETHODCALLTYPE
IStartMenuSiteImpl_QueryInterface(IN OUT IStartMenuSite *iface,
                                  IN REFIID riid,
                                  OUT LPVOID *ppvObj)
{
    IStartMenuSiteImpl *This;

    if (ppvObj == NULL)
        return E_POINTER;

    This = IStartMenuSiteImpl_from_IStartMenuSite(iface);

    if (IsEqualIID(riid,
                   &IID_IUnknown))
    {
        *ppvObj = IUnknown_from_IStartMenuSiteImpl(This);
    }
    else if (IsEqualIID(riid,
                        &IID_IServiceProvider))
    {
        *ppvObj = IServiceProvider_from_IStartMenuSiteImpl(This);
    }
    else if (IsEqualIID(riid,
                        &IID_IStartMenuCallback) ||
             IsEqualIID(riid,
                        &IID_IOleWindow))
    {
        *ppvObj = IStartMenuCallback_from_IStartMenuSiteImpl(This);
    }
    else if (IsEqualIID(riid,
                        &IID_IOleCommandTarget))
    {
        *ppvObj = IOleCommandTarget_from_IStartMenuSiteImpl(This);
    }
    else
    {
        DbgPrint("IStartMenuSite::QueryInterface queried unsupported interface: "
                 "{0x%8x,0x%4x,0x%4x,{0x%2x,0x%2x,0x%2x,0x%2x,0x%2x,0x%2x,0x%2x,0x%2x}}\n",
                 riid->Data1, riid->Data2, riid->Data3, riid->Data4[0], riid->Data4[1],
                 riid->Data4[2], riid->Data4[3], riid->Data4[4], riid->Data4[5],
                 riid->Data4[6], riid->Data4[7]);
        *ppvObj = NULL;
        return E_NOINTERFACE;
    }

    IStartMenuSiteImpl_AddRef(iface);
    return S_OK;
}

static const IStartMenuSiteVtbl IStartMenuSiteImpl_Vtbl =
{
    /*** IUnknown methods ***/
    IStartMenuSiteImpl_QueryInterface,
    IStartMenuSiteImpl_AddRef,
    IStartMenuSiteImpl_Release,
    /*** IStartMenuSite methods ***/
};

/*******************************************************************/

METHOD_IUNKNOWN_INHERITED_ADDREF(IServiceProvider, IStartMenuSite)
METHOD_IUNKNOWN_INHERITED_RELEASE(IServiceProvider, IStartMenuSite)
METHOD_IUNKNOWN_INHERITED_QUERYINTERFACE(IServiceProvider, IStartMenuSite)

static HRESULT STDMETHODCALLTYPE
IStartMenuSiteImpl_QueryService(IN OUT IServiceProvider *iface,
                                IN REFGUID guidService,
                                IN REFIID riid,
                                OUT PVOID *ppvObject)
{
    IStartMenuSiteImpl *This = IStartMenuSiteImpl_from_IServiceProvider(iface);

    if (IsEqualGUID(guidService,
                    &SID_SMenuPopup))
    {
        return IStartMenuSiteImpl_QueryInterface(IStartMenuSite_from_IStartMenuSiteImpl(This),
                                                 riid,
                                                 ppvObject);
    }

    return E_NOINTERFACE;
}

static const IServiceProviderVtbl IServiceProviderImpl_Vtbl =
{
    /*** IUnknown methods ***/
    METHOD_IUNKNOWN_INHERITED_QUERYINTERFACE_NAME(IServiceProvider, IStartMenuSite),
    METHOD_IUNKNOWN_INHERITED_ADDREF_NAME(IServiceProvider, IStartMenuSite),
    METHOD_IUNKNOWN_INHERITED_RELEASE_NAME(IServiceProvider, IStartMenuSite),
    /*** IServiceProvider methods ***/
    IStartMenuSiteImpl_QueryService
};

/*******************************************************************/

METHOD_IUNKNOWN_INHERITED_ADDREF(IStartMenuCallback, IStartMenuSite)
METHOD_IUNKNOWN_INHERITED_RELEASE(IStartMenuCallback, IStartMenuSite)
METHOD_IUNKNOWN_INHERITED_QUERYINTERFACE(IStartMenuCallback, IStartMenuSite)

static HRESULT STDMETHODCALLTYPE
IStartMenuSiteImpl_GetWindow(IN OUT IStartMenuCallback *iface,
                             OUT HWND *phwnd)
{
    IStartMenuSiteImpl *This = IStartMenuSiteImpl_from_IStartMenuCallback(iface);
    DbgPrint("IStartMenuCallback::GetWindow\n");

    *phwnd = ITrayWindow_GetHWND(This->Tray);
    if (*phwnd != NULL)
        return S_OK;

    return E_FAIL;
}

static HRESULT STDMETHODCALLTYPE
IStartMenuSiteImpl_ContextSensitiveHelp(IN OUT IStartMenuCallback *iface,

⌨️ 快捷键说明

复制代码 Ctrl + C
搜索代码 Ctrl + F
全屏模式 F11
切换主题 Ctrl + Shift + D
显示快捷键 ?
增大字号 Ctrl + =
减小字号 Ctrl + -