//
|
// Copyright (C) 2014 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.
|
//
|
|
#ifndef UPDATE_ENGINE_UPDATE_MANAGER_REAL_DEVICE_POLICY_PROVIDER_H_
|
#define UPDATE_ENGINE_UPDATE_MANAGER_REAL_DEVICE_POLICY_PROVIDER_H_
|
|
#include <memory>
|
#include <set>
|
#include <string>
|
#include <utility>
|
|
#include <brillo/message_loops/message_loop.h>
|
#include <gtest/gtest_prod.h> // for FRIEND_TEST
|
#include <policy/libpolicy.h>
|
#if USE_DBUS
|
#include <session_manager/dbus-proxies.h>
|
#endif // USE_DBUS
|
|
#include "update_engine/update_manager/device_policy_provider.h"
|
#include "update_engine/update_manager/generic_variables.h"
|
|
namespace chromeos_update_manager {
|
|
// DevicePolicyProvider concrete implementation.
|
class RealDevicePolicyProvider : public DevicePolicyProvider {
|
public:
|
#if USE_DBUS
|
RealDevicePolicyProvider(
|
std::unique_ptr<org::chromium::SessionManagerInterfaceProxyInterface>
|
session_manager_proxy,
|
policy::PolicyProvider* policy_provider)
|
: policy_provider_(policy_provider),
|
session_manager_proxy_(std::move(session_manager_proxy)) {}
|
#endif // USE_DBUS
|
explicit RealDevicePolicyProvider(policy::PolicyProvider* policy_provider)
|
: policy_provider_(policy_provider) {}
|
~RealDevicePolicyProvider();
|
|
// Initializes the provider and returns whether it succeeded.
|
bool Init();
|
|
Variable<bool>* var_device_policy_is_loaded() override {
|
return &var_device_policy_is_loaded_;
|
}
|
|
Variable<std::string>* var_release_channel() override {
|
return &var_release_channel_;
|
}
|
|
Variable<bool>* var_release_channel_delegated() override {
|
return &var_release_channel_delegated_;
|
}
|
|
Variable<bool>* var_update_disabled() override {
|
return &var_update_disabled_;
|
}
|
|
Variable<std::string>* var_target_version_prefix() override {
|
return &var_target_version_prefix_;
|
}
|
|
Variable<RollbackToTargetVersion>* var_rollback_to_target_version() override {
|
return &var_rollback_to_target_version_;
|
}
|
|
Variable<int>* var_rollback_allowed_milestones() override {
|
return &var_rollback_allowed_milestones_;
|
}
|
|
Variable<base::TimeDelta>* var_scatter_factor() override {
|
return &var_scatter_factor_;
|
}
|
|
Variable<std::set<chromeos_update_engine::ConnectionType>>*
|
var_allowed_connection_types_for_update() override {
|
return &var_allowed_connection_types_for_update_;
|
}
|
|
Variable<std::string>* var_owner() override { return &var_owner_; }
|
|
Variable<bool>* var_http_downloads_enabled() override {
|
return &var_http_downloads_enabled_;
|
}
|
|
Variable<bool>* var_au_p2p_enabled() override { return &var_au_p2p_enabled_; }
|
|
Variable<bool>* var_allow_kiosk_app_control_chrome_version() override {
|
return &var_allow_kiosk_app_control_chrome_version_;
|
}
|
|
Variable<std::string>* var_auto_launched_kiosk_app_id() override {
|
return &var_auto_launched_kiosk_app_id_;
|
}
|
|
Variable<WeeklyTimeIntervalVector>* var_disallowed_time_intervals() override {
|
return &var_disallowed_time_intervals_;
|
}
|
|
private:
|
FRIEND_TEST(UmRealDevicePolicyProviderTest, RefreshScheduledTest);
|
FRIEND_TEST(UmRealDevicePolicyProviderTest, NonExistentDevicePolicyReloaded);
|
FRIEND_TEST(UmRealDevicePolicyProviderTest, ValuesUpdated);
|
|
// A static handler for the PropertyChangedCompleted signal from the session
|
// manager used as a callback.
|
void OnPropertyChangedCompletedSignal(const std::string& success);
|
|
// Called when the signal in UpdateEngineLibcrosProxyResolvedInterface is
|
// connected.
|
void OnSignalConnected(const std::string& interface_name,
|
const std::string& signal_name,
|
bool successful);
|
|
// Schedules a call to periodically refresh the device policy.
|
void RefreshDevicePolicyAndReschedule();
|
|
// Reloads the device policy and updates all the exposed variables.
|
void RefreshDevicePolicy();
|
|
// Updates the async variable |var| based on the result value of the method
|
// passed, which is a DevicePolicy getter method.
|
template <typename T>
|
void UpdateVariable(AsyncCopyVariable<T>* var,
|
bool (policy::DevicePolicy::*getter_method)(T*) const);
|
|
// Updates the async variable |var| based on the result value of the getter
|
// method passed, which is a wrapper getter on this class.
|
template <typename T>
|
void UpdateVariable(AsyncCopyVariable<T>* var,
|
bool (RealDevicePolicyProvider::*getter_method)(T*)
|
const);
|
|
// Wrapper for DevicePolicy::GetRollbackToTargetVersion() that converts the
|
// result to RollbackToTargetVersion.
|
bool ConvertRollbackToTargetVersion(
|
RollbackToTargetVersion* rollback_to_target_version) const;
|
|
// Wrapper for DevicePolicy::GetScatterFactorInSeconds() that converts the
|
// result to a base::TimeDelta. It returns the same value as
|
// GetScatterFactorInSeconds().
|
bool ConvertScatterFactor(base::TimeDelta* scatter_factor) const;
|
|
// Wrapper for DevicePolicy::GetAllowedConnectionTypesForUpdate() that
|
// converts the result to a set of ConnectionType elements instead of strings.
|
bool ConvertAllowedConnectionTypesForUpdate(
|
std::set<chromeos_update_engine::ConnectionType>* allowed_types) const;
|
|
// Wrapper for DevicePolicy::GetUpdateTimeRestrictions() that converts
|
// the DevicePolicy::WeeklyTimeInterval structs to WeeklyTimeInterval objects,
|
// which offer more functionality.
|
bool ConvertDisallowedTimeIntervals(
|
WeeklyTimeIntervalVector* disallowed_intervals_out) const;
|
|
// Used for fetching information about the device policy.
|
policy::PolicyProvider* policy_provider_;
|
|
// Used to schedule refreshes of the device policy.
|
brillo::MessageLoop::TaskId scheduled_refresh_{
|
brillo::MessageLoop::kTaskIdNull};
|
|
#if USE_DBUS
|
// The DBus (mockable) session manager proxy.
|
std::unique_ptr<org::chromium::SessionManagerInterfaceProxyInterface>
|
session_manager_proxy_;
|
#endif // USE_DBUS
|
|
// Variable exposing whether the policy is loaded.
|
AsyncCopyVariable<bool> var_device_policy_is_loaded_{"policy_is_loaded",
|
false};
|
|
// Variables mapping the exposed methods from the policy::DevicePolicy.
|
AsyncCopyVariable<std::string> var_release_channel_{"release_channel"};
|
AsyncCopyVariable<bool> var_release_channel_delegated_{
|
"release_channel_delegated"};
|
AsyncCopyVariable<bool> var_update_disabled_{"update_disabled"};
|
AsyncCopyVariable<std::string> var_target_version_prefix_{
|
"target_version_prefix"};
|
AsyncCopyVariable<RollbackToTargetVersion> var_rollback_to_target_version_{
|
"rollback_to_target_version"};
|
AsyncCopyVariable<int> var_rollback_allowed_milestones_{
|
"rollback_allowed_milestones"};
|
AsyncCopyVariable<base::TimeDelta> var_scatter_factor_{"scatter_factor"};
|
AsyncCopyVariable<std::set<chromeos_update_engine::ConnectionType>>
|
var_allowed_connection_types_for_update_{
|
"allowed_connection_types_for_update"};
|
AsyncCopyVariable<std::string> var_owner_{"owner"};
|
AsyncCopyVariable<bool> var_http_downloads_enabled_{"http_downloads_enabled"};
|
AsyncCopyVariable<bool> var_au_p2p_enabled_{"au_p2p_enabled"};
|
AsyncCopyVariable<bool> var_allow_kiosk_app_control_chrome_version_{
|
"allow_kiosk_app_control_chrome_version"};
|
AsyncCopyVariable<WeeklyTimeIntervalVector> var_disallowed_time_intervals_{
|
"update_time_restrictions"};
|
AsyncCopyVariable<std::string> var_auto_launched_kiosk_app_id_{
|
"auto_launched_kiosk_app_id"};
|
|
DISALLOW_COPY_AND_ASSIGN(RealDevicePolicyProvider);
|
};
|
|
} // namespace chromeos_update_manager
|
|
#endif // UPDATE_ENGINE_UPDATE_MANAGER_REAL_DEVICE_POLICY_PROVIDER_H_
|