#include "custom_profile.h"

//------------------------------------------------------------------------------

#include <stdint.h>

#include <string.h>

#include "nordic_common.h"

#include "nrf.h"

#include "app_error.h"

#include "nrf51_bitfields.h"

#include "ble.h"

#include "ble_hci.h"

#include "ble_srv_common.h"

#include "ble_advdata.h"

#include "ble_advertising.h"

#include "ble_bas.h"

#include "ble_dis.h"

#include "ble_conn_params.h"

#include "softdevice_handler.h"

#include "app_timer.h"

#include "device_manager.h"

#include "pstorage.h"

#include "app_trace.h"

#include "app_gpiote.h"

#include "nrf_delay.h"

#ifdef BLE_DFU_APP_SUPPORT

#include "ble_dfu.h"

#include "dfu_app_handler.h"

#endif // BLE_DFU_APP_SUPPORT

//------------------------------------------------------------------------------

// Application specific includes

#include "custom_board.h"

#include "custom_service.h"

//------------------------------------------------------------------------------

// Building the scan response

//

// This is a little pre-processor magic to build a scan response using a macro

// defined at the project level. The purpose is so that in development the

// build number can be easily incremented and used to identify the firmware

// rev on your device

#define QUOTE(name) #name

#define STR(macro) QUOTE(macro)

static char _scanResponse[128];

#define SCAN_RESPONSE_BASE "Template App Build "

//------------------------------------------------------------------------------

// Custom Profile variables

//

// Custom Service structure

static ble_cs_t _cs;

// Battery service struct

static ble_bas_t _bas;

// DFU Service

#ifdef BLE_DFU_APP_SUPPORT

static ble_dfu_t _dfus;

#endif

// Handle for the current connection

static uint16_t _conn_handle = BLE_CONN_HANDLE_INVALID;

// Device Manager application instance ID

static dm_application_instance_t _app_handle;

////////////////////////////////////////////////////////////////////////////////

//

// GAP Initialization

static void gap_params_init(void)

{

// Create the scan response name

memset(_scanResponse, 0, sizeof(_scanResponse));

memcpy(_scanResponse, SCAN_RESPONSE_BASE, strlen(SCAN_RESPONSE_BASE));

memcpy(_scanResponse + strlen(SCAN_RESPONSE_BASE), STR(BUILD_NUM), 4);

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 *)_scanResponse,

strlen(_scanResponse));

APP_ERROR_CHECK(err_code);

err_code = sd_ble_gap_appearance_set(BLE_APPEARANCE_UNKNOWN);

APP_ERROR_CHECK(err_code);

err_code = sd_ble_gap_tx_power_set(4);

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);

}

#ifdef BLE_DFU_APP_SUPPORT

// {

/**@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(_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 (_conn_handle != BLE_CONN_HANDLE_INVALID)

{

// Disconnect from peer.

err_code = sd_ble_gap_disconnect(_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);

nrf_delay_ms(500);

}

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

// }

#endif // BLE_DFU_APP_SUPPORT

//============================================================================//

// This is where we initialize our services and register them with the stack.

//

// The services that we create are:

//

// 1) Custom Service

// 2) Battery Service

// 3) Device Information Service

// 4) DFU Service (if enabled via macro)

static void services_init(void)

{

uint32_t err_code;

ble_bas_init_t bas_init;

ble_dis_init_t dis_init;

//--------------------------------------------------------------------------

// Set up the Custom Service

err_code = CS_Init(&_cs);

APP_ERROR_CHECK(err_code);

//--------------------------------------------------------------------------

// Initialize Battery Service.

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

//--------------------------------------------------------------------------

// Here the sec level for the Battery Service can be changed/increased.

BLE_GAP_CONN_SEC_MODE_SET_OPEN(&bas_init.battery_level_char_attr_md.cccd_write_perm);

BLE_GAP_CONN_SEC_MODE_SET_OPEN(&bas_init.battery_level_char_attr_md.read_perm);

BLE_GAP_CONN_SEC_MODE_SET_NO_ACCESS(&bas_init.battery_level_char_attr_md.write_perm);

BLE_GAP_CONN_SEC_MODE_SET_OPEN(&bas_init.battery_level_report_read_perm);

bas_init.evt_handler = NULL;

bas_init.support_notification = true;

bas_init.p_report_ref = NULL;

bas_init.initial_batt_level = 100;

err_code = ble_bas_init(&_bas, &bas_init);

APP_ERROR_CHECK(err_code);

//--------------------------------------------------------------------------

// Initialize Device Information Service.

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

ble_srv_ascii_to_utf8(&dis_init.manufact_name_str, (char *)MANUFACTURER_NAME);

ble_srv_ascii_to_utf8(&dis_init.model_num_str, (char *)MODEL_NUMBER);

ble_srv_ascii_to_utf8(&dis_init.serial_num_str, (char *)SERIAL_NUMBER);

ble_srv_ascii_to_utf8(&dis_init.hw_rev_str, (char *)HARDWARE_REVISION);

ble_srv_ascii_to_utf8(&dis_init.fw_rev_str, (char *)FIRMWARE_REVISION);

BLE_GAP_CONN_SEC_MODE_SET_OPEN(&dis_init.dis_attr_md.read_perm);

BLE_GAP_CONN_SEC_MODE_SET_NO_ACCESS(&dis_init.dis_attr_md.write_perm);

err_code = ble_dis_init(&dis_init);

APP_ERROR_CHECK(err_code);

#ifdef BLE_DFU_APP_SUPPORT

/** @snippet [DFU BLE Service initialization] */

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(&_dfus, &dfus_init);

APP_ERROR_CHECK(err_code);

dfu_app_reset_prepare_set(reset_prepare);

