mirror of
https://github.com/gnif/LookingGlass.git
synced 2026-08-22 07:01:30 +00:00
aea73cd2b3
Create and advertise one reliable input stream for each possible LGMP client, then bind endpoints to active input subscribers. Drain endpoints fairly while prioritizing the current owner's lane so its RELEASE remains the ordering barrier. Reject transport changes within an activation generation. Gracefully retire missing or stalled subscribers without reusing an active reservation. Keep excess clients on the queue and release HID ownership from the input worker when its stream retires.
147 lines
4.2 KiB
C++
147 lines
4.2 KiB
C++
/**
|
|
* Looking Glass
|
|
* Copyright © 2017-2026 The Looking Glass Authors
|
|
* https://looking-glass.io
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License as published by the Free
|
|
* Software Foundation; either version 2 of the License, or (at your option)
|
|
* any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
* more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc., 59
|
|
* Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "Atomic.h"
|
|
#include "CSRWLock.h"
|
|
#include "transport/IInputSource.h"
|
|
#include "common/KVMFRInput.h"
|
|
#include "common/LGMPConfig.h"
|
|
|
|
#include <Windows.h>
|
|
|
|
#include <stdint.h>
|
|
|
|
extern "C" {
|
|
#include "lgmp/host.h"
|
|
#include "lgmp/stream.h"
|
|
}
|
|
|
|
class CLGMPHost;
|
|
class IInputTarget;
|
|
|
|
class CLGMPInputTransport final : public IInputSource
|
|
{
|
|
private:
|
|
static constexpr ULONGLONG OWNER_LEASE_MS = 500;
|
|
static constexpr ULONGLONG ACTIVE_POLL_MS = 50;
|
|
static constexpr ULONGLONG LOG_INTERVAL_MS = 5000;
|
|
static constexpr unsigned DRAIN_LIMIT = 256;
|
|
|
|
enum class MessageTransport : uint8_t
|
|
{
|
|
NONE,
|
|
QUEUE,
|
|
STREAM,
|
|
};
|
|
|
|
struct StreamEndpoint
|
|
{
|
|
PLGMPHostStream stream;
|
|
LGMPStreamDescriptor descriptor;
|
|
uint32_t clientID;
|
|
uint32_t epoch;
|
|
bool draining;
|
|
};
|
|
|
|
struct Statistics
|
|
{
|
|
ULONGLONG lastLog;
|
|
uint64_t messages;
|
|
uint64_t reports;
|
|
uint64_t drainLimit;
|
|
uint64_t malformedSize;
|
|
uint64_t malformedMessage;
|
|
uint64_t sequenceErrors;
|
|
uint64_t nonOwner;
|
|
uint64_t deliveryFailures;
|
|
uint64_t claims;
|
|
uint64_t releases;
|
|
unsigned maxDrain;
|
|
};
|
|
|
|
CLGMPHost& m_host;
|
|
|
|
PLGMPHostQueue m_queue = nullptr;
|
|
PLGMPMemory m_statusMemory[LGMP_Q_INPUT_LEN] = {};
|
|
StreamEndpoint m_streamEndpoint[
|
|
KVMFR_INPUT_STREAM_ENDPOINT_COUNT] = {};
|
|
IInputTarget * m_target = nullptr;
|
|
|
|
CSRWLock m_lifecycleLock;
|
|
CSRWLock m_statusLock;
|
|
CSRWLock m_streamLock;
|
|
HANDLE m_stopEvent = nullptr;
|
|
HANDLE m_pollTimer = nullptr;
|
|
HANDLE m_thread = nullptr;
|
|
|
|
uint32_t m_ownerClientID = 0;
|
|
uint32_t m_ownerGeneration = 0;
|
|
uint32_t m_ownerSequence = 0;
|
|
MessageTransport m_ownerTransport = MessageTransport::NONE;
|
|
ULONGLONG m_ownerDeadline = 0;
|
|
InputTargetState m_targetState;
|
|
uint32_t m_endpointGeneration = 0;
|
|
uint32_t m_streamGeneration = 0;
|
|
unsigned m_streamDrainCursor = 0;
|
|
uint32_t m_statusSerial = 0;
|
|
bool m_statusDirty = false;
|
|
std::atomic<bool> m_statusFailed = false;
|
|
Statistics m_statistics = {};
|
|
|
|
bool Initialize();
|
|
void DeInit();
|
|
bool ReconcileStreams(bool& changed);
|
|
void ResetStreams();
|
|
InputSourceId Owner() const;
|
|
void UpdateTargetState(const InputTargetState& state);
|
|
bool PublishStatus();
|
|
void FlushStatus();
|
|
void LogStatistics(ULONGLONG now);
|
|
bool DrainQueueMessages(bool& received);
|
|
bool DrainStreamMessages(bool& received);
|
|
bool ProcessMessage(uint32_t sourceClientID,
|
|
const KVMFRInputMessage& message, MessageTransport transport);
|
|
bool ValidatePayload(const KVMFRInputMessage& message) const;
|
|
bool IsOwner(uint32_t sourceClientID, uint32_t generation) const;
|
|
bool Claim(uint32_t sourceClientID,
|
|
const KVMFRInputMessage& message, MessageTransport transport);
|
|
void RenewLease();
|
|
void ReleaseOwner(bool reset);
|
|
void CheckOwner();
|
|
void Thread();
|
|
|
|
static DWORD CALLBACK ThreadProc(void * context);
|
|
|
|
friend class CLGMPTransport;
|
|
|
|
public:
|
|
explicit CLGMPInputTransport(CLGMPHost& host) :
|
|
m_host(host) {}
|
|
~CLGMPInputTransport() override;
|
|
|
|
CLGMPInputTransport(const CLGMPInputTransport&) = delete;
|
|
CLGMPInputTransport& operator=(const CLGMPInputTransport&) = delete;
|
|
|
|
bool Start(IInputTarget& target) override;
|
|
void Stop() override;
|
|
};
|