Mirror/Momentum-Firmware
1
0
Fork 0
mirror of https://github.com/Next-Flip/Momentum-Firmware.git synced 2026-05-12 23:58:36 -07:00
Code Issues Packages Projects Releases Wiki Activity

Fix internal storage bugs with restoring U2F keys

Browse Source
This commit is contained in:
Willy-JL
2023-04-28 19:50:04 +01:00
parent 8a5d88fc9a
commit 7adb7e2ded
1 changed files with 57 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

100
applications/services/storage/storages/storage_int.c
View File

@@ -193,57 +193,71 @@ static void storage_int_lfs_mount(LFSData* lfs_data, StorageData* storage) {
was_fingerprint_outdated; was_fingerprint_outdated;
if(need_format) { if(need_format) {
err = lfs_mount(lfs, &lfs_data->config); // Backup U2F keys
if(err == 0) { lfs_file_t file;
uint8_t* cnt = NULL;
uint32_t cnt_size;
uint8_t* key = NULL;
uint32_t key_size;
if(lfs_mount(lfs, &lfs_data->config) == 0) {
FURI_LOG_I(TAG, "Factory reset: Mounted for backup"); FURI_LOG_I(TAG, "Factory reset: Mounted for backup");
// Backup U2F keys if(lfs_file_open(lfs, &file, ".cnt.u2f", LFS_O_RDONLY) == 0) {
lfs_file_t file; cnt_size = file.ctz.size;
lfs_file_open(lfs, &file, ".cnt.u2f", LFS_O_RDWR | LFS_O_CREAT); cnt = malloc(cnt_size);
uint8_t cnt[file.ctz.size]; if(lfs_file_read(lfs, &file, cnt, cnt_size) != (int32_t)cnt_size) {
lfs_file_read(lfs, &file, cnt, sizeof(cnt)); free(cnt);
lfs_file_close(lfs, &file); cnt = NULL;
lfs_file_open(lfs, &file, ".key.u2f", LFS_O_RDWR | LFS_O_CREAT);
uint8_t key[file.ctz.size];
lfs_file_read(lfs, &file, key, sizeof(key));
lfs_file_close(lfs, &file);
err = lfs_unmount(lfs);
if(err == 0) {
FURI_LOG_E(TAG, "Factory reset: Unmounted after backup");
// Format storage
err = lfs_format(lfs, &lfs_data->config);
if(err == 0) {
FURI_LOG_I(TAG, "Factory reset: Format successful, trying to mount");
furi_hal_rtc_reset_flag(FuriHalRtcFlagFactoryReset);
err = lfs_mount(lfs, &lfs_data->config);
if(err == 0) {
FURI_LOG_I(TAG, "Factory reset: Mounted");
// Restore U2F keys
lfs_file_open(lfs, &file, ".cnt.u2f", LFS_O_RDWR | LFS_O_CREAT);
lfs_file_write(lfs, &file, cnt, sizeof(cnt));
lfs_file_close(lfs, &file);
lfs_file_open(lfs, &file, ".key.u2f", LFS_O_RDWR | LFS_O_CREAT);
lfs_file_write(lfs, &file, key, sizeof(key));
lfs_file_close(lfs, &file);
storage->status = StorageStatusOK;
} else {
FURI_LOG_E(TAG, "Factory reset: Mount after format failed");
storage->status = StorageStatusNotMounted;
}
} else {
FURI_LOG_E(TAG, "Factory reset: Format failed");
storage->status = StorageStatusNoFS;
} }
lfs_file_close(lfs, &file);
if(lfs_file_open(lfs, &file, ".key.u2f", LFS_O_RDONLY) == 0) {
key_size = file.ctz.size;
key = malloc(key_size);
if(lfs_file_read(lfs, &file, key, key_size) != (int32_t)key_size) {
free(key);
key = NULL;
}
lfs_file_close(lfs, &file);
}
}
if(lfs_unmount(lfs) == 0) {
FURI_LOG_E(TAG, "Factory reset: Unmounted after backup");
} else { } else {
FURI_LOG_E(TAG, "Factory reset: Unmount after backup failed"); FURI_LOG_E(TAG, "Factory reset: Unmount after backup failed");
storage->status = StorageStatusNotMounted;
} }
} else { } else {
FURI_LOG_E(TAG, "Factory reset: Mount for backup failed"); FURI_LOG_E(TAG, "Factory reset: Mount for backup failed");
storage->status = StorageStatusNotMounted; }
// Format storage
if(lfs_format(lfs, &lfs_data->config) == 0) {
FURI_LOG_I(TAG, "Factory reset: Format successful, trying to mount");
furi_hal_rtc_reset_flag(FuriHalRtcFlagFactoryReset);
if(lfs_mount(lfs, &lfs_data->config) == 0) {
FURI_LOG_I(TAG, "Factory reset: Mounted");
storage->status = StorageStatusOK;
// Restore U2F keys
if(cnt != NULL && key != NULL) {
if(lfs_file_open(lfs, &file, ".cnt.u2f", LFS_O_WRONLY | LFS_O_CREAT) == 0) {
lfs_file_write(lfs, &file, cnt, cnt_size);
lfs_file_close(lfs, &file);
if(lfs_file_open(lfs, &file, ".key.u2f", LFS_O_WRONLY | LFS_O_CREAT) == 0) {
lfs_file_write(lfs, &file, key, key_size);
lfs_file_close(lfs, &file);
}
}
}
if(cnt != NULL) free(cnt);
if(key != NULL) free(key);
} else {
FURI_LOG_E(TAG, "Factory reset: Mount after format failed");
storage->status = StorageStatusNotMounted;
}
} else {
FURI_LOG_E(TAG, "Factory reset: Format failed");
storage->status = StorageStatusNoFS;
} }
} else { } else {
// Normal // Normal