capability.h 33.4 KB
Newer Older
Kenton Varda's avatar
Kenton Varda committed
1 2
// Copyright (c) 2013-2014 Sandstorm Development Group, Inc. and contributors
// Licensed under the MIT License:
3
//
Kenton Varda's avatar
Kenton Varda committed
4 5 6 7 8 9
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
10
//
Kenton Varda's avatar
Kenton Varda committed
11 12
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
13
//
Kenton Varda's avatar
Kenton Varda committed
14 15 16 17 18 19 20
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.
21 22 23 24

#ifndef CAPNP_CAPABILITY_H_
#define CAPNP_CAPABILITY_H_

25 26 27 28
#if defined(__GNUC__) && !CAPNP_HEADER_WARNINGS
#pragma GCC system_header
#endif

29 30 31 32
#if CAPNP_LITE
#error "RPC APIs, including this header, are not available in lite mode."
#endif

33
#include <kj/async.h>
34
#include "any.h"
35
#include "pointer-helpers.h"
36 37 38

namespace capnp {

39
template <typename Results>
40 41
class Response;

42
template <typename T>
43 44 45 46 47 48 49 50
class RemotePromise: public kj::Promise<Response<T>>, public T::Pipeline {
  // A Promise which supports pipelined calls.  T is typically a struct type.  T must declare
  // an inner "mix-in" type "Pipeline" which implements pipelining; RemotePromise simply
  // multiply-inherits that type along with Promise<Response<T>>.  T::Pipeline must be movable,
  // but does not need to be copyable (i.e. just like Promise<T>).
  //
  // The promise is for an owned pointer so that the RPC system can allocate the MessageReader
  // itself.
51 52

public:
53 54
  inline RemotePromise(kj::Promise<Response<T>>&& promise, typename T::Pipeline&& pipeline)
      : kj::Promise<Response<T>>(kj::mv(promise)),
55 56
        T::Pipeline(kj::mv(pipeline)) {}
  inline RemotePromise(decltype(nullptr))
57 58
      : kj::Promise<Response<T>>(nullptr),
        T::Pipeline(nullptr) {}
59 60 61 62 63
  KJ_DISALLOW_COPY(RemotePromise);
  RemotePromise(RemotePromise&& other) = default;
  RemotePromise& operator=(RemotePromise&& other) = default;
};

64
class LocalClient;
65 66
namespace _ { // private
struct RawSchema;
67
struct RawBrandedSchema;
68
extern const RawSchema NULL_INTERFACE_SCHEMA;  // defined in schema.c++
69
class CapabilityServerSetBase;
70 71 72 73 74 75 76 77 78 79
}  // namespace _ (private)

struct Capability {
  // A capability without type-safe methods.  Typed capability clients wrap `Client` and typed
  // capability servers subclass `Server` to dispatch to the regular, typed methods.

  class Client;
  class Server;

  struct _capnpPrivate {
80
    struct IsInterface;
81 82 83
    static constexpr uint64_t typeId = 0x3;
    static constexpr Kind kind = Kind::INTERFACE;
    static constexpr _::RawSchema const* schema = &_::NULL_INTERFACE_SCHEMA;
84 85 86 87 88 89

    static const _::RawBrandedSchema* const brand;
    // Can't quite declare this one inline without including generated-header-support.h. Avoiding
    // for now by declaring out-of-line.
    // TODO(cleanup): Split RawSchema stuff into its own header that can be included here, or
    //   something.
90 91 92
  };
};

93 94
// =======================================================================================

95 96 97 98
class RequestHook;
class ResponseHook;
class PipelineHook;
class ClientHook;
99

100
template <typename Params, typename Results>
101 102 103 104 105 106 107
class Request: public Params::Builder {
  // A call that hasn't been sent yet.  This class extends a Builder for the call's "Params"
  // structure with a method send() that actually sends it.
  //
  // Given a Cap'n Proto method `foo(a :A, b :B): C`, the generated client interface will have
  // a method `Request<FooParams, C> startFoo()` (as well as a convenience method
  // `RemotePromise<C> foo(A::Reader a, B::Reader b)`).
108 109

public:
110 111
  inline Request(typename Params::Builder builder, kj::Own<RequestHook>&& hook)
      : Params::Builder(builder), hook(kj::mv(hook)) {}
112

113 114
  RemotePromise<Results> send();
  // Send the call and return a promise for the results.
115

116 117
private:
  kj::Own<RequestHook> hook;
118 119

