Bluetooth: BAP: Add bt_bap_ep_get_conn

Add a new function, bt_bap_ep_get_conn, which returns
the ACL connection for the endpoint. This works because
endpoints are specific to an ACL in BAP. The function
returns a pointer with a new reference similar to the
lookup functions from conn.h

The conn pointer was not added to the bt_bap_ep_info
struct, as doing so would be more likely to cause
reference leaks if the caller did not care about the
conn pointer when using bt_bap_ep_get_info. It can be
added later if that is requested.

Signed-off-by: Emil Gydesen <[email protected]>

Author: Thalley

include/zephyr/bluetooth/audio/bap.h

@@ -882,6 +882,21 @@ struct bt_bap_ep_info {
  */
 int bt_bap_ep_get_info(const struct bt_bap_ep *ep, struct bt_bap_ep_info *info);
 
+/**
+ * @brief Get the pointer to the ACL connection of an endpoint
+ *
+ * The caller gets a new reference to the connection object, if not NULL, which must be
+ * released with bt_conn_unref() once done using the object.
+ *
+ * @param ep The endpoint to get the ACL connection of
+ *
+ * @return The ACL connection pointer.
+ *         Will always be NULL for broadcast endpoints.
+ *         Will be NULL for Unicast Server endpoints if the endpoint is not configured by a client.
+ *         Will be NULL for Unicast Client endpoints if @p does not match a discovered endpoint.
+ */
+struct bt_conn *bt_bap_ep_get_conn(const struct bt_bap_ep *ep);
+
 /**
  * @brief Basic Audio Profile stream structure.
  *

subsys/bluetooth/audio/ascs.c

@@ -3263,4 +3263,17 @@ bool bt_ascs_has_ep(const struct bt_bap_ep *ep)
 {
 	return PART_OF_ARRAY(ascs.ase_pool, ep);
 }
+
+struct bt_conn *bt_ascs_ep_get_conn(const struct bt_bap_ep *ep)
+{
+	struct bt_ascs_ase *ase = CONTAINER_OF(ep, struct bt_ascs_ase, ep);
+
+	__ASSERT_NO_MSG(bt_ascs_has_ep(ep));
+
+	if (ase->conn == NULL) {
+		return NULL;
+	}
+
+	return bt_conn_ref(ase->conn);
+}
 #endif /* BT_BAP_UNICAST_SERVER */

subsys/bluetooth/audio/ascs_internal.h

@@ -365,5 +365,6 @@ void bt_ascs_foreach_ep(struct bt_conn *conn, bt_bap_ep_func_t func, void *user_
 int bt_ascs_register(uint8_t snk_cnt, uint8_t src_cnt);
 int bt_ascs_unregister(void);
 bool bt_ascs_has_ep(const struct bt_bap_ep *ep);
+struct bt_conn *bt_ascs_ep_get_conn(const struct bt_bap_ep *ep);
 
 #endif /* BT_ASCS_INTERNAL_H */

subsys/bluetooth/audio/bap_internal.h

@@ -147,3 +147,5 @@ bool bt_bap_broadcast_sink_has_ep(const struct bt_bap_ep *ep);
 bool bt_bap_broadcast_source_has_ep(const struct bt_bap_ep *ep);
 bool bt_bap_unicast_client_has_ep(const struct bt_bap_ep *ep);
 bool bt_bap_unicast_server_has_ep(const struct bt_bap_ep *ep);
+struct bt_conn *bt_bap_unicast_client_ep_get_conn(const struct bt_bap_ep *ep);
+struct bt_conn *bt_bap_unicast_server_ep_get_conn(const struct bt_bap_ep *ep);

subsys/bluetooth/audio/bap_stream.c

@@ -173,6 +173,31 @@ int bt_bap_ep_get_info(const struct bt_bap_ep *ep, struct bt_bap_ep_info *info)
 	return 0;
 }
 
+struct bt_conn *bt_bap_ep_get_conn(const struct bt_bap_ep *ep)
+{
+	struct bt_conn *conn;
+
+	if (ep == NULL) {
+		LOG_DBG("ep is NULL");
+
+		return NULL;
+	}
+
+	if ((IS_ENABLED(CONFIG_BT_BAP_BROADCAST_SOURCE) && bt_bap_broadcast_source_has_ep(ep)) ||
+	    (IS_ENABLED(CONFIG_BT_BAP_BROADCAST_SINK) && bt_bap_broadcast_sink_has_ep(ep))) {
+		conn = NULL;
+	} else if (IS_ENABLED(CONFIG_BT_BAP_UNICAST_CLIENT) && bt_bap_unicast_client_has_ep(ep)) {
+		conn = bt_bap_unicast_client_ep_get_conn(ep);
+	} else if (IS_ENABLED(CONFIG_BT_BAP_UNICAST_SERVER) && bt_bap_unicast_server_has_ep(ep)) {
+		conn = bt_bap_unicast_server_ep_get_conn(ep);
+	} else {
+		LOG_DBG("Invalid endpoint %p", ep);
+		conn = NULL;
+	}
+
+	return conn;
+}
+
 enum bt_bap_ascs_reason bt_audio_verify_qos(const struct bt_bap_qos_cfg *qos)
 {
 	if (qos->interval < BT_ISO_SDU_INTERVAL_MIN ||

subsys/bluetooth/audio/bap_unicast_client.c

@@ -460,6 +460,41 @@ bool bt_bap_unicast_client_has_ep(const struct bt_bap_ep *ep)
 	return false;
 }
 
+struct bt_conn *bt_bap_unicast_client_ep_get_conn(const struct bt_bap_ep *ep)
+{
+	for (size_t i = 0U; i < ARRAY_SIZE(uni_cli_insts); i++) {
+#if CONFIG_BT_BAP_UNICAST_CLIENT_ASE_SNK_COUNT > 0
+		if (PART_OF_ARRAY(uni_cli_insts[i].snks, ep)) {
+			ARRAY_FOR_EACH_PTR(uni_cli_insts[i].snks, client_ep) {
+				if (&client_ep->ep == ep) {
+					if (client_ep->handle == BAP_HANDLE_UNUSED) {
+						return NULL;
+					}
+
+					return bt_conn_lookup_index((uint8_t)i);
+				}
+			}
+		}
+#endif /* CONFIG_BT_BAP_UNICAST_CLIENT_ASE_SNK_COUNT > 0 */
+
+#if CONFIG_BT_BAP_UNICAST_CLIENT_ASE_SRC_COUNT > 0
+		if (PART_OF_ARRAY(uni_cli_insts[i].srcs, ep)) {
+			ARRAY_FOR_EACH_PTR(uni_cli_insts[i].srcs, client_ep) {
+				if (&client_ep->ep == ep) {
+					if (client_ep->handle == BAP_HANDLE_UNUSED) {
+						return NULL;
+					}
+
+					return bt_conn_lookup_index((uint8_t)i);
+				}
+			}
+		}
+#endif /* CONFIG_BT_BAP_UNICAST_CLIENT_ASE_SRC_COUNT > 0 */
+	}
+
+	return NULL;
+}
+
 static void unicast_client_ep_init(struct bt_bap_ep *ep, uint16_t handle, uint8_t dir)
 {
 	struct bt_bap_unicast_client_ep *client_ep;

subsys/bluetooth/audio/bap_unicast_server.c

@@ -227,3 +227,8 @@ bool bt_bap_unicast_server_has_ep(const struct bt_bap_ep *ep)
 {
 	return bt_ascs_has_ep(ep);
 }
+
+struct bt_conn *bt_bap_unicast_server_ep_get_conn(const struct bt_bap_ep *ep)
+{
+	return bt_ascs_ep_get_conn(ep);
+}