mirror of
https://github.com/yurisieucuti/treble_evolution.git
synced 2024-12-05 01:37:02 +00:00
111 lines
3.5 KiB
C++
111 lines
3.5 KiB
C++
#include <string>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <sys/mount.h>
|
|
#include <sys/sysmacros.h>
|
|
#include <sys/xattr.h>
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include <libfiemap/image_manager.h>
|
|
#include <android-base/file.h>
|
|
|
|
using namespace std::chrono_literals;
|
|
using namespace std::string_literals;
|
|
using android::fiemap::IImageManager;
|
|
|
|
|
|
std::string getNextSlot() {
|
|
std::string current_slot;
|
|
std::string next_slot = "a";
|
|
if(android::base::ReadFileToString("/metadata/phh/img", ¤t_slot) &&
|
|
current_slot.c_str()[0] == 'a') {
|
|
next_slot = "b";
|
|
}
|
|
return next_slot;
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
mkdir("/metadata/gsi/phh", 0771);
|
|
chown("/metadata/gsi/phh", 0, 1000);
|
|
mkdir("/data/gsi/phh", 0771);
|
|
chown("/data/gsi/phh", 0, 1000);
|
|
|
|
auto imgManager = IImageManager::Open("phh", 0ms);
|
|
if(argc>=2 && strcmp(argv[1], "unmap") == 0) {
|
|
fprintf(stderr, "Unmapping backing image returned %s\n", imgManager->UnmapImageDevice("system_otaphh_a") ? "true" : "false");
|
|
fprintf(stderr, "Unmapping backing image returned %s\n", imgManager->UnmapImageDevice("system_otaphh_b") ? "true" : "false");
|
|
return 0;
|
|
}
|
|
if(argc>=2 && strcmp(argv[1], "switch-slot") == 0) {
|
|
std::string next_slot = getNextSlot();
|
|
mkdir("/metadata/phh", 0700);
|
|
android::base::WriteStringToFile(next_slot, "/metadata/phh/img");
|
|
return 0;
|
|
}
|
|
if(argc>=2 && strcmp(argv[1], "new-slot") == 0) {
|
|
std::string next_slot = getNextSlot();
|
|
|
|
std::string imageName = "system_otaphh_"s + next_slot;
|
|
|
|
fprintf(stderr, "Unmapping backing image returned %s\n", imgManager->UnmapImageDevice(imageName) ? "true" : "false");
|
|
fprintf(stderr, "Deleting backing image returned %s\n", imgManager->DeleteBackingImage(imageName) ? "true" : "false");
|
|
auto backRes = imgManager->CreateBackingImage(imageName, 4*1024*1024*1024LL, IImageManager::CREATE_IMAGE_DEFAULT, nullptr);
|
|
if(backRes.is_ok()) {
|
|
fprintf(stderr, "Creating system image succeeded\n");
|
|
} else {
|
|
fprintf(stderr, "Creating system image failed\n");
|
|
return -1;
|
|
}
|
|
|
|
std::string blockDev;
|
|
fprintf(stderr, "Mapping backing image returned %s\n", imgManager->MapImageDevice(imageName, 0ms, &blockDev) ? "true" : "false");
|
|
fprintf(stderr, "blockdev is %s\n", blockDev.c_str());
|
|
printf("%s\n", blockDev.c_str());
|
|
|
|
struct stat sb;
|
|
for(int i=0; i<10; i++) {
|
|
if(!stat(blockDev.c_str(), &sb)) break;
|
|
sleep(1);
|
|
}
|
|
|
|
if(!S_ISBLK(sb.st_mode)) {
|
|
fprintf(stderr, "blockDev wasn't block dev\n");
|
|
return -1;
|
|
}
|
|
|
|
unlink("/dev/phh-ota");
|
|
mknod("/dev/phh-ota", 0664 | S_IFBLK, makedev(major(sb.st_rdev), minor(sb.st_rdev)));
|
|
chmod("/dev/phh-ota", 0664);
|
|
// Allow system uid to write there
|
|
chown("/dev/phh-ota", 0, 1000);
|
|
const char *dstContext = "u:object_r:phhota_dev:s0";
|
|
setxattr("/dev/phh-ota", "security.selinux", dstContext, strlen(dstContext), 0);
|
|
|
|
return 0;
|
|
}
|
|
if(argc>=2 && strcmp(argv[1], "delete-other-slot") == 0) {
|
|
const char* current_slot = getenv("PHH_OTA_SLOT");
|
|
if(current_slot == NULL) {
|
|
imgManager->UnmapImageDevice("system_otaphh_a");
|
|
imgManager->DeleteBackingImage("system_otaphh_a");
|
|
imgManager->UnmapImageDevice("system_otaphh_b");
|
|
imgManager->DeleteBackingImage("system_otaphh_b");
|
|
return 0;
|
|
}
|
|
if(current_slot[0] == 'a') {
|
|
imgManager->UnmapImageDevice("system_otaphh_b");
|
|
imgManager->DeleteBackingImage("system_otaphh_b");
|
|
return 0;
|
|
}
|
|
if(current_slot[0] == 'b') {
|
|
imgManager->UnmapImageDevice("system_otaphh_a");
|
|
imgManager->DeleteBackingImage("system_otaphh_a");
|
|
return 0;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
return 1;
|
|
}
|