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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
| | # -*- encoding: binary -*-
# Copyright (C) 2013, Eric Wong <normalperson@yhbt.net> and all contributors
# License: GPLv3 or later (https://www.gnu.org/licenses/gpl-3.0.txt)
require_relative 'stream_file'
# Writes a Rack response to your client using the HTTP/1.1 specification.
# You use it by simply doing:
#
# status, headers, body = rack_app.call(env)
# http_response_write(status, headers, body)
#
# Most header correctness (including Content-Length and Content-Type)
# is the job of Rack, with the exception of the "Date" header.
module Yahns::HttpResponse # :nodoc:
include Unicorn::HttpResponse
# avoid GC overhead for frequently used-strings:
CONN_KA = "Connection: keep-alive\r\n\r\n"
CONN_CLOSE = "Connection: close\r\n\r\n"
Z = ""
RESPONSE_START = "HTTP/1.1 "
def response_start
@response_start_sent ? Z : RESPONSE_START
end
def response_wait_write(rv)
# call the kgio_wait_readable or kgio_wait_writable method
ok = __send__("kgio_#{rv}") and return ok
k = self.class
k.logger.info("fd=#{fileno} ip=#@kgio_addr timeout on :#{rv} after "\
"#{k.client_timeout}s")
nil
end
def err_response(code)
"#{response_start}#{CODES[code]}\r\n\r\n"
end
def response_header_blocked(ret, header, body, alive, offset, count)
if body.respond_to?(:to_path)
alive = Yahns::StreamFile.new(body, alive, offset, count)
body = nil
end
wbuf = Yahns::Wbuf.new(body, alive)
rv = wbuf.wbuf_write(self, header)
body.each { |chunk| rv = wbuf.wbuf_write(self, chunk) } if body
wbuf_maybe(wbuf, rv, alive)
end
def wbuf_maybe(wbuf, rv, alive)
case rv # trysendfile return value
when nil
case alive
when :ignore
@state = :ignore
when true, false
http_response_done(alive)
end
else
@state = wbuf
rv
end
end
def http_response_done(alive)
@input = @input.close if @input
if alive
@response_start_sent = false
# @hs.buf will have data if the client pipelined
if @hs.buf.empty?
@state = :headers
:wait_readable
else
@state = :pipelined
# may need to wait for readability if SSL,
# only need writability if plain TCP
:wait_readwrite
end
else
# shutdown is needed in case the app forked, we rescue here since
# StreamInput may issue shutdown as well
shutdown rescue nil
nil # trigger close
end
end
# writes the rack_response to socket as an HTTP response
# returns :wait_readable, :wait_writable, :forget, or nil
def http_response_write(status, headers, body)
status = CODES[status.to_i] || status
offset = 0
count = hijack = nil
k = self.class
alive = @hs.next? && k.persistent_connections
if @hs.headers?
buf = "#{response_start}#{status}\r\nDate: #{httpdate}\r\n"
headers.each do |key, value|
case key
when %r{\ADate\z}
next
when %r{\AContent-Range\z}i
if %r{\Abytes (\d+)-(\d+)/\d+\z} =~ value
offset = $1.to_i
count = $2.to_i - offset + 1
end
when %r{\AConnection\z}i
# allow Rack apps to tell us they want to drop the client
alive = !!(value =~ /\bclose\b/i)
when "rack.hijack"
hijack = value
body = nil # ensure we do not close body
else
if value =~ /\n/
# avoiding blank, key-only cookies with /\n+/
buf << value.split(/\n+/).map! { |v| "#{key}: #{v}\r\n" }.join
else
buf << "#{key}: #{value}\r\n"
end
end
end
buf << (alive ? CONN_KA : CONN_CLOSE)
case rv = kgio_trywrite(buf)
when nil # all done, likely
break
when String
buf = rv # hope the skb grows
when :wait_writable, :wait_readable
if k.output_buffering
alive = hijack ? hijack : alive
rv = response_header_blocked(rv, buf, body, alive, offset, count)
body = nil # ensure we do not close body in ensure
return rv
else
response_wait_write(rv) or return
end
end while true
end
if hijack
hijack.call(self)
return :ignore
end
if body.respond_to?(:to_path)
@state = body = Yahns::StreamFile.new(body, alive, offset, count)
return step_write
end
wbuf = rv = nil
body.each do |chunk|
if wbuf
rv = wbuf.wbuf_write(self, chunk)
else
case rv = kgio_trywrite(chunk)
when nil # all done, likely and good!
break
when String
chunk = rv # hope the skb grows when we loop into the trywrite
when :wait_writable, :wait_readable
if k.output_buffering
wbuf = Yahns::Wbuf.new(body, alive)
rv = wbuf.wbuf_write(self, chunk)
break
else
response_wait_write(rv) or return
end
end while true
end
end
# if we buffered the write body, we must return :wait_writable
# (or :wait_readable for SSL) and hit Yahns::HttpClient#step_write
if wbuf
body = nil # ensure we do not close the body in ensure
wbuf_maybe(wbuf, rv, alive)
else
http_response_done(alive)
end
ensure
body.respond_to?(:close) and body.close
end
end
|