#include <stdint.h>

#include <string.h>

#include "ble.h"

#include "ble_hci.h"

#include "ble_advdata.h"

#include "ble_dfu.h"

#include "dfu_app_handler.h"

#include "ble_conn_params.h"

#include "softdevice_handler.h"

#include "app_timer.h"

#include "device_manager.h"

#include "pstorage.h"

#include "nrf_delay.h"

#include "ble_nus.h"

#include "bta_ble.h"

#define IS_SRVC_CHANGED_CHARACT_PRESENT 1 /**< Include or not the service_changed characteristic. if not enabled, the server's database cannot be changed for the lifetime of the device*/

#define DEVICE_NAME "Anonymous" /**< Name of device. Will be included in the advertising data. */

#define MANUFACTURER_NAME "NordicSemiconductor" /**< Manufacturer. Will be passed to Device Information Service. */

#define APP_ADV_INTERVAL 300 /**< The advertising interval (in units of 0.625 ms. This value corresponds to 25 ms). */

#define APP_ADV_TIMEOUT_IN_SECONDS 180 /**< The advertising timeout in units of seconds. */

#define MIN_CONN_INTERVAL MSEC_TO_UNITS(400, UNIT_1_25_MS) /**< Minimum acceptable connection interval (0.4 seconds). */

#define MAX_CONN_INTERVAL MSEC_TO_UNITS(650, UNIT_1_25_MS) /**< Maximum acceptable connection interval (0.65 second). */

#define SLAVE_LATENCY 0 /**< Slave latency. */

#define CONN_SUP_TIMEOUT MSEC_TO_UNITS(4000, UNIT_10_MS) /**< Connection supervisory timeout (4 seconds). */

#define FIRST_CONN_PARAMS_UPDATE_DELAY APP_TIMER_TICKS(5000, APP_TIMER_PRESCALER) /**< Time from initiating event (connect or start of notification) to first time sd_ble_gap_conn_param_update is called (5 seconds). */

#define NEXT_CONN_PARAMS_UPDATE_DELAY APP_TIMER_TICKS(30000, APP_TIMER_PRESCALER)/**< Time between each call to sd_ble_gap_conn_param_update after the first call (30 seconds). */

#define MAX_CONN_PARAMS_UPDATE_COUNT 3 /**< Number of attempts before giving up the connection parameter negotiation. */

#define SEC_PARAM_BOND 1 /**< Perform bonding. */

#define SEC_PARAM_MITM 0 /**< Man In The Middle protection not required. */

#define SEC_PARAM_IO_CAPABILITIES BLE_GAP_IO_CAPS_NONE /**< No I/O capabilities. */

#define SEC_PARAM_OOB 0 /**< Out Of Band data not available. */

#define SEC_PARAM_MIN_KEY_SIZE 7 /**< Minimum encryption key size. */

#define SEC_PARAM_MAX_KEY_SIZE 16 /**< Maximum encryption key size. */

#define DFU_REV_MAJOR 0x00 /** DFU Major revision number to be exposed. */

#define DFU_REV_MINOR 0x01 /** DFU Minor revision number to be exposed. */

#define DFU_REVISION ((DFU_REV_MAJOR << 8) | DFU_REV_MINOR) /** DFU Revision number to be exposed. Combined of major and minor versions. */

#define APP_SERVICE_HANDLE_START 0x000C /**< Handle of first application specific service when when service changed characteristic is present. */

#define BLE_HANDLE_MAX 0xFFFF /**< Max handle value in BLE. */

STATIC_ASSERT(IS_SRVC_CHANGED_CHARACT_PRESENT); /** When having DFU Service support in application the Service Changed Characteristic should always be present. */

static uint16_t m_conn_handle = BLE_CONN_HANDLE_INVALID; /**< Handle of the current connection. */

static dm_application_instance_t m_app_handle; /**< Application identifier allocated by device manager */

static ble_nus_t m_nus; /**< Structure to identify the Nordic UART Service. */

static ble_dfu_t m_dfus; /**< Structure used to identify the DFU service. */

static ble_gap_sec_params_t m_sec_params; /**< Security requirements for this application. */

