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 } 119