nrf51822/source/nRF5xCharacteristicDescriptorDiscoverer.cpp

#include "nRF5xCharacteristicDescriptorDiscoverer.h"
#include "ble_err.h"
#include "mbed-drivers/mbed_error.h"
#include "ble/DiscoveredCharacteristicDescriptor.h"



namespace { 
    void emptyDiscoveryCallback(const CharacteristicDescriptorDiscovery::DiscoveryCallbackParams_t*) { }
    void emptyTerminationCallback(const CharacteristicDescriptorDiscovery::TerminationCallbackParams_t*) { }
}

nRF5xCharacteristicDescriptorDiscoverer::nRF5xCharacteristicDescriptorDiscoverer(size_t concurrentConnectionsCount) : 
    maximumConcurrentConnectionsCount(concurrentConnectionsCount), 
    discoveryRunning(new Discovery[concurrentConnectionsCount]) {

}

nRF5xCharacteristicDescriptorDiscoverer::~nRF5xCharacteristicDescriptorDiscoverer() { 
    delete [] discoveryRunning;
}

ble_error_t nRF5xCharacteristicDescriptorDiscoverer::launch(
    const DiscoveredCharacteristic& characteristic, 
    const CharacteristicDescriptorDiscovery::DiscoveryCallback_t& discoveryCallback,
    const CharacteristicDescriptorDiscovery::TerminationCallback_t& terminationCallback
) {
    Gap::Handle_t connHandle = characteristic.getConnectionHandle();
    // it is ok to deduce that the start handle for descriptors is after 
    // the characteristic declaration and the characteristic value declaration
    // see BLUETOOTH SPECIFICATION Version 4.2 [Vol 3, Part G] (3.3)
    Gap::Handle_t descriptorStartHandle = characteristic.getDeclHandle() + 2;
    Gap::Handle_t descriptorEndHandle = characteristic.getLastHandle();

    // check if their is any descriptor to discover
    if (descriptorEndHandle < descriptorStartHandle) { 
        CharacteristicDescriptorDiscovery::TerminationCallbackParams_t termParams = { 
            characteristic
        };
        terminationCallback.call(&termParams);
        return BLE_ERROR_NONE;
    }

    // check if we can run this discovery
    if (isConnectionInUse(connHandle)) { 
        return BLE_STACK_BUSY;
    }

    // get a new discovery slot, if none are available, just return
    Discovery* discovery = getAvailableDiscoverySlot();
    if(discovery == NULL) {
        return BLE_STACK_BUSY; 
    }

    // try to launch the discovery 
    ble_gattc_handle_range_t discoveryRange = {
        descriptorStartHandle,
        descriptorEndHandle
    };
    uint32_t err = sd_ble_gattc_descriptors_discover(characteristic.getConnectionHandle(), &discoveryRange);
    switch(err) { 
        case NRF_SUCCESS:
            // commit the new discovery to its slot
            *discovery = Discovery(
                characteristic,
                discoveryCallback, 
                terminationCallback
            );

            return BLE_ERROR_NONE;            
        case BLE_ERROR_INVALID_CONN_HANDLE:
            return BLE_ERROR_INVALID_PARAM;
        case NRF_ERROR_INVALID_ADDR:
            return BLE_ERROR_PARAM_OUT_OF_RANGE;
        case NRF_ERROR_BUSY:
            return BLE_STACK_BUSY;
        default:
            return BLE_ERROR_UNSPECIFIED;
    }
}

bool nRF5xCharacteristicDescriptorDiscoverer::isActive(const DiscoveredCharacteristic& characteristic) const {
    return findRunningDiscovery(characteristic) != NULL;
}

void nRF5xCharacteristicDescriptorDiscoverer::requestTerminate(const DiscoveredCharacteristic& characteristic) {
    Discovery* discovery = findRunningDiscovery(characteristic);
    if(discovery) { 
        discovery->onDiscovery = emptyDiscoveryCallback;
        // call terminate anyway
        discovery->terminate();
        discovery->onTerminate = emptyTerminationCallback;
    }
}

void nRF5xCharacteristicDescriptorDiscoverer::process(uint16_t handle, const ble_gattc_evt_desc_disc_rsp_t& descriptors) {
    Discovery* discovery = findRunningDiscovery(handle);
    if(!discovery) { 
        error("logic error in nRF5xCharacteristicDescriptorDiscoverer::process !!!");
    }

    for (uint16_t i = 0; i < descriptors.count; ++i) {
        discovery->process(
            descriptors.descs[i].handle, UUID(descriptors.descs[i].uuid.uuid)
        );
    }
}

void nRF5xCharacteristicDescriptorDiscoverer::terminate(uint16_t handle) {
    Discovery* discovery = findRunningDiscovery(handle);
    if(!discovery) { 
        error("logic error in nRF5xCharacteristicDescriptorDiscoverer::process !!!");
    }
    discovery->terminate();
    removeDiscovery(discovery);
}

nRF5xCharacteristicDescriptorDiscoverer::Discovery* 
nRF5xCharacteristicDescriptorDiscoverer::findRunningDiscovery(const DiscoveredCharacteristic& characteristic) {
    for(size_t i = 0; i < maximumConcurrentConnectionsCount; ++i) { 
        if(discoveryRunning[i].characteristic == characteristic) { 
            return &discoveryRunning[i];
        }
    }
    return NULL;
}

nRF5xCharacteristicDescriptorDiscoverer::Discovery* 
nRF5xCharacteristicDescriptorDiscoverer::findRunningDiscovery(const DiscoveredCharacteristic& characteristic) const {
    for(size_t i = 0; i < maximumConcurrentConnectionsCount; ++i) { 
        if(discoveryRunning[i].characteristic == characteristic) { 
            return &discoveryRunning[i];
        }
    }
    return NULL;
}

nRF5xCharacteristicDescriptorDiscoverer::Discovery* 
nRF5xCharacteristicDescriptorDiscoverer::findRunningDiscovery(uint16_t handle) {
    for(size_t i = 0; i < maximumConcurrentConnectionsCount; ++i) { 
        if(discoveryRunning[i].characteristic.getConnectionHandle() == handle) { 
            return &discoveryRunning[i];
        }
    }
    return NULL;
}     

void nRF5xCharacteristicDescriptorDiscoverer::removeDiscovery(Discovery* discovery) {
    for(size_t i = 0; i < maximumConcurrentConnectionsCount; ++i) { 
        if(&discoveryRunning[i] == discovery) { 
            discoveryRunning[i] = Discovery();
        }
    }
}

nRF5xCharacteristicDescriptorDiscoverer::Discovery* 
nRF5xCharacteristicDescriptorDiscoverer::getAvailableDiscoverySlot() {
    for(size_t i = 0; i < maximumConcurrentConnectionsCount; ++i) { 
        if(discoveryRunning[i] == Discovery()) {
            return &discoveryRunning[i];
        }
    }     
    return NULL;
}

bool nRF5xCharacteristicDescriptorDiscoverer::isConnectionInUse(uint16_t connHandle) {
     return findRunningDiscovery(connHandle) != NULL;
}