/**@brief Function for the GAP initialization.

*

* @details This function sets up all the necessary GAP (Generic Access Profile) parameters of the

* device including the device name, appearance, and the preferred connection parameters.

*/

static void gap_params_init(void)

{

uint32_t err_code;

ble_gap_conn_params_t gap_conn_params;

ble_gap_conn_sec_mode_t sec_mode;

BLE_GAP_CONN_SEC_MODE_SET_OPEN(&sec_mode);

err_code = sd_ble_gap_device_name_set(&sec_mode,

(const uint8_t *)DEVICE_NAME,

strlen(DEVICE_NAME));

APP_ERROR_CHECK(err_code);

err_code = sd_ble_gap_appearance_set(BLE_APPEARANCE_UNKNOWN);

APP_ERROR_CHECK(err_code);

memset(&gap_conn_params, 0, sizeof(gap_conn_params));

gap_conn_params.min_conn_interval = MIN_CONN_INTERVAL;

gap_conn_params.max_conn_interval = MAX_CONN_INTERVAL;

gap_conn_params.slave_latency = SLAVE_LATENCY;

gap_conn_params.conn_sup_timeout = CONN_SUP_TIMEOUT;

err_code = sd_ble_gap_ppcp_set(&gap_conn_params);

APP_ERROR_CHECK(err_code);

}

/**@brief Function for stopping advertising.

*/

static void advertising_stop(void)

{

uint32_t err_code;

err_code = sd_ble_gap_adv_stop();

APP_ERROR_CHECK(err_code);

}

/**@brief Function for loading application-specific context after establishing a secure connection.

*

* @details This function will load the application context and check if the ATT table is marked as

* changed. If the ATT table is marked as changed, a Service Changed Indication

* is sent to the peer if the Service Changed CCCD is set to indicate.

*

* @param[in] p_handle The Device Manager handle that identifies the connection for which the context

* should be loaded.

*/

static void app_context_load(dm_handle_t const * p_handle)

{

uint32_t err_code;

static uint32_t context_data;

dm_application_context_t context;

context.len = sizeof(context_data);

context.p_data = (uint8_t *)&context_data;

err_code = dm_application_context_get(p_handle, &context);

if (err_code == NRF_SUCCESS)

{

// Send Service Changed Indication if ATT table has changed.

if ((context_data & (DFU_APP_ATT_TABLE_CHANGED << DFU_APP_ATT_TABLE_POS)) != 0)

{

err_code = sd_ble_gatts_service_changed(m_conn_handle, APP_SERVICE_HANDLE_START, BLE_HANDLE_MAX);

if ((err_code != NRF_SUCCESS) &&

(err_code != BLE_ERROR_INVALID_CONN_HANDLE) &&

(err_code != NRF_ERROR_INVALID_STATE) &&

(err_code != BLE_ERROR_NO_TX_BUFFERS) &&

(err_code != NRF_ERROR_BUSY) &&

(err_code != BLE_ERROR_GATTS_SYS_ATTR_MISSING))

{

APP_ERROR_HANDLER(err_code);

}

}

err_code = dm_application_context_delete(p_handle);

APP_ERROR_CHECK(err_code);

}

else if (err_code == DM_NO_APP_CONTEXT)

{

// No context available. Ignore.

}

else

{

APP_ERROR_HANDLER(err_code);

}

}

/** @snippet [DFU BLE Reset prepare] */

/**@brief Function for preparing for system reset.

*

* @details This function implements @ref dfu_app_reset_prepare_t. It will be called by

* @ref dfu_app_handler.c before entering the bootloader/DFU.

* This allows the current running application to shut down gracefully.

*/

static void reset_prepare(void)

{

uint32_t err_code;

if (m_conn_handle != BLE_CONN_HANDLE_INVALID)

{

// Disconnect from peer.

err_code = sd_ble_gap_disconnect(m_conn_handle, BLE_HCI_REMOTE_USER_TERMINATED_CONNECTION);

APP_ERROR_CHECK(err_code);

}

else

{

// If not connected, the device will be advertising. Hence stop the advertising.

advertising_stop();

}

err_code = ble_conn_params_stop();

APP_ERROR_CHECK(err_code);

bta_bleResetPrepare();

nrf_delay_ms(500);

}

