/*
* @implemented
*/
NTSTATUS
NTAPI
PsRemoveCreateThreadNotifyRoutine(IN PCREATE_THREAD_NOTIFY_ROUTINE NotifyRoutine)
{
ULONG i;
PEX_CALLBACK_ROUTINE_BLOCK CallBack;
PAGED_CODE();
/* Loop all callbacks */
for (i = 0; i < PSP_MAX_CREATE_THREAD_NOTIFY; i++)
{
/* Reference this slot */
CallBack = ExReferenceCallBackBlock(&PspThreadNotifyRoutine[i]);
if (CallBack)
{
/* Check for a match */
if (ExGetCallBackBlockRoutine(CallBack) == (PVOID)NotifyRoutine)
{
/* Try removing it if it matches */
if (ExCompareExchangeCallBack(&PspThreadNotifyRoutine[i],
NULL,
CallBack))
{
/* We removed it, now dereference the block */
InterlockedDecrement((PLONG)&PspThreadNotifyRoutineCount);
ExDereferenceCallBackBlock(&PspThreadNotifyRoutine[i],
CallBack);
/* Wait for active callbacks */
ExWaitForCallBacks(CallBack);
/* Free the callback and return */
ExFreeCallBack(CallBack);
return STATUS_SUCCESS;
}
}
/* Dereference the callback */
ExDereferenceCallBackBlock(&PspThreadNotifyRoutine[i], CallBack);
}
}
/* Nothing found to remove */
return STATUS_PROCEDURE_NOT_FOUND;
}
NTSTATUS
ExCallCallBack (
IN OUT PEX_CALLBACK CallBack,
IN PVOID Argument1,
IN PVOID Argument2
)
/*++
Routine Description:
This function calls the callback thats inside a callback structure
Arguments:
CallBack - Call back that needs to be called through
Argument1 - Caller provided argument to pass on
Argument2 - Caller provided argument to pass on
Return Value:
NTSTATUS - Status returned by callback or STATUS_SUCCESS if theres wasn't one
--*/
{
PEX_CALLBACK_ROUTINE_BLOCK CallBackBlock;
NTSTATUS Status;
CallBackBlock = ExReferenceCallBackBlock (CallBack);
if (CallBackBlock) {
//
// Call the function
//
Status = CallBackBlock->Function (CallBackBlock->Context, Argument1, Argument2);
ExDereferenceCallBackBlock (CallBack, CallBackBlock);
} else {
Status = STATUS_SUCCESS;
}
return Status;
}
/*
* @implemented
*/
NTSTATUS
NTAPI
PsSetCreateProcessNotifyRoutine(IN PCREATE_PROCESS_NOTIFY_ROUTINE NotifyRoutine,
IN BOOLEAN Remove)
{
ULONG i;
PEX_CALLBACK_ROUTINE_BLOCK CallBack;
PAGED_CODE();
/* Check if we're removing */
if (Remove)
{
/* Loop all the routines */
for (i = 0; i < PSP_MAX_CREATE_PROCESS_NOTIFY; i++)
{
/* Reference the callback block */
CallBack = ExReferenceCallBackBlock(&PspProcessNotifyRoutine[i]);
if (!CallBack) continue;
/* Check it this is a matching block */
if (ExGetCallBackBlockRoutine(CallBack) != (PVOID)NotifyRoutine)
{
/* It's not, try the next one */
continue;
}
/* It is, clear the current routine */
if (ExCompareExchangeCallBack(&PspProcessNotifyRoutine[i],
NULL,
CallBack))
{
/* Decrement the number of routines */
InterlockedDecrement((PLONG)&PspProcessNotifyRoutineCount);
/* Dereference the block */
ExDereferenceCallBackBlock(&PspProcessNotifyRoutine[i],
CallBack);
/* Wait for actice callbacks */
ExWaitForCallBacks(CallBack);
/* Free the callback and exit */
ExFreeCallBack (CallBack);
return STATUS_SUCCESS;
}
/* Dereference the block */
ExDereferenceCallBackBlock(&PspProcessNotifyRoutine[i],
CallBack);
}
/* We didn't find any matching block */
return STATUS_PROCEDURE_NOT_FOUND;
}
else
{
/* Allocate a callback */
CallBack = ExAllocateCallBack((PVOID)NotifyRoutine, NULL);
if (!CallBack) return STATUS_INSUFFICIENT_RESOURCES;
/* Loop all callbacks */
for (i = 0; i < PSP_MAX_CREATE_PROCESS_NOTIFY; i++)
{
/* Add this routine if it's an empty slot */
if (ExCompareExchangeCallBack(&PspProcessNotifyRoutine[i],
CallBack,
NULL))
{
/* Found and inserted into an empty slot, return */
InterlockedIncrement((PLONG)&PspProcessNotifyRoutineCount);
return STATUS_SUCCESS;
}
}
/* We didn't find a free slot, free the callback and fail */
ExFreeCallBack(CallBack);
return STATUS_INVALID_PARAMETER;
}
}
