| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef MOJO_PUBLIC_CPP_BINDINGS_LIB_INTERFACE_PTR_INTERNAL_H_ | 5 #ifndef MOJO_PUBLIC_CPP_BINDINGS_LIB_INTERFACE_PTR_INTERNAL_H_ |
| 6 #define MOJO_PUBLIC_CPP_BINDINGS_LIB_INTERFACE_PTR_INTERNAL_H_ | 6 #define MOJO_PUBLIC_CPP_BINDINGS_LIB_INTERFACE_PTR_INTERNAL_H_ |
| 7 | 7 |
| 8 #include <assert.h> | 8 #include <assert.h> |
| 9 #include <stdio.h> | 9 #include <stdio.h> |
| 10 | 10 |
| 11 #include "mojo/public/cpp/bindings/lib/filter_chain.h" | 11 #include "mojo/public/cpp/bindings/lib/filter_chain.h" |
| 12 #include "mojo/public/cpp/bindings/lib/message_header_validator.h" | 12 #include "mojo/public/cpp/bindings/lib/message_header_validator.h" |
| 13 #include "mojo/public/cpp/bindings/lib/router.h" | 13 #include "mojo/public/cpp/bindings/lib/router.h" |
| 14 #include "mojo/public/cpp/environment/environment.h" | 14 |
| 15 struct MojoAsyncWaiter; |
| 15 | 16 |
| 16 namespace mojo { | 17 namespace mojo { |
| 17 namespace internal { | 18 namespace internal { |
| 18 | 19 |
| 19 template <typename Interface> | 20 template <typename Interface> |
| 20 class InterfacePtrState { | 21 class InterfacePtrState { |
| 21 public: | 22 public: |
| 22 InterfacePtrState() : proxy_(NULL), router_(NULL) {} | 23 InterfacePtrState() : proxy_(NULL), router_(NULL), waiter_(NULL) {} |
| 23 | 24 |
| 24 ~InterfacePtrState() { | 25 ~InterfacePtrState() { |
| 25 // Destruction order matters here. We delete |proxy_| first, even though | 26 // Destruction order matters here. We delete |proxy_| first, even though |
| 26 // |router_| may have a reference to it, so that |~Interface| may have a | 27 // |router_| may have a reference to it, so that |~Interface| may have a |
| 27 // shot at generating new outbound messages (ie, invoking client methods). | 28 // shot at generating new outbound messages (ie, invoking client methods). |
| 28 delete proxy_; | 29 delete proxy_; |
| 29 delete router_; | 30 delete router_; |
| 30 } | 31 } |
| 31 | 32 |
| 32 Interface* instance() const { return proxy_; } | 33 Interface* instance() { |
| 34 ConfigureProxyIfNecessary(); |
| 33 | 35 |
| 34 Router* router() const { return router_; } | 36 // This will be NULL if the object is not bound. |
| 37 return proxy_; |
| 38 } |
| 35 | 39 |
| 36 void Swap(InterfacePtrState* other) { | 40 void Swap(InterfacePtrState* other) { |
| 37 std::swap(other->proxy_, proxy_); | 41 std::swap(other->proxy_, proxy_); |
| 38 std::swap(other->router_, router_); | 42 std::swap(other->router_, router_); |
| 43 handle_.swap(other->handle_); |
| 44 std::swap(other->waiter_, waiter_); |
| 39 } | 45 } |
| 40 | 46 |
| 41 void ConfigureProxy( | 47 void Bind(ScopedMessagePipeHandle handle, const MojoAsyncWaiter* waiter) { |
| 42 ScopedMessagePipeHandle handle, | |
| 43 const MojoAsyncWaiter* waiter = Environment::GetDefaultAsyncWaiter()) { | |
| 44 assert(!proxy_); | 48 assert(!proxy_); |
| 45 assert(!router_); | 49 assert(!router_); |
| 50 assert(!handle_.is_valid()); |
| 51 assert(!waiter_); |
| 46 | 52 |
| 47 FilterChain filters; | 53 handle_ = handle.Pass(); |
| 48 filters.Append<MessageHeaderValidator>(); | 54 waiter_ = waiter; |
| 49 filters.Append<typename Interface::Client::RequestValidator_>(); | |
| 50 filters.Append<typename Interface::ResponseValidator_>(); | |
| 51 | |
| 52 router_ = new Router(handle.Pass(), filters.Pass(), waiter); | |
| 53 ProxyWithStub* proxy = new ProxyWithStub(router_); | |
| 54 router_->set_incoming_receiver(&proxy->stub); | |
| 55 | |
| 56 proxy_ = proxy; | |
| 57 } | 55 } |
| 58 | 56 |
| 59 bool WaitForIncomingMethodCall() { | 57 bool WaitForIncomingMethodCall() { |
| 58 ConfigureProxyIfNecessary(); |
| 59 |
| 60 assert(router_); | 60 assert(router_); |
| 61 return router_->WaitForIncomingMessage(); | 61 return router_->WaitForIncomingMessage(); |
| 62 } | 62 } |
| 63 | 63 |
| 64 ScopedMessagePipeHandle PassMessagePipe() { |
| 65 if (router_) |
| 66 return router_->PassMessagePipe(); |
| 67 |
| 68 waiter_ = NULL; |
| 69 return handle_.Pass(); |
| 70 } |
| 71 |
| 64 void set_client(typename Interface::Client* client) { | 72 void set_client(typename Interface::Client* client) { |
| 73 ConfigureProxyIfNecessary(); |
| 74 |
| 65 assert(proxy_); | 75 assert(proxy_); |
| 66 proxy_->stub.set_sink(client); | 76 proxy_->stub.set_sink(client); |
| 67 } | 77 } |
| 68 | 78 |
| 79 bool encountered_error() const { |
| 80 return router_ ? router_->encountered_error() : false; |
| 81 } |
| 82 |
| 83 void set_error_handler(ErrorHandler* error_handler) { |
| 84 ConfigureProxyIfNecessary(); |
| 85 |
| 86 assert(router_); |
| 87 router_->set_error_handler(error_handler); |
| 88 } |
| 89 |
| 90 Router* router_for_testing() { |
| 91 ConfigureProxyIfNecessary(); |
| 92 return router_; |
| 93 } |
| 94 |
| 69 private: | 95 private: |
| 70 class ProxyWithStub : public Interface::Proxy_ { | 96 class ProxyWithStub : public Interface::Proxy_ { |
| 71 public: | 97 public: |
| 72 explicit ProxyWithStub(MessageReceiverWithResponder* receiver) | 98 explicit ProxyWithStub(MessageReceiverWithResponder* receiver) |
| 73 : Interface::Proxy_(receiver) { | 99 : Interface::Proxy_(receiver) { |
| 74 } | 100 } |
| 75 typename Interface::Client::Stub_ stub; | 101 typename Interface::Client::Stub_ stub; |
| 76 private: | 102 private: |
| 77 MOJO_DISALLOW_COPY_AND_ASSIGN(ProxyWithStub); | 103 MOJO_DISALLOW_COPY_AND_ASSIGN(ProxyWithStub); |
| 78 }; | 104 }; |
| 79 | 105 |
| 106 void ConfigureProxyIfNecessary() { |
| 107 // The proxy has been configured. |
| 108 if (proxy_) { |
| 109 assert(router_); |
| 110 return; |
| 111 } |
| 112 // The object hasn't been bound. |
| 113 if (!waiter_) { |
| 114 assert(!handle_.is_valid()); |
| 115 return; |
| 116 } |
| 117 |
| 118 FilterChain filters; |
| 119 filters.Append<MessageHeaderValidator>(); |
| 120 filters.Append<typename Interface::Client::RequestValidator_>(); |
| 121 filters.Append<typename Interface::ResponseValidator_>(); |
| 122 |
| 123 router_ = new Router(handle_.Pass(), filters.Pass(), waiter_); |
| 124 waiter_ = NULL; |
| 125 |
| 126 ProxyWithStub* proxy = new ProxyWithStub(router_); |
| 127 router_->set_incoming_receiver(&proxy->stub); |
| 128 |
| 129 proxy_ = proxy; |
| 130 } |
| 131 |
| 80 ProxyWithStub* proxy_; | 132 ProxyWithStub* proxy_; |
| 81 Router* router_; | 133 Router* router_; |
| 82 | 134 |
| 135 // |proxy_| and |router_| are not initialized until read/write with the |
| 136 // message pipe handle is needed. Before that, |handle_| and |waiter_| store |
| 137 // the arguments of Bind(). |
| 138 ScopedMessagePipeHandle handle_; |
| 139 const MojoAsyncWaiter* waiter_; |
| 140 |
| 83 MOJO_DISALLOW_COPY_AND_ASSIGN(InterfacePtrState); | 141 MOJO_DISALLOW_COPY_AND_ASSIGN(InterfacePtrState); |
| 84 }; | 142 }; |
| 85 | 143 |
| 86 } // namespace internal | 144 } // namespace internal |
| 87 } // namespace mojo | 145 } // namespace mojo |
| 88 | 146 |
| 89 #endif // MOJO_PUBLIC_CPP_BINDINGS_LIB_INTERFACE_PTR_INTERNAL_H_ | 147 #endif // MOJO_PUBLIC_CPP_BINDINGS_LIB_INTERFACE_PTR_INTERNAL_H_ |
| OLD | NEW |