另类附值指定参数的方法(不用attr_accessible 或 attr_protected)

最新推荐文章于 2016-09-14 11:32:42 发布

最新推荐文章于 2016-09-14 11:32:42 发布 · 144 阅读

ROR中级班专栏收录该内容

11 篇文章

订阅专栏

本文探讨了在Ruby on Rails应用中如何安全地将表单数据赋值给模型，防止非法参数篡改，介绍了几种实现方法包括使用attr_protected、attr_accessible及自定义方法split_off。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Users < ApplicationController
    def create
      @user = User.create params[:user]
    end
end

这类方法是我们常用的表单到模型的附值了。

也是经常出现漏洞的地方。具体有哪些漏洞就不说，网上有很多介绍。

通常我们为了解决这个问题都在Model中用attr_protected 或 attr_accessible 来解决。

另外还可以通过精确挑选需要的值来进行操作。

class Users < ApplicationController
    def create
        @user = User.new
        @user.login = params[:user][:login]
        @user.password = params[:user][:password]
        @user.password_confirmation = params[:user][:password_confirmation]
        @user.save
        #user is 创建了, 但is_admin 权限没有附值
    end
end

为了简化代码，我们只用到了下面五行解决

class Users < ApplicationController
    def create
      @user = User.create(
        params.split_off(:login, :password, :password_confirmation) )
    end
end

这是一DRY应用，split_off 在helper中

class Hash
    def split_off(*keys)
      h = {}
      keys.each { |key| h[key] = self[key] }
      h
    end

排除的版本

   def split_off!(*keys)
      h = {}
      keys.each { |key| h[key] = delete key }
      h
   end
end

允许一个不同的action (和一个block) ，如果他们key中没有value:

class Hash
    def split_off(*keys)
      h = {}
      if block_given?
        keys.each do |key|
          val = self[key]
          h[key] = val.nil? ? yield(key) : val
        end
      else
        keys.each { |key| h[key] = self[key] }
      end
      h
    end

    def split_off!(*keys)
      h = {}
      if block_given?
        keys.each do |key|
          val = delete(key)
          h[key] = val.nil? ? yield(key) : val
        end
      else
        keys.each { |key| h[key] = delete(key) }
      end
      h
    end
end