Support/testcase/tstSupSem-Zombie.cpp

	tstSupSem-Zombie.cpp revision e64031e20c39650a7bc902a3e1aba613b9415dee
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync/* $Id$ */
ad27e1d5e48ca41245120c331cc88b50464813cevboxsync/** @file
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync * Support Library Testcase - Ring-3 Semaphore interface - Zombie bugs.
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync */
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync/*
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync * Copyright (C) 2009 Oracle Corporation
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync *
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync * This file is part of VirtualBox Open Source Edition (OSE), as
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync * available from http://www.virtualbox.org. This file is free software;
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync * you can redistribute it and/or modify it under the terms of the GNU
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync * General Public License (GPL) as published by the Free Software
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync * Foundation, in version 2 as it comes in the "COPYING" file of the
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync *
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync * The contents of this file may alternatively be used under the terms
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync * of the Common Development and Distribution License Version 1.0
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync * VirtualBox OSE distribution, in which case the provisions of the
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync * CDDL are applicable instead of those of the GPL.
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync *
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync * You may elect to license modified versions of this file under the
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync * terms and conditions of either the GPL or the CDDL or both.
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync */
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync/*******************************************************************************
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync*   Header Files                                                               *
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync*******************************************************************************/
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync#include <VBox/sup.h>
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync#include <VBox/param.h>
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync#include <iprt/env.h>
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync#include <iprt/err.h>
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync#include <iprt/initterm.h>
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync#include <iprt/process.h>
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync#include <iprt/stream.h>
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync#include <iprt/string.h>
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync#include <iprt/test.h>
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync#include <iprt/time.h>
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync#include <iprt/thread.h>
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync/*******************************************************************************
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync*   Structures and Typedefs                                                    *
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync*******************************************************************************/
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsyncstatic PSUPDRVSESSION   g_pSession;
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsyncstatic RTTEST           g_hTest;
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsyncstatic DECLCALLBACK(int) tstSupSemSRETimed(RTTHREAD hSelf, void *pvUser)
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync{
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    SUPSEMEVENT hEvent = (SUPSEMEVENT)pvUser;
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    RTThreadUserSignal(hSelf);
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    int rc = SUPSemEventWaitNoResume(g_pSession, hEvent, 120*1000);
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    AssertReleaseMsgFailed(("%Rrc\n", rc));
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    return rc;
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync}
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsyncstatic DECLCALLBACK(int) tstSupSemMRETimed(RTTHREAD hSelf, void *pvUser)
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync{
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync    SUPSEMEVENTMULTI hEventMulti = (SUPSEMEVENTMULTI)pvUser;
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync    RTThreadUserSignal(hSelf);
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync    int rc = SUPSemEventMultiWaitNoResume(g_pSession, hEventMulti, 120*1000);
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync    AssertReleaseMsgFailed(("%Rrc\n", rc));
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync    return rc;
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync}
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsyncstatic DECLCALLBACK(int) tstSupSemSREInf(RTTHREAD hSelf, void *pvUser)
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync{
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    SUPSEMEVENT hEvent = (SUPSEMEVENT)pvUser;
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    RTThreadUserSignal(hSelf);
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync    int rc = SUPSemEventWaitNoResume(g_pSession, hEvent, RT_INDEFINITE_WAIT);
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    AssertReleaseMsgFailed(("%Rrc\n", rc));
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    return rc;
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync}
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsyncstatic DECLCALLBACK(int) tstSupSemMREInf(RTTHREAD hSelf, void *pvUser)
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync{
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    SUPSEMEVENTMULTI hEventMulti = (SUPSEMEVENTMULTI)pvUser;
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync    RTThreadUserSignal(hSelf);
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync    int rc = SUPSemEventMultiWaitNoResume(g_pSession, hEventMulti, RT_INDEFINITE_WAIT);
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    AssertReleaseMsgFailed(("%Rrc\n", rc));
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    return rc;
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync}
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsyncstatic int mainChild(void)
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync{
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    /*
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync     * Init.
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync     */
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    int rc = RTR3InitAndSUPLib();
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync    if (RT_FAILURE(rc))
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync    {
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync        RTPrintf("tstSupSem-Zombie-Child: fatal error: RTR3InitAndSUPLib failed with rc=%Rrc\n", rc);
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync        return 1;
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    }
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    RTTEST hTest;
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    rc = RTTestCreate("tstSupSem-Zombie-Child", &hTest);
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync    if (RT_FAILURE(rc))
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    {
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync        RTPrintf("tstSupSem-Zombie-Child: fatal error: RTTestCreate failed with rc=%Rrc\n", rc);
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync        return 1;
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync    }
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    g_hTest = hTest;
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    PSUPDRVSESSION pSession;
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    rc = SUPR3Init(&pSession);
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    if (RT_FAILURE(rc))
c89ab46314a2ed24bfd30a78fabcd673f66a85c5vboxsync    {
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync        RTTestFailed(hTest, "SUPR3Init failed with rc=%Rrc\n", rc);
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync        return RTTestSummaryAndDestroy(hTest);
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    }
e1bb6985ff821f55a24e0959c809a15871e58c5avboxsync    g_pSession = pSession;

    /*
     * A semaphore of each kind and throw a bunch of threads on them.
     */
    SUPSEMEVENT hEvent = NIL_SUPSEMEVENT;
    RTTESTI_CHECK_RC(rc = SUPSemEventCreate(pSession, &hEvent), VINF_SUCCESS);
    if (RT_SUCCESS(rc))
    {
        SUPSEMEVENTMULTI hEventMulti = NIL_SUPSEMEVENT;
        RTTESTI_CHECK_RC(SUPSemEventMultiCreate(pSession, &hEventMulti), VINF_SUCCESS);
        if (RT_SUCCESS(rc))
        {
            for (uint32_t cThreads = 0; cThreads < 5; cThreads++)
            {
                RTTHREAD hThread;
                RTTESTI_CHECK_RC(RTThreadCreate(&hThread, tstSupSemSRETimed, (void *)hEvent,      0, RTTHREADTYPE_TIMER, 0 /*fFlags*/, "IntSRE"), VINF_SUCCESS);
                RTTESTI_CHECK_RC(RTThreadCreate(&hThread, tstSupSemMRETimed, (void *)hEventMulti, 0, RTTHREADTYPE_TIMER, 0 /*fFlags*/, "IntMRE"), VINF_SUCCESS);
                RTTESTI_CHECK_RC(RTThreadCreate(&hThread, tstSupSemSREInf,   (void *)hEvent,      0, RTTHREADTYPE_TIMER, 0 /*fFlags*/, "IntSRE"), VINF_SUCCESS);
                RTTESTI_CHECK_RC(RTThreadCreate(&hThread, tstSupSemMREInf,   (void *)hEventMulti, 0, RTTHREADTYPE_TIMER, 0 /*fFlags*/, "IntMRE"), VINF_SUCCESS);
                RTThreadSleep(2);
            }
            RTThreadSleep(50);

            /*
             * This is where the test really starts...
             */
            return 0;
        }
    }

    return RTTestSummaryAndDestroy(hTest);
}