static void nus_data_handler(ble_nus_t * p_nus, uint8_t * p_data, uint16_t length)

{

bta_bleUartReceived(p_data, length);

}

/**@brief Function for initializing services that will be used by the application.

*

*/

static void services_init(void)

{

uint32_t err_code;

ble_dfu_init_t dfus_init;

// Initialize the Device Firmware Update Service.

memset(&dfus_init, 0, sizeof(dfus_init));

dfus_init.evt_handler = dfu_app_on_dfu_evt;

dfus_init.error_handler = NULL;

dfus_init.evt_handler = dfu_app_on_dfu_evt;

dfus_init.revision = DFU_REVISION;

err_code = ble_dfu_init(&m_dfus, &dfus_init);

APP_ERROR_CHECK(err_code);

dfu_app_reset_prepare_set(reset_prepare);

dfu_app_dm_appl_instance_set(m_app_handle);

// Initialize the Nordic Uart Service.

ble_nus_init_t nus_init;

memset(&nus_init, 0, sizeof(nus_init));

nus_init.data_handler = nus_data_handler;

err_code = ble_nus_init(&m_nus, &nus_init);

APP_ERROR_CHECK(err_code);

}

/**@brief Function for handling the Connection Parameters Module.

*

* @details This function will be called for all events in the Connection Parameters Module which

* are passed to the application.

* @note All this function does is to disconnect. This could have been done by simply

* setting the disconnect_on_fail config parameter, but instead we use the event

* handler mechanism to demonstrate its use.

*

* @param[in] p_evt Event received from the Connection Parameters Module.

*/

static void on_conn_params_evt(ble_conn_params_evt_t * p_evt)

{

uint32_t err_code;

if (p_evt->evt_type == BLE_CONN_PARAMS_EVT_FAILED)

{

err_code = sd_ble_gap_disconnect(m_conn_handle, BLE_HCI_CONN_INTERVAL_UNACCEPTABLE);

APP_ERROR_CHECK(err_code);

}

}

/**@brief Function for handling a Connection Parameters error.

*

* @param[in] nrf_error Error code containing information about what went wrong.

*/

static void conn_params_error_handler(uint32_t nrf_error)

{

APP_ERROR_HANDLER(nrf_error);

}

/**@brief Function for initializing the Connection Parameters module.

*/

static void conn_params_init(void)

{

uint32_t err_code;

ble_conn_params_init_t cp_init;

memset(&cp_init, 0, sizeof(cp_init));

cp_init.p_conn_params = NULL;

cp_init.first_conn_params_update_delay = FIRST_CONN_PARAMS_UPDATE_DELAY;

cp_init.next_conn_params_update_delay = NEXT_CONN_PARAMS_UPDATE_DELAY;

cp_init.max_conn_params_update_count = MAX_CONN_PARAMS_UPDATE_COUNT;

cp_init.start_on_notify_cccd_handle = BLE_GATT_HANDLE_INVALID;

cp_init.disconnect_on_fail = false;

cp_init.evt_handler = on_conn_params_evt;

cp_init.error_handler = conn_params_error_handler;

err_code = ble_conn_params_init(&cp_init);

APP_ERROR_CHECK(err_code);

}

void advertising_start(void)

{

uint32_t err_code;

ble_gap_adv_params_t adv_params;

// Start advertising.

memset(&adv_params, 0, sizeof(adv_params));

adv_params.type = BLE_GAP_ADV_TYPE_ADV_IND;

adv_params.p_peer_addr = NULL;

adv_params.fp = BLE_GAP_ADV_FP_ANY;

adv_params.interval = APP_ADV_INTERVAL;

adv_params.timeout = APP_ADV_TIMEOUT_IN_SECONDS;

err_code = sd_ble_gap_adv_start(&adv_params);

APP_ERROR_CHECK(err_code);

}

/**@brief Function for handling the Application's BLE Stack events.

*

* @param[in] p_ble_evt Bluetooth stack event.

*/

static void on_ble_evt(ble_evt_t * p_ble_evt)

