[NTOS:SE] Capture the groups length when creating a token
[reactos.git] / dll / shellext / fontext / CFontMenu.cpp
1 /*
2 * PROJECT: ReactOS Font Shell Extension
3 * LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
4 * PURPOSE: CFontMenu implementation
5 * COPYRIGHT: Copyright 2019,2020 Mark Jansen (mark.jansen@reactos.org)
6 */
7
8 #include "precomp.h"
9
10 WINE_DEFAULT_DEBUG_CHANNEL(fontext);
11
12 static CLIPFORMAT g_cfHIDA;
13
14 HRESULT _GetCidlFromDataObject(IDataObject *pDataObject, CIDA** ppcida)
15 {
16 if (g_cfHIDA == NULL)
17 {
18 g_cfHIDA = (CLIPFORMAT)RegisterClipboardFormatW(CFSTR_SHELLIDLIST);
19 }
20
21 FORMATETC fmt = { g_cfHIDA, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
22 STGMEDIUM medium;
23
24 HRESULT hr = pDataObject->GetData(&fmt, &medium);
25 if (FAILED_UNEXPECTEDLY(hr))
26 return hr;
27
28 LPVOID lpSrc = GlobalLock(medium.hGlobal);
29 SIZE_T cbSize = GlobalSize(medium.hGlobal);
30
31 *ppcida = (CIDA *)::CoTaskMemAlloc(cbSize);
32 if (*ppcida)
33 {
34 memcpy(*ppcida, lpSrc, cbSize);
35 hr = S_OK;
36 }
37 else
38 {
39 hr = E_FAIL;
40 }
41 ReleaseStgMedium(&medium);
42 return hr;
43 }
44
45 const char* DFM_TO_STR(UINT uMsg)
46 {
47 switch(uMsg)
48 {
49 case DFM_MERGECONTEXTMENU: return "DFM_MERGECONTEXTMENU";
50 case DFM_INVOKECOMMAND: return "DFM_INVOKECOMMAND";
51 case DFM_MODIFYQCMFLAGS: return "DFM_MODIFYQCMFLAGS";
52 case DFM_MERGECONTEXTMENU_TOP: return "DFM_MERGECONTEXTMENU_TOP";
53 case DFM_MERGECONTEXTMENU_BOTTOM: return "DFM_MERGECONTEXTMENU_BOTTOM";
54 case DFM_GETHELPTEXTW: return "DFM_GETHELPTEXTW";
55 case DFM_GETVERBW: return "DFM_GETVERBW";
56 case DFM_GETVERBA: return "DFM_GETVERBA";
57 case DFM_WM_INITMENUPOPUP: return "DFM_WM_INITMENUPOPUP";
58 case DFM_INVOKECOMMANDEX: return "DFM_INVOKECOMMANDEX";
59 case DFM_GETDEFSTATICID: return "DFM_GETDEFSTATICID";
60 case 3: return "MENU_BEGIN";
61 case 4: return "MENU_END";
62 default: return "";
63 }
64 }
65
66
67 static void RunFontViewer(HWND hwnd, const FontPidlEntry* fontEntry)
68 {
69 WCHAR FontViewerPath[MAX_PATH] = L"%SystemRoot%\\System32\\fontview.exe";
70 WCHAR FontPathArg[MAX_PATH + 3];
71
72 CStringW Path = g_FontCache->Filename(fontEntry, true);
73 if (!Path.IsEmpty())
74 {
75 // '/d' disables the install button
76 StringCchPrintfW(FontPathArg, _countof(FontPathArg), L"/d %s", Path.GetString());
77 PathQuoteSpacesW(FontPathArg + 3);
78
79 SHELLEXECUTEINFOW si = { sizeof(si) };
80 si.fMask = SEE_MASK_DOENVSUBST;
81 si.hwnd = hwnd;
82 si.lpFile = FontViewerPath;
83 si.lpParameters = FontPathArg;
84 si.nShow = SW_SHOWNORMAL;
85 ShellExecuteExW(&si);
86 }
87 }
88
89 static HRESULT CALLBACK FontFolderMenuCallback(IShellFolder *psf, HWND hwnd, IDataObject *pdtobj,
90 UINT uMsg, WPARAM wParam, LPARAM lParam)
91 {
92 TRACE("FontFolderMenuCallback(%u {%s})\n", uMsg, DFM_TO_STR(uMsg));
93 switch (uMsg)
94 {
95 case DFM_MERGECONTEXTMENU:
96 {
97 QCMINFO *pqcminfo = (QCMINFO *)lParam;
98
99 CStringW menuText(MAKEINTRESOURCEW(IDS_FONT_PREVIEW));
100 MENUITEMINFOW cmi = { sizeof(cmi) };
101 cmi.fMask = MIIM_ID | MIIM_STRING | MIIM_STATE;
102 cmi.fType = MFT_STRING;
103 cmi.fState = MFS_DEFAULT;
104 cmi.wID = pqcminfo->idCmdFirst++;
105 cmi.dwTypeData = (LPWSTR)menuText.GetString();
106 InsertMenuItemW(pqcminfo->hmenu, pqcminfo->indexMenu, TRUE, &cmi);
107
108 return S_OK;
109 }
110 case DFM_INVOKECOMMAND:
111 // Preview is the only item we can handle
112 if (wParam == 0)
113 {
114 CComHeapPtr<CIDA> cida;
115 HRESULT hr = _GetCidlFromDataObject(pdtobj, &cida);
116 if (FAILED_UNEXPECTEDLY(hr))
117 return hr;
118
119 for (UINT n = 0; n < cida->cidl; ++n)
120 {
121 const FontPidlEntry* fontEntry = _FontFromIL(HIDA_GetPIDLItem(cida, n));
122 RunFontViewer(hwnd, fontEntry);
123 }
124 return S_OK;
125 }
126 return S_FALSE;
127
128 case DFM_INVOKECOMMANDEX:
129 return E_NOTIMPL;
130 case DFM_GETDEFSTATICID: // Required for Windows 7 to pick a default
131 return S_FALSE;
132 }
133 return E_NOTIMPL;
134 }
135
136
137 HRESULT _CFontMenu_CreateInstance(HWND hwnd, UINT cidl, PCUITEMID_CHILD_ARRAY apidl,
138 IShellFolder *psf, REFIID riid, LPVOID* ppvOut)
139 {
140 if (cidl > 0)
141 {
142 HKEY keys[1] = {0};
143 int nkeys = 0;
144 CComPtr<IContextMenu> spMenu;
145
146 // Use the default context menu handler, but augment it from the callbacks
147 HRESULT hr = CDefFolderMenu_Create2(NULL, hwnd, cidl, apidl, psf, FontFolderMenuCallback, nkeys, keys, &spMenu);
148
149 if (FAILED_UNEXPECTEDLY(hr))
150 return hr;
151
152 // See if the requested interface (e.g. IContextMenu3) is also available
153 return spMenu->QueryInterface(riid, ppvOut);
154 }
155
156 // We can't create a background menu
157 return E_FAIL;
158 }
159