Developing.ru

Есть программа N, написанная сторонним производителем *.exe .

Я хочу поменять в ней нужный элемент меню на свой, т.е. название оставить прежнее, а функцию обработки сменить на свою.

Могу я как-нибудь узнать HANDLE меню и HANDLE самого элемента меню, чтобы этот элемент меню удалить, а свой добавить? Как это сделать?

Узнать-то в принципе можно, если есть хэндл соотв.окна: GetMenu и далее рекурсивно GetSubMenu, GetMenuInfo пока не найдем что нужно (напр., по тексту menu item). И удалить/добавить свой menu item для чужого процесса, скорее всего, тоже можно, хотя и не проверял. Но вот внедрить свой обработчик в чужую программу нельзя.

Можно. Всего надо:

1. Открыть exe-файл.
2. Найти адрес начала и адрес конца обработчка.
3. Залить всё NOP-ами.
4. Поверх NOP-ов написать свой код (на асме, конечно)
5. Сохранить exe-файл.

2Eugie: мне кажется, что это реально, т.к. этот процесс пользуется моей dll -> я нахожусь тоже в этом процессе -> подменить обработчик элемента меню несложно

2Romeo: тоже вариант, но мне он не подходит

Верно, обработчик действительно реально заменить программно, но всё будет работать только до того момента, пока твоя DLL подгружена. После её выгрузки код, на который ссылается обработчик станет невалидным и первый же клик на соответствующем айтеме меню приведёт к крашу всего комплекса.

Пока обработчик ссылается на мой код, длл выгружена не будет.

Так бы оно и было, если бы DLL была COM-овской. Что-то у меня необъяснимое ощущение, что DLL-ка загружается комплексом, как плагин. Т.е. делает LoadLibrary, потом GetProcAddress, вызывает полученный метод, а потом делает FreeLibrary. Это, конечно, утрированный пример, но технология, думаю, именно такова. А, поскольку, меняя стандартный обработчик на свой, dllcounter ты не изменяешь, то при вызове FreeLibrary произойдёт выгрузка и никаких других вариантов быть просто не может.

2Eugie: мне кажется, что это реально, т.к. этот процесс пользуется моей dll -> я нахожусь тоже в этом процессе -> подменить обработчик элемента меню несложно

В этом случае действительно можно: при загрузке dll ставить хук типа WH_GETMESSAGE и ловить сообщение WM_COMMAND с заданным id. Ессно, при выгрузке dll хук нужно удалять.

Теперь другая проблема вылезла

В общем заменил я пункт меню на свой, а вот чтобы теперь к нему добавить вызываемую функцию по этому пункту меню...
Ведь существует карта сообщений у программы. Я ее не знаю.. Мой новый элемент меню имеет свой ID. Как мне изменить эту карту так, чтобы она по этому ID вызывала мою функцию??? Как к этой карте получить доступ?

Вот пример, как это можно сделать (код для dll: когда она загружена, перехватывается обращение к пункту меню с именем "&About ...", вместо него вызывается MessageBox):

Код: Выделить всё

HHOOK hkMsg;
int nTargetCmdID;

// Searching for the main window of the current app.
BOOL CALLBACK EnumWindowsProc(
  HWND hwnd,      // handle to parent window
  LPARAM lParam   // application-defined value
)
{
	DWORD dwPID, dwTID = GetWindowThreadProcessId(hwnd, &dwPID);
	DWORD dwCurPID = GetCurrentProcessId();
	BOOL bProceed = dwCurPID != dwPID;
	if (!bProceed) 
		*(HWND*)lParam = hwnd;
	return bProceed;
}


HWND GetAppTopWindow()
{
	HWND hwnd = NULL;
	EnumWindows(EnumWindowsProc, (LPARAM)&hwnd);
	return hwnd;
}

int FindMenuItemID(HMENU hmMenu, LPCTSTR lpszMenuItemText)
{
	int id = -1;
	TCHAR buf[100];
	MENUITEMINFO mii;
	mii.cbSize = sizeof(MENUITEMINFO);
	mii.fMask = MIIM_STRING;
	mii.dwTypeData = buf;
	mii.cch = sizeof(buf)/sizeof(TCHAR);

	for(int i = 0; i < GetMenuItemCount(hmMenu); i++)
	{
		GetMenuItemInfo(hmMenu, i, TRUE, &mii);
		if (strcmp(mii.dwTypeData, lpszMenuItemText) == 0)
			return GetMenuItemID(hmMenu, i);
	  mii.fMask = MIIM_SUBMENU;
		GetMenuItemInfo(hmMenu, i, TRUE, &mii);
		if (mii.hSubMenu)
			id = FindMenuItemID(mii.hSubMenu, lpszMenuItemText);
	}

	return id;
}

int GetMenuCmdMsgID(LPCTSTR lpszMenuItemText)
{
	HWND hwnd = GetAppTopWindow();
	HMENU hmenu = GetMenu(hwnd);
	return FindMenuItemID(hmenu, lpszMenuItemText);
}

LRESULT CALLBACK GetMsgProc(
  int code,       // hook code
  WPARAM wParam,  // removal option
  LPARAM lParam   // message
)
{
	if (code == HC_ACTION)
	{
		MSG* pmsg = (MSG*)lParam;
		wParam = PM_REMOVE;

	  nTargetCmdID = GetMenuCmdMsgID("&About ...");
		if(pmsg->hwnd == GetAppTopWindow() &&
		   pmsg->message == WM_COMMAND &&
		   HIWORD(pmsg->wParam) == 0 &&
		   LOWORD(pmsg->wParam) == nTargetCmdID)
		{
			pmsg->wParam = -1; // to prevent passing to standard handler
			MessageBox(NULL, "Target menu command called.", "Info", MB_OK);
			return 0;
		}

	}
	return CallNextHookEx(hkMsg, code, wParam, lParam);
}

void InstallHook()
{
	hkMsg = SetWindowsHookEx(WH_GETMESSAGE, GetMsgProc, NULL, GetCurrentThreadId()); 
}

void UninstallHook()
{
	UnhookWindowsHookEx(hkMsg); 
}

BOOL APIENTRY DllMain( HANDLE hModule, 
                       DWORD  ul_reason_for_call, 
                       LPVOID lpReserved
					 )
{
    switch (ul_reason_for_call)
	{
		case DLL_PROCESS_ATTACH:
			InstallHook();
			break;
		case DLL_PROCESS_DETACH:
			UninstallHook();
			break;
		case DLL_THREAD_ATTACH:
			break;
		case DLL_THREAD_DETACH:
			break;
    }
    return TRUE;
}

Developing.ru

Заменить пункт меню на свой через HANDLE

Заменить пункт меню на свой через HANDLE