123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823 |
- #include "adafruit_ble.h"
- #include <stdio.h>
- #include <stdlib.h>
- #include <alloca.h>
- #include <util/delay.h>
- #include <util/atomic.h>
- #include "debug.h"
- #include "pincontrol.h"
- #include "timer.h"
- #include "action_util.h"
- #include "ringbuffer.hpp"
- #include <string.h>
- #ifndef AdafruitBleResetPin
- #define AdafruitBleResetPin D4
- #endif
- #ifndef AdafruitBleCSPin
- #define AdafruitBleCSPin B4
- #endif
- #ifndef AdafruitBleIRQPin
- #define AdafruitBleIRQPin E6
- #endif
- #define SAMPLE_BATTERY
- #define ConnectionUpdateInterval 1000
- static struct {
- bool is_connected;
- bool initialized;
- bool configured;
- #define ProbedEvents 1
- #define UsingEvents 2
- bool event_flags;
- #ifdef SAMPLE_BATTERY
- uint16_t last_battery_update;
- uint32_t vbat;
- #endif
- uint16_t last_connection_update;
- } state;
- #define SdepMaxPayload 16
- struct sdep_msg {
- uint8_t type;
- uint8_t cmd_low;
- uint8_t cmd_high;
- struct __attribute__((packed)) {
- uint8_t len:7;
- uint8_t more:1;
- };
- uint8_t payload[SdepMaxPayload];
- } __attribute__((packed));
- enum queue_type {
- QTKeyReport,
- QTConsumer,
- #ifdef MOUSE_ENABLE
- QTMouseMove,
- #endif
- };
- struct queue_item {
- enum queue_type queue_type;
- uint16_t added;
- union __attribute__((packed)) {
- struct __attribute__((packed)) {
- uint8_t modifier;
- uint8_t keys[6];
- } key;
- uint16_t consumer;
- struct __attribute__((packed)) {
- int8_t x, y, scroll, pan;
- uint8_t buttons;
- } mousemove;
- };
- };
- static RingBuffer<queue_item, 40> send_buf;
- static RingBuffer<uint16_t, 2> resp_buf;
- static bool process_queue_item(struct queue_item *item, uint16_t timeout);
- enum sdep_type {
- SdepCommand = 0x10,
- SdepResponse = 0x20,
- SdepAlert = 0x40,
- SdepError = 0x80,
- SdepSlaveNotReady = 0xfe,
- SdepSlaveOverflow = 0xff,
- };
- enum ble_cmd {
- BleInitialize = 0xbeef,
- BleAtWrapper = 0x0a00,
- BleUartTx = 0x0a01,
- BleUartRx = 0x0a02,
- };
- enum ble_system_event_bits {
- BleSystemConnected = 0,
- BleSystemDisconnected = 1,
- BleSystemUartRx = 8,
- BleSystemMidiRx = 10,
- };
- #define SpiBusSpeed 4000000
- #define SdepTimeout 150
- #define SdepShortTimeout 10
- #define SdepBackOff 25
- #define BatteryUpdateInterval 10000
- static bool at_command(const char *cmd, char *resp, uint16_t resplen,
- bool verbose, uint16_t timeout = SdepTimeout);
- static bool at_command_P(const char *cmd, char *resp, uint16_t resplen,
- bool verbose = false);
- struct SPI_Settings {
- uint8_t spcr, spsr;
- };
- static struct SPI_Settings spi;
- void SPI_init(struct SPI_Settings *spi) {
- spi->spcr = _BV(SPE) | _BV(MSTR);
- spi->spsr = _BV(SPI2X);
- static_assert(SpiBusSpeed == F_CPU / 2, "hard coded at 4Mhz");
- ATOMIC_BLOCK(ATOMIC_RESTORESTATE) {
-
- digitalWrite(B0, PinLevelHigh);
- pinMode(B0, PinDirectionOutput);
- SPCR |= _BV(MSTR);
- SPCR |= _BV(SPE);
- pinMode(B1 , PinDirectionOutput);
- pinMode(B2 , PinDirectionOutput);
- }
- }
- static inline void SPI_begin(struct SPI_Settings*spi) {
- SPCR = spi->spcr;
- SPSR = spi->spsr;
- }
- static inline uint8_t SPI_TransferByte(uint8_t data) {
- SPDR = data;
- asm volatile("nop");
- while (!(SPSR & _BV(SPIF))) {
- ;
- }
- return SPDR;
- }
- static inline void spi_send_bytes(const uint8_t *buf, uint8_t len) {
- if (len == 0) return;
- const uint8_t *end = buf + len;
- while (buf < end) {
- SPDR = *buf;
- while (!(SPSR & _BV(SPIF))) {
- ;
- }
- ++buf;
- }
- }
- static inline uint16_t spi_read_byte(void) {
- return SPI_TransferByte(0x00 );
- }
- static inline void spi_recv_bytes(uint8_t *buf, uint8_t len) {
- const uint8_t *end = buf + len;
- if (len == 0) return;
- while (buf < end) {
- SPDR = 0;
- while (!(SPSR & _BV(SPIF))) {
- ;
- }
- *buf = SPDR;
- ++buf;
- }
- }
- #if 0
- static void dump_pkt(const struct sdep_msg *msg) {
- print("pkt: type=");
- print_hex8(msg->type);
- print(" cmd=");
- print_hex8(msg->cmd_high);
- print_hex8(msg->cmd_low);
- print(" len=");
- print_hex8(msg->len);
- print(" more=");
- print_hex8(msg->more);
- print("\n");
- }
- #endif
- static bool sdep_send_pkt(const struct sdep_msg *msg, uint16_t timeout) {
- SPI_begin(&spi);
- digitalWrite(AdafruitBleCSPin, PinLevelLow);
- uint16_t timerStart = timer_read();
- bool success = false;
- bool ready = false;
- do {
- ready = SPI_TransferByte(msg->type) != SdepSlaveNotReady;
- if (ready) {
- break;
- }
-
- digitalWrite(AdafruitBleCSPin, PinLevelHigh);
- _delay_us(SdepBackOff);
- digitalWrite(AdafruitBleCSPin, PinLevelLow);
- } while (timer_elapsed(timerStart) < timeout);
- if (ready) {
-
- spi_send_bytes(&msg->cmd_low,
- sizeof(*msg) - (1 + sizeof(msg->payload)) + msg->len);
- success = true;
- }
- digitalWrite(AdafruitBleCSPin, PinLevelHigh);
- return success;
- }
- static inline void sdep_build_pkt(struct sdep_msg *msg, uint16_t command,
- const uint8_t *payload, uint8_t len,
- bool moredata) {
- msg->type = SdepCommand;
- msg->cmd_low = command & 0xff;
- msg->cmd_high = command >> 8;
- msg->len = len;
- msg->more = (moredata && len == SdepMaxPayload) ? 1 : 0;
- static_assert(sizeof(*msg) == 20, "msg is correctly packed");
- memcpy(msg->payload, payload, len);
- }
- static bool sdep_recv_pkt(struct sdep_msg *msg, uint16_t timeout) {
- bool success = false;
- uint16_t timerStart = timer_read();
- bool ready = false;
- do {
- ready = digitalRead(AdafruitBleIRQPin);
- if (ready) {
- break;
- }
- _delay_us(1);
- } while (timer_elapsed(timerStart) < timeout);
- if (ready) {
- SPI_begin(&spi);
- digitalWrite(AdafruitBleCSPin, PinLevelLow);
- do {
-
- msg->type = spi_read_byte();
- if (msg->type == SdepSlaveNotReady || msg->type == SdepSlaveOverflow) {
-
- digitalWrite(AdafruitBleCSPin, PinLevelHigh);
- _delay_us(SdepBackOff);
- digitalWrite(AdafruitBleCSPin, PinLevelLow);
- continue;
- }
-
- spi_recv_bytes(&msg->cmd_low, sizeof(*msg) - (1 + sizeof(msg->payload)));
-
- if (msg->len <= SdepMaxPayload) {
- spi_recv_bytes(msg->payload, msg->len);
- }
- success = true;
- break;
- } while (timer_elapsed(timerStart) < timeout);
- digitalWrite(AdafruitBleCSPin, PinLevelHigh);
- }
- return success;
- }
- static void resp_buf_read_one(bool greedy) {
- uint16_t last_send;
- if (!resp_buf.peek(last_send)) {
- return;
- }
- if (digitalRead(AdafruitBleIRQPin)) {
- struct sdep_msg msg;
- again:
- if (sdep_recv_pkt(&msg, SdepTimeout)) {
- if (!msg.more) {
-
- resp_buf.get(last_send);
- dprintf("recv latency %dms\n", TIMER_DIFF_16(timer_read(), last_send));
- }
- if (greedy && resp_buf.peek(last_send) && digitalRead(AdafruitBleIRQPin)) {
- goto again;
- }
- }
- } else if (timer_elapsed(last_send) > SdepTimeout * 2) {
- dprintf("waiting_for_result: timeout, resp_buf size %d\n",
- (int)resp_buf.size());
-
- resp_buf.get(last_send);
- }
- }
- static void send_buf_send_one(uint16_t timeout = SdepTimeout) {
- struct queue_item item;
-
- if (!resp_buf.empty()) {
- return;
- }
- if (!send_buf.peek(item)) {
- return;
- }
- if (process_queue_item(&item, timeout)) {
-
- send_buf.get(item);
- dprintf("send_buf_send_one: have %d remaining\n", (int)send_buf.size());
- } else {
- dprint("failed to send, will retry\n");
- _delay_ms(SdepTimeout);
- resp_buf_read_one(true);
- }
- }
- static void resp_buf_wait(const char *cmd) {
- bool didPrint = false;
- while (!resp_buf.empty()) {
- if (!didPrint) {
- dprintf("wait on buf for %s\n", cmd);
- didPrint = true;
- }
- resp_buf_read_one(true);
- }
- }
- static bool ble_init(void) {
- state.initialized = false;
- state.configured = false;
- state.is_connected = false;
- pinMode(AdafruitBleIRQPin, PinDirectionInput);
- pinMode(AdafruitBleCSPin, PinDirectionOutput);
- digitalWrite(AdafruitBleCSPin, PinLevelHigh);
- SPI_init(&spi);
-
- pinMode(AdafruitBleResetPin, PinDirectionOutput);
- digitalWrite(AdafruitBleResetPin, PinLevelHigh);
- digitalWrite(AdafruitBleResetPin, PinLevelLow);
- _delay_ms(10);
- digitalWrite(AdafruitBleResetPin, PinLevelHigh);
- _delay_ms(1000);
- state.initialized = true;
- return state.initialized;
- }
- static inline uint8_t min(uint8_t a, uint8_t b) {
- return a < b ? a : b;
- }
- static bool read_response(char *resp, uint16_t resplen, bool verbose) {
- char *dest = resp;
- char *end = dest + resplen;
- while (true) {
- struct sdep_msg msg;
- if (!sdep_recv_pkt(&msg, 2 * SdepTimeout)) {
- dprint("sdep_recv_pkt failed\n");
- return false;
- }
- if (msg.type != SdepResponse) {
- *resp = 0;
- return false;
- }
- uint8_t len = min(msg.len, end - dest);
- if (len > 0) {
- memcpy(dest, msg.payload, len);
- dest += len;
- }
- if (!msg.more) {
-
- break;
- }
- }
-
- *dest = 0;
-
-
- --dest;
- while (dest > resp && (dest[0] == '\n' || dest[0] == '\r')) {
- *dest = 0;
- --dest;
- }
-
- char *last_line = strrchr(resp, '\n');
- if (last_line) {
- ++last_line;
- } else {
- last_line = resp;
- }
- bool success = false;
- static const char kOK[] PROGMEM = "OK";
- success = !strcmp_P(last_line, kOK );
- if (verbose || !success) {
- dprintf("result: %s\n", resp);
- }
- return success;
- }
- static bool at_command(const char *cmd, char *resp, uint16_t resplen,
- bool verbose, uint16_t timeout) {
- const char *end = cmd + strlen(cmd);
- struct sdep_msg msg;
- if (verbose) {
- dprintf("ble send: %s\n", cmd);
- }
- if (resp) {
-
-
-
- resp_buf_wait(cmd);
- *resp = 0;
- }
-
- while (end - cmd > SdepMaxPayload) {
- sdep_build_pkt(&msg, BleAtWrapper, (uint8_t *)cmd, SdepMaxPayload, true);
- if (!sdep_send_pkt(&msg, timeout)) {
- return false;
- }
- cmd += SdepMaxPayload;
- }
- sdep_build_pkt(&msg, BleAtWrapper, (uint8_t *)cmd, end - cmd, false);
- if (!sdep_send_pkt(&msg, timeout)) {
- return false;
- }
- if (resp == NULL) {
- auto now = timer_read();
- while (!resp_buf.enqueue(now)) {
- resp_buf_read_one(false);
- }
- auto later = timer_read();
- if (TIMER_DIFF_16(later, now) > 0) {
- dprintf("waited %dms for resp_buf\n", TIMER_DIFF_16(later, now));
- }
- return true;
- }
- return read_response(resp, resplen, verbose);
- }
- bool at_command_P(const char *cmd, char *resp, uint16_t resplen, bool verbose) {
- auto cmdbuf = (char *)alloca(strlen_P(cmd) + 1);
- strcpy_P(cmdbuf, cmd);
- return at_command(cmdbuf, resp, resplen, verbose);
- }
- bool adafruit_ble_is_connected(void) {
- return state.is_connected;
- }
- bool adafruit_ble_enable_keyboard(void) {
- char resbuf[128];
- if (!state.initialized && !ble_init()) {
- return false;
- }
- state.configured = false;
-
- static const char kEcho[] PROGMEM = "ATE=0";
-
- static const char kGapDevName[] PROGMEM =
- "AT+GAPDEVNAME=" STR(PRODUCT) " " STR(DESCRIPTION);
-
- static const char kHidEnOn[] PROGMEM = "AT+BLEHIDEN=1";
-
-
-
-
-
- static const char kGapIntervals[] PROGMEM = "AT+GAPINTERVALS=10,30,,";
-
- static const char kATZ[] PROGMEM = "ATZ";
-
- static const char kPower[] PROGMEM = "AT+BLEPOWERLEVEL=-12";
- static PGM_P const configure_commands[] PROGMEM = {
- kEcho,
- kGapIntervals,
- kGapDevName,
- kHidEnOn,
- kPower,
- kATZ,
- };
- uint8_t i;
- for (i = 0; i < sizeof(configure_commands) / sizeof(configure_commands[0]);
- ++i) {
- PGM_P cmd;
- memcpy_P(&cmd, configure_commands + i, sizeof(cmd));
- if (!at_command_P(cmd, resbuf, sizeof(resbuf))) {
- dprintf("failed BLE command: %S: %s\n", cmd, resbuf);
- goto fail;
- }
- }
- state.configured = true;
-
-
- state.last_connection_update = timer_read();
- fail:
- return state.configured;
- }
- static void set_connected(bool connected) {
- if (connected != state.is_connected) {
- if (connected) {
- print("****** BLE CONNECT!!!!\n");
- } else {
- print("****** BLE DISCONNECT!!!!\n");
- }
- state.is_connected = connected;
-
-
-
-
-
-
-
-
- }
- }
- void adafruit_ble_task(void) {
- char resbuf[48];
- if (!state.configured && !adafruit_ble_enable_keyboard()) {
- return;
- }
- resp_buf_read_one(true);
- send_buf_send_one(SdepShortTimeout);
- if (resp_buf.empty() && (state.event_flags & UsingEvents) &&
- digitalRead(AdafruitBleIRQPin)) {
-
- if (at_command_P(PSTR("AT+EVENTSTATUS"), resbuf, sizeof(resbuf))) {
- uint32_t mask = strtoul(resbuf, NULL, 16);
- if (mask & BleSystemConnected) {
- set_connected(true);
- } else if (mask & BleSystemDisconnected) {
- set_connected(false);
- }
- }
- }
- if (timer_elapsed(state.last_connection_update) > ConnectionUpdateInterval) {
- bool shouldPoll = true;
- if (!(state.event_flags & ProbedEvents)) {
-
-
-
-
-
-
- if (at_command_P(PSTR("AT+EVENTENABLE=0x1"), resbuf, sizeof(resbuf))) {
- at_command_P(PSTR("AT+EVENTENABLE=0x2"), resbuf, sizeof(resbuf));
- state.event_flags |= UsingEvents;
- }
- state.event_flags |= ProbedEvents;
-
-
- } else {
- shouldPoll = false;
- }
- static const char kGetConn[] PROGMEM = "AT+GAPGETCONN";
- state.last_connection_update = timer_read();
- if (at_command_P(kGetConn, resbuf, sizeof(resbuf))) {
- set_connected(atoi(resbuf));
- }
- }
- #ifdef SAMPLE_BATTERY
-
-
-
- if (timer_elapsed(state.last_battery_update) > BatteryUpdateInterval &&
- resp_buf.empty()) {
- state.last_battery_update = timer_read();
- if (at_command_P(PSTR("AT+HWVBAT"), resbuf, sizeof(resbuf))) {
- state.vbat = atoi(resbuf);
- }
- }
- #endif
- }
- static bool process_queue_item(struct queue_item *item, uint16_t timeout) {
- char cmdbuf[48];
- char fmtbuf[64];
-
- state.last_connection_update = timer_read();
- #if 1
- if (TIMER_DIFF_16(state.last_connection_update, item->added) > 0) {
- dprintf("send latency %dms\n",
- TIMER_DIFF_16(state.last_connection_update, item->added));
- }
- #endif
- switch (item->queue_type) {
- case QTKeyReport:
- strcpy_P(fmtbuf,
- PSTR("AT+BLEKEYBOARDCODE=%02x-00-%02x-%02x-%02x-%02x-%02x-%02x"));
- snprintf(cmdbuf, sizeof(cmdbuf), fmtbuf, item->key.modifier,
- item->key.keys[0], item->key.keys[1], item->key.keys[2],
- item->key.keys[3], item->key.keys[4], item->key.keys[5]);
- return at_command(cmdbuf, NULL, 0, true, timeout);
- case QTConsumer:
- strcpy_P(fmtbuf, PSTR("AT+BLEHIDCONTROLKEY=0x%04x"));
- snprintf(cmdbuf, sizeof(cmdbuf), fmtbuf, item->consumer);
- return at_command(cmdbuf, NULL, 0, true, timeout);
- #ifdef MOUSE_ENABLE
- case QTMouseMove:
- strcpy_P(fmtbuf, PSTR("AT+BLEHIDMOUSEMOVE=%d,%d,%d,%d"));
- snprintf(cmdbuf, sizeof(cmdbuf), fmtbuf, item->mousemove.x,
- item->mousemove.y, item->mousemove.scroll, item->mousemove.pan);
- if (!at_command(cmdbuf, NULL, 0, true, timeout)) {
- return false;
- }
- strcpy_P(cmdbuf, PSTR("AT+BLEHIDMOUSEBUTTON="));
- if (item->mousemove.buttons & MOUSE_BTN1) {
- strcat(cmdbuf, "L");
- }
- if (item->mousemove.buttons & MOUSE_BTN2) {
- strcat(cmdbuf, "R");
- }
- if (item->mousemove.buttons & MOUSE_BTN3) {
- strcat(cmdbuf, "M");
- }
- if (item->mousemove.buttons == 0) {
- strcat(cmdbuf, "0");
- }
- return at_command(cmdbuf, NULL, 0, true, timeout);
- #endif
- default:
- return true;
- }
- }
- bool adafruit_ble_send_keys(uint8_t hid_modifier_mask, uint8_t *keys,
- uint8_t nkeys) {
- struct queue_item item;
- bool didWait = false;
- item.queue_type = QTKeyReport;
- item.key.modifier = hid_modifier_mask;
- item.added = timer_read();
- while (nkeys >= 0) {
- item.key.keys[0] = keys[0];
- item.key.keys[1] = nkeys >= 1 ? keys[1] : 0;
- item.key.keys[2] = nkeys >= 2 ? keys[2] : 0;
- item.key.keys[3] = nkeys >= 3 ? keys[3] : 0;
- item.key.keys[4] = nkeys >= 4 ? keys[4] : 0;
- item.key.keys[5] = nkeys >= 5 ? keys[5] : 0;
- if (!send_buf.enqueue(item)) {
- if (!didWait) {
- dprint("wait for buf space\n");
- didWait = true;
- }
- send_buf_send_one();
- continue;
- }
- if (nkeys <= 6) {
- return true;
- }
- nkeys -= 6;
- keys += 6;
- }
- return true;
- }
- bool adafruit_ble_send_consumer_key(uint16_t keycode, int hold_duration) {
- struct queue_item item;
- item.queue_type = QTConsumer;
- item.consumer = keycode;
- while (!send_buf.enqueue(item)) {
- send_buf_send_one();
- }
- return true;
- }
- #ifdef MOUSE_ENABLE
- bool adafruit_ble_send_mouse_move(int8_t x, int8_t y, int8_t scroll,
- int8_t pan, uint8_t buttons) {
- struct queue_item item;
- item.queue_type = QTMouseMove;
- item.mousemove.x = x;
- item.mousemove.y = y;
- item.mousemove.scroll = scroll;
- item.mousemove.pan = pan;
- item.mousemove.buttons = buttons;
- while (!send_buf.enqueue(item)) {
- send_buf_send_one();
- }
- return true;
- }
- #endif
- uint32_t adafruit_ble_read_battery_voltage(void) {
- return state.vbat;
- }
- bool adafruit_ble_set_mode_leds(bool on) {
- if (!state.configured) {
- return false;
- }
-
- at_command_P(on ? PSTR("AT+HWMODELED=1") : PSTR("AT+HWMODELED=0"), NULL, 0);
-
-
-
- at_command_P(on && state.is_connected ? PSTR("AT+HWGPIO=19,1")
- : PSTR("AT+HWGPIO=19,0"),
- NULL, 0);
- return true;
- }
- bool adafruit_ble_set_power_level(int8_t level) {
- char cmd[46];
- if (!state.configured) {
- return false;
- }
- snprintf(cmd, sizeof(cmd), "AT+BLEPOWERLEVEL=%d", level);
- return at_command(cmd, NULL, 0, false);
- }
|