SQL防火墙是数据库层面的防火墙功能,可以防止恶意SQL注入。可以用来学习一些定义好的SQL规则,并将这些规则储存在数据库中作为白名单,学习完成后,可以限制用户执行这些定义规则之外的风险操作。

前提条件

PolarDB PostgreSQL版版本为PostgreSQL 11。

注意事项

sql_firewall插件需要使用共享内存,如果集群启动时没有加载sql_firewall库(默认不加载),则会创建插件失败。如果需要安装sql_firewall插件,请在shared_preload_libraries参数中添加sql_firewall

学习模式、预警模式与防火墙模式

流程图

SQL防火墙支持如下三种模式:

  • 学习模式:防火墙记录用户执行的SQL,并添加到常用SQL白名单。
  • 预警模式:防火墙对用户将执行的SQL进行判断,如果SQL不在白名单中,仍然会执行该SQL,但是会告警用户该SQL不在白名单中。
  • 防火墙模式:防火墙对用户将执行的SQL进行判断,如果SQL不在白名单中,防火墙会拒绝执行该SQL并返回错误提示。

SQL防火墙的使用步骤

  1. 打开防火墙的学习模式,让防火墙记录用户执行的SQL,并加入白名单。这个过程建议持续一段较长的时间,尽量让防火墙学习到所有可能执行的SQL。
  2. 切换防火墙为预警模式,这个过程防火墙会对用户的一些不在白名单的SQL进行告警,用户结合自己的业务判断是否为风险SQL,如果这些SQL确实是用户需要的业务语句,则记录这些SQL,之后打开学习模式进行二次学习。
  3. 经过前两步,用户常用SQL已经被记录完毕,打开防火墙模式,此时不在白名单的SQL均会被拒绝执行。

操作说明

  • 创建插件
    create extension sql_firewall;
    说明
    • sql_firewall插件需要使用共享内存,因此您在集群启动时没有加载sql_firewall库(默认不加载),则会创建插件失败。
    • 如果想正常使用插件功能,您需要在postgresql.conf文件中添加shared_preload_libraries = 'sql_firewall'语句,并重启集群。
  • 删除插件
    drop extension sql_firewall;
  • 切换模式

    在控制台找到sql_firewall.firewall参数,修改参数值并重启集群。

    sql_firewall.firewall取值如下:

    • disable:关闭SQL防火墙。
    • learning:学习模式。
    • permissive:预警模式。
    • enforcing:防火墙模式。
  • 功能函数
    • sql_firewall_reset()

      清空白名单。该函数只有在防火墙关闭模式下,polar_superuser权限的用户可执行该函数。

    • sql_firewall_stat_reset()

      清空统计信息。该函数只有在防火墙关闭模式下,polar_superuser权限的用户可执行该函数。

  • 视图函数
    • sql_firewall.sql_firewall_statements

      展示目前数据库中所有的白名单及其被调用的次数。

       select * from sql_firewall.sql_firewall_statements;
      结果如下:
           userid |  queryid   |              query              | calls
          --------+------------+---------------------------------+-------
               10 | 3294787656 | select * from k1 where uid = ?; |     4
          (1 row)
    • sql_firewall.sql_firewall_stat

      展示预警模式下的警告数量(sql_warning)和防火墙模式下的错误数量(sql_error)。

      select * from sql_firewall.sql_firewall_stat;
           sql_warning | sql_error
          -------------+-----------
                     2 |         1
          (1 row)

示例

--预警模式

    select * from sql_firewall.sql_firewall_statements;
    WARNING:  Prohibited SQL statement
     userid |  queryid   |              query              | calls
    --------+------------+---------------------------------+-------
         10 | 3294787656 | select * from k1 where uid = 1; |     1
    (1 row)

     select * from k1 where uid = 1;
     uid |    uname
    -----+-------------
       1 | Park Gyu-ri
    (1 row)

     select * from k1 where uid = 3;
     uid |   uname
    -----+-----------
       3 | Goo Ha-ra
    (1 row)

     select * from k1 where uid = 3 or 1 = 1;
    WARNING:  Prohibited SQL statement
     uid |     uname
    -----+----------------
       1 | Park Gyu-ri
       2 | Nicole Jung
       3 | Goo Ha-ra
       4 | Han Seung-yeon
       5 | Kang Ji-young
    (5 rows)

--防火墙模式

    select * from k1 where uid = 3;
     uid |   uname
    -----+-----------
       3 | Goo Ha-ra
    (1 row)

     select * from k1 where uid = 3 or 1 = 1;
    ERROR:  Prohibited SQL statement