; +

; Сохранение контекста.

; o Eax: ID NtUserEnumDisplayMonitors.

; o Esi: адрес возврата.

; o Edi: ссылка на стек.

;

TsSave proc C

	xor ecx,ecx

	push esp	; Параметр для калбэка - ссылка на стек для его восстановления.

	Call @f	; Ссылка на калбэк.

; Калбэк.

; typedef BOOL (CALLBACK* MONITORENUMPROC)(HMONITOR, HDC, LPRECT, LPARAM);

	mov esp,dword ptr [esp + 4*4]	; Восстанавливаем стек, ссылка для восстановления передаётся параметром.

	xor eax,eax

	retn

@@:

	push ecx

	push ecx

	mov edx,esp

; BOOL

; NtUserEnumDisplayMonitors(

;     IN HDC             hdc,

;     IN LPCRECT         lprcClip,

;     IN MONITORENUMPROC lpfnEnum,

;     IN LPARAM          dwData)

	Int 2EH	; NtUserEnumDisplayMonitors

; При восстановлении контекста возвращается не ноль.

	.if !Eax	; Калбэк не был вызван изза исчерпания лимита вызовов/вложенности(0x2C).

	add esp,4*4	; Удаляем параметры сервиса.

	retn

	.else

	mov esp,edi

	add eax,3*4	; Ссылка на стек, который был при вызове TsLoad(для передачи параметров.

	jmp esi

	.endif

TsSave endp



; +

; Восстановление контекста.

;

TsLoad proc C	; stdcall

	xor eax,eax

	mov edx,3*4

	push eax

	push eax

	push esp

	mov ecx,esp

	Int 2BH

	add esp,3*4

	retn		; В случае ошибки возвратит STATUS_NO_CALLBACK_ACTIVE.

TsLoad endp