RTDECL(int) RTMpOnPair(RTCPUID idCpu1, RTCPUID idCpu2, uint32_t fFlags, PFNRTMPWORKER pfnWorker, void *pvUser1, void *pvUser2)
{
int rc;
RTMPARGS Args;
RTSOLCPUSET CpuSet;
RTTHREADPREEMPTSTATE PreemptState = RTTHREADPREEMPTSTATE_INITIALIZER;
AssertReturn(idCpu1 != idCpu2, VERR_INVALID_PARAMETER);
AssertReturn(!(fFlags & RTMPON_F_VALID_MASK), VERR_INVALID_FLAGS);
Args.pfnWorker = pfnWorker;
Args.pvUser1 = pvUser1;
Args.pvUser2 = pvUser2;
Args.idCpu = idCpu1;
Args.idCpu2 = idCpu2;
Args.cHits = 0;
for (int i = 0; i < IPRT_SOL_SET_WORDS; i++)
CpuSet.auCpus[i] = 0;
BT_SET(CpuSet.auCpus, idCpu1);
BT_SET(CpuSet.auCpus, idCpu2);
/*
* Check that both CPUs are online before doing the broadcast call.
*/
RTThreadPreemptDisable(&PreemptState);
if ( RTMpIsCpuOnline(idCpu1)
&& RTMpIsCpuOnline(idCpu2))
{
rtMpSolCrossCall(&CpuSet, rtMpSolOnPairCpuWrapper, &Args);
Assert(Args.cHits <= 2);
if (Args.cHits == 2)
rc = VINF_SUCCESS;
else if (Args.cHits == 1)
rc = VERR_NOT_ALL_CPUS_SHOWED;
else if (Args.cHits == 0)
rc = VERR_CPU_OFFLINE;
else
rc = VERR_CPU_IPE_1;
}
/*
* A CPU must be present to be considered just offline.
*/
else if ( RTMpIsCpuPresent(idCpu1)
&& RTMpIsCpuPresent(idCpu2))
rc = VERR_CPU_OFFLINE;
else
rc = VERR_CPU_NOT_FOUND;
RTThreadPreemptRestore(&PreemptState);
return rc;
}
RTDECL(int) RTMpOnSpecific(RTCPUID idCpu, PFNRTMPWORKER pfnWorker, void *pvUser1, void *pvUser2)
{
RTMPARGS Args;
RT_ASSERT_INTS_ON();
if (idCpu >= ncpus)
return VERR_CPU_NOT_FOUND;
if (RT_UNLIKELY(!RTMpIsCpuOnline(idCpu)))
return RTMpIsCpuPresent(idCpu) ? VERR_CPU_OFFLINE : VERR_CPU_NOT_FOUND;
Args.pfnWorker = pfnWorker;
Args.pvUser1 = pvUser1;
Args.pvUser2 = pvUser2;
Args.idCpu = idCpu;
Args.cHits = 0;
RTTHREADPREEMPTSTATE PreemptState = RTTHREADPREEMPTSTATE_INITIALIZER;
RTThreadPreemptDisable(&PreemptState);
RTSOLCPUSET CpuSet;
for (int i = 0; i < IPRT_SOL_SET_WORDS; i++)
CpuSet.auCpus[i] = 0;
BT_SET(CpuSet.auCpus, idCpu);
rtMpSolCrossCall(&CpuSet, rtMpSolOnSpecificCpuWrapper, &Args);
RTThreadPreemptRestore(&PreemptState);
Assert(ASMAtomicUoReadU32(&Args.cHits) <= 1);
return ASMAtomicUoReadU32(&Args.cHits) == 1
? VINF_SUCCESS
: VERR_CPU_NOT_FOUND;
}
RTDECL(int) RTMpOnPair(RTCPUID idCpu1, RTCPUID idCpu2, uint32_t fFlags, PFNRTMPWORKER pfnWorker, void *pvUser1, void *pvUser2)
{
int rc;
AssertReturn(idCpu1 != idCpu2, VERR_INVALID_PARAMETER);
AssertReturn(!(fFlags & RTMPON_F_VALID_MASK), VERR_INVALID_FLAGS);
if ((fFlags & RTMPON_F_CONCURRENT_EXEC) && !g_pfnrtKeIpiGenericCall)
return VERR_NOT_SUPPORTED;
/*
* Check that both CPUs are online before doing the broadcast call.
*/
if ( RTMpIsCpuOnline(idCpu1)
&& RTMpIsCpuOnline(idCpu2))
{
/*
* The broadcast IPI isn't quite as bad as it could have been, because
* it looks like windows doesn't synchronize CPUs on the way out, they
* seems to get back to normal work while the pair is still busy.
*/
uint32_t cHits = 0;
if (g_pfnrtKeIpiGenericCall)
rc = rtMpCallUsingBroadcastIpi(pfnWorker, pvUser1, pvUser2, rtmpNtOnPairBroadcastIpiWrapper, idCpu1, idCpu2, &cHits);
else
rc = rtMpCallUsingDpcs(pfnWorker, pvUser1, pvUser2, RT_NT_CPUID_PAIR, idCpu1, idCpu2, &cHits);
if (RT_SUCCESS(rc))
{
Assert(cHits <= 2);
if (cHits == 2)
rc = VINF_SUCCESS;
else if (cHits == 1)
rc = VERR_NOT_ALL_CPUS_SHOWED;
else if (cHits == 0)
rc = VERR_CPU_OFFLINE;
else
rc = VERR_CPU_IPE_1;
}
}
/*
* A CPU must be present to be considered just offline.
*/
else if ( RTMpIsCpuPresent(idCpu1)
&& RTMpIsCpuPresent(idCpu2))
rc = VERR_CPU_OFFLINE;
else
rc = VERR_CPU_NOT_FOUND;
return rc;
}
RTDECL(PRTCPUSET) RTMpGetPresentSet(PRTCPUSET pSet)
{
#ifdef RT_STRICT
RTCPUID cCpusPresent = 0;
#endif
RTCpuSetEmpty(pSet);
RTCPUID cCpus = RTMpGetCount();
for (RTCPUID idCpu = 0; idCpu < cCpus; idCpu++)
if (RTMpIsCpuPresent(idCpu))
{
RTCpuSetAdd(pSet, idCpu);
#ifdef RT_STRICT
cCpusPresent++;
#endif
}
Assert(cCpusPresent == RTMpGetPresentCount());
return pSet;
}
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);
}
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;
}
RTDECL(int) RTMpOnPair(RTCPUID idCpu1, RTCPUID idCpu2, uint32_t fFlags, PFNRTMPWORKER pfnWorker, void *pvUser1, void *pvUser2)
{
IPRT_LINUX_SAVE_EFL_AC();
int rc;
RTTHREADPREEMPTSTATE PreemptState = RTTHREADPREEMPTSTATE_INITIALIZER;
AssertReturn(idCpu1 != idCpu2, VERR_INVALID_PARAMETER);
AssertReturn(!(fFlags & RTMPON_F_VALID_MASK), VERR_INVALID_FLAGS);
/*
* Check that both CPUs are online before doing the broadcast call.
*/
RTThreadPreemptDisable(&PreemptState);
if ( RTMpIsCpuOnline(idCpu1)
&& RTMpIsCpuOnline(idCpu2))
{
/*
* Use the smp_call_function variant taking a cpu mask where available,
* falling back on broadcast with filter. Slight snag if one of the
* CPUs is the one we're running on, we must do the call and the post
* call wait ourselves.
*/
#if LINUX_VERSION_CODE >= KERNEL_VERSION(2, 6, 27)
cpumask_t DstCpuMask;
#endif
RTCPUID idCpuSelf = RTMpCpuId();
bool const fCallSelf = idCpuSelf == idCpu1 || idCpuSelf == idCpu2;
RTMPARGS Args;
Args.pfnWorker = pfnWorker;
Args.pvUser1 = pvUser1;
Args.pvUser2 = pvUser2;
Args.idCpu = idCpu1;
Args.idCpu2 = idCpu2;
Args.cHits = 0;
#if LINUX_VERSION_CODE >= KERNEL_VERSION(2, 6, 28)
cpumask_clear(&DstCpuMask);
cpumask_set_cpu(idCpu1, &DstCpuMask);
cpumask_set_cpu(idCpu2, &DstCpuMask);
#elif LINUX_VERSION_CODE >= KERNEL_VERSION(2, 6, 27)
cpus_clear(DstCpuMask);
cpu_set(idCpu1, DstCpuMask);
cpu_set(idCpu2, DstCpuMask);
#endif
#if LINUX_VERSION_CODE >= KERNEL_VERSION(2, 6, 29)
smp_call_function_many(&DstCpuMask, rtmpLinuxWrapperPostInc, &Args, !fCallSelf /* wait */);
rc = 0;
#elif LINUX_VERSION_CODE >= KERNEL_VERSION(2, 6, 28)
rc = smp_call_function_many(&DstCpuMask, rtmpLinuxWrapperPostInc, &Args, !fCallSelf /* wait */);
#elif LINUX_VERSION_CODE >= KERNEL_VERSION(2, 6, 27)
rc = smp_call_function_mask(DstCpuMask, rtmpLinuxWrapperPostInc, &Args, !fCallSelf /* wait */);
#else /* older kernels */
rc = smp_call_function(rtMpLinuxOnPairWrapper, &Args, 0 /* retry */, !fCallSelf /* wait */);
#endif /* older kernels */
Assert(rc == 0);
/* Call ourselves if necessary and wait for the other party to be done. */
if (fCallSelf)
{
uint32_t cLoops = 0;
rtmpLinuxWrapper(&Args);
while (ASMAtomicReadU32(&Args.cHits) < 2)
{
if ((cLoops & 0x1ff) == 0 && !RTMpIsCpuOnline(idCpuSelf == idCpu1 ? idCpu2 : idCpu1))
break;
cLoops++;
ASMNopPause();
}
}
Assert(Args.cHits <= 2);
if (Args.cHits == 2)
rc = VINF_SUCCESS;
else if (Args.cHits == 1)
rc = VERR_NOT_ALL_CPUS_SHOWED;
else if (Args.cHits == 0)
rc = VERR_CPU_OFFLINE;
else
rc = VERR_CPU_IPE_1;
}
/*
* A CPU must be present to be considered just offline.
*/
else if ( RTMpIsCpuPresent(idCpu1)
&& RTMpIsCpuPresent(idCpu2))
rc = VERR_CPU_OFFLINE;
else
rc = VERR_CPU_NOT_FOUND;
RTThreadPreemptRestore(&PreemptState);;
IPRT_LINUX_RESTORE_EFL_AC();
return rc;
}
