Оберточная библиотека-перехватчик

Доброго времени суток всем!
Недавно я выкладывал свою реализацию перехватчика(тут).
Получился он, чего уж греха таить, громоздким и коряво работающим. В 1 же из комментариев отписался уважаемый k_d и упомянул о своей обертке над mhook(MHook vs Zuma).
Конечно, меня заинтересовала альтернатива, работающая с перехватом не только __cdecl'а но и, более того, позволяющая ставить перехват практически в любом месте кода.
Решение понравилось и я решил переписать свою библиотеку, которую использую для различного рода перехватов, начисто, с использованием материала от k_d. Текущий вариант умеет перехватывать все, до чего дотянутся загребуще-экспериментальные ручки программиста и парсить аргументы из стека(по крайней мере, потенциально идея это все умеет. Наверное.)
Представляю Вашему вниманию то, что получилось.

Для того, что бы скомпилировать рабочую библиотеку, понадобится MHook, обертка от k_d, средневыпрямленные руки, голова, вытащенная из-под стола и уточка, с которой можно посоветоваться: .

Итак, поехали!

I) Функционал:

1) Вывод логов в файл:

void debug_msg(const char* func_name, const char* txt,...)
{
	va_list args;
	FILE *file;
	char file_name[256]="";
	strcpy(file_name,"C:\VariadicDump\");
	strcat(file_name,func_name);
	strcat(file_name,".txt");
	fopen_s(&file,file_name, "a");
	va_start(args,txt);
	vfprintf(file,txt,args);
	va_end (args);
	fprintf(file, "n");
	fclose(file);
}

Использование:

debug_msg("Advanced", "--%s arg list started--", funcName);

При funcName = «Hello, world, you read this message so many times» получим файл C:VariadicDumpAdvanced.txt со следующим содержимым:


--Hello, world, you read this message so many times arg list started--


2) Парс lua_State (только для инициализированного и частного для данной функции! Полезен, к примеру, в ММО Runes of Magic для перехвата чата, в других случаях не использовал):

void Parse_LuaState(lua_State *L, const char *func_name)
{
	int n = lua_gettop(L);   
	std::vector<const char*> names;
	for(int i = 0; i < n; i++)
	{
		size_t arg2Len = 0;
		names.push_back(luaL_checklstring(L, i, &arg2Len));
	}
	string str = "";
	for (unsigned int i=0; i<names.size(); i++)
	{
		str.append(names[i]);
		str.append("; ");
	}
	debug_named_msg(func_name, str.c_str());
}

Использование:

void lua_hook(lua_State *L)
{
Parse_LuaState(L, "Lua");
}

К примеру, в *L было 2 аргумента — тип чата и текст(1 и «hello, world, are you hate to hear it again?»). Функция запишет в «C:VariadicDumpLua.txt» следующее:


1; hello, world, are you hate to hear it again?


3) Парсинг аргументов из стека:

void parseArgs(int *ptr, int size, char* funcName)
{
	debug_msg("Advanced", "--%s arg list started--", funcName);
	for(int i=0; i*4<size; i++)
	{
		debug_msg("Advanced", "  |---Element %d: %d", i, ptr[i]);
	}
	debug_msg("Advanced", "--arg list finished--n");
}

Использование:

void con_hook(Context *context)
{
	int *ptr = (int*)(void*)(context->ESP+4);
	parseArgs(ptr, con_arg_amount, __FUNCTION__);
}

В результате будут выведены аргументы из расчета esp + index*4 name##_arg_amount/4 раз.
По-хорошему, там правильнее и красивее передавать кол-во аргументов, но это не сложно сделать и самому в случае надобности.

4) Адрес основного модуля перехватываемого приложения:

int GetMainModuleAddress()
{
	DWORD dwAddress = NULL;
	HANDLE hthSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE | TH32CS_SNAPMODULE32, _getpid());
	if (hthSnapshot) 
	{
		MODULEENTRY32 me = { sizeof(me) };
		if (Module32First(hthSnapshot, &me))
		{
			CloseHandle(hthSnapshot);
			dwAddress = (DWORD)me.modBaseAddr;
		}
	}
	return dwAddress;
}

