OLD | NEW |
| (Empty) |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #ifndef MOJO_PUBLIC_CPP_BINDINGS_LIB_ROUTER_H_ | |
6 #define MOJO_PUBLIC_CPP_BINDINGS_LIB_ROUTER_H_ | |
7 | |
8 #include <map> | |
9 | |
10 #include "mojo/public/cpp/bindings/lib/connector.h" | |
11 #include "mojo/public/cpp/bindings/lib/filter_chain.h" | |
12 #include "mojo/public/cpp/bindings/lib/shared_data.h" | |
13 #include "mojo/public/cpp/environment/environment.h" | |
14 | |
15 namespace mojo { | |
16 namespace internal { | |
17 | |
18 class Router : public MessageReceiverWithResponder { | |
19 public: | |
20 Router(ScopedMessagePipeHandle message_pipe, | |
21 FilterChain filters, | |
22 const MojoAsyncWaiter* waiter = Environment::GetDefaultAsyncWaiter()); | |
23 ~Router() override; | |
24 | |
25 // Sets the receiver to handle messages read from the message pipe that do | |
26 // not have the kMessageIsResponse flag set. | |
27 void set_incoming_receiver(MessageReceiverWithResponder* receiver) { | |
28 incoming_receiver_ = receiver; | |
29 } | |
30 | |
31 // Sets the error handler to receive notifications when an error is | |
32 // encountered while reading from the pipe or waiting to read from the pipe. | |
33 void set_error_handler(ErrorHandler* error_handler) { | |
34 connector_.set_error_handler(error_handler); | |
35 } | |
36 | |
37 // Returns true if an error was encountered while reading from the pipe or | |
38 // waiting to read from the pipe. | |
39 bool encountered_error() const { return connector_.encountered_error(); } | |
40 | |
41 void CloseMessagePipe() { connector_.CloseMessagePipe(); } | |
42 | |
43 ScopedMessagePipeHandle PassMessagePipe() { | |
44 return connector_.PassMessagePipe(); | |
45 } | |
46 | |
47 // MessageReceiver implementation: | |
48 bool Accept(Message* message) override; | |
49 bool AcceptWithResponder(Message* message, | |
50 MessageReceiver* responder) override; | |
51 | |
52 // Blocks the current thread for the first incoming method call, i.e., either | |
53 // a call to a client method or a callback method. | |
54 bool WaitForIncomingMessage() { return connector_.WaitForIncomingMessage(); } | |
55 | |
56 // Sets this object to testing mode. | |
57 // In testing mode: | |
58 // - the object is more tolerant of unrecognized response messages; | |
59 // - the connector continues working after seeing errors from its incoming | |
60 // receiver. | |
61 void EnableTestingMode(); | |
62 | |
63 private: | |
64 typedef std::map<uint64_t, MessageReceiver*> ResponderMap; | |
65 | |
66 class HandleIncomingMessageThunk : public MessageReceiver { | |
67 public: | |
68 HandleIncomingMessageThunk(Router* router); | |
69 ~HandleIncomingMessageThunk() override; | |
70 | |
71 // MessageReceiver implementation: | |
72 bool Accept(Message* message) override; | |
73 | |
74 private: | |
75 Router* router_; | |
76 }; | |
77 | |
78 bool HandleIncomingMessage(Message* message); | |
79 | |
80 HandleIncomingMessageThunk thunk_; | |
81 FilterChain filters_; | |
82 Connector connector_; | |
83 SharedData<Router*> weak_self_; | |
84 MessageReceiverWithResponder* incoming_receiver_; | |
85 ResponderMap responders_; | |
86 uint64_t next_request_id_; | |
87 bool testing_mode_; | |
88 }; | |
89 | |
90 } // namespace internal | |
91 } // namespace mojo | |
92 | |
93 #endif // MOJO_PUBLIC_CPP_BINDINGS_LIB_ROUTER_H_ | |
OLD | NEW |