socket_base.hpp 6.2 KB
Newer Older
1
/*
2
    Copyright (c) 2007-2010 iMatix Corporation
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22

    This file is part of 0MQ.

    0MQ is free software; you can redistribute it and/or modify it under
    the terms of the Lesser GNU General Public License as published by
    the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.

    0MQ is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    Lesser GNU General Public License for more details.

    You should have received a copy of the Lesser GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef __ZMQ_SOCKET_BASE_HPP_INCLUDED__
#define __ZMQ_SOCKET_BASE_HPP_INCLUDED__

23
#include <set>
24
#include <map>
Martin Sustrik's avatar
Martin Sustrik committed
25
#include <vector>
Martin Sustrik's avatar
Martin Sustrik committed
26
#include <string>
27

28 29
#include "../bindings/c/zmq.h"

Martin Sustrik's avatar
Martin Sustrik committed
30
#include "i_endpoint.hpp"
31
#include "object.hpp"
32
#include "yarray_item.hpp"
33
#include "mutex.hpp"
34
#include "options.hpp"
Martin Sustrik's avatar
Martin Sustrik committed
35
#include "stdint.hpp"
36
#include "atomic_counter.hpp"
37
#include "stdint.hpp"
38 39 40 41

namespace zmq
{

42 43
    class socket_base_t :
        public object_t, public i_endpoint, public yarray_item_t
44 45 46
    {
    public:

47
        socket_base_t (class app_thread_t *parent_);
48

49
        //  Interface for communication with the API layer.
50
        int setsockopt (int option_, const void *optval_,
51
            size_t optvallen_);
52 53
        int bind (const char *addr_);
        int connect (const char *addr_);
54
        int send (zmq_msg_t *msg_, int flags_);
55
        int flush ();
56
        int recv (zmq_msg_t *msg_, int flags_);
57
        int close ();
58

59 60 61 62 63
        //  When another owned object wants to send command to this object
        //  it calls this function to let it know it should not shut down
        //  before the command is delivered.
        void inc_seqnum ();

64 65 66 67 68 69 70 71 72 73
        //  This function is used by the polling mechanism to determine
        //  whether the socket belongs to the application thread the poll
        //  is called from.
        class app_thread_t *get_thread ();

        //  These functions are used by the polling mechanism to determine
        //  which events are to be reported from this socket.
        bool has_in ();
        bool has_out ();

74
        //  The list of sessions cannot be accessed via inter-thread
75 76
        //  commands as it is unacceptable to wait for the completion of the
        //  action till user application yields control of the application
77
        //  thread to 0MQ. Locking is used instead.
78 79 80
        //  There are two distinct types of sessions: those identified by name
        //  and those identified by ordinal number. Thus two sets of session
        //  management functions.
81
        bool register_session (const char *name_, class session_t *session_);
82
        void unregister_session (const char *name_);
83
        class session_t *find_session (const char *name_);
84 85 86
        uint64_t register_session (class session_t *session_);
        void unregister_session (uint64_t ordinal_);
        class session_t *find_session (uint64_t ordinal_);
87

Martin Sustrik's avatar
Martin Sustrik committed
88
        //  i_endpoint interface implementation.
89
        void attach_pipes (class reader_t *inpipe_, class writer_t *outpipe_);
Martin Sustrik's avatar
Martin Sustrik committed
90 91
        void detach_inpipe (class reader_t *pipe_);
        void detach_outpipe (class writer_t *pipe_);
92 93
        void kill (class reader_t *pipe_);
        void revive (class reader_t *pipe_);
Martin Sustrik's avatar
Martin Sustrik committed
94

95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110
    protected:

        //  Destructor is protected. Socket is closed using 'close' function.
        virtual ~socket_base_t ();

        //  Pipe management is done by individual socket types.
        virtual void xattach_pipes (class reader_t *inpipe_,
            class writer_t *outpipe_) = 0;
        virtual void xdetach_inpipe (class reader_t *pipe_) = 0;
        virtual void xdetach_outpipe (class writer_t *pipe_) = 0;
        virtual void xkill (class reader_t *pipe_) = 0;
        virtual void xrevive (class reader_t *pipe_) = 0;

        //  Actual algorithms are to be defined by individual socket types.
        virtual int xsetsockopt (int option_, const void *optval_,
            size_t optvallen_) = 0;
111
        virtual int xsend (zmq_msg_t *msg_, int options_) = 0;
112
        virtual int xflush () = 0;
113
        virtual int xrecv (zmq_msg_t *msg_, int options_) = 0;
114 115
        virtual bool xhas_in () = 0;
        virtual bool xhas_out () = 0;
Martin Sustrik's avatar
Martin Sustrik committed
116

117 118 119
        //  Socket options.
        options_t options;

120 121 122
    private:

        //  Handlers for incoming commands.
123
        void process_own (class owned_t *object_);
124
        void process_bind (class reader_t *in_pipe_, class writer_t *out_pipe_);
125
        void process_term_req (class owned_t *object_);
126
        void process_term_ack ();
127
        void process_seqnum ();
128 129 130

        //  List of all I/O objects owned by this socket. The socket is
        //  responsible for deallocating them before it quits.
131
        typedef std::set <class owned_t*> io_objects_t;
132 133 134 135 136 137
        io_objects_t io_objects;

        //  Number of I/O objects that were already asked to terminate
        //  but haven't acknowledged it yet.
        int pending_term_acks;

Martin Sustrik's avatar
Martin Sustrik committed
138 139 140
        //  Number of messages received since last command processing.
        int ticks;

141 142 143
        //  Application thread the socket lives in.
        class app_thread_t *app_thread;

144 145 146 147
        //  If true, socket is already shutting down. No new work should be
        //  started.
        bool shutting_down;

148 149 150 151 152 153
        //  Sequence number of the last command sent to this object.
        atomic_counter_t sent_seqnum;

        //  Sequence number of the last command processed by this object.
        uint64_t processed_seqnum;

154 155 156 157 158 159 160 161 162
        //  Lists of existing sessions. This lists are never referenced from
        //  within the socket, instead they are used by I/O objects owned by
        //  the socket. As those objects can live in different threads,
        //  the access is synchronised by mutex.
        typedef std::map <std::string, session_t*> named_sessions_t;
        named_sessions_t named_sessions;
        typedef std::map <uint64_t, session_t*> unnamed_sessions_t;
        unnamed_sessions_t unnamed_sessions;
        uint64_t next_ordinal;
163 164
        mutex_t sessions_sync;

165 166 167 168 169 170 171
        socket_base_t (const socket_base_t&);
        void operator = (const socket_base_t&);
    };

}

#endif