Turn on mass assignment protection

[rails.git] / app / models / client_application.rb

diff --git a/app/models/client_application.rb b/app/models/client_application.rb
index 09eec40d3e5f020a4445282a485a312d98a6590c..b1f4022928fad8f6bf27a85de8a00ac1a694483c 100644 (file)
--- a/app/models/client_application.rb
+++ b/app/models/client_application.rb
@@ -1,15 +1,24 @@
 require 'oauth'
 require 'oauth'

+

 class ClientApplication < ActiveRecord::Base
   belongs_to :user
   has_many :tokens, :class_name => "OauthToken"
   has_many :access_tokens
 class ClientApplication < ActiveRecord::Base
   belongs_to :user
   has_many :tokens, :class_name => "OauthToken"
   has_many :access_tokens

+  has_many :oauth2_verifiers
+  has_many :oauth_tokens
+

   validates_presence_of :name, :url, :key, :secret
   validates_uniqueness_of :key
   validates_presence_of :name, :url, :key, :secret
   validates_uniqueness_of :key

-  before_validation_on_create :generate_keys
-  

   validates_format_of :url, :with => /\Ahttp(s?):\/\/(\w+:{0,1}\w*@)?(\S+)(:[0-9]+)?(\/|\/([\w#!:.?+=&%@!\-\/]))?/i
   validates_format_of :support_url, :with => /\Ahttp(s?):\/\/(\w+:{0,1}\w*@)?(\S+)(:[0-9]+)?(\/|\/([\w#!:.?+=&%@!\-\/]))?/i, :allow_blank=>true
   validates_format_of :url, :with => /\Ahttp(s?):\/\/(\w+:{0,1}\w*@)?(\S+)(:[0-9]+)?(\/|\/([\w#!:.?+=&%@!\-\/]))?/i
   validates_format_of :support_url, :with => /\Ahttp(s?):\/\/(\w+:{0,1}\w*@)?(\S+)(:[0-9]+)?(\/|\/([\w#!:.?+=&%@!\-\/]))?/i, :allow_blank=>true

-  validates_format_of :callback_url, :with => /\Ahttp(s?):\/\/(\w+:{0,1}\w*@)?(\S+)(:[0-9]+)?(\/|\/([\w#!:.?+=&%@!\-\/]))?/i, :allow_blank=>true
+  validates_format_of :callback_url, :with => /\A[a-z][a-z0-9.+-]*:\/\/(\w+:{0,1}\w*@)?(\S+)(:[0-9]+)?(\/|\/([\w#!:.?+=&%@!\-\/]))?/i, :allow_blank=>true
+
+  attr_accessible :name, :url, :support_url, :callback_url,
+                  :allow_read_prefs, :allow_write_prefs,
+                  :allow_write_diary, :allow_write_api,
+                  :allow_read_gpx, :allow_write_gpx
+
+  before_validation :generate_keys, :on => :create

 
   attr_accessor :token_callback_url
   
 
   attr_accessor :token_callback_url
   


@@ -25,15 +34,10 @@ class ClientApplication < ActiveRecord::Base
   def self.verify_request(request, options = {}, &block)
     begin
       signature = OAuth::Signature.build(request, options, &block)
   def self.verify_request(request, options = {}, &block)
     begin
       signature = OAuth::Signature.build(request, options, &block)

-      logger.info "Signature Base String: #{signature.signature_base_string}"
-      logger.info "Consumer: #{signature.send :consumer_key}"
-      logger.info "Token: #{signature.send :token}"

       return false unless OauthNonce.remember(signature.request.nonce, signature.request.timestamp)
       value = signature.verify
       return false unless OauthNonce.remember(signature.request.nonce, signature.request.timestamp)
       value = signature.verify

-      logger.info "Signature verification returned: #{value.to_s}"

       value
     rescue OAuth::Signature::UnknownSignatureMethod => e
       value
     rescue OAuth::Signature::UnknownSignatureMethod => e

-      logger.info "ERROR"+e.to_s

       false
     end
   end
       false
     end
   end


@@ -50,19 +54,23 @@ class ClientApplication < ActiveRecord::Base
     @oauth_client ||= OAuth::Consumer.new(key, secret)
   end
     
     @oauth_client ||= OAuth::Consumer.new(key, secret)
   end
     

-  def create_request_token
-    RequestToken.create :client_application => self, :callback_url => self.token_callback_url
+  def create_request_token(params={})
+    params = { :client_application => self, :callback_url => self.token_callback_url }
+    permissions.each do |p|
+      params[p] = true
+    end
+    RequestToken.create(params, :without_protection => true)

   end
 
   def access_token_for_user(user)
   end
 
   def access_token_for_user(user)

-    unless token = access_tokens.find(:first, :conditions => { :user_id => user.id, :invalidated_at => nil })
+    unless token = access_tokens.valid.where(:user_id => user).first

       params = { :user => user }
 
       permissions.each do |p|
         params[p] = true
       end
 
       params = { :user => user }
 
       permissions.each do |p|
         params[p] = true
       end
 

-      token = access_tokens.create(params)
+      token = access_tokens.create(params, :without_protection => true)

     end
     
     token
     end
     
     token


@@ -82,8 +90,7 @@ protected
                  :allow_write_api, :allow_read_gpx, :allow_write_gpx ]
 
   def generate_keys
                  :allow_write_api, :allow_read_gpx, :allow_write_gpx ]
 
   def generate_keys

-    oauth_client = oauth_server.generate_consumer_credentials
-    self.key = oauth_client.key
-    self.secret = oauth_client.secret
+    self.key = OAuth::Helper.generate_key(40)[0,40]
+    self.secret = OAuth::Helper.generate_key(40)[0,40]

   end
 end
   end
 end