mirror of
https://github.com/DarkFlippers/unleashed-firmware
synced 2024-12-04 02:09:12 +00:00
5b36c5465d
State callbacks assumed they were invoked from the thread that changed its state, but this wasn't true for FuriThreadStateStarting in the past, and now it's not true for FuriThreadStateStopped either. Now it is safe to release the thread memory form the state callback once it switches to FuriThreadStateStopped. Therefore, pending deletion calls can be removed. Co-authored-by: Aleksandr Kutuzov <alleteam@gmail.com>
556 lines
18 KiB
C
556 lines
18 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/update_operation.h>
|
|
|
|
#define TAG "UpdWorker"
|
|
|
|
static const char* update_task_stage_descr[] = {
|
|
[UpdateTaskStageProgress] = "...",
|
|
[UpdateTaskStageReadManifest] = "Loading update manifest",
|
|
[UpdateTaskStageValidateDFUImage] = "Checking DFU file",
|
|
[UpdateTaskStageFlashWrite] = "Writing flash",
|
|
[UpdateTaskStageFlashValidate] = "Validating flash",
|
|
[UpdateTaskStageRadioImageValidate] = "Checking radio FW",
|
|
[UpdateTaskStageRadioErase] = "Uninstalling radio FW",
|
|
[UpdateTaskStageRadioWrite] = "Writing radio FW",
|
|
[UpdateTaskStageRadioInstall] = "Installing radio FW",
|
|
[UpdateTaskStageRadioBusy] = "Core 2 busy",
|
|
[UpdateTaskStageOBValidation] = "Validating opt. bytes",
|
|
[UpdateTaskStageIntBackup] = "Backing up configuration",
|
|
[UpdateTaskStageIntRestore] = "Restoring configuration",
|
|
[UpdateTaskStageResourcesFileCleanup] = "Cleaning up files",
|
|
[UpdateTaskStageResourcesDirCleanup] = "Cleaning up directories",
|
|
[UpdateTaskStageResourcesFileUnpack] = "Extracting resources",
|
|
[UpdateTaskStageSplashscreenInstall] = "Installing splashscreen",
|
|
[UpdateTaskStageCompleted] = "Restarting...",
|
|
[UpdateTaskStageError] = "Error",
|
|
[UpdateTaskStageOBError] = "OB, report",
|
|
};
|
|
|
|
static const struct {
|
|
UpdateTaskStage stage;
|
|
uint8_t percent_min, percent_max;
|
|
const char* descr;
|
|
} update_task_error_detail[] = {
|
|
{
|
|
.stage = UpdateTaskStageReadManifest,
|
|
.percent_min = 0,
|
|
.percent_max = 13,
|
|
.descr = "Wrong Updater HW",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageReadManifest,
|
|
.percent_min = 14,
|
|
.percent_max = 20,
|
|
.descr = "Manifest pointer error",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageReadManifest,
|
|
.percent_min = 21,
|
|
.percent_max = 30,
|
|
.descr = "Manifest load error",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageReadManifest,
|
|
.percent_min = 31,
|
|
.percent_max = 40,
|
|
.descr = "Wrong package version",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageReadManifest,
|
|
.percent_min = 41,
|
|
.percent_max = 50,
|
|
.descr = "HW Target mismatch",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageReadManifest,
|
|
.percent_min = 51,
|
|
.percent_max = 60,
|
|
.descr = "No DFU file",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageReadManifest,
|
|
.percent_min = 61,
|
|
.percent_max = 80,
|
|
.descr = "No Radio file",
|
|
},
|
|
#ifndef FURI_RAM_EXEC
|
|
{
|
|
.stage = UpdateTaskStageIntBackup,
|
|
.percent_min = 0,
|
|
.percent_max = 100,
|
|
.descr = "FS R/W error",
|
|
},
|
|
#else
|
|
{
|
|
.stage = UpdateTaskStageRadioImageValidate,
|
|
.percent_min = 0,
|
|
.percent_max = 98,
|
|
.descr = "FS Read error",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageRadioImageValidate,
|
|
.percent_min = 99,
|
|
.percent_max = 100,
|
|
.descr = "CRC mismatch",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageRadioErase,
|
|
.percent_min = 0,
|
|
.percent_max = 30,
|
|
.descr = "Stack remove: cmd error",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageRadioErase,
|
|
.percent_min = 31,
|
|
.percent_max = 100,
|
|
.descr = "Stack remove: wait failed",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageRadioWrite,
|
|
.percent_min = 0,
|
|
.percent_max = 100,
|
|
.descr = "Stack write: error",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageRadioInstall,
|
|
.percent_min = 0,
|
|
.percent_max = 10,
|
|
.descr = "Stack install: cmd error",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageRadioInstall,
|
|
.percent_min = 11,
|
|
.percent_max = 100,
|
|
.descr = "Stack install: wait failed",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageRadioBusy,
|
|
.percent_min = 0,
|
|
.percent_max = 10,
|
|
.descr = "Failed to start C2",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageRadioBusy,
|
|
.percent_min = 11,
|
|
.percent_max = 20,
|
|
.descr = "C2 FUS switch failed",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageRadioBusy,
|
|
.percent_min = 21,
|
|
.percent_max = 30,
|
|
.descr = "FUS operation failed",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageRadioBusy,
|
|
.percent_min = 31,
|
|
.percent_max = 100,
|
|
.descr = "C2 Stach switch failed",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageOBValidation,
|
|
.percent_min = 0,
|
|
.percent_max = 100,
|
|
.descr = "Uncorr. value mismatch",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageValidateDFUImage,
|
|
.percent_min = 0,
|
|
.percent_max = 1,
|
|
.descr = "Failed to open DFU file",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageValidateDFUImage,
|
|
.percent_min = 1,
|
|
.percent_max = 97,
|
|
.descr = "DFU file read error",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageValidateDFUImage,
|
|
.percent_min = 98,
|
|
.percent_max = 100,
|
|
.descr = "DFU file CRC mismatch",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageFlashWrite,
|
|
.percent_min = 0,
|
|
.percent_max = 100,
|
|
.descr = "Flash write error",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageFlashValidate,
|
|
.percent_min = 0,
|
|
.percent_max = 100,
|
|
.descr = "Flash compare error",
|
|
},
|
|
#endif
|
|
#ifndef FURI_RAM_EXEC
|
|
{
|
|
.stage = UpdateTaskStageIntRestore,
|
|
.percent_min = 0,
|
|
.percent_max = 100,
|
|
.descr = "SD card I/O error",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageResourcesFileCleanup,
|
|
.percent_min = 0,
|
|
.percent_max = 100,
|
|
.descr = "SD card I/O error",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageResourcesDirCleanup,
|
|
.percent_min = 0,
|
|
.percent_max = 100,
|
|
.descr = "SD card I/O error",
|
|
},
|
|
{
|
|
.stage = UpdateTaskStageResourcesFileUnpack,
|
|
.percent_min = 0,
|
|
.percent_max = 100,
|
|
.descr = "SD card I/O error",
|
|
},
|
|
#endif
|
|
};
|
|
|
|
static const char* update_task_get_error_message(UpdateTaskStage stage, uint8_t percent) {
|
|
for(size_t i = 0; i < COUNT_OF(update_task_error_detail); i++) {
|
|
if(update_task_error_detail[i].stage == stage &&
|
|
percent >= update_task_error_detail[i].percent_min &&
|
|
percent <= update_task_error_detail[i].percent_max) {
|
|
return update_task_error_detail[i].descr;
|
|
}
|
|
}
|
|
return "Unknown error";
|
|
}
|
|
|
|
typedef struct {
|
|
UpdateTaskStageGroup group;
|
|
uint8_t weight;
|
|
} UpdateTaskStageGroupMap;
|
|
|
|
#define STAGE_DEF(GROUP, WEIGHT) \
|
|
{ \
|
|
.group = (GROUP), \
|
|
.weight = (WEIGHT), \
|
|
}
|
|
|
|
static const UpdateTaskStageGroupMap update_task_stage_progress[] = {
|
|
[UpdateTaskStageProgress] = STAGE_DEF(UpdateTaskStageGroupMisc, 0),
|
|
|
|
[UpdateTaskStageReadManifest] = STAGE_DEF(UpdateTaskStageGroupPreUpdate, 45),
|
|
[UpdateTaskStageIntBackup] = STAGE_DEF(UpdateTaskStageGroupPreUpdate, 5),
|
|
|
|
[UpdateTaskStageRadioImageValidate] = STAGE_DEF(UpdateTaskStageGroupRadio, 15),
|
|
[UpdateTaskStageRadioErase] = STAGE_DEF(UpdateTaskStageGroupRadio, 25),
|
|
[UpdateTaskStageRadioWrite] = STAGE_DEF(UpdateTaskStageGroupRadio, 40),
|
|
[UpdateTaskStageRadioInstall] = STAGE_DEF(UpdateTaskStageGroupRadio, 30),
|
|
[UpdateTaskStageRadioBusy] = STAGE_DEF(UpdateTaskStageGroupRadio, 5),
|
|
|
|
[UpdateTaskStageOBValidation] = STAGE_DEF(UpdateTaskStageGroupOptionBytes, 2),
|
|
|
|
[UpdateTaskStageValidateDFUImage] = STAGE_DEF(UpdateTaskStageGroupFirmware, 33),
|
|
[UpdateTaskStageFlashWrite] = STAGE_DEF(UpdateTaskStageGroupFirmware, 100),
|
|
[UpdateTaskStageFlashValidate] = STAGE_DEF(UpdateTaskStageGroupFirmware, 20),
|
|
|
|
[UpdateTaskStageIntRestore] = STAGE_DEF(UpdateTaskStageGroupPostUpdate, 5),
|
|
|
|
[UpdateTaskStageResourcesFileCleanup] = STAGE_DEF(UpdateTaskStageGroupResources, 100),
|
|
[UpdateTaskStageResourcesDirCleanup] = STAGE_DEF(UpdateTaskStageGroupResources, 50),
|
|
[UpdateTaskStageResourcesFileUnpack] = STAGE_DEF(UpdateTaskStageGroupResources, 255),
|
|
[UpdateTaskStageSplashscreenInstall] = STAGE_DEF(UpdateTaskStageGroupSplashscreen, 5),
|
|
|
|
[UpdateTaskStageCompleted] = STAGE_DEF(UpdateTaskStageGroupMisc, 1),
|
|
[UpdateTaskStageError] = STAGE_DEF(UpdateTaskStageGroupMisc, 1),
|
|
[UpdateTaskStageOBError] = STAGE_DEF(UpdateTaskStageGroupMisc, 1),
|
|
};
|
|
|
|
static UpdateTaskStageGroup update_task_get_task_groups(UpdateTask* update_task) {
|
|
UpdateTaskStageGroup ret = UpdateTaskStageGroupPreUpdate | UpdateTaskStageGroupPostUpdate;
|
|
UpdateManifest* manifest = update_task->manifest;
|
|
if(!furi_string_empty(manifest->radio_image)) {
|
|
ret |= UpdateTaskStageGroupRadio;
|
|
}
|
|
if(update_manifest_has_obdata(manifest)) {
|
|
ret |= UpdateTaskStageGroupOptionBytes;
|
|
}
|
|
if(!furi_string_empty(manifest->firmware_dfu_image)) {
|
|
ret |= UpdateTaskStageGroupFirmware;
|
|
}
|
|
if(!furi_string_empty(manifest->resource_bundle)) {
|
|
ret |= UpdateTaskStageGroupResources;
|
|
}
|
|
if(!furi_string_empty(manifest->splash_file)) {
|
|
ret |= UpdateTaskStageGroupSplashscreen;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
static void update_task_calc_completed_stages(UpdateTask* update_task) {
|
|
uint32_t completed_stages_points = 0;
|
|
for(UpdateTaskStage past_stage = UpdateTaskStageProgress;
|
|
past_stage < update_task->state.stage;
|
|
++past_stage) {
|
|
const UpdateTaskStageGroupMap* grp_descr = &update_task_stage_progress[past_stage];
|
|
if((grp_descr->group & update_task->state.groups) == 0) {
|
|
continue;
|
|
}
|
|
completed_stages_points += grp_descr->weight;
|
|
}
|
|
update_task->state.completed_stages_points = completed_stages_points;
|
|
}
|
|
|
|
void update_task_set_progress(UpdateTask* update_task, UpdateTaskStage stage, uint8_t progress) {
|
|
if(stage != UpdateTaskStageProgress) {
|
|
FURI_LOG_I(TAG, "Stage %d, progress %d", stage, progress);
|
|
/* do not override more specific error states */
|
|
if((stage >= UpdateTaskStageError) && (update_task->state.stage >= UpdateTaskStageError)) {
|
|
return;
|
|
}
|
|
/* Build error message with code "[stage_idx-stage_percent]" */
|
|
if(stage >= UpdateTaskStageError) {
|
|
furi_string_printf(
|
|
update_task->state.status,
|
|
"%s\n#[%d-%d]",
|
|
update_task_get_error_message(
|
|
update_task->state.stage, update_task->state.stage_progress),
|
|
update_task->state.stage,
|
|
update_task->state.stage_progress);
|
|
} else {
|
|
furi_string_set(update_task->state.status, update_task_stage_descr[stage]);
|
|
}
|
|
/* Store stage update */
|
|
update_task->state.stage = stage;
|
|
/* If we are still alive, sum completed stages weights */
|
|
if((stage > UpdateTaskStageProgress) && (stage < UpdateTaskStageCompleted)) {
|
|
update_task_calc_completed_stages(update_task);
|
|
}
|
|
}
|
|
|
|
/* Store stage progress for all non-error updates - to provide details on error state */
|
|
if(!update_stage_is_error(stage)) {
|
|
update_task->state.stage_progress = progress;
|
|
}
|
|
|
|
/* Calculate "overall" progress, based on stage weights */
|
|
uint32_t adapted_progress = 1;
|
|
if(update_task->state.total_progress_points != 0) {
|
|
if(stage < UpdateTaskStageCompleted) {
|
|
adapted_progress = MIN(
|
|
(update_task->state.completed_stages_points +
|
|
(update_task_stage_progress[update_task->state.stage].weight * progress / 100)) *
|
|
100 / (update_task->state.total_progress_points),
|
|
100u);
|
|
|
|
} else {
|
|
adapted_progress = update_task->state.overall_progress;
|
|
}
|
|
}
|
|
update_task->state.overall_progress = adapted_progress;
|
|
|
|
if(update_task->status_change_cb) {
|
|
(update_task->status_change_cb)(
|
|
furi_string_get_cstr(update_task->state.status),
|
|
adapted_progress,
|
|
update_stage_is_error(update_task->state.stage),
|
|
update_task->status_change_cb_state);
|
|
}
|
|
}
|
|
|
|
static void update_task_close_file(UpdateTask* update_task) {
|
|
furi_assert(update_task);
|
|
if(!storage_file_is_open(update_task->file)) {
|
|
return;
|
|
}
|
|
|
|
storage_file_close(update_task->file);
|
|
}
|
|
|
|
static bool update_task_check_file_exists(UpdateTask* update_task, FuriString* filename) {
|
|
furi_assert(update_task);
|
|
FuriString* tmp_path;
|
|
tmp_path = furi_string_alloc_set(update_task->update_path);
|
|
path_append(tmp_path, furi_string_get_cstr(filename));
|
|
bool exists = storage_file_exists(update_task->storage, furi_string_get_cstr(tmp_path));
|
|
furi_string_free(tmp_path);
|
|
return exists;
|
|
}
|
|
|
|
bool update_task_open_file(UpdateTask* update_task, FuriString* filename) {
|
|
furi_assert(update_task);
|
|
update_task_close_file(update_task);
|
|
|
|
FuriString* tmp_path;
|
|
tmp_path = furi_string_alloc_set(update_task->update_path);
|
|
path_append(tmp_path, furi_string_get_cstr(filename));
|
|
bool open_success = storage_file_open(
|
|
update_task->file, furi_string_get_cstr(tmp_path), FSAM_READ, FSOM_OPEN_EXISTING);
|
|
furi_string_free(tmp_path);
|
|
return open_success;
|
|
}
|
|
|
|
static void
|
|
update_task_worker_thread_cb(FuriThread* thread, FuriThreadState state, void* context) {
|
|
UNUSED(context);
|
|
|
|
if(state != FuriThreadStateStopped) {
|
|
return;
|
|
}
|
|
|
|
if(furi_thread_get_return_code(thread) == UPDATE_TASK_NOERR) {
|
|
furi_delay_ms(UPDATE_DELAY_OPERATION_OK);
|
|
furi_hal_power_reset();
|
|
}
|
|
}
|
|
|
|
UpdateTask* update_task_alloc(void) {
|
|
UpdateTask* update_task = malloc(sizeof(UpdateTask));
|
|
|
|
update_task->state.stage = UpdateTaskStageProgress;
|
|
update_task->state.stage_progress = 0;
|
|
update_task->state.overall_progress = 0;
|
|
update_task->state.status = furi_string_alloc();
|
|
|
|
update_task->manifest = update_manifest_alloc();
|
|
update_task->storage = furi_record_open(RECORD_STORAGE);
|
|
update_task->file = storage_file_alloc(update_task->storage);
|
|
update_task->status_change_cb = NULL;
|
|
update_task->boot_mode = furi_hal_rtc_get_boot_mode();
|
|
update_task->update_path = furi_string_alloc();
|
|
|
|
FuriThread* thread = update_task->thread =
|
|
furi_thread_alloc_ex("UpdateWorker", 5120, NULL, update_task);
|
|
|
|
furi_thread_set_state_callback(thread, update_task_worker_thread_cb);
|
|
#ifdef FURI_RAM_EXEC
|
|
UNUSED(update_task_worker_backup_restore);
|
|
furi_thread_set_callback(thread, update_task_worker_flash_writer);
|
|
#else
|
|
UNUSED(update_task_worker_flash_writer);
|
|
furi_thread_set_callback(thread, update_task_worker_backup_restore);
|
|
#endif
|
|
|
|
return update_task;
|
|
}
|
|
|
|
void update_task_free(UpdateTask* update_task) {
|
|
furi_assert(update_task);
|
|
|
|
furi_thread_join(update_task->thread);
|
|
|
|
furi_thread_free(update_task->thread);
|
|
update_task_close_file(update_task);
|
|
storage_file_free(update_task->file);
|
|
update_manifest_free(update_task->manifest);
|
|
|
|
furi_record_close(RECORD_STORAGE);
|
|
furi_string_free(update_task->update_path);
|
|
|
|
free(update_task);
|
|
}
|
|
|
|
bool update_task_parse_manifest(UpdateTask* update_task) {
|
|
furi_assert(update_task);
|
|
update_task->state.stage_progress = 0;
|
|
update_task->state.overall_progress = 0;
|
|
update_task->state.total_progress_points = 0;
|
|
update_task->state.completed_stages_points = 0;
|
|
update_task->state.groups = 0;
|
|
|
|
update_task_set_progress(update_task, UpdateTaskStageReadManifest, 0);
|
|
bool result = false;
|
|
FuriString* manifest_path;
|
|
manifest_path = furi_string_alloc();
|
|
|
|
do {
|
|
update_task_set_progress(update_task, UpdateTaskStageProgress, 13);
|
|
if(!furi_hal_version_do_i_belong_here()) {
|
|
break;
|
|
}
|
|
|
|
update_task_set_progress(update_task, UpdateTaskStageProgress, 20);
|
|
if(!update_operation_get_current_package_manifest_path(
|
|
update_task->storage, manifest_path)) {
|
|
break;
|
|
}
|
|
|
|
path_extract_dirname(furi_string_get_cstr(manifest_path), update_task->update_path);
|
|
update_task_set_progress(update_task, UpdateTaskStageProgress, 30);
|
|
|
|
UpdateManifest* manifest = update_task->manifest;
|
|
if(!update_manifest_init(manifest, furi_string_get_cstr(manifest_path))) {
|
|
break;
|
|
}
|
|
|
|
update_task_set_progress(update_task, UpdateTaskStageProgress, 40);
|
|
if(manifest->manifest_version < UPDATE_OPERATION_MIN_MANIFEST_VERSION) {
|
|
break;
|
|
}
|
|
|
|
update_task_set_progress(update_task, UpdateTaskStageProgress, 50);
|
|
/* Check target only if it's set - skipped for pre-production samples */
|
|
if(furi_hal_version_get_hw_target() &&
|
|
(manifest->target != furi_hal_version_get_hw_target())) {
|
|
break;
|
|
}
|
|
|
|
update_task->state.groups = update_task_get_task_groups(update_task);
|
|
for(size_t stage_counter = 0; stage_counter < COUNT_OF(update_task_stage_progress);
|
|
++stage_counter) {
|
|
const UpdateTaskStageGroupMap* grp_descr = &update_task_stage_progress[stage_counter];
|
|
if((grp_descr->group & update_task->state.groups) != 0) {
|
|
update_task->state.total_progress_points += grp_descr->weight;
|
|
}
|
|
}
|
|
|
|
update_task_set_progress(update_task, UpdateTaskStageProgress, 60);
|
|
if((update_task->state.groups & UpdateTaskStageGroupFirmware) &&
|
|
!update_task_check_file_exists(update_task, manifest->firmware_dfu_image)) {
|
|
break;
|
|
}
|
|
|
|
update_task_set_progress(update_task, UpdateTaskStageProgress, 80);
|
|
if((update_task->state.groups & UpdateTaskStageGroupRadio) &&
|
|
(!update_task_check_file_exists(update_task, manifest->radio_image) ||
|
|
(manifest->radio_version.version.type == 0))) {
|
|
break;
|
|
}
|
|
|
|
update_task_set_progress(update_task, UpdateTaskStageProgress, 100);
|
|
result = true;
|
|
} while(false);
|
|
|
|
furi_string_free(manifest_path);
|
|
return result;
|
|
}
|
|
|
|
void update_task_set_progress_cb(UpdateTask* update_task, updateProgressCb cb, void* state) {
|
|
update_task->status_change_cb = cb;
|
|
update_task->status_change_cb_state = state;
|
|
}
|
|
|
|
void update_task_start(UpdateTask* update_task) {
|
|
furi_assert(update_task);
|
|
furi_thread_start(update_task->thread);
|
|
}
|
|
|
|
bool update_task_is_running(UpdateTask* update_task) {
|
|
furi_assert(update_task);
|
|
return furi_thread_get_state(update_task->thread) == FuriThreadStateRunning;
|
|
}
|
|
|
|
UpdateTaskState const* update_task_get_state(UpdateTask* update_task) {
|
|
furi_assert(update_task);
|
|
return &update_task->state;
|
|
}
|
|
|
|
UpdateManifest const* update_task_get_manifest(UpdateTask* update_task) {
|
|
furi_assert(update_task);
|
|
return update_task->manifest;
|
|
}
|