ZyreNode.h

Go to the documentation of this file.

#pragma once

#include <swarmio/transport/BasicNode.h>
#include <string>
#include <sstream>
#include <atomic>

namespace swarmio::transport::zyre
{
    class ZyreEndpoint;

    class SWARMIO_API ZyreNode final : public BasicNode
    {
        private:

            friend ZyreEndpoint;

            unsigned char _message_counter[24];

            unsigned char _key[32];

            unsigned char _pubkey[32];

            std::string _address;

            std::atomic<bool> _online;

            std::atomic<bool> _verified;

            void SetOnline(bool online)
            {
                _online = online;
            }

        public:

            ZyreNode(const std::string& uuid, const std::string& name, const std::string& deviceClass, const std::string& address)
                : BasicNode(uuid, name, deviceClass), _address(address), _online(false) { }

            bool IsOnline() const override
            {
                return _online;
            }

            const std::string& GetAddress() const
            {
                return _address;
            }

            const unsigned char* GetKey() const
            {
                return _key;
            }

            const unsigned char* GetPubKey() const
            {
                return _pubkey;
            }

            void SetKeys(unsigned char* k, unsigned char* pk)
            {
                memcpy(_key, k, 32);
                memcpy(_pubkey, pk, 32);
            }

            const unsigned char* GetCtr() const
            {
                return _message_counter;
            }

            const bool GetVerified() const
            {
                return _verified;
            }
        
            void SetVerified()
            {
                _verified = true;
            }

            void IncrementCtr() const
            {
                for (int i = 23; i >= 0; i--) if (++const_cast <ZyreNode*>(this)->_message_counter[i]) break;
            }

            void SetCtr(unsigned char* c) const
            {
                memcpy(const_cast <ZyreNode*>(this)->_message_counter, c, 24);
            }

            virtual std::string GetDescription() const override
            {
                std::ostringstream stream;
                stream << "online=" << (_online ? "true" : "false") << ", "
                       << "address=" << _address;
                return stream.str();
            }
    };
}