Function _tmain

EnableCustomKernelSigners/_tmain.cpp:271–297

}

int _tmain(int argc, PTSTR argv[]) {
    LSTATUS Status;
    OwnedResource<GenericHandleTraits> hToken;
    BOOL bShutdownPrivilegeEnable;

    if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ALL_ACCESS, hToken.GetAddress())) {
        _tprintf_s(TEXT("[-] Failed to open current process's token. CODE: 0x%.8X\n"), GetLastError());
        return -1;
    }

    Status = EnableCksCheckPrivilege(hToken, SE_SHUTDOWN_NAME, &bShutdownPrivilegeEnable);
    if (Status != ERROR_SUCCESS) {
        _tprintf_s(TEXT("[-] Failed to check " SE_SHUTDOWN_NAME ". CODE: 0x%.8X\n"), Status);
        return -1;
    }

    if (bShutdownPrivilegeEnable == FALSE && (Status = EnableCksSetPrivilege(hToken, SE_SHUTDOWN_NAME, TRUE)) != ERROR_SUCCESS) {
        _tprintf_s(TEXT("[-] Failed to enable " SE_SHUTDOWN_NAME " . CODE: 0x%.8X\n"), Status);
        return -1;
    }

    if (argc == 2 && _tcsicmp(argv[1], TEXT("--setupmode")) == 0) {
        return EnableCksFinal(EnableCksSetupMode(), true);
    } else {
        return EnableCksFinal(EnableCksNonSetupMode(), false);
    }
}