80 lines
2.2 KiB
Ruby
80 lines
2.2 KiB
Ruby
module ActiveSupport
|
|
# MessageVerifier makes it easy to generate and verify messages which are signed
|
|
# to prevent tampering.
|
|
#
|
|
# This is useful for cases like remember-me tokens and auto-unsubscribe links where the
|
|
# session store isn't suitable or available.
|
|
#
|
|
# Remember Me:
|
|
# cookies[:remember_me] = @verifier.generate([@user.id, 2.weeks.from_now])
|
|
#
|
|
# In the authentication filter:
|
|
#
|
|
# id, time = @verifier.verify(cookies[:remember_me])
|
|
# if time < Time.now
|
|
# self.current_user = User.find(id)
|
|
# end
|
|
#
|
|
class MessageVerifier
|
|
class InvalidSignature < StandardError; end
|
|
|
|
def initialize(secret, digest = 'SHA1')
|
|
@secret = secret
|
|
@digest = digest
|
|
end
|
|
|
|
def verify(signed_message)
|
|
raise InvalidSignature if signed_message.blank?
|
|
|
|
data, digest = signed_message.split("--")
|
|
if data.present? && digest.present? && secure_compare(digest, generate_digest(data))
|
|
Marshal.load(ActiveSupport::Base64.decode64(data))
|
|
else
|
|
raise InvalidSignature
|
|
end
|
|
end
|
|
|
|
def generate(value)
|
|
data = ActiveSupport::Base64.encode64s(Marshal.dump(value))
|
|
"#{data}--#{generate_digest(data)}"
|
|
end
|
|
|
|
private
|
|
if "foo".respond_to?(:force_encoding)
|
|
# constant-time comparison algorithm to prevent timing attacks
|
|
def secure_compare(a, b)
|
|
a = a.dup.force_encoding(Encoding::BINARY)
|
|
b = b.dup.force_encoding(Encoding::BINARY)
|
|
|
|
if a.length == b.length
|
|
result = 0
|
|
for i in 0..(a.length - 1)
|
|
result |= a[i].ord ^ b[i].ord
|
|
end
|
|
result == 0
|
|
else
|
|
false
|
|
end
|
|
end
|
|
else
|
|
# For <= 1.8.6
|
|
def secure_compare(a, b)
|
|
if a.length == b.length
|
|
result = 0
|
|
for i in 0..(a.length - 1)
|
|
result |= a[i] ^ b[i]
|
|
end
|
|
result == 0
|
|
else
|
|
false
|
|
end
|
|
end
|
|
end
|
|
|
|
def generate_digest(data)
|
|
require 'openssl' unless defined?(OpenSSL)
|
|
OpenSSL::HMAC.hexdigest(OpenSSL::Digest::Digest.new(@digest), @secret, data)
|
|
end
|
|
end
|
|
end
|