xnu-344.32.tar.gz

[apple/xnu.git] / iokit / Tests / TestDevice.cpp

/*
 * Copyright (c) 1998-2000 Apple Computer, Inc. All rights reserved.
 *
 * @APPLE_LICENSE_HEADER_START@
 * 
 * The contents of this file constitute Original Code as defined in and
 * are subject to the Apple Public Source License Version 1.1 (the
 * "License").  You may not use this file except in compliance with the
 * License.  Please obtain a copy of the License at
 * http://www.apple.com/publicsource and read it before using this file.
 * 
 * This Original Code and all software distributed under the License are
 * distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, EITHER
 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT.  Please see the
 * License for the specific language governing rights and limitations
 * under the License.
 * 
 * @APPLE_LICENSE_HEADER_END@
 */
#if DEBUG

#include "Tests.h"

#include <IOKit/IOCommandQueue.h>
#include <IOKit/IOInterruptEventSource.h>
#include <IOKit/IOWorkLoop.h>

#include <mach/sync_policy.h>

#define super OSObject

static TestDevice *sDevice;

static mach_timespec_t hundredMill = { 0, 100000000 };
static semaphore_port_t completeSema;

OSDefineMetaClassAndStructors(TestDevice, OSObject)

kern_return_t
TestDevice::enqueueCommand(bool sleep,
                           TestDeviceAction act, int tag, void *dataP)
{
    return commQ->enqueueCommand(sleep, (void *) act, (void *) tag, dataP);
}

bool TestDevice::init()
{
    if ( !super::init() )
        return false;

    workLoop = IOWorkLoop::workLoop();
    if ( !workLoop )
        return false;

    commQ = IOCommandQueue::commandQueue
        (this, (IOCommandQueueAction) &rawCommandOccurred, 8);
    if (!commQ || kIOReturnSuccess != workLoop->addEventSource(commQ))
        return false;

    intES = IOInterruptEventSource::interruptEventSource
        (this, (IOInterruptEventAction) &interruptAction);
    if (!intES || kIOReturnSuccess != workLoop->addEventSource(intES))
        return false;

    return true;
}

void TestDevice::free()
{
    if (intES)    intES->release();
    if (commQ)    commQ->release();
    if (workLoop) workLoop->release();

    super::free(); 
}

void
TestDevice::rawCommandOccurred
    (void *field0, void *field1, void *field2, void *)
{
    (*(TestDeviceAction) field0)(this, (int) field1, field2);
}

void
TestDevice::interruptAction(IOInterruptEventSource *, int count)
{
    logPrintf(("I(%d, %d) ", count, ++intCount));
}

void
TestDevice::producer1Action(int tag)
{
    logPrintf(("C1(%d) ", tag));
}

void
TestDevice::producer2Action(int tag, void *count)
{
    logPrintf(("C2(%d,%d) ", tag, (int) count));
    if ( !(tag % 10) )
        IOSleep(1000);
}

void
TestDevice::alarm()
{
    intES->interruptOccurred(0, 0, 0);
    IOScheduleFunc((IOThreadFunc) alarm, (void *) this, hundredMill, 1);
}

static void producer(void *inProducerId)
{
    int producerId = (int) inProducerId;
    TestDeviceAction command;
    int i;

    semaphore_wait(completeSema);

    if (producerId & 1)
        command = (TestDeviceAction) sDevice->producer1Action;
    else
        command = (TestDeviceAction) sDevice->producer2Action;

    for (i = 0; i < 5 * (producerId << 1); i++) {
        sDevice->enqueueCommand
                (true, command, i, (void *) (i % (producerId + 1)));
        if ( !(i % (producerId + 1)) )
            /* cthread_yield() */;
        logPrintf(("TestDevice(%d): %d\n", producerId, i));
    }

    logPrintf(("TestDevice: producer %d exiting\n", producerId));
    semaphore_signal(completeSema);

    IOExitThread(producerId);
}

void testWorkLoop()
{
    int i;

    sDevice = new TestDevice;
    if (!sDevice || !sDevice->init()) {
        if (sDevice) sDevice->free();
        logPrintf(("TestDevice: couldn't create device instance\n"));
        return;
    }

    IOSleep(1000);

    IOScheduleFunc((IOThreadFunc) sDevice->alarm, sDevice, hundredMill, 1);

    IOSleep(2000);

    if (KERN_SUCCESS
    !=  semaphore_create(kernel_task, &completeSema, SYNC_POLICY_FIFO, 4))
        return;

    IOCreateThread(producer, (void *) 4);
    IOCreateThread(producer, (void *) 3);
    IOCreateThread(producer, (void *) 2);
    IOCreateThread(producer, (void *) 1);

    IOSleep(2000);

    for (i = 0; i < 4; i++)
        semaphore_wait(completeSema);

    IOUnscheduleFunc((IOThreadFunc) sDevice->alarm, sDevice);

    sDevice->free(); sDevice = 0;

    logPrintf(("TestDevice: exiting\n"));
}

#endif /* DEBUG */