Added some functions to BLESecurity

src/ble_security.rs

@@ -1,4 +1,5 @@
 use crate::enums;
+
 pub struct BLESecurity {
   passkey: u32,
 }
@@ -31,4 +32,16 @@ impl BLESecurity {
     unsafe { esp_idf_sys::ble_hs_cfg.sm_io_cap = iocap as _ };
     self
   }
+
+  /// If we are the initiator of the security procedure this sets the keys we will distribute.
+  pub fn set_security_init_key(&mut self, init_key: enums::PairKeyDist) -> &mut Self {
+    unsafe { esp_idf_sys::ble_hs_cfg.sm_our_key_dist = init_key.bits() };
+    self
+  }
+
+  /// Set the keys we are willing to accept during pairing.
+  pub fn set_security_resp_key(&mut self, resp_key: enums::PairKeyDist) -> &mut Self {
+    unsafe { esp_idf_sys::ble_hs_cfg.sm_their_key_dist = init_key.bits() };
+    self
+  }
 }

src/enums.rs

@@ -1,3 +1,4 @@
+use bitflags::bitflags;
 use esp_idf_sys::*;
 
 #[repr(u8)]
@@ -79,3 +80,14 @@ pub enum PowerType {
   /// For default, if not set other, it will use default value
   Default = esp_ble_power_type_t_ESP_BLE_PWR_TYPE_DEFAULT as _,
 }
+
+bitflags! {
+  #[repr(transparent)]
+  #[derive(Debug, Clone, Copy, PartialEq, Eq)]
+  pub struct PairKeyDist: u8 {
+    const ENC = BLE_SM_PAIR_KEY_DIST_ENC as _;
+    const ID = BLE_SM_PAIR_KEY_DIST_ID as _;
+    const SIGN = BLE_SM_PAIR_KEY_DIST_SIGN as _;
+    const LINK = BLE_SM_PAIR_KEY_DIST_LINK as _;
+  }
+}