// Copyright 2018 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.

#ifndef SRC_DEVICES_BLOCK_DRIVERS_CORE_MANAGER_H_
#define SRC_DEVICES_BLOCK_DRIVERS_CORE_MANAGER_H_

#include <fuchsia/hardware/block/cpp/banjo.h>
#include <lib/zx/fifo.h>
#include <lib/zx/vmo.h>
#include <threads.h>
#include <zircon/types.h>

#include <condition_variable>
#include <mutex>

#include "server.h"

// Manager controls the state of a background thread (or threads) servicing Fifo
// requests.
class Manager {
 public:
  DISALLOW_COPY_ASSIGN_AND_MOVE(Manager);

  Manager();
  ~Manager();

  // Launches the Fifo server in a background thread.
  //
  // Returns an error if the block server cannot be created.
  // Returns an error if the Fifo server is already running.
  zx_status_t StartServer(ddk::BlockProtocolClient* protocol, zx::fifo* out_fifo);

  // Ensures the FIFO server has terminated.
  //
  // When this function returns, it is guaranteed that the next call to |StartServer()|
  // won't see an already running Fifo server.
  zx_status_t CloseFifoServer();

  // Attaches a VMO to the currently executing server, if one is running.
  //
  // Returns an error if a server is not currently running.
  zx_status_t AttachVmo(zx::vmo vmo, vmoid_t* out_vmoid);

 private:
  enum class ThreadState : uint32_t {
    // No server is currently executing.
    None,
    // The server is executing right now.
    Running,
    // The server has finished executing, and is ready to be joined.
    Joinable,
  };

  // Queries if the Fifo Server is running, possibly cleaning up the old server's
  // thread if one exists.
  bool IsFifoServerRunning();

  // Joins the completed server thread and clean up all resources it may have used.
  void JoinServer();

  // Frees the Fifo server, cleaning up "server_" and setting the thread state to none.
  //
  // Precondition: No background thread is executing.
  void FreeServer();

  // Runs the server until |blockserver_shutdown| is invoked on |server_|, or the client
  // closes their end of the Fifo.
  static int RunServer(void* arg);

  ThreadState GetState() const {
    std::scoped_lock lock(mutex_);
    return state_;
  }

  void SetState(ThreadState state) {
    std::scoped_lock lock(mutex_);
    state_ = state;
    condition_.notify_all();
  }

  thrd_t thread_;
  mutable std::mutex mutex_;
  std::condition_variable_any condition_;
  ThreadState state_ TA_GUARDED(mutex_) = ThreadState::None;

  std::unique_ptr<Server> server_;
};

#endif  // SRC_DEVICES_BLOCK_DRIVERS_CORE_MANAGER_H_