//
|
// Copyright (C) 2015 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_COMMON_FAKE_BOOT_CONTROL_H_
|
#define UPDATE_ENGINE_COMMON_FAKE_BOOT_CONTROL_H_
|
|
#include <map>
|
#include <string>
|
#include <vector>
|
|
#include <base/time/time.h>
|
|
#include "update_engine/common/boot_control_interface.h"
|
|
namespace chromeos_update_engine {
|
|
// Implements a fake bootloader control interface used for testing.
|
class FakeBootControl : public BootControlInterface {
|
public:
|
FakeBootControl() {
|
SetNumSlots(num_slots_);
|
// The current slot should be bootable.
|
is_bootable_[current_slot_] = true;
|
}
|
|
// BootControlInterface overrides.
|
unsigned int GetNumSlots() const override { return num_slots_; }
|
BootControlInterface::Slot GetCurrentSlot() const override {
|
return current_slot_;
|
}
|
|
bool GetPartitionDevice(const std::string& partition_name,
|
BootControlInterface::Slot slot,
|
std::string* device) const override {
|
if (slot >= num_slots_)
|
return false;
|
auto part_it = devices_[slot].find(partition_name);
|
if (part_it == devices_[slot].end())
|
return false;
|
*device = part_it->second;
|
return true;
|
}
|
|
bool IsSlotBootable(BootControlInterface::Slot slot) const override {
|
return slot < num_slots_ && is_bootable_[slot];
|
}
|
|
bool MarkSlotUnbootable(BootControlInterface::Slot slot) override {
|
if (slot >= num_slots_)
|
return false;
|
is_bootable_[slot] = false;
|
return true;
|
}
|
|
bool SetActiveBootSlot(Slot slot) override { return true; }
|
|
bool MarkBootSuccessfulAsync(base::Callback<void(bool)> callback) override {
|
// We run the callback directly from here to avoid having to setup a message
|
// loop in the test environment.
|
callback.Run(true);
|
return true;
|
}
|
|
bool InitPartitionMetadata(Slot slot,
|
const PartitionMetadata& partition_metadata,
|
bool update_metadata) override {
|
return true;
|
}
|
|
void Cleanup() override {}
|
|
// Setters
|
void SetNumSlots(unsigned int num_slots) {
|
num_slots_ = num_slots;
|
is_bootable_.resize(num_slots_, false);
|
devices_.resize(num_slots_);
|
}
|
|
void SetCurrentSlot(BootControlInterface::Slot slot) { current_slot_ = slot; }
|
|
void SetPartitionDevice(const std::string& partition_name,
|
BootControlInterface::Slot slot,
|
const std::string& device) {
|
DCHECK(slot < num_slots_);
|
devices_[slot][partition_name] = device;
|
}
|
|
void SetSlotBootable(BootControlInterface::Slot slot, bool bootable) {
|
DCHECK(slot < num_slots_);
|
is_bootable_[slot] = bootable;
|
}
|
|
private:
|
BootControlInterface::Slot num_slots_{2};
|
BootControlInterface::Slot current_slot_{0};
|
|
std::vector<bool> is_bootable_;
|
std::vector<std::map<std::string, std::string>> devices_;
|
|
DISALLOW_COPY_AND_ASSIGN(FakeBootControl);
|
};
|
|
} // namespace chromeos_update_engine
|
|
#endif // UPDATE_ENGINE_COMMON_FAKE_BOOT_CONTROL_H_
|
