mechanism.hpp 4.89 KB
Newer Older
1
/*
2
    Copyright (c) 2007-2016 Contributors as noted in the AUTHORS file
3

4
    This file is part of libzmq, the ZeroMQ core engine in C++.
5

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

11 12 13 14 15 16 17 18 19 20 21 22 23 24
    As a special exception, the Contributors give you permission to link
    this library with independent modules to produce an executable,
    regardless of the license terms of these independent modules, and to
    copy and distribute the resulting executable under terms of your choice,
    provided that you also meet, for each linked independent module, the
    terms and conditions of the license of that module. An independent
    module is a module which is not derived from or based on this library.
    If you modify this library, you must extend this exception to your
    version of the library.

    libzmq 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 GNU Lesser General Public
    License for more details.
25 26 27 28 29 30 31 32

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

#ifndef __ZMQ_MECHANISM_HPP_INCLUDED__
#define __ZMQ_MECHANISM_HPP_INCLUDED__

33
#include "stdint.hpp"
34 35
#include "options.hpp"
#include "blob.hpp"
36
#include "metadata.hpp"
37 38 39

namespace zmq
{
40 41
class msg_t;
class session_base_t;
42

43 44
//  Abstract class representing security mechanism.
//  Different mechanism extends this class.
45

46 47 48 49
class mechanism_t
{
  public:
    enum status_t
50
    {
51 52 53 54
        handshaking,
        ready,
        error
    };
55

56
    mechanism_t (const options_t &options_);
57

58
    virtual ~mechanism_t ();
59

60 61
    //  Prepare next handshake command that is to be sent to the peer.
    virtual int next_handshake_command (msg_t *msg_) = 0;
62

63 64
    //  Process the handshake command received from the peer.
    virtual int process_handshake_command (msg_t *msg_) = 0;
65

66
    virtual int encode (msg_t *) { return 0; }
67

68
    virtual int decode (msg_t *) { return 0; }
69

70 71
    //  Notifies mechanism about availability of ZAP message.
    virtual int zap_msg_available () { return 0; }
72

73 74
    //  Returns the status of this mechanism.
    virtual status_t status () const = 0;
75

76
    void set_peer_routing_id (const void *id_ptr_, size_t id_size_);
77

78
    void peer_routing_id (msg_t *msg_);
79

80
    void set_user_id (const void *user_id_, size_t size_);
81

82
    const blob_t &get_user_id () const;
83

84 85 86 87
    const metadata_t::dict_t &get_zmtp_properties ()
    {
        return _zmtp_properties;
    }
88

89
    const metadata_t::dict_t &get_zap_properties () { return _zap_properties; }
90

91 92 93
  protected:
    //  Only used to identify the socket for the Socket-Type
    //  property in the wire protocol.
94
    const char *socket_type_string (int socket_type_) const;
95

96 97 98 99 100 101
    static size_t add_property (unsigned char *ptr_,
                                size_t ptr_capacity_,
                                const char *name_,
                                const void *value_,
                                size_t value_len_);
    static size_t property_len (const char *name_, size_t value_len_);
102

103 104
    size_t add_basic_properties (unsigned char *ptr_,
                                 size_t ptr_capacity_) const;
105
    size_t basic_properties_len () const;
106

107
    void make_command_with_basic_properties (msg_t *msg_,
108 109
                                             const char *prefix_,
                                             size_t prefix_len_) const;
110

111 112 113 114 115
    //  Parses a metadata.
    //  Metadata consists of a list of properties consisting of
    //  name and value as size-specified strings.
    //  Returns 0 on success and -1 on error, in which case errno is set.
    int parse_metadata (const unsigned char *ptr_,
116 117
                        size_t length_,
                        bool zap_flag_ = false);
118

119 120 121 122 123 124 125 126 127
    //  This is called by parse_property method whenever it
    //  parses a new property. The function should return 0
    //  on success and -1 on error, in which case it should
    //  set errno. Signaling error prevents parser from
    //  parsing remaining data.
    //  Derived classes are supposed to override this
    //  method to handle custom processing.
    virtual int
    property (const std::string &name_, const void *value_, size_t length_);
128

129 130 131
    const options_t options;

  private:
132
    //  Properties received from ZMTP peer.
133
    metadata_t::dict_t _zmtp_properties;
134

135
    //  Properties received from ZAP server.
136
    metadata_t::dict_t _zap_properties;
137

138
    blob_t _routing_id;
139

140
    blob_t _user_id;
141

142 143
    //  Returns true iff socket associated with the mechanism
    //  is compatible with a given socket type 'type_'.
144
    bool check_socket_type (const char *type_, size_t len_) const;
145
};
146
}
147 148

#endif