Changeset 715

Timestamp:

12/05/08 04:37:43 (4 years ago)

Author:

lowjoel

Message:

Defined a GetHKeyPath function which translates the HKEY parameter from IShellExtInit::Initialize to a registry key name. This marks the beginning of us allowing our shell extension to be integrated into the Explorer Recycle Bin context menu (okay, it's a little hackish, but let's see how it plays out. All that I needed was a debugger into the shell extension, no other hacks :P)

Location:

branches/eraser6/ShellExt

Files:

• CtxMenu.cpp (modified) (6 diffs)
• CtxMenu.h (modified) (3 diffs)
• ShellExt_i.c (modified) (1 diff)
• ShellExt_i.h (modified) (1 diff)

branches/eraser6/ShellExt/CtxMenu.cpp

@@ -25 +25 @@
 #include <sstream>
 
+extern "C"
+{
+    typedef LONG NTSTATUS;
+    enum KEY_INFORMATION_CLASS
+    {
+        KeyBasicInformation,
+        KeyNodeInformation,
+        KeyFullInformation,
+        KeyNameInformation,
+        KeyCachedInformation,
+        KeyVirtualizationInformation
+    };
+
+    struct KEY_BASIC_INFORMATION
+    {
+        LARGE_INTEGER LastWriteTime;
+        ULONG TitleIndex;
+        ULONG NameLength;
+        WCHAR Name[1];
+    };
+
+    typedef NTSTATUS (*pZwQueryKey)(HANDLE KeyHandle, KEY_INFORMATION_CLASS KeyInformationClass,
+        PVOID KeyInformation, ULONG Length, PULONG ResultLength);
+    pZwQueryKey ZwQueryKey;
+}
+
 template<typename handleType> class Handle
 {
@@ -51 +77 @@
 
     HRESULT CCtxMenu::Initialize(LPCITEMIDLIST pidlFolder, LPDATAOBJECT pDataObj,
-                                 HKEY /*hProgID*/)
-    {
-        m_itemID      = 0;
-        FORMATETC fmt = { CF_HDROP, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
-        STGMEDIUM stg = { TYMED_HGLOBAL };
-        HDROP     hDrop;
-
-        //Check pidlFolder for the drop path, if it exists.
-        if (pidlFolder != NULL)
-        {
+                                 HKEY hProgID)
+    {
+        //Initialise member variables.
+        m_itemID = 0;
+        
+        //Determine where the shell extension was invoked from.
+        if (GetHKeyPath(hProgID) == L"{645FF040-5081-101B-9F08-00AA002F954E}")
+        {
+            InvokeReason = INVOKEREASON_RECYCLEBIN;
+
+            //We can't do much other processing: the LPDATAOBJECT parameter contains
+            //data that is a private type so we don't know how to query for it.
+            return S_OK;
+        }
+
+        //Check pidlFolder for the drop path, if it exists. This is for drag-and-drop
+        //context menus.
+        else if (pidlFolder != NULL)
+        {
+            InvokeReason = INVOKEREASON_DRAGDROP;
+
             //Translate the drop path to a location on the filesystem.
             wchar_t dropTargetPath[MAX_PATH];
@@ -69 +106 @@
         }
 
+        //Okay, everything else is a simple context menu for a set of selected files/
+        //folders/drives.
+        else
+            InvokeReason = INVOKEREASON_FILEFOLDER;
+
         //Look for CF_HDROP data in the data object.
+        FORMATETC fmt = { CF_HDROP, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
+        STGMEDIUM stg = { TYMED_HGLOBAL };
         if (FAILED(pDataObj->GetData(&fmt, &stg)))
             //Nope! Return an "invalid argument" error back to Explorer.
-            return E_INVALIDARG;
+            return S_OK;
 
         //Get a pointer to the actual data.
-        hDrop = static_cast<HDROP>(GlobalLock(stg.hGlobal));
-
-        //Make sure it worked.
+        HDROP hDrop = static_cast<HDROP>(GlobalLock(stg.hGlobal));
         if (hDrop == NULL)
             return E_INVALIDARG;
@@ -90 +132 @@
         }
 
+        //Collect all the files which have been selected.
         HRESULT hr = S_OK;
         WCHAR buffer[MAX_PATH] = {0};
@@ -104 +147 @@
         }
 
+        //Clean up.
         GlobalUnlock(stg.hGlobal);
         ReleaseStgMedium(&stg);
         return hr;
     }
-    /*
-    +-------------------+
-    |                   |
-    |                   |
-    |                   |
-    |                   |
-    +-------------------+    +-------------------+
-    |(ICON) Eraser v6 > |    | Erase selected    | //--> erase the files immediately using defaults
-    +-------------------+    | Schedule Selected | //--> open the scheduler menu, with files/folders filled in
-    |                   |    +-------------------+
-    |                   |    | Secure move       | //--> secure move the files
-    |                   |    +-------------------+ // Eraser.Manager Algorithms popup
-    |                   |    |(*) Customise      | // set algorithm for this query only
-    +-------------------+    +-------------------+
-    */
 
     HRESULT CCtxMenu::QueryContextMenu(HMENU hmenu, UINT uMenuIndex, UINT uidFirstCmd,
@@ -534 +563 @@
     }
 
+    std::wstring CCtxMenu::GetHKeyPath(HKEY handle)
+    {
+        ZwQueryKey = reinterpret_cast<pZwQueryKey>(GetProcAddress(
+            LoadLibrary(L"Ntdll.dll"), "ZwQueryKey"));
+        ULONG keyInfoSize = sizeof(KEY_BASIC_INFORMATION);
+        KEY_BASIC_INFORMATION* keyInfo = NULL;
+        NTSTATUS queryResult = ERROR_MORE_DATA;
+
+        while (queryResult == ERROR_MORE_DATA)
+        {
+            delete[] keyInfo;
+            keyInfo = reinterpret_cast<KEY_BASIC_INFORMATION*>(
+                new char[keyInfoSize += 512]);
+            ::ZeroMemory(keyInfo, keyInfoSize);
+            queryResult = ZwQueryKey(handle, KeyBasicInformation, keyInfo,
+                keyInfoSize, &keyInfoSize);
+        }
+
+        std::wstring result(keyInfo->Name, keyInfoSize -
+            sizeof(KEY_BASIC_INFORMATION) + 1);
+        delete[] keyInfo;
+        return result;
+    }
+
     MENUITEMINFO* CCtxMenu::GetSeparator()
     {

branches/eraser6/ShellExt/CtxMenu.h

@@ -49 +49 @@
         }
 
+        /// The place where the context menu extension was invoked.
+        enum InvokeReasons
+        {
+            INVOKEREASON_FILEFOLDER,
+            INVOKEREASON_DRAGDROP,
+            INVOKEREASON_RECYCLEBIN
+        };
+
         enum CEraserLPVERBS
         {
@@ -75 +83 @@
         CEraserLPVERBS GetApplicableActions();
 
+        static std::wstring GetHKeyPath(HKEY handle);
         static MENUITEMINFO* GetSeparator();
         static HICON GetMenuIcon();
@@ -81 +90 @@
 
     protected:
+        InvokeReasons InvokeReason;
+
         UINT        m_itemID;
         string_list m_szSelectedFiles;

branches/eraser6/ShellExt/ShellExt_i.c

@@ -7 +7 @@
 
  /* File created by MIDL compiler version 7.00.0500 */
-/* at Tue Dec 02 22:04:21 2008
+/* at Fri Dec 05 11:15:23 2008
  */
 /* Compiler settings for .\DllMain.idl:
-    Oicf, W1, Zp8, env=Win64 (32b run)
+    Oicf, W1, Zp8, env=Win32 (32b run)
     protocol : dce , ms_ext, c_ext, robust
-    error checks: allocation ref bounds_check enum stub_data 
+    error checks: stub_data 
     VC __declspec() decoration level: 
          __declspec(uuid()), __declspec(selectany), __declspec(novtable)

branches/eraser6/ShellExt/ShellExt_i.h

@@ -5 +5 @@
 
  /* File created by MIDL compiler version 7.00.0500 */
-/* at Tue Dec 02 22:04:21 2008
+/* at Fri Dec 05 11:15:23 2008
  */
 /* Compiler settings for .\DllMain.idl:
-    Oicf, W1, Zp8, env=Win64 (32b run)
+    Oicf, W1, Zp8, env=Win32 (32b run)
     protocol : dce , ms_ext, c_ext, robust
-    error checks: allocation ref bounds_check enum stub_data 
+    error checks: stub_data 
     VC __declspec() decoration level: 
          __declspec(uuid()), __declspec(selectany), __declspec(novtable)