iokit/Tests/TestDevice.cpp

   1 /*
   2  * Copyright (c) 1998-2000 Apple Computer, Inc. All rights reserved.
   3  *
   4  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. The rights granted to you under the License
  10  * may not be used to create, or enable the creation or redistribution of,
  11  * unlawful or unlicensed copies of an Apple operating system, or to
  12  * circumvent, violate, or enable the circumvention or violation of, any
  13  * terms of an Apple operating system software license agreement.
  14  *
  15  * Please obtain a copy of the License at
  16  * http://www.opensource.apple.com/apsl/ and read it before using this file.
  17  *
  18  * The Original Code and all software distributed under the License are
  19  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  20  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  21  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  22  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  23  * Please see the License for the specific language governing rights and
  24  * limitations under the License.
  25  *
  26  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  27  */
  28 #if DEBUG
  29
  30 #include "Tests.h"
  31
  32 #include <IOKit/IOCommandQueue.h>
  33 #include <IOKit/IOInterruptEventSource.h>
  34 #include <IOKit/IOWorkLoop.h>
  35
  36 #include <mach/sync_policy.h>
  37
  38 #define super OSObject
  39
  40 static TestDevice *sDevice;
  41
  42 static mach_timespec_t hundredMill = { 0, 100000000 };
  43 static semaphore_port_t completeSema;
  44
  45 OSDefineMetaClassAndStructors(TestDevice, OSObject)
  46
  47 kern_return_t
  48 TestDevice::enqueueCommand(bool sleep,
  49     TestDeviceAction act, int tag, void *dataP)
  50 {
  51         return commQ->enqueueCommand(sleep, (void *) act, (void *) tag, dataP);
  52 }
  53
  54 bool
  55 TestDevice::init()
  56 {
  57         if (!super::init()) {
  58                 return false;
  59         }
  60
  61         workLoop = IOWorkLoop::workLoop();
  62         if (!workLoop) {
  63                 return false;
  64         }
  65
  66         commQ = IOCommandQueue::commandQueue
  67             (this, (IOCommandQueueAction) & rawCommandOccurred, 8);
  68         if (!commQ || kIOReturnSuccess != workLoop->addEventSource(commQ)) {
  69                 return false;
  70         }
  71
  72         intES = IOInterruptEventSource::interruptEventSource
  73             (this, (IOInterruptEventAction) & interruptAction);
  74         if (!intES || kIOReturnSuccess != workLoop->addEventSource(intES)) {
  75                 return false;
  76         }
  77
  78         return true;
  79 }
  80
  81 void
  82 TestDevice::free()
  83 {
  84         if (intES) {
  85                 intES->release();
  86         }
  87         if (commQ) {
  88                 commQ->release();
  89         }
  90         if (workLoop) {
  91                 workLoop->release();
  92         }
  93
  94         super::free();
  95 }
  96
  97 void
  98 TestDevice::rawCommandOccurred
  99 (void *field0, void *field1, void *field2, void *)
 100 {
 101         (*(TestDeviceAction) field0)(this, (int) field1, field2);
 102 }
 103
 104 void
 105 TestDevice::interruptAction(IOInterruptEventSource *, int count)
 106 {
 107         logPrintf(("I(%d, %d) ", count, ++intCount));
 108 }
 109
 110 void
 111 TestDevice::producer1Action(int tag)
 112 {
 113         logPrintf(("C1(%d) ", tag));
 114 }
 115
 116 void
 117 TestDevice::producer2Action(int tag, void *count)
 118 {
 119         logPrintf(("C2(%d,%d) ", tag, (int) count));
 120         if (!(tag % 10)) {
 121                 IOSleep(1000);
 122         }
 123 }
 124
 125 void
 126 TestDevice::alarm()
 127 {
 128         intES->interruptOccurred(0, 0, 0);
 129         IOScheduleFunc((IOThreadFunc) alarm, (void *) this, hundredMill, 1);
 130 }
 131
 132 static void
 133 producer(void *inProducerId)
 134 {
 135         int producerId = (int) inProducerId;
 136         TestDeviceAction command;
 137         int i;
 138
 139         semaphore_wait(completeSema);
 140
 141         if (producerId & 1) {
 142                 command = (TestDeviceAction) sDevice->producer1Action;
 143         } else {
 144                 command = (TestDeviceAction) sDevice->producer2Action;
 145         }
 146
 147         for (i = 0; i < 5 * (producerId << 1); i++) {
 148                 sDevice->enqueueCommand
 149                 (true, command, i, (void *) (i % (producerId + 1)));
 150                 if (!(i % (producerId + 1))) {
 151                         /* cthread_yield() */;
 152                 }
 153                 logPrintf(("TestDevice(%d): %d\n", producerId, i));
 154         }
 155
 156         logPrintf(("TestDevice: producer %d exiting\n", producerId));
 157         semaphore_signal(completeSema);
 158
 159         IOExitThread(producerId);
 160 }
 161
 162 void
 163 testWorkLoop()
 164 {
 165         int i;
 166
 167         sDevice = new TestDevice;
 168         if (!sDevice || !sDevice->init()) {
 169                 if (sDevice) {
 170                         sDevice->free();
 171                 }
 172                 logPrintf(("TestDevice: couldn't create device instance\n"));
 173                 return;
 174         }
 175
 176         IOSleep(1000);
 177
 178         IOScheduleFunc((IOThreadFunc) sDevice->alarm, sDevice, hundredMill, 1);
 179
 180         IOSleep(2000);
 181
 182         if (KERN_SUCCESS
 183             != semaphore_create(kernel_task, &completeSema, SYNC_POLICY_FIFO, 4)) {
 184                 return;
 185         }
 186
 187         IOCreateThread(producer, (void *) 4);
 188         IOCreateThread(producer, (void *) 3);
 189         IOCreateThread(producer, (void *) 2);
 190         IOCreateThread(producer, (void *) 1);
 191
 192         IOSleep(2000);
 193
 194         for (i = 0; i < 4; i++) {
 195                 semaphore_wait(completeSema);
 196         }
 197
 198         IOUnscheduleFunc((IOThreadFunc) sDevice->alarm, sDevice);
 199
 200         sDevice->free(); sDevice = 0;
 201
 202         logPrintf(("TestDevice: exiting\n"));
 203 }
 204
 205 #endif /* DEBUG */