|
|
@@ -0,0 +1,794 @@
|
|
|
+/*
|
|
|
+ * SPDX-FileCopyrightText: 2021-2022 Espressif Systems (Shanghai) CO LTD
|
|
|
+ *
|
|
|
+ * SPDX-License-Identifier: Unlicense OR CC0-1.0
|
|
|
+ */
|
|
|
+
|
|
|
+#include <assert.h>
|
|
|
+#include <string.h>
|
|
|
+#include "host/ble_hs.h"
|
|
|
+#include "coc_blecent.h"
|
|
|
+
|
|
|
+static void *peer_svc_mem;
|
|
|
+static struct os_mempool peer_svc_pool;
|
|
|
+
|
|
|
+static void *peer_chr_mem;
|
|
|
+static struct os_mempool peer_chr_pool;
|
|
|
+
|
|
|
+static void *peer_dsc_mem;
|
|
|
+static struct os_mempool peer_dsc_pool;
|
|
|
+
|
|
|
+static void *peer_mem;
|
|
|
+static struct os_mempool peer_pool;
|
|
|
+static SLIST_HEAD(, peer) peers;
|
|
|
+
|
|
|
+static struct peer_svc *
|
|
|
+peer_svc_find_range(struct peer *peer, uint16_t attr_handle);
|
|
|
+static struct peer_svc *
|
|
|
+peer_svc_find(struct peer *peer, uint16_t svc_start_handle,
|
|
|
+ struct peer_svc **out_prev);
|
|
|
+int
|
|
|
+peer_svc_is_empty(const struct peer_svc *svc);
|
|
|
+
|
|
|
+uint16_t
|
|
|
+chr_end_handle(const struct peer_svc *svc, const struct peer_chr *chr);
|
|
|
+int
|
|
|
+chr_is_empty(const struct peer_svc *svc, const struct peer_chr *chr);
|
|
|
+static struct peer_chr *
|
|
|
+peer_chr_find(const struct peer_svc *svc, uint16_t chr_def_handle,
|
|
|
+ struct peer_chr **out_prev);
|
|
|
+static void
|
|
|
+peer_disc_chrs(struct peer *peer);
|
|
|
+
|
|
|
+static int
|
|
|
+peer_dsc_disced(uint16_t conn_handle, const struct ble_gatt_error *error,
|
|
|
+ uint16_t chr_val_handle, const struct ble_gatt_dsc *dsc,
|
|
|
+ void *arg);
|
|
|
+
|
|
|
+struct peer *
|
|
|
+peer_find(uint16_t conn_handle)
|
|
|
+{
|
|
|
+ struct peer *peer;
|
|
|
+
|
|
|
+ SLIST_FOREACH(peer, &peers, next) {
|
|
|
+ if (peer->conn_handle == conn_handle) {
|
|
|
+ return peer;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return NULL;
|
|
|
+}
|
|
|
+
|
|
|
+static void
|
|
|
+peer_disc_complete(struct peer *peer, int rc)
|
|
|
+{
|
|
|
+ peer->disc_prev_chr_val = 0;
|
|
|
+
|
|
|
+ /* Notify caller that discovery has completed. */
|
|
|
+ if (peer->disc_cb != NULL) {
|
|
|
+ peer->disc_cb(peer, rc, peer->disc_cb_arg);
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+static struct peer_dsc *
|
|
|
+peer_dsc_find_prev(const struct peer_chr *chr, uint16_t dsc_handle)
|
|
|
+{
|
|
|
+ struct peer_dsc *prev;
|
|
|
+ struct peer_dsc *dsc;
|
|
|
+
|
|
|
+ prev = NULL;
|
|
|
+ SLIST_FOREACH(dsc, &chr->dscs, next) {
|
|
|
+ if (dsc->dsc.handle >= dsc_handle) {
|
|
|
+ break;
|
|
|
+ }
|
|
|
+
|
|
|
+ prev = dsc;
|
|
|
+ }
|
|
|
+
|
|
|
+ return prev;
|
|
|
+}
|
|
|
+
|
|
|
+static struct peer_dsc *
|
|
|
+peer_dsc_find(const struct peer_chr *chr, uint16_t dsc_handle,
|
|
|
+ struct peer_dsc **out_prev)
|
|
|
+{
|
|
|
+ struct peer_dsc *prev;
|
|
|
+ struct peer_dsc *dsc;
|
|
|
+
|
|
|
+ prev = peer_dsc_find_prev(chr, dsc_handle);
|
|
|
+ if (prev == NULL) {
|
|
|
+ dsc = SLIST_FIRST(&chr->dscs);
|
|
|
+ } else {
|
|
|
+ dsc = SLIST_NEXT(prev, next);
|
|
|
+ }
|
|
|
+
|
|
|
+ if (dsc != NULL && dsc->dsc.handle != dsc_handle) {
|
|
|
+ dsc = NULL;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (out_prev != NULL) {
|
|
|
+ *out_prev = prev;
|
|
|
+ }
|
|
|
+ return dsc;
|
|
|
+}
|
|
|
+
|
|
|
+static int
|
|
|
+peer_dsc_add(struct peer *peer, uint16_t chr_val_handle,
|
|
|
+ const struct ble_gatt_dsc *gatt_dsc)
|
|
|
+{
|
|
|
+ struct peer_dsc *prev;
|
|
|
+ struct peer_dsc *dsc;
|
|
|
+ struct peer_svc *svc;
|
|
|
+ struct peer_chr *chr;
|
|
|
+
|
|
|
+ svc = peer_svc_find_range(peer, chr_val_handle);
|
|
|
+ if (svc == NULL) {
|
|
|
+ /* Can't find service for discovered descriptor; this shouldn't
|
|
|
+ * happen.
|
|
|
+ */
|
|
|
+ assert(0);
|
|
|
+ return BLE_HS_EUNKNOWN;
|
|
|
+ }
|
|
|
+
|
|
|
+ chr = peer_chr_find(svc, chr_val_handle, NULL);
|
|
|
+ if (chr == NULL) {
|
|
|
+ /* Can't find characteristic for discovered descriptor; this shouldn't
|
|
|
+ * happen.
|
|
|
+ */
|
|
|
+ assert(0);
|
|
|
+ return BLE_HS_EUNKNOWN;
|
|
|
+ }
|
|
|
+
|
|
|
+ dsc = peer_dsc_find(chr, gatt_dsc->handle, &prev);
|
|
|
+ if (dsc != NULL) {
|
|
|
+ /* Descriptor already discovered. */
|
|
|
+ return 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ dsc = os_memblock_get(&peer_dsc_pool);
|
|
|
+ if (dsc == NULL) {
|
|
|
+ /* Out of memory. */
|
|
|
+ return BLE_HS_ENOMEM;
|
|
|
+ }
|
|
|
+ memset(dsc, 0, sizeof * dsc);
|
|
|
+
|
|
|
+ dsc->dsc = *gatt_dsc;
|
|
|
+
|
|
|
+ if (prev == NULL) {
|
|
|
+ SLIST_INSERT_HEAD(&chr->dscs, dsc, next);
|
|
|
+ } else {
|
|
|
+ SLIST_NEXT(prev, next) = dsc;
|
|
|
+ }
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|
|
|
+
|
|
|
+static void
|
|
|
+peer_disc_dscs(struct peer *peer)
|
|
|
+{
|
|
|
+ struct peer_chr *chr;
|
|
|
+ struct peer_svc *svc;
|
|
|
+ int rc;
|
|
|
+
|
|
|
+ /* Search through the list of discovered characteristics for the first
|
|
|
+ * characteristic that contains undiscovered descriptors. Then, discover
|
|
|
+ * all descriptors belonging to that characteristic.
|
|
|
+ */
|
|
|
+ SLIST_FOREACH(svc, &peer->svcs, next) {
|
|
|
+ SLIST_FOREACH(chr, &svc->chrs, next) {
|
|
|
+ if (!chr_is_empty(svc, chr) &&
|
|
|
+ SLIST_EMPTY(&chr->dscs) &&
|
|
|
+ peer->disc_prev_chr_val <= chr->chr.def_handle) {
|
|
|
+
|
|
|
+ rc = ble_gattc_disc_all_dscs(peer->conn_handle,
|
|
|
+ chr->chr.val_handle,
|
|
|
+ chr_end_handle(svc, chr),
|
|
|
+ peer_dsc_disced, peer);
|
|
|
+ if (rc != 0) {
|
|
|
+ peer_disc_complete(peer, rc);
|
|
|
+ }
|
|
|
+
|
|
|
+ peer->disc_prev_chr_val = chr->chr.val_handle;
|
|
|
+ return;
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /* All descriptors discovered. */
|
|
|
+ peer_disc_complete(peer, 0);
|
|
|
+}
|
|
|
+
|
|
|
+static int
|
|
|
+peer_dsc_disced(uint16_t conn_handle, const struct ble_gatt_error *error,
|
|
|
+ uint16_t chr_val_handle, const struct ble_gatt_dsc *dsc,
|
|
|
+ void *arg)
|
|
|
+{
|
|
|
+ struct peer *peer;
|
|
|
+ int rc;
|
|
|
+
|
|
|
+ peer = arg;
|
|
|
+ assert(peer->conn_handle == conn_handle);
|
|
|
+
|
|
|
+ switch (error->status) {
|
|
|
+ case 0:
|
|
|
+ rc = peer_dsc_add(peer, chr_val_handle, dsc);
|
|
|
+ break;
|
|
|
+
|
|
|
+ case BLE_HS_EDONE:
|
|
|
+ /* All descriptors in this characteristic discovered; start discovering
|
|
|
+ * descriptors in the next characteristic.
|
|
|
+ */
|
|
|
+ if (peer->disc_prev_chr_val > 0) {
|
|
|
+ peer_disc_dscs(peer);
|
|
|
+ }
|
|
|
+ rc = 0;
|
|
|
+ break;
|
|
|
+
|
|
|
+ default:
|
|
|
+ /* Error; abort discovery. */
|
|
|
+ rc = error->status;
|
|
|
+ break;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (rc != 0) {
|
|
|
+ /* Error; abort discovery. */
|
|
|
+ peer_disc_complete(peer, rc);
|
|
|
+ }
|
|
|
+
|
|
|
+ return rc;
|
|
|
+}
|
|
|
+
|
|
|
+uint16_t
|
|
|
+chr_end_handle(const struct peer_svc *svc, const struct peer_chr *chr)
|
|
|
+{
|
|
|
+ const struct peer_chr *next_chr;
|
|
|
+
|
|
|
+ next_chr = SLIST_NEXT(chr, next);
|
|
|
+ if (next_chr != NULL) {
|
|
|
+ return next_chr->chr.def_handle - 1;
|
|
|
+ } else {
|
|
|
+ return svc->svc.end_handle;
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+int
|
|
|
+chr_is_empty(const struct peer_svc *svc, const struct peer_chr *chr)
|
|
|
+{
|
|
|
+ return chr_end_handle(svc, chr) <= chr->chr.val_handle;
|
|
|
+}
|
|
|
+
|
|
|
+static struct peer_chr *
|
|
|
+peer_chr_find_prev(const struct peer_svc *svc, uint16_t chr_val_handle)
|
|
|
+{
|
|
|
+ struct peer_chr *prev;
|
|
|
+ struct peer_chr *chr;
|
|
|
+
|
|
|
+ prev = NULL;
|
|
|
+ SLIST_FOREACH(chr, &svc->chrs, next) {
|
|
|
+ if (chr->chr.val_handle >= chr_val_handle) {
|
|
|
+ break;
|
|
|
+ }
|
|
|
+
|
|
|
+ prev = chr;
|
|
|
+ }
|
|
|
+
|
|
|
+ return prev;
|
|
|
+}
|
|
|
+
|
|
|
+static struct peer_chr *
|
|
|
+peer_chr_find(const struct peer_svc *svc, uint16_t chr_val_handle,
|
|
|
+ struct peer_chr **out_prev)
|
|
|
+{
|
|
|
+ struct peer_chr *prev;
|
|
|
+ struct peer_chr *chr;
|
|
|
+
|
|
|
+ prev = peer_chr_find_prev(svc, chr_val_handle);
|
|
|
+ if (prev == NULL) {
|
|
|
+ chr = SLIST_FIRST(&svc->chrs);
|
|
|
+ } else {
|
|
|
+ chr = SLIST_NEXT(prev, next);
|
|
|
+ }
|
|
|
+
|
|
|
+ if (chr != NULL && chr->chr.val_handle != chr_val_handle) {
|
|
|
+ chr = NULL;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (out_prev != NULL) {
|
|
|
+ *out_prev = prev;
|
|
|
+ }
|
|
|
+ return chr;
|
|
|
+}
|
|
|
+
|
|
|
+static void
|
|
|
+peer_chr_delete(struct peer_chr *chr)
|
|
|
+{
|
|
|
+ struct peer_dsc *dsc;
|
|
|
+
|
|
|
+ while ((dsc = SLIST_FIRST(&chr->dscs)) != NULL) {
|
|
|
+ SLIST_REMOVE_HEAD(&chr->dscs, next);
|
|
|
+ os_memblock_put(&peer_dsc_pool, dsc);
|
|
|
+ }
|
|
|
+
|
|
|
+ os_memblock_put(&peer_chr_pool, chr);
|
|
|
+}
|
|
|
+
|
|
|
+static int
|
|
|
+peer_chr_add(struct peer *peer, uint16_t svc_start_handle,
|
|
|
+ const struct ble_gatt_chr *gatt_chr)
|
|
|
+{
|
|
|
+ struct peer_chr *prev;
|
|
|
+ struct peer_chr *chr;
|
|
|
+ struct peer_svc *svc;
|
|
|
+
|
|
|
+ svc = peer_svc_find(peer, svc_start_handle, NULL);
|
|
|
+ if (svc == NULL) {
|
|
|
+ /* Can't find service for discovered characteristic; this shouldn't
|
|
|
+ * happen.
|
|
|
+ */
|
|
|
+ assert(0);
|
|
|
+ return BLE_HS_EUNKNOWN;
|
|
|
+ }
|
|
|
+
|
|
|
+ chr = peer_chr_find(svc, gatt_chr->def_handle, &prev);
|
|
|
+ if (chr != NULL) {
|
|
|
+ /* Characteristic already discovered. */
|
|
|
+ return 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ chr = os_memblock_get(&peer_chr_pool);
|
|
|
+ if (chr == NULL) {
|
|
|
+ /* Out of memory. */
|
|
|
+ return BLE_HS_ENOMEM;
|
|
|
+ }
|
|
|
+ memset(chr, 0, sizeof * chr);
|
|
|
+
|
|
|
+ chr->chr = *gatt_chr;
|
|
|
+
|
|
|
+ if (prev == NULL) {
|
|
|
+ SLIST_INSERT_HEAD(&svc->chrs, chr, next);
|
|
|
+ } else {
|
|
|
+ SLIST_NEXT(prev, next) = chr;
|
|
|
+ }
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|
|
|
+
|
|
|
+static int
|
|
|
+peer_chr_disced(uint16_t conn_handle, const struct ble_gatt_error *error,
|
|
|
+ const struct ble_gatt_chr *chr, void *arg)
|
|
|
+{
|
|
|
+ struct peer *peer;
|
|
|
+ int rc;
|
|
|
+
|
|
|
+ peer = arg;
|
|
|
+ assert(peer->conn_handle == conn_handle);
|
|
|
+
|
|
|
+ switch (error->status) {
|
|
|
+ case 0:
|
|
|
+ rc = peer_chr_add(peer, peer->cur_svc->svc.start_handle, chr);
|
|
|
+ break;
|
|
|
+
|
|
|
+ case BLE_HS_EDONE:
|
|
|
+ /* All characteristics in this service discovered; start discovering
|
|
|
+ * characteristics in the next service.
|
|
|
+ */
|
|
|
+ if (peer->disc_prev_chr_val > 0) {
|
|
|
+ peer_disc_chrs(peer);
|
|
|
+ }
|
|
|
+ rc = 0;
|
|
|
+ break;
|
|
|
+
|
|
|
+ default:
|
|
|
+ rc = error->status;
|
|
|
+ break;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (rc != 0) {
|
|
|
+ /* Error; abort discovery. */
|
|
|
+ peer_disc_complete(peer, rc);
|
|
|
+ }
|
|
|
+
|
|
|
+ return rc;
|
|
|
+}
|
|
|
+
|
|
|
+static void
|
|
|
+peer_disc_chrs(struct peer *peer)
|
|
|
+{
|
|
|
+ struct peer_svc *svc;
|
|
|
+ int rc;
|
|
|
+
|
|
|
+ /* Search through the list of discovered service for the first service that
|
|
|
+ * contains undiscovered characteristics. Then, discover all
|
|
|
+ * characteristics belonging to that service.
|
|
|
+ */
|
|
|
+ SLIST_FOREACH(svc, &peer->svcs, next) {
|
|
|
+ if (!peer_svc_is_empty(svc) && SLIST_EMPTY(&svc->chrs)) {
|
|
|
+ peer->cur_svc = svc;
|
|
|
+ rc = ble_gattc_disc_all_chrs(peer->conn_handle,
|
|
|
+ svc->svc.start_handle,
|
|
|
+ svc->svc.end_handle,
|
|
|
+ peer_chr_disced, peer);
|
|
|
+ if (rc != 0) {
|
|
|
+ peer_disc_complete(peer, rc);
|
|
|
+ }
|
|
|
+ return;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /* All characteristics discovered. */
|
|
|
+ peer_disc_dscs(peer);
|
|
|
+}
|
|
|
+
|
|
|
+int
|
|
|
+peer_svc_is_empty(const struct peer_svc *svc)
|
|
|
+{
|
|
|
+ return svc->svc.end_handle <= svc->svc.start_handle;
|
|
|
+}
|
|
|
+
|
|
|
+static struct peer_svc *
|
|
|
+peer_svc_find_prev(struct peer *peer, uint16_t svc_start_handle)
|
|
|
+{
|
|
|
+ struct peer_svc *prev;
|
|
|
+ struct peer_svc *svc;
|
|
|
+
|
|
|
+ prev = NULL;
|
|
|
+ SLIST_FOREACH(svc, &peer->svcs, next) {
|
|
|
+ if (svc->svc.start_handle >= svc_start_handle) {
|
|
|
+ break;
|
|
|
+ }
|
|
|
+
|
|
|
+ prev = svc;
|
|
|
+ }
|
|
|
+
|
|
|
+ return prev;
|
|
|
+}
|
|
|
+
|
|
|
+static struct peer_svc *
|
|
|
+peer_svc_find(struct peer *peer, uint16_t svc_start_handle,
|
|
|
+ struct peer_svc **out_prev)
|
|
|
+{
|
|
|
+ struct peer_svc *prev;
|
|
|
+ struct peer_svc *svc;
|
|
|
+
|
|
|
+ prev = peer_svc_find_prev(peer, svc_start_handle);
|
|
|
+ if (prev == NULL) {
|
|
|
+ svc = SLIST_FIRST(&peer->svcs);
|
|
|
+ } else {
|
|
|
+ svc = SLIST_NEXT(prev, next);
|
|
|
+ }
|
|
|
+
|
|
|
+ if (svc != NULL && svc->svc.start_handle != svc_start_handle) {
|
|
|
+ svc = NULL;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (out_prev != NULL) {
|
|
|
+ *out_prev = prev;
|
|
|
+ }
|
|
|
+ return svc;
|
|
|
+}
|
|
|
+
|
|
|
+static struct peer_svc *
|
|
|
+peer_svc_find_range(struct peer *peer, uint16_t attr_handle)
|
|
|
+{
|
|
|
+ struct peer_svc *svc;
|
|
|
+
|
|
|
+ SLIST_FOREACH(svc, &peer->svcs, next) {
|
|
|
+ if (svc->svc.start_handle <= attr_handle &&
|
|
|
+ svc->svc.end_handle >= attr_handle) {
|
|
|
+
|
|
|
+ return svc;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return NULL;
|
|
|
+}
|
|
|
+
|
|
|
+const struct peer_svc *
|
|
|
+peer_svc_find_uuid(const struct peer *peer, const ble_uuid_t *uuid)
|
|
|
+{
|
|
|
+ const struct peer_svc *svc;
|
|
|
+
|
|
|
+ SLIST_FOREACH(svc, &peer->svcs, next) {
|
|
|
+ if (ble_uuid_cmp(&svc->svc.uuid.u, uuid) == 0) {
|
|
|
+ return svc;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return NULL;
|
|
|
+}
|
|
|
+
|
|
|
+const struct peer_chr *
|
|
|
+peer_chr_find_uuid(const struct peer *peer, const ble_uuid_t *svc_uuid,
|
|
|
+ const ble_uuid_t *chr_uuid)
|
|
|
+{
|
|
|
+ const struct peer_svc *svc;
|
|
|
+ const struct peer_chr *chr;
|
|
|
+
|
|
|
+ svc = peer_svc_find_uuid(peer, svc_uuid);
|
|
|
+ if (svc == NULL) {
|
|
|
+ return NULL;
|
|
|
+ }
|
|
|
+
|
|
|
+ SLIST_FOREACH(chr, &svc->chrs, next) {
|
|
|
+ if (ble_uuid_cmp(&chr->chr.uuid.u, chr_uuid) == 0) {
|
|
|
+ return chr;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return NULL;
|
|
|
+}
|
|
|
+
|
|
|
+const struct peer_dsc *
|
|
|
+peer_dsc_find_uuid(const struct peer *peer, const ble_uuid_t *svc_uuid,
|
|
|
+ const ble_uuid_t *chr_uuid, const ble_uuid_t *dsc_uuid)
|
|
|
+{
|
|
|
+ const struct peer_chr *chr;
|
|
|
+ const struct peer_dsc *dsc;
|
|
|
+
|
|
|
+ chr = peer_chr_find_uuid(peer, svc_uuid, chr_uuid);
|
|
|
+ if (chr == NULL) {
|
|
|
+ return NULL;
|
|
|
+ }
|
|
|
+
|
|
|
+ SLIST_FOREACH(dsc, &chr->dscs, next) {
|
|
|
+ if (ble_uuid_cmp(&dsc->dsc.uuid.u, dsc_uuid) == 0) {
|
|
|
+ return dsc;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return NULL;
|
|
|
+}
|
|
|
+
|
|
|
+static int
|
|
|
+peer_svc_add(struct peer *peer, const struct ble_gatt_svc *gatt_svc)
|
|
|
+{
|
|
|
+ struct peer_svc *prev;
|
|
|
+ struct peer_svc *svc;
|
|
|
+
|
|
|
+ svc = peer_svc_find(peer, gatt_svc->start_handle, &prev);
|
|
|
+ if (svc != NULL) {
|
|
|
+ /* Service already discovered. */
|
|
|
+ return 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ svc = os_memblock_get(&peer_svc_pool);
|
|
|
+ if (svc == NULL) {
|
|
|
+ /* Out of memory. */
|
|
|
+ return BLE_HS_ENOMEM;
|
|
|
+ }
|
|
|
+ memset(svc, 0, sizeof * svc);
|
|
|
+
|
|
|
+ svc->svc = *gatt_svc;
|
|
|
+ SLIST_INIT(&svc->chrs);
|
|
|
+
|
|
|
+ if (prev == NULL) {
|
|
|
+ SLIST_INSERT_HEAD(&peer->svcs, svc, next);
|
|
|
+ } else {
|
|
|
+ SLIST_INSERT_AFTER(prev, svc, next);
|
|
|
+ }
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|
|
|
+
|
|
|
+static void
|
|
|
+peer_svc_delete(struct peer_svc *svc)
|
|
|
+{
|
|
|
+ struct peer_chr *chr;
|
|
|
+
|
|
|
+ while ((chr = SLIST_FIRST(&svc->chrs)) != NULL) {
|
|
|
+ SLIST_REMOVE_HEAD(&svc->chrs, next);
|
|
|
+ peer_chr_delete(chr);
|
|
|
+ }
|
|
|
+
|
|
|
+ os_memblock_put(&peer_svc_pool, svc);
|
|
|
+}
|
|
|
+
|
|
|
+static int
|
|
|
+peer_svc_disced(uint16_t conn_handle, const struct ble_gatt_error *error,
|
|
|
+ const struct ble_gatt_svc *service, void *arg)
|
|
|
+{
|
|
|
+ struct peer *peer;
|
|
|
+ int rc;
|
|
|
+
|
|
|
+ peer = arg;
|
|
|
+ assert(peer->conn_handle == conn_handle);
|
|
|
+
|
|
|
+ switch (error->status) {
|
|
|
+ case 0:
|
|
|
+ rc = peer_svc_add(peer, service);
|
|
|
+ break;
|
|
|
+
|
|
|
+ case BLE_HS_EDONE:
|
|
|
+ /* All services discovered; start discovering characteristics. */
|
|
|
+ if (peer->disc_prev_chr_val > 0) {
|
|
|
+ peer_disc_chrs(peer);
|
|
|
+ }
|
|
|
+ rc = 0;
|
|
|
+ break;
|
|
|
+
|
|
|
+ default:
|
|
|
+ rc = error->status;
|
|
|
+ break;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (rc != 0) {
|
|
|
+ /* Error; abort discovery. */
|
|
|
+ peer_disc_complete(peer, rc);
|
|
|
+ }
|
|
|
+
|
|
|
+ return rc;
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+int
|
|
|
+peer_disc_all(uint16_t conn_handle, peer_disc_fn *disc_cb, void *disc_cb_arg)
|
|
|
+{
|
|
|
+ struct peer_svc *svc;
|
|
|
+ struct peer *peer;
|
|
|
+ int rc;
|
|
|
+
|
|
|
+ peer = peer_find(conn_handle);
|
|
|
+ if (peer == NULL) {
|
|
|
+ return BLE_HS_ENOTCONN;
|
|
|
+ }
|
|
|
+
|
|
|
+ /* Undiscover everything first. */
|
|
|
+ while ((svc = SLIST_FIRST(&peer->svcs)) != NULL) {
|
|
|
+ SLIST_REMOVE_HEAD(&peer->svcs, next);
|
|
|
+ peer_svc_delete(svc);
|
|
|
+ }
|
|
|
+
|
|
|
+ peer->disc_prev_chr_val = 1;
|
|
|
+ peer->disc_cb = disc_cb;
|
|
|
+ peer->disc_cb_arg = disc_cb_arg;
|
|
|
+
|
|
|
+ rc = ble_gattc_disc_all_svcs(conn_handle, peer_svc_disced, peer);
|
|
|
+ if (rc != 0) {
|
|
|
+ return rc;
|
|
|
+ }
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|
|
|
+
|
|
|
+int
|
|
|
+peer_delete(uint16_t conn_handle)
|
|
|
+{
|
|
|
+ struct peer_svc *svc;
|
|
|
+ struct peer *peer;
|
|
|
+ int rc;
|
|
|
+
|
|
|
+ peer = peer_find(conn_handle);
|
|
|
+ if (peer == NULL) {
|
|
|
+ return BLE_HS_ENOTCONN;
|
|
|
+ }
|
|
|
+
|
|
|
+ SLIST_REMOVE(&peers, peer, peer, next);
|
|
|
+
|
|
|
+ while ((svc = SLIST_FIRST(&peer->svcs)) != NULL) {
|
|
|
+ SLIST_REMOVE_HEAD(&peer->svcs, next);
|
|
|
+ peer_svc_delete(svc);
|
|
|
+ }
|
|
|
+
|
|
|
+ rc = os_memblock_put(&peer_pool, peer);
|
|
|
+ if (rc != 0) {
|
|
|
+ return BLE_HS_EOS;
|
|
|
+ }
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|
|
|
+
|
|
|
+int
|
|
|
+peer_add(uint16_t conn_handle)
|
|
|
+{
|
|
|
+ struct peer *peer;
|
|
|
+
|
|
|
+ /* Make sure the connection handle is unique. */
|
|
|
+ peer = peer_find(conn_handle);
|
|
|
+ if (peer != NULL) {
|
|
|
+ return BLE_HS_EALREADY;
|
|
|
+ }
|
|
|
+
|
|
|
+ peer = os_memblock_get(&peer_pool);
|
|
|
+ if (peer == NULL) {
|
|
|
+ /* Out of memory. */
|
|
|
+ return BLE_HS_ENOMEM;
|
|
|
+ }
|
|
|
+
|
|
|
+ memset(peer, 0, sizeof * peer);
|
|
|
+ peer->conn_handle = conn_handle;
|
|
|
+
|
|
|
+ SLIST_INSERT_HEAD(&peers, peer, next);
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|
|
|
+
|
|
|
+static void
|
|
|
+peer_free_mem(void)
|
|
|
+{
|
|
|
+ free(peer_mem);
|
|
|
+ peer_mem = NULL;
|
|
|
+
|
|
|
+ free(peer_svc_mem);
|
|
|
+ peer_svc_mem = NULL;
|
|
|
+
|
|
|
+ free(peer_chr_mem);
|
|
|
+ peer_chr_mem = NULL;
|
|
|
+
|
|
|
+ free(peer_dsc_mem);
|
|
|
+ peer_dsc_mem = NULL;
|
|
|
+}
|
|
|
+
|
|
|
+int
|
|
|
+peer_init(int max_peers, int max_svcs, int max_chrs, int max_dscs)
|
|
|
+{
|
|
|
+ int rc;
|
|
|
+
|
|
|
+ /* Free memory first in case this function gets called more than once. */
|
|
|
+ peer_free_mem();
|
|
|
+
|
|
|
+ peer_mem = malloc(
|
|
|
+ OS_MEMPOOL_BYTES(max_peers, sizeof (struct peer)));
|
|
|
+ if (peer_mem == NULL) {
|
|
|
+ rc = BLE_HS_ENOMEM;
|
|
|
+ goto err;
|
|
|
+ }
|
|
|
+
|
|
|
+ rc = os_mempool_init(&peer_pool, max_peers,
|
|
|
+ sizeof (struct peer), peer_mem,
|
|
|
+ "peer_pool");
|
|
|
+ if (rc != 0) {
|
|
|
+ rc = BLE_HS_EOS;
|
|
|
+ goto err;
|
|
|
+ }
|
|
|
+
|
|
|
+ peer_svc_mem = malloc(
|
|
|
+ OS_MEMPOOL_BYTES(max_svcs, sizeof (struct peer_svc)));
|
|
|
+ if (peer_svc_mem == NULL) {
|
|
|
+ rc = BLE_HS_ENOMEM;
|
|
|
+ goto err;
|
|
|
+ }
|
|
|
+
|
|
|
+ rc = os_mempool_init(&peer_svc_pool, max_svcs,
|
|
|
+ sizeof (struct peer_svc), peer_svc_mem,
|
|
|
+ "peer_svc_pool");
|
|
|
+ if (rc != 0) {
|
|
|
+ rc = BLE_HS_EOS;
|
|
|
+ goto err;
|
|
|
+ }
|
|
|
+
|
|
|
+ peer_chr_mem = malloc(
|
|
|
+ OS_MEMPOOL_BYTES(max_chrs, sizeof (struct peer_chr)));
|
|
|
+ if (peer_chr_mem == NULL) {
|
|
|
+ rc = BLE_HS_ENOMEM;
|
|
|
+ goto err;
|
|
|
+ }
|
|
|
+
|
|
|
+ rc = os_mempool_init(&peer_chr_pool, max_chrs,
|
|
|
+ sizeof (struct peer_chr), peer_chr_mem,
|
|
|
+ "peer_chr_pool");
|
|
|
+ if (rc != 0) {
|
|
|
+ rc = BLE_HS_EOS;
|
|
|
+ goto err;
|
|
|
+ }
|
|
|
+
|
|
|
+ peer_dsc_mem = malloc(
|
|
|
+ OS_MEMPOOL_BYTES(max_dscs, sizeof (struct peer_dsc)));
|
|
|
+ if (peer_dsc_mem == NULL) {
|
|
|
+ rc = BLE_HS_ENOMEM;
|
|
|
+ goto err;
|
|
|
+ }
|
|
|
+
|
|
|
+ rc = os_mempool_init(&peer_dsc_pool, max_dscs,
|
|
|
+ sizeof (struct peer_dsc), peer_dsc_mem,
|
|
|
+ "peer_dsc_pool");
|
|
|
+ if (rc != 0) {
|
|
|
+ rc = BLE_HS_EOS;
|
|
|
+ goto err;
|
|
|
+ }
|
|
|
+
|
|
|
+ return 0;
|
|
|
+
|
|
|
+err:
|
|
|
+ peer_free_mem();
|
|
|
+ return rc;
|
|
|
+}
|