dalli 4.3.4 → 5.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/CHANGELOG.md +23 -25
- data/README.md +3 -15
- data/lib/dalli/client.rb +19 -56
- data/lib/dalli/instrumentation.rb +1 -1
- data/lib/dalli/options.rb +1 -1
- data/lib/dalli/pipelined_getter.rb +14 -11
- data/lib/dalli/protocol/base.rb +17 -30
- data/lib/dalli/protocol/connection_manager.rb +7 -1
- data/lib/dalli/protocol/meta.rb +3 -7
- data/lib/dalli/protocol/{meta/request_formatter.rb → request_formatter.rb} +8 -21
- data/lib/dalli/protocol/response_buffer.rb +12 -27
- data/lib/dalli/protocol/{meta/response_processor.rb → response_processor.rb} +22 -10
- data/lib/dalli/protocol/server_config_parser.rb +1 -1
- data/lib/dalli/ring.rb +2 -2
- data/lib/dalli/servers_arg_normalizer.rb +1 -1
- data/lib/dalli/version.rb +2 -2
- data/lib/dalli.rb +0 -2
- metadata +7 -13
- data/lib/dalli/protocol/binary/request_formatter.rb +0 -117
- data/lib/dalli/protocol/binary/response_header.rb +0 -36
- data/lib/dalli/protocol/binary/response_processor.rb +0 -229
- data/lib/dalli/protocol/binary/sasl_authentication.rb +0 -60
- data/lib/dalli/protocol/binary.rb +0 -200
- data/lib/dalli/protocol_deprecations.rb +0 -45
- /data/lib/dalli/protocol/{meta/key_regularizer.rb → key_regularizer.rb} +0 -0
|
@@ -147,6 +147,14 @@ module Dalli
|
|
|
147
147
|
true
|
|
148
148
|
end
|
|
149
149
|
|
|
150
|
+
def tokens_from_header_buffer(buf)
|
|
151
|
+
header = header_from_buffer(buf)
|
|
152
|
+
tokens = header.split
|
|
153
|
+
header_len = header.bytesize + TERMINATOR.length
|
|
154
|
+
body_len = body_len_from_tokens(tokens)
|
|
155
|
+
[tokens, header_len, body_len]
|
|
156
|
+
end
|
|
157
|
+
|
|
150
158
|
def full_response_from_buffer(tokens, body, resp_size)
|
|
151
159
|
value = @value_marshaller.retrieve(body, bitflags_from_tokens(tokens))
|
|
152
160
|
[resp_size, tokens.first == VA, cas_from_tokens(tokens), key_from_tokens(tokens), value]
|
|
@@ -162,15 +170,11 @@ module Dalli
|
|
|
162
170
|
# The remaining three values in the array are the ResponseHeader,
|
|
163
171
|
# key, and value.
|
|
164
172
|
##
|
|
165
|
-
def getk_response_from_buffer(buf
|
|
166
|
-
#
|
|
167
|
-
|
|
168
|
-
return [0, nil, nil, nil, nil] unless term_idx
|
|
173
|
+
def getk_response_from_buffer(buf)
|
|
174
|
+
# There's no header in the buffer, so don't advance
|
|
175
|
+
return [0, nil, nil, nil, nil] unless contains_header?(buf)
|
|
169
176
|
|
|
170
|
-
|
|
171
|
-
tokens = header.split
|
|
172
|
-
header_len = header.bytesize + TERMINATOR.length
|
|
173
|
-
body_len = body_len_from_tokens(tokens)
|
|
177
|
+
tokens, header_len, body_len = tokens_from_header_buffer(buf)
|
|
174
178
|
|
|
175
179
|
# We have a complete response that has no body.
|
|
176
180
|
# This is either the response to the terminating
|
|
@@ -181,14 +185,22 @@ module Dalli
|
|
|
181
185
|
resp_size = header_len + body_len + TERMINATOR.length
|
|
182
186
|
# The header is in the buffer, but the body is not. As we don't have
|
|
183
187
|
# a complete response, don't advance the buffer
|
|
184
|
-
return [0, nil, nil, nil, nil] unless buf.bytesize >=
|
|
188
|
+
return [0, nil, nil, nil, nil] unless buf.bytesize >= resp_size
|
|
185
189
|
|
|
186
190
|
# The full response is in our buffer, so parse it and return
|
|
187
191
|
# the values
|
|
188
|
-
body = buf.
|
|
192
|
+
body = buf.slice(header_len, body_len)
|
|
189
193
|
full_response_from_buffer(tokens, body, resp_size)
|
|
190
194
|
end
|
|
191
195
|
|
|
196
|
+
def contains_header?(buf)
|
|
197
|
+
buf.include?(TERMINATOR)
|
|
198
|
+
end
|
|
199
|
+
|
|
200
|
+
def header_from_buffer(buf)
|
|
201
|
+
buf.split(TERMINATOR, 2).first
|
|
202
|
+
end
|
|
203
|
+
|
|
192
204
|
def error_on_unexpected!(expected_codes)
|
|
193
205
|
tokens = next_line_to_tokens
|
|
194
206
|
|
|
@@ -6,7 +6,7 @@ module Dalli
|
|
|
6
6
|
module Protocol
|
|
7
7
|
##
|
|
8
8
|
# Dalli::Protocol::ServerConfigParser parses a server string passed to
|
|
9
|
-
# a Dalli::Protocol::
|
|
9
|
+
# a Dalli::Protocol::Meta instance into the hostname, port, weight, and
|
|
10
10
|
# socket_type.
|
|
11
11
|
##
|
|
12
12
|
class ServerConfigParser
|
data/lib/dalli/ring.rb
CHANGED
|
@@ -23,9 +23,9 @@ module Dalli
|
|
|
23
23
|
|
|
24
24
|
attr_accessor :servers, :continuum
|
|
25
25
|
|
|
26
|
-
def initialize(servers_arg,
|
|
26
|
+
def initialize(servers_arg, options)
|
|
27
27
|
@servers = servers_arg.map do |s|
|
|
28
|
-
|
|
28
|
+
Dalli::Protocol::Meta.new(s, options)
|
|
29
29
|
end
|
|
30
30
|
@continuum = nil
|
|
31
31
|
@continuum = build_continuum(servers) if servers.size > 1
|
|
@@ -16,7 +16,7 @@ module Dalli
|
|
|
16
16
|
# weight are optional (e.g. 'localhost', 'abc.com:12345', 'example.org:22222:3')
|
|
17
17
|
# * A colon separated string of (UNIX socket, weight) where the weight is optional
|
|
18
18
|
# (e.g. '/var/run/memcached/socket', '/tmp/xyz:3') (not supported on Windows)
|
|
19
|
-
# * A URI with a 'memcached' protocol
|
|
19
|
+
# * A URI with a 'memcached' protocol (e.g. 'memcached://localhost:11211')
|
|
20
20
|
#
|
|
21
21
|
# The methods in this module do not validate the format of individual server strings, but
|
|
22
22
|
# rather normalize the argument into a compact array, wherein each array entry corresponds
|
data/lib/dalli/version.rb
CHANGED
data/lib/dalli.rb
CHANGED
|
@@ -62,7 +62,6 @@ require_relative 'dalli/version'
|
|
|
62
62
|
require_relative 'dalli/instrumentation'
|
|
63
63
|
|
|
64
64
|
require_relative 'dalli/compressor'
|
|
65
|
-
require_relative 'dalli/protocol_deprecations'
|
|
66
65
|
require_relative 'dalli/client'
|
|
67
66
|
require_relative 'dalli/key_manager'
|
|
68
67
|
require_relative 'dalli/pipelined_getter'
|
|
@@ -71,7 +70,6 @@ require_relative 'dalli/pipelined_deleter'
|
|
|
71
70
|
require_relative 'dalli/ring'
|
|
72
71
|
require_relative 'dalli/protocol'
|
|
73
72
|
require_relative 'dalli/protocol/base'
|
|
74
|
-
require_relative 'dalli/protocol/binary'
|
|
75
73
|
require_relative 'dalli/protocol/connection_manager'
|
|
76
74
|
require_relative 'dalli/protocol/meta'
|
|
77
75
|
require_relative 'dalli/protocol/response_buffer'
|
metadata
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
--- !ruby/object:Gem::Specification
|
|
2
2
|
name: dalli
|
|
3
3
|
version: !ruby/object:Gem::Version
|
|
4
|
-
version:
|
|
4
|
+
version: 5.0.0
|
|
5
5
|
platform: ruby
|
|
6
6
|
authors:
|
|
7
7
|
- Peter M. Goldstein
|
|
@@ -49,24 +49,18 @@ files:
|
|
|
49
49
|
- lib/dalli/pipelined_setter.rb
|
|
50
50
|
- lib/dalli/protocol.rb
|
|
51
51
|
- lib/dalli/protocol/base.rb
|
|
52
|
-
- lib/dalli/protocol/binary.rb
|
|
53
|
-
- lib/dalli/protocol/binary/request_formatter.rb
|
|
54
|
-
- lib/dalli/protocol/binary/response_header.rb
|
|
55
|
-
- lib/dalli/protocol/binary/response_processor.rb
|
|
56
|
-
- lib/dalli/protocol/binary/sasl_authentication.rb
|
|
57
52
|
- lib/dalli/protocol/connection_manager.rb
|
|
53
|
+
- lib/dalli/protocol/key_regularizer.rb
|
|
58
54
|
- lib/dalli/protocol/meta.rb
|
|
59
|
-
- lib/dalli/protocol/
|
|
60
|
-
- lib/dalli/protocol/meta/request_formatter.rb
|
|
61
|
-
- lib/dalli/protocol/meta/response_processor.rb
|
|
55
|
+
- lib/dalli/protocol/request_formatter.rb
|
|
62
56
|
- lib/dalli/protocol/response_buffer.rb
|
|
57
|
+
- lib/dalli/protocol/response_processor.rb
|
|
63
58
|
- lib/dalli/protocol/server_config_parser.rb
|
|
64
59
|
- lib/dalli/protocol/string_marshaller.rb
|
|
65
60
|
- lib/dalli/protocol/ttl_sanitizer.rb
|
|
66
61
|
- lib/dalli/protocol/value_compressor.rb
|
|
67
62
|
- lib/dalli/protocol/value_marshaller.rb
|
|
68
63
|
- lib/dalli/protocol/value_serializer.rb
|
|
69
|
-
- lib/dalli/protocol_deprecations.rb
|
|
70
64
|
- lib/dalli/ring.rb
|
|
71
65
|
- lib/dalli/servers_arg_normalizer.rb
|
|
72
66
|
- lib/dalli/socket.rb
|
|
@@ -77,7 +71,7 @@ licenses:
|
|
|
77
71
|
- MIT
|
|
78
72
|
metadata:
|
|
79
73
|
bug_tracker_uri: https://github.com/petergoldstein/dalli/issues
|
|
80
|
-
changelog_uri: https://github.com/petergoldstein/dalli/blob/
|
|
74
|
+
changelog_uri: https://github.com/petergoldstein/dalli/blob/main/CHANGELOG.md
|
|
81
75
|
rubygems_mfa_required: 'true'
|
|
82
76
|
rdoc_options: []
|
|
83
77
|
require_paths:
|
|
@@ -86,14 +80,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
|
|
|
86
80
|
requirements:
|
|
87
81
|
- - ">="
|
|
88
82
|
- !ruby/object:Gem::Version
|
|
89
|
-
version: '3.
|
|
83
|
+
version: '3.3'
|
|
90
84
|
required_rubygems_version: !ruby/object:Gem::Requirement
|
|
91
85
|
requirements:
|
|
92
86
|
- - ">="
|
|
93
87
|
- !ruby/object:Gem::Version
|
|
94
88
|
version: '0'
|
|
95
89
|
requirements: []
|
|
96
|
-
rubygems_version: 4.0.
|
|
90
|
+
rubygems_version: 4.0.6
|
|
97
91
|
specification_version: 4
|
|
98
92
|
summary: High performance memcached client for Ruby
|
|
99
93
|
test_files: []
|
|
@@ -1,117 +0,0 @@
|
|
|
1
|
-
# frozen_string_literal: true
|
|
2
|
-
|
|
3
|
-
module Dalli
|
|
4
|
-
module Protocol
|
|
5
|
-
class Binary
|
|
6
|
-
##
|
|
7
|
-
# Class that encapsulates logic for formatting binary protocol requests
|
|
8
|
-
# to memcached.
|
|
9
|
-
##
|
|
10
|
-
class RequestFormatter
|
|
11
|
-
REQUEST = 0x80
|
|
12
|
-
|
|
13
|
-
OPCODES = {
|
|
14
|
-
get: 0x00,
|
|
15
|
-
set: 0x01,
|
|
16
|
-
add: 0x02,
|
|
17
|
-
replace: 0x03,
|
|
18
|
-
delete: 0x04,
|
|
19
|
-
incr: 0x05,
|
|
20
|
-
decr: 0x06,
|
|
21
|
-
flush: 0x08,
|
|
22
|
-
noop: 0x0A,
|
|
23
|
-
version: 0x0B,
|
|
24
|
-
getkq: 0x0D,
|
|
25
|
-
append: 0x0E,
|
|
26
|
-
prepend: 0x0F,
|
|
27
|
-
stat: 0x10,
|
|
28
|
-
setq: 0x11,
|
|
29
|
-
addq: 0x12,
|
|
30
|
-
replaceq: 0x13,
|
|
31
|
-
deleteq: 0x14,
|
|
32
|
-
incrq: 0x15,
|
|
33
|
-
decrq: 0x16,
|
|
34
|
-
flushq: 0x18,
|
|
35
|
-
appendq: 0x19,
|
|
36
|
-
prependq: 0x1A,
|
|
37
|
-
touch: 0x1C,
|
|
38
|
-
gat: 0x1D,
|
|
39
|
-
auth_negotiation: 0x20,
|
|
40
|
-
auth_request: 0x21,
|
|
41
|
-
auth_continue: 0x22
|
|
42
|
-
}.freeze
|
|
43
|
-
|
|
44
|
-
REQ_HEADER_FORMAT = 'CCnCCnNNQ'
|
|
45
|
-
|
|
46
|
-
KEY_ONLY = 'a*'
|
|
47
|
-
TTL_AND_KEY = 'Na*'
|
|
48
|
-
KEY_AND_VALUE = 'a*a*'
|
|
49
|
-
INCR_DECR = 'NNNNNa*'
|
|
50
|
-
TTL_ONLY = 'N'
|
|
51
|
-
NO_BODY = ''
|
|
52
|
-
|
|
53
|
-
BODY_FORMATS = {
|
|
54
|
-
get: KEY_ONLY,
|
|
55
|
-
getkq: KEY_ONLY,
|
|
56
|
-
delete: KEY_ONLY,
|
|
57
|
-
deleteq: KEY_ONLY,
|
|
58
|
-
stat: KEY_ONLY,
|
|
59
|
-
|
|
60
|
-
append: KEY_AND_VALUE,
|
|
61
|
-
prepend: KEY_AND_VALUE,
|
|
62
|
-
appendq: KEY_AND_VALUE,
|
|
63
|
-
prependq: KEY_AND_VALUE,
|
|
64
|
-
auth_request: KEY_AND_VALUE,
|
|
65
|
-
auth_continue: KEY_AND_VALUE,
|
|
66
|
-
|
|
67
|
-
set: 'NNa*a*',
|
|
68
|
-
setq: 'NNa*a*',
|
|
69
|
-
add: 'NNa*a*',
|
|
70
|
-
addq: 'NNa*a*',
|
|
71
|
-
replace: 'NNa*a*',
|
|
72
|
-
replaceq: 'NNa*a*',
|
|
73
|
-
|
|
74
|
-
incr: INCR_DECR,
|
|
75
|
-
decr: INCR_DECR,
|
|
76
|
-
incrq: INCR_DECR,
|
|
77
|
-
decrq: INCR_DECR,
|
|
78
|
-
|
|
79
|
-
flush: TTL_ONLY,
|
|
80
|
-
flushq: TTL_ONLY,
|
|
81
|
-
|
|
82
|
-
noop: NO_BODY,
|
|
83
|
-
auth_negotiation: NO_BODY,
|
|
84
|
-
version: NO_BODY,
|
|
85
|
-
|
|
86
|
-
touch: TTL_AND_KEY,
|
|
87
|
-
gat: TTL_AND_KEY
|
|
88
|
-
}.freeze
|
|
89
|
-
FORMAT = BODY_FORMATS.transform_values { |v| REQ_HEADER_FORMAT + v }
|
|
90
|
-
|
|
91
|
-
# rubocop:disable Metrics/ParameterLists
|
|
92
|
-
def self.standard_request(opkey:, key: nil, value: nil, opaque: 0, cas: 0, bitflags: nil, ttl: nil)
|
|
93
|
-
extra_len = (bitflags.nil? ? 0 : 4) + (ttl.nil? ? 0 : 4)
|
|
94
|
-
key_len = key.nil? ? 0 : key.bytesize
|
|
95
|
-
value_len = value.nil? ? 0 : value.bytesize
|
|
96
|
-
header = [REQUEST, OPCODES[opkey], key_len, extra_len, 0, 0, extra_len + key_len + value_len, opaque, cas]
|
|
97
|
-
body = [bitflags, ttl, key, value].compact
|
|
98
|
-
(header + body).pack(FORMAT[opkey])
|
|
99
|
-
end
|
|
100
|
-
# rubocop:enable Metrics/ParameterLists
|
|
101
|
-
|
|
102
|
-
def self.decr_incr_request(opkey:, key: nil, count: nil, initial: nil, expiry: nil)
|
|
103
|
-
extra_len = 20
|
|
104
|
-
(h, l) = as_8byte_uint(count)
|
|
105
|
-
(dh, dl) = as_8byte_uint(initial)
|
|
106
|
-
header = [REQUEST, OPCODES[opkey], key.bytesize, extra_len, 0, 0, key.bytesize + extra_len, 0, 0]
|
|
107
|
-
body = [h, l, dh, dl, expiry, key]
|
|
108
|
-
(header + body).pack(FORMAT[opkey])
|
|
109
|
-
end
|
|
110
|
-
|
|
111
|
-
def self.as_8byte_uint(val)
|
|
112
|
-
[val >> 32, val & 0xFFFFFFFF]
|
|
113
|
-
end
|
|
114
|
-
end
|
|
115
|
-
end
|
|
116
|
-
end
|
|
117
|
-
end
|
|
@@ -1,36 +0,0 @@
|
|
|
1
|
-
# frozen_string_literal: true
|
|
2
|
-
|
|
3
|
-
module Dalli
|
|
4
|
-
module Protocol
|
|
5
|
-
class Binary
|
|
6
|
-
##
|
|
7
|
-
# Class that encapsulates data parsed from a memcached response header.
|
|
8
|
-
##
|
|
9
|
-
class ResponseHeader
|
|
10
|
-
SIZE = 24
|
|
11
|
-
FMT = '@2nCCnNNQ'
|
|
12
|
-
|
|
13
|
-
attr_reader :key_len, :extra_len, :data_type, :status, :body_len, :opaque, :cas
|
|
14
|
-
|
|
15
|
-
def initialize(buf)
|
|
16
|
-
raise ArgumentError, "Response buffer must be at least #{SIZE} bytes" unless buf.bytesize >= SIZE
|
|
17
|
-
|
|
18
|
-
@key_len, @extra_len, @data_type, @status, @body_len, @opaque, @cas = buf.unpack(FMT)
|
|
19
|
-
end
|
|
20
|
-
|
|
21
|
-
def ok?
|
|
22
|
-
status.zero?
|
|
23
|
-
end
|
|
24
|
-
|
|
25
|
-
def not_found?
|
|
26
|
-
status == 1
|
|
27
|
-
end
|
|
28
|
-
|
|
29
|
-
NOT_STORED_STATUSES = [2, 5].freeze
|
|
30
|
-
def not_stored?
|
|
31
|
-
NOT_STORED_STATUSES.include?(status)
|
|
32
|
-
end
|
|
33
|
-
end
|
|
34
|
-
end
|
|
35
|
-
end
|
|
36
|
-
end
|
|
@@ -1,229 +0,0 @@
|
|
|
1
|
-
# frozen_string_literal: true
|
|
2
|
-
|
|
3
|
-
module Dalli
|
|
4
|
-
module Protocol
|
|
5
|
-
class Binary
|
|
6
|
-
##
|
|
7
|
-
# Class that encapsulates logic for processing binary protocol responses
|
|
8
|
-
# from memcached. Includes logic for pulling data from an IO source
|
|
9
|
-
# and parsing into local values. Handles errors on unexpected values.
|
|
10
|
-
##
|
|
11
|
-
class ResponseProcessor
|
|
12
|
-
# Response codes taken from:
|
|
13
|
-
# https://github.com/memcached/memcached/wiki/BinaryProtocolRevamped#response-status
|
|
14
|
-
RESPONSE_CODES = {
|
|
15
|
-
0 => 'No error',
|
|
16
|
-
1 => 'Key not found',
|
|
17
|
-
2 => 'Key exists',
|
|
18
|
-
3 => 'Value too large',
|
|
19
|
-
4 => 'Invalid arguments',
|
|
20
|
-
5 => 'Item not stored',
|
|
21
|
-
6 => 'Incr/decr on a non-numeric value',
|
|
22
|
-
7 => 'The vbucket belongs to another server',
|
|
23
|
-
8 => 'Authentication error',
|
|
24
|
-
9 => 'Authentication continue',
|
|
25
|
-
0x20 => 'Authentication required',
|
|
26
|
-
0x81 => 'Unknown command',
|
|
27
|
-
0x82 => 'Out of memory',
|
|
28
|
-
0x83 => 'Not supported',
|
|
29
|
-
0x84 => 'Internal error',
|
|
30
|
-
0x85 => 'Busy',
|
|
31
|
-
0x86 => 'Temporary failure'
|
|
32
|
-
}.freeze
|
|
33
|
-
|
|
34
|
-
def initialize(io_source, value_marshaller)
|
|
35
|
-
@io_source = io_source
|
|
36
|
-
@value_marshaller = value_marshaller
|
|
37
|
-
end
|
|
38
|
-
|
|
39
|
-
def read(num_bytes)
|
|
40
|
-
@io_source.read(num_bytes)
|
|
41
|
-
end
|
|
42
|
-
|
|
43
|
-
def read_response
|
|
44
|
-
resp_header = ResponseHeader.new(read_header)
|
|
45
|
-
body = read(resp_header.body_len) if resp_header.body_len.positive?
|
|
46
|
-
[resp_header, body]
|
|
47
|
-
end
|
|
48
|
-
|
|
49
|
-
def unpack_response_body(resp_header, body, parse_as_stored_value)
|
|
50
|
-
extra_len = resp_header.extra_len
|
|
51
|
-
key_len = resp_header.key_len
|
|
52
|
-
bitflags = extra_len.positive? ? body.unpack1('N') : 0x0
|
|
53
|
-
key = body.byteslice(extra_len, key_len).force_encoding(Encoding::UTF_8) if key_len.positive?
|
|
54
|
-
value = body.byteslice((extra_len + key_len)..-1)
|
|
55
|
-
value = @value_marshaller.retrieve(value, bitflags) if parse_as_stored_value
|
|
56
|
-
[key, value]
|
|
57
|
-
end
|
|
58
|
-
|
|
59
|
-
def read_header
|
|
60
|
-
read(ResponseHeader::SIZE) || raise(Dalli::NetworkError, 'No response')
|
|
61
|
-
end
|
|
62
|
-
|
|
63
|
-
def raise_on_not_ok!(resp_header)
|
|
64
|
-
return if resp_header.ok?
|
|
65
|
-
|
|
66
|
-
raise Dalli::DalliError, "Response error #{resp_header.status}: #{RESPONSE_CODES[resp_header.status]}"
|
|
67
|
-
end
|
|
68
|
-
|
|
69
|
-
def get(cache_nils: false)
|
|
70
|
-
resp_header, body = read_response
|
|
71
|
-
|
|
72
|
-
return false if resp_header.not_stored? # Not stored, normal status for add operation
|
|
73
|
-
return cache_nils ? ::Dalli::NOT_FOUND : nil if resp_header.not_found?
|
|
74
|
-
|
|
75
|
-
raise_on_not_ok!(resp_header)
|
|
76
|
-
return true unless body
|
|
77
|
-
|
|
78
|
-
unpack_response_body(resp_header, body, true).last
|
|
79
|
-
end
|
|
80
|
-
|
|
81
|
-
##
|
|
82
|
-
# Response for a storage operation. Returns the cas on success. False
|
|
83
|
-
# if the value wasn't stored. And raises an error on all other error
|
|
84
|
-
# codes from memcached.
|
|
85
|
-
##
|
|
86
|
-
def storage_response
|
|
87
|
-
resp_header, = read_response
|
|
88
|
-
return nil if resp_header.not_found?
|
|
89
|
-
return false if resp_header.not_stored? # Not stored, normal status for add operation
|
|
90
|
-
|
|
91
|
-
raise_on_not_ok!(resp_header)
|
|
92
|
-
resp_header.cas
|
|
93
|
-
end
|
|
94
|
-
|
|
95
|
-
def delete
|
|
96
|
-
resp_header, = read_response
|
|
97
|
-
return false if resp_header.not_found? || resp_header.not_stored?
|
|
98
|
-
|
|
99
|
-
raise_on_not_ok!(resp_header)
|
|
100
|
-
true
|
|
101
|
-
end
|
|
102
|
-
|
|
103
|
-
def data_cas_response
|
|
104
|
-
resp_header, body = read_response
|
|
105
|
-
return [nil, resp_header.cas] if resp_header.not_found?
|
|
106
|
-
return [nil, false] if resp_header.not_stored?
|
|
107
|
-
|
|
108
|
-
raise_on_not_ok!(resp_header)
|
|
109
|
-
return [nil, resp_header.cas] unless body
|
|
110
|
-
|
|
111
|
-
[unpack_response_body(resp_header, body, true).last, resp_header.cas]
|
|
112
|
-
end
|
|
113
|
-
|
|
114
|
-
# Returns the new value for the key, if found and updated
|
|
115
|
-
def decr_incr
|
|
116
|
-
body = generic_response
|
|
117
|
-
body ? body.unpack1('Q>') : body
|
|
118
|
-
end
|
|
119
|
-
|
|
120
|
-
def stats
|
|
121
|
-
hash = {}
|
|
122
|
-
loop do
|
|
123
|
-
resp_header, body = read_response
|
|
124
|
-
# This is the response to the terminating noop / end of stat
|
|
125
|
-
return hash if resp_header.ok? && resp_header.key_len.zero?
|
|
126
|
-
|
|
127
|
-
# Ignore any responses with non-zero status codes,
|
|
128
|
-
# such as errors from set operations. That allows
|
|
129
|
-
# this code to be used at the end of a multi
|
|
130
|
-
# block to clear any error responses from inside the multi.
|
|
131
|
-
next unless resp_header.ok?
|
|
132
|
-
|
|
133
|
-
key, value = unpack_response_body(resp_header, body, true)
|
|
134
|
-
hash[key] = value
|
|
135
|
-
end
|
|
136
|
-
end
|
|
137
|
-
|
|
138
|
-
def flush
|
|
139
|
-
no_body_response
|
|
140
|
-
end
|
|
141
|
-
|
|
142
|
-
def reset
|
|
143
|
-
generic_response
|
|
144
|
-
end
|
|
145
|
-
|
|
146
|
-
def version
|
|
147
|
-
generic_response
|
|
148
|
-
end
|
|
149
|
-
|
|
150
|
-
def consume_all_responses_until_noop
|
|
151
|
-
loop do
|
|
152
|
-
resp_header, = read_response
|
|
153
|
-
# This is the response to the terminating noop / end of stat
|
|
154
|
-
return true if resp_header.ok? && resp_header.key_len.zero?
|
|
155
|
-
end
|
|
156
|
-
end
|
|
157
|
-
|
|
158
|
-
def generic_response
|
|
159
|
-
resp_header, body = read_response
|
|
160
|
-
|
|
161
|
-
return false if resp_header.not_stored? # Not stored, normal status for add operation
|
|
162
|
-
return nil if resp_header.not_found?
|
|
163
|
-
|
|
164
|
-
raise_on_not_ok!(resp_header)
|
|
165
|
-
return true unless body
|
|
166
|
-
|
|
167
|
-
unpack_response_body(resp_header, body, false).last
|
|
168
|
-
end
|
|
169
|
-
|
|
170
|
-
def no_body_response
|
|
171
|
-
resp_header, = read_response
|
|
172
|
-
return false if resp_header.not_stored? # Not stored, possible status for append/prepend/delete
|
|
173
|
-
|
|
174
|
-
raise_on_not_ok!(resp_header)
|
|
175
|
-
true
|
|
176
|
-
end
|
|
177
|
-
|
|
178
|
-
def validate_auth_format(extra_len, count)
|
|
179
|
-
return if extra_len.zero?
|
|
180
|
-
|
|
181
|
-
raise Dalli::NetworkError, "Unexpected message format: #{extra_len} #{count}"
|
|
182
|
-
end
|
|
183
|
-
|
|
184
|
-
def auth_response(buf = read_header)
|
|
185
|
-
resp_header = ResponseHeader.new(buf)
|
|
186
|
-
body_len = resp_header.body_len
|
|
187
|
-
validate_auth_format(resp_header.extra_len, body_len)
|
|
188
|
-
content = read(body_len) if body_len.positive?
|
|
189
|
-
[resp_header.status, content]
|
|
190
|
-
end
|
|
191
|
-
|
|
192
|
-
##
|
|
193
|
-
# This method returns an array of values used in a pipelined
|
|
194
|
-
# getk process. The first value is the number of bytes by
|
|
195
|
-
# which to advance the pointer in the buffer. If the
|
|
196
|
-
# complete response is found in the buffer, this will
|
|
197
|
-
# be the response size. Otherwise it is zero.
|
|
198
|
-
#
|
|
199
|
-
# The remaining three values in the array are the ResponseHeader,
|
|
200
|
-
# key, and value.
|
|
201
|
-
##
|
|
202
|
-
def getk_response_from_buffer(buf, offset = 0)
|
|
203
|
-
# There's no header in the buffer, so don't advance
|
|
204
|
-
return [0, nil, nil, nil, nil] unless buf && buf.bytesize >= offset + ResponseHeader::SIZE
|
|
205
|
-
|
|
206
|
-
resp_header = ResponseHeader.new(buf.byteslice(offset, ResponseHeader::SIZE))
|
|
207
|
-
body_len = resp_header.body_len
|
|
208
|
-
|
|
209
|
-
# We have a complete response that has no body.
|
|
210
|
-
# This is either the response to the terminating
|
|
211
|
-
# noop or, if the status is not zero, an intermediate
|
|
212
|
-
# error response that needs to be discarded.
|
|
213
|
-
return [ResponseHeader::SIZE, resp_header.ok?, resp_header.cas, nil, nil] if body_len.zero?
|
|
214
|
-
|
|
215
|
-
resp_size = ResponseHeader::SIZE + body_len
|
|
216
|
-
# The header is in the buffer, but the body is not. As we don't have
|
|
217
|
-
# a complete response, don't advance the buffer
|
|
218
|
-
return [0, nil, nil, nil, nil] unless buf.bytesize >= offset + resp_size
|
|
219
|
-
|
|
220
|
-
# The full response is in our buffer, so parse it and return
|
|
221
|
-
# the values
|
|
222
|
-
body = buf.byteslice(offset + ResponseHeader::SIZE, body_len)
|
|
223
|
-
key, value = unpack_response_body(resp_header, body, true)
|
|
224
|
-
[resp_size, resp_header.ok?, resp_header.cas, key, value]
|
|
225
|
-
end
|
|
226
|
-
end
|
|
227
|
-
end
|
|
228
|
-
end
|
|
229
|
-
end
|
|
@@ -1,60 +0,0 @@
|
|
|
1
|
-
# frozen_string_literal: true
|
|
2
|
-
|
|
3
|
-
module Dalli
|
|
4
|
-
module Protocol
|
|
5
|
-
class Binary
|
|
6
|
-
##
|
|
7
|
-
# Code to support SASL authentication
|
|
8
|
-
##
|
|
9
|
-
module SaslAuthentication
|
|
10
|
-
def perform_auth_negotiation
|
|
11
|
-
write(RequestFormatter.standard_request(opkey: :auth_negotiation))
|
|
12
|
-
|
|
13
|
-
status, content = response_processor.auth_response
|
|
14
|
-
return [status, []] if content.nil?
|
|
15
|
-
|
|
16
|
-
# Substitute spaces for the \x00 returned by
|
|
17
|
-
# memcached as a separator for easier
|
|
18
|
-
content&.tr!("\u0000", ' ')
|
|
19
|
-
mechanisms = content&.split
|
|
20
|
-
[status, mechanisms]
|
|
21
|
-
end
|
|
22
|
-
|
|
23
|
-
PLAIN_AUTH = 'PLAIN'
|
|
24
|
-
|
|
25
|
-
def supported_mechanisms!(mechanisms)
|
|
26
|
-
unless mechanisms.include?(PLAIN_AUTH)
|
|
27
|
-
raise NotImplementedError,
|
|
28
|
-
'Dalli only supports the PLAIN authentication mechanism'
|
|
29
|
-
end
|
|
30
|
-
[PLAIN_AUTH]
|
|
31
|
-
end
|
|
32
|
-
|
|
33
|
-
def authenticate_with_plain
|
|
34
|
-
write(RequestFormatter.standard_request(opkey: :auth_request,
|
|
35
|
-
key: PLAIN_AUTH,
|
|
36
|
-
value: "\x0#{username}\x0#{password}"))
|
|
37
|
-
@response_processor.auth_response
|
|
38
|
-
end
|
|
39
|
-
|
|
40
|
-
def authenticate_connection
|
|
41
|
-
Dalli.logger.info { "Dalli/SASL authenticating as #{username}" }
|
|
42
|
-
|
|
43
|
-
status, mechanisms = perform_auth_negotiation
|
|
44
|
-
return Dalli.logger.debug('Authentication not required/supported by server') if status == 0x81
|
|
45
|
-
|
|
46
|
-
supported_mechanisms!(mechanisms)
|
|
47
|
-
status, content = authenticate_with_plain
|
|
48
|
-
|
|
49
|
-
return Dalli.logger.info("Dalli/SASL: #{content}") if status.zero?
|
|
50
|
-
|
|
51
|
-
raise Dalli::DalliError, "Error authenticating: 0x#{status.to_s(16)}" unless status == 0x21
|
|
52
|
-
|
|
53
|
-
raise NotImplementedError, 'No two-step authentication mechanisms supported'
|
|
54
|
-
# (step, msg) = sasl.receive('challenge', content)
|
|
55
|
-
# raise Dalli::NetworkError, "Authentication failed" if sasl.failed? || step != 'response'
|
|
56
|
-
end
|
|
57
|
-
end
|
|
58
|
-
end
|
|
59
|
-
end
|
|
60
|
-
end
|