{

printf("on_ble_evt %d", p_ble_evt->header.evt_id);

uint32_t err_code;

static ble_gap_sec_keyset_t s_sec_keyset;

ble_gap_enc_info_t * p_enc_info;

switch (p_ble_evt->header.evt_id)

{

case BLE_GAP_EVT_CONNECTED:

m_conn_handle = p_ble_evt->evt.gap_evt.conn_handle;

bta_bleConnected();

break;

case BLE_GAP_EVT_DISCONNECTED:

m_conn_handle = BLE_CONN_HANDLE_INVALID;

bta_bleDisconnected();

break;

case BLE_GAP_EVT_SEC_PARAMS_REQUEST:

s_sec_keyset.keys_periph.p_enc_key = NULL;

err_code = sd_ble_gap_sec_params_reply(m_conn_handle,

BLE_GAP_SEC_STATUS_SUCCESS,

&m_sec_params,

&s_sec_keyset);

APP_ERROR_CHECK(err_code);

break;

case BLE_GATTS_EVT_SYS_ATTR_MISSING:

err_code = sd_ble_gatts_sys_attr_set(m_conn_handle, NULL, 0, 0);

APP_ERROR_CHECK(err_code);

break;

case BLE_GAP_EVT_AUTH_STATUS:

// TODO: Adoptation to s110v8.0.0, is this needed anymore ?

break;

case BLE_GAP_EVT_SEC_INFO_REQUEST:

if (s_sec_keyset.keys_periph.p_enc_key != NULL)

{

p_enc_info = &s_sec_keyset.keys_periph.p_enc_key->enc_info;

err_code = sd_ble_gap_sec_info_reply(m_conn_handle, p_enc_info, NULL, NULL);

APP_ERROR_CHECK(err_code);

}

else

{

// No keys found for this device.

err_code = sd_ble_gap_sec_info_reply(m_conn_handle, NULL, NULL, NULL);

APP_ERROR_CHECK(err_code);

}

break;

case BLE_GAP_EVT_TIMEOUT:

if (p_ble_evt->evt.gap_evt.params.timeout.src == BLE_GAP_TIMEOUT_SRC_ADVERTISING)

{

bta_bleAdvertiseTimeout();

}

break;

default:

// No implementation needed.

break;

}

}

/**@brief Function for dispatching a BLE stack event to all modules with a BLE stack event handler.

*

* @details This function is called from the BLE Stack event interrupt handler after a BLE stack

* event has been received.

*

* @param[in] p_ble_evt Bluetooth stack event.

*/

static void ble_evt_dispatch(ble_evt_t * p_ble_evt)

{

dm_ble_evt_handler(p_ble_evt);

ble_conn_params_on_ble_evt(p_ble_evt);

ble_dfu_on_ble_evt(&m_dfus, p_ble_evt);

on_ble_evt(p_ble_evt);

ble_nus_on_ble_evt(&m_nus, p_ble_evt);

}

/**@brief Function for dispatching a system event to interested modules.

*

* @details This function is called from the System event interrupt handler after a system

* event has been received.

*

* @param[in] sys_evt System stack event.

*/

static void sys_evt_dispatch(uint32_t sys_evt)

{

pstorage_sys_event_handler(sys_evt);

}

/**@brief Function for initializing the BLE stack.

*

* @details Initializes the SoftDevice and the BLE event interrupt.

*/

static void ble_stack_init(void)

{

uint32_t err_code;

// Initialize the SoftDevice handler module.

SOFTDEVICE_HANDLER_INIT(NRF_CLOCK_LFCLKSRC_XTAL_20_PPM, NULL);

// Enable BLE stack.

ble_enable_params_t ble_enable_params;

memset(&ble_enable_params, 0, sizeof(ble_enable_params));

ble_enable_params.gatts_enable_params.service_changed = IS_SRVC_CHANGED_CHARACT_PRESENT;

err_code = sd_ble_enable(&ble_enable_params);

APP_ERROR_CHECK(err_code);

// Register with the SoftDevice handler module for BLE events.

err_code = softdevice_ble_evt_handler_set(ble_evt_dispatch);

APP_ERROR_CHECK(err_code);

// Register with the SoftDevice handler module for BLE events.

err_code = softdevice_sys_evt_handler_set(sys_evt_dispatch);

APP_ERROR_CHECK(err_code);

}

/**@brief Function for handling the Device Manager events.

*

* @param[in] p_evt Data associated to the device manager event.

*/

