mirror of
https://github.com/DarkFlippers/unleashed-firmware
synced 2024-12-19 09:13:11 +00:00
416 lines
15 KiB
C
416 lines
15 KiB
C
#include "hid_service.h"
|
|
#include "app_common.h"
|
|
#include <ble/ble.h>
|
|
|
|
#include <furi.h>
|
|
|
|
#define TAG "BtHid"
|
|
|
|
typedef struct {
|
|
uint16_t svc_handle;
|
|
uint16_t protocol_mode_char_handle;
|
|
uint16_t report_char_handle[HID_SVC_REPORT_COUNT];
|
|
uint16_t report_ref_desc_handle[HID_SVC_REPORT_COUNT];
|
|
uint16_t report_map_char_handle;
|
|
uint16_t info_char_handle;
|
|
uint16_t ctrl_point_char_handle;
|
|
// led state
|
|
uint16_t led_state_char_handle;
|
|
uint16_t led_state_desc_handle;
|
|
HidLedStateEventCallback led_state_event_callback;
|
|
void* led_state_ctx;
|
|
} HIDSvc;
|
|
|
|
static HIDSvc* hid_svc = NULL;
|
|
|
|
static SVCCTL_EvtAckStatus_t hid_svc_event_handler(void* event) {
|
|
SVCCTL_EvtAckStatus_t ret = SVCCTL_EvtNotAck;
|
|
hci_event_pckt* event_pckt = (hci_event_pckt*)(((hci_uart_pckt*)event)->data);
|
|
evt_blecore_aci* blecore_evt = (evt_blecore_aci*)event_pckt->data;
|
|
// aci_gatt_attribute_modified_event_rp0* attribute_modified;
|
|
if(event_pckt->evt == HCI_VENDOR_SPECIFIC_DEBUG_EVT_CODE) {
|
|
if(blecore_evt->ecode == ACI_GATT_ATTRIBUTE_MODIFIED_VSEVT_CODE) {
|
|
// Process modification events
|
|
ret = SVCCTL_EvtAckFlowEnable;
|
|
} else if(blecore_evt->ecode == ACI_GATT_SERVER_CONFIRMATION_VSEVT_CODE) {
|
|
// Process notification confirmation
|
|
ret = SVCCTL_EvtAckFlowEnable;
|
|
} else if(blecore_evt->ecode == ACI_GATT_WRITE_PERMIT_REQ_VSEVT_CODE) {
|
|
// Process write request
|
|
aci_gatt_write_permit_req_event_rp0* req =
|
|
(aci_gatt_write_permit_req_event_rp0*)blecore_evt->data;
|
|
|
|
furi_check(hid_svc->led_state_event_callback && hid_svc->led_state_ctx);
|
|
|
|
// this check is likely to be incorrect, it will actually work in our case
|
|
// but we need to investigate gatt api to see what is the rules
|
|
// that specify attibute handle value from char handle (or the reverse)
|
|
if(req->Attribute_Handle == (hid_svc->led_state_char_handle + 1)) {
|
|
hid_svc->led_state_event_callback(req->Data[0], hid_svc->led_state_ctx);
|
|
aci_gatt_write_resp(
|
|
req->Connection_Handle,
|
|
req->Attribute_Handle,
|
|
0x00, /* write_status = 0 (no error))*/
|
|
0x00, /* err_code */
|
|
req->Data_Length,
|
|
req->Data);
|
|
aci_gatt_write_char_value(
|
|
req->Connection_Handle,
|
|
hid_svc->led_state_char_handle,
|
|
req->Data_Length,
|
|
req->Data);
|
|
ret = SVCCTL_EvtAckFlowEnable;
|
|
}
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
void hid_svc_start() {
|
|
tBleStatus status;
|
|
hid_svc = malloc(sizeof(HIDSvc));
|
|
Service_UUID_t svc_uuid = {};
|
|
Char_Desc_Uuid_t desc_uuid = {};
|
|
Char_UUID_t char_uuid = {};
|
|
|
|
// Register event handler
|
|
SVCCTL_RegisterSvcHandler(hid_svc_event_handler);
|
|
// Add service
|
|
svc_uuid.Service_UUID_16 = HUMAN_INTERFACE_DEVICE_SERVICE_UUID;
|
|
/**
|
|
* Add Human Interface Device Service
|
|
*/
|
|
status = aci_gatt_add_service(
|
|
UUID_TYPE_16,
|
|
&svc_uuid,
|
|
PRIMARY_SERVICE,
|
|
2 + /* protocol mode */
|
|
(4 * HID_SVC_INPUT_REPORT_COUNT) + (3 * HID_SVC_OUTPUT_REPORT_COUNT) +
|
|
(3 * HID_SVC_FEATURE_REPORT_COUNT) + 1 + 2 + 2 + 2 +
|
|
4, /* Service + Report Map + HID Information + HID Control Point + LED state */
|
|
&hid_svc->svc_handle);
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to add HID service: %d", status);
|
|
}
|
|
// Add Protocol mode characteristics
|
|
char_uuid.Char_UUID_16 = PROTOCOL_MODE_CHAR_UUID;
|
|
status = aci_gatt_add_char(
|
|
hid_svc->svc_handle,
|
|
UUID_TYPE_16,
|
|
&char_uuid,
|
|
1,
|
|
CHAR_PROP_READ | CHAR_PROP_WRITE_WITHOUT_RESP,
|
|
ATTR_PERMISSION_NONE,
|
|
GATT_NOTIFY_ATTRIBUTE_WRITE,
|
|
10,
|
|
CHAR_VALUE_LEN_CONSTANT,
|
|
&hid_svc->protocol_mode_char_handle);
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to add protocol mode characteristic: %d", status);
|
|
}
|
|
// Update Protocol mode characteristic
|
|
uint8_t protocol_mode = 1;
|
|
status = aci_gatt_update_char_value(
|
|
hid_svc->svc_handle, hid_svc->protocol_mode_char_handle, 0, 1, &protocol_mode);
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to update protocol mode characteristic: %d", status);
|
|
}
|
|
|
|
#if(HID_SVC_REPORT_COUNT != 0)
|
|
for(uint8_t i = 0; i < HID_SVC_REPORT_COUNT; i++) {
|
|
if(i < HID_SVC_INPUT_REPORT_COUNT) { //-V547
|
|
uint8_t buf[2] = {i + 1, 1}; // 1 input
|
|
char_uuid.Char_UUID_16 = REPORT_CHAR_UUID;
|
|
status = aci_gatt_add_char(
|
|
hid_svc->svc_handle,
|
|
UUID_TYPE_16,
|
|
&char_uuid,
|
|
HID_SVC_REPORT_MAX_LEN,
|
|
CHAR_PROP_READ | CHAR_PROP_NOTIFY,
|
|
ATTR_PERMISSION_NONE,
|
|
GATT_DONT_NOTIFY_EVENTS,
|
|
10,
|
|
CHAR_VALUE_LEN_VARIABLE,
|
|
&(hid_svc->report_char_handle[i]));
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to add report characteristic: %d", status);
|
|
}
|
|
|
|
desc_uuid.Char_UUID_16 = REPORT_REFERENCE_DESCRIPTOR_UUID;
|
|
status = aci_gatt_add_char_desc(
|
|
hid_svc->svc_handle,
|
|
hid_svc->report_char_handle[i],
|
|
UUID_TYPE_16,
|
|
&desc_uuid,
|
|
HID_SVC_REPORT_REF_LEN,
|
|
HID_SVC_REPORT_REF_LEN,
|
|
buf,
|
|
ATTR_PERMISSION_NONE,
|
|
ATTR_ACCESS_READ_WRITE,
|
|
GATT_DONT_NOTIFY_EVENTS,
|
|
MIN_ENCRY_KEY_SIZE,
|
|
CHAR_VALUE_LEN_CONSTANT,
|
|
&(hid_svc->report_ref_desc_handle[i]));
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to add report reference descriptor: %d", status);
|
|
}
|
|
} else if((i - HID_SVC_INPUT_REPORT_COUNT) < HID_SVC_OUTPUT_REPORT_COUNT) {
|
|
uint8_t buf[2] = {i + 1, 2}; // 2 output
|
|
char_uuid.Char_UUID_16 = REPORT_CHAR_UUID;
|
|
status = aci_gatt_add_char(
|
|
hid_svc->svc_handle,
|
|
UUID_TYPE_16,
|
|
&char_uuid,
|
|
HID_SVC_REPORT_MAX_LEN,
|
|
CHAR_PROP_READ | CHAR_PROP_NOTIFY,
|
|
ATTR_PERMISSION_NONE,
|
|
GATT_DONT_NOTIFY_EVENTS,
|
|
10,
|
|
CHAR_VALUE_LEN_VARIABLE,
|
|
&(hid_svc->report_char_handle[i]));
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to add report characteristic: %d", status);
|
|
}
|
|
|
|
desc_uuid.Char_UUID_16 = REPORT_REFERENCE_DESCRIPTOR_UUID;
|
|
status = aci_gatt_add_char_desc(
|
|
hid_svc->svc_handle,
|
|
hid_svc->report_char_handle[i],
|
|
UUID_TYPE_16,
|
|
&desc_uuid,
|
|
HID_SVC_REPORT_REF_LEN,
|
|
HID_SVC_REPORT_REF_LEN,
|
|
buf,
|
|
ATTR_PERMISSION_NONE,
|
|
ATTR_ACCESS_READ_WRITE,
|
|
GATT_DONT_NOTIFY_EVENTS,
|
|
MIN_ENCRY_KEY_SIZE,
|
|
CHAR_VALUE_LEN_CONSTANT,
|
|
&(hid_svc->report_ref_desc_handle[i]));
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to add report reference descriptor: %d", status);
|
|
}
|
|
} else {
|
|
uint8_t buf[2] = {i + 1, 3}; // 3 feature
|
|
char_uuid.Char_UUID_16 = REPORT_CHAR_UUID;
|
|
status = aci_gatt_add_char(
|
|
hid_svc->svc_handle,
|
|
UUID_TYPE_16,
|
|
&char_uuid,
|
|
HID_SVC_REPORT_MAX_LEN,
|
|
CHAR_PROP_READ | CHAR_PROP_NOTIFY,
|
|
ATTR_PERMISSION_NONE,
|
|
GATT_DONT_NOTIFY_EVENTS,
|
|
10,
|
|
CHAR_VALUE_LEN_VARIABLE,
|
|
&(hid_svc->report_char_handle[i]));
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to add report characteristic: %d", status);
|
|
}
|
|
|
|
desc_uuid.Char_UUID_16 = REPORT_REFERENCE_DESCRIPTOR_UUID;
|
|
status = aci_gatt_add_char_desc(
|
|
hid_svc->svc_handle,
|
|
hid_svc->report_char_handle[i],
|
|
UUID_TYPE_16,
|
|
&desc_uuid,
|
|
HID_SVC_REPORT_REF_LEN,
|
|
HID_SVC_REPORT_REF_LEN,
|
|
buf,
|
|
ATTR_PERMISSION_NONE,
|
|
ATTR_ACCESS_READ_WRITE,
|
|
GATT_DONT_NOTIFY_EVENTS,
|
|
MIN_ENCRY_KEY_SIZE,
|
|
CHAR_VALUE_LEN_CONSTANT,
|
|
&(hid_svc->report_ref_desc_handle[i]));
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to add report reference descriptor: %d", status);
|
|
}
|
|
}
|
|
}
|
|
#endif
|
|
// Add led state output report
|
|
char_uuid.Char_UUID_16 = REPORT_CHAR_UUID;
|
|
status = aci_gatt_add_char(
|
|
hid_svc->svc_handle,
|
|
UUID_TYPE_16,
|
|
&char_uuid,
|
|
1,
|
|
CHAR_PROP_READ | CHAR_PROP_WRITE_WITHOUT_RESP | CHAR_PROP_WRITE,
|
|
ATTR_PERMISSION_NONE,
|
|
GATT_NOTIFY_ATTRIBUTE_WRITE | GATT_NOTIFY_WRITE_REQ_AND_WAIT_FOR_APPL_RESP,
|
|
10,
|
|
CHAR_VALUE_LEN_CONSTANT,
|
|
&(hid_svc->led_state_char_handle));
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to add led state characteristic: %d", status);
|
|
}
|
|
|
|
// Add led state char descriptor specifying it is an output report
|
|
uint8_t buf[2] = {HID_SVC_REPORT_COUNT + 1, 2};
|
|
desc_uuid.Char_UUID_16 = REPORT_REFERENCE_DESCRIPTOR_UUID;
|
|
status = aci_gatt_add_char_desc(
|
|
hid_svc->svc_handle,
|
|
hid_svc->led_state_char_handle,
|
|
UUID_TYPE_16,
|
|
&desc_uuid,
|
|
HID_SVC_REPORT_REF_LEN,
|
|
HID_SVC_REPORT_REF_LEN,
|
|
buf,
|
|
ATTR_PERMISSION_NONE,
|
|
ATTR_ACCESS_READ_WRITE,
|
|
GATT_DONT_NOTIFY_EVENTS,
|
|
MIN_ENCRY_KEY_SIZE,
|
|
CHAR_VALUE_LEN_CONSTANT,
|
|
&(hid_svc->led_state_desc_handle));
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to add led state descriptor: %d", status);
|
|
}
|
|
// Add Report Map characteristic
|
|
char_uuid.Char_UUID_16 = REPORT_MAP_CHAR_UUID;
|
|
status = aci_gatt_add_char(
|
|
hid_svc->svc_handle,
|
|
UUID_TYPE_16,
|
|
&char_uuid,
|
|
HID_SVC_REPORT_MAP_MAX_LEN,
|
|
CHAR_PROP_READ,
|
|
ATTR_PERMISSION_NONE,
|
|
GATT_DONT_NOTIFY_EVENTS,
|
|
10,
|
|
CHAR_VALUE_LEN_VARIABLE,
|
|
&hid_svc->report_map_char_handle);
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to add report map characteristic: %d", status);
|
|
}
|
|
|
|
// Add Information characteristic
|
|
char_uuid.Char_UUID_16 = HID_INFORMATION_CHAR_UUID;
|
|
status = aci_gatt_add_char(
|
|
hid_svc->svc_handle,
|
|
UUID_TYPE_16,
|
|
&char_uuid,
|
|
HID_SVC_INFO_LEN,
|
|
CHAR_PROP_READ,
|
|
ATTR_PERMISSION_NONE,
|
|
GATT_DONT_NOTIFY_EVENTS,
|
|
10,
|
|
CHAR_VALUE_LEN_CONSTANT,
|
|
&hid_svc->info_char_handle);
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to add information characteristic: %d", status);
|
|
}
|
|
// Add Control Point characteristic
|
|
char_uuid.Char_UUID_16 = HID_CONTROL_POINT_CHAR_UUID;
|
|
status = aci_gatt_add_char(
|
|
hid_svc->svc_handle,
|
|
UUID_TYPE_16,
|
|
&char_uuid,
|
|
HID_SVC_CONTROL_POINT_LEN,
|
|
CHAR_PROP_WRITE_WITHOUT_RESP,
|
|
ATTR_PERMISSION_NONE,
|
|
GATT_NOTIFY_ATTRIBUTE_WRITE,
|
|
10,
|
|
CHAR_VALUE_LEN_CONSTANT,
|
|
&hid_svc->ctrl_point_char_handle);
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to add control point characteristic: %d", status);
|
|
}
|
|
|
|
hid_svc->led_state_event_callback = NULL;
|
|
hid_svc->led_state_ctx = NULL;
|
|
}
|
|
|
|
bool hid_svc_update_report_map(const uint8_t* data, uint16_t len) {
|
|
furi_assert(data);
|
|
furi_assert(hid_svc);
|
|
|
|
tBleStatus status = aci_gatt_update_char_value(
|
|
hid_svc->svc_handle, hid_svc->report_map_char_handle, 0, len, data);
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed updating report map characteristic: %d", status);
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool hid_svc_update_input_report(uint8_t input_report_num, uint8_t* data, uint16_t len) {
|
|
furi_assert(data);
|
|
furi_assert(hid_svc);
|
|
|
|
tBleStatus status = aci_gatt_update_char_value(
|
|
hid_svc->svc_handle, hid_svc->report_char_handle[input_report_num], 0, len, data);
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed updating report characteristic: %d", status);
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool hid_svc_update_info(uint8_t* data, uint16_t len) {
|
|
furi_assert(data);
|
|
furi_assert(hid_svc);
|
|
|
|
tBleStatus status =
|
|
aci_gatt_update_char_value(hid_svc->svc_handle, hid_svc->info_char_handle, 0, len, data);
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed updating info characteristic: %d", status);
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void hid_svc_register_led_state_callback(HidLedStateEventCallback callback, void* context) {
|
|
furi_assert(hid_svc);
|
|
furi_assert(callback);
|
|
furi_assert(context);
|
|
|
|
hid_svc->led_state_event_callback = callback;
|
|
hid_svc->led_state_ctx = context;
|
|
}
|
|
|
|
bool hid_svc_is_started() {
|
|
return hid_svc != NULL;
|
|
}
|
|
|
|
void hid_svc_stop() {
|
|
tBleStatus status;
|
|
if(hid_svc) {
|
|
// Delete characteristics
|
|
status = aci_gatt_del_char(hid_svc->svc_handle, hid_svc->report_map_char_handle);
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to delete Report Map characteristic: %d", status);
|
|
}
|
|
#if(HID_SVC_INPUT_REPORT_COUNT != 0)
|
|
for(uint8_t i = 0; i < HID_SVC_REPORT_COUNT; i++) {
|
|
status = aci_gatt_del_char(hid_svc->svc_handle, hid_svc->report_char_handle[i]);
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to delete Report characteristic: %d", status);
|
|
}
|
|
}
|
|
#endif
|
|
status = aci_gatt_del_char(hid_svc->svc_handle, hid_svc->protocol_mode_char_handle);
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to delete Protocol Mode characteristic: %d", status);
|
|
}
|
|
status = aci_gatt_del_char(hid_svc->svc_handle, hid_svc->info_char_handle);
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to delete Information characteristic: %d", status);
|
|
}
|
|
status = aci_gatt_del_char(hid_svc->svc_handle, hid_svc->ctrl_point_char_handle);
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to delete Control Point characteristic: %d", status);
|
|
}
|
|
status = aci_gatt_del_char(hid_svc->svc_handle, hid_svc->led_state_char_handle);
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to delete led state characteristic: %d", status);
|
|
}
|
|
// Delete service
|
|
status = aci_gatt_del_service(hid_svc->svc_handle);
|
|
if(status) {
|
|
FURI_LOG_E(TAG, "Failed to delete HID service: %d", status);
|
|
}
|
|
// Delete buffer size mutex
|
|
free(hid_svc);
|
|
hid_svc = NULL;
|
|
}
|
|
}
|