summaryrefslogtreecommitdiffstatshomepage
path: root/ports/zephyr/modbluetooth_zephyr.c
diff options
context:
space:
mode:
Diffstat (limited to 'ports/zephyr/modbluetooth_zephyr.c')
-rw-r--r--ports/zephyr/modbluetooth_zephyr.c616
1 files changed, 602 insertions, 14 deletions
diff --git a/ports/zephyr/modbluetooth_zephyr.c b/ports/zephyr/modbluetooth_zephyr.c
index 8947bdf535..c1b8ddfd8b 100644
--- a/ports/zephyr/modbluetooth_zephyr.c
+++ b/ports/zephyr/modbluetooth_zephyr.c
@@ -31,8 +31,12 @@
#if MICROPY_PY_BLUETOOTH
+#include <zephyr/types.h>
#include <zephyr/bluetooth/bluetooth.h>
#include <zephyr/bluetooth/hci.h>
+#include <zephyr/bluetooth/conn.h>
+#include <zephyr/bluetooth/uuid.h>
+#include <zephyr/bluetooth/gatt.h>
#include "extmod/modbluetooth.h"
#define DEBUG_printf(...) // printk("BLE: " __VA_ARGS__)
@@ -44,6 +48,23 @@
#define ERRNO_BLUETOOTH_NOT_ACTIVE MP_ENODEV
+#define MP_BLUETOOTH_ZEPHYR_MAX_SERVICES (8)
+
+/* This masks Permission bits from GATT API */
+#define GATT_PERM_MASK (BT_GATT_PERM_READ | \
+ BT_GATT_PERM_READ_AUTHEN | \
+ BT_GATT_PERM_READ_ENCRYPT | \
+ BT_GATT_PERM_WRITE | \
+ BT_GATT_PERM_WRITE_AUTHEN | \
+ BT_GATT_PERM_WRITE_ENCRYPT | \
+ BT_GATT_PERM_PREPARE_WRITE)
+
+#define GATT_PERM_ENC_READ_MASK (BT_GATT_PERM_READ_ENCRYPT | \
+ BT_GATT_PERM_READ_AUTHEN)
+
+#define GATT_PERM_ENC_WRITE_MASK (BT_GATT_PERM_WRITE_ENCRYPT | \
+ BT_GATT_PERM_WRITE_AUTHEN)
+
enum {
MP_BLUETOOTH_ZEPHYR_BLE_STATE_OFF,
MP_BLUETOOTH_ZEPHYR_BLE_STATE_ACTIVE,
@@ -56,9 +77,42 @@ enum {
MP_BLUETOOTH_ZEPHYR_GAP_SCAN_STATE_ACTIVE,
};
+union uuid_u {
+ struct bt_uuid uuid;
+ struct bt_uuid_16 u16;
+ struct bt_uuid_32 u32;
+ struct bt_uuid_128 u128;
+};
+
+struct add_characteristic {
+ uint8_t properties;
+ uint8_t permissions;
+ const struct bt_uuid *uuid;
+};
+
+struct add_descriptor {
+ uint8_t permissions;
+ const struct bt_uuid *uuid;
+};
+
+typedef struct _mp_bt_zephyr_conn_t {
+ struct bt_conn *conn;
+ struct _mp_bt_zephyr_conn_t *next;
+} mp_bt_zephyr_conn_t;
+
typedef struct _mp_bluetooth_zephyr_root_pointers_t {
+ // list of objects to be tracked by the gc
+ mp_obj_t objs_list;
+
// Characteristic (and descriptor) value storage.
mp_gatts_db_t gatts_db;
+
+ // Service definitions.
+ size_t n_services;
+ struct bt_gatt_service *services[MP_BLUETOOTH_ZEPHYR_MAX_SERVICES];
+
+ // active connections
+ mp_bt_zephyr_conn_t *connections;
} mp_bluetooth_zephyr_root_pointers_t;
static int mp_bluetooth_zephyr_ble_state;
@@ -69,6 +123,98 @@ static struct k_timer mp_bluetooth_zephyr_gap_scan_timer;
static struct bt_le_scan_cb mp_bluetooth_zephyr_gap_scan_cb_struct;
#endif
+static struct bt_data bt_ad_data[8];
+static size_t bt_ad_len = 0;
+static struct bt_data bt_sd_data[8];
+static size_t bt_sd_len = 0;
+
+static mp_bt_zephyr_conn_t *mp_bt_zephyr_next_conn;
+
+static mp_bt_zephyr_conn_t *mp_bt_zephyr_find_connection(uint8_t conn_handle);
+static void mp_bt_zephyr_insert_connection(mp_bt_zephyr_conn_t *connection);
+static void mp_bt_zephyr_remove_connection(uint8_t conn_handle);
+static void mp_bt_zephyr_connected(struct bt_conn *connected, uint8_t err);
+static void mp_bt_zephyr_disconnected(struct bt_conn *disconn, uint8_t reason);
+static struct bt_uuid *create_zephyr_uuid(const mp_obj_bluetooth_uuid_t *uuid);
+static void gatt_db_add(const struct bt_gatt_attr *pattern, struct bt_gatt_attr *attr, size_t user_data_len);
+static void add_service(const struct bt_uuid *u, struct bt_gatt_attr *attr);
+static void add_characteristic(struct add_characteristic *ch, struct bt_gatt_attr *attr_chrc, struct bt_gatt_attr *attr_value);
+static void add_ccc(struct bt_gatt_attr *attr, struct bt_gatt_attr *attr_desc);
+static void add_cep(const struct bt_gatt_attr *attr_chrc, struct bt_gatt_attr *attr_desc);
+static void add_descriptor(struct bt_gatt_attr *chrc, struct add_descriptor *d, struct bt_gatt_attr *attr_desc);
+static void mp_bt_zephyr_gatt_indicate_done(struct bt_conn *conn, struct bt_gatt_indicate_params *params, uint8_t err);
+static struct bt_gatt_attr *mp_bt_zephyr_find_attr_by_handle(uint16_t value_handle);
+
+static struct bt_conn_cb mp_bt_zephyr_conn_callbacks = {
+ .connected = mp_bt_zephyr_connected,
+ .disconnected = mp_bt_zephyr_disconnected,
+};
+
+static mp_bt_zephyr_conn_t *mp_bt_zephyr_find_connection(uint8_t conn_handle) {
+ struct bt_conn_info info;
+ for (mp_bt_zephyr_conn_t *connection = MP_STATE_PORT(bluetooth_zephyr_root_pointers)->connections; connection != NULL; connection = connection->next) {
+ if (connection->conn) {
+ bt_conn_get_info(connection->conn, &info);
+ if (info.id == conn_handle) {
+ return connection;
+ }
+ }
+ }
+ return NULL;
+}
+
+static void mp_bt_zephyr_insert_connection(mp_bt_zephyr_conn_t *connection) {
+ connection->next = MP_STATE_PORT(bluetooth_zephyr_root_pointers)->connections;
+ MP_STATE_PORT(bluetooth_zephyr_root_pointers)->connections = connection;
+}
+
+static void mp_bt_zephyr_remove_connection(uint8_t conn_handle) {
+ struct bt_conn_info info;
+ mp_bt_zephyr_conn_t *prev = NULL;
+ for (mp_bt_zephyr_conn_t *connection = MP_STATE_PORT(bluetooth_zephyr_root_pointers)->connections; connection != NULL; connection = connection->next) {
+ if (connection->conn) {
+ bt_conn_get_info(connection->conn, &info);
+ if (info.id == conn_handle) {
+ // unlink this item and the gc will eventually collect it
+ if (prev != NULL) {
+ prev->next = connection->next;
+ } else {
+ // move the start pointer
+ MP_STATE_PORT(bluetooth_zephyr_root_pointers)->connections = connection->next;
+ }
+ break;
+ } else {
+ prev = connection;
+ }
+ }
+ }
+}
+
+static void mp_bt_zephyr_connected(struct bt_conn *conn, uint8_t err) {
+ struct bt_conn_info info;
+ bt_conn_get_info(conn, &info);
+
+ if (err) {
+ uint8_t addr[6] = {0};
+ DEBUG_printf("Connection from central failed (err %u)\n", err);
+ mp_bluetooth_gap_on_connected_disconnected(MP_BLUETOOTH_IRQ_CENTRAL_DISCONNECT, info.id, 0xff, addr);
+ } else {
+ DEBUG_printf("Central connected with id %d\n", info.id);
+ mp_bt_zephyr_next_conn->conn = bt_conn_ref(conn);
+ mp_bluetooth_gap_on_connected_disconnected(MP_BLUETOOTH_IRQ_CENTRAL_CONNECT, info.id, info.le.dst->type, info.le.dst->a.val);
+ mp_bt_zephyr_insert_connection(mp_bt_zephyr_next_conn);
+ }
+}
+
+static void mp_bt_zephyr_disconnected(struct bt_conn *conn, uint8_t reason) {
+ struct bt_conn_info info;
+ bt_conn_get_info(conn, &info);
+ DEBUG_printf("Central disconnected (id %d reason %u)\n", info.id, reason);
+ bt_conn_unref(conn);
+ mp_bt_zephyr_remove_connection(info.id);
+ mp_bluetooth_gap_on_connected_disconnected(MP_BLUETOOTH_IRQ_CENTRAL_DISCONNECT, info.id, info.le.dst->type, info.le.dst->a.val);
+}
+
static int bt_err_to_errno(int err) {
// Zephyr uses errno codes directly, but they are negative.
return -err;
@@ -128,6 +274,11 @@ int mp_bluetooth_init(void) {
MP_STATE_PORT(bluetooth_zephyr_root_pointers) = m_new0(mp_bluetooth_zephyr_root_pointers_t, 1);
mp_bluetooth_gatts_db_create(&MP_STATE_PORT(bluetooth_zephyr_root_pointers)->gatts_db);
+ MP_STATE_PORT(bluetooth_zephyr_root_pointers)->connections = NULL;
+ mp_bt_zephyr_next_conn = NULL;
+
+ MP_STATE_PORT(bluetooth_zephyr_root_pointers)->objs_list = mp_obj_new_list(0, NULL);
+
#if MICROPY_PY_BLUETOOTH_ENABLE_CENTRAL_MODE
mp_bluetooth_zephyr_gap_scan_state = MP_BLUETOOTH_ZEPHYR_GAP_SCAN_STATE_INACTIVE;
k_timer_init(&mp_bluetooth_zephyr_gap_scan_timer, gap_scan_cb_timeout, NULL);
@@ -137,6 +288,9 @@ int mp_bluetooth_init(void) {
#endif
if (mp_bluetooth_zephyr_ble_state == MP_BLUETOOTH_ZEPHYR_BLE_STATE_OFF) {
+
+ bt_conn_cb_register(&mp_bt_zephyr_conn_callbacks);
+
// bt_enable can only be called once.
int ret = bt_enable(NULL);
if (ret) {
@@ -151,15 +305,22 @@ int mp_bluetooth_init(void) {
return 0;
}
-void mp_bluetooth_deinit(void) {
+int mp_bluetooth_deinit(void) {
DEBUG_printf("mp_bluetooth_deinit %d\n", mp_bluetooth_zephyr_ble_state);
if (mp_bluetooth_zephyr_ble_state == MP_BLUETOOTH_ZEPHYR_BLE_STATE_OFF
|| mp_bluetooth_zephyr_ble_state == MP_BLUETOOTH_ZEPHYR_BLE_STATE_SUSPENDED) {
- return;
+ return 0;
}
mp_bluetooth_gap_advertise_stop();
+ #if CONFIG_BT_GATT_DYNAMIC_DB
+ for (size_t i = 0; i < MP_STATE_PORT(bluetooth_zephyr_root_pointers)->n_services; ++i) {
+ bt_gatt_service_unregister(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->services[i]);
+ MP_STATE_PORT(bluetooth_zephyr_root_pointers)->services[i] = NULL;
+ }
+ #endif
+
#if MICROPY_PY_BLUETOOTH_ENABLE_CENTRAL_MODE
mp_bluetooth_gap_scan_stop();
bt_le_scan_cb_unregister(&mp_bluetooth_zephyr_gap_scan_cb_struct);
@@ -170,6 +331,8 @@ void mp_bluetooth_deinit(void) {
mp_bluetooth_zephyr_ble_state = MP_BLUETOOTH_ZEPHYR_BLE_STATE_SUSPENDED;
MP_STATE_PORT(bluetooth_zephyr_root_pointers) = NULL;
+ mp_bt_zephyr_next_conn = NULL;
+ return 0;
}
bool mp_bluetooth_is_active(void) {
@@ -191,7 +354,7 @@ void mp_bluetooth_get_current_address(uint8_t *addr_type, uint8_t *addr) {
}
void mp_bluetooth_set_address_mode(uint8_t addr_mode) {
- // TODO: implement
+ mp_raise_OSError(MP_EOPNOTSUPP);
}
size_t mp_bluetooth_gap_get_device_name(const uint8_t **buf) {
@@ -232,15 +395,11 @@ int mp_bluetooth_gap_advertise_start(bool connectable, int32_t interval_us, cons
mp_bluetooth_gap_advertise_stop();
- struct bt_data bt_ad_data[8];
- size_t bt_ad_len = 0;
if (adv_data) {
bt_ad_len = MP_ARRAY_SIZE(bt_ad_data);
mp_bluetooth_prepare_bt_data(adv_data, adv_data_len, bt_ad_data, &bt_ad_len);
}
- struct bt_data bt_sd_data[8];
- size_t bt_sd_len = 0;
if (sr_data) {
bt_sd_len = MP_ARRAY_SIZE(bt_sd_data);
mp_bluetooth_prepare_bt_data(sr_data, sr_data_len, bt_sd_data, &bt_sd_len);
@@ -259,6 +418,10 @@ int mp_bluetooth_gap_advertise_start(bool connectable, int32_t interval_us, cons
.peer = NULL,
};
+ // pre-allocate a new connection structure as we cannot allocate this inside the connection callback
+ mp_bt_zephyr_next_conn = m_new0(mp_bt_zephyr_conn_t, 1);
+ mp_obj_list_append(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->objs_list, MP_OBJ_FROM_PTR(mp_bt_zephyr_next_conn));
+
return bt_err_to_errno(bt_le_adv_start(&param, bt_ad_data, bt_ad_len, bt_sd_data, bt_sd_len));
}
@@ -271,6 +434,8 @@ void mp_bluetooth_gap_advertise_stop(void) {
}
int mp_bluetooth_gatts_register_service_begin(bool append) {
+ #if CONFIG_BT_GATT_DYNAMIC_DB
+
if (!mp_bluetooth_is_active()) {
return ERRNO_BLUETOOTH_NOT_ACTIVE;
}
@@ -280,25 +445,190 @@ int mp_bluetooth_gatts_register_service_begin(bool append) {
return MP_EOPNOTSUPP;
}
+ // Unregister and unref any previous service definitions.
+ for (size_t i = 0; i < MP_STATE_PORT(bluetooth_zephyr_root_pointers)->n_services; ++i) {
+ bt_gatt_service_unregister(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->services[i]);
+ MP_STATE_PORT(bluetooth_zephyr_root_pointers)->services[i] = NULL;
+ }
+ MP_STATE_PORT(bluetooth_zephyr_root_pointers)->n_services = 0;
+
// Reset the gatt characteristic value db.
mp_bluetooth_gatts_db_reset(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->gatts_db);
+ MP_STATE_PORT(bluetooth_zephyr_root_pointers)->connections = NULL;
+ MP_STATE_PORT(bluetooth_zephyr_root_pointers)->objs_list = mp_obj_new_list(0, NULL);
+ mp_bt_zephyr_next_conn = NULL;
+ return 0;
+
+ #else
return MP_EOPNOTSUPP;
+ #endif
}
int mp_bluetooth_gatts_register_service_end(void) {
- return MP_EOPNOTSUPP;
+ return 0;
}
int mp_bluetooth_gatts_register_service(mp_obj_bluetooth_uuid_t *service_uuid, mp_obj_bluetooth_uuid_t **characteristic_uuids, uint16_t *characteristic_flags, mp_obj_bluetooth_uuid_t **descriptor_uuids, uint16_t *descriptor_flags, uint8_t *num_descriptors, uint16_t *handles, size_t num_characteristics) {
+ #if CONFIG_BT_GATT_DYNAMIC_DB
+ if (MP_STATE_PORT(bluetooth_zephyr_root_pointers)->n_services >= MP_BLUETOOTH_ZEPHYR_MAX_SERVICES) {
+ return MP_E2BIG;
+ }
+
+ // first of all allocate the entire memory for all the attributes that this service is composed of
+ // 1 for the service itself, 2 for each characteristic (the declaration and the value), and one for each descriptor
+ size_t total_descriptors = 0;
+ for (size_t i = 0; i < num_characteristics; ++i) {
+ total_descriptors += num_descriptors[i];
+ // we have to add the CCC manually
+ if (characteristic_flags[i] & (MP_BLUETOOTH_CHARACTERISTIC_FLAG_NOTIFY | MP_BLUETOOTH_CHARACTERISTIC_FLAG_INDICATE)) {
+ total_descriptors += 1;
+ }
+ }
+ size_t total_attributes = 1 + (num_characteristics * 2) + total_descriptors;
+
+ // allocate one extra so that we can know later where the final attribute is
+ struct bt_gatt_attr *svc_attributes = m_new(struct bt_gatt_attr, total_attributes + 1);
+ mp_obj_list_append(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->objs_list, MP_OBJ_FROM_PTR(svc_attributes));
+
+ size_t handle_index = 0;
+ size_t descriptor_index = 0;
+ size_t attr_index = 0;
+ // bitfield of the handles we should ignore, should be more than enough for most applications
+ uint64_t attrs_to_ignore = 0;
+ uint64_t attrs_are_chrs = 0;
+ uint64_t chr_has_ccc = 0;
+
+ add_service(create_zephyr_uuid(service_uuid), &svc_attributes[attr_index]);
+ attr_index += 1;
+
+ for (size_t i = 0; i < num_characteristics; ++i) {
+
+ struct add_characteristic add_char;
+ add_char.uuid = create_zephyr_uuid(characteristic_uuids[i]);
+ add_char.permissions = 0;
+ add_char.properties = 0;
+ if (characteristic_flags[i] & MP_BLUETOOTH_CHARACTERISTIC_FLAG_READ) {
+ add_char.permissions |= BT_GATT_PERM_READ;
+ add_char.properties |= BT_GATT_CHRC_READ;
+ }
+ if (characteristic_flags[i] & MP_BLUETOOTH_CHARACTERISTIC_FLAG_NOTIFY) {
+ add_char.properties |= BT_GATT_CHRC_NOTIFY;
+ }
+ if (characteristic_flags[i] & MP_BLUETOOTH_CHARACTERISTIC_FLAG_INDICATE) {
+ add_char.properties |= BT_GATT_CHRC_INDICATE;
+ }
+ if (characteristic_flags[i] & (MP_BLUETOOTH_CHARACTERISTIC_FLAG_WRITE | MP_BLUETOOTH_CHARACTERISTIC_FLAG_WRITE_NO_RESPONSE)) {
+ add_char.permissions |= BT_GATT_PERM_WRITE;
+ add_char.properties |= (BT_GATT_CHRC_WRITE | BT_GATT_CHRC_WRITE_WITHOUT_RESP);
+ }
+
+ add_characteristic(&add_char, &svc_attributes[attr_index], &svc_attributes[attr_index + 1]);
+
+ struct bt_gatt_attr *curr_char = &svc_attributes[attr_index];
+ attrs_are_chrs |= (1 << attr_index);
+ if (characteristic_flags[i] & (MP_BLUETOOTH_CHARACTERISTIC_FLAG_NOTIFY | MP_BLUETOOTH_CHARACTERISTIC_FLAG_INDICATE)) {
+ chr_has_ccc |= (1 << attr_index);
+ }
+ attr_index += 1;
+ attrs_to_ignore |= (1 << attr_index); // ignore the value handle
+ attr_index += 1;
+
+ if (num_descriptors[i] > 0) {
+ for (size_t j = 0; j < num_descriptors[i]; ++j) {
+
+ struct add_descriptor add_desc;
+ add_desc.uuid = create_zephyr_uuid(descriptor_uuids[descriptor_index]);
+ add_desc.permissions = 0;
+ if (descriptor_flags[descriptor_index] & MP_BLUETOOTH_CHARACTERISTIC_FLAG_READ) {
+ add_desc.permissions |= BT_GATT_PERM_READ;
+ }
+ if (descriptor_flags[descriptor_index] & (MP_BLUETOOTH_CHARACTERISTIC_FLAG_WRITE | MP_BLUETOOTH_CHARACTERISTIC_FLAG_WRITE_NO_RESPONSE)) {
+ add_desc.permissions |= BT_GATT_PERM_WRITE;
+ }
+
+ add_descriptor(curr_char, &add_desc, &svc_attributes[attr_index]);
+ attr_index += 1;
+
+ descriptor_index++;
+ }
+ }
+
+ // to support indications and notifications we must add the CCC descriptor manually
+ if (characteristic_flags[i] & (MP_BLUETOOTH_CHARACTERISTIC_FLAG_NOTIFY | MP_BLUETOOTH_CHARACTERISTIC_FLAG_INDICATE)) {
+ struct add_descriptor add_desc;
+ mp_obj_bluetooth_uuid_t ccc_uuid;
+ ccc_uuid.base.type = &mp_type_bluetooth_uuid;
+ ccc_uuid.data[0] = BT_UUID_GATT_CCC_VAL & 0xff;
+ ccc_uuid.data[1] = (BT_UUID_GATT_CCC_VAL >> 8) & 0xff;
+ ccc_uuid.type = MP_BLUETOOTH_UUID_TYPE_16;
+ add_desc.uuid = create_zephyr_uuid(&ccc_uuid);
+ add_desc.permissions = BT_GATT_PERM_READ | BT_GATT_PERM_WRITE;
+
+ attrs_to_ignore |= (1 << attr_index);
+
+ add_descriptor(curr_char, &add_desc, &svc_attributes[attr_index]);
+ attr_index += 1;
+ }
+ }
+
+ struct bt_gatt_service *service = m_new(struct bt_gatt_service, 1);
+ mp_obj_list_append(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->objs_list, MP_OBJ_FROM_PTR(service));
+ service->attrs = svc_attributes;
+ service->attr_count = attr_index;
+ // invalidate the last attribute uuid pointer so that we new this is the end of attributes for this service
+ svc_attributes[attr_index].uuid = NULL;
+
+ // Note: advertising must be stopped for gatts registration to work
+
+ int err = bt_gatt_service_register(service);
+ if (err) {
+ return bt_err_to_errno(err);
+ }
+
+ // now that the service has been registered, we can assign the handles for the characteristics and the descriptors
+ // we are not interested in the handle of the service itself, so we start the loop from index 1
+ for (int i = 1; i < total_attributes; i++) {
+ // store all the relevant handles (characteristics and descriptors defined in Python)
+ if (!((uint64_t)(attrs_to_ignore >> i) & (uint64_t)0x01)) {
+ if (svc_attributes[i].user_data == NULL) {
+ mp_bluetooth_gatts_db_create_entry(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->gatts_db, svc_attributes[i].handle, MP_BLUETOOTH_DEFAULT_ATTR_LEN);
+ mp_bluetooth_gatts_db_entry_t *entry = mp_bluetooth_gatts_db_lookup(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->gatts_db, svc_attributes[i].handle);
+ svc_attributes[i].user_data = entry->data;
+ } else if (((uint64_t)(attrs_are_chrs >> i) & (uint64_t)0x01)) {
+ if (svc_attributes[i + 1].user_data == NULL) {
+ mp_bluetooth_gatts_db_create_entry(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->gatts_db, svc_attributes[i].handle, MP_BLUETOOTH_DEFAULT_ATTR_LEN);
+ mp_bluetooth_gatts_db_entry_t *entry = mp_bluetooth_gatts_db_lookup(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->gatts_db, svc_attributes[i].handle);
+ svc_attributes[i + 1].user_data = entry->data;
+
+ if (((uint64_t)(chr_has_ccc >> i) & (uint64_t)0x01)) {
+ // create another database entry for the ccc of this characteristic
+ mp_bluetooth_gatts_db_create_entry(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->gatts_db, svc_attributes[i].handle + 2, 1);
+ }
+ }
+ }
+ handles[handle_index++] = svc_attributes[i].handle;
+ }
+ }
+
+ MP_STATE_PORT(bluetooth_zephyr_root_pointers)->services[MP_STATE_PORT(bluetooth_zephyr_root_pointers)->n_services++] = service;
+
+ return 0;
+
+ #else
return MP_EOPNOTSUPP;
+ #endif
}
int mp_bluetooth_gap_disconnect(uint16_t conn_handle) {
if (!mp_bluetooth_is_active()) {
return ERRNO_BLUETOOTH_NOT_ACTIVE;
}
- return MP_EOPNOTSUPP;
+ mp_bt_zephyr_conn_t *connection = mp_bt_zephyr_find_connection(conn_handle);
+ if (connection) {
+ return bt_conn_disconnect(connection->conn, BT_HCI_ERR_REMOTE_USER_TERM_CONN);
+ }
+ return MP_ENOENT;
}
int mp_bluetooth_gatts_read(uint16_t value_handle, const uint8_t **value, size_t *value_len) {
@@ -312,24 +642,155 @@ int mp_bluetooth_gatts_write(uint16_t value_handle, const uint8_t *value, size_t
if (!mp_bluetooth_is_active()) {
return ERRNO_BLUETOOTH_NOT_ACTIVE;
}
- if (send_update) {
- return MP_EOPNOTSUPP;
+
+ int err = mp_bluetooth_gatts_db_write(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->gatts_db, value_handle, value, value_len);
+
+ if ((err == 0) && send_update) {
+ struct bt_gatt_attr *attr_val = mp_bt_zephyr_find_attr_by_handle(value_handle + 1);
+ mp_bluetooth_gatts_db_entry_t *ccc_entry = mp_bluetooth_gatts_db_lookup(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->gatts_db, value_handle + 2);
+
+ if (ccc_entry && (ccc_entry->data[0] == BT_GATT_CCC_NOTIFY)) {
+ err = bt_gatt_notify(NULL, attr_val, value, value_len);
+ } else if (ccc_entry && (ccc_entry->data[0] == BT_GATT_CCC_INDICATE)) {
+ struct bt_gatt_indicate_params params = {
+ .uuid = NULL,
+ .attr = attr_val,
+ .func = mp_bt_zephyr_gatt_indicate_done,
+ .destroy = NULL,
+ .data = value,
+ .len = value_len
+ };
+ err = bt_gatt_indicate(NULL, &params);
+ }
+ }
+ return err;
+}
+
+static void mp_bt_zephyr_gatt_indicate_done(struct bt_conn *conn, struct bt_gatt_indicate_params *params, uint8_t err) {
+ struct bt_conn_info info;
+ bt_conn_get_info(conn, &info);
+ uint16_t chr_handle = params->attr->handle - 1;
+ mp_bluetooth_gatts_on_indicate_complete(info.id, chr_handle, err);
+}
+
+static ssize_t mp_bt_zephyr_gatts_attr_read(struct bt_conn *conn, const struct bt_gatt_attr *attr, void *buf, uint16_t len, uint16_t offset) {
+ // we receive the value handle, but to look up in the gatts db we need the characteristic handle, and that is is the value handle minus 1
+ uint16_t _handle = attr->handle - 1;
+
+ DEBUG_printf("BLE attr read for handle %d\n", attr->handle);
+
+ mp_bluetooth_gatts_db_entry_t *entry = mp_bluetooth_gatts_db_lookup(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->gatts_db, _handle);
+ if (!entry) {
+ // it could be a descriptor instead
+ _handle = attr->handle;
+ entry = mp_bluetooth_gatts_db_lookup(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->gatts_db, _handle);
+ if (!entry) {
+ return BT_GATT_ERR(BT_ATT_ERR_INVALID_HANDLE);
+ }
+ }
+
+ return bt_gatt_attr_read(conn, attr, buf, len, offset, entry->data, entry->data_len);
+}
+
+static ssize_t mp_bt_zephyr_gatts_attr_write(struct bt_conn *conn, const struct bt_gatt_attr *attr, const void *buf, uint16_t len, uint16_t offset, uint8_t flags) {
+ struct bt_conn_info info;
+ bt_conn_get_info(conn, &info);
+
+ DEBUG_printf("BLE attr write for handle %d\n", attr->handle);
+
+ // the characteristic handle is the value handle minus 1
+ uint16_t _handle = attr->handle - 1;
+
+ mp_bluetooth_gatts_db_entry_t *entry = mp_bluetooth_gatts_db_lookup(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->gatts_db, _handle);
+ if (!entry) {
+ // it could be a descriptor instead
+ _handle = attr->handle;
+ entry = mp_bluetooth_gatts_db_lookup(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->gatts_db, _handle);
+ if (!entry) {
+ return BT_GATT_ERR(BT_ATT_ERR_INVALID_HANDLE);
+ }
+ }
+
+ // Don't write anything if prepare flag is set
+ if (flags & BT_GATT_WRITE_FLAG_PREPARE) {
+ return 0;
+ }
+
+ if (offset > entry->data_alloc) {
+ return BT_GATT_ERR(BT_ATT_ERR_INVALID_OFFSET);
+ }
+
+ if ((offset + len) > entry->data_alloc) {
+ return BT_GATT_ERR(BT_ATT_ERR_INVALID_ATTRIBUTE_LEN);
+ }
+
+ if (entry->append) {
+ offset = entry->data_len;
+ }
+
+ // copy the data into the buffer in the gatts database
+ memcpy(&entry->data[offset], buf, len);
+ entry->data_len = offset + len;
+
+ mp_bluetooth_gatts_on_write(info.id, _handle);
+
+ return len;
+}
+
+static struct bt_gatt_attr *mp_bt_zephyr_find_attr_by_handle(uint16_t value_handle) {
+ for (int i = 0; i < MP_STATE_PORT(bluetooth_zephyr_root_pointers)->n_services; i++) {
+ int j = 0;
+ while (MP_STATE_PORT(bluetooth_zephyr_root_pointers)->services[i]->attrs[j].uuid != NULL) {
+ if (MP_STATE_PORT(bluetooth_zephyr_root_pointers)->services[i]->attrs[j].handle == value_handle) {
+ return &MP_STATE_PORT(bluetooth_zephyr_root_pointers)->services[i]->attrs[j];
+ }
+ j++;
+ }
}
- return mp_bluetooth_gatts_db_write(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->gatts_db, value_handle, value, value_len);
+ return NULL;
}
int mp_bluetooth_gatts_notify_indicate(uint16_t conn_handle, uint16_t value_handle, int gatts_op, const uint8_t *value, size_t value_len) {
if (!mp_bluetooth_is_active()) {
return ERRNO_BLUETOOTH_NOT_ACTIVE;
}
- return MP_EOPNOTSUPP;
+
+ int err = MP_ENOENT;
+ mp_bt_zephyr_conn_t *connection = mp_bt_zephyr_find_connection(conn_handle);
+
+ if (connection) {
+ struct bt_gatt_attr *attr_val = mp_bt_zephyr_find_attr_by_handle(value_handle + 1);
+
+ if (attr_val) {
+ switch (gatts_op) {
+ case MP_BLUETOOTH_GATTS_OP_NOTIFY: {
+ err = bt_gatt_notify(connection->conn, attr_val, value, value_len);
+ break;
+ }
+ case MP_BLUETOOTH_GATTS_OP_INDICATE: {
+ struct bt_gatt_indicate_params params = {
+ .uuid = NULL,
+ .attr = attr_val,
+ .func = mp_bt_zephyr_gatt_indicate_done,
+ .destroy = NULL,
+ .data = value,
+ .len = value_len
+ };
+ err = bt_gatt_indicate(connection->conn, &params);
+ break;
+ }
+ }
+ }
+ }
+
+ return err;
}
int mp_bluetooth_gatts_set_buffer(uint16_t value_handle, size_t len, bool append) {
if (!mp_bluetooth_is_active()) {
return ERRNO_BLUETOOTH_NOT_ACTIVE;
}
- return MP_EOPNOTSUPP;
+ return mp_bluetooth_gatts_db_resize(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->gatts_db, value_handle, len, append);
}
int mp_bluetooth_get_preferred_mtu(void) {
@@ -404,6 +865,133 @@ int mp_bluetooth_gap_peripheral_connect_cancel(void) {
#endif // MICROPY_PY_BLUETOOTH_ENABLE_CENTRAL_MODE
+// Note: modbluetooth UUIDs store their data in LE.
+static struct bt_uuid *create_zephyr_uuid(const mp_obj_bluetooth_uuid_t *uuid) {
+ struct bt_uuid *result = (struct bt_uuid *)m_new(union uuid_u, 1);
+ mp_obj_list_append(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->objs_list, MP_OBJ_FROM_PTR(result));
+ if (uuid->type == MP_BLUETOOTH_UUID_TYPE_16) {
+ bt_uuid_create(result, uuid->data, 2);
+ } else if (uuid->type == MP_BLUETOOTH_UUID_TYPE_32) {
+ bt_uuid_create(result, uuid->data, 4);
+ } else { // MP_BLUETOOTH_UUID_TYPE_128
+ bt_uuid_create(result, uuid->data, 16);
+ }
+ return result;
+}
+
+static void gatt_db_add(const struct bt_gatt_attr *pattern, struct bt_gatt_attr *attr, size_t user_data_len) {
+ const union uuid_u *u = CONTAINER_OF(pattern->uuid, union uuid_u, uuid);
+ size_t uuid_size = sizeof(u->u16);
+
+ if (u->uuid.type == BT_UUID_TYPE_32) {
+ uuid_size = sizeof(u->u32);
+ } else if (u->uuid.type == BT_UUID_TYPE_128) {
+ uuid_size = sizeof(u->u128);
+ }
+
+ memcpy(attr, pattern, sizeof(*attr));
+
+ // Store the UUID.
+ attr->uuid = (const struct bt_uuid *)m_new(union uuid_u, 1);
+ mp_obj_list_append(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->objs_list, MP_OBJ_FROM_PTR(attr->uuid));
+ memcpy((void *)attr->uuid, &u->uuid, uuid_size);
+
+ // Copy user_data to the buffer.
+ if (user_data_len) {
+ attr->user_data = m_new(uint8_t, user_data_len);
+ mp_obj_list_append(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->objs_list, MP_OBJ_FROM_PTR(attr->user_data));
+ memcpy(attr->user_data, pattern->user_data, user_data_len);
+ }
+}
+
+static void add_service(const struct bt_uuid *u, struct bt_gatt_attr *attr) {
+ union uuid_u *uuid = (union uuid_u *)u;
+
+ size_t uuid_size = sizeof(uuid->u16);
+
+ if (uuid->uuid.type == BT_UUID_TYPE_32) {
+ uuid_size = sizeof(uuid->u32);
+ } else if (uuid->uuid.type == BT_UUID_TYPE_128) {
+ uuid_size = sizeof(uuid->u128);
+ }
+
+ gatt_db_add(&(struct bt_gatt_attr)BT_GATT_PRIMARY_SERVICE(&uuid->uuid), attr, uuid_size);
+}
+
+static void add_characteristic(struct add_characteristic *ch, struct bt_gatt_attr *attr_chrc, struct bt_gatt_attr *attr_value) {
+ struct bt_gatt_chrc *chrc_data;
+
+ // Add Characteristic Declaration
+ gatt_db_add(&(struct bt_gatt_attr)
+ BT_GATT_ATTRIBUTE(BT_UUID_GATT_CHRC,
+ BT_GATT_PERM_READ,
+ bt_gatt_attr_read_chrc, NULL,
+ (&(struct bt_gatt_chrc) {})), attr_chrc, sizeof(*chrc_data));
+
+ // Allow prepare writes
+ ch->permissions |= BT_GATT_PERM_PREPARE_WRITE;
+
+ // Add Characteristic Value
+ gatt_db_add(&(struct bt_gatt_attr)
+ BT_GATT_ATTRIBUTE(ch->uuid,
+ ch->permissions & GATT_PERM_MASK,
+ mp_bt_zephyr_gatts_attr_read, mp_bt_zephyr_gatts_attr_write, NULL), attr_value, 0);
+
+ chrc_data = attr_chrc->user_data;
+ chrc_data->properties = ch->properties;
+ chrc_data->uuid = attr_value->uuid;
+}
+
+static void ccc_cfg_changed(const struct bt_gatt_attr *attr, uint16_t value) {
+ mp_bluetooth_gatts_db_entry_t *entry = mp_bluetooth_gatts_db_lookup(MP_STATE_PORT(bluetooth_zephyr_root_pointers)->gatts_db, attr->handle);
+ entry->data[0] = value;
+}
+
+static struct bt_gatt_attr ccc_definition = BT_GATT_CCC(ccc_cfg_changed, BT_GATT_PERM_READ | BT_GATT_PERM_WRITE);
+
+static void add_ccc(struct bt_gatt_attr *attr, struct bt_gatt_attr *attr_desc) {
+ struct bt_gatt_chrc *chrc = attr->user_data;
+
+ // Check characteristic properties
+ if (!(chrc->properties & (BT_GATT_CHRC_NOTIFY | BT_GATT_CHRC_INDICATE))) {
+ mp_raise_OSError(MP_EINVAL);
+ }
+
+ // Add CCC descriptor to GATT database
+ gatt_db_add(&ccc_definition, attr_desc, 0);
+}
+
+static void add_cep(const struct bt_gatt_attr *attr_chrc, struct bt_gatt_attr *attr_desc) {
+ struct bt_gatt_chrc *chrc = attr_chrc->user_data;
+ struct bt_gatt_cep cep_value;
+
+ // Extended Properties bit shall be set
+ if (!(chrc->properties & BT_GATT_CHRC_EXT_PROP)) {
+ mp_raise_OSError(MP_EINVAL);
+ }
+
+ cep_value.properties = 0x0000;
+
+ // Add CEP descriptor to GATT database
+ gatt_db_add(&(struct bt_gatt_attr)BT_GATT_CEP(&cep_value), attr_desc, sizeof(cep_value));
+}
+
+static void add_descriptor(struct bt_gatt_attr *chrc, struct add_descriptor *d, struct bt_gatt_attr *attr_desc) {
+ if (!bt_uuid_cmp(d->uuid, BT_UUID_GATT_CEP)) {
+ add_cep(chrc, attr_desc);
+ } else if (!bt_uuid_cmp(d->uuid, BT_UUID_GATT_CCC)) {
+ add_ccc(chrc, attr_desc);
+ } else {
+ // Allow prepare writes
+ d->permissions |= BT_GATT_PERM_PREPARE_WRITE;
+
+ gatt_db_add(&(struct bt_gatt_attr)
+ BT_GATT_DESCRIPTOR(d->uuid,
+ d->permissions & GATT_PERM_MASK,
+ mp_bt_zephyr_gatts_attr_read, mp_bt_zephyr_gatts_attr_write, NULL), attr_desc, 0);
+ }
+}
+
MP_REGISTER_ROOT_POINTER(struct _mp_bluetooth_zephyr_root_pointers_t *bluetooth_zephyr_root_pointers);
#endif // MICROPY_PY_BLUETOOTH