* Copyright (c) 2020 Huawei Device Co., Ltd.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "securec.h"
#include "commonutil.h"
#include "distribution.h"
#include "log.h"
#include "parsedata.h"
#include "jsonutil.h"
#include "auth_info.h"
#include "add_auth_info.h"
#include "build_object.h"
#include "mem_stat.h"
#include "huks_adapter.h"
#include "sec_clone_server.h"
#define LIST_TRUST_PEER_DEF_COUNT 0
#define FREE_SEND_DATA_FUNC(name) \
{ \
if (((name)->cipher.val != NULL) && ((name)->cipher.length > 0)) { \
FREE((name)->cipher.val); \
(name)->cipher.val = NULL; \
} \
break; \
}
#define FREE_SEC_SEND_DATA_FUNC(name) \
{ \
if ((name)->val != NULL) { \
FREE((name)->val); \
(name)->val = NULL; \
} \
break; \
}
#ifdef DESC
#undef DESC
#endif
#define DESC(...) 1
static void encap_inform_message(int32_t error_code, struct message *send);
static int32_t deserialize_message(const struct uint8_buff *data, struct message *receive);
static int32_t build_send_data_by_struct(const struct message *message, void **send_data, uint32_t *send_data_len);
static void destroy_receive_data_struct(const struct message *message);
static void destroy_send_data(struct message *message);
static void set_result(struct hichain *hichain, uint16_t rcv_msg_code,
uint16_t snd_msg_code, int32_t error_code, int32_t errorCodeRecv);
static int32_t check_identity(const struct session_identity *identity);
static int32_t check_call_back(const struct hc_call_back *call_back);
static int32_t check_auth_info(const struct hc_user_info *user_info);
static int32_t GetErrorCode(const struct uint8_buff *data, int32_t *errorCode);
static int32_t delete_base_key(struct service_id service_id, struct operation_parameter para);
static int32_t delete_public_key(hc_handle handle, struct service_id service_id, int32_t user_type);
#if !(defined(_CUT_STS_) || defined(_CUT_STS_SERVER_) || defined(_CUT_EXCHANGE_) || defined(_CUT_EXCHANGE_SERVER_))
static void build_self_lt_key_pair(const struct hichain *hichain);
#endif
#if DESC("api")
DLL_API_PUBLIC hc_handle get_instance(const struct session_identity *identity, enum hc_type type,
const struct hc_call_back *call_back)
{
LOGI("Begin get instance");
if (check_identity(identity) != HC_OK) {
LOGE("Identity error");
return NULL;
}
if (check_call_back(call_back) != HC_OK) {
LOGE("Call back error");
return NULL;
}
#if !(defined(_CUT_STS_) || defined(_CUT_STS_SERVER_) || defined(_CUT_EXCHANGE_) || defined(_CUT_EXCHANGE_SERVER_))
int32_t ret = key_info_init();
if (ret != HC_OK) {
LOGE("Call key info init failed, status=%d", ret);
return NULL;
}
#endif
struct hichain *hichain = (struct hichain *)MALLOC(sizeof(struct hichain));
if (hichain == NULL) {
LOGE("Alloc memory failed");
return NULL;
}
(void)memset_s(hichain, sizeof(*hichain), 0, sizeof(*hichain));
hichain->identity = *identity;
hichain->type = type;
hichain->state = INIT_STATE;
hichain->last_state = INIT_STATE;
hichain->cb = *call_back;
#if !(defined(_CUT_STS_) || defined(_CUT_STS_SERVER_) || defined(_CUT_EXCHANGE_) || defined(_CUT_EXCHANGE_SERVER_))
build_self_lt_key_pair(hichain);
#endif
LOGI("Get instance success");
return hichain;
}
DLL_API_PUBLIC void destroy(hc_handle *handle)
{
LOGI("Begin destroy");
check_ptr_return(handle);
check_ptr_return(*handle);
struct hichain *hichain = (struct hichain *)*handle;
if (hichain->pake_server != NULL) {
destroy_pake_server(hichain->pake_server);
}
if (hichain->pake_client != NULL) {
destroy_pake_client(hichain->pake_client);
}
if (hichain->sts_server != NULL) {
destroy_sts_server(hichain->sts_server);
}
if (hichain->sts_client != NULL) {
destroy_sts_client(hichain->sts_client);
}
if (hichain->auth_info != NULL) {
destroy_auth_client(hichain->auth_info);
}
if (hichain->sec_clone_server != NULL) {
destroy_sec_clone_server(hichain->sec_clone_server);
}
FREE(hichain);
*handle = NULL;
LOGI("End destroy");
}
DLL_API_PUBLIC int32_t receive_data(hc_handle handle, struct uint8_buff *data)
{
LOGI("Begin receive data");
check_ptr_return_val(handle, HC_INPUT_ERROR);
check_ptr_return_val(data, HC_INPUT_ERROR);
check_ptr_return_val(data->val, HC_INPUT_ERROR);
LOGI("Receive data from peer");
struct hichain *hichain = (struct hichain *)handle;
struct message receive = { 0, 0, 0 };
struct message send = { INFORM_MESSAGE, 0, 0 };
void *send_data = NULL;
uint32_t send_data_len = 0;
int32_t ret = deserialize_message(data, &receive);
if (ret != HC_OK) {
goto inform;
}
struct header_analysis nav = navigate_message(receive.msg_code);
ret = check_message_support(hichain, &nav, &receive);
if (ret != HC_OK) {
goto inform;
}
ret = build_object(hichain, nav.modular, !nav.is_request_msg, NULL);
if (ret != HC_OK) {
goto inform;
}
ret = proc_message(hichain, &nav, &receive, &send);
if (ret != HC_OK) {
goto inform;
}
ret = connect_message(hichain, &nav, &send);
inform:
encap_inform_message(ret, &send);
ret = build_send_data_by_struct(&send, &send_data, &send_data_len);
if (ret == HC_OK) {
DBG_OUT("Send data to peer");
hichain->cb.transmit(&hichain->identity, send_data, send_data_len);
FREE(send_data);
} else if (ret == HC_NO_MESSAGE_TO_SEND) {
LOGI("Had no message to send");
ret = HC_OK;
} else {
LOGE("build send data failed, error code is %d", ret);
}
int32_t errorCode = HC_OK;
GetErrorCode(data, &errorCode);
set_result(hichain, receive.msg_code, send.msg_code, ret, errorCode);
destroy_receive_data_struct(&receive);
destroy_send_data(&send);
LOGI("End receive data");
return ret;
}
static int32_t triggered_sts_client(struct hichain *hichain, int32_t operation_code);
int32_t remove_auth_info(hc_handle handle, const struct operation_parameter *params,
const struct hc_auth_id *auth_id, int32_t user_type)
#if (defined(_SUPPORT_SEC_CLONE_) || defined(_SUPPORT_SEC_CLONE_SERVER_))
{
LOGI("Begin remove auth info");
check_ptr_return_val(handle, HC_INPUT_ERROR);
check_ptr_return_val(params, HC_INPUT_ERROR);
check_ptr_return_val(auth_id, HC_INPUT_ERROR);
struct hichain *hichain = (struct hichain *)handle;
int32_t ret = build_object(hichain, STS_MODULAR, true, params);
if (ret != HC_OK) {
LOGE("Build sts client sub object failed, error code is %d", ret);
return ret;
}
struct auth_info_cache auth_info = {
.user_type = user_type,
.auth_id = *auth_id
};
ret = build_object(hichain, REMOVE_MODULAR, true, &auth_info);
if (ret != HC_OK) {
LOGE("Build remove client sub object failed, error code is %d", ret);
return ret;
}
ret = triggered_sts_client(hichain, REMOVE_AUTHINFO);
LOGI("Triggered sts client error code is %d", ret);
LOGI("End remove auth info");
return ret;
}
#else
{
LOGE("Secclone has been cut, remove auth info not support");
(void)handle;
(void)params;
(void)auth_id;
(void)user_type;
return HC_UNSUPPORT;
}
#endif
static int32_t triggered_pake_client(struct hichain *hichain, int32_t operation_code);
DLL_API_PUBLIC int32_t start_pake(hc_handle handle, const struct operation_parameter *params)
{
LOGI("Begin start pake");
check_ptr_return_val(handle, HC_INPUT_ERROR);
check_ptr_return_val(params, HC_INPUT_ERROR);
struct hichain *hichain = (struct hichain *)handle;
int32_t ret = build_object(hichain, PAKE_MODULAR, true, params);
if (ret != HC_OK) {
LOGE("Build pake client sub object failed, error code is %d", ret);
return ret;
}
ret = triggered_pake_client(hichain, BIND);
LOGI("Triggered pake client error code is %d", ret);
LOGI("End start pake");
return ret;
}
#ifndef _CUT_API_
DLL_API_PUBLIC int32_t authenticate_peer(hc_handle handle, struct operation_parameter *params)
{
LOGI("Begin authenticate peer");
check_ptr_return_val(handle, HC_INPUT_ERROR);
check_ptr_return_val(params, HC_INPUT_ERROR);
struct hichain *hichain = (struct hichain *)handle;
int32_t ret = build_object(hichain, STS_MODULAR, true, params);
if (ret != HC_OK) {
LOGE("Build sts client sub object failed, error code is %d", ret);
return ret;
}
ret = triggered_sts_client(hichain, AUTHENTICATE);
LOGI("Triggered sts client error code is %d", ret);
LOGI("End authenticate peer");
return ret;
}
DLL_API_PUBLIC int32_t delete_local_auth_info(hc_handle handle, struct hc_user_info *user_info)
{
LOGI("Begin delete local auth info");
check_ptr_return_val(handle, HC_INPUT_ERROR);
if (check_auth_info(user_info) != HC_OK) {
LOGE("User info is error");
return HC_INPUT_ERROR;
}
struct hichain *hichain = (struct hichain *)handle;
struct service_id service_id = generate_service_id(&hichain->identity);
if (service_id.length == 0) {
LOGE("Generate service id failed");
return HC_GEN_SERVICE_ID_FAILED;
}
struct hc_pin pin = { 0, {0} };
struct operation_parameter para;
(void)memset_s(¶, sizeof(para), 0, sizeof(para));
hichain->cb.get_protocol_params(&hichain->identity, REMOVE_ALL_AUTHINFO, &pin, ¶);
if (para.self_auth_id.length > 0) {
if (memcmp(para.self_auth_id.auth_id, user_info->auth_id.auth_id, para.self_auth_id.length) == 0) {
int32_t ret_base = delete_base_key(service_id, para);
int32_t ret_accessor = delete_public_key(handle, service_id, KEY_ALIAS_ACCESSOR_PK);
int32_t ret_controller = delete_public_key(handle, service_id, KEY_ALIAS_CONTROLLER_PK);
if ((ret_base != HC_OK) || (ret_accessor != HC_OK) || (ret_controller != HC_OK)) {
LOGE("delete all key failed");
return ERROR_CODE_FAILED;
}
return HC_OK;
}
}
enum huks_key_alias_type alias_type = (user_info->user_type == HC_USER_TYPE_ACCESSORY ?
KEY_ALIAS_ACCESSOR_PK : KEY_ALIAS_CONTROLLER_PK);
struct hc_key_alias alias = generate_key_alias(&service_id, &(user_info->auth_id), alias_type);
if (alias.length == 0) {
LOGE("Generate key alias failed");
return HC_GEN_ALIAS_FAILED;
}
int32_t ret = check_lt_public_key_exist(&alias);
if (ret != HC_OK) {
LOGE("Check lt public key not exist!");
return HC_OK;
}
ret = delete_lt_public_key(&alias);
if (ret != HC_OK) {
LOGE("delete lt public key is %d", ret);
return ret;
}
LOGI("End delete local auth info");
return HC_OK;
}
DLL_API_PUBLIC int32_t is_trust_peer(hc_handle handle, struct hc_user_info *user_info)
{
LOGI("Begin is trust peer");
check_ptr_return_val(handle, HC_NOT_TRUST_PEER);
if (check_auth_info(user_info) != HC_OK) {
LOGE("User info is error");
return HC_NOT_TRUST_PEER;
}
struct hichain *hichain = (struct hichain *)handle;
struct service_id service_id = generate_service_id(&hichain->identity);
if (service_id.length == 0) {
LOGE("Generate service id failed");
return HC_GEN_SERVICE_ID_FAILED;
}
enum huks_key_alias_type alias_type = (user_info->user_type == HC_USER_TYPE_ACCESSORY ?
KEY_ALIAS_ACCESSOR_PK : KEY_ALIAS_CONTROLLER_PK);
struct hc_key_alias alias = generate_key_alias(&service_id, &(user_info->auth_id), alias_type);
if (alias.length == 0) {
LOGE("Generate key alias failed");
return HC_GEN_ALIAS_FAILED;
}
int32_t ret = check_lt_public_key_exist(&alias);
if (ret != ERROR_CODE_SUCCESS) {
LOGE("Check lt public key exist is %d", ret);
return HC_NOT_TRUST_PEER;
}
struct huks_key_type key_type;
struct hc_auth_id auth_id;
(void)memset_s(&key_type, sizeof(key_type), 0, sizeof(key_type));
ret = get_lt_key_info(&alias, &key_type, &auth_id);
if (ret != ERROR_CODE_SUCCESS) {
LOGE("Check lt public key exist is %d", ret);
return HC_NOT_TRUST_PEER;
}
LOGI("End is trust peer");
if (key_type.user_type == (uint8_t)HC_USER_TYPE_ACCESSORY) {
return HC_ACCESSORY_TRUST_PEER;
}
if (key_type.pair_type == (uint8_t)HC_PAIR_TYPE_BIND) {
return HC_BINDED_TRUST_PEER;
} else {
return HC_AUTHED_TRUST_PEER;
}
}
DLL_API_PUBLIC uint32_t list_trust_peers(hc_handle handle, int32_t trust_user_type,
struct hc_auth_id *owner_auth_id, struct hc_auth_id **auth_id_list)
{
LOGI("Begin list trust peers");
struct hichain *hichain = (struct hichain *)handle;
check_ptr_return_val(hichain, LIST_TRUST_PEER_DEF_COUNT);
check_ptr_return_val(auth_id_list, LIST_TRUST_PEER_DEF_COUNT);
check_ptr_return_val(*auth_id_list, LIST_TRUST_PEER_DEF_COUNT);
if ((trust_user_type != HC_USER_TYPE_ACCESSORY) && (trust_user_type != HC_USER_TYPE_CONTROLLER)) {
LOGE("user type is not support");
return LIST_TRUST_PEER_DEF_COUNT;
}
if (owner_auth_id != NULL) {
struct service_id srv_id = generate_service_id(&hichain->identity);
if (srv_id.length == 0) {
LOGE("Generate service id failed");
return 0;
}
enum huks_key_alias_type alias_type = (trust_user_type == HC_USER_TYPE_ACCESSORY ?
KEY_ALIAS_ACCESSOR_PK : KEY_ALIAS_CONTROLLER_PK);
struct hc_key_alias owner_alias = generate_key_alias(&srv_id, owner_auth_id, alias_type);
if (owner_alias.length == 0) {
LOGE("Generate key alias failed");
return 0;
}
if (check_lt_public_key_exist(&owner_alias) != ERROR_CODE_SUCCESS) {
LOGE("not found this owner");
return 0;
}
if (check_key_alias_is_owner(&owner_alias) != ERROR_CODE_SUCCESS) {
LOGE("hc_auth_id is not owner");
return 0;
}
}
uint32_t count = LIST_TRUST_PEER_DEF_COUNT;
int32_t ret = get_lt_public_key_list(owner_auth_id, trust_user_type, *auth_id_list, &count);
LOGI("End list trust peers");
if ((ret != ERROR_CODE_SUCCESS) || (count == LIST_TRUST_PEER_DEF_COUNT)) {
return LIST_TRUST_PEER_DEF_COUNT;
}
return count;
}
#endif
#endif
struct msg_result_map {
enum message_code msg_code;
enum hc_result result;
enum hichain_state state;
};
static void set_result_by_map(struct hichain *hichain, const struct msg_result_map *map, int32_t errorCodeRecv)
{
if (map == NULL) {
return;
}
if (map->state != OPERATION_STATE) {
goto out;
}
if ((hichain->operation_code != AUTHENTICATE) && (hichain->operation_code != AUTH_KEY_AGREEMENT)) {
goto out;
}
if (hichain->state != OVER_STATE) {
hichain->last_state = hichain->state;
hichain->state = OVER_STATE;
}
hichain->cb.set_service_result(&hichain->identity, END_SUCCESS, errorCodeRecv);
return;
out:
if (hichain->state != map->state) {
hichain->last_state = hichain->state;
hichain->state = map->state;
}
hichain->cb.set_service_result(&hichain->identity, map->result, errorCodeRecv);
}
const struct msg_result_map *select_result_map(uint16_t rcv_msg_code, const struct msg_result_map *map, uint32_t n)
{
for (uint32_t i = 0; i < n; i++) {
if (rcv_msg_code == map[i].msg_code) {
return &map[i];
}
}
return NULL;
}
static void set_result(struct hichain *hichain, uint16_t rcv_msg_code,
uint16_t snd_msg_code, int32_t error_code, int32_t errorCodeRecv)
{
if (error_code != HC_OK) {
LOGE("Error code is not ok, and set end failed");
goto error;
}
if (snd_msg_code == INFORM_MESSAGE) {
LOGE("Send an inform message, and set end failed");
goto error;
}
const struct msg_result_map map[] = { { PAKE_REQUEST, KEY_AGREEMENT_PROCESSING, KEY_AGREEMENT_STATE },
{ PAKE_RESPONSE, KEY_AGREEMENT_PROCESSING, KEY_AGREEMENT_STATE },
{ PAKE_CLIENT_CONFIRM, KEY_AGREEMENT_END, OPERATION_STATE },
{ PAKE_SERVER_CONFIRM_RESPONSE, KEY_AGREEMENT_END, OPERATION_STATE },
{ EXCHANGE_REQUEST, END_SUCCESS, OVER_STATE },
{ EXCHANGE_RESPONSE, END_SUCCESS, OVER_STATE },
{ AUTH_START_REQUEST, KEY_AGREEMENT_PROCESSING, KEY_AGREEMENT_STATE },
{ AUTH_START_RESPONSE, KEY_AGREEMENT_PROCESSING, KEY_AGREEMENT_STATE },
{ AUTH_ACK_REQUEST, KEY_AGREEMENT_END, OPERATION_STATE },
{ AUTH_ACK_RESPONSE, KEY_AGREEMENT_END, OPERATION_STATE },
{ ADD_AUTHINFO_REQUEST, END_SUCCESS, OVER_STATE },
{ ADD_AUTHINFO_RESPONSE, END_SUCCESS, OVER_STATE },
{ REMOVE_AUTHINFO_REQUEST, END_SUCCESS, OVER_STATE },
{ REMOVE_AUTHINFO_RESPONSE, END_SUCCESS, OVER_STATE },
{ SEC_CLONE_START_REQUEST, OPERATION_PROCESSING, OPERATION_STATE },
{ SEC_CLONE_ACK_REQUEST, END_SUCCESS, OVER_STATE },
{ INFORM_MESSAGE, END_FAILED, OVER_STATE },
{ INVALID_MESSAGE, KEY_AGREEMENT_PROCESSING, KEY_AGREEMENT_STATE } };
const struct msg_result_map *map_ptr = select_result_map(rcv_msg_code, map,
sizeof(map) / sizeof(struct msg_result_map));
set_result_by_map(hichain, map_ptr, errorCodeRecv);
return;
error:
hichain->last_state = hichain->state;
hichain->state = OVER_STATE;
hichain->cb.set_service_result(&hichain->identity, END_FAILED, errorCodeRecv);
}
static void encap_inform_message(int32_t error_code, struct message *send)
{
if ((error_code == HC_OK) || (send->msg_code != INFORM_MESSAGE) || (send->payload != NULL)) {
return;
}
#if (defined(_CUT_EXCHANGE_) || defined(_CUT_EXCHANGE_SERVER_))
if (error_code == HC_UNSUPPORT) {
send->msg_code = INVALID_MESSAGE;
return;
}
#endif
int32_t *err = (int32_t *)MALLOC(sizeof(int32_t));
if (err == NULL) {
LOGE("Malloc for encape inform message failed");
return;
}
*err = error_code;
send->payload = err;
}
static int32_t triggered_pake_client(struct hichain *hichain, int32_t operation_code)
#if !(defined(_CUT_PAKE_) || defined(_CUT_PAKE_CLIENT_))
{
hichain->operation_code = operation_code;
hichain->pake_client->operation_code = operation_code;
struct message send = {
.msg_code = PAKE_REQUEST,
.rsv = 0,
.payload = NULL
};
int32_t ret = send_pake_start_request(hichain->pake_client, &send);
if (ret != HC_OK) {
LOGE("Object %u build sts start request failed, error code is %d", pake_client_sn(hichain->pake_client), ret);
return HC_BUILD_SEND_DATA_FAILED;
}
void *send_data = NULL;
uint32_t send_data_len = 0;
ret = build_send_data_by_struct(&send, &send_data, &send_data_len);
if (ret != HC_OK) {
LOGW("build send data failed, error code is %d", ret);
} else {
DBG_OUT("send_data:%s", (uint8_t *)send_data);
hichain->cb.transmit(&hichain->identity, send_data, send_data_len);
FREE(send_data);
}
set_result(hichain, INVALID_MESSAGE, PAKE_REQUEST, ret, HC_OK);
destroy_send_data(&send);
return ret;
}
#else
{
(void)hichain;
(void)operation_code;
LOGE("Pake client has been cut, triggered pake client not support");
return HC_UNSUPPORT;
}
#endif
static int32_t triggered_sts_client(struct hichain *hichain, int32_t operation_code)
#if !(defined(_CUT_STS_) || defined(_CUT_STS_CLIENT_))
{
hichain->operation_code = operation_code;
hichain->sts_client->operation_code = operation_code;
struct message send = {
.msg_code = AUTH_START_REQUEST,
.rsv = 0,
.payload = NULL
};
int32_t ret = send_sts_start_request(hichain->sts_client, &send);
if (ret != HC_OK) {
LOGE("Object %u build sts start request failed, error code is %d", sts_client_sn(hichain->sts_client), ret);
return HC_BUILD_SEND_DATA_FAILED;
}
void *send_data = NULL;
uint32_t send_data_len = 0;
ret = build_send_data_by_struct(&send, &send_data, &send_data_len);
if (ret != HC_OK) {
LOGW("build send data failed, error code is %d", ret);
} else {
DBG_OUT("send_data:%s", (uint8_t *)send_data);
hichain->cb.transmit(&hichain->identity, send_data, send_data_len);
FREE(send_data);
}
set_result(hichain, INVALID_MESSAGE, AUTH_START_REQUEST, ret, HC_OK);
destroy_send_data(&send);
return ret;
}
#else
{
(void)hichain;
(void)operation_code;
LOGE("stsclient has been cut, triggered_sts_client not support");
return HC_UNSUPPORT;
}
#endif
static int32_t build_struct_by_receive_data(uint32_t msg_code, const char *payload_data,
enum json_object_data_type type, struct message *message);
static int32_t deserialize_message(const struct uint8_buff *data, struct message *receive)
{
struct pass_through_data *pass_through_data = parse_data((const char *)data->val);
if (pass_through_data == NULL) {
LOGE("Parse data failed");
return HC_BUILD_OBJECT_FAILED;
}
#if (defined(_CUT_EXCHANGE_) || defined(_CUT_EXCHANGE_SERVER_))
if (pass_through_data->message_code == EXCHANGE_REQUEST) {
free_data(pass_through_data);
pass_through_data = NULL;
return HC_UNSUPPORT;
}
#endif
int32_t ret = build_struct_by_receive_data(pass_through_data->message_code, pass_through_data->payload_data,
JSON_STRING_DATA, receive);
if (ret != HC_OK) {
LOGE("Build struct by receive data failed, error code is %d", ret);
}
free_data(pass_through_data);
pass_through_data = NULL;
return ret;
}
typedef void *(*parse_message_func)(const char *pay_load, enum json_object_data_type type);
struct parse_message_map {
enum message_code msg_code;
parse_message_func parse_message;
};
static int32_t build_struct_by_receive_data(uint32_t msg_code, const char *payload_data,
enum json_object_data_type type, struct message *message)
{
const struct parse_message_map map[] = { { PAKE_RESPONSE, parse_pake_response },
{ PAKE_SERVER_CONFIRM_RESPONSE, parse_pake_server_confirm },
{ AUTH_START_RESPONSE, parse_auth_start_response },
{ AUTH_ACK_RESPONSE, parse_auth_ack_response },
{ REMOVE_AUTHINFO_REQUEST, parse_rmv_auth_info_request },
{ REMOVE_AUTHINFO_RESPONSE, parse_rmv_auth_info_response },
{ EXCHANGE_RESPONSE, parse_exchange_response },
{ SEC_CLONE_START_REQUEST, sec_clone_parse_client_request },
{ SEC_CLONE_ACK_REQUEST, sec_clone_parse_client_ack } };
for (uint32_t i = 0; i < sizeof(map) / sizeof(struct parse_message_map); i++) {
if (map[i].msg_code != msg_code) {
continue;
}
void *payload = map[i].parse_message(payload_data, type);
if (payload == NULL) {
return HC_BUILD_OBJECT_FAILED;
}
message->msg_code = map[i].msg_code;
message->payload = payload;
return HC_OK;
}
LOGE("Unsupport parse 0x%04x message", message->msg_code);
return HC_UNKNOW_MESSAGE;
}
typedef void (*free_message_func)(void *obj);
struct free_message_map {
enum message_code msg_code;
free_message_func free_message;
};
static void destroy_receive_data_struct(const struct message *message)
{
const struct free_message_map map[] = { { PAKE_REQUEST, free_pake_request },
{ PAKE_RESPONSE, free_pake_response },
{ PAKE_CLIENT_CONFIRM, free_pake_client_confirm },
{ PAKE_SERVER_CONFIRM_RESPONSE, free_pake_server_confirm },
{ AUTH_START_REQUEST, free_auth_start_request },
{ AUTH_START_RESPONSE, free_auth_start_response },
{ AUTH_ACK_REQUEST, free_auth_ack_request },
{ AUTH_ACK_RESPONSE, free_auth_ack_response },
{ REMOVE_AUTHINFO_REQUEST, free_rmv_auth_info_request },
{ REMOVE_AUTHINFO_RESPONSE, free_rmv_auth_info_response },
{ EXCHANGE_REQUEST, free_exchange_request },
{ EXCHANGE_RESPONSE, free_exchange_response },
{ SEC_CLONE_START_REQUEST, sec_clone_free_client_request },
{ SEC_CLONE_ACK_REQUEST, sec_clone_free_client_ack } };
for (uint32_t i = 0; i < sizeof(map) / sizeof(struct free_message_map); i++) {
if (map[i].msg_code == message->msg_code) {
map[i].free_message(message->payload);
}
}
}
typedef char *(*make_message_func)(void *data);
struct make_message_map {
enum message_code msg_code;
make_message_func make_message;
};
static int32_t build_send_data_by_struct(const struct message *message, void **send_data, uint32_t *send_data_len)
{
const struct make_message_map map[] = { { PAKE_REQUEST, make_pake_request },
{ PAKE_CLIENT_CONFIRM, make_pake_client_confirm },
{ AUTH_START_REQUEST, make_auth_start_request },
{ AUTH_ACK_REQUEST, make_auth_ack_request },
{ REMOVE_AUTHINFO_REQUEST, make_rmv_auth_info_request },
{ REMOVE_AUTHINFO_RESPONSE, make_rmv_auth_info_response },
{ EXCHANGE_REQUEST, make_exchange_request },
{ SEC_CLONE_START_RESPONSE, sec_clone_make_srv_proof },
{ SEC_CLONE_ACK_RESPONSE, sec_clone_make_clone_ret },
{ INFORM_MESSAGE, make_inform_message } };
if (message->msg_code == INVALID_MESSAGE) {
return HC_NO_MESSAGE_TO_SEND;
}
if (message->payload == NULL) {
LOGE("Message payload is null");
return HC_BUILD_SEND_DATA_FAILED;
}
for (uint32_t i = 0; i < sizeof(map) / sizeof(struct make_message_map); i++) {
if (map[i].msg_code != message->msg_code) {
continue;
}
*send_data = map[i].make_message(message->payload);
if (*send_data == NULL) {
return HC_BUILD_SEND_DATA_FAILED;
}
*send_data_len = strlen(*send_data);
return HC_OK;
}
LOGE("Unsupport encape 0x%04x message", message->msg_code);
return HC_INNER_ERROR;
}
static void destroy_send_data(struct message *message)
{
if (message->payload == NULL) {
return;
}
switch (message->msg_code) {
case ADD_AUTHINFO_RESPONSE: {
add_response_data *add_res_data = message->payload;
FREE_SEND_DATA_FUNC(add_res_data);
}
case ADD_AUTHINFO_REQUEST: {
add_request_data *add_req_data = message->payload;
FREE_SEND_DATA_FUNC(add_req_data);
}
case REMOVE_AUTHINFO_RESPONSE: {
remove_response_data *rmv_auth_info_res_data = message->payload;
FREE_SEND_DATA_FUNC(rmv_auth_info_res_data);
}
case REMOVE_AUTHINFO_REQUEST: {
remove_request_data *rmv_auth_info_req_data = message->payload;
FREE_SEND_DATA_FUNC(rmv_auth_info_req_data);
}
case EXCHANGE_RESPONSE: {
exchange_response_data *exchange_response = message->payload;
FREE_SEND_DATA_FUNC(exchange_response);
}
case EXCHANGE_REQUEST: {
exchange_request_data *exchange_requeset = message->payload;
FREE_SEND_DATA_FUNC(exchange_requeset);
}
case SEC_CLONE_START_RESPONSE: {
struct uint8_buff *data = message->payload;
FREE_SEC_SEND_DATA_FUNC(data);
}
case SEC_CLONE_ACK_RESPONSE: {
struct uint8_buff *data = message->payload;
FREE_SEC_SEND_DATA_FUNC(data);
}
default:
break;
}
if (message->payload != NULL) {
FREE(message->payload);
message->payload = NULL;
}
}
static int32_t check_identity(const struct session_identity *identity)
{
check_ptr_return_val(identity, HC_INPUT_ERROR);
if (identity->package_name.length > HC_PACKAGE_NAME_BUFF_LEN) {
LOGE("Package name length error, %u > %u", identity->package_name.length, HC_PACKAGE_NAME_BUFF_LEN);
return HC_INPUT_ERROR;
}
if (identity->service_type.length > HC_SERVICE_TYPE_BUFF_LEN) {
LOGE("Service type length error, %u > %u", identity->service_type.length, HC_SERVICE_TYPE_BUFF_LEN);
return HC_INPUT_ERROR;
}
return HC_OK;
}
static int32_t check_call_back(const struct hc_call_back *call_back)
{
check_ptr_return_val(call_back, HC_INPUT_ERROR);
check_ptr_return_val(call_back->transmit, HC_INPUT_ERROR);
check_ptr_return_val(call_back->get_protocol_params, HC_INPUT_ERROR);
check_ptr_return_val(call_back->set_session_key, HC_INPUT_ERROR);
check_ptr_return_val(call_back->set_service_result, HC_INPUT_ERROR);
check_ptr_return_val(call_back->confirm_receive_request, HC_INPUT_ERROR);
return HC_OK;
}
static int32_t check_auth_info(const struct hc_user_info *user_info)
{
check_ptr_return_val(user_info, HC_INPUT_ERROR);
if (user_info->auth_id.length > HC_AUTH_ID_BUFF_LEN) {
LOGE("Auth id length is error, %u > %u", user_info->auth_id.length, HC_AUTH_ID_BUFF_LEN);
return HC_INPUT_ERROR;
}
if ((user_info->user_type != HC_USER_TYPE_CONTROLLER) && (user_info->user_type != HC_USER_TYPE_ACCESSORY)) {
LOGE("User type %d is not controller or accessory", user_info->user_type);
return HC_INPUT_ERROR;
}
return HC_OK;
}
static int32_t delete_base_key(struct service_id service_id, struct operation_parameter para)
{
LOGI("delete base key");
struct hc_key_alias alias_list[HC_BASE_KEY_NUM];
int32_t size = sizeof(struct hc_key_alias);
int32_t length = HC_BASE_KEY_NUM * size;
(void)memset_s(alias_list, length, 0, length);
int32_t pos = 0;
struct hc_key_alias base_alias = generate_key_alias(&service_id, &(para.peer_auth_id), KEY_ALIAS_KEK);
(void)memcpy_s(alias_list + pos, length - pos * size, &base_alias, size);
pos++;
base_alias = generate_key_alias(&service_id, &(para.peer_auth_id), KEY_ALIAS_DEK);
(void)memcpy_s(alias_list + pos, length - pos * size, &base_alias, size);
pos++;
base_alias = generate_key_alias(&service_id, &(para.peer_auth_id), KEY_ALIAS_LT_KEY_PAIR);
(void)memcpy_s(alias_list + pos, length - pos * size, &base_alias, size);
pos++;
base_alias = generate_key_alias(&service_id, &(para.self_auth_id), KEY_ALIAS_LT_KEY_PAIR);
(void)memcpy_s(alias_list + pos, length - pos * size, &base_alias, size);
pos++;
base_alias = generate_key_alias(&service_id, &(para.self_auth_id), KEY_ALIAS_TMP);
(void)memcpy_s(alias_list + pos, length - pos * size, &base_alias, size);
for (uint32_t loop = 0; loop < HC_BASE_KEY_NUM; loop++) {
if (alias_list[loop].length == 0) {
LOGE("Generate key alias failed");
continue;
}
int32_t ret = check_lt_public_key_exist(&alias_list[loop]);
if (ret != HC_OK) {
LOGE("not found key alias %d", loop);
continue;
}
ret = delete_lt_public_key(&alias_list[loop]);
if (ret != HC_OK) {
LOGE("delete auth_alias public key is %d", ret);
return ret;
}
}
return HC_OK;
}
static int32_t delete_public_key(hc_handle handle, struct service_id service_id, int32_t user_type)
{
LOGI("delete public key");
uint32_t length = HC_PUB_KEY_ALIAS_MAX_NUM * sizeof(struct hc_auth_id);
struct hc_auth_id *auth_id_list = (struct hc_auth_id *)MALLOC(length);
if (auth_id_list == NULL) {
LOGE("malloc auth id list failed");
return HC_MALLOC_FAILED;
}
(void)memset_s(auth_id_list, length, 0, length);
uint32_t peers_num = list_trust_peers(handle, user_type, NULL, &auth_id_list);
LOGI("peers_num %u", peers_num);
for (uint32_t loop = 0; loop < peers_num; loop++) {
struct hc_key_alias key_alias = generate_key_alias(&service_id, &auth_id_list[loop], user_type);
if (key_alias.length == 0) {
LOGE("Generate key alias failed");
continue;
}
int32_t ret = check_lt_public_key_exist(&key_alias);
if (ret != HC_OK) {
continue;
}
ret = delete_lt_public_key(&key_alias);
if (ret != HC_OK) {
LOGE("delete key_alias public key is %d", ret);
if (auth_id_list != NULL) {
FREE(auth_id_list);
auth_id_list = NULL;
}
return ret;
}
}
if (auth_id_list != NULL) {
FREE(auth_id_list);
auth_id_list = NULL;
}
return HC_OK;
}
static int32_t ParseInformMessage(const char *payload, enum json_object_data_type dataType, int32_t *errorCode);
static int32_t GetErrorCode(const struct uint8_buff *data, int32_t *errorCode)
{
check_ptr_return_val(data, HC_INPUT_ERROR);
check_ptr_return_val(errorCode, HC_INPUT_ERROR);
struct pass_through_data *passThroughData = parse_data((const char *)data->val);
if (passThroughData == NULL) {
LOGE("Parse data failed");
return HC_INPUT_ERROR;
}
if (passThroughData->message_code == INFORM_MESSAGE) {
int32_t ret = ParseInformMessage(passThroughData->payload_data, JSON_STRING_DATA, errorCode);
free_data(passThroughData);
return ret;
}
free_data(passThroughData);
*errorCode = HC_OK;
return HC_OK;
}
static int32_t ParseInformMessage(const char *payload, enum json_object_data_type dataType, int32_t *errorCode)
{
check_ptr_return_val(payload, HC_INPUT_ERROR);
json_pobject obj = parse_payload(payload, dataType);
if (obj == NULL) {
LOGE("Parse inform message payload failed");
return HC_INPUT_ERROR;
}
*errorCode = get_json_int(obj, FIELD_ERROR_CODE);
if (*errorCode == -1) {
free_payload(obj, dataType);
LOGE("Parse inform message error code is not exist");
return HC_INPUT_ERROR;
}
free_payload(obj, dataType);
LOGI("receive error code: %d", *errorCode);
return HC_OK;
}
#if !(defined(_CUT_STS_) || defined(_CUT_STS_SERVER_)|| defined(_CUT_EXCHANGE_)|| defined(_CUT_EXCHANGE_SERVER_))
static void build_self_lt_key_pair(const struct hichain *hichain)
{
struct hc_pin pin = { 0, {0} };
struct operation_parameter para;
(void)memset_s(¶, sizeof(para), 0, sizeof(para));
hichain->cb.get_protocol_params(&hichain->identity, GENERATE_KEY_PAIR, &pin, ¶);
if (para.self_auth_id.length > 0) {
struct service_id service_id = generate_service_id(&hichain->identity);
if (service_id.length == 0) {
LOGE("Generate service id failed");
return;
}
#if (defined(_SUPPORT_SEC_CLONE_) || defined(_SUPPORT_SEC_CLONE_SERVER_))
struct hc_key_alias alias = generate_key_alias(&service_id, ¶.self_auth_id, KEY_ALIAS_LT_KEY_PAIR);
#else
struct hc_key_alias alias = generate_key_alias(&service_id, ¶.self_auth_id, KEY_ALIAS_ACCESSOR_PK);
#endif
if (alias.length == 0) {
LOGE("Generate key alias failed");
return;
}
int32_t ret = check_lt_public_key_exist(&alias);
if (ret != HC_OK) {
ret = generate_lt_key_pair(&alias, ¶.self_auth_id);
if (ret != HC_OK) {
LOGE("Generate self ltpk return value is %d", ret);
return;
}
DBG_OUT("Generate self ltpk ok");
}
}
}
#endif