Rename some RequestForgeryProtection methods. The class method is now #protect_from_forgery, and the default parameter is now 'authenticity_token'. [Rick]

git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@7596 5ecf4fe2-1ee6-0310-87b1-e25e094e27de
This commit is contained in:
Rick Olson
2007-09-23 18:14:44 +00:00
parent da0725aae3
commit c619003854
8 changed files with 50 additions and 44 deletions

View File

@@ -1,5 +1,7 @@
*SVN*
* Rename some RequestForgeryProtection methods. The class method is now #protect_from_forgery, and the default parameter is now 'authenticity_token'. [Rick]
* Merge csrf_killer plugin into rails. Adds RequestForgeryProtection model that verifies session-specific _tokens for non-GET requests. [Rick]
* Secure #sanitize, #strip_tags, and #strip_links helpers against xss attacks. Closes #8877. [Rick, lifofifo, Jacques Distler]

View File

@@ -327,7 +327,7 @@ module ActionController #:nodoc:
@@resource_action_separator = "/"
cattr_accessor :resource_action_separator
# Sets the token parameter name for RequestForgery. Calling #verify_token sets it to :_token by default
# Sets the token parameter name for RequestForgery. Calling #protect_from_forgery sets it to :authenticity_token by default
@@request_forgery_protection_token = nil
cattr_accessor :request_forgery_protection_token

View File

@@ -1,43 +1,43 @@
module ActionController #:nodoc:
class InvalidToken < ActionControllerError; end
# Protect a controller's actions with the #verify_token method. Failure to validate will result in a ActionController::InvalidToken
# Protect a controller's actions with the #protect_from_forgery method. Failure to validate will result in a ActionController::InvalidToken
# exception. Customize the error message through the use of rescue_templates and rescue_action_in_public.
#
# class FooController < ApplicationController
# # uses the cookie session store
# verify_token :except => :index
# protect_from_forgery :except => :index
#
# # uses one of the other session stores that uses a session_id value.
# verify_token :secret => 'my-little-pony', :except => :index
# protect_from_forgery :secret => 'my-little-pony', :except => :index
# end
#
# Valid Options:
#
# * <tt>:only/:except</tt> - passed to the before_filter call. Set which actions are verified.
# * <tt>:secret</tt> - Custom salt used to generate the form_token. Leave this off if you are using the cookie session store.
# * <tt>:secret</tt> - Custom salt used to generate the form_authenticity_token. Leave this off if you are using the cookie session store.
# * <tt>:digest</tt> - Message digest used for hashing. Defaults to 'SHA1'
module RequestForgeryProtection
def self.included(base)
base.class_eval do
class_inheritable_accessor :verify_token_options
self.verify_token_options = {}
helper_method :form_token
class_inheritable_accessor :request_forgery_protection_options
self.request_forgery_protection_options = {}
helper_method :form_authenticity_token
end
base.extend(ClassMethods)
end
module ClassMethods
def verify_token(options = {})
self.request_forgery_protection_token ||= :_token
before_filter :verify_request_token, :only => options.delete(:only), :except => options.delete(:except)
verify_token_options.update(options)
def protect_from_forgery(options = {})
self.request_forgery_protection_token ||= :authenticity_token
before_filter :verify_authenticity_token, :only => options.delete(:only), :except => options.delete(:except)
request_forgery_protection_options.update(options)
end
end
protected
# The actual before_filter that is used. Modify this to change how you handle unverified requests.
def verify_request_token
def verify_authenticity_token
verified_request? || raise(ActionController::InvalidToken)
end
@@ -45,9 +45,12 @@ module ActionController #:nodoc:
#
# * is the format restricted? By default, only HTML and AJAX requests are checked.
# * is it a GET request? Gets should be safe and idempotent
# * Does the form_token match the given _token value from the params?
# * Does the form_authenticity_token match the given _token value from the params?
def verified_request?
request_forgery_protection_token.nil? || request.method == :get || !verifiable_request_format? || form_token == params[request_forgery_protection_token]
request_forgery_protection_token.nil? ||
request.method == :get ||
!verifiable_request_format? ||
form_authenticity_token == params[request_forgery_protection_token]
end
def verifiable_request_format?
@@ -55,19 +58,27 @@ module ActionController #:nodoc:
end
# Sets the token value for the current session. Pass a :secret option in #verify_token to add a custom salt to the hash.
def form_token
@form_token ||= verify_token_options[:secret] ? token_from_session_id : token_from_cookie_session
def form_authenticity_token
@form_authenticity_token ||= if request_forgery_protection_options[:secret]
authenticity_token_from_session_id
else
authenticity_token_from_cookie_session
end
end
# Generates a unique digest using the session_id and the CSRF secret.
def token_from_session_id
key = verify_token_options[:secret].respond_to?(:call) ? verify_token_options[:secret].call(@session) : verify_token_options[:secret]
digest = verify_token_options[:digest] || 'SHA1'
def authenticity_token_from_session_id
key = if request_forgery_protection_options[:secret].respond_to?(:call)
request_forgery_protection_options[:secret].call(@session)
else
request_forgery_protection_options[:secret]
end
digest = request_forgery_protection_options[:digest] ||= 'SHA1'
OpenSSL::HMAC.hexdigest(OpenSSL::Digest::Digest.new(digest), key.to_s, session.session_id.to_s)
end
# No secret was given, so assume this is a cookie session store.
def token_from_cookie_session
def authenticity_token_from_cookie_session
session[:csrf_id] ||= CGI::Session.generate_unique_id
session.dbman.generate_digest(session[:csrf_id])
end

View File

