/* Copyright Statement:
 * *
 * * This software/firmware and related documentation ("MediaTek Software") are
 * * protected under relevant copyright laws. The information contained herein
 * * is confidential and proprietary to MediaTek Inc. and/or its licensors.
 * * Without the prior written permission of MediaTek inc. and/or its licensors,
 * * any reproduction, modification, use or disclosure of MediaTek Software,
 * * and information contained herein, in whole or in part, shall be strictly prohibited.
 * *
 * * MediaTek Inc. (C) 2016. All rights reserved.
 * *
 * * BY OPENING THIS FILE, RECEIVER HEREBY UNEQUIVOCALLY ACKNOWLEDGES AND AGREES
 * * THAT THE SOFTWARE/FIRMWARE AND ITS DOCUMENTATIONS ("MEDIATEK SOFTWARE")
 * * RECEIVED FROM MEDIATEK AND/OR ITS REPRESENTATIVES ARE PROVIDED TO RECEIVER ON
 * * AN "AS-IS" BASIS ONLY. MEDIATEK EXPRESSLY DISCLAIMS ANY AND ALL WARRANTIES,
 * * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE IMPLIED WARRANTIES OF
 * * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE OR NONINFRINGEMENT.
 * * NEITHER DOES MEDIATEK PROVIDE ANY WARRANTY WHATSOEVER WITH RESPECT TO THE
 * * SOFTWARE OF ANY THIRD PARTY WHICH MAY BE USED BY, INCORPORATED IN, OR
 * * SUPPLIED WITH THE MEDIATEK SOFTWARE, AND RECEIVER AGREES TO LOOK ONLY TO SUCH
 * * THIRD PARTY FOR ANY WARRANTY CLAIM RELATING THERETO. RECEIVER EXPRESSLY ACKNOWLEDGES
 * * THAT IT IS RECEIVER'S SOLE RESPONSIBILITY TO OBTAIN FROM ANY THIRD PARTY ALL PROPER LICENSES
 * * CONTAINED IN MEDIATEK SOFTWARE. MEDIATEK SHALL ALSO NOT BE RESPONSIBLE FOR ANY MEDIATEK
 * * SOFTWARE RELEASES MADE TO RECEIVER'S SPECIFICATION OR TO CONFORM TO A PARTICULAR
 * * STANDARD OR OPEN FORUM. RECEIVER'S SOLE AND EXCLUSIVE REMEDY AND MEDIATEK'S ENTIRE AND
 * * CUMULATIVE LIABILITY WITH RESPECT TO THE MEDIATEK SOFTWARE RELEASED HEREUNDER WILL BE,
 * * AT MEDIATEK'S OPTION, TO REVISE OR REPLACE THE MEDIATEK SOFTWARE AT ISSUE,
 * * OR REFUND ANY SOFTWARE LICENSE FEES OR SERVICE CHARGE PAID BY RECEIVER TO
 * * MEDIATEK FOR SUCH MEDIATEK SOFTWARE AT ISSUE.
 * *
 * * The following software/firmware and/or related documentation ("MediaTek Software")
 * * have been modified by MediaTek Inc. All revisions are subject to any receiver's
 * * applicable license agreements with MediaTek Inc.
 * */
#ifdef MTK_MESH_SUPPORT
#if (SMP_DEBUG == TRUE)
#include <stdio.h>
#endif
#include "bt_target.h"
#include "stack/smp/p_256_ecc_pp.h"
#include "stack/smp/smp_int.h"
#include "mtk_btm_ble_api.h"
#endif

#ifdef MTK_MESH_SUPPORT
EXPORT_SYMBOL void smp_generate_key(uint8_t pub_key[BT_OCTET32_LEN * 2], uint8_t pri_key[BT_OCTET32_LEN]) {
  Point public_key;
  BT_OCTET32 private_key;

  SMP_TRACE_DEBUG("%s", __func__);
  memcpy(private_key, pri_key, BT_OCTET32_LEN);
  ECC_PointMult(&public_key, &(curve_p256.G), (uint32_t*)private_key);
  memcpy(pub_key, public_key.x, BT_OCTET32_LEN);
  memcpy(&pub_key[BT_OCTET32_LEN], public_key.y, BT_OCTET32_LEN);

#if (SMP_DEBUG == TRUE)
  smp_debug_print_nbyte_little_endian (pri_key, "private", BT_OCTET32_LEN);
  smp_debug_print_nbyte_little_endian (public_key.x, "public(x)", BT_OCTET32_LEN);
  smp_debug_print_nbyte_little_endian (public_key.y, "public(y)", BT_OCTET32_LEN);
#endif
}

EXPORT_SYMBOL void smp_calculate_dhkey(const uint8_t pub_key[BT_OCTET32_LEN * 2], const uint8_t pri_key[BT_OCTET32_LEN], uint8_t secret[BT_OCTET32_LEN]) {
  Point peer_publ_key, new_publ_key;

  SMP_TRACE_DEBUG("%s", __func__);

  memcpy(peer_publ_key.x, pub_key, BT_OCTET32_LEN);
  memcpy(peer_publ_key.y, &pub_key[BT_OCTET32_LEN], BT_OCTET32_LEN);
  ECC_PointMult(&new_publ_key, &peer_publ_key, (uint32_t*)pri_key);
  memcpy(secret, new_publ_key.x, BT_OCTET32_LEN);

#if (SMP_DEBUG == TRUE)
  smp_debug_print_nbyte_little_endian(pri_key, "private", BT_OCTET32_LEN);
  smp_debug_print_nbyte_little_endian(pub_key, "public(x)", BT_OCTET32_LEN);
  smp_debug_print_nbyte_little_endian(&pub_key[32], "public(y)", BT_OCTET32_LEN);
  smp_debug_print_nbyte_little_endian(secret, "secret", BT_OCTET32_LEN);
#endif
}
#endif