policy_manager/policy_manager-inl.h

// Copyright (c) 2014 The Chromium OS 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 CHROMEOS_PLATFORM_UPDATE_ENGINE_POLICY_MANAGER_POLICY_MANAGER_INL_H_
#define CHROMEOS_PLATFORM_UPDATE_ENGINE_POLICY_MANAGER_POLICY_MANAGER_INL_H_

#include <string>

#include <base/bind.h>

#include "update_engine/policy_manager/evaluation_context.h"
#include "update_engine/policy_manager/event_loop.h"

namespace chromeos_policy_manager {

template<typename R, typename... Args>
EvalStatus PolicyManager::EvaluatePolicy(
    EvaluationContext* ec,
    EvalStatus (Policy::*policy_method)(EvaluationContext* ec,
                                        State* state,
                                        std::string* error,
                                        R* result,
                                        Args... args) const,
    R* result, Args... args) {
  std::string error;

  // First try calling the actual policy.
  EvalStatus status = (policy_.get()->*policy_method)(ec, state_.get(), &error,
                                                      result, args...);

  if (status == EvalStatus::kFailed) {
    LOG(WARNING) << "PolicyRequest() failed with error: " << error;
    error.clear();
    status = (default_policy_.*policy_method)(ec, state_.get(), &error,
                                              result, args...);

    if (status == EvalStatus::kFailed) {
      LOG(WARNING) << "Request to DefaultPolicy also failed, passing error.";
    }
  }
  // TODO(deymo): Log the actual state used from the EvaluationContext.
  return status;
}

template<typename R, typename... Args>
void PolicyManager::OnPolicyReadyToEvaluate(
    scoped_refptr<EvaluationContext> ec,
    base::Callback<void(EvalStatus status, const R& result)> callback,
    EvalStatus (Policy::*policy_method)(EvaluationContext* ec,
                                        State* state,
                                        std::string* error,
                                        R* result,
                                        Args... args) const,
    Args... args) {
  ec->ResetEvaluation();
  R result;
  EvalStatus status = EvaluatePolicy(ec, policy_method, &result, args...);

  if (status != EvalStatus::kAskMeAgainLater) {
    // AsyncPolicyRequest finished.
    callback.Run(status, result);
    return;
  }
  // Re-schedule the policy request based on used variables.
  base::Closure closure = base::Bind(
      &PolicyManager::OnPolicyReadyToEvaluate<R, Args...>,
      base::Unretained(this), ec, callback, policy_method, args...);

  if (!ec->RunOnValueChangeOrTimeout(closure)) {
    // The policy method didn't use any non-const variable nor there's any
    // time-based event that will change the status of evaluation. We call the
    // callback with EvalStatus::kAskMeAgainLater.
    LOG(ERROR) << "Policy implementation didn't use any non-const variable "
                  "but returned kAskMeAgainLater.";
    callback.Run(EvalStatus::kAskMeAgainLater, result);
    return;
  }
}

template<typename R, typename... Args>
EvalStatus PolicyManager::PolicyRequest(
    EvalStatus (Policy::*policy_method)(EvaluationContext* ec,
                                        State* state,
                                        std::string* error,
                                        R* result,
                                        Args... args) const,
    R* result, Args... args) {
  scoped_refptr<EvaluationContext> ec(new EvaluationContext(clock_));
  // A PolicyRequest allways consists on a single evaluation on a new
  // EvaluationContext.
  return EvaluatePolicy(ec, policy_method, result, args...);
}

template<typename R, typename... Args>
void PolicyManager::AsyncPolicyRequest(
    base::Callback<void(EvalStatus, const R& result)> callback,
    EvalStatus (Policy::*policy_method)(EvaluationContext* ec,
                                        State* state,
                                        std::string* error,
                                        R* result,
                                        Args... args) const,
    Args... args) {
  scoped_refptr<EvaluationContext> ec = new EvaluationContext(clock_);
  base::Closure closure = base::Bind(
      &PolicyManager::OnPolicyReadyToEvaluate<R, Args...>,
      base::Unretained(this), ec, callback, policy_method, args...);
  RunFromMainLoop(closure);
}

}  // namespace chromeos_policy_manager

#endif  // CHROMEOS_PLATFORM_UPDATE_ENGINE_POLICY_MANAGER_POLICY_MANAGER_INL_H_