  friend class Capability::Client;
120
  friend struct DynamicCapability;
121 122
  template <typename, typename>
  friend class CallContext;
123
  friend class RequestHook;
124 125
};

126 127
template <typename Results>
class Response: public Results::Reader {
128 129 130 131
  // A completed call.  This class extends a Reader for the call's answer structure.  The Response
  // is move-only -- once it goes out-of-scope, the underlying message will be freed.

public:
132
  inline Response(typename Results::Reader reader, kj::Own<ResponseHook>&& hook)
133
      : Results::Reader(reader), hook(kj::mv(hook)) {}
134 135

private:
136
  kj::Own<ResponseHook> hook;
137 138 139 140 141 142 143 144 145

  template <typename, typename>
  friend class Request;
};

class Capability::Client {
  // Base type for capability clients.

public:
146 147 148
  typedef Capability Reads;
  typedef Capability Calls;

149
  Client(decltype(nullptr));
150 151 152 153
  // If you need to declare a Client before you have anything to assign to it (perhaps because
  // the assignment is going to occur in an if/else scope), you can start by initializing it to
  // `nullptr`.  The resulting client is not meant to be called and throws exceptions from all
  // methods.
154

155
  template <typename T, typename = kj::EnableIf<kj::canConvert<T*, Capability::Server*>()>>
156
  Client(kj::Own<T>&& server);
157 158 159
  // Make a client capability that wraps the given server capability.  The server's methods will
  // only be executed in the given EventLoop, regardless of what thread calls the client's methods.

160
  template <typename T, typename = kj::EnableIf<kj::canConvert<T*, Client*>()>>
161
  Client(kj::Promise<T>&& promise);
162 163 164 165 166 167
  // Make a client from a promise for a future client.  The resulting client queues calls until the
  // promise resolves.

  Client(kj::Exception&& exception);
  // Make a broken client that throws the given exception from all calls.

168 169 170 171
  Client(Client& other);
  Client& operator=(Client& other);
  // Copies by reference counting.  Warning:  This refcounting is not thread-safe.  All copies of
  // the client must remain in one thread.
172 173 174

  Client(Client&&) = default;
  Client& operator=(Client&&) = default;
175
  // Move constructor avoids reference counting.
176

177
  explicit Client(kj::Own<ClientHook>&& hook);
178 179
  // For use by the RPC implementation:  Wrap a ClientHook.

180
  template <typename T>
181
  typename T::Client castAs();
182 183 184 185 186
  // Reinterpret the capability as implementing the given interface.  Note that no error will occur
  // here if the capability does not actually implement this interface, but later method calls will
  // fail.  It's up to the application to decide how indicate that additional interfaces are
  // supported.
  //
Kenton Varda's avatar
Kenton Varda committed
187
  // TODO(perf):  GCC 4.8 / Clang 3.3:  rvalue-qualified version for better performance.
188 189

  template <typename T>
190
  typename T::Client castAs(InterfaceSchema schema);
191 192
  // Dynamic version.  `T` must be `DynamicCapability`, and you must `#include <capnp/dynamic.h>`.

193
  kj::Promise<void> whenResolved();
194 195 196 197 198 199
  // If the capability is actually only a promise, the returned promise resolves once the
  // capability itself has resolved to its final destination (or propagates the exception if
  // the capability promise is rejected).  This is mainly useful for error-checking in the case
  // where no calls are being made.  There is no reason to wait for this before making calls; if
  // the capability does not resolve, the call results will propagate the error.

200 201 202 203 204 205
  Request<AnyPointer, AnyPointer> typelessRequest(
      uint64_t interfaceId, uint16_t methodId,
      kj::Maybe<MessageSize> sizeHint);
  // Make a request without knowing the types of the params or results. You specify the type ID
  // and method number manually.

206
  // TODO(someday):  method(s) for Join
207

208 209 210 211 212
protected:
  Client() = default;

