pinebuds/services/tota/app_tota.cpp

249 lines
6.8 KiB
C++

/***************************************************************************
*
* Copyright 2015-2019 BES.
* All rights reserved. All unpublished rights reserved.
*
* No part of this work may be used or reproduced in any form or by any
* means, or stored in a database or retrieval system, without prior written
* permission of BES.
*
* Use of this work is governed by a license granted by BES.
* This work contains confidential and proprietary information of
* BES. which is protected by copyright, trade secret,
* trademark and other intellectual property rights.
*
****************************************************************************/
#include "hal_trace.h"
#include "hal_timer.h"
#include "app_audio.h"
#include "app_utils.h"
#include "hal_aud.h"
#include "hal_norflash.h"
#include "pmu.h"
#include "string.h"
#include "stdio.h"
#include "stdarg.h"
#include "cmsis_os.h"
#include "app_tota.h"
#include "app_tota_cmd_code.h"
#include "app_tota_cmd_handler.h"
#include "app_tota_data_handler.h"
#include "app_spp_tota.h"
#include "cqueue.h"
#include "app_ble_rx_handler.h"
#include "rwapp_config.h"
#include "btapp.h"
#include "app_bt.h"
#include "apps.h"
#include "app_thread.h"
#include "cqueue.h"
#include "hal_location.h"
#include "app_hfp.h"
#include "bt_drv_reg_op.h"
#if defined(IBRT)
#include "app_tws_ibrt.h"
#endif
#include "cmsis.h"
#include "app_battery.h"
#include "crc32.h"
#include "factory_section.h"
#include "app_ibrt_rssi.h"
#include "app_spp_tota_general_service.h"
#include "tota_stream_data_transfer.h"
#include "app_tota_flash_program.h"
#include "app_tota_audio_dump.h"
#include "app_tota_mic.h"
#include "app_tota_anc.h"
#include "app_tota_general.h"
#include "app_tota_custom.h"
#include "app_tota_conn.h"
#include "aes.h"
typedef struct
{
uint8_t connectedType;
APP_TOTA_TRANSMISSION_PATH_E dataPath;
} APP_TOTA_ENV_T;
static APP_TOTA_ENV_T app_tota_env=
{
0,
};
bool app_is_in_tota_mode(void)
{
return app_tota_env.connectedType;
}
void app_tota_init(void)
{
TOTA_LOG_DBG(0,"Init application test over the air.");
app_spp_tota_init();
app_spp_tota_gen_init();
app_tota_cmd_handler_init();
app_tota_stream_data_transfer_init();
/* register callback modules */
app_tota_mic_init();
app_tota_anc_init();
app_tota_audio_dump_init();
app_tota_general_init();
app_tota_custom_init();
app_tota_flash_init();
/* set module to access spp callback */
//tota_callback_module_set(APP_TOTA_AUDIO_DUMP);
tota_callback_module_set(APP_TOTA_ANC);
#if (BLE_APP_TOTA)
app_ble_rx_handler_init();
#endif
}
void app_tota_connected(uint8_t connType)
{
TOTA_LOG_DBG(0,"Tota is connected.");
app_tota_env.connectedType |= connType;
}
void app_tota_disconnected(uint8_t disconnType)
{
TOTA_LOG_DBG(0,"Tota is disconnected.");
app_tota_env.connectedType &= disconnType;
}
void app_tota_general_connected(uint8_t connType)
{
TOTA_LOG_DBG(0,"Tota gen is connected.");
app_tota_env.connectedType |= connType;
}
void app_tota_update_datapath(APP_TOTA_TRANSMISSION_PATH_E dataPath)
{
app_tota_env.dataPath = dataPath;
}
APP_TOTA_TRANSMISSION_PATH_E app_tota_get_datapath(void)
{
return app_tota_env.dataPath;
}
/*---------------------------------------------------------------------------------------------------------------------------*/
static bool _is_tota_connect = false;
/**/
void tota_connected_handle()
{
_is_tota_connect = true;
}
/**/
void tota_disconnected_handle()
{
_is_tota_connect = false;
}
bool is_tota_connected()
{
return _is_tota_connect;
}
/*
** @func: encrypt && decrypt
*/
static uint8_t encrypt_data[MAX_SPP_PACKET_SIZE];
static uint8_t decrypt_data[MAX_SPP_PACKET_SIZE];
static __unused uint8_t key[ENCRYPT_KEY_SIZE] = {0x2b, 0x7e, 0x15, 0x16, 0x28, 0xae, 0xd2, 0xa6, 0xab, 0xf7, 0x15, 0x88, 0x09, 0xcf, 0x4f, 0x3c};
static __unused uint8_t iv[ENCRYPT_KEY_SIZE] = {0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f};
void tota_set_encrypt_key_from_hash_key(uint8_t * hash_key)
{
for (uint8_t i = 0; i < ENCRYPT_KEY_SIZE; i ++)
{
key[i] = hash_key[2*i];
}
TOTA_LOG_DBG(0,"aes key:");
DUMP8("%02x ", key, ENCRYPT_KEY_SIZE);
}
void test_aes_encode_decode()
{
uint8_t w[16] = {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1};
uint8_t z[16] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
uint8_t x[16] = {1, 1, 1, 1, 1, 1, 1, 1};
uint8_t dw[16]= {0x69, 0x0a, 0x6c, 0x5e, 0xd6, 0x66, 0x51, 0x25, 0x97, 0xfa, 0x1e, 0x6e, 0xcc, 0xfc, 0x1b, 0xdb};
uint8_t dz[16]= {0x50, 0xfe, 0x67, 0xcc, 0x99, 0x6d, 0x32, 0xb6, 0xda, 0x09, 0x37, 0xe9, 0x9b, 0xaf, 0xec, 0x60};
uint32_t length = 0;
tota_encrypt_packet(w, 16, &length);
osDelay(50);
tota_encrypt_packet(z, 16, &length);
osDelay(50);
tota_decrypt_packet(dw, 16, &length);
osDelay(50);
tota_decrypt_packet(dz, 16, &length);
tota_encrypt_packet(x, 8, &length);
osDelay(50);
}
uint8_t * tota_encrypt_packet(uint8_t * in, uint32_t inLen, uint32_t * poutLen)
{
uint32_t morebytes = inLen % ENCRYPT_KEY_SIZE? ENCRYPT_KEY_SIZE: 0;
uint32_t outLen = (inLen/ENCRYPT_KEY_SIZE)*ENCRYPT_KEY_SIZE + morebytes;
TOTA_LOG_DBG(0,"raw data:");
DUMP8("0x%02x, ", in, inLen);
#if defined(TOTA)
AES128_CBC_encrypt_buffer(encrypt_data, in, inLen, key, iv);
#endif
*poutLen = outLen;
TOTA_LOG_DBG(2,"encrypt data: %u -> %u", inLen, outLen);
DUMP8("0x%02x, ", encrypt_data, *poutLen);
return encrypt_data;
}
/*
** inLen must 16 times
*/
uint8_t * tota_decrypt_packet(uint8_t * in, uint32_t inLen, uint32_t * poutLen)
{
uint32_t morebytes = inLen % ENCRYPT_KEY_SIZE? ENCRYPT_KEY_SIZE: 0;
uint32_t outLen = (inLen/ENCRYPT_KEY_SIZE)*ENCRYPT_KEY_SIZE + morebytes;
TOTA_LOG_DBG(0,"raw data:");
DUMP8("0x%02x, ", in, inLen);
#if defined(TOTA)
AES128_CBC_decrypt_buffer(decrypt_data, in, inLen, key, iv);
#endif
*poutLen = outLen;
TOTA_LOG_DBG(2,"decrypt data: %u -> %u", inLen, outLen);
DUMP8("0x%02x, ", decrypt_data, *poutLen);
return decrypt_data;
}
//TODO:
static char strBuf[MAX_SPP_PACKET_SIZE-4];
void tota_printf(const char * format, ...)
{
va_list vlist;
va_start(vlist, format);
vsprintf(strBuf, format, vlist);
va_end(vlist);
app_tota_send_command(OP_TOTA_STRING, (uint8_t*)strBuf, strlen(strBuf), app_tota_get_datapath());
}
/*---------------------------------------------------------------------------------------------------------------------------*/
static void app_tota_demo_cmd_handler(APP_TOTA_CMD_CODE_E funcCode, uint8_t* ptrParam, uint32_t paramLen)
{
TOTA_LOG_DBG(2,"Func code 0x%x, param len %d", funcCode, paramLen);
TOTA_LOG_DBG(0,"Param content:");
DUMP8("%02x ", ptrParam, paramLen);
}
TOTA_COMMAND_TO_ADD(OP_TOTA_DEMO_CMD, app_tota_demo_cmd_handler, false, 0, NULL );