@@ -424,7 +424,7 @@ module ActionView
if request_forgery_protection_token.nil?
''
else
tag(:input, :type => "hidden", :name => request_forgery_protection_token.to_s, :value => form_token)
tag(:input, :type => "hidden", :name => request_forgery_protection_token.to_s, :value => form_authenticity_token)
end
end
end

View File

@@ -745,7 +745,7 @@ module ActionView
else
js_options['parameters'] = "'"
end
js_options['parameters'] << "_token=' + encodeURIComponent('#{escape_javascript form_token}')"
js_options['parameters'] << "#{request_forgery_protection_token}=' + encodeURIComponent('#{escape_javascript form_authenticity_token}')"
end
options_for_javascript(js_options)

View File

@@ -474,7 +474,7 @@ module ActionView
if request_forgery_protection_token
submit_function << "var s = document.createElement('input'); s.setAttribute('type', 'hidden'); "
submit_function << "s.setAttribute('name', '_token'); s.setAttribute('value', '#{escape_javascript form_token}'); f.appendChild(s);"
submit_function << "s.setAttribute('name', '#{request_forgery_protection_token}'); s.setAttribute('value', '#{escape_javascript form_authenticity_token}'); f.appendChild(s);"
end
submit_function << "f.submit();"
end

View File

@@ -5,7 +5,7 @@ ActionController::Routing::Routes.draw do |map|
end
class RequestForgeryProtectionController < ActionController::Base
verify_token :only => :index, :secret => 'abc'
protect_from_forgery :only => :index, :secret => 'abc'
def index
render :inline => "<%= form_tag('/') {} %>"
@@ -27,7 +27,7 @@ class RequestForgeryProtectionControllerTest < Test::Unit::TestCase
def session_id() '123' end
end
@token = OpenSSL::HMAC.hexdigest(OpenSSL::Digest::Digest.new('SHA1'), 'abc', '123')
ActionController::Base.request_forgery_protection_token = :_token
ActionController::Base.request_forgery_protection_token = :authenticity_token
end
def teardown
@@ -36,7 +36,7 @@ class RequestForgeryProtectionControllerTest < Test::Unit::TestCase
def test_should_render_form_with_token_tag
get :index
assert_select 'form>div>input[name=?][value=?]', '_token', @token
assert_select 'form>div>input[name=?][value=?]', 'authenticity_token', @token
end
# Replace this with your real tests.
@@ -75,17 +75,17 @@ class RequestForgeryProtectionControllerTest < Test::Unit::TestCase
end
def test_should_allow_post_with_token
post :index, :_token => @token
post :index, :authenticity_token => @token
assert_response :success
end
def test_should_allow_put_with_token
put :index, :_token => @token
put :index, :authenticity_token => @token
assert_response :success
end
def test_should_allow_delete_with_token
delete :index, :_token => @token
delete :index, :authenticity_token => @token
assert_response :success
end
@@ -107,7 +107,7 @@ end
# no token is given, assume the cookie store is used
class CsrfCookieMonsterController < ActionController::Base
verify_token :only => :index
protect_from_forgery :only => :index
def index
render :inline => "<%= form_tag('/') {} %>"
@@ -137,7 +137,7 @@ class CsrfCookieMonsterControllerTest < Test::Unit::TestCase
attr_reader :dbman
end
@token = Digest::SHA1.hexdigest("secure")
ActionController::Base.request_forgery_protection_token = :_token
ActionController::Base.request_forgery_protection_token = :authenticity_token
end
def teardown
@@ -146,7 +146,7 @@ class CsrfCookieMonsterControllerTest < Test::Unit::TestCase
def test_should_render_form_with_token_tag
get :index
assert_select 'form>div>input[name=?][value=?]', '_token', @token
assert_select 'form>div>input[name=?][value=?]', 'authenticity_token', @token
end
# Replace this with your real tests.
@@ -185,17 +185,17 @@ class CsrfCookieMonsterControllerTest < Test::Unit::TestCase
end
def test_should_allow_post_with_token
post :index, :_token => @token
post :index, :authenticity_token => @token
assert_response :success
end
def test_should_allow_put_with_token
put :index, :_token => @token
put :index, :authenticity_token => @token
assert_response :success
end
def test_should_allow_delete_with_token
delete :index, :_token => @token
delete :index, :authenticity_token => @token
assert_response :success
end

View File

@@ -289,7 +289,7 @@ class TextHelperTest < Test::Unit::TestCase
assert_sanitized raw, %{src="javascript:bang" <img width="5">foo</img>, <span>bar</span>}
end
ActionView::Base.sanitized_allowed_tags.each do |tag_name|
ActionView::Helpers::TextHelper.sanitized_allowed_tags.each do |tag_name|
define_method "test_should_allow_#{tag_name}_tag" do
assert_sanitized "start <#{tag_name} title=\"1\" onclick=\"foo\">foo <bad>bar</bad> baz</#{tag_name}> end", %(start <#{tag_name} title="1">foo bar baz</#{tag_name}> end)
end
@@ -551,11 +551,4 @@ class TextHelperTest < Test::Unit::TestCase
def assert_sanitized(text, expected = nil)
assert_equal((expected || text), sanitize(text))
end
# pull in configuration values from ActionView::Base
[:sanitized_protocol_separator, :sanitized_protocol_attributes, :sanitized_bad_tags, :sanitized_allowed_tags, :sanitized_allowed_attributes, :sanitized_allowed_protocols, :sanitized_allowed_css_properties, :sanitized_allowed_css_keywords, :sanitized_shorthand_css_properties, :sanitized_uri_attributes].each do |attr|
define_method attr do
ActionView::Base.send(attr)
end
end
end