  template <typename Params, typename Results>
  Request<Params, Results> newCall(uint64_t interfaceId, uint16_t methodId,
213
                                   kj::Maybe<MessageSize> sizeHint);
214

215
private:
216
  kj::Own<ClientHook> hook;
217

218
  static kj::Own<ClientHook> makeLocalClient(kj::Own<Capability::Server>&& server);
219

Kenton Varda's avatar
Kenton Varda committed
220 221
  template <typename, Kind>
  friend struct _::PointerHelpers;
222 223 224 225 226 227
  friend struct DynamicCapability;
  friend class Orphanage;
  friend struct DynamicStruct;
  friend struct DynamicList;
  template <typename, Kind>
  friend struct List;
228
  friend class _::CapabilityServerSetBase;
229
  friend class ClientHook;
230 231 232
};

// =======================================================================================
233 234 235 236
// Local capabilities

class CallContextHook;

237
template <typename Params, typename Results>
238
class CallContext: public kj::DisallowConstCopy {
239
  // Wrapper around CallContextHook with a specific return type.
240 241 242
  //
  // Methods of this class may only be called from within the server's event loop, not from other
  // threads.
Kenton Varda's avatar
Kenton Varda committed
243 244
  //
  // The CallContext becomes invalid as soon as the call reports completion.
245 246

public:
247 248 249 250 251 252 253 254 255 256 257
  explicit CallContext(CallContextHook& hook);

  typename Params::Reader getParams();
  // Get the params payload.

  void releaseParams();
  // Release the params payload.  getParams() will throw an exception after this is called.
  // Releasing the params may allow the RPC system to free up buffer space to handle other
  // requests.  Long-running asynchronous methods should try to call this as early as is
  // convenient.

258 259
  typename Results::Builder getResults(kj::Maybe<MessageSize> sizeHint = nullptr);
  typename Results::Builder initResults(kj::Maybe<MessageSize> sizeHint = nullptr);
260 261
  void setResults(typename Results::Reader value);
  void adoptResults(Orphan<Results>&& value);
262
  Orphanage getResultsOrphanage(kj::Maybe<MessageSize> sizeHint = nullptr);
263 264
  // Manipulate the results payload.  The "Return" message (part of the RPC protocol) will
  // typically be allocated the first time one of these is called.  Some RPC systems may
265 266 267
  // allocate these messages in a limited space (such as a shared memory segment), therefore the
  // application should delay calling these as long as is convenient to do so (but don't delay
  // if doing so would require extra copies later).
Kenton Varda's avatar
Kenton Varda committed
268
  //
269 270 271 272 273
  // `sizeHint` indicates a guess at the message size.  This will usually be used to decide how
  // much space to allocate for the first message segment (don't worry: only space that is actually
  // used will be sent on the wire).  If omitted, the system decides.  The message root pointer
  // should not be included in the size.  So, if you are simply going to copy some existing message
  // directly into the results, just call `.totalSize()` and pass that in.
274

275 276 277 278 279 280 281 282 283 284 285 286 287
  template <typename SubParams>
  kj::Promise<void> tailCall(Request<SubParams, Results>&& tailRequest);
  // Resolve the call by making a tail call.  `tailRequest` is a request that has been filled in
  // but not yet sent.  The context will send the call, then fill in the results with the result
  // of the call.  If tailCall() is used, {get,init,set,adopt}Results (above) *must not* be called.
  //
  // The RPC implementation may be able to optimize a tail call to another machine such that the
  // results never actually pass through this machine.  Even if no such optimization is possible,
  // `tailCall()` may allow pipelined calls to be forwarded optimistically to the new call site.
  //
  // In general, this should be the last thing a method implementation calls, and the promise
  // returned from `tailCall()` should then be returned by the method implementation.

288
  void allowCancellation();
289 290 291 292 293 294 295 296 297 298
  // Indicate that it is OK for the RPC system to discard its Promise for this call's result if
  // the caller cancels the call, thereby transitively canceling any asynchronous operations the
  // call implementation was performing.  This is not done by default because it could represent a
  // security risk:  applications must be carefully written to ensure that they do not end up in
  // a bad state if an operation is canceled at an arbitrary point.  However, for long-running
  // method calls that hold significant resources, prompt cancellation is often useful.
  //
  // Keep in mind that asynchronous cancellation cannot occur while the method is synchronously
  // executing on a local thread.  The method must perform an asynchronous operation or call
  // `EventLoop::current().runLater()` to yield control.
Kenton Varda's avatar
Kenton Varda committed
299
  //
300 301 302 303 304 305 306 307 308
  // Note:  You might think that we should offer `onCancel()` and/or `isCanceled()` methods that
  // provide notification when the caller cancels the request without forcefully killing off the
  // promise chain.  Unfortunately, this composes poorly with promise forking:  the canceled
  // path may be just one branch of a fork of the result promise.  The other branches still want
  // the call to continue.  Promise forking is used within the Cap'n Proto implementation -- in
  // particular each pipelined call forks the result promise.  So, if a caller made a pipelined
  // call and then dropped the original object, the call should not be canceled, but it would be
  // excessively complicated for the framework to avoid notififying of cancellation as long as
  // pipelined calls still exist.
309

310
private:
311 312 313
  CallContextHook* hook;

