#include "TestShmem.h"

#include "IPDLUnitTests.h"      // fail etc.


namespace mozilla {
namespace _ipdltest {

//-----------------------------------------------------------------------------
// Parent

void
TestShmemParent::Main()
{
    Shmem mem;
    size_t size = 12345;
    if (!AllocShmem(size, SharedMemory::TYPE_BASIC, &mem))
        fail("can't alloc shmem");

    if (mem.Size<char>() != size)
        fail("shmem is wrong size: expected %lu, got %lu",
             size, mem.Size<char>());

    char* ptr = mem.get<char>();
    memcpy(ptr, "Hello!", sizeof("Hello!"));
    if (!SendGive(mem, size))
        fail("can't send Give()");

    // uncomment the following line for a (nondeterministic) surprise!
    //char c1 = *ptr;  (void)c1;

    // uncomment the following line for a deterministic surprise!
    //char c2 = *mem.get<char>(); (void)c2;
}


bool
TestShmemParent::RecvTake(Shmem& mem, const size_t& expectedSize)
{
    if (mem.Size<char>() != expectedSize)
        fail("expected shmem size %lu, but it has size %lu",
             expectedSize, mem.Size<char>());

    if (strcmp(mem.get<char>(), "And yourself!"))
        fail("expected message was not written");

    if (!DeallocShmem(mem))
        fail("DeallocShmem");

    Close();

    return true;
}

//-----------------------------------------------------------------------------
// Child

bool
TestShmemChild::RecvGive(Shmem& mem, const size_t& expectedSize)
{
    if (mem.Size<char>() != expectedSize)
        fail("expected shmem size %lu, but it has size %lu",
             expectedSize, mem.Size<char>());

    if (strcmp(mem.get<char>(), "Hello!"))
        fail("expected message was not written");

    memcpy(mem.get<char>(), "And yourself!", sizeof("And yourself!"));

    if (!SendTake(mem, expectedSize))
        fail("can't send Take()");

    return true;
}


} // namespace _ipdltest
} // namespace mozilla