1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
| | # -*- encoding: binary -*-
# :enddoc:
# no stable API here
require 'unicorn_http'
# TODO: remove redundant names
Unicorn.const_set(:HttpRequest, Unicorn::HttpParser)
class Unicorn::HttpParser
# default parameters we merge into the request env for Rack handlers
DEFAULTS = {
"rack.errors" => $stderr,
"rack.multiprocess" => true,
"rack.multithread" => false,
"rack.run_once" => false,
"rack.version" => [1, 1],
"SCRIPT_NAME" => "",
# this is not in the Rack spec, but some apps may rely on it
"SERVER_SOFTWARE" => "Unicorn #{Unicorn::Const::UNICORN_VERSION}"
}
NULL_IO = StringIO.new("")
attr_accessor :response_start_sent
# :stopdoc:
# A frozen format for this is about 15% faster
REMOTE_ADDR = 'REMOTE_ADDR'.freeze
RACK_INPUT = 'rack.input'.freeze
@@input_class = Unicorn::TeeInput
@@check_client_connection = false
def self.input_class
@@input_class
end
def self.input_class=(klass)
@@input_class = klass
end
def self.check_client_connection
@@check_client_connection
end
def self.check_client_connection=(bool)
@@check_client_connection = bool
end
# :startdoc:
# Does the majority of the IO processing. It has been written in
# Ruby using about 8 different IO processing strategies.
#
# It is currently carefully constructed to make sure that it gets
# the best possible performance for the common case: GET requests
# that are fully complete after a single read(2)
#
# Anyone who thinks they can make it faster is more than welcome to
# take a crack at it.
#
# returns an environment hash suitable for Rack if successful
# This does minimal exception trapping and it is up to the caller
# to handle any socket errors (e.g. user aborted upload).
def read(socket)
clear
e = env
# From http://www.ietf.org/rfc/rfc3875:
# "Script authors should be aware that the REMOTE_ADDR and
# REMOTE_HOST meta-variables (see sections 4.1.8 and 4.1.9)
# may not identify the ultimate source of the request. They
# identify the client for the immediate request to the server;
# that client may be a proxy, gateway, or other intermediary
# acting on behalf of the actual source client."
e[REMOTE_ADDR] = socket.kgio_addr
# short circuit the common case with small GET requests first
socket.kgio_read!(16384, buf)
if parse.nil?
# Parser is not done, queue up more data to read and continue parsing
# an Exception thrown from the parser will throw us out of the loop
false until add_parse(socket.kgio_read!(16384))
end
# detect if the socket is valid by writing a partial response:
if @@check_client_connection && headers?
@response_start_sent = true
Unicorn::Const::HTTP_RESPONSE_START.each { |c| socket.write(c) }
end
e[RACK_INPUT] = 0 == content_length ?
NULL_IO : @@input_class.new(socket, self)
hijack_setup(e, socket)
e.merge!(DEFAULTS)
end
# Rack 1.5.0 (protocol version 1.2) adds hijack request support
if ((Rack::VERSION[0] << 8) | Rack::VERSION[1]) >= 0x0102
DEFAULTS["rack.hijack?"] = true
DEFAULTS["rack.version"] = [1, 2]
RACK_HIJACK = "rack.hijack".freeze
RACK_HIJACK_IO = "rack.hijack_io".freeze
def hijacked?
env.include?(RACK_HIJACK_IO)
end
def hijack_setup(e, socket)
e[RACK_HIJACK] = proc { e[RACK_HIJACK_IO] = socket }
end
else
# old Rack, do nothing.
def hijack_setup(e, _)
end
def hijacked?
false
end
end
end
|