Gather trace data by waiting on handles with a fixed timeout

mojo/public/cpp/bindings/lib/connector.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "mojo/public/cpp/bindings/lib/connector.h"
 
 #include "mojo/public/cpp/bindings/error_handler.h"
 #include "mojo/public/cpp/environment/logging.h"
 
 namespace mojo {
@@ skipping 27 matching lines @@
 void Connector::CloseMessagePipe() {
   CancelWait();
   Close(message_pipe_.Pass());
 }
 
 ScopedMessagePipeHandle Connector::PassMessagePipe() {
   CancelWait();
   return message_pipe_.Pass();
 }
 
-bool Connector::WaitForIncomingMessage() {
+bool Connector::WaitForIncomingMessage(MojoDeadline deadline) {
   if (error_)
     return false;
 
-  MojoResult rv = Wait(message_pipe_.get(),
-                       MOJO_HANDLE_SIGNAL_READABLE,
-                       MOJO_DEADLINE_INDEFINITE,
-                       nullptr);
+  MojoResult rv =
+      Wait(message_pipe_.get(), MOJO_HANDLE_SIGNAL_READABLE, deadline, nullptr);
+  if (rv == MOJO_RESULT_SHOULD_WAIT)
+    return false;
   if (rv != MOJO_RESULT_OK) {
     NotifyError();
     return false;
   }
   mojo_ignore_result(ReadSingleMessage(&rv));
   return (rv == MOJO_RESULT_OK);
 }
 
 bool Connector::Accept(Message* message) {
   MOJO_CHECK(message_pipe_.is_valid());
@@ skipping 132 matching lines @@
 
 void Connector::NotifyError() {
   error_ = true;
   CancelWait();
   if (error_handler_)
     error_handler_->OnConnectionError();
 }
 
 }  // namespace internal
 }  // namespace mojo