123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441 |
- #include "nrf_bootloader_fw_activation.h"
- #include "nrf_dfu_settings.h"
- #include "nrf_dfu_mbr.h"
- #include "nrf_bootloader_info.h"
- #include "crc32.h"
- #include "nrf_log.h"
- #include "nrf_log_ctrl.h"
- #include "nrf_dfu_utils.h"
- #include "nrf_bootloader_wdt.h"
- static volatile bool m_flash_write_done;
- static uint32_t image_copy(uint32_t dst_addr,
- uint32_t src_addr,
- uint32_t size,
- uint32_t progress_update_step)
- {
- if (src_addr == dst_addr)
- {
- NRF_LOG_DEBUG("No copy needed");
- return NRF_SUCCESS;
- }
- ASSERT(src_addr >= dst_addr);
- ASSERT(progress_update_step > 0);
- if (size != 0)
- {
- ASSERT((dst_addr % CODE_PAGE_SIZE) == 0);
- }
- uint32_t max_safe_progress_upd_step = (src_addr - dst_addr)/CODE_PAGE_SIZE;
- ASSERT(max_safe_progress_upd_step > 0);
- uint32_t ret_val = NRF_SUCCESS;
- uint32_t pages_left = CEIL_DIV(size, CODE_PAGE_SIZE);
-
- nrf_bootloader_wdt_init();
- progress_update_step = MIN(progress_update_step, max_safe_progress_upd_step);
- while (size > 0)
- {
- uint32_t pages;
- uint32_t bytes;
- if (pages_left <= progress_update_step)
- {
- pages = pages_left;
- bytes = size;
- }
- else
- {
- pages = progress_update_step;
- bytes = progress_update_step * CODE_PAGE_SIZE;
- }
-
- ret_val = nrf_dfu_flash_erase(dst_addr, pages, NULL);
- if (ret_val != NRF_SUCCESS)
- {
- return ret_val;
- }
-
- NRF_LOG_DEBUG("Copying 0x%x to 0x%x, size: 0x%x", src_addr, dst_addr, bytes);
- ret_val = nrf_dfu_flash_store(dst_addr,
- (uint32_t *)src_addr,
- ALIGN_NUM(sizeof(uint32_t), bytes),
- NULL);
- if (ret_val != NRF_SUCCESS)
- {
- return ret_val;
- }
- pages_left -= pages;
- size -= bytes;
- dst_addr += bytes;
- src_addr += bytes;
- s_dfu_settings.write_offset += bytes;
-
- ret_val = nrf_dfu_settings_write_and_backup(NULL);
- if (ret_val != NRF_SUCCESS)
- {
- NRF_LOG_ERROR("Failed to write image copying progress to settings page.");
- return ret_val;
- }
- }
- return ret_val;
- }
- static uint32_t app_activate(void)
- {
-
- uint32_t const image_size = s_dfu_settings.bank_1.image_size;
- uint32_t src_addr = s_dfu_settings.progress.update_start_address;
- uint32_t ret_val = NRF_SUCCESS;
- uint32_t target_addr = nrf_dfu_bank0_start_addr() + s_dfu_settings.write_offset;
- uint32_t length_left = (image_size - s_dfu_settings.write_offset);
- uint32_t crc;
- NRF_LOG_DEBUG("Enter nrf_dfu_app_continue");
- src_addr += s_dfu_settings.write_offset;
- if (src_addr == target_addr)
- {
- length_left = 0;
- }
- ret_val = image_copy(target_addr, src_addr, length_left, NRF_BL_FW_COPY_PROGRESS_STORE_STEP);
- if (ret_val != NRF_SUCCESS)
- {
- NRF_LOG_ERROR("Failed to copy firmware.");
- return ret_val;
- }
-
- crc = crc32_compute((uint8_t*)nrf_dfu_bank0_start_addr(), image_size, NULL);
- if (crc == s_dfu_settings.bank_1.image_crc)
- {
- NRF_LOG_DEBUG("Setting app as valid");
- s_dfu_settings.bank_0.bank_code = NRF_DFU_BANK_VALID_APP;
- s_dfu_settings.bank_0.image_crc = crc;
- s_dfu_settings.bank_0.image_size = image_size;
- }
- else
- {
- NRF_LOG_ERROR("CRC computation failed for copied app: "
- "src crc: 0x%08x, res crc: 0x%08x",
- s_dfu_settings.bank_1.image_crc,
- crc);
- }
- return ret_val;
- }
- static uint32_t sd_activate(void)
- {
- uint32_t ret_val = NRF_SUCCESS;
- uint32_t target_addr = nrf_dfu_softdevice_start_address() + s_dfu_settings.write_offset;
- uint32_t src_addr = s_dfu_settings.progress.update_start_address;
- uint32_t sd_size = s_dfu_settings.sd_size;
- uint32_t length_left = ALIGN_TO_PAGE(sd_size - s_dfu_settings.write_offset);
- NRF_LOG_DEBUG("Enter nrf_bootloader_dfu_sd_continue");
- if (SD_MAGIC_NUMBER_GET(src_addr) != SD_MAGIC_NUMBER)
- {
- NRF_LOG_ERROR("Source address does not contain a valid SoftDevice.")
- return NRF_ERROR_INTERNAL;
- }
-
- src_addr += s_dfu_settings.write_offset;
- if (s_dfu_settings.write_offset == sd_size)
- {
- NRF_LOG_DEBUG("SD already copied");
- return NRF_SUCCESS;
- }
- if (s_dfu_settings.write_offset == 0)
- {
- NRF_LOG_DEBUG("Updating SD. Old SD ver: %d, New ver: %d",
- SD_VERSION_GET(MBR_SIZE) / 1000000, SD_VERSION_GET(src_addr) / 1000000);
- }
- ret_val = image_copy(target_addr, src_addr, length_left, NRF_BL_FW_COPY_PROGRESS_STORE_STEP);
- if (ret_val != NRF_SUCCESS)
- {
- NRF_LOG_ERROR("Failed to copy firmware.");
- return ret_val;
- }
- ret_val = nrf_dfu_settings_write_and_backup(NULL);
- return ret_val;
- }
- static uint32_t bl_activate(void)
- {
- uint32_t ret_val = NRF_ERROR_INVALID_DATA;
- nrf_dfu_bank_t * p_bank = &s_dfu_settings.bank_1;
- uint32_t len = p_bank->image_size;
- uint32_t src_addr = s_dfu_settings.progress.update_start_address;
- if (p_bank->bank_code == NRF_DFU_BANK_VALID_SD_BL)
- {
- src_addr += s_dfu_settings.sd_size;
- len -= s_dfu_settings.sd_size;
- }
- else if (src_addr == 0)
- {
- src_addr = nrf_dfu_bank1_start_addr();
- }
- NRF_LOG_DEBUG("Verifying BL: Addr: 0x%08x, Src: 0x%08x, Len: 0x%08x", BOOTLOADER_START_ADDR, src_addr, len);
-
-
-
- #if defined(NRF52832_XXAA) && defined(BLE_STACK_SUPPORT_REQD)
- if ((p_bank->bank_code == NRF_DFU_BANK_VALID_SD_BL) &&
- (memcmp((void *)BOOTLOADER_START_ADDR, (void *)(src_addr - 0x4000), len) == 0))
- {
- ret_val = NRF_SUCCESS;
- }
- #endif
-
- if ((ret_val != NRF_SUCCESS) &&
- (memcmp((void *)BOOTLOADER_START_ADDR, (void *)src_addr, len) == 0))
- {
- ret_val = NRF_SUCCESS;
- }
-
- if (ret_val == NRF_SUCCESS)
- {
- NRF_LOG_DEBUG("No bootloader copy needed, bootloader update complete.");
- }
- else
- {
- NRF_LOG_DEBUG("Copying bootloader: Src: 0x%08x, Len: 0x%08x", src_addr, len);
- NRF_LOG_FLUSH();
- nrf_bootloader_wdt_feed();
-
-
-
- ret_val = nrf_dfu_mbr_copy_bl((uint32_t*)src_addr, len);
- if (ret_val != NRF_SUCCESS)
- {
- NRF_LOG_ERROR("Request to copy BL failed");
- }
- }
- return ret_val;
- }
- static uint32_t sd_bl_activate()
- {
- uint32_t ret_val = NRF_SUCCESS;
- NRF_LOG_DEBUG("Enter nrf_dfu_sd_bl_continue");
- ret_val = sd_activate();
- if (ret_val != NRF_SUCCESS)
- {
- NRF_LOG_ERROR("SD+BL: SD copy failed");
- return ret_val;
- }
- ret_val = bl_activate();
- if (ret_val != NRF_SUCCESS)
- {
- NRF_LOG_ERROR("SD+BL: BL copy failed");
- return ret_val;
- }
- return ret_val;
- }
- static void flash_write_callback(void * p_context)
- {
- UNUSED_PARAMETER(p_context);
- m_flash_write_done = true;
- }
- nrf_bootloader_fw_activation_result_t nrf_bootloader_fw_activate(void)
- {
- nrf_bootloader_fw_activation_result_t result;
- uint32_t ret_val = NRF_SUCCESS;
- nrf_dfu_bank_t * p_bank = &s_dfu_settings.bank_1;
- bool sd_update = false;
- NRF_LOG_DEBUG("Enter nrf_bootloader_fw_activate");
- switch (p_bank->bank_code)
- {
- case NRF_DFU_BANK_VALID_APP:
- NRF_LOG_DEBUG("Valid App");
- ret_val = app_activate();
- break;
- case NRF_DFU_BANK_VALID_SD:
- NRF_LOG_DEBUG("Valid SD");
- ret_val = sd_activate();
- sd_update = true;
- break;
- case NRF_DFU_BANK_VALID_BL:
- NRF_LOG_DEBUG("Valid BL");
- ret_val = bl_activate();
- break;
- case NRF_DFU_BANK_VALID_SD_BL:
- NRF_LOG_DEBUG("Valid SD + BL");
- ret_val = sd_bl_activate();
- sd_update = true;
- break;
- case NRF_DFU_BANK_INVALID:
- default:
- NRF_LOG_INFO("No firmware to activate.");
- return ACTIVATION_NONE;
- }
- if (ret_val != NRF_SUCCESS)
- {
- NRF_LOG_ERROR("Activation failed with error %d (bank code: 0x%x)", ret_val, p_bank->bank_code);
- result = ACTIVATION_ERROR;
- }
-
- nrf_dfu_bank_invalidate(p_bank);
- m_flash_write_done = false;
- ret_val = nrf_dfu_settings_write_and_backup(flash_write_callback);
- ASSERT(m_flash_write_done);
- if (ret_val == NRF_SUCCESS)
- {
- result = ACTIVATION_SUCCESS;
- if (sd_update && (s_dfu_settings.bank_0.bank_code == NRF_DFU_BANK_VALID_APP))
- {
-
- NRF_LOG_DEBUG("A SoftDevice has just been activated. It's likely that an application will come immediately");
- result = ACTIVATION_SUCCESS_EXPECT_ADDITIONAL_UPDATE;
- }
- }
- else
- {
- NRF_LOG_ERROR("Could not write settings.");
- result = ACTIVATION_ERROR;
- }
- return result;
- }
|