From 86c26db3ff40b87a8d058a67ac5c71f71a3818b6 Mon Sep 17 00:00:00 2001 From: Damien George Date: Thu, 20 Feb 2020 14:43:28 +1100 Subject: [PATCH] extmod/btstack: Implement service registration. Work done in collaboration with Jim Mussared aka @jimmo. --- extmod/btstack/modbluetooth_btstack.c | 122 ++++++++++++++++++++++++++ extmod/modbluetooth.h | 1 + 2 files changed, 123 insertions(+) diff --git a/extmod/btstack/modbluetooth_btstack.c b/extmod/btstack/modbluetooth_btstack.c index 3e3852e5c2..e7f9014048 100644 --- a/extmod/btstack/modbluetooth_btstack.c +++ b/extmod/btstack/modbluetooth_btstack.c @@ -37,6 +37,41 @@ volatile int mp_bluetooth_btstack_state = MP_BLUETOOTH_BTSTACK_STATE_OFF; +static btstack_packet_callback_registration_t hci_event_callback_registration; + +STATIC void att_server_packet_handler(uint8_t packet_type, uint16_t channel, uint8_t *packet, uint16_t size) { + // printf("att_server_packet_handler: packet_type: %u, channel: %u, packet: %p, size: %u\n", packet_type, channel, packet, size); + if (packet_type == HCI_EVENT_PACKET) { + uint8_t event_type = hci_event_packet_get_type(packet); + if (event_type == ATT_EVENT_CONNECTED) { + printf(" --> att connected\n"); + } else if (event_type == ATT_EVENT_DISCONNECTED) { + printf(" --> att disconnected\n"); + } else if (event_type == HCI_EVENT_LE_META) { + printf(" --> att le meta\n"); + } else if (event_type == HCI_EVENT_DISCONNECTION_COMPLETE) { + printf(" --> att disconnect complete\n"); + } else { + printf(" --> att type: unknown (%u)\n", event_type); + } + } +} + +STATIC void hci_packet_handler(uint8_t packet_type, uint16_t channel, uint8_t *packet, uint16_t size) { + // printf("hci_packet_handler: packet_type: %u, channel: %u, packet: %p, size: %u\n", packet_type, channel, packet, size); + if (packet_type == HCI_EVENT_PACKET) { + uint8_t event_type = hci_event_packet_get_type(packet); + if (event_type == HCI_EVENT_TRANSPORT_PACKET_SENT || event_type == HCI_EVENT_COMMAND_COMPLETE) { + } else if (event_type == BTSTACK_EVENT_NR_CONNECTIONS_CHANGED) { + printf(" --> hci type: # conns changed\n"); + } else if (event_type == HCI_EVENT_VENDOR_SPECIFIC) { + printf(" --> hci type: vendor specific\n"); + } else { + printf(" --> hci type: unknown (%u)\n", event_type); + } + } +} + int mp_bluetooth_init(void) { btstack_memory_init(); @@ -58,6 +93,15 @@ int mp_bluetooth_init(void) { mp_bluetooth_btstack_port_start(); mp_bluetooth_btstack_state = MP_BLUETOOTH_BTSTACK_STATE_ACTIVE; + + // Register for HCI events. + memset(&hci_event_callback_registration, 0, sizeof(btstack_packet_callback_registration_t)); + hci_event_callback_registration.callback = &hci_packet_handler; + hci_add_event_handler(&hci_event_callback_registration); + + // Register for ATT event. + att_server_register_packet_handler(&att_server_packet_handler); + return 0; } @@ -118,14 +162,92 @@ void mp_bluetooth_gap_advertise_stop(void) { } int mp_bluetooth_gatts_register_service_begin(bool append) { + if (!append) { + // This will reset the DB. + // Becase the DB is statically allocated, there's no problem with just re-initing it. + // Note this would be a memory leak if we enabled HAVE_MALLOC (there's no API to free the existing db). + att_db_util_init(); + + att_db_util_add_service_uuid16(GAP_SERVICE_UUID); + att_db_util_add_characteristic_uuid16(GAP_DEVICE_NAME_UUID, ATT_PROPERTY_READ, ATT_SECURITY_NONE, ATT_SECURITY_NONE, (uint8_t *)"MPY BTSTACK", 11); + + att_db_util_add_service_uuid16(0x1801); + att_db_util_add_characteristic_uuid16(0x2a05, ATT_PROPERTY_READ, ATT_SECURITY_NONE, ATT_SECURITY_NONE, NULL, 0); + + } + return 0; } +STATIC uint16_t att_read_callback(hci_con_handle_t connection_handle, uint16_t att_handle, uint16_t offset, uint8_t *buffer, uint16_t buffer_size) { + printf("btstack: att_read_callback (handle: %u, offset: %u, buffer: %p, size: %u)\n", att_handle, offset, buffer, buffer_size); + return 0; +} + +STATIC int att_write_callback(hci_con_handle_t connection_handle, uint16_t att_handle, uint16_t transaction_mode, uint16_t offset, uint8_t *buffer, uint16_t buffer_size) { + printf("btstack: att_write_callback (handle: %u, mode: %u, offset: %u, buffer: %p, size: %u)\n", att_handle, transaction_mode, offset, buffer, buffer_size); + return 0; +} + +// Note: modbluetooth UUIDs store their data in LE. +STATIC inline uint16_t get_uuid16(const mp_obj_bluetooth_uuid_t *uuid) { + return (uuid->data[1] << 8) | uuid->data[0]; +} + int mp_bluetooth_gatts_register_service(mp_obj_bluetooth_uuid_t *service_uuid, mp_obj_bluetooth_uuid_t **characteristic_uuids, uint8_t *characteristic_flags, mp_obj_bluetooth_uuid_t **descriptor_uuids, uint8_t *descriptor_flags, uint8_t *num_descriptors, uint16_t *handles, size_t num_characteristics) { + // TODO: It's not clear which endian btstack needs for uuids. + + uint16_t service_handle; + if (service_uuid->type == MP_BLUETOOTH_UUID_TYPE_16) { + service_handle = att_db_util_add_service_uuid16(get_uuid16(service_uuid)); + } else if (service_uuid->type == MP_BLUETOOTH_UUID_TYPE_128) { + service_handle = att_db_util_add_service_uuid128(service_uuid->data); + } else { + return MP_EINVAL; + } + + printf("Registered service with handle %u\n", service_handle); + + size_t handle_index = 0; + size_t descriptor_index = 0; + + for (size_t i = 0; i < num_characteristics; ++i) { + uint16_t props = characteristic_flags[i] | ATT_PROPERTY_DYNAMIC; + uint16_t read_permission = ATT_SECURITY_NONE; + uint16_t write_permission = ATT_SECURITY_NONE; + if (characteristic_uuids[i]->type == MP_BLUETOOTH_UUID_TYPE_16) { + handles[handle_index] = att_db_util_add_characteristic_uuid16(get_uuid16(characteristic_uuids[i]), props, read_permission, write_permission, NULL, 0); + } else if (characteristic_uuids[i]->type == MP_BLUETOOTH_UUID_TYPE_128) { + handles[handle_index] = att_db_util_add_characteristic_uuid128(characteristic_uuids[i]->data, props, read_permission, write_permission, NULL, 0); + } else { + return MP_EINVAL; + } + printf("Registered char with handle %u\n", handles[handle_index]); + ++handle_index; + + for (size_t j = 0; j < num_descriptors[i]; ++j) { + uint16_t props = descriptor_flags[descriptor_index] | ATT_PROPERTY_DYNAMIC; + uint16_t read_permission = ATT_SECURITY_NONE; + uint16_t write_permission = ATT_SECURITY_NONE; + + if (descriptor_uuids[descriptor_index]->type == MP_BLUETOOTH_UUID_TYPE_16) { + handles[handle_index] = att_db_util_add_descriptor_uuid16(get_uuid16(descriptor_uuids[descriptor_index]), props, read_permission, write_permission, NULL, 0); + } else if (descriptor_uuids[descriptor_index]->type == MP_BLUETOOTH_UUID_TYPE_128) { + handles[handle_index] = att_db_util_add_descriptor_uuid128(descriptor_uuids[descriptor_index]->data, props, read_permission, write_permission, NULL, 0); + } else { + return MP_EINVAL; + } + printf("Registered desc with handle %u\n", handles[handle_index]); + ++descriptor_index; + ++handle_index; + } + } + return 0; } int mp_bluetooth_gatts_register_service_end(void) { + att_server_init(att_db_util_get_address(), &att_read_callback, &att_write_callback); return 0; } diff --git a/extmod/modbluetooth.h b/extmod/modbluetooth.h index 661130802a..ed280e3897 100644 --- a/extmod/modbluetooth.h +++ b/extmod/modbluetooth.h @@ -63,6 +63,7 @@ // Advertisement packet lengths #define MP_BLUETOOTH_GAP_ADV_MAX_LEN (32) +// These match the spec values for these flags so can be passed directly to the stack. #define MP_BLUETOOTH_CHARACTERISTIC_FLAG_READ (1 << 1) #define MP_BLUETOOTH_CHARACTERISTIC_FLAG_WRITE (1 << 3) #define MP_BLUETOOTH_CHARACTERISTIC_FLAG_NOTIFY (1 << 4)