cannam@132: // Copyright (c) 2013-2014 Sandstorm Development Group, Inc. and contributors cannam@132: // Licensed under the MIT License: cannam@132: // cannam@132: // Permission is hereby granted, free of charge, to any person obtaining a copy cannam@132: // of this software and associated documentation files (the "Software"), to deal cannam@132: // in the Software without restriction, including without limitation the rights cannam@132: // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell cannam@132: // copies of the Software, and to permit persons to whom the Software is cannam@132: // furnished to do so, subject to the following conditions: cannam@132: // cannam@132: // The above copyright notice and this permission notice shall be included in cannam@132: // all copies or substantial portions of the Software. cannam@132: // cannam@132: // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR cannam@132: // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, cannam@132: // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE cannam@132: // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER cannam@132: // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, cannam@132: // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN cannam@132: // THE SOFTWARE. cannam@132: cannam@132: #ifndef CAPNP_RPC_TWOPARTY_H_ cannam@132: #define CAPNP_RPC_TWOPARTY_H_ cannam@132: cannam@132: #if defined(__GNUC__) && !defined(CAPNP_HEADER_WARNINGS) cannam@132: #pragma GCC system_header cannam@132: #endif cannam@132: cannam@132: #include "rpc.h" cannam@132: #include "message.h" cannam@132: #include cannam@132: #include cannam@132: cannam@132: namespace capnp { cannam@132: cannam@132: namespace rpc { cannam@132: namespace twoparty { cannam@132: typedef VatId SturdyRefHostId; // For backwards-compatibility with version 0.4. cannam@132: } cannam@132: } cannam@132: cannam@132: typedef VatNetwork cannam@132: TwoPartyVatNetworkBase; cannam@132: cannam@132: class TwoPartyVatNetwork: public TwoPartyVatNetworkBase, cannam@132: private TwoPartyVatNetworkBase::Connection { cannam@132: // A `VatNetwork` that consists of exactly two parties communicating over an arbitrary byte cannam@132: // stream. This is used to implement the common case of a client/server network. cannam@132: // cannam@132: // See `ez-rpc.h` for a simple interface for setting up two-party clients and servers. cannam@132: // Use `TwoPartyVatNetwork` only if you need the advanced features. cannam@132: cannam@132: public: cannam@132: TwoPartyVatNetwork(kj::AsyncIoStream& stream, rpc::twoparty::Side side, cannam@132: ReaderOptions receiveOptions = ReaderOptions()); cannam@132: KJ_DISALLOW_COPY(TwoPartyVatNetwork); cannam@132: cannam@132: kj::Promise onDisconnect() { return disconnectPromise.addBranch(); } cannam@132: // Returns a promise that resolves when the peer disconnects. cannam@132: cannam@132: rpc::twoparty::Side getSide() { return side; } cannam@132: cannam@132: // implements VatNetwork ----------------------------------------------------- cannam@132: cannam@132: kj::Maybe> connect( cannam@132: rpc::twoparty::VatId::Reader ref) override; cannam@132: kj::Promise> accept() override; cannam@132: cannam@132: private: cannam@132: class OutgoingMessageImpl; cannam@132: class IncomingMessageImpl; cannam@132: cannam@132: kj::AsyncIoStream& stream; cannam@132: rpc::twoparty::Side side; cannam@132: MallocMessageBuilder peerVatId; cannam@132: ReaderOptions receiveOptions; cannam@132: bool accepted = false; cannam@132: cannam@132: kj::Maybe> previousWrite; cannam@132: // Resolves when the previous write completes. This effectively serves as the write queue. cannam@132: // Becomes null when shutdown() is called. cannam@132: cannam@132: kj::Own>> acceptFulfiller; cannam@132: // Fulfiller for the promise returned by acceptConnectionAsRefHost() on the client side, or the cannam@132: // second call on the server side. Never fulfilled, because there is only one connection. cannam@132: cannam@132: kj::ForkedPromise disconnectPromise = nullptr; cannam@132: cannam@132: class FulfillerDisposer: public kj::Disposer { cannam@132: // Hack: TwoPartyVatNetwork is both a VatNetwork and a VatNetwork::Connection. When the RPC cannam@132: // system detects (or initiates) a disconnection, it drops its reference to the Connection. cannam@132: // When all references have been dropped, then we want disconnectPromise to be fulfilled. cannam@132: // So we hand out Owns with this disposer attached, so that we can detect when cannam@132: // they are dropped. cannam@132: cannam@132: public: cannam@132: mutable kj::Own> fulfiller; cannam@132: mutable uint refcount = 0; cannam@132: cannam@132: void disposeImpl(void* pointer) const override; cannam@132: }; cannam@132: FulfillerDisposer disconnectFulfiller; cannam@132: cannam@132: kj::Own asConnection(); cannam@132: // Returns a pointer to this with the disposer set to disconnectFulfiller. cannam@132: cannam@132: // implements Connection ----------------------------------------------------- cannam@132: cannam@132: rpc::twoparty::VatId::Reader getPeerVatId() override; cannam@132: kj::Own newOutgoingMessage(uint firstSegmentWordSize) override; cannam@132: kj::Promise>> receiveIncomingMessage() override; cannam@132: kj::Promise shutdown() override; cannam@132: }; cannam@132: cannam@132: class TwoPartyServer: private kj::TaskSet::ErrorHandler { cannam@132: // Convenience class which implements a simple server which accepts connections on a listener cannam@132: // socket and serices them as two-party connections. cannam@132: cannam@132: public: cannam@132: explicit TwoPartyServer(Capability::Client bootstrapInterface); cannam@132: cannam@132: void accept(kj::Own&& connection); cannam@132: // Accepts the connection for servicing. cannam@132: cannam@132: kj::Promise listen(kj::ConnectionReceiver& listener); cannam@132: // Listens for connections on the given listener. The returned promise never resolves unless an cannam@132: // exception is thrown while trying to accept. You may discard the returned promise to cancel cannam@132: // listening. cannam@132: cannam@132: private: cannam@132: Capability::Client bootstrapInterface; cannam@132: kj::TaskSet tasks; cannam@132: cannam@132: struct AcceptedConnection; cannam@132: cannam@132: void taskFailed(kj::Exception&& exception) override; cannam@132: }; cannam@132: cannam@132: class TwoPartyClient { cannam@132: // Convenience class which implements a simple client. cannam@132: cannam@132: public: cannam@132: explicit TwoPartyClient(kj::AsyncIoStream& connection); cannam@132: TwoPartyClient(kj::AsyncIoStream& connection, Capability::Client bootstrapInterface, cannam@132: rpc::twoparty::Side side = rpc::twoparty::Side::CLIENT); cannam@132: cannam@132: Capability::Client bootstrap(); cannam@132: // Get the server's bootstrap interface. cannam@132: cannam@132: inline kj::Promise onDisconnect() { return network.onDisconnect(); } cannam@132: cannam@132: private: cannam@132: TwoPartyVatNetwork network; cannam@132: RpcSystem rpcSystem; cannam@132: }; cannam@132: cannam@132: } // namespace capnp cannam@132: cannam@132: #endif // CAPNP_RPC_TWOPARTY_H_