mirror of
https://github.com/espressif/esp-idf.git
synced 2024-10-05 20:47:46 -04:00
c86c83ca4d
- Added blecent_throughput(client) and bleprph_throughput(server) examples to demonstrate application throughput.
828 lines
30 KiB
C
828 lines
30 KiB
C
/*
|
|
* Copyright 2020 Espressif Systems (Shanghai) PTE LTD
|
|
*
|
|
* Licensed to the Apache Software Foundation (ASF) under one
|
|
* or more contributor license agreements. See the NOTICE file
|
|
* distributed with this work for additional information
|
|
* regarding copyright ownership. The ASF licenses this file
|
|
* to you under the Apache License, Version 2.0 (the
|
|
* "License"); you may not use this file except in compliance
|
|
* with the License. You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing,
|
|
* software distributed under the License is distributed on an
|
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
* KIND, either express or implied. See the License for the
|
|
* specific language governing permissions and limitations
|
|
* under the License.
|
|
*/
|
|
|
|
#include "esp_log.h"
|
|
#include "nvs_flash.h"
|
|
/* BLE */
|
|
#include "esp_nimble_hci.h"
|
|
#include "nimble/nimble_port.h"
|
|
#include "nimble/nimble_port_freertos.h"
|
|
#include "host/ble_hs.h"
|
|
#include "host/util/util.h"
|
|
#include "console/console.h"
|
|
#include "services/gap/ble_svc_gap.h"
|
|
#include "gattc.h"
|
|
#include "esp_timer.h"
|
|
#include "linenoise/linenoise.h"
|
|
#include "esp_console.h"
|
|
#include "driver/uart.h"
|
|
#include "esp_vfs_dev.h"
|
|
#include "argtable3/argtable3.h"
|
|
#include "cmd_system.h"
|
|
#include "../src/ble_hs_hci_priv.h"
|
|
|
|
/* 0000xxxx-8c26-476f-89a7-a108033a69c7 */
|
|
#define THRPT_UUID_DECLARE(uuid16) \
|
|
((const ble_uuid_t *) (&(ble_uuid128_t) BLE_UUID128_INIT( \
|
|
0xc7, 0x69, 0x3a, 0x03, 0x08, 0xa1, 0xa7, 0x89, \
|
|
0x6f, 0x47, 0x26, 0x8c, uuid16, uuid16 >> 8, 0x00, 0x00 \
|
|
)))
|
|
|
|
/* 0000xxxx-8c26-476f-89a7-a108033a69c6 */
|
|
#define THRPT_UUID_DECLARE_ALT(uuid16) \
|
|
((const ble_uuid_t *) (&(ble_uuid128_t) BLE_UUID128_INIT( \
|
|
0xc6, 0x69, 0x3a, 0x03, 0x08, 0xa1, 0xa7, 0x89, \
|
|
0x6f, 0x47, 0x26, 0x8c, uuid16, uuid16 >> 8, 0x00, 0x00 \
|
|
)))
|
|
|
|
#define THRPT_SVC 0x0001
|
|
#define THRPT_CHR_READ_WRITE 0x0006
|
|
#define THRPT_CHR_NOTIFY 0x000a
|
|
#define THRPT_LONG_CHR_READ_WRITE 0x000b
|
|
#define THRPT_LONG_CHR_READ_WRITE_ALT 0x001a
|
|
#define THRPT_CHR_READ_WRITE_ALT 0x001f
|
|
|
|
/* Throughput cases */
|
|
#define READ_THROUGHPUT 1
|
|
#define WRITE_THROUGHPUT 2
|
|
#define NOTIFY_THROUGHPUT 3
|
|
|
|
#define READ_THROUGHPUT_PAYLOAD 500
|
|
#define WRITE_THROUGHPUT_PAYLOAD 500
|
|
#define LL_PACKET_TIME 2120
|
|
#define LL_PACKET_LENGTH 251
|
|
static const char *tag = "blecent_throughput";
|
|
static int blecent_gap_event(struct ble_gap_event *event, void *arg);
|
|
static uint8_t peer_addr[6];
|
|
static SemaphoreHandle_t xSemaphore;
|
|
static int mbuf_len_total;
|
|
static int failure_count;
|
|
static int conn_params_def[] = {40, 40, 0, 500, 80, 80};
|
|
static struct ble_gap_upd_params conn_params = {
|
|
/** Minimum value for connection interval in 1.25ms units */
|
|
.itvl_min = CONFIG_EXAMPLE_CONN_ITVL_MIN,
|
|
/** Maximum value for connection interval in 1.25ms units */
|
|
.itvl_max = CONFIG_EXAMPLE_CONN_ITVL_MAX,
|
|
/** Connection latency */
|
|
.latency = CONFIG_EXAMPLE_CONN_LATENCY,
|
|
/** Supervision timeout in 10ms units */
|
|
.supervision_timeout = CONFIG_EXAMPLE_CONN_TIMEOUT,
|
|
/** Minimum length of connection event in 0.625ms units */
|
|
.min_ce_len = CONFIG_EXAMPLE_CONN_CE_LEN_MIN,
|
|
/** Maximum length of connection event in 0.625ms units */
|
|
.max_ce_len = CONFIG_EXAMPLE_CONN_CE_LEN_MAX,
|
|
};
|
|
|
|
static int mtu_def = 512;
|
|
/* test_data accepts test_name and test_time from CLI */
|
|
static int test_data[] = {1, 600};
|
|
void ble_store_config_init(void);
|
|
|
|
static int
|
|
blecent_notify(uint16_t conn_handle, uint16_t val_handle,
|
|
ble_gatt_attr_fn *cb, struct peer *peer, int test_time)
|
|
{
|
|
uint8_t value[2] = {1, 0};/*To subscribe to notifications*/
|
|
int rc;
|
|
|
|
rc = ble_gattc_write_flat(conn_handle, val_handle,
|
|
value, sizeof value, NULL, &test_time);
|
|
if (rc != 0) {
|
|
ESP_LOGE(tag, "Error: Failed to subscribe to characteristic; "
|
|
"rc = %d", rc);
|
|
goto err;
|
|
}
|
|
|
|
return 0;
|
|
err:
|
|
/* Terminate the connection. */
|
|
return ble_gap_terminate(peer->conn_handle, BLE_ERR_REM_USER_CONN_TERM);
|
|
}
|
|
|
|
static int
|
|
blecent_repeat_write(uint16_t conn_handle,
|
|
const struct ble_gatt_error *error,
|
|
struct ble_gatt_attr *attr,
|
|
void *arg)
|
|
{
|
|
if (error->status == 0) {
|
|
xSemaphoreGive(xSemaphore);
|
|
ESP_LOGD(tag, " attr_handle=%d value=", attr->handle);
|
|
MODLOG_DFLT(INFO, "\n");
|
|
} else {
|
|
failure_count++;
|
|
xSemaphoreGive(xSemaphore);
|
|
ESP_LOGE(tag, " Error writing error code = %d", error->status);
|
|
}
|
|
return error->status;
|
|
}
|
|
|
|
static int blecent_write(uint16_t conn_handle, uint16_t val_handle,
|
|
ble_gatt_attr_fn *cb, struct peer *peer, int test_time)
|
|
{
|
|
int64_t start_time, end_time, write_time = 0;
|
|
int write_count = 0;
|
|
uint8_t value[WRITE_THROUGHPUT_PAYLOAD] = {0};
|
|
int rc;
|
|
|
|
value[0] = rand();
|
|
failure_count = 0;
|
|
start_time = esp_timer_get_time();
|
|
|
|
while (write_time < test_time * 1000) {
|
|
/* Wait till the previous write is complete. For first time Semaphore
|
|
* is already available */
|
|
xSemaphoreTake(xSemaphore, portMAX_DELAY);
|
|
rc = ble_gattc_write_flat(conn_handle, val_handle,
|
|
&value, sizeof value, blecent_repeat_write, NULL);
|
|
if (rc != 0) {
|
|
ESP_LOGE(tag, "Error: Failed to write characteristic; rc=%d\n",
|
|
rc);
|
|
goto err;
|
|
}
|
|
|
|
end_time = esp_timer_get_time();
|
|
write_time = (end_time - start_time) / 1000 ;
|
|
write_count += 1;
|
|
}
|
|
|
|
/* Each successful write is of WRITE_THROUGHPUT_PAYLOAD Bytes of
|
|
* application data. */
|
|
printf("\n****************************************************************\n");
|
|
ESP_LOGI(tag, "Application Write throughput = %d bps, write count = %d,"
|
|
"failure count = %d",
|
|
((write_count - failure_count) * 8 * WRITE_THROUGHPUT_PAYLOAD) / test_time,
|
|
write_count, failure_count);
|
|
printf("\n****************************************************************\n");
|
|
|
|
return 0;
|
|
err:
|
|
/* Terminate the connection. */
|
|
return ble_gap_terminate(peer->conn_handle, BLE_ERR_REM_USER_CONN_TERM);
|
|
}
|
|
|
|
static int
|
|
blecent_repeat_read(uint16_t conn_handle,
|
|
const struct ble_gatt_error *error,
|
|
struct ble_gatt_attr *attr,
|
|
void *arg)
|
|
{
|
|
if (error->status == 0) {
|
|
xSemaphoreGive(xSemaphore);
|
|
ESP_LOGD(tag, " attr_handle=%d value=", attr->handle);
|
|
mbuf_len_total += OS_MBUF_PKTLEN(attr->om);
|
|
} else {
|
|
ESP_LOGE(tag, " Read failed, callback error code = %d", error->status );
|
|
xSemaphoreGive(xSemaphore);
|
|
}
|
|
return error->status;
|
|
}
|
|
|
|
static int blecent_read(uint16_t conn_handle, uint16_t val_handle,
|
|
ble_gatt_attr_fn *cb, struct peer *peer, int test_time)
|
|
{
|
|
int rc, read_count = 0;
|
|
int64_t start_time, end_time, read_time = 0;
|
|
/* Keep track of number of bytes read from char */
|
|
mbuf_len_total = 0;
|
|
start_time = esp_timer_get_time();
|
|
|
|
ESP_LOGD(tag, " Throughput read started :val_handle=%d test_time=%d", val_handle,
|
|
test_time);
|
|
|
|
while (read_time < (test_time * 1000)) {
|
|
/* Wait till the previous read is complete. For first time use Semaphore
|
|
* is already available */
|
|
xSemaphoreTake(xSemaphore, portMAX_DELAY);
|
|
|
|
rc = ble_gattc_read(peer->conn_handle, val_handle,
|
|
blecent_repeat_read, (void *) &peer);
|
|
if (rc != 0) {
|
|
ESP_LOGE(tag, "Error: Failed to read characteristic; rc=%d",
|
|
rc);
|
|
goto err;
|
|
}
|
|
|
|
end_time = esp_timer_get_time();
|
|
read_time = (end_time - start_time) / 1000 ;
|
|
read_count += 1;
|
|
}
|
|
|
|
/* Application data throughput */
|
|
printf("\n****************************************************************\n");
|
|
ESP_LOGI(tag, "Application Read throughput = %d bps, Read op counter = %d",
|
|
(mbuf_len_total * 8) / (test_time), read_count);
|
|
printf("\n****************************************************************\n");
|
|
return 0;
|
|
|
|
err:
|
|
/* Terminate the connection. */
|
|
vTaskDelay(100 / portTICK_PERIOD_MS);
|
|
return ble_gap_terminate(peer->conn_handle, BLE_ERR_REM_USER_CONN_TERM);
|
|
}
|
|
|
|
static void throughput_task(void *arg)
|
|
{
|
|
struct peer *peer = (struct peer *)arg;
|
|
const struct peer_chr *chr;
|
|
const struct peer_dsc *dsc;
|
|
int rc = 0;
|
|
|
|
while (1) {
|
|
vTaskDelay(4000 / portTICK_PERIOD_MS);
|
|
ESP_LOGI(tag, "Format for throughput demo:: throughput read 100");
|
|
printf(" ==================================================================\n");
|
|
printf(" | Steps to test nimble throughput |\n");
|
|
printf(" | |\n");
|
|
printf(" | 1. Enter throughput [--Type] [--Test time] |\n");
|
|
printf(" | Type: read/write/notify. |\n");
|
|
printf(" | Test time: Enter value in seconds. |\n");
|
|
printf(" | |\n");
|
|
printf(" | e.g. throughput read 600 |\n");
|
|
printf(" | |\n");
|
|
printf(" | ** Enter 'throughput read 60' for reading char for 60 seconds |\n");
|
|
printf(" | OR 'throughput write 60' for writing to char for 60 seconds |\n");
|
|
printf(" | OR 'throughput notify 60' for notifications (for 60 seconds)**|\n");
|
|
printf(" | |\n");
|
|
printf(" =================================================================\n\n");
|
|
/* XXX Delay ? */
|
|
vTaskDelay(1000 / portTICK_PERIOD_MS);
|
|
|
|
if (!cli_receive_key(test_data)) {
|
|
/* No command supplied, start with reading */
|
|
test_data[0] = READ_THROUGHPUT;
|
|
test_data[1] = 60;
|
|
ESP_LOGI(tag, "No command received from user, start with READ op"
|
|
" with 60 seconds test time");
|
|
}
|
|
scli_reset_queue();
|
|
|
|
switch (test_data[0]) {
|
|
|
|
case READ_THROUGHPUT:
|
|
/* Read the characteristic supporting long read support
|
|
* `THRPT_LONG_CHR_READ_WRITE` (0x000b) */
|
|
chr = peer_chr_find_uuid(peer,
|
|
THRPT_UUID_DECLARE(THRPT_SVC),
|
|
THRPT_UUID_DECLARE(THRPT_LONG_CHR_READ_WRITE));
|
|
if (chr == NULL) {
|
|
ESP_LOGE(tag, "Peer does not support "
|
|
"LONG_READ (0x000b) characteristic ");
|
|
break;
|
|
}
|
|
|
|
if (test_data[1] > 0) {
|
|
rc = blecent_read(peer->conn_handle, chr->chr.val_handle,
|
|
blecent_repeat_read, (void *) peer, test_data[1]);
|
|
if (rc != 0) {
|
|
ESP_LOGE(tag, "Error while reading from GATTS; rc = %d", rc);
|
|
}
|
|
} else {
|
|
ESP_LOGE(tag, "Please enter non-zero value for test time in seconds!!");
|
|
}
|
|
break;
|
|
|
|
case WRITE_THROUGHPUT:
|
|
chr = peer_chr_find_uuid(peer,
|
|
THRPT_UUID_DECLARE(THRPT_SVC),
|
|
THRPT_UUID_DECLARE(THRPT_CHR_READ_WRITE));
|
|
if (chr == NULL) {
|
|
ESP_LOGE(tag, "Error: Peer doesn't support the READ "
|
|
"WRITE characteristic (0x0006) ");
|
|
break;
|
|
}
|
|
|
|
if (test_data[1] > 0) {
|
|
rc = blecent_write(peer->conn_handle, chr->chr.val_handle,
|
|
blecent_repeat_write, (void *) peer, test_data[1]);
|
|
if (rc != 0) {
|
|
ESP_LOGE(tag, "Error while writing data; rc = %d", rc);
|
|
}
|
|
} else {
|
|
ESP_LOGE(tag, "Please enter non-zero value for test time in seconds!!");
|
|
}
|
|
break;
|
|
|
|
case NOTIFY_THROUGHPUT:
|
|
chr = peer_chr_find_uuid(peer,
|
|
THRPT_UUID_DECLARE(THRPT_SVC),
|
|
THRPT_UUID_DECLARE(THRPT_CHR_NOTIFY));
|
|
if (chr == NULL) {
|
|
ESP_LOGE(tag, "Error: Peer doesn't support the NOTIFY "
|
|
"characteristic (0x000a) ");
|
|
break;
|
|
}
|
|
dsc = peer_dsc_find_uuid(peer,
|
|
THRPT_UUID_DECLARE(THRPT_SVC),
|
|
THRPT_UUID_DECLARE(THRPT_CHR_NOTIFY),
|
|
BLE_UUID16_DECLARE(BLE_GATT_DSC_CLT_CFG_UUID16));
|
|
if (dsc == NULL) {
|
|
ESP_LOGE(tag, "Error: Peer lacks a CCCD for the Notify "
|
|
"Status characteristic\n");
|
|
break;
|
|
}
|
|
|
|
rc = blecent_notify(peer->conn_handle, dsc->dsc.handle,
|
|
NULL, (void *) peer, test_data[1]);
|
|
if (rc != 0) {
|
|
ESP_LOGE(tag, "Subscribing to notification failed; rc = %d ", rc);
|
|
} else {
|
|
ESP_LOGI(tag, "Subscribed to notifications. Throughput number"
|
|
" can be seen on peripheral terminal after %d seconds",
|
|
test_data[1]);
|
|
}
|
|
vTaskDelay(test_data[1]*1000 / portTICK_PERIOD_MS);
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
vTaskDelay(5000 / portTICK_PERIOD_MS);
|
|
}
|
|
vTaskDelete(NULL);
|
|
}
|
|
|
|
static void
|
|
blecent_read_write_subscribe(const struct peer *peer)
|
|
{
|
|
xTaskCreate(throughput_task, "throughput_task", 4096, (void *) peer, 10, NULL);
|
|
return;
|
|
}
|
|
|
|
/**
|
|
* Called when service discovery of the specified peer has completed.
|
|
*/
|
|
static void
|
|
blecent_on_disc_complete(const struct peer *peer, int status, void *arg)
|
|
{
|
|
|
|
if (status != 0) {
|
|
/* Service discovery failed. Terminate the connection. */
|
|
ESP_LOGE(tag, "Error: Service discovery failed; status=%d "
|
|
"conn_handle=%d\n", status, peer->conn_handle);
|
|
ble_gap_terminate(peer->conn_handle, BLE_ERR_REM_USER_CONN_TERM);
|
|
return;
|
|
}
|
|
|
|
/* Service discovery has completed successfully. Now we have a complete
|
|
* list of services, characteristics, and descriptors that the peer
|
|
* supports.
|
|
*/
|
|
ESP_LOGI(tag, "Service discovery complete; status=%d "
|
|
"conn_handle=%d\n", status, peer->conn_handle);
|
|
|
|
/* Now perform three GATT procedures against the peer: read,
|
|
* write, and subscribe to notifications depending upon user input.
|
|
*/
|
|
blecent_read_write_subscribe(peer);
|
|
}
|
|
|
|
/**
|
|
* Initiates the GAP general discovery procedure.
|
|
*/
|
|
static void
|
|
blecent_scan(void)
|
|
{
|
|
uint8_t own_addr_type;
|
|
struct ble_gap_disc_params disc_params;
|
|
int rc;
|
|
|
|
/* Figure out address to use while advertising (no privacy for now) */
|
|
rc = ble_hs_id_infer_auto(0, &own_addr_type);
|
|
if (rc != 0) {
|
|
ESP_LOGE(tag, "error determining address type; rc=%d", rc);
|
|
return;
|
|
}
|
|
|
|
/* Tell the controller to filter duplicates; we don't want to process
|
|
* repeated advertisements from the same device.
|
|
*/
|
|
disc_params.filter_duplicates = 1;
|
|
|
|
/**
|
|
* Perform a passive scan. I.e., don't send follow-up scan requests to
|
|
* each advertiser.
|
|
*/
|
|
disc_params.passive = 1;
|
|
|
|
/* Use defaults for the rest of the parameters. */
|
|
disc_params.itvl = 0;
|
|
disc_params.window = 0;
|
|
disc_params.filter_policy = 0;
|
|
disc_params.limited = 0;
|
|
|
|
rc = ble_gap_disc(own_addr_type, BLE_HS_FOREVER, &disc_params,
|
|
blecent_gap_event, NULL);
|
|
if (rc != 0) {
|
|
ESP_LOGE(tag, "Error initiating GAP discovery procedure; rc=%d\n",
|
|
rc);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Indicates whether we should try to connect to the sender of the specified
|
|
* advertisement. The function returns a positive result if the device
|
|
* advertises connectability and support for the THRPT service i.e. 0x0001.
|
|
*/
|
|
static int
|
|
blecent_should_connect(const struct ble_gap_disc_desc *disc)
|
|
{
|
|
struct ble_hs_adv_fields fields;
|
|
int rc;
|
|
int i;
|
|
|
|
rc = ble_hs_adv_parse_fields(&fields, disc->data, disc->length_data);
|
|
if (rc != 0) {
|
|
return rc;
|
|
}
|
|
|
|
if (strlen(CONFIG_EXAMPLE_PEER_ADDR) && (strncmp(CONFIG_EXAMPLE_PEER_ADDR, "ADDR_ANY", strlen("ADDR_ANY")) != 0)) {
|
|
ESP_LOGI(tag, "Peer address from menuconfig: %s", CONFIG_EXAMPLE_PEER_ADDR);
|
|
/* Convert string to address */
|
|
sscanf(CONFIG_EXAMPLE_PEER_ADDR, "%hhx:%hhx:%hhx:%hhx:%hhx:%hhx",
|
|
&peer_addr[5], &peer_addr[4], &peer_addr[3],
|
|
&peer_addr[2], &peer_addr[1], &peer_addr[0]);
|
|
if (memcmp(peer_addr, disc->addr.val, sizeof(disc->addr.val)) != 0) {
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
ESP_LOGI(tag, "connect; fields.num_uuids128 =%d", fields.num_uuids128);
|
|
for (i = 0; i < fields.num_uuids128; i++) {
|
|
if ((memcmp(&fields.uuids128[i], THRPT_UUID_DECLARE(THRPT_SVC),
|
|
sizeof(ble_uuid128_t))) == 0 ) {
|
|
ESP_LOGI(tag, "blecent_should_connect 'THRPT' success");
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
char serv_name[] = "nimble_prph";
|
|
if (fields.name != NULL) {
|
|
ESP_LOGI(tag, "Device Name = %s", (char *)fields.name);
|
|
|
|
if (memcmp(fields.name, serv_name, fields.name_len) == 0) {
|
|
ESP_LOGI(tag, "central connect to `nimble_prph` success");
|
|
return 1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* Connects to the sender of the specified advertisement of it looks
|
|
* interesting. A device is "interesting" if it advertises connectability and
|
|
* support for the Alert Notification service.
|
|
*/
|
|
static void
|
|
blecent_connect_if_interesting(const struct ble_gap_disc_desc *disc)
|
|
{
|
|
uint8_t own_addr_type;
|
|
int rc;
|
|
|
|
/* Don't do anything if we don't care about this advertiser. */
|
|
if (!blecent_should_connect(disc)) {
|
|
return;
|
|
}
|
|
|
|
/* Scanning must be stopped before a connection can be initiated. */
|
|
rc = ble_gap_disc_cancel();
|
|
if (rc != 0) {
|
|
MODLOG_DFLT(DEBUG, "Failed to cancel scan; rc=%d\n", rc);
|
|
return;
|
|
}
|
|
|
|
/* Figure out address to use for connect (no privacy for now) */
|
|
rc = ble_hs_id_infer_auto(0, &own_addr_type);
|
|
if (rc != 0) {
|
|
ESP_LOGE(tag, "error determining address type; rc=%d\n", rc);
|
|
return;
|
|
}
|
|
|
|
/* Try to connect the the advertiser. Allow 30 seconds (30000 ms) for
|
|
* timeout.
|
|
*/
|
|
rc = ble_gap_connect(own_addr_type, &disc->addr, 30000, NULL,
|
|
blecent_gap_event, NULL);
|
|
if (rc != 0) {
|
|
ESP_LOGE(tag, "Error: Failed to connect to device; addr_type=%d "
|
|
"addr=%s; rc=%d\n",
|
|
disc->addr.type, addr_str(disc->addr.val), rc);
|
|
return;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* The nimble host executes this callback when a GAP event occurs. The
|
|
* application associates a GAP event callback with each connection that is
|
|
* established. central uses the same callback for all connections.
|
|
*
|
|
* @param event The event being signalled.
|
|
* @param arg Application-specified argument; unused by
|
|
* gattc.
|
|
*
|
|
* @return 0 if the application successfully handled the
|
|
* event; nonzero on failure. The semantics
|
|
* of the return code is specific to the
|
|
* particular GAP event being signalled.
|
|
*/
|
|
static int
|
|
blecent_gap_event(struct ble_gap_event *event, void *arg)
|
|
{
|
|
struct ble_gap_conn_desc desc;
|
|
struct ble_hs_adv_fields fields;
|
|
int rc;
|
|
|
|
|
|
switch (event->type) {
|
|
case BLE_GAP_EVENT_DISC:
|
|
ESP_LOGI(tag, "Event DISC ");
|
|
rc = ble_hs_adv_parse_fields(&fields, event->disc.data,
|
|
event->disc.length_data);
|
|
if (rc != 0) {
|
|
return 0;
|
|
}
|
|
|
|
/* An advertisment report was received during GAP discovery. */
|
|
print_adv_fields(&fields);
|
|
|
|
/* Try to connect to the advertiser if it looks interesting. */
|
|
blecent_connect_if_interesting(&event->disc);
|
|
return 0;
|
|
|
|
case BLE_GAP_EVENT_CONNECT:
|
|
/* A new connection was established or a connection attempt failed. */
|
|
if (event->connect.status == 0) {
|
|
/* Connection successfully established. */
|
|
/* XXX Set packet length in controller for better throughput */
|
|
ESP_LOGI(tag, "Connection established ");
|
|
rc = ble_hs_hci_util_set_data_len(event->connect.conn_handle,
|
|
LL_PACKET_LENGTH, LL_PACKET_TIME);
|
|
if (rc != 0) {
|
|
ESP_LOGE(tag, "Set packet length failed; rc = %d", rc);
|
|
}
|
|
|
|
rc = ble_att_set_preferred_mtu(mtu_def);
|
|
if (rc != 0) {
|
|
ESP_LOGE(tag, "Failed to set preferred MTU; rc = %d", rc);
|
|
}
|
|
|
|
rc = ble_gattc_exchange_mtu(event->connect.conn_handle, NULL, NULL);
|
|
if (rc != 0) {
|
|
ESP_LOGE(tag, "Failed to negotiate MTU; rc = %d", rc);
|
|
}
|
|
|
|
rc = ble_gap_conn_find(event->connect.conn_handle, &desc);
|
|
assert(rc == 0);
|
|
print_conn_desc(&desc);
|
|
|
|
rc = ble_gap_update_params(event->connect.conn_handle, &conn_params);
|
|
if (rc != 0) {
|
|
ESP_LOGE(tag, "Failed to update params; rc = %d", rc);
|
|
}
|
|
|
|
/* Remember peer. */
|
|
rc = peer_add(event->connect.conn_handle);
|
|
if (rc != 0) {
|
|
ESP_LOGE(tag, "Failed to add peer; rc = %d", rc);
|
|
return 0;
|
|
}
|
|
|
|
/* Perform service discovery. */
|
|
rc = peer_disc_all(event->connect.conn_handle,
|
|
blecent_on_disc_complete, NULL);
|
|
if (rc != 0) {
|
|
ESP_LOGE(tag, "Failed to discover services; rc = %d", rc);
|
|
return 0;
|
|
}
|
|
} else {
|
|
/* Connection attempt failed; resume scanning. */
|
|
ESP_LOGE(tag, "Error: Connection failed; status = %d",
|
|
event->connect.status);
|
|
blecent_scan();
|
|
}
|
|
|
|
return 0;
|
|
|
|
case BLE_GAP_EVENT_DISCONNECT:
|
|
/* Connection terminated. */
|
|
ESP_LOGI(tag, "disconnect; reason=%d ", event->disconnect.reason);
|
|
print_conn_desc(&event->disconnect.conn);
|
|
ESP_LOGI(tag, "\n");
|
|
|
|
/* Forget about peer. */
|
|
peer_delete(event->disconnect.conn.conn_handle);
|
|
/* Resume scanning. */
|
|
blecent_scan();
|
|
return 0;
|
|
|
|
case BLE_GAP_EVENT_DISC_COMPLETE:
|
|
ESP_LOGI(tag, "discovery complete; reason = %d",
|
|
event->disc_complete.reason);
|
|
return 0;
|
|
|
|
case BLE_GAP_EVENT_ENC_CHANGE:
|
|
/* Encryption has been enabled or disabled for this connection. */
|
|
ESP_LOGI(tag, "encryption change event; status = %d ",
|
|
event->enc_change.status);
|
|
rc = ble_gap_conn_find(event->enc_change.conn_handle, &desc);
|
|
assert(rc == 0);
|
|
print_conn_desc(&desc);
|
|
return 0;
|
|
|
|
case BLE_GAP_EVENT_NOTIFY_RX:
|
|
/* Peer sent us a notification or indication. */
|
|
mbuf_len_total = mbuf_len_total + OS_MBUF_PKTLEN(event->notify_rx.om);
|
|
ESP_LOGI(tag, "received %s; conn_handle = %d attr_handle = %d "
|
|
"attr_len = %d ; Total length = %d",
|
|
event->notify_rx.indication ?
|
|
"indication" :
|
|
"notification",
|
|
event->notify_rx.conn_handle,
|
|
event->notify_rx.attr_handle,
|
|
OS_MBUF_PKTLEN(event->notify_rx.om),
|
|
mbuf_len_total);
|
|
/* Attribute data is contained in event->notify_rx.attr_data. */
|
|
return 0;
|
|
|
|
case BLE_GAP_EVENT_MTU:
|
|
ESP_LOGI(tag, "mtu update event; conn_handle = %d cid = %d mtu = %d",
|
|
event->mtu.conn_handle,
|
|
event->mtu.channel_id,
|
|
event->mtu.value);
|
|
return 0;
|
|
|
|
default:
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
static void
|
|
blecent_on_reset(int reason)
|
|
{
|
|
ESP_LOGE(tag, "Resetting state; reason=%d\n", reason);
|
|
}
|
|
|
|
static void
|
|
blecent_on_sync(void)
|
|
{
|
|
int rc;
|
|
bool yes = false;
|
|
|
|
/* Make sure we have proper identity address set (public preferred) */
|
|
rc = ble_hs_util_ensure_addr(0);
|
|
assert(rc == 0);
|
|
|
|
ESP_LOGI(tag, "Do you want to configure connection params ? ");
|
|
ESP_LOGI(tag, "If yes then enter in this format: `Insert Yes` ");
|
|
if (scli_receive_yesno(&yes)) {
|
|
if (yes) {
|
|
ESP_LOGI(tag, " Enter preferred MTU, format:: `MTU 512` ");
|
|
if (scli_receive_key(&mtu_def)) {
|
|
ESP_LOGI(tag, "MTU provided by user= %d", mtu_def);
|
|
} else {
|
|
ESP_LOGD(tag, "No input for setting MTU; use default mtu = %d", mtu_def);
|
|
}
|
|
|
|
scli_reset_queue();
|
|
ESP_LOGI(tag, "For Conn param: Enter `min_itvl` `max_itvl`"
|
|
"`latency` `timeout` `min_ce` `max_ce` in same order");
|
|
ESP_LOGI(tag, "Enter conn_update in this format:: `conn 6 120 0 500 0 0`");
|
|
|
|
if (scli_receive_key(conn_params_def)) {
|
|
/** Minimum value for connection interval in 1.25ms units */
|
|
conn_params.itvl_min = conn_params_def[0];
|
|
/** Maximum value for connection interval in 1.25ms units */
|
|
conn_params.itvl_max = conn_params_def[1];
|
|
/** Connection latency */
|
|
conn_params.latency = conn_params_def[2];
|
|
/** Supervision timeout in 10ms units */
|
|
conn_params.supervision_timeout = conn_params_def[3];
|
|
/** Minimum length of connection event in 0.625ms units */
|
|
conn_params.min_ce_len = conn_params_def[4];
|
|
/** Maximum length of connection event in 0.625ms units */
|
|
conn_params.max_ce_len = conn_params_def[5];
|
|
|
|
} else {
|
|
ESP_LOGD(tag, "no input by user for conn param; use default ");
|
|
}
|
|
scli_reset_queue();
|
|
}
|
|
}
|
|
/* Begin scanning for a peripheral to connect to. */
|
|
blecent_scan();
|
|
}
|
|
|
|
void blecent_host_task(void *param)
|
|
{
|
|
ESP_LOGI(tag, "BLE Host Task Started");
|
|
xSemaphore = xSemaphoreCreateBinary();
|
|
xSemaphoreGive(xSemaphore);
|
|
|
|
/* This function will return only when nimble_port_stop() is executed */
|
|
nimble_port_run();
|
|
vSemaphoreDelete(xSemaphore);
|
|
nimble_port_freertos_deinit();
|
|
}
|
|
|
|
void
|
|
app_main(void)
|
|
{
|
|
int rc;
|
|
/* Initialize NVS — it is used to store PHY calibration data */
|
|
esp_err_t ret = nvs_flash_init();
|
|
if (ret == ESP_ERR_NVS_NO_FREE_PAGES || ret == ESP_ERR_NVS_NEW_VERSION_FOUND) {
|
|
ESP_ERROR_CHECK(nvs_flash_erase());
|
|
ret = nvs_flash_init();
|
|
}
|
|
ESP_ERROR_CHECK(ret);
|
|
ESP_ERROR_CHECK(esp_nimble_hci_and_controller_init());
|
|
|
|
nimble_port_init();
|
|
|
|
/* Configure the host. */
|
|
ble_hs_cfg.reset_cb = blecent_on_reset;
|
|
ble_hs_cfg.sync_cb = blecent_on_sync;
|
|
ble_hs_cfg.store_status_cb = ble_store_util_status_rr;
|
|
|
|
/* Initialize data structures to track connected peers. */
|
|
rc = peer_init(MYNEWT_VAL(BLE_MAX_CONNECTIONS), 64, 64, 64);
|
|
assert(rc == 0);
|
|
|
|
/* Set the default device name. */
|
|
rc = ble_svc_gap_device_name_set("gattc-throughput");
|
|
assert(rc == 0);
|
|
|
|
/* XXX Need to have template for store */
|
|
ble_store_config_init();
|
|
|
|
/* Before starting with blecent host task, let us get CLI task up and
|
|
* running */
|
|
esp_console_repl_config_t repl_config = ESP_CONSOLE_REPL_CONFIG_DEFAULT();
|
|
esp_console_dev_uart_config_t uart_config = ESP_CONSOLE_DEV_UART_CONFIG_DEFAULT();
|
|
esp_console_repl_t *repl = NULL;
|
|
repl_config.prompt = "Throughput : ";
|
|
ESP_ERROR_CHECK(esp_console_new_repl_uart(&uart_config, &repl_config, &repl));
|
|
register_system();
|
|
esp_console_register_help_command();
|
|
/* Register commands */
|
|
ble_register_cli();
|
|
|
|
/* As BLE CLI has been registered, let us start nimble host task */
|
|
nimble_port_freertos_init(blecent_host_task);
|
|
|
|
printf("\n ===============================================================================================\n");
|
|
printf(" | Steps to test nimble throughput |\n");
|
|
printf(" | |\n");
|
|
printf(" | 1. Print 'help' to gain overview of commands |\n");
|
|
printf(" | |\n");
|
|
printf("\n ===============================================================================================\n");
|
|
|
|
const char *prompt = LOG_COLOR_I "Throughput demo >> " LOG_RESET_COLOR;
|
|
|
|
while (true) {
|
|
/* Get a line using linenoise.
|
|
* The line is returned when ENTER is pressed.
|
|
*/
|
|
char *line = linenoise(prompt);
|
|
if (line == NULL) { /* Ignore empty lines */
|
|
continue;
|
|
}
|
|
/* Add the command to the history */
|
|
linenoiseHistoryAdd(line);
|
|
|
|
/* Try to run the command */
|
|
int ret;
|
|
esp_err_t err = esp_console_run(line, &ret);
|
|
if (err == ESP_ERR_NOT_FOUND) {
|
|
printf("Unrecognized command\n");
|
|
} else if (err == ESP_ERR_INVALID_ARG) {
|
|
// command was empty
|
|
} else if (err == ESP_OK && ret != ESP_OK) {
|
|
printf("Command returned non-zero error code: 0x%x (%s)\n", ret, esp_err_to_name(ret));
|
|
} else if (err != ESP_OK) {
|
|
printf("Internal error: %s\n", esp_err_to_name(err));
|
|
}
|
|
/* linenoise allocates line buffer on the heap, so need to free it */
|
|
linenoiseFree(line);
|
|
}
|
|
}
|