static uint32_t device_manager_evt_handler(dm_handle_t const * p_handle,

dm_event_t const * p_event,

ret_code_t event_result)

{

APP_ERROR_CHECK(event_result);

if (p_event->event_id == DM_EVT_LINK_SECURED)

{

app_context_load(p_handle);

}

return NRF_SUCCESS;

}

/**@brief Function for the Device Manager initialization.

*/

static void device_manager_init(bool clearBonds)

{

uint32_t err_code;

dm_init_param_t init_data;

dm_application_param_t register_param;

// Initialize persistent storage module.

err_code = pstorage_init();

APP_ERROR_CHECK(err_code);

init_data.clear_persistent_data = clearBonds ;

APP_ERROR_CHECK(err_code);

err_code = dm_init(&init_data);

APP_ERROR_CHECK(err_code);

memset(&register_param.sec_param, 0, sizeof(ble_gap_sec_params_t));

register_param.sec_param.bond = SEC_PARAM_BOND;

register_param.sec_param.mitm = SEC_PARAM_MITM;

register_param.sec_param.io_caps = SEC_PARAM_IO_CAPABILITIES;

register_param.sec_param.oob = SEC_PARAM_OOB;

register_param.sec_param.min_key_size = SEC_PARAM_MIN_KEY_SIZE;

register_param.sec_param.max_key_size = SEC_PARAM_MAX_KEY_SIZE;

register_param.evt_handler = device_manager_evt_handler;

register_param.service_type = DM_PROTOCOL_CNTXT_GATT_SRVR_ID;

err_code = dm_register(&m_app_handle, &register_param);

APP_ERROR_CHECK(err_code);

}

static void advertising_init(uint8_t *data, uint8_t size)

{

#define COMPANY_IDENTIFIER (0x1000) /* reserver ID */

static ble_advdata_manuf_data_t manuf_specific_data; /**< Manufacturer specific data. */

uint32_t err_code;

ble_advdata_t advdata;

ble_advdata_t scanrsp;

uint8_t flags = BLE_GAP_ADV_FLAGS_LE_ONLY_LIMITED_DISC_MODE;

ble_uuid_t adv_uuids[] = {{BLE_UUID_NUS_SERVICE,m_nus.uuid_type}};

memset(&advdata, 0, sizeof(advdata));

advdata.name_type = BLE_ADVDATA_FULL_NAME;

advdata.include_appearance = false;

advdata.flags = flags;

memset(&scanrsp, 0, sizeof(scanrsp));

scanrsp.uuids_complete.uuid_cnt = sizeof(adv_uuids) / sizeof(adv_uuids[0]);

scanrsp.uuids_complete.p_uuids = adv_uuids;

manuf_specific_data.company_identifier = COMPANY_IDENTIFIER;

manuf_specific_data.data.p_data = &data[0];

manuf_specific_data.data.size = size;

advdata.p_manuf_specific_data = &manuf_specific_data;

err_code = ble_advdata_set(&advdata, &scanrsp);

APP_ERROR_CHECK(err_code);

}

void bta_bleInit(bool clearBonds)

{

ble_stack_init();

device_manager_init(clearBonds);

gap_params_init();

services_init();

conn_params_init();

}

void bta_bleAdvertiseBegin(uint8_t *data, uint8_t size)

{

advertising_init(data, size);

advertising_start();

}

void bta_bleAdvertiseStop(void)

{

sd_ble_gap_adv_stop();

}

void bta_bleDisconnect(void)

{

sd_ble_gap_disconnect(m_nus.conn_handle, BLE_HCI_REMOTE_USER_TERMINATED_CONNECTION);

}

void bta_bleSetDeviceName(char * name){

uint32_t err_code;

ble_gap_conn_sec_mode_t sec_mode;

BLE_GAP_CONN_SEC_MODE_SET_OPEN(&sec_mode);

err_code = sd_ble_gap_device_name_set(&sec_mode, (const uint8_t *)name, strlen(name));

APP_ERROR_CHECK(err_code);

}

void bta_bleUartTransmit(uint8_t * p_data, uint16_t length){

uint32_t err_code=ble_nus_string_send(&m_nus, p_data, length);

APP_ERROR_CHECK(err_code);

}