Commit 161470432e1753dfc561ebb4189624967a1e4cd7

Authored by Ábner Silva de Oliveira
1 parent b6d92d25

added random as a possible value for make_order_with_parameters

lib/noosfero/api/helpers.rb
... ... @@ -164,12 +164,18 @@ require 'grape'
164 164 def make_order_with_parameters(object, method, params)
165 165 order = "created_at DESC"
166 166 unless params[:order].blank?
167   - field_name, direction = params[:order].split(' ')
168   - assoc = object.class.reflect_on_association(method.to_sym)
169   - if !field_name.blank? and assoc
170   - if assoc.klass.attribute_names.include? field_name
171   - if direction.present? and ['ASC','DESC'].include? direction.upcase
172   - order = "#{field_name} #{direction.upcase}"
  167 + if params[:order].include? '\'' or params[:order].include? '"'
  168 + order = "created_at DESC"
  169 + elsif ['RANDOM()', 'RANDOM'].include? params[:order].upcase
  170 + order = 'RANDOM()'
  171 + else
  172 + field_name, direction = params[:order].split(' ')
  173 + assoc = object.class.reflect_on_association(method.to_sym)
  174 + if !field_name.blank? and assoc
  175 + if assoc.klass.attribute_names.include? field_name
  176 + if direction.present? and ['ASC','DESC'].include? direction.upcase
  177 + order = "#{field_name} #{direction.upcase}"
  178 + end
173 179 end
174 180 end
175 181 end
... ...
test/unit/api/helpers_test.rb
... ... @@ -182,6 +182,17 @@ class APIHelpersTest < ActiveSupport::TestCase
182 182 assert_equal "created_at DESC", make_order_with_parameters(environment, "articles", params)
183 183 end
184 184  
  185 + should 'make_order_with_parameters return RANDOM() if random is passed' do
  186 + environment = Environment.new
  187 + params = {:order => "random"} # quote used to check sql injection vunerabillity
  188 + assert_equal "RANDOM()", make_order_with_parameters(environment, "articles", params)
  189 + end
  190 +
  191 + should 'make_order_with_parameters return RANDOM() if random function is passed' do
  192 + environment = Environment.new
  193 + params = {:order => "random()"} # quote used to check sql injection vunerabillity
  194 + assert_equal "RANDOM()", make_order_with_parameters(environment, "articles", params)
  195 + end
185 196  
186 197 should 'render not_found if endpoint is unavailable' do
187 198 Noosfero::API::API.stubs(:endpoint_unavailable?).returns(true)
... ...