PostgreSQL 用美元引用的字符串常量来防止 SQL 注入

ben*_*diy 6 postgresql stored-procedures sanitization sql-injection

我可以使用 PostgreSQL 的带美元引号的字符串常量安全地防止 SQL 注入吗?

我知道处理动态查询的最好方法是使用参数化查询在应用程序层中生成它们,这不是这个问题的内容。所有的业务逻辑都在存储过程中。

我有一个存储过程,它接受参数并生成查询、运行它、格式化结果并将其作为文本块返回。这个函数传递了一个表名、列名和 WHERE 参数。传递给函数的 WHERE 参数来自用户在数据库中输入的数据。我想确保对刺进行消毒,以便构建的查询是安全的。

使用 PostgreSQL 的美元引号字符串常量,我应该能够安全地清理除“ $$ ”之外的所有字符串输入。但是,如果我对 "$" 进行字符串替换以对其进行转义,我应该能够进行安全的字符串比较。

存储过程:

function_name(tablename text, colnames text[], whereparam text)
--Build dynamic query...
Run Code Online (Sandbox Code Playgroud)

函数调用:

SELECT 
  function_name('tablename', ARRAY['col1', 'col2', 'col3'], 'AND replace(col1, ''$'', ''/$'') =  $$' || replace(alt_string_col, '$', '/$') || '$$ ')
FROM alttable
WHERE alt_id = 123;
Run Code Online (Sandbox Code Playgroud)

查询生成:

SELECT col1, col2, col3 FROM tablename WHERE 1=1 AND replace(col1, '$', '/$') =  $$un/safe'user /$/$ data;$$
Run Code Online (Sandbox Code Playgroud)

由于我在将 col1 字段与转义的用户数据进行比较之前对其进行了转义,因此即使用户输入“un/safe'user $$ data;” 在字段 alt_string_col 中,双美元符号不会中断查询并且比较通过。

这是在 PostgreSQL 存储过程中转义字符串的安全方法吗?

编辑1

感谢 Erwin Brandstetter。使用USING子句 for EXECUTEI 即将创建一个可以这样调用的函数:

SELECT function_name(
        'tablename',
        ARRAY['col1', 'col2', 'col3'], 
        ARRAY[' AND col1 =  $1 ', ' OR col2 = $5 '],
        quote_literal(alt_string_col)::text, --Text 1-4
        NULL::text,
        NULL::text,
        NULL::text,
        alt_active_col::boolean, --Bool 1-4
        NULL::boolean,
        NULL::boolean,
        NULL::boolean,
        NULL::integer, --Int 1-4
        NULL::integer,
        NULL::integer,
        NULL::integer
        )
FROM alttable 
WHERE alt_id = 123;
Run Code Online (Sandbox Code Playgroud)

它为可以传入的 WHERE 子句提供了一些灵活性。

在存储过程中,我有这样的EXECUTE语句。

  FOR results IN EXECUTE(builtquery) USING 
    textParm1, 
    textParm2, 
    textParm3, 
    textParm4, 
    boolParm1, 
    boolParm2, 
    boolParm3, 
    boolParm4, 
    intParm1, 
    intParm2, 
    intParm3, 
    intParm4
  LOOP
    -- Do some stuff
  END LOOP;
Run Code Online (Sandbox Code Playgroud)

Erw*_*ter 5

用于quote_ident()在连接标识符时防止 SQL 注入。或者format()在 Postgres 9.1 或更高版本中。

在 plpgsql 代码中使用USINGfor 子句EXECUTE传递。或者至少quote_literal()

要确保表名存在(并且在连接时自动在必要时被引用和模式限定),请使用特殊数据类型regclass

有关使用 plpgsql 执行动态 SQL 的更多信息:

从 PostgreSQL 9.0 开始,您还可以DO语句中使用匿名代码块来执行动态 SQL。