NimBLESecurity.h

/*
 * NimBLESecurity.h
 *
 *  Created: on Feb 22 2020
 *      Author H2zero
 *
 * Originally:
 *
 * BLESecurity.h
 *
 *  Created on: Dec 17, 2017
 *      Author: chegewara
 */
 
#ifndef COMPONENTS_NIMBLESECURITY_H_
#define COMPONENTS_NIMBLESECURITY_H_
 
#include "nimconfig.h"
#if defined(CONFIG_BT_ENABLED)
 
#if defined(CONFIG_NIMBLE_CPP_IDF)
#include "host/ble_gap.h"
#else
#include "nimble/nimble/host/include/host/ble_gap.h"
#endif
 
/****  FIX COMPILATION ****/
#undef min
#undef max
/**************************/
 
#include <stdint.h>
 
#define ESP_LE_AUTH_NO_BOND                 0x00                                                          /* relate to BTM_LE_AUTH_NO_BOND in stack/btm_api.h */
#define ESP_LE_AUTH_BOND                    0x01                                                    /* relate to BTM_LE_AUTH_BOND in stack/btm_api.h */
#define ESP_LE_AUTH_REQ_MITM                (1 << 2)                                                /* relate to BTM_LE_AUTH_REQ_MITM in stack/btm_api.h */
#define ESP_LE_AUTH_REQ_BOND_MITM           (ESP_LE_AUTH_BOND | ESP_LE_AUTH_REQ_MITM)
#define ESP_LE_AUTH_REQ_SC_ONLY             (1 << 3)                                                /* relate to BTM_LE_AUTH_REQ_SC_ONLY in stack/btm_api.h */
#define ESP_LE_AUTH_REQ_SC_BOND             (ESP_LE_AUTH_BOND | ESP_LE_AUTH_REQ_SC_ONLY)              /* relate to BTM_LE_AUTH_REQ_SC_BOND in stack/btm_api.h */
#define ESP_LE_AUTH_REQ_SC_MITM             (ESP_LE_AUTH_REQ_MITM | ESP_LE_AUTH_REQ_SC_ONLY)          /* relate to BTM_LE_AUTH_REQ_SC_MITM in stack/btm_api.h */
#define ESP_LE_AUTH_REQ_SC_MITM_BOND        (ESP_LE_AUTH_REQ_MITM | ESP_LE_AUTH_REQ_SC_ONLY | ESP_LE_AUTH_BOND)     /* relate to BTM_LE_AUTH_REQ_SC_MITM_BOND in stack/btm_api.h */
 
#define ESP_IO_CAP_OUT                      0            /* relate to BTM_IO_CAP_OUT in stack/btm_api.h */
#define ESP_IO_CAP_IO                       1           /* relate to BTM_IO_CAP_IO in stack/btm_api.h */
#define ESP_IO_CAP_IN                       2           /* relate to BTM_IO_CAP_IN in stack/btm_api.h */
#define ESP_IO_CAP_NONE                     3        /* relate to BTM_IO_CAP_NONE in stack/btm_api.h */
#define ESP_IO_CAP_KBDISP                   4       /* relate to BTM_IO_CAP_KBDISP in stack/btm_api.h */
 
#define ESP_BLE_ENC_KEY_MASK    (1 << 0)            /* relate to BTM_BLE_ENC_KEY_MASK in stack/btm_api.h */
#define ESP_BLE_ID_KEY_MASK     (1 << 1)            /* relate to BTM_BLE_ID_KEY_MASK in stack/btm_api.h */
#define ESP_BLE_CSR_KEY_MASK    (1 << 2)            /* relate to BTM_BLE_CSR_KEY_MASK in stack/btm_api.h */
#define ESP_BLE_LINK_KEY_MASK   (1 << 3)            /* relate to BTM_BLE_LINK_KEY_MASK in stack/btm_api.h */
 
typedef uint8_t   esp_ble_auth_req_t;         
typedef uint8_t   esp_ble_io_cap_t;               
class NimBLESecurity {
public:
    NimBLESecurity();
    virtual ~NimBLESecurity();
    void setAuthenticationMode(esp_ble_auth_req_t auth_req);
    void setCapability(esp_ble_io_cap_t iocap);
    void setInitEncryptionKey(uint8_t init_key);
    void setRespEncryptionKey(uint8_t resp_key);
    void setKeySize(uint8_t key_size = 16);
    void setStaticPIN(uint32_t pin);
    //static char* esp_key_type_to_str(esp_ble_key_type_t key_type);
/*
private:
    esp_ble_auth_req_t m_authReq;
    esp_ble_io_cap_t m_iocap;
    uint8_t m_initKey;
    uint8_t m_respKey;
    uint8_t m_keySize;
*/
}; // BLESecurity
 
 
class NimBLESecurityCallbacks {
public:
    virtual ~NimBLESecurityCallbacks() {};
 
    virtual uint32_t onPassKeyRequest() = 0;
 
    virtual void onPassKeyNotify(uint32_t pass_key) = 0;
 
    virtual bool onSecurityRequest() = 0 ;
    virtual void onAuthenticationComplete(ble_gap_conn_desc*) = 0;
 
    virtual bool onConfirmPIN(uint32_t pin) = 0;
}; // BLESecurityCallbacks
 
#endif // CONFIG_BT_ENABLED
#endif // COMPONENTS_NIMBLESECURITY_H_