/* SPDX-License-Identifier: MPL-2.0 */

#include "precompiled.hpp"

#include <string.h>

#include "macros.hpp"

#include "dish.hpp"

#include "err.hpp"

zmq::dish_t::dish_t (class ctx_t *parent_, uint32_t tid_, int sid_) :

    socket_base_t (parent_, tid_, sid_, true), _has_message (false)

{

    options.type = ZMQ_DISH;

    //  When socket is being closed down we don't want to wait till pending

    //  subscription commands are sent to the wire.

    options.linger.store (0);

    const int rc = _message.init ();

    errno_assert (rc == 0);

}

zmq::dish_t::~dish_t ()

{

    const int rc = _message.close ();

    errno_assert (rc == 0);

}

void zmq::dish_t::xattach_pipe (pipe_t *pipe_,

                                bool subscribe_to_all_,

                                bool locally_initiated_)

{

    LIBZMQ_UNUSED (subscribe_to_all_);

    LIBZMQ_UNUSED (locally_initiated_);

    zmq_assert (pipe_);

    _fq.attach (pipe_);

    _dist.attach (pipe_);

    //  Send all the cached subscriptions to the new upstream peer.

    send_subscriptions (pipe_);

}

void zmq::dish_t::xread_activated (pipe_t *pipe_)

{

    _fq.activated (pipe_);

}

void zmq::dish_t::xwrite_activated (pipe_t *pipe_)

{

    _dist.activated (pipe_);

}

void zmq::dish_t::xpipe_terminated (pipe_t *pipe_)

{

    _fq.pipe_terminated (pipe_);

    _dist.pipe_terminated (pipe_);

}

void zmq::dish_t::xhiccuped (pipe_t *pipe_)

{

    //  Send all the cached subscriptions to the hiccuped pipe.

    send_subscriptions (pipe_);

}

int zmq::dish_t::xjoin (const char *group_)

{

    const std::string group = std::string (group_);

    if (group.length () > ZMQ_GROUP_MAX_LENGTH) {

        errno = EINVAL;

        return -1;

    }

    //  User cannot join same group twice

    if (!_subscriptions.insert (group).second) {

        errno = EINVAL;

        return -1;

    }

    msg_t msg;

    int rc = msg.init_join ();

    errno_assert (rc == 0);

    rc = msg.set_group (group_);

    errno_assert (rc == 0);

    int err = 0;

    rc = _dist.send_to_all (&msg);

    if (rc != 0)

        err = errno;

    const int rc2 = msg.close ();

    errno_assert (rc2 == 0);

    if (rc != 0)

        errno = err;

    return rc;

}

int zmq::dish_t::xleave (const char *group_)

{

    const std::string group = std::string (group_);

    if (group.length () > ZMQ_GROUP_MAX_LENGTH) {

        errno = EINVAL;

        return -1;

    }

    if (0 == _subscriptions.erase (group)) {

        errno = EINVAL;

        return -1;

    }

    msg_t msg;

    int rc = msg.init_leave ();

    errno_assert (rc == 0);

    rc = msg.set_group (group_);

    errno_assert (rc == 0);

    int err = 0;

    rc = _dist.send_to_all (&msg);

    if (rc != 0)

        err = errno;

    const int rc2 = msg.close ();

    errno_assert (rc2 == 0);

    if (rc != 0)

        errno = err;

    return rc;

}

int zmq::dish_t::xsend (msg_t *msg_)

{

    LIBZMQ_UNUSED (msg_);

    errno = ENOTSUP;

    return -1;

}

bool zmq::dish_t::xhas_out ()

{

    //  Subscription can be added/removed anytime.

    return true;

}

int zmq::dish_t::xrecv (msg_t *msg_)

{

    //  If there's already a message prepared by a previous call to zmq_poll,

    //  return it straight ahead.

    if (_has_message) {

        const int rc = msg_->move (_message);

        errno_assert (rc == 0);

        _has_message = false;

        return 0;

    }

    return xxrecv (msg_);

}

int zmq::dish_t::xxrecv (msg_t *msg_)

