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

#include "bluetooth_ble.hpp"

#include <atomic>
#include <mutex>
#include <cstring>
#include "../btdrv_mitm_flags.hpp"

#include "../btdrv_mitm_logging.hpp"

namespace ams::bluetooth::ble {

    namespace {

        std::atomic<bool> g_isInitialized(false);

        os::Mutex g_eventDataLock(false);
        u8 g_eventDataBuffer[0x400];
        BleEventType g_currentEventType;

        os::SystemEventType g_btBleSystemEvent;
        os::SystemEventType g_btBleSystemEventFwd;
        os::SystemEventType g_btBleSystemEventUser;

    }

    bool IsInitialized(void) {
        return g_isInitialized;
    }

    os::SystemEventType *GetSystemEvent(void) {
        return &g_btBleSystemEvent;
    }

    os::SystemEventType *GetForwardEvent(void) {
        return &g_btBleSystemEventFwd;
    }

    os::SystemEventType *GetUserForwardEvent(void) {
        return &g_btBleSystemEventUser;
    }

    Result Initialize(Handle eventHandle) {
        //os::AttachReadableHandleToSystemEvent(&g_btBleSystemEvent, eventHandle, false, os::EventClearMode_AutoClear);
        os::AttachReadableHandleToSystemEvent(&g_btBleSystemEvent, eventHandle, false, os::EventClearMode_ManualClear);

        R_TRY(os::CreateSystemEvent(&g_btBleSystemEventFwd, os::EventClearMode_AutoClear, true));
        R_TRY(os::CreateSystemEvent(&g_btBleSystemEventUser, os::EventClearMode_AutoClear, true));

        g_isInitialized = true;

        return ams::ResultSuccess();
    }

    void Finalize(void) {
        os::DestroySystemEvent(&g_btBleSystemEventUser);
        os::DestroySystemEvent(&g_btBleSystemEventFwd);

        g_isInitialized = false;
    }

    Result GetEventInfo(ncm::ProgramId program_id, BleEventType *type, u8* buffer, size_t size) {
        std::scoped_lock lk(g_eventDataLock); 
        {
            *type = g_currentEventType;
            std::memcpy(buffer, g_eventDataBuffer, size);
        }
        
        return ams::ResultSuccess();
    }

    void HandleEvent(void) {
        std::scoped_lock lk(g_eventDataLock);

        R_ABORT_UNLESS(btdrvGetBleManagedEventInfo(&g_currentEventType, g_eventDataBuffer, sizeof(g_eventDataBuffer)));

        BTDRV_LOG_FMT("[%02d] BLE Event", g_currentEventType);
        
        // Signal our forwarder events
        //if (!g_redirectEvents || g_preparingForSleep)
        if (!g_redirectEvents)
            os::SignalSystemEvent(&g_btBleSystemEventFwd);
        else
            os::SignalSystemEvent(&g_btBleSystemEventUser);
    }

}