Ничего особо, нужен, к примеру, в Рифте — там виртуальный адрес рассчитывается следующим образом:

#define RECALC(name)
	name##_Detour = (t##name)((int)name##_Detour-0x400000+rift);

где name##_Detour — перехватываемая функция.

5) Функции attach/detach:

Простая обертка для установки/удаления трамплина. Нужны чисто для того, что бы не лазить по коду в поисках DllMain :)

Последняя в списке, но не последняя в наших сердцах и полезностью в функционале — unhook'er:

BOOL UnHookFunction(DWORD addr, unsigned char *lpBackup)
{
	DWORD dwAddr = addr;
	if (WriteProcessMemory(GetCurrentProcess(), (LPVOID)dwAddr, lpBackup, 6, 0))
		return TRUE;
	return FALSE;
}

Просто и со вкусом восстанавливает код, в котором стоял трамплин.

II) Макросы

Макросы — основная фишка данного проекта, которая, собственно, и позволила свернуть рабочую часть до неприлично малых размеров.

1) RECALC:

#define RECALC(name)
	name##_Detour = (t##name)((int)name##_Detour-0x400000+rift);

Упоминался выше, нужен для пересчета адреса.

2) DETACH:

#define DETACH(name1)
	UnHookFunction((DWORD)name1##_Detour,name1##_var);

Обертка над восста… освобождением кода от трамплина.

3) ATR/ATR_R:

#define ATR(name)
	name##_var = RegHook((long)name##_Detour, (long)(void*)name##_hook)

#define ATR_R(name)
	RECALC(name)
	name##_var = RegHook((long)name##_Detour, (long)(void*)name##_hook)

Два брата-близнеца для установки трамплина. RegHook — функция обертки над Mhook, описание можно прочесть в исходниках либо в . В той же статье указано, что есть Context и чем закусывать, что бы не бегать голышом по набережной и не звать ежика из тумана(шучу, этого просто так вам никто не расскажет).

Используется в функции attach();

4) DTR:

#define DTR(name)
	DETACH(name)

Обертка над оберткой.
Спорная надобность, но читается красивее рядом с ATR ^^ (DETACH — реликт эпохи, жаль удалять).

Используется в функции detach();

5) RF_O_UP_FUNC_CONTEXT:

#define RF_O_UP_FUNC_CONTEXT(name1, adres, args)
	typedef void (__cdecl * t##name1 ) ();
	t##name1 name1##_Detour = ( t##name1 ) ( adres ); /*перехватываемая функция*/
	void name1##_hook(Context *context); /*вызывается при переходе из предыдущей функции; весь функционал перехватчика- здесь*/
	BYTE *name1##_var;  /*здесь хранится код оригинальной функции*/
	int name1##_arg_amount = args; /*суммарный размер аргументов, которые нужно отобразить(см. выше)*/

Самый хитровыдернутый макрос в этой деревне. Собственно, он и занимается объявлением перехватчика и перехватываемой функции, объявляет переменную для сохранения исходного кода от трамплина и т.д.

Пример:

RF_O_UP_FUNC_CONTEXT(con, 0x60D710, 0x8);

Объявляем 2 функции — con_Detour и con_hook, привязываем перехватываемую функцию к адресу 0x60D710 и указываем, что если нам нужно будет проанализировать аргументы, то мы получим 2 штуки в выводе.

III) Пример использования

RF_O_UP_FUNC_CONTEXT(con, 0x60D710, 0x8);

void lua_hook(Context *context)
{
	int *ptr = (int*)(void*)(context->ESP+4); //помещаем адрес 1 аргумента из стека в указатель ptr
	parseArgs(ptr, con_arg_amount, __FUNCTION__); // парсим аргументы согласно макросу, 2 штуки - чужих портсигаров нам не надо!
	debug_msg("LuaDump", "%s", ptr[0]); //выводим 1 аргумент отдельно - а вот захотелось.Да.
}

Для работы dll могут понадобиться либы от lua, а также mhook + обертка от k_d.

Здесь можно скачать сорцы библиотеки.

Также еще раз благодарю k_d за отлично проделанную работу и идею!

Всем спасибо, всего доброго! И да восторжествует уточка! Кря!

Автор: Demogor

Источник