/*
 * Copyright (C) 2022 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.
 */

#include "host/libs/command_util/util.h"

#include "sys/time.h"
#include "sys/types.h"

#include <string>

#include "common/libs/fs/shared_buf.h"
#include "common/libs/fs/shared_fd.h"
#include "common/libs/fs/shared_select.h"
#include "common/libs/utils/result.h"
#include "host/libs/command_util/runner/defs.h"
#include "host/libs/config/cuttlefish_config.h"

namespace cuttlefish {
namespace {

bool IsShortAction(const LauncherAction action) {
  switch (action) {
    case LauncherAction::kFail:
    case LauncherAction::kPowerwash:
    case LauncherAction::kRestart:
    case LauncherAction::kStatus:
    case LauncherAction::kStop:
      return true;
    default:
      return false;
  };
}

template <typename T>
static Result<void> WriteAllBinaryResult(const SharedFD& fd, const T* t) {
  ssize_t n = WriteAllBinary(fd, t);
  CF_EXPECTF(n > 0, "Write error: {}", fd->StrError());
  CF_EXPECT(n == sizeof(*t), "Unexpected EOF on write");
  return {};
}

template <typename T>
static Result<void> ReadExactBinaryResult(const SharedFD& fd, T* t) {
  ssize_t n = ReadExactBinary(fd, t);
  CF_EXPECTF(n > 0, "Read error: {}", fd->StrError());
  CF_EXPECT(n == sizeof(*t), "Unexpected EOF on read");
  return {};
}

}  // namespace

Result<RunnerExitCodes> ReadExitCode(SharedFD monitor_socket) {
  RunnerExitCodes exit_codes;
  CF_EXPECT(ReadExactBinaryResult(monitor_socket, &exit_codes),
            "Error reading RunnerExitCodes");
  return exit_codes;
}

Result<SharedFD> GetLauncherMonitorFromInstance(
    const CuttlefishConfig::InstanceSpecific& instance_config,
    const int timeout_seconds) {
  std::string monitor_path = instance_config.launcher_monitor_socket_path();
  CF_EXPECT(!monitor_path.empty(), "No path to launcher monitor found");

  SharedFD monitor_socket = SharedFD::SocketLocalClient(
      monitor_path.c_str(), false, SOCK_STREAM, timeout_seconds);
  CF_EXPECT(monitor_socket->IsOpen(),
            "Unable to connect to launcher monitor at "
                << monitor_path << ":" << monitor_socket->StrError());
  return monitor_socket;
}

Result<SharedFD> GetLauncherMonitor(const CuttlefishConfig& config,
                                    const int instance_num,
                                    const int timeout_seconds) {
  auto instance_config = config.ForInstance(instance_num);
  return GetLauncherMonitorFromInstance(instance_config, timeout_seconds);
}

Result<LauncherActionInfo> ReadLauncherActionFromFd(SharedFD monitor_socket) {
  LauncherAction action;
  CF_EXPECT(ReadExactBinaryResult(monitor_socket, &action),
            "Error reading LauncherAction");
  if (IsShortAction(action)) {
    return LauncherActionInfo{
        .action = action,
        .extended_action = {},
    };
  }
  std::uint32_t length = 0;
  CF_EXPECT(ReadExactBinaryResult(monitor_socket, &length),
            "Error reading proto length");
  if (length == 0) {
    return LauncherActionInfo{
        .action = action,
        .extended_action = {},
    };
  }
  std::string serialized_data(length, 0);
  ssize_t n =
      ReadExact(monitor_socket, serialized_data.data(), serialized_data.size());
  CF_EXPECTF(n > 0, "Read error: {}", monitor_socket->StrError());
  CF_EXPECT(n == serialized_data.size(), "Unexpected EOF on read");

  run_cvd::ExtendedLauncherAction extended_action;
  CF_EXPECT(extended_action.ParseFromString(serialized_data),
            "Failed to parse ExtendedLauncherAction proto");

  return LauncherActionInfo{
      .action = action,
      .extended_action = std::move(extended_action),
  };
}

Result<void> WaitForRead(SharedFD monitor_socket, const int timeout_seconds) {
  // Perform a select with a timeout to guard against launcher hanging
  SharedFDSet read_set;
  read_set.Set(monitor_socket);
  struct timeval timeout = {timeout_seconds, 0};
  int select_result = Select(&read_set, nullptr, nullptr,
                             timeout_seconds <= 0 ? nullptr : &timeout);
  CF_EXPECT(select_result != 0,
            "Timeout expired waiting for launcher monitor to respond");
  CF_EXPECT(
      select_result > 0,
      "Failed communication with the launcher monitor: " << strerror(errno));
  return {};
}

Result<void> RunLauncherAction(SharedFD monitor_socket, LauncherAction action,
                               std::optional<int> timeout_seconds) {
  CF_EXPECTF(IsShortAction(action),
             "PerformActionRequest doesn't support extended action \"{}\"",
             static_cast<const char>(action));
  CF_EXPECT(WriteAllBinaryResult(monitor_socket, &action),
            "Error writing LauncherAction");

  if (timeout_seconds.has_value()) {
    CF_EXPECT(WaitForRead(monitor_socket, timeout_seconds.value()));
  }
  LauncherResponse response;
  CF_EXPECT(ReadExactBinaryResult(monitor_socket, &response),
            "Error reading LauncherResponse");
  CF_EXPECT_EQ((int)response, (int)LauncherResponse::kSuccess);
  return {};
}

Result<void> RunLauncherAction(
    SharedFD monitor_socket,
    const run_cvd::ExtendedLauncherAction& extended_action,
    std::optional<int> timeout_seconds) {
  const std::string serialized_data = extended_action.SerializeAsString();
  CF_EXPECT(!serialized_data.empty(), "failed to serialize proto");

  const LauncherAction action = LauncherAction::kExtended;
  CF_EXPECT(WriteAllBinaryResult(monitor_socket, &action),
            "Error writing LauncherAction");
  const std::uint32_t length = serialized_data.size();
  CF_EXPECT(WriteAllBinaryResult(monitor_socket, &length),
            "Error writing proto length");
  if (!serialized_data.empty()) {
    ssize_t n = WriteAll(monitor_socket, serialized_data.data(),
                         serialized_data.size());
    CF_EXPECTF(n > 0, "Write error: {}", monitor_socket->StrError());
    CF_EXPECT(n == serialized_data.size(), "Unexpected EOF on write");
  }

  if (timeout_seconds.has_value()) {
    CF_EXPECT(WaitForRead(monitor_socket, timeout_seconds.value()));
  }
  LauncherResponse response;
  CF_EXPECT(ReadExactBinaryResult(monitor_socket, &response),
            "Error reading LauncherResponse");
  CF_EXPECT_EQ((int)response, (int)LauncherResponse::kSuccess);
  return {};
}

}  // namespace cuttlefish