343 lines
14 KiB
C
343 lines
14 KiB
C
|
/**
|
|||
|
|
********************************************************************
|
|||
|
|
* @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 <dji_logger.h>
|
|||
|
|
#include <dji_platform.h>
|
|||
|
|
#include <utils/util_misc.h>
|
|||
|
|
#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****/
|