This is a port of the Android bootctrl HAL and Qualcomms GPT based implementation for musl/glibc based systems running mainline linux Currently it doesn't work because the gpt hdr which is read doesn't seem to have valid data.merge-requests/7/head
commit
2f5bf10269
@ -0,0 +1,2 @@ |
|||||||
|
.vscode/ |
||||||
|
build/ |
@ -0,0 +1,92 @@ |
|||||||
|
/*
|
||||||
|
* Copyright (C) 2016 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. |
||||||
|
*/ |
||||||
|
|
||||||
|
#include <string> |
||||||
|
|
||||||
|
#include "boot_control.h" |
||||||
|
#include "BootControl.h" |
||||||
|
|
||||||
|
BootControl::BootControl(boot_control_module *module) : mModule(module) { |
||||||
|
} |
||||||
|
|
||||||
|
// Methods from ::android::hardware::boot::V1_0::IBootControl follow.
|
||||||
|
unsigned int BootControl::getNumberSlots() { |
||||||
|
return mModule->getNumberSlots(mModule); |
||||||
|
} |
||||||
|
|
||||||
|
unsigned int BootControl::getCurrentSlot() { |
||||||
|
return mModule->getCurrentSlot(mModule); |
||||||
|
} |
||||||
|
|
||||||
|
int BootControl::markBootSuccessful() { |
||||||
|
int ret = mModule->markBootSuccessful(mModule); |
||||||
|
return ret; |
||||||
|
} |
||||||
|
|
||||||
|
int BootControl::setActiveBootSlot(unsigned int slot) { |
||||||
|
int ret = mModule->setActiveBootSlot(mModule, slot); |
||||||
|
return ret; |
||||||
|
} |
||||||
|
|
||||||
|
int BootControl::setSlotAsUnbootable(unsigned int slot) { |
||||||
|
int ret = mModule->setSlotAsUnbootable(mModule, slot); |
||||||
|
return ret; |
||||||
|
} |
||||||
|
|
||||||
|
BoolResult BootControl::isSlotBootable(unsigned int slot) { |
||||||
|
int32_t ret = mModule->isSlotBootable(mModule, slot); |
||||||
|
if (ret < 0) { |
||||||
|
return BoolResult::INVALID_SLOT; |
||||||
|
} |
||||||
|
return ret ? BoolResult::TRUE : BoolResult::FALSE; |
||||||
|
} |
||||||
|
|
||||||
|
BoolResult BootControl::isSlotMarkedSuccessful(unsigned int slot) { |
||||||
|
int32_t ret = mModule->isSlotMarkedSuccessful(mModule, slot); |
||||||
|
if (ret < 0) { |
||||||
|
return BoolResult::INVALID_SLOT; |
||||||
|
} |
||||||
|
return ret ? BoolResult::TRUE : BoolResult::FALSE; |
||||||
|
} |
||||||
|
|
||||||
|
std::string BootControl::getSuffix(unsigned int slot) { |
||||||
|
std::string ans; |
||||||
|
const char *suffix = mModule->getSuffix(mModule, slot); |
||||||
|
if (suffix) { |
||||||
|
ans = std::string(suffix); |
||||||
|
} |
||||||
|
return ans; |
||||||
|
} |
||||||
|
|
||||||
|
BootControl* BootControlInit() { |
||||||
|
boot_control_module* module; |
||||||
|
|
||||||
|
// For devices that don't build a standalone libhardware bootctrl impl for recovery,
|
||||||
|
// we simulate the hw_get_module() by accessing it from the current process directly.
|
||||||
|
const boot_control_module* hw_module = &HAL_MODULE_INFO_SYM; |
||||||
|
module = reinterpret_cast<boot_control_module*>(const_cast<boot_control_module*>(hw_module)); |
||||||
|
module->init(module); |
||||||
|
return new BootControl(module); |
||||||
|
} |
||||||
|
|
||||||
|
int main (int argc, char * argv []) { |
||||||
|
BootControl *bootctl = BootControlInit(); |
||||||
|
printf("======= Current slot: %d\n", bootctl->getCurrentSlot()); |
||||||
|
printf("======= isslotbootable: a = %d, b = %d\n", bootctl->isSlotBootable(0), |
||||||
|
bootctl->isSlotBootable(1)); |
||||||
|
printf("======= isSlotMarkedSuccessful: a = %d, b = %d\n", bootctl->isSlotMarkedSuccessful(0), |
||||||
|
bootctl->isSlotMarkedSuccessful(1)); |
||||||
|
} |
@ -0,0 +1,45 @@ |
|||||||
|
/*
|
||||||
|
* Copyright (C) 2016 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 ANDROID_HARDWARE_BOOT_V1_0_BOOTCONTROL_H |
||||||
|
#define ANDROID_HARDWARE_BOOT_V1_0_BOOTCONTROL_H |
||||||
|
|
||||||
|
/**
|
||||||
|
* A result encapsulating whether a function returned true, false or |
||||||
|
* failed due to an invalid slot number |
||||||
|
*/ |
||||||
|
enum class BoolResult : int { |
||||||
|
FALSE = 0, |
||||||
|
TRUE = 1, |
||||||
|
INVALID_SLOT = -1 /* -1 */, |
||||||
|
}; |
||||||
|
|
||||||
|
struct BootControl { |
||||||
|
BootControl(boot_control_module* module); |
||||||
|
// Methods from ::android::hardware::boot::V1_0::IBootControl follow.
|
||||||
|
unsigned int getNumberSlots(); |
||||||
|
unsigned int getCurrentSlot(); |
||||||
|
int markBootSuccessful(); |
||||||
|
int setActiveBootSlot(unsigned int slot); |
||||||
|
int setSlotAsUnbootable(unsigned int slot); |
||||||
|
BoolResult isSlotBootable(unsigned int slot); |
||||||
|
BoolResult isSlotMarkedSuccessful(unsigned int slot); |
||||||
|
std::string getSuffix(unsigned int slot); |
||||||
|
private: |
||||||
|
boot_control_module* mModule; |
||||||
|
}; |
||||||
|
|
||||||
|
|
||||||
|
#endif // ANDROID_HARDWARE_BOOT_V1_0_BOOTCONTROL_H
|
@ -0,0 +1,15 @@ |
|||||||
|
cmake_minimum_required(VERSION 3.10) |
||||||
|
|
||||||
|
# set the project name |
||||||
|
project(qbootctl) |
||||||
|
|
||||||
|
set(BootCtrl_Files BootControl.cpp |
||||||
|
boot_control_impl_qcom.cpp |
||||||
|
gpt-utils.cpp) |
||||||
|
|
||||||
|
include_directories(include) |
||||||
|
|
||||||
|
# add the executable |
||||||
|
add_executable(qbootctl ${BootCtrl_Files}) |
||||||
|
|
||||||
|
target_link_libraries(qbootctl libz.so) |
@ -0,0 +1,7 @@ |
|||||||
|
Qualcomm bootctl HAL for Linux |
||||||
|
|
||||||
|
This HAL was pulled from AOSP source code and bastardised to build and run on a musl/glibc system. This may or may not render any hardware you run it on unusable, you have been warned. |
||||||
|
|
||||||
|
# Dependencies |
||||||
|
|
||||||
|
* zlib-dev |
@ -0,0 +1,136 @@ |
|||||||
|
/*
|
||||||
|
* 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 ANDROID_INCLUDE_HARDWARE_BOOT_CONTROL_H |
||||||
|
#define ANDROID_INCLUDE_HARDWARE_BOOT_CONTROL_H |
||||||
|
|
||||||
|
#define BOOT_CONTROL_MODULE_API_VERSION_0_1 HARDWARE_MODULE_API_VERSION(0, 1) |
||||||
|
|
||||||
|
/**
|
||||||
|
* The id of this module |
||||||
|
*/ |
||||||
|
#define BOOT_CONTROL_HARDWARE_MODULE_ID "bootctrl" |
||||||
|
|
||||||
|
/*
|
||||||
|
* The Boot Control HAL is designed to allow for managing sets of redundant |
||||||
|
* partitions, called slots, that can be booted from independantly. Slots |
||||||
|
* are sets of partitions whose names differ only by a given suffix. |
||||||
|
* They are identified here by a 0 indexed number, and associated with their |
||||||
|
* suffix, which can be appended to the base name for any particular partition |
||||||
|
* to find the one associated with that slot. The bootloader must pass the suffix |
||||||
|
* of the currently active slot either through a kernel command line property at |
||||||
|
* androidboot.slot_suffix, or the device tree at /firmware/android/slot_suffix. |
||||||
|
* The primary use of this set up is to allow for background updates while the |
||||||
|
* device is running, and to provide a fallback in the event that the update fails. |
||||||
|
*/ |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Every hardware module must have a data structure named HAL_MODULE_INFO_SYM |
||||||
|
* and the fields of this data structure must begin with hw_module_t |
||||||
|
* followed by module specific information. |
||||||
|
*/ |
||||||
|
struct boot_control_module { |
||||||
|
|
||||||
|
/*
|
||||||
|
* (*init)() perform any initialization tasks needed for the HAL. |
||||||
|
* This is called only once. |
||||||
|
*/ |
||||||
|
void (*init)(struct boot_control_module *module); |
||||||
|
|
||||||
|
/*
|
||||||
|
* (*getNumberSlots)() returns the number of available slots. |
||||||
|
* For instance, a system with a single set of partitions would return |
||||||
|
* 1, a system with A/B would return 2, A/B/C -> 3... |
||||||
|
*/ |
||||||
|
unsigned (*getNumberSlots)(struct boot_control_module *module); |
||||||
|
|
||||||
|
/*
|
||||||
|
* (*getCurrentSlot)() returns the value letting the system know |
||||||
|
* whether the current slot is A or B. The meaning of A and B is |
||||||
|
* left up to the implementer. It is assumed that if the current slot |
||||||
|
* is A, then the block devices underlying B can be accessed directly |
||||||
|
* without any risk of corruption. |
||||||
|
* The returned value is always guaranteed to be strictly less than the |
||||||
|
* value returned by getNumberSlots. Slots start at 0 and |
||||||
|
* finish at getNumberSlots() - 1 |
||||||
|
*/ |
||||||
|
unsigned (*getCurrentSlot)(struct boot_control_module *module); |
||||||
|
|
||||||
|
/*
|
||||||
|
* (*markBootSuccessful)() marks the current slot |
||||||
|
* as having booted successfully |
||||||
|
* |
||||||
|
* Returns 0 on success, -errno on error. |
||||||
|
*/ |
||||||
|
int (*markBootSuccessful)(struct boot_control_module *module); |
||||||
|
|
||||||
|
/*
|
||||||
|
* (*setActiveBootSlot)() marks the slot passed in parameter as |
||||||
|
* the active boot slot (see getCurrentSlot for an explanation |
||||||
|
* of the "slot" parameter). This overrides any previous call to |
||||||
|
* setSlotAsUnbootable. |
||||||
|
* Returns 0 on success, -errno on error. |
||||||
|
*/ |
||||||
|
int (*setActiveBootSlot)(struct boot_control_module *module, unsigned slot); |
||||||
|
|
||||||
|
/*
|
||||||
|
* (*setSlotAsUnbootable)() marks the slot passed in parameter as |
||||||
|
* an unbootable. This can be used while updating the contents of the slot's |
||||||
|
* partitions, so that the system will not attempt to boot a known bad set up. |
||||||
|
* Returns 0 on success, -errno on error. |
||||||
|
*/ |
||||||
|
int (*setSlotAsUnbootable)(struct boot_control_module *module, unsigned slot); |
||||||
|
|
||||||
|
/*
|
||||||
|
* (*isSlotBootable)() returns if the slot passed in parameter is |
||||||
|
* bootable. Note that slots can be made unbootable by both the |
||||||
|
* bootloader and by the OS using setSlotAsUnbootable. |
||||||
|
* Returns 1 if the slot is bootable, 0 if it's not, and -errno on |
||||||
|
* error. |
||||||
|
*/ |
||||||
|
int (*isSlotBootable)(struct boot_control_module *module, unsigned slot); |
||||||
|
|
||||||
|
/*
|
||||||
|
* (*getSuffix)() returns the string suffix used by partitions that |
||||||
|
* correspond to the slot number passed in parameter. The returned string |
||||||
|
* is expected to be statically allocated and not need to be freed. |
||||||
|
* Returns NULL if slot does not match an existing slot. |
||||||
|
*/ |
||||||
|
const char* (*getSuffix)(struct boot_control_module *module, unsigned slot); |
||||||
|
|
||||||
|
/*
|
||||||
|
* (*isSlotMarkedSucessful)() returns if the slot passed in parameter has |
||||||
|
* been marked as successful using markBootSuccessful. |
||||||
|
* Returns 1 if the slot has been marked as successful, 0 if it's |
||||||
|
* not the case, and -errno on error. |
||||||
|
*/ |
||||||
|
int (*isSlotMarkedSuccessful)(struct boot_control_module *module, unsigned slot); |
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns the active slot to boot into on the next boot. If |
||||||
|
* setActiveBootSlot() has been called, the getter function should return |
||||||
|
* the same slot as the one provided in the last setActiveBootSlot() call. |
||||||
|
*/ |
||||||
|
unsigned (*getActiveBootSlot)(struct boot_control_module *module); |
||||||
|
|
||||||
|
void* reserved[30]; |
||||||
|
}; |
||||||
|
|
||||||
|
extern const struct boot_control_module HAL_MODULE_INFO_SYM; |
||||||
|
|
||||||
|
|
||||||
|
#endif // ANDROID_INCLUDE_HARDWARE_BOOT_CONTROL_H
|
@ -0,0 +1,717 @@ |
|||||||
|
/*
|
||||||
|
* Copyright (c) 2016, The Linux Foundation. All rights reserved. |
||||||
|
* |
||||||
|
* Redistribution and use in source and binary forms, with or without |
||||||
|
* modification, are permitted provided that the following conditions are |
||||||
|
* met: |
||||||
|
* * Redistributions of source code must retain the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer. |
||||||
|
* * Redistributions in binary form must reproduce the above |
||||||
|
* copyright notice, this list of conditions and the following |
||||||
|
* disclaimer in the documentation and/or other materials provided |
||||||
|
* with the distribution. |
||||||
|
* * Neither the name of The Linux Foundation nor the names of its |
||||||
|
* contributors may be used to endorse or promote products derived |
||||||
|
* from this software without specific prior written permission. |
||||||
|
* |
||||||
|
* THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESS OR IMPLIED |
||||||
|
* WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF |
||||||
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT |
||||||
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS |
||||||
|
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
||||||
|
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
||||||
|
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR |
||||||
|
* BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, |
||||||
|
* WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE |
||||||
|
* OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN |
||||||
|
* IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||||
|
*/ |
||||||
|
#include <map> |
||||||
|
#include <list> |
||||||
|
#include <string> |
||||||
|
#include <vector> |
||||||
|
#include <errno.h> |
||||||
|
#include <regex> |
||||||
|
#include <stdio.h> |
||||||
|
#include <string.h> |
||||||
|
#include <unistd.h> |
||||||
|
#include <dirent.h> |
||||||
|
#include <sys/types.h> |
||||||
|
#include <sys/stat.h> |
||||||
|
#include <fcntl.h> |
||||||
|
#include <limits.h> |
||||||
|
|
||||||
|
#include "boot_control.h" |
||||||
|
#include "gpt-utils.h" |
||||||
|
|
||||||
|
#define BOOTDEV_DIR "/dev/disk/by-partlabel" |
||||||
|
#define BOOT_IMG_PTN_NAME "boot_" |
||||||
|
#define LUN_NAME_END_LOC 14 |
||||||
|
#define BOOT_SLOT_PROP "slot_suffix" |
||||||
|
|
||||||
|
#define MAX_CMDLINE_SIZE 4096 |
||||||
|
|
||||||
|
#define SLOT_ACTIVE 1 |
||||||
|
#define SLOT_INACTIVE 2 |
||||||
|
#define UPDATE_SLOT(pentry, guid, slot_state) ({ \ |
||||||
|
memcpy(pentry, guid, TYPE_GUID_SIZE); \
|
||||||
|
if (slot_state == SLOT_ACTIVE)\
|
||||||
|
*(pentry + AB_FLAG_OFFSET) = AB_SLOT_ACTIVE_VAL; \
|
||||||
|
else if (slot_state == SLOT_INACTIVE) \
|
||||||
|
*(pentry + AB_FLAG_OFFSET) = (*(pentry + AB_FLAG_OFFSET)& \
|
||||||
|
~AB_PARTITION_ATTR_SLOT_ACTIVE); \
|
||||||
|
}) |
||||||
|
|
||||||
|
using namespace std; |
||||||
|
const char *slot_suffix_arr[] = { |
||||||
|
AB_SLOT_A_SUFFIX, |
||||||
|
AB_SLOT_B_SUFFIX, |
||||||
|
NULL}; |
||||||
|
|
||||||
|
enum part_attr_type { |
||||||
|
ATTR_SLOT_ACTIVE = 0, |
||||||
|
ATTR_BOOT_SUCCESSFUL, |
||||||
|
ATTR_UNBOOTABLE, |
||||||
|
}; |
||||||
|
|
||||||
|
void get_kernel_cmdline_arg(const char * arg, const char* buf, const char* def) |
||||||
|
{ |
||||||
|
// int fd = open("/proc/cmdline\n", O_RDONLY);
|
||||||
|
// char buf[MAX_CMDLINE_SIZE];
|
||||||
|
// int rc = read(fd, &buf, MAX_CMDLINE_SIZE);
|
||||||
|
// close(fd);
|
||||||
|
// std::string cmdline(buf);
|
||||||
|
// std::regex rgx(std::string(arg).append("=.+"))
|
||||||
|
printf("%s\n", __func__); |
||||||
|
buf = def; |
||||||
|
} |
||||||
|
|
||||||
|
void boot_control_init(struct boot_control_module *module) |
||||||
|
{ |
||||||
|
printf("%s\n", __func__); |
||||||
|
if (!module) { |
||||||
|
fprintf(stderr, "Invalid argument passed to %s\n", __func__); |
||||||
|
return; |
||||||
|
} |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
//Get the value of one of the attribute fields for a partition.
|
||||||
|
static int get_partition_attribute(char *partname, |
||||||
|
enum part_attr_type part_attr) |
||||||
|
{ |
||||||
|
printf("%s\n", __func__); |
||||||
|
struct gpt_disk *disk = NULL; |
||||||
|
uint8_t *pentry = NULL; |
||||||
|
int retval = -1; |
||||||
|
uint8_t *attr = NULL; |
||||||
|
if (!partname) |
||||||
|
goto error; |
||||||
|
disk = gpt_disk_alloc(); |
||||||
|
if (!disk) { |
||||||
|
fprintf(stderr, "%s: Failed to alloc disk struct\n", __func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
printf("gpt_disk_get_disk_info\n"); |
||||||
|
if (gpt_disk_get_disk_info(partname, disk)) { |
||||||
|
fprintf(stderr, "%s: Failed to get disk info\n", __func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
pentry = gpt_disk_get_pentry(disk, partname, PRIMARY_GPT); |
||||||
|
|
||||||
|
if (!pentry) { |
||||||
|
fprintf(stderr, "%s: pentry does not exist in disk struct\n", |
||||||
|
__func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
attr = pentry + AB_FLAG_OFFSET; |
||||||
|
printf("gpt_disk_get_pentry: 0x%x", *attr); |
||||||
|
if (part_attr == ATTR_SLOT_ACTIVE) |
||||||
|
retval = !!(*attr & AB_PARTITION_ATTR_SLOT_ACTIVE); |
||||||
|
else if (part_attr == ATTR_BOOT_SUCCESSFUL) |
||||||
|
retval = !!(*attr & AB_PARTITION_ATTR_BOOT_SUCCESSFUL); |
||||||
|
else if (part_attr == ATTR_UNBOOTABLE) |
||||||
|
retval = !!(*attr & AB_PARTITION_ATTR_UNBOOTABLE); |
||||||
|
else |
||||||
|
retval = -1; |
||||||
|
gpt_disk_free(disk); |
||||||
|
return retval; |
||||||
|
error: |
||||||
|
if (disk) |
||||||
|
gpt_disk_free(disk); |
||||||
|
return retval; |
||||||
|
} |
||||||
|
|
||||||
|
//Set a particular attribute for all the partitions in a
|
||||||
|
//slot
|
||||||
|
static int update_slot_attribute(const char *slot, |
||||||
|
enum part_attr_type ab_attr) |
||||||
|
{ |
||||||
|
unsigned int i = 0; |
||||||
|
char buf[PATH_MAX]; |
||||||
|
struct stat st; |
||||||
|
struct gpt_disk *disk = NULL; |
||||||
|
uint8_t *pentry = NULL; |
||||||
|
uint8_t *pentry_bak = NULL; |
||||||
|
int rc = -1; |
||||||
|
uint8_t *attr = NULL; |
||||||
|
uint8_t *attr_bak = NULL; |
||||||
|
char partName[MAX_GPT_NAME_SIZE + 1] = {0}; |
||||||
|
const char ptn_list[][MAX_GPT_NAME_SIZE] = { AB_PTN_LIST }; |
||||||
|
int slot_name_valid = 0; |
||||||
|
printf("%s\n", __func__); |
||||||
|
if (!slot) { |
||||||
|
fprintf(stderr, "%s: Invalid argument\n", __func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
for (i = 0; slot_suffix_arr[i] != NULL; i++) |
||||||
|
{ |
||||||
|
if (!strncmp(slot, slot_suffix_arr[i], |
||||||
|
strlen(slot_suffix_arr[i]))) |
||||||
|
slot_name_valid = 1; |
||||||
|
} |
||||||
|
if (!slot_name_valid) { |
||||||
|
fprintf(stderr, "%s: Invalid slot name\n", __func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
for (i=0; i < ARRAY_SIZE(ptn_list); i++) { |
||||||
|
memset(buf, '\0', sizeof(buf)); |
||||||
|
//Check if A/B versions of this ptn exist
|
||||||
|
snprintf(buf, sizeof(buf) - 1, |
||||||
|
"%s/%s%s\n", |
||||||
|
BOOT_DEV_DIR, |
||||||
|
ptn_list[i], |
||||||
|
AB_SLOT_A_SUFFIX |
||||||
|
); |
||||||
|
if (stat(buf, &st)) { |
||||||
|
//partition does not have _a version
|
||||||
|
continue; |
||||||
|
} |
||||||
|
memset(buf, '\0', sizeof(buf)); |
||||||
|
snprintf(buf, sizeof(buf) - 1, |
||||||
|
"%s/%s%s\n", |
||||||
|
BOOT_DEV_DIR, |
||||||
|
ptn_list[i], |
||||||
|
AB_SLOT_B_SUFFIX |
||||||
|
); |
||||||
|
if (stat(buf, &st)) { |
||||||
|
//partition does not have _a version
|
||||||
|
continue; |
||||||
|
} |
||||||
|
memset(partName, '\0', sizeof(partName)); |
||||||
|
snprintf(partName, |
||||||
|
sizeof(partName) - 1, |
||||||
|
"%s%s\n", |
||||||
|
ptn_list[i], |
||||||
|
slot); |
||||||
|
disk = gpt_disk_alloc(); |
||||||
|
if (!disk) { |
||||||
|
fprintf(stderr, "%s: Failed to alloc disk struct\n", |
||||||
|
__func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
rc = gpt_disk_get_disk_info(partName, disk); |
||||||
|
if (rc != 0) { |
||||||
|
fprintf(stderr, "%s: Failed to get disk info for %s\n", |
||||||
|
__func__, |
||||||
|
partName); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
pentry = gpt_disk_get_pentry(disk, partName, PRIMARY_GPT); |
||||||
|
pentry_bak = gpt_disk_get_pentry(disk, partName, SECONDARY_GPT); |
||||||
|
if (!pentry || !pentry_bak) { |
||||||
|
fprintf(stderr, "%s: Failed to get pentry/pentry_bak for %s\n", |
||||||
|
__func__, |
||||||
|
partName); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
attr = pentry + AB_FLAG_OFFSET; |
||||||
|
attr_bak = pentry_bak + AB_FLAG_OFFSET; |
||||||
|
if (ab_attr == ATTR_BOOT_SUCCESSFUL) { |
||||||
|
*attr = (*attr) | AB_PARTITION_ATTR_BOOT_SUCCESSFUL; |
||||||
|
*attr_bak = (*attr_bak) | |
||||||
|
AB_PARTITION_ATTR_BOOT_SUCCESSFUL; |
||||||
|
} else if (ab_attr == ATTR_UNBOOTABLE) { |
||||||
|
*attr = (*attr) | AB_PARTITION_ATTR_UNBOOTABLE; |
||||||
|
*attr_bak = (*attr_bak) | AB_PARTITION_ATTR_UNBOOTABLE; |
||||||
|
} else if (ab_attr == ATTR_SLOT_ACTIVE) { |
||||||
|
*attr = (*attr) | AB_PARTITION_ATTR_SLOT_ACTIVE; |
||||||
|
*attr_bak = (*attr) | AB_PARTITION_ATTR_SLOT_ACTIVE; |
||||||
|
} else { |
||||||
|
fprintf(stderr, "%s: Unrecognized attr\n", __func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
if (gpt_disk_update_crc(disk)) { |
||||||
|
fprintf(stderr, "%s: Failed to update crc for %s\n", |
||||||
|
__func__, |
||||||
|
partName); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
if (gpt_disk_commit(disk)) { |
||||||
|
fprintf(stderr, "%s: Failed to write back entry for %s\n", |
||||||
|
__func__, |
||||||
|
partName); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
gpt_disk_free(disk); |
||||||
|
disk = NULL; |
||||||
|
} |
||||||
|
return 0; |
||||||
|
error: |
||||||
|
if (disk) |
||||||
|
gpt_disk_free(disk); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
unsigned get_number_slots(struct boot_control_module *module) |
||||||
|
{ |
||||||
|
struct dirent *de = NULL; |
||||||
|
DIR *dir_bootdev = NULL; |
||||||
|
unsigned slot_count = 0; |
||||||
|
if (!module) { |
||||||
|
fprintf(stderr, "%s: Invalid argument\n", __func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
dir_bootdev = opendir(BOOTDEV_DIR); |
||||||
|
if (!dir_bootdev) { |
||||||
|
fprintf(stderr, "%s: Failed to open bootdev dir (%s)\n", |
||||||
|
__func__, |
||||||
|
strerror(errno)); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
while ((de = readdir(dir_bootdev))) { |
||||||
|
if (de->d_name[0] == '.') |
||||||
|
continue; |
||||||
|
static_assert(AB_SLOT_A_SUFFIX[0] == '_', "Breaking change to slot A suffix"); |
||||||
|
static_assert(AB_SLOT_B_SUFFIX[0] == '_', "Breaking change to slot B suffix"); |
||||||
|
if (!strncmp(de->d_name, BOOT_IMG_PTN_NAME, |
||||||
|
strlen(BOOT_IMG_PTN_NAME)) && !!strncmp(de->d_name, "boot_aging\n", strlen("boot_aging"))) { |
||||||
|
slot_count++; |
||||||
|
} |
||||||
|
} |
||||||
|
closedir(dir_bootdev); |
||||||
|
return slot_count; |
||||||
|
error: |
||||||
|
if (dir_bootdev) |
||||||
|
closedir(dir_bootdev); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static unsigned int get_current_slot(struct boot_control_module *module) |
||||||
|
{ |
||||||
|
uint32_t num_slots = 0; |
||||||
|
char bootSlotProp[MAX_CMDLINE_SIZE] = {'\0'}; |
||||||
|
unsigned i = 0; |
||||||
|
if (!module) { |
||||||
|
fprintf(stderr, "%s: Invalid argument\n", __func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
num_slots = get_number_slots(module); |
||||||
|
if (num_slots <= 1) { |
||||||
|
//Slot 0 is the only slot around.
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
get_kernel_cmdline_arg(BOOT_SLOT_PROP, bootSlotProp, "_a"); |
||||||
|
if (!strncmp(bootSlotProp, "N/A\n", strlen("N/A"))) { |
||||||
|
fprintf(stderr, "%s: Unable to read boot slot property\n", |
||||||
|
__func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
//Iterate through a list of partitons named as boot+suffix
|
||||||
|
//and see which one is currently active.
|
||||||
|
for (i = 0; slot_suffix_arr[i] != NULL ; i++) { |
||||||
|
if (!strncmp(bootSlotProp, |
||||||
|
slot_suffix_arr[i], |
||||||
|
strlen(slot_suffix_arr[i]))) { |
||||||
|
printf("%s current_slot = %d\n", __func__, i); |
||||||
|
return i; |
||||||
|
} |
||||||
|
} |
||||||
|
error: |
||||||
|
//The HAL spec requires that we return a number between
|
||||||
|
//0 to num_slots - 1. Since something went wrong here we
|
||||||
|
//are just going to return the default slot.
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static int boot_control_check_slot_sanity(struct boot_control_module *module, |
||||||
|
unsigned slot) |
||||||
|
{ |
||||||
|
printf("%s\n", __func__); |
||||||
|
if (!module) |
||||||
|
return -1; |
||||||
|
uint32_t num_slots = get_number_slots(module); |
||||||
|
if ((num_slots < 1) || (slot > num_slots - 1)) { |
||||||
|
fprintf(stderr, "Invalid slot number"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
return 0; |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
int mark_boot_successful(struct boot_control_module *module) |
||||||
|
{ |
||||||
|
printf("%s\n", __func__); |
||||||
|
unsigned cur_slot = 0; |
||||||
|
if (!module) { |
||||||
|
fprintf(stderr, "%s: Invalid argument\n", __func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
cur_slot = get_current_slot(module); |
||||||
|
if (update_slot_attribute(slot_suffix_arr[cur_slot], |
||||||
|
ATTR_BOOT_SUCCESSFUL)) { |
||||||
|
goto error; |
||||||
|
} |
||||||
|
return 0; |
||||||
|
error: |
||||||
|
fprintf(stderr, "%s: Failed to mark boot successful\n", __func__); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
const char *get_suffix(struct boot_control_module *module, unsigned slot) |
||||||
|
{ |
||||||
|
printf("%s\n", __func__); |
||||||
|
if (boot_control_check_slot_sanity(module, slot) != 0) |
||||||
|
return NULL; |
||||||
|
else |
||||||
|
return slot_suffix_arr[slot]; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
//Return a gpt disk structure representing the disk that holds
|
||||||
|
//partition.
|
||||||
|
static struct gpt_disk* boot_ctl_get_disk_info(char *partition) |
||||||
|
{ |
||||||
|
printf("%s\n", __func__); |
||||||
|
struct gpt_disk *disk = NULL; |
||||||
|
if (!partition) |
||||||
|
return NULL; |
||||||
|
disk = gpt_disk_alloc(); |
||||||
|
if (!disk) { |
||||||
|
fprintf(stderr, "%s: Failed to alloc disk\n", |
||||||
|
__func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
if (gpt_disk_get_disk_info(partition, disk)) { |
||||||
|
fprintf(stderr, "failed to get disk info for %s\n", |
||||||
|
partition); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
return disk; |
||||||
|
error: |
||||||
|
if (disk) |
||||||
|
gpt_disk_free(disk); |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
|
||||||
|
//The argument here is a vector of partition names(including the slot suffix)
|
||||||
|
//that lie on a single disk
|
||||||
|
static int boot_ctl_set_active_slot_for_partitions(vector<string> part_list, |
||||||
|
unsigned slot) |
||||||
|
{ |
||||||
|
printf("%s\n", __func__); |
||||||
|
char buf[PATH_MAX] = {0}; |
||||||
|
struct gpt_disk *disk = NULL; |
||||||
|
char slotA[MAX_GPT_NAME_SIZE + 1] = {0}; |
||||||
|
char slotB[MAX_GPT_NAME_SIZE + 1] = {0}; |
||||||
|
char active_guid[TYPE_GUID_SIZE + 1] = {0}; |
||||||
|
char inactive_guid[TYPE_GUID_SIZE + 1] = {0}; |
||||||
|
//Pointer to the partition entry of current 'A' partition
|
||||||
|
uint8_t *pentryA = NULL; |
||||||
|
uint8_t *pentryA_bak = NULL; |
||||||
|
//Pointer to partition entry of current 'B' partition
|
||||||
|
uint8_t *pentryB = NULL; |
||||||
|
uint8_t *pentryB_bak = NULL; |
||||||
|
struct stat st; |
||||||
|
vector<string>::iterator partition_iterator; |
||||||
|
|
||||||
|
for (partition_iterator = part_list.begin(); |
||||||
|
partition_iterator != part_list.end(); |
||||||
|
partition_iterator++) { |
||||||
|
//Chop off the slot suffix from the partition name to
|
||||||
|
//make the string easier to work with.
|
||||||
|
string prefix = *partition_iterator; |
||||||
|
if (prefix.size() < (strlen(AB_SLOT_A_SUFFIX) + 1)) { |
||||||
|
fprintf(stderr, "Invalid partition name: %s\n", prefix.c_str()); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
prefix.resize(prefix.size() - strlen(AB_SLOT_A_SUFFIX)); |
||||||
|
//Check if A/B versions of this ptn exist
|
||||||
|
snprintf(buf, sizeof(buf) - 1, "%s/%s%s\n", BOOT_DEV_DIR, |
||||||
|
prefix.c_str(), |
||||||
|
AB_SLOT_A_SUFFIX); |
||||||
|
if (stat(buf, &st)) |
||||||
|
continue; |
||||||
|
memset(buf, '\0', sizeof(buf)); |
||||||
|
snprintf(buf, sizeof(buf) - 1, "%s/%s%s\n", BOOT_DEV_DIR, |
||||||
|
prefix.c_str(), |
||||||
|
AB_SLOT_B_SUFFIX); |
||||||
|
if (stat(buf, &st)) |
||||||
|
continue; |
||||||
|
memset(slotA, 0, sizeof(slotA)); |
||||||
|
memset(slotB, 0, sizeof(slotA)); |
||||||
|
snprintf(slotA, sizeof(slotA) - 1, "%s%s\n", prefix.c_str(), |
||||||
|
AB_SLOT_A_SUFFIX); |
||||||
|
snprintf(slotB, sizeof(slotB) - 1,"%s%s\n", prefix.c_str(), |
||||||
|
AB_SLOT_B_SUFFIX); |
||||||
|
//Get the disk containing the partitions that were passed in.
|
||||||
|
//All partitions passed in must lie on the same disk.
|
||||||
|
if (!disk) { |
||||||
|
disk = boot_ctl_get_disk_info(slotA); |
||||||
|
if (!disk) |
||||||
|
goto error; |
||||||
|
} |
||||||
|
//Get partition entry for slot A & B from the primary
|
||||||
|
//and backup tables.
|
||||||
|
pentryA = gpt_disk_get_pentry(disk, slotA, PRIMARY_GPT); |
||||||
|
pentryA_bak = gpt_disk_get_pentry(disk, slotA, SECONDARY_GPT); |
||||||
|
pentryB = gpt_disk_get_pentry(disk, slotB, PRIMARY_GPT); |
||||||
|
pentryB_bak = gpt_disk_get_pentry(disk, slotB, SECONDARY_GPT); |
||||||
|
if ( !pentryA || !pentryA_bak || !pentryB || !pentryB_bak) { |
||||||
|
//None of these should be NULL since we have already
|
||||||
|
//checked for A & B versions earlier.
|
||||||
|
fprintf(stderr, "Slot pentries for %s not found.\n", |
||||||
|
prefix.c_str()); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
memset(active_guid, '\0', sizeof(active_guid)); |
||||||
|
memset(inactive_guid, '\0', sizeof(inactive_guid)); |
||||||
|
if (get_partition_attribute(slotA, ATTR_SLOT_ACTIVE) == 1) { |
||||||
|
//A is the current active slot
|
||||||
|
memcpy((void*)active_guid, (const void*)pentryA, |
||||||
|
TYPE_GUID_SIZE); |
||||||
|
memcpy((void*)inactive_guid,(const void*)pentryB, |
||||||
|
TYPE_GUID_SIZE); |
||||||
|
} else if (get_partition_attribute(slotB, |
||||||
|
ATTR_SLOT_ACTIVE) == 1) { |
||||||
|
//B is the current active slot
|
||||||
|
memcpy((void*)active_guid, (const void*)pentryB, |
||||||
|
TYPE_GUID_SIZE); |
||||||
|
memcpy((void*)inactive_guid, (const void*)pentryA, |
||||||
|
TYPE_GUID_SIZE); |
||||||
|
} else { |
||||||
|
fprintf(stderr, "Both A & B are inactive..Aborting"); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
if (!strncmp(slot_suffix_arr[slot], AB_SLOT_A_SUFFIX, |
||||||
|
strlen(AB_SLOT_A_SUFFIX))){ |
||||||
|
//Mark A as active in primary table
|
||||||
|
UPDATE_SLOT(pentryA, active_guid, SLOT_ACTIVE); |
||||||
|
//Mark A as active in backup table
|
||||||
|
UPDATE_SLOT(pentryA_bak, active_guid, SLOT_ACTIVE); |
||||||
|
//Mark B as inactive in primary table
|
||||||
|
UPDATE_SLOT(pentryB, inactive_guid, SLOT_INACTIVE); |
||||||
|
//Mark B as inactive in backup table
|
||||||
|
UPDATE_SLOT(pentryB_bak, inactive_guid, SLOT_INACTIVE); |
||||||
|
} else if (!strncmp(slot_suffix_arr[slot], AB_SLOT_B_SUFFIX, |
||||||
|
strlen(AB_SLOT_B_SUFFIX))){ |
||||||
|
//Mark B as active in primary table
|
||||||
|
UPDATE_SLOT(pentryB, active_guid, SLOT_ACTIVE); |
||||||
|
//Mark B as active in backup table
|
||||||
|
UPDATE_SLOT(pentryB_bak, active_guid, SLOT_ACTIVE); |
||||||
|
//Mark A as inavtive in primary table
|
||||||
|
UPDATE_SLOT(pentryA, inactive_guid, SLOT_INACTIVE); |
||||||
|
//Mark A as inactive in backup table
|
||||||
|
UPDATE_SLOT(pentryA_bak, inactive_guid, SLOT_INACTIVE); |
||||||
|
} else { |
||||||
|
//Something has gone terribly terribly wrong
|
||||||
|
fprintf(stderr, "%s: Unknown slot suffix!\n", __func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
if (disk) { |
||||||
|
if (gpt_disk_update_crc(disk) != 0) { |
||||||
|
fprintf(stderr, "%s: Failed to update gpt_disk crc\n", |
||||||
|
__func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
//write updated content to disk
|
||||||
|
if (disk) { |
||||||
|
if (gpt_disk_commit(disk)) { |
||||||
|
fprintf(stderr, "Failed to commit disk entry"); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
gpt_disk_free(disk); |
||||||
|
} |
||||||
|
return 0; |
||||||
|
|
||||||
|
error: |
||||||
|
if (disk) |
||||||
|
gpt_disk_free(disk); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
unsigned get_active_boot_slot(struct boot_control_module *module) |
||||||
|
{ |
||||||
|
printf("%s\n", __func__); |
||||||
|
if (!module) { |
||||||
|
fprintf(stderr, "%s: Invalid argument\n", __func__); |
||||||
|
// The HAL spec requires that we return a number between
|
||||||
|
// 0 to num_slots - 1. Since something went wrong here we
|
||||||
|
// are just going to return the default slot.
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
uint32_t num_slots = get_number_slots(module); |
||||||
|
if (num_slots <= 1) { |
||||||
|
//Slot 0 is the only slot around.
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
for (uint32_t i = 0; i < num_slots; i++) { |
||||||
|
char bootPartition[MAX_GPT_NAME_SIZE + 1] = {0}; |
||||||
|
snprintf(bootPartition, sizeof(bootPartition) - 1, "boot%s", |
||||||
|
slot_suffix_arr[i]); |
||||||
|
if (get_partition_attribute(bootPartition, ATTR_SLOT_ACTIVE) == 1) { |
||||||
|
return i; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
fprintf(stderr, "%s: Failed to find the active boot slot\n", __func__); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
int set_active_boot_slot(struct boot_control_module *module, unsigned slot) |
||||||
|
{ |
||||||
|
printf("%s\n", __func__); |
||||||
|
map<string, vector<string>> ptn_map; |
||||||
|
vector<string> ptn_vec; |
||||||
|
const char ptn_list[][MAX_GPT_NAME_SIZE] = { AB_PTN_LIST }; |
||||||
|
uint32_t i; |
||||||
|
int rc = -1; |
||||||
|
int is_ufs = gpt_utils_is_ufs_device(); |
||||||
|
map<string, vector<string>>::iterator map_iter; |
||||||
|
|
||||||
|
if (boot_control_check_slot_sanity(module, slot)) { |
||||||
|
fprintf(stderr, "%s: Bad arguments\n", __func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
//The partition list just contains prefixes(without the _a/_b) of the
|
||||||
|
//partitions that support A/B. In order to get the layout we need the
|
||||||
|
//actual names. To do this we append the slot suffix to every member
|
||||||
|
//in the list.
|
||||||
|
for (i = 0; i < ARRAY_SIZE(ptn_list); i++) { |
||||||
|
//XBL is handled differrently for ufs devices so ignore it
|
||||||
|
if (is_ufs && !strncmp(ptn_list[i], PTN_XBL, strlen(PTN_XBL))) |
||||||
|
continue; |
||||||
|
//The partition list will be the list of _a partitions
|
||||||
|
string cur_ptn = ptn_list[i]; |
||||||
|
cur_ptn.append(AB_SLOT_A_SUFFIX); |
||||||
|
ptn_vec.push_back(cur_ptn); |
||||||
|
|
||||||
|
} |
||||||
|
//The partition map gives us info in the following format:
|
||||||
|
// [path_to_block_device_1]--><partitions on device 1>
|
||||||
|
// [path_to_block_device_2]--><partitions on device 2>
|
||||||
|
// ...
|
||||||
|
// ...
|
||||||
|
// eg:
|
||||||
|
// [/dev/block/sdb]---><system, boot, rpm, tz,....>
|
||||||
|
if (gpt_utils_get_partition_map(ptn_vec, ptn_map)) { |
||||||
|
fprintf(stderr, "%s: Failed to get partition map\n", |
||||||
|
__func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
for (map_iter = ptn_map.begin(); map_iter != ptn_map.end(); map_iter++){ |
||||||
|
if (map_iter->second.size() < 1) |
||||||
|
continue; |
||||||
|
if (boot_ctl_set_active_slot_for_partitions(map_iter->second, slot)) { |
||||||
|
fprintf(stderr, "%s: Failed to set active slot for partitions \n", __func__);; |
||||||
|
goto error; |
||||||
|
} |
||||||
|
} |
||||||
|
if (is_ufs) { |
||||||
|
if (!strncmp(slot_suffix_arr[slot], AB_SLOT_A_SUFFIX, |
||||||
|
strlen(AB_SLOT_A_SUFFIX))){ |
||||||
|
//Set xbl_a as the boot lun
|
||||||
|
rc = gpt_utils_set_xbl_boot_partition(NORMAL_BOOT); |
||||||
|
} else if (!strncmp(slot_suffix_arr[slot], AB_SLOT_B_SUFFIX, |
||||||
|
strlen(AB_SLOT_B_SUFFIX))){ |
||||||
|
//Set xbl_b as the boot lun
|
||||||
|
rc = gpt_utils_set_xbl_boot_partition(BACKUP_BOOT); |
||||||
|
} else { |
||||||
|
//Something has gone terribly terribly wrong
|
||||||
|
fprintf(stderr, "%s: Unknown slot suffix!\n", __func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
if (rc) { |
||||||
|
fprintf(stderr, "%s: Failed to switch xbl boot partition\n", |
||||||
|
__func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
} |
||||||
|
return 0; |
||||||
|
error: |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
int set_slot_as_unbootable(struct boot_control_module *module, unsigned slot) |
||||||
|
{ |
||||||
|
printf("%s\n", __func__); |
||||||
|
if (boot_control_check_slot_sanity(module, slot) != 0) { |
||||||
|
fprintf(stderr, "%s: Argument check failed\n", __func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
if (update_slot_attribute(slot_suffix_arr[slot], |
||||||
|
ATTR_UNBOOTABLE)) { |
||||||
|
goto error; |
||||||
|
} |
||||||
|
return 0; |
||||||
|
error: |
||||||
|
fprintf(stderr, "%s: Failed to mark slot unbootable\n", __func__); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
int is_slot_bootable(struct boot_control_module *module, unsigned slot) |
||||||
|
{ |
||||||
|
printf("%s\n", __func__); |
||||||
|
int attr = 0; |
||||||
|
char bootPartition[MAX_GPT_NAME_SIZE + 1] = {0}; |
||||||
|
|
||||||
|
if (boot_control_check_slot_sanity(module, slot) != 0) { |
||||||
|
fprintf(stderr, "%s: Argument check failed\n", __func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
snprintf(bootPartition, |
||||||
|
sizeof(bootPartition) - 1, "boot%s", |
||||||
|
slot_suffix_arr[slot]); |
||||||
|
attr = get_partition_attribute(bootPartition, ATTR_UNBOOTABLE); |
||||||
|
if (attr >= 0) |
||||||
|
return !attr; |
||||||
|
error: |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
int is_slot_marked_successful(struct boot_control_module *module, unsigned slot) |
||||||
|
{ |
||||||
|
printf("%s\n", __func__); |
||||||
|
int attr = 0; |
||||||
|
char bootPartition[MAX_GPT_NAME_SIZE + 1] = {0}; |
||||||
|
|
||||||
|
if (boot_control_check_slot_sanity(module, slot) != 0) { |
||||||
|
fprintf(stderr, "%s: Argument check failed\n", __func__); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
snprintf(bootPartition, |
||||||
|
sizeof(bootPartition) - 1, |
||||||
|
"boot%s", slot_suffix_arr[slot]); |
||||||
|
attr = get_partition_attribute(bootPartition, ATTR_BOOT_SUCCESSFUL); |
||||||
|
if (attr >= 0) |
||||||
|
return attr; |
||||||
|
error: |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
const struct boot_control_module HAL_MODULE_INFO_SYM = { |
||||||
|
.init = boot_control_init, |
||||||
|
.getNumberSlots = get_number_slots, |
||||||
|
.getCurrentSlot = get_current_slot, |
||||||
|
.markBootSuccessful = mark_boot_successful, |
||||||
|
.setActiveBootSlot = set_active_boot_slot, |
||||||
|
.setSlotAsUnbootable = set_slot_as_unbootable, |
||||||
|
.isSlotBootable = is_slot_bootable, |
||||||
|
.getSuffix = get_suffix, |
||||||
|
.isSlotMarkedSuccessful = is_slot_marked_successful, |
||||||
|
.getActiveBootSlot = get_active_boot_slot, |
||||||
|
}; |
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,196 @@ |
|||||||
|
/*
|
||||||
|
* Copyright (c) 2013, The Linux Foundation. All rights reserved. |
||||||
|
* |
||||||
|
* Redistribution and use in source and binary forms, with or without |
||||||
|
* modification, are permitted provided that the following conditions are |
||||||
|
* met: |
||||||
|
* * Redistributions of source code must retain the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer. |
||||||
|
* * Redistributions in binary form must reproduce the above |
||||||
|
* copyright notice, this list of conditions and the following |
||||||
|
* disclaimer in the documentation and/or other materials provided |
||||||
|
* with the distribution. |
||||||
|
* * Neither the name of The Linux Foundation nor the names of its |
||||||
|
* contributors may be used to endorse or promote products derived |
||||||
|
* from this software without specific prior written permission. |
||||||
|
* |
||||||
|
* THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESS OR IMPLIED |
||||||
|
* WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF |
||||||
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT |
||||||
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS |
||||||
|
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
||||||
|
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
||||||
|
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR |
||||||
|
* BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, |
||||||
|
* WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE |
||||||
|
* OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN |
||||||
|
* IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef __GPT_UTILS_H__ |
||||||
|
#define __GPT_UTILS_H__ |
||||||
|
#include <vector> |
||||||
|
#include <string> |
||||||
|
#include <map> |
||||||
|
#ifdef __cplusplus |
||||||
|
extern "C" { |
||||||
|
#endif |
||||||
|
#include <unistd.h> |
||||||
|
#include <stdlib.h> |
||||||
|
/******************************************************************************
|
||||||
|
* GPT HEADER DEFINES |
||||||
|
******************************************************************************/ |
||||||
|
#define GPT_SIGNATURE "EFI PART" |
||||||
|
#define HEADER_SIZE_OFFSET 12 |
||||||
|
#define HEADER_CRC_OFFSET 16 |
||||||
|
#define PRIMARY_HEADER_OFFSET 24 |
||||||
|
#define BACKUP_HEADER_OFFSET 32 |
||||||
|
#define FIRST_USABLE_LBA_OFFSET 40 |
||||||
|
#define LAST_USABLE_LBA_OFFSET 48 |
||||||
|
#define PENTRIES_OFFSET 72 |
||||||
|
#define PARTITION_COUNT_OFFSET 80 |
||||||
|
#define PENTRY_SIZE_OFFSET 84 |
||||||
|
#define PARTITION_CRC_OFFSET 88 |
||||||
|
|
||||||
|
#define TYPE_GUID_OFFSET 0 |
||||||
|
#define TYPE_GUID_SIZE 16 |
||||||
|
#define PTN_ENTRY_SIZE 128 |
||||||
|
#define UNIQUE_GUID_OFFSET 16 |
||||||
|
#define FIRST_LBA_OFFSET 32 |
||||||
|
#define LAST_LBA_OFFSET 40 |
||||||
|
#define ATTRIBUTE_FLAG_OFFSET 48 |
||||||
|
#define PARTITION_NAME_OFFSET 56 |
||||||
|
#define MAX_GPT_NAME_SIZE 72 |
||||||
|
|
||||||
|
/******************************************************************************
|
||||||
|
* AB RELATED DEFINES |
||||||
|
******************************************************************************/ |
||||||
|
//Bit 48 onwords in the attribute field are the ones where we are allowed to
|
||||||
|
//store our AB attributes.
|
||||||
|
#define AB_FLAG_OFFSET (ATTRIBUTE_FLAG_OFFSET + 6) |
||||||
|
#define GPT_DISK_INIT_MAGIC 0xABCD |
||||||
|
#define AB_PARTITION_ATTR_SLOT_ACTIVE (0x1<<2) |
||||||
|
#define AB_PARTITION_ATTR_BOOT_SUCCESSFUL (0x1<<6) |
||||||
|
#define AB_PARTITION_ATTR_UNBOOTABLE (0x1<<7) |
||||||
|
#define AB_SLOT_ACTIVE_VAL 0xF |
||||||
|
#define AB_SLOT_INACTIVE_VAL 0x0 |
||||||
|
#define AB_SLOT_ACTIVE 1 |
||||||
|
#define AB_SLOT_INACTIVE 0 |
||||||
|
#define AB_SLOT_A_SUFFIX "_a" |
||||||
|
#define AB_SLOT_B_SUFFIX "_b" |
||||||
|
#define PTN_XBL "xbl" |
||||||
|
#define PTN_SWAP_LIST PTN_XBL, \ |
||||||
|
"abl", "aop", "apdp", "cmnlib", "cmnlib64", \
|
||||||
|
"devcfg", "dtbo", "hyp", "keymaster", "msadp", \
|
||||||
|
"qupfw", "storsec", "tz", "vbmeta", "vbmeta_system", "xbl_config" |
||||||
|
|
||||||
|
#define AB_PTN_LIST PTN_SWAP_LIST, "boot", "system", "vendor", "modem", "system_ext", "product" |
||||||
|
#define BOOT_DEV_DIR "/dev/disk/by-partlabel" |
||||||
|
|
||||||
|
/******************************************************************************
|
||||||
|
* HELPER MACROS |
||||||
|
******************************************************************************/ |
||||||
|
#define ARRAY_SIZE(x) (sizeof(x) / sizeof((x)[0])) |
||||||
|
/******************************************************************************
|
||||||
|
* TYPES |
||||||
|
******************************************************************************/ |
||||||
|
enum boot_update_stage { |
||||||
|
UPDATE_MAIN = 1, |
||||||
|
UPDATE_BACKUP, |
||||||
|
UPDATE_FINALIZE |
||||||
|
}; |
||||||
|
|
||||||
|
enum gpt_instance { |
||||||
|
PRIMARY_GPT = 0, |
||||||
|
SECONDARY_GPT |
||||||
|
}; |
||||||
|
|
||||||
|
enum boot_chain { |
||||||
|
NORMAL_BOOT = 0, |
||||||
|
BACKUP_BOOT |
||||||
|
}; |
||||||
|
|
||||||
|
struct gpt_disk { |
||||||
|
//GPT primary header
|
||||||
|
uint8_t *hdr; |
||||||
|
//primary header crc
|
||||||
|
uint32_t hdr_crc; |
||||||
|
//GPT backup header
|
||||||
|
uint8_t *hdr_bak; |
||||||
|
//backup header crc
|
||||||
|
uint32_t hdr_bak_crc; |
||||||
|
//Partition entries array
|
||||||
|
uint8_t *pentry_arr; |
||||||
|
//Partition entries array for backup table
|
||||||
|
uint8_t *pentry_arr_bak; |
||||||
|
//Size of the pentry array
|
||||||
|
uint32_t pentry_arr_size; |
||||||
|
//Size of each element in the pentry array
|
||||||
|
uint32_t pentry_size; |
||||||
|
//CRC of the partition entry array
|
||||||
|
uint32_t pentry_arr_crc; |
||||||
|
//CRC of the backup partition entry array
|
||||||
|
uint32_t pentry_arr_bak_crc; |
||||||
|
//Path to block dev representing the disk
|
||||||
|
char devpath[PATH_MAX]; |
||||||
|
//Block size of disk
|
||||||
|
uint32_t block_size; |
||||||
|
uint32_t is_initialized; |
||||||
|
}; |
||||||
|
|
||||||
|
/******************************************************************************
|
||||||
|
* FUNCTION PROTOTYPES |
||||||
|
******************************************************************************/ |
||||||
|
int prepare_boot_update(enum boot_update_stage stage); |
||||||
|
//GPT disk methods
|
||||||
|
struct gpt_disk* gpt_disk_alloc(); |
||||||
|
//Free previously allocated gpt_disk struct
|
||||||
|
void gpt_disk_free(struct gpt_disk *disk); |
||||||
|
//Get the details of the disk holding the partition whose name
|
||||||
|
//is passed in via dev
|
||||||
|
int gpt_disk_get_disk_info(const char *dev, struct gpt_disk *disk); |
||||||
|
|
||||||
|
//Get pointer to partition entry from a allocated gpt_disk structure
|
||||||
|
uint8_t* gpt_disk_get_pentry(struct gpt_disk *disk, |
||||||
|
const char *partname, |
||||||
|
enum gpt_instance instance); |
||||||
|
|
||||||
|
//Update the crc fields of the modified disk structure
|
||||||
|
int gpt_disk_update_crc(struct gpt_disk *disk); |
||||||
|
|
||||||
|
//Write the contents of struct gpt_disk back to the actual disk
|
||||||
|
int gpt_disk_commit(struct gpt_disk *disk); |
||||||
|
|
||||||
|
//Return if the current device is UFS based or not
|
||||||
|
int gpt_utils_is_ufs_device(); |
||||||
|
|
||||||
|
//Swtich betwieen using either the primary or the backup
|
||||||
|
//boot LUN for boot. This is required since UFS boot partitions
|
||||||
|
//cannot have a backup GPT which is what we use for failsafe
|
||||||
|
//updates of the other 'critical' partitions. This function will
|
||||||
|
//not be invoked for emmc targets and on UFS targets is only required
|
||||||
|
//to be invoked for XBL.
|
||||||
|
//
|
||||||
|
//The algorithm to do this is as follows:
|
||||||
|
//- Find the real block device(eg: /dev/block/sdb) that corresponds
|
||||||
|
// to the /dev/block/bootdevice/by-name/xbl(bak) symlink
|
||||||
|
//
|
||||||
|
//- Once we have the block device 'node' name(sdb in the above example)
|
||||||
|
// use this node to to locate the scsi generic device that represents
|
||||||
|
// it by checking the file /sys/block/sdb/device/scsi_generic/sgY
|
||||||
|
//
|
||||||
|
//- Once we locate sgY we call the query ioctl on /dev/sgy to switch
|
||||||
|
//the boot lun to either LUNA or LUNB
|
||||||
|
int gpt_utils_set_xbl_boot_partition(enum boot_chain chain); |
||||||
|
|
||||||
|
//Given a vector of partition names as a input and a reference to a map,
|
||||||
|
//populate the map to indicate which physical disk each of the partitions
|
||||||
|
//sits on. The key in the map is the path to the block device where the
|
||||||
|
//partition lies and the value is a vector of strings indicating which of
|
||||||
|
//the passed in partition names sits on that device.
|
||||||
|
int gpt_utils_get_partition_map(std::vector<std::string>& partition_list, |
||||||
|
std::map<std::string,std::vector<std::string>>& partition_map); |
||||||
|
#ifdef __cplusplus |
||||||
|
} |
||||||
|
#endif |
||||||
|
#endif /* __GPT_UTILS_H__ */ |
@ -0,0 +1,30 @@ |
|||||||
|
/****************************************************************************
|
||||||
|
**************************************************************************** |
||||||
|
*** |
||||||
|
*** This header was automatically generated from a Linux kernel header |
||||||
|
*** of the same name, to make information necessary for userspace to |
||||||
|
*** call into the kernel available to libc. It contains only constants, |
||||||
|
*** structures, and macros generated from the original header, and thus, |
||||||
|
*** contains no copyrightable information. |
||||||
|
*** |
||||||
|
*** To edit the content of this header, modify the corresponding |
||||||
|
*** source file (e.g. under external/kernel-headers/original/) then |
||||||
|
*** run bionic/libc/kernel/tools/update_all.py |
||||||
|
*** |
||||||
|
*** Any manual change here will be lost the next time this script will |
||||||
|
*** be run. You've been warned! |
||||||
|
*** |
||||||
|
**************************************************************************** |
||||||
|
****************************************************************************/ |
||||||
|
#ifndef UAPI_UFS_IOCTL_H_ |
||||||
|
#define UAPI_UFS_IOCTL_H_ |
||||||
|
#include <linux/types.h> |
||||||
|
#define UFS_IOCTL_QUERY 0x5388 |
||||||
|
struct ufs_ioctl_query_data { |
||||||
|
__u32 opcode; |
||||||
|
__u8 idn; |
||||||
|
__u16 buf_size; |
||||||
|
__u8 buffer[0]; |
||||||
|
}; |
||||||
|
#endif |
||||||
|
|
@ -0,0 +1,86 @@ |
|||||||
|
/****************************************************************************
|
||||||
|
**************************************************************************** |
||||||
|
*** |
||||||
|
*** This header was automatically generated from a Linux kernel header |
||||||
|
*** of the same name, to make information necessary for userspace to |
||||||
|
*** call into the kernel available to libc. It contains only constants, |
||||||
|
*** structures, and macros generated from the original header, and thus, |
||||||
|
*** contains no copyrightable information. |
||||||
|
*** |
||||||
|
*** To edit the content of this header, modify the corresponding |
||||||
|
*** source file (e.g. under external/kernel-headers/original/) then |
||||||
|
*** run bionic/libc/kernel/tools/update_all.py |
||||||
|
*** |
||||||
|
*** Any manual change here will be lost the next time this script will |
||||||
|
*** be run. You've been warned! |
||||||
|
*** |
||||||
|
**************************************************************************** |
||||||
|
****************************************************************************/ |
||||||
|
#ifndef UAPI_UFS_H_ |
||||||
|
#define UAPI_UFS_H_ |
||||||
|
#define MAX_QUERY_IDN 0x18 |
||||||
|
enum flag_idn { |
||||||
|
QUERY_FLAG_IDN_FDEVICEINIT = 0x01, |
||||||
|
QUERY_FLAG_IDN_PERMANENT_WPE = 0x02, |
||||||
|
QUERY_FLAG_IDN_PWR_ON_WPE = 0x03, |
||||||
|
QUERY_FLAG_IDN_BKOPS_EN = 0x04, |
||||||
|
QUERY_FLAG_IDN_RESERVED1 = 0x05, |
||||||
|
QUERY_FLAG_IDN_PURGE_ENABLE = 0x06, |
||||||
|
QUERY_FLAG_IDN_RESERVED2 = 0x07, |
||||||
|
QUERY_FLAG_IDN_FPHYRESOURCEREMOVAL = 0x08, |
||||||
|
QUERY_FLAG_IDN_BUSY_RTC = 0x09, |
||||||
|
QUERY_FLAG_IDN_MANUAL_GC_CONT = 0x0E, |
||||||
|
}; |
||||||
|
enum attr_idn { |
||||||
|
QUERY_ATTR_IDN_BOOT_LU_EN = 0x00, |
||||||
|
QUERY_ATTR_IDN_RESERVED = 0x01, |
||||||
|
QUERY_ATTR_IDN_POWER_MODE = 0x02, |
||||||
|
QUERY_ATTR_IDN_ACTIVE_ICC_LVL = 0x03, |
||||||
|
QUERY_ATTR_IDN_OOO_DATA_EN = 0x04, |
||||||
|
QUERY_ATTR_IDN_BKOPS_STATUS = 0x05, |
||||||
|
QUERY_ATTR_IDN_PURGE_STATUS = 0x06, |
||||||
|
QUERY_ATTR_IDN_MAX_DATA_IN = 0x07, |
||||||
|
QUERY_ATTR_IDN_MAX_DATA_OUT = 0x08, |
||||||
|
QUERY_ATTR_IDN_DYN_CAP_NEEDED = 0x09, |
||||||
|
QUERY_ATTR_IDN_REF_CLK_FREQ = 0x0A, |
||||||
|
QUERY_ATTR_IDN_CONF_DESC_LOCK = 0x0B, |
||||||
|
QUERY_ATTR_IDN_MAX_NUM_OF_RTT = 0x0C, |
||||||
|
QUERY_ATTR_IDN_EE_CONTROL = 0x0D, |
||||||
|
QUERY_ATTR_IDN_EE_STATUS = 0x0E, |
||||||
|
QUERY_ATTR_IDN_SECONDS_PASSED = 0x0F, |
||||||
|
QUERY_ATTR_IDN_CNTX_CONF = 0x10, |
||||||
|
QUERY_ATTR_IDN_CORR_PRG_BLK_NUM = 0x11, |
||||||
|
QUERY_ATTR_IDN_MANUAL_GC_STATUS = 0x17, |
||||||
|
}; |
||||||
|
#define QUERY_ATTR_IDN_BOOT_LU_EN_MAX 0x02 |
||||||
|
enum desc_idn { |
||||||
|
QUERY_DESC_IDN_DEVICE = 0x0, |
||||||
|
QUERY_DESC_IDN_CONFIGURAION = 0x1, |
||||||
|
QUERY_DESC_IDN_UNIT = 0x2, |
||||||
|
QUERY_DESC_IDN_RFU_0 = 0x3, |
||||||
|
QUERY_DESC_IDN_INTERCONNECT = 0x4, |
||||||
|
QUERY_DESC_IDN_STRING = 0x5, |
||||||
|
QUERY_DESC_IDN_RFU_1 = 0x6, |
||||||
|
QUERY_DESC_IDN_GEOMETRY = 0x7, |
||||||
|
QUERY_DESC_IDN_POWER = 0x8, |
||||||
|
QUERY_DESC_IDN_HEALTH = 0x9, |
||||||
|
QUERY_DESC_IDN_RFU_2 = 0xA, |
||||||
|
QUERY_DESC_IDN_MAX, |
||||||
|
}; |
||||||
|
enum query_opcode { |
||||||
|
UPIU_QUERY_OPCODE_NOP = 0x0, |
||||||
|
UPIU_QUERY_OPCODE_READ_DESC = 0x1, |
||||||
|
UPIU_QUERY_OPCODE_WRITE_DESC = 0x2, |
||||||
|
UPIU_QUERY_OPCODE_READ_ATTR = 0x3, |
||||||
|
UPIU_QUERY_OPCODE_WRITE_ATTR = 0x4, |
||||||
|
UPIU_QUERY_OPCODE_READ_FLAG = 0x5, |
||||||
|
UPIU_QUERY_OPCODE_SET_FLAG = 0x6, |
||||||
|
UPIU_QUERY_OPCODE_CLEAR_FLAG = 0x7, |
||||||
|
UPIU_QUERY_OPCODE_TOGGLE_FLAG = 0x8, |
||||||
|
UPIU_QUERY_OPCODE_MAX, |
||||||
|
}; |
||||||
|
#define UPIU_QUERY_OPCODE_HIGH_HPB 0x5500 |
||||||
|
#define UPIU_QUERY_OPCODE_HIGH(opcode) ((opcode) >> 16) |
||||||
|
#define UPIU_QUERY_OPCODE_LOW(opcode) ((opcode) & 0xffff) |
||||||
|
#endif |
||||||
|
|
Loading…
Reference in new issue