Class: HTTP::Protocol::HTTP1::Connection
- Inherits:
-
Object
- Object
- HTTP::Protocol::HTTP1::Connection
- Defined in:
- lib/http/protocol/http1/connection.rb
Constant Summary collapse
- CRLF =
"\r\n".freeze
- HTTP10 =
"HTTP/1.0".freeze
- HTTP11 =
"HTTP/1.1".freeze
- HEAD =
"HEAD".freeze
- CONNECT =
"CONNECT".freeze
Instance Attribute Summary collapse
-
#count ⇒ Object
readonly
The number of requests processed.
-
#persistent ⇒ Object
readonly
Whether the connection is persistent.
-
#stream ⇒ Object
readonly
Returns the value of attribute stream.
Instance Method Summary collapse
-
#close ⇒ Object
Close the connection and underlying stream.
- #each_line ⇒ Object
-
#hijack ⇒ IO
Effectively close the connection and return the underlying IO.
-
#initialize(stream, persistent = true) ⇒ Connection
constructor
A new instance of Connection.
- #persistent?(version, headers) ⇒ Boolean
- #read_body(headers, remainder = false) ⇒ Object
- #read_chunk ⇒ Object
- #read_chunked_body ⇒ Object
- #read_fixed_body(length) ⇒ Object
- #read_headers ⇒ Object
- #read_line ⇒ Object
- #read_remainder_body ⇒ Object
- #read_request ⇒ Object
- #read_request_body(headers) ⇒ Object
- #read_response(method) ⇒ Object
- #read_response_body(method, status, headers) ⇒ Object
- #read_tunnel_body ⇒ Object
- #write_body(body, chunked = true) ⇒ Object
- #write_body_and_close(body) ⇒ Object
- #write_body_head(body) ⇒ Object
- #write_chunk(chunk) ⇒ Object
- #write_chunked_body(body) ⇒ Object
- #write_empty_body(body) ⇒ Object
- #write_fixed_length_body(body, length) ⇒ Object
- #write_headers(headers) ⇒ Object
-
#write_persistent_header(version) ⇒ Object
Write the appropriate header for connection persistence.
- #write_request(authority, method, path, version, headers) ⇒ Object
- #write_response(version, status, headers, body = nil, head = false) ⇒ Object
Constructor Details
#initialize(stream, persistent = true) ⇒ Connection
Returns a new instance of Connection.
31 32 33 34 35 36 37 |
# File 'lib/http/protocol/http1/connection.rb', line 31 def initialize(stream, persistent = true) @stream = stream @persistent = persistent @count = 0 end |
Instance Attribute Details
#count ⇒ Object (readonly)
The number of requests processed.
45 46 47 |
# File 'lib/http/protocol/http1/connection.rb', line 45 def count @count end |
#persistent ⇒ Object (readonly)
Whether the connection is persistent.
42 43 44 |
# File 'lib/http/protocol/http1/connection.rb', line 42 def persistent @persistent end |
#stream ⇒ Object (readonly)
Returns the value of attribute stream.
39 40 41 |
# File 'lib/http/protocol/http1/connection.rb', line 39 def stream @stream end |
Instance Method Details
#close ⇒ Object
Close the connection and underlying stream.
83 84 85 |
# File 'lib/http/protocol/http1/connection.rb', line 83 def close @stream.close end |
#each_line ⇒ Object
118 119 120 121 122 |
# File 'lib/http/protocol/http1/connection.rb', line 118 def each_line while line = read_line yield line end end |
#hijack ⇒ IO
Effectively close the connection and return the underlying IO.
74 75 76 77 78 79 80 |
# File 'lib/http/protocol/http1/connection.rb', line 74 def hijack @persistent = false @stream.flush return @stream end |
#persistent?(version, headers) ⇒ Boolean
47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
# File 'lib/http/protocol/http1/connection.rb', line 47 def persistent?(version, headers) if version == HTTP10 if connection = headers[CONNECTION] return connection.include?(KEEP_ALIVE) else return false end else if connection = headers[CONNECTION] return !connection.include?(CLOSE) else return true end end end |
#read_body(headers, remainder = false) ⇒ Object
343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 |
# File 'lib/http/protocol/http1/connection.rb', line 343 def read_body(headers, remainder = false) # 3. If a Transfer-Encoding header field is present and the chunked # transfer coding (Section 4.1) is the final encoding, the message # body length is determined by reading and decoding the chunked # data until the transfer coding indicates the data is complete. if transfer_encoding = headers.delete(TRANSFER_ENCODING) # If a message is received with both a Transfer-Encoding and a # Content-Length header field, the Transfer-Encoding overrides the # Content-Length. Such a message might indicate an attempt to # perform request smuggling (Section 9.5) or response splitting # (Section 9.4) and ought to be handled as an error. A sender MUST # remove the received Content-Length field prior to forwarding such # a message downstream. if headers[CONTENT_LENGTH] raise BadRequest, "Message contains both transfer encoding and content length!" end if transfer_encoding.last == CHUNKED return read_chunked_body else # If a Transfer-Encoding header field is present in a response and # the chunked transfer coding is not the final encoding, the # message body length is determined by reading the connection until # it is closed by the server. If a Transfer-Encoding header field # is present in a request and the chunked transfer coding is not # the final encoding, the message body length cannot be determined # reliably; the server MUST respond with the 400 (Bad Request) # status code and then close the connection. return read_body_remainder end end # 5. If a valid Content-Length header field is present without # Transfer-Encoding, its decimal value defines the expected message # body length in octets. If the sender closes the connection or # the recipient times out before the indicated number of octets are # received, the recipient MUST consider the message to be # incomplete and close the connection. if content_length = headers.delete(CONTENT_LENGTH) length = Integer(content_length) if length > 0 return read_fixed_body(length) elsif length == 0 return nil else raise BadRequest, "Invalid content length: #{content_length}" end end if remainder # 7. Otherwise, this is a response message without a declared message # body length, so the message body length is determined by the # number of octets received prior to the server closing the # connection. return read_remainder_body end end |
#read_chunk ⇒ Object
177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 |
# File 'lib/http/protocol/http1/connection.rb', line 177 def read_chunk length = self.read_line.to_i(16) if length == 0 self.read_line return nil end # Read the data: chunk = @stream.read(length) # Consume the trailing CRLF: crlf = @stream.read(2) return chunk end |
#read_chunked_body ⇒ Object
288 289 290 291 292 293 294 295 296 297 |
# File 'lib/http/protocol/http1/connection.rb', line 288 def read_chunked_body buffer = String.new.b while chunk = read_chunk buffer << chunk chunk.clear end return buffer end |
#read_fixed_body(length) ⇒ Object
299 300 301 |
# File 'lib/http/protocol/http1/connection.rb', line 299 def read_fixed_body(length) @stream.read(length) end |
#read_headers ⇒ Object
163 164 165 166 167 168 169 170 171 172 173 174 175 |
# File 'lib/http/protocol/http1/connection.rb', line 163 def read_headers fields = [] self.each_line do |line| if line =~ /^([a-zA-Z\-\d]+):\s*(.+?)\s*$/ fields << [$1, $2] else break end end return Headers.new(fields) end |
#read_line ⇒ Object
124 125 126 127 128 129 130 131 132 |
# File 'lib/http/protocol/http1/connection.rb', line 124 def read_line # To support Ruby 2.3, we do the following which is pretty inefficient. Ruby 2.4+ can do the following: # @stream.gets(CRLF, chomp: true) or raise EOFError if line = @stream.gets(CRLF) return line.chomp!(CRLF) else raise EOFError end end |
#read_remainder_body ⇒ Object
307 308 309 |
# File 'lib/http/protocol/http1/connection.rb', line 307 def read_remainder_body @stream.read end |
#read_request ⇒ Object
134 135 136 137 138 139 140 141 142 143 144 145 |
# File 'lib/http/protocol/http1/connection.rb', line 134 def read_request method, path, version = read_line.split(/\s+/, 3) headers = read_headers @persistent = persistent?(version, headers) body = read_request_body(headers) @count += 1 return headers.delete(HOST), method, path, version, headers, body end |
#read_request_body(headers) ⇒ Object
337 338 339 340 341 |
# File 'lib/http/protocol/http1/connection.rb', line 337 def read_request_body(headers) # 6. If this is a request message and none of the above are true, then # the message body length is zero (no message body is present). return read_body(headers) end |
#read_response(method) ⇒ Object
147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 |
# File 'lib/http/protocol/http1/connection.rb', line 147 def read_response(method) version, status, reason = read_line.split(/\s+/, 3) status = Integer(status) headers = read_headers @persistent = persistent?(version, headers) body = read_response_body(method, status, headers) @count += 1 return version, status, reason, headers, body end |
#read_response_body(method, status, headers) ⇒ Object
314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 |
# File 'lib/http/protocol/http1/connection.rb', line 314 def read_response_body(method, status, headers) # RFC 7230 3.3.3 # 1. Any response to a HEAD request and any response with a 1xx # (Informational), 204 (No Content), or 304 (Not Modified) status # code is always terminated by the first empty line after the # header fields, regardless of the header fields present in the # message, and thus cannot contain a message body. if method == "HEAD" or (status >= 100 and status < 200) or status == 204 or status == 304 return nil end # 2. Any 2xx (Successful) response to a CONNECT request implies that # the connection will become a tunnel immediately after the empty # line that concludes the header fields. A client MUST ignore any # Content-Length or Transfer-Encoding header fields received in # such a message. if method == "CONNECT" and status == 200 return read_tunnel_body end return read_body(headers, true) end |
#read_tunnel_body ⇒ Object
303 304 305 |
# File 'lib/http/protocol/http1/connection.rb', line 303 def read_tunnel_body read_remainder_body end |
#write_body(body, chunked = true) ⇒ Object
266 267 268 269 270 271 272 273 274 275 276 |
# File 'lib/http/protocol/http1/connection.rb', line 266 def write_body(body, chunked = true) if body.nil? or body.empty? write_empty_body(body) elsif length = body.length write_fixed_length_body(body, length) elsif chunked write_chunked_body(body) else write_body_and_close(body) end end |
#write_body_and_close(body) ⇒ Object
252 253 254 255 256 257 258 259 260 261 262 263 264 |
# File 'lib/http/protocol/http1/connection.rb', line 252 def write_body_and_close(body) # We can't be persistent because we don't know the data length: @persistent = false @stream.write("\r\n") body.each do |chunk| @stream.write(chunk) @stream.flush end @stream.stream.close_write end |
#write_body_head(body) ⇒ Object
278 279 280 281 282 283 284 285 286 |
# File 'lib/http/protocol/http1/connection.rb', line 278 def write_body_head(body) if body.nil? or body.empty? @stream.write("content-length: 0\r\n\r\n") elsif length = body.length @stream.write("content-length: #{length}\r\n\r\n") else @stream.write("\r\n") end end |
#write_chunk(chunk) ⇒ Object
195 196 197 198 199 200 201 202 203 204 |
# File 'lib/http/protocol/http1/connection.rb', line 195 def write_chunk(chunk) if chunk.nil? @stream.write("0\r\n\r\n") elsif !chunk.empty? @stream.write("#{chunk.bytesize.to_s(16).upcase}\r\n") @stream.write(chunk) @stream.write(CRLF) @stream.flush end end |
#write_chunked_body(body) ⇒ Object
236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 |
# File 'lib/http/protocol/http1/connection.rb', line 236 def write_chunked_body(body) @stream.write("transfer-encoding: chunked\r\n\r\n") body.each do |chunk| next if chunk.size == 0 @stream.write("#{chunk.bytesize.to_s(16).upcase}\r\n") @stream.write(chunk) @stream.write(CRLF) @stream.flush end @stream.write("0\r\n\r\n") @stream.flush end |
#write_empty_body(body) ⇒ Object
206 207 208 209 210 211 212 213 |
# File 'lib/http/protocol/http1/connection.rb', line 206 def write_empty_body(body) @stream.write("content-length: 0\r\n\r\n") # Consume the input body. body.each{} if body @stream.flush end |
#write_fixed_length_body(body, length) ⇒ Object
215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 |
# File 'lib/http/protocol/http1/connection.rb', line 215 def write_fixed_length_body(body, length) @stream.write("content-length: #{length}\r\n\r\n") chunk_length = 0 body.each do |chunk| chunk_length += chunk.bytesize if chunk_length > length raise ArgumentError, "Trying to write #{chunk_length} bytes, but content length was #{length} bytes!" end @stream.write(chunk) end @stream.flush if chunk_length != length raise ArgumentError, "Wrote #{chunk_length} bytes, but content length was #{length} bytes!" end end |
#write_headers(headers) ⇒ Object
112 113 114 115 116 |
# File 'lib/http/protocol/http1/connection.rb', line 112 def write_headers(headers) headers.each do |name, value| @stream.write("#{name}: #{value}\r\n") end end |
#write_persistent_header(version) ⇒ Object
Write the appropriate header for connection persistence.
64 65 66 67 68 69 70 |
# File 'lib/http/protocol/http1/connection.rb', line 64 def write_persistent_header(version) if version == HTTP10 @stream.write("connection: keep-alive\r\n") if @persistent else @stream.write("connection: close\r\n") unless @persistent end end |
#write_request(authority, method, path, version, headers) ⇒ Object
87 88 89 90 91 92 93 94 95 |
# File 'lib/http/protocol/http1/connection.rb', line 87 def write_request(, method, path, version, headers) @stream.write("#{method} #{path} #{version}\r\n") @stream.write("host: #{}\r\n") write_headers(headers) write_persistent_header(version) @stream.flush end |
#write_response(version, status, headers, body = nil, head = false) ⇒ Object
97 98 99 100 101 102 103 104 105 106 107 108 109 110 |
# File 'lib/http/protocol/http1/connection.rb', line 97 def write_response(version, status, headers, body = nil, head = false) @stream.write("#{version} #{status}\r\n") write_headers(headers) write_persistent_header(version) if head write_body_head(body) else write_body(body, version == HTTP11) end @stream.flush end |