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();

        virtual void output(string_ref 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, string_ref message) override;
        void update(CliComm::UpdateType type, string_ref machine,
                    string_ref name, string_ref 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();

        void output(string_ref message) override;

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

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

        void output(string_ref 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();

        void output(string_ref message) override;

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

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

} // namespace openmsx

#endif