host_server.h

Go to the documentation of this file.

/*
 * Copyright (C) 2021 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#pragma once
 
#include <stdint.h>
 
#include <atomic>
#include <memory>
#include <mutex>
#include <string>
#include <thread>
#include <unordered_map>
#include <vector>
 
#include "absl/log/log.h"
#include "fruit/fruit.h"
#include "teeui/utils.h"
 
#include "cuttlefish/common/libs/concurrency/multiplexer.h"
#include "cuttlefish/common/libs/concurrency/semaphore.h"
#include "cuttlefish/common/libs/concurrency/thread_safe_queue.h"
#include "cuttlefish/common/libs/confui/protocol.h"
#include "cuttlefish/common/libs/fs/shared_fd.h"
#include "cuttlefish/host/commands/kernel_log_monitor/utils.h"
#include "cuttlefish/host/libs/config/logging.h"
#include "cuttlefish/host/libs/confui/host_mode_ctrl.h"
#include "cuttlefish/host/libs/confui/host_renderer.h"
#include "cuttlefish/host/libs/confui/server_common.h"
#include "cuttlefish/host/libs/confui/session.h"
 
namespace cuttlefish {
namespace confui {
struct PipeConnectionPair {
  SharedFD from_guest_;
  SharedFD to_guest_;
};
 
class HostServer {
 public:
  INJECT(HostServer(HostModeCtrl& host_mode_ctrl, ConfUiRenderer& host_renderer,
                    const PipeConnectionPair& fd_pair));
 
  void Start();  // start this server itself
  virtual ~HostServer() {}
 
  // implement input interfaces. called by webRTC
  void TouchEvent(int x, int y, bool is_down);
  void UserAbortEvent();
 
 private:
  HostServer() = delete;
 
  [[noreturn]] void MainLoop();
  void HalCmdFetcherLoop();
 
  bool IsVirtioConsoleOpen() const;
  // If !IsVirtioConsoleOpen(), LOG(FATAL) and return false
  bool CheckVirtioConsole();
  std::shared_ptr<Session> CreateSession(const std::string& session_name);
  void SendUserSelection(std::unique_ptr<ConfUiMessage>& input);
 
  void Transition(std::unique_ptr<ConfUiMessage>& input_ptr);
  std::string GetCurrentSessionId() {
    if (curr_session_) {
      return curr_session_->GetId();
    }
    return SESSION_ANY;
  }
 
  std::string GetCurrentState() {
    if (!curr_session_) {
      return {"kInvalid"};
    }
    return ToString(curr_session_->GetState());
  }
 
  const uint32_t display_num_;
  ConfUiRenderer& host_renderer_;
  HostModeCtrl& host_mode_ctrl_;
 
  std::shared_ptr<Session> curr_session_;
 
  SharedFD from_guest_fifo_fd_;
  SharedFD to_guest_fifo_fd_;
 
  using Multiplexer =
      Multiplexer<std::unique_ptr<ConfUiMessage>,
                  ThreadSafeQueue<std::unique_ptr<ConfUiMessage>>>;
  /*
   * Multiplexer has N queues. When pop(), it is going to sleep until
   * there's at least one item in at least one queue. The lower the Q
   * index is, the higher the priority is.
   *
   * For HostServer, we have a queue for the user input events, and
   * another for hal cmd/msg queues
   */
  Multiplexer input_multiplexer_;
  int hal_cmd_q_id_;         // Q id in input_multiplexer_
  int user_input_evt_q_id_;  // Q id in input_multiplexer_
 
  std::thread main_loop_thread_;
  std::thread hal_input_fetcher_thread_;
};
 
}  // end of namespace confui
}  // end of namespace cuttlefish