2008-05-18 06:22:34 +02:00
|
|
|
require 'abstract_unit'
|
2007-12-21 08:48:59 +01:00
|
|
|
require 'digest/sha1'
|
|
|
|
|
|
|
|
ActionController::Routing::Routes.draw do |map|
|
|
|
|
map.connect ':controller/:action/:id'
|
|
|
|
end
|
|
|
|
|
|
|
|
# common controller actions
|
|
|
|
module RequestForgeryProtectionActions
|
|
|
|
def index
|
|
|
|
render :inline => "<%= form_tag('/') {} %>"
|
|
|
|
end
|
|
|
|
|
|
|
|
def show_button
|
|
|
|
render :inline => "<%= button_to('New', '/') {} %>"
|
|
|
|
end
|
|
|
|
|
2008-05-18 06:22:34 +02:00
|
|
|
def remote_form
|
|
|
|
render :inline => "<% form_remote_tag(:url => '/') {} %>"
|
|
|
|
end
|
|
|
|
|
2007-12-21 08:48:59 +01:00
|
|
|
def unsafe
|
|
|
|
render :text => 'pwn'
|
|
|
|
end
|
|
|
|
|
|
|
|
def rescue_action(e) raise e end
|
|
|
|
end
|
|
|
|
|
|
|
|
# sample controllers
|
|
|
|
class RequestForgeryProtectionController < ActionController::Base
|
|
|
|
include RequestForgeryProtectionActions
|
|
|
|
protect_from_forgery :only => :index
|
|
|
|
end
|
|
|
|
|
2009-02-04 21:26:08 +01:00
|
|
|
class FreeCookieController < RequestForgeryProtectionController
|
2007-12-21 08:48:59 +01:00
|
|
|
self.allow_forgery_protection = false
|
|
|
|
|
|
|
|
def index
|
|
|
|
render :inline => "<%= form_tag('/') {} %>"
|
|
|
|
end
|
|
|
|
|
|
|
|
def show_button
|
|
|
|
render :inline => "<%= button_to('New', '/') {} %>"
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
# common test methods
|
|
|
|
|
|
|
|
module RequestForgeryProtectionTests
|
|
|
|
def teardown
|
|
|
|
ActionController::Base.request_forgery_protection_token = nil
|
|
|
|
end
|
|
|
|
|
2008-11-24 22:53:39 +01:00
|
|
|
|
2007-12-21 08:48:59 +01:00
|
|
|
def test_should_render_form_with_token_tag
|
2008-11-24 22:53:39 +01:00
|
|
|
get :index
|
|
|
|
assert_select 'form>div>input[name=?][value=?]', 'authenticity_token', @token
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_render_button_to_with_token_tag
|
|
|
|
get :show_button
|
|
|
|
assert_select 'form>div>input[name=?][value=?]', 'authenticity_token', @token
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_render_remote_form_with_only_one_token_parameter
|
|
|
|
get :remote_form
|
|
|
|
assert_equal 1, @response.body.scan(@token).size
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_allow_get
|
|
|
|
get :index
|
|
|
|
assert_response :success
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_allow_post_without_token_on_unsafe_action
|
|
|
|
post :unsafe
|
|
|
|
assert_response :success
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_not_allow_html_post_without_token
|
|
|
|
@request.env['CONTENT_TYPE'] = Mime::URL_ENCODED_FORM.to_s
|
2009-03-16 15:55:30 +01:00
|
|
|
assert_raise(ActionController::InvalidAuthenticityToken) { post :index, :format => :html }
|
2007-12-21 08:48:59 +01:00
|
|
|
end
|
|
|
|
|
2008-11-24 22:53:39 +01:00
|
|
|
def test_should_not_allow_html_put_without_token
|
|
|
|
@request.env['CONTENT_TYPE'] = Mime::URL_ENCODED_FORM.to_s
|
2009-03-16 15:55:30 +01:00
|
|
|
assert_raise(ActionController::InvalidAuthenticityToken) { put :index, :format => :html }
|
2007-12-21 08:48:59 +01:00
|
|
|
end
|
|
|
|
|
2008-11-24 22:53:39 +01:00
|
|
|
def test_should_not_allow_html_delete_without_token
|
|
|
|
@request.env['CONTENT_TYPE'] = Mime::URL_ENCODED_FORM.to_s
|
2009-03-16 15:55:30 +01:00
|
|
|
assert_raise(ActionController::InvalidAuthenticityToken) { delete :index, :format => :html }
|
2007-12-21 08:48:59 +01:00
|
|
|
end
|
2008-06-02 08:35:38 +02:00
|
|
|
|
2008-11-24 22:53:39 +01:00
|
|
|
def test_should_allow_api_formatted_post_without_token
|
|
|
|
assert_nothing_raised do
|
2008-06-02 08:35:38 +02:00
|
|
|
post :index, :format => 'xml'
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_not_allow_api_formatted_put_without_token
|
2008-11-24 22:53:39 +01:00
|
|
|
assert_nothing_raised do
|
2008-06-02 08:35:38 +02:00
|
|
|
put :index, :format => 'xml'
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2008-11-24 22:53:39 +01:00
|
|
|
def test_should_allow_api_formatted_delete_without_token
|
|
|
|
assert_nothing_raised do
|
2008-06-02 08:35:38 +02:00
|
|
|
delete :index, :format => 'xml'
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_not_allow_api_formatted_post_sent_as_url_encoded_form_without_token
|
2009-03-16 15:55:30 +01:00
|
|
|
assert_raise(ActionController::InvalidAuthenticityToken) do
|
2008-06-02 08:35:38 +02:00
|
|
|
@request.env['CONTENT_TYPE'] = Mime::URL_ENCODED_FORM.to_s
|
|
|
|
post :index, :format => 'xml'
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_not_allow_api_formatted_put_sent_as_url_encoded_form_without_token
|
2009-03-16 15:55:30 +01:00
|
|
|
assert_raise(ActionController::InvalidAuthenticityToken) do
|
2008-06-02 08:35:38 +02:00
|
|
|
@request.env['CONTENT_TYPE'] = Mime::URL_ENCODED_FORM.to_s
|
|
|
|
put :index, :format => 'xml'
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_not_allow_api_formatted_delete_sent_as_url_encoded_form_without_token
|
2009-03-16 15:55:30 +01:00
|
|
|
assert_raise(ActionController::InvalidAuthenticityToken) do
|
2008-06-02 08:35:38 +02:00
|
|
|
@request.env['CONTENT_TYPE'] = Mime::URL_ENCODED_FORM.to_s
|
|
|
|
delete :index, :format => 'xml'
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_not_allow_api_formatted_post_sent_as_multipart_form_without_token
|
2009-03-16 15:55:30 +01:00
|
|
|
assert_raise(ActionController::InvalidAuthenticityToken) do
|
2008-06-02 08:35:38 +02:00
|
|
|
@request.env['CONTENT_TYPE'] = Mime::MULTIPART_FORM.to_s
|
|
|
|
post :index, :format => 'xml'
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_not_allow_api_formatted_put_sent_as_multipart_form_without_token
|
2009-03-16 15:55:30 +01:00
|
|
|
assert_raise(ActionController::InvalidAuthenticityToken) do
|
2008-06-02 08:35:38 +02:00
|
|
|
@request.env['CONTENT_TYPE'] = Mime::MULTIPART_FORM.to_s
|
|
|
|
put :index, :format => 'xml'
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_not_allow_api_formatted_delete_sent_as_multipart_form_without_token
|
2009-03-16 15:55:30 +01:00
|
|
|
assert_raise(ActionController::InvalidAuthenticityToken) do
|
2008-06-02 08:35:38 +02:00
|
|
|
@request.env['CONTENT_TYPE'] = Mime::MULTIPART_FORM.to_s
|
|
|
|
delete :index, :format => 'xml'
|
|
|
|
end
|
|
|
|
end
|
2009-09-05 09:01:46 +02:00
|
|
|
|
2008-11-24 22:53:39 +01:00
|
|
|
def test_should_allow_xhr_post_without_token
|
|
|
|
assert_nothing_raised { xhr :post, :index }
|
|
|
|
end
|
2007-12-21 08:48:59 +01:00
|
|
|
|
2008-11-24 22:53:39 +01:00
|
|
|
def test_should_allow_xhr_put_without_token
|
|
|
|
assert_nothing_raised { xhr :put, :index }
|
2007-12-21 08:48:59 +01:00
|
|
|
end
|
|
|
|
|
2008-11-24 22:53:39 +01:00
|
|
|
def test_should_allow_xhr_delete_without_token
|
|
|
|
assert_nothing_raised { xhr :delete, :index }
|
2007-12-21 08:48:59 +01:00
|
|
|
end
|
|
|
|
|
2009-09-05 09:01:46 +02:00
|
|
|
def test_should_allow_xhr_post_with_encoded_form_content_type_without_token
|
|
|
|
@request.env['CONTENT_TYPE'] = Mime::URL_ENCODED_FORM.to_s
|
|
|
|
assert_nothing_raised { xhr :post, :index }
|
|
|
|
end
|
|
|
|
|
2007-12-21 08:48:59 +01:00
|
|
|
def test_should_allow_post_with_token
|
|
|
|
post :index, :authenticity_token => @token
|
|
|
|
assert_response :success
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_allow_put_with_token
|
|
|
|
put :index, :authenticity_token => @token
|
|
|
|
assert_response :success
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_allow_delete_with_token
|
|
|
|
delete :index, :authenticity_token => @token
|
|
|
|
assert_response :success
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_allow_post_with_xml
|
2008-06-02 08:35:38 +02:00
|
|
|
@request.env['CONTENT_TYPE'] = Mime::XML.to_s
|
2007-12-21 08:48:59 +01:00
|
|
|
post :index, :format => 'xml'
|
|
|
|
assert_response :success
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_allow_put_with_xml
|
2008-06-02 08:35:38 +02:00
|
|
|
@request.env['CONTENT_TYPE'] = Mime::XML.to_s
|
2007-12-21 08:48:59 +01:00
|
|
|
put :index, :format => 'xml'
|
|
|
|
assert_response :success
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_allow_delete_with_xml
|
2008-06-02 08:35:38 +02:00
|
|
|
@request.env['CONTENT_TYPE'] = Mime::XML.to_s
|
2007-12-21 08:48:59 +01:00
|
|
|
delete :index, :format => 'xml'
|
|
|
|
assert_response :success
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
# OK let's get our test on
|
|
|
|
|
2009-02-04 21:26:08 +01:00
|
|
|
class RequestForgeryProtectionControllerTest < ActionController::TestCase
|
2007-12-21 08:48:59 +01:00
|
|
|
include RequestForgeryProtectionTests
|
|
|
|
def setup
|
|
|
|
@controller = RequestForgeryProtectionController.new
|
|
|
|
@request = ActionController::TestRequest.new
|
2008-11-24 22:53:39 +01:00
|
|
|
@request.format = :html
|
2007-12-21 08:48:59 +01:00
|
|
|
@response = ActionController::TestResponse.new
|
2009-02-04 21:26:08 +01:00
|
|
|
@token = "cf50faa3fe97702ca1ae"
|
2007-12-21 08:48:59 +01:00
|
|
|
|
2009-02-04 21:26:08 +01:00
|
|
|
ActiveSupport::SecureRandom.stubs(:base64).returns(@token)
|
2007-12-21 08:48:59 +01:00
|
|
|
ActionController::Base.request_forgery_protection_token = :authenticity_token
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2009-02-04 21:26:08 +01:00
|
|
|
class FreeCookieControllerTest < ActionController::TestCase
|
2007-12-21 08:48:59 +01:00
|
|
|
def setup
|
|
|
|
@controller = FreeCookieController.new
|
|
|
|
@request = ActionController::TestRequest.new
|
|
|
|
@response = ActionController::TestResponse.new
|
2009-02-04 21:26:08 +01:00
|
|
|
@token = "cf50faa3fe97702ca1ae"
|
|
|
|
|
|
|
|
ActiveSupport::SecureRandom.stubs(:base64).returns(@token)
|
2007-12-21 08:48:59 +01:00
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_not_render_form_with_token_tag
|
|
|
|
get :index
|
|
|
|
assert_select 'form>div>input[name=?][value=?]', 'authenticity_token', @token, false
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_not_render_button_to_with_token_tag
|
|
|
|
get :show_button
|
|
|
|
assert_select 'form>div>input[name=?][value=?]', 'authenticity_token', @token, false
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_should_allow_all_methods_without_token
|
|
|
|
[:post, :put, :delete].each do |method|
|
|
|
|
assert_nothing_raised { send(method, :index)}
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|