152 lines
6.0 KiB
C
152 lines
6.0 KiB
C
/**
|
||
********************************************************************
|
||
* @file test_upgrade_common_file_transfer.c
|
||
* @version V2.0.0
|
||
* @date 3/12/20
|
||
* @brief
|
||
*
|
||
* @copyright (c) 2018-2019 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 "psdk_logger.h"
|
||
#include <utils/util_md5.h>
|
||
#include "test_upgrade_platform_opt.h"
|
||
|
||
/* Private constants ---------------------------------------------------------*/
|
||
#define PSDK_TEST_FILE_MD5_BUFFER_SIZE 256
|
||
|
||
/* Private types -------------------------------------------------------------*/
|
||
|
||
/* Private values -------------------------------------------------------------*/
|
||
static T_PsdkUpgradeFileInfo s_upgradeFileInfo = {0};
|
||
static uint32_t s_alreadyTransferFileSize = 0;
|
||
|
||
/* Private functions declaration ---------------------------------------------*/
|
||
static T_PsdkReturnCode PsdkTestFile_GetUpgradeFileMd5(uint8_t md5[PSDK_MD5_BUFFER_LEN]);
|
||
|
||
/* Exported functions definition ---------------------------------------------*/
|
||
T_PsdkReturnCode PsdkTestCommonFileTransfer_Start(const T_PsdkUpgradeFileInfo *fileInfo)
|
||
{
|
||
T_PsdkReturnCode returnCode;
|
||
|
||
s_upgradeFileInfo.fileSize = 0;
|
||
memset(s_upgradeFileInfo.fileName, 0, sizeof(s_upgradeFileInfo.fileName));
|
||
s_alreadyTransferFileSize = 0;
|
||
|
||
returnCode = PsdkTest_CreateUpgradeProgramFile(fileInfo);
|
||
if (returnCode != PSDK_ERROR_SYSTEM_MODULE_CODE_SUCCESS) {
|
||
PsdkLogger_UserLogError("Create upgrade program file error");
|
||
return returnCode;
|
||
}
|
||
|
||
s_upgradeFileInfo = *fileInfo;
|
||
|
||
return PSDK_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
|
||
}
|
||
|
||
T_PsdkReturnCode PsdkTestCommonFileTransfer_Transfer(const uint8_t *data, uint16_t dataLen)
|
||
{
|
||
T_PsdkReturnCode returnCode;
|
||
|
||
if (s_alreadyTransferFileSize >= s_upgradeFileInfo.fileSize) {
|
||
PsdkLogger_UserLogError("Already transfer file size is more than file real size");
|
||
return PSDK_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
|
||
}
|
||
|
||
returnCode = PsdkTest_WriteUpgradeProgramFile(s_alreadyTransferFileSize, data, dataLen);
|
||
if (returnCode != PSDK_ERROR_SYSTEM_MODULE_CODE_SUCCESS) {
|
||
PsdkLogger_UserLogError("Write upgrade program file error, return code = 0x%08llX", returnCode);
|
||
return PSDK_ERROR_SYSTEM_MODULE_CODE_SYSTEM_ERROR;
|
||
}
|
||
|
||
s_alreadyTransferFileSize += dataLen;
|
||
|
||
return PSDK_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
|
||
}
|
||
|
||
T_PsdkReturnCode PsdkTestCommonFileTransfer_Finish(const uint8_t md5[PSDK_MD5_BUFFER_LEN])
|
||
{
|
||
uint8_t localFileMd5[PSDK_MD5_BUFFER_LEN] = {0};
|
||
T_PsdkReturnCode returnCode;
|
||
|
||
if (s_alreadyTransferFileSize != s_upgradeFileInfo.fileSize) {
|
||
PsdkLogger_UserLogError("Transfer finish error, transfer file size is not equal");
|
||
return PSDK_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
|
||
}
|
||
|
||
returnCode = PsdkTestFile_GetUpgradeFileMd5(localFileMd5);
|
||
if (returnCode != PSDK_ERROR_SYSTEM_MODULE_CODE_SUCCESS) {
|
||
PsdkLogger_UserLogError("Get file md5 error, return code = 0x%08llX", returnCode);
|
||
goto out;
|
||
}
|
||
|
||
if (memcmp(md5, localFileMd5, PSDK_MD5_BUFFER_LEN) == 0) {
|
||
returnCode = PSDK_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
|
||
} else {
|
||
returnCode = PSDK_ERROR_SYSTEM_MODULE_CODE_SYSTEM_ERROR;
|
||
}
|
||
|
||
out:
|
||
PsdkTest_CloseUpgradeProgramFile();
|
||
s_upgradeFileInfo.fileSize = 0;
|
||
memset(s_upgradeFileInfo.fileName, 0, sizeof(s_upgradeFileInfo.fileName));
|
||
s_alreadyTransferFileSize = 0;
|
||
return returnCode;
|
||
}
|
||
|
||
/* Private functions definition-----------------------------------------------*/
|
||
static T_PsdkReturnCode PsdkTestFile_GetUpgradeFileMd5(uint8_t md5[PSDK_MD5_BUFFER_LEN])
|
||
{
|
||
uint8_t fileBuffer[PSDK_TEST_FILE_MD5_BUFFER_SIZE] = {0};
|
||
T_PsdkReturnCode returnCode;
|
||
uint32_t offset;
|
||
MD5_CTX fileMd5Ctx;
|
||
uint16_t realLen = 0;
|
||
|
||
offset = 0;
|
||
UtilMd5_Init(&fileMd5Ctx);
|
||
while (s_upgradeFileInfo.fileSize - offset > PSDK_TEST_FILE_MD5_BUFFER_SIZE) {
|
||
returnCode = PsdkTest_ReadUpgradeProgramFile(offset, PSDK_TEST_FILE_MD5_BUFFER_SIZE,
|
||
fileBuffer, &realLen);
|
||
if (returnCode != PSDK_ERROR_SYSTEM_MODULE_CODE_SUCCESS || realLen != PSDK_TEST_FILE_MD5_BUFFER_SIZE) {
|
||
PsdkLogger_UserLogError("Get file data error, return code = 0x%08llX", returnCode);
|
||
return returnCode;
|
||
}
|
||
|
||
UtilMd5_Update(&fileMd5Ctx, fileBuffer, PSDK_TEST_FILE_MD5_BUFFER_SIZE);
|
||
|
||
offset += PSDK_TEST_FILE_MD5_BUFFER_SIZE;
|
||
}
|
||
|
||
returnCode = PsdkTest_ReadUpgradeProgramFile(offset, s_upgradeFileInfo.fileSize - offset, fileBuffer, &realLen);
|
||
if (returnCode != PSDK_ERROR_SYSTEM_MODULE_CODE_SUCCESS || realLen != s_upgradeFileInfo.fileSize - offset) {
|
||
PsdkLogger_UserLogError("Get file data error, return code = 0x%08llX", returnCode);
|
||
return returnCode;
|
||
}
|
||
|
||
UtilMd5_Update(&fileMd5Ctx, fileBuffer, s_upgradeFileInfo.fileSize - offset);
|
||
UtilMd5_Final(&fileMd5Ctx, md5);
|
||
|
||
return PSDK_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
|
||
}
|
||
|
||
/****************** (C) COPYRIGHT DJI Innovations *****END OF FILE****/
|