KernelTests: Update tests for dataqueue usage changes

google · May 18, 2015 · d5c1d66 · d5c1d66
1 parent 536b896
commit d5c1d66
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Tests/KernelTests/main.m b/Tests/KernelTests/main.m
@@ -193,7 +193,7 @@ - (void)beginListening {
 
   /// Begin listening for events
   queueMemory = (IODataQueueMemory *)address;
-  while (IODataQueueWaitForAvailableData(queueMemory, receivePort) == kIOReturnSuccess) {
+  do {
     while (IODataQueueDataAvailable(queueMemory)) {
       dataSize = sizeof(vdata);
       kr = IODataQueueDequeue(queueMemory, &vdata, &dataSize);
@@ -227,7 +227,7 @@ - (void)beginListening {
         TFAILINFO("Error receiving data: %d", kr);
       }
     }
-  }
+  }  while (IODataQueueWaitForAvailableData(queueMemory, receivePort) == kIOReturnSuccess);
 
   IOConnectUnmapMemory(self.connection, kIODefaultMemoryType, mach_task_self(), address);
   mach_port_destroy(mach_task_self(), receivePort);