  friend class Capability::Server;
314
  friend struct DynamicCapability;
315 316
};

317 318 319 320
class Capability::Server {
  // Objects implementing a Cap'n Proto interface must subclass this.  Typically, such objects
  // will instead subclass a typed Server interface which will take care of implementing
  // dispatchCall().
321 322

public:
323 324
  typedef Capability Serves;

325
  virtual kj::Promise<void> dispatchCall(uint64_t interfaceId, uint16_t methodId,
326
                                         CallContext<AnyPointer, AnyPointer> context) = 0;
327 328 329 330
  // Call the given method.  `params` is the input struct, and should be released as soon as it
  // is no longer needed.  `context` may be used to allocate the output struct and deal with
  // cancellation.

331
  // TODO(someday):  Method which can optionally be overridden to implement Join when the object is
332
  //   a proxy.
333 334

protected:
335 336 337 338 339 340 341 342 343 344 345
  inline Capability::Client thisCap();
  // Get a capability pointing to this object, much like the `this` keyword.
  //
  // The effect of this method is undefined if:
  // - No capability client has been created pointing to this object. (This is always the case in
  //   the server's constructor.)
  // - The capability client pointing at this object has been destoryed. (This is always the case
  //   in the server's destructor.)
  // - Multiple capability clients have been created around the same server (possible if the server
  //   is refcounted, which is not recommended since the client itself provides refcounting).

346 347
  template <typename Params, typename Results>
  CallContext<Params, Results> internalGetTypedContext(
348
      CallContext<AnyPointer, AnyPointer> typeless);
349 350 351 352 353 354
  kj::Promise<void> internalUnimplemented(const char* actualInterfaceName,
                                          uint64_t requestedTypeId);
  kj::Promise<void> internalUnimplemented(const char* interfaceName,
                                          uint64_t typeId, uint16_t methodId);
  kj::Promise<void> internalUnimplemented(const char* interfaceName, const char* methodName,
                                          uint64_t typeId, uint16_t methodId);
355 356 357 358

private:
  ClientHook* thisHook = nullptr;
  friend class LocalClient;
359 360
};

361 362 363 364 365 366 367 368 369 370 371 372 373 374
// =======================================================================================

namespace _ {  // private

class CapabilityServerSetBase {
public:
  Capability::Client addInternal(kj::Own<Capability::Server>&& server, void* ptr);
  kj::Promise<void*> getLocalServerInternal(Capability::Client& client);
};

}  // namespace _ (private)

template <typename T>
class CapabilityServerSet: private _::CapabilityServerSetBase {
375 376
  // Allows a server to recognize its own capabilities when passed back to it, and obtain the
  // underlying Server objects associated with them.
377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401
  //
  // All objects in the set must have the same interface type T. The objects may implement various
  // interfaces derived from T (and in fact T can be `capnp::Capability` to accept all objects),
  // but note that if you compile with RTTI disabled then you will not be able to down-cast through
  // virtual inheritance, and all inheritance between server interfaces is virtual. So, with RTTI
  // disabled, you will likely need to set T to be the most-derived Cap'n Proto interface type,
  // and you server class will need to be directly derived from that, so that you can use
  // static_cast (or kj::downcast) to cast to it after calling getLocalServer(). (If you compile
  // with RTTI, then you can freely dynamic_cast and ignore this issue!)

public:
  CapabilityServerSet() = default;
  KJ_DISALLOW_COPY(CapabilityServerSet);

  typename T::Client add(kj::Own<typename T::Server>&& server);
  // Create a new capability Client for the given Server and also add this server to the set.

  kj::Promise<kj::Maybe<typename T::Server&>> getLocalServer(typename T::Client& client);
  // Given a Client pointing to a server previously passed to add(), return the corresponding
  // Server. This returns a promise because if the input client is itself a promise, this must
  // wait for it to resolve. Keep in mind that the server will be deleted when all clients are
  // gone, so the caller should make sure to keep the client alive (hence why this method only
  // accepts an lvalue input).
};

402
// =======================================================================================
403 404
// Hook interfaces which must be implemented by the RPC system.  Applications never call these
// directly; the RPC system implements them and the types defined earlier in this file wrap them.
405

406 407
class RequestHook {
  // Hook interface implemented by RPC system representing a request being built.
408

409
public:
410
  virtual RemotePromise<AnyPointer> send() = 0;
411
  // Send the call and return a promise for the result.
412

413
  virtual const void* getBrand() = 0;
414 415 416
  // Returns a void* that identifies who made this request.  This can be used by an RPC adapter to
  // discover when tail call is going to be sent over its own connection and therefore can be
  // optimized into a remote tail call.
417 418 419 420 421

