NEW: release DJI Payload-SDK version 3.0
Signed-off-by: DJI-Martin <DJI-Martin@dji.com>
This commit is contained in:
DJI-Martin
342
samples/sample_c/module_sample/upgrade/test_upgrade.c
Normal file
342
samples/sample_c/module_sample/upgrade/test_upgrade.c
Normal file
@ -0,0 +1,342 @@
|
||||
/**
|
||||
********************************************************************
|
||||
* @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****/
|
||||
58
samples/sample_c/module_sample/upgrade/test_upgrade.h
Normal file
58
samples/sample_c/module_sample/upgrade/test_upgrade.h
Normal file
@ -0,0 +1,58 @@
|
||||
/**
|
||||
********************************************************************
|
||||
* @file test_upgrade.h
|
||||
* @brief This is the header file for "test_upgrade.c", defining the structure and
|
||||
* (exported) function prototypes.
|
||||
*
|
||||
* @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.
|
||||
*
|
||||
*********************************************************************
|
||||
*/
|
||||
|
||||
/* Define to prevent recursive inclusion -------------------------------------*/
|
||||
#ifndef TEST_UPGRADE_H
|
||||
#define TEST_UPGRADE_H
|
||||
|
||||
/* Includes ------------------------------------------------------------------*/
|
||||
#include <dji_upgrade.h>
|
||||
#include "test_upgrade_platform_opt.h"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
/* Exported constants --------------------------------------------------------*/
|
||||
|
||||
/* Exported types ------------------------------------------------------------*/
|
||||
typedef struct {
|
||||
T_DjiFirmwareVersion firmwareVersion;
|
||||
E_DjiFirmwareTransferType transferType;
|
||||
//For linux: need replace program before reboot system
|
||||
//For mcu: don't need replace program before reboot system, replace program in loader
|
||||
bool needReplaceProgramBeforeReboot;
|
||||
} T_DjiTestUpgradeConfig;
|
||||
|
||||
/* Exported functions --------------------------------------------------------*/
|
||||
T_DjiReturnCode DjiTest_UpgradeStartService(const T_DjiTestUpgradePlatformOpt *upgradePlatformOpt,
|
||||
T_DjiTestUpgradeConfig testUpgradeConfig);
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
||||
#endif
|
||||
/************************ (C) COPYRIGHT DJI Innovations *******END OF FILE******/
|
||||
@ -0,0 +1,149 @@
|
||||
/**
|
||||
********************************************************************
|
||||
* @file test_upgrade_common_file_transfer.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 "test_upgrade_common_file_transfer.h"
|
||||
#include "dji_logger.h"
|
||||
#include <utils/util_md5.h>
|
||||
#include "test_upgrade_platform_opt.h"
|
||||
|
||||
/* Private constants ---------------------------------------------------------*/
|
||||
#define DJI_TEST_FILE_MD5_BUFFER_SIZE 256
|
||||
|
||||
/* Private types -------------------------------------------------------------*/
|
||||
|
||||
/* Private values -------------------------------------------------------------*/
|
||||
static T_DjiUpgradeFileInfo s_upgradeFileInfo = {0};
|
||||
static uint32_t s_alreadyTransferFileSize = 0;
|
||||
|
||||
/* Private functions declaration ---------------------------------------------*/
|
||||
static T_DjiReturnCode DjiTestFile_GetUpgradeFileMd5(uint8_t md5[DJI_MD5_BUFFER_LEN]);
|
||||
|
||||
/* Exported functions definition ---------------------------------------------*/
|
||||
T_DjiReturnCode DjiTestCommonFileTransfer_Start(const T_DjiUpgradeFileInfo *fileInfo)
|
||||
{
|
||||
T_DjiReturnCode returnCode;
|
||||
|
||||
s_upgradeFileInfo.fileSize = 0;
|
||||
memset(s_upgradeFileInfo.fileName, 0, sizeof(s_upgradeFileInfo.fileName));
|
||||
s_alreadyTransferFileSize = 0;
|
||||
|
||||
returnCode = DjiTest_CreateUpgradeProgramFile(fileInfo);
|
||||
if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) {
|
||||
USER_LOG_ERROR("Create upgrade program file error");
|
||||
return returnCode;
|
||||
}
|
||||
|
||||
s_upgradeFileInfo = *fileInfo;
|
||||
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
|
||||
}
|
||||
|
||||
T_DjiReturnCode DjiTestCommonFileTransfer_Transfer(const uint8_t *data, uint16_t dataLen)
|
||||
{
|
||||
T_DjiReturnCode returnCode;
|
||||
|
||||
if (s_alreadyTransferFileSize >= s_upgradeFileInfo.fileSize) {
|
||||
USER_LOG_ERROR("Already transfer file size is more than file real size");
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
|
||||
}
|
||||
|
||||
returnCode = DjiTest_WriteUpgradeProgramFile(s_alreadyTransferFileSize, data, dataLen);
|
||||
if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) {
|
||||
USER_LOG_ERROR("Write upgrade program file error, return code = 0x%08llX", returnCode);
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_SYSTEM_ERROR;
|
||||
}
|
||||
|
||||
s_alreadyTransferFileSize += dataLen;
|
||||
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
|
||||
}
|
||||
|
||||
T_DjiReturnCode DjiTestCommonFileTransfer_Finish(const uint8_t md5[DJI_MD5_BUFFER_LEN])
|
||||
{
|
||||
uint8_t localFileMd5[DJI_MD5_BUFFER_LEN] = {0};
|
||||
T_DjiReturnCode returnCode;
|
||||
|
||||
if (s_alreadyTransferFileSize != s_upgradeFileInfo.fileSize) {
|
||||
USER_LOG_ERROR("Transfer finish error, transfer file size is not equal");
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
|
||||
}
|
||||
|
||||
returnCode = DjiTestFile_GetUpgradeFileMd5(localFileMd5);
|
||||
if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) {
|
||||
USER_LOG_ERROR("Get file md5 error, return code = 0x%08llX", returnCode);
|
||||
goto out;
|
||||
}
|
||||
|
||||
if (memcmp(md5, localFileMd5, DJI_MD5_BUFFER_LEN) == 0) {
|
||||
returnCode = DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
|
||||
} else {
|
||||
returnCode = DJI_ERROR_SYSTEM_MODULE_CODE_SYSTEM_ERROR;
|
||||
}
|
||||
|
||||
out:
|
||||
DjiTest_CloseUpgradeProgramFile();
|
||||
s_upgradeFileInfo.fileSize = 0;
|
||||
memset(s_upgradeFileInfo.fileName, 0, sizeof(s_upgradeFileInfo.fileName));
|
||||
s_alreadyTransferFileSize = 0;
|
||||
return returnCode;
|
||||
}
|
||||
|
||||
/* Private functions definition-----------------------------------------------*/
|
||||
static T_DjiReturnCode DjiTestFile_GetUpgradeFileMd5(uint8_t md5[DJI_MD5_BUFFER_LEN])
|
||||
{
|
||||
uint8_t fileBuffer[DJI_TEST_FILE_MD5_BUFFER_SIZE] = {0};
|
||||
T_DjiReturnCode returnCode;
|
||||
uint32_t offset;
|
||||
MD5_CTX fileMd5Ctx;
|
||||
uint16_t realLen = 0;
|
||||
|
||||
offset = 0;
|
||||
UtilMd5_Init(&fileMd5Ctx);
|
||||
while (s_upgradeFileInfo.fileSize - offset > DJI_TEST_FILE_MD5_BUFFER_SIZE) {
|
||||
returnCode = DjiTest_ReadUpgradeProgramFile(offset, DJI_TEST_FILE_MD5_BUFFER_SIZE,
|
||||
fileBuffer, &realLen);
|
||||
if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS || realLen != DJI_TEST_FILE_MD5_BUFFER_SIZE) {
|
||||
USER_LOG_ERROR("Get file data error, return code = 0x%08llX", returnCode);
|
||||
return returnCode;
|
||||
}
|
||||
|
||||
UtilMd5_Update(&fileMd5Ctx, fileBuffer, DJI_TEST_FILE_MD5_BUFFER_SIZE);
|
||||
|
||||
offset += DJI_TEST_FILE_MD5_BUFFER_SIZE;
|
||||
}
|
||||
|
||||
returnCode = DjiTest_ReadUpgradeProgramFile(offset, s_upgradeFileInfo.fileSize - offset, fileBuffer, &realLen);
|
||||
if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS || realLen != s_upgradeFileInfo.fileSize - offset) {
|
||||
USER_LOG_ERROR("Get file data error, return code = 0x%08llX", returnCode);
|
||||
return returnCode;
|
||||
}
|
||||
|
||||
UtilMd5_Update(&fileMd5Ctx, fileBuffer, s_upgradeFileInfo.fileSize - offset);
|
||||
UtilMd5_Final(&fileMd5Ctx, md5);
|
||||
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
|
||||
}
|
||||
|
||||
/****************** (C) COPYRIGHT DJI Innovations *****END OF FILE****/
|
||||
@ -0,0 +1,51 @@
|
||||
/**
|
||||
********************************************************************
|
||||
* @file test_upgrade_common_file_transfer.h
|
||||
* @brief This is the header file for "test_upgrade_common_file_transfer.c", defining the structure and
|
||||
* (exported) function prototypes.
|
||||
*
|
||||
* @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.
|
||||
*
|
||||
*********************************************************************
|
||||
*/
|
||||
|
||||
/* Define to prevent recursive inclusion -------------------------------------*/
|
||||
#ifndef TEST_UPGRADE_COMMON_FILE_TRANSFER_H
|
||||
#define TEST_UPGRADE_COMMON_FILE_TRANSFER_H
|
||||
|
||||
/* Includes ------------------------------------------------------------------*/
|
||||
#include <dji_upgrade.h>
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
/* Exported constants --------------------------------------------------------*/
|
||||
|
||||
/* Exported types ------------------------------------------------------------*/
|
||||
|
||||
/* Exported functions --------------------------------------------------------*/
|
||||
T_DjiReturnCode DjiTestCommonFileTransfer_Start(const T_DjiUpgradeFileInfo *fileInfo);
|
||||
T_DjiReturnCode DjiTestCommonFileTransfer_Transfer(const uint8_t *data, uint16_t dataLen);
|
||||
T_DjiReturnCode DjiTestCommonFileTransfer_Finish(const uint8_t md5[DJI_MD5_BUFFER_LEN]);
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
||||
#endif // TEST_UPGRADE_COMMON_FILE_TRANSFER_LINUX_H
|
||||
/************************ (C) COPYRIGHT DJI Innovations *******END OF FILE******/
|
||||
@ -0,0 +1,139 @@
|
||||
/**
|
||||
********************************************************************
|
||||
* @file test_upgrade_platform_opt.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 "test_upgrade_platform_opt.h"
|
||||
#include <dji_logger.h>
|
||||
|
||||
/* Private constants ---------------------------------------------------------*/
|
||||
|
||||
/* Private types -------------------------------------------------------------*/
|
||||
|
||||
/* Private values -------------------------------------------------------------*/
|
||||
static T_DjiTestUpgradePlatformOpt s_upgradePlatformOpt = {0};
|
||||
|
||||
/* Private functions declaration ---------------------------------------------*/
|
||||
|
||||
/* Exported functions definition ---------------------------------------------*/
|
||||
T_DjiReturnCode DjiTest_RegUpgradePlatformOpt(const T_DjiTestUpgradePlatformOpt *upgradePlatformOpt)
|
||||
{
|
||||
if (upgradePlatformOpt->rebootSystem == NULL) {
|
||||
USER_LOG_ERROR("rebootSystem callback can't be NULL");
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
|
||||
}
|
||||
if (upgradePlatformOpt->cleanUpgradeProgramFileStoreArea == NULL) {
|
||||
USER_LOG_ERROR("cleanUpgradeProgramFileStoreArea callback can't be NULL");
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
|
||||
}
|
||||
if (upgradePlatformOpt->createUpgradeProgramFile == NULL) {
|
||||
USER_LOG_ERROR("createUpgradeProgramFile callback can't be NULL");
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
|
||||
}
|
||||
if (upgradePlatformOpt->readUpgradeProgramFile == NULL) {
|
||||
USER_LOG_ERROR("readUpgradeProgramFile callback can't be NULL");
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
|
||||
}
|
||||
if (upgradePlatformOpt->writeUpgradeProgramFile == NULL) {
|
||||
USER_LOG_ERROR("writeUpgradeProgramFile callback can't be NULL");
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
|
||||
}
|
||||
if (upgradePlatformOpt->closeUpgradeProgramFile == NULL) {
|
||||
USER_LOG_ERROR("closeUpgradeProgramFile callback can't be NULL");
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
|
||||
}
|
||||
if (upgradePlatformOpt->replaceOldProgram == NULL) {
|
||||
USER_LOG_ERROR("replaceOldProgram callback can't be NULL");
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
|
||||
}
|
||||
if (upgradePlatformOpt->setUpgradeRebootState == NULL) {
|
||||
USER_LOG_ERROR("setUpgradeRebootState callback can't be NULL");
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
|
||||
}
|
||||
if (upgradePlatformOpt->getUpgradeRebootState == NULL) {
|
||||
USER_LOG_ERROR("getUpgradeRebootState callback can't be NULL");
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
|
||||
}
|
||||
if (upgradePlatformOpt->cleanUpgradeRebootState == NULL) {
|
||||
USER_LOG_ERROR("cleanUpgradeRebootState callback can't be NULL");
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
|
||||
}
|
||||
|
||||
s_upgradePlatformOpt = *upgradePlatformOpt;
|
||||
|
||||
return DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
|
||||
}
|
||||
|
||||
T_DjiReturnCode DjiTest_RebootSystem(void)
|
||||
{
|
||||
return s_upgradePlatformOpt.rebootSystem();
|
||||
}
|
||||
|
||||
T_DjiReturnCode DjiTest_CleanUpgradeProgramFileStoreArea(void)
|
||||
{
|
||||
return s_upgradePlatformOpt.cleanUpgradeProgramFileStoreArea();
|
||||
}
|
||||
|
||||
T_DjiReturnCode DjiTest_CreateUpgradeProgramFile(const T_DjiUpgradeFileInfo *fileInfo)
|
||||
{
|
||||
return s_upgradePlatformOpt.createUpgradeProgramFile(fileInfo);
|
||||
}
|
||||
|
||||
T_DjiReturnCode DjiTest_WriteUpgradeProgramFile(uint32_t offset, const uint8_t *data, uint16_t dataLen)
|
||||
{
|
||||
return s_upgradePlatformOpt.writeUpgradeProgramFile(offset, data, dataLen);
|
||||
}
|
||||
|
||||
T_DjiReturnCode DjiTest_ReadUpgradeProgramFile(uint32_t offset, uint16_t readDataLen, uint8_t *data,
|
||||
uint16_t *realLen)
|
||||
{
|
||||
return s_upgradePlatformOpt.readUpgradeProgramFile(offset, readDataLen, data, realLen);
|
||||
}
|
||||
|
||||
T_DjiReturnCode DjiTest_CloseUpgradeProgramFile(void)
|
||||
{
|
||||
return s_upgradePlatformOpt.closeUpgradeProgramFile();
|
||||
}
|
||||
|
||||
T_DjiReturnCode DjiTest_ReplaceOldProgram(void)
|
||||
{
|
||||
return s_upgradePlatformOpt.replaceOldProgram();
|
||||
}
|
||||
|
||||
T_DjiReturnCode DjiTest_SetUpgradeRebootState(const T_DjiUpgradeEndInfo *upgradeEndInfo)
|
||||
{
|
||||
return s_upgradePlatformOpt.setUpgradeRebootState(upgradeEndInfo);
|
||||
}
|
||||
|
||||
T_DjiReturnCode DjiTest_GetUpgradeRebootState(bool *isUpgradeReboot, T_DjiUpgradeEndInfo *upgradeEndInfo)
|
||||
{
|
||||
return s_upgradePlatformOpt.getUpgradeRebootState(isUpgradeReboot, upgradeEndInfo);
|
||||
}
|
||||
|
||||
T_DjiReturnCode DjiTest_CleanUpgradeRebootState(void)
|
||||
{
|
||||
return s_upgradePlatformOpt.cleanUpgradeRebootState();
|
||||
}
|
||||
|
||||
/****************** (C) COPYRIGHT DJI Innovations *****END OF FILE****/
|
||||
@ -0,0 +1,84 @@
|
||||
/**
|
||||
********************************************************************
|
||||
* @file test_upgrade_platform_opt.h
|
||||
* @brief This is the header file for "test_upgrade_platform_opt.c", defining the structure and
|
||||
* (exported) function prototypes.
|
||||
*
|
||||
* @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.
|
||||
*
|
||||
*********************************************************************
|
||||
*/
|
||||
|
||||
/* Define to prevent recursive inclusion -------------------------------------*/
|
||||
#ifndef TEST_UPGRADE_PLATFORM_OPT_H
|
||||
#define TEST_UPGRADE_PLATFORM_OPT_H
|
||||
|
||||
#include <dji_typedef.h>
|
||||
#include <dji_upgrade.h>
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
/* Includes ------------------------------------------------------------------*/
|
||||
|
||||
|
||||
/* Exported constants --------------------------------------------------------*/
|
||||
|
||||
/* Exported types ------------------------------------------------------------*/
|
||||
typedef struct {
|
||||
T_DjiReturnCode (*rebootSystem)(void);
|
||||
|
||||
T_DjiReturnCode (*cleanUpgradeProgramFileStoreArea)(void);
|
||||
T_DjiReturnCode (*createUpgradeProgramFile)(const T_DjiUpgradeFileInfo *fileInfo);
|
||||
T_DjiReturnCode (*writeUpgradeProgramFile)(uint32_t offset, const uint8_t *data, uint16_t dataLen);
|
||||
T_DjiReturnCode (*readUpgradeProgramFile)(uint32_t offset, uint16_t readDataLen, uint8_t *data,
|
||||
uint16_t *realLen);
|
||||
T_DjiReturnCode (*closeUpgradeProgramFile)(void);
|
||||
|
||||
T_DjiReturnCode (*replaceOldProgram)(void);
|
||||
|
||||
T_DjiReturnCode (*setUpgradeRebootState)(const T_DjiUpgradeEndInfo *upgradeEndInfo);
|
||||
T_DjiReturnCode (*getUpgradeRebootState)(bool *isUpgradeReboot, T_DjiUpgradeEndInfo *upgradeEndInfo);
|
||||
T_DjiReturnCode (*cleanUpgradeRebootState)(void);
|
||||
} T_DjiTestUpgradePlatformOpt;
|
||||
|
||||
/* Exported functions --------------------------------------------------------*/
|
||||
T_DjiReturnCode DjiTest_RegUpgradePlatformOpt(const T_DjiTestUpgradePlatformOpt *upgradePlatformOpt);
|
||||
|
||||
T_DjiReturnCode DjiTest_RebootSystem(void);
|
||||
|
||||
T_DjiReturnCode DjiTest_CleanUpgradeProgramFileStoreArea(void);
|
||||
T_DjiReturnCode DjiTest_CreateUpgradeProgramFile(const T_DjiUpgradeFileInfo *fileInfo);
|
||||
T_DjiReturnCode DjiTest_WriteUpgradeProgramFile(uint32_t offset, const uint8_t *data, uint16_t dataLen);
|
||||
T_DjiReturnCode DjiTest_ReadUpgradeProgramFile(uint32_t offset, uint16_t readDataLen, uint8_t *data,
|
||||
uint16_t *realLen);
|
||||
T_DjiReturnCode DjiTest_CloseUpgradeProgramFile(void);
|
||||
|
||||
T_DjiReturnCode DjiTest_ReplaceOldProgram(void);
|
||||
|
||||
T_DjiReturnCode DjiTest_SetUpgradeRebootState(const T_DjiUpgradeEndInfo *upgradeEndInfo);
|
||||
T_DjiReturnCode DjiTest_GetUpgradeRebootState(bool *isUpgradeReboot, T_DjiUpgradeEndInfo *upgradeEndInfo);
|
||||
T_DjiReturnCode DjiTest_CleanUpgradeRebootState(void);
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
||||
#endif // TEST_UPGRADE_PLATFORM_OPT_H
|
||||
|
||||
/************************ (C) COPYRIGHT DJI Innovations *******END OF FILE******/
|
||||
Reference in New Issue
Block a user