mirrors/futurerestore
1
0
Fork 0
mirror of https://github.com/tihmstar/futurerestore.git synced 2025-04-20 14:31:41 +00:00
Code Issues Projects Releases Wiki Activity

Nightly.link doesn't support partialzip :(

Browse source
This commit is contained in:
Cryptiiiic 2022-10-08 02:56:22 -07:00
parent 2160321938
commit 93b74bdefe
No known key found for this signature in database
GPG key ID: 6027B509EFE3A76B
2 changed files with 65 additions and 41 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

105
src/futurerestore.cpp
View file

@ -29,6 +29,8 @@ extern "C" {
#include "restore.h"
#include "endianness.h"
#include "tsschecker.h"
#include "download.h"
#include <zip.h>
}
#ifdef safe_mkdir
@ -2387,53 +2389,31 @@ std::string futurerestore::getGeneratorFromSHSH2(plist_t shsh2) {
#ifndef WIN32
void futurerestore::checkForUpdates() {
info("Checking for updates...\n\n");
std::string num = futurerestoreTempPath + "/latest_build_num.txt";
std::string sha = futurerestoreTempPath + "/latest_build_sha.txt";
info("Checking for updates...\n");
std::string url = "https://nightly.link/futurerestore/futurerestore/workflows/ci/main/Versioning.zip";
bool fail = false;
if(downloadPartialzip(url.c_str(), "latest_build_num.txt", num.c_str())) {
char *buffer = nullptr;
uint32_t sz = 0;
if(download_to_buffer(url.c_str(), &buffer, &sz)) {
fail = true;
} else {
std::string num_str = extractZipFileToString(buffer, "latest_build_num.txt", sz);
if(num_str.empty()) {
fail = true;
} else {
this->latest_num = num_str;
}
}
if(downloadPartialzip(url.c_str(), "latest_build_sha.txt", sha.c_str())) {
if(download_to_buffer(url.c_str(), &buffer, &sz)) {
fail = true;
} else {
std::string sha_str = extractZipFileToString(buffer, "latest_build_sha.txt", sz);
if(sha_str.empty()) {
fail = true;
} else {
this->latest_sha = sha_str;
}
}
struct stat st{0};
if(fail || stat(num.c_str(), &st) == -1 || stat(sha.c_str(), &st) == -1) {
info("ERROR: failed to check for futurerestore updates! continuing...\n");
return;
}
char *num_data = nullptr;
size_t num_size = 0;
std::ifstream numFileStream(num, std::ios::binary | std::ios::in | std::ios::ate);
retassure(numFileStream.good(), "%s: failed init file stream for %s!\n", __func__, num.c_str());
num_size = numFileStream.tellg();
numFileStream.seekg(0, std::ios_base::beg);
std::allocator<uint8_t> alloc;
retassure(num_data = (char *)alloc.allocate(num_size),
"%s: failed to allocate memory for %s\n", __func__, num.c_str());
numFileStream.read(reinterpret_cast<char*>(num_data),
num_size);
retassure(numFileStream.good(), "%s: failed to read file stream for %s!\n", __func__, num.c_str());
retassure(*(uint64_t *) (num_data) != 0,
"%s: failed to load num versioning file for %s with the size %zu!\n",
__func__, num.c_str(), num_size);
this->latest_num = std::string(num_data);
char *sha_data = nullptr;
size_t sha_size = 0;
std::ifstream shaFileStream(sha, std::ios::binary | std::ios::in | std::ios::ate);
retassure(shaFileStream.good(), "%s: failed init file stream for %s!\n", __func__, sha.c_str());
sha_size = shaFileStream.tellg();
shaFileStream.seekg(0, std::ios_base::beg);
retassure(sha_data = (char *)alloc.allocate(sha_size),
"%s: failed to allocate memory for %s\n", __func__, sha.c_str());
shaFileStream.read(reinterpret_cast<char*>(sha_data),
sha_size);
retassure(shaFileStream.good(), "%s: failed to read file stream for %s!\n", __func__, sha.c_str());
retassure(*(uint64_t *) (sha_data) != 0,
"%s: failed to load sha versioning file for %s with the size %zu!\n",
__func__, sha.c_str(), sha_size);
this->latest_sha = std::string(sha_data);
if(this->latest_num.empty() || this->latest_sha.empty()) {
info("ERROR: failed to check for futurerestore updates! continuing...\n");
return;
@ -2454,4 +2434,47 @@ void futurerestore::checkForUpdates() {
}
}
std::string futurerestore::extractZipFileToString(char *zip_buffer, const char *file, uint32_t sz) {
bool fail = false;
std::allocator<uint8_t> alloc;
std::string str;
zip_error_t *err = nullptr;
zip_source_t *target_source_zip = zip_source_buffer_create(zip_buffer, sz, 0, err);
zip_t *target_zip = zip_open_from_source(target_source_zip, 0, err);
if(target_zip == nullptr) {
fail = true;
} else {
int idx = zip_name_locate(target_zip, file, 0);
if(idx < 0) {
fail = true;
} else {
struct zip_stat zstat;
zip_stat_init(&zstat);
if(zip_stat_index(target_zip, idx, 0, &zstat) != 0) {
fail = true;
} else {
struct zip_file* target_file = zip_fopen_index(target_zip, idx, 0);
if (target_file == nullptr) {
fail = true;
} else {
unsigned char *target_buffer = (unsigned char*)alloc.allocate(zstat.size+1);
if(target_buffer == nullptr) {
fail = true;
} else {
if (zip_fread(target_file, target_buffer, zstat.size) != zstat.size) {
fail = true;
} else {
target_buffer[zstat.size] = '\0';
str = std::string(reinterpret_cast<const char *>(target_buffer));
}
}
free(target_buffer);
}
zip_fclose(target_file);
}
}
}
return fail ? "" : str;
}
#endif

1
src/futurerestore.hpp
View file

@ -191,6 +191,7 @@ public:
static unsigned char *getDigestOfElementInManifest(const char *element, const char *manifeststr, const char *boardConfig, int isUpdateInstall);
static bool elemExists(const char *element, const char *manifeststr, const char *boardConfig, int isUpdateInstall);
static std::string getGeneratorFromSHSH2(plist_t shsh2);
static std::string extractZipFileToString(char *zip_buffer, const char *file, uint32_t sz);
};