/* $Id: tstSemMutex.cpp 8245 2008-04-21 17:24:28Z vboxsync $ */ /** @file * IPRT Testcase - Simple Semaphore Smoke Test. */ /* * Copyright (C) 2006-2007 Sun Microsystems, Inc. * * This file is part of VirtualBox Open Source Edition (OSE), as * available from http://www.virtualbox.org. This file is free software; * you can redistribute it and/or modify it under the terms of the GNU * General Public License (GPL) as published by the Free Software * Foundation, in version 2 as it comes in the "COPYING" file of the * VirtualBox OSE distribution. VirtualBox OSE is distributed in the * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind. * * The contents of this file may alternatively be used under the terms * of the Common Development and Distribution License Version 1.0 * (CDDL) only, as it comes in the "COPYING.CDDL" file of the * VirtualBox OSE distribution, in which case the provisions of the * CDDL are applicable instead of those of the GPL. * * You may elect to license modified versions of this file under the * terms and conditions of either the GPL or the CDDL or both. * * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa * Clara, CA 95054 USA or visit http://www.sun.com if you need * additional information or have any questions. */ /******************************************************************************* * Header Files * *******************************************************************************/ #include #include #include #include #include #include #include #include /******************************************************************************* * Global Variables * *******************************************************************************/ static RTSEMMUTEX g_hMutex = NIL_RTSEMMUTEX; static bool volatile g_fTerminate; static bool g_fYield; static bool g_fQuiet; static uint32_t volatile g_cbConcurrent; static uint32_t volatile g_cErrors; int PrintError(const char *pszFormat, ...) { ASMAtomicIncU32(&g_cErrors); RTPrintf("tstSemMutex: FAILURE - "); va_list va; va_start(va, pszFormat); RTPrintfV(pszFormat, va); va_end(va); return 1; } int ThreadTest1(RTTHREAD ThreadSelf, void *pvUser) { uint64_t *pu64 = (uint64_t *)pvUser; for (;;) { int rc = RTSemMutexRequestNoResume(g_hMutex, RT_INDEFINITE_WAIT); if (RT_FAILURE(rc)) { PrintError("%x: RTSemMutexRequestNoResume failed with %Rrc\n", rc); break; } if (ASMAtomicIncU32(&g_cbConcurrent) != 1) { PrintError("g_cbConcurrent=%d after request!\n", g_cbConcurrent); break; } /* * Check for fairness: The values of the threads should not differ too much */ (*pu64)++; /* * Check for correctness: Give other threads a chance. If the implementation is * correct, no other thread will be able to enter this lock now. */ if (g_fYield) RTThreadYield(); if (ASMAtomicDecU32(&g_cbConcurrent) != 0) { PrintError("g_cbConcurrent=%d before release!\n", g_cbConcurrent); break; } rc = RTSemMutexRelease(g_hMutex); if (RT_FAILURE(rc)) { PrintError("%x: RTSemMutexRelease failed with %Rrc\n", rc); break; } if (g_fTerminate) break; } if (!g_fQuiet) RTPrintf("tstSemMutex: Thread %08x exited with %lld\n", ThreadSelf, *pu64); return VINF_SUCCESS; } static int Test1(unsigned cThreads, unsigned cSeconds, bool fYield, bool fQuiet) { int rc; unsigned i; uint64_t g_au64[32]; RTTHREAD aThreads[RT_ELEMENTS(g_au64)]; AssertRelease(cThreads <= RT_ELEMENTS(g_au64)); /* * Init globals. */ g_fYield = fYield; g_fQuiet = fQuiet; g_fTerminate = false; rc = RTSemMutexCreate(&g_hMutex); if (RT_FAILURE(rc)) return PrintError("RTSemMutexCreate failed (rc=%Rrc)\n", rc); /* * Create the threads and let them block on the mutex. */ rc = RTSemMutexRequest(g_hMutex, RT_INDEFINITE_WAIT); if (RT_FAILURE(rc)) return PrintError("RTSemMutexRequest failed (rc=%Rrc)\n", rc); for (i = 0; i < cThreads; i++) { g_au64[i] = 0; rc = RTThreadCreate(&aThreads[i], ThreadTest1, &g_au64[i], 0, RTTHREADTYPE_DEFAULT, RTTHREADFLAGS_WAITABLE, "test"); if (RT_FAILURE(rc)) return PrintError("RTThreadCreate failed for thread %u (rc=%Rrc)\n", i, rc); } if (!fQuiet) RTPrintf("tstSemMutex: %zu Threads created. Racing them for %u seconds (%s) ...\n", cThreads, cSeconds, g_fYield ? "yielding" : "no yielding"); uint64_t u64StartTS = RTTimeNanoTS(); rc = RTSemMutexRelease(g_hMutex); if (RT_FAILURE(rc)) PrintError("RTSemMutexRelease failed (rc=%Rrc)\n", rc); RTThreadSleep(cSeconds * 1000); ASMAtomicXchgBool(&g_fTerminate, true); uint64_t ElapsedNS = RTTimeNanoTS() - u64StartTS; for (i = 0; i < cThreads; i++) { rc = RTThreadWait(aThreads[i], 5000, NULL); if (RT_FAILURE(rc)) PrintError("RTThreadWait failed for thread %u (rc=%Rrc)\n", i, rc); } rc = RTSemMutexDestroy(g_hMutex); if (RT_FAILURE(rc)) PrintError("RTSemMutexDestroy failed - %Rrc\n", rc); g_hMutex = NIL_RTSEMMUTEX; if (g_cErrors) RTThreadSleep(100); /* * Collect and display the results. */ uint64_t Total = g_au64[0]; for (i = 1; i < cThreads; i++) Total += g_au64[i]; uint64_t Normal = Total / cThreads; uint64_t MaxDiviation = 0; for (i = 0; i < cThreads; i++) { uint64_t Delta = RT_ABS((int64_t)(g_au64[i] - Normal)); if (Delta > Normal / 2) RTPrintf("tstSemMutex: Warning! Thread %d diviates by more than 50%% - %llu (it) vs. %llu (avg)\n", i, g_au64[i], Normal); if (Delta > MaxDiviation) MaxDiviation = Delta; } RTPrintf("tstSemMutex: Threads: %u Total: %llu Per Sec: %llu Avg: %llu ns Max div: %llu%%\n", cThreads, Total, Total / cSeconds, ElapsedNS / Total, MaxDiviation * 100 / Normal ); return 0; } int main(int argc, char **argv) { int rc = RTR3Init(false, 0); if (RT_FAILURE(rc)) { RTPrintf("tstSemMutex: RTR3Init failed (rc=%Rrc)\n", rc); return 1; } RTPrintf("tstSemMutex: TESTING...\n"); if (argc == 1) { /* threads, seconds, yield, quiet */ Test1( 1, 1, true, false); Test1( 2, 1, true, false); Test1( 10, 1, true, false); Test1( 10, 10, false, false); RTPrintf("tstSemMutex: benchmarking...\n"); for (unsigned cThreads = 1; cThreads < 32; cThreads++) Test1(cThreads, 2, false, true); /** @todo add a testcase where some stuff times out. */ } else { /* threads, seconds, yield, quiet */ RTPrintf("tstSemMutex: benchmarking...\n"); Test1( 1, 3, false, true); Test1( 1, 3, false, true); Test1( 1, 3, false, true); Test1( 2, 3, false, true); Test1( 2, 3, false, true); Test1( 2, 3, false, true); Test1( 3, 3, false, true); Test1( 3, 3, false, true); Test1( 3, 3, false, true); } if (!g_cErrors) RTPrintf("tstSemMutex: SUCCESS\n"); else RTPrintf("tstSemMutex: FAILURE - %u errors\n", g_cErrors); return g_cErrors != 0; }