  template <typename T, typename U>
  inline static kj::Own<RequestHook> from(Request<T, U>&& request) {
    return kj::mv(request.hook);
  }
422 423
};

424 425 426 427
class ResponseHook {
  // Hook interface implemented by RPC system representing a response.
  //
  // At present this class has no methods.  It exists only for garbage collection -- when the
428
  // ResponseHook is destroyed, the results can be freed.
429 430

public:
Kenton Varda's avatar
Kenton Varda committed
431 432
  virtual ~ResponseHook() noexcept(false);
  // Just here to make sure the type is dynamic.
433 434
};

435
// class PipelineHook is declared in any.h because it is needed there.
436

437
class ClientHook {
438
public:
439 440
  ClientHook();

441
  virtual Request<AnyPointer, AnyPointer> newCall(
442
      uint64_t interfaceId, uint16_t methodId, kj::Maybe<MessageSize> sizeHint) = 0;
Kenton Varda's avatar
Kenton Varda committed
443 444
  // Start a new call, allowing the client to allocate request/response objects as it sees fit.
  // This version is used when calls are made from application code in the local process.
445

Kenton Varda's avatar
Kenton Varda committed
446 447
  struct VoidPromiseAndPipeline {
    kj::Promise<void> promise;
448
    kj::Own<PipelineHook> pipeline;
Kenton Varda's avatar
Kenton Varda committed
449 450 451
  };

  virtual VoidPromiseAndPipeline call(uint64_t interfaceId, uint16_t methodId,
452
                                      kj::Own<CallContextHook>&& context) = 0;
Kenton Varda's avatar
Kenton Varda committed
453
  // Call the object, but the caller controls allocation of the request/response objects.  If the
454
  // callee insists on allocating these objects itself, it must make a copy.  This version is used
Kenton Varda's avatar
Kenton Varda committed
455
  // when calls come in over the network via an RPC system.  During the call, the context object
456 457 458 459
  // may be used from any thread so long as it is only used from one thread at a time.  Note that
  // even if the returned `Promise<void>` is discarded, the call may continue executing if any
  // pipelined calls are waiting for it; the call is only truly done when the CallContextHook is
  // destroyed.
Kenton Varda's avatar
Kenton Varda committed
460
  //
461 462
  // Since the caller of this method chooses the CallContext implementation, it is the caller's
  // responsibility to ensure that the returned promise is not canceled unless allowed via
463
  // the context's `allowCancellation()`.
464
  //
Kenton Varda's avatar
Kenton Varda committed
465 466
  // The call must not begin synchronously, as the caller may hold arbitrary mutexes.

467
  virtual kj::Maybe<ClientHook&> getResolved() = 0;
468 469 470 471 472
  // If this ClientHook is a promise that has already resolved, returns the inner, resolved version
  // of the capability.  The caller may permanently replace this client with the resolved one if
  // desired.  Returns null if the client isn't a promise or hasn't resolved yet -- use
  // `whenMoreResolved()` to distinguish between them.

473
  virtual kj::Maybe<kj::Promise<kj::Own<ClientHook>>> whenMoreResolved() = 0;
Kenton Varda's avatar
Kenton Varda committed
474 475
  // If this client is a settled reference (not a promise), return nullptr.  Otherwise, return a
  // promise that eventually resolves to a new client that is closer to being the final, settled
476 477
  // client (i.e. the value eventually returned by `getResolved()`).  Calling this repeatedly
  // should eventually produce a settled client.
Kenton Varda's avatar
Kenton Varda committed
478

479
  kj::Promise<void> whenResolved();
480 481
  // Repeatedly calls whenMoreResolved() until it returns nullptr.

482
  virtual kj::Own<ClientHook> addRef() = 0;
483
  // Return a new reference to the same capability.
484

485
  virtual const void* getBrand() = 0;
486 487 488
  // Returns a void* that identifies who made this client.  This can be used by an RPC adapter to
  // discover when a capability it needs to marshal is one that it created in the first place, and
  // therefore it can transfer the capability without proxying.
489 490 491 492 493