/**
 * The parent main routine.
 * @param   argv0       The executable name (or whatever).
 */
static int mainParent(const char *argv0)
{
    /*
     * Init.
     */
    RTTEST hTest;
    int rc = RTTestInitAndCreate("tstSupSem-Zombie", &hTest);
    if (rc)
        return rc;
    RTTestBanner(hTest);

    /*
     * Spin of the child process which may or may not turn into a zombie
     */
    for (uint32_t iPass = 0; iPass < 32; iPass++)
    {
        RTTestSubF(hTest, "Pass %u", iPass);

        RTPROCESS hProcess;
        const char *apszArgs[3] = { argv0, "--child", NULL };
        RTTESTI_CHECK_RC_OK(rc = RTProcCreate(argv0, apszArgs, RTENV_DEFAULT, 0 /*fFlags*/, &hProcess));
        if (RT_SUCCESS(rc))
        {
            /*
             * Wait for 60 seconds then give up.
             */
            RTPROCSTATUS    Status;
            uint64_t        StartTS = RTTimeMilliTS();
            for (;;)
            {
                rc = RTProcWait(hProcess, RTPROCWAIT_FLAGS_NOBLOCK, &Status);
                if (RT_SUCCESS(rc))
                    break;
                uint64_t cElapsed = RTTimeMilliTS() - StartTS;
                if (cElapsed > 60*1000)
                    break;
                RTThreadSleep(cElapsed < 60 ? 30 : cElapsed < 200 ? 10 : 100);
            }
            RTTESTI_CHECK_RC_OK(rc);
            if (    RT_SUCCESS(rc)
                &&  (   Status.enmReason != RTPROCEXITREASON_NORMAL
                     || Status.iStatus != 0))
            {
                RTTestIFailed("child %d (%#x) reason %d\n", Status.iStatus, Status.iStatus, Status.enmReason);
                rc = VERR_PERMISSION_DENIED;
            }
        }
        /* one zombie process is enough. */
        if (RT_FAILURE(rc))
            break;
    }

    return RTTestSummaryAndDestroy(hTest);
}


int main(int argc, char **argv)
{
    if (    argc == 2
        &&  !strcmp(argv[1], "--child"))
        return mainChild();
    return mainParent(argv[0]);
}