{

    do {

        //  Get a message using fair queueing algorithm.

        const int rc = _fq.recv (msg_);

        //  If there's no message available, return immediately.

        //  The same when error occurs.

        if (rc != 0)

            return -1;

        //  Skip non matching messages

    } while (0 == _subscriptions.count (std::string (msg_->group ())));

    //  Found a matching message

    return 0;

}

bool zmq::dish_t::xhas_in ()

{

    //  If there's already a message prepared by a previous call to zmq_poll,

    //  return straight ahead.

    if (_has_message)

        return true;

    const int rc = xxrecv (&_message);

    if (rc != 0) {

        errno_assert (errno == EAGAIN);

        return false;

    }

    //  Matching message found

    _has_message = true;

    return true;

}

void zmq::dish_t::send_subscriptions (pipe_t *pipe_)

{

    for (subscriptions_t::iterator it = _subscriptions.begin (),

                                   end = _subscriptions.end ();

         it != end; ++it) {

        msg_t msg;

        int rc = msg.init_join ();

        errno_assert (rc == 0);

        rc = msg.set_group (it->c_str ());

        errno_assert (rc == 0);

        //  Send it to the pipe.

        pipe_->write (&msg);

    }

    pipe_->flush ();

}

zmq::dish_session_t::dish_session_t (io_thread_t *io_thread_,

                                     bool connect_,

                                     socket_base_t *socket_,

                                     const options_t &options_,

                                     address_t *addr_) :

    session_base_t (io_thread_, connect_, socket_, options_, addr_),

    _state (group)

{

}

zmq::dish_session_t::~dish_session_t ()

{

}

int zmq::dish_session_t::push_msg (msg_t *msg_)

{

    if (_state == group) {

        if ((msg_->flags () & msg_t::more) != msg_t::more) {

            errno = EFAULT;

            return -1;

        }

        if (msg_->size () > ZMQ_GROUP_MAX_LENGTH) {

            errno = EFAULT;

            return -1;

        }

        _group_msg = *msg_;

        _state = body;

        const int rc = msg_->init ();

        errno_assert (rc == 0);

        return 0;

    }

    const char *group_setting = msg_->group ();

    int rc;

    if (group_setting[0] != 0)

        goto has_group;

    //  Set the message group

    rc = msg_->set_group (static_cast<char *> (_group_msg.data ()),

                          _group_msg.size ());

    errno_assert (rc == 0);

    //  We set the group, so we don't need the group_msg anymore

    rc = _group_msg.close ();

    errno_assert (rc == 0);

has_group:

    //  Thread safe socket doesn't support multipart messages

    if ((msg_->flags () & msg_t::more) == msg_t::more) {

        errno = EFAULT;

        return -1;

    }

    //  Push message to dish socket

    rc = session_base_t::push_msg (msg_);

    if (rc == 0)

        _state = group;

    return rc;

}

int zmq::dish_session_t::pull_msg (msg_t *msg_)

{

    int rc = session_base_t::pull_msg (msg_);

    if (rc != 0)

        return rc;

    if (!msg_->is_join () && !msg_->is_leave ())

        return rc;

    const int group_length = static_cast<int> (strlen (msg_->group ()));

    msg_t command;

    int offset;

    if (msg_->is_join ()) {

        rc = command.init_size (group_length + 5);

        errno_assert (rc == 0);

        offset = 5;

        memcpy (command.data (), "\4JOIN", 5);

    } else {

        rc = command.init_size (group_length + 6);

        errno_assert (rc == 0);

        offset = 6;

        memcpy (command.data (), "\5LEAVE", 6);

    }

    command.set_flags (msg_t::command);

    char *command_data = static_cast<char *> (command.data ());

    //  Copy the group

    memcpy (command_data + offset, msg_->group (), group_length);

    //  Close the join message

    rc = msg_->close ();

    errno_assert (rc == 0);

    *msg_ = command;

    return 0;

}

void zmq::dish_session_t::reset ()

{

    session_base_t::reset ();

    _state = group;

}