/** ******************************************************************** * @file test_upgrade.c * @brief * * @copyright (c) 2021 DJI. All rights reserved. * * All information contained herein is, and remains, the property of DJI. * The intellectual and technical concepts contained herein are proprietary * to DJI and may be covered by U.S. and foreign patents, patents in process, * and protected by trade secret or copyright law. Dissemination of this * information, including but not limited to data and other proprietary * material(s) incorporated within the information, in any form, is strictly * prohibited without the express written consent of DJI. * * If you receive this source code without DJI’s authorization, you may not * further disseminate the information, and you must immediately remove the * source code and notify DJI of its removal. DJI reserves the right to pursue * legal actions against you for any loss(es) or damage(s) caused by your * failure to do so. * ********************************************************************* */ /* Includes ------------------------------------------------------------------*/ #include #include #include #include "test_upgrade_common_file_transfer.h" #include "test_upgrade_platform_opt.h" #include "test_upgrade.h" /* Private constants ---------------------------------------------------------*/ #define UPGRADE_TASK_STACK_SIZE (2048) #define DJI_TEST_UPGRADE_TASK_FREQ (50) #define DJI_TEST_ENTER_UPGRADE_WAIT_TIME (10) //wait 10s for enter upgrade process #define DJI_TEST_UPGRADE_REBOOT_TIMEOUT (30) //reboot timeout 30s /* Private types -------------------------------------------------------------*/ /* Private values -------------------------------------------------------------*/ static T_DjiUpgradeState s_upgradeState = {0}; static T_DjiMutexHandle s_upgradeStateMutex = {0}; static T_DjiTaskHandle s_upgradeProcessThread; static T_DjiTaskHandle s_enterUpgradeModeProcessThread; static bool s_isNeedEnterUpgradeModeProcess = false; static bool s_isNeedReplaceProgramBeforeReboot = false; /* Private functions declaration ---------------------------------------------*/ static T_DjiReturnCode DjiTest_EnterUpgradeMode(uint16_t *waitTime); static T_DjiReturnCode DjiTest_CheckFirmware(void); static T_DjiReturnCode DjiTest_StartUpgrade(void); static T_DjiReturnCode DjiTest_FinishUpgrade(void); static void *DjiTest_UpgradeProcessTask(void *arg); static void *DjiTest_EnterUpgradeModeProcessTask(void *arg); /* Exported functions definition ---------------------------------------------*/ T_DjiReturnCode DjiTest_UpgradeStartService(const T_DjiTestUpgradePlatformOpt *upgradePlatformOpt, T_DjiTestUpgradeConfig testUpgradeConfig) { T_DjiOsalHandler *osalHandler = DjiPlatform_GetOsalHandler(); T_DjiReturnCode returnCode; bool isUpgradeReboot = false; T_DjiUpgradeEndInfo upgradeEndInfo = {0}; T_DjiUpgradeConfig upgradeConfig = { .currentFirmwareVersion = testUpgradeConfig.firmwareVersion, .firmwareTransferInfo = { .transferType = testUpgradeConfig.transferType, .ftpTransferInfo.port = 21, .dcftpFileTransferOpt = { .start = DjiTestCommonFileTransfer_Start, .transfer = DjiTestCommonFileTransfer_Transfer, .finish = DjiTestCommonFileTransfer_Finish, } } }; s_isNeedReplaceProgramBeforeReboot = testUpgradeConfig.needReplaceProgramBeforeReboot; T_DjiUpgradeHandler s_upgradeHandler = { .EnterUpgradeMode = DjiTest_EnterUpgradeMode, .CheckFirmware = DjiTest_CheckFirmware, .StartUpgrade = DjiTest_StartUpgrade, .FinishUpgrade = DjiTest_FinishUpgrade }; returnCode = DjiTest_RegUpgradePlatformOpt(upgradePlatformOpt); if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) { USER_LOG_ERROR("Reg upgrade platform opt error, return code = 0x%08llX", returnCode); return returnCode; } returnCode = osalHandler->MutexCreate(&s_upgradeStateMutex); if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) { USER_LOG_ERROR("Create mutex error"); return returnCode; } returnCode = DjiTest_GetUpgradeRebootState(&isUpgradeReboot, &upgradeEndInfo); if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) { USER_LOG_ERROR("Get upgrade reboot state error"); isUpgradeReboot = false; } returnCode = DjiTest_CleanUpgradeRebootState(); if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) { USER_LOG_ERROR("Clean upgrade reboot state error"); } osalHandler->MutexLock(s_upgradeStateMutex); if (isUpgradeReboot == true) { s_upgradeState.upgradeStage = DJI_UPGRADE_STAGE_END; s_upgradeState.upgradeEndInfo = upgradeEndInfo; } else { s_upgradeState.upgradeStage = DJI_UPGRADE_STAGE_IDLE; } osalHandler->MutexUnlock(s_upgradeStateMutex); returnCode = DjiUpgrade_Init(&upgradeConfig); if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) { USER_LOG_ERROR("DjiUpgrade_Init error, return code = %d", returnCode); return returnCode; } returnCode = DjiUpgrade_RegHandler(&s_upgradeHandler); if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) { USER_LOG_ERROR("DjiUpgrade_RegHandler error, return code = %d", returnCode); return returnCode; } returnCode = DjiUpgrade_EnableLocalUpgrade(); if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) { USER_LOG_ERROR("DjiUpgrade_EnableLocalUpgrade error, return code = %d", returnCode); return returnCode; } if (osalHandler->TaskCreate("upgrade_task", DjiTest_UpgradeProcessTask, UPGRADE_TASK_STACK_SIZE, NULL, &s_upgradeProcessThread) != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) { USER_LOG_ERROR("Dji upgrade test task create error."); return DJI_ERROR_SYSTEM_MODULE_CODE_UNKNOWN; } if (osalHandler->TaskCreate("enter_upgrade_mode_task", DjiTest_EnterUpgradeModeProcessTask, UPGRADE_TASK_STACK_SIZE, NULL, &s_enterUpgradeModeProcessThread) != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) { USER_LOG_ERROR("Dji upgrade test task create error."); return DJI_ERROR_SYSTEM_MODULE_CODE_UNKNOWN; } return DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS; } /* Private functions definition-----------------------------------------------*/ static T_DjiReturnCode DjiTest_EnterUpgradeMode(uint16_t *waitTime) { // need 10s for upgrade preprocess work. *waitTime = DJI_TEST_ENTER_UPGRADE_WAIT_TIME; // enable is need enter upgrade mode process, the process is in DjiTest_EnterUpgradeModeProcessTask s_isNeedEnterUpgradeModeProcess = true; return DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS; } static T_DjiReturnCode DjiTest_CheckFirmware(void) { // you can do decrypt and check firmware in this stage return DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS; } static T_DjiReturnCode DjiTest_StartUpgrade(void) { T_DjiOsalHandler *osalHandler = DjiPlatform_GetOsalHandler(); osalHandler->MutexLock(s_upgradeStateMutex); s_upgradeState.upgradeOngoingInfo.upgradeProgress = 0; s_upgradeState.upgradeStage = DJI_UPGRADE_STAGE_ONGOING; osalHandler->MutexUnlock(s_upgradeStateMutex); return DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS; } static T_DjiReturnCode DjiTest_FinishUpgrade(void) { T_DjiOsalHandler *osalHandler = DjiPlatform_GetOsalHandler(); osalHandler->MutexLock(s_upgradeStateMutex); s_upgradeState.upgradeStage = DJI_UPGRADE_STAGE_IDLE; osalHandler->MutexUnlock(s_upgradeStateMutex); return DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS; } #ifndef __CC_ARM #pragma GCC diagnostic push #pragma GCC diagnostic ignored "-Wmissing-noreturn" #pragma GCC diagnostic ignored "-Wreturn-type" #endif static void *DjiTest_EnterUpgradeModeProcessTask(void *arg) { T_DjiOsalHandler *osalHandler = DjiPlatform_GetOsalHandler(); T_DjiReturnCode returnCode; USER_UTIL_UNUSED(arg); while (1) { if (s_isNeedEnterUpgradeModeProcess) { // prepare enter upgrade mode // you can do some thing before enter upgrade mode. // clear upgrade program file store area returnCode = DjiTest_CleanUpgradeProgramFileStoreArea(); if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) { USER_LOG_ERROR("Clean upgrade file dir error, please check dir permission"); } s_isNeedEnterUpgradeModeProcess = false; USER_LOG_INFO("Clean upgrade store area"); } osalHandler->TaskSleepMs(1000 / DJI_TEST_UPGRADE_TASK_FREQ); } } #ifndef __CC_ARM #pragma GCC diagnostic pop #endif #ifndef __CC_ARM #pragma GCC diagnostic push #pragma GCC diagnostic ignored "-Wmissing-noreturn" #pragma GCC diagnostic ignored "-Wreturn-type" #endif static void *DjiTest_UpgradeProcessTask(void *arg) { T_DjiOsalHandler *osalHandler = DjiPlatform_GetOsalHandler(); T_DjiUpgradeState tempUpgradeState; T_DjiUpgradeEndInfo upgradeEndInfo; T_DjiReturnCode returnCode; USER_UTIL_UNUSED(arg); while (1) { osalHandler->MutexLock(s_upgradeStateMutex); tempUpgradeState = s_upgradeState; osalHandler->MutexUnlock(s_upgradeStateMutex); if (tempUpgradeState.upgradeStage == DJI_UPGRADE_STAGE_ONGOING) { if (s_isNeedReplaceProgramBeforeReboot) { // Step 1 : Replace old program returnCode = DjiTest_ReplaceOldProgram(); if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) { USER_LOG_ERROR("Replace firmware error, return code = 0x%08llX", returnCode); osalHandler->MutexLock(s_upgradeStateMutex); s_upgradeState.upgradeStage = DJI_UPGRADE_STAGE_END; s_upgradeState.upgradeEndInfo.upgradeEndState = DJI_UPGRADE_END_STATE_UNKNOWN_ERROR; osalHandler->MutexUnlock(s_upgradeStateMutex); continue; } osalHandler->TaskSleepMs(1000); osalHandler->MutexLock(s_upgradeStateMutex); s_upgradeState.upgradeStage = DJI_UPGRADE_STAGE_ONGOING; s_upgradeState.upgradeOngoingInfo.upgradeProgress = 20; DjiUpgrade_PushUpgradeState(&s_upgradeState); osalHandler->MutexUnlock(s_upgradeStateMutex); // Step 2 : Clean upgrade program file store area returnCode = DjiTest_CleanUpgradeProgramFileStoreArea(); if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) { USER_LOG_ERROR("Clean upgrade file dir error, please check dir permission"); osalHandler->MutexLock(s_upgradeStateMutex); s_upgradeState.upgradeStage = DJI_UPGRADE_STAGE_END; s_upgradeState.upgradeEndInfo.upgradeEndState = DJI_UPGRADE_END_STATE_UNKNOWN_ERROR; osalHandler->MutexUnlock(s_upgradeStateMutex); continue; } osalHandler->TaskSleepMs(1000); osalHandler->MutexLock(s_upgradeStateMutex); s_upgradeState.upgradeStage = DJI_UPGRADE_STAGE_ONGOING; s_upgradeState.upgradeOngoingInfo.upgradeProgress = 30; DjiUpgrade_PushUpgradeState(&s_upgradeState); osalHandler->MutexUnlock(s_upgradeStateMutex); } //attention emulation upgrade progress, user don't need this process do { osalHandler->TaskSleepMs(1000); osalHandler->MutexLock(s_upgradeStateMutex); s_upgradeState.upgradeStage = DJI_UPGRADE_STAGE_ONGOING; s_upgradeState.upgradeOngoingInfo.upgradeProgress += 10; tempUpgradeState = s_upgradeState; DjiUpgrade_PushUpgradeState(&s_upgradeState); osalHandler->MutexUnlock(s_upgradeStateMutex); } while (tempUpgradeState.upgradeOngoingInfo.upgradeProgress < 100); // Step 3 : Reboot device osalHandler->MutexLock(s_upgradeStateMutex); s_upgradeState.upgradeStage = DJI_UPGRADE_STAGE_DEVICE_REBOOT; s_upgradeState.upgradeRebootInfo.rebootTimeout = DJI_TEST_UPGRADE_REBOOT_TIMEOUT; DjiUpgrade_PushUpgradeState(&s_upgradeState); osalHandler->MutexUnlock(s_upgradeStateMutex); osalHandler->TaskSleepMs(1000); // sleep 1000ms to ensure push send terminal. upgradeEndInfo.upgradeEndState = DJI_UPGRADE_END_STATE_SUCCESS; returnCode = DjiTest_SetUpgradeRebootState(&upgradeEndInfo); if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) { USER_LOG_ERROR("Set Upgrade reboot state error"); osalHandler->MutexLock(s_upgradeStateMutex); s_upgradeState.upgradeStage = DJI_UPGRADE_STAGE_END; s_upgradeState.upgradeEndInfo.upgradeEndState = DJI_UPGRADE_END_STATE_UNKNOWN_ERROR; osalHandler->MutexUnlock(s_upgradeStateMutex); continue; } returnCode = DjiTest_RebootSystem(); if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) { USER_LOG_ERROR("Reboot system error"); osalHandler->MutexLock(s_upgradeStateMutex); s_upgradeState.upgradeStage = DJI_UPGRADE_STAGE_END; s_upgradeState.upgradeEndInfo.upgradeEndState = DJI_UPGRADE_END_STATE_UNKNOWN_ERROR; osalHandler->MutexUnlock(s_upgradeStateMutex); continue; } while (1) { osalHandler->TaskSleepMs(500); } } else if (s_upgradeState.upgradeStage == DJI_UPGRADE_STAGE_END) { osalHandler->MutexLock(s_upgradeStateMutex); DjiUpgrade_PushUpgradeState(&s_upgradeState); osalHandler->MutexUnlock(s_upgradeStateMutex); } osalHandler->TaskSleepMs(500); } } #ifndef __CC_ARM #pragma GCC diagnostic pop #endif /****************** (C) COPYRIGHT DJI Innovations *****END OF FILE****/