  virtual void* getLocalServer(_::CapabilityServerSetBase& capServerSet);
  // If this is a local capability created through `capServerSet`, return the underlying Server.
  // Otherwise, return nullptr. Default implementation (which everyone except LocalClient should
  // use) always returns nullptr.
494 495

  static kj::Own<ClientHook> from(Capability::Client client) { return kj::mv(client.hook); }
496 497
};

498 499 500
class CallContextHook {
  // Hook interface implemented by RPC system to manage a call on the server side.  See
  // CallContext<T>.
501 502

public:
503
  virtual AnyPointer::Reader getParams() = 0;
504
  virtual void releaseParams() = 0;
505
  virtual AnyPointer::Builder getResults(kj::Maybe<MessageSize> sizeHint) = 0;
506
  virtual kj::Promise<void> tailCall(kj::Own<RequestHook>&& request) = 0;
507
  virtual void allowCancellation() = 0;
508

509
  virtual kj::Promise<AnyPointer::Pipeline> onTailCall() = 0;
510 511 512
  // If `tailCall()` is called, resolves to the PipelineHook from the tail call.  An
  // implementation of `ClientHook::call()` is allowed to call this at most once.

513 514 515 516 517
  virtual ClientHook::VoidPromiseAndPipeline directTailCall(kj::Own<RequestHook>&& request) = 0;
  // Call this when you would otherwise call onTailCall() immediately followed by tailCall().
  // Implementations of tailCall() should typically call directTailCall() and then fulfill the
  // promise fulfiller for onTailCall() with the returned pipeline.

518
  virtual kj::Own<CallContextHook> addRef() = 0;
519
};
520

521
kj::Own<ClientHook> newLocalPromiseClient(kj::Promise<kj::Own<ClientHook>>&& promise);
Kenton Varda's avatar
Kenton Varda committed
522 523 524 525
// Returns a ClientHook that queues up calls until `promise` resolves, then forwards them to
// the new client.  This hook's `getResolved()` and `whenMoreResolved()` methods will reflect the
// redirection to the eventual replacement client.

526 527 528 529 530 531 532
kj::Own<ClientHook> newBrokenCap(kj::StringPtr reason);
kj::Own<ClientHook> newBrokenCap(kj::Exception&& reason);
// Helper function that creates a capability which simply throws exceptions when called.

kj::Own<PipelineHook> newBrokenPipeline(kj::Exception&& reason);
// Helper function that creates a pipeline which simply throws exceptions when called.

533 534 535 536
Request<AnyPointer, AnyPointer> newBrokenRequest(
    kj::Exception&& reason, kj::Maybe<MessageSize> sizeHint);
// Helper function that creates a Request object that simply throws exceptions when sent.

537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552
// =======================================================================================
// Extend PointerHelpers for interfaces

namespace _ {  // private

template <typename T>
struct PointerHelpers<T, Kind::INTERFACE> {
  static inline typename T::Client get(PointerReader reader) {
    return typename T::Client(reader.getCapability());
  }
  static inline typename T::Client get(PointerBuilder builder) {
    return typename T::Client(builder.getCapability());
  }
  static inline void set(PointerBuilder builder, typename T::Client&& value) {
    builder.setCapability(kj::mv(value.Capability::Client::hook));
  }
553
  static inline void set(PointerBuilder builder, typename T::Client& value) {
554 555
    builder.setCapability(value.Capability::Client::hook->addRef());
  }
556 557 558 559 560 561 562 563 564 565
  static inline void adopt(PointerBuilder builder, Orphan<T>&& value) {
    builder.adopt(kj::mv(value.builder));
  }
  static inline Orphan<T> disown(PointerBuilder builder) {
    return Orphan<T>(builder.disown());
  }
};

}  // namespace _ (private)

566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642
// =======================================================================================
// Extend List for interfaces

template <typename T>
struct List<T, Kind::INTERFACE> {
  List() = delete;

  class Reader {
  public:
    typedef List<T> Reads;

    Reader() = default;
    inline explicit Reader(_::ListReader reader): reader(reader) {}

    inline uint size() const { return reader.size() / ELEMENTS; }
    inline typename T::Client operator[](uint index) const {
      KJ_IREQUIRE(index < size());
      return typename T::Client(reader.getPointerElement(index * ELEMENTS).getCapability());
    }

    typedef _::IndexingIterator<const Reader, typename T::Client> Iterator;
    inline Iterator begin() const { return Iterator(this, 0); }
    inline Iterator end() const { return Iterator(this, size()); }

