IDropTarget :: Метод отбрасывания

Я пытаюсь перехватить COM-метод IDropTarget :: Drop, но это не так, до этого я перехватывал метод IFileOperation :: RenameItems, он работает нормально, а для перехвата RenameItems я использовал следующий код:

#include "stdafx.h"#include <windows.h>
#include "MinHook.h"#include <Winternl.h>
#include <stdio.h>
#include <shlwapi.h>
#include <tchar.h>
#include <string.h>
#include <psapi.h>
#include <strsafe.h>
#include <Shobjidl.h>
#include <Shellapi.h>
#include <Shlguid.h>

#if defined _M_X64
#pragma comment(lib, "libMinHook.x64.lib")
bool __is32bitMachine = false ;
#elif defined _M_IX86
#pragma comment(lib, "libMinHook.x86.lib")
bool __is32bitMachine = true;
#endif

PVOID GetInterfaceMethod(PVOID intf, DWORD methodIndex)
{
if (__is32bitMachine)
return *(PVOID*)(*(DWORD_PTR*)intf + methodIndex * 4);
else
return *(PVOID*)(*(DWORD_PTR*)intf + methodIndex * 8);
}

typedef HRESULT (STDMETHODCALLTYPE *RenameItemsNext)(IFileOperation *pThis, IUnknown *pUnkItems, LPCWSTR pszNewName);
RenameItemsNext Real_RenameItems = NULL;
RenameItemsNext Actual_RenameItems ;

HRESULT STDMETHODCALLTYPE RenameItemsCallback(IFileOperation *pThis, IUnknown *pUnkItems, LPCWSTR pszNewName)
{
IID IID_ShellItemArray;
IID IID_DataObject;
CLSIDFromString(L"{b63ea76d-1f85-456f-a19c-48159efa858b}", &IID_ShellItemArray);
CLSIDFromString(L"{0000010E-0000-0000-C000-000000000046}", &IID_DataObject);

void *shellItemArray;
void *dataObject;
LPWSTR dstFN;

if (pUnkItems->QueryInterface(IID_ShellItemArray, &shellItemArray) == S_OK)
{
DWORD numItems = 0;
IShellItem *shell;
IShellItemArray *pItems = (IShellItemArray *) pUnkItems;
pItems->GetCount(&numItems);
for (int i = 0; i < numItems; i++)
{
pItems->GetItemAt(i, &shell);
shell->GetDisplayName(SIGDN_FILESYSPATH, &dstFN);
MessageBoxW(NULL, dstFN, L"Renamed From", MB_OK);
}
MessageBoxW(NULL, pszNewName, L"Renamed To", MB_OK);
}

if (pUnkItems->QueryInterface(IID_DataObject, &dataObject) == S_OK)
{
Real_SHCreateShell = (SHCreateShellNext)GetProcAddress(LoadLibraryA("shell32.dll"), "SHCreateShellItemArrayFromDataObject");
if (Real_SHCreateShell != NULL)
{
if (Real_SHCreateShell((IDataObject*)dataObject, IID_ShellItemArray, &shellItemArray) == S_OK)
{
DWORD numItems = 0;
IShellItem *shell;
IShellItemArray *pItems = (IShellItemArray *)shellItemArray;
pItems->GetCount(&numItems);
if (pItems->GetCount(&numItems) == S_OK)
{
for (int i = 0; i < numItems; i++)
{
if (pItems->GetItemAt(i, &shell) == S_OK)
{
shell->GetDisplayName(SIGDN_FILESYSPATH, &dstFN);
MessageBoxW(NULL, dstFN, L"Renamed From", MB_OK);
}
}
}
}
}
MessageBoxW(NULL, dstFN, L"Renamed To", MB_OK);
}
return Real_RenameItems(pThis, pUnkItems, pszNewName);
}

typedef HRESULT (WINAPI *COCREATEINSTANCE)(REFCLSID, LPUNKNOWN, DWORD, REFIID, LPVOID*);
COCREATEINSTANCE Real_CoCreateInstance = NULL;

HRESULT WINAPI CoCreateInstanceCallback(REFCLSID rclsid, LPUNKNOWN pUnkOuter, DWORD dwClsContext, REFIID riid, LPVOID *ppv)
{
const char *IFileOperation_GUID = "{3AD05575-8857-4850-9277-11B85BDB8E09}";
char GUIDString[64];

HRESULT HR = Real_CoCreateInstance(rclsid, pUnkOuter, dwClsContext, riid, ppv);

sprintf_s(GUIDString, 64, "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}\0",
rclsid.Data1, rclsid.Data2, rclsid.Data3,
rclsid.Data4[0], rclsid.Data4[1],
rclsid.Data4[2], rclsid.Data4[3],
rclsid.Data4[4], rclsid.Data4[5],
rclsid.Data4[6], rclsid.Data4[7]);

if (strcmp(GUIDString, IFileOperation_GUID) == 0)
{
if(Real_RenameItems == NULL)
{
Actual_RenameItems = (RenameItemsNext) GetInterfaceMethod(*ppv, 13);

if (MH_CreateHook(Actual_RenameItems, &RenameItemsCallback, reinterpret_cast<void**>(&Real_RenameItems)) != MH_OK)
{
MessageBoxW(NULL, L"Failed CreateHook Real_RenameItems", L"Info!", MB_ICONWARNING|MB_OK);
}
if (MH_EnableHook(Actual_RenameItems) != MH_OK)
{
MessageBoxW(NULL, L"Failed EnableHook Real_RenameItems", L"Info!", MB_ICONWARNING|MB_OK);
}
}
}
return HR;
}

BOOL APIENTRY DllMain(HMODULE hModule, DWORD  ul_reason_for_call, LPVOID lpReserved)
{
switch (ul_reason_for_call)
{
case DLL_PROCESS_ATTACH:
if (MH_Initialize() != MH_OK)
{
MessageBoxW(NULL, L"Failed Initialize", L"Info!", MB_ICONWARNING|MB_OK);
}
if (MH_CreateHook(&CoCreateInstance, &CoCreateInstanceCallback, reinterpret_cast<void**>(&Real_CoCreateInstance)) != MH_OK)
{
MessageBoxW(NULL, L"Failed MH_CreateHook CoCreateInstance", L"Info!", MB_ICONWARNING|MB_OK);
}
if (MH_EnableHook(&CoCreateInstance) != MH_OK)
{
MessageBoxW(NULL, L"Failed MH_EnableHook CoCreateInstance", L"Info!", MB_ICONWARNING|MB_OK);
}
break;

case DLL_PROCESS_DETACH:
if (MH_Uninitialize() != MH_OK)
{
}
if (MH_DisableHook(Actual_RenameItems) != MH_OK)
{
}
if (MH_DisableHook(&CoCreateInstance) != MH_OK)
{
}
break;
}
return TRUE;
}

Приведенный выше код работает нормально для IOFileOperation :: RenameItems, и я пытаюсь реализовать тот же код для метода IDropTarget :: Drop, но происходит то, что сама функция CoCreateInstance не вызывает операцию удаления, поэтому любая может помочь мне как реализовать это IDropTarget :: Drop Hooking …….

2

Решение

Обычно IDropTarget реализуется внутри клиентского приложения и регистрируется с помощью вызова RegisterDropTarget.
Попробуй подключить функцию RegisterDropTarget, может она поможет

Обновить: на самом деле имя функции — RegisterDragDrop

1

Другие решения

Других решений пока нет …

По вопросам рекламы ammmcru@yandex.ru
Adblock
detector