ZyreEndpoint.h

Go to the documentation of this file.

#pragma once

#include <swarmio/transport/BasicEndpoint.h>
#include <swarmio/transport/zyre/ZyreNode.h>
#include <swarmio/transport/zyre/ZyreControlSocket.h>
#include <g3log/g3log.hpp>
#include <readerwriterqueue.h>
#include <zyre.h>
#include <map>
#include <list>
#include <thread>
#include <shared_mutex>
#include <sodium.h>
#define CONTEXT "CPSwarm"

namespace swarmio::transport::zyre
{
class SWARMIO_API ZyreEndpoint : public BasicEndpoint
{
private:
    zyre_t *_zyre;

    ZyreControlSocket _control;

    std::map<std::string, ZyreNode> _nodes;

    std::shared_timed_mutex _mutex;

    std::thread *_worker = nullptr;

    std::string _uuid;

    bool security_enabled = false;
    // Keys
    unsigned char my_publickey[crypto_box_PUBLICKEYBYTES];
    unsigned char my_secretkey[crypto_box_SECRETKEYBYTES];
    unsigned char bcast_publickey[crypto_box_PUBLICKEYBYTES];
    unsigned char bcast_secretkey[crypto_box_SECRETKEYBYTES];
    unsigned char signature[crypto_sign_SECRETKEYBYTES];
    unsigned char server_public[crypto_sign_PUBLICKEYBYTES];
    unsigned char certificate[crypto_box_PUBLICKEYBYTES + crypto_sign_SECRETKEYBYTES];

    bool isJoining = false;
    std::string configFilePath = "config.cfg";
    void Process();

    void Deliver(moodycamel::BlockingReaderWriterQueue<zyre_event_t *> *queue);

protected:
    virtual void Send(const void *data, size_t size, const Node *node) override;

public:
    ZyreEndpoint(const char *name, const char *deviceClass);

    void SetPort(uint16_t port);

    void SetInterface(const char *ifname);

    void SetConfig(std::string cfg);

    virtual void Start() override;

    virtual void Stop() override;

    virtual ~ZyreEndpoint() override;

    virtual const std::string &GetUUID() override
    {
        return _uuid;
    }

    std::list<const ZyreNode *> GetNodes();

    virtual const Node *NodeForUUID(const std::string &uuid) override;
};
} // namespace swarmio::transport::zyre