source bootctrl changes

[GitHub/moto-9609/twrp_device_motorola_troika.git] / bootctrl / BootControl.cpp

/*\r
 * Copyright (C) 2020 The LineageOS Project\r
 *\r
 * Licensed under the Apache License, Version 2.0 (the "License");\r
 * you may not use this file except in compliance with the License.\r
 * You may obtain a copy of the License at\r
 *\r
 *      http://www.apache.org/licenses/LICENSE-2.0\r
 *\r
 * Unless required by applicable law or agreed to in writing, software\r
 * distributed under the License is distributed on an "AS IS" BASIS,\r
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
 * See the License for the specific language governing permissions and\r
 * limitations under the License.\r
 */\r
\r
#include "BootControl.h"\r
\r
#include <fstream>\r
\r
namespace android {\r
namespace hardware {\r
namespace boot {\r
namespace V1_0 {\r
namespace implementation {\r
\r
bool BootControl::readMetadata(bctl_metadata_t& data) {\r
    std::fstream in(mBlkDevice, std::ios::binary | std::ios::in);\r
\r
    if (in.fail()) {\r
        return false;\r
    }\r
\r
    in.seekg(BCTL_METADATA_OFFSET);\r
\r
    if (in.fail()) {\r
        return false;\r
    }\r
\r
    in.read(reinterpret_cast<char*>(&data), sizeof(bctl_metadata_t));\r
\r
    if (!validateMetadata(data))\r
        return false;\r
\r
    return !in.eof() && !in.fail();\r
}\r
\r
bool BootControl::writeMetadata(bctl_metadata_t& data) {\r
    if (!validateMetadata(data))\r
        return false;\r
\r
    // We use std::ios::in | std::ios::out even though we only write so that\r
    // we don't truncate or append to the file, but rather overwrite the file\r
    // in the exact place that we want to write the struct to.\r
    std::fstream out(mBlkDevice, std::ios::binary | std::ios::in | std::ios::out);\r
\r
    if (out.fail()) {\r
        return false;\r
    }\r
\r
    out.seekp(BCTL_METADATA_OFFSET);\r
\r
    if (out.fail()) {\r
        return false;\r
    }\r
\r
    out.write(reinterpret_cast<const char*>(&data), sizeof(bctl_metadata_t));\r
\r
    return !out.eof() && !out.fail();\r
}\r
\r
bool BootControl::validateMetadata(bctl_metadata_t& data) {\r
    if (data.slot_info[0].magic != BCTL_METADATA_MAGIC || data.slot_info[1].magic != BCTL_METADATA_MAGIC) {\r
        return false;\r
    }\r
    \r
    return true;\r
}\r
\r
void BootControl::resetMetadata(bctl_metadata_t& data) {\r
    // reset to defaults\r
    data.slot_info[0].magic = BCTL_METADATA_MAGIC;\r
    data.slot_info[0].bootable = 1;\r
    data.slot_info[0].is_active = 1;\r
    data.slot_info[0].boot_successful = 0;\r
    data.slot_info[0].tries_remaining = 7;\r
\r
    data.slot_info[1].magic = BCTL_METADATA_MAGIC;\r
    data.slot_info[1].bootable = 1;\r
    data.slot_info[1].is_active = 0;\r
    data.slot_info[1].boot_successful = 0;\r
    data.slot_info[1].tries_remaining = 7;\r
}\r
\r
// Methods from ::android::hardware::boot::V1_0::IBootControl follow.\r
Return<uint32_t> BootControl::getNumberSlots() {\r
    return 2;\r
}\r
\r
Return<uint32_t> BootControl::getCurrentSlot() {\r
    bctl_metadata_t data;\r
\r
    if (readMetadata(data)) {\r
        // This is a clever hack because if slot b is active,\r
        // is_active will be 0 and if slot a is active, is_active\r
        // will be 1. In other words, the "not" value of slot A's\r
        // is_active var lines up to the current active slot index.\r
        return !data.slot_info[0].is_active;\r
    }\r
\r
    return 0;\r
}\r
\r
Return<void> BootControl::markBootSuccessful(markBootSuccessful_cb _hidl_cb) {\r
    bctl_metadata_t data;\r
    bool success = false;\r
    int active_slot;\r
\r
    if (readMetadata(data)) {\r
        active_slot = !data.slot_info[0].is_active;\r
        \r
        data.slot_info[active_slot].boot_successful = 1;\r
        data.slot_info[active_slot].tries_remaining = 0;\r
\r
        if (success)\r
            if (writeMetadata(data)) {\r
                _hidl_cb(CommandResult{true, ""});\r
            } else {\r
                _hidl_cb(CommandResult{false, "Failed to write metadata"});\r
            }\r
        else {\r
            _hidl_cb(CommandResult{false, "No active slot"});\r
        }\r
    } else {\r
        _hidl_cb(CommandResult{false, "Failed to read metadata"});\r
    }\r
\r
    return Void();\r
}\r
\r
Return<void> BootControl::setActiveBootSlot(uint32_t slot, setActiveBootSlot_cb _hidl_cb) {\r
    bctl_metadata_t data;\r
    int slot2 = (slot == 0) ? 1 : 0;\r
\r
    if (slot < 2) {\r
        if (readMetadata(data)) {\r
            data.slot_info[slot].bootable = 1;\r
            data.slot_info[slot].is_active = 1;\r
            data.slot_info[slot].boot_successful = 0;\r
            data.slot_info[slot].tries_remaining = 7;\r
            \r
            data.slot_info[slot2].bootable = 1;\r
            data.slot_info[slot2].is_active = 0;\r
            data.slot_info[slot2].boot_successful = 0;\r
            data.slot_info[slot2].tries_remaining = 7;\r
\r
            if (writeMetadata(data)) {\r
                _hidl_cb(CommandResult{true, ""});\r
            } else {\r
                _hidl_cb(CommandResult{false, "Failed to write metadata"});\r
            }\r
        } else {\r
            _hidl_cb(CommandResult{false, "Failed to read metadata"});\r
        }\r
    } else {\r
        _hidl_cb(CommandResult{false, "Invalid slot"});\r
    }\r
\r
    return Void();\r
}\r
\r
Return<void> BootControl::setSlotAsUnbootable(uint32_t slot, setSlotAsUnbootable_cb _hidl_cb) {\r
    bctl_metadata_t data;\r
\r
    if (slot < 2) {\r
        if (readMetadata(data)) {\r
            data.slot_info[slot].bootable = 0;\r
\r
            if (writeMetadata(data)) {\r
                _hidl_cb(CommandResult{true, ""});\r
            } else {\r
                _hidl_cb(CommandResult{false, "Failed to write metadata"});\r
            }\r
        } else {\r
            _hidl_cb(CommandResult{false, "Failed to read metadata"});\r
        }\r
    } else {\r
        _hidl_cb(CommandResult{false, "Invalid slot"});\r
    }\r
\r
    return Void();\r
}\r
\r
Return<BoolResult> BootControl::isSlotBootable(uint32_t slot) {\r
    bctl_metadata_t data;\r
    BoolResult ret = BoolResult::FALSE;\r
\r
    if (slot < 2) {\r
        if (readMetadata(data)) {\r
            ret = static_cast<BoolResult>(data.slot_info[slot].bootable);\r
        } else {\r
            ret = BoolResult::FALSE;\r
        }\r
    } else {\r
        ret = BoolResult::INVALID_SLOT;\r
    }\r
\r
    return ret;\r
}\r
\r
Return<BoolResult> BootControl::isSlotMarkedSuccessful(uint32_t slot) {\r
    bctl_metadata_t data;\r
    BoolResult ret = BoolResult::FALSE;\r
\r
    if (slot < 2) {\r
        if (readMetadata(data)) {\r
            ret = static_cast<BoolResult>(data.slot_info[slot].boot_successful);\r
        } else {\r
            ret = BoolResult::FALSE;\r
        }\r
    } else {\r
        ret = BoolResult::INVALID_SLOT;\r
    }\r
\r
    return ret;\r
}\r
\r
Return<void> BootControl::getSuffix(uint32_t slot, getSuffix_cb _hidl_cb) {\r
    if (slot < 2) {\r
        if (slot == 0) {\r
            _hidl_cb(SLOT_SUFFIX_A);\r
        } else {\r
            _hidl_cb(SLOT_SUFFIX_B);\r
        }\r
    } else {\r
        // default to slot A\r
        _hidl_cb(SLOT_SUFFIX_A);\r
    }\r
\r
    return Void();\r
}\r
\r
}  // namespace implementation\r
}  // namespace V1_0\r
}  // namespace boot\r
}  // namespace hardware\r
}  // namespace android\r