From: Eric Wong <normalperson@yhbt.net>
To: mongrel-unicorn@rubyforge.org
Subject: [PATCH 2/1] ignore normal Rack response at request-time hijack
Date: Tue, 22 Jan 2013 23:57:28 +0000 [thread overview]
Message-ID: <20130122235728.GA8717@dcvr.yhbt.net> (raw)
In-Reply-To: <20130122114943.GA8455@dcvr.yhbt.net>
Once a connection is hijacked, we ignore it completely and leave
the connection at the mercy of the application.
---
Pushed to "hijack" of git://bogomips.org/unicorn
lib/unicorn/http_request.rb | 8 ++++++++
lib/unicorn/http_server.rb | 2 ++
t/hijack.ru | 7 ++++++-
t/t0200-rack-hijack.sh | 2 +-
4 files changed, 17 insertions(+), 2 deletions(-)
diff --git a/lib/unicorn/http_request.rb b/lib/unicorn/http_request.rb
index 3bc64ed..3795b3b 100644
--- a/lib/unicorn/http_request.rb
+++ b/lib/unicorn/http_request.rb
@@ -106,6 +106,10 @@ class Unicorn::HttpParser
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
@@ -113,5 +117,9 @@ class Unicorn::HttpParser
# old Rack, do nothing.
def hijack_setup(e, _)
end
+
+ def hijacked?
+ false
+ end
end
end
diff --git a/lib/unicorn/http_server.rb b/lib/unicorn/http_server.rb
index 2d8e4e1..cc0a705 100644
--- a/lib/unicorn/http_server.rb
+++ b/lib/unicorn/http_server.rb
@@ -550,11 +550,13 @@ class Unicorn::HttpServer
# in 3 easy steps: read request, call app, write app response
def process_client(client)
status, headers, body = @app.call(env = @request.read(client))
+ return if @request.hijacked?
if 100 == status.to_i
client.write(expect_100_response)
env.delete(Unicorn::Const::HTTP_EXPECT)
status, headers, body = @app.call(env)
+ return if @request.hijacked?
end
@request.headers? or headers = nil
http_response_write(client, status, headers, body,
diff --git a/t/hijack.ru b/t/hijack.ru
index 105e0d7..fcb0b6d 100644
--- a/t/hijack.ru
+++ b/t/hijack.ru
@@ -17,7 +17,12 @@ run lambda { |env|
io = env["rack.hijack"].call
if io.respond_to?(:read_nonblock) &&
env["rack.hijack_io"].respond_to?(:read_nonblock)
- return [ 200, {}, [ "hijack.OK\n" ] ]
+
+ # exercise both, since we Rack::Lint may use different objects
+ env["rack.hijack_io"].write("HTTP/1.0 200 OK\r\n\r\n")
+ io.write("request.hijacked")
+ io.close
+ return [ 500, {}, DieIfUsed.new ]
end
end
[ 500, {}, [ "hijack BAD\n" ] ]
diff --git a/t/t0200-rack-hijack.sh b/t/t0200-rack-hijack.sh
index 23a9ee4..f772071 100755
--- a/t/t0200-rack-hijack.sh
+++ b/t/t0200-rack-hijack.sh
@@ -9,7 +9,7 @@ t_begin "setup and start" && {
}
t_begin "check request hijack" && {
- test "xhijack.OK" = x"$(curl -sSfv http://$listen/hijack_req)"
+ test "xrequest.hijacked" = x"$(curl -sSfv http://$listen/hijack_req)"
}
t_begin "check response hijack" && {
--
Eric Wong
_______________________________________________
Unicorn mailing list - mongrel-unicorn@rubyforge.org
http://rubyforge.org/mailman/listinfo/mongrel-unicorn
Do not quote signatures (like this one) or top post when replying
prev parent reply other threads:[~2013-01-22 23:57 UTC|newest]
Thread overview: 2+ messages / expand[flat|nested] mbox.gz Atom feed top
2013-01-22 11:49 [PATCH] support for Rack hijack in request and response Eric Wong
2013-01-22 23:57 ` Eric Wong [this message]
Reply instructions:
You may reply publicly to this message via plain-text email
using any one of the following methods:
* Save the following mbox file, import it into your mail client,
and reply-to-all from there: mbox
Avoid top-posting and favor interleaved quoting:
https://en.wikipedia.org/wiki/Posting_style#Interleaved_style
List information: https://yhbt.net/unicorn/
* Reply using the --to, --cc, and --in-reply-to
switches of git-send-email(1):
git send-email \
--in-reply-to=20130122235728.GA8717@dcvr.yhbt.net \
--to=normalperson@yhbt.net \
--cc=mongrel-unicorn@rubyforge.org \
/path/to/YOUR_REPLY
https://kernel.org/pub/software/scm/git/docs/git-send-email.html
* If your mail client supports setting the In-Reply-To header
via mailto: links, try the mailto: link
Be sure your reply has a Subject: header at the top and a blank line
before the message body.
Code repositories for project(s) associated with this public inbox
https://yhbt.net/unicorn.git/
This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox;
as well as URLs for read-only IMAP folder(s) and NNTP newsgroup(s).