gecko/ipc/ipdl/test/cxx/TestRPCShutdownRace.cpp

145 lines
3.4 KiB
C++

#include "TestRPCShutdownRace.h"
#include "IPDLUnitTests.h" // fail etc.
#include "IPDLUnitTestSubprocess.h"
template<>
struct RunnableMethodTraits<mozilla::_ipdltest::TestRPCShutdownRaceParent>
{
static void RetainCallee(mozilla::_ipdltest::TestRPCShutdownRaceParent* obj) { }
static void ReleaseCallee(mozilla::_ipdltest::TestRPCShutdownRaceParent* obj) { }
};
namespace mozilla {
namespace _ipdltest {
//-----------------------------------------------------------------------------
// parent
namespace {
// NB: this test does its own shutdown, rather than going through
// QuitParent(), because it's testing degenerate edge cases
void DeleteSubprocess()
{
delete gSubprocess;
gSubprocess = NULL;
}
void Done()
{
passed(__FILE__);
QuitParent();
}
} // namespace <anon>
TestRPCShutdownRaceParent::TestRPCShutdownRaceParent()
{
MOZ_COUNT_CTOR(TestRPCShutdownRaceParent);
}
TestRPCShutdownRaceParent::~TestRPCShutdownRaceParent()
{
MOZ_COUNT_DTOR(TestRPCShutdownRaceParent);
}
void
TestRPCShutdownRaceParent::Main()
{
if (!SendStart())
fail("sending Start");
}
bool
TestRPCShutdownRaceParent::RecvStartDeath()
{
// this will be ordered before the OnMaybeDequeueOne event of
// Orphan in the queue
MessageLoop::current()->PostTask(
FROM_HERE,
NewRunnableMethod(this,
&TestRPCShutdownRaceParent::StartShuttingDown));
return true;
}
void
TestRPCShutdownRaceParent::StartShuttingDown()
{
// NB: we sleep here to try and avoid receiving the Orphan message
// while waiting for the CallExit() reply. if we fail at that, it
// will cause the test to pass spuriously, because there won't be
// an OnMaybeDequeueOne task for Orphan
PR_Sleep(2000);
if (CallExit())
fail("connection was supposed to be interrupted");
Close();
delete static_cast<TestRPCShutdownRaceParent*>(gParentActor);
gParentActor = NULL;
XRE_GetIOMessageLoop()->PostTask(FROM_HERE,
NewRunnableFunction(DeleteSubprocess));
// this is ordered after the OnMaybeDequeueOne event in the queue
MessageLoop::current()->PostTask(FROM_HERE,
NewRunnableFunction(Done));
// |this| has been deleted, be mindful
}
bool
TestRPCShutdownRaceParent::RecvOrphan()
{
// it would be nice to fail() here, but we'll process this message
// while waiting for the reply CallExit(). The OnMaybeDequeueOne
// task will still be in the queue, it just wouldn't have had any
// work to do, if we hadn't deleted ourself
return true;
}
//-----------------------------------------------------------------------------
// child
TestRPCShutdownRaceChild::TestRPCShutdownRaceChild()
{
MOZ_COUNT_CTOR(TestRPCShutdownRaceChild);
}
TestRPCShutdownRaceChild::~TestRPCShutdownRaceChild()
{
MOZ_COUNT_DTOR(TestRPCShutdownRaceChild);
}
bool
TestRPCShutdownRaceChild::RecvStart()
{
if (!SendStartDeath())
fail("sending StartDeath");
// See comment in StartShuttingDown(): we want to send Orphan()
// while the parent is in its PR_Sleep()
PR_Sleep(1000);
if (!SendOrphan())
fail("sending Orphan");
return true;
}
bool
TestRPCShutdownRaceChild::AnswerExit()
{
_exit(0);
NS_RUNTIMEABORT("unreached");
return false;
}
} // namespace _ipdltest
} // namespace mozilla