 597ee5b939
			
		
	
	
		597ee5b939
		
			
		
	
	
	
	
		
			
			* Updater: Migrating to new manifest path convention * RPC: Added update preparation status to RPC * RPC: bumped protobuf submodule * Bumped protobuf_version.h * FuriCore: add missing include. Lib: make mlib smaller * Explicitly tell where we have doubles and fix random in animations * makefile: added -DLFS_NO_DEBUG * Updater: path len constant dedup * Updater: checking for hardware version match before parsing manifest * LD: moved _DRIVER_CONTEXT sections to .bss, where they belong. * LD: avoiding PROBGITS warning, moved _CONTEXT to data * Updater: Added version check on update package - refusing to install outdated Co-authored-by: あく <alleteam@gmail.com>
		
			
				
	
	
		
			138 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			138 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| #include "update_task.h"
 | |
| #include "update_task_i.h"
 | |
| 
 | |
| #include <furi.h>
 | |
| #include <furi_hal.h>
 | |
| #include <storage/storage.h>
 | |
| #include <toolbox/path.h>
 | |
| #include <update_util/dfu_file.h>
 | |
| #include <update_util/lfs_backup.h>
 | |
| #include <update_util/update_operation.h>
 | |
| #include <toolbox/tar/tar_archive.h>
 | |
| #include <toolbox/crc32_calc.h>
 | |
| 
 | |
| #define TAG "UpdWorkerBackup"
 | |
| 
 | |
| #define CHECK_RESULT(x) \
 | |
|     if(!(x)) {          \
 | |
|         break;          \
 | |
|     }
 | |
| 
 | |
| #define EXT_PATH "/ext"
 | |
| 
 | |
| static bool update_task_pre_update(UpdateTask* update_task) {
 | |
|     bool success = false;
 | |
|     string_t backup_file_path;
 | |
|     string_init(backup_file_path);
 | |
|     path_concat(
 | |
|         string_get_cstr(update_task->update_path), LFS_BACKUP_DEFAULT_FILENAME, backup_file_path);
 | |
| 
 | |
|     update_task_set_progress(update_task, UpdateTaskStageLfsBackup, 0);
 | |
|     /* to avoid bootloops */
 | |
|     furi_hal_rtc_set_boot_mode(FuriHalRtcBootModeNormal);
 | |
|     if((success = lfs_backup_create(update_task->storage, string_get_cstr(backup_file_path)))) {
 | |
|         furi_hal_rtc_set_boot_mode(FuriHalRtcBootModeUpdate);
 | |
|     }
 | |
| 
 | |
|     string_clear(backup_file_path);
 | |
|     return success;
 | |
| }
 | |
| 
 | |
| typedef struct {
 | |
|     UpdateTask* update_task;
 | |
|     int32_t total_files, processed_files;
 | |
| } TarUnpackProgress;
 | |
| 
 | |
| static bool update_task_resource_unpack_cb(const char* name, bool is_directory, void* context) {
 | |
|     UNUSED(name);
 | |
|     UNUSED(is_directory);
 | |
|     TarUnpackProgress* unpack_progress = context;
 | |
|     unpack_progress->processed_files++;
 | |
|     update_task_set_progress(
 | |
|         unpack_progress->update_task,
 | |
|         UpdateTaskStageProgress,
 | |
|         unpack_progress->processed_files * 100 / (unpack_progress->total_files + 1));
 | |
|     return true;
 | |
| }
 | |
| 
 | |
| static bool update_task_post_update(UpdateTask* update_task) {
 | |
|     bool success = false;
 | |
| 
 | |
|     string_t file_path;
 | |
|     string_init(file_path);
 | |
| 
 | |
|     TarArchive* archive = tar_archive_alloc(update_task->storage);
 | |
|     do {
 | |
|         path_concat(
 | |
|             string_get_cstr(update_task->update_path), LFS_BACKUP_DEFAULT_FILENAME, file_path);
 | |
| 
 | |
|         update_task_set_progress(update_task, UpdateTaskStageLfsRestore, 0);
 | |
|         update_operation_disarm();
 | |
| 
 | |
|         CHECK_RESULT(lfs_backup_unpack(update_task->storage, string_get_cstr(file_path)));
 | |
| 
 | |
|         if(update_task->state.groups & UpdateTaskStageGroupResources) {
 | |
|             TarUnpackProgress progress = {
 | |
|                 .update_task = update_task,
 | |
|                 .total_files = 0,
 | |
|                 .processed_files = 0,
 | |
|             };
 | |
|             update_task_set_progress(update_task, UpdateTaskStageResourcesUpdate, 0);
 | |
| 
 | |
|             path_concat(
 | |
|                 string_get_cstr(update_task->update_path),
 | |
|                 string_get_cstr(update_task->manifest->resource_bundle),
 | |
|                 file_path);
 | |
| 
 | |
|             tar_archive_set_file_callback(archive, update_task_resource_unpack_cb, &progress);
 | |
|             CHECK_RESULT(
 | |
|                 tar_archive_open(archive, string_get_cstr(file_path), TAR_OPEN_MODE_READ));
 | |
| 
 | |
|             progress.total_files = tar_archive_get_entries_count(archive);
 | |
|             if(progress.total_files > 0) {
 | |
|                 CHECK_RESULT(tar_archive_unpack_to(archive, EXT_PATH));
 | |
|             }
 | |
|         }
 | |
|         success = true;
 | |
|     } while(false);
 | |
| 
 | |
|     tar_archive_free(archive);
 | |
|     string_clear(file_path);
 | |
|     return success;
 | |
| }
 | |
| 
 | |
| int32_t update_task_worker_backup_restore(void* context) {
 | |
|     furi_assert(context);
 | |
|     UpdateTask* update_task = context;
 | |
|     bool success = false;
 | |
| 
 | |
|     FuriHalRtcBootMode boot_mode = furi_hal_rtc_get_boot_mode();
 | |
|     if((boot_mode != FuriHalRtcBootModePreUpdate) && (boot_mode != FuriHalRtcBootModePostUpdate)) {
 | |
|         /* no idea how we got here. Clear to normal boot */
 | |
|         update_operation_disarm();
 | |
|         return UPDATE_TASK_NOERR;
 | |
|     }
 | |
| 
 | |
|     if(!update_task_parse_manifest(update_task)) {
 | |
|         return UPDATE_TASK_FAILED;
 | |
|     }
 | |
| 
 | |
|     /* Waiting for BT service to 'start', so we don't race for boot mode flag */
 | |
|     furi_record_open("bt");
 | |
|     furi_record_close("bt");
 | |
| 
 | |
|     if(boot_mode == FuriHalRtcBootModePreUpdate) {
 | |
|         success = update_task_pre_update(update_task);
 | |
|     } else if(boot_mode == FuriHalRtcBootModePostUpdate) {
 | |
|         success = update_task_post_update(update_task);
 | |
|     }
 | |
| 
 | |
|     if(!success) {
 | |
|         update_task_set_progress(update_task, UpdateTaskStageError, 0);
 | |
|         return UPDATE_TASK_FAILED;
 | |
|     }
 | |
| 
 | |
|     update_task_set_progress(update_task, UpdateTaskStageCompleted, 100);
 | |
|     return UPDATE_TASK_NOERR;
 | |
| }
 |