  private:
    _::ListReader reader;
    template <typename U, Kind K>
    friend struct _::PointerHelpers;
    template <typename U, Kind K>
    friend struct List;
    friend class Orphanage;
    template <typename U, Kind K>
    friend struct ToDynamic_;
  };

  class Builder {
  public:
    typedef List<T> Builds;

    Builder() = delete;
    inline Builder(decltype(nullptr)) {}
    inline explicit Builder(_::ListBuilder builder): builder(builder) {}

    inline operator Reader() { return Reader(builder.asReader()); }
    inline Reader asReader() { return Reader(builder.asReader()); }

    inline uint size() const { return builder.size() / ELEMENTS; }
    inline typename T::Client operator[](uint index) {
      KJ_IREQUIRE(index < size());
      return typename T::Client(builder.getPointerElement(index * ELEMENTS).getCapability());
    }
    inline void set(uint index, typename T::Client value) {
      KJ_IREQUIRE(index < size());
      builder.getPointerElement(index * ELEMENTS).setCapability(kj::mv(value.hook));
    }
    inline void adopt(uint index, Orphan<T>&& value) {
      KJ_IREQUIRE(index < size());
      builder.getPointerElement(index * ELEMENTS).adopt(kj::mv(value));
    }
    inline Orphan<T> disown(uint index) {
      KJ_IREQUIRE(index < size());
      return Orphan<T>(builder.getPointerElement(index * ELEMENTS).disown());
    }

    typedef _::IndexingIterator<Builder, typename T::Client> Iterator;
    inline Iterator begin() { return Iterator(this, 0); }
    inline Iterator end() { return Iterator(this, size()); }

  private:
    _::ListBuilder builder;
    friend class Orphanage;
    template <typename U, Kind K>
    friend struct ToDynamic_;
  };

private:
  inline static _::ListBuilder initPointer(_::PointerBuilder builder, uint size) {
643
    return builder.initList(ElementSize::POINTER, size * ELEMENTS);
644 645
  }
  inline static _::ListBuilder getFromPointer(_::PointerBuilder builder, const word* defaultValue) {
646
    return builder.getList(ElementSize::POINTER, defaultValue);
647 648 649
  }
  inline static _::ListReader getFromPointer(
      const _::PointerReader& reader, const word* defaultValue) {
650
    return reader.getList(ElementSize::POINTER, defaultValue);
651 652 653 654 655 656 657 658
  }

