logiops/src/logid/DeviceFinder.cpp

#include <hid/DeviceMonitor.h>
#include <hidpp/SimpleDispatcher.h>
#include <hidpp/Device.h>
#include <hidpp10/Error.h>
#include <hidpp10/IReceiver.h>
#include <hidpp20/Error.h>
#include <cstring>
#include <unistd.h>
#include <thread>
#include <fstream>
#include <sstream>

#include "DeviceFinder.h"
#include "util.h"
#include "Device.h"

void stopAndDeletePairedDevice (PairedDevice &pairedDevice)
{
    log_printf(INFO, "%s (Device %d on %s) disconnected", pairedDevice.device->name.c_str(), pairedDevice.device->index, pairedDevice.device->path.c_str()); 
    pairedDevice.device->stop();
    pairedDevice.associatedThread.join();
    delete(pairedDevice.device);
}

DeviceFinder::~DeviceFinder()
{
    this->devicesMutex.lock();
        for (auto it = this->devices.begin(); it != this->devices.end(); it++) {
            for (auto jt = it->second.begin(); jt != it->second.end(); jt++) {
                stopAndDeletePairedDevice(jt->second);
            }
        }
    this->devicesMutex.unlock();
}

void DeviceFinder::insertNewDevice(const std::string &path, HIDPP::DeviceIndex index)
{
    Device *device = new Device(path, index);

    this->devicesMutex.lock();
        log_printf(INFO, "%s detected: device %d on %s", device->name.c_str(), index, path.c_str());
        auto pathBucket = this->devices.emplace(path, std::map<HIDPP::DeviceIndex, PairedDevice>()).first;
        pathBucket->second.emplace(index, PairedDevice{
            device,
            std::thread([device]() {
                device->start();
            })
        });
    this->devicesMutex.unlock();
}

void DeviceFinder::stopAndDeleteAllDevicesIn (const std::string &path)
{
    this->devicesMutex.lock();
        auto pathBucket = this->devices.find(path);
        if (pathBucket != this->devices.end())
        {
            for (auto& indexBucket : pathBucket->second) {
                stopAndDeletePairedDevice(indexBucket.second);
            }
            this->devices.erase(pathBucket);
        }
    this->devicesMutex.unlock();

    log_printf(WARN, "Attempted to disconnect not previously connected devices on %s", path.c_str());
}

void DeviceFinder::stopAndDeleteDevice (const std::string &path, HIDPP::DeviceIndex index)
{
    this->devicesMutex.lock();
        auto pathBucket = this->devices.find(path);
        if (pathBucket != this->devices.end())
        {
            auto indexBucket = pathBucket->second.find(index);
            if (indexBucket != pathBucket->second.end())
            {
                stopAndDeletePairedDevice(indexBucket->second);
                pathBucket->second.erase(indexBucket);
            }
        }
    this->devicesMutex.unlock();

    log_printf(WARN, "Attempted to disconnect not previously connected device %d on %s", index, path.c_str());
}
    

void DeviceFinder::addDevice(const char *path)
{
    std::string string_path(path);
    // Asynchronously scan device
    std::thread{[=]()
    {
        const int max_tries = 10;
        const int try_delay = 250000;

        //Check if device is an HID++ device and handle it accordingly
        try
        {
            HIDPP::SimpleDispatcher dispatcher(string_path.c_str());
            bool has_receiver_index = false;
            for(HIDPP::DeviceIndex index: {
                HIDPP::DefaultDevice, HIDPP::CordedDevice,
                HIDPP::WirelessDevice1, HIDPP::WirelessDevice2,
                HIDPP::WirelessDevice3, HIDPP::WirelessDevice4,
                HIDPP::WirelessDevice5, HIDPP::WirelessDevice6})
            {

                if(!has_receiver_index && index == HIDPP::WirelessDevice1)
                    break;
                for(int i = 0; i < max_tries; i++)
                {
                    try
                    {
                        HIDPP::Device d(&dispatcher, index);
                        auto version = d.protocolVersion();
                        uint major, minor;
                        std::tie(major, minor) = version;
                        if(index == HIDPP::DefaultDevice && version == std::make_tuple(1, 0))
                            has_receiver_index = true;
                        if(major > 1) // HID++ 2.0 devices only
                        {
                            this->insertNewDevice(string_path, index);
                        }
                        break;
                    }
                    catch(HIDPP10::Error &e)
                    {
                        if(e.errorCode() != HIDPP10::Error::UnknownDevice)
                        {
                            if(i == max_tries - 1)
                                log_printf(ERROR, "Error while querying %s, wireless device %d: %s", string_path.c_str(), index, e.what());
                            else usleep(try_delay);
                        }
                        else break;
                    }
                    catch(HIDPP20::Error &e)
                    {
                        if(e.errorCode() != HIDPP20::Error::UnknownDevice)
                        {
                            if(i == max_tries - 1)
                                log_printf(ERROR, "Error while querying %s, device %d: %s", string_path.c_str(), index, e.what());
                            else usleep(try_delay);
                        }
                        else break;
                    }
                    catch(HIDPP::Dispatcher::TimeoutError &e)
                    {
                        if(i == max_tries - 1)
                            log_printf(ERROR, "Device %s (index %d) timed out.", string_path.c_str(), index);
                        else usleep(try_delay);
                    }
                    catch(std::runtime_error &e)
                    {
                        if(i == max_tries - 1)
                            log_printf(ERROR, "Runtime error on device %d on %s: %s", index, string_path.c_str(), e.what());
                        else usleep(try_delay);
                    }
                }
            }
        }
        catch(HIDPP::Dispatcher::NoHIDPPReportException &e) { }
        catch(std::system_error &e) { log_printf(WARN, "Failed to open %s: %s", string_path.c_str(), e.what()); }

    }}.detach();
}

void DeviceFinder::removeDevice(const char* path)
{
    this->stopAndDeleteAllDevicesIn(std::string(path));
}