#define NDEBUG
#include <debug.h>
+typedef struct _EVENTPACK
+{
+ PEVENTHOOK pEH;
+ LONG idObject;
+ LONG idChild;
+} EVENTPACK, *PEVENTPACK;
+
static PEVENTTABLE GlobalEvents = NULL;
-static ULONG EventSys[EVENT_SYSTEM_MINIMIZEEND+1] = {0};
-static ULONG EventObj[( EVENT_OBJECT_ACCELERATORCHANGE - EVENT_OBJECT_CREATE) +1] = {0};
/* PRIVATE FUNCTIONS *********************************************************/
static
VOID
FASTCALL
-IntEventUpCount(ULONG eventMin, ULONG eventMax)
+IntSetSrvEventMask( UINT EventMin, UINT EventMax)
{
- INT i, Min, Max;
-
- if ( eventMin >= EVENT_SYSTEM_SOUND && eventMax <= EVENT_SYSTEM_MINIMIZEEND)
+ UINT event;
+ DPRINT("SetSrvEventMask 1\n");
+ for ( event = EventMin; event <= EventMax; event++)
{
- for (i = eventMin; i < eventMax; i++)
+ if ((event >= EVENT_SYSTEM_SOUND && event <= EVENT_SYSTEM_MINIMIZEEND) ||
+ (event >= EVENT_CONSOLE_CARET && event <= EVENT_CONSOLE_END_APPLICATION) ||
+ (event >= EVENT_OBJECT_CREATE && event <= EVENT_OBJECT_ACCELERATORCHANGE))
{
- gpsi->SrvEventActivity |= GetMaskFromEvent(i);
- EventSys[i]++;
+ gpsi->SrvEventActivity |= GetMaskFromEvent(event);
}
- }
- if ( eventMin >= EVENT_OBJECT_CREATE && eventMax <= EVENT_OBJECT_ACCELERATORCHANGE)
- {
- for (i = eventMin; i < eventMax; i++)
+ if (event > EVENT_SYSTEM_MINIMIZEEND && event < EVENT_CONSOLE_CARET)
{
- gpsi->SrvEventActivity |= GetMaskFromEvent(i);
- EventObj[i - EVENT_OBJECT_CREATE]++;
+ event = EVENT_CONSOLE_CARET-1;
+ gpsi->SrvEventActivity |= GetMaskFromEvent(event);
}
- }
- if ( eventMin >= EVENT_SYSTEM_SOUND && eventMax <= EVENT_OBJECT_ACCELERATORCHANGE)
- {
- Max = EVENT_SYSTEM_MINIMIZEEND;
- for (i = eventMin; i < Max; i++)
+ if (event > EVENT_CONSOLE_END_APPLICATION && event < EVENT_OBJECT_CREATE )
{
- gpsi->SrvEventActivity |= GetMaskFromEvent(i);
- EventSys[i]++;
+ event = EVENT_OBJECT_CREATE-1;
+ gpsi->SrvEventActivity |= GetMaskFromEvent(event);
}
- Min = EVENT_OBJECT_CREATE;
- for (i = Min; i < eventMax; i++)
+ if (event > EVENT_OBJECT_ACCELERATORCHANGE && event < EVENT_MAX)
{
- gpsi->SrvEventActivity |= GetMaskFromEvent(i);
- EventObj[i - EVENT_OBJECT_CREATE]++;
- }
+ event = EVENT_MAX-1;
+ gpsi->SrvEventActivity |= GetMaskFromEvent(event);
+ break;
+ }
}
-}
-
-static
-VOID
-FASTCALL
-IntEventDownCount(ULONG eventMin, ULONG eventMax)
-{
- INT i, Min, Max;
-
- if ( eventMin >= EVENT_SYSTEM_SOUND && eventMax <= EVENT_SYSTEM_MINIMIZEEND)
- {
- for (i = eventMin; i < eventMax; i++)
- {
- EventSys[i]--;
- if (!EventSys[i]) gpsi->SrvEventActivity &= ~GetMaskFromEvent(i);
- }
- }
- if ( eventMin >= EVENT_OBJECT_CREATE && eventMax <= EVENT_OBJECT_ACCELERATORCHANGE)
- {
- for (i = eventMin; i < eventMax; i++)
- {
- EventObj[i - EVENT_OBJECT_CREATE]--;
- if (!EventObj[i - EVENT_OBJECT_CREATE])
- gpsi->SrvEventActivity &= ~GetMaskFromEvent(i);
- }
- }
- if ( eventMin >= EVENT_SYSTEM_SOUND && eventMax <= EVENT_OBJECT_ACCELERATORCHANGE)
- {
- Max = EVENT_SYSTEM_MINIMIZEEND;
- for (i = eventMin; i < Max; i++)
- {
- EventSys[i]--;
- if (!EventSys[i]) gpsi->SrvEventActivity &= ~GetMaskFromEvent(i);
- }
- Min = EVENT_OBJECT_CREATE;
- for (i = Min; i < eventMax; i++)
- {
- EventObj[i - EVENT_OBJECT_CREATE]--;
- if (!EventObj[i - EVENT_OBJECT_CREATE])
- gpsi->SrvEventActivity &= ~GetMaskFromEvent(i);
- }
- }
-}
-
-
-static
-DWORD
-FASTCALL
-TimeStamp(VOID)
-{
- return (DWORD)((ULONGLONG)SharedUserData->TickCountLowDeprecated * SharedUserData->TickCountMultiplier / 16777216);
+ if (!gpsi->SrvEventActivity)
+ gpsi->SrvEventActivity |= SRV_EVENT_RUNNING; // Set something.
+ DPRINT("SetSrvEventMask 2 : %x\n", gpsi->SrvEventActivity);
}
static
{
NTSTATUS Status;
ULONG_PTR uResult;
+ EVENTPACK EP;
+
+ EP.pEH = pEH;
+ EP.idObject = idObject;
+ EP.idChild = idChild;
/* FIXME should get timeout from
* HKEY_CURRENT_USER\Control Panel\Desktop\LowLevelHooksTimeout */
Status = co_MsqSendMessage(((PW32THREAD)pEH->Thread->Tcb.Win32Thread)->MessageQueue,
hwnd,
event,
- idObject,
- idChild,
+ 0,
+ (LPARAM)&EP,
5000,
TRUE,
MSQ_ISEVENT,
return NT_SUCCESS(Status) ? uResult : 0;
}
+
static
BOOL
FASTCALL
{
RemoveEntryList(&pEH->Chain);
GlobalEvents->Counts--;
+ if (!GlobalEvents->Counts) gpsi->SrvEventActivity = 0;
UserDeleteObject(pEH->Self, otEvent);
return TRUE;
}
LONG idObject,
LONG idChild)
{
-
- PEVENTHOOK pEH = UserHeapAlloc(sizeof(EVENTHOOK));
-
- if ((gpsi->SrvEventActivity & GetMaskFromEvent(event))) return 0; // No events to run.
-
-
- if ((pEH->Thread != PsGetCurrentThread()) && (pEH->Thread != NULL))
- {
- // Post it in message queue.
- return IntCallLowLevelEvent(pEH, event, hwnd, idObject, idChild);
- }
-
-
- LRESULT Result = co_IntCallEventProc(pEH->Self,
- event,
- hwnd,
- idObject,
- idChild,
- (DWORD)(NtCurrentTeb()->Cid).UniqueThread,
- TimeStamp(),
- pEH->Proc);
+ PEVENTHOOK pEH;
+ LRESULT Result;
+ PEVENTPACK pEP = (PEVENTPACK)idChild;
+
+ pEH = pEP->pEH;
+
+ Result = co_IntCallEventProc( pEH->Self,
+ event,
+ hwnd,
+ pEP->idObject,
+ pEP->idChild,
+ (DWORD)(NtCurrentTeb()->Cid).UniqueThread,
+ (DWORD)EngGetTickCount(),
+ pEH->Proc);
return Result;
}
+VOID
+FASTCALL
+IntNotifyWinEvent(
+ DWORD Event,
+ PWINDOW_OBJECT Window,
+ LONG idObject,
+ LONG idChild)
+{
+ PEVENTHOOK pEH;
+ LRESULT Result;
+
+ if (!GlobalEvents->Counts) return;
+
+ pEH = (PEVENTHOOK)GlobalEvents->Events.Flink;
+
+ do
+ {
+ UserReferenceObject(pEH);
+ // Must be inside the event window.
+ if ( (pEH->eventMin <= Event) && (pEH->eventMax >= Event))
+ {
+ if ((pEH->Thread != PsGetCurrentThread()) && (pEH->Thread != NULL))
+ { // if all process || all thread || other thread same process
+ if (!(pEH->idProcess) || !(pEH->idThread) ||
+ ((DWORD)(NtCurrentTeb()->Cid).UniqueProcess == pEH->idProcess))
+ {
+ Result = IntCallLowLevelEvent(pEH, Event, Window->hSelf, idObject, idChild);
+ }
+ }// if ^skip own thread && ((Pid && CPid == Pid && ^skip own process) || all process)
+ else if ( !(pEH->Flags & WINEVENT_SKIPOWNTHREAD) &&
+ ( ((pEH->idProcess &&
+ (DWORD)(NtCurrentTeb()->Cid).UniqueProcess == pEH->idProcess) &&
+ !(pEH->Flags & WINEVENT_SKIPOWNPROCESS)) ||
+ !pEH->idProcess ) )
+ {
+ Result = co_IntCallEventProc( pEH->Self,
+ Event,
+ Window->hSelf,
+ idObject,
+ idChild,
+ (DWORD)(NtCurrentTeb()->Cid).UniqueThread,
+ (DWORD)EngGetTickCount(),
+ pEH->Proc);
+ }
+ }
+ UserDereferenceObject(pEH);
+
+ pEH = (PEVENTHOOK)pEH->Chain.Flink;
+ } while (pEH != (PEVENTHOOK)&GlobalEvents->Events.Flink);
+}
VOID
STDCALL
LONG idObject,
LONG idChild)
{
+ PWINDOW_OBJECT Window = NULL;
+ USER_REFERENCE_ENTRY Ref;
UserEnterExclusive();
- UNIMPLEMENTED
+
+ /* Validate input */
+ if (hWnd && (hWnd != INVALID_HANDLE_VALUE) && !(Window = UserGetWindowObject(hWnd)))
+ {
+ return;
+ }
+
+ if (gpsi->SrvEventActivity & GetMaskFromEvent(Event))
+ {
+ UserRefObjectCo(Window, &Ref);
+ IntNotifyWinEvent( Event, Window, idObject, idChild);
+ UserDerefObjectCo(Window);
+ }
UserLeave();
}
UNICODE_STRING ModuleName;
NTSTATUS Status;
HANDLE Handle;
+ PETHREAD Thread = NULL;
UserEnterExclusive();
- DPRINT1("WARNING! Use at your own risk! Function is UNIMPLEMENTED!\n");
-
if ( !GlobalEvents )
{
GlobalEvents = ExAllocatePoolWithTag(PagedPool, sizeof(EVENTTABLE), TAG_HOOK);
InitializeListHead(&GlobalEvents->Events);
}
+ if (eventMin > eventMax)
+ {
+ SetLastWin32Error(ERROR_INVALID_HOOK_FILTER);
+ goto SetEventExit;
+ }
+
+ if (!lpfnWinEventProc)
+ {
+ SetLastWin32Error(ERROR_INVALID_FILTER_PROC);
+ goto SetEventExit;
+ }
+
+ if ((dwflags & WINEVENT_INCONTEXT) && !hmodWinEventProc)
+ {
+ SetLastWin32Error(ERROR_HOOK_NEEDS_HMOD);
+ goto SetEventExit;
+ }
+
+ if (idThread)
+ {
+ Status = PsLookupThreadByThreadId((HANDLE)(DWORD_PTR)idThread, &Thread);
+ if (!NT_SUCCESS(Status))
+ {
+ SetLastWin32Error(ERROR_INVALID_THREAD_ID);
+ goto SetEventExit;
+ }
+ }
+
pEH = UserCreateObject(gHandleTable, &Handle, otEvent, sizeof(EVENTHOOK));
if (pEH)
{
GlobalEvents->Counts++;
pEH->Self = Handle;
- pEH->Thread = PsGetCurrentThread();
+ if (Thread)
+ pEH->Thread = Thread;
+ else
+ pEH->Thread = PsGetCurrentThread();
pEH->eventMin = eventMin;
pEH->eventMax = eventMax;
pEH->idProcess = idProcess;
pEH->Ansi = FALSE;
pEH->Flags = dwflags;
- if ((dwflags & WINEVENT_INCONTEXT) && !hmodWinEventProc)
- {
- SetLastWin32Error(ERROR_HOOK_NEEDS_HMOD);
- goto SetEventExit;
- }
-
- if (eventMin > eventMax)
- {
- SetLastWin32Error(ERROR_INVALID_HOOK_FILTER);
- goto SetEventExit;
- }
if (NULL != hmodWinEventProc)
{
- Status = MmCopyFromCaller(&ModuleName, puString, sizeof(UNICODE_STRING));
+ Status = MmCopyFromCaller(&ModuleName,
+ puString,
+ sizeof(UNICODE_STRING));
+
if (! NT_SUCCESS(Status))
{
UserDereferenceObject(pEH);
SetLastNtError(Status);
goto SetEventExit;
}
+
pEH->ModuleName.Buffer = ExAllocatePoolWithTag(PagedPool,
ModuleName.MaximumLength,
TAG_HOOK);
+
if (NULL == pEH->ModuleName.Buffer)
{
UserDereferenceObject(pEH);
SetLastWin32Error(ERROR_NOT_ENOUGH_MEMORY);
goto SetEventExit;
}
+
pEH->ModuleName.MaximumLength = ModuleName.MaximumLength;
+
Status = MmCopyFromCaller(pEH->ModuleName.Buffer,
ModuleName.Buffer,
ModuleName.MaximumLength);
+
if (! NT_SUCCESS(Status))
{
ExFreePool(pEH->ModuleName.Buffer);
SetLastNtError(Status);
goto SetEventExit;
}
+
pEH->ModuleName.Length = ModuleName.Length;
+
pEH->Proc = (void *)((char *)lpfnWinEventProc - (char *)hmodWinEventProc);
}
else
pEH->Proc = lpfnWinEventProc;
Ret = Handle;
- /*
- Now we are good, set the Events and counts.
- */
- IntEventUpCount(eventMin, eventMax);
+ IntSetSrvEventMask( eventMin, eventMax);
}
SetEventExit:
+ if (Thread) ObDereferenceObject(Thread);
UserLeave();
return Ret;
}
-
BOOL
STDCALL
NtUserUnhookWinEvent(
UserEnterExclusive();
- DPRINT1("WARNING! Use at your own risk! Function is UNIMPLEMENTED!\n");
-
pEH = (PEVENTHOOK)UserGetObject(gHandleTable, hWinEventHook, otEvent);
if (pEH)
{
- IntEventDownCount(pEH->eventMin, pEH->eventMax);
Ret = IntRemoveEvent(pEH);
}