login_manager/fake_container_manager.cc

// Copyright 2016 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "login_manager/fake_container_manager.h"

#include <utility>

#include <base/logging.h>

namespace login_manager {

FakeContainerManager::FakeContainerManager(pid_t pid) : pid_(pid) {}

bool FakeContainerManager::HandleExit(const siginfo_t& status) {
  return running_ && status.si_pid == pid_;
}

void FakeContainerManager::RequestJobExit(ArcContainerStopReason reason) {
  LOG_IF(FATAL, !running_) << "Trying to stop an already stopped container";
  running_ = false;
  std::move(exit_callback_).Run(pid_, reason);
}

void FakeContainerManager::EnsureJobExit(base::TimeDelta timeout) {}

bool FakeContainerManager::StartContainer(const std::vector<std::string>& env,
                                          ExitCallback exit_callback) {
  LOG_IF(FATAL, running_) << "Trying to start an already started container";
  exit_callback_ = std::move(exit_callback);
  running_ = true;
  return true;
}

StatefulMode FakeContainerManager::GetStatefulMode() const {
  return stateful_mode_;
}

void FakeContainerManager::SetStatefulMode(StatefulMode mode) {
  stateful_mode_ = mode;
}

bool FakeContainerManager::GetContainerPID(pid_t* pid_out) const {
  if (!running_)
    return false;
  *pid_out = pid_;
  return true;
}

void FakeContainerManager::SimulateCrash() {
  LOG_IF(FATAL, !running_) << "Trying to crash an already stopped container";
  running_ = false;
  std::move(exit_callback_).Run(pid_, ArcContainerStopReason::CRASH);
}

}  // namespace login_manager