summaryrefslogtreecommitdiffstats
path: root/mojo/public/cpp/bindings/lib/interface_impl_internal.h
blob: 79b598da9343c46749e995c8ee5dc8b92ce9a627 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
// Copyright 2014 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.

#ifndef MOJO_PUBLIC_CPP_BINDINGS_LIB_INTERFACE_IMPL_INTERNAL_H_
#define MOJO_PUBLIC_CPP_BINDINGS_LIB_INTERFACE_IMPL_INTERNAL_H_

#include "mojo/public/cpp/bindings/error_handler.h"
#include "mojo/public/cpp/bindings/interface_ptr.h"
#include "mojo/public/cpp/bindings/lib/filter_chain.h"
#include "mojo/public/cpp/bindings/lib/message_header_validator.h"
#include "mojo/public/cpp/system/macros.h"

namespace mojo {
namespace internal {

template <typename Interface>
class InterfaceImplBase : public Interface {
 public:
  virtual ~InterfaceImplBase() {}
  virtual void OnConnectionEstablished() = 0;
  virtual void OnConnectionError() = 0;
};

template <typename Interface>
class InterfaceImplState : public ErrorHandler {
 public:
  typedef typename Interface::Client Client;

  explicit InterfaceImplState(InterfaceImplBase<Interface>* instance)
      : router_(NULL),
        proxy_(NULL) {
    assert(instance);
    stub_.set_sink(instance);
  }

  virtual ~InterfaceImplState() {
    delete proxy_;
    if (router_) {
      router_->set_error_handler(NULL);
      delete router_;
    }
  }

  void BindProxy(
      InterfacePtr<Interface>* ptr,
      const MojoAsyncWaiter* waiter = GetDefaultAsyncWaiter()) {
    MessagePipe pipe;
    ptr->Bind(pipe.handle0.Pass(), waiter);
    Bind(pipe.handle1.Pass(), waiter);
  }

  void Bind(ScopedMessagePipeHandle handle,
            const MojoAsyncWaiter* waiter) {
    assert(!router_);

    FilterChain filters;
    filters.Append<MessageHeaderValidator>();
    filters.Append<typename Interface::RequestValidator_>();
    filters.Append<typename Interface::Client::ResponseValidator_>();

    router_ = new Router(handle.Pass(), filters.Pass(), waiter);
    router_->set_incoming_receiver(&stub_);
    router_->set_error_handler(this);

    proxy_ = new typename Client::Proxy_(router_);

    instance()->OnConnectionEstablished();
  }

  Router* router() { return router_; }
  Client* client() { return proxy_; }

 private:
  InterfaceImplBase<Interface>* instance() {
    return static_cast<InterfaceImplBase<Interface>*>(stub_.sink());
  }

  virtual void OnConnectionError() MOJO_OVERRIDE {
    instance()->OnConnectionError();
  }

  Router* router_;
  typename Client::Proxy_* proxy_;
  typename Interface::Stub_ stub_;

  MOJO_DISALLOW_COPY_AND_ASSIGN(InterfaceImplState);
};

}  // namespace internal
}  // namespace mojo

#endif  // MOJO_PUBLIC_CPP_BINDINGS_LIB_INTERFACE_IMPL_INTERNAL_H_