Function CallInSystemContext

Kernel-Bridge/API/Callable.cpp:11–56

namespace Callable
{
    bool CallInSystemContext(bool(*Callback)(void* Arg), void* Arg, bool Wait)
    {
        HANDLE hThread = NULL;

        struct PARAMS {
            bool(*Callback)(PVOID Arg);
            PVOID Arg;
            bool Result;
        } Params = {};
        Params.Callback = Callback;
        Params.Arg = Arg;

        OBJECT_ATTRIBUTES ObjectAttributes;
        InitializeObjectAttributes(&ObjectAttributes, NULL, OBJ_KERNEL_HANDLE, NULL, NULL);
        NTSTATUS Status = PsCreateSystemThread(
            &hThread,
            GENERIC_ALL,
            &ObjectAttributes,
            NULL,
            NULL,
            [](PVOID Arg)
            {
                PARAMS* Params = reinterpret_cast<PARAMS*>(Arg);
                Params->Result = Params->Callback(Params->Arg);
                PsTerminateSystemThread(STATUS_SUCCESS);
            },
            &Params
        );

        if (NT_SUCCESS(Status))
        {
            if (Wait)
            {
                ZwWaitForSingleObject(hThread, FALSE, NULL);
                ZwClose(hThread);
                return Params.Result;
            }
            else
            {
                ZwClose(hThread);
                return true;
            }
        }

        return false;
    }

    bool ForEachCpu(bool(*Callback)(void* Arg, unsigned int ProcessorNumber), void* Arg)
    {