tools/tests/testkext/testthreadcall.cpp

   1 /*
   2  *  testthreadcall.cpp
   3  *  testkext
   4  *
   5  */
   6
   7 #include "testthreadcall.h"
   8
   9 #include <kern/thread_call.h>
  10 #include <pexpert/pexpert.h>
  11
  12 #define super IOService
  13 OSDefineMetaClassAndStructors(testthreadcall, super);
  14
  15 extern "C" {
  16 static void thread_call_test_func(thread_call_param_t param0,
  17     thread_call_param_t param1);
  18
  19 static void thread_call_test_func2(thread_call_param_t param0,
  20     thread_call_param_t param1);
  21 }
  22
  23 static int my_event;
  24
  25 bool
  26 testthreadcall::start( IOService * provider )
  27 {
  28         boolean_t ret;
  29         uint64_t deadline;
  30         int sleepret;
  31         uint32_t kernel_configuration;
  32
  33         IOLog("%s\n", __PRETTY_FUNCTION__);
  34
  35         if (!super::start(provider)) {
  36                 return false;
  37         }
  38
  39         kernel_configuration = PE_i_can_has_kernel_configuration();
  40         IOLog("%s: Assertions %s\n", __PRETTY_FUNCTION__,
  41             (kernel_configuration & kPEICanHasAssertions) ? "enabled" : "disabled");
  42         IOLog("%s: Statistics %s\n", __PRETTY_FUNCTION__,
  43             (kernel_configuration & kPEICanHasStatistics) ? "enabled" : "disabled");
  44         IOLog("%s: Diagnostic API %s\n", __PRETTY_FUNCTION__,
  45             (kernel_configuration & kPEICanHasDiagnosticAPI) ? "enabled" : "disabled");
  46
  47         IOLog("Attempting thread_call_allocate\n");
  48         tcall = thread_call_allocate(thread_call_test_func, this);
  49         IOLog("thread_call_t %p\n", tcall);
  50
  51         tlock = IOSimpleLockAlloc();
  52         IOLog("tlock %p\n", tlock);
  53
  54         clock_interval_to_deadline(5, NSEC_PER_SEC, &deadline);
  55         IOLog("%d sec deadline is %llu\n", 5, deadline);
  56
  57         ret = thread_call_enter_delayed(tcall, deadline);
  58
  59         IOLog("Attempting thread_call_allocate\n");
  60         tcall2 = thread_call_allocate(thread_call_test_func2, this);
  61         IOLog("thread_call_t %p\n", tcall);
  62
  63         tlock2 = IOLockAlloc();
  64         IOLog("tlock2 %p\n", tlock2);
  65
  66         clock_interval_to_deadline(2, NSEC_PER_SEC, &deadline);
  67         IOLog("%d sec deadline is %llu\n", 2, deadline);
  68
  69         ret = thread_call_enter_delayed(tcall2, deadline);
  70
  71         IOLockLock(tlock2);
  72
  73         clock_interval_to_deadline(3, NSEC_PER_SEC, &deadline);
  74         IOLog("%d sec deadline is %llu\n", 3, deadline);
  75         sleepret = IOLockSleepDeadline(tlock2, &my_event, deadline, THREAD_INTERRUPTIBLE);
  76         IOLog("IOLockSleepDeadline(&my_event, %llu) returned %d, expected 0\n", deadline, sleepret);
  77
  78         IOLockUnlock(tlock2);
  79
  80         clock_interval_to_deadline(4, NSEC_PER_SEC, &deadline);
  81         IOLog("%d sec deadline is %llu\n", 4, deadline);
  82
  83         ret = thread_call_enter_delayed(tcall2, deadline);
  84
  85         IOLockLock(tlock2);
  86
  87         clock_interval_to_deadline(3, NSEC_PER_SEC, &deadline);
  88         IOLog("%d sec deadline is %llu\n", 3, deadline);
  89         sleepret = IOLockSleepDeadline(tlock2, &my_event, deadline, THREAD_INTERRUPTIBLE);
  90         IOLog("IOLockSleepDeadline(&my_event, %llu) returned %d, expected 1\n", deadline, sleepret);
  91
  92         IOLockUnlock(tlock2);
  93
  94         return true;
  95 }
  96
  97 static void
  98 thread_call_test_func(thread_call_param_t param0,
  99     thread_call_param_t param1)
 100 {
 101         testthreadcall *self = (testthreadcall *)param0;
 102
 103         IOLog("thread_call_test_func %p %p\n", param0, param1);
 104
 105         IOSimpleLockLock(self->tlock);
 106         IOSimpleLockUnlock(self->tlock);
 107 }
 108
 109 static void
 110 thread_call_test_func2(thread_call_param_t param0,
 111     thread_call_param_t param1)
 112 {
 113         testthreadcall *self = (testthreadcall *)param0;
 114
 115         IOLog("thread_call_test_func2 %p %p\n", param0, param1);
 116
 117         IOLockWakeup(self->tlock2, &my_event, false);
 118 }