1 files changed, 61 insertions, 0 deletions
diff --git a/profiling/client/src/CommandHandler.hpp b/profiling/client/src/CommandHandler.hpp
new file mode 100644
index 0000000000..b097f9ef20
--- /dev/null
+++ b/profiling/client/src/CommandHandler.hpp
@@ -0,0 +1,61 @@
+//
+// Copyright © 2019 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include "IProfilingConnection.hpp"
+#include <common/include/PacketVersionResolver.hpp>
+
+#include <common/include/CommandHandlerRegistry.hpp>
+
+#include <atomic>
+#include <thread>
+
+namespace arm
+{
+
+namespace pipe
+{
+
+class CommandHandler
+{
+public:
+    CommandHandler(uint32_t timeout,
+                   bool stopAfterTimeout,
+                   arm::pipe::CommandHandlerRegistry& commandHandlerRegistry,
+                   arm::pipe::PacketVersionResolver& packetVersionResolver)
+        : m_Timeout(timeout),
+          m_StopAfterTimeout(stopAfterTimeout),
+          m_IsRunning(false),
+          m_KeepRunning(false),
+          m_CommandThread(),
+          m_CommandHandlerRegistry(commandHandlerRegistry),
+          m_PacketVersionResolver(packetVersionResolver)
+    {}
+    ~CommandHandler() { Stop(); }
+
+    void SetTimeout(uint32_t timeout) { m_Timeout.store(timeout); }
+    void SetStopAfterTimeout(bool stopAfterTimeout) { m_StopAfterTimeout.store(stopAfterTimeout); }
+
+    void Start(IProfilingConnection& profilingConnection);
+    void Stop();
+    bool IsRunning() const { return m_IsRunning.load(); }
+
+private:
+    void HandleCommands(IProfilingConnection& profilingConnection);
+
+    std::atomic<uint32_t> m_Timeout;
+    std::atomic<bool>     m_StopAfterTimeout;
+    std::atomic<bool>     m_IsRunning;
+    std::atomic<bool>     m_KeepRunning;
+    std::thread           m_CommandThread;
+
+    arm::pipe::CommandHandlerRegistry& m_CommandHandlerRegistry;
+    arm::pipe::PacketVersionResolver&  m_PacketVersionResolver;
+};
+
+} // namespace pipe
+
+} // namespace arm