  template <typename U, Kind k>
  friend struct List;
  template <typename U, Kind K>
  friend struct _::PointerHelpers;
};

659 660 661
// =======================================================================================
// Inline implementation details

662 663
template <typename Params, typename Results>
RemotePromise<Results> Request<Params, Results>::send() {
664
  auto typelessPromise = hook->send();
665
  hook = nullptr;  // prevent reuse
666 667 668 669

  // Convert the Promise to return the correct response type.
  // Explicitly upcast to kj::Promise to make clear that calling .then() doesn't invalidate the
  // Pipeline part of the RemotePromise.
670 671
  auto typedPromise = kj::implicitCast<kj::Promise<Response<AnyPointer>>&>(typelessPromise)
      .then([](Response<AnyPointer>&& response) -> Response<Results> {
672
        return Response<Results>(response.getAs<Results>(), kj::mv(response.hook));
673 674 675
      });

  // Wrap the typeless pipeline in a typed wrapper.
676
  typename Results::Pipeline typedPipeline(
677
      kj::mv(kj::implicitCast<AnyPointer::Pipeline&>(typelessPromise)));
678

679
  return RemotePromise<Results>(kj::mv(typedPromise), kj::mv(typedPipeline));
680 681
}

682
inline Capability::Client::Client(kj::Own<ClientHook>&& hook): hook(kj::mv(hook)) {}
683
template <typename T, typename>
684 685
inline Capability::Client::Client(kj::Own<T>&& server)
    : hook(makeLocalClient(kj::mv(server))) {}
686
template <typename T, typename>
687 688 689 690
inline Capability::Client::Client(kj::Promise<T>&& promise)
    : hook(newLocalPromiseClient(promise.then([](T&& t) { return kj::mv(t.hook); }))) {}
inline Capability::Client::Client(Client& other): hook(other.hook->addRef()) {}
inline Capability::Client& Capability::Client::operator=(Client& other) {
691 692 693
  hook = other.hook->addRef();
  return *this;
}
694
template <typename T>
695
inline typename T::Client Capability::Client::castAs() {
696 697
  return typename T::Client(hook->addRef());
}
698
inline kj::Promise<void> Capability::Client::whenResolved() {
699 700
  return hook->whenResolved();
}
701 702 703 704 705
inline Request<AnyPointer, AnyPointer> Capability::Client::typelessRequest(
    uint64_t interfaceId, uint16_t methodId,
    kj::Maybe<MessageSize> sizeHint) {
  return newCall<AnyPointer, AnyPointer>(interfaceId, methodId, sizeHint);
}
706 707
template <typename Params, typename Results>
inline Request<Params, Results> Capability::Client::newCall(
708 709
    uint64_t interfaceId, uint16_t methodId, kj::Maybe<MessageSize> sizeHint) {
  auto typeless = hook->newCall(interfaceId, methodId, sizeHint);
710 711
  return Request<Params, Results>(typeless.template getAs<Params>(), kj::mv(typeless.hook));
}
712

713 714 715 716 717 718 719 720 721 722 723
template <typename Params, typename Results>
inline CallContext<Params, Results>::CallContext(CallContextHook& hook): hook(&hook) {}
template <typename Params, typename Results>
inline typename Params::Reader CallContext<Params, Results>::getParams() {
  return hook->getParams().template getAs<Params>();
}
template <typename Params, typename Results>
inline void CallContext<Params, Results>::releaseParams() {
  hook->releaseParams();
}
template <typename Params, typename Results>
Kenton Varda's avatar
Kenton Varda committed
724
inline typename Results::Builder CallContext<Params, Results>::getResults(
725
    kj::Maybe<MessageSize> sizeHint) {
726
  // `template` keyword needed due to: http://llvm.org/bugs/show_bug.cgi?id=17401
727
  return hook->getResults(sizeHint).template getAs<Results>();
728
}
729
template <typename Params, typename Results>
Kenton Varda's avatar
Kenton Varda committed
730
inline typename Results::Builder CallContext<Params, Results>::initResults(
731
    kj::Maybe<MessageSize> sizeHint) {
732
  // `template` keyword needed due to: http://llvm.org/bugs/show_bug.cgi?id=17401
733
  return hook->getResults(sizeHint).template initAs<Results>();
734
}
735 736
template <typename Params, typename Results>
inline void CallContext<Params, Results>::setResults(typename Results::Reader value) {
737
  hook->getResults(value.totalSize()).template setAs<Results>(value);
738
}
739 740
template <typename Params, typename Results>
inline void CallContext<Params, Results>::adoptResults(Orphan<Results>&& value) {
741
  hook->getResults(nullptr).adopt(kj::mv(value));
742
}
743
template <typename Params, typename Results>
744 745 746
inline Orphanage CallContext<Params, Results>::getResultsOrphanage(
    kj::Maybe<MessageSize> sizeHint) {
  return Orphanage::getForMessageContaining(hook->getResults(sizeHint));
747
}
748
template <typename Params, typename Results>
749 750 751 752 753 754
template <typename SubParams>
inline kj::Promise<void> CallContext<Params, Results>::tailCall(
    Request<SubParams, Results>&& tailRequest) {
  return hook->tailCall(kj::mv(tailRequest.hook));
}
template <typename Params, typename Results>
755 756
inline void CallContext<Params, Results>::allowCancellation() {
  hook->allowCancellation();
757
}
758

759 760
template <typename Params, typename Results>
CallContext<Params, Results> Capability::Server::internalGetTypedContext(
761
    CallContext<AnyPointer, AnyPointer> typeless) {
762 763 764
  return CallContext<Params, Results>(*typeless.hook);
}

765 766
Capability::Client Capability::Server::thisCap() {
  return Client(thisHook->addRef());
767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789
}

template <typename T>
typename T::Client CapabilityServerSet<T>::add(kj::Own<typename T::Server>&& server) {
  void* ptr = reinterpret_cast<void*>(server.get());
  // Clang insists that `castAs` is a template-dependent member and therefore we need the
  // `template` keyword here, but AFAICT this is wrong: addImpl() is not a template.
  return addInternal(kj::mv(server), ptr).template castAs<T>();
}

template <typename T>
kj::Promise<kj::Maybe<typename T::Server&>> CapabilityServerSet<T>::getLocalServer(
    typename T::Client& client) {
  return getLocalServerInternal(client)
      .then([](void* server) -> kj::Maybe<typename T::Server&> {
    if (server == nullptr) {
      return nullptr;
    } else {
      return *reinterpret_cast<typename T::Server*>(server);
    }
  });
}

790 791 792
}  // namespace capnp

#endif  // CAPNP_CAPABILITY_H_