【腾讯云代码分析】经典案例分享：SQL注入漏洞

擒兽一号

发布于 2024-09-27 14:29:12

540

发布于 2024-09-27 14:29:12

官网地址：（点击最下方【阅读原文】可直达）https://tca.tencent.com/

官网介绍：https://cloud.tencent.com/product/tcap 官方开源：https://github.com/Tencent/CodeAnalysis 国内镜像：https://git.code.tencent.com/Tencent_Open_Source/CodeAnalysis

规则介绍

▼

SQL注入漏洞是一种常见的Web应用程序安全漏洞，攻击者可以通过构造特殊的输入，将恶意的SQL代码注入到Web应用程序中，从而导致数据泄露、篡改或删除等安全问题。

下列案例对应代码分析TCA-Armory-L工具中的规则：sqli——检测代码中可能存在的SQL注入漏洞，并提供警告或建议修复的信息。

经典案例

▼

代码案例

protected AttackResult injectableQuery(String accountName) {
    String query = "";
    try (Connection connection = dataSource.getConnection()) {
        query = "SELECT * FROM user_data WHERE first_name = 'John' and last_name = '" + accountName + "'";
        try (Statement statement = connection.createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_UPDATABLE)) {
            ResultSet results = statement.executeQuery(query);
            // Do something...
        }
    }
}

问题解析

这段代码存在SQL注入漏洞。漏洞的原因是在构造SQL查询语句时，直接将用户输入的accountName拼接到SQL语句中，没有进行任何验证或过滤。攻击者可以通过构造特殊的accountName值，将恶意的SQL代码注入到查询语句中，从而导致数据泄露、篡改或删除等安全问题。

例如，如果攻击者将accountName设置为Doe' OR '1' = '1，那么构造的SQL查询语句将变为：

SELECT * FROM user_data WHERE first_name = 'John' and last_name = 'Doe' OR '1' = '1'

这个查询语句将返回user_data表中所有的记录，因为'1' = '1'总是为真。这显然是不符合预期的行为，可能导致数据泄露。

AI修复建议

验证用户输入：对用户输入进行严格的验证，确保输入符合预期的格式和范围。
过滤特殊字符：对用户输入进行过滤，去除可能导致SQL注入的特殊字符，如单引号、双引号、分号等。
使用预编译SQL语句：使用预编译SQL语句，将用户输入作为参数传递给SQL语句，而不是直接将用户输入拼接到SQL语句中。这样可以避免用户输入被解释为SQL代码，从而防止SQL注入漏洞。

protected AttackResult injectableQuery(String accountName) {
    String query = "SELECT * FROM accounts WHERE account_name = ?";
    try (Connection connection = dataSource.getConnection()) {
        PreparedStatement preparedStatement = connection.prepareStatement(query);
        preparedStatement.setString(1, accountName);
        ResultSet results = preparedStatement.executeQuery();

        if ((results != null) && (results.first())) {
            ResultSetMetaData resultsMetaData = results.getMetaData();
            StringBuilder output = new StringBuilder();
            // ... 其他代码
        }
    } catch (SQLException e) {
        // 异常处理
    }
    // 返回结果或其他操作
}