197 lines
6.6 KiB
C
197 lines
6.6 KiB
C
/*
|
|
* This file is part of the MicroPython project, http://micropython.org/
|
|
*
|
|
* The MIT License (MIT)
|
|
*
|
|
* Copyright (c) 2017 Glenn Ruben Bakke
|
|
* Copyright (c) 2018 Artur Pacholec
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
* THE SOFTWARE.
|
|
*/
|
|
|
|
#include <string.h>
|
|
|
|
#include "ble.h"
|
|
#include "ble_uart.h"
|
|
#include "ringbuffer.h"
|
|
#include "py/mphal.h"
|
|
#include "py/runtime.h"
|
|
#include "lib/utils/interrupt_char.h"
|
|
#include "shared-bindings/bleio/Adapter.h"
|
|
#include "shared-bindings/bleio/Characteristic.h"
|
|
#include "shared-bindings/bleio/Device.h"
|
|
#include "shared-bindings/bleio/Service.h"
|
|
#include "shared-bindings/bleio/UUID.h"
|
|
|
|
#if (MICROPY_PY_BLE_NUS == 1)
|
|
|
|
static const char default_name[] = "CP-REPL"; // max 8 chars or uuid won't fit in adv data
|
|
static const char NUS_UUID[] = "6e400001-b5a3-f393-e0a9-e50e24dcca9e";
|
|
|
|
#define NUS_RX_UUID 0x0002
|
|
#define NUS_TX_UUID 0x0003
|
|
#define BUFFER_SIZE 128
|
|
|
|
ringBuffer_typedef(uint8_t, ringbuffer_t);
|
|
|
|
static bleio_device_obj_t m_device;
|
|
static bleio_service_obj_t *m_nus;
|
|
static bleio_characteristic_obj_t *m_tx_chara;
|
|
static bleio_characteristic_obj_t *m_rx_chara;
|
|
|
|
static volatile bool m_cccd_enabled;
|
|
|
|
static uint8_t m_rx_ring_buffer_data[BUFFER_SIZE];
|
|
static ringbuffer_t m_rx_ring_buffer = {
|
|
.size = sizeof(m_rx_ring_buffer_data) + 1,
|
|
.elems = m_rx_ring_buffer_data,
|
|
};
|
|
|
|
STATIC void on_ble_evt(ble_evt_t *ble_evt, void *param) {
|
|
switch (ble_evt->header.evt_id) {
|
|
case BLE_GAP_EVT_DISCONNECTED:
|
|
{
|
|
mp_obj_t device_obj = MP_OBJ_FROM_PTR(&m_device);
|
|
mp_call_function_0(mp_load_attr(device_obj, qstr_from_str("start_advertising")));
|
|
break;
|
|
}
|
|
|
|
case BLE_GATTS_EVT_WRITE:
|
|
{
|
|
ble_gatts_evt_write_t *write = &ble_evt->evt.gatts_evt.params.write;
|
|
|
|
if (write->handle == m_tx_chara->cccd_handle) {
|
|
m_cccd_enabled = true;
|
|
} else if (write->handle == m_rx_chara->handle) {
|
|
for (size_t i = 0; i < write->len; ++i) {
|
|
#if MICROPY_KBD_EXCEPTION
|
|
if (write->data[i] == mp_interrupt_char) {
|
|
mp_keyboard_interrupt();
|
|
} else
|
|
#endif
|
|
{
|
|
bufferWrite(&m_rx_ring_buffer, write->data[i]);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void ble_uart_init(void) {
|
|
mp_obj_t device_obj = MP_OBJ_FROM_PTR(&m_device);
|
|
m_device.base.type = &bleio_device_type;
|
|
m_device.service_list = mp_obj_new_list(0, NULL);
|
|
m_device.notif_handler = mp_const_none;
|
|
m_device.conn_handler = mp_const_none;
|
|
m_device.conn_handle = 0xFFFF;
|
|
m_device.is_peripheral = true;
|
|
m_device.name = mp_obj_new_str(default_name, strlen(default_name), false);
|
|
common_hal_bleio_adapter_get_address(&m_device.address);
|
|
|
|
mp_obj_t nus_uuid_str = mp_obj_new_str(NUS_UUID, strlen(NUS_UUID), false);
|
|
mp_obj_t nus_uuid_obj = bleio_uuid_type.make_new(&bleio_uuid_type, 1, 0, &nus_uuid_str);
|
|
mp_obj_t nus_obj = bleio_service_type.make_new(&bleio_service_type, 1, 0, &nus_uuid_obj);
|
|
m_nus = MP_OBJ_TO_PTR(nus_obj);
|
|
mp_call_function_1(mp_load_attr(device_obj, qstr_from_str("add_service")), nus_obj);
|
|
|
|
mp_obj_t tx_uuid_int = mp_obj_new_int(NUS_TX_UUID);
|
|
mp_obj_t tx_uuid_obj = bleio_uuid_type.make_new(&bleio_uuid_type, 1, 0, &tx_uuid_int);
|
|
mp_obj_t tx_obj = bleio_characteristic_type.make_new(&bleio_characteristic_type, 1, 0, &tx_uuid_obj);
|
|
m_tx_chara = MP_OBJ_TO_PTR(tx_obj);
|
|
m_tx_chara->uuid->type = UUID_TYPE_128BIT;
|
|
m_tx_chara->uuid->uuid_vs_idx = m_nus->uuid->uuid_vs_idx;
|
|
m_tx_chara->props.notify = true;
|
|
mp_call_function_1(mp_load_attr(nus_obj, qstr_from_str("add_characteristic")), tx_obj);
|
|
|
|
mp_obj_t rx_uuid_int = mp_obj_new_int(NUS_RX_UUID);
|
|
mp_obj_t rx_uuid_obj = bleio_uuid_type.make_new(&bleio_uuid_type, 1, 0, &rx_uuid_int);
|
|
mp_obj_t rx_obj = bleio_characteristic_type.make_new(&bleio_characteristic_type, 1, 0, &rx_uuid_obj);
|
|
m_rx_chara = MP_OBJ_TO_PTR(rx_obj);
|
|
m_rx_chara->uuid->type = UUID_TYPE_128BIT;
|
|
m_rx_chara->uuid->uuid_vs_idx = m_nus->uuid->uuid_vs_idx;
|
|
m_rx_chara->props.write = true;
|
|
m_rx_chara->props.write_wo_resp = true;
|
|
mp_call_function_1(mp_load_attr(nus_obj, qstr_from_str("add_characteristic")), rx_obj);
|
|
|
|
mp_call_function_0(mp_load_attr(device_obj, qstr_from_str("start_advertising")));
|
|
|
|
ble_drv_add_event_handler(on_ble_evt, &m_device);
|
|
|
|
m_cccd_enabled = false;
|
|
|
|
while (!m_cccd_enabled) {
|
|
#ifdef MICROPY_VM_HOOK_LOOP
|
|
MICROPY_VM_HOOK_LOOP
|
|
#endif
|
|
}
|
|
}
|
|
|
|
bool ble_uart_connected(void) {
|
|
return (m_device.conn_handle != BLE_CONN_HANDLE_INVALID);
|
|
}
|
|
|
|
char ble_uart_rx_chr(void) {
|
|
while (isBufferEmpty(&m_rx_ring_buffer)) {
|
|
#ifdef MICROPY_VM_HOOK_LOOP
|
|
MICROPY_VM_HOOK_LOOP
|
|
#endif
|
|
}
|
|
|
|
uint8_t byte;
|
|
bufferRead(&m_rx_ring_buffer, byte);
|
|
return (int)byte;
|
|
}
|
|
|
|
bool ble_uart_stdin_any(void) {
|
|
return !isBufferEmpty(&m_rx_ring_buffer);
|
|
}
|
|
|
|
void ble_uart_stdout_tx_str(const char *text) {
|
|
mp_hal_stdout_tx_strn(text, strlen(text));
|
|
}
|
|
|
|
int mp_hal_stdin_rx_chr(void) {
|
|
return ble_uart_rx_chr();
|
|
}
|
|
|
|
void mp_hal_stdout_tx_strn(const char *str, size_t len) {
|
|
size_t send_len;
|
|
|
|
while (len > 0) {
|
|
if (len >= BLE_GATT_ATT_MTU_DEFAULT - 3) {
|
|
send_len = (BLE_GATT_ATT_MTU_DEFAULT - 3);
|
|
} else {
|
|
send_len = len;
|
|
}
|
|
|
|
mp_buffer_info_t bufinfo = {
|
|
.buf = (uint8_t*)str,
|
|
.len = send_len,
|
|
};
|
|
|
|
common_hal_bleio_characteristic_write_value(m_tx_chara, &bufinfo);
|
|
|
|
len -= send_len;
|
|
str += send_len;
|
|
}
|
|
}
|
|
|
|
#endif // MICROPY_PY_BLE_NUS
|