int main()
{
RTR3InitExeNoArguments(0);
RTPrintf("tstMp-1: TESTING...\n");
/*
* Present and possible CPUs.
*/
RTCPUID cCpus = RTMpGetCount();
if (cCpus > 0)
RTPrintf("tstMp-1: RTMpGetCount -> %d\n", (int)cCpus);
else
{
RTPrintf("tstMp-1: FAILURE: RTMpGetCount -> %d\n", (int)cCpus);
g_cErrors++;
cCpus = 1;
}
RTCPUSET Set;
PRTCPUSET pSet = RTMpGetSet(&Set);
if (pSet == &Set)
{
if ((RTCPUID)RTCpuSetCount(&Set) != cCpus)
{
RTPrintf("tstMp-1: FAILURE: RTMpGetSet returned a set with a different cpu count; %d, expected %d\n",
RTCpuSetCount(&Set), cCpus);
g_cErrors++;
}
RTPrintf("tstMp-1: Possible CPU mask:\n");
for (int iCpu = 0; iCpu < RTCPUSET_MAX_CPUS; iCpu++)
{
RTCPUID idCpu = RTMpCpuIdFromSetIndex(iCpu);
if (RTCpuSetIsMemberByIndex(&Set, iCpu))
{
RTPrintf("tstMp-1: %2d - id %d: %u/%u MHz", iCpu, (int)idCpu,
RTMpGetCurFrequency(idCpu), RTMpGetMaxFrequency(idCpu));
if (RTMpIsCpuPresent(idCpu))
RTPrintf(RTMpIsCpuOnline(idCpu) ? " online\n" : " offline\n");
else
{
if (!RTMpIsCpuOnline(idCpu))
RTPrintf(" absent\n");
else
{
RTPrintf(" online but absent!\n");
RTPrintf("tstMp-1: FAILURE: Cpu with index %d is report as !RTIsCpuPresent while RTIsCpuOnline returns true!\n", iCpu);
g_cErrors++;
}
}
if (!RTMpIsCpuPossible(idCpu))
{
RTPrintf("tstMp-1: FAILURE: Cpu with index %d is returned by RTCpuSet but not RTMpIsCpuPossible!\n", iCpu);
g_cErrors++;
}
}
else if (RTMpIsCpuPossible(idCpu))
{
RTPrintf("tstMp-1: FAILURE: Cpu with index %d is returned by RTMpIsCpuPossible but not RTCpuSet!\n", iCpu);
g_cErrors++;
}
else if (RTMpGetCurFrequency(idCpu) != 0)
{
RTPrintf("tstMp-1: FAILURE: RTMpGetCurFrequency(%d[idx=%d]) didn't return 0 as it should\n", (int)idCpu, iCpu);
g_cErrors++;
}
else if (RTMpGetMaxFrequency(idCpu) != 0)
{
RTPrintf("tstMp-1: FAILURE: RTMpGetMaxFrequency(%d[idx=%d]) didn't return 0 as it should\n", (int)idCpu, iCpu);
g_cErrors++;
}
}
}
else
{
RTPrintf("tstMp-1: FAILURE: RTMpGetSet -> %p, expected %p\n", pSet, &Set);
g_cErrors++;
RTCpuSetEmpty(&Set);
RTCpuSetAdd(&Set, RTMpCpuIdFromSetIndex(0));
}
/*
* Online CPUs.
*/
RTCPUID cCpusOnline = RTMpGetOnlineCount();
if (cCpusOnline > 0)
{
if (cCpusOnline <= cCpus)
RTPrintf("tstMp-1: RTMpGetOnlineCount -> %d\n", (int)cCpusOnline);
else
{
RTPrintf("tstMp-1: FAILURE: RTMpGetOnlineCount -> %d, expected <= %d\n", (int)cCpusOnline, (int)cCpus);
g_cErrors++;
cCpusOnline = cCpus;
}
}
else
{
RTPrintf("tstMp-1: FAILURE: RTMpGetOnlineCount -> %d\n", (int)cCpusOnline);
g_cErrors++;
cCpusOnline = 1;
}
RTCPUSET SetOnline;
pSet = RTMpGetOnlineSet(&SetOnline);
if (pSet == &SetOnline)
{
if (RTCpuSetCount(&SetOnline) <= 0)
{
RTPrintf("tstMp-1: FAILURE: RTMpGetOnlineSet returned an empty set!\n");
g_cErrors++;
}
else if ((RTCPUID)RTCpuSetCount(&SetOnline) > cCpus)
{
RTPrintf("tstMp-1: FAILURE: RTMpGetOnlineSet returned a too high value; %d, expected <= %d\n",
RTCpuSetCount(&SetOnline), cCpus);
g_cErrors++;
}
RTPrintf("tstMp-1: Online CPU mask:\n");
for (int iCpu = 0; iCpu < RTCPUSET_MAX_CPUS; iCpu++)
if (RTCpuSetIsMemberByIndex(&SetOnline, iCpu))
{
RTCPUID idCpu = RTMpCpuIdFromSetIndex(iCpu);
RTPrintf("tstMp-1: %2d - id %d: %u/%u MHz %s\n", iCpu, (int)idCpu, RTMpGetCurFrequency(idCpu),
RTMpGetMaxFrequency(idCpu), RTMpIsCpuOnline(idCpu) ? "online" : "offline");
if (!RTCpuSetIsMemberByIndex(&Set, iCpu))
{
RTPrintf("tstMp-1: FAILURE: online cpu with index %2d is not a member of the possible cpu set!\n", iCpu);
g_cErrors++;
}
}
/* There isn't any sane way of testing RTMpIsCpuOnline really... :-/ */
}
else
{
RTPrintf("tstMp-1: FAILURE: RTMpGetOnlineSet -> %p, expected %p\n", pSet, &Set);
g_cErrors++;
}
/*
* Present CPUs.
*/
RTCPUID cCpusPresent = RTMpGetPresentCount();
if (cCpusPresent > 0)
{
if ( cCpusPresent <= cCpus
&& cCpusPresent >= cCpusOnline)
RTPrintf("tstMp-1: RTMpGetPresentCount -> %d\n", (int)cCpusPresent);
else
{
RTPrintf("tstMp-1: FAILURE: RTMpGetPresentCount -> %d, expected <= %d and >= %d\n", (int)cCpusPresent, (int)cCpus, (int)cCpusOnline);
g_cErrors++;
}
}
else
{
RTPrintf("tstMp-1: FAILURE: RTMpGetPresentCount -> %d\n", (int)cCpusPresent);
g_cErrors++;
cCpusPresent = 1;
}
RTCPUSET SetPresent;
pSet = RTMpGetPresentSet(&SetPresent);
if (pSet == &SetPresent)
{
if (RTCpuSetCount(&SetPresent) <= 0)
{
RTPrintf("tstMp-1: FAILURE: RTMpGetPresentSet returned an empty set!\n");
g_cErrors++;
}
else if ((RTCPUID)RTCpuSetCount(&SetPresent) != cCpusPresent)
{
RTPrintf("tstMp-1: FAILURE: RTMpGetPresentSet returned a bad value; %d, expected = %d\n",
RTCpuSetCount(&SetPresent), cCpusPresent);
g_cErrors++;
}
RTPrintf("tstMp-1: Present CPU mask:\n");
for (int iCpu = 0; iCpu < RTCPUSET_MAX_CPUS; iCpu++)
if (RTCpuSetIsMemberByIndex(&SetPresent, iCpu))
{
RTCPUID idCpu = RTMpCpuIdFromSetIndex(iCpu);
RTPrintf("tstMp-1: %2d - id %d: %u/%u MHz %s\n", iCpu, (int)idCpu, RTMpGetCurFrequency(idCpu),
RTMpGetMaxFrequency(idCpu), RTMpIsCpuPresent(idCpu) ? "present" : "absent");
if (!RTCpuSetIsMemberByIndex(&Set, iCpu))
{
RTPrintf("tstMp-1: FAILURE: online cpu with index %2d is not a member of the possible cpu set!\n", iCpu);
g_cErrors++;
}
}
/* There isn't any sane way of testing RTMpIsCpuPresent really... :-/ */
}
else
{
RTPrintf("tstMp-1: FAILURE: RTMpGetPresentSet -> %p, expected %p\n", pSet, &Set);
g_cErrors++;
}
/* Find an online cpu for the next test. */
RTCPUID idCpuOnline;
for (idCpuOnline = 0; idCpuOnline < RTCPUSET_MAX_CPUS; idCpuOnline++)
if (RTMpIsCpuOnline(idCpuOnline))
break;
/*
* Quick test of RTMpGetDescription.
*/
char szBuf[64];
int rc = RTMpGetDescription(idCpuOnline, &szBuf[0], sizeof(szBuf));
if (RT_SUCCESS(rc))
{
RTPrintf("tstMp-1: RTMpGetDescription -> '%s'\n", szBuf);
size_t cch = strlen(szBuf);
rc = RTMpGetDescription(idCpuOnline, &szBuf[0], cch);
if (rc != VERR_BUFFER_OVERFLOW)
{
RTPrintf("tstMp-1: FAILURE: RTMpGetDescription -> %Rrc, expected VERR_BUFFER_OVERFLOW\n", rc);
g_cErrors++;
}
rc = RTMpGetDescription(idCpuOnline, &szBuf[0], cch + 1);
if (RT_FAILURE(rc))
{
RTPrintf("tstMp-1: FAILURE: RTMpGetDescription -> %Rrc, expected VINF_SUCCESS\n", rc);
g_cErrors++;
}
}
else
{
RTPrintf("tstMp-1: FAILURE: RTMpGetDescription -> %Rrc\n", rc);
g_cErrors++;
}
if (!g_cErrors)
RTPrintf("tstMp-1: SUCCESS\n", g_cErrors);
else
RTPrintf("tstMp-1: FAILURE - %d errors\n", g_cErrors);
return !!g_cErrors;
}
int main()
{
RTTEST hTest;
RTEXITCODE rcExit = RTTestInitAndCreate("tstRTMp-1", &hTest);
if (rcExit != RTEXITCODE_SUCCESS)
return rcExit;
RTTestBanner(hTest);
/*
* Present and possible CPUs.
*/
RTCPUID cCpus = RTMpGetCount();
if (cCpus > 0)
RTTestIPrintf(RTTESTLVL_ALWAYS, "RTMpGetCount -> %u\n", cCpus);
else
{
RTTestIFailed("RTMpGetCount returned zero");
cCpus = 1;
}
RTCPUID cCoreCpus = RTMpGetCoreCount();
if (cCoreCpus > 0)
RTTestIPrintf(RTTESTLVL_ALWAYS, "RTMpGetCoreCount -> %d\n", (int)cCoreCpus);
else
{
RTTestIFailed("RTMpGetCoreCount returned zero");
cCoreCpus = 1;
}
RTTESTI_CHECK(cCoreCpus <= cCpus);
RTCPUSET Set;
PRTCPUSET pSet = RTMpGetSet(&Set);
RTTESTI_CHECK(pSet == &Set);
if (pSet == &Set)
{
RTTESTI_CHECK((RTCPUID)RTCpuSetCount(&Set) == cCpus);
RTTestIPrintf(RTTESTLVL_ALWAYS, "Possible CPU mask:\n");
for (int iCpu = 0; iCpu < RTCPUSET_MAX_CPUS; iCpu++)
{
RTCPUID idCpu = RTMpCpuIdFromSetIndex(iCpu);
if (RTCpuSetIsMemberByIndex(&Set, iCpu))
{
RTTestIPrintf(RTTESTLVL_ALWAYS, "%2d - id %d: %u/%u MHz", iCpu, (int)idCpu,
RTMpGetCurFrequency(idCpu), RTMpGetMaxFrequency(idCpu));
if (RTMpIsCpuPresent(idCpu))
RTTestIPrintf(RTTESTLVL_ALWAYS, RTMpIsCpuOnline(idCpu) ? " online\n" : " offline\n");
else
{
if (!RTMpIsCpuOnline(idCpu))
RTTestIPrintf(RTTESTLVL_ALWAYS, " absent\n");
else
{
RTTestIPrintf(RTTESTLVL_ALWAYS, " online but absent!\n");
RTTestIFailed("Cpu with index %d is report as !RTIsCpuPresent while RTIsCpuOnline returns true!\n", iCpu);
}
}
if (!RTMpIsCpuPossible(idCpu))
RTTestIFailed("Cpu with index %d is returned by RTCpuSet but not RTMpIsCpuPossible!\n", iCpu);
}
else if (RTMpIsCpuPossible(idCpu))
RTTestIFailed("Cpu with index %d is returned by RTMpIsCpuPossible but not RTCpuSet!\n", iCpu);
else if (RTMpGetCurFrequency(idCpu) != 0)
RTTestIFailed("RTMpGetCurFrequency(%d[idx=%d]) didn't return 0 as it should\n", (int)idCpu, iCpu);
else if (RTMpGetMaxFrequency(idCpu) != 0)
RTTestIFailed("RTMpGetMaxFrequency(%d[idx=%d]) didn't return 0 as it should\n", (int)idCpu, iCpu);
}
}
else
{
RTCpuSetEmpty(&Set);
RTCpuSetAdd(&Set, RTMpCpuIdFromSetIndex(0));
}
/*
* Online CPUs.
*/
RTCPUID cCpusOnline = RTMpGetOnlineCount();
if (cCpusOnline > 0)
{
if (cCpusOnline <= cCpus)
RTTestIPrintf(RTTESTLVL_ALWAYS, "RTMpGetOnlineCount -> %d\n", (int)cCpusOnline);
else
{
RTTestIFailed("RTMpGetOnlineCount -> %d, expected <= %d\n", (int)cCpusOnline, (int)cCpus);
cCpusOnline = cCpus;
}
}
else
{
RTTestIFailed("RTMpGetOnlineCount -> %d\n", (int)cCpusOnline);
cCpusOnline = 1;
}
RTCPUID cCoresOnline = RTMpGetOnlineCoreCount();
if (cCoresOnline > 0)
RTTestIPrintf(RTTESTLVL_ALWAYS, "RTMpGetOnlineCoreCount -> %d\n", (int)cCoresOnline);
else
{
RTTestIFailed("RTMpGetOnlineCoreCount -> %d, expected <= %d\n", (int)cCoresOnline, (int)cCpusOnline);
cCoresOnline = 1;
}
RTTESTI_CHECK(cCoresOnline <= cCpusOnline);
RTCPUSET SetOnline;
pSet = RTMpGetOnlineSet(&SetOnline);
if (pSet == &SetOnline)
{
if (RTCpuSetCount(&SetOnline) <= 0)
RTTestIFailed("RTMpGetOnlineSet returned an empty set!\n");
else if ((RTCPUID)RTCpuSetCount(&SetOnline) > cCpus)
RTTestIFailed("RTMpGetOnlineSet returned a too high value; %d, expected <= %d\n",
RTCpuSetCount(&SetOnline), cCpus);
RTTestIPrintf(RTTESTLVL_ALWAYS, "Online CPU mask:\n");
for (int iCpu = 0; iCpu < RTCPUSET_MAX_CPUS; iCpu++)
if (RTCpuSetIsMemberByIndex(&SetOnline, iCpu))
{
RTCPUID idCpu = RTMpCpuIdFromSetIndex(iCpu);
RTTestIPrintf(RTTESTLVL_ALWAYS, "%2d - id %d: %u/%u MHz %s\n", iCpu, (int)idCpu, RTMpGetCurFrequency(idCpu),
RTMpGetMaxFrequency(idCpu), RTMpIsCpuOnline(idCpu) ? "online" : "offline");
if (!RTCpuSetIsMemberByIndex(&Set, iCpu))
RTTestIFailed("online cpu with index %2d is not a member of the possible cpu set!\n", iCpu);
}
/* There isn't any sane way of testing RTMpIsCpuOnline really... :-/ */
}
else
RTTestIFailed("RTMpGetOnlineSet -> %p, expected %p\n", pSet, &Set);
/*
* Present CPUs.
*/
RTCPUID cCpusPresent = RTMpGetPresentCount();
if (cCpusPresent > 0)
{
if ( cCpusPresent <= cCpus
&& cCpusPresent >= cCpusOnline)
RTTestIPrintf(RTTESTLVL_ALWAYS, "RTMpGetPresentCount -> %d\n", (int)cCpusPresent);
else
RTTestIFailed("RTMpGetPresentCount -> %d, expected <= %d and >= %d\n",
(int)cCpusPresent, (int)cCpus, (int)cCpusOnline);
}
else
{
RTTestIFailed("RTMpGetPresentCount -> %d\n", (int)cCpusPresent);
cCpusPresent = 1;
}
RTCPUSET SetPresent;
pSet = RTMpGetPresentSet(&SetPresent);
if (pSet == &SetPresent)
{
if (RTCpuSetCount(&SetPresent) <= 0)
RTTestIFailed("RTMpGetPresentSet returned an empty set!\n");
else if ((RTCPUID)RTCpuSetCount(&SetPresent) != cCpusPresent)
RTTestIFailed("RTMpGetPresentSet returned a bad value; %d, expected = %d\n",
RTCpuSetCount(&SetPresent), cCpusPresent);
RTTestIPrintf(RTTESTLVL_ALWAYS, "Present CPU mask:\n");
for (int iCpu = 0; iCpu < RTCPUSET_MAX_CPUS; iCpu++)
if (RTCpuSetIsMemberByIndex(&SetPresent, iCpu))
{
RTCPUID idCpu = RTMpCpuIdFromSetIndex(iCpu);
RTTestIPrintf(RTTESTLVL_ALWAYS, "%2d - id %d: %u/%u MHz %s\n", iCpu, (int)idCpu, RTMpGetCurFrequency(idCpu),
RTMpGetMaxFrequency(idCpu), RTMpIsCpuPresent(idCpu) ? "present" : "absent");
if (!RTCpuSetIsMemberByIndex(&Set, iCpu))
RTTestIFailed("online cpu with index %2d is not a member of the possible cpu set!\n", iCpu);
}
/* There isn't any sane way of testing RTMpIsCpuPresent really... :-/ */
}
else
RTTestIFailed("RTMpGetPresentSet -> %p, expected %p\n", pSet, &Set);
/* Find an online cpu for the next test. */
RTCPUID idCpuOnline;
for (idCpuOnline = 0; idCpuOnline < RTCPUSET_MAX_CPUS; idCpuOnline++)
if (RTMpIsCpuOnline(idCpuOnline))
break;
/*
* Quick test of RTMpGetDescription.
*/
char szBuf[64];
int rc = RTMpGetDescription(idCpuOnline, &szBuf[0], sizeof(szBuf));
if (RT_SUCCESS(rc))
{
RTTestIPrintf(RTTESTLVL_ALWAYS, "RTMpGetDescription -> '%s'\n", szBuf);
size_t cch = strlen(szBuf);
rc = RTMpGetDescription(idCpuOnline, &szBuf[0], cch);
if (rc != VERR_BUFFER_OVERFLOW)
RTTestIFailed("RTMpGetDescription -> %Rrc, expected VERR_BUFFER_OVERFLOW\n", rc);
rc = RTMpGetDescription(idCpuOnline, &szBuf[0], cch + 1);
if (RT_FAILURE(rc))
RTTestIFailed("RTMpGetDescription -> %Rrc, expected VINF_SUCCESS\n", rc);
}
else
RTTestIFailed("RTMpGetDescription -> %Rrc\n", rc);
return RTTestSummaryAndDestroy(hTest);
}
/**
* The timer callback for an omni-timer.
*
* This is responsible for queueing the DPCs for the other CPUs and
* perform the callback on the CPU on which it is called.
*
* @param pDpc The DPC object.
* @param pvUser Pointer to the sub-timer.
* @param SystemArgument1 Some system stuff.
* @param SystemArgument2 Some system stuff.
*/
static void _stdcall rtTimerNtOmniMasterCallback(IN PKDPC pDpc, IN PVOID pvUser, IN PVOID SystemArgument1, IN PVOID SystemArgument2)
{
PRTTIMERNTSUBTIMER pSubTimer = (PRTTIMERNTSUBTIMER)pvUser;
PRTTIMER pTimer = pSubTimer->pParent;
int iCpuSelf = RTMpCpuIdToSetIndex(RTMpCpuId());
AssertPtr(pTimer);
#ifdef RT_STRICT
if (KeGetCurrentIrql() < DISPATCH_LEVEL)
RTAssertMsg2Weak("rtTimerNtOmniMasterCallback: Irql=%d expected >=%d\n", KeGetCurrentIrql(), DISPATCH_LEVEL);
if (pSubTimer - &pTimer->aSubTimers[0] != iCpuSelf)
RTAssertMsg2Weak("rtTimerNtOmniMasterCallback: iCpuSelf=%d pSubTimer=%p / %d\n", iCpuSelf, pSubTimer, pSubTimer - &pTimer->aSubTimers[0]);
#endif
/*
* Check that we haven't been suspended before scheduling the other DPCs
* and doing the callout.
*/
if ( !ASMAtomicUoReadBool(&pTimer->fSuspended)
&& pTimer->u32Magic == RTTIMER_MAGIC)
{
RTCPUSET OnlineSet;
RTMpGetOnlineSet(&OnlineSet);
ASMAtomicWriteHandle(&pSubTimer->hActiveThread, RTThreadNativeSelf());
if (pTimer->u64NanoInterval)
{
/*
* Recurring timer.
*/
for (int iCpu = 0; iCpu < RTCPUSET_MAX_CPUS; iCpu++)
if ( RTCpuSetIsMemberByIndex(&OnlineSet, iCpu)
&& iCpuSelf != iCpu)
KeInsertQueueDpc(&pTimer->aSubTimers[iCpu].NtDpc, 0, 0);
uint64_t iTick = ++pSubTimer->iTick;
rtTimerNtRearmInternval(pTimer, iTick, &pTimer->aSubTimers[RTMpCpuIdToSetIndex(pTimer->idCpu)].NtDpc);
pTimer->pfnTimer(pTimer, pTimer->pvUser, iTick);
}
else
{
/*
* Single shot timers gets complicated wrt to fSuspended maintance.
*/
uint32_t cCpus = 0;
for (int iCpu = 0; iCpu < RTCPUSET_MAX_CPUS; iCpu++)
if (RTCpuSetIsMemberByIndex(&OnlineSet, iCpu))
cCpus++;
ASMAtomicAddS32(&pTimer->cOmniSuspendCountDown, cCpus);
for (int iCpu = 0; iCpu < RTCPUSET_MAX_CPUS; iCpu++)
if ( RTCpuSetIsMemberByIndex(&OnlineSet, iCpu)
&& iCpuSelf != iCpu)
if (!KeInsertQueueDpc(&pTimer->aSubTimers[iCpu].NtDpc, 0, 0))
ASMAtomicDecS32(&pTimer->cOmniSuspendCountDown); /* already queued and counted. */
if (ASMAtomicDecS32(&pTimer->cOmniSuspendCountDown) <= 0)
ASMAtomicWriteBool(&pTimer->fSuspended, true);
pTimer->pfnTimer(pTimer, pTimer->pvUser, ++pSubTimer->iTick);
}
ASMAtomicWriteHandle(&pSubTimer->hActiveThread, NIL_RTNATIVETHREAD);
}
NOREF(pDpc); NOREF(SystemArgument1); NOREF(SystemArgument2);
}
