r0drv/nt/initterm-r0drv-nt.cpp

	initterm-r0drv-nt.cpp revision c2ac210bd84591123bb8803712887e2b016cb78f
1N/A/* $Id$ */
1N/A/** @file
1N/A * IPRT - Initialization & Termination, R0 Driver, NT.
1N/A */
1N/A
1N/A/*
1N/A * Copyright (C) 2006-2007 Oracle Corporation
1N/A *
1N/A * This file is part of VirtualBox Open Source Edition (OSE), as
1N/A * available from http://www.virtualbox.org. This file is free software;
1N/A * you can redistribute it and/or modify it under the terms of the GNU
1N/A * General Public License (GPL) as published by the Free Software
1N/A * Foundation, in version 2 as it comes in the "COPYING" file of the
1N/A * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
1N/A * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
1N/A *
1N/A * The contents of this file may alternatively be used under the terms
1N/A * of the Common Development and Distribution License Version 1.0
1N/A * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
1N/A * VirtualBox OSE distribution, in which case the provisions of the
1N/A * CDDL are applicable instead of those of the GPL.
1N/A *
1N/A * You may elect to license modified versions of this file under the
1N/A * terms and conditions of either the GPL or the CDDL or both.
1N/A */
1N/A
1N/A/*******************************************************************************
1N/A*   Header Files                                                               *
1N/A*******************************************************************************/
1N/A#include "the-nt-kernel.h"
1N/A#include <iprt/asm-amd64-x86.h>
1N/A#include <iprt/assert.h>
1N/A#include <iprt/err.h>
1N/A#include <iprt/mp.h>
1N/A#include <iprt/string.h>
1N/A#include "internal/initterm.h"
1N/A#include "internal-r0drv-nt.h"
1N/A
1N/A
1N/A/*******************************************************************************
1N/A*   Global Variables                                                           *
1N/A*******************************************************************************/
1N/A/** The Nt CPU set.
1N/A * KeQueryActiveProcssors() cannot be called at all IRQLs and therefore we'll
1N/A * have to cache it. Fortunately, Nt doesn't really support taking CPUs offline
1N/A * or online. It's first with W2K8 that support for CPU hotplugging was added.
1N/A * Once we start caring about this, we'll simply let the native MP event callback
1N/A * and update this variable as CPUs comes online. (The code is done already.)
1N/A */
1N/ARTCPUSET                    g_rtMpNtCpuSet;
1N/A
1N/A/** ExSetTimerResolution, introduced in W2K. */
1N/APFNMYEXSETTIMERRESOLUTION   g_pfnrtNtExSetTimerResolution;
1N/A/** KeFlushQueuedDpcs, introduced in XP. */
1N/APFNMYKEFLUSHQUEUEDDPCS      g_pfnrtNtKeFlushQueuedDpcs;
1N/A/** HalRequestIpi, introduced in ??. */
1N/APFNHALREQUESTIPI            g_pfnrtNtHalRequestIpi;
1N/A/** HalSendSoftwareInterrupt */
1N/APFNHALSENDSOFTWAREINTERRUPT g_pfnrtNtHalSendSoftwareInterrupt;
1N/A/** SendIpi handler based on Windows version */
1N/APFNRTSENDIPI                g_pfnrtSendIpi;
1N/A/** KeIpiGenericCall - Windows Server 2003+ only */
1N/APFNRTKEIPIGENERICCALL       g_pfnrtKeIpiGenericCall;
1N/A
1N/A/** Offset of the _KPRCB::QuantumEnd field. 0 if not found. */
1N/Auint32_t                    g_offrtNtPbQuantumEnd;
1N/A/** Size of the _KPRCB::QuantumEnd field. 0 if not found. */
1N/Auint32_t                    g_cbrtNtPbQuantumEnd;
1N/A/** Offset of the _KPRCB::DpcQueueDepth field. 0 if not found. */
1N/Auint32_t                    g_offrtNtPbDpcQueueDepth;
1N/A
1N/A
1N/A
1N/Aint rtR0InitNative(void)
1N/A{
1N/A    /*
1N/A     * Init the Nt cpu set.
1N/A     */
1N/A#ifdef IPRT_TARGET_NT4
1N/A    KAFFINITY ActiveProcessors = (UINT64_C(1) << KeNumberProcessors) - UINT64_C(1);
1N/A#else
1N/A    KAFFINITY ActiveProcessors = KeQueryActiveProcessors();
1N/A#endif
1N/A    RTCpuSetEmpty(&g_rtMpNtCpuSet);
1N/A#ifdef RT_WITH_LOTS_OF_CPUS
1N/A# error "port me"
1N/A#endif
1N/A    RTCpuSetFromU64(&g_rtMpNtCpuSet, ActiveProcessors);
1N/A
1N/A#ifdef IPRT_TARGET_NT4
1N/A    g_pfnrtNtExSetTimerResolution = NULL;
1N/A    g_pfnrtNtKeFlushQueuedDpcs = NULL;
1N/A    g_pfnrtNtHalRequestIpi = NULL;
1N/A    g_pfnrtNtHalSendSoftwareInterrupt = NULL;
1N/A    g_pfnrtKeIpiGenericCall = NULL;
1N/A#else
1N/A    /*
1N/A     * Initialize the function pointers.
1N/A     */
1N/A    UNICODE_STRING RoutineName;
1N/A    RtlInitUnicodeString(&RoutineName, L"ExSetTimerResolution");
1N/A    g_pfnrtNtExSetTimerResolution = (PFNMYEXSETTIMERRESOLUTION)MmGetSystemRoutineAddress(&RoutineName);
1N/A
1N/A    RtlInitUnicodeString(&RoutineName, L"KeFlushQueuedDpcs");
1N/A    g_pfnrtNtKeFlushQueuedDpcs = (PFNMYKEFLUSHQUEUEDDPCS)MmGetSystemRoutineAddress(&RoutineName);
1N/A
1N/A    RtlInitUnicodeString(&RoutineName, L"HalRequestIpi");
1N/A    g_pfnrtNtHalRequestIpi = (PFNHALREQUESTIPI)MmGetSystemRoutineAddress(&RoutineName);
1N/A
1N/A    RtlInitUnicodeString(&RoutineName, L"HalSendSoftwareInterrupt");
1N/A    g_pfnrtNtHalSendSoftwareInterrupt = (PFNHALSENDSOFTWAREINTERRUPT)MmGetSystemRoutineAddress(&RoutineName);
1N/A
    RtlInitUnicodeString(&RoutineName, L"KeIpiGenericCall");
    g_pfnrtKeIpiGenericCall = (PFNRTKEIPIGENERICCALL)MmGetSystemRoutineAddress(&RoutineName);
#endif

    /*
     * Get some info that might come in handy below.
     */
    ULONG MajorVersion = 0;
    ULONG MinorVersion = 0;
    ULONG BuildNumber  = 0;
    BOOLEAN fChecked = PsGetVersion(&MajorVersion, &MinorVersion, &BuildNumber, NULL);

    g_pfnrtSendIpi = rtMpSendIpiDummy;
#ifndef IPRT_TARGET_NT4
    if (    g_pfnrtNtHalRequestIpi
        &&  MajorVersion == 6
        &&  MinorVersion == 0)
    {
        /* Vista or Windows Server 2008 */
        g_pfnrtSendIpi = rtMpSendIpiVista;
    }
    else
    if (    g_pfnrtNtHalSendSoftwareInterrupt
        &&  MajorVersion == 6
        &&  MinorVersion == 1)
    {
        /* Windows 7 or Windows Server 2008 R2 */
        g_pfnrtSendIpi = rtMpSendIpiWin7;
    }
    /* Windows XP should send always send an IPI -> VERIFY */
#endif
    KIRQL OldIrql;
    KeRaiseIrql(DISPATCH_LEVEL, &OldIrql); /* make sure we stay on the same cpu */

    union
    {
        uint32_t auRegs[4];
        char szVendor[4*3+1];
    } u;
    ASMCpuId(0, &u.auRegs[3], &u.auRegs[0], &u.auRegs[2], &u.auRegs[1]);
    u.szVendor[4*3] = '\0';

    /*
     * HACK ALERT (and d�j� vu warning)!
     *
     * Try find _KPRCB::QuantumEnd and _KPRCB::[DpcData.]DpcQueueDepth.
     * For purpose of verification we use the VendorString member (12+1 chars).
     *
     * The offsets was initially derived by poking around with windbg
     * (dt _KPRCB, !prcb ++, and such like). Systematic harvesting is now done
     * by means of dia2dump, grep and the symbol packs. Typically:
     *      dia2dump -type _KDPC_DATA -type _KPRCB EXE\ntkrnlmp.pdb | grep -wE "QuantumEnd|DpcData|DpcQueueDepth|VendorString"
     */
    /** @todo array w/ data + script for extracting a row. (save space + readability; table will be short.) */
    __try
    {
#if defined(RT_ARCH_X86)
        PKPCR    pPcr   = (PKPCR)__readfsdword(RT_OFFSETOF(KPCR,SelfPcr));
        uint8_t *pbPrcb = (uint8_t *)pPcr->Prcb;

        if (    BuildNumber == 2600                             /* XP SP2 */
            &&  !memcmp(&pbPrcb[0x900], &u.szVendor[0], 4*3))
        {
            g_offrtNtPbQuantumEnd    = 0x88c;
            g_cbrtNtPbQuantumEnd     = 4;
            g_offrtNtPbDpcQueueDepth = 0x870;
        }
        /* WindowsVista.6002.090410-1830.x86fre.Symbols.exe
           WindowsVista.6002.090410-1830.x86chk.Symbols.exe
           WindowsVista.6002.090130-1715.x86fre.Symbols.exe
           WindowsVista.6002.090130-1715.x86chk.Symbols.exe */
        else if (   BuildNumber == 6002
                 && !memcmp(&pbPrcb[0x1c2c], &u.szVendor[0], 4*3))
        {
            g_offrtNtPbQuantumEnd    = 0x1a41;
            g_cbrtNtPbQuantumEnd     = 1;
            g_offrtNtPbDpcQueueDepth = 0x19e0 + 0xc;
        }
        else if (   BuildNumber == 3790                         /* Server 2003 SP2 */
                 && !memcmp(&pbPrcb[0xb60], &u.szVendor[0], 4*3))
        {
            g_offrtNtPbQuantumEnd    = 0x981;
            g_cbrtNtPbQuantumEnd     = 1;
            g_offrtNtPbDpcQueueDepth = 0x920 + 0xc;
        }

        /** @todo more */
        //pbQuantumEnd = (uint8_t volatile *)pPcr->Prcb + 0x1a41;

#elif defined(RT_ARCH_AMD64)
        PKPCR    pPcr   = (PKPCR)__readgsqword(RT_OFFSETOF(KPCR,Self));
        uint8_t *pbPrcb = (uint8_t *)pPcr->CurrentPrcb;

        if (    BuildNumber == 3790                             /* XP64 / W2K3-AMD64 SP1 */
            &&  !memcmp(&pbPrcb[0x22b4], &u.szVendor[0], 4*3))
        {
            g_offrtNtPbQuantumEnd    = 0x1f75;
            g_cbrtNtPbQuantumEnd     = 1;
            g_offrtNtPbDpcQueueDepth = 0x1f00 + 0x18;
        }
        else if (   BuildNumber == 6000                         /* Vista/AMD64 */
                 && !memcmp(&pbPrcb[0x38bc], &u.szVendor[0], 4*3))
        {
            g_offrtNtPbQuantumEnd    = 0x3375;
            g_cbrtNtPbQuantumEnd     = 1;
            g_offrtNtPbDpcQueueDepth = 0x3300 + 0x18;
        }
        /* WindowsVista.6002.090410-1830.amd64fre.Symbols
           WindowsVista.6002.090130-1715.amd64fre.Symbols
           WindowsVista.6002.090410-1830.amd64chk.Symbols */
        else if (   BuildNumber == 6002
                 && !memcmp(&pbPrcb[0x399c], &u.szVendor[0], 4*3))
        {
            g_offrtNtPbQuantumEnd    = 0x3475;
            g_cbrtNtPbQuantumEnd     = 1;
            g_offrtNtPbDpcQueueDepth = 0x3400 + 0x18;
        }
        /* Windows7.7600.16539.amd64fre.win7_gdr.100226-1909 */
        else if (    BuildNumber == 7600
                 && !memcmp(&pbPrcb[0x4bb8], &u.szVendor[0], 4*3))
        {
            g_offrtNtPbQuantumEnd    = 0x21d9;
            g_cbrtNtPbQuantumEnd     = 1;
            g_offrtNtPbDpcQueueDepth = 0x2180 + 0x18;
        }

#else
# error "port me"
#endif
    }
    __except(EXCEPTION_EXECUTE_HANDLER) /** @todo this handler doesn't seem to work... Because of Irql? */
    {
        g_offrtNtPbQuantumEnd    = 0;
        g_cbrtNtPbQuantumEnd     = 0;
        g_offrtNtPbDpcQueueDepth = 0;
    }

    KeLowerIrql(OldIrql);

#ifndef IN_GUEST /** @todo fix above for all Nt versions. */
    if (!g_offrtNtPbQuantumEnd && !g_offrtNtPbDpcQueueDepth)
        DbgPrint("IPRT: Neither _KPRCB::QuantumEnd nor _KPRCB::DpcQueueDepth was not found! Kernel %u.%u %u %s\n",
                 MajorVersion, MinorVersion, BuildNumber, fChecked ? "checked" : "free");
# ifdef DEBUG
    else
        DbgPrint("IPRT: _KPRCB:{.QuantumEnd=%x/%d, .DpcQueueDepth=%x/%d} Kernel %ul.%ul %ul %s\n",
                 g_offrtNtPbQuantumEnd, g_cbrtNtPbQuantumEnd, g_offrtNtPbDpcQueueDepth,
                 MajorVersion, MinorVersion, BuildNumber, fChecked ? "checked" : "free");
# endif
#endif

    return VINF_SUCCESS;
}


void rtR0TermNative(void)
{
}