dfu_app_dm_appl_instance_set(_app_handle);

/** @snippet [DFU BLE Service initialization] */

#endif // BLE_DFU_APP_SUPPORT

}

//============================================================================//

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(_conn_handle, BLE_HCI_CONN_INTERVAL_UNACCEPTABLE);

APP_ERROR_CHECK(err_code);

}

}

//============================================================================//

static void conn_params_error_handler(uint32_t nrf_error)

{

APP_ERROR_HANDLER(nrf_error);

}

//============================================================================//

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 = _cs.status_handles.cccd_handle;

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);

}

//============================================================================//

static void on_adv_evt(ble_adv_evt_t ble_adv_evt)

{

switch (ble_adv_evt)

{

case BLE_ADV_EVT_FAST:

//

break;

case BLE_ADV_EVT_IDLE:

// Go to system-off mode. This function will not return; wakeup will cause a reset.

//err_code = sd_power_system_off();

//APP_ERROR_CHECK(err_code);

break;

default:

break;

}

}

//============================================================================//

static void on_ble_evt(ble_evt_t * p_ble_evt)

{

switch (p_ble_evt->header.evt_id)

{

case BLE_GAP_EVT_CONNECTED:

_conn_handle = p_ble_evt->evt.gap_evt.conn_handle;

break;

case BLE_GAP_EVT_DISCONNECTED:

_conn_handle = BLE_CONN_HANDLE_INVALID;

break;

default:

// No implementation needed.

break;

}

}

//============================================================================//

static void ble_evt_dispatch(ble_evt_t * p_ble_evt)

{

dm_ble_evt_handler(p_ble_evt);

ble_cs_on_ble_evt(&_cs, p_ble_evt);

ble_bas_on_ble_evt(&_bas, p_ble_evt);

ble_conn_params_on_ble_evt(p_ble_evt);

#ifdef BLE_DFU_APP_SUPPORT

/** @snippet [Propagating BLE Stack events to DFU Service] */

ble_dfu_on_ble_evt(&_dfus, p_ble_evt);

/** @snippet [Propagating BLE Stack events to DFU Service] */

#endif // BLE_DFU_APP_SUPPORT

on_ble_evt(p_ble_evt);

ble_advertising_on_ble_evt(p_ble_evt);

}

//============================================================================//

static void sys_evt_dispatch(uint32_t sys_evt)

{

pstorage_sys_event_handler(sys_evt);

ble_advertising_on_sys_evt(sys_evt);

}

//============================================================================//

void CP_StackInit(void)

{

uint32_t err_code;

// Initialize the SoftDevice handler module.

SOFTDEVICE_HANDLER_INIT(NRF_CLOCK_LFCLKSRC_XTAL_20_PPM, NULL);

#ifdef S110

// 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.attr_tab_size = BLE_GATTS_ATTR_TAB_SIZE_MIN;

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);

#endif

// 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);

}

//============================================================================//

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);

#ifdef BLE_DFU_APP_SUPPORT

if (p_event->event_id == DM_EVT_LINK_SECURED)

{

app_context_load(p_handle);

}

#endif // BLE_DFU_APP_SUPPORT

return NRF_SUCCESS;

}

//============================================================================//

static void device_manager_init(void)

{

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);

// TODO: integrate bonding some other way

// Clear all bonded centrals if the Bonds Delete button is pushed.

//err_code = bsp_button_is_pressed(BOND_DELETE_ALL_BUTTON_ID,&(init_data.clear_persistent_data));

//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(&_app_handle, &register_param);

APP_ERROR_CHECK(err_code);

}

//============================================================================//

static void advertising_init(void)

{

uint32_t err_code;

ble_advdata_t advdata;

// Build advertising data struct to pass into @ref ble_advertising_init.

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

advdata.name_type = BLE_ADVDATA_FULL_NAME;

//advdata.include_appearance = true;

advdata.flags = BLE_GAP_ADV_FLAGS_LE_ONLY_GENERAL_DISC_MODE;

//advdata.uuids_complete.uuid_cnt = sizeof(m_adv_uuids) / sizeof(m_adv_uuids[0]);

//advdata.uuids_complete.p_uuids = m_adv_uuids;

err_code = ble_advdata_set(&advdata, NULL);

APP_ERROR_CHECK(err_code);

ble_adv_modes_config_t options = {0};

options.ble_adv_fast_enabled = BLE_ADV_FAST_ENABLED;

options.ble_adv_fast_interval = APP_ADV_INTERVAL;

options.ble_adv_fast_timeout = APP_ADV_TIMEOUT_IN_SECONDS;

err_code = ble_advertising_init(&advdata, 0, &options, on_adv_evt, NULL);

APP_ERROR_CHECK(err_code);

}

//============================================================================//

void CP_Init(void)

{

device_manager_init();

gap_params_init();

advertising_init();

services_init();

conn_params_init();

}

//============================================================================//

void CP_AdvertisingStart(void)

{

if (_conn_handle == BLE_CONN_HANDLE_INVALID)

{

ble_advertising_start(BLE_ADV_MODE_FAST);

}

}

//============================================================================//

void CP_BatteryServiceUpdate(uint8_t level)

{

uint32_t err_code;

err_code = ble_bas_battery_level_update(&_bas, level);

if ((err_code != NRF_SUCCESS) &&

(err_code != NRF_ERROR_INVALID_STATE) &&

(err_code != BLE_ERROR_NO_TX_BUFFERS) &&

(err_code != BLE_ERROR_GATTS_SYS_ATTR_MISSING)

)

{

APP_ERROR_HANDLER(err_code);

}

}

//============================================================================//

void CP_SetStatus(uint8_t status)

{

CS_SendStatusUpdate(&_cs, status);

}