### Python 实现 SQL 注入攻击的代码示例 以下是一个简单的示例，展示如何通过用户输入构造恶意 SQL 查询来模拟 SQL 注入攻击。此代码仅用于学习和理解 SQL 注入的工作原理，请勿在生产环境中使用或尝试。 ```python # 模拟一个存在漏洞的数据库查询函数 def vulnerable_query(username): # 构造 SQL 查询字符串，未对用户输入进行任何验证或转义 query = f"SELECT * FROM users WHERE username = '{username}'" print("执行的查询:", query) # 假设这里执行了查询 return query # 用户输入的恶意数据 malicious_input = "admin' -- " # 调用函数并传入恶意输入 vulnerable_query(malicious_input) ``` 在这个例子中，`malicious_input` 包含了一个 SQL 注入攻击的有效载荷：`admin' -- `。这个有效载荷会关闭原本的查询条件，并注释掉后续的 SQL 语句[^1]。最终生成的查询可能如下所示： ```sql SELECT * FROM users WHERE username = 'admin' -- ' ``` 由于 `--` 是 SQL 中的注释符号，因此查询的实际效果是只检查 `username = 'admin'`，而忽略了后续的条件。这可能导致未经授权访问系统中的数据。 --- ### 使用预编译语句防止 SQL 注入 为了防止上述攻击，可以使用预编译语句（Prepared Statements）。以下是使用 Python 的 `psycopg2` 库实现安全查询的示例： ```python import psycopg2 # 安全的查询函数 def secure_query(username): # 连接到数据库 connection = psycopg2.connect(database="testdb", user="postgres", password="password", host="127.0.0.1", port="5432") cursor = connection.cursor() # 使用预编译语句防止 SQL 注入 query = "SELECT * FROM users WHERE username = %s" cursor.execute(query, (username,)) # 获取结果 result = cursor.fetchall() print("查询结果:", result) # 关闭连接 cursor.close() connection.close() # 测试安全查询 secure_query("admin' -- ") ``` 在这个例子中，`%s` 是一个占位符，实际的值会被正确地转义并作为参数传递给数据库[^2]。即使用户输入包含恶意字符，也无法修改原始查询的逻辑。 --- ### 使用 `psycopg.sql` 模块编写更安全的查询 如果需要动态生成表名或其他 SQL 元素，可以使用 `psycopg2.sql` 模块来确保安全性： ```python from psycopg2 import sql import psycopg2 def count_rows_securely(table_name: str) -> int: connection = psycopg2.connect(database="testdb", user="postgres", password="password", host="127.0.0.1", port="5432") with connection.cursor() as cursor: stmt = sql.SQL("SELECT COUNT(*) FROM {table}").format( table=sql.Identifier(table_name), ) cursor.execute(stmt) result = cursor.fetchone() rowcount, = result return rowcount # 测试安全查询 print("行数:", count_rows_securely("users")) ``` 在这里，`sql.Identifier` 确保表名被正确处理，避免潜在的 SQL 注入风险[^3]。 --- ### 总结 SQL 注入是一种常见的安全威胁，可以通过使用预编译语句、参数化查询以及专门的安全模块（如 `psycopg.sql`）来有效防止。上述示例展示了如何利用 Python 构造恶意 SQL 查询，同时也提供了防止此类攻击的方法。 ---