examples/fidl/llcpp/server/main.cc

// Copyright 2020 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <iostream>

// [START fidl_includes]
#include <fidl/fuchsia.examples/cpp/wire.h>
// [END fidl_includes]

// [START includes]
#include <lib/async-loop/cpp/loop.h>
#include <lib/async-loop/default.h>
#include <lib/fdio/directory.h>
#include <lib/fidl/llcpp/server.h>
#include <lib/stdcompat/optional.h>
#include <lib/svc/outgoing.h>
#include <zircon/process.h>
#include <zircon/processargs.h>
#include <zircon/status.h>
// [END includes]

class EchoImpl;

// [START impl]
// An implementation of the Echo protocol. Protocols are implemented in LLCPP by
// creating a subclass of the ::Interface class for the protocol.
class EchoImpl final : public fidl::WireServer<fuchsia_examples::Echo> {
 public:
  // [START bind_server]
  // Bind this implementation to a channel.
  EchoImpl(async_dispatcher_t* dispatcher, fidl::ServerEnd<fuchsia_examples::Echo> request)
      : binding_(fidl::BindServer(dispatcher, std::move(request), this,
                                  // This is a fidl::OnUnboundFn<EchoImpl>.
                                  [this](EchoImpl* impl, fidl::UnbindInfo info,
                                         fidl::ServerEnd<fuchsia_examples::Echo> server_end) {
                                    if (info.is_peer_closed()) {
                                      std::cout << "Client disconnected" << std::endl;
                                    } else if (!info.is_user_initiated()) {
                                      std::cerr << "server error: " << info << std::endl;
                                    }
                                    delete this;
                                  })) {}
  // [END bind_server]

  // Handle a SendString request by sending on OnString "event" (an unsolicited server-to-client
  // message) back on the same channel.
  //
  // For fire-and-forget methods like this one, the completer is normally not used but its
  // Close(zx_status_t) method can be used to close the channel (either if the connection is "done"
  // or it encountered an unrecoverable error).
  void SendString(SendStringRequestView request, SendStringCompleter::Sync& completer) override {
    fidl::Status status = fidl::WireSendEvent(binding_)->OnString(request->value);
    ZX_ASSERT(status.ok());
  }

  // Handle an EchoString request by responding with the request value. For two-way methods (those
  // with a response) like this one, the completer is used to send the response.
  void EchoString(EchoStringRequestView request, EchoStringCompleter::Sync& completer) override {
    completer.Reply(request->value);
  }

 private:
  // A reference back to the Binding that this class is bound to, which is used
  // to send events to the client.
  fidl::ServerBindingRef<fuchsia_examples::Echo> binding_;
};
// [END impl]

// [START main]
int main(int argc, char** argv) {
  // Initialize the async loop. The Echo server will use the dispatcher of this
  // loop to listen for incoming requests.
  async::Loop loop(&kAsyncLoopConfigAttachToCurrentThread);
  async_dispatcher_t* dispatcher = loop.dispatcher();

  // Create an Outgoing class which will serve requests from the /svc/ directory.
  svc::Outgoing outgoing(loop.dispatcher());
  zx_status_t status = outgoing.ServeFromStartupInfo();
  if (status != ZX_OK) {
    std::cerr << "error: ServeFromStartupInfo returned: " << status << " ("
              << zx_status_get_string(status) << ")" << std::endl;
    return -1;
  }

  // Register a handler for components trying to connect to fuchsia.examples.Echo.
  status = outgoing.svc_dir()->AddEntry(
      fidl::DiscoverableProtocolName<fuchsia_examples::Echo>,
      fbl::MakeRefCounted<fs::Service>(
          [dispatcher](fidl::ServerEnd<fuchsia_examples::Echo> request) mutable {
            std::cout << "Incoming connection for "
                      << fidl::DiscoverableProtocolName<fuchsia_examples::Echo> << std::endl;
            // [START create_server]
            // Create an instance of our EchoImpl that destroys itself when the connection closes.
            new EchoImpl(dispatcher, std::move(request));
            // [END create_server]
            return ZX_OK;
          }));

  std::cout << "Running echo server" << std::endl;
  loop.Run();
  return 0;
}
// [END main]