MissionControl/btdrv-mitm/source/bluetooth/bluetooth_hid_events.cpp

85 lines
2.8 KiB
C++

#include "bluetooth_hid_events.hpp"
#include "../controllermanager.hpp"
#include "../btdrv_mitm_flags.hpp"
#include "../btdrv_mitm_logging.hpp"
ams::os::ThreadType g_bt_hid_event_task_thread;
alignas(ams::os::ThreadStackAlignment) u8 g_bt_hid_event_task_stack[0x2000];
u8 g_bt_hid_event_data_buffer[0x480];
HidEventType g_current_bt_hid_event_type;
ams::os::SystemEventType g_btHidSystemEvent;
ams::os::SystemEventType g_btHidSystemEventFwd;
ams::os::SystemEventType g_btHidSystemEventUser;
void handleConnectionStateEvent(HidEventData *eventData) {
switch (eventData->connectionState.state) {
case HidConnectionState_Connected:
ams::mitm::btdrv::attachDeviceHandler(&eventData->connectionState.address);
BTDRV_LOG_FMT("device connected");
break;
case HidConnectionState_Disconnected:
ams::mitm::btdrv::removeDeviceHandler(&eventData->connectionState.address);
BTDRV_LOG_FMT("device disconnected");
break;
default:
break;
}
BTDRV_LOG_DATA(&eventData->connectionState.address, sizeof(BluetoothAddress));
}
void BluetoothHidEventThreadFunc(void *arg) {
HidEventData *eventData = reinterpret_cast<HidEventData *>(g_bt_hid_event_data_buffer);
while (true) {
// Wait for real bluetooth event
ams::os::WaitSystemEvent(&g_btHidSystemEvent);
BTDRV_LOG_FMT("hid event fired");
R_ABORT_UNLESS(btdrvGetHidEventInfo(&g_current_bt_hid_event_type, g_bt_hid_event_data_buffer, sizeof(g_bt_hid_event_data_buffer)));
switch (g_current_bt_hid_event_type) {
case HidEvent_ConnectionState:
handleConnectionStateEvent(eventData);
break;
default:
break;
}
// Signal our forwarder events
//os::SignalSystemEvent(&g_btHidSystemEventFwd);
if (!g_redirectEvents || g_preparingForSleep) {
ams::os::SignalSystemEvent(&g_btHidSystemEventFwd);
}
else {
ams::os::SignalSystemEvent(&g_btHidSystemEventUser);
}
}
}
ams::Result InitializeBluetoothHidEvents(void) {
R_TRY(ams::os::CreateSystemEvent(&g_btHidSystemEventFwd, ams::os::EventClearMode_AutoClear, true));
R_TRY(ams::os::CreateSystemEvent(&g_btHidSystemEventUser, ams::os::EventClearMode_AutoClear, true));
return ams::ResultSuccess();
}
ams::Result StartBluetoothHidEventThread(void) {
R_TRY(ams::os::CreateThread(&g_bt_hid_event_task_thread,
BluetoothHidEventThreadFunc,
nullptr,
g_bt_hid_event_task_stack,
sizeof(g_bt_hid_event_task_stack),
9
//38 // priority of btm sysmodule + 1
));
ams::os::StartThread(&g_bt_hid_event_task_thread);
return ams::ResultSuccess();
}