CliConnection.hh

Go to the documentation of this file.

#ifndef CLICONNECTION_HH
#define CLICONNECTION_HH

#include "CliListener.hh"
#include "EventListener.hh"
#include "Socket.hh"
#include "CliComm.hh"
#include "AdhocCliCommParser.hh"
#include "Poller.hh"
#include <mutex>
#include <string>
#include <thread>

namespace openmsx {

class CommandController;
class EventDistributor;

class CliConnection : public CliListener, private EventListener
{
public:
        void setUpdateEnable(CliComm::UpdateType type, bool value) {
                updateEnabled[type] = value;
        }
        bool getUpdateEnable(CliComm::UpdateType type) const {
                return updateEnabled[type];
        }

        void start();

protected:
        CliConnection(CommandController& commandController,
                      EventDistributor& eventDistributor);
        ~CliConnection() override;

        virtual void output(std::string_view message) = 0;

        void end();

        virtual void close() = 0;

        void startOutput();

        AdhocCliCommParser parser;
        Poller poller;

private:
        virtual void run() = 0;

        void execute(const std::string& command);

        // CliListener
        void log(CliComm::LogLevel level, std::string_view message) override;
        void update(CliComm::UpdateType type, std::string_view machine,
                    std::string_view name, std::string_view value) override;

        // EventListener
        int signalEvent(const std::shared_ptr<const Event>& event) override;

        CommandController& commandController;
        EventDistributor& eventDistributor;

        std::thread thread;

        bool updateEnabled[CliComm::NUM_UPDATES];
};

class StdioConnection final : public CliConnection
{
public:
        StdioConnection(CommandController& commandController,
                        EventDistributor& eventDistributor);
        ~StdioConnection() override;

        void output(std::string_view message) override;

private:
        void close() override;
        void run() override;
};

#ifdef _WIN32
class PipeConnection final : public CliConnection
{
public:
        PipeConnection(CommandController& commandController,
                       EventDistributor& eventDistributor,
                       std::string_view name);
        ~PipeConnection() override;

        void output(std::string_view message) override;

private:
        void close() override;
        void run() override;

        HANDLE pipeHandle;
        HANDLE shutdownEvent;
};
#endif

class SocketConnection final : public CliConnection
{
public:
        SocketConnection(CommandController& commandController,
                         EventDistributor& eventDistributor,
                         SOCKET sd);
        ~SocketConnection() override;

        void output(std::string_view message) override;

private:
        void close() override;
        void run() override;
        void closeSocket();

        std::mutex sdMutex;
        SOCKET sd;
        bool established;
};